moved log to ResponseWrapper

README.md

@@ -0,0 +1,24 @@
+A plugin to override A and AAAA records with a custom value. 
+I made this to override the public IP address of my server with its VPN IP address.
+
+There is probably a better way to do this, but I couldn't find it.
+
+# Usage
+Add this to your `plugin.cfg` beetween `minimal` and `template` :
+```
+override:git.kapelle.org/niklas/coredns-override
+```
+
+In your `Corefile`:
+```
+. {
+    override {
+        12.34.56.78 10.0.0.1
+        98.76.54.32 10.0.0.2
+    }
+
+    forward . 1.1.1.1
+}
+```
+
+

ResponseWrapper.go

@@ -7,7 +7,7 @@ import (
 	"github.com/miekg/dns"
 )
 
-var log = clog.NewWithPlugin("overide")
+var log = clog.NewWithPlugin("override")
 
 type Rule struct {
 	Search   net.IP
@@ -22,6 +22,18 @@ type ResponseWrapper struct {
 func (r *ResponseWrapper) WriteMsg(res *dns.Msg) error {
 	for _, r := range r.Rules {
 		for _, rr := range res.Answer {
+			overideRR(r, rr)
+		}
+
+		for _, rr := range res.Extra {
+			overideRR(r, rr)
+		}
+	}
+
+	return r.ResponseWriter.WriteMsg(res)
+}
+
+func overideRR(r Rule, rr dns.RR) {
 	if rr.Header().Rrtype == dns.TypeA {
 		a := rr.(*dns.A)
 		if a.A.Equal(r.Search) {
@@ -33,10 +45,6 @@ func (r *ResponseWrapper) WriteMsg(res *dns.Msg) error {
 			a.AAAA = r.Override
 		}
 	}
-		}
-	}
-
-	return r.ResponseWriter.WriteMsg(res)
 }
 
 func (r *ResponseWrapper) Write(buf []byte) (int, error) {

setup.go

@@ -13,8 +13,6 @@ func init() {
 }
 
 func setup(c *caddy.Controller) error {
-	c.Next()
-
 	rules, err := parseRules(c)
 
 	if err != nil {
@@ -31,10 +29,25 @@ func setup(c *caddy.Controller) error {
 func parseRules(c *caddy.Controller) ([]Rule, error) {
 	var rules []Rule
 
-	rules = append(rules, Rule{
-		Search:   net.ParseIP("45.9.63.244"),
-		Override: net.ParseIP("10.0.0.1"),
-	})
+	for c.Next() {
+		for c.NextBlock() {
+			searchIP := net.ParseIP(c.Val())
+			if searchIP == nil {
+				return nil, c.Errf("invalid search IP: %s", c.Val())
+			}
+
+			if !c.NextArg() {
+				return nil, c.ArgErr()
+			}
+
+			overrideIP := net.ParseIP(c.Val())
+			if overrideIP == nil {
+				return nil, c.Errf("invalid override IP: %s", c.Val())
+			}
+
+			rules = append(rules, Rule{Search: searchIP, Override: overrideIP})
+		}
+	}
 
 	return rules, nil
 }