gfwleak/modikai-fpdns-server
Archived
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix logics in cname/ns chain, add rd0/max-recursion-queries features

Browse Source
This commit is contained in:
MDK
2024-02-21 21:44:30 +08:00
parent e9345e9fa7
commit 6f4d7c0758
2 changed files with 139 additions and 47 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
datastruct/datastruct.go
View File

@@ -2,12 +2,14 @@ package datastruct
type ResolverFp struct { type ResolverFp struct {
Dnssec bool `json:"dnssec"` Dnssec bool `json:"dnssec"`
RDbit bool `json:"rd0"`
QnameEncode bool `json:"0x20"` QnameEncode bool `json:"0x20"`
QueryMerge int `json:"merge_dup"` QueryMerge int `json:"merge_dup"`
MaxNsDepth int `json:"max_ns_depth"` MaxNsDepth int `json:"max_ns_depth"`
MaxCnameDepth int `json:"max_cname_depth"` MaxCnameDepth int `json:"max_cname_depth"`
RetryLimit int `json:"retry_limit"` RetryLimit int `json:"retry_limit"`
FetchLimit int `json:"fetch_limit"` FetchLimit int `json:"fetch_limit"`
MaxQueryNum int `json:"max_query_num"`
TimeoutStart int64 `json:"timeout_start"` TimeoutStart int64 `json:"timeout_start"`
TimeoutEnd int64 `json:"timeout_end"` TimeoutEnd int64 `json:"timeout_end"`
} }

112
fpdns_server.go
View File

