Merge branch 'main' of github.com:vulncheck-oss/go-exploit

Author: lobsterjerusalem

go.mod

@@ -13,7 +13,7 @@ require (
 	golang.org/x/crypto v0.43.0
 	golang.org/x/net v0.46.0
 	golang.org/x/text v0.30.0
-	modernc.org/sqlite v1.39.1
+	modernc.org/sqlite v1.40.0
 )
 
 require (

go.sum

@@ -146,8 +146,8 @@ modernc.org/opt v0.1.4 h1:2kNGMRiUjrp4LcaPuLY2PzUfqM/w9N23quVwhKt5Qm8=
 modernc.org/opt v0.1.4/go.mod h1:03fq9lsNfvkYSfxrfUhZCWPk1lm4cq4N+Bh//bEtgns=
 modernc.org/sortutil v1.2.1 h1:+xyoGf15mM3NMlPDnFqrteY07klSFxLElE2PVuWIJ7w=
 modernc.org/sortutil v1.2.1/go.mod h1:7ZI3a3REbai7gzCLcotuw9AC4VZVpYMjDzETGsSMqJE=
-modernc.org/sqlite v1.39.1 h1:H+/wGFzuSCIEVCvXYVHX5RQglwhMOvtHSv+VtidL2r4=
-modernc.org/sqlite v1.39.1/go.mod h1:9fjQZ0mB1LLP0GYrp39oOJXx/I2sxEnZtzCmEQIKvGE=
+modernc.org/sqlite v1.40.0 h1:bNWEDlYhNPAUdUdBzjAvn8icAs/2gaKlj4vM+tQ6KdQ=
+modernc.org/sqlite v1.40.0/go.mod h1:9fjQZ0mB1LLP0GYrp39oOJXx/I2sxEnZtzCmEQIKvGE=
 modernc.org/strutil v1.2.1 h1:UneZBkQA+DX2Rp35KcM69cSsNES9ly8mQWD71HKlOA0=
 modernc.org/strutil v1.2.1/go.mod h1:EHkiggD70koQxjVdSBM3JKM7k6L0FbGE5eymy9i3B9A=
 modernc.org/token v1.1.0 h1:Xl7Ap9dKaEs5kLoOQeQmPWevfnk/DM5qcLcYlA8ys6Y=

random/random.go

@@ -152,6 +152,44 @@ func RandIPv4() net.IP {
 	)
 }
 
+// RandIPv4Private Generates a random private IPv4 address.
+func RandIPv4Private() net.IP {
+	n, _ := rand.Int(rand.Reader, big.NewInt(3))
+	switch n.Int64() {
+	case 0:
+		return net.IPv4(
+			10,
+			byte(RandIntRange(1, 256)),
+			byte(RandIntRange(1, 256)),
+			byte(RandIntRange(1, 256)),
+		)
+	case 1:
+		return net.IPv4(
+			172,
+			byte(RandIntRange(16, 32)),
+			byte(RandIntRange(1, 256)),
+			byte(RandIntRange(1, 256)),
+		)
+	default:
+		return net.IPv4(
+			192,
+			168,
+			byte(RandIntRange(1, 256)),
+			byte(RandIntRange(1, 256)),
+		)
+	}
+}
+
+// RandomIPv4Loopback generates IPv4 Loopback address
+func RandIPv4Loopback() net.IP {
+	return net.IPv4(
+		127,
+		byte(RandIntRange(1, 256)),
+		byte(RandIntRange(1, 256)),
+		byte(RandIntRange(1, 256)),
+	)
+}
+
 // RandIPv6 generates a random IPv6 address.
 func RandIPv6() net.IP {
 	ip := make(net.IP, net.IPv6len)

random/random_test.go

@@ -192,3 +192,43 @@ func TestRandIPv6(t *testing.T) {
 		}
 	}
 }
+
+func TestRandIPv4Private(t *testing.T) {
+	for range 100 {
+		r := RandIPv4Private()
+		parsed := net.ParseIP(r.String())
+		if parsed == nil {
+			t.Error("Generated IP is nil: " + r.String())
+		}
+		if parsed.To4() == nil {
+			t.Error("Generated IP is not a valid IPv4 address: " + r.String())
+		}
+
+		// Check if the IP is private
+		isPrivate := r.IsPrivate()
+
+		if !isPrivate {
+			t.Error("Generated IP is not in a private range: " + r.String())
+		}
+	}
+}
+
+func TestRandIPv4Loopback(t *testing.T) {
+	for range 5 {
+		r := RandIPv4Loopback()
+		parsed := net.ParseIP(r.String())
+		if parsed == nil {
+			t.Error("Generated IP is nil: " + r.String())
+		}
+		if parsed.To4() == nil {
+			t.Error("Generated IP is not a valid IPv4 address: " + r.String())
+		}
+
+		// Check if the IP is loopback
+		isLoopback := r.IsLoopback()
+
+		if !isLoopback {
+			t.Error("Generated IP is not the expected loopback address:" + r.String())
+		}
+	}
+}