diff --git a/vendor/github.com/packer-community/winrmcp/winrmcp/winrmcp.go b/vendor/github.com/packer-community/winrmcp/winrmcp/winrmcp.go index 844c5a2f6..d41061b4b 100644 --- a/vendor/github.com/packer-community/winrmcp/winrmcp/winrmcp.go +++ b/vendor/github.com/packer-community/winrmcp/winrmcp/winrmcp.go @@ -4,6 +4,7 @@ import ( "errors" "fmt" "io" + "net/http" "os" "path/filepath" "time" @@ -24,6 +25,7 @@ type Config struct { CACertBytes []byte OperationTimeout time.Duration MaxOperationsPerShell int + TransportDecorator func(*http.Transport) http.RoundTripper } type Auth struct { @@ -41,6 +43,10 @@ func New(addr string, config *Config) (*Winrmcp, error) { } params := winrm.DefaultParameters() + if config.TransportDecorator != nil { + params.TransportDecorator = config.TransportDecorator + } + if config.OperationTimeout.Seconds() > 0 { params.Timeout = iso8601.FormatDuration(config.OperationTimeout) }