增加参数范围校验

2025-07-24 18:02:14 +08:00
parent 1f2709326f
commit 7e4bc7f87b
3 changed files with 133 additions and 25 deletions
@@ -0,0 +1,70 @@
+package com.sunyard.util;
+
+import java.util.regex.Pattern;
+
+public class AddressUtil {
+    // 处理带端口号的地址
+    public static boolean isValidAddressWithPort(String address) {
+        if (address == null) {
+            return false;
+        }
+
+        // 分离 IP:port 或 [IPv6]:port
+        String[] parts;
+        if (address.startsWith("[") && address.contains("]:")) {
+            parts = address.split("\\]:", 2);
+            parts[0] = parts[0] + "]"; // 补全 IPv6 的括号
+        } else {
+            parts = address.split(":", 2);
+        }
+
+        if (parts.length != 2) {
+            return false;
+        }
+
+        String ipPart = parts[0];
+        String portPart = parts[1];
+
+        return isValidIP(ipPart) &&
+                isValidPort(portPart);
+    }
+
+    // IPv4 正则（允许 0-255）
+    private static final Pattern IPV4_PATTERN = Pattern.compile(
+            "^((25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\\.){3}(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)$"
+    );
+
+    // IPv6 正则（包含缩写形式）
+    private static final Pattern IPV6_PATTERN = Pattern.compile(
+            "^(?:[0-9a-fA-F]{1,4}:){7}[0-9a-fA-F]{1,4}$"
+    );
+
+    // 简化域名正则（允许字母、数字、连字符和点）
+    private static final Pattern DOMAIN_PATTERN = Pattern.compile(
+            "^([a-zA-Z0-9-]{1,63}\\.)+[a-zA-Z]{2,63}$"
+    );
+
+    public static boolean isValidIP(String ip) {
+        // 检查 IPv4 或 IPv6
+        return IPV4_PATTERN.matcher(ip).matches() ||
+                IPV6_PATTERN.matcher(ip).matches();
+    }
+
+    public static boolean isValidDomain(String domain) {
+        return DOMAIN_PATTERN.matcher(domain).matches();
+    }
+
+    public static boolean isValidPort(int port) {
+        return port >= 0 && port <= 65535;
+    }
+
+    // 字符串形式的端口校验
+    public static boolean isValidPort(String portStr) {
+        try {
+            int port = Integer.parseInt(portStr);
+            return port >= 0 && port <= 65535;
+        } catch (NumberFormatException e) {
+            return false;
+        }
+    }
+}