V2.7 增加agent执行器

libs/server.go

@@ -0,0 +1,150 @@
+/************************************************************
+** @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)
+	}
+}