Merge branch 'fasthttp' into ctxCancel

frontend/http/params_test.go

@@ -68,21 +68,18 @@ func mapArrayEqual(boxed map[string][]string, unboxed *queryParams) (bool, strin
 // Also note that any error that is encountered during parsing is returned as a
 // ClientError, as this method is expected to be used to parse client-provided
 // data.
-func parseURLData(urlData []byte) (*queryParams, error) {
+func parseURLData(urlData []byte) *queryParams {
 	i := bytes.IndexByte(urlData, '?')
 	if i >= 0 {
 		urlData = urlData[i+1:]
 	}
 	q := &queryParams{new(fasthttp.Args)}
 	q.ParseBytes(urlData)
-	return q, nil
+	return q
 }
 
 func TestParseEmptyURLData(t *testing.T) {
-	parsedQuery, err := parseURLData(nil)
-	if err != nil {
-		t.Fatal(err)
-	}
+	parsedQuery := parseURLData(nil)
 	if parsedQuery == nil {
 		t.Fatal("Parsed query must not be nil")
 	}
@@ -90,10 +87,7 @@ func TestParseEmptyURLData(t *testing.T) {
 
 func TestParseValidURLData(t *testing.T) {
 	for parseIndex, parseVal := range ValidAnnounceArguments {
-		parsedQueryObj, err := parseURLData([]byte("/announce?" + parseVal.Encode()))
-		if err != nil {
-			t.Fatal(err)
-		}
+		parsedQueryObj := parseURLData([]byte("/announce?" + parseVal.Encode()))
 
 		if eq, exp := mapArrayEqual(parseVal, parsedQueryObj); !eq {
 			t.Fatalf("Incorrect parse at item %d.\n Expected=%v\n Received=%v\n", parseIndex, parseVal, exp)
@@ -101,9 +95,9 @@ func TestParseValidURLData(t *testing.T) {
 	}
 }
 
-func TestParseShouldNotPanicURLData(t *testing.T) {
+func TestParseShouldNotPanicURLData(_ *testing.T) {
 	for _, parseStr := range shouldNotPanicQueries {
-		_, _ = parseURLData(parseStr)
+		_ = parseURLData(parseStr)
 	}
 }
 
@@ -115,11 +109,7 @@ func BenchmarkParseQuery(b *testing.B) {
 	b.ResetTimer()
 	for bCount := 0; bCount < b.N; bCount++ {
 		i := bCount % len(announceStrings)
-		parsedQueryObj, err := parseURLData(announceStrings[i])
-		if err != nil {
-			b.Error(err, i)
-			b.Log(parsedQueryObj)
-		}
+		_ = parseURLData(announceStrings[i])
 	}
 }
 

frontend/udp/connection_id.go

@@ -2,9 +2,9 @@ package udp
 
 import (
 	"crypto/hmac"
+	cr "crypto/rand"
 	"encoding/binary"
 	"hash"
-	"math/rand"
 	"net/netip"
 	"time"
 
@@ -70,18 +70,25 @@ func NewConnectionIDGenerator(key []byte, maxClockSkew time.Duration) *Connectio
 		buff:         make([]byte, buffLen),
 		scratch:      make([]byte, scratchLen),
 		maxClockSkew: int64(maxClockSkew),
-		s:            rand.Uint64(),
 	}
 }
 
 // reset resets the generator.
 // This is called by other methods of the generator, it's not necessary to call
 // it after getting a generator from a pool.
-func (g *ConnectionIDGenerator) reset() {
+func (g *ConnectionIDGenerator) reset(init bool) {
 	g.mac.Reset()
 	g.connID = g.connID[:connIDLen]
 	g.buff = g.buff[:buffLen]
 	g.scratch = g.scratch[:0]
+	if init {
+		r := make([]byte, 8)
+		if _, err := cr.Read(r); err == nil {
+			g.s = binary.BigEndian.Uint64(r)
+		} else {
+			g.s = uint64(time.Now().UnixNano())
+		}
+	}
 }
 
 // Generate generates an 8-byte connection ID as described in BEP 15 for the
@@ -102,7 +109,7 @@ func (g *ConnectionIDGenerator) reset() {
 // will be reused, so it must not be referenced after returning the generator
 // to a pool and will be overwritten be subsequent calls to Generate!
 func (g *ConnectionIDGenerator) Generate(ip netip.Addr, now time.Time) (out []byte) {
-	g.reset()
+	g.reset(true)
 	var r uint64
 	r, g.s = xorshift.XorShift64S(g.s)
 	g.buff[0] = byte(r)
@@ -123,7 +130,7 @@ func (g *ConnectionIDGenerator) Generate(ip netip.Addr, now time.Time) (out []by
 
 // Validate validates the given connection ID for an IP and the current time.
 func (g *ConnectionIDGenerator) Validate(connectionID []byte, ip netip.Addr, now time.Time) bool {
-	g.reset()
+	g.reset(false)
 	nowTS := now.Unix()
 	g.buff[0] = connectionID[0]
 	// connectionID contains only 2 bytes of timestamp, so we clean little 16 bits to place it and rehash.

frontend/udp/connection_id_test.go

@@ -13,7 +13,6 @@ import (
 
 	"github.com/cespare/xxhash/v2"
 	"github.com/sot-tech/mochi/pkg/log"
-	_ "github.com/sot-tech/mochi/pkg/randseed"
 	"github.com/stretchr/testify/require"
 )
 

frontend/udp/frontend_test.go

@@ -7,7 +7,6 @@ import (
 	"github.com/sot-tech/mochi/middleware"
 	"github.com/sot-tech/mochi/pkg/conf"
 	"github.com/sot-tech/mochi/pkg/log"
-	_ "github.com/sot-tech/mochi/pkg/randseed"
 	"github.com/sot-tech/mochi/storage"
 	_ "github.com/sot-tech/mochi/storage/memory"
 )

frontend/udp/params_test.go

@@ -88,7 +88,9 @@ func TestParseInvalidURLData(t *testing.T) {
 
 func TestParseShouldNotPanicURLData(t *testing.T) {
 	for _, parseStr := range shouldNotPanicQueries {
-		_, _ = parseQuery(parseStr)
+		if _, err := parseQuery(parseStr); err != nil {
+			t.Error(err)
+		}
 	}
 }