1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114

<?php

if (!defined('ABSPATH')) exit;
if (!class_exists('BVIPStoreCallback')) :

require_once dirname( __FILE__ ) . '/../../protect/wp/ipstore.php';

class BVIPStoreCallback extends BVCallbackBase {
    public $db;

    public function __construct($callback_handler) {
        $this->db = $callback_handler->db;
    }

    public function updateBVTableContent($table, $value, $filter) {
        $this->db->query("UPDATE $table SET $value $filter;");
    }

    public function insertBVTableContent($table, $fields, $value) {
        $this->db->query("INSERT INTO $table $fields values $value;");
    }

    public function deleteIPs($table, $rmfilters) {
        if (is_array($rmfilters)) {
            foreach ($rmfilters as $rmfilter) {
                $rmfilter = base64_decode($rmfilter);
                $this->db->deleteBVTableContent($table, $rmfilter);
            }
        }
    }

    public function insertIPs($table, $fields, $values) {
        if (is_array($values)) {
            foreach ($values as $value) {
                $value = base64_decode($value);
                $this->insertBVTableContent($table, $fields, $value);
            }
        }
    }

    public function updateIPs($table, $value, $filters) {
        if (is_array($filters)) {
            foreach ($filters as $filter) {
                $filter = base64_decode($filter);
                $this->updateBVTableContent($table, $value, $filter);
            }
        }
    }

    public function getIPs($table, $auto_increment_offset, $type, $category) {
        $query = "SELECT `start_ip_range` FROM $table WHERE id < $auto_increment_offset AND `type` = $type AND ";
        $query .= ($category == BVIPStore::FW) ? "`is_fw` = true;" : "`is_lp` = true;";
        return $this->db->getCol($query);
    }

    public function getIPStoreOffset($table, $auto_increment_offset) {
        $db = $this->db;
        return intval($db->getVar("SELECT MAX(id) FROM $table WHERE id < $auto_increment_offset"));
    }

    public function getIPStoreInfo($table, $auto_increment_offset) {
            $db = $this->db;
            $info = array();
            $info['fw_blacklisted_ips'] = $this->getIPs($table, $auto_increment_offset, BVIPStore::BLACKLISTED, BVIPStore::FW);
            $info['lp_blacklisted_ips'] = $this->getIPs($table, $auto_increment_offset, BVIPStore::BLACKLISTED, BVIPStore::LP);
            $info['fw_whitelisted_ips'] = $this->getIPs($table, $auto_increment_offset, BVIPStore::WHITELISTED, BVIPStore::FW);
            $info['lp_whitelisted_ips'] = $this->getIPs($table, $auto_increment_offset, BVIPStore::WHITELISTED, BVIPStore::LP);
            $info['ip_store_offset'] = $this->getIPStoreOffset($table, $auto_increment_offset);
            $info['country_ips_size'] = intval($db->getVar("SELECT COUNT(id) FROM $table WHERE id >= $auto_increment_offset"));
            return $info;
    }

    public function process($request) {
        $db = $this->db;
        $params = $request->params;
        $table = $params['table'];
        $bvTable = $db->getBVTable($table);
        $auto_increment_offset = $params['auto_increment_offset'];
        if (!$db->isTablePresent($bvTable)) {
            $resp = array("info" => false);
        } else {
            switch ($request->method) {
            case "ipstrinfo":
                $info = $this->getIPStoreInfo($bvTable, $auto_increment_offset);
                $resp = array("info" => $info);
                break;
            case "insrtips":
                $values = $params['values'];
                $fields = $params['fields'];
                if (array_key_exists('rmfilter', $params)) {
                    $db->deleteBVTableContent($table, $params['rmfilter']);
                }
                $this->insertIPs($bvTable, $fields, $values);
                $resp = array("offset" => $this->getIPStoreOffset($bvTable, $auto_increment_offset));
                break;
            case "dltips":
                $rmfilters = $params['rmfilters'];
                $this->deleteIPs($table, $rmfilters);
                $resp = array("offset" => $this->getIPStoreOffset($bvTable, $auto_increment_offset));
                break;
            case "updtips":
                $value = $params['value'];
                $filters = $params['filters'];
                $this->updateIPs($bvTable, $value, $filters);
                $resp = array("offset" => $this->getIPStoreOffset($bvTable, $auto_increment_offset));
                break;
            default:
                $resp = false;
            }
            return $resp;
        }
    }
}
endif;