我终于想出了如何做到这一点,我在 IRC 频道中从 schmichael 那里得到了提示。谢谢大家!
编辑:一点解释:
我遇到的问题的很大一部分是我没有意识到本地net.Listener(不仅仅是本地net.Conn)需要设置来接收本地请求并在转发字节之前创建net.Conn。
此外,同时存在端口转发和反向端口转发,我之前没有详细考虑过常规端口转发也会发回字节的事实,因此我没有实现将远程读取器复制到本地写入器,但它是非常需要。
下面是一个尝试将这段代码的本质联系起来:
- 监听本地端口 9000。
- 尝试从本地端口 9000 读取时:(
listener.Accept()),
- 接受连接并返回本地
io.Reader 和io.Writer 和,
- 连接到远程服务器,
- 连接到远程端口 9999,返回
io.Reader 和 io.Writer。
- 不断复制本地
io.Reader字节到远程io.Writer,
- 不断将远程
io.Reader 字节复制到本地io.Writer。
代码如下:
package main
// Forward from local port 9000 to remote port 9999
import (
"io"
"log"
"net"
"golang.org/x/crypto/ssh"
)
var (
username = "root"
password = "password"
serverAddrString = "192.168.1.100:22"
localAddrString = "localhost:9000"
remoteAddrString = "localhost:9999"
)
func forward(localConn net.Conn, config *ssh.ClientConfig) {
// Setup sshClientConn (type *ssh.ClientConn)
sshClientConn, err := ssh.Dial("tcp", serverAddrString, config)
if err != nil {
log.Fatalf("ssh.Dial failed: %s", err)
}
// Setup sshConn (type net.Conn)
sshConn, err := sshClientConn.Dial("tcp", remoteAddrString)
// Copy localConn.Reader to sshConn.Writer
go func() {
_, err = io.Copy(sshConn, localConn)
if err != nil {
log.Fatalf("io.Copy failed: %v", err)
}
}()
// Copy sshConn.Reader to localConn.Writer
go func() {
_, err = io.Copy(localConn, sshConn)
if err != nil {
log.Fatalf("io.Copy failed: %v", err)
}
}()
}
func main() {
// Setup SSH config (type *ssh.ClientConfig)
config := &ssh.ClientConfig{
User: username,
Auth: []ssh.AuthMethod{
ssh.Password(password),
},
}
// Setup localListener (type net.Listener)
localListener, err := net.Listen("tcp", localAddrString)
if err != nil {
log.Fatalf("net.Listen failed: %v", err)
}
for {
// Setup localConn (type net.Conn)
localConn, err := localListener.Accept()
if err != nil {
log.Fatalf("listen.Accept failed: %v", err)
}
go forward(localConn, config)
}
}