From 205021afecbe261b4f2ae088ea1baba1c88a4fb9 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=BD=B1=E8=88=9E=E8=80=85?= <shadow1ng@qq.com>
Date: Fri, 7 Jan 2022 16:54:23 +0800
Subject: [PATCH] update http

---
 WebScan/lib/client.go | 2 +-
 WebScan/lib/eval.go   | 3 +--
 2 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/WebScan/lib/client.go b/WebScan/lib/client.go
index 87e5de0..5906638 100644
--- a/WebScan/lib/client.go
+++ b/WebScan/lib/client.go
@@ -40,7 +40,7 @@ func InitHttpClient(ThreadsNum int, DownProxy string, Timeout time.Duration) err
 		IdleConnTimeout:     keepAlive,
 		TLSClientConfig:     &tls.Config{InsecureSkipVerify: true},
 		TLSHandshakeTimeout: 5 * time.Second,
-		DisableKeepAlives:   false,
+		DisableKeepAlives:   true,
 	}
 	if DownProxy != "" {
 		if DownProxy == "1" {
diff --git a/WebScan/lib/eval.go b/WebScan/lib/eval.go
index aff9d9e..eb056c1 100644
--- a/WebScan/lib/eval.go
+++ b/WebScan/lib/eval.go
@@ -591,9 +591,8 @@ func getRespBody(oResp *http.Response) ([]byte, error) {
 			body = append(body, buf...)
 		}
 	} else {
-		raw, err := ioutil.ReadAll(oResp.Body)
+		raw, err := ioutil.ReadAll(io.LimitReader(oResp.Body, int64(3<<20)))
 		if err != nil {
-			//utils.Logger.Error(err)
 			return nil, err
 		}
 		defer oResp.Body.Close()