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('BVIPStore')) :

    class BVIPStore {

        public $db;
        public static $name = 'ip_store';

        #TYPE
        const BLACKLISTED = 1;
        const WHITELISTED = 2;

        #CATEGORY
        const FW = 3;
        const LP = 4;

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

        function init() {
            add_action('clear_ip_store', array($this, 'clearConfig'));
        }

        public function clearConfig() {
            $this->db->dropBVTable(BVIPStore::$name);
        }

        public function hasIPv6Support() {
            return defined('AF_INET6');
        }

        public static function isValidIP($ip) {
            return filter_var($ip, FILTER_VALIDATE_IP) !== false;
        }

        public function bvInetPton($ip) {
            $pton = $this->isValidIP($ip) ? ($this->hasIPv6Support() ? inet_pton($ip) : $this->_bvInetPton($ip)) : false;
            return $pton;
        }

        public function _bvInetPton($ip) {
            if (preg_match('/^(?:\d{1,3}(?:\.|$)){4}/', $ip)) {
                $octets = explode('.', $ip);
                $bin = chr($octets[0]) . chr($octets[1]) . chr($octets[2]) . chr($octets[3]);
                return $bin;
            }

            if (preg_match('/^((?:[\da-f]{1,4}(?::|)){0,8})(::)?((?:[\da-f]{1,4}(?::|)){0,8})$/i', $ip)) {
                if ($ip === '::') {
                    return "\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0";
                }
                $colon_count = substr_count($ip, ':');
                $dbl_colon_pos = strpos($ip, '::');
                if ($dbl_colon_pos !== false) {
                    $ip = str_replace('::', str_repeat(':0000',
                        (($dbl_colon_pos === 0 || $dbl_colon_pos === strlen($ip) - 2) ? 9 : 8) - $colon_count) . ':', $ip);
                    $ip = trim($ip, ':');
                }

                $ip_groups = explode(':', $ip);
                $ipv6_bin = '';
                foreach ($ip_groups as $ip_group) {
                    $ipv6_bin .= pack('H*', str_pad($ip_group, 4, '0', STR_PAD_LEFT));
                }

                return strlen($ipv6_bin) === 16 ? $ipv6_bin : false;
            }

            if (preg_match('/^(?:\:(?:\:0{1,4}){0,4}\:|(?:0{1,4}\:){5})ffff\:(\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3})$/i', $ip, $matches)) {
                $octets = explode('.', $matches[1]);
                return chr($octets[0]) . chr($octets[1]) . chr($octets[2]) . chr($octets[3]);
            }

            return false;
        }

        public function isLPIPBlacklisted($ip) {
            return $this->checkIPPresent($ip, BVIPStore::BLACKLISTED, BVIPStore::LP);
        }

        public function isLPIPWhitelisted($ip) {
            return $this->checkIPPresent($ip, BVIPStore::WHITELISTED, BVIPStore::LP);
        }


        public function isFWIPBlacklisted($ip) {
            return $this->checkIPPresent($ip, BVIPStore::BLACKLISTED, BVIPStore::FW);
        }

        public function isFWIPWhitelisted($ip) {
            return $this->checkIPPresent($ip, BVIPStore::WHITELISTED, BVIPStore::FW);
        }

        public function checkIPPresent($ip, $type, $category) {
            $db = $this->db;
            $table = $db->getBVTable(BVIPStore::$name);
            if ($db->isTablePresent($table)) {
                $binIP = $this->bvInetPton($ip);
                if ($binIP !== false) {
                    $category_str = ($category == BVIPStore::FW) ? "`is_fw` = true" : "`is_lp` = true";
                    $query_str = "SELECT * FROM $table WHERE %s >= `start_ip_range` && %s <= `end_ip_range` && " . $category_str . " && `type` = %d LIMIT 1;";
                    $query = $db->prepare($query_str, array($binIP, $binIP, $type));
                    if ($db->getVar($query) > 0)
                        return true;
                }
                return false;
            }
            return false;
        }

    }
endif;