增加安全中间件

app/middleware/SecurityMiddleware.php

@@ -326,7 +326,7 @@ class SecurityMiddleware
      */
     protected function handleAttack(Request $request, array $attacks): void
     {
-        $ip = $request->ip();
+        $ip = $this->getRealIp($request);
 
         // 记录攻击日志
         foreach ($attacks as $attack) {
@@ -340,6 +340,47 @@ class SecurityMiddleware
     }
 
     /**
+     * 获取真实客户端IP（支持Swoole模式）
+     */
+    protected function getRealIp(Request $request): string
+    {
+        $ip = '0.0.0.0';
+
+        // 1. 优先从Swoole的header中获取
+        if ($this->isSwooleMode()) {
+            // Swoole模式下，真实IP通常在x-real-ip或x-forwarded-for中
+            $ip = $request->header('x-real-ip', '');
+
+            if (empty($ip)) {
+                // 如果有多个代理，x-forwarded-for是逗号分隔的列表，取第一个
+                $xff = $request->header('x-forwarded-for', '');
+                if (!empty($xff)) {
+                    $ips = explode(',', $xff);
+                    $ip = trim($ips[0]);
+                }
+            }
+        }
+
+        // 2. 如果Swoole模式获取失败，尝试常规方法
+        if (empty($ip) || !filter_var($ip, FILTER_VALIDATE_IP)) {
+            $ip = $request->ip();
+        }
+
+        // 3. 验证IP格式
+        return filter_var($ip, FILTER_VALIDATE_IP) ? $ip : '0.0.0.0';
+    }
+
+    /**
+     * 检查是否运行在Swoole模式
+     */
+    protected function isSwooleMode(): bool
+    {
+        return extension_loaded('swoole') &&
+            defined('SWOOLE_VERSION') &&
+            (php_sapi_name() === 'cli' || isset($_SERVER['SWOOLE_SERVER']));
+    }
+
+    /**
      * 记录攻击日志到数据库
      */
     protected function logAttack(string $ip, Request $request, array $attack): void
@@ -449,18 +490,10 @@ class SecurityMiddleware
      */
     protected function blockResponse(): Response
     {
-        throw new \Exception('访问被拒绝', 403);
-        // return response()
-        //     ->json([
-        //         'code' => 403,
-        //         'message' => '访问被拒绝',
-        //         'data' => null
-        //     ])
-        //     ->code(403)
-        //     ->header([
-        //         'Content-Type' => 'application/json',
-        //         'Retry-After' => 3600
-        //     ]);
+        return Response::create([
+            'code' => 403,
+            'msg'  => '访问被拒绝'
+        ], 'json')->code(403);
     }
 
     /**
@@ -468,13 +501,9 @@ class SecurityMiddleware
      */
     protected function errorResponse(): Response
     {
-        throw new \Exception('请求参数错误', 400);
-        // return response()
-        //     ->json([
-        //         'code' => 400,
-        //         'message' => '请求参数错误',
-        //         'data' => null
-        //     ])
-        //     ->code(400);
+        return Response::create([
+            'code' => 400,
+            'msg'  => '请求参数错误'
+        ], 'json')->code(400);
     }
 }