do not allow auto redirect to http client

Author: buzztaiki

aws-es-proxy.go

@@ -52,6 +52,7 @@ type proxy struct {
 	fileRequest  *os.File
 	fileResponse *os.File
 	credentials  *credentials.Credentials
+	client       *http.Client
 }
 
 func newProxy(args ...interface{}) *proxy {
@@ -64,6 +65,14 @@ func newProxy(args ...interface{}) *proxy {
 	}
 }
 
+var client = &http.Client{
+	CheckRedirect: noRedirect,
+}
+
+func noRedirect(req *http.Request, via []*http.Request) error {
+	return http.ErrUseLastResponse
+}
+
 func (p *proxy) parseEndpoint() error {
 	var link *url.URL
 	var err error
@@ -149,7 +158,7 @@ func (p *proxy) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 		signer.Sign(req, payload, p.service, p.region, time.Now())
 	}
 
-	resp, err := http.DefaultClient.Do(req)
+	resp, err := client.Do(req)
 	if err != nil {
 		log.Fatalln(err.Error())
 		http.Error(w, err.Error(), http.StatusBadRequest)