fix memory leak

main.go

@@ -29,7 +29,7 @@ var (
 )
 
 var (
-    headerReg, bodyReg string
+	headerReg, bodyReg string
 )
 
 var (
@@ -162,12 +162,12 @@ func main() {
 
 	var ipSource <-chan net.IP
 	if ipList != "" {
-        list, err := utils.LoadInput(ipList)
-        if err != nil {
+		list, err := utils.LoadInput(ipList)
+		if err != nil {
 			utils.EPrintln("[E] IP list", err)
 			return
-        }
-        ipSource = generators.RandomHostsFromListGen(list)
+		}
+		ipSource = generators.RandomHostsFromListGen(list)
 	} else if cidrNetwork == "" {
 		ipGenerator := generators.NewIPGenerator(4, randomIPsCount)
 		ipSource = ipGenerator.Generate()

services/http.go

@@ -40,6 +40,7 @@ func NewHTTPService(ports []int, connTimeout time.Duration, paths []string, head
 	transport := http.DefaultTransport.(*http.Transport).Clone()
 	transport.TLSClientConfig.InsecureSkipVerify = true
 	transport.DialContext = network.DialContextFunc(svc.connTimeout)
+	transport.DisableKeepAlives = true // less memory leak, less errors
 
 	svc.client = &http.Client{
 		Transport: transport,
@@ -66,37 +67,37 @@ func NewHTTPService(ports []int, connTimeout time.Duration, paths []string, head
 }
 
 func (s *HTTPService) check(uri url.URL) (bool, error) {
-	r, err := s.client.Get(uri.String())
+	response, err := s.client.Get(uri.String())
 	if err != nil {
 		return false, err
 	}
 
-	defer r.Body.Close()
+	defer response.Body.Close()
 
-	if r.StatusCode > 400 {
+	if response.StatusCode > 400 {
 		return false, nil
 	}
 
-	for k, values := range r.Header {
+	for k, values := range response.Header {
 		for _, v := range values {
 			if s.headerReg != nil && s.headerReg.MatchString(k+": "+v) {
 				return true, nil
 			}
 		}
 	}
 
-	if r.ContentLength == -1 || r.ContentLength > 1024*1024 {
+	if response.ContentLength == -1 || response.ContentLength > 1024*1024 {
 		return false, nil
 	}
 
-	reader := io.LimitReader(r.Body, 1024*1024)
-	b, err := io.ReadAll(reader)
+	reader := io.LimitReader(response.Body, 1024*1024)
+	body, err := io.ReadAll(reader)
 
 	if err != nil {
 		return false, nil
 	}
 
-	if s.bodyReg != nil && s.bodyReg.Match(b) {
+	if s.bodyReg != nil && s.bodyReg.Match(body) {
 		return true, nil
 	}