浏览代码

add protection

andre 4 年之前
父节点
当前提交
bdaed7f0d9
共有 2 个文件被更改,包括 12 次插入1 次删除
  1. 11 0
      action.php
  2. 1 1
      admin.php

+ 11 - 0
action.php

@@ -77,6 +77,17 @@ class action {
         return $list;
     }
 
+    public function isBlocked($ip) {
+        $dbAction = $this->db->prepare("SELECT ip FROM list UNION SELECT ip FROM blacklist ORDER BY ip ASC");
+        $dbAction->bind_param('i',$ip);
+        $dbAction->execute();
+        $dbAction->store_result();
+        if($dbAction->num_rows() == 0) {
+            return false;
+        }
+        return true;
+    }
+
     private function isListed($ip) {
         $dbAction = $this->db->prepare("SELECT ip FROM list WHERE ip = ?");
         $dbAction->bind_param('i',$ip);

+ 1 - 1
admin.php

@@ -40,7 +40,7 @@ $stats = true;
 $tok = new token;
 $act = new action;
 
-if($act->isListed($_SERVER['REMOTE_ADDR'])) {
+if($act->isBlocked($_SERVER['REMOTE_ADDR'])) {
     http_response_code(403);
     exit;
 }