Allow for QOS/DSCP to be set

speedtest.go

@@ -29,6 +29,7 @@ var (
 	noDownload   = kingpin.Flag("no-download", "Disable download test.").Bool()
 	noUpload     = kingpin.Flag("no-upload", "Disable upload test.").Bool()
 	pingMode     = kingpin.Flag("ping-mode", "Select a method for Ping. (support icmp/tcp/http)").Default("http").String()
+	dscp	     = kingpin.Flag("dscp", "type of service field in IP header").Short('Q').Int()
 	debug        = kingpin.Flag("debug", "Enable debug mode.").Short('d').Bool()
 )
 
@@ -52,6 +53,7 @@ func main() {
 			Keyword:      *search,
 			NoDownload:   *noDownload,
 			NoUpload:     *noUpload,
+			Dscp:         *dscp,
 		}))
 	speedtestClient.SetNThread(*thread)
 

speedtest/speedtest.go

@@ -7,6 +7,7 @@ import (
 	"net/url"
 	"runtime"
 	"strings"
+	"syscall"
 	"time"
 )
 
@@ -52,6 +53,7 @@ type UserConfig struct {
 
 	NoDownload bool
 	NoUpload   bool
+	Dscp       int
 }
 
 func parseAddr(addr string) (string, string) {
@@ -119,13 +121,31 @@ func (s *Speedtest) NewUserConfig(uc *UserConfig) {
 		}
 	}
 
+	setDscp := func(network, address string, c syscall.RawConn) error {
+		return c.Control(func(fd uintptr) {
+			level := syscall.IPPROTO_IP
+			optname := syscall.IP_TOS
+			if network != "tcp4" {
+				level = syscall.IPPROTO_IPV6
+				optname = syscall.IPV6_TCLASS
+			}
+			err := syscall.SetsockoptInt(int(fd), level, optname, uc.Dscp)
+			if err != nil {
+				dbg.Printf("syscall.SetsockoptInt: %s", err)
+				return
+			}
+		})
+	}
+
 	s.tcpDialer = &net.Dialer{
+		Control:   setDscp,
 		LocalAddr: tcpSource,
 		Timeout:   30 * time.Second,
 		KeepAlive: 30 * time.Second,
 	}
 
 	s.ipDialer = &net.Dialer{
+		Control: setDscp,
 		LocalAddr: icmpSource,
 		Timeout:   30 * time.Second,
 		KeepAlive: 30 * time.Second,