- config

- sanity checks
  - parsing of timetables
- removed unused global variables (moved at config level)
- added new stat: DNS request for each "CONFIG"
- implemented configuration from ip groups / timetables

01.conf.go

@@ -5,7 +5,9 @@ import (
 	"fmt"
 	"io/ioutil"
 	"log"
+	"net"
 	"os"
+	"strconv"
 	"strings"
 
 	"github.com/miekg/dns"
@@ -46,13 +48,15 @@ func init() {
 
 	configs := MyConf["configs"].(map[string]interface{})
 
-	defaultConf := configs["default"].(map[string]interface{})
+	if len(configs) == 0 {
+		log.Println("you shall set at least default config")
+		os.Exit(1)
+	}
 
-	ZabovUpDNS = defaultConf["upstream"].(string)
-	ZabovSingleBL = defaultConf["singlefilters"].(string)
-	ZabovDoubleBL = defaultConf["doublefilters"].(string)
-	ZabovAddBL = defaultConf["blackholeip"].(string)
-	ZabovHostsFile = defaultConf["hostsfile"].(string)
+	if configs["default"] == nil {
+		log.Println("default config is required")
+		os.Exit(1)
+	}
 
 	zabovString := ZabovAddr + ":" + ZabovPort
 
@@ -60,13 +64,11 @@ func init() {
 	MyDNS.Addr = zabovString
 	MyDNS.Net = ZabovType
 
-	ZabovDNSArray = fileByLines(ZabovUpDNS)
-
 	ZabovConfigs = map[string]ZabovConfig{}
 	ZabovIPGroups = []ZabovIPGroup{}
-	ZabovTimetables = map[string]ZabovTimetable{}
+	ZabovTimetables = map[string]*ZabovTimetable{}
 	ZabovIPAliases = map[string]string{}
-	ZabovDNSArrays = map[string][]string{}
+
 	IPAliasesRaw := MyConf["ipaliases"].(map[string]interface{})
 
 	for alias, ip := range IPAliasesRaw {
@@ -84,11 +86,8 @@ func init() {
 		conf.ZabovAddBL = confRaw["blackholeip"].(string)
 		conf.ZabovHostsFile = confRaw["hostsfile"].(string)
 
-		ZabovDNSArrays[name] = fileByLines(conf.ZabovUpDNS)
+		conf.ZabovDNSArray = fileByLines(conf.ZabovUpDNS)
 		ZabovConfigs[name] = conf
-		if name == "default" {
-			ZabovConfigDefault = conf
-		}
 		ZabovCreateKDB(name)
 	}
 
@@ -111,13 +110,13 @@ func init() {
 
 		_, ok := ZabovConfigs[timetable.cfgin]
 		if !ok {
-			log.Println("inexistent cfgin:", timetable.cfgin)
+			log.Println("timetable: inexistent cfgin:", timetable.cfgin)
 			os.Exit(1)
 		}
 
 		_, ok = ZabovConfigs[timetable.cfgout]
 		if !ok {
-			log.Println("inexistent cfgout:", timetable.cfgout)
+			log.Println("timetable: inexistent cfgout:", timetable.cfgout)
 			os.Exit(1)
 		}
 
@@ -126,11 +125,36 @@ func init() {
 		for i := range tables {
 			table := tables[i].(map[string]interface{})
 			var ttEntry ZabovTimetableEntry
-			ttEntry.times = strings.Split(table["times"].(string), ";")
-			ttEntry.days = strings.Split(table["days"].(string), ";")
-			timetable.table = append(timetable.table, ttEntry)
+			ttEntry.times = []*ZabovTimeRange{}
+			for _, tRaw := range strings.Split(table["times"].(string), ";") {
+				tRawArr := strings.Split(tRaw, "-")
+				if len(tRawArr) > 1 {
+					startArr := strings.Split(tRawArr[0], ":")
+					stopArr := strings.Split(tRawArr[1], ":")
+
+					if len(startArr) > 1 && len(stopArr) > 1 {
+						hourStart, _ := strconv.Atoi(startArr[0])
+						minuteStart, _ := strconv.Atoi(startArr[1])
+						start := ZabovTime{hour: hourStart, minute: minuteStart}
+
+						hourStop, _ := strconv.Atoi(stopArr[0])
+						minuteStop, _ := strconv.Atoi(stopArr[1])
+						stop := ZabovTime{hour: hourStop, minute: minuteStop}
+						t := ZabovTimeRange{start: start, stop: stop}
+						ttEntry.times = append(ttEntry.times, &t)
+					}
+				}
+
+			}
+
+			ttEntry.days = map[string]bool{}
+			for _, day := range strings.Split(table["days"].(string), ";") {
+				ttEntry.days[day] = true
+			}
+
+			timetable.table = append(timetable.table, &ttEntry)
 		}
-		ZabovTimetables[name] = timetable
+		ZabovTimetables[name] = &timetable
 	}
 
 	IPGroups := MyConf["ipgroups"].([]interface{})
@@ -141,15 +165,16 @@ func init() {
 		var groupStruct ZabovIPGroup
 		groupMap := IPGroups[i].(map[string]interface{})
 		IPsRaw := groupMap["ips"].([]interface{})
-		groupStruct.ips = []string{}
+		groupStruct.ips = []net.IP{}
 		for x := range IPsRaw {
-			ip := IPsRaw[x].(string)
-			fmt.Println("adding IP ", ip)
+			ipRaw := IPsRaw[x].(string)
+			ip := net.ParseIP(ipRaw)
+			fmt.Println("adding IP ", ipRaw)
 
-			alias, ok := ZabovIPAliases[ip]
+			alias, ok := ZabovIPAliases[ipRaw]
 			if ok {
-				fmt.Println("IP alias: ", ip, alias)
-				ip = alias
+				fmt.Println("IP alias: ", ipRaw, alias)
+				ip = net.ParseIP(alias)
 			}
 			groupStruct.ips = append(groupStruct.ips, ip)
 		}
@@ -164,9 +189,9 @@ func init() {
 		}
 		ZabovIPGroups = append(ZabovIPGroups, groupStruct)
 	}
-	fmt.Println("ZabovConfigs:", ZabovConfigs)
-	fmt.Println("ZabovTimetables:", ZabovTimetables)
-	fmt.Println("ZabovIPAliases:", ZabovIPAliases)
-	fmt.Println("ZabovIPGroups:", ZabovIPGroups)
+	//fmt.Println("ZabovConfigs:", ZabovConfigs)
+	//fmt.Println("ZabovTimetables:", ZabovTimetables)
+	//fmt.Println("ZabovIPAliases:", ZabovIPAliases)
+	//fmt.Println("ZabovIPGroups:", ZabovIPGroups)
 
 }

adlist_hosts.go

@@ -89,6 +89,7 @@ func getDoubleFilters(urls urlsMap) {
 	for url, configs := range urls {
 		DoubleIndexFilter(url, configs)
 	}
+	fmt.Println("getDoubleFilters: DONE!")
 
 }
 

adlist_single.go

@@ -87,6 +87,7 @@ func getSingleFilters(urls urlsMap) {
 	for url, configs := range urls {
 		SingleIndexFilter(url, configs)
 	}
+	fmt.Println("getSingleFilters: DONE!")
 
 }
 

config.json

@@ -7,18 +7,19 @@
         "killfilettl": 12
     },
     "ipaliases":{
-        "pc8":"192.168.178.29"
+        "pc8":"192.168.178.29",
+        "localhost":"127.0.0.1"
     },
     "ipgroups":[
         {
-            "ips":["192.168.178.30/32", "192.168.178.31/32", "pc8"],
+            "ips":["localhost", "::1", "192.168.178.30", "192.168.178.31", "pc8"],
             "cfg":"",
             "timetable":"tt_children"
         }
     ],
     "timetables":{
         "tt_children":{
-            "tables":[{"times":"8:30-12:30;18:30-22:30", "days":"Mo;Tu;We;Th;Fr;Sa;Su"}],
+            "tables":[{"times":"00:00-05:00;8:30-12:30;18:30-22:59", "days":"Mo;Tu;We;Th;Fr;Sa;Su"}],
             "cfgin":"children_restricted",
             "cfgout":"children"
         }

dns_client.go

@@ -83,8 +83,7 @@ func oneTimeDNS(config string) (dns string) {
 
 	rand.Seed(time.Now().Unix())
 
-	ZabovDNSArray := ZabovDNSArrays[config]
-	upl := ZabovDNSArray
+	upl := ZabovConfigs[config].ZabovDNSArray
 
 	if len(upl) < 1 {
 		fmt.Println("No DNS defined, using default 127.0.0.53:53. Hope it works!")

dns_handler.go

@@ -1,12 +1,68 @@
 package main
 
 import (
+	"fmt"
 	"net"
 	"strings"
+	"time"
 
 	"github.com/miekg/dns"
 )
 
+var weekdays []string
+
+func init() {
+	weekdays = []string{"Su", "Mo", "Tu", "We", "Th", "Fr", "Sa"}
+}
+
+func getCurTime() (time.Time, error) {
+	return time.Parse("15:04", time.Now().Format("15:04"))
+}
+
+func confFromTimeTable(timetable string) string {
+	tt := ZabovTimetables[timetable]
+	if tt == nil {
+		fmt.Println("confFromTimeTable: return default")
+		return "default"
+	}
+	for _, ttentry := range tt.table {
+		now := time.Now()
+		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)) {
+					incrementStats("TIMETABLE IN: "+timetable, 1)
+					fmt.Println("confFromTimeTable: return IN", tt.cfgin)
+					return tt.cfgin
+				}
+			}
+		}
+	}
+	incrementStats("TIMETABLE OUT: "+timetable, 1)
+	fmt.Println("confFromTimeTable: return OUT", tt.cfgout)
+	return tt.cfgout
+}
+
+func confFromIP(clientIP net.IP) string {
+
+	for _, ipgroup := range ZabovIPGroups {
+		for _, ip := range ipgroup.ips {
+			if clientIP.Equal(ip) {
+				if len(ipgroup.timetable) > 0 {
+					return confFromTimeTable(ipgroup.timetable)
+				}
+				fmt.Println("confFromIP: ipgroup.cfg")
+				return ipgroup.cfg
+			}
+		}
+	}
+	fmt.Println("confFromIP: return default")
+	return "default"
+}
 func (mydns *handler) ServeDNS(w dns.ResponseWriter, r *dns.Msg) {
 	go incrementStats("TotalQueries", 1)
 
@@ -20,8 +76,10 @@ func (mydns *handler) ServeDNS(w dns.ResponseWriter, r *dns.Msg) {
 	msg := dns.Msg{}
 	msg.SetReply(r)
 
-	config := "default" // TODO: get config from client IP & timetable
+	config := confFromIP(net.ParseIP(remIP))
+	incrementStats("CONFIG: "+config, 1)
 
+	ZabovConfig := ZabovConfigs[config]
 	switch r.Question[0].Qtype {
 	case dns.TypeA:
 		msg.Authoritative = true
@@ -33,7 +91,7 @@ func (mydns *handler) ServeDNS(w dns.ResponseWriter, r *dns.Msg) {
 
 			msg.Answer = append(msg.Answer, &dns.A{
 				Hdr: dns.RR_Header{Name: domain, Rrtype: dns.TypeA, Class: dns.ClassINET, Ttl: 60},
-				A:   net.ParseIP(ZabovAddBL),
+				A:   net.ParseIP(ZabovConfig.ZabovAddBL),
 			})
 		} else {
 			ret := ForwardQuery(r, config)

main.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"log"
+	"net"
 
 	"github.com/miekg/dns"
 )
@@ -9,53 +10,30 @@ import (
 //MyDNS is my dns server
 var MyDNS *dns.Server
 
-//ZabovUpDNS keeps the name of upstream DNSs
-var ZabovUpDNS string
-
-//ZabovSingleBL list of urls returning a file with just names of domains
-var ZabovSingleBL string
-
-//ZabovDoubleBL list of urls returning a file with  IP<space>domain
-var ZabovDoubleBL string
-
-//ZabovAddBL is the IP we want to send all the clients to. Usually is 127.0.0.1
-var ZabovAddBL string
-
-//ZabovCacheTTL is the amount of hours we cache records of DNS
+//ZabovCacheTTL is the amount of hours we cache records of DNS (global)
 var ZabovCacheTTL int
 
-//ZabovKillTTL is the amount of hours we cache the killfile
+//ZabovKillTTL is the amount of hours we cache the killfile (global)
 var ZabovKillTTL int
 
-//ZabovHostsFile is the file we use to keep our hosts
-var ZabovHostsFile string
-
-//ZabovDNSArray is the array containing all the DNS we mention
-var ZabovDNSArray []string
-
 type handler struct{}
 
-//ZabovDNSArrays contains the arrays containing all the DNS we mention
-var ZabovDNSArrays map[string][]string
-
 // ZabovConfig contains all Zabov configs
 type ZabovConfig struct {
-	ZabovUpDNS     string // json:upstream -> ZabovDNSArray
-	ZabovSingleBL  string // json:singlefilters
-	ZabovDoubleBL  string // json:doublefilters
-	ZabovAddBL     string // json:blackholeip
-	ZabovHostsFile string // json:hostsfile
+	ZabovSingleBL  string   // json:singlefilters -> ZabovSingleBL list of urls returning a file with just names of domains
+	ZabovDoubleBL  string   // json:doublefilters -> ZabovDoubleBL list of urls returning a file with  IP<space>domain
+	ZabovAddBL     string   // json:blackholeip  -> ZabovAddBL is the IP we want to send all the clients to. Usually is 127.0.0.1
+	ZabovHostsFile string   // json:hostsfile -> ZabovHostsFile is the file we use to keep our hosts
+	ZabovUpDNS     string   // json:upstream -> ZabovUpDNS keeps the name of upstream DNSs
+	ZabovDNSArray  []string // contains all the DNS we mention, parsed from ZabovUpDNS file
 }
 
 // ZabovConfigs contains all Zabov configs
 var ZabovConfigs map[string]ZabovConfig
 
-// ZabovConfigDefault contains only "default" config
-var ZabovConfigDefault ZabovConfig
-
 // ZabovIPGroup contains Zabov groups of IPs
 type ZabovIPGroup struct {
-	ips       []string // IPs in this group
+	ips       []net.IP // IPs in this group
 	cfg       string   // config name to be used if there is no timetable
 	timetable string   // timetable name to be used for this group; timetable SHALL reference to config name to use
 }
@@ -63,21 +41,33 @@ type ZabovIPGroup struct {
 // ZabovIPGroups contains an array of all Zabov groups of IP rules
 var ZabovIPGroups []ZabovIPGroup
 
+// ZabovTime contains Zabov single time
+type ZabovTime struct {
+	hour   int
+	minute int
+}
+
+// ZabovTimeRange contains Zabov single time range
+type ZabovTimeRange struct {
+	start ZabovTime
+	stop  ZabovTime
+}
+
 // ZabovTimetableEntry contains Zabov single time table entry
 type ZabovTimetableEntry struct {
-	times []string
-	days  []string
+	times []*ZabovTimeRange
+	days  map[string]bool
 }
 
 // ZabovTimetable contains a Zabov time table
 type ZabovTimetable struct {
-	table  []ZabovTimetableEntry
+	table  []*ZabovTimetableEntry
 	cfgin  string // configuration name to be used if "inside" timetable
 	cfgout string // configuration name to be used if "outiside" timetable
 }
 
 // ZabovTimetables contains all Zabov time tables, by name
-var ZabovTimetables map[string]ZabovTimetable
+var ZabovTimetables map[string]*ZabovTimetable
 
 // ZabovIPAliases contains an array of all Zabov IP aliases
 var ZabovIPAliases map[string]string