Merge pull request #152 from onSec-fr/main

Add a global flag to customize the User-Agent

Author: rvazarkar

client/rest/http.go

@@ -128,8 +128,13 @@ func NewRequest(
 			req.Header.Set("Accept", "application/json")
 		}
 
-		// set azurehound as user-agent
-		req.Header.Set("User-Agent", constants.UserAgent())
+		// set azurehound as user-agent, use custom if set in config
+		ua := config.UserAgent.Value()
+		if s, ok := ua.(string); ok && s != "" {
+			req.Header.Set("User-Agent", s)
+		} else {
+			req.Header.Set("User-Agent", constants.UserAgent())
+		}
 		return req, nil
 	}
 }

config/config.go

@@ -354,6 +354,14 @@ var (
 		Default:    "",
 	}
 
+	UserAgent = Config{
+		Name:       "user-agent",
+		Shorthand:  "U",
+		Usage:      "Custom User-Agent header",
+		Persistent: true,
+		Default:    "",
+	}
+
 	GlobalConfig = []Config{
 		ConfigFile,
 		VerbosityLevel,
@@ -363,6 +371,7 @@ var (
 		Proxy,
 		RefreshToken,
 		Pprof,
+		UserAgent,
 	}
 
 	AzureConfig = []Config{