Improve internal tooling

cmds/trafficgen/main.go

@@ -0,0 +1,98 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+	"net/http"
+	"os"
+	"time"
+
+	"github.com/miekg/dns"
+
+	"github.com/safing/portbase/log"
+)
+
+const (
+	dnsResolver = "1.1.1.1:53"
+)
+
+var (
+	url      string
+	lookup   string
+	n        int
+	waitMsec int
+)
+
+func init() {
+	flag.StringVar(&url, "url", "", "send HTTP HEAD requests to this url")
+	flag.StringVar(&lookup, "lookup", "", fmt.Sprintf("query %s for this domains", dnsResolver))
+	flag.IntVar(&n, "n", 1000, "how many requests to make")
+	flag.IntVar(&waitMsec, "w", 100, "how many ms to wait between requests")
+}
+
+func main() {
+	// Parse flags
+	flag.Parse()
+	if url == "" && lookup == "" {
+		flag.Usage()
+		os.Exit(1)
+	}
+
+	// Start logging.
+	err := log.Start()
+	if err != nil {
+		fmt.Printf("failed to start logging: %s\n", err)
+		os.Exit(1)
+	}
+	defer log.Shutdown()
+	log.SetLogLevel(log.TraceLevel)
+	log.Info("starting traffic generator")
+
+	// Execute requests
+	waitDuration := time.Duration(waitMsec) * time.Millisecond
+	for i := 0; i < n; i++ {
+		makeHTTPRequest()
+		lookupDomain()
+		time.Sleep(waitDuration)
+	}
+}
+
+func makeHTTPRequest() {
+	if url == "" {
+		return
+	}
+
+	// Create a new client so that the connection won't be shared with other requests.
+	client := http.Client{
+		CheckRedirect: func(*http.Request, []*http.Request) error { return http.ErrUseLastResponse },
+	}
+	start := time.Now()
+	resp, err := client.Head(url)
+	if err != nil {
+		log.Errorf("http request failed after %s: %s", time.Since(start).Round(time.Millisecond), err)
+		return
+	}
+	defer resp.Body.Close()
+
+	log.Infof("http response after %s: %d", time.Since(start).Round(time.Millisecond), resp.StatusCode)
+}
+
+func lookupDomain() {
+	if lookup == "" {
+		return
+	}
+
+	// Create DNS query.
+	dnsQuery := new(dns.Msg)
+	dnsQuery.SetQuestion(dns.Fqdn(lookup), dns.TypeA)
+
+	// Send request.
+	start := time.Now()
+	reply, err := dns.Exchange(dnsQuery, dnsResolver)
+	if err != nil {
+		log.Errorf("dns request failed after %s: %s", time.Since(start).Round(time.Millisecond), err)
+		return
+	}
+
+	log.Infof("dns response after %s: %s", time.Since(start).Round(time.Millisecond), dns.RcodeToString[reply.Rcode])
+}

cmds/trafficgen/pack

@@ -0,0 +1,18 @@
+#!/bin/bash
+
+baseDir="$( cd "$(dirname "$0")" && pwd )"
+cd "$baseDir"
+
+echo "building..."
+GOOS=windows go build
+
+dstFile="../../../portmaster-windows-kext/install/MINGW/amd64/trafficgen.exe"
+if [[ -d $(dirname "$dstFile") ]]; then
+  # make path absolute
+  dstFile="$(cd "$(dirname "$dstFile")" && pwd)/$(basename "$dstFile")"
+  # copy
+  echo "copying to $dstFile"
+  cp trafficgen.exe "$dstFile"
+else
+  echo "not copying to $dstFile (dir does not exist)"
+fi