Merge pull request #34 from sneal/stop-on-connection-timeout

Issue: #27
cloudfoundry · Jun 27, 2024 · 38e21bb · 38e21bb
2 parents 72bb051 + 724200c
commit 38e21bb
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/remotemanager/winrm_clientfactory.go b/remotemanager/winrm_clientfactory.go
@@ -18,6 +18,12 @@ func NewWinRmClientFactory(host, username, password string) *WinRMClientFactory
 
 func (f *WinRMClientFactory) Build(timeout time.Duration) (WinRMClient, error) {
 	endpoint := winrm.NewEndpoint(f.host, WinRmPort, false, true, nil, nil, nil, timeout)
-	client, err := winrm.NewClient(endpoint, f.username, f.password)
+	params := winrm.NewParameters(
+		winrm.DefaultParameters.Timeout,
+		winrm.DefaultParameters.Locale,
+		winrm.DefaultParameters.EnvelopeSize,
+	)
+	params.AllowTimeout = true
+	client, err := winrm.NewClientWithParameters(endpoint, f.username, f.password, params)
 	return client, err
 }