SensitiveMiddleware.php 2.9 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182
  1. <?php
  2. declare(strict_types=1);
  3. namespace App\Middleware\Auth;
  4. use App\Tools\PublicData;
  5. use Hyperf\HttpServer\Contract\RequestInterface;
  6. use Hyperf\HttpServer\Contract\ResponseInterface as HttpResponse;
  7. use Psr\Container\ContainerInterface;
  8. use Psr\Http\Message\ResponseInterface;
  9. use Psr\Http\Message\ServerRequestInterface;
  10. use Psr\Http\Server\MiddlewareInterface;
  11. use Psr\Http\Server\RequestHandlerInterface;
  12. use Hyperf\HttpMessage\Stream\SwooleStream;
  13. use Hyperf\Di\Annotation\Inject;
  14. use Hyperf\Redis\Redis;
  15. class SensitiveMiddleware implements MiddlewareInterface
  16. {
  17. protected ContainerInterface $container;
  18. protected RequestInterface $request;
  19. protected HttpResponse $response;
  20. #[Inject]
  21. protected Redis $redis;
  22. public function __construct( RequestInterface $request, HttpResponse $response)
  23. {
  24. $this->request = $request;
  25. $this->response = $response;
  26. }
  27. public function process(ServerRequestInterface $request, RequestHandlerInterface $handler): ResponseInterface
  28. {
  29. try {
  30. $badWords = $this->redis->sMembers('black_word'); //黑名单
  31. $whiteWords = $this->redis->sMembers('white_word');//白名单
  32. // 获取所有请求参数并拼接成文本
  33. $params = $this->request->all();
  34. $concatenated = "";
  35. if($params){
  36. foreach ($params as $value) {
  37. if (is_array($value)) {
  38. // 如果值是数组,将数组元素用逗号连接
  39. $concatenated.= json_encode($value);
  40. } else {
  41. // 如果不是数组,直接拼接
  42. $concatenated.= $value;
  43. }
  44. }
  45. }
  46. // 遍历违禁词,检查是否在文本中存在
  47. foreach ($badWords as $badWord) {
  48. // 判断该违禁词是否在白名单中,如果在则跳过
  49. if (in_array($badWord, $whiteWords)) {
  50. continue;
  51. }
  52. if (str_contains($concatenated, $badWord)) {
  53. // 处理找到违禁词的情况,例如返回错误信息
  54. $message = '发现违禁词: '. $badWord;
  55. return $this->response->json(
  56. [
  57. 'code' => 0,
  58. 'data' => [],
  59. 'message' => $message
  60. ]
  61. );
  62. }
  63. }
  64. return $handler->handle($request);
  65. }catch (\Exception $e){
  66. return $this->response->json(
  67. [
  68. 'code' => $e->getCode(),
  69. 'data' => [],
  70. 'message' => '敏感词检测失败:'.$e->getMessage(),
  71. ]
  72. );
  73. }
  74. return false;
  75. }
  76. }