iuu/go-ssh-tunnel
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

更新 sshtunnel/ssh.go

Browse Source 
新增密钥登陆功能
This commit is contained in:
建超
2025-02-21 10:35:13 +00:00
parent d10b746649
commit 0a6aa35508
1 changed files with 146 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

148
sshtunnel/ssh.go
View File

@@ -46,8 +46,39 @@ func (st *SSHTunnel) GetSSHClient() (*ssh.Client, error) {
}
var auth []ssh.AuthMethod
auth = make([]ssh.AuthMethod, 0)
auth = append(auth, ssh.Password(st.config.Pass))
// 如果配置中有密钥字符串，使用密钥认证
if st.config.Key != "" {
//fmt.Println(st.config.Key)
if st.config.KeyPass != "" {
// 使用 passphrase 解密密钥字符串
passphrase := "admin" // 提供密码来解密密钥
// 解析加密的密钥字符串
signer, err := ssh.ParsePrivateKeyWithPassphrase([]byte(st.config.Key), []byte(passphrase))
if err != nil {
return nil, fmt.Errorf("解析私钥失败: %v", err)
}
auth = append(auth, ssh.PublicKeys(signer))
} else {
//signer, err := ssh.ParsePrivateKeyWithPassphrase([]byte(st.config.Key), []byte(passphrase))
//if err != nil {
// return nil, fmt.Errorf("解析私钥失败: %v", err)
//}
// 解析密钥字符串
signer, err := ssh.ParsePrivateKey([]byte(st.config.Key))
if err != nil {
return nil, fmt.Errorf("解析私钥失败: %v", err)
}
auth = append(auth, ssh.PublicKeys(signer))
}
} else {
// 否则使用密码认证
auth = append(auth, ssh.Password(st.config.Pass))
}
// 创建 SSH 客户端配置
sc := &ssh.ClientConfig{
User: st.config.User,
Auth: auth,
@@ -55,15 +86,128 @@ func (st *SSHTunnel) GetSSHClient() (*ssh.Client, error) {
return nil
},
}
// 尝试连接 SSH 服务器
var err error
st.client, err = ssh.Dial("tcp", st.config.Addr, sc)
if err != nil {
return nil, err
}
log.Printf("连接到服务器成功: %s", st.config.Addr)
return st.client, err
return st.client, nil
}
//func (st *SSHTunnel) GetSSHClient() (*ssh.Client, error) {
// if st.client != nil {
// return st.client, nil
// }
// var auth []ssh.AuthMethod
// auth = make([]ssh.AuthMethod, 0)
// auth = append(auth, ssh.Password(st.config.Pass))
//
// sc := &ssh.ClientConfig{
// User: st.config.User,
// Auth: auth,
// HostKeyCallback: func(hostname string, remote net.Addr, key ssh.PublicKey) error {
// return nil
// },
// }
// var err error
// st.client, err = ssh.Dial("tcp", st.config.Addr, sc)
// if err != nil {
// return nil, err
// }
// log.Printf("连接到服务器成功: %s", st.config.Addr)
// return st.client, err
//}
//func (st *SSHTunnel) GetSSHClient() (*ssh.Client, error) {
// if st.client != nil {
// return st.client, nil
// }
// var auth []ssh.AuthMethod
// auth = make([]ssh.AuthMethod, 0)
//
// // 如果配置中有密钥文件路径，使用密钥认证
// if st.config.Key != "" {
// key, err := ioutil.ReadFile(st.config.Key)
// if err != nil {
// return nil, fmt.Errorf("读取密钥文件失败: %v", err)
// }
//
// // 使用私钥进行认证
// signer, err := ssh.ParsePrivateKey(key)
// if err != nil {
// return nil, fmt.Errorf("解析私钥失败: %v", err)
// }
// auth = append(auth, ssh.PublicKeys(signer))
// } else {
// // 否则使用密码认证
// auth = append(auth, ssh.Password(st.config.Pass))
// }
//
// // 创建 SSH 客户端配置
// sc := &ssh.ClientConfig{
// User: st.config.User,
// Auth: auth,
// HostKeyCallback: func(hostname string, remote net.Addr, key ssh.PublicKey) error {
// return nil
// },
// }
//
// // 尝试连接 SSH 服务器
// var err error
// st.client, err = ssh.Dial("tcp", st.config.Addr, sc)
// if err != nil {
// return nil, err
// }
//
// log.Printf("连接到服务器成功: %s", st.config.Addr)
// return st.client, nil
//}
//func (st *SSHTunnel) GetSSHClient() (*ssh.Client, error) {
// if st.client != nil {
// return st.client, nil
// }
// var auth []ssh.AuthMethod
// auth = make([]ssh.AuthMethod, 0)
//
// // 如果配置中有密钥字符串，使用密钥认证
// if st.config.Key != "" {
// fmt.Println(st.config.Key)
// // 解析密钥字符串
// signer, err := ssh.ParsePrivateKey([]byte(st.config.Key))
// if err != nil {
// return nil, fmt.Errorf("解析私钥失败: %v", err)
// }
// auth = append(auth, ssh.PublicKeys(signer))
// } else {
// // 否则使用密码认证
// auth = append(auth, ssh.Password(st.config.Pass))
// }
//
// // 创建 SSH 客户端配置
// sc := &ssh.ClientConfig{
// User: st.config.User,
// Auth: auth,
// HostKeyCallback: func(hostname string, remote net.Addr, key ssh.PublicKey) error {
// return nil
// },
// }
//
// // 尝试连接 SSH 服务器
// var err error
// st.client, err = ssh.Dial("tcp", st.config.Addr, sc)
// if err != nil {
// return nil, err
// }
//
// log.Printf("连接到服务器成功: %s", st.config.Addr)
// return st.client, nil
//}
func (st *SSHTunnel) connect(t Tunnel) {
tid := fmt.Sprintf("%s-%s", t.Local, t.Remote)
ll, err := net.Listen("tcp", t.Local)