From 91cfa42579db86715c5e9ab745eb9c4afb6ca50f Mon Sep 17 00:00:00 2001
From: Leeon123 <leeon123ex@gmail.com>
Date: Sun, 2 Aug 2020 15:18:14 +0800
Subject: [PATCH] Fix

---
 httpflood.go | 56 ++++++++++++++++++++++++++++------------------------
 1 file changed, 30 insertions(+), 26 deletions(-)

diff --git a/httpflood.go b/httpflood.go
index efb5fe1..e562a6f 100644
--- a/httpflood.go
+++ b/httpflood.go
@@ -130,38 +130,42 @@ func flood(ip, port, page, mode string) {
 			header += acceptall[rand.Intn(len(acceptall))]
 			header += referers[rand.Intn(len(referers))] + "\r\n"
 		} else {
-			fi, err := os.Open(os.Args[7])
-			if err != nil {
-				fmt.Printf("Error: %s\n", err)
-				return
-			}
-			defer fi.Close()
-			br := bufio.NewReader(fi)
-			for {
-				a, _, c := br.ReadLine()
-				if c == io.EOF {
-					break
+			func() {
+				fi, err := os.Open(os.Args[7])
+				if err != nil {
+					fmt.Printf("Error: %s\n", err)
+					return
 				}
-				header += string(a) + "\r\n"
-			}
+				defer fi.Close()
+				br := bufio.NewReader(fi)
+				for {
+					a, _, c := br.ReadLine()
+					if c == io.EOF {
+						break
+					}
+					header += string(a) + "\r\n"
+				}
+			}()
 		}
 	} else if mode == "post" {
 		data := ""
 		if os.Args[7] != "nil" {
-			fi, err := os.Open(os.Args[7])
-			if err != nil {
-				fmt.Printf("Error: %s\n", err)
-				return
-			}
-			defer fi.Close()
-			br := bufio.NewReader(fi)
-			for {
-				a, _, c := br.ReadLine()
-				if c == io.EOF {
-					break
+			func() {
+				fi, err := os.Open(os.Args[7])
+				if err != nil {
+					fmt.Printf("Error: %s\n", err)
+					return
 				}
-				data += string(a) + "\r\n"
-			}
+				defer fi.Close()
+				br := bufio.NewReader(fi)
+				for {
+					a, _, c := br.ReadLine()
+					if c == io.EOF {
+						break
+					}
+					header += string(a) + "\r\n"
+				}
+			}()
 
 		} else {
 			for x := 0; x > 100; x++ {