write knownhosts like openssh

trzsz · Mar 3, 2024 · cee1fa4 · cee1fa4
1 parent 82a80cb
commit cee1fa4
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/tssh/login.go b/tssh/login.go
@@ -249,7 +249,13 @@ func writeKnownHost(path, host string, remote net.Addr, key ssh.PublicKey) error
 	if err := ensureNewline(file); err != nil {
 		return err
 	}
-	return knownhosts.WriteKnownHost(file, host, remote, key)
+
+	hostNormalized := knownhosts.Normalize(host)
+	if strings.ContainsAny(hostNormalized, "\t ") {
+		return fmt.Errorf("host '%s' contains spaces", hostNormalized)
+	}
+	line := knownhosts.Line([]string{hostNormalized}, key) + "\n"
+	return writeAll(file, []byte(line))
 }
 
 func addHostKey(path, host string, remote net.Addr, key ssh.PublicKey, ask bool) error {