2 Commits

Author SHA1 Message Date
87d9dca1ce removed unused function 2020-12-30 00:24:59 +01:00
4ddaa5a4c0 improved blacklist parsing 2020-12-29 22:34:30 +01:00

View File

@@ -1,6 +1,7 @@
package main package main
import ( import (
"errors"
"io/ioutil" "io/ioutil"
"log" "log"
"net" "net"
@@ -26,12 +27,10 @@ func loadBlacklist(config []configBlacklist) map[string]bool {
} }
domains := parseRawBlacklist(element, *raw) domains := parseRawBlacklist(element, *raw)
log.Printf("Added %d blocked domains", len(domains))
list = append(list, domains...) list = append(list, domains...)
} }
// list = removeDuplicates(list)
// sort.Strings(list)
domainMap := make(map[string]bool) domainMap := make(map[string]bool)
for _, e := range list { for _, e := range list {
domainMap[e] = true domainMap[e] = true
@@ -40,38 +39,47 @@ func loadBlacklist(config []configBlacklist) map[string]bool {
return domainMap return domainMap
} }
func removeDuplicates(elements []string) []string { func requestBacklist(blacklist configBlacklist) (*string, error) {
encountered := map[string]bool{} if blacklist.URL != "" {
result := []string{} return getBlacklistFromURL(blacklist.URL)
for v := range elements {
if !encountered[elements[v]] {
encountered[elements[v]] = true
result = append(result, elements[v])
}
} }
return result return nil, errors.New("No blacklist provided")
} }
func requestBacklist(blacklist configBlacklist) (*string, error) { func getBlacklistFromURL(url string) (*string, error) {
// Request list // Request list
resp, err := http.Get(blacklist.URL) resp, err := http.Get(url)
if err != nil { if err != nil {
return nil, err return nil, err
} }
defer resp.Body.Close() defer resp.Body.Close()
if resp.StatusCode != 200 {
log.Printf("Got %d status code. Continueing anyway.", resp.StatusCode)
}
body, err := ioutil.ReadAll(resp.Body) body, err := ioutil.ReadAll(resp.Body)
bodyString := string(body) bodyString := string(body)
log.Printf("Downloaded blacklist %s", blacklist.URL) log.Printf("Downloaded blacklist %s", url)
return &bodyString, err return &bodyString, err
} }
func parseRawBlacklist(blacklist configBlacklist, raw string) []string { func parseRawBlacklist(blacklist configBlacklist, raw string) []string {
switch blacklist.Format {
case "host":
return parseHostFormat(raw)
default:
log.Printf("Failed to parse blacklist. Format not supported: %s", blacklist.Format)
log.Println("Supported types are: host")
return make([]string, 0)
}
}
func parseHostFormat(raw string) []string {
finalList := make([]string, 0) finalList := make([]string, 0)
reg := regexp.MustCompile(`(?mi)^\s*(#*)\s*(\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3})\s+([a-zA-Z0-9\.\- ]+)$`) reg := regexp.MustCompile(`(?mi)^\s*(#*)\s*(\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3})\s+([a-zA-Z0-9\.\- ]+)$`)
matches := reg.FindAllStringSubmatch(raw, -1) matches := reg.FindAllStringSubmatch(raw, -1)