greatape/greataped/utility/httpsig/handler.go

38 wiersze
865 B
Go

package httpsig
import (
"fmt"
"net/http"
"strings"
)
func RequireSignature(h http.Handler, v *Verifier, realm string) (
out http.Handler) {
var challenge_params []string
if realm != "" {
challenge_params = append(challenge_params,
fmt.Sprintf("realm=%q", realm))
}
if headers := v.RequiredHeaders(); len(headers) > 0 {
challenge_params = append(challenge_params,
fmt.Sprintf("headers=%q", strings.Join(headers, " ")))
}
challenge := "Signature"
if len(challenge_params) > 0 {
challenge += fmt.Sprintf(" %s", strings.Join(challenge_params, ", "))
}
return http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
err := v.Verify(req)
if err != nil {
w.Header()["WWW-Authenticate"] = []string{challenge}
w.WriteHeader(http.StatusUnauthorized)
fmt.Fprintln(w, err.Error())
return
}
h.ServeHTTP(w, req)
})
}