support set chaos in config file (#11)

Author: WangXiangUSTC

handler.go

@@ -28,6 +28,10 @@ func (k Kubernetes) ServeDNS(ctx context.Context, w dns.ResponseWriter, r *dns.M
 	}
 
 	if k.IsNameError(err) {
+		if len(zone) == 0 {
+			return plugin.NextOrFailure(k.Name(), k.Next, ctx, w, r)
+		}
+
 		if k.Fall.Through(state.Name()) {
 			return plugin.NextOrFailure(k.Name(), k.Next, ctx, w, r)
 		}

kubernetes.go

@@ -278,12 +278,25 @@ func (k *Kubernetes) InitKubeCache(ctx context.Context) (err error) {
 	k.opts.endpointNameMode = k.endpointNameMode
 	k.APIConn = newdnsController(ctx, kubeClient, k.opts)
 
+	// get IP for chaos Pod
+	for _, pods := range k.podMap {
+		for name := range pods {
+			podInfo := pods[name]
+			pod, err := k.getPodFromCluster(podInfo.Namespace, podInfo.Name)
+			if err != nil {
+				return err
+			}
+			podInfo.IP = pod.Status.PodIP
+			podInfo.LastUpdateTime = time.Now()
+			k.ipPodMap[pod.Status.PodIP] = podInfo
+		}
+	}
+
 	return err
 }
 
 // Records looks up services in kubernetes.
 func (k *Kubernetes) Records(ctx context.Context, state request.Request, exact bool) ([]msg.Service, error) {
-	log.Infof("k8s Records, sourceIP: %s", state.IP())
 	r, e := parseRequest(state.Name(), state.Zone)
 	if e != nil {
 		return nil, e

setup.go

@@ -288,6 +288,34 @@ func ParseStanza(c *caddy.Controller) (*Kubernetes, error) {
 				}
 				k8s.grpcPort = port
 			}
+		case "chaos":
+			/*
+				the sample config:
+					chaos error outer busybox.busybox-0 busybox.busybox-1
+					chaos random inner busybox.busybox-2 busybox.busybox-3
+			*/
+			args := c.RemainingArgs()
+			if len(args) >= 3 {
+				for i := 2; i < len(args); i++ {
+					items := strings.SplitN(args[i], ".", 2)
+					if len(items) != 2 {
+						return nil, c.ArgErr()
+					}
+
+					if _, ok := k8s.podMap[items[0]]; !ok {
+						k8s.podMap[items[0]] = make(map[string]*PodInfo)
+					}
+					k8s.podMap[items[0]][items[1]] = &PodInfo{
+						Namespace: items[0],
+						Name:      items[1],
+						Action:    args[0],
+						Scope:     args[1],
+					}
+				}
+			} else {
+				return nil, c.ArgErr()
+			}
+
 		default:
 			return nil, c.Errf("unknown property '%s'", c.Val())
 		}