Question

我在golang中用crypto / ssh包编写了一个小型SSH服务器。

它支持返回交互式shell并立即执行命令。

以下是服务器的最小示例：

package main

import (
    "fmt"
    "io/ioutil"
    "log"
    "net"
    "os/exec"

    "golang.org/x/crypto/ssh"
)

func main() {
    c := &ssh.ServerConfig{
        PasswordCallback: func(c ssh.ConnMetadata, pass []byte) (*ssh.Permissions, error) {
            if c.User() == "foo" && string(pass) == "bar" {
                return nil, nil
            }
            return nil, fmt.Errorf("password rejected for %q", c.User())
        },
    }

    keyBytes, _ := ioutil.ReadFile("key")
    key, _ := ssh.ParsePrivateKey(keyBytes)
    c.AddHostKey(key)

    listener, _ := net.Listen("tcp", "0.0.0.0:2200")
    for {
        tcpConn, _ := listener.Accept()
        _, chans, reqs, _ := ssh.NewServerConn(tcpConn, c)
        go ssh.DiscardRequests(reqs)
        go handleChannels(chans)
    }
}

func handleChannels(chans <-chan ssh.NewChannel) {
    for newChannel := range chans {
        go handleChannel(newChannel)
    }
}

func handleChannel(newChannel ssh.NewChannel) {
    channel, requests, _ := newChannel.Accept()
    for req := range requests {
        switch req.Type {
        case "shell":
            go handleShell(channel)
        case "exec":
            go handleExec(channel, req)
        }
    }
}

func handleShell(c ssh.Channel) {}
func handleExec(c ssh.Channel, r *ssh.Request) {
    cmdString, args, _ := parseCommand(r.Payload)
    log.Printf("exec: %s\n", cmdString)
    for i := range args {
        log.Printf("arg %d: %s\n", i, args[i])
    }
    cmd := exec.Command(cmdString, args...)
    cmd.Run()
}

func parseCommand(b []byte) (string, []string, error) {
    cmdString := strings.TrimSpace(string(b))
    cmdArray := strings.Split(cmdString, " ")

    cmd := strings.Trim(cmdArray[0], " ")
    args := cmdArray[1:]

    return cmd, args, nil
}

如果我按如下方式运行服务器并执行scp：

scp -P 2200 test.file foo@localhost:~/

调用handleExec函数。

cmdString的输出显示：

2015/11/22 17:49:14 exec: scp
2015/11/22 17:49:14 arg 0: -t
2015/11/22 17:49:14 arg 1: ~/

但是如何实现handleExec函数来实际保存我通过scp传递的文件/目录？

Answer 1

我刚刚遇到了在我的ssh服务器上执行scp和自定义命令的问题，因为它没有记录如何执行此操作我将一些代码拼凑在一起来自crypto.ssh（https://github.com/golang/crypto/blob/master/ssh/session.go和{{3它适用于OpenSSH和crypto.ssh客户端。例如，您可以在客户端上调用session.Run（）并处理例如scp或自定义命令。

doReturn(returnValue).when(mockedObject).methodName()

您需要将runYourCommand替换为执行命令的任何函数，并将退出代码设置为您的命令/进程返回的任何内容。

Golang SSH-Server：如何使用scp处理文件传输？

1 个答案: