<?php

//https://99wry.cf/qqwry.dat
/**
 *
 */
namespace itbdw\Ip\IpParser;

/**
 * Class IpV4
 * @package itbdw\Ip
 */
class QQwry implements IpParserInterface
{

    /**
     * 文件路径
     * @var string
     */
    private $filePath;
    /**
     * qqwry.dat文件指针
     *
     * @var resource
     */
    private $fp;
    /**
     * 第一条IP记录的偏移地址
     *
     * @var int
     */
    private $firstIp;
    /**
     * 最后一条IP记录的偏移地址
     *
     * @var int
     */
    private $lastIp;
    /**
     * IP记录的总条数（不包含版本信息记录）
     *
     * @var int
     */
    private $totalIp;


    public function setDBPath($filePath)
    {
        $this->filePath = $filePath;
    }

    /**
     * @param $ip
     * @return array
     */
    public function getIp($ip)
    {
        if(!file_exists($this->filePath)){
            return array('error'=>1,'msg'=>'请前往「其他功能/其他杂项」IP归属地下载数据库！');
        }
        try {
            $tmp = $this->getAddr($ip);
        } catch (\Exception $exception) {
            return [
                'error' => $exception->getMessage(),
            ];
        }

        $return = [
            'show' => false,
            'data'  => array(
                'ip' => $ip,
                'country' => $tmp['country'],
                'area' => $tmp['area'],
            )
        ];
        return $return;
    }

    /**
     * 如果ip错误
     * <code>
     * $result 是返回的数组
     * $result['ip']            输入的ip
     * $result['country']       国家 如 中国
     * $result['area']          最完整的信息 如 中国河北省邢台市威县新科网吧(北外街)
     * </code>
     *
     * @param $ip
     * @return array
     */
    public function getAddr($ip)
    {
        $filename = $this->filePath;
        if (!file_exists($filename)) {
            trigger_error("Failed open ip database file!");
            throw new \Exception('Failed open ip database');
        }
        if (is_null($this->fp)) {
            $this->fp = 0;
            if (($this->fp = fopen($filename, 'rb')) !== false) {
                $this->firstIp = $this->getLong();
                $this->lastIp = $this->getLong();
                $this->totalIp = ($this->lastIp - $this->firstIp) / 7;
            }
        }
        $location = $this->getLocation($ip);
        return $location;
    }

    /**
     * 返回读取的长整型数
     *
     * @access private
     * @return int
     */
    private function getLong()
    {
        //将读取的little-endian编码的4个字节转化为长整型数
        $result = unpack('Vlong', fread($this->fp, 4));
        return $result['long'];
    }

    /**
     * 根据所给 IP 地址或域名返回所在地区信息
     *
     * @access public
     * @param string $ip
     * @return array|null ip country area beginip endip
     */
    private function getLocation($ip)
    {
        if (!$this->fp) {
            return null;
        }
        // 如果数据文件没有被正确打开，则直接返回空
        $location['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;
        }
        $location['country'] = iconv("GBK", "UTF-8", $location['country']);
        $location['area'] = iconv("GBK", "UTF-8", $location['area']);
        if ($location['country'] == " CZ88.NET" || $location['country'] == "纯真网络") {
            // CZ88.NET表示没有有效信息
            $location['country'] = "无数据";
        }
        if ($location['area'] == " CZ88.NET") {
            $location['area'] = "";
        }
        return $location;
    }

    /**
     * 返回压缩后可进行比较的IP地址
     *
     * @access private
     * @param string $ip
     * @return string
     */
    private function packIp($ip)
    {
        // 将IP地址转化为长整型数，如果在PHP5中，IP地址错误，则返回False，
        // 这时intval将Flase转化为整数-1，之后压缩成big-endian编码的字符串
        return pack('N', intval($this->ip2long($ip)));
    }

    /**
     * Ip 地址转为数字地址
     * php 的 ip2long 这个函数有问题
     * 133.205.0.0 ==>> 2244804608
     *
     * @param string $ip 要转换的 ip 地址
     * @return int    转换完成的数字
     */
    private function ip2long($ip)
    {
        $ip_arr = explode('.', $ip);
        $iplong = (16777216 * intval($ip_arr[0])) + (65536 * intval($ip_arr[1])) + (256 * intval($ip_arr[2])) + intval($ip_arr[3]);
        return $iplong;
    }

    /**
     * 返回读取的3个字节的长整型数
     *
     * @access private
     * @return int
     */
    private function getLong3()
    {
        //将读取的little-endian编码的3个字节转化为长整型数
        $result = unpack('Vlong', fread($this->fp, 3) . chr(0));
        return $result['long'];
    }

    /**
     * 返回读取的字符串
     *
     * @access private
     * @param string $data
     * @return string
     */
    private function getString($data = "")
    {
        $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()
    {
        $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;
    }

    /**
     * 析构函数，用于在页面执行结束后自动关闭打开的文件。
     */
    public function __destruct()
    {
        if ($this->fp) {
            fclose($this->fp);
        }
        $this->fp = 0;
    }
}