diff options
| -rw-r--r-- | hosts/hosts.go | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/hosts/hosts.go b/hosts/hosts.go index d31ed5e..63433cb 100644 --- a/hosts/hosts.go +++ b/hosts/hosts.go @@ -17,8 +17,12 @@ var knownHostsFile = filepath.Join(xdg.DataHome, "hose", "known_hosts") // Set sets the public key of a remote host. // It replaces or creates an entry in the known hosts file. -func Set(host net.Addr, pubkey [32]byte) error { - addr, err := netip.ParseAddr(host.String()) +func Set(hostport net.Addr, pubkey [32]byte) error { + host, _, err := net.SplitHostPort(hostport.String()) + if err != nil { + return err + } + addr, err := netip.ParseAddr(host) if err != nil { return err } |