You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
75 lines
1.4 KiB
Go
75 lines
1.4 KiB
Go
9 years ago
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"github.com/miekg/dns"
|
||
|
|
"testing"
|
||
|
|
//"os"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestParseNet(t *testing.T) {
|
||
|
|
nets := parse_net("net_china.txt")
|
||
|
|
fmt.Printf("get %d networks\n", len(nets))
|
||
|
|
fmt.Printf("1st %s\n", nets[0].String())
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestQuery(t *testing.T) {
|
||
|
|
ip_region = parse_net("net_china.txt")
|
||
|
|
var c *dns.Client
|
||
|
|
for _, srv := range []string{
|
||
|
|
"tcp:114.114.114.114:53",
|
||
|
|
"udp:8.8.8.8:53",
|
||
|
|
"udp:192.168.41.1:53",
|
||
|
|
"udp:4.2.2.2:53",
|
||
|
|
} {
|
||
|
|
proto, addr, err := parse_addr(srv)
|
||
|
|
if err != nil {
|
||
|
|
t.Error(err)
|
||
|
|
}
|
||
|
|
if proto == "tcp" {
|
||
|
|
c = client_tcp
|
||
|
|
} else {
|
||
|
|
c = client_udp
|
||
|
|
}
|
||
|
|
upsrv := &UpstreamServer{
|
||
|
|
Addr: addr,
|
||
|
|
Proto: proto,
|
||
|
|
client: c,
|
||
|
|
}
|
||
|
|
DefaultServer = append(DefaultServer, upsrv)
|
||
|
|
}
|
||
|
|
blacklist_file = "blacklist.txt"
|
||
|
|
a, err := load_domain(blacklist_file)
|
||
|
|
if err != nil {
|
||
|
|
fmt.Println(err)
|
||
|
|
} else {
|
||
|
|
Blacklist_ips = a
|
||
|
|
}
|
||
|
|
logger = NewLogger("", true)
|
||
|
|
for _, dn := range []string{
|
||
|
|
"www.google.com",
|
||
|
|
"www.sina.com.cn",
|
||
|
|
"www.taobao.com",
|
||
|
|
"www.ifeng.com",
|
||
|
|
"twitter.com",
|
||
|
|
"www.facebook.com",
|
||
|
|
"plus.google.com",
|
||
|
|
"drive.google.com",
|
||
|
|
"dongtaiwang.com",
|
||
|
|
"www.ratafee.nl",
|
||
|
|
"cc.ratafee.nl",
|
||
|
|
"noddcade.xx.ffs.aafde",
|
||
|
|
"ndfddcade.xx.ffs.aafde",
|
||
|
|
"sddf32dsf.comd.ffdasdf.fdsd3eaaaaa",
|
||
|
|
"www.google.com.hk",
|
||
|
|
} {
|
||
|
|
m := new(dns.Msg)
|
||
|
|
m.SetQuestion(dns.Fqdn(dn), dns.TypeA)
|
||
|
|
m1 := query(m)
|
||
|
|
if m1 == nil {
|
||
|
|
t.Error("query failed")
|
||
|
|
}
|
||
|
|
fmt.Printf("%s\n", m1)
|
||
|
|
}
|
||
|
|
}
|