forked from imroc/req
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
* Add SetDigestAuth for Request. * Add SetCommonDigestAuth for Client. * Add OnAfterResponse for Request
- Loading branch information
Showing
6 changed files
with
363 additions
and
39 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,277 @@ | ||
| package req | ||
|
|
||
| import ( | ||
| "crypto/md5" | ||
| "crypto/rand" | ||
| "crypto/sha256" | ||
| "crypto/sha512" | ||
| "errors" | ||
| "fmt" | ||
| "github.com/imroc/req/v3/internal/header" | ||
| "hash" | ||
| "io" | ||
| "net/http" | ||
| "strings" | ||
| ) | ||
|
|
||
| var ( | ||
| errDigestBadChallenge = errors.New("digest: challenge is bad") | ||
| errDigestCharset = errors.New("digest: unsupported charset") | ||
| errDigestAlgNotSupported = errors.New("digest: algorithm is not supported") | ||
| errDigestQopNotSupported = errors.New("digest: no supported qop in list") | ||
| errDigestNoQop = errors.New("digest: qop must be specified") | ||
| ) | ||
|
|
||
| var hashFuncs = map[string]func() hash.Hash{ | ||
| "": md5.New, | ||
| "MD5": md5.New, | ||
| "MD5-sess": md5.New, | ||
| "SHA-256": sha256.New, | ||
| "SHA-256-sess": sha256.New, | ||
| "SHA-512-256": sha512.New, | ||
| "SHA-512-256-sess": sha512.New, | ||
| } | ||
|
|
||
| // create response middleware for http digest authentication. | ||
| func handleDigestAuthFunc(username, password string) ResponseMiddleware { | ||
| return func(client *Client, resp *Response) error { | ||
| if resp.Err != nil || resp.StatusCode != http.StatusUnauthorized { | ||
| return nil | ||
| } | ||
| auth, err := createDigestAuth(resp.Response, username, password) | ||
| if err != nil { | ||
| return err | ||
| } | ||
| r := resp.Request | ||
| req := *r.RawRequest | ||
| if req.Body != nil { | ||
| err = parseRequestBody(client, r) // re-setup body | ||
| if err != nil { | ||
| return err | ||
| } | ||
| if r.GetBody != nil { | ||
| body, err := r.GetBody() | ||
| if err != nil { | ||
| return err | ||
| } | ||
| req.Body = body | ||
| req.GetBody = r.GetBody | ||
| } | ||
| } | ||
| if req.Header == nil { | ||
| req.Header = make(http.Header) | ||
| } | ||
| req.Header.Set(header.Authorization, auth) | ||
| resp.Response, err = client.GetTransport().RoundTrip(&req) | ||
| return err | ||
| } | ||
| } | ||
|
|
||
| func createDigestAuth(resp *http.Response, username, password string) (auth string, err error) { | ||
| chal := resp.Header.Get(header.WwwAuthenticate) | ||
| if chal == "" { | ||
| return "", errDigestBadChallenge | ||
| } | ||
|
|
||
| c, err := parseChallenge(chal) | ||
| if err != nil { | ||
| return "", err | ||
| } | ||
|
|
||
| // Form credentials based on the challenge | ||
| cr := newCredentials(resp.Request.URL.RequestURI(), resp.Request.Method, username, password, c) | ||
| auth, err = cr.authorize() | ||
| return | ||
| } | ||
|
|
||
| func newCredentials(digestURI, method, username, password string, c *challenge) *credentials { | ||
| return &credentials{ | ||
| username: username, | ||
| userhash: c.userhash, | ||
| realm: c.realm, | ||
| nonce: c.nonce, | ||
| digestURI: digestURI, | ||
| algorithm: c.algorithm, | ||
| sessionAlg: strings.HasSuffix(c.algorithm, "-sess"), | ||
| opaque: c.opaque, | ||
| messageQop: c.qop, | ||
| nc: 0, | ||
| method: method, | ||
| password: password, | ||
| } | ||
| } | ||
|
|
||
| type challenge struct { | ||
| realm string | ||
| domain string | ||
| nonce string | ||
| opaque string | ||
| stale string | ||
| algorithm string | ||
| qop string | ||
| userhash string | ||
| } | ||
|
|
||
| func parseChallenge(input string) (*challenge, error) { | ||
| const ws = " \n\r\t" | ||
| const qs = `"` | ||
| s := strings.Trim(input, ws) | ||
| if !strings.HasPrefix(s, "Digest ") { | ||
| return nil, errDigestBadChallenge | ||
| } | ||
| s = strings.Trim(s[7:], ws) | ||
| sl := strings.Split(s, ", ") | ||
| c := &challenge{} | ||
| var r []string | ||
| for i := range sl { | ||
| r = strings.SplitN(sl[i], "=", 2) | ||
| if len(r) != 2 { | ||
| return nil, errDigestBadChallenge | ||
| } | ||
| switch r[0] { | ||
| case "realm": | ||
| c.realm = strings.Trim(r[1], qs) | ||
| case "domain": | ||
| c.domain = strings.Trim(r[1], qs) | ||
| case "nonce": | ||
| c.nonce = strings.Trim(r[1], qs) | ||
| case "opaque": | ||
| c.opaque = strings.Trim(r[1], qs) | ||
| case "stale": | ||
| c.stale = r[1] | ||
| case "algorithm": | ||
| c.algorithm = r[1] | ||
| case "qop": | ||
| c.qop = strings.Trim(r[1], qs) | ||
| case "charset": | ||
| if strings.ToUpper(strings.Trim(r[1], qs)) != "UTF-8" { | ||
| return nil, errDigestCharset | ||
| } | ||
| case "userhash": | ||
| c.userhash = strings.Trim(r[1], qs) | ||
| default: | ||
| return nil, errDigestBadChallenge | ||
| } | ||
| } | ||
| return c, nil | ||
| } | ||
|
|
||
| type credentials struct { | ||
| username string | ||
| userhash string | ||
| realm string | ||
| nonce string | ||
| digestURI string | ||
| algorithm string | ||
| sessionAlg bool | ||
| cNonce string | ||
| opaque string | ||
| messageQop string | ||
| nc int | ||
| method string | ||
| password string | ||
| } | ||
|
|
||
| func (c *credentials) authorize() (string, error) { | ||
| if _, ok := hashFuncs[c.algorithm]; !ok { | ||
| return "", errDigestAlgNotSupported | ||
| } | ||
|
|
||
| if err := c.validateQop(); err != nil { | ||
| return "", err | ||
| } | ||
|
|
||
| resp, err := c.resp() | ||
| if err != nil { | ||
| return "", err | ||
| } | ||
|
|
||
| sl := make([]string, 0, 10) | ||
| if c.userhash == "true" { | ||
| // RFC 7616 3.4.4 | ||
| c.username = c.h(fmt.Sprintf("%s:%s", c.username, c.realm)) | ||
| sl = append(sl, fmt.Sprintf(`userhash=%s`, c.userhash)) | ||
| } | ||
| sl = append(sl, fmt.Sprintf(`username="%s"`, c.username)) | ||
| sl = append(sl, fmt.Sprintf(`realm="%s"`, c.realm)) | ||
| sl = append(sl, fmt.Sprintf(`nonce="%s"`, c.nonce)) | ||
| sl = append(sl, fmt.Sprintf(`uri="%s"`, c.digestURI)) | ||
| sl = append(sl, fmt.Sprintf(`response="%s"`, resp)) | ||
| sl = append(sl, fmt.Sprintf(`algorithm=%s`, c.algorithm)) | ||
| if c.opaque != "" { | ||
| sl = append(sl, fmt.Sprintf(`opaque="%s"`, c.opaque)) | ||
| } | ||
| if c.messageQop != "" { | ||
| sl = append(sl, fmt.Sprintf("qop=%s", c.messageQop)) | ||
| sl = append(sl, fmt.Sprintf("nc=%08x", c.nc)) | ||
| sl = append(sl, fmt.Sprintf(`cnonce="%s"`, c.cNonce)) | ||
| } | ||
|
|
||
| return fmt.Sprintf("Digest %s", strings.Join(sl, ", ")), nil | ||
| } | ||
|
|
||
| func (c *credentials) validateQop() error { | ||
| // Currently only supporting auth quality of protection. TODO: add auth-int support | ||
| if c.messageQop == "" { | ||
| return errDigestNoQop | ||
| } | ||
| possibleQops := strings.Split(c.messageQop, ", ") | ||
| var authSupport bool | ||
| for _, qop := range possibleQops { | ||
| if qop == "auth" { | ||
| authSupport = true | ||
| break | ||
| } | ||
| } | ||
| if !authSupport { | ||
| return errDigestQopNotSupported | ||
| } | ||
|
|
||
| c.messageQop = "auth" | ||
|
|
||
| return nil | ||
| } | ||
|
|
||
| func (c *credentials) h(data string) string { | ||
| hfCtor := hashFuncs[c.algorithm] | ||
| hf := hfCtor() | ||
| _, _ = hf.Write([]byte(data)) // Hash.Write never returns an error | ||
| return fmt.Sprintf("%x", hf.Sum(nil)) | ||
| } | ||
|
|
||
| func (c *credentials) resp() (string, error) { | ||
| c.nc++ | ||
|
|
||
| b := make([]byte, 16) | ||
| _, err := io.ReadFull(rand.Reader, b) | ||
| if err != nil { | ||
| return "", err | ||
| } | ||
| c.cNonce = fmt.Sprintf("%x", b)[:32] | ||
|
|
||
| ha1 := c.ha1() | ||
| ha2 := c.ha2() | ||
|
|
||
| return c.kd(ha1, fmt.Sprintf("%s:%08x:%s:%s:%s", | ||
| c.nonce, c.nc, c.cNonce, c.messageQop, ha2)), nil | ||
| } | ||
|
|
||
| func (c *credentials) kd(secret, data string) string { | ||
| return c.h(fmt.Sprintf("%s:%s", secret, data)) | ||
| } | ||
|
|
||
| // RFC 7616 3.4.2 | ||
| func (c *credentials) ha1() string { | ||
| ret := c.h(fmt.Sprintf("%s:%s:%s", c.username, c.realm, c.password)) | ||
| if c.sessionAlg { | ||
| return c.h(fmt.Sprintf("%s:%s:%s", ret, c.nonce, c.cNonce)) | ||
| } | ||
|
|
||
| return ret | ||
| } | ||
|
|
||
| // RFC 7616 3.4.3 | ||
| func (c *credentials) ha2() string { | ||
| // currently no auth-int support | ||
| return c.h(fmt.Sprintf("%s:%s", c.method, c.digestURI)) | ||
| } |
Oops, something went wrong.