@@ -8,8 +8,8 @@ import (
"net/http" "net/http"
"strconv" "strconv"
"strings" "strings"
"time"
"sync" "sync"
"time"
"fpdns_server/datastruct" "fpdns_server/datastruct"
@@ -74,6 +74,10 @@ func task_parser(name string, qtype uint16) int8 {
return 11 return 11
} else if strings.Contains(subdomain, "timeout") { } else if strings.Contains(subdomain, "timeout") {
return 12 return 12
} else if strings.Contains(subdomain, "recursion") {
return 13
} else if strings.Contains(subdomain, "maxq") {
return 14 // max recursion queries
} }
} }
return -1 return -1
@@ -103,7 +107,7 @@ func query_handler(w dns.ResponseWriter, r *dns.Msg) {
_, task_existed := datastruct.Task_map[random_token] _, task_existed := datastruct.Task_map[random_token]
if !init_existed && !task_existed { if !init_existed && !task_existed {
datastruct.Init_set[random_token] = struct{}{} datastruct.Init_set[random_token] = struct{}{}
//fmt.Printf("test initialized (token = %v)\n", random_token) fmt.Printf("test initialized (token = %v)\n", random_token)
break break
} }
} }
@@ -128,7 +132,7 @@ func query_handler(w dns.ResponseWriter, r *dns.Msg) {
if ok { if ok {
delete(datastruct.Init_set, token) delete(datastruct.Init_set, token)
datastruct.Task_map[token] = &datastruct.ResolverFp{} datastruct.Task_map[token] = &datastruct.ResolverFp{}
//fmt.Printf("Task Created (token = %v)\n", token) fmt.Printf("Task Created (token = %v)\n", token)
if r.IsEdns0() != nil { if r.IsEdns0() != nil {
datastruct.Task_map[token].Dnssec = r.IsEdns0().Do() datastruct.Task_map[token].Dnssec = r.IsEdns0().Do()
} }
@@ -158,7 +162,7 @@ func query_handler(w dns.ResponseWriter, r *dns.Msg) {
delete(datastruct.Task_map, token) delete(datastruct.Task_map, token)
datastruct.Result_map[token] = *result datastruct.Result_map[token] = *result
reply_code = "200.200.200.200" reply_code = "200.200.200.200"
//fmt.Printf("Task terminated (token = %v)\n", token) fmt.Printf("Task terminated (token = %v)\n", token)
} else { } else {
m.MsgHdr.Rcode = dns.RcodeServerFailure m.MsgHdr.Rcode = dns.RcodeServerFailure
break break
@@ -190,7 +194,7 @@ func query_handler(w dns.ResponseWriter, r *dns.Msg) {
} }
if fp, ok := datastruct.Task_map[token]; ok { if fp, ok := datastruct.Task_map[token]; ok {
fp.QueryMerge += 1 fp.QueryMerge += 1
//fmt.Printf("MergeTest (token = %v) current_status: %v\n", token, fp.QueryMerge) fmt.Printf("MergeTest (token = %v) current_status: %v\n", token, fp.QueryMerge)
} }
case 7: case 7:
respond_flag = false respond_flag = false
@@ -200,7 +204,7 @@ func query_handler(w dns.ResponseWriter, r *dns.Msg) {
} }
if fp, ok := datastruct.Task_map[token]; ok { if fp, ok := datastruct.Task_map[token]; ok {
fp.FetchLimit += 1 fp.FetchLimit += 1
//fmt.Printf("FetchTest (token = %v) current_status: %v\n", token, fp.FetchLimit) fmt.Printf("FetchTest (token = %v) current_status: %v\n", token, fp.FetchLimit)
} }
case 8: case 8:
wrong_ns := &dns.NS{ wrong_ns := &dns.NS{
@@ -216,6 +220,35 @@ func query_handler(w dns.ResponseWriter, r *dns.Msg) {
case 9: case 9:
var cname_target string var cname_target string
token, err := retrieve_token(subdomain) token, err := retrieve_token(subdomain)
if err == nil {
if fp, ok := datastruct.Task_map[token]; ok {
labels := strings.Split(subdomain, "-")
if labels[0] == "cname" {
cname_target = "1-" + subdomain
} else {
iter_cnt, err := strconv.Atoi(labels[0])
if err == nil {
if iter_cnt > fp.MaxNsDepth {
fp.MaxNsDepth = iter_cnt
fmt.Printf("CnameChainTest (token = %v) status_update: %v\n", token, fp.MaxCnameDepth)
}
}
iter_cnt += 1
cname_target = strconv.Itoa(iter_cnt) + "-" + strings.Join(labels[1:], "-")
}
cname_target = dns.Fqdn(cname_target + "." + CONFIG_SLD)
cname_redir := &dns.NS{
Hdr: dns.RR_Header{Name: name, Rrtype: dns.TypeNS, Class: dns.ClassINET, Ttl: 60},
Ns: cname_target,
}
m.Answer = append(m.Answer, cname_redir)
} else {
m.Rcode = dns.RcodeNameError
}
} else {
m.Rcode = dns.RcodeNameError
}
/*
labels := strings.Split(subdomain, "-") labels := strings.Split(subdomain, "-")
if labels[0] == "cname" { if labels[0] == "cname" {
cname_target = "1-" + subdomain cname_target = "1-" + subdomain
@@ -225,7 +258,7 @@ func query_handler(w dns.ResponseWriter, r *dns.Msg) {
if fp, ok := datastruct.Task_map[token]; ok { if fp, ok := datastruct.Task_map[token]; ok {
if iter_cnt > fp.MaxCnameDepth { if iter_cnt > fp.MaxCnameDepth {
fp.MaxCnameDepth = iter_cnt fp.MaxCnameDepth = iter_cnt
//fmt.Printf("CnameChainTest (token = %v) status_update: %v\n", token, fp.MaxCnameDepth) fmt.Printf("CnameChainTest (token = %v) status_update: %v\n", token, fp.MaxCnameDepth)
} }
} }
} }
@@ -239,9 +272,39 @@ func query_handler(w dns.ResponseWriter, r *dns.Msg) {
Target: cname_target, Target: cname_target,
} }
m.Answer = append(m.Answer, cname_redir) m.Answer = append(m.Answer, cname_redir)
*/
case 10: case 10:
var ns_target string var ns_target string
token, err := retrieve_token(subdomain) token, err := retrieve_token(subdomain)
if err == nil {
if fp, ok := datastruct.Task_map[token]; ok {
labels := strings.Split(subdomain, "-")
if labels[0] == "ns" {
ns_target = "1-" + subdomain
} else {
iter_cnt, err := strconv.Atoi(labels[0])
if err == nil {
if iter_cnt > fp.MaxNsDepth {
fp.MaxNsDepth = iter_cnt
fmt.Printf("NsChainTest (token = %v) status_update: %v\n", token, fp.MaxNsDepth)
}
}
iter_cnt += 1
ns_target = strconv.Itoa(iter_cnt) + "-" + strings.Join(labels[1:], "-")
}
ns_target = dns.Fqdn(ns_target + "." + CONFIG_SLD)
ns_delegation := &dns.NS{
Hdr: dns.RR_Header{Name: name, Rrtype: dns.TypeNS, Class: dns.ClassINET, Ttl: 60},
Ns: ns_target,
}
m.Ns = append(m.Ns, ns_delegation)
} else {
m.Rcode = dns.RcodeNameError
}
} else {
m.Rcode = dns.RcodeNameError
}
/*
labels := strings.Split(subdomain, "-") labels := strings.Split(subdomain, "-")
if labels[0] == "ns" { if labels[0] == "ns" {
ns_target = "1-" + subdomain ns_target = "1-" + subdomain
@@ -251,7 +314,7 @@ func query_handler(w dns.ResponseWriter, r *dns.Msg) {
if fp, ok := datastruct.Task_map[token]; ok { if fp, ok := datastruct.Task_map[token]; ok {
if iter_cnt > fp.MaxNsDepth { if iter_cnt > fp.MaxNsDepth {
fp.MaxNsDepth = iter_cnt fp.MaxNsDepth = iter_cnt
//fmt.Printf("NsChainTest (token = %v) status_update: %v\n", token, fp.MaxNsDepth) fmt.Printf("NsChainTest (token = %v) status_update: %v\n", token, fp.MaxNsDepth)
} }
} }
} }
@@ -265,6 +328,7 @@ func query_handler(w dns.ResponseWriter, r *dns.Msg) {
Ns: ns_target, Ns: ns_target,
} }
m.Ns = append(m.Ns, ns_delegation) m.Ns = append(m.Ns, ns_delegation)
*/
case 11: case 11:
respond_flag = false respond_flag = false
token, err := retrieve_token(subdomain) token, err := retrieve_token(subdomain)
@@ -273,7 +337,7 @@ func query_handler(w dns.ResponseWriter, r *dns.Msg) {
} }
if fp, ok := datastruct.Task_map[token]; ok { if fp, ok := datastruct.Task_map[token]; ok {
fp.RetryLimit += 1 fp.RetryLimit += 1
//fmt.Printf("RetryTest (token = %v) current_status: %v\n", token, fp.RetryLimit) fmt.Printf("RetryTest (token = %v) current_status: %v\n", token, fp.RetryLimit)
} }
case 12: case 12:
respond_flag = false respond_flag = false
@@ -284,10 +348,36 @@ func query_handler(w dns.ResponseWriter, r *dns.Msg) {
if fp, ok := datastruct.Task_map[token]; ok { if fp, ok := datastruct.Task_map[token]; ok {
if fp.TimeoutStart == 0 { if fp.TimeoutStart == 0 {
fp.TimeoutStart = time.Now().UnixMilli() fp.TimeoutStart = time.Now().UnixMilli()
//fmt.Printf("TimeoutTest (token = %v) start_timestamp: %v\n", token, fp.TimeoutStart) fmt.Printf("TimeoutTest (token = %v) start_timestamp: %v\n", token, fp.TimeoutStart)
} else { } else {
fp.TimeoutEnd = time.Now().UnixMilli() fp.TimeoutEnd = time.Now().UnixMilli()
//fmt.Printf("TimeoutTest (token = %v) timestamp_update: %v\n", token, fp.TimeoutEnd) fmt.Printf("TimeoutTest (token = %v) timestamp_update: %v\n", token, fp.TimeoutEnd)
}
}
case 13:
token, err := retrieve_token(subdomain)
if err != nil {
break
}
if fp, ok := datastruct.Task_map[token]; ok {
fp.RDbit = true
}
m.MsgHdr.Rcode = dns.RcodeNameError
case 14:
token, err := retrieve_token(subdomain)
if err != nil {
break
}
if fp, ok := datastruct.Task_map[token]; ok {
fp.MaxQueryNum += 1
fmt.Printf("MaxQueryNum (token = %v) timestamp_update: %v\n", token, fp.TimeoutEnd)
for i := 0; i < 10; i++ {
ns_target := dns.Fqdn(strconv.Itoa(i) + subdomain + "." + CONFIG_SLD)
ns_delegation := &dns.NS{
Hdr: dns.RR_Header{Name: name, Rrtype: dns.TypeNS, Class: dns.ClassINET, Ttl: 60},
Ns: ns_target,
}
m.Ns = append(m.Ns, ns_delegation)
} }
} }
case 0: case 0: