From 61bed9275526243288101b12637a5c9df0472448 Mon Sep 17 00:00:00 2001
From: "cn.yimian.xyz" <i@iotcat.me>
Date: Mon, 2 Nov 2020 16:16:45 +0800
Subject: [PATCH] add ip confine

---
 getIp.php   | 18 ++++++++++++++++++
 getRank.php | 18 ++++++++++++++++++
 index.php   | 11 +++++++++--
 3 files changed, 45 insertions(+), 2 deletions(-)
 create mode 100644 getIp.php
 create mode 100644 getRank.php

diff --git a/getIp.php b/getIp.php
new file mode 100644
index 0000000..435eff8
--- /dev/null
+++ b/getIp.php
@@ -0,0 +1,18 @@
+<?php
+
+
+$redis = new redis(); 
+$redis->connect('redis',6379);
+
+$keys = $redis->keys('api/img/_ip/*');
+
+
+$res = array();
+foreach($keys as $key){
+    $res[$key] = $redis->get($key);
+}
+
+arsort($res);
+
+foreach($res as $a => $b)
+echo "$a => $b <br>";
diff --git a/getRank.php b/getRank.php
new file mode 100644
index 0000000..7ebb3f7
--- /dev/null
+++ b/getRank.php
@@ -0,0 +1,18 @@
+<?php
+
+
+$redis = new redis(); 
+$redis->connect('redis',6379);
+
+$keys = $redis->keys('api/img/_num/*');
+
+
+$res = array();
+foreach($keys as $key){
+    $res[$key] = $redis->get($key);
+}
+
+arsort($res);
+
+foreach($res as $a => $b)
+echo "$a => $b <br>";
diff --git a/index.php b/index.php
index a9ce8b7..b7b4e5c 100755
--- a/index.php
+++ b/index.php
@@ -67,37 +67,44 @@ $__from = get_from();
 if($__from == '') $__from = 'http://blank.com/';
 $__from = parse_url($__from)['host'];
 
-if(!in_array($__from, $whiteList) && $type == "moe"){
+if($type == "moe"){
     $redis = new redis(); 
     $redis->connect('redis',6379);
 
 
     $__time = 'api/img/_time/'.$__from;
     $__num = 'api/img/_num/'.$__from;
+    $__ip = 'api/img/_ip/'.$__from.'/'.getIp();
 
     if(!$redis->exists($__time) || !$redis->exists($__num)){
         $redis->set($__time, time());
         $redis->set($__num, -200);
+        $redis->set($__ip, 0);
     }
 
     $_time = $redis->get($__time);
     $_num = $redis->get($__num);
+    $_ip = $redis->get($__ip);
 
     if(time() - $_time > 60*60*24){
         $redis->set($__time, time());
         $redis->set($__num, 0);
+        $redis->set($__ip, 0);
         $_num = $redis->get($__num);
+        $_ip = $redis->get($__ip);
         $_time = $redis->get($__time);
     } 
 
     
     $redis->set($__num, $_num+1);
+    $redis->set($__ip, $_ip+1);
 
-    if($_num > 50) {
+    if((!in_array($__from, $whiteList) && $_num > 50) || $_ip > 100) {
         header("Location: https://api.vvhan.com/api/acgimg");
         yimian__log("log_api", array("api" => "img", "timestamp" => date('Y-m-d H:i:s', time()), "ip" => ip2long(getIp()), "_from" => get_from(), "content" => 'https://api.vvhan.com/api/acgimg')); 
         die();
     }
+
 }