Skip to content

Commit 13792df

Browse files
committed
add rsa_pss and hs sign method to identify
1 parent 250ebc2 commit 13792df

File tree

1 file changed

+11
-3
lines changed

1 file changed

+11
-3
lines changed

generates/jwt_access.go

Lines changed: 11 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -56,11 +56,13 @@ func (a *JWTAccessGenerate) Token(data *oauth2.GenerateBasic, isGenRefresh bool)
5656
if err != nil {
5757
return "", "", err
5858
}
59-
} else if a.isRs() {
59+
} else if a.isRsOrPS() {
6060
key, err = jwt.ParseRSAPrivateKeyFromPEM(a.SignedKey)
6161
if err != nil {
6262
return "", "", err
6363
}
64+
} else if a.isHs() {
65+
key = a.SignedKey
6466
} else {
6567
return "", "", errs.New("unsupported sign method")
6668
}
@@ -81,6 +83,12 @@ func (a *JWTAccessGenerate) isEs() bool {
8183
return strings.HasPrefix(a.SignedMethod.Alg(), "ES")
8284
}
8385

84-
func (a *JWTAccessGenerate) isRs() bool {
85-
return strings.HasPrefix(a.SignedMethod.Alg(), "RS")
86+
func (a *JWTAccessGenerate) isRsOrPS() bool {
87+
isRs := strings.HasPrefix(a.SignedMethod.Alg(), "RS")
88+
isPs := strings.HasPrefix(a.SignedMethod.Alg(), "PS")
89+
return isRs || isPs
90+
}
91+
92+
func (a *JWTAccessGenerate) isHs() bool {
93+
return strings.HasPrefix(a.SignedMethod.Alg(), "HS")
8694
}

0 commit comments

Comments
 (0)