added protocol choice

Author: utkusen

README.md

@@ -42,7 +42,7 @@ You can download the pre-built binaries from the [releases](https://github.com/u
 
 # Usage
 
-reqstress requires 3 parameters to run: 
+reqstress requires 4 parameters to run: 
 
 `-r` : Path of the request file. For example: `-r request.txt`. Request file should contain a raw HTTP request. For example:
 
@@ -68,4 +68,6 @@ log=admin&pwd=asdadsasdads
 
 `-w` : The number of workers to run. The default value is 500. You can increase or decrease this by testing out the capability of your system.
 
-`-d` : Duration of the test (in seconds). Default is infinite.
+`-d` : Duration of the test (in seconds). Default is infinite.
+
+`-p` : Protocol to attack. Can be http or https. Default is https.

main.go

@@ -20,6 +20,7 @@ var average_time time.Duration
 var numOfSuccess int
 var numOfFail int
 var numOfnon200 int
+var urlStr string
 
 func sendRequest(wg *sync.WaitGroup, req *fasthttp.Request, resp *fasthttp.Response, client *fasthttp.Client) {
 	defer wg.Done()
@@ -47,6 +48,7 @@ func main() {
 	requestFile := flag.String("r", "", "Path of request file")
 	numWorker := flag.Int("w", 500, "Number of worker. Default: 500")
 	duration := flag.Int("d", 0, "Test duration. Default: infinite")
+	protocol := flag.String("p", "https", "Protol to attack. http or https.")
 	flag.Parse()
 	if *requestFile == "" {
 		fmt.Println("Please specify all arguments!")
@@ -63,7 +65,12 @@ func main() {
 		panic(err)
 	}
 	var wg sync.WaitGroup
-	urlStr := "http://" + httpRequest.Host + httpRequest.RequestURI
+	if *protocol == "http" {
+		urlStr = "http://" + httpRequest.Host + httpRequest.RequestURI
+	} else {
+		urlStr = "https://" + httpRequest.Host + httpRequest.RequestURI
+	}
+
 	bodyBytes, _ := ioutil.ReadAll(httpRequest.Body)
 	bodyString := string(bodyBytes)
 	req := fasthttp.AcquireRequest()