177e32b62e
add a test to see if we copy the rcode correctly. Some minor cleanup in import ordering and renaming NewUpstream to New as we already are in the upstream package.
132 lines
2.5 KiB
Go
132 lines
2.5 KiB
Go
package file
|
|
|
|
import (
|
|
"os"
|
|
"path"
|
|
|
|
"github.com/coredns/coredns/core/dnsserver"
|
|
"github.com/coredns/coredns/plugin"
|
|
"github.com/coredns/coredns/plugin/pkg/parse"
|
|
"github.com/coredns/coredns/plugin/pkg/upstream"
|
|
|
|
"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 plugin.Error("file", err)
|
|
}
|
|
|
|
// Add startup functions to notify the master(s).
|
|
for _, n := range zones.Names {
|
|
z := zones.Z[n]
|
|
c.OnStartup(func() error {
|
|
z.StartupOnce.Do(func() {
|
|
if len(z.TransferTo) > 0 {
|
|
z.Notify()
|
|
}
|
|
z.Reload()
|
|
})
|
|
return nil
|
|
})
|
|
}
|
|
for _, n := range zones.Names {
|
|
z := zones.Z[n]
|
|
c.OnShutdown(z.OnShutdown)
|
|
}
|
|
|
|
dnsserver.GetConfig(c).AddPlugin(func(next plugin.Handler) plugin.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{}
|
|
|
|
config := dnsserver.GetConfig(c)
|
|
|
|
for c.Next() {
|
|
// 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
|
|
}
|
|
|
|
if !path.IsAbs(fileName) && config.Root != "" {
|
|
fileName = path.Join(config.Root, fileName)
|
|
}
|
|
|
|
reader, err := os.Open(fileName)
|
|
if err != nil {
|
|
// bail out
|
|
return Zones{}, err
|
|
}
|
|
|
|
for i := range origins {
|
|
origins[i] = plugin.Host(origins[i]).Normalize()
|
|
zone, err := Parse(reader, origins[i], fileName, 0)
|
|
if err == nil {
|
|
z[origins[i]] = zone
|
|
} else {
|
|
return Zones{}, err
|
|
}
|
|
names = append(names, origins[i])
|
|
}
|
|
|
|
noReload := false
|
|
upstr := upstream.Upstream{}
|
|
t := []string{}
|
|
var e error
|
|
|
|
for c.NextBlock() {
|
|
switch c.Val() {
|
|
case "transfer":
|
|
t, _, e = parse.Transfer(c, false)
|
|
if e != nil {
|
|
return Zones{}, e
|
|
}
|
|
|
|
case "no_reload":
|
|
noReload = true
|
|
|
|
case "upstream":
|
|
args := c.RemainingArgs()
|
|
upstr, err = upstream.New(args)
|
|
if err != nil {
|
|
return Zones{}, err
|
|
}
|
|
|
|
default:
|
|
return Zones{}, c.Errf("unknown property '%s'", c.Val())
|
|
}
|
|
|
|
for _, origin := range origins {
|
|
if t != nil {
|
|
z[origin].TransferTo = append(z[origin].TransferTo, t...)
|
|
}
|
|
z[origin].NoReload = noReload
|
|
z[origin].Upstream = upstr
|
|
}
|
|
}
|
|
}
|
|
return Zones{Z: z, Names: names}, nil
|
|
}
|