gitee.com/ks-custle/core-gm@v0.0.0-20230922171213-b83bdd97b62c/gmhttp/client.go (about) 1 // Copyright 2009 The Go Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style 3 // license that can be found in the LICENSE file. 4 5 // HTTP client. See RFC 7230 through 7235. 6 // 7 // This is the high-level Client interface. 8 // The low-level implementation is in transport.go. 9 10 package gmhttp 11 12 import ( 13 "context" 14 "encoding/base64" 15 "errors" 16 "fmt" 17 "io" 18 "log" 19 "net/url" 20 "reflect" 21 "sort" 22 "strings" 23 "sync" 24 "time" 25 26 "gitee.com/ks-custle/core-gm/gmhttp/internal/ascii" 27 tls "gitee.com/ks-custle/core-gm/gmtls" 28 ) 29 30 // A Client is an HTTP client. Its zero value (DefaultClient) is a 31 // usable client that uses DefaultTransport. 32 // 33 // The Client's Transport typically has internal state (cached TCP 34 // connections), so Clients should be reused instead of created as 35 // needed. Clients are safe for concurrent use by multiple goroutines. 36 // 37 // A Client is higher-level than a RoundTripper (such as Transport) 38 // and additionally handles HTTP details such as cookies and 39 // redirects. 40 // 41 // When following redirects, the Client will forward all headers set on the 42 // initial Request except: 43 // 44 // • when forwarding sensitive headers like "Authorization", 45 // "WWW-Authenticate", and "Cookie" to untrusted targets. 46 // These headers will be ignored when following a redirect to a domain 47 // that is not a subdomain match or exact match of the initial domain. 48 // For example, a redirect from "foo.com" to either "foo.com" or "sub.foo.com" 49 // will forward the sensitive headers, but a redirect to "bar.com" will not. 50 // 51 // • when forwarding the "Cookie" header with a non-nil cookie Jar. 52 // Since each redirect may mutate the state of the cookie jar, 53 // a redirect may possibly alter a cookie set in the initial request. 54 // When forwarding the "Cookie" header, any mutated cookies will be omitted, 55 // with the expectation that the Jar will insert those mutated cookies 56 // with the updated values (assuming the origin matches). 57 // If Jar is nil, the initial cookies are forwarded without change. 58 type Client struct { 59 // Transport specifies the mechanism by which individual 60 // HTTP requests are made. 61 // If nil, DefaultTransport is used. 62 Transport RoundTripper 63 64 // CheckRedirect specifies the policy for handling redirects. 65 // If CheckRedirect is not nil, the client calls it before 66 // following an HTTP redirect. The arguments req and via are 67 // the upcoming request and the requests made already, oldest 68 // first. If CheckRedirect returns an error, the Client's Get 69 // method returns both the previous Response (with its Body 70 // closed) and CheckRedirect's error (wrapped in a url.Error) 71 // instead of issuing the Request req. 72 // As a special case, if CheckRedirect returns ErrUseLastResponse, 73 // then the most recent response is returned with its body 74 // unclosed, along with a nil error. 75 // 76 // If CheckRedirect is nil, the Client uses its default policy, 77 // which is to stop after 10 consecutive requests. 78 CheckRedirect func(req *Request, via []*Request) error 79 80 // Jar specifies the cookie jar. 81 // 82 // The Jar is used to insert relevant cookies into every 83 // outbound Request and is updated with the cookie values 84 // of every inbound Response. The Jar is consulted for every 85 // redirect that the Client follows. 86 // 87 // If Jar is nil, cookies are only sent if they are explicitly 88 // set on the Request. 89 Jar CookieJar 90 91 // Timeout specifies a time limit for requests made by this 92 // Client. The timeout includes connection time, any 93 // redirects, and reading the response body. The timer remains 94 // running after Get, Head, Post, or Do return and will 95 // interrupt reading of the Response.Body. 96 // 97 // A Timeout of zero means no timeout. 98 // 99 // The Client cancels requests to the underlying Transport 100 // as if the Request's Context ended. 101 // 102 // For compatibility, the Client will also use the deprecated 103 // CancelRequest method on Transport if found. New 104 // RoundTripper implementations should use the Request's Context 105 // for cancellation instead of implementing CancelRequest. 106 Timeout time.Duration 107 } 108 109 // DefaultClient is the default Client and is used by Get, Head, and Post. 110 var DefaultClient = &Client{} 111 112 // RoundTripper is an interface representing the ability to execute a 113 // single HTTP transaction, obtaining the Response for a given Request. 114 // 115 // A RoundTripper must be safe for concurrent use by multiple 116 // goroutines. 117 type RoundTripper interface { 118 // RoundTrip executes a single HTTP transaction, returning 119 // a Response for the provided Request. 120 // 121 // RoundTrip should not attempt to interpret the response. In 122 // particular, RoundTrip must return err == nil if it obtained 123 // a response, regardless of the response's HTTP status code. 124 // A non-nil err should be reserved for failure to obtain a 125 // response. Similarly, RoundTrip should not attempt to 126 // handle higher-level protocol details such as redirects, 127 // authentication, or cookies. 128 // 129 // RoundTrip should not modify the request, except for 130 // consuming and closing the Request's Body. RoundTrip may 131 // read fields of the request in a separate goroutine. Callers 132 // should not mutate or reuse the request until the Response's 133 // Body has been closed. 134 // 135 // RoundTrip must always close the body, including on errors, 136 // but depending on the implementation may do so in a separate 137 // goroutine even after RoundTrip returns. This means that 138 // callers wanting to reuse the body for subsequent requests 139 // must arrange to wait for the Close call before doing so. 140 // 141 // The Request's URL and Header fields must be initialized. 142 RoundTrip(*Request) (*Response, error) 143 } 144 145 // refererForURL returns a referer without any authentication info or 146 // an empty string if lastReq scheme is https and newReq scheme is http. 147 func refererForURL(lastReq, newReq *url.URL) string { 148 // https://tools.ietf.org/html/rfc7231#section-5.5.2 149 // "Clients SHOULD NOT include a Referer header field in a 150 // (non-secure) HTTP request if the referring page was 151 // transferred with a secure protocol." 152 if lastReq.Scheme == "https" && newReq.Scheme == "http" { 153 return "" 154 } 155 referer := lastReq.String() 156 if lastReq.User != nil { 157 // This is not very efficient, but is the best we can 158 // do without: 159 // - introducing a new method on URL 160 // - creating a race condition 161 // - copying the URL struct manually, which would cause 162 // maintenance problems down the line 163 auth := lastReq.User.String() + "@" 164 referer = strings.Replace(referer, auth, "", 1) 165 } 166 return referer 167 } 168 169 // didTimeout is non-nil only if err != nil. 170 func (c *Client) send(req *Request, deadline time.Time) (resp *Response, didTimeout func() bool, err error) { 171 if c.Jar != nil { 172 for _, cookie := range c.Jar.Cookies(req.URL) { 173 req.AddCookie(cookie) 174 } 175 } 176 resp, didTimeout, err = send(req, c.transport(), deadline) 177 if err != nil { 178 return nil, didTimeout, err 179 } 180 if c.Jar != nil { 181 if rc := resp.Cookies(); len(rc) > 0 { 182 c.Jar.SetCookies(req.URL, rc) 183 } 184 } 185 return resp, nil, nil 186 } 187 188 func (c *Client) deadline() time.Time { 189 if c.Timeout > 0 { 190 return time.Now().Add(c.Timeout) 191 } 192 return time.Time{} 193 } 194 195 func (c *Client) transport() RoundTripper { 196 if c.Transport != nil { 197 return c.Transport 198 } 199 return DefaultTransport 200 } 201 202 // send issues an HTTP request. 203 // Caller should close resp.Body when done reading from it. 204 func send(ireq *Request, rt RoundTripper, deadline time.Time) (resp *Response, didTimeout func() bool, err error) { 205 req := ireq // req is either the original request, or a modified fork 206 207 if rt == nil { 208 _ = req.closeBody() 209 return nil, alwaysFalse, errors.New("http: no Client.Transport or DefaultTransport") 210 } 211 212 if req.URL == nil { 213 _ = req.closeBody() 214 return nil, alwaysFalse, errors.New("http: nil Request.URL") 215 } 216 217 if req.RequestURI != "" { 218 _ = req.closeBody() 219 return nil, alwaysFalse, errors.New("http: Request.RequestURI can't be set in client requests") 220 } 221 222 // forkReq forks req into a shallow clone of ireq the first 223 // time it's called. 224 forkReq := func() { 225 if ireq == req { 226 req = new(Request) 227 *req = *ireq // shallow clone 228 } 229 } 230 231 // Most the callers of send (Get, Post, et al) don't need 232 // Headers, leaving it uninitialized. We guarantee to the 233 // Transport that this has been initialized, though. 234 if req.Header == nil { 235 forkReq() 236 req.Header = make(Header) 237 } 238 239 if u := req.URL.User; u != nil && req.Header.Get("Authorization") == "" { 240 username := u.Username() 241 password, _ := u.Password() 242 forkReq() 243 req.Header = cloneOrMakeHeader(ireq.Header) 244 req.Header.Set("Authorization", "Basic "+basicAuth(username, password)) 245 } 246 247 if !deadline.IsZero() { 248 forkReq() 249 } 250 stopTimer, didTimeout := setRequestCancel(req, rt, deadline) 251 252 resp, err = rt.RoundTrip(req) 253 if err != nil { 254 stopTimer() 255 if resp != nil { 256 log.Printf("RoundTripper returned a response & error; ignoring response") 257 } 258 if tlsErr, ok := err.(tls.RecordHeaderError); ok { 259 // If we get a bad TLS record header, check to see if the 260 // response looks like HTTP and give a more helpful error. 261 // See golang.org/issue/11111. 262 if string(tlsErr.RecordHeader[:]) == "HTTP/" { 263 err = errors.New("http: server gave HTTP response to HTTPS client") 264 } 265 } 266 return nil, didTimeout, err 267 } 268 if resp == nil { 269 return nil, didTimeout, fmt.Errorf("http: RoundTripper implementation (%T) returned a nil *Response with a nil error", rt) 270 } 271 if resp.Body == nil { 272 // The documentation on the Body field says “The http Client and Transport 273 // guarantee that Body is always non-nil, even on responses without a body 274 // or responses with a zero-length body.” Unfortunately, we didn't document 275 // that same constraint for arbitrary RoundTripper implementations, and 276 // RoundTripper implementations in the wild (mostly in tests) assume that 277 // they can use a nil Body to mean an empty one (similar to Request.Body). 278 // (See https://golang.org/issue/38095.) 279 // 280 // If the ContentLength allows the Body to be empty, fill in an empty one 281 // here to ensure that it is non-nil. 282 if resp.ContentLength > 0 && req.Method != "HEAD" { 283 return nil, didTimeout, fmt.Errorf("http: RoundTripper implementation (%T) returned a *Response with content length %d but a nil Body", rt, resp.ContentLength) 284 } 285 resp.Body = io.NopCloser(strings.NewReader("")) 286 } 287 if !deadline.IsZero() { 288 resp.Body = &cancelTimerBody{ 289 stop: stopTimer, 290 rc: resp.Body, 291 reqDidTimeout: didTimeout, 292 } 293 } 294 return resp, nil, nil 295 } 296 297 // timeBeforeContextDeadline reports whether the non-zero Time t is 298 // before ctx's deadline, if any. If ctx does not have a deadline, it 299 // always reports true (the deadline is considered infinite). 300 func timeBeforeContextDeadline(t time.Time, ctx context.Context) bool { 301 d, ok := ctx.Deadline() 302 if !ok { 303 return true 304 } 305 return t.Before(d) 306 } 307 308 // knownRoundTripperImpl reports whether rt is a RoundTripper that's 309 // maintained by the Go team and known to implement the latest 310 // optional semantics (notably contexts). The Request is used 311 // to check whether this particular request is using an alternate protocol, 312 // in which case we need to check the RoundTripper for that protocol. 313 func knownRoundTripperImpl(rt RoundTripper, req *Request) bool { 314 switch t := rt.(type) { 315 case *Transport: 316 if altRT := t.alternateRoundTripper(req); altRT != nil { 317 return knownRoundTripperImpl(altRT, req) 318 } 319 return true 320 case *http2Transport, http2noDialH2RoundTripper: 321 return true 322 } 323 // There's a very minor chance of a false positive with this. 324 // Instead of detecting our golang.org/x/net/http2.Transport, 325 // it might detect a Transport type in a different http2 326 // package. But I know of none, and the only problem would be 327 // some temporarily leaked goroutines if the transport didn't 328 // support contexts. So this is a good enough heuristic: 329 if reflect.TypeOf(rt).String() == "*http2.Transport" { 330 return true 331 } 332 return false 333 } 334 335 // setRequestCancel sets req.Cancel and adds a deadline context to req 336 // if deadline is non-zero. The RoundTripper's type is used to 337 // determine whether the legacy CancelRequest behavior should be used. 338 // 339 // As background, there are three ways to cancel a request: 340 // First was Transport.CancelRequest. (deprecated) 341 // Second was Request.Cancel. 342 // Third was Request.Context. 343 // This function populates the second and third, and uses the first if it really needs to. 344 func setRequestCancel(req *Request, rt RoundTripper, deadline time.Time) (stopTimer func(), didTimeout func() bool) { 345 if deadline.IsZero() { 346 return nop, alwaysFalse 347 } 348 knownTransport := knownRoundTripperImpl(rt, req) 349 oldCtx := req.Context() 350 351 if req.Cancel == nil && knownTransport { 352 // If they already had a Request.Context that's 353 // expiring sooner, do nothing: 354 if !timeBeforeContextDeadline(deadline, oldCtx) { 355 return nop, alwaysFalse 356 } 357 358 var cancelCtx func() 359 req.ctx, cancelCtx = context.WithDeadline(oldCtx, deadline) 360 return cancelCtx, func() bool { return time.Now().After(deadline) } 361 } 362 initialReqCancel := req.Cancel // the user's original Request.Cancel, if any 363 364 var cancelCtx func() 365 if oldCtx := req.Context(); timeBeforeContextDeadline(deadline, oldCtx) { 366 req.ctx, cancelCtx = context.WithDeadline(oldCtx, deadline) 367 } 368 369 cancel := make(chan struct{}) 370 req.Cancel = cancel 371 372 doCancel := func() { 373 // The second way in the func comment above: 374 close(cancel) 375 // The first way, used only for RoundTripper 376 // implementations written before Go 1.5 or Go 1.6. 377 type canceler interface{ CancelRequest(*Request) } 378 if v, ok := rt.(canceler); ok { 379 v.CancelRequest(req) 380 } 381 } 382 383 stopTimerCh := make(chan struct{}) 384 var once sync.Once 385 stopTimer = func() { 386 once.Do(func() { 387 close(stopTimerCh) 388 if cancelCtx != nil { 389 cancelCtx() 390 } 391 }) 392 } 393 394 timer := time.NewTimer(time.Until(deadline)) 395 var timedOut atomicBool 396 397 go func() { 398 select { 399 case <-initialReqCancel: 400 doCancel() 401 timer.Stop() 402 case <-timer.C: 403 timedOut.setTrue() 404 doCancel() 405 case <-stopTimerCh: 406 timer.Stop() 407 } 408 }() 409 410 return stopTimer, timedOut.isSet 411 } 412 413 // See 2 (end of page 4) https://www.ietf.org/rfc/rfc2617.txt 414 // "To receive authorization, the client sends the userid and password, 415 // separated by a single colon (":") character, within a base64 416 // encoded string in the credentials." 417 // It is not meant to be urlencoded. 418 func basicAuth(username, password string) string { 419 auth := username + ":" + password 420 return base64.StdEncoding.EncodeToString([]byte(auth)) 421 } 422 423 // Get issues a GET to the specified URL. If the response is one of 424 // the following redirect codes, Get follows the redirect, up to a 425 // maximum of 10 redirects: 426 // 427 // 301 (Moved Permanently) 428 // 302 (Found) 429 // 303 (See Other) 430 // 307 (Temporary Redirect) 431 // 308 (Permanent Redirect) 432 // 433 // An error is returned if there were too many redirects or if there 434 // was an HTTP protocol error. A non-2xx response doesn't cause an 435 // error. Any returned error will be of type *url.Error. The url.Error 436 // value's Timeout method will report true if the request timed out. 437 // 438 // When err is nil, resp always contains a non-nil resp.Body. 439 // Caller should close resp.Body when done reading from it. 440 // 441 // Get is a wrapper around DefaultClient.Get. 442 // 443 // To make a request with custom headers, use NewRequest and 444 // DefaultClient.Do. 445 // 446 // To make a request with a specified context.Context, use NewRequestWithContext 447 // and DefaultClient.Do. 448 func Get(url string) (resp *Response, err error) { 449 return DefaultClient.Get(url) 450 } 451 452 // Get issues a GET to the specified URL. If the response is one of the 453 // following redirect codes, Get follows the redirect after calling the 454 // Client's CheckRedirect function: 455 // 456 // 301 (Moved Permanently) 457 // 302 (Found) 458 // 303 (See Other) 459 // 307 (Temporary Redirect) 460 // 308 (Permanent Redirect) 461 // 462 // An error is returned if the Client's CheckRedirect function fails 463 // or if there was an HTTP protocol error. A non-2xx response doesn't 464 // cause an error. Any returned error will be of type *url.Error. The 465 // url.Error value's Timeout method will report true if the request 466 // timed out. 467 // 468 // When err is nil, resp always contains a non-nil resp.Body. 469 // Caller should close resp.Body when done reading from it. 470 // 471 // To make a request with custom headers, use NewRequest and Client.Do. 472 // 473 // To make a request with a specified context.Context, use NewRequestWithContext 474 // and Client.Do. 475 func (c *Client) Get(url string) (resp *Response, err error) { 476 req, err := NewRequest("GET", url, nil) 477 if err != nil { 478 return nil, err 479 } 480 return c.Do(req) 481 } 482 483 func alwaysFalse() bool { return false } 484 485 // ErrUseLastResponse can be returned by Client.CheckRedirect hooks to 486 // control how redirects are processed. If returned, the next request 487 // is not sent and the most recent response is returned with its body 488 // unclosed. 489 var ErrUseLastResponse = errors.New("gitee.com/ks-custle/core-gm/gmhttp: use last response") 490 491 // checkRedirect calls either the user's configured CheckRedirect 492 // function, or the default. 493 func (c *Client) checkRedirect(req *Request, via []*Request) error { 494 fn := c.CheckRedirect 495 if fn == nil { 496 fn = defaultCheckRedirect 497 } 498 return fn(req, via) 499 } 500 501 // redirectBehavior describes what should happen when the 502 // client encounters a 3xx status code from the server 503 func redirectBehavior(reqMethod string, resp *Response, ireq *Request) (redirectMethod string, shouldRedirect, includeBody bool) { 504 switch resp.StatusCode { 505 case 301, 302, 303: 506 redirectMethod = reqMethod 507 shouldRedirect = true 508 includeBody = false 509 510 // RFC 2616 allowed automatic redirection only with GET and 511 // HEAD requests. RFC 7231 lifts this restriction, but we still 512 // restrict other methods to GET to maintain compatibility. 513 // See Issue 18570. 514 if reqMethod != "GET" && reqMethod != "HEAD" { 515 redirectMethod = "GET" 516 } 517 case 307, 308: 518 redirectMethod = reqMethod 519 shouldRedirect = true 520 includeBody = true 521 522 // Treat 307 and 308 specially, since they're new in 523 // Go 1.8, and they also require re-sending the request body. 524 if resp.Header.Get("Location") == "" { 525 // 308s have been observed in the wild being served 526 // without Location headers. Since Go 1.7 and earlier 527 // didn't follow these codes, just stop here instead 528 // of returning an error. 529 // See Issue 17773. 530 shouldRedirect = false 531 break 532 } 533 if ireq.GetBody == nil && ireq.outgoingLength() != 0 { 534 // We had a request body, and 307/308 require 535 // re-sending it, but GetBody is not defined. So just 536 // return this response to the user instead of an 537 // error, like we did in Go 1.7 and earlier. 538 shouldRedirect = false 539 } 540 } 541 return redirectMethod, shouldRedirect, includeBody 542 } 543 544 // urlErrorOp returns the (*url.Error).Op value to use for the 545 // provided (*Request).Method value. 546 func urlErrorOp(method string) string { 547 if method == "" { 548 return "Get" 549 } 550 if lowerMethod, ok := ascii.ToLower(method); ok { 551 return method[:1] + lowerMethod[1:] 552 } 553 return method 554 } 555 556 // Do sends an HTTP request and returns an HTTP response, following 557 // policy (such as redirects, cookies, auth) as configured on the 558 // client. 559 // 560 // An error is returned if caused by client policy (such as 561 // CheckRedirect), or failure to speak HTTP (such as a network 562 // connectivity problem). A non-2xx status code doesn't cause an 563 // error. 564 // 565 // If the returned error is nil, the Response will contain a non-nil 566 // Body which the user is expected to close. If the Body is not both 567 // read to EOF and closed, the Client's underlying RoundTripper 568 // (typically Transport) may not be able to re-use a persistent TCP 569 // connection to the server for a subsequent "keep-alive" request. 570 // 571 // The request Body, if non-nil, will be closed by the underlying 572 // Transport, even on errors. 573 // 574 // On error, any Response can be ignored. A non-nil Response with a 575 // non-nil error only occurs when CheckRedirect fails, and even then 576 // the returned Response.Body is already closed. 577 // 578 // Generally Get, Post, or PostForm will be used instead of Do. 579 // 580 // If the server replies with a redirect, the Client first uses the 581 // CheckRedirect function to determine whether the redirect should be 582 // followed. If permitted, a 301, 302, or 303 redirect causes 583 // subsequent requests to use HTTP method GET 584 // (or HEAD if the original request was HEAD), with no body. 585 // A 307 or 308 redirect preserves the original HTTP method and body, 586 // provided that the Request.GetBody function is defined. 587 // The NewRequest function automatically sets GetBody for common 588 // standard library body types. 589 // 590 // Any returned error will be of type *url.Error. The url.Error 591 // value's Timeout method will report true if the request timed out. 592 func (c *Client) Do(req *Request) (*Response, error) { 593 return c.do(req) 594 } 595 596 var testHookClientDoResult func(retres *Response, reterr error) 597 598 func (c *Client) do(req *Request) (retres *Response, reterr error) { 599 if testHookClientDoResult != nil { 600 defer func() { testHookClientDoResult(retres, reterr) }() 601 } 602 if req.URL == nil { 603 _ = req.closeBody() 604 return nil, &url.Error{ 605 Op: urlErrorOp(req.Method), 606 Err: errors.New("http: nil Request.URL"), 607 } 608 } 609 610 var ( 611 deadline = c.deadline() 612 reqs []*Request 613 resp *Response 614 copyHeaders = c.makeHeadersCopier(req) 615 reqBodyClosed = false // have we closed the current req.Body? 616 617 // Redirect behavior: 618 redirectMethod string 619 includeBody bool 620 ) 621 uerr := func(err error) error { 622 // the body may have been closed already by c.send() 623 if !reqBodyClosed { 624 _ = req.closeBody() 625 } 626 var urlStr string 627 if resp != nil && resp.Request != nil { 628 urlStr = stripPassword(resp.Request.URL) 629 } else { 630 urlStr = stripPassword(req.URL) 631 } 632 return &url.Error{ 633 Op: urlErrorOp(reqs[0].Method), 634 URL: urlStr, 635 Err: err, 636 } 637 } 638 for { 639 // For all but the first request, create the next 640 // request hop and replace req. 641 if len(reqs) > 0 { 642 loc := resp.Header.Get("Location") 643 if loc == "" { 644 resp.closeBody() 645 return nil, uerr(fmt.Errorf("%d response missing Location header", resp.StatusCode)) 646 } 647 u, err := req.URL.Parse(loc) 648 if err != nil { 649 resp.closeBody() 650 return nil, uerr(fmt.Errorf("failed to parse Location header %q: %v", loc, err)) 651 } 652 host := "" 653 if req.Host != "" && req.Host != req.URL.Host { 654 // If the caller specified a custom Host header and the 655 // redirect location is relative, preserve the Host header 656 // through the redirect. See issue #22233. 657 if u, _ := url.Parse(loc); u != nil && !u.IsAbs() { 658 host = req.Host 659 } 660 } 661 ireq := reqs[0] 662 req = &Request{ 663 Method: redirectMethod, 664 Response: resp, 665 URL: u, 666 Header: make(Header), 667 Host: host, 668 Cancel: ireq.Cancel, 669 ctx: ireq.ctx, 670 } 671 if includeBody && ireq.GetBody != nil { 672 req.Body, err = ireq.GetBody() 673 if err != nil { 674 resp.closeBody() 675 return nil, uerr(err) 676 } 677 req.ContentLength = ireq.ContentLength 678 } 679 680 // Copy original headers before setting the Referer, 681 // in case the user set Referer on their first request. 682 // If they really want to override, they can do it in 683 // their CheckRedirect func. 684 copyHeaders(req) 685 686 // Add the Referer header from the most recent 687 // request URL to the new one, if it's not https->http: 688 if ref := refererForURL(reqs[len(reqs)-1].URL, req.URL); ref != "" { 689 req.Header.Set("Referer", ref) 690 } 691 err = c.checkRedirect(req, reqs) 692 693 // Sentinel error to let users select the 694 // previous response, without closing its 695 // body. See Issue 10069. 696 if err == ErrUseLastResponse { 697 return resp, nil 698 } 699 700 // Close the previous response's body. But 701 // read at least some of the body so if it's 702 // small the underlying TCP connection will be 703 // re-used. No need to check for errors: if it 704 // fails, the Transport won't reuse it anyway. 705 const maxBodySlurpSize = 2 << 10 706 if resp.ContentLength == -1 || resp.ContentLength <= maxBodySlurpSize { 707 _, _ = io.CopyN(io.Discard, resp.Body, maxBodySlurpSize) 708 } 709 _ = resp.Body.Close() 710 711 if err != nil { 712 // Special case for Go 1 compatibility: return both the response 713 // and an error if the CheckRedirect function failed. 714 // See https://golang.org/issue/3795 715 // The resp.Body has already been closed. 716 ue := uerr(err) 717 ue.(*url.Error).URL = loc 718 return resp, ue 719 } 720 } 721 722 reqs = append(reqs, req) 723 var err error 724 var didTimeout func() bool 725 if resp, didTimeout, err = c.send(req, deadline); err != nil { 726 // c.send() always closes req.Body 727 reqBodyClosed = true 728 if !deadline.IsZero() && didTimeout() { 729 err = &httpError{ 730 err: err.Error() + " (Client.Timeout exceeded while awaiting headers)", 731 timeout: true, 732 } 733 } 734 return nil, uerr(err) 735 } 736 737 var shouldRedirect bool 738 redirectMethod, shouldRedirect, includeBody = redirectBehavior(req.Method, resp, reqs[0]) 739 if !shouldRedirect { 740 return resp, nil 741 } 742 743 _ = req.closeBody() 744 } 745 } 746 747 // makeHeadersCopier makes a function that copies headers from the 748 // initial Request, ireq. For every redirect, this function must be called 749 // so that it can copy headers into the upcoming Request. 750 func (c *Client) makeHeadersCopier(ireq *Request) func(*Request) { 751 // The headers to copy are from the very initial request. 752 // We use a closured callback to keep a reference to these original headers. 753 var ( 754 ireqhdr = cloneOrMakeHeader(ireq.Header) 755 icookies map[string][]*Cookie 756 ) 757 if c.Jar != nil && ireq.Header.Get("Cookie") != "" { 758 icookies = make(map[string][]*Cookie) 759 for _, c := range ireq.Cookies() { 760 icookies[c.Name] = append(icookies[c.Name], c) 761 } 762 } 763 764 preq := ireq // The previous request 765 return func(req *Request) { 766 // If Jar is present and there was some initial cookies provided 767 // via the request header, then we may need to alter the initial 768 // cookies as we follow redirects since each redirect may end up 769 // modifying a pre-existing cookie. 770 // 771 // Since cookies already set in the request header do not contain 772 // information about the original domain and path, the logic below 773 // assumes any new set cookies override the original cookie 774 // regardless of domain or path. 775 // 776 // See https://golang.org/issue/17494 777 if c.Jar != nil && icookies != nil { 778 var changed bool 779 resp := req.Response // The response that caused the upcoming redirect 780 for _, c := range resp.Cookies() { 781 if _, ok := icookies[c.Name]; ok { 782 delete(icookies, c.Name) 783 changed = true 784 } 785 } 786 if changed { 787 ireqhdr.Del("Cookie") 788 var ss []string 789 for _, cs := range icookies { 790 for _, c := range cs { 791 ss = append(ss, c.Name+"="+c.Value) 792 } 793 } 794 sort.Strings(ss) // Ensure deterministic headers 795 ireqhdr.Set("Cookie", strings.Join(ss, "; ")) 796 } 797 } 798 799 // Copy the initial request's Header values 800 // (at least the safe ones). 801 for k, vv := range ireqhdr { 802 if shouldCopyHeaderOnRedirect(k, preq.URL, req.URL) { 803 req.Header[k] = vv 804 } 805 } 806 807 preq = req // Update previous Request with the current request 808 } 809 } 810 811 //goland:noinspection GoUnusedParameter 812 func defaultCheckRedirect(req *Request, via []*Request) error { 813 if len(via) >= 10 { 814 return errors.New("stopped after 10 redirects") 815 } 816 return nil 817 } 818 819 // Post issues a POST to the specified URL. 820 // 821 // Caller should close resp.Body when done reading from it. 822 // 823 // If the provided body is an io.Closer, it is closed after the 824 // request. 825 // 826 // Post is a wrapper around DefaultClient.Post. 827 // 828 // To set custom headers, use NewRequest and DefaultClient.Do. 829 // 830 // See the Client.Do method documentation for details on how redirects 831 // are handled. 832 // 833 // To make a request with a specified context.Context, use NewRequestWithContext 834 // and DefaultClient.Do. 835 func Post(url, contentType string, body io.Reader) (resp *Response, err error) { 836 return DefaultClient.Post(url, contentType, body) 837 } 838 839 // Post issues a POST to the specified URL. 840 // 841 // Caller should close resp.Body when done reading from it. 842 // 843 // If the provided body is an io.Closer, it is closed after the 844 // request. 845 // 846 // To set custom headers, use NewRequest and Client.Do. 847 // 848 // To make a request with a specified context.Context, use NewRequestWithContext 849 // and Client.Do. 850 // 851 // See the Client.Do method documentation for details on how redirects 852 // are handled. 853 func (c *Client) Post(url, contentType string, body io.Reader) (resp *Response, err error) { 854 req, err := NewRequest("POST", url, body) 855 if err != nil { 856 return nil, err 857 } 858 req.Header.Set("Content-Type", contentType) 859 return c.Do(req) 860 } 861 862 // PostForm issues a POST to the specified URL, with data's keys and 863 // values URL-encoded as the request body. 864 // 865 // The Content-Type header is set to application/x-www-form-urlencoded. 866 // To set other headers, use NewRequest and DefaultClient.Do. 867 // 868 // When err is nil, resp always contains a non-nil resp.Body. 869 // Caller should close resp.Body when done reading from it. 870 // 871 // PostForm is a wrapper around DefaultClient.PostForm. 872 // 873 // See the Client.Do method documentation for details on how redirects 874 // are handled. 875 // 876 // To make a request with a specified context.Context, use NewRequestWithContext 877 // and DefaultClient.Do. 878 // 879 //goland:noinspection GoUnusedExportedFunction 880 func PostForm(url string, data url.Values) (resp *Response, err error) { 881 return DefaultClient.PostForm(url, data) 882 } 883 884 // PostForm issues a POST to the specified URL, 885 // with data's keys and values URL-encoded as the request body. 886 // 887 // The Content-Type header is set to application/x-www-form-urlencoded. 888 // To set other headers, use NewRequest and Client.Do. 889 // 890 // When err is nil, resp always contains a non-nil resp.Body. 891 // Caller should close resp.Body when done reading from it. 892 // 893 // See the Client.Do method documentation for details on how redirects 894 // are handled. 895 // 896 // To make a request with a specified context.Context, use NewRequestWithContext 897 // and Client.Do. 898 func (c *Client) PostForm(url string, data url.Values) (resp *Response, err error) { 899 return c.Post(url, "application/x-www-form-urlencoded", strings.NewReader(data.Encode())) 900 } 901 902 // Head issues a HEAD to the specified URL. If the response is one of 903 // the following redirect codes, Head follows the redirect, up to a 904 // maximum of 10 redirects: 905 // 906 // 301 (Moved Permanently) 907 // 302 (Found) 908 // 303 (See Other) 909 // 307 (Temporary Redirect) 910 // 308 (Permanent Redirect) 911 // 912 // # Head is a wrapper around DefaultClient.Head 913 // 914 // To make a request with a specified context.Context, use NewRequestWithContext 915 // and DefaultClient.Do. 916 // 917 //goland:noinspection GoUnusedExportedFunction 918 func Head(url string) (resp *Response, err error) { 919 return DefaultClient.Head(url) 920 } 921 922 // Head issues a HEAD to the specified URL. If the response is one of the 923 // following redirect codes, Head follows the redirect after calling the 924 // Client's CheckRedirect function: 925 // 926 // 301 (Moved Permanently) 927 // 302 (Found) 928 // 303 (See Other) 929 // 307 (Temporary Redirect) 930 // 308 (Permanent Redirect) 931 // 932 // To make a request with a specified context.Context, use NewRequestWithContext 933 // and Client.Do. 934 func (c *Client) Head(url string) (resp *Response, err error) { 935 req, err := NewRequest("HEAD", url, nil) 936 if err != nil { 937 return nil, err 938 } 939 return c.Do(req) 940 } 941 942 // CloseIdleConnections closes any connections on its Transport which 943 // were previously connected from previous requests but are now 944 // sitting idle in a "keep-alive" state. It does not interrupt any 945 // connections currently in use. 946 // 947 // If the Client's Transport does not have a CloseIdleConnections method 948 // then this method does nothing. 949 func (c *Client) CloseIdleConnections() { 950 type closeIdler interface { 951 CloseIdleConnections() 952 } 953 if tr, ok := c.transport().(closeIdler); ok { 954 tr.CloseIdleConnections() 955 } 956 } 957 958 // cancelTimerBody is an io.ReadCloser that wraps rc with two features: 959 // 1. On Read error or close, the stop func is called. 960 // 2. On Read failure, if reqDidTimeout is true, the error is wrapped and 961 // marked as net.Error that hit its timeout. 962 type cancelTimerBody struct { 963 stop func() // stops the time.Timer waiting to cancel the request 964 rc io.ReadCloser 965 reqDidTimeout func() bool 966 } 967 968 func (b *cancelTimerBody) Read(p []byte) (n int, err error) { 969 n, err = b.rc.Read(p) 970 if err == nil { 971 return n, nil 972 } 973 if err == io.EOF { 974 return n, err 975 } 976 if b.reqDidTimeout() { 977 err = &httpError{ 978 err: err.Error() + " (Client.Timeout or context cancellation while reading body)", 979 timeout: true, 980 } 981 } 982 return n, err 983 } 984 985 func (b *cancelTimerBody) Close() error { 986 err := b.rc.Close() 987 b.stop() 988 return err 989 } 990 991 func shouldCopyHeaderOnRedirect(headerKey string, initial, dest *url.URL) bool { 992 switch CanonicalHeaderKey(headerKey) { 993 case "Authorization", "Www-Authenticate", "Cookie", "Cookie2": 994 // Permit sending auth/cookie headers from "foo.com" 995 // to "sub.foo.com". 996 997 // Note that we don't send all cookies to subdomains 998 // automatically. This function is only used for 999 // Cookies set explicitly on the initial outgoing 1000 // client request. Cookies automatically added via the 1001 // CookieJar mechanism continue to follow each 1002 // cookie's scope as set by Set-Cookie. But for 1003 // outgoing requests with the Cookie header set 1004 // directly, we don't know their scope, so we assume 1005 // it's for *.domain.com. 1006 1007 ihost := canonicalAddr(initial) 1008 dhost := canonicalAddr(dest) 1009 return isDomainOrSubdomain(dhost, ihost) 1010 } 1011 // All other headers are copied: 1012 return true 1013 } 1014 1015 // isDomainOrSubdomain reports whether sub is a subdomain (or exact 1016 // match) of the parent domain. 1017 // 1018 // Both domains must already be in canonical form. 1019 func isDomainOrSubdomain(sub, parent string) bool { 1020 if sub == parent { 1021 return true 1022 } 1023 // If sub is "foo.example.com" and parent is "example.com", 1024 // that means sub must end in "."+parent. 1025 // Do it without allocating. 1026 if !strings.HasSuffix(sub, parent) { 1027 return false 1028 } 1029 return sub[len(sub)-len(parent)-1] == '.' 1030 } 1031 1032 func stripPassword(u *url.URL) string { 1033 _, passSet := u.User.Password() 1034 if passSet { 1035 return strings.Replace(u.String(), u.User.String()+"@", u.User.Username()+":***@", 1) 1036 } 1037 return u.String() 1038 }