使用winrm-cli在Golang中远程执行Windows命令

winrm-cli是一个Golang库，允许通过Windows远程管理(WinRM)协议在远程Windows机器上执行命令。下面我将详细介绍如何使用这个库。

安装winrm-cli

首先需要安装winrm-cli库：

go get github.com/masterzen/winrm-cli

基本用法

1. 创建WinRM客户端

package main

import (
	"fmt"
	"os"
	
	"github.com/masterzen/winrm"
)

func main() {
	// 配置WinRM客户端参数
	endpoint := "https://192.168.1.100:5986/wsman" // WinRM端点
	username := "administrator"                      // 管理员用户名
	password := "yourpassword"                      // 密码
	
	// 创建WinRM客户端
	client, err := winrm.NewClient(
		winrm.NewEndpoint(
			endpoint, // 端点URL
			5986,     // 端口
			false,    // 是否使用HTTPS
			false,    // 不验证TLS证书
			nil,      // CA证书
			nil,      // 客户端证书
			nil,      // 客户端私钥
			0,        // 超时时间(0表示默认)
		),
		username,
		password,
	)
	if err != nil {
		fmt.Printf("创建WinRM客户端失败: %v\n", err)
		os.Exit(1)
	}
	
	// 测试连接
	_, err = client.Run("echo Hello, WinRM!", os.Stdout, os.Stderr)
	if err != nil {
		fmt.Printf("连接测试失败: %v\n", err)
		os.Exit(1)
	}
	
	fmt.Println("连接成功!")
}

2. 执行远程命令

func executeRemoteCommand(client *winrm.Client, command string) (string, error) {
	var stdout, stderr bytes.Buffer
	exitCode, err := client.Run(command, &stdout, &stderr)
	
	if err != nil {
		return "", fmt.Errorf("执行命令失败: %v, 退出码: %d, 错误输出: %s", 
			err, exitCode, stderr.String())
	}
	
	if stderr.Len() > 0 {
		return stdout.String(), fmt.Errorf("命令执行有错误输出: %s", stderr.String())
	}
	
	return stdout.String(), nil
}

// 使用示例
output, err := executeRemoteCommand(client, "ipconfig")
if err != nil {
    fmt.Printf("错误: %v\n", err)
} else {
    fmt.Printf("命令输出:\n%s\n", output)
}

3. 执行PowerShell脚本

func executePowerShell(client *winrm.Client, script string) (string, error) {
	// 将脚本转换为Base64编码，避免特殊字符问题
	encodedScript := base64.StdEncoding.EncodeToString([]byte(script))
	command := fmt.Sprintf("powershell -encodedcommand %s", encodedScript)
	
	return executeRemoteCommand(client, command)
}

// 使用示例
psScript := `Get-Process | Where-Object { $_.CPU -gt 100 } | Select-Object Name, CPU`
output, err := executePowerShell(client, psScript)
if err != nil {
    fmt.Printf("执行PowerShell失败: %v\n", err)
} else {
    fmt.Printf("PowerShell输出:\n%s\n", output)
}

高级功能

1. 文件传输

func uploadFile(client *winrm.Client, localPath, remotePath string) error {
	content, err := os.ReadFile(localPath)
	if err != nil {
		return fmt.Errorf("读取本地文件失败: %v", err)
	}
	
	// 将文件内容转换为Base64
	encodedContent := base64.StdEncoding.EncodeToString(content)
	
	// 创建PowerShell脚本解码并保存文件
	script := fmt.Sprintf(`
		$content = "%s"
		$bytes = [Convert]::FromBase64String($content)
		[IO.File]::WriteAllBytes("%s", $bytes)
	`, encodedContent, remotePath)
	
	_, err = executePowerShell(client, script)
	return err
}

// 使用示例
err := uploadFile(client, "localfile.txt", "C:\\temp\\remotefile.txt")
if err != nil {
    fmt.Printf("文件上传失败: %v\n", err)
}

2. 长时间运行命令

func runLongCommand(client *winrm.Client, command string) error {
	shell, err := client.CreateShell()
	if err != nil {
		return fmt.Errorf("创建Shell失败: %v", err)
	}
	defer shell.Close()
	
	cmd, err := shell.Execute(command)
	if err != nil {
		return fmt.Errorf("执行命令失败: %v", err)
	}
	defer cmd.Close()
	
	var stdout, stderr bytes.Buffer
	go io.Copy(&stdout, cmd.Stdout)
	go io.Copy(&stderr, cmd.Stderr)
	
	cmd.Wait()
	
	if stderr.Len() > 0 {
		return fmt.Errorf("命令执行有错误输出: %s", stderr.String())
	}
	
	fmt.Printf("命令输出:\n%s\n", stdout.String())
	return nil
}

安全注意事项

1. 使用HTTPS：在生产环境中始终使用HTTPS而不是HTTP
2. 证书验证：应该验证服务器证书，不要跳过验证
3. 凭据安全：不要将凭据硬编码在代码中，使用环境变量或配置管理工具
4. 最小权限：使用具有最小必要权限的账户

完整示例

package main

import (
	"bytes"
	"encoding/base64"
	"fmt"
	"os"
	
	"github.com/masterzen/winrm"
)

func main() {
	// 从环境变量获取凭据
	endpoint := os.Getenv("WINRM_ENDPOINT")
	username := os.Getenv("WINRM_USERNAME")
	password := os.Getenv("WINRM_PASSWORD")
	
	if endpoint == "" || username == "" || password == "" {
		fmt.Println("请设置WINRM_ENDPOINT, WINRM_USERNAME和WINRM_PASSWORD环境变量")
		os.Exit(1)
	}
	
	// 创建WinRM客户端
	client, err := winrm.NewClient(
		winrm.NewEndpoint(
			endpoint,
			5986,
			true,  // 使用HTTPS
			false, // 不验证证书(生产环境应该为true)
			nil, nil, nil, 0),
		username,
		password,
	)
	if err != nil {
		fmt.Printf("创建WinRM客户端失败: %v\n", err)
		os.Exit(1)
	}
	
	// 执行系统信息命令
	output, err := executeRemoteCommand(client, "systeminfo")
	if err != nil {
		fmt.Printf("获取系统信息失败: %v\n", err)
	} else {
		fmt.Printf("系统信息:\n%s\n", output)
	}
	
	// 执行PowerShell命令
	psOutput, err := executePowerShell(client, "Get-Service | Where-Object { $_.Status -eq 'Running' }")
	if err != nil {
		fmt.Printf("执行PowerShell失败: %v\n", err)
	} else {
		fmt.Printf("运行中的服务:\n%s\n", psOutput)
	}
}

// executeRemoteCommand 和 executePowerShell 函数同上文示例

winrm-cli库提供了强大的功能来远程管理Windows服务器。通过合理使用，可以构建自动化运维工具、部署脚本等。记得始终遵循安全最佳实践来保护你的远程管理连接。