traefik/vendor/github.com/xenolf/lego/providers/dns/iij/iij.go

240 lines
5.6 KiB
Go
Raw Normal View History

2018-09-14 08:06:03 +00:00
// Package iij implements a DNS provider for solving the DNS-01 challenge using IIJ DNS.
package iij
import (
"fmt"
2018-09-17 13:16:03 +00:00
"strconv"
2018-09-14 08:06:03 +00:00
"strings"
"time"
"github.com/iij/doapi"
"github.com/iij/doapi/protocol"
"github.com/xenolf/lego/acme"
"github.com/xenolf/lego/platform/config/env"
)
// Config is used to configure the creation of the DNSProvider
type Config struct {
2018-09-17 13:16:03 +00:00
AccessKey string
SecretKey string
DoServiceCode string
PropagationTimeout time.Duration
PollingInterval time.Duration
TTL int
}
// NewDefaultConfig returns a default configuration for the DNSProvider
func NewDefaultConfig() *Config {
return &Config{
TTL: env.GetOrDefaultInt("IIJ_TTL", 300),
PropagationTimeout: env.GetOrDefaultSecond("IIJ_PROPAGATION_TIMEOUT", 2*time.Minute),
PollingInterval: env.GetOrDefaultSecond("IIJ_POLLING_INTERVAL", 4*time.Second),
}
2018-09-14 08:06:03 +00:00
}
// DNSProvider implements the acme.ChallengeProvider interface
type DNSProvider struct {
api *doapi.API
config *Config
}
// NewDNSProvider returns a DNSProvider instance configured for IIJ DO
func NewDNSProvider() (*DNSProvider, error) {
values, err := env.Get("IIJ_API_ACCESS_KEY", "IIJ_API_SECRET_KEY", "IIJ_DO_SERVICE_CODE")
if err != nil {
2018-09-17 13:16:03 +00:00
return nil, fmt.Errorf("iij: %v", err)
2018-09-14 08:06:03 +00:00
}
2018-09-17 13:16:03 +00:00
config := NewDefaultConfig()
config.AccessKey = values["IIJ_API_ACCESS_KEY"]
config.SecretKey = values["IIJ_API_SECRET_KEY"]
config.DoServiceCode = values["IIJ_DO_SERVICE_CODE"]
return NewDNSProviderConfig(config)
2018-09-14 08:06:03 +00:00
}
2018-09-17 13:16:03 +00:00
// NewDNSProviderConfig takes a given config
// and returns a custom configured DNSProvider instance
2018-09-14 08:06:03 +00:00
func NewDNSProviderConfig(config *Config) (*DNSProvider, error) {
2018-09-17 13:16:03 +00:00
if config.SecretKey == "" || config.AccessKey == "" || config.DoServiceCode == "" {
return nil, fmt.Errorf("iij: credentials missing")
}
2018-09-14 08:06:03 +00:00
return &DNSProvider{
api: doapi.NewAPI(config.AccessKey, config.SecretKey),
config: config,
}, nil
}
// Timeout returns the timeout and interval to use when checking for DNS propagation.
2018-09-17 13:16:03 +00:00
func (d *DNSProvider) Timeout() (timeout, interval time.Duration) {
return d.config.PropagationTimeout, d.config.PollingInterval
2018-09-14 08:06:03 +00:00
}
// Present creates a TXT record using the specified parameters
2018-09-17 13:16:03 +00:00
func (d *DNSProvider) Present(domain, token, keyAuth string) error {
2018-09-14 08:06:03 +00:00
_, value, _ := acme.DNS01Record(domain, keyAuth)
2018-09-17 13:16:03 +00:00
err := d.addTxtRecord(domain, value)
2018-09-25 13:06:03 +00:00
if err != nil {
return fmt.Errorf("iij: %v", err)
}
return nil
2018-09-14 08:06:03 +00:00
}
// CleanUp removes the TXT record matching the specified parameters
2018-09-17 13:16:03 +00:00
func (d *DNSProvider) CleanUp(domain, token, keyAuth string) error {
2018-09-14 08:06:03 +00:00
_, value, _ := acme.DNS01Record(domain, keyAuth)
2018-09-17 13:16:03 +00:00
err := d.deleteTxtRecord(domain, value)
2018-09-25 13:06:03 +00:00
if err != nil {
return fmt.Errorf("iij: %v", err)
}
return nil
2018-09-14 08:06:03 +00:00
}
2018-09-17 13:16:03 +00:00
func (d *DNSProvider) addTxtRecord(domain, value string) error {
zones, err := d.listZones()
2018-09-14 08:06:03 +00:00
if err != nil {
return err
}
owner, zone, err := splitDomain(domain, zones)
if err != nil {
return err
}
request := protocol.RecordAdd{
2018-09-17 13:16:03 +00:00
DoServiceCode: d.config.DoServiceCode,
2018-09-14 08:06:03 +00:00
ZoneName: zone,
Owner: owner,
2018-09-17 13:16:03 +00:00
TTL: strconv.Itoa(d.config.TTL),
2018-09-14 08:06:03 +00:00
RecordType: "TXT",
RData: value,
}
response := &protocol.RecordAddResponse{}
2018-09-17 13:16:03 +00:00
if err := doapi.Call(*d.api, request, response); err != nil {
2018-09-14 08:06:03 +00:00
return err
}
2018-09-17 13:16:03 +00:00
return d.commit()
2018-09-14 08:06:03 +00:00
}
2018-09-17 13:16:03 +00:00
func (d *DNSProvider) deleteTxtRecord(domain, value string) error {
zones, err := d.listZones()
2018-09-14 08:06:03 +00:00
if err != nil {
return err
}
owner, zone, err := splitDomain(domain, zones)
if err != nil {
return err
}
2018-09-17 13:16:03 +00:00
id, err := d.findTxtRecord(owner, zone, value)
2018-09-14 08:06:03 +00:00
if err != nil {
return err
}
request := protocol.RecordDelete{
2018-09-17 13:16:03 +00:00
DoServiceCode: d.config.DoServiceCode,
2018-09-14 08:06:03 +00:00
ZoneName: zone,
RecordID: id,
}
response := &protocol.RecordDeleteResponse{}
2018-09-17 13:16:03 +00:00
if err := doapi.Call(*d.api, request, response); err != nil {
2018-09-14 08:06:03 +00:00
return err
}
2018-09-17 13:16:03 +00:00
return d.commit()
2018-09-14 08:06:03 +00:00
}
2018-09-17 13:16:03 +00:00
func (d *DNSProvider) commit() error {
2018-09-14 08:06:03 +00:00
request := protocol.Commit{
2018-09-17 13:16:03 +00:00
DoServiceCode: d.config.DoServiceCode,
2018-09-14 08:06:03 +00:00
}
response := &protocol.CommitResponse{}
2018-09-17 13:16:03 +00:00
return doapi.Call(*d.api, request, response)
2018-09-14 08:06:03 +00:00
}
2018-09-17 13:16:03 +00:00
func (d *DNSProvider) findTxtRecord(owner, zone, value string) (string, error) {
2018-09-14 08:06:03 +00:00
request := protocol.RecordListGet{
2018-09-17 13:16:03 +00:00
DoServiceCode: d.config.DoServiceCode,
2018-09-14 08:06:03 +00:00
ZoneName: zone,
}
response := &protocol.RecordListGetResponse{}
2018-09-17 13:16:03 +00:00
if err := doapi.Call(*d.api, request, response); err != nil {
2018-09-14 08:06:03 +00:00
return "", err
}
var id string
for _, record := range response.RecordList {
if record.Owner == owner && record.RecordType == "TXT" && record.RData == "\""+value+"\"" {
id = record.Id
}
}
if id == "" {
return "", fmt.Errorf("%s record in %s not found", owner, zone)
}
return id, nil
}
2018-09-17 13:16:03 +00:00
func (d *DNSProvider) listZones() ([]string, error) {
2018-09-14 08:06:03 +00:00
request := protocol.ZoneListGet{
2018-09-17 13:16:03 +00:00
DoServiceCode: d.config.DoServiceCode,
2018-09-14 08:06:03 +00:00
}
response := &protocol.ZoneListGetResponse{}
2018-09-17 13:16:03 +00:00
if err := doapi.Call(*d.api, request, response); err != nil {
2018-09-14 08:06:03 +00:00
return nil, err
}
return response.ZoneList, nil
}
func splitDomain(domain string, zones []string) (string, string, error) {
parts := strings.Split(strings.Trim(domain, "."), ".")
var owner string
var zone string
for i := 0; i < len(parts)-1; i++ {
zone = strings.Join(parts[i:], ".")
if zoneContains(zone, zones) {
baseOwner := strings.Join(parts[0:i], ".")
if len(baseOwner) > 0 {
baseOwner = "." + baseOwner
}
owner = "_acme-challenge" + baseOwner
break
}
}
if len(owner) == 0 {
return "", "", fmt.Errorf("%s not found", domain)
}
return owner, zone, nil
}
func zoneContains(zone string, zones []string) bool {
for _, z := range zones {
if zone == z {
return true
}
}
return false
}