2016-01-22 01:28:15 +01:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
|
|
|
"fmt"
|
|
|
|
"net/http"
|
|
|
|
|
2016-01-25 07:26:53 +01:00
|
|
|
jwtgo "github.com/dgrijalva/jwt-go"
|
2016-01-22 01:28:15 +01:00
|
|
|
)
|
|
|
|
|
|
|
|
type authHandler struct {
|
|
|
|
handler http.Handler
|
|
|
|
}
|
|
|
|
|
|
|
|
// AuthHandler -
|
|
|
|
// Verify if authorization header has signature version '2', reject it cleanly.
|
|
|
|
func AuthHandler(h http.Handler) http.Handler {
|
|
|
|
return authHandler{h}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Ignore request if authorization header is not valid.
|
|
|
|
func (h authHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
2016-01-24 04:44:32 +01:00
|
|
|
// Let the top level caller handle if the requests should be
|
|
|
|
// allowed.
|
|
|
|
if r.Header.Get("Authorization") == "" {
|
2016-01-22 01:28:15 +01:00
|
|
|
h.handler.ServeHTTP(w, r)
|
|
|
|
return
|
|
|
|
}
|
2016-01-25 07:26:53 +01:00
|
|
|
jwt := InitJWT()
|
|
|
|
token, err := jwtgo.ParseFromRequest(r, func(token *jwtgo.Token) (interface{}, error) {
|
|
|
|
if _, ok := token.Method.(*jwtgo.SigningMethodRSA); !ok {
|
2016-01-22 01:28:15 +01:00
|
|
|
return nil, fmt.Errorf("Unexpected signing method: %v", token.Header["alg"])
|
|
|
|
}
|
2016-01-25 07:26:53 +01:00
|
|
|
return jwt.PublicKey, nil
|
2016-01-22 01:28:15 +01:00
|
|
|
})
|
|
|
|
if err != nil || !token.Valid {
|
|
|
|
w.WriteHeader(http.StatusUnauthorized)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
h.handler.ServeHTTP(w, r)
|
|
|
|
}
|