/
jwt.go
103 lines (86 loc) · 2.8 KB
/
jwt.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
// Package jwt is able to generate and validate json web tokens.
// Follows https://tools.ietf.org/html/draft-ietf-oauth-json-web-token-32
package jwt
import (
"crypto/rsa"
"crypto/sha256"
"fmt"
"strings"
"github.com/dgrijalva/jwt-go"
"github.com/pkg/errors"
)
// RS256JWTStrategy is responsible for generating and validating JWT challenges
type RS256JWTStrategy struct {
PrivateKey *rsa.PrivateKey
}
// Generate generates a new authorize code or returns an error. set secret
func (j *RS256JWTStrategy) Generate(claims jwt.Claims, header Mapper) (string, string, error) {
if header == nil || claims == nil {
return "", "", errors.New("Either claims or header is nil.")
}
token := jwt.NewWithClaims(jwt.SigningMethodRS256, claims)
token.Header = assign(token.Header, header.ToMap())
var sig, sstr string
var err error
if sstr, err = token.SigningString(); err != nil {
return "", "", errors.Wrap(err, "")
}
if sig, err = token.Method.Sign(sstr, j.PrivateKey); err != nil {
return "", "", errors.Wrap(err, "")
}
return fmt.Sprintf("%s.%s", sstr, sig), sig, nil
}
// Validate validates a token and returns its signature or an error if the token is not valid.
func (j *RS256JWTStrategy) Validate(token string) (string, error) {
if _, err := j.Decode(token); err != nil {
return "", errors.Wrap(err, "")
}
return j.GetSignature(token)
}
// Decode will decode a JWT token
func (j *RS256JWTStrategy) Decode(token string) (*jwt.Token, error) {
// Parse the token.
parsedToken, err := jwt.Parse(token, func(t *jwt.Token) (interface{}, error) {
if _, ok := t.Method.(*jwt.SigningMethodRSA); !ok {
return nil, errors.Errorf("Unexpected signing method: %v", t.Header["alg"])
}
return &j.PrivateKey.PublicKey, nil
})
if err != nil {
return nil, errors.Errorf("Couldn't parse token: %v", err)
} else if !parsedToken.Valid {
return nil, errors.Errorf("Token is invalid")
}
return parsedToken, err
}
// GetSignature will return the signature of a token
func (j *RS256JWTStrategy) GetSignature(token string) (string, error) {
split := strings.Split(token, ".")
if len(split) != 3 {
return "", errors.New("Header, body and signature must all be set")
}
return split[2], nil
}
// Hash will return a given hash based on the byte input or an error upon fail
func (j *RS256JWTStrategy) Hash(in []byte) ([]byte, error) {
// SigningMethodRS256
hash := sha256.New()
_, err := hash.Write(in)
if err != nil {
return []byte{}, errors.Wrap(err, "")
}
return hash.Sum([]byte{}), nil
}
// GetSigningMethodLength will return the length of the signing method
func (j *RS256JWTStrategy) GetSigningMethodLength() int {
return jwt.SigningMethodRS256.Hash.Size()
}
func assign(a, b map[string]interface{}) map[string]interface{} {
for k, w := range b {
if _, ok := a[k]; ok {
continue
}
a[k] = w
}
return a
}