forked from loweel/zabov
316 lines
8.7 KiB
Go
316 lines
8.7 KiB
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 !domainInWhiteListfile(fqdn, config) && 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)
|
|
|
|
}
|