feat: 增加IP数据库调用

This commit is contained in:
Ying
2023-07-04 18:42:41 +08:00
parent 78160b8058
commit 4c1f46e36a
5 changed files with 267 additions and 15 deletions

View File

@@ -15,11 +15,9 @@ namespace app\admin\controller\system;
use app\admin\service\UserService;
use app\AdminController;
use app\common\exception\OperateException;
use app\common\library\Ip2Region;
use app\common\model\system\User as UserModel;
use app\common\model\system\UserGroup as UserGroupModel;
use support\Response;
use system\Random;
use think\db\exception\DataNotFoundException;
use think\db\exception\DbException;
use think\db\exception\ModelNotFoundException;

View File

@@ -12,9 +12,9 @@ declare (strict_types=1);
namespace app\admin\service;
use app\common\exception\OperateException;
use app\common\library\Ip2Region;
use app\common\model\system\User;
use app\common\model\system\UserGroup;
use system\IpLocation;
use system\Random;
use think\db\exception\DataNotFoundException;
use think\db\exception\DbException;
@@ -54,14 +54,15 @@ class UserService
$list = $model->where($conditions)->order("id asc")->limit($limit)->page($page)->select();
// 循环处理数据
$userGroup = (new UserGroup)->select()->toArray();
$qqWry = new IpLocation();
foreach ($list as $key => $value) {
$value->hidden(['pwd', 'salt']);
try {
$region = Ip2Region::instance()->memorySearch($value['login_ip']);
$region = $qqWry->getLocation($value['login_ip']);
$region = $region['country'] . ' ' . $region['area'];
} catch (\Exception $e) {
$region = ['region' => '未知|未知|未知'];
$region = 'unKnown';
}
$region = explode('|',$region['region']);
$list[$key]['region'] = $region;
$result = list_search($userGroup,['id'=> $value['group_id']]);
if (!empty($result)) {

View File

@@ -7,7 +7,6 @@ declare (strict_types = 1);
* Time: 上午9:11
*/
namespace SensitiveHelper;
use app\common\controller\common\model\system\Tags;
class SensitiveHelper
{
@@ -81,14 +80,6 @@ class SensitiveHelper
*/
public function setTree($sensitiveWords = null, bool $type = true)
{
// 默认从数据库读取
if (empty($sensitiveWords)) {
$sensitiveWords = Tags::where([
'type'=> $type,
'status'=> true,
])->column('name');
}
$this->wordTree = new HashMap();
foreach ($sensitiveWords as $word) {
$this->buildWordToTree($word);

BIN
extend/conf/ip/qqwry.dat Normal file

Binary file not shown.

View File

@@ -0,0 +1,262 @@
<?php
// +----------------------------------------------------------------------
// | ThinkPHP [ WE CAN DO IT JUST THINK IT ]
// +----------------------------------------------------------------------
// | Copyright (c) 2009 http://thinkphp.cn All rights reserved.
// +----------------------------------------------------------------------
// | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
// +----------------------------------------------------------------------
// | Author: liu21st <liu21st@gmail.com>
// +----------------------------------------------------------------------
namespace system;
/**
* IP 地理位置查询类 修改自 CoolCode.CN
* 由于使用UTF8编码 如果使用纯真IP地址库的话 需要对返回结果进行编码转换
* @author liu21st <liu21st@gmail.com>
*/
class IpLocation
{
/**
* QQWry.Dat文件指针
*
* @var resource
*/
private $fp;
/**
* 第一条IP记录的偏移地址
*
* @var int
*/
private int $firstIp;
/**
* 最后一条IP记录的偏移地址
*
* @var int
*/
private int $lastIp;
/**
* IP记录的总条数不包含版本信息记录
*
* @var int
*/
private int $totalIp;
/**
* 构造函数,打开 QQWry.Dat 文件并初始化类中的信息
*
* @param string $filePath
* @throws \Exception
*/
public function __construct(string $filePath = '')
{
if (empty($filePath)) {
$filePath = root_path() . 'extend/conf/ip/qqwry.dat';
}
if (!is_file($filePath)) {
throw new \Exception('文件不存在');
}
$this->fp = fopen($filePath, 'rb');
$this->firstIp = $this->getLong();
$this->lastIp = $this->getLong();
$this->totalIp = ($this->lastIp - $this->firstIp) / 7;
}
/**
* 返回读取的长整型数
*
* @access private
* @return int
*/
private function getLong(): int
{
//将读取的little-endian编码的4个字节转化为长整型数
$result = unpack('Vlong', fread($this->fp, 4));
return $result['long'];
}
/**
* 返回读取的3个字节的长整型数
*
* @access private
* @return int
*/
private function getLong3(): int
{
//将读取的little-endian编码的3个字节转化为长整型数
$result = unpack('Vlong', fread($this->fp, 3) . chr(0));
return $result['long'];
}
/**
* 返回压缩后可进行比较的IP地址
*
* @access private
* @param string $ip
* @return string
*/
private function packIp(string $ip): string
{
// 将IP地址转化为长整型数如果在PHP5中IP地址错误则返回False
// 这时intval将Flase转化为整数-1之后压缩成big-endian编码的字符串
return pack('N', intval(ip2long($ip)));
}
/**
* 返回读取的字符串
*
* @access private
* @param string $data
* @return string
*/
private function getString(string $data = ""): string
{
$char = fread($this->fp, 1);
while (ord($char) > 0) {
// 字符串按照C格式保存以\0结束
$data .= $char; // 将读取的字符连接到给定字符串之后
$char = fread($this->fp, 1);
}
return $data;
}
/**
* 返回地区信息
*
* @access private
* @return string
*/
private function getArea(): string
{
$byte = fread($this->fp, 1); // 标志字节
switch (ord($byte)) {
case 0: // 没有区域信息
$area = "";
break;
case 1:
case 2: // 标志字节为1或2表示区域信息被重定向
fseek($this->fp, $this->getLong3());
$area = $this->getString();
break;
default: // 否则,表示区域信息没有被重定向
$area = $this->getString($byte);
break;
}
return $area;
}
/**
* 根据所给 IP 地址或域名返回所在地区信息
*
* @access public
* @param string $ip
* @return array|null
*/
public function getLocation(string $ip = ''): ?array
{
if (!$this->fp) {
return null;
}
// 如果数据文件没有被正确打开,则直接返回空
if (empty($ip)) {
$ip = request()->ip();
}
$location['ip'] = gethostbyname($ip); // 将输入的域名转化为IP地址
$ip = $this->packIp($location['ip']); // 将输入的IP地址转化为可比较的IP地址
// 不合法的IP地址会被转化为255.255.255.255
// 对分搜索
$l = 0; // 搜索的下边界
$u = $this->totalIp; // 搜索的上边界
$findIp = $this->lastIp; // 如果没有找到就返回最后一条IP记录QQWry.Dat的版本信息
while ($l <= $u) {
// 当上边界小于下边界时,查找失败
$i = floor(($l + $u) / 2); // 计算近似中间记录
fseek($this->fp, $this->firstIp + $i * 7);
$beginIp = strrev(fread($this->fp, 4)); // 获取中间记录的开始IP地址
// strrev函数在这里的作用是将little-endian的压缩IP地址转化为big-endian的格式
// 以便用于比较,后面相同。
if ($ip < $beginIp) { // 用户的IP小于中间记录的开始IP地址时
$u = $i - 1; // 将搜索的上边界修改为中间记录减一
} else {
fseek($this->fp, $this->getLong3());
$endIp = strrev(fread($this->fp, 4)); // 获取中间记录的结束IP地址
if ($ip > $endIp) { // 用户的IP大于中间记录的结束IP地址时
$l = $i + 1; // 将搜索的下边界修改为中间记录加一
} else {
// 用户的IP在中间记录的IP范围内时
$findIp = $this->firstIp + $i * 7;
break; // 则表示找到结果,退出循环
}
}
}
//获取查找到的IP地理位置信息
fseek($this->fp, $findIp);
$location['beginip'] = long2ip($this->getLong()); // 用户IP所在范围的开始地址
$offset = $this->getLong3();
fseek($this->fp, $offset);
$location['endip'] = long2ip($this->getLong()); // 用户IP所在范围的结束地址
$byte = fread($this->fp, 1); // 标志字节
switch (ord($byte)) {
case 1: // 标志字节为1表示国家和区域信息都被同时重定向
$countryOffset = $this->getLong3(); // 重定向地址
fseek($this->fp, $countryOffset);
$byte = fread($this->fp, 1); // 标志字节
switch (ord($byte)) {
case 2: // 标志字节为2表示国家信息又被重定向
fseek($this->fp, $this->getLong3());
$location['country'] = $this->getString();
fseek($this->fp, $countryOffset + 4);
$location['area'] = $this->getArea();
break;
default: // 否则,表示国家信息没有被重定向
$location['country'] = $this->getString($byte);
$location['area'] = $this->getArea();
break;
}
break;
case 2: // 标志字节为2表示国家信息被重定向
fseek($this->fp, $this->getLong3());
$location['country'] = $this->getString();
fseek($this->fp, $offset + 8);
$location['area'] = $this->getArea();
break;
default: // 否则,表示国家信息没有被重定向
$location['country'] = $this->getString($byte);
$location['area'] = $this->getArea();
break;
}
if (trim($location['country']) == 'CZ88.NET') {
// CZ88.NET表示没有有效信息
$location['country'] = '未知';
}
if (trim($location['area']) == 'CZ88.NET') {
$location['area'] = '';
}
// 编码转换
foreach ($location as $key => $item) {
$location[$key] = iconv('GBK', 'UTF-8', $item);
}
return $location;
}
/**
* 析构函数,用于在页面执行结束后自动关闭打开的文件。
*
*/
public function __destruct()
{
if ($this->fp) {
fclose($this->fp);
}
$this->fp = 0;
}
}