duckdns: fix subsubdomain (#676)

This commit is contained in:
Nick Maliwacki 2018-10-16 12:28:49 -07:00 committed by Ludovic Fernandez
parent 4d21f8eec1
commit 6de343314c
2 changed files with 94 additions and 3 deletions

View file

@ -7,8 +7,12 @@ import (
"fmt" "fmt"
"io/ioutil" "io/ioutil"
"net/http" "net/http"
"net/url"
"strconv"
"strings"
"time" "time"
"github.com/miekg/dns"
"github.com/xenolf/lego/acme" "github.com/xenolf/lego/acme"
"github.com/xenolf/lego/platform/config/env" "github.com/xenolf/lego/platform/config/env"
) )
@ -96,9 +100,16 @@ func (d *DNSProvider) Timeout() (timeout, interval time.Duration) {
// To update the TXT record we just need to make one simple get request. // To update the TXT record we just need to make one simple get request.
// In DuckDNS you only have one TXT record shared with the domain and all sub domains. // In DuckDNS you only have one TXT record shared with the domain and all sub domains.
func updateTxtRecord(domain, token, txt string, clear bool) error { func updateTxtRecord(domain, token, txt string, clear bool) error {
u := fmt.Sprintf("https://www.duckdns.org/update?domains=%s&token=%s&clear=%t&txt=%s", domain, token, clear, txt) u, _ := url.Parse("https://www.duckdns.org/update")
response, err := acme.HTTPClient.Get(u) query := u.Query()
query.Set("domains", getMainDomain(domain))
query.Set("token", token)
query.Set("clear", strconv.FormatBool(clear))
query.Set("txt", txt)
u.RawQuery = query.Encode()
response, err := acme.HTTPClient.Get(u.String())
if err != nil { if err != nil {
return err return err
} }
@ -115,3 +126,23 @@ func updateTxtRecord(domain, token, txt string, clear bool) error {
} }
return nil return nil
} }
// DuckDNS only lets you write to your subdomain
// so it must be in format subdomain.duckdns.org
// not in format subsubdomain.subdomain.duckdns.org
// so strip off everything that is not top 3 levels
func getMainDomain(domain string) string {
domain = acme.UnFqdn(domain)
split := dns.Split(domain)
if strings.HasSuffix(strings.ToLower(domain), "duckdns.org") {
if len(split) < 3 {
return ""
}
firstSubDomainIndex := split[len(split)-3]
return domain[firstSubDomainIndex:]
}
return domain[split[len(split)-1]:]
}

View file

@ -4,6 +4,7 @@ import (
"testing" "testing"
"time" "time"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require" "github.com/stretchr/testify/require"
"github.com/xenolf/lego/platform/tester" "github.com/xenolf/lego/platform/tester"
) )
@ -86,6 +87,65 @@ func TestNewDNSProviderConfig(t *testing.T) {
} }
} }
func Test_getMainDomain(t *testing.T) {
testCases := []struct {
desc string
domain string
expected string
}{
{
desc: "empty",
domain: "",
expected: "",
},
{
desc: "missing sub domain",
domain: "duckdns.org",
expected: "",
},
{
desc: "explicit domain: sub domain",
domain: "sub.duckdns.org",
expected: "sub.duckdns.org",
},
{
desc: "explicit domain: subsub domain",
domain: "my.sub.duckdns.org",
expected: "sub.duckdns.org",
},
{
desc: "explicit domain: subsubsub domain",
domain: "my.sub.sub.duckdns.org",
expected: "sub.duckdns.org",
},
{
desc: "only subname: sub domain",
domain: "sub",
expected: "sub",
},
{
desc: "only subname: subsub domain",
domain: "my.sub",
expected: "sub",
},
{
desc: "only subname: subsubsub domain",
domain: "my.sub.sub",
expected: "sub",
},
}
for _, test := range testCases {
test := test
t.Run(test.desc, func(t *testing.T) {
t.Parallel()
wDomain := getMainDomain(test.domain)
assert.Equal(t, test.expected, wDomain)
})
}
}
func TestLivePresent(t *testing.T) { func TestLivePresent(t *testing.T) {
if !envTest.IsLiveTest() { if !envTest.IsLiveTest() {
t.Skip("skipping live test") t.Skip("skipping live test")
@ -108,7 +168,7 @@ func TestLiveCleanUp(t *testing.T) {
provider, err := NewDNSProvider() provider, err := NewDNSProvider()
require.NoError(t, err) require.NoError(t, err)
time.Sleep(10 * time.Second) time.Sleep(1 * time.Second)
err = provider.CleanUp(envTest.GetDomain(), "", "123d==") err = provider.CleanUp(envTest.GetDomain(), "", "123d==")
require.NoError(t, err) require.NoError(t, err)