libplugin

libplugin/util.go

@@ -0,0 +1,138 @@
+package libplugin
+
+import (
+	"fmt"
+	"io"
+	"net"
+	"os"
+	"strconv"
+
+	"github.com/sirupsen/logrus"
+)
+
+func WriterToFile(writer io.Writer) (*os.File, error) {
+	r, w, err := os.Pipe()
+	if err != nil {
+		return nil, err
+	}
+
+	go io.Copy(writer, r)
+
+	return w, nil
+}
+
+func AuthMethodTypeToName(a AuthMethod) string {
+	switch a {
+	case AuthMethod_NONE:
+		return "none"
+	case AuthMethod_PASSWORD:
+		return "password"
+	case AuthMethod_PUBLICKEY:
+		return "publickey"
+	case AuthMethod_KEYBOARD_INTERACTIVE:
+		return "keyboard-interactive"
+	}
+	return ""
+}
+
+func AuthMethodFromName(n string) AuthMethod {
+	switch n {
+	case "none":
+		return AuthMethod_NONE
+	case "password":
+		return AuthMethod_PASSWORD
+	case "publickey":
+		return AuthMethod_PUBLICKEY
+	case "keyboard-interactive":
+		return AuthMethod_KEYBOARD_INTERACTIVE
+	}
+	return -1
+}
+
+func ConfigStdioLogrus(p SshPiperPlugin, logger *logrus.Logger) {
+	if logger == nil {
+		logger = logrus.StandardLogger()
+	}
+	logger.SetOutput(p.GetLoggerOutput())
+	logger.SetFormatter(&logrus.TextFormatter{ForceColors: true})
+}
+
+// SplitHostPortForSSH is the modified version of net.SplitHostPort but return port 22 is no port is specified
+func SplitHostPortForSSH(addr string) (host string, port int, err error) {
+	host = addr
+	h, p, err := net.SplitHostPort(host)
+	if err == nil {
+		host = h
+		port, err = strconv.Atoi(p)
+
+		if err != nil {
+			return
+		}
+	} else if host != "" {
+		// test valid after concat :22
+		if _, _, err = net.SplitHostPort(host + ":22"); err == nil {
+			port = 22
+		}
+	}
+
+	if host == "" {
+		err = fmt.Errorf("empty addr")
+	}
+
+	return
+}
+
+// DialForSSH is the modified version of net.Dial, would add ":22" automaticlly
+func DialForSSH(addr string) (net.Conn, error) {
+
+	if _, _, err := net.SplitHostPort(addr); err != nil && addr != "" {
+		// test valid after concat :22
+		if _, _, err := net.SplitHostPort(addr + ":22"); err == nil {
+			addr += ":22"
+		}
+	}
+
+	return net.Dial("tcp", addr)
+}
+
+func CreateNoneAuth(password []byte) *Upstream_None {
+	return &Upstream_None{
+		None: &UpstreamNoneAuth{},
+	}
+}
+
+func CreatePasswordAuth(password []byte) *Upstream_Password {
+	return CreatePasswordAuthFromString(string(password))
+}
+
+func CreatePasswordAuthFromString(password string) *Upstream_Password {
+	return &Upstream_Password{
+		Password: &UpstreamPasswordAuth{
+			Password: password,
+		},
+	}
+}
+
+func CreatePrivateKeyAuth(key []byte) *Upstream_PrivateKey {
+	return &Upstream_PrivateKey{
+		PrivateKey: &UpstreamPrivateKeyAuth{
+			PrivateKey: key,
+		},
+	}
+}
+
+func CreateRemoteSignerAuth(meta string) *Upstream_RemoteSigner {
+	return &Upstream_RemoteSigner{
+		RemoteSigner: &UpstreamRemoteSignerAuth{
+			Meta: meta,
+		},
+	}
+}
+
+func CreateNextPluginAuth(meta map[string]string) *Upstream_NextPlugin {
+	return &Upstream_NextPlugin{
+		NextPlugin: &UpstreamNextPluginAuth{
+			Meta: meta,
+		},
+	}
+}