wfIpLocator.php — Wordfence Security – Firewall, Malware Scan, and Login Security 8.0.5

wordfence / lib / wfIpLocator.php

wordfence / lib Last commit date

wfIpLocator.php

110 lines

1	<?php
2
3	require_once __DIR__ . '/../vendor/autoload.php';
4
5	require_once __DIR__ . '/wfIpLocation.php';
6
7	use Wordfence\MmdbReader\Database;
8	use Wordfence\MmdbReader\Exception\MmdbThrowable;
9
10	class wfIpLocator {
11
12	const SOURCE_BUNDLED = 0;
13	const SOURCE_WFLOGS = 1;
14
15	const DATABASE_FILE_NAME = 'GeoLite2-Country.mmdb';
16
17	private static $instances = array();
18
19	private $database;
20	private $preferred;
21
22	private function __construct($database, $preferred) {
23	$this->database = $database;
24	$this->preferred = $preferred;
25	}
26
27	public function isPreferred() {
28	return $this->preferred;
29	}
30
31	private static function logError($message) {
32	if (class_exists('wfUtils'))
33	wfUtils::check_and_log_last_error('ip_locator', 'IP Location Error:', $message, 0);
34	}
35
36	public function locate($ip) {
37	if ($this->database !== null) {
38	try {
39	$record = $this->database->search($ip);
40	if ($record !== null)
41	return new wfIpLocation($record);
42	}
43	catch (MmdbThrowable $t) {
44	self::logError('Failed to locate IP address: ' . $t->getMessage());
45	}
46	}
47	return null;
48	}
49
50	public function getCountryCode($ip, $default = '') {
51	$record = $this->locate($ip);
52	if ($record !== null)
53	return $record->getCountryCode();
54	return $default;
55	}
56
57	public function getDatabaseVersion() {
58	if ($this->database !== null) {
59	try {
60	return $this->database->getMetadata()->getBuildEpoch();
61	}
62	catch (MmdbThrowable $t) {
63	self::logError('Failed to retrieve database version: ' . $t->getMessage());
64	}
65	}
66	return null;
67	}
68
69	private static function getDatabaseDirectory($source) {
70	switch ($source) {
71	case self::SOURCE_BUNDLED:
72	return WFWAF_LOG_PATH;
73	case self::SOURCE_BUNDLED:
74	default:
75	return __DIR__;
76	}
77	}
78
79	private static function initializeDatabase($preferredSource, &$isPreferred) {
80	$sources = array();
81	if ($preferredSource !== self::SOURCE_BUNDLED)
82	$sources[] = $preferredSource;
83	$sources[] = self::SOURCE_BUNDLED;
84	$isPreferred = true;
85	foreach ($sources as $source) {
86	$directory = self::getDatabaseDirectory($source);
87	try {
88	$path = $directory . '/' . self::DATABASE_FILE_NAME;
89	if (file_exists($path)) //Preemptive check to prevent warnings
90	return Database::open($path);
91	}
92	catch (MmdbThrowable $t) {
93	self::logError('Failed to initialize IP location database: ' . $t->getMessage());
94	}
95	$preferred = false;
96	}
97	return null;
98	}
99
100	public static function getInstance($preferredSource = null) {
101	if ($preferredSource === null)
102	$preferredSource = self::SOURCE_WFLOGS;
103	if (!array_key_exists($preferredSource, self::$instances)) {
104	$database = self::initializeDatabase($preferredSource, $isPreferred);
105	self::$instances[$preferredSource] = new wfIpLocator($database, $isPreferred);
106	}
107	return self::$instances[$preferredSource];
108	}
109
110	}