forked from TrueCloudLab/lego
74 lines
1.7 KiB
Go
74 lines
1.7 KiB
Go
package acme
|
|
|
|
import (
|
|
"fmt"
|
|
"net"
|
|
"net/http"
|
|
"strings"
|
|
)
|
|
|
|
type httpChallenge struct {
|
|
jws *jws
|
|
validate validateFunc
|
|
iface string
|
|
port string
|
|
done chan bool
|
|
}
|
|
|
|
func (s *httpChallenge) Solve(chlng challenge, domain string) error {
|
|
|
|
logf("[INFO][%s] acme: Trying to solve HTTP-01", domain)
|
|
|
|
s.done = make(chan bool)
|
|
|
|
// Generate the Key Authorization for the challenge
|
|
keyAuth, err := getKeyAuthorization(chlng.Token, &s.jws.privKey.PublicKey)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Allow for CLI port override
|
|
port := "80"
|
|
if s.port != "" {
|
|
port = s.port
|
|
}
|
|
|
|
iface := ""
|
|
if s.iface != "" {
|
|
iface = s.iface
|
|
}
|
|
|
|
listener, err := net.Listen("tcp", net.JoinHostPort(iface, port))
|
|
if err != nil {
|
|
return fmt.Errorf("Could not start HTTP server for challenge -> %v", err)
|
|
}
|
|
|
|
path := "/.well-known/acme-challenge/" + chlng.Token
|
|
|
|
go s.serve(listener, path, keyAuth, domain)
|
|
|
|
err = s.validate(s.jws, domain, chlng.URI, challenge{Resource: "challenge", Type: chlng.Type, Token: chlng.Token, KeyAuthorization: keyAuth})
|
|
listener.Close()
|
|
<-s.done
|
|
|
|
return err
|
|
}
|
|
|
|
func (s *httpChallenge) serve(listener net.Listener, path, keyAuth, domain string) {
|
|
// The handler validates the HOST header and request type.
|
|
// For validation it then writes the token the server returned with the challenge
|
|
mux := http.NewServeMux()
|
|
mux.HandleFunc(path, func(w http.ResponseWriter, r *http.Request) {
|
|
if strings.HasPrefix(r.Host, domain) && r.Method == "GET" {
|
|
w.Header().Add("Content-Type", "text/plain")
|
|
w.Write([]byte(keyAuth))
|
|
logf("[INFO][%s] Served key authentication", domain)
|
|
} else {
|
|
logf("[INFO] Received request for domain %s with method %s", r.Host, r.Method)
|
|
w.Write([]byte("TEST"))
|
|
}
|
|
})
|
|
|
|
http.Serve(listener, mux)
|
|
s.done <- true
|
|
}
|