sashakoshka/go-gemini
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix hosts not being added to known hosts file

Browse Source
This commit is contained in:
adnano 2020-09-27 16:06:17 -04:00
parent c4af352e87
commit 32a9fcba0c
2 changed files with 7 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/client/client.go
View File

@ -37,7 +37,7 @@ func init() {
return nil return nil
} else if userTrustsCertificatePermanently() { } else if userTrustsCertificatePermanently() {
// Add the certificate to the known hosts file // Add the certificate to the known hosts file
knownHosts.Add(cert) knownHosts.Add(req.Hostname(), cert)
return nil return nil
} }
} }

15
tofu.go
View File

@ -60,15 +60,12 @@ func (k *KnownHosts) LoadFrom(path string) error {
// Add adds a certificate to the list of known hosts. // Add adds a certificate to the list of known hosts.
// If KnownHosts was loaded from a file, Add will append to the file. // If KnownHosts was loaded from a file, Add will append to the file.
func (k *KnownHosts) Add(cert *x509.Certificate) { func (k *KnownHosts) Add(hostname string, cert *x509.Certificate) {
// Add an entry per hostname host := NewKnownHost(hostname, cert)
for _, name := range cert.DNSNames { k.hosts = append(k.hosts, host)
host := NewKnownHost(name, cert) // Append to the file
k.hosts = append(k.hosts, host) if k.file != nil {
// Append to the file host.Write(k.file)
if k.file != nil {
host.Write(k.file)
}
} }
} }