60 lines
996 B
Go
60 lines
996 B
Go
|
package main
|
||
|
|
||
|
import (
|
||
|
"github.com/hoisie/redis"
|
||
|
// "github.com/miekg/dns"
|
||
|
"fmt"
|
||
|
"io/ioutil"
|
||
|
)
|
||
|
|
||
|
type HostsQueryFaild struct {
|
||
|
domain string
|
||
|
}
|
||
|
|
||
|
func (e HostsQueryFaild) Error() string {
|
||
|
return e.domain + " hosts match failed"
|
||
|
}
|
||
|
|
||
|
func readLocalHostsFile(file string) map[string]string {
|
||
|
var hosts = make(map[string]string)
|
||
|
f, _ := os.Open(file)
|
||
|
scanner := bufio.NewScanner(f)
|
||
|
for scanner.Scan() {
|
||
|
|
||
|
line := scanner.Text()
|
||
|
line = strings.TrimSpace(line)
|
||
|
|
||
|
if strings.HasPrefix(line, "#") || line == "" {
|
||
|
continue
|
||
|
}
|
||
|
|
||
|
sli := strings.Split(line, " ")
|
||
|
if len(sli) == 1 {
|
||
|
sli = strings.Split(line, "\t")
|
||
|
}
|
||
|
|
||
|
if len(sli) < 2 {
|
||
|
continue
|
||
|
}
|
||
|
|
||
|
domain := sli[len(sli)-1]
|
||
|
ip := sli[0]
|
||
|
if !isDomain(domain) || !isIP(ip) {
|
||
|
continue
|
||
|
}
|
||
|
|
||
|
hosts[domain] = ip
|
||
|
}
|
||
|
return hosts
|
||
|
}
|
||
|
|
||
|
func isDomain(domain string) bool {
|
||
|
match, _ := regexp.MatchString("^[a-z]", domain)
|
||
|
return match
|
||
|
}
|
||
|
|
||
|
func isIP(ip string) bool {
|
||
|
match, _ := regexp.MatchString("^[1-9]", ip)
|
||
|
return match
|
||
|
}
|