-
-
Save corny/d9c1a75b369b4a43cb688c47d0cd1c97 to your computer and use it in GitHub Desktop.
SSH tunnelling in Golang
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
package main | |
import ( | |
"fmt" | |
"io" | |
"io/ioutil" | |
"log" | |
"net" | |
"os" | |
"golang.org/x/crypto/ssh" | |
) | |
type SSHtunnel struct { | |
Local string | |
Server string | |
Remote string | |
Config *ssh.ClientConfig | |
} | |
func (tunnel *SSHtunnel) Start() error { | |
listener, err := net.Listen("tcp", tunnel.Local) | |
if err != nil { | |
return err | |
} | |
defer listener.Close() | |
sshClient, err := ssh.Dial("tcp", tunnel.Server, tunnel.Config) | |
if err != nil { | |
return err | |
} | |
for { | |
conn, err := listener.Accept() | |
if err != nil { | |
return err | |
} | |
go tunnel.forward(sshClient, conn) | |
} | |
} | |
func (tunnel *SSHtunnel) forward(sshClient *ssh.Client, localConn net.Conn) { | |
remoteConn, err := sshClient.Dial("tcp", tunnel.Remote) | |
if err != nil { | |
fmt.Printf("Remote dial error: %s\n", err) | |
return | |
} | |
copyConn := func(writer, reader net.Conn) { | |
_, err := io.Copy(writer, reader) | |
if err != nil { | |
fmt.Printf("io.Copy error: %s", err) | |
} | |
log.Println("closing channel", remoteConn.Close()) | |
} | |
go copyConn(localConn, remoteConn) | |
go copyConn(remoteConn, localConn) | |
} | |
// Liest einen SSH-Key ein | |
func getKeyFile(path string) (ssh.Signer, error) { | |
buf, err := ioutil.ReadFile(path) | |
if err != nil { | |
return nil, err | |
} | |
return ssh.ParsePrivateKey(buf) | |
} | |
func main() { | |
key, err := getKeyFile(os.Getenv("HOME") + "/.ssh/id_rsa") | |
if err != nil { | |
panic(err) | |
} | |
sshConfig := &ssh.ClientConfig{ | |
User: "root", | |
Auth: []ssh.AuthMethod{ssh.PublicKeys(key)}, | |
HostKeyCallback: func(hostname string, remote net.Addr, key ssh.PublicKey) error { | |
return nil | |
}, | |
} | |
tunnel := &SSHtunnel{ | |
Config: sshConfig, | |
Server: "example.com:22", | |
Local: "localhost:9100", | |
Remote: "localhost:9100", | |
} | |
tunnel.Start() | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment