Fix IPv4 boundary checking

gofiber · Mar 18, 2024 · 7ca5fab · 7ca5fab
1 parent dbfbf0b
commit 7ca5fab
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/ips.go b/ips.go
@@ -21,7 +21,7 @@ func IsIPv4(s string) bool {
 
 		for ci = 0; ci < len(s) && '0' <= s[ci] && s[ci] <= '9'; ci++ {
 			n = n*10 + int(s[ci]-'0')
-			if n >= 0xFF {
+			if n > 0xFF {
 				return false
 			}
 		}

diff --git a/ips_test.go b/ips_test.go
@@ -14,6 +14,7 @@ import (
 func Test_IsIPv4(t *testing.T) {
 	t.Parallel()
 
+	require.Equal(t, true, IsIPv4("255.255.255.255"))
 	require.Equal(t, true, IsIPv4("174.23.33.100"))
 	require.Equal(t, true, IsIPv4("127.0.0.1"))
 	require.Equal(t, true, IsIPv4("0.0.0.0"))
@@ -30,6 +31,7 @@ func Test_IsIPv4(t *testing.T) {
 	require.Equal(t, false, IsIPv4("189.12.34.260"))
 	require.Equal(t, false, IsIPv4("189.12.260.260"))
 	require.Equal(t, false, IsIPv4("189.260.260.260"))
+	require.Equal(t, false, IsIPv4("255.255.255.256"))
 	require.Equal(t, false, IsIPv4("999.999.999.999"))
 	require.Equal(t, false, IsIPv4("9999.9999.9999.9999"))