diff options
| author | MDK <[email protected]> | 2024-06-03 19:22:35 +0800 |
|---|---|---|
| committer | MDK <[email protected]> | 2024-06-03 19:22:35 +0800 |
| commit | 4b8110b6b018b41ffbddce4dfd3c4af16bd80ba4 (patch) | |
| tree | 375fe780e929555c6fd2b5226055acbed867b3dc /injection_probe.go | |
Diffstat (limited to 'injection_probe.go')
| -rw-r--r-- | injection_probe.go | 121 |
1 files changed, 121 insertions, 0 deletions
diff --git a/injection_probe.go b/injection_probe.go new file mode 100644 index 0000000..44ae277 --- /dev/null +++ b/injection_probe.go @@ -0,0 +1,121 @@ +package main
+
+import (
+ "flag"
+ "fmt"
+ "strconv"
+ "strings"
+ "sync"
+ "time"
+
+ "github.com/miekg/dns"
+ "github.com/rs/zerolog"
+ "github.com/rs/zerolog/log"
+)
+
+func dnsQuery(ip, domain string) (*dns.Msg, error) {
+ domain = dns.Fqdn(domain)
+ addr := ip + ":53"
+ m := new(dns.Msg)
+ m.SetQuestion(domain, dns.TypeA)
+ m.RecursionDesired = true
+ res, err := dns.Exchange(m, addr)
+ return res, err
+}
+
+func main() {
+ domain := flag.String("domain", "", "Target domain")
+ targetIP := flag.String("server", "", "server IP address")
+ packetsPerSecond := flag.Int("speed", 0, "Packets per second")
+ packetNum := flag.Int("num", 1000, "total packets")
+ expectedResult := flag.String("truth", "", "Expected result")
+ random_flag := flag.Bool("random", false, "random subdomain")
+ flag.Parse()
+
+ if *domain == "" || *targetIP == "" || *packetsPerSecond == 0 {
+ flag.Usage()
+ return
+ }
+
+ currentTime := strconv.FormatInt(time.Now().Unix(), 10)
+ totalRequests := *packetNum
+ responseTimes := make([]int64, 0, totalRequests)
+ successfulRequests := 0
+ validResponses := 0
+
+ zerolog.TimeFieldFormat = zerolog.TimeFormatUnix
+
+ var query_tasks sync.WaitGroup
+ fmt.Println("testing......")
+ for i := 0; i < totalRequests; i++ {
+ query_tasks.Add(1)
+ go func() {
+ var prefix string
+ start := time.Now()
+ if *random_flag {
+ prefix = strings.Join([]string{currentTime, strconv.Itoa(i), ""}, "-")
+ } else {
+ prefix = ""
+ }
+ query_dn := prefix + (*domain)
+ res, err := dnsQuery(*targetIP, query_dn)
+ elapsed := time.Since(start).Microseconds()
+ if err == nil {
+ successfulRequests += 1
+ log.Log().Msg("gfw worked!")
+ responseTimes = append(responseTimes, elapsed)
+ if *expectedResult == "" {
+ if *random_flag {
+ if (len(res.Answer) != 1) || (res.MsgHdr.Rcode != dns.RcodeSuccess) {
+ validResponses += 1
+ log.Log().Msg("gfw escaped!")
+ //fmt.Printf("answer: %d rcode: %d\n", len(res.Answer), res.MsgHdr.Rcode)
+ }
+ } else {
+ if len(res.Answer) > 1 {
+ validResponses += 1
+ log.Log().Msg("gfw escaped!")
+ //fmt.Println(elapsed)
+ }
+ }
+ } else {
+ if a, ok := res.Answer[0].(*dns.A); ok {
+ //fmt.Println(a.A.String())
+ if a.A.String() == *expectedResult {
+ validResponses += 1
+ log.Log().Msg("gfw escaped!")
+ //fmt.Println(elapsed)
+ }
+ }
+ }
+ }
+
+ query_tasks.Done()
+ }()
+ time.Sleep(time.Second / time.Duration(*packetsPerSecond))
+ }
+
+ query_tasks.Wait()
+
+ var sum_rtt int64
+ var max_rtt int64
+ var min_rtt int64
+ sum_rtt = 0
+ max_rtt = 0
+ min_rtt = 3000000
+ for _, rtt := range responseTimes {
+ sum_rtt += rtt
+ if rtt < min_rtt {
+ min_rtt = rtt
+ }
+ if rtt > max_rtt {
+ max_rtt = rtt
+ }
+ }
+ avg_rtt := float64(sum_rtt) / float64(len(responseTimes))
+
+ valid_rate := float64(validResponses) / float64(successfulRequests)
+
+ fmt.Println(avg_rtt, max_rtt, min_rtt, successfulRequests, validResponses, valid_rate)
+
+}
|