initial commit
This commit is contained in:
21
datastruct/datastruct.go
Normal file
21
datastruct/datastruct.go
Normal file
@@ -0,0 +1,21 @@
|
||||
package datastruct
|
||||
|
||||
type ResolverFp struct {
|
||||
Dnssec bool `json:"dnssec"`
|
||||
QnameEncode bool `json:"0x20"`
|
||||
QueryMerge int `json:"merge_dup"`
|
||||
MaxNsDepth int `json:"max_ns_depth"`
|
||||
MaxCnameDepth int `json:"max_cname_depth"`
|
||||
RetryLimit int `json:"retry_limit"`
|
||||
FetchLimit int `json:"fetch_limit"`
|
||||
TimeoutStart int64 `json:"timeout_start"`
|
||||
TimeoutEnd int64 `json:"timeout_end"`
|
||||
}
|
||||
|
||||
type InitSet map[int]struct{}
|
||||
type TaskMap map[int]*ResolverFp
|
||||
type ResultMap map[int]ResolverFp
|
||||
|
||||
var Init_set = make(InitSet)
|
||||
var Task_map = make(TaskMap)
|
||||
var Result_map = make(ResultMap)
|
||||
BIN
fpdns_server
Executable file
BIN
fpdns_server
Executable file
Binary file not shown.
517
fpdns_server.go
Normal file
517
fpdns_server.go
Normal file
@@ -0,0 +1,517 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"flag"
|
||||
"fmt"
|
||||
"math/rand"
|
||||
"net"
|
||||
"net/http"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
"sync"
|
||||
|
||||
"fpdns_server/datastruct"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
"github.com/miekg/dns"
|
||||
)
|
||||
|
||||
var CONFIG_SLD string
|
||||
var UniqueCounter uint32 = 0xfffffffa
|
||||
var mutex_lock sync.Mutex
|
||||
|
||||
func InttoIPv4(n uint32) net.IP {
|
||||
b0 := (n >> 24) & 0xff
|
||||
b1 := (n >> 16) & 0xff
|
||||
b2 := (n >> 8) & 0xff
|
||||
b3 := n & 0xff
|
||||
return net.IPv4(byte(b0), byte(b1), byte(b2), byte(b3))
|
||||
}
|
||||
|
||||
func TtlParser(domain string) uint32 {
|
||||
subdomain := strings.ToLower(strings.Split(domain, ".")[0])
|
||||
ttl, _ := strconv.Atoi(strings.Split(subdomain, "-")[0])
|
||||
return uint32(ttl)
|
||||
}
|
||||
|
||||
func retrieve_token(label string) (int, error) {
|
||||
fields := strings.Split(label, "-")
|
||||
token, err := strconv.Atoi(fields[len(fields)-1])
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
return token, nil
|
||||
}
|
||||
|
||||
func task_parser(name string, qtype uint16) int8 {
|
||||
labels := strings.Split(name, ".")
|
||||
subdomain := strings.ToLower(labels[0])
|
||||
if qtype == dns.TypeA {
|
||||
if len(labels) == 4 && (subdomain == "ns1" || subdomain == "ns2") {
|
||||
return 0
|
||||
} else if strings.Contains(subdomain, "start") {
|
||||
return 1
|
||||
} else if strings.Contains(subdomain, "init") {
|
||||
return 2
|
||||
} else if strings.Contains(subdomain, "end") {
|
||||
return 3
|
||||
} else if strings.Contains(subdomain, "echo") {
|
||||
return 4 // basic echodns
|
||||
} else if strings.Contains(subdomain, "ttl") {
|
||||
return 5 // ttl test
|
||||
} else if strings.Contains(subdomain, "merge") {
|
||||
return 6
|
||||
} else if strings.Contains(subdomain, "fetch") {
|
||||
return 7
|
||||
} else if strings.Contains(subdomain, "bailiwick") {
|
||||
return 8 // bailiwick verification
|
||||
} else if strings.Contains(subdomain, "cname-chain") {
|
||||
return 9 // unlimited ns chain
|
||||
} else if strings.Contains(subdomain, "ns-chain") {
|
||||
return 10 // unlimited cname chain
|
||||
} else if strings.Contains(subdomain, "retry") {
|
||||
return 11
|
||||
} else if strings.Contains(subdomain, "timeout") {
|
||||
return 12
|
||||
}
|
||||
}
|
||||
return -1
|
||||
}
|
||||
|
||||
func query_handler(w dns.ResponseWriter, r *dns.Msg) {
|
||||
var ip net.IP
|
||||
respond_flag := true
|
||||
if addr, ok := w.RemoteAddr().(*net.UDPAddr); ok {
|
||||
ip = addr.IP
|
||||
}
|
||||
m := new(dns.Msg)
|
||||
m.SetReply(r)
|
||||
m.Compress = true
|
||||
m.Authoritative = true
|
||||
name := m.Question[0].Name
|
||||
subdomain := strings.Split(name, ".")[0]
|
||||
qtype := m.Question[0].Qtype
|
||||
mutex_lock.Lock()
|
||||
switch task_parser(name, qtype) {
|
||||
case 1:
|
||||
var random_token int
|
||||
// may cause dead loop -> attention!
|
||||
for {
|
||||
random_token = rand.Intn(1 << 32)
|
||||
_, init_existed := datastruct.Init_set[random_token]
|
||||
_, task_existed := datastruct.Task_map[random_token]
|
||||
if !init_existed && !task_existed {
|
||||
datastruct.Init_set[random_token] = struct{}{}
|
||||
//fmt.Printf("test initialized (token = %v)\n", random_token)
|
||||
break
|
||||
}
|
||||
}
|
||||
a := &dns.A{
|
||||
Hdr: dns.RR_Header{Name: name, Rrtype: dns.TypeA, Class: dns.ClassINET, Ttl: 60},
|
||||
A: InttoIPv4(uint32(random_token)),
|
||||
}
|
||||
m.Answer = append(m.Answer, a)
|
||||
case 2:
|
||||
var reply_code string
|
||||
var ok bool
|
||||
token, err := retrieve_token(subdomain)
|
||||
if err != nil {
|
||||
m.MsgHdr.Rcode = dns.RcodeServerFailure
|
||||
break
|
||||
}
|
||||
//mutex_lock.Lock()
|
||||
if _, ok = datastruct.Init_set[token]; ok {
|
||||
delete(datastruct.Init_set, token)
|
||||
}
|
||||
//mutex_lock.Unlock()
|
||||
if ok {
|
||||
delete(datastruct.Init_set, token)
|
||||
datastruct.Task_map[token] = &datastruct.ResolverFp{}
|
||||
//fmt.Printf("Task Created (token = %v)\n", token)
|
||||
if r.IsEdns0() != nil {
|
||||
datastruct.Task_map[token].Dnssec = r.IsEdns0().Do()
|
||||
}
|
||||
if name != strings.ToLower(name) {
|
||||
datastruct.Task_map[token].QnameEncode = true
|
||||
} else {
|
||||
datastruct.Task_map[token].QnameEncode = false
|
||||
}
|
||||
reply_code = "200.200.200.200"
|
||||
} else {
|
||||
m.MsgHdr.Rcode = dns.RcodeServerFailure
|
||||
break
|
||||
}
|
||||
a := &dns.A{
|
||||
Hdr: dns.RR_Header{Name: name, Rrtype: dns.TypeA, Class: dns.ClassINET, Ttl: 60},
|
||||
A: net.ParseIP(reply_code),
|
||||
}
|
||||
m.Answer = append(m.Answer, a)
|
||||
case 3:
|
||||
var reply_code string
|
||||
token, err := retrieve_token(subdomain)
|
||||
if err != nil {
|
||||
m.MsgHdr.Rcode = dns.RcodeServerFailure
|
||||
break
|
||||
}
|
||||
if result, ok := datastruct.Task_map[token]; ok {
|
||||
delete(datastruct.Task_map, token)
|
||||
datastruct.Result_map[token] = *result
|
||||
reply_code = "200.200.200.200"
|
||||
//fmt.Printf("Task terminated (token = %v)\n", token)
|
||||
} else {
|
||||
m.MsgHdr.Rcode = dns.RcodeServerFailure
|
||||
break
|
||||
}
|
||||
a := &dns.A{
|
||||
Hdr: dns.RR_Header{Name: name, Rrtype: dns.TypeA, Class: dns.ClassINET, Ttl: 60},
|
||||
A: net.ParseIP(reply_code),
|
||||
}
|
||||
m.Answer = append(m.Answer, a)
|
||||
case 4:
|
||||
a := &dns.A{
|
||||
Hdr: dns.RR_Header{Name: name, Rrtype: dns.TypeA, Class: dns.ClassINET, Ttl: 60},
|
||||
A: ip,
|
||||
}
|
||||
m.Answer = append(m.Answer, a)
|
||||
case 5:
|
||||
query_ttl := TtlParser(name)
|
||||
//fmt.Println(query_ttl)
|
||||
a := &dns.A{
|
||||
Hdr: dns.RR_Header{Name: name, Rrtype: dns.TypeA, Class: dns.ClassINET, Ttl: query_ttl},
|
||||
A: ip,
|
||||
}
|
||||
m.Answer = append(m.Answer, a)
|
||||
case 6:
|
||||
respond_flag = false
|
||||
token, err := retrieve_token(subdomain)
|
||||
if err != nil {
|
||||
break
|
||||
}
|
||||
if fp, ok := datastruct.Task_map[token]; ok {
|
||||
fp.QueryMerge += 1
|
||||
//fmt.Printf("MergeTest (token = %v) current_status: %v\n", token, fp.QueryMerge)
|
||||
}
|
||||
case 7:
|
||||
respond_flag = false
|
||||
token, err := retrieve_token(subdomain)
|
||||
if err != nil {
|
||||
break
|
||||
}
|
||||
if fp, ok := datastruct.Task_map[token]; ok {
|
||||
fp.FetchLimit += 1
|
||||
//fmt.Printf("FetchTest (token = %v) current_status: %v\n", token, fp.FetchLimit)
|
||||
}
|
||||
case 8:
|
||||
wrong_ns := &dns.NS{
|
||||
Hdr: dns.RR_Header{Name: name, Rrtype: dns.TypeNS, Class: dns.ClassINET, Ttl: 60},
|
||||
Ns: dns.Fqdn("dns.baidu.com"),
|
||||
}
|
||||
wrong_ns_ip := &dns.A{
|
||||
Hdr: dns.RR_Header{Name: "dns.baidu.com.", Rrtype: dns.TypeA, Class: dns.ClassINET, Ttl: 60},
|
||||
A: net.ParseIP("159.75.200.247"),
|
||||
}
|
||||
m.Ns = append(m.Ns, wrong_ns)
|
||||
m.Extra = append(m.Extra, wrong_ns_ip)
|
||||
case 9:
|
||||
var cname_target string
|
||||
token, err := retrieve_token(subdomain)
|
||||
labels := strings.Split(subdomain, "-")
|
||||
if labels[0] == "cname" {
|
||||
cname_target = "1-" + subdomain
|
||||
} else {
|
||||
iter_cnt, _ := strconv.Atoi(labels[0])
|
||||
if err == nil {
|
||||
if fp, ok := datastruct.Task_map[token]; ok {
|
||||
if iter_cnt > fp.MaxCnameDepth {
|
||||
fp.MaxCnameDepth = 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 = strings.Join([]string{labels[0], labels[1], strconv.Itoa(iter_cnt)}, "-")
|
||||
}
|
||||
cname_target = dns.Fqdn(cname_target + "." + CONFIG_SLD)
|
||||
cname_redir := &dns.CNAME{
|
||||
Hdr: dns.RR_Header{Name: name, Rrtype: dns.TypeCNAME, Class: dns.ClassINET, Ttl: 60},
|
||||
Target: cname_target,
|
||||
}
|
||||
m.Answer = append(m.Answer, cname_redir)
|
||||
case 10:
|
||||
var ns_target string
|
||||
token, err := retrieve_token(subdomain)
|
||||
labels := strings.Split(subdomain, "-")
|
||||
if labels[0] == "ns" {
|
||||
ns_target = "1-" + subdomain
|
||||
} else {
|
||||
iter_cnt, _ := strconv.Atoi(labels[0])
|
||||
if err == nil {
|
||||
if fp, ok := datastruct.Task_map[token]; ok {
|
||||
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 = strings.Join([]string{labels[0], labels[1], strconv.Itoa(iter_cnt)}, "-")
|
||||
}
|
||||
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)
|
||||
case 11:
|
||||
respond_flag = false
|
||||
token, err := retrieve_token(subdomain)
|
||||
if err != nil {
|
||||
break
|
||||
}
|
||||
if fp, ok := datastruct.Task_map[token]; ok {
|
||||
fp.RetryLimit += 1
|
||||
//fmt.Printf("RetryTest (token = %v) current_status: %v\n", token, fp.RetryLimit)
|
||||
}
|
||||
case 12:
|
||||
respond_flag = false
|
||||
token, err := retrieve_token(subdomain)
|
||||
if err != nil {
|
||||
break
|
||||
}
|
||||
if fp, ok := datastruct.Task_map[token]; ok {
|
||||
if fp.TimeoutStart == 0 {
|
||||
fp.TimeoutStart = time.Now().UnixMilli()
|
||||
//fmt.Printf("TimeoutTest (token = %v) start_timestamp: %v\n", token, fp.TimeoutStart)
|
||||
} else {
|
||||
fp.TimeoutEnd = time.Now().UnixMilli()
|
||||
//fmt.Printf("TimeoutTest (token = %v) timestamp_update: %v\n", token, fp.TimeoutEnd)
|
||||
}
|
||||
}
|
||||
case 0:
|
||||
a := &dns.A{
|
||||
Hdr: dns.RR_Header{Name: name, Rrtype: dns.TypeA, Class: dns.ClassINET, Ttl: 3600},
|
||||
A: net.ParseIP("159.75.200.247"),
|
||||
}
|
||||
m.Answer = append(m.Answer, a)
|
||||
case -1:
|
||||
m.MsgHdr.Rcode = dns.RcodeNameError
|
||||
}
|
||||
mutex_lock.Unlock()
|
||||
if respond_flag {
|
||||
w.WriteMsg(m)
|
||||
}
|
||||
}
|
||||
|
||||
/*
|
||||
func strategyMaker(name string, qtype uint16) int8 {
|
||||
var subdomain string
|
||||
labels := strings.Split(name, ".")
|
||||
if len(labels) >= 4 {
|
||||
subdomain = strings.ToLower(labels[len(labels)-4])
|
||||
}
|
||||
|
||||
if qtype == dns.TypeA {
|
||||
if len(labels) == 4 && (subdomain == "ns1" || subdomain == "ns2") {
|
||||
return 0
|
||||
} else if strings.Contains(subdomain, "fwd") {
|
||||
return 1 // return rdns ip in cname
|
||||
} else if strings.Contains(subdomain, "rdns") {
|
||||
return 2 // return honey cname record
|
||||
} else if strings.Contains(subdomain, "honey") {
|
||||
return 3 // return timestamp in a record
|
||||
} else if strings.Contains(subdomain, "echo") {
|
||||
return 4 // basic echodns
|
||||
} else if strings.Contains(subdomain, "ttl") {
|
||||
return 5 // ttl test
|
||||
} else if strings.Contains(subdomain, "wrong-id") {
|
||||
return 6 // return response with wrong txid
|
||||
} else if strings.Contains(subdomain, "wrong-rec") {
|
||||
return 7 // return response with wrong records
|
||||
} else if strings.Contains(subdomain, "bailiwick") {
|
||||
return 8 // bailiwick verification
|
||||
} else if strings.Contains(subdomain, "cname-chain") {
|
||||
return 9 // unlimited ns chain
|
||||
} else if strings.Contains(subdomain, "ns-chain") {
|
||||
return 10 // unlimited cname chain
|
||||
} else if strings.Contains(subdomain, "attack") {
|
||||
return 11
|
||||
}
|
||||
}
|
||||
return -1
|
||||
}
|
||||
|
||||
|
||||
func handleReflect(w dns.ResponseWriter, r *dns.Msg) {
|
||||
var (
|
||||
ip net.IP
|
||||
name string
|
||||
qtype uint16
|
||||
)
|
||||
m := new(dns.Msg)
|
||||
m.SetReply(r)
|
||||
m.Compress = true
|
||||
m.Authoritative = true
|
||||
if addr, ok := w.RemoteAddr().(*net.UDPAddr); ok {
|
||||
ip = addr.IP
|
||||
}
|
||||
name = m.Question[0].Name
|
||||
subdomain := strings.ToLower(strings.Split(name, ".")[0])
|
||||
qtype = m.Question[0].Qtype
|
||||
switch strategyMaker(name, qtype) {
|
||||
case 1:
|
||||
cname_subdomain := "rdns-" + strings.Replace(ip.String(), ".", "-", -1)
|
||||
cname_fqdn := dns.Fqdn(cname_subdomain + "." + CONFIG_SLD)
|
||||
cname := &dns.CNAME{
|
||||
Hdr: dns.RR_Header{Name: name, Rrtype: dns.TypeCNAME, Class: dns.ClassINET, Ttl: 14400},
|
||||
Target: cname_fqdn,
|
||||
}
|
||||
//fmt.Println(name+" "+cname_fqdn)
|
||||
m.Answer = append(m.Answer, cname)
|
||||
case 2:
|
||||
cname_fqdn := dns.Fqdn("honey." + CONFIG_SLD)
|
||||
cname := &dns.CNAME{
|
||||
Hdr: dns.RR_Header{Name: name, Rrtype: dns.TypeCNAME, Class: dns.ClassINET, Ttl: 14400},
|
||||
Target: cname_fqdn,
|
||||
}
|
||||
//fmt.Println(cname_fqdn)
|
||||
m.Answer = append(m.Answer, cname)
|
||||
case 3:
|
||||
//time_str := strconv.FormatInt(time.Now().UnixMicro(), 10)
|
||||
//time_int, _ := strconv.Atoi(time_str[5 : len(time_str)-2])
|
||||
//time_int += rand.Intn(10000)
|
||||
UniqueCounter += 1
|
||||
timestamp := InttoIPv4(uint32(UniqueCounter))
|
||||
a := &dns.A{
|
||||
Hdr: dns.RR_Header{Name: name, Rrtype: dns.TypeA, Class: dns.ClassINET, Ttl: 14400},
|
||||
A: timestamp,
|
||||
}
|
||||
m.Answer = append(m.Answer, a)
|
||||
case 4:
|
||||
a := &dns.A{
|
||||
Hdr: dns.RR_Header{Name: name, Rrtype: dns.TypeA, Class: dns.ClassINET, Ttl: 60},
|
||||
A: ip,
|
||||
}
|
||||
m.Answer = append(m.Answer, a)
|
||||
case 5:
|
||||
query_ttl := TtlParser(name)
|
||||
//fmt.Println(query_ttl)
|
||||
a := &dns.A{
|
||||
Hdr: dns.RR_Header{Name: name, Rrtype: dns.TypeA, Class: dns.ClassINET, Ttl: query_ttl},
|
||||
A: ip,
|
||||
}
|
||||
m.Answer = append(m.Answer, a)
|
||||
case 6:
|
||||
m.MsgHdr.Id += 1
|
||||
a := &dns.A{
|
||||
Hdr: dns.RR_Header{Name: name, Rrtype: dns.TypeA, Class: dns.ClassINET, Ttl: 60},
|
||||
A: ip,
|
||||
}
|
||||
m.Answer = append(m.Answer, a)
|
||||
case 7:
|
||||
wrong_answer := &dns.AAAA{
|
||||
Hdr: dns.RR_Header{Name: "www.example.com.", Rrtype: dns.TypeAAAA, Class: dns.ClassINET, Ttl: 60},
|
||||
AAAA: net.ParseIP("fe80::7526:a2ae:a0b8:946d"),
|
||||
}
|
||||
m.Answer = append(m.Answer, wrong_answer)
|
||||
case 8:
|
||||
wrong_ns := &dns.NS{
|
||||
Hdr: dns.RR_Header{Name: name, Rrtype: dns.TypeNS, Class: dns.ClassINET, Ttl: 60},
|
||||
Ns: dns.Fqdn("dns.baidu.com"),
|
||||
}
|
||||
wrong_ns_ip := &dns.A{
|
||||
Hdr: dns.RR_Header{Name: "dns.baidu.com.", Rrtype: dns.TypeA, Class: dns.ClassINET, Ttl: 60},
|
||||
A: net.ParseIP("159.75.200.247"),
|
||||
}
|
||||
m.Ns = append(m.Ns, wrong_ns)
|
||||
m.Extra = append(m.Extra, wrong_ns_ip)
|
||||
case 9:
|
||||
cname_target := ""
|
||||
labels := strings.Split(subdomain, "-")
|
||||
if labels[len(labels)-1] == "chain" {
|
||||
cname_target = subdomain + "-0"
|
||||
} else {
|
||||
iter_cnt, _ := strconv.Atoi(labels[len(labels)-1])
|
||||
iter_cnt += 1
|
||||
labels = append(labels[:len(labels)-1], strconv.Itoa(iter_cnt))
|
||||
cname_target = strings.Join(labels, "-")
|
||||
//cname_target = strings.Join([]string{labels[0], labels[1], strconv.Itoa(iter_cnt)}, "-")
|
||||
}
|
||||
cname_target = dns.Fqdn(cname_target + "." + CONFIG_SLD)
|
||||
cname_redir := &dns.CNAME{
|
||||
Hdr: dns.RR_Header{Name: name, Rrtype: dns.TypeCNAME, Class: dns.ClassINET, Ttl: 60},
|
||||
Target: cname_target,
|
||||
}
|
||||
m.Answer = append(m.Answer, cname_redir)
|
||||
case 10:
|
||||
ns_target := ""
|
||||
labels := strings.Split(subdomain, "-")
|
||||
if labels[len(labels)-1] == "chain" {
|
||||
ns_target = subdomain + "-0"
|
||||
} else {
|
||||
iter_cnt, _ := strconv.Atoi(labels[len(labels)-1])
|
||||
iter_cnt += 1
|
||||
labels = append(labels[:len(labels)-1], strconv.Itoa(iter_cnt))
|
||||
ns_target = strings.Join(labels, "-")
|
||||
//ns_target = strings.Join([]string{labels[0], labels[1], strconv.Itoa(iter_cnt)}, "-")
|
||||
}
|
||||
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)
|
||||
case 11:
|
||||
ns_record := &dns.NS{
|
||||
Hdr: dns.RR_Header{Name: name, Rrtype: dns.TypeNS, Class: dns.ClassINET, Ttl: 60},
|
||||
Ns: "shiyan7.jthmfgz.icu.",
|
||||
}
|
||||
m.Ns = append(m.Ns, ns_record)
|
||||
case 0:
|
||||
a := &dns.A{
|
||||
Hdr: dns.RR_Header{Name: name, Rrtype: dns.TypeA, Class: dns.ClassINET, Ttl: 3600},
|
||||
A: net.ParseIP("159.75.200.247"),
|
||||
}
|
||||
m.Answer = append(m.Answer, a)
|
||||
case -1:
|
||||
m.MsgHdr.Rcode = dns.RcodeNameError
|
||||
}
|
||||
w.WriteMsg(m)
|
||||
}
|
||||
*/
|
||||
|
||||
func getResults(c *gin.Context) {
|
||||
token, err := strconv.Atoi(c.Param("token"))
|
||||
if err != nil {
|
||||
c.IndentedJSON(http.StatusNotFound, gin.H{"message": "wrong token"})
|
||||
return
|
||||
}
|
||||
if data, ok := datastruct.Result_map[token]; ok {
|
||||
c.IndentedJSON(http.StatusOK, data)
|
||||
delete(datastruct.Result_map, token)
|
||||
return
|
||||
}
|
||||
c.IndentedJSON(http.StatusNotFound, gin.H{"message": "result not found"})
|
||||
}
|
||||
|
||||
func main() {
|
||||
flag.StringVar(&CONFIG_SLD, "sld", "", "configure sld for echo dns server")
|
||||
flag.Parse()
|
||||
if CONFIG_SLD == "" {
|
||||
panic("Please configure the SLD for the echo dns server!\n")
|
||||
}
|
||||
|
||||
router := gin.Default()
|
||||
router.GET("/results/:token", getResults)
|
||||
|
||||
go router.Run(":8888")
|
||||
|
||||
dns.HandleFunc(dns.Fqdn(CONFIG_SLD), query_handler)
|
||||
server := &dns.Server{Addr: ":53", Net: "udp"}
|
||||
if err := server.ListenAndServe(); err != nil {
|
||||
fmt.Println("Failed to set up dns server!")
|
||||
panic(err)
|
||||
}
|
||||
}
|
||||
BIN
fpdns_server_nolog
Executable file
BIN
fpdns_server_nolog
Executable file
Binary file not shown.
37
go.mod
Normal file
37
go.mod
Normal file
@@ -0,0 +1,37 @@
|
||||
module fpdns_server
|
||||
|
||||
go 1.20
|
||||
|
||||
require (
|
||||
github.com/gin-gonic/gin v1.9.1
|
||||
github.com/miekg/dns v1.1.56
|
||||
)
|
||||
|
||||
require (
|
||||
github.com/bytedance/sonic v1.9.1 // indirect
|
||||
github.com/chenzhuoyu/base64x v0.0.0-20221115062448-fe3a3abad311 // indirect
|
||||
github.com/gabriel-vasile/mimetype v1.4.2 // indirect
|
||||
github.com/gin-contrib/sse v0.1.0 // indirect
|
||||
github.com/go-playground/locales v0.14.1 // indirect
|
||||
github.com/go-playground/universal-translator v0.18.1 // indirect
|
||||
github.com/go-playground/validator/v10 v10.14.0 // indirect
|
||||
github.com/goccy/go-json v0.10.2 // indirect
|
||||
github.com/json-iterator/go v1.1.12 // indirect
|
||||
github.com/klauspost/cpuid/v2 v2.2.4 // indirect
|
||||
github.com/leodido/go-urn v1.2.4 // indirect
|
||||
github.com/mattn/go-isatty v0.0.19 // indirect
|
||||
github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect
|
||||
github.com/modern-go/reflect2 v1.0.2 // indirect
|
||||
github.com/pelletier/go-toml/v2 v2.0.8 // indirect
|
||||
github.com/twitchyliquid64/golang-asm v0.15.1 // indirect
|
||||
github.com/ugorji/go/codec v1.2.11 // indirect
|
||||
golang.org/x/arch v0.3.0 // indirect
|
||||
golang.org/x/crypto v0.13.0 // indirect
|
||||
golang.org/x/mod v0.12.0 // indirect
|
||||
golang.org/x/net v0.15.0 // indirect
|
||||
golang.org/x/sys v0.12.0 // indirect
|
||||
golang.org/x/text v0.13.0 // indirect
|
||||
golang.org/x/tools v0.13.0 // indirect
|
||||
google.golang.org/protobuf v1.30.0 // indirect
|
||||
gopkg.in/yaml.v3 v3.0.1 // indirect
|
||||
)
|
||||
92
go.sum
Normal file
92
go.sum
Normal file
@@ -0,0 +1,92 @@
|
||||
github.com/bytedance/sonic v1.5.0/go.mod h1:ED5hyg4y6t3/9Ku1R6dU/4KyJ48DZ4jPhfY1O2AihPM=
|
||||
github.com/bytedance/sonic v1.9.1 h1:6iJ6NqdoxCDr6mbY8h18oSO+cShGSMRGCEo7F2h0x8s=
|
||||
github.com/bytedance/sonic v1.9.1/go.mod h1:i736AoUSYt75HyZLoJW9ERYxcy6eaN6h4BZXU064P/U=
|
||||
github.com/chenzhuoyu/base64x v0.0.0-20211019084208-fb5309c8db06/go.mod h1:DH46F32mSOjUmXrMHnKwZdA8wcEefY7UVqBKYGjpdQY=
|
||||
github.com/chenzhuoyu/base64x v0.0.0-20221115062448-fe3a3abad311 h1:qSGYFH7+jGhDF8vLC+iwCD4WpbV1EBDSzWkJODFLams=
|
||||
github.com/chenzhuoyu/base64x v0.0.0-20221115062448-fe3a3abad311/go.mod h1:b583jCggY9gE99b6G5LEC39OIiVsWj+R97kbl5odCEk=
|
||||
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
|
||||
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||
github.com/gabriel-vasile/mimetype v1.4.2 h1:w5qFW6JKBz9Y393Y4q372O9A7cUSequkh1Q7OhCmWKU=
|
||||
github.com/gabriel-vasile/mimetype v1.4.2/go.mod h1:zApsH/mKG4w07erKIaJPFiX0Tsq9BFQgN3qGY5GnNgA=
|
||||
github.com/gin-contrib/sse v0.1.0 h1:Y/yl/+YNO8GZSjAhjMsSuLt29uWRFHdHYUb5lYOV9qE=
|
||||
github.com/gin-contrib/sse v0.1.0/go.mod h1:RHrZQHXnP2xjPF+u1gW/2HnVO7nvIa9PG3Gm+fLHvGI=
|
||||
github.com/gin-gonic/gin v1.9.1 h1:4idEAncQnU5cB7BeOkPtxjfCSye0AAm1R0RVIqJ+Jmg=
|
||||
github.com/gin-gonic/gin v1.9.1/go.mod h1:hPrL7YrpYKXt5YId3A/Tnip5kqbEAP+KLuI3SUcPTeU=
|
||||
github.com/go-playground/assert/v2 v2.2.0 h1:JvknZsQTYeFEAhQwI4qEt9cyV5ONwRHC+lYKSsYSR8s=
|
||||
github.com/go-playground/locales v0.14.1 h1:EWaQ/wswjilfKLTECiXz7Rh+3BjFhfDFKv/oXslEjJA=
|
||||
github.com/go-playground/locales v0.14.1/go.mod h1:hxrqLVvrK65+Rwrd5Fc6F2O76J/NuW9t0sjnWqG1slY=
|
||||
github.com/go-playground/universal-translator v0.18.1 h1:Bcnm0ZwsGyWbCzImXv+pAJnYK9S473LQFuzCbDbfSFY=
|
||||
github.com/go-playground/universal-translator v0.18.1/go.mod h1:xekY+UJKNuX9WP91TpwSH2VMlDf28Uj24BCp08ZFTUY=
|
||||
github.com/go-playground/validator/v10 v10.14.0 h1:vgvQWe3XCz3gIeFDm/HnTIbj6UGmg/+t63MyGU2n5js=
|
||||
github.com/go-playground/validator/v10 v10.14.0/go.mod h1:9iXMNT7sEkjXb0I+enO7QXmzG6QCsPWY4zveKFVRSyU=
|
||||
github.com/goccy/go-json v0.10.2 h1:CrxCmQqYDkv1z7lO7Wbh2HN93uovUHgrECaO5ZrCXAU=
|
||||
github.com/goccy/go-json v0.10.2/go.mod h1:6MelG93GURQebXPDq3khkgXZkazVtN9CRI+MGFi0w8I=
|
||||
github.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk=
|
||||
github.com/google/go-cmp v0.5.5 h1:Khx7svrCpmxxtHBq5j2mp/xVjsi8hQMfNLvJFAlrGgU=
|
||||
github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
|
||||
github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg=
|
||||
github.com/json-iterator/go v1.1.12 h1:PV8peI4a0ysnczrg+LtxykD8LfKY9ML6u2jnxaEnrnM=
|
||||
github.com/json-iterator/go v1.1.12/go.mod h1:e30LSqwooZae/UwlEbR2852Gd8hjQvJoHmT4TnhNGBo=
|
||||
github.com/klauspost/cpuid/v2 v2.0.9/go.mod h1:FInQzS24/EEf25PyTYn52gqo7WaD8xa0213Md/qVLRg=
|
||||
github.com/klauspost/cpuid/v2 v2.2.4 h1:acbojRNwl3o09bUq+yDCtZFc1aiwaAAxtcn8YkZXnvk=
|
||||
github.com/klauspost/cpuid/v2 v2.2.4/go.mod h1:RVVoqg1df56z8g3pUjL/3lE5UfnlrJX8tyFgg4nqhuY=
|
||||
github.com/leodido/go-urn v1.2.4 h1:XlAE/cm/ms7TE/VMVoduSpNBoyc2dOxHs5MZSwAN63Q=
|
||||
github.com/leodido/go-urn v1.2.4/go.mod h1:7ZrI8mTSeBSHl/UaRyKQW1qZeMgak41ANeCNaVckg+4=
|
||||
github.com/mattn/go-isatty v0.0.19 h1:JITubQf0MOLdlGRuRq+jtsDlekdYPia9ZFsB8h/APPA=
|
||||
github.com/mattn/go-isatty v0.0.19/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
|
||||
github.com/miekg/dns v1.1.56 h1:5imZaSeoRNvpM9SzWNhEcP9QliKiz20/dA2QabIGVnE=
|
||||
github.com/miekg/dns v1.1.56/go.mod h1:cRm6Oo2C8TY9ZS/TqsSrseAcncm74lfK5G+ikN2SWWY=
|
||||
github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
|
||||
github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd h1:TRLaZ9cD/w8PVh93nsPXa1VrQ6jlwL5oN8l14QlcNfg=
|
||||
github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
|
||||
github.com/modern-go/reflect2 v1.0.2 h1:xBagoLtFs94CBntxluKeaWgTMpvLxC4ur3nMaC9Gz0M=
|
||||
github.com/modern-go/reflect2 v1.0.2/go.mod h1:yWuevngMOJpCy52FWWMvUC8ws7m/LJsjYzDa0/r8luk=
|
||||
github.com/pelletier/go-toml/v2 v2.0.8 h1:0ctb6s9mE31h0/lhu+J6OPmVeDxJn+kYnJc2jZR9tGQ=
|
||||
github.com/pelletier/go-toml/v2 v2.0.8/go.mod h1:vuYfssBdrU2XDZ9bYydBu6t+6a6PYNcZljzZR9VXg+4=
|
||||
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
|
||||
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
|
||||
github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw=
|
||||
github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo=
|
||||
github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
|
||||
github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
|
||||
github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
|
||||
github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU=
|
||||
github.com/stretchr/testify v1.8.1/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4=
|
||||
github.com/stretchr/testify v1.8.2/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4=
|
||||
github.com/stretchr/testify v1.8.3 h1:RP3t2pwF7cMEbC1dqtB6poj3niw/9gnV4Cjg5oW5gtY=
|
||||
github.com/stretchr/testify v1.8.3/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo=
|
||||
github.com/twitchyliquid64/golang-asm v0.15.1 h1:SU5vSMR7hnwNxj24w34ZyCi/FmDZTkS4MhqMhdFk5YI=
|
||||
github.com/twitchyliquid64/golang-asm v0.15.1/go.mod h1:a1lVb/DtPvCB8fslRZhAngC2+aY1QWCk3Cedj/Gdt08=
|
||||
github.com/ugorji/go/codec v1.2.11 h1:BMaWp1Bb6fHwEtbplGBGJ498wD+LKlNSl25MjdZY4dU=
|
||||
github.com/ugorji/go/codec v1.2.11/go.mod h1:UNopzCgEMSXjBc6AOMqYvWC1ktqTAfzJZUZgYf6w6lg=
|
||||
golang.org/x/arch v0.0.0-20210923205945-b76863e36670/go.mod h1:5om86z9Hs0C8fWVUuoMHwpExlXzs5Tkyp9hOrfG7pp8=
|
||||
golang.org/x/arch v0.3.0 h1:02VY4/ZcO/gBOH6PUaoiptASxtXU10jazRCP865E97k=
|
||||
golang.org/x/arch v0.3.0/go.mod h1:5om86z9Hs0C8fWVUuoMHwpExlXzs5Tkyp9hOrfG7pp8=
|
||||
golang.org/x/crypto v0.13.0 h1:mvySKfSWJ+UKUii46M40LOvyWfN0s2U+46/jDd0e6Ck=
|
||||
golang.org/x/crypto v0.13.0/go.mod h1:y6Z2r+Rw4iayiXXAIxJIDAJ1zMW4yaTpebo8fPOliYc=
|
||||
golang.org/x/mod v0.12.0 h1:rmsUpXtvNzj340zd98LZ4KntptpfRHwpFOHG188oHXc=
|
||||
golang.org/x/mod v0.12.0/go.mod h1:iBbtSCu2XBx23ZKBPSOrRkjjQPZFPuis4dIYUhu/chs=
|
||||
golang.org/x/net v0.15.0 h1:ugBLEUaxABaB5AJqW9enI0ACdci2RUd4eP51NTBvuJ8=
|
||||
golang.org/x/net v0.15.0/go.mod h1:idbUs1IY1+zTqbi8yxTbhexhEEk5ur9LInksu6HrEpk=
|
||||
golang.org/x/sync v0.3.0 h1:ftCYgMx6zT/asHUrPw8BLLscYtGznsLAnjq5RH9P66E=
|
||||
golang.org/x/sys v0.0.0-20220704084225-05e143d24a9e/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.12.0 h1:CM0HF96J0hcLAwsHPJZjfdNzs0gftsLfgKt57wWHJ0o=
|
||||
golang.org/x/sys v0.12.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/text v0.13.0 h1:ablQoSUd0tRdKxZewP80B+BaqeKJuVhuRxj/dkrun3k=
|
||||
golang.org/x/text v0.13.0/go.mod h1:TvPlkZtksWOMsz7fbANvkp4WM8x/WCo/om8BMLbz+aE=
|
||||
golang.org/x/tools v0.13.0 h1:Iey4qkscZuv0VvIt8E0neZjtPVQFSc870HQ448QgEmQ=
|
||||
golang.org/x/tools v0.13.0/go.mod h1:HvlwmtVNQAhOuCjW7xxvovg8wbNq7LwfXh/k7wXUl58=
|
||||
golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543 h1:E7g+9GITq07hpfrRu66IVDexMakfv52eLZ2CXBWiKr4=
|
||||
golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
||||
google.golang.org/protobuf v1.26.0-rc.1/go.mod h1:jlhhOSvTdKEhbULTjvd4ARK9grFBp09yW+WbY/TyQbw=
|
||||
google.golang.org/protobuf v1.30.0 h1:kPPoIgf3TsEvrm0PFe15JQ+570QVxYzEvvHqChK+cng=
|
||||
google.golang.org/protobuf v1.30.0/go.mod h1:HV8QOd/L58Z+nl8r43ehVNZIU/HEI6OcFqwMG9pJV4I=
|
||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=
|
||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
|
||||
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||
rsc.io/pdf v0.1.1/go.mod h1:n8OzWcQ6Sp37PL01nO98y4iUCRdTGarVfzxY20ICaU4=
|
||||
Reference in New Issue
Block a user