151 lines
3.1 KiB
Go
151 lines
3.1 KiB
Go
/************************************************************
|
|
** @Description: server
|
|
** @Author: george hao
|
|
** @Date: 2019-07-02 11:16
|
|
** @Last Modified by: george hao
|
|
** @Last Modified time: 2019-07-02 11:16
|
|
*************************************************************/
|
|
package libs
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/george518/PPGo_Job/common"
|
|
"github.com/george518/PPGo_Job/models"
|
|
"github.com/linxiaozhi/go-telnet"
|
|
"github.com/pkg/errors"
|
|
"golang.org/x/crypto/ssh"
|
|
"io/ioutil"
|
|
"net"
|
|
"net/rpc"
|
|
"net/rpc/jsonrpc"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
func RemoteCommandByTelnetPassword(servers *models.TaskServer) error {
|
|
|
|
addr := fmt.Sprintf("%s:%d", servers.ServerIp, servers.Port)
|
|
conn, err := gote.DialTimeout("tcp", addr, time.Second*10)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
defer conn.Close()
|
|
|
|
buf := make([]byte, 4096)
|
|
_, err = conn.Read(buf)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
_, err = conn.Write([]byte(servers.ServerAccount + "\r\n"))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
_, err = conn.Read(buf)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
_, err = conn.Write([]byte(servers.Password + "\r\n"))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
_, err = conn.Read(buf)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
str := GbkAsUtf8(string(buf[:]))
|
|
|
|
if strings.Contains(str, ">") {
|
|
return nil
|
|
}
|
|
|
|
return errors.Errorf("连接失败!")
|
|
}
|
|
|
|
func RemoteCommandByPassword(servers *models.TaskServer) error {
|
|
var (
|
|
auth []ssh.AuthMethod
|
|
addr string
|
|
clientConfig *ssh.ClientConfig
|
|
)
|
|
|
|
auth = make([]ssh.AuthMethod, 0)
|
|
auth = append(auth, ssh.Password(servers.Password))
|
|
|
|
clientConfig = &ssh.ClientConfig{
|
|
User: servers.ServerAccount,
|
|
Auth: auth,
|
|
HostKeyCallback: func(hostname string, remote net.Addr, key ssh.PublicKey) error {
|
|
return nil
|
|
},
|
|
Timeout: 5 * time.Second,
|
|
}
|
|
|
|
addr = fmt.Sprintf("%s:%d", servers.ServerIp, servers.Port)
|
|
client, err := ssh.Dial("tcp", addr, clientConfig)
|
|
if err == nil {
|
|
defer client.Close()
|
|
}
|
|
return err
|
|
}
|
|
|
|
func RemoteCommandByKey(servers *models.TaskServer) error {
|
|
key, err := ioutil.ReadFile(servers.PrivateKeySrc)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
signer, err := ssh.ParsePrivateKey(key)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
addr := fmt.Sprintf("%s:%d", servers.ServerIp, servers.Port)
|
|
config := &ssh.ClientConfig{
|
|
User: servers.ServerAccount,
|
|
Auth: []ssh.AuthMethod{
|
|
// Use the PublicKeys method for remote authentication.
|
|
ssh.PublicKeys(signer),
|
|
},
|
|
//HostKeyCallback: ssh.FixedHostKey(hostKey),
|
|
HostKeyCallback: func(hostname string, remote net.Addr, key ssh.PublicKey) error {
|
|
return nil
|
|
},
|
|
Timeout: 5 * time.Second,
|
|
}
|
|
|
|
client, err := ssh.Dial("tcp", addr, config)
|
|
if err == nil {
|
|
client.Close()
|
|
}
|
|
return err
|
|
}
|
|
|
|
func RemoteAgent(servers *models.TaskServer) error {
|
|
|
|
conn, err := net.Dial("tcp", servers.ServerIp+":"+strconv.Itoa(servers.Port))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer conn.Close()
|
|
client := rpc.NewClientWithCodec(jsonrpc.NewClientCodec(conn))
|
|
var reply *common.RpcResult
|
|
defer client.Close()
|
|
|
|
ping := "ping"
|
|
err = client.Call("RpcTask.HeartBeat", ping, &reply)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if reply.Status == 200 {
|
|
return nil
|
|
} else {
|
|
return fmt.Errorf("链接错误：%v", reply.Message)
|
|
}
|
|
}
|