155 lines
4.4 KiB
Go
155 lines
4.4 KiB
Go
|
package resolver
|
||
|
|
||
|
import (
|
||
|
"errors"
|
||
|
"fmt"
|
||
|
"sort"
|
||
|
"strconv"
|
||
|
"time"
|
||
|
|
||
|
"github.com/xenolf/lego/acme"
|
||
|
"github.com/xenolf/lego/acme/api"
|
||
|
"github.com/xenolf/lego/challenge"
|
||
|
"github.com/xenolf/lego/challenge/dns01"
|
||
|
"github.com/xenolf/lego/challenge/http01"
|
||
|
"github.com/xenolf/lego/challenge/tlsalpn01"
|
||
|
"github.com/xenolf/lego/log"
|
||
|
)
|
||
|
|
||
|
type byType []acme.Challenge
|
||
|
|
||
|
func (a byType) Len() int { return len(a) }
|
||
|
func (a byType) Swap(i, j int) { a[i], a[j] = a[j], a[i] }
|
||
|
func (a byType) Less(i, j int) bool { return a[i].Type > a[j].Type }
|
||
|
|
||
|
type SolverManager struct {
|
||
|
core *api.Core
|
||
|
solvers map[challenge.Type]solver
|
||
|
}
|
||
|
|
||
|
func NewSolversManager(core *api.Core) *SolverManager {
|
||
|
return &SolverManager{
|
||
|
solvers: map[challenge.Type]solver{},
|
||
|
core: core,
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// SetHTTP01Provider specifies a custom provider p that can solve the given HTTP-01 challenge.
|
||
|
func (c *SolverManager) SetHTTP01Provider(p challenge.Provider) error {
|
||
|
c.solvers[challenge.HTTP01] = http01.NewChallenge(c.core, validate, p)
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
// SetTLSALPN01Provider specifies a custom provider p that can solve the given TLS-ALPN-01 challenge.
|
||
|
func (c *SolverManager) SetTLSALPN01Provider(p challenge.Provider) error {
|
||
|
c.solvers[challenge.TLSALPN01] = tlsalpn01.NewChallenge(c.core, validate, p)
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
// SetDNS01Provider specifies a custom provider p that can solve the given DNS-01 challenge.
|
||
|
func (c *SolverManager) SetDNS01Provider(p challenge.Provider, opts ...dns01.ChallengeOption) error {
|
||
|
c.solvers[challenge.DNS01] = dns01.NewChallenge(c.core, validate, p, opts...)
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
// Remove Remove a challenge type from the available solvers.
|
||
|
func (c *SolverManager) Remove(chlgType challenge.Type) {
|
||
|
delete(c.solvers, chlgType)
|
||
|
}
|
||
|
|
||
|
// Checks all challenges from the server in order and returns the first matching solver.
|
||
|
func (c *SolverManager) chooseSolver(authz acme.Authorization) solver {
|
||
|
// Allow to have a deterministic challenge order
|
||
|
sort.Sort(byType(authz.Challenges))
|
||
|
|
||
|
domain := challenge.GetTargetedDomain(authz)
|
||
|
for _, chlg := range authz.Challenges {
|
||
|
if solvr, ok := c.solvers[challenge.Type(chlg.Type)]; ok {
|
||
|
log.Infof("[%s] acme: use %s solver", domain, chlg.Type)
|
||
|
return solvr
|
||
|
}
|
||
|
log.Infof("[%s] acme: Could not find solver for: %s", domain, chlg.Type)
|
||
|
}
|
||
|
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
func validate(core *api.Core, domain string, chlg acme.Challenge) error {
|
||
|
chlng, err := core.Challenges.New(chlg.URL)
|
||
|
if err != nil {
|
||
|
return fmt.Errorf("failed to initiate challenge: %v", err)
|
||
|
}
|
||
|
|
||
|
valid, err := checkChallengeStatus(chlng)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
if valid {
|
||
|
log.Infof("[%s] The server validated our request", domain)
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
// After the path is sent, the ACME server will access our server.
|
||
|
// Repeatedly check the server for an updated status on our request.
|
||
|
for {
|
||
|
authz, err := core.Authorizations.Get(chlng.AuthorizationURL)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
valid, err := checkAuthorizationStatus(authz)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
if valid {
|
||
|
log.Infof("[%s] The server validated our request", domain)
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
ra, err := strconv.Atoi(chlng.RetryAfter)
|
||
|
if err != nil {
|
||
|
// The ACME server MUST return a Retry-After.
|
||
|
// If it doesn't, we'll just poll hard.
|
||
|
// Boulder does not implement the ability to retry challenges or the Retry-After header.
|
||
|
// https://github.com/letsencrypt/boulder/blob/master/docs/acme-divergences.md#section-82
|
||
|
ra = 5
|
||
|
}
|
||
|
time.Sleep(time.Duration(ra) * time.Second)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func checkChallengeStatus(chlng acme.ExtendedChallenge) (bool, error) {
|
||
|
switch chlng.Status {
|
||
|
case acme.StatusValid:
|
||
|
return true, nil
|
||
|
case acme.StatusPending, acme.StatusProcessing:
|
||
|
return false, nil
|
||
|
case acme.StatusInvalid:
|
||
|
return false, chlng.Error
|
||
|
default:
|
||
|
return false, errors.New("the server returned an unexpected state")
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func checkAuthorizationStatus(authz acme.Authorization) (bool, error) {
|
||
|
switch authz.Status {
|
||
|
case acme.StatusValid:
|
||
|
return true, nil
|
||
|
case acme.StatusPending, acme.StatusProcessing:
|
||
|
return false, nil
|
||
|
case acme.StatusDeactivated, acme.StatusExpired, acme.StatusRevoked:
|
||
|
return false, fmt.Errorf("the authorization state %s", authz.Status)
|
||
|
case acme.StatusInvalid:
|
||
|
for _, chlg := range authz.Challenges {
|
||
|
if chlg.Status == acme.StatusInvalid && chlg.Error != nil {
|
||
|
return false, chlg.Error
|
||
|
}
|
||
|
}
|
||
|
return false, fmt.Errorf("the authorization state %s", authz.Status)
|
||
|
default:
|
||
|
return false, errors.New("the server returned an unexpected state")
|
||
|
}
|
||
|
}
|