105 lines
2.7 KiB
Go
105 lines
2.7 KiB
Go
package main
|
|
|
|
import (
|
|
"errors"
|
|
"flag"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/packer-community/winrmcp/winrmcp"
|
|
)
|
|
|
|
var usage = `
|
|
Usage: winrmcp [options] [-help | <from> <to>]
|
|
|
|
Copy a local file or directory to a remote directory.
|
|
|
|
Options:
|
|
|
|
-user Name of the user to authenticate as
|
|
-pass Password to authenticate with
|
|
-addr=localhost:5985 Host and port of the remote machine
|
|
-https Use HTTPS in preference to HTTP
|
|
-insecure Do not validate the HTTPS certificate chain
|
|
-cacert Filename of CA cert to validate against
|
|
-tlsservername Server name to validate against when using https
|
|
-op-timeout=60s Timeout duration of each WinRM operation
|
|
-max-ops-per-shell=15 Max number of operations per WinRM shell
|
|
|
|
`
|
|
|
|
func main() {
|
|
if hasSwitch("-help") {
|
|
fmt.Print(usage)
|
|
os.Exit(0)
|
|
}
|
|
if err := runMain(); err != nil {
|
|
fmt.Println(err.Error())
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
func runMain() error {
|
|
flags := flag.NewFlagSet("cli", flag.ContinueOnError)
|
|
flags.Usage = func() { fmt.Print(usage) }
|
|
addr := flags.String("addr", "localhost:5985", "winrm remote host:port")
|
|
user := flags.String("user", "", "winrm admin username")
|
|
pass := flags.String("pass", "", "winrm admin password")
|
|
https := flags.Bool("https", false, "use https instead of http")
|
|
insecure := flags.Bool("insecure", false, "do not validate https certificate chain")
|
|
tlsservername := flags.String("tlsservername", "", "server name to validate against when using https")
|
|
cacert := flags.String("cacert", "", "ca certificate to validate against")
|
|
opTimeout := flags.Duration("op-timeout", time.Second*60, "operation timeout")
|
|
maxOpsPerShell := flags.Int("max-ops-per-shell", 15, "max operations per shell")
|
|
flags.Parse(os.Args[1:])
|
|
|
|
var certBytes []byte
|
|
var err error
|
|
if *cacert != "" {
|
|
certBytes, err = ioutil.ReadFile(*cacert)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
os.Exit(1)
|
|
}
|
|
} else {
|
|
certBytes = nil
|
|
}
|
|
|
|
client, err := winrmcp.New(*addr, &winrmcp.Config{
|
|
Auth: winrmcp.Auth{User: *user, Password: *pass},
|
|
Https: *https,
|
|
Insecure: *insecure,
|
|
TLSServerName: *tlsservername,
|
|
CACertBytes: certBytes,
|
|
OperationTimeout: *opTimeout,
|
|
MaxOperationsPerShell: *maxOpsPerShell,
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
args := flags.Args()
|
|
if len(args) < 1 {
|
|
return errors.New("Source directory is required.")
|
|
}
|
|
if len(args) < 2 {
|
|
return errors.New("Remote directory is required.")
|
|
}
|
|
if len(args) > 2 {
|
|
return errors.New("Too many arguments.")
|
|
}
|
|
|
|
return client.Copy(args[0], args[1])
|
|
}
|
|
|
|
func hasSwitch(name string) bool {
|
|
for _, arg := range os.Args[1:] {
|
|
if arg == name {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|