1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96
|
// Copyright (C) 2022 Andrew Ayer
//
// Permission is hereby granted, free of charge, to any person obtaining a
// copy of this software and associated documentation files (the "Software"),
// to deal in the Software without restriction, including without limitation
// the rights to use, copy, modify, merge, publish, distribute, sublicense,
// and/or sell copies of the Software, and to permit persons to whom the
// Software is furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included
// in all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
// THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR
// OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
// ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
// OTHER DEALINGS IN THE SOFTWARE.
//
// Except as contained in this notice, the name(s) of the above copyright
// holders shall not be used in advertising or otherwise to promote the
// sale, use or other dealings in this Software without prior written
// authorization.
package cert
import (
"encoding/base64"
"golang.org/x/crypto/acme"
"golang.org/x/crypto/acme/autocert"
"os"
"path/filepath"
)
func getAutocertCache() autocert.Cache {
if cacheDir := os.Getenv("AUTOCERT_CACHE_DIR"); cacheDir != "" {
return autocert.DirCache(cacheDir)
} else if os.Getuid() == 0 {
return autocert.DirCache("/var/lib/autocert-cache")
} else if dataDir := os.Getenv("XDG_DATA_HOME"); dataDir != "" {
return autocert.DirCache(filepath.Join(dataDir, "autocert-cache"))
} else if homeDir, err := os.UserHomeDir(); err == nil {
return autocert.DirCache(filepath.Join(homeDir, ".local/share/autocert-cache"))
} else {
return nil
}
}
func getAutocertEAB() *acme.ExternalAccountBinding {
kidString := os.Getenv("AUTOCERT_EAB_KID")
keyString := os.Getenv("AUTOCERT_EAB_KEY")
if kidString == "" || keyString == "" {
return nil
}
key, err := base64.RawURLEncoding.DecodeString(keyString)
if err != nil {
return nil
}
return &acme.ExternalAccountBinding{
KID: kidString,
Key: key,
}
}
func getCertificateAutomatically(hostPolicy autocert.HostPolicy) GetCertificateFunc {
manager := &autocert.Manager{
Client: &acme.Client{
DirectoryURL: os.Getenv("AUTOCERT_ACME_SERVER"),
},
Prompt: autocert.AcceptTOS,
Cache: getAutocertCache(),
HostPolicy: hostPolicy,
Email: os.Getenv("AUTOCERT_EMAIL"),
ExternalAccountBinding: getAutocertEAB(),
}
return manager.GetCertificate
}
// Returns a GetCertificateFunc that automatically obtains certificates using ACME
// for the given hostnames. Various environment variables can be used to customize the ACME
// client. See the [go-listener README] for details.
//
// [go-listener README]: https://pkg.go.dev/src.agwa.name/go-listener#readme-acme-configuration
func GetCertificateAutomatically(hostnames []string) GetCertificateFunc {
if hostnames == nil {
return getCertificateAutomatically(nil)
} else {
return getCertificateAutomatically(autocert.HostWhitelist(hostnames...))
}
}
|