feat: Limit read http response size (#903)

jeessy2 · Nov 4, 2023 · 857f2a6 · 857f2a6
1 parent b0c050c
commit 857f2a6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/config/config.go b/config/config.go
@@ -193,7 +193,8 @@ func (conf *DnsConfig) getIpv4AddrFromUrl() string {
 			continue
 		}
 		defer resp.Body.Close()
-		body, err := io.ReadAll(resp.Body)
+		lr := io.LimitReader(resp.Body, 1024000)
+		body, err := io.ReadAll(lr)
 		if err != nil {
 			log.Println("读取IPv4结果失败! 接口: ", url)
 			continue
@@ -327,7 +328,8 @@ func (conf *DnsConfig) getIpv6AddrFromUrl() string {
 		}
 
 		defer resp.Body.Close()
-		body, err := io.ReadAll(resp.Body)
+		lr := io.LimitReader(resp.Body, 1024000)
+		body, err := io.ReadAll(lr)
 		if err != nil {
 			log.Println("读取IPv6结果失败! 接口: ", url)
 			continue

diff --git a/util/http_util.go b/util/http_util.go
@@ -34,7 +34,8 @@ func GetHTTPResponseOrg(resp *http.Response, url string, err error) ([]byte, err
 	}
 
 	defer resp.Body.Close()
-	body, err := io.ReadAll(resp.Body)
+	lr := io.LimitReader(resp.Body, 1024000)
+	body, err := io.ReadAll(lr)
 
 	if err != nil {
 		log.Printf("请求接口%s失败! ERROR: %s\n", url, err)