346 lines
No EOL
11 KiB
PHP
Executable file
346 lines
No EOL
11 KiB
PHP
Executable file
<?php
|
|
|
|
/**
|
|
*
|
|
* DB-IP.com database query and management class
|
|
*
|
|
* Copyright (C) 2022 db-ip.com
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, write to the Free Software
|
|
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
*
|
|
*/
|
|
|
|
|
|
|
|
use PDO;
|
|
use Exception;
|
|
use PDOException;
|
|
|
|
|
|
class DBIPException extends Exception {
|
|
|
|
}
|
|
|
|
class DBIP {
|
|
|
|
const VERSION = 4;
|
|
|
|
const TYPE_COUNTRY_LITE = 1;
|
|
const TYPE_CITY_LITE = 2;
|
|
const TYPE_COUNTRY_V4 = 3;
|
|
const TYPE_LOCATION_V4 = 4;
|
|
const TYPE_ISP_V4 = 5;
|
|
const TYPE_FULL_V4 = 6;
|
|
|
|
const TYPE_COUNTRY_V3 = 7;
|
|
const TYPE_LOCATION_V3 = 8;
|
|
const TYPE_ISP_V3 = 9;
|
|
const TYPE_FULL_V3 = 10;
|
|
|
|
const TYPE_LOCATION_V2 = 11;
|
|
const TYPE_ISP_V2 = 12;
|
|
const TYPE_FULL_V2 = 13;
|
|
|
|
const TYPE_COUNTRY_IPINFO = 15;
|
|
|
|
const TYPE_COUNTRY = self::TYPE_COUNTRY_V4;
|
|
const TYPE_LOCATION = self::TYPE_LOCATION_V4;
|
|
const TYPE_ISP = self::TYPE_ISP_V4;
|
|
const TYPE_FULL = self::TYPE_FULL_V4;
|
|
|
|
private $db;
|
|
protected $dbQuoteChar = "";
|
|
|
|
public function __construct(PDO $db, $dbQuoteChar = "") {
|
|
$this->db = $db;
|
|
$this->db->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);
|
|
$this->dbQuoteChar = $dbQuoteChar;
|
|
}
|
|
|
|
public function importFromCsv($fileName, $type, $tableName = "dbip_lookup", $progressCallback = null) {
|
|
switch ($type) {
|
|
case self::TYPE_COUNTRY_LITE:
|
|
$fields = array("ip_start", "ip_end", "country");
|
|
$fieldsprepare = $fields;
|
|
break;
|
|
case self::TYPE_COUNTRY_IPINFO:
|
|
$fields = array("ip_start", "ip_end", "country", "d1", "d2", "d3");
|
|
$fieldsprepare = array("ip_start", "ip_end", "country");
|
|
break;
|
|
case self::TYPE_CITY_LITE:
|
|
$fields = array("ip_start", "ip_end", "continent", "country", "stateprov", "city", "latitude", "longitude");
|
|
break;
|
|
case self::TYPE_COUNTRY_V4:
|
|
case self::TYPE_COUNTRY_V3:
|
|
$fields = array("ip_start", "ip_end", "continent", "country");
|
|
break;
|
|
case self::TYPE_LOCATION_V4:
|
|
$fields = array("ip_start", "ip_end", "continent", "country", "stateprov_code", "stateprov", "district", "city", "zipcode", "latitude", "longitude", "geoname_id", "timezone_offset", "timezone_name", "weather_code");
|
|
break;
|
|
case self::TYPE_ISP_V4:
|
|
$fields = array("ip_start", "ip_end", "continent", "country", "isp_name", "as_number", "usage_type", "connection_type", "organization_name");
|
|
break;
|
|
case self::TYPE_FULL_V4:
|
|
$fields = array("ip_start", "ip_end", "continent", "country", "stateprov_code", "stateprov", "district", "city", "zipcode", "latitude", "longitude", "geoname_id", "timezone_offset", "timezone_name", "weather_code", "isp_name", "as_number", "usage_type", "connection_type", "organization_name");
|
|
break;
|
|
case self::TYPE_LOCATION_V3:
|
|
$fields = array("ip_start", "ip_end", "continent", "country", "stateprov", "district", "city", "zipcode", "latitude", "longitude", "geoname_id", "timezone_offset", "timezone_name", "weather_code");
|
|
break;
|
|
case self::TYPE_ISP_V3:
|
|
$fields = array("ip_start", "ip_end", "continent", "country", "isp_name", "as_number", "connection_type", "organization_name");
|
|
break;
|
|
case self::TYPE_FULL_V3:
|
|
$fields = array("ip_start", "ip_end", "continent", "country", "stateprov", "district", "city", "zipcode", "latitude", "longitude", "geoname_id", "timezone_offset", "timezone_name", "weather_code", "isp_name", "as_number", "connection_type", "organization_name");
|
|
break;
|
|
case self::TYPE_LOCATION_V2:
|
|
$fields = array("ip_start", "ip_end", "country", "stateprov", "district", "city", "zipcode", "latitude", "longitude", "geoname_id", "timezone_offset", "timezone_name");
|
|
break;
|
|
case self::TYPE_ISP_V2:
|
|
$fields = array("ip_start", "ip_end", "country", "isp_name", "connection_type", "organization_name");
|
|
break;
|
|
case self::TYPE_FULL_V2:
|
|
$fields = array("ip_start", "ip_end", "country", "stateprov", "district", "city", "zipcode", "latitude", "longitude", "geoname_id", "timezone_offset", "timezone_name", "isp_name", "connection_type", "organization_name");
|
|
break;
|
|
default:
|
|
throw new DBIPException("invalid database type");
|
|
}
|
|
if (!file_exists($fileName)) {
|
|
throw new DBIPException("file {$fileName} does not exists");
|
|
}
|
|
$q = $this->prepareImport($tableName, $fieldsprepare);
|
|
if (preg_match('/\.gz(\..+)?$/', $fileName)) {
|
|
$openName = "compress.zlib://" . $fileName;
|
|
} else {
|
|
$openName = $fileName;
|
|
}
|
|
$f = @fopen($openName, "r");
|
|
if (!is_resource($f)) {
|
|
throw new DBIPException("cannot open {$fileName} for reading");
|
|
}
|
|
$nrecs = 0;
|
|
while ($r = fgetcsv($f, 1024, ",", '"', "\0")) {
|
|
foreach ($r as $k => $v) {
|
|
if (!isset($fields[$k])) {
|
|
throw new DBIPException("invalid CSV record for {$r[0]}");
|
|
}
|
|
switch ($fields[$k]) {
|
|
case "latitude":
|
|
case "longitude":
|
|
case "timezone_offset":
|
|
$r[$k] = (float)$v;
|
|
break;
|
|
case "d1":
|
|
case "d2":
|
|
case "d3":
|
|
unset($r[$k]);
|
|
break;
|
|
case "isp_name":
|
|
case "organization_name":
|
|
$r[$k] = mb_substr($v, 0, 128);
|
|
break;
|
|
case "city":
|
|
case "district":
|
|
case "stateprov":
|
|
$r[$k] = mb_substr($v, 0, 80);
|
|
break;
|
|
case "zipcode":
|
|
$r[$k] = mb_substr($v, 0, 20);
|
|
break;
|
|
case "timezone_name":
|
|
$r[$k] = mb_substr($v, 0, 64);
|
|
break;
|
|
case "connection_type":
|
|
case "stateprov_code":
|
|
case "usage_type":
|
|
case "geoname_id":
|
|
case "as_number":
|
|
if (!$r[$k]) $r[$k] = null;
|
|
break;
|
|
default:
|
|
}
|
|
}
|
|
|
|
try {
|
|
/*$r[] = self::addrType($r[0]);
|
|
$r[0] = inet_pton($r[0]);
|
|
$r[1] = inet_pton($r[1]);*/
|
|
|
|
$set = [
|
|
0 => inet_pton($r[0]),
|
|
1 => inet_pton($r[1]),
|
|
2 => $r[2],
|
|
3 => self::addrType($r[0])
|
|
];
|
|
$this->importRow($q, $set);
|
|
} catch (Exception $e) {
|
|
throw new Exception("cannot import record #{$nrecs}: {$e->getMessage()}");
|
|
}
|
|
if ((($nrecs % 100) === 0) && is_callable($progressCallback)) {
|
|
call_user_func($progressCallback, $nrecs);
|
|
}
|
|
$nrecs++;
|
|
}
|
|
fclose($f);
|
|
$this->finalizeImport();
|
|
return $nrecs;
|
|
}
|
|
|
|
public function lookup($addr, $tableName = "dbip_lookup") {
|
|
if ($ret = $this->doLookup($tableName, self::addrType($addr), inet_pton($addr))) {
|
|
$ret->ip_start = inet_ntop($ret->ip_start);
|
|
$ret->ip_end = inet_ntop($ret->ip_end);
|
|
return $ret;
|
|
} else {
|
|
throw new DBIPException("address not found");
|
|
}
|
|
}
|
|
|
|
protected function prepareImport($tableName, array $fields) {
|
|
try {
|
|
if ($this->db->query("select count(*) as cnt from {$this->dbQuoteChar}{$tableName}{$this->dbQuoteChar}")->fetchObject()->cnt) {
|
|
throw new DBIPException("table {$tableName} is not empty");
|
|
}
|
|
$this->db->setAttribute(PDO::ATTR_EMULATE_PREPARES, false);
|
|
$this->db->beginTransaction();
|
|
$q = $this->db->prepare("insert into {$this->dbQuoteChar}{$tableName}{$this->dbQuoteChar} (" . implode(",", $fields) . ",addr_type) values (" . implode(",", array_fill(0, count($fields), "?")) . ",?)");
|
|
return $q;
|
|
} catch (PDOException $e) {
|
|
throw new DBIPException("database error: {$e->getMessage()}");
|
|
}
|
|
}
|
|
|
|
protected function finalizeImport() {
|
|
try{
|
|
$this->db->commit();
|
|
}catch (Exception $e) {
|
|
throw new Exception("commit import record: {$e->getMessage()}");
|
|
}
|
|
|
|
}
|
|
|
|
protected function importRow($res, $row) {
|
|
return $res->execute($row);
|
|
}
|
|
|
|
protected function doLookup($tableName, $addrType, $addrStart) {
|
|
$q = $this->db->prepare("select * from {$this->dbQuoteChar}{$tableName}{$this->dbQuoteChar} where addr_type = ? and ip_start <= ? order by ip_start desc limit 1");
|
|
$q->execute(array($addrType, $addrStart));
|
|
return $q->fetchObject();
|
|
}
|
|
|
|
static private function addrType($addr) {
|
|
if (ip2long($addr) !== false) {
|
|
return "ipv4";
|
|
} else if (preg_match('/^[0-9a-fA-F:]+$/', $addr) && @inet_pton($addr)) {
|
|
return "ipv6";
|
|
}
|
|
throw new DBIPException("unknown address type for {$addr}");
|
|
}
|
|
|
|
}
|
|
|
|
/**
|
|
* DBIP MySQL class
|
|
*
|
|
* If you are using a MySQL database backend, please prefer the DBIP base class which uses the PDO mysql driver.
|
|
* This is only meant to be used on PHP installations where PDO is disabled and the old mysql_* interface is available.
|
|
*/
|
|
class DBIPMySQL extends DBIP {
|
|
|
|
private $db;
|
|
|
|
public function __construct($db) {
|
|
$this->db = $db;
|
|
}
|
|
|
|
protected function prepareImport($tableName, array $fields) {
|
|
$q = mysql_query("select count(*) as cnt from {$this->dbQuoteChar}{$tableName}{$this->dbQuoteChar}", $this->db);
|
|
$r = mysql_fetch_object($q);
|
|
mysql_free_result($q);
|
|
if ($r->cnt) {
|
|
throw new DBIPException("table {$tableName} is not empty");
|
|
}
|
|
return array($tableName, $fields);
|
|
}
|
|
|
|
protected function finalizeImport() {
|
|
}
|
|
|
|
protected function importRow($res, $row) {
|
|
$vals = array();
|
|
foreach ($row as $val) {
|
|
$vals[] = "'" . mysql_real_escape_string($val) . "'";
|
|
}
|
|
if (!mysql_query("insert into {$this->dbQuoteChar}{$res[0]}{$this->dbQuoteChar} (" . implode(",", $res[1]) . ",addr_type) values (" . implode(",", $vals) . ")", $this->db)) {
|
|
throw new DBIPException("database error: cannot insert row");
|
|
}
|
|
return true;
|
|
}
|
|
|
|
protected function doLookup($tableName, $addrType, $addrStart) {
|
|
$q = mysql_query("select * from {$this->dbQuoteChar}{$tableName}{$this->dbQuoteChar} where addr_type = '{$addrType}' and ip_start <= '" . mysql_real_escape_string($addrStart) . "' order by ip_start desc limit 1", $this->db);
|
|
$r = mysql_fetch_object($q);
|
|
mysql_free_result($q);
|
|
return $r;
|
|
}
|
|
|
|
}
|
|
|
|
/**
|
|
* DBIP MySQLI class
|
|
*
|
|
* If you are using a MySQL database backend, please prefer the DBIP base class which uses the PDO mysql driver.
|
|
* This is only meant to be used on PHP installations where PDO is disabled and the old mysqli_* interface is available.
|
|
*/
|
|
class DBIPMySQLI extends DBIP {
|
|
|
|
private $db;
|
|
|
|
public function __construct($db) {
|
|
$this->db = $db;
|
|
}
|
|
|
|
protected function prepareImport($tableName, array $fields) {
|
|
$q = mysqli_query($this->db, "select count(*) as cnt from {$this->dbQuoteChar}{$tableName}{$this->dbQuoteChar}");
|
|
$r = mysqli_fetch_object($q);
|
|
mysqli_free_result($q);
|
|
if ($r->cnt) {
|
|
throw new DBIPException("table {$tableName} is not empty");
|
|
}
|
|
return array($tableName, $fields);
|
|
}
|
|
|
|
protected function finalizeImport() {
|
|
}
|
|
|
|
protected function importRow($res, $row) {
|
|
$vals = array();
|
|
foreach ($row as $val) {
|
|
$vals[] = "'" . mysqli_real_escape_string($this->db, $val) . "'";
|
|
}
|
|
if (!mysqli_query($this->db, "insert into {$this->dbQuoteChar}{$res[0]}{$this->dbQuoteChar} (" . implode(",", $res[1]) . ",addr_type) values (" . implode(",", $vals) . ")")) {
|
|
throw new DBIPException("database error: cannot insert row");
|
|
}
|
|
return true;
|
|
}
|
|
|
|
protected function doLookup($tableName, $addrType, $addrStart) {
|
|
$q = mysqli_query($this->db, "select * from {$this->dbQuoteChar}{$tableName}{$this->dbQuoteChar} where addr_type = '{$addrType}' and ip_start <= '" . mysqli_real_escape_string($this->db, $addrStart) . "' order by ip_start desc limit 1");
|
|
$r = mysqli_fetch_object($q);
|
|
mysqli_free_result($q);
|
|
return $r;
|
|
}
|
|
|
|
} |