如何使用 Golang 创建简单的 SSH 端口转发？-Golang-PHP中文网

如何使用 Golang 创建简单的 SSH 端口转发？

DDD

发布： 2024-11-04 09:01:02

原创

290 人浏览过

How to Create a Simple SSH Port Forward Using Golang?

Golang 中的简单 SSH 端口转发

问题：

以编程方式在 Golang 中通过 SSH 创建基本 TCP 端口转发熟悉 Ruby 的用户。

解决方案：

要应对这一挑战，必须了解所涉及的关键步骤：

建立本地侦听器 (net.Listener) 接受本地端口上的连接。
尝试连接时，接受连接并获取本地 io.Reader 和 io.Writer 对象。
建立远程连接到服务器和远程端口，获取其 io.Reader 和 io.Writer 对象。
在本地和远程 io.Readers 和 io.Writer 之间持续双向传输字节。

代码：

<code class="go">package main

import (
    "io"
    "log"
    "net"
    "golang.org/x/crypto/ssh"
)

// Configuration parameters for SSH connection
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) {
    // Establish SSH connection to remote server
    sshClientConn, err := ssh.Dial("tcp", serverAddrString, config)
    if err != nil {
        log.Fatalf("ssh.Dial failed: %s", err)
    }

    // Connect to remote port
    sshConn, err := sshClientConn.Dial("tcp", remoteAddrString)
    if err != nil {
        log.Fatalf("sshConn.Dial failed: %s", err)
    }

    // Create goroutines for bidirectional data transfer
    go func() {
        _, err = io.Copy(sshConn, localConn)
        if err != nil {
            log.Fatalf("io.Copy failed: %v", err)
        }

    }()

    go func() {
        _, err = io.Copy(localConn, sshConn)
        if err != nil {
            log.Fatalf("io.Copy failed: %v", err)
        }

    }()
}

func main() {
    // Configure SSH client
    config := &ssh.ClientConfig{
        User: username,
        Auth: []ssh.AuthMethod{
            ssh.Password(password),
        },
    }

    // Create local listener
    localListener, err := net.Listen("tcp", localAddrString)
    if err != nil {
        log.Fatalf("net.Listen failed: %v", err)
    }

    // Monitor listener for new connections
    for {
        localConn, err := localListener.Accept()
        if err != nil {
            log.Fatalf("localListener.Accept failed: %v", err)
        }

        go forward(localConn, config)
    }
}</code>

登录后复制

以上是如何使用 Golang 创建简单的 SSH 端口转发？的详细内容。更多信息请关注PHP中文网其他相关文章！