Edit File by line

<?php

[0] Fix | Delete

[1] Fix | Delete

require_once __DIR__ . '/../vendor/autoload.php';

[2] Fix | Delete

[3] Fix | Delete

require_once __DIR__ . '/wfIpLocation.php';

[4] Fix | Delete

[5] Fix | Delete

use Wordfence\MmdbReader\Database;

[6] Fix | Delete

use Wordfence\MmdbReader\Exception\MmdbThrowable;

[7] Fix | Delete

[8] Fix | Delete

class wfIpLocator {

[9] Fix | Delete

[10] Fix | Delete

const SOURCE_BUNDLED = 0;

[11] Fix | Delete

const SOURCE_WFLOGS = 1;

[12] Fix | Delete

[13] Fix | Delete

const DATABASE_FILE_NAME = 'GeoLite2-Country.mmdb';

[14] Fix | Delete

[15] Fix | Delete

private static $instances = array();

[16] Fix | Delete

[17] Fix | Delete

private $database;

[18] Fix | Delete

private $preferred;

[19] Fix | Delete

[20] Fix | Delete

private function __construct($database, $preferred) {

[21] Fix | Delete

$this->database = $database;

[22] Fix | Delete

$this->preferred = $preferred;

[23] Fix | Delete

}

[24] Fix | Delete

[25] Fix | Delete

public function isPreferred() {

[26] Fix | Delete

return $this->preferred;

[27] Fix | Delete

}

[28] Fix | Delete

[29] Fix | Delete

private static function logError($message) {

[30] Fix | Delete

if (class_exists('wfUtils'))

[31] Fix | Delete

wfUtils::check_and_log_last_error('ip_locator', 'IP Location Error:', $message, 0);

[32] Fix | Delete

}

[33] Fix | Delete

[34] Fix | Delete

public function locate($ip) {

[35] Fix | Delete

if ($this->database !== null) {

[36] Fix | Delete

try {

[37] Fix | Delete

$record = $this->database->search($ip);

[38] Fix | Delete

if ($record !== null)

[39] Fix | Delete

return new wfIpLocation($record);

[40] Fix | Delete

}

[41] Fix | Delete

catch (MmdbThrowable $t) {

[42] Fix | Delete

self::logError('Failed to locate IP address: ' . $t->getMessage());

[43] Fix | Delete

}

[44] Fix | Delete

}

[45] Fix | Delete

return null;

[46] Fix | Delete

}

[47] Fix | Delete

[48] Fix | Delete

public function getCountryCode($ip, $default = '') {

[49] Fix | Delete

$record = $this->locate($ip);

[50] Fix | Delete

if ($record !== null)

[51] Fix | Delete

return $record->getCountryCode();

[52] Fix | Delete

return $default;

[53] Fix | Delete

}

[54] Fix | Delete

[55] Fix | Delete

public function getDatabaseVersion() {

[56] Fix | Delete

if ($this->database !== null) {

[57] Fix | Delete

try {

[58] Fix | Delete

return $this->database->getMetadata()->getBuildEpoch();

[59] Fix | Delete

}

[60] Fix | Delete

catch (MmdbThrowable $t) {

[61] Fix | Delete

self::logError('Failed to retrieve database version: ' . $t->getMessage());

[62] Fix | Delete

}

[63] Fix | Delete

}

[64] Fix | Delete

return null;

[65] Fix | Delete

}

[66] Fix | Delete

[67] Fix | Delete

private static function getDatabaseDirectory($source) {

[68] Fix | Delete

switch ($source) {

[69] Fix | Delete

case self::SOURCE_BUNDLED:

[70] Fix | Delete

return WFWAF_LOG_PATH;

[71] Fix | Delete

case self::SOURCE_BUNDLED:

[72] Fix | Delete

default:

[73] Fix | Delete

return __DIR__;

[74] Fix | Delete

}

[75] Fix | Delete

}

[76] Fix | Delete

[77] Fix | Delete

private static function initializeDatabase($preferredSource, &$isPreferred) {

[78] Fix | Delete

$sources = array();

[79] Fix | Delete

if ($preferredSource !== self::SOURCE_BUNDLED)

[80] Fix | Delete

$sources[] = $preferredSource;

[81] Fix | Delete

$sources[] = self::SOURCE_BUNDLED;

[82] Fix | Delete

$isPreferred = true;

[83] Fix | Delete

foreach ($sources as $source) {

[84] Fix | Delete

$directory = self::getDatabaseDirectory($source);

[85] Fix | Delete

try {

[86] Fix | Delete

$path = $directory . '/' . self::DATABASE_FILE_NAME;

[87] Fix | Delete

if (file_exists($path)) //Preemptive check to prevent warnings

[88] Fix | Delete

return Database::open($path);

[89] Fix | Delete

}

[90] Fix | Delete

catch (MmdbThrowable $t) {

[91] Fix | Delete

self::logError('Failed to initialize IP location database: ' . $t->getMessage());

[92] Fix | Delete

}

[93] Fix | Delete

$preferred = false;

[94] Fix | Delete

}

[95] Fix | Delete

return null;

[96] Fix | Delete

}

[97] Fix | Delete

[98] Fix | Delete

public static function getInstance($preferredSource = null) {

[99] Fix | Delete

if ($preferredSource === null)

[100] Fix | Delete

$preferredSource = self::SOURCE_WFLOGS;

[101] Fix | Delete

if (!array_key_exists($preferredSource, self::$instances)) {

[102] Fix | Delete

$database = self::initializeDatabase($preferredSource, $isPreferred);

[103] Fix | Delete

self::$instances[$preferredSource] = new wfIpLocator($database, $isPreferred);

[104] Fix | Delete

}

[105] Fix | Delete

return self::$instances[$preferredSource];

[106] Fix | Delete

}

[107] Fix | Delete

[108] Fix | Delete

}

[109] Fix | Delete