ssh_agent.go

package laptop_booter

import (
	"fmt"
	"net"
	"os"
	"time"

	"log"

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

func SSHConfigFromAgent(username string) (closer func(), clientConfig *ssh.ClientConfig, err error) {
	var signers []ssh.Signer
	agentConn, err := net.Dial("unix", os.Getenv("SSH_AUTH_SOCK"))
	if err != nil {
		return nil, nil, fmt.Errorf("cannot connect to ssh-agent, check if SSH_AUTH_SOCK is set: %w", err)
	}
	sshAgent := agent.NewClient(agentConn)
	signers, err = sshAgent.Signers()
	log.Printf("Signers available in the SSH Agent: %+v, user: %v", len(signers), username)
	if err != nil {
		return nil, nil, err
	}
	closer = func() {
		agentConn.Close()
	}
	clientConfig = &ssh.ClientConfig{
		User:            username,
		Auth:            []ssh.AuthMethod{ssh.PublicKeys(signers...)},
		Timeout:         5 * time.Second,
		HostKeyCallback: ssh.InsecureIgnoreHostKey(),
	}
	return
}