71 lines
1.4 KiB
PHP
71 lines
1.4 KiB
PHP
<?php
|
|
|
|
require_once(dirname(__FILE__) . '/../vendor/autoload.php');
|
|
|
|
use GeoIp2\Database\Reader;
|
|
|
|
if (!defined('WFWAF_RUN_COMPLETE')) {
|
|
class wfWAFGeoIP2 {
|
|
private $_reader;
|
|
|
|
/**
|
|
* Returns the singleton wfWAFGeoIP2.
|
|
*
|
|
* @return wfWAFGeoIP2
|
|
*/
|
|
public static function shared() {
|
|
static $_geoip = null;
|
|
if ($_geoip === null) {
|
|
$_geoip = new wfWAFGeoIP2();
|
|
}
|
|
return $_geoip;
|
|
}
|
|
|
|
public function __construct() {
|
|
try {
|
|
if (file_exists(WFWAF_LOG_PATH . '/GeoLite2-Country.mmdb')) {
|
|
$this->_reader = new Reader(WFWAF_LOG_PATH . '/GeoLite2-Country.mmdb');
|
|
return;
|
|
}
|
|
}
|
|
catch (Exception $e) {
|
|
//Fall through to bundled copy
|
|
}
|
|
|
|
$this->_reader = new Reader(__DIR__ . '/../lib/GeoLite2-Country.mmdb'); //Can throw, but we don't catch it because it means the installation is likely corrupt and needs fixed anyway
|
|
}
|
|
|
|
/**
|
|
* Returns the database version in use. This is the timestamp of when it was packaged.
|
|
*
|
|
* @return null|int
|
|
*/
|
|
public function version() {
|
|
try {
|
|
return $this->_reader->metadata()->buildEpoch;
|
|
}
|
|
catch (Exception $e) {
|
|
//Fall through
|
|
}
|
|
return null;
|
|
}
|
|
|
|
/**
|
|
* Returns the country code for the IP if known.
|
|
*
|
|
* @param string $ip
|
|
* @return null|string
|
|
*/
|
|
public function countryCode($ip) {
|
|
try {
|
|
$record = $this->_reader->country($ip);
|
|
return $record->country->isoCode;
|
|
}
|
|
catch (Exception $e) {
|
|
//Fall through
|
|
}
|
|
return null;
|
|
}
|
|
}
|
|
}
|