diff --git a/client/rest/http.go b/client/rest/http.go index cc729fe7..52e8aba9 100644 --- a/client/rest/http.go +++ b/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 } } diff --git a/config/config.go b/config/config.go index dae48594..1ef1efd1 100644 --- a/config/config.go +++ b/config/config.go @@ -355,6 +355,14 @@ var ( Default: "", } + UserAgent = Config{ + Name: "user-agent", + Shorthand: "U", + Usage: "Custom User-Agent header", + Persistent: true, + Default: "", + } + GlobalConfig = []Config{ ConfigFile, VerbosityLevel, @@ -364,6 +372,7 @@ var ( Proxy, RefreshToken, Pprof, + UserAgent, } AzureConfig = []Config{