fix: authreq header

This commit is contained in:
Jon Lundy 2023-01-22 11:11:01 -07:00
parent c7f56789de
commit 2b9f0ffa12
Signed by untrusted user who does not match committer: xuu
GPG Key ID: C63E6D61F3035024

View File

@ -2,6 +2,7 @@ package authreq
import ( import (
"bytes" "bytes"
"context"
"crypto/ed25519" "crypto/ed25519"
"encoding/base64" "encoding/base64"
"fmt" "fmt"
@ -14,9 +15,10 @@ import (
"github.com/golang-jwt/jwt/v4" "github.com/golang-jwt/jwt/v4"
) )
var SignatureLifetime = 90 * time.Minute var SignatureLifetime = 30 * time.Minute
var AuthHeader = "Authorization"
func Sign(req http.Request, key ed25519.PrivateKey) (http.Request, error) { func Sign(req *http.Request, key ed25519.PrivateKey) (*http.Request, error) {
pub := enc([]byte(key.Public().(ed25519.PublicKey))) pub := enc([]byte(key.Public().(ed25519.PublicKey)))
h := fnv.New128a() h := fnv.New128a()
@ -44,14 +46,14 @@ func Sign(req http.Request, key ed25519.PrivateKey) (http.Request, error) {
return req, err return req, err
} }
req.Header.Set("Authorization", sig) req.Header.Set(AuthHeader, sig)
return req, nil return req, nil
} }
func Authorization(hdlr http.Handler) http.Handler { func Authorization(hdlr http.Handler) http.Handler {
return http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) { return http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
auth := req.Header.Get("Authorizaton") auth := req.Header.Get(AuthHeader)
if auth == "" { if auth == "" {
rw.WriteHeader(http.StatusUnauthorized) rw.WriteHeader(http.StatusUnauthorized)
return return
@ -82,7 +84,7 @@ func Authorization(hdlr http.Handler) http.Handler {
pub, err := dec(c.Issuer) pub, err := dec(c.Issuer)
return ed25519.PublicKey(pub), err return ed25519.PublicKey(pub), err
}, },
jwt.WithValidMethods([]string{"EdDSA"}), jwt.WithValidMethods([]string{jwt.SigningMethodEdDSA.Alg()}),
jwt.WithJSONNumber(), jwt.WithJSONNumber(),
) )
if err != nil { if err != nil {
@ -97,6 +99,8 @@ func Authorization(hdlr http.Handler) http.Handler {
return return
} }
req = req.WithContext(context.WithValue(req.Context(), contextKey, c))
if c.Subject != subject { if c.Subject != subject {
rw.WriteHeader(http.StatusForbidden) rw.WriteHeader(http.StatusForbidden)
return return
@ -113,3 +117,15 @@ func dec(s string) ([]byte, error) {
s = strings.TrimSpace(s) s = strings.TrimSpace(s)
return base64.RawURLEncoding.DecodeString(s) return base64.RawURLEncoding.DecodeString(s)
} }
var contextKey = struct{ name string }{"jwtClaim"}
func FromContext(ctx context.Context) *jwt.RegisteredClaims {
if v := ctx.Value(contextKey); v != nil {
if c, ok := v.(*jwt.RegisteredClaims); ok {
return c
}
}
return nil
}