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

Implement default known hosts path

Browse Source
This commit is contained in:
adnano 2020-09-26 13:35:56 -04:00
parent 1b3f9a0655
commit 769b6ff4d9
2 changed files with 38 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -50,7 +50,7 @@ clients. Here is a simple client using TOFU to authenticate certificates:
```go ```go
client := &gemini.Client{ client := &gemini.Client{
KnownHosts: gemini.LoadKnownHosts(".local/share/gemini/known_hosts"), KnownHosts: gemini.LoadKnownHosts(),
TrustCertificate: func(cert *x509.Certificate, knownHosts *gemini.KnownHosts) error { TrustCertificate: func(cert *x509.Certificate, knownHosts *gemini.KnownHosts) error {
// If the certificate is in the known hosts list, allow the connection // If the certificate is in the known hosts list, allow the connection
if err := knownHosts.Lookup(cert); { if err := knownHosts.Lookup(cert); {

38
tofu.go
View File

@ -20,10 +20,22 @@ type KnownHosts struct {
file *os.File file *os.File
} }
// LoadKnownHosts loads the known hosts from the default known hosts path.
// The default path is $XDG_DATA_HOME/gemini/known_hosts
// It creates the path and any of its parent directories if they do not exist.
// The returned KnownHosts appends to the file whenever a certificate is added.
func LoadKnownHosts() (*KnownHosts, error) {
path, err := defaultKnownHostsPath()
if err != nil {
return nil, err
}
return LoadKnownHostsFrom(path)
}
// LoadKnownHosts loads the known hosts from the provided path. // LoadKnownHosts loads the known hosts from the provided path.
// It creates the path and any of its parent directories if they do not exist. // It creates the path and any of its parent directories if they do not exist.
// The returned KnownHosts appends to the file whenever a certificate is added. // The returned KnownHosts appends to the file whenever a certificate is added.
func LoadKnownHosts(path string) (*KnownHosts, error) { func LoadKnownHostsFrom(path string) (*KnownHosts, error) {
if dir := filepath.Dir(path); dir != "." { if dir := filepath.Dir(path); dir != "." {
err := os.MkdirAll(dir, 0755) err := os.MkdirAll(dir, 0755)
if err != nil { if err != nil {
@ -149,3 +161,27 @@ func Fingerprint(cert *x509.Certificate) string {
} }
return buf.String() return buf.String()
} }
// defaultKnownHostsPath returns the default known_hosts path.
// The default path is $XDG_DATA_HOME/gemini/known_hosts
func defaultKnownHostsPath() (string, error) {
dataDir, err := userDataDir()
if err != nil {
return "", err
}
return filepath.Join(dataDir, "gemini", "known_hosts"), nil
}
// userDataDir returns the user data directory.
func userDataDir() (string, error) {
dataDir, ok := os.LookupEnv("XDG_DATA_HOME")
if ok {
return dataDir, nil
}
home, err := os.UserHomeDir()
if err != nil {
return "", err
}
return filepath.Join(home, ".local", "share"), nil
}