From 0d3f55a926e42ffa45febc4cc722f3d7873206e0 Mon Sep 17 00:00:00 2001
From: Sam Anthony <sam@samanthony.xyz>
Date: Wed, 16 Apr 2025 17:27:20 -0400
Subject: call key decoding functions when parsing known_hosts file

---
 hosts/hosts.go | 25 +++++++++----------------
 1 file changed, 9 insertions(+), 16 deletions(-)

(limited to 'hosts')

diff --git a/hosts/hosts.go b/hosts/hosts.go
index 3dc3f33..0412e4a 100644
--- a/hosts/hosts.go
+++ b/hosts/hosts.go
@@ -2,7 +2,7 @@ package hosts
 
 import (
 	"bufio"
-	"encoding/hex"
+	"bytes"
 	"errors"
 	"fmt"
 	"github.com/adrg/xdg"
@@ -10,7 +10,6 @@ import (
 	"os"
 	"path/filepath"
 	"slices"
-	"strings"
 
 	"git.samanthony.xyz/hose/key"
 	"git.samanthony.xyz/hose/util"
@@ -57,7 +56,7 @@ func Load() ([]Host, error) {
 
 	scanner := bufio.NewScanner(f)
 	for line := 1; scanner.Scan(); line++ {
-		host, err := parseHost(scanner.Text())
+		host, err := parseHost(scanner.Bytes())
 		if err != nil {
 			return hosts, fmt.Errorf("error parsing known hosts file: %s:%d: %v", knownHostsFile, line, err)
 		}
@@ -71,30 +70,24 @@ func Load() ([]Host, error) {
 }
 
 // parseHost parses a line of the known hosts file.
-func parseHost(s string) (Host, error) {
-	fields := strings.Fields(s)
+func parseHost(b []byte) (Host, error) {
+	fields := bytes.Fields(b)
 	if len(fields) != 3 {
 		return Host{}, fmt.Errorf("expected 3 fields; got %d", len(fields))
 	}
 
-	addr, err := netip.ParseAddr(fields[0])
+	addr, err := netip.ParseAddr(string(fields[0]))
 	if err != nil {
 		return Host{}, err
 	}
 
-	var boxPubKey key.BoxPublicKey
-	if hex.DecodedLen(len(fields[1])) != len(boxPubKey) {
-		return Host{}, fmt.Errorf("malformed box public key: %s", fields[1])
-	}
-	if _, err := hex.Decode(boxPubKey[:], []byte(fields[1])); err != nil {
+	boxPubKey, err := key.DecodeBoxPublicKey(fields[1])
+	if err != nil {
 		return Host{}, err
 	}
 
-	var sigPubKey key.SigPublicKey
-	if hex.DecodedLen(len(fields[2])) != len(sigPubKey) {
-		return Host{}, fmt.Errorf("malformed signature public key: %s", fields[2])
-	}
-	if _, err := hex.Decode(sigPubKey[:], []byte(fields[2])); err != nil {
+	sigPubKey, err := key.DecodeSigPublicKey(fields[2])
+	if err != nil {
 		return Host{}, err
 	}
 
-- 
cgit v1.2.3