added default headers option to shared library payload;

bogdanfinn · Sep 3, 2023 · a2937f0 · a2937f0
1 parent a82dd73
commit a2937f0
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 0 deletions.
diff --git a/cffi_src/factory.go b/cffi_src/factory.go
@@ -360,6 +360,10 @@ func getTlsClient(requestInput RequestInput, sessionId string, withSession bool)
 		options = append(options, tls_client.WithInsecureSkipVerify())
 	}
 
+	if requestInput.DefaultHeaders != nil && len(requestInput.DefaultHeaders) != 0 {
+		options = append(options, tls_client.WithDefaultHeaders(requestInput.DefaultHeaders))
+	}
+
 	if requestInput.ServerNameOverwrite != nil && *requestInput.ServerNameOverwrite != "" {
 		options = append(options, tls_client.WithServerNameOverwrite(*requestInput.ServerNameOverwrite))
 	}

diff --git a/cffi_src/types.go b/cffi_src/types.go
@@ -55,6 +55,7 @@ type RequestInput struct {
 	ForceHttp1                  bool                `json:"forceHttp1"`
 	HeaderOrder                 []string            `json:"headerOrder"`
 	Headers                     map[string]string   `json:"headers"`
+	DefaultHeaders              map[string][]string `json:"defaultHeaders"`
 	InsecureSkipVerify          bool                `json:"insecureSkipVerify"`
 	IsByteRequest               bool                `json:"isByteRequest"`
 	IsByteResponse              bool                `json:"isByteResponse"`

diff --git a/client.go b/client.go
@@ -66,6 +66,7 @@ func NewHttpClient(logger Logger, options ...HttpClientOption) (HttpClient, erro
 		followRedirects:    true,
 		badPinHandler:      nil,
 		customRedirectFunc: nil,
+		defaultHeaders:     make(http.Header),
 		clientProfile:      DefaultClientProfile,
 		timeout:            time.Duration(DefaultTimeoutSeconds) * time.Second,
 	}