68 lines
1.4 KiB
Go
68 lines
1.4 KiB
Go
// must pass cert-manager DNS conformance tests
|
|
package sthome
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"sync"
|
|
|
|
"github.com/cert-manager/cert-manager/pkg/acme/webhook"
|
|
acme "github.com/cert-manager/cert-manager/pkg/acme/webhook/apis/acme/v1alpha1"
|
|
"github.com/miekg/dns"
|
|
"k8s.io/client-go/rest"
|
|
)
|
|
|
|
type sthomeSolver struct {
|
|
name string
|
|
server *dns.Server
|
|
txtRecords map[string]string
|
|
sync.RWMutex
|
|
}
|
|
|
|
func (e *sthomeSolver) Name() string {
|
|
return e.name
|
|
}
|
|
|
|
func (e *sthomeSolver) Present(ch *acme.ChallengeRequest) error {
|
|
e.Lock()
|
|
e.txtRecords[ch.ResolvedFQDN] = ch.Key
|
|
e.Unlock()
|
|
return nil
|
|
}
|
|
|
|
func (e *sthomeSolver) CleanUp(ch *acme.ChallengeRequest) error {
|
|
e.Lock()
|
|
delete(e.txtRecords, ch.ResolvedFQDN)
|
|
e.Unlock()
|
|
return nil
|
|
}
|
|
|
|
func (e *sthomeSolver) Initialize(kubeClientConfig *rest.Config, stopCh <-chan struct{}) error {
|
|
go func(done <-chan struct{}) {
|
|
<-done
|
|
if err := e.server.Shutdown(); err != nil {
|
|
fmt.Fprintf(os.Stderr, "%s\n", err.Error())
|
|
}
|
|
}(stopCh)
|
|
go func() {
|
|
if err := e.server.ListenAndServe(); err != nil {
|
|
fmt.Fprintf(os.Stderr, "%s\n", err.Error())
|
|
os.Exit(1)
|
|
}
|
|
}()
|
|
return nil
|
|
}
|
|
|
|
func New(port string) webhook.Solver {
|
|
e := &sthomeSolver{
|
|
name: "sthome",
|
|
txtRecords: make(map[string]string),
|
|
}
|
|
e.server = &dns.Server{
|
|
Addr: ":" + port,
|
|
Net: "udp",
|
|
Handler: dns.HandlerFunc(e.handleDNSRequest),
|
|
}
|
|
return e
|
|
}
|