iSharkFly-Docs/packer-cn
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Updated vendor pacakge for winrmcp

Browse Source
This commit is contained in:
Chris Bednarski 2016-05-06 22:57:29 -07:00
parent da4491f65e
commit e38c1122ab
3 changed files with 23 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Godeps/Godeps.json generated
View File

@ -1,7 +1,7 @@
{ {
"ImportPath": "github.com/mitchellh/packer", "ImportPath": "github.com/mitchellh/packer",
"GoVersion": "go1.6", "GoVersion": "go1.6",
"GodepVersion": "v62", "GodepVersion": "v63",
"Deps": [ "Deps": [
{ {
"ImportPath": "github.com/ActiveState/tail", "ImportPath": "github.com/ActiveState/tail",

20
vendor/github.com/packer-community/winrmcp/winrmcp/cp.go generated vendored
View File

@ -10,18 +10,21 @@ import (
"sync" "sync"
"github.com/masterzen/winrm/winrm" "github.com/masterzen/winrm/winrm"
"github.com/mitchellh/packer/common/uuid" "github.com/nu7hatch/gouuid"
) )
func doCopy(client *winrm.Client, config *Config, in io.Reader, toPath string) error { func doCopy(client *winrm.Client, config *Config, in io.Reader, toPath string) error {
tempFile := fmt.Sprintf("winrmcp-%s.tmp", uuid.TimeOrderedUUID()) tempFile, err := tempFileName()
if err != nil {
return errors.New(fmt.Sprintf("Error generating unique filename: %v", err))
}
tempPath := "$env:TEMP\\" + tempFile tempPath := "$env:TEMP\\" + tempFile
if os.Getenv("WINRMCP_DEBUG") != "" { if os.Getenv("WINRMCP_DEBUG") != "" {
log.Printf("Copying file to %s\n", tempPath) log.Printf("Copying file to %s\n", tempPath)
} }
err := uploadContent(client, config.MaxOperationsPerShell, "%TEMP%\\"+tempFile, in) err = uploadContent(client, config.MaxOperationsPerShell, "%TEMP%\\"+tempFile, in)
if err != nil { if err != nil {
return errors.New(fmt.Sprintf("Error uploading file to %s: %v", tempPath, err)) return errors.New(fmt.Sprintf("Error uploading file to %s: %v", tempPath, err))
} }
@ -113,7 +116,7 @@ func restoreContent(client *winrm.Client, fromPath, toPath string) error {
defer shell.Close() defer shell.Close()
script := fmt.Sprintf(` script := fmt.Sprintf(`
$tmp_file_path = [System.IO.Path]::GetFullPath("%s") $tmp_file_path = [System.IO.Path]::GetFullPath("%s")
$dest_file_path = [System.IO.Path]::GetFullPath("%s") $dest_file_path = [System.IO.Path]::GetFullPath("%s".Trim("'"))
if (Test-Path $dest_file_path) { if (Test-Path $dest_file_path) {
rm $dest_file_path rm $dest_file_path
} }
@ -198,3 +201,12 @@ func appendContent(shell *winrm.Shell, filePath, content string) error {
return nil return nil
} }
func tempFileName() (string, error) {
uniquePart, err := uuid.NewV4()
if err != nil {
return "", err
}
return fmt.Sprintf("winrmcp-%s.tmp", uniquePart), nil
}

7
vendor/github.com/packer-community/winrmcp/winrmcp/winrmcp.go generated vendored
View File

@ -42,7 +42,12 @@ func New(addr string, config *Config) (*Winrmcp, error) {
config = &Config{} config = &Config{}
} }
params := winrm.DefaultParameters() params := winrm.NewParameters(
winrm.DefaultParameters.Timeout,
winrm.DefaultParameters.Locale,
winrm.DefaultParameters.EnvelopeSize,
)
if config.TransportDecorator != nil { if config.TransportDecorator != nil {
params.TransportDecorator = config.TransportDecorator params.TransportDecorator = config.TransportDecorator
} }