vikarti.anatra/safing-portmaster
1
0
Fork 0
mirror of https://github.com/safing/portmaster synced 2025-09-02 18:49:14 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Improve trafficgen tool

Browse source
This commit is contained in:
Daniel 2020-12-11 17:02:58 +01:00
parent f32e11e0ff
commit bf64996b2f
1 changed files with 10 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
cmds/trafficgen/main.go
View file

@ -26,7 +26,7 @@ var (
func init() { func init() {
flag.StringVar(&url, "url", "", "send HTTP HEAD requests to this url") flag.StringVar(&url, "url", "", "send HTTP HEAD requests to this url")
flag.StringVar(&lookup, "lookup", "", fmt.Sprintf("query %s for this domains", dnsResolver)) flag.StringVar(&lookup, "lookup", "", fmt.Sprintf("query %s for this domains", dnsResolver))
flag.IntVar(&n, "n", 1000, "how many requests to make") flag.IntVar(&n, "n", 10, "how many requests to make")
flag.IntVar(&waitMsec, "w", 100, "how many ms to wait between requests") flag.IntVar(&waitMsec, "w", 100, "how many ms to wait between requests")
} }
@ -50,14 +50,14 @@ func main() {
// Execute requests // Execute requests
waitDuration := time.Duration(waitMsec) * time.Millisecond waitDuration := time.Duration(waitMsec) * time.Millisecond
for i := 0; i < n; i++ { for i := 1; i <= n; i++ {
makeHTTPRequest() makeHTTPRequest(i)
lookupDomain() lookupDomain(i)
time.Sleep(waitDuration) time.Sleep(waitDuration)
} }
} }
func makeHTTPRequest() { func makeHTTPRequest(i int) {
if url == "" { if url == "" {
return return
} }
@ -69,15 +69,15 @@ func makeHTTPRequest() {
start := time.Now() start := time.Now()
resp, err := client.Head(url) resp, err := client.Head(url)
if err != nil { if err != nil {
log.Errorf("http request failed after %s: %s", time.Since(start).Round(time.Millisecond), err) log.Errorf("http request #%d failed after %s: %s", i, time.Since(start).Round(time.Millisecond), err)
return return
} }
defer resp.Body.Close() defer resp.Body.Close()
log.Infof("http response after %s: %d", time.Since(start).Round(time.Millisecond), resp.StatusCode) log.Infof("http response #%d after %s: %d", i, time.Since(start).Round(time.Millisecond), resp.StatusCode)
} }
func lookupDomain() { func lookupDomain(i int) {
if lookup == "" { if lookup == "" {
return return
} }
@ -90,9 +90,9 @@ func lookupDomain() {
start := time.Now() start := time.Now()
reply, err := dns.Exchange(dnsQuery, dnsResolver) reply, err := dns.Exchange(dnsQuery, dnsResolver)
if err != nil { if err != nil {
log.Errorf("dns request failed after %s: %s", time.Since(start).Round(time.Millisecond), err) log.Errorf("dns request #%d failed after %s: %s", i, time.Since(start).Round(time.Millisecond), err)
return return
} }
log.Infof("dns response after %s: %s", time.Since(start).Round(time.Millisecond), dns.RcodeToString[reply.Rcode]) log.Infof("dns response #%d after %s: %s", i, time.Since(start).Round(time.Millisecond), dns.RcodeToString[reply.Rcode])
} }