| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736 |
- // Copyright (C) 2019-2023 Nicola Murino
- //
- // This program is free software: you can redistribute it and/or modify
- // it under the terms of the GNU Affero General Public License as published
- // by the Free Software Foundation, version 3.
- //
- // This program is distributed in the hope that it will be useful,
- // but WITHOUT ANY WARRANTY; without even the implied warranty of
- // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- // GNU Affero General Public License for more details.
- //
- // You should have received a copy of the GNU Affero General Public License
- // along with this program. If not, see <https://www.gnu.org/licenses/>.
- // Package acme provides automatic access to certificates from Let's Encrypt and any other ACME-based CA
- // The code here is largely coiped from https://github.com/go-acme/lego/tree/master/cmd
- // This package is intended to provide basic functionality for obtaining and renewing certificates
- // and implements the "HTTP-01" and "TLSALPN-01" challenge types.
- // For more advanced features use external tools such as "lego"
- package acme
- import (
- "crypto"
- "crypto/x509"
- "encoding/json"
- "encoding/pem"
- "errors"
- "fmt"
- "math/rand"
- "net/url"
- "os"
- "path/filepath"
- "strconv"
- "strings"
- "time"
- "github.com/go-acme/lego/v4/certcrypto"
- "github.com/go-acme/lego/v4/certificate"
- "github.com/go-acme/lego/v4/challenge"
- "github.com/go-acme/lego/v4/challenge/http01"
- "github.com/go-acme/lego/v4/challenge/tlsalpn01"
- "github.com/go-acme/lego/v4/lego"
- "github.com/go-acme/lego/v4/log"
- "github.com/go-acme/lego/v4/providers/http/webroot"
- "github.com/go-acme/lego/v4/registration"
- "github.com/robfig/cron/v3"
- "github.com/drakkan/sftpgo/v2/internal/common"
- "github.com/drakkan/sftpgo/v2/internal/ftpd"
- "github.com/drakkan/sftpgo/v2/internal/httpd"
- "github.com/drakkan/sftpgo/v2/internal/logger"
- "github.com/drakkan/sftpgo/v2/internal/telemetry"
- "github.com/drakkan/sftpgo/v2/internal/util"
- "github.com/drakkan/sftpgo/v2/internal/version"
- "github.com/drakkan/sftpgo/v2/internal/webdavd"
- )
- const (
- logSender = "acme"
- )
- var (
- config *Configuration
- scheduler *cron.Cron
- logMode int
- )
- // GetCertificates tries to obtain the certificates for the configured domains
- func GetCertificates() error {
- if config == nil {
- return errors.New("acme is disabled")
- }
- return config.getCertificates()
- }
- // HTTP01Challenge defines the configuration for HTTP-01 challenge type
- type HTTP01Challenge struct {
- Port int `json:"port" mapstructure:"port"`
- WebRoot string `json:"webroot" mapstructure:"webroot"`
- ProxyHeader string `json:"proxy_header" mapstructure:"proxy_header"`
- }
- func (c *HTTP01Challenge) isEnabled() bool {
- return c.Port > 0 || c.WebRoot != ""
- }
- func (c *HTTP01Challenge) validate() error {
- if !c.isEnabled() {
- return nil
- }
- if c.WebRoot != "" {
- if !filepath.IsAbs(c.WebRoot) {
- return fmt.Errorf("invalid HTTP-01 challenge web root, please set an absolute path")
- }
- _, err := os.Stat(c.WebRoot)
- if err != nil {
- return fmt.Errorf("invalid HTTP-01 challenge web root: %w", err)
- }
- } else {
- if c.Port > 65535 {
- return fmt.Errorf("invalid HTTP-01 challenge port: %d", c.Port)
- }
- }
- return nil
- }
- // TLSALPN01Challenge defines the configuration for TLSALPN-01 challenge type
- type TLSALPN01Challenge struct {
- Port int `json:"port" mapstructure:"port"`
- }
- func (c *TLSALPN01Challenge) isEnabled() bool {
- return c.Port > 0
- }
- func (c *TLSALPN01Challenge) validate() error {
- if !c.isEnabled() {
- return nil
- }
- if c.Port > 65535 {
- return fmt.Errorf("invalid TLSALPN-01 challenge port: %d", c.Port)
- }
- return nil
- }
- // Configuration holds the ACME configuration
- type Configuration struct {
- Email string `json:"email" mapstructure:"email"`
- KeyType string `json:"key_type" mapstructure:"key_type"`
- CertsPath string `json:"certs_path" mapstructure:"certs_path"`
- CAEndpoint string `json:"ca_endpoint" mapstructure:"ca_endpoint"`
- // if a certificate is to be valid for multiple domains specify the names separated by commas,
- // for example: example.com,www.example.com
- Domains []string `json:"domains" mapstructure:"domains"`
- RenewDays int `json:"renew_days" mapstructure:"renew_days"`
- HTTP01Challenge HTTP01Challenge `json:"http01_challenge" mapstructure:"http01_challenge"`
- TLSALPN01Challenge TLSALPN01Challenge `json:"tls_alpn01_challenge" mapstructure:"tls_alpn01_challenge"`
- accountConfigPath string
- accountKeyPath string
- lockPath string
- tempDir string
- }
- // Initialize validates and set the configuration
- func (c *Configuration) Initialize(configDir string, checkRenew bool) error {
- common.SetCertAutoReloadMode(true)
- config = nil
- setLogMode(checkRenew)
- c.checkDomains()
- if len(c.Domains) == 0 {
- acmeLog(logger.LevelInfo, "no domains configured, acme disabled")
- return nil
- }
- if c.Email == "" || !util.IsEmailValid(c.Email) {
- return fmt.Errorf("invalid email address %#v", c.Email)
- }
- if c.RenewDays < 1 {
- return fmt.Errorf("invalid number of days remaining before renewal: %d", c.RenewDays)
- }
- supportedKeyTypes := []string{
- string(certcrypto.EC256),
- string(certcrypto.EC384),
- string(certcrypto.RSA2048),
- string(certcrypto.RSA4096),
- string(certcrypto.RSA8192),
- }
- if !util.Contains(supportedKeyTypes, c.KeyType) {
- return fmt.Errorf("invalid key type %#v", c.KeyType)
- }
- caURL, err := url.Parse(c.CAEndpoint)
- if err != nil {
- return fmt.Errorf("invalid CA endopoint: %w", err)
- }
- if !util.IsFileInputValid(c.CertsPath) {
- return fmt.Errorf("invalid certs path %#v", c.CertsPath)
- }
- if !filepath.IsAbs(c.CertsPath) {
- c.CertsPath = filepath.Join(configDir, c.CertsPath)
- }
- err = os.MkdirAll(c.CertsPath, 0700)
- if err != nil {
- return fmt.Errorf("unable to create certs path %#v: %w", c.CertsPath, err)
- }
- c.tempDir = filepath.Join(c.CertsPath, "temp")
- err = os.MkdirAll(c.CertsPath, 0700)
- if err != nil {
- return fmt.Errorf("unable to create certs temp path %#v: %w", c.tempDir, err)
- }
- serverPath := strings.NewReplacer(":", "_", "/", string(os.PathSeparator)).Replace(caURL.Host)
- accountPath := filepath.Join(c.CertsPath, serverPath)
- err = os.MkdirAll(accountPath, 0700)
- if err != nil {
- return fmt.Errorf("unable to create account path %#v: %w", accountPath, err)
- }
- c.accountConfigPath = filepath.Join(accountPath, c.Email+".json")
- c.accountKeyPath = filepath.Join(accountPath, c.Email+".key")
- c.lockPath = filepath.Join(c.CertsPath, "lock")
- if err = c.validateChallenges(); err != nil {
- return err
- }
- acmeLog(logger.LevelInfo, "configured domains: %+v", c.Domains)
- common.SetCertAutoReloadMode(false)
- config = c
- if checkRenew {
- return startScheduler()
- }
- return nil
- }
- func (c *Configuration) validateChallenges() error {
- if !c.HTTP01Challenge.isEnabled() && !c.TLSALPN01Challenge.isEnabled() {
- return fmt.Errorf("no challenge type defined")
- }
- if err := c.HTTP01Challenge.validate(); err != nil {
- return err
- }
- if err := c.TLSALPN01Challenge.validate(); err != nil {
- return err
- }
- return nil
- }
- func (c *Configuration) checkDomains() {
- var domains []string
- for _, domain := range c.Domains {
- domain = strings.TrimSpace(domain)
- if domain == "" {
- continue
- }
- if d, ok := isDomainValid(domain); ok {
- domains = append(domains, d)
- }
- }
- c.Domains = util.RemoveDuplicates(domains, true)
- }
- func (c *Configuration) setLockTime() error {
- lockTime := fmt.Sprintf("%v", util.GetTimeAsMsSinceEpoch(time.Now()))
- err := os.WriteFile(c.lockPath, []byte(lockTime), 0600)
- if err != nil {
- acmeLog(logger.LevelError, "unable to save lock time to %#v: %v", c.lockPath, err)
- return fmt.Errorf("unable to save lock time: %w", err)
- }
- acmeLog(logger.LevelDebug, "lock time saved: %#v", lockTime)
- return nil
- }
- func (c *Configuration) getLockTime() (time.Time, error) {
- content, err := os.ReadFile(c.lockPath)
- if err != nil {
- if os.IsNotExist(err) {
- acmeLog(logger.LevelDebug, "lock file %#v not found", c.lockPath)
- return time.Time{}, nil
- }
- acmeLog(logger.LevelError, "unable to read lock file %#v: %v", c.lockPath, err)
- return time.Time{}, err
- }
- msec, err := strconv.ParseInt(strings.TrimSpace(string(content)), 10, 64)
- if err != nil {
- acmeLog(logger.LevelError, "unable to parse lock time: %v", err)
- return time.Time{}, fmt.Errorf("unable to parse lock time: %w", err)
- }
- return util.GetTimeFromMsecSinceEpoch(msec), nil
- }
- func (c *Configuration) saveAccount(account *account) error {
- jsonBytes, err := json.MarshalIndent(account, "", "\t")
- if err != nil {
- return err
- }
- err = os.WriteFile(c.accountConfigPath, jsonBytes, 0600)
- if err != nil {
- acmeLog(logger.LevelError, "unable to save account to file %#v: %v", c.accountConfigPath, err)
- return fmt.Errorf("unable to save account: %w", err)
- }
- return nil
- }
- func (c *Configuration) getAccount(privateKey crypto.PrivateKey) (account, error) {
- _, err := os.Stat(c.accountConfigPath)
- if err != nil && os.IsNotExist(err) {
- acmeLog(logger.LevelDebug, "account does not exist")
- return account{Email: c.Email, key: privateKey}, nil
- }
- var account account
- fileBytes, err := os.ReadFile(c.accountConfigPath)
- if err != nil {
- acmeLog(logger.LevelError, "unable to read account from file %#v: %v", c.accountConfigPath, err)
- return account, fmt.Errorf("unable to read account from file: %w", err)
- }
- err = json.Unmarshal(fileBytes, &account)
- if err != nil {
- acmeLog(logger.LevelError, "invalid account file content: %v", err)
- return account, fmt.Errorf("unable to parse account file as JSON: %w", err)
- }
- account.key = privateKey
- if account.Registration == nil || account.Registration.Body.Status == "" {
- acmeLog(logger.LevelInfo, "couldn't load account but got a key. Try to look the account up")
- reg, err := c.tryRecoverRegistration(privateKey)
- if err != nil {
- acmeLog(logger.LevelError, "unable to look the account up: %v", err)
- return account, fmt.Errorf("unable to look the account up: %w", err)
- }
- account.Registration = reg
- err = c.saveAccount(&account)
- if err != nil {
- return account, err
- }
- }
- return account, nil
- }
- func (c *Configuration) loadPrivateKey() (crypto.PrivateKey, error) {
- keyBytes, err := os.ReadFile(c.accountKeyPath)
- if err != nil {
- acmeLog(logger.LevelError, "unable to read account key from file %#v: %v", c.accountKeyPath, err)
- return nil, fmt.Errorf("unable to read account key: %w", err)
- }
- keyBlock, _ := pem.Decode(keyBytes)
- var privateKey crypto.PrivateKey
- switch keyBlock.Type {
- case "RSA PRIVATE KEY":
- privateKey, err = x509.ParsePKCS1PrivateKey(keyBlock.Bytes)
- case "EC PRIVATE KEY":
- privateKey, err = x509.ParseECPrivateKey(keyBlock.Bytes)
- default:
- err = fmt.Errorf("unknown private key type %#v", keyBlock.Type)
- }
- if err != nil {
- acmeLog(logger.LevelError, "unable to parse private key from file %#v: %v", c.accountKeyPath, err)
- return privateKey, fmt.Errorf("unable to parse private key: %w", err)
- }
- return privateKey, nil
- }
- func (c *Configuration) generatePrivateKey() (crypto.PrivateKey, error) {
- privateKey, err := certcrypto.GeneratePrivateKey(certcrypto.KeyType(c.KeyType))
- if err != nil {
- acmeLog(logger.LevelError, "unable to generate private key: %v", err)
- return nil, fmt.Errorf("unable to generate private key: %w", err)
- }
- certOut, err := os.Create(c.accountKeyPath)
- if err != nil {
- acmeLog(logger.LevelError, "unable to save private key to file %#v: %v", c.accountKeyPath, err)
- return nil, fmt.Errorf("unable to save private key: %w", err)
- }
- defer certOut.Close()
- pemKey := certcrypto.PEMBlock(privateKey)
- err = pem.Encode(certOut, pemKey)
- if err != nil {
- acmeLog(logger.LevelError, "unable to encode private key: %v", err)
- return nil, fmt.Errorf("unable to encode private key: %w", err)
- }
- acmeLog(logger.LevelDebug, "new account private key generated")
- return privateKey, nil
- }
- func (c *Configuration) getPrivateKey() (crypto.PrivateKey, error) {
- _, err := os.Stat(c.accountKeyPath)
- if err != nil && os.IsNotExist(err) {
- acmeLog(logger.LevelDebug, "private key file %#v does not exist, generating new private key", c.accountKeyPath)
- return c.generatePrivateKey()
- }
- acmeLog(logger.LevelDebug, "loading private key from file %#v, stat error: %v", c.accountKeyPath, err)
- return c.loadPrivateKey()
- }
- func (c *Configuration) loadCertificatesForDomain(domain string) ([]*x509.Certificate, error) {
- domain = sanitizedDomain(domain)
- acmeLog(logger.LevelDebug, "loading certificates for domain %#v", domain)
- content, err := os.ReadFile(filepath.Join(c.CertsPath, domain+".crt"))
- if err != nil {
- acmeLog(logger.LevelError, "unable to load certificates for domain %#v: %v", domain, err)
- return nil, fmt.Errorf("unable to load certificates for domain %#v: %w", domain, err)
- }
- certs, err := certcrypto.ParsePEMBundle(content)
- if err != nil {
- acmeLog(logger.LevelError, "unable to parse certificates for domain %#v: %v", domain, err)
- return certs, fmt.Errorf("unable to parse certificates for domain %#v: %w", domain, err)
- }
- return certs, nil
- }
- func (c *Configuration) needRenewal(x509Cert *x509.Certificate, domain string) bool {
- if x509Cert.IsCA {
- acmeLog(logger.LevelError, "certificate bundle starts with a CA certificate, cannot renew domain %v", domain)
- return false
- }
- notAfter := int(time.Until(x509Cert.NotAfter).Hours() / 24.0)
- if notAfter > c.RenewDays {
- acmeLog(logger.LevelDebug, "the certificate for domain %#v expires in %d days, no renewal", domain, notAfter)
- return false
- }
- return true
- }
- func (c *Configuration) setup() (*account, *lego.Client, error) {
- privateKey, err := c.getPrivateKey()
- if err != nil {
- return nil, nil, err
- }
- account, err := c.getAccount(privateKey)
- if err != nil {
- return nil, nil, err
- }
- config := lego.NewConfig(&account)
- config.CADirURL = c.CAEndpoint
- config.Certificate.KeyType = certcrypto.KeyType(c.KeyType)
- config.UserAgent = fmt.Sprintf("SFTPGo/%v", version.Get().Version)
- client, err := lego.NewClient(config)
- if err != nil {
- acmeLog(logger.LevelError, "unable to get ACME client: %v", err)
- return nil, nil, fmt.Errorf("unable to get ACME client: %w", err)
- }
- err = c.setupChalleges(client)
- if err != nil {
- return nil, nil, err
- }
- return &account, client, nil
- }
- func (c *Configuration) setupChalleges(client *lego.Client) error {
- client.Challenge.Remove(challenge.DNS01)
- if c.HTTP01Challenge.isEnabled() {
- if c.HTTP01Challenge.WebRoot != "" {
- acmeLog(logger.LevelDebug, "configuring HTTP-01 web root challenge, path %#v", c.HTTP01Challenge.WebRoot)
- providerServer, err := webroot.NewHTTPProvider(c.HTTP01Challenge.WebRoot)
- if err != nil {
- acmeLog(logger.LevelError, "unable to create HTTP-01 web root challenge provider from path %#v: %v",
- c.HTTP01Challenge.WebRoot, err)
- return fmt.Errorf("unable to create HTTP-01 web root challenge provider: %w", err)
- }
- err = client.Challenge.SetHTTP01Provider(providerServer)
- if err != nil {
- acmeLog(logger.LevelError, "unable to set HTTP-01 challenge provider: %v", err)
- return fmt.Errorf("unable to set HTTP-01 challenge provider: %w", err)
- }
- } else {
- acmeLog(logger.LevelDebug, "configuring HTTP-01 challenge, port %d", c.HTTP01Challenge.Port)
- providerServer := http01.NewProviderServer("", fmt.Sprintf("%d", c.HTTP01Challenge.Port))
- if c.HTTP01Challenge.ProxyHeader != "" {
- acmeLog(logger.LevelDebug, "setting proxy header to \"%s\"", c.HTTP01Challenge.ProxyHeader)
- providerServer.SetProxyHeader(c.HTTP01Challenge.ProxyHeader)
- }
- err := client.Challenge.SetHTTP01Provider(providerServer)
- if err != nil {
- acmeLog(logger.LevelError, "unable to set HTTP-01 challenge provider: %v", err)
- return fmt.Errorf("unable to set HTTP-01 challenge provider: %w", err)
- }
- }
- } else {
- client.Challenge.Remove(challenge.HTTP01)
- }
- if c.TLSALPN01Challenge.isEnabled() {
- acmeLog(logger.LevelDebug, "configuring TLSALPN-01 challenge, port %d", c.TLSALPN01Challenge.Port)
- err := client.Challenge.SetTLSALPN01Provider(tlsalpn01.NewProviderServer("", fmt.Sprintf("%d", c.TLSALPN01Challenge.Port)))
- if err != nil {
- acmeLog(logger.LevelError, "unable to set TLSALPN-01 challenge provider: %v", err)
- return fmt.Errorf("unable to set TLSALPN-01 challenge provider: %w", err)
- }
- } else {
- client.Challenge.Remove(challenge.TLSALPN01)
- }
- return nil
- }
- func (c *Configuration) register(client *lego.Client) (*registration.Resource, error) {
- return client.Registration.Register(registration.RegisterOptions{TermsOfServiceAgreed: true})
- }
- func (c *Configuration) tryRecoverRegistration(privateKey crypto.PrivateKey) (*registration.Resource, error) {
- config := lego.NewConfig(&account{key: privateKey})
- config.CADirURL = c.CAEndpoint
- config.UserAgent = fmt.Sprintf("SFTPGo/%v", version.Get().Version)
- client, err := lego.NewClient(config)
- if err != nil {
- acmeLog(logger.LevelError, "unable to get the ACME client: %v", err)
- return nil, err
- }
- return client.Registration.ResolveAccountByKey()
- }
- func (c *Configuration) obtainAndSaveCertificate(client *lego.Client, domain string) error {
- domains := getDomains(domain)
- acmeLog(logger.LevelInfo, "requesting certificates for domains %+v", domains)
- request := certificate.ObtainRequest{
- Domains: domains,
- Bundle: true,
- MustStaple: false,
- PreferredChain: "",
- AlwaysDeactivateAuthorizations: false,
- }
- cert, err := client.Certificate.Obtain(request)
- if err != nil {
- acmeLog(logger.LevelError, "unable to obtain certificates for domains %+v: %v", domains, err)
- return fmt.Errorf("unable to obtain certificates: %w", err)
- }
- domain = sanitizedDomain(domain)
- err = os.WriteFile(filepath.Join(c.CertsPath, domain+".crt"), cert.Certificate, 0600)
- if err != nil {
- acmeLog(logger.LevelError, "unable to save certificate for domain %v: %v", domain, err)
- return fmt.Errorf("unable to save certificate: %w", err)
- }
- err = os.WriteFile(filepath.Join(c.CertsPath, domain+".key"), cert.PrivateKey, 0600)
- if err != nil {
- acmeLog(logger.LevelError, "unable to save private key for domain %v: %v", domain, err)
- return fmt.Errorf("unable to save private key: %w", err)
- }
- jsonBytes, err := json.MarshalIndent(cert, "", "\t")
- if err != nil {
- acmeLog(logger.LevelError, "unable to marshal certificate resources for domain %v: %v", domain, err)
- return err
- }
- err = os.WriteFile(filepath.Join(c.CertsPath, domain+".json"), jsonBytes, 0600)
- if err != nil {
- acmeLog(logger.LevelError, "unable to save certificate resources for domain %v: %v", domain, err)
- return fmt.Errorf("unable to save certificate resources: %w", err)
- }
- acmeLog(logger.LevelInfo, "certificates for domains %+v saved", domains)
- return nil
- }
- func (c *Configuration) getCertificates() error {
- account, client, err := c.setup()
- if err != nil {
- return err
- }
- if account.Registration == nil {
- reg, err := c.register(client)
- if err != nil {
- acmeLog(logger.LevelError, "unable to register account: %v", err)
- return fmt.Errorf("unable to register account: %w", err)
- }
- account.Registration = reg
- err = c.saveAccount(account)
- if err != nil {
- return err
- }
- }
- for _, domain := range c.Domains {
- err = c.obtainAndSaveCertificate(client, domain)
- if err != nil {
- return err
- }
- }
- return nil
- }
- func (c *Configuration) notifyCertificateRenewal(domain string, err error) {
- if domain == "" {
- domain = strings.Join(c.Domains, ",")
- }
- params := common.EventParams{
- Name: domain,
- Event: "Certificate renewal",
- Timestamp: time.Now().UnixNano(),
- }
- if err != nil {
- params.Status = 2
- params.AddError(err)
- } else {
- params.Status = 1
- }
- common.HandleCertificateEvent(params)
- }
- func (c *Configuration) renewCertificates() error {
- lockTime, err := c.getLockTime()
- if err != nil {
- return err
- }
- acmeLog(logger.LevelDebug, "certificate renew lock time %v", lockTime)
- if lockTime.Add(-30*time.Second).Before(time.Now()) && lockTime.Add(5*time.Minute).After(time.Now()) {
- acmeLog(logger.LevelInfo, "certificate renew skipped, lock time too close: %v", lockTime)
- return nil
- }
- err = c.setLockTime()
- if err != nil {
- c.notifyCertificateRenewal("", err)
- return err
- }
- account, client, err := c.setup()
- if err != nil {
- c.notifyCertificateRenewal("", err)
- return err
- }
- if account.Registration == nil {
- acmeLog(logger.LevelError, "cannot renew certificates, your account is not registered")
- err = errors.New("cannot renew certificates, your account is not registered")
- c.notifyCertificateRenewal("", err)
- return err
- }
- var errRenew error
- needReload := false
- for _, domain := range c.Domains {
- certificates, err := c.loadCertificatesForDomain(domain)
- if err != nil {
- c.notifyCertificateRenewal(domain, err)
- errRenew = err
- continue
- }
- cert := certificates[0]
- if !c.needRenewal(cert, domain) {
- continue
- }
- err = c.obtainAndSaveCertificate(client, domain)
- if err != nil {
- c.notifyCertificateRenewal(domain, err)
- errRenew = err
- } else {
- c.notifyCertificateRenewal(domain, nil)
- needReload = true
- }
- }
- if needReload {
- // at least one certificate has been renewed, sends a reload to all services that may be using certificates
- err = ftpd.ReloadCertificateMgr()
- acmeLog(logger.LevelInfo, "ftpd certificate manager reloaded , error: %v", err)
- err = httpd.ReloadCertificateMgr()
- acmeLog(logger.LevelInfo, "httpd certificates manager reloaded , error: %v", err)
- err = webdavd.ReloadCertificateMgr()
- acmeLog(logger.LevelInfo, "webdav certificates manager reloaded , error: %v", err)
- err = telemetry.ReloadCertificateMgr()
- acmeLog(logger.LevelInfo, "telemetry certificates manager reloaded , error: %v", err)
- }
- return errRenew
- }
- func isDomainValid(domain string) (string, bool) {
- isValid := false
- for _, d := range strings.Split(domain, ",") {
- d = strings.TrimSpace(d)
- if d != "" {
- isValid = true
- break
- }
- }
- return domain, isValid
- }
- func getDomains(domain string) []string {
- var domains []string
- delimiter := ","
- if !strings.Contains(domain, ",") && strings.Contains(domain, " ") {
- delimiter = " "
- }
- for _, d := range strings.Split(domain, delimiter) {
- d = strings.TrimSpace(d)
- if d != "" {
- domains = append(domains, d)
- }
- }
- return domains
- }
- func sanitizedDomain(domain string) string {
- return strings.NewReplacer(":", "_", "*", "_", ",", "_", " ", "_").Replace(domain)
- }
- func stopScheduler() {
- if scheduler != nil {
- scheduler.Stop()
- scheduler = nil
- }
- }
- func startScheduler() error {
- stopScheduler()
- randSecs := rand.Intn(59)
- scheduler = cron.New(cron.WithLocation(time.UTC), cron.WithLogger(cron.DiscardLogger))
- _, err := scheduler.AddFunc(fmt.Sprintf("@every 12h0m%ds", randSecs), renewCertificates)
- if err != nil {
- return fmt.Errorf("unable to schedule certificates renewal: %w", err)
- }
- acmeLog(logger.LevelInfo, "starting scheduler, initial certificates check in %d seconds", randSecs)
- initialTimer := time.NewTimer(time.Duration(randSecs) * time.Second)
- go func() {
- <-initialTimer.C
- renewCertificates()
- }()
- scheduler.Start()
- return nil
- }
- func renewCertificates() {
- if config != nil {
- if err := config.renewCertificates(); err != nil {
- acmeLog(logger.LevelError, "unable to renew certificates: %v", err)
- }
- }
- }
- func setLogMode(checkRenew bool) {
- if checkRenew {
- logMode = 1
- } else {
- logMode = 2
- }
- log.Logger = &logger.LegoAdapter{
- LogToConsole: logMode != 1,
- }
- }
- func acmeLog(level logger.LogLevel, format string, v ...any) {
- if logMode == 1 {
- logger.Log(level, logSender, "", format, v...)
- } else {
- switch level {
- case logger.LevelDebug:
- logger.DebugToConsole(format, v...)
- case logger.LevelInfo:
- logger.InfoToConsole(format, v...)
- case logger.LevelWarn:
- logger.WarnToConsole(format, v...)
- default:
- logger.ErrorToConsole(format, v...)
- }
- }
- }
|