From b3805999f357843cd138f5fab3719795fb880910 Mon Sep 17 00:00:00 2001
From: LuitelSamikshya <85764322+LuitelSamikshya@users.noreply.github.com>
Date: Sat, 27 Nov 2021 17:10:27 -0600
Subject: [PATCH] Unsafe flag with base template (#1279)

* unsafe flag with base template
---
 v2/pkg/protocols/http/build_request.go | 6 ++++--
 v2/pkg/protocols/http/request.go       | 3 +--
 2 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/v2/pkg/protocols/http/build_request.go b/v2/pkg/protocols/http/build_request.go
index 8c8a2dbc..be1d1311 100644
--- a/v2/pkg/protocols/http/build_request.go
+++ b/v2/pkg/protocols/http/build_request.go
@@ -307,10 +307,12 @@ func (r *requestGenerator) fillRequest(req *http.Request, values map[string]inte
 		}
 		req.Body = ioutil.NopCloser(strings.NewReader(body))
 	}
-	setHeader(req, "User-Agent", uarand.GetRandom())
+	if !r.request.Unsafe {
+		setHeader(req, "User-Agent", uarand.GetRandom())
+	}
 
 	// Only set these headers on non-raw requests
-	if len(r.request.Raw) == 0 {
+	if len(r.request.Raw) == 0 && !r.request.Unsafe {
 		setHeader(req, "Accept", "*/*")
 		setHeader(req, "Accept-Language", "en")
 	}
diff --git a/v2/pkg/protocols/http/request.go b/v2/pkg/protocols/http/request.go
index a0bce62e..a38b56b9 100644
--- a/v2/pkg/protocols/http/request.go
+++ b/v2/pkg/protocols/http/request.go
@@ -310,9 +310,8 @@ func (request *Request) executeRequest(reqURL string, generatedRequest *generate
 		if request.options.Options.Debug || request.options.Options.DebugRequests {
 			gologger.Info().Msgf("[%s] Dumped HTTP request for %s\n\n", request.options.TemplateID, reqURL)
 			gologger.Print().Msgf("%s", dumpedRequestString)
-		}
+		} 
 	}
-
 	var formedURL string
 	var hostname string
 	timeStart := time.Now()