@
2Nfree 用这个办法可以输入 sudo 的密码,之前写过一个模仿 ansiable 的程序是这么做的,里面部分代码:
package utils
import (
"batch-execution-tool/core"
"bytes"
"fmt"
"
golang.org/x/crypto/ssh"
"os"
"sync"
)
func ExecuteScripts() {
var wg sync.WaitGroup
defer wg.Wait()
for _, ip := range core.Config.ServersIP {
wg.Add(1)
go func(ip string) {
defer wg.Done()
SSHClient, errSSH := createSSHConnect(ip)
if errSSH != nil {
core.Logger.Sugar().Errorf("创建 SSH 链接失败:\n%v", errSSH)
return
}
for _, script := range core.Config.Scripts {
for name := range script {
err := exec(SSHClient, ip, name)
if err != nil {
core.Logger.Sugar().Errorf("在 IP 地址为:%v 的服务器,出现错误:\n%v", ip, err)
} else {
core.Logger.Sugar().Debugf("在 IP 地址为:%v 的服务器,执行脚本 %v 成功", ip, name)
}
}
}
}(ip)
}
}
func exec(client *ssh.Client, ip, script string) error {
core.Logger.Sugar().Debugf("正在 IP 地址为:%v 的服务器中执行脚本:%s", ip, script)
session, err := client.NewSession()
if err != nil {
return err
}
var output bytes.Buffer
session.Stderr = &output
remoteDir := "scripts"
if core.Config.Auth.Username == "root" {
if err := session.Start("bash " + remoteDir + "/" + script + " > " + remoteDir + "/" + script + ".log"); err != nil {
return fmt.Errorf("执行脚本 %s 失败: \n%s:%s", script, err, output.String())
}
if err := session.Wait(); err != nil {
return fmt.Errorf("执行脚本 %s 失败: \n%s:%s", script, err, output.String())
}
} else {
if err := session.RequestPty("xterm", 80, 40, ssh.TerminalModes{}); err != nil {
return fmt.Errorf("请求 TTY 出错:%v ,\nos.Stderr:%v", err, os.Stderr)
}
stdin, err := session.StdinPipe()
defer stdin.Close()
if err != nil {
return fmt.Errorf("创建标准管道输出错误: \n%v", err)
}
if err := session.Start("sudo bash " + remoteDir + "/" + script + " > " + remoteDir + "/" + script + ".log"); err != nil {
return fmt.Errorf("执行脚本 %s 失败: \n%s:%s", script, err, output.String())
}
_, err = fmt.Fprintln(stdin, core.Config.Auth.Password)
if err != nil {
return fmt.Errorf("使用标准管道输如密码错误: \n%v", err)
}
if err := session.Wait(); err != nil {
return fmt.Errorf("执行脚本 %s 失败: \n%s:%s", script, err, output.String())
}
}
session.Close()
return nil
}