2dd8a687b3
Stop the caddy message and start our own init notifications. Log the version of CoreDNS when starting up. Fix all middleware's setup functions so that return the error prefixed with *which* middleware was failing; leads to better debuggable errors when starting up.
143 lines
3 KiB
Go
143 lines
3 KiB
Go
package file
|
|
|
|
import (
|
|
"fmt"
|
|
"net"
|
|
"os"
|
|
|
|
"github.com/miekg/coredns/core/dnsserver"
|
|
"github.com/miekg/coredns/middleware"
|
|
|
|
"github.com/mholt/caddy"
|
|
)
|
|
|
|
func init() {
|
|
caddy.RegisterPlugin("file", caddy.Plugin{
|
|
ServerType: "dns",
|
|
Action: setup,
|
|
})
|
|
}
|
|
|
|
func setup(c *caddy.Controller) error {
|
|
zones, err := fileParse(c)
|
|
if err != nil {
|
|
return middleware.Error("file", err)
|
|
}
|
|
|
|
// Add startup functions to notify the master(s).
|
|
for _, n := range zones.Names {
|
|
c.OnStartup(func() error {
|
|
zones.Z[n].StartupOnce.Do(func() {
|
|
if len(zones.Z[n].TransferTo) > 0 {
|
|
zones.Z[n].Notify()
|
|
}
|
|
zones.Z[n].Reload(nil)
|
|
})
|
|
return nil
|
|
})
|
|
}
|
|
|
|
dnsserver.GetConfig(c).AddMiddleware(func(next dnsserver.Handler) dnsserver.Handler {
|
|
return File{Next: next, Zones: zones}
|
|
})
|
|
|
|
return nil
|
|
}
|
|
|
|
func fileParse(c *caddy.Controller) (Zones, error) {
|
|
z := make(map[string]*Zone)
|
|
names := []string{}
|
|
origins := []string{}
|
|
|
|
for c.Next() {
|
|
if c.Val() == "file" {
|
|
// file db.file [zones...]
|
|
if !c.NextArg() {
|
|
return Zones{}, c.ArgErr()
|
|
}
|
|
fileName := c.Val()
|
|
|
|
origins = make([]string, len(c.ServerBlockKeys))
|
|
copy(origins, c.ServerBlockKeys)
|
|
args := c.RemainingArgs()
|
|
if len(args) > 0 {
|
|
origins = args
|
|
}
|
|
|
|
reader, err := os.Open(fileName)
|
|
if err != nil {
|
|
// bail out
|
|
return Zones{}, err
|
|
}
|
|
|
|
for i, _ := range origins {
|
|
origins[i] = middleware.Host(origins[i]).Normalize()
|
|
zone, err := Parse(reader, origins[i], fileName)
|
|
if err == nil {
|
|
z[origins[i]] = zone
|
|
} else {
|
|
return Zones{}, err
|
|
}
|
|
names = append(names, origins[i])
|
|
}
|
|
|
|
noReload := false
|
|
for c.NextBlock() {
|
|
t, _, e := TransferParse(c)
|
|
if e != nil {
|
|
return Zones{}, e
|
|
}
|
|
switch c.Val() {
|
|
case "no_reload":
|
|
noReload = true
|
|
}
|
|
// discard from, here, maybe check and show log when we do?
|
|
for _, origin := range origins {
|
|
if t != nil {
|
|
z[origin].TransferTo = append(z[origin].TransferTo, t...)
|
|
}
|
|
z[origin].NoReload = noReload
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return Zones{Z: z, Names: names}, nil
|
|
}
|
|
|
|
// TransferParse parses transfer statements: 'transfer to [address...]'.
|
|
// Exported so secondary can use this as well.
|
|
func TransferParse(c *caddy.Controller) (tos, froms []string, err error) {
|
|
what := c.Val()
|
|
if !c.NextArg() {
|
|
return nil, nil, c.ArgErr()
|
|
}
|
|
value := c.Val()
|
|
switch what {
|
|
case "transfer":
|
|
if value == "to" {
|
|
tos = c.RemainingArgs()
|
|
for i, _ := range tos {
|
|
if tos[i] != "*" {
|
|
if x := net.ParseIP(tos[i]); x == nil {
|
|
return nil, nil, fmt.Errorf("must specify an IP addres: `%s'", tos[i])
|
|
}
|
|
tos[i] = middleware.Addr(tos[i]).Normalize()
|
|
}
|
|
}
|
|
}
|
|
if value == "from" {
|
|
froms = c.RemainingArgs()
|
|
for i, _ := range froms {
|
|
if froms[i] != "*" {
|
|
if x := net.ParseIP(froms[i]); x == nil {
|
|
return nil, nil, fmt.Errorf("must specify an IP addres: `%s'", froms[i])
|
|
}
|
|
froms[i] = middleware.Addr(froms[i]).Normalize()
|
|
} else {
|
|
return nil, nil, fmt.Errorf("can't use '*' in transfer from")
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return
|
|
}
|