dns64/setup.go

package dns64

import (
	"net"
	"strconv"

	"github.com/coredns/coredns/core/dnsserver"
	"github.com/coredns/coredns/plugin"
	clog "github.com/coredns/coredns/plugin/pkg/log"
	"github.com/coredns/coredns/plugin/pkg/parse"
	"github.com/coredns/proxy"

	"github.com/mholt/caddy"
)

var log = clog.NewWithPlugin("dns64")

func init() {
	caddy.RegisterPlugin("dns64", caddy.Plugin{
		ServerType: "dns",
		Action:     setup,
	})
}

func setup(c *caddy.Controller) error {
	prxy, pref, translateAll, err := dns64Parse(c)
	if err != nil {
		return plugin.Error("dns64", err)
	}

	dnsserver.GetConfig(c).AddPlugin(func(next plugin.Handler) plugin.Handler {
		return DNS64{Next: next, Proxy: prxy, Prefix: pref, translateAll: translateAll}
	})

	return nil
}

func dns64Parse(c *caddy.Controller) (proxy.Proxy, *net.IPNet, bool, error) {
	prxy := proxy.Proxy{}
	_, pref, _ := net.ParseCIDR("64:ff9b::/96")
	translateAll := false

	for c.Next() {
		args := c.RemainingArgs()
		if len(args) > 0 {
			return prxy, pref, translateAll, c.ArgErr()
		}

		for c.NextBlock() {
			switch c.Val() {
			case "upstream":
				args := c.RemainingArgs()
				if len(args) == 0 {
					return prxy, pref, translateAll, c.ArgErr()
				}
				ups, err := parse.HostPortOrFile(args...)
				if err != nil {
					return prxy, pref, translateAll, err
				}
				prxy = proxy.NewLookup(ups)
				log.Infof("Upstream %v", ups)
			case "prefix":
				if !c.NextArg() {
					return prxy, pref, translateAll, c.ArgErr()
				}
				var err error
				_, pref, err = net.ParseCIDR(c.Val())

				// test for valid prefix
				n, total := pref.Mask.Size()
				if total != 128 {
					return prxy, pref, translateAll, c.Errf("'%s' not a valid IPv6 address", pref)
				}
				if n%8 != 0 || n < 32 || n > 96 {
					return prxy, pref, translateAll, c.Errf("'%s' not a valid prefix length", pref)
				}

				if err != nil {
					return prxy, pref, translateAll, err
				}
				log.Infof("Prefix %v", pref)
			case "translateAll":
				args := c.RemainingArgs()
				if len(args) > 0 {
					translateAll, _ = strconv.ParseBool(args[0])
				}
			default:
				return prxy, pref, translateAll, c.Errf("unknown property '%s'", c.Val())
			}
		}
	}
	return prxy, pref, translateAll, nil
}