golangで~/.ssh/configを読み込んでssh接続する(ProxyCommand対応)

自作のsshクライアントコマンドで~/.ssh/configを読み込んで処理させる機能を追加したかったのだが、ProxyCommandがネックになっていたため今まで実装を見送っていた。 で、久しぶりになんか情報ないかなと探してみたところ、net.Pipe()を利用したら処理できるという情報を見かけた。

ProxyCommandはローカルで実行するコマンドになるので、そのコマンドの出力をPipeで繋げていけばいいようだ。なるほど…(´・ω・｀)。 というわけで、さっそく実装してみた。こんな感じのコードで.ssh/configから指定したホストの情報を取得して、かつProxyCommandでのプロキシ経由の接続ができるようだ。 ソースコードは こちら にも上げてある。

ssh_term_ssh_config_proxy_command.go


package main

import (
    "fmt"
    "io"
    "io/ioutil"
    "net"
    "os"
    "os/exec"
    "os/signal"
    "os/user"
    "strings"
    "syscall"

    "github.com/kevinburke/ssh_config"
    "golang.org/x/crypto/ssh"
    "golang.org/x/crypto/ssh/terminal"
)

var (
    host             = "test_cent7"    // テスト用ホスト
    defaultSshConfig = "~/.ssh/config" // ssh/config
)

func main() {
    // current user
    usr, _ := user.Current()
    confFile := strings.Replace(defaultSshConfig, "~", usr.HomeDir, 1)

    // .ssh/configを開いてDecode
    f, _ := os.Open(confFile)
    cfg, _ := ssh_config.Decode(f)

    // 情報の取得
    user, _ := cfg.Get(host, "User")
    addr, _ := cfg.Get(host, "HostName")
    port, _ := cfg.Get(host, "Port")
    keyPath, _ := cfg.Get(host, "IdentityFile")
    proxyCommand, _ := cfg.Get(host, "ProxyCommand")

    // keyPathを置換
    keyPath = strings.Replace(keyPath, "~", usr.HomeDir, 1)

    // ProxyCommandを置換
    proxyCommand = strings.Replace(proxyCommand, "%h", addr, -1)
    proxyCommand = strings.Replace(proxyCommand, "%p", port, -1)

    // sshConfigの生成
    auth := []ssh.AuthMethod{}
    key, _ := ioutil.ReadFile(keyPath)
    signer, _ := ssh.ParsePrivateKey(key)
    auth = append(auth, ssh.PublicKeys(signer))
    sshConfig := &ssh.ClientConfig{
        User:            user,
        Auth:            auth,
        HostKeyCallback: ssh.InsecureIgnoreHostKey(), // FIXME
    }

    fmt.Println(addr)
    fmt.Println(port)
    fmt.Println(user)
    fmt.Println(keyPath)
    fmt.Println(proxyCommand)

    // net.Pipeの作成
    cli, srv := net.Pipe()
    cmd := exec.Command("bash", "-c", proxyCommand)

    // 標準入出力の指定
    cmd.Stdin = srv
    cmd.Stdout = srv
    cmd.Stderr = os.Stderr

    // コマンドの実行
    if err := cmd.Start(); err != nil {
        os.Exit(1)
    }

    // // proxy経由での接続を実施
    conn, incomingChannels, incomingRequests, err := ssh.NewClientConn(cli, net.JoinHostPort(addr, port), sshConfig)
    if err != nil {
        os.Exit(1)
    }

    // proxy経由でのsshClientの作成
    client := ssh.NewClient(conn, incomingChannels, incomingRequests)
    // client, _ := ssh.Dial("tcp", net.JoinHostPort(addr, port), sshConfig)

    // ↓共通コード
    // Create Session
    session, err := client.NewSession()
    defer session.Close()

    // キー入力を接続先が認識できる形式に変換する(ここがキモ)
    fd := int(os.Stdin.Fd())
    state, err := terminal.MakeRaw(fd)
    if err != nil {
        fmt.Println(err)
    }
    defer terminal.Restore(fd, state)

    // ターミナルサイズの取得
    w, h, err := terminal.GetSize(fd)
    if err != nil {
        fmt.Println(err)
    }

    modes := ssh.TerminalModes{
        ssh.ECHO:          1,
        ssh.TTY_OP_ISPEED: 14400,
        ssh.TTY_OP_OSPEED: 14400,
    }

    err = session.RequestPty("xterm", h, w, modes)
    if err != nil {
        fmt.Println(err)
    }

    // log := new(bytes.Buffer)
    logFile, _ := os.OpenFile("./ssh_term_with_log.log", os.O_RDWR|os.O_CREATE|os.O_APPEND, 0600)

    session.Stdout = io.MultiWriter(os.Stdout, logFile)
    session.Stderr = io.MultiWriter(os.Stderr, logFile)
    session.Stdin = os.Stdin

    err = session.Shell()
    if err != nil {
        fmt.Println(err)
    }

    // ターミナルサイズの変更検知・処理
    signal_chan := make(chan os.Signal, 1)
    signal.Notify(signal_chan, syscall.SIGWINCH)
    go func() {
        for {
            s := <-signal_chan
            switch s {
            case syscall.SIGWINCH:
                fd := int(os.Stdout.Fd())
                w, h, _ = terminal.GetSize(fd)
                session.WindowChange(h, w)
            }
        }
    }()

    err = session.Wait()
    if err != nil {
        fmt.Println(err)
    }

}

参考