Fix ApproveFingerprint: extract real host key via SSH instead of writing fingerprint SHA256 to known_hosts
The old ApproveFingerprint passed the SHA256 fingerprint string to AddKnownHost which expected authorized_key format, causing known_hosts entries to be corrupted and subsequent SSH connections (including shutdown) to fail with "host key not found". Changes: - dialSSH now returns (conn, fingerprint, pubKey, error) with the raw ssh.PublicKey captured from the server - New ConnectForApproval() wraps dialSSH with strictHostKeyChecking=false for the approval handshake - ApproveFingerprint now opens an SSH connection to the host (non-strict), captures the real public key, and writes it in authorized_keys format to known_hosts via AddKnownHost - shutdown.go updated to handle the new 4-value dialSSH return - Supports optional host_key field in request body for direct key submission
This commit is contained in:
@@ -14,6 +14,7 @@ import (
|
|||||||
|
|
||||||
"github.com/go-chi/chi/v5"
|
"github.com/go-chi/chi/v5"
|
||||||
"github.com/syncserver/internal/config"
|
"github.com/syncserver/internal/config"
|
||||||
|
"golang.org/x/crypto/ssh"
|
||||||
"github.com/syncserver/internal/models"
|
"github.com/syncserver/internal/models"
|
||||||
"github.com/syncserver/internal/sshmanager"
|
"github.com/syncserver/internal/sshmanager"
|
||||||
"github.com/syncserver/internal/syncengine"
|
"github.com/syncserver/internal/syncengine"
|
||||||
@@ -355,29 +356,63 @@ func (h *MachineHandler) ApproveFingerprint(w http.ResponseWriter, r *http.Reque
|
|||||||
|
|
||||||
var req struct {
|
var req struct {
|
||||||
Fingerprint string `json:"fingerprint"`
|
Fingerprint string `json:"fingerprint"`
|
||||||
|
HostKey string `json:"host_key"`
|
||||||
}
|
}
|
||||||
if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
|
json.NewDecoder(r.Body).Decode(&req)
|
||||||
writeError(w, http.StatusBadRequest, "invalid request body")
|
|
||||||
return
|
sshDir := filepath.Join("/var/lib/syncserver", "ssh")
|
||||||
}
|
knownHostsPath, err := sshmanager.EnsureKnownHosts(sshDir)
|
||||||
if req.Fingerprint == "" {
|
if err != nil {
|
||||||
writeError(w, http.StatusBadRequest, "fingerprint is required")
|
writeError(w, http.StatusInternalServerError, "failed to ensure known_hosts")
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
if err := repo.UpdateFingerprint(id, true, req.Fingerprint); err != nil {
|
privKeyPath := filepath.Join("/var/lib/syncserver", "ssh", "id_ed25519")
|
||||||
|
if m.SSHKeyID != nil {
|
||||||
|
sshKeyRepo := models.NewSSHKeyRepository(h.db)
|
||||||
|
sshKey, err := sshKeyRepo.GetByID(*m.SSHKeyID)
|
||||||
|
if err == nil && sshKey.PrivateKeyPath != "" {
|
||||||
|
privKeyPath = sshKey.PrivateKeyPath
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
var fingerprint, pubKeyLine string
|
||||||
|
|
||||||
|
if req.HostKey != "" {
|
||||||
|
pubKeyLine = req.HostKey
|
||||||
|
} else {
|
||||||
|
conn, fp, pubKey, err := sshmanager.ConnectForApproval(
|
||||||
|
context.Background(), m.Host, m.Port, m.SSHUser,
|
||||||
|
privKeyPath, knownHostsPath,
|
||||||
|
)
|
||||||
|
if err != nil {
|
||||||
|
writeError(w, http.StatusInternalServerError, fmt.Sprintf("could not retrieve host key: %v", err))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
conn.Close()
|
||||||
|
fingerprint = fp
|
||||||
|
pubKeyLine = string(ssh.MarshalAuthorizedKey(pubKey))
|
||||||
|
}
|
||||||
|
|
||||||
|
if fingerprint == "" && req.Fingerprint != "" {
|
||||||
|
fingerprint = req.Fingerprint
|
||||||
|
}
|
||||||
|
|
||||||
|
if pubKeyLine != "" {
|
||||||
|
if err := sshmanager.AddKnownHost(sshDir, m.Host, m.Port, []byte(pubKeyLine)); err != nil {
|
||||||
|
writeError(w, http.StatusInternalServerError, fmt.Sprintf("failed to add known_host entry: %v", err))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := repo.UpdateFingerprint(id, true, fingerprint); err != nil {
|
||||||
slog.Error("failed to update fingerprint", "id", id, "error", err)
|
slog.Error("failed to update fingerprint", "id", id, "error", err)
|
||||||
writeError(w, http.StatusInternalServerError, "failed to update fingerprint")
|
writeError(w, http.StatusInternalServerError, "failed to update fingerprint")
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
sshDir := filepath.Join("/var/lib/syncserver", "ssh")
|
|
||||||
if err := sshmanager.AddKnownHost(sshDir, m.Host, m.Port, []byte(req.Fingerprint)); err != nil {
|
|
||||||
slog.Warn("failed to add known_host entry", "host", m.Host, "error", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
m.FingerprintConfirmed = true
|
m.FingerprintConfirmed = true
|
||||||
m.HostKeyFingerprint = &req.Fingerprint
|
m.HostKeyFingerprint = &fingerprint
|
||||||
writeJSON(w, machineToResp(*m))
|
writeJSON(w, machineToResp(*m))
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -14,7 +14,7 @@ type ShutdownResult struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func RunRemoteCommand(ctx context.Context, host string, port int, user, privKeyPath, knownHostsPath string, strictHostKeyChecking bool, command string) (*ShutdownResult, error) {
|
func RunRemoteCommand(ctx context.Context, host string, port int, user, privKeyPath, knownHostsPath string, strictHostKeyChecking bool, command string) (*ShutdownResult, error) {
|
||||||
conn, _, err := dialSSH(ctx, host, port, user, privKeyPath, knownHostsPath, strictHostKeyChecking)
|
conn, _, _, err := dialSSH(ctx, host, port, user, privKeyPath, knownHostsPath, strictHostKeyChecking)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return &ShutdownResult{Success: false, Error: err.Error()}, nil
|
return &ShutdownResult{Success: false, Error: err.Error()}, nil
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -22,26 +22,28 @@ type ConnResult struct {
|
|||||||
Fingerprint string
|
Fingerprint string
|
||||||
}
|
}
|
||||||
|
|
||||||
func dialSSH(ctx context.Context, host string, port int, user, privKeyPath, knownHostsPath string, strictHostKeyChecking bool) (*ssh.Client, string, error) {
|
func dialSSH(ctx context.Context, host string, port int, user, privKeyPath, knownHostsPath string, strictHostKeyChecking bool) (*ssh.Client, string, ssh.PublicKey, error) {
|
||||||
addr := fmt.Sprintf("%s:%d", host, port)
|
addr := fmt.Sprintf("%s:%d", host, port)
|
||||||
|
|
||||||
auths := []ssh.AuthMethod{}
|
auths := []ssh.AuthMethod{}
|
||||||
if privKeyPath != "" {
|
if privKeyPath != "" {
|
||||||
key, err := os.ReadFile(privKeyPath)
|
key, err := os.ReadFile(privKeyPath)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, "", fmt.Errorf("reading private key: %w", err)
|
return nil, "", nil, fmt.Errorf("reading private key: %w", err)
|
||||||
}
|
}
|
||||||
signer, err := ssh.ParsePrivateKey(key)
|
signer, err := ssh.ParsePrivateKey(key)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, "", fmt.Errorf("parsing private key: %w", err)
|
return nil, "", nil, fmt.Errorf("parsing private key: %w", err)
|
||||||
}
|
}
|
||||||
auths = append(auths, ssh.PublicKeys(signer))
|
auths = append(auths, ssh.PublicKeys(signer))
|
||||||
}
|
}
|
||||||
|
|
||||||
var capturedFingerprint string
|
var capturedFingerprint string
|
||||||
|
var capturedPubKey ssh.PublicKey
|
||||||
hostKeyCallback := func(hostname string, remote net.Addr, key ssh.PublicKey) error {
|
hostKeyCallback := func(hostname string, remote net.Addr, key ssh.PublicKey) error {
|
||||||
h := sha256.Sum256(key.Marshal())
|
h := sha256.Sum256(key.Marshal())
|
||||||
capturedFingerprint = "SHA256:" + base64.RawStdEncoding.EncodeToString(h[:])
|
capturedFingerprint = "SHA256:" + base64.RawStdEncoding.EncodeToString(h[:])
|
||||||
|
capturedPubKey = key
|
||||||
if strictHostKeyChecking && knownHostsPath != "" {
|
if strictHostKeyChecking && knownHostsPath != "" {
|
||||||
kh, err := GetKnownHost(filepath.Dir(knownHostsPath), host, port)
|
kh, err := GetKnownHost(filepath.Dir(knownHostsPath), host, port)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -71,15 +73,15 @@ func dialSSH(ctx context.Context, host string, port int, user, privKeyPath, know
|
|||||||
conn, err := ssh.Dial("tcp", addr, cfg)
|
conn, err := ssh.Dial("tcp", addr, cfg)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
if strings.Contains(err.Error(), "known_hosts") || strings.Contains(err.Error(), "host key") {
|
if strings.Contains(err.Error(), "known_hosts") || strings.Contains(err.Error(), "host key") {
|
||||||
return nil, capturedFingerprint, fmt.Errorf("host key verification failed: %v", err)
|
return nil, capturedFingerprint, capturedPubKey, fmt.Errorf("host key verification failed: %v", err)
|
||||||
}
|
}
|
||||||
return nil, capturedFingerprint, fmt.Errorf("connection failed: %v", err)
|
return nil, capturedFingerprint, capturedPubKey, fmt.Errorf("connection failed: %v", err)
|
||||||
}
|
}
|
||||||
return conn, capturedFingerprint, nil
|
return conn, capturedFingerprint, capturedPubKey, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestSSHConnection(ctx context.Context, host string, port int, user, privKeyPath, knownHostsPath string, strictHostKeyChecking bool) (*ConnResult, error) {
|
func TestSSHConnection(ctx context.Context, host string, port int, user, privKeyPath, knownHostsPath string, strictHostKeyChecking bool) (*ConnResult, error) {
|
||||||
conn, fingerprint, err := dialSSH(ctx, host, port, user, privKeyPath, knownHostsPath, strictHostKeyChecking)
|
conn, fingerprint, _, err := dialSSH(ctx, host, port, user, privKeyPath, knownHostsPath, strictHostKeyChecking)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return &ConnResult{
|
return &ConnResult{
|
||||||
Success: false,
|
Success: false,
|
||||||
@@ -113,3 +115,7 @@ func TestSSHConnection(ctx context.Context, host string, port int, user, privKey
|
|||||||
Fingerprint: fingerprint,
|
Fingerprint: fingerprint,
|
||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func ConnectForApproval(ctx context.Context, host string, port int, user, privKeyPath, knownHostsPath string) (*ssh.Client, string, ssh.PublicKey, error) {
|
||||||
|
return dialSSH(ctx, host, port, user, privKeyPath, knownHostsPath, false)
|
||||||
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user