[golang]-使用私钥登录服务器执行命令

最新推荐文章于 2024-05-23 10:05:02 发布

爷来辣

最新推荐文章于 2024-05-23 10:05:02 发布

阅读量397

点赞数

分类专栏： golang

仅供学习参考转载请标明出处。个人笔记为自己记录如果对大家有帮助那也挺好写的马虎如果有不对的地方谢谢指正各位大佬别喷哦

本文链接：https://blog.csdn.net/xujiamin0022016/article/details/128095189

版权

SSH 私钥 Golang 自动化运维远程命令执行

关键词由CSDN通过智能技术生成

golang 专栏收录该内容

63 篇文章 2 订阅

订阅专栏

导语：需要通过ssh私钥登录到服务器上执行命令并返回结果

直接上代码

package main

import (
	"fmt"
	"io/ioutil"
	"log"
	"time"

	"github.com/mitchellh/go-homedir"
	"golang.org/x/crypto/ssh"
)

func main() {
	sshHost := "192.168.10.82"
	sshUser := "root"
	sshPassword := ""
	sshType := "key"         //password 或者 key
	sshKeyPath := "./id_rsa" //ssh id_rsa.id 路径"
	sshPort := 22
	//if sshUser == "" {  // 设置默认用户
	//	sshUser = "root"
	//}
	//
	//if sshKeyPath == "" { // 设置默认私钥路径
	//	userHome, _ := os.UserHomeDir()
	//	sshKeyPath = fmt.Sprintf("%s/.ssh/id_rsa", userHome)
	//}

	//创建sshp登陆配置
	config := &ssh.ClientConfig{
		Timeout:         time.Second, //ssh 连接time out 时间一秒钟, 如果ssh验证错误 会在一秒内返回
		User:            sshUser,
		HostKeyCallback: ssh.InsecureIgnoreHostKey(), //这个可以， 但是不够安全
		//HostKeyCallback: hostKeyCallBackFunc(h.Host),
	}
	if sshType == "password" {
		config.Auth = []ssh.AuthMethod{ssh.Password(sshPassword)}
	} else {
		config.Auth = []ssh.AuthMethod{publicKeyAuthFunc(sshKeyPath)}
	}

	//dial 获取ssh client
	addr := fmt.Sprintf("%s:%d", sshHost, sshPort)
	sshClient, err := ssh.Dial("tcp", addr, config)
	if err != nil {
		log.Fatal("创建ssh client 失败", err)
	}
	defer sshClient.Close()

	//创建ssh-session
	session, err := sshClient.NewSession()
	if err != nil {
		log.Fatal("创建ssh session 失败", err)
	}
	defer session.Close()
	//执行远程命令
	//combo, err := session.CombinedOutput("whoami;cd / ; ls -al;echo https://github.com/dejavuzhou/felix")  // ls -al 可能返回会卡住
	combo, err := session.CombinedOutput("whoami;pwd")
	if err != nil {
		log.Fatal("远程执行cmd 失败", err)
	}
	log.Println("命令输出:", string(combo))

}

func publicKeyAuthFunc(kPath string) ssh.AuthMethod {
	keyPath, err := homedir.Expand(kPath)
	if err != nil {
		log.Fatal("find key's home dir failed", err)
	}
	key, err := ioutil.ReadFile(keyPath)
	if err != nil {
		log.Fatal("ssh key file read failed", err)
	}
	// Create the Signer for this private key.
	signer, err := ssh.ParsePrivateKey(key)
	if err != nil {
		log.Fatal("ssh key signer failed", err)
	}
	return ssh.PublicKeys(signer)
}