Make winrm protocol options configurable

masterzen · Jun 11, 2022 · 689d380 · 689d380
1 parent 4d91010
commit 689d380
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 4 deletions.
diff --git a/parameters.go b/parameters.go
@@ -8,6 +8,7 @@ type Parameters struct {
 	Timeout            string
 	Locale             string
 	EnvelopeSize       int
+	RequestOptions     map[string]string
 	TransportDecorator func() Transporter
 	Dial               func(network, addr string) (net.Conn, error)
 }
@@ -23,5 +24,11 @@ func NewParameters(timeout, locale string, envelopeSize int) *Parameters {
 		Timeout:      timeout,
 		Locale:       locale,
 		EnvelopeSize: envelopeSize,
+		RequestOptions: map[string]string{
+			"WINRS_NOPROFILE":         "FALSE",
+			"WINRS_CODEPAGE":          "65001",
+			"WINRS_CONSOLEMODE_STDIN": "TRUE",
+			"WINRS_SKIP_CMD_SHELL":    "FALSE",
+		},
 	}
 }
diff --git a/request.go b/request.go
@@ -33,8 +33,8 @@ func NewOpenShellRequest(uri string, params *Parameters) *soap.SoapMessage {
 	defaultHeaders(message, uri, params).
 		Action("http://schemas.xmlsoap.org/ws/2004/09/transfer/Create").
 		ResourceURI("http://schemas.microsoft.com/wbem/wsman/1/windows/shell/cmd").
-		AddOption(soap.NewHeaderOption("WINRS_NOPROFILE", "TRUE")).
-		AddOption(soap.NewHeaderOption("WINRS_CODEPAGE", "65001")).
+		AddOption(soap.NewHeaderOption("WINRS_NOPROFILE", params.RequestOptions["WINRS_NOPROFILE"])).
+		AddOption(soap.NewHeaderOption("WINRS_CODEPAGE", params.RequestOptions["WINRS_CODEPAGE"])).
 		Build()
 
 	body := message.CreateBodyElement("Shell", soap.DOM_NS_WIN_SHELL)
@@ -73,8 +73,8 @@ func NewExecuteCommandRequest(uri, shellID, command string, arguments []string,
 		Action("http://schemas.microsoft.com/wbem/wsman/1/windows/shell/Command").
 		ResourceURI("http://schemas.microsoft.com/wbem/wsman/1/windows/shell/cmd").
 		ShellId(shellID).
-		AddOption(soap.NewHeaderOption("WINRS_CONSOLEMODE_STDIN", "FALSE")).
-		AddOption(soap.NewHeaderOption("WINRS_SKIP_CMD_SHELL", "FALSE")).
+		AddOption(soap.NewHeaderOption("WINRS_CONSOLEMODE_STDIN", params.RequestOptions["WINRS_CONSOLEMODE_STDIN"])).
+		AddOption(soap.NewHeaderOption("WINRS_SKIP_CMD_SHELL", params.RequestOptions["WINRS_SKIP_CMD_SHELL"])).
 		Build()
 
 	body := message.CreateBodyElement("CommandLine", soap.DOM_NS_WIN_SHELL)