4ae29a449c
* geoip: read source IP from EDNS0 subnet if provided This patch implements EDNS backend processing (similar in powerdns: https://doc.powerdns.com/authoritative/settings.html#setting-edns-subnet-processing). This feature comes very handy to test whether your geo config is working properly. Signed-off-by: Balazs Nagy <julsevern@gmail.com>
57 lines
1.1 KiB
Go
57 lines
1.1 KiB
Go
package geoip
|
|
|
|
import (
|
|
"github.com/coredns/caddy"
|
|
"github.com/coredns/coredns/core/dnsserver"
|
|
"github.com/coredns/coredns/plugin"
|
|
)
|
|
|
|
const pluginName = "geoip"
|
|
|
|
func init() { plugin.Register(pluginName, setup) }
|
|
|
|
func setup(c *caddy.Controller) error {
|
|
geoip, err := geoipParse(c)
|
|
if err != nil {
|
|
return plugin.Error(pluginName, err)
|
|
}
|
|
|
|
dnsserver.GetConfig(c).AddPlugin(func(next plugin.Handler) plugin.Handler {
|
|
geoip.Next = next
|
|
return geoip
|
|
})
|
|
|
|
return nil
|
|
}
|
|
|
|
func geoipParse(c *caddy.Controller) (*GeoIP, error) {
|
|
var dbPath string
|
|
var edns0 bool
|
|
|
|
for c.Next() {
|
|
if !c.NextArg() {
|
|
return nil, c.ArgErr()
|
|
}
|
|
if dbPath != "" {
|
|
return nil, c.Errf("configuring multiple databases is not supported")
|
|
}
|
|
dbPath = c.Val()
|
|
// There shouldn't be any more arguments.
|
|
if len(c.RemainingArgs()) != 0 {
|
|
return nil, c.ArgErr()
|
|
}
|
|
|
|
for c.NextBlock() {
|
|
if c.Val() != "edns-subnet" {
|
|
return nil, c.Errf("unknown property %q", c.Val())
|
|
}
|
|
edns0 = true
|
|
}
|
|
}
|
|
|
|
geoIP, err := newGeoIP(dbPath, edns0)
|
|
if err != nil {
|
|
return geoIP, c.Err(err.Error())
|
|
}
|
|
return geoIP, nil
|
|
}
|