zabov/dns_handler.go

package main

import (
	"fmt"
	"log"
	"net"
	"os"
	"path"
	"strings"
	"time"

	"github.com/miekg/dns"
)

var reqTypes map[uint16]string

var weekdays []string

type logItem struct {
	clientIP  string
	name      string
	reqType   uint16
	config    string
	timetable string
	killed    string
}

// logChannel used by logging thread
var logChannel chan logItem

func init() {

	weekdays = []string{"Su", "Mo", "Tu", "We", "Th", "Fr", "Sa"}

	if len(ZabovDebugDBPath) > 0 {
		os.MkdirAll(ZabovDebugDBPath, 0755)
	}

	reqTypes = map[uint16]string{
		dns.TypeNone:       "TypeNone",
		dns.TypeA:          "TypeA",
		dns.TypeNS:         "TypeNS",
		dns.TypeMD:         "TypeMD",
		dns.TypeMF:         "TypeMF",
		dns.TypeCNAME:      "TypeCNAME",
		dns.TypeSOA:        "TypeSOA",
		dns.TypeMB:         "TypeMB",
		dns.TypeMG:         "TypeMG",
		dns.TypeMR:         "TypeMR",
		dns.TypeNULL:       "TypeNULL",
		dns.TypePTR:        "TypePTR",
		dns.TypeHINFO:      "TypeHINFO",
		dns.TypeMINFO:      "TypeMINFO",
		dns.TypeMX:         "TypeMX",
		dns.TypeTXT:        "TypeTXT",
		dns.TypeRP:         "TypeRP",
		dns.TypeAFSDB:      "TypeAFSDB",
		dns.TypeX25:        "TypeX25",
		dns.TypeISDN:       "TypeISDN",
		dns.TypeRT:         "TypeRT",
		dns.TypeNSAPPTR:    "TypeNSAPPTR",
		dns.TypeSIG:        "TypeSIG",
		dns.TypeKEY:        "TypeKEY",
		dns.TypePX:         "TypePX",
		dns.TypeGPOS:       "TypeGPOS",
		dns.TypeAAAA:       "TypeAAAA",
		dns.TypeLOC:        "TypeLOC",
		dns.TypeNXT:        "TypeNXT",
		dns.TypeEID:        "TypeEID",
		dns.TypeNIMLOC:     "TypeNIMLOC",
		dns.TypeSRV:        "TypeSRV",
		dns.TypeATMA:       "TypeATMA",
		dns.TypeNAPTR:      "TypeNAPTR",
		dns.TypeKX:         "TypeKX",
		dns.TypeCERT:       "TypeCERT",
		dns.TypeDNAME:      "TypeDNAME",
		dns.TypeOPT:        "TypeOPT",
		dns.TypeAPL:        "TypeAPL",
		dns.TypeDS:         "TypeDS",
		dns.TypeSSHFP:      "TypeSSHFP",
		dns.TypeRRSIG:      "TypeRRSIG",
		dns.TypeNSEC:       "TypeNSEC",
		dns.TypeDNSKEY:     "TypeDNSKEY",
		dns.TypeDHCID:      "TypeDHCID",
		dns.TypeNSEC3:      "TypeNSEC3",
		dns.TypeNSEC3PARAM: "TypeNSEC3PARAM",
		dns.TypeTLSA:       "TypeTLSA",
		dns.TypeSMIMEA:     "TypeSMIMEA",
		dns.TypeHIP:        "TypeHIP",
		dns.TypeNINFO:      "TypeNINFO",
		dns.TypeRKEY:       "TypeRKEY",
		dns.TypeTALINK:     "TypeTALINK",
		dns.TypeCDS:        "TypeCDS",
		dns.TypeCDNSKEY:    "TypeCDNSKEY",
		dns.TypeOPENPGPKEY: "TypeOPENPGPKEY",
		dns.TypeCSYNC:      "TypeCSYNC",
		dns.TypeSPF:        "TypeSPF",
		dns.TypeUINFO:      "TypeUINFO",
		dns.TypeUID:        "TypeUID",
		dns.TypeGID:        "TypeGID",
		dns.TypeUNSPEC:     "TypeUNSPEC",
		dns.TypeNID:        "TypeNID",
		dns.TypeL32:        "TypeL32",
		dns.TypeL64:        "TypeL64",
		dns.TypeLP:         "TypeLP",
		dns.TypeEUI48:      "TypeEUI48",
		dns.TypeEUI64:      "TypeEUI64",
		dns.TypeURI:        "TypeURI",
		dns.TypeCAA:        "TypeCAA",
		dns.TypeAVC:        "TypeAVC",
		dns.TypeTKEY:       "TypeTKEY",
		dns.TypeTSIG:       "TypeTSIG",
		dns.TypeIXFR:       "TypeIXFR",
		dns.TypeAXFR:       "TypeAXFR",
		dns.TypeMAILB:      "TypeMAILB",
		dns.TypeMAILA:      "TypeMAILA",
		dns.TypeANY:        "TypeANY",
		dns.TypeTA:         "TypeTA",
		dns.TypeDLV:        "TypeDLV",
		dns.TypeReserved:   "TypeReserved"}

	fmt.Println("Local Time:", getLocalTime().Format(time.ANSIC))

	if len(ZabovDebugDBPath) > 0 {
		logChannel = make(chan logItem, 1024)
		go logWriteThread()
	}
}

func logWriteThread() {
	for item := range logChannel {
		var header string
		d := time.Now().Format("2006-01-02")
		logpath := path.Join(ZabovDebugDBPath, strings.Replace(item.clientIP, ":", "_", -1)+"-"+d+".log")

		_, err1 := os.Stat(logpath)
		if os.IsNotExist(err1) {
			header = strings.Join([]string{"time", "clientIP", "name", "reqType", "config", "timetable", "killed"}, "\t")
		}
		f, err := os.OpenFile(logpath, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
		if err == nil {
			reqTypeName, err := reqTypes[item.reqType]
			if !err {
				reqTypeName = fmt.Sprintf("%d", item.reqType)
			}
			ct := time.Now().Format(time.RFC3339)
			log := strings.Join([]string{ct, item.clientIP, strings.TrimRight(item.name, "."), reqTypeName, item.config, item.timetable, item.killed}, "\t")
			if len(header) > 0 {
				f.Write([]byte(header))
				f.Write([]byte("\n"))
			}
			f.Write([]byte(log))
			f.Write([]byte("\n"))
			f.Close()
		}
	}
}

func logQuery(clientIP string, name string, reqType uint16, config string, timetable string, killed string) {
	if len(ZabovDebugDBPath) > 0 {
		k := logItem{clientIP: clientIP, name: name, reqType: reqType, config: config, timetable: timetable, killed: killed}

		logChannel <- k

	}
}

func getLocalTime() time.Time {
	return time.Now().Local()
}

func confFromTimeTable(timetable string) string {
	tt := ZabovTimetables[timetable]
	if tt == nil {
		if ZabovDebug {
			log.Println("confFromTimeTable: return default")
		}
		return "default"
	}
	for _, ttentry := range tt.table {
		now := getLocalTime()

		nowHour := now.Hour()
		nowMinute := now.Minute()
		weekday := weekdays[now.Weekday()]
		if ttentry.days == nil || len(ttentry.days) == 0 || ttentry.days[weekday] || ttentry.days[strings.ToLower(weekday)] {
			for _, t := range ttentry.times {

				if (nowHour > t.start.hour || (nowHour == t.start.hour && nowMinute >= t.start.minute)) &&
					(nowHour < t.stop.hour || (nowHour == t.stop.hour && nowMinute <= t.stop.minute)) {
					go incrementStats("TIMETABLE IN: "+timetable, 1)
					if ZabovDebug {
						log.Println("confFromTimeTable: return IN", tt.cfgin)
					}
					return tt.cfgin
				}
			}
		}
	}
	go incrementStats("TIMETABLE OUT: "+timetable, 1)
	if ZabovDebug {
		log.Println("confFromTimeTable: return OUT", tt.cfgout)
	}
	return tt.cfgout
}

func confFromIP(clientIP net.IP) (string, string) {
	for _, ipgroup := range ZabovIPGroups {
		for _, ip := range ipgroup.ips {
			if clientIP.Equal(ip) {
				if len(ipgroup.timetable) > 0 {
					return confFromTimeTable(ipgroup.timetable), ipgroup.timetable
				}
				if ZabovDebug {
					log.Println("confFromIP: ipgroup.cfg", ipgroup.cfg)
				}
				return ipgroup.cfg, ""
			}
		}
	}
	if len(ZabovDefaultTimetable) > 0 {
		return confFromTimeTable(ZabovDefaultTimetable), ZabovDefaultTimetable
	}

	if ZabovDebug {
		log.Println("confFromIP: return default")
	}
	return "default", ""
}
func (mydns *handler) ServeDNS(w dns.ResponseWriter, r *dns.Msg) {
	go incrementStats("TotalQueries", 1)

	remIP, _, e := net.SplitHostPort(w.RemoteAddr().String())
	if e != nil {
		go incrementStats("CLIENT ERROR: "+remIP, 1)
	} else {
		go incrementStats("CLIENT: "+remIP, 1)
	}

	msg := dns.Msg{}
	msg.SetReply(r)

	config, timetable := confFromIP(net.ParseIP(remIP))

	if ZabovDebug {
		log.Println("REQUEST:", remIP, config)
	}
	ZabovConfig := ZabovConfigs[config]
	QType := r.Question[0].Qtype
	switch QType {
	case dns.TypeA:
		msg.Authoritative = true
		domain := msg.Question[0].Name
		fqdn := strings.TrimRight(domain, ".")

		if ZabovDebug {
			log.Println("TypeA: fqdn:", fqdn)
		}

		if len(ZabovIPAliases[fqdn]) > 0 {
			config = "__aliases__"
			msg.Answer = append(msg.Answer, &dns.A{
				Hdr: dns.RR_Header{Name: domain, Rrtype: dns.TypeA, Class: dns.ClassINET, Ttl: 60},
				A:   net.ParseIP(ZabovIPAliases[fqdn]),
			})
			go logQuery(remIP, fqdn, QType, config, timetable, "alias")
			break
		}
		if len(ZabovLocalResponder) > 0 {
			if !strings.Contains(fqdn, ".") ||
				(len(ZabovLocalDomain) > 0 && strings.HasSuffix(fqdn, ZabovLocalDomain)) {
				config = localresponderConfigName
				ret := ForwardQuery(r, config, true)
				w.WriteMsg(ret)
				go logQuery(remIP, fqdn, QType, config, timetable, "localresponder")
				break
			}

		}
		if domainInKillfile(fqdn, config) {
			go incrementStats("Killed", 1)

			msg.Answer = append(msg.Answer, &dns.A{
				Hdr: dns.RR_Header{Name: domain, Rrtype: dns.TypeA, Class: dns.ClassINET, Ttl: 60},
				A:   ZabovConfig.ZabovAddBL,
			})
			go logQuery(remIP, fqdn, QType, config, timetable, "killed")
		} else {
			go logQuery(remIP, fqdn, QType, config, timetable, "forwarded")
			ret := ForwardQuery(r, config, !ZabovConfig.ZabovCache)
			w.WriteMsg(ret)
		}
	case dns.TypePTR:
		if ZabovDebug {
			log.Println("TypePTR: Name:", msg.Question[0].Name)
		}

		if len(ZabovLocalResponder) > 0 {
			// if set use local responder for reverse lookup (suffix ".in-addr.arpa.")
			config = localresponderConfigName
		}
		ret := ForwardQuery(r, config, true)
		w.WriteMsg(ret)
		go logQuery(remIP, msg.Question[0].Name, QType, config, timetable, "localresponder")
	default:
		ret := ForwardQuery(r, config, !ZabovConfig.ZabovCache)
		w.WriteMsg(ret)
		if len(ZabovDebugDBPath) > 0 {
			go logQuery(remIP, msg.Question[0].Name, QType, config, timetable, "forwarded")
		}
	}
	go incrementStats("CONFIG: "+config, 1)
	w.WriteMsg(&msg)

}