use strings.CutPrefix

icholy · Apr 24, 2024 · f19583a · f19583a
1 parent 1fc1d2b
commit f19583a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/challenge.go b/challenge.go
@@ -34,10 +34,11 @@ func (c *Challenge) SupportsQOP(qop string) bool {
 
 // ParseChallenge parses the WWW-Authenticate header challenge
 func ParseChallenge(s string) (*Challenge, error) {
-	if !IsDigest(s) {
+	s, ok := strings.CutPrefix(s, Prefix)
+	if !ok {
 		return nil, errors.New("digest: invalid challenge prefix")
 	}
-	pp, err := param.Parse(s[len(Prefix):])
+	pp, err := param.Parse(s)
 	if err != nil {
 		return nil, err
 	}

diff --git a/credentials.go b/credentials.go
@@ -26,10 +26,11 @@ type Credentials struct {
 
 // ParseCredentials parses the Authorization header value into credentials
 func ParseCredentials(s string) (*Credentials, error) {
-	if !IsDigest(s) {
+	s, ok := strings.CutPrefix(s, Prefix)
+	if !ok {
 		return nil, errors.New("digest: invalid credentials prefix")
 	}
-	pp, err := param.Parse(s[len(Prefix):])
+	pp, err := param.Parse(s)
 	if err != nil {
 		return nil, err
 	}