2020-11-13 20:51:07 +08:00
|
|
|
package middleware
|
|
|
|
|
|
|
|
import (
|
2021-01-05 21:05:40 +08:00
|
|
|
"crypto/subtle"
|
2020-11-13 20:51:07 +08:00
|
|
|
"fmt"
|
|
|
|
"net/http"
|
|
|
|
)
|
|
|
|
|
|
|
|
// BasicAuth implements a simple middleware handler for adding basic http auth to a route.
|
|
|
|
func BasicAuth(realm string, creds map[string]string) func(next http.Handler) http.Handler {
|
|
|
|
return func(next http.Handler) http.Handler {
|
|
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
|
|
user, pass, ok := r.BasicAuth()
|
|
|
|
if !ok {
|
|
|
|
basicAuthFailed(w, realm)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
credPass, credUserOk := creds[user]
|
2021-01-05 21:05:40 +08:00
|
|
|
if !credUserOk || subtle.ConstantTimeCompare([]byte(pass), []byte(credPass)) != 1 {
|
2020-11-13 20:51:07 +08:00
|
|
|
basicAuthFailed(w, realm)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
next.ServeHTTP(w, r)
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func basicAuthFailed(w http.ResponseWriter, realm string) {
|
|
|
|
w.Header().Add("WWW-Authenticate", fmt.Sprintf(`Basic realm="%s"`, realm))
|
|
|
|
w.WriteHeader(http.StatusUnauthorized)
|
|
|
|
}
|