431 lines
16 KiB
C++
431 lines
16 KiB
C++
#pragma once
|
||
#include <string>
|
||
#include <winsock2.h>
|
||
#include <iphlpapi.h>
|
||
#include <ws2tcpip.h>
|
||
#include <WinInet.h>
|
||
#include "logger.h"
|
||
#include "jsoncpp/json.h"
|
||
|
||
#ifndef _WIN64
|
||
#ifdef _DEBUG
|
||
#pragma comment(lib, "jsoncpp/jsoncppd.lib")
|
||
#else
|
||
#pragma comment(lib, "jsoncpp/jsoncpp.lib")
|
||
#endif
|
||
#else
|
||
#ifdef _DEBUG
|
||
#pragma comment(lib, "jsoncpp/jsoncpp_x64d.lib")
|
||
#else
|
||
#pragma comment(lib, "jsoncpp/jsoncpp_x64.lib")
|
||
#endif
|
||
#endif
|
||
|
||
#pragma comment(lib, "wininet.lib")
|
||
#pragma comment(lib, "Iphlpapi.lib")
|
||
|
||
// UTF-8 转 ANSI（当前系统代码页）
|
||
inline std::string Utf8ToAnsi(const std::string& utf8)
|
||
{
|
||
if (utf8.empty()) return "";
|
||
|
||
// UTF-8 -> UTF-16
|
||
int wideLen = MultiByteToWideChar(CP_UTF8, 0, utf8.c_str(), -1, NULL, 0);
|
||
if (wideLen <= 0) return utf8;
|
||
|
||
std::wstring wideStr(wideLen, 0);
|
||
MultiByteToWideChar(CP_UTF8, 0, utf8.c_str(), -1, &wideStr[0], wideLen);
|
||
|
||
// UTF-16 -> ANSI
|
||
int ansiLen = WideCharToMultiByte(CP_ACP, 0, wideStr.c_str(), -1, NULL, 0, NULL, NULL);
|
||
if (ansiLen <= 0) return utf8;
|
||
|
||
std::string ansiStr(ansiLen, 0);
|
||
WideCharToMultiByte(CP_ACP, 0, wideStr.c_str(), -1, &ansiStr[0], ansiLen, NULL, NULL);
|
||
|
||
// 去掉末尾的 null 字符
|
||
if (!ansiStr.empty() && ansiStr.back() == '\0') {
|
||
ansiStr.pop_back();
|
||
}
|
||
|
||
return ansiStr;
|
||
}
|
||
|
||
// 检测字符串是否可能是 UTF-8 编码（包含多字节序列）
|
||
inline bool IsLikelyUtf8(const std::string& str)
|
||
{
|
||
for (size_t i = 0; i < str.size(); i++) {
|
||
unsigned char c = str[i];
|
||
if (c >= 0x80) {
|
||
// 2字节序列: 110xxxxx 10xxxxxx
|
||
if ((c & 0xE0) == 0xC0 && i + 1 < str.size()) {
|
||
if ((str[i+1] & 0xC0) == 0x80) return true;
|
||
}
|
||
// 3字节序列: 1110xxxx 10xxxxxx 10xxxxxx
|
||
else if ((c & 0xF0) == 0xE0 && i + 2 < str.size()) {
|
||
if ((str[i+1] & 0xC0) == 0x80 && (str[i+2] & 0xC0) == 0x80) return true;
|
||
}
|
||
// 4字节序列: 11110xxx 10xxxxxx 10xxxxxx 10xxxxxx
|
||
else if ((c & 0xF8) == 0xF0 && i + 3 < str.size()) {
|
||
if ((str[i+1] & 0xC0) == 0x80 && (str[i+2] & 0xC0) == 0x80 && (str[i+3] & 0xC0) == 0x80) return true;
|
||
}
|
||
}
|
||
}
|
||
return false;
|
||
}
|
||
|
||
// 安全转换：检测到 UTF-8 才转换，否则原样返回
|
||
inline std::string SafeUtf8ToAnsi(const std::string& str)
|
||
{
|
||
if (str.empty()) return str;
|
||
return IsLikelyUtf8(str) ? Utf8ToAnsi(str) : str;
|
||
}
|
||
|
||
inline void splitIpPort(const std::string& input, std::string& ip, std::string& port)
|
||
{
|
||
size_t pos = input.find(':');
|
||
if (pos != std::string::npos) {
|
||
ip = input.substr(0, pos);
|
||
port = input.substr(pos + 1);
|
||
} else {
|
||
ip = input;
|
||
port = "";
|
||
}
|
||
}
|
||
|
||
/**
|
||
* IPConverter: IP 操作类，用于获取公网IP，获取IP对应的地理位置等.
|
||
* 目前是通过调用公开的互联网API完成，假如该查询网站不可访问则需要重新适配.
|
||
*/
|
||
class IPConverter
|
||
{
|
||
public:
|
||
virtual ~IPConverter() {}
|
||
|
||
virtual std::string IPtoAddress(const std::string& ip)
|
||
{
|
||
return "implement me";
|
||
}
|
||
|
||
/**
|
||
* 判断给定的 IP 地址是否是局域网（内网）IP
|
||
* @param ipAddress IP 地址字符串（如 "192.168.1.1"）
|
||
* @return 如果是局域网 IP，返回 true; 否则返回 false
|
||
*/
|
||
bool IsPrivateIP(const std::string& ipAddress)
|
||
{
|
||
// 将 IP 地址字符串转换为二进制格式
|
||
in_addr addr;
|
||
if (inet_pton(AF_INET, ipAddress.c_str(), &addr) != 1) {
|
||
Mprintf("Invalid IP address: %s\n", ipAddress.c_str());
|
||
return false;
|
||
}
|
||
|
||
// 将二进制 IP 地址转换为无符号整数
|
||
unsigned long ip = ntohl(addr.s_addr);
|
||
|
||
// 检查 IP 地址是否在局域网范围内
|
||
if ((ip >= 0x0A000000 && ip <= 0x0AFFFFFF) || // 10.0.0.0/8
|
||
(ip >= 0xAC100000 && ip <= 0xAC1FFFFF) || // 172.16.0.0/12
|
||
(ip >= 0xC0A80000 && ip <= 0xC0A8FFFF) || // 192.168.0.0/16
|
||
(ip >= 0x7F000000 && ip <= 0x7FFFFFFF)) { // 127.0.0.0/8
|
||
return true;
|
||
}
|
||
|
||
return false;
|
||
}
|
||
|
||
/**
|
||
* 判断给定的 IP 地址是否是公网 IP
|
||
*/
|
||
bool IsPublicIP(const std::string& ipAddress)
|
||
{
|
||
in_addr addr;
|
||
if (inet_pton(AF_INET, ipAddress.c_str(), &addr) != 1) return false;
|
||
unsigned long h = ntohl(addr.s_addr);
|
||
if ((h & 0xFF000000) == 0x0A000000) return false; // 10.0.0.0/8
|
||
if ((h & 0xFFF00000) == 0xAC100000) return false; // 172.16.0.0/12
|
||
if ((h & 0xFFFF0000) == 0xC0A80000) return false; // 192.168.0.0/16
|
||
if ((h & 0xFF000000) == 0x7F000000) return false; // 127.0.0.0/8
|
||
if ((h & 0xFF000000) == 0x00000000) return false; // 0.0.0.0/8
|
||
if ((h & 0xFFFF0000) == 0xA9FE0000) return false; // 169.254.0.0/16
|
||
if ((h & 0xFFC00000) == 0x64400000) return false; // 100.64.0.0/10
|
||
if ((h & 0xF0000000) == 0xE0000000) return false; // 224.0.0.0/4
|
||
if ((h & 0xF0000000) == 0xF0000000) return false; // 240.0.0.0/4
|
||
return true;
|
||
}
|
||
|
||
/**
|
||
* 获取本机网卡的公网 IP 和私有 IP
|
||
*/
|
||
void GetLocalIPs(std::string& publicIP, std::string& privateIP)
|
||
{
|
||
publicIP.clear();
|
||
privateIP.clear();
|
||
ULONG outBufLen = 15000;
|
||
IP_ADAPTER_ADDRESSES* pAddresses = (IP_ADAPTER_ADDRESSES*)malloc(outBufLen);
|
||
if (GetAdaptersAddresses(AF_INET, GAA_FLAG_INCLUDE_PREFIX, nullptr, pAddresses, &outBufLen) == ERROR_BUFFER_OVERFLOW) {
|
||
free(pAddresses);
|
||
pAddresses = (IP_ADAPTER_ADDRESSES*)malloc(outBufLen);
|
||
}
|
||
if (GetAdaptersAddresses(AF_INET, GAA_FLAG_INCLUDE_PREFIX, nullptr, pAddresses, &outBufLen) == NO_ERROR) {
|
||
for (IP_ADAPTER_ADDRESSES* pCurr = pAddresses; pCurr; pCurr = pCurr->Next) {
|
||
if (pCurr->OperStatus != IfOperStatusUp) continue;
|
||
if (pCurr->IfType == IF_TYPE_SOFTWARE_LOOPBACK) continue;
|
||
for (IP_ADAPTER_UNICAST_ADDRESS* pUnicast = pCurr->FirstUnicastAddress; pUnicast; pUnicast = pUnicast->Next) {
|
||
if (pUnicast->Address.lpSockaddr->sa_family != AF_INET) continue;
|
||
char addrBuf[INET_ADDRSTRLEN] = { 0 };
|
||
sockaddr_in* sa_in = (sockaddr_in*)pUnicast->Address.lpSockaddr;
|
||
inet_ntop(AF_INET, &sa_in->sin_addr, addrBuf, sizeof(addrBuf));
|
||
std::string ip = addrBuf;
|
||
if (ip.substr(0, 4) == "127.") continue;
|
||
if (IsPublicIP(ip)) { if (publicIP.empty()) publicIP = ip; }
|
||
else { if (privateIP.empty()) privateIP = ip; }
|
||
if (!publicIP.empty() && !privateIP.empty()) break;
|
||
}
|
||
if (!publicIP.empty() && !privateIP.empty()) break;
|
||
}
|
||
}
|
||
free(pAddresses);
|
||
}
|
||
|
||
// 获取本机地理位置
|
||
std::string GetLocalLocation()
|
||
{
|
||
return GetGeoLocation(getPublicIP());
|
||
}
|
||
|
||
// 获取 IP 地址地理位置 (多API备用)
|
||
virtual std::string GetGeoLocation(const std::string& IP)
|
||
{
|
||
if (IP.empty()) return "";
|
||
|
||
std::string ip = IP;
|
||
if (isLocalIP(ip)) {
|
||
ip = getPublicIP();
|
||
if (ip.empty())
|
||
return "";
|
||
}
|
||
|
||
// 初始化 WinINet
|
||
HINTERNET hInternet = InternetOpen("IP Geolocation", INTERNET_OPEN_TYPE_DIRECT, NULL, NULL, 0);
|
||
if (hInternet == NULL) {
|
||
Mprintf("InternetOpen failed! %d\n", GetLastError());
|
||
return "";
|
||
}
|
||
|
||
// 设置超时
|
||
DWORD timeout = 5000;
|
||
InternetSetOptionA(hInternet, INTERNET_OPTION_CONNECT_TIMEOUT, &timeout, sizeof(timeout));
|
||
InternetSetOptionA(hInternet, INTERNET_OPTION_SEND_TIMEOUT, &timeout, sizeof(timeout));
|
||
InternetSetOptionA(hInternet, INTERNET_OPTION_RECEIVE_TIMEOUT, &timeout, sizeof(timeout));
|
||
|
||
// API配置: {url格式, 城市字段, 国家字段, 成功条件字段, 成功条件值, 是否HTTPS}
|
||
struct GeoApiConfig {
|
||
const char* urlFmt; // URL格式 (%s = IP)
|
||
const char* cityField; // 城市字段名
|
||
const char* countryField;// 国家字段名
|
||
const char* checkField; // 校验字段 (空=不校验)
|
||
const char* checkValue; // 校验值 (空=检查非error)
|
||
bool useHttps;
|
||
};
|
||
|
||
static const GeoApiConfig apis[] = {
|
||
// ip-api.com: 45次/分钟
|
||
{"http://ip-api.com/json/%s?fields=status,country,city", "city", "country", "status", "success", false},
|
||
// ipinfo.io: 1000次/月
|
||
{"http://ipinfo.io/%s/json", "city", "country", "", "", false},
|
||
// ipapi.co: 1000次/天
|
||
{"https://ipapi.co/%s/json/", "city", "country_name", "error", "", true},
|
||
};
|
||
|
||
std::string location;
|
||
for (const auto& api : apis) {
|
||
location = TryGeoApi(hInternet, ip, api.urlFmt, api.cityField, api.countryField,
|
||
api.checkField, api.checkValue, api.useHttps);
|
||
if (!location.empty()) break;
|
||
}
|
||
|
||
InternetCloseHandle(hInternet);
|
||
|
||
if (location.empty() && IsPrivateIP(ip)) {
|
||
return "Local Area Network";
|
||
}
|
||
return location;
|
||
}
|
||
|
||
private:
|
||
// 通用API查询函数
|
||
std::string TryGeoApi(HINTERNET hInternet, const std::string& ip,
|
||
const char* urlFmt, const char* cityField, const char* countryField,
|
||
const char* checkField, const char* checkValue, bool useHttps)
|
||
{
|
||
char urlBuf[256];
|
||
sprintf_s(urlBuf, urlFmt, ip.c_str());
|
||
|
||
DWORD flags = INTERNET_FLAG_RELOAD;
|
||
if (useHttps) flags |= INTERNET_FLAG_SECURE;
|
||
|
||
HINTERNET hConnect = InternetOpenUrlA(hInternet, urlBuf, NULL, 0, flags, 0);
|
||
if (!hConnect) return "";
|
||
|
||
// 检查HTTP状态码
|
||
DWORD statusCode = 0;
|
||
DWORD statusSize = sizeof(statusCode);
|
||
if (HttpQueryInfo(hConnect, HTTP_QUERY_STATUS_CODE | HTTP_QUERY_FLAG_NUMBER, &statusCode, &statusSize, NULL)) {
|
||
if (statusCode == 429 || statusCode >= 400) {
|
||
InternetCloseHandle(hConnect);
|
||
return ""; // 429或其他错误，尝试下一个API
|
||
}
|
||
}
|
||
|
||
std::string readBuffer;
|
||
char buffer[4096];
|
||
DWORD bytesRead;
|
||
while (InternetReadFile(hConnect, buffer, sizeof(buffer), &bytesRead) && bytesRead > 0) {
|
||
readBuffer.append(buffer, bytesRead);
|
||
}
|
||
InternetCloseHandle(hConnect);
|
||
|
||
// 备用: 检查响应体中的错误信息
|
||
if (readBuffer.find("Rate limit") != std::string::npos ||
|
||
readBuffer.find("rate limit") != std::string::npos) {
|
||
return "";
|
||
}
|
||
|
||
Json::Value json;
|
||
Json::Reader reader;
|
||
if (!reader.parse(readBuffer, json)) return "";
|
||
|
||
// 校验条件
|
||
if (checkField && checkField[0]) {
|
||
if (checkValue && checkValue[0]) {
|
||
// 检查字段==值
|
||
if (json[checkField].asString() != checkValue) return "";
|
||
} else {
|
||
// 检查字段不为true (error字段)
|
||
if (json[checkField].asBool()) return "";
|
||
}
|
||
}
|
||
|
||
std::string city = Utf8ToAnsi(json[cityField].asString());
|
||
std::string country = Utf8ToAnsi(json[countryField].asString());
|
||
|
||
if (!city.empty() && !country.empty()) return city + ", " + country;
|
||
if (!city.empty()) return city;
|
||
if (!country.empty()) return country;
|
||
return "";
|
||
}
|
||
|
||
public:
|
||
|
||
bool isLoopbackAddress(const std::string& ip)
|
||
{
|
||
return (ip == "127.0.0.1" || ip == "::1");
|
||
}
|
||
|
||
bool isLocalIP(const std::string& ip)
|
||
{
|
||
if (isLoopbackAddress(ip)) return true; // 先检查回环地址
|
||
|
||
ULONG outBufLen = 15000;
|
||
IP_ADAPTER_ADDRESSES* pAddresses = (IP_ADAPTER_ADDRESSES*)malloc(outBufLen);
|
||
if (GetAdaptersAddresses(AF_UNSPEC, GAA_FLAG_INCLUDE_PREFIX, nullptr, pAddresses, &outBufLen) == ERROR_BUFFER_OVERFLOW) {
|
||
free(pAddresses);
|
||
pAddresses = (IP_ADAPTER_ADDRESSES*)malloc(outBufLen);
|
||
}
|
||
|
||
if (GetAdaptersAddresses(AF_UNSPEC, GAA_FLAG_INCLUDE_PREFIX, nullptr, pAddresses, &outBufLen) == NO_ERROR) {
|
||
for (IP_ADAPTER_ADDRESSES* pCurrAddresses = pAddresses; pCurrAddresses; pCurrAddresses = pCurrAddresses->Next) {
|
||
for (IP_ADAPTER_UNICAST_ADDRESS* pUnicast = pCurrAddresses->FirstUnicastAddress; pUnicast; pUnicast = pUnicast->Next) {
|
||
char addressBuffer[INET6_ADDRSTRLEN] = { 0 };
|
||
getnameinfo(pUnicast->Address.lpSockaddr, pUnicast->Address.iSockaddrLength, addressBuffer, sizeof(addressBuffer), nullptr, 0, NI_NUMERICHOST);
|
||
|
||
if (ip == addressBuffer) {
|
||
free(pAddresses);
|
||
return true;
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
free(pAddresses);
|
||
return false;
|
||
}
|
||
|
||
// 获取公网IP, 获取失败返回空
|
||
std::string getPublicIP()
|
||
{
|
||
clock_t t = clock();
|
||
|
||
// 多个候选查询源
|
||
static const std::vector<std::string> urls = {
|
||
"https://checkip.amazonaws.com", // 全球最稳
|
||
"https://api.ipify.org", // 主流高可用
|
||
"https://ipinfo.io/ip", // 备用方案
|
||
"https://icanhazip.com", // 轻量快速
|
||
"https://ifconfig.me/ip" // 末位兜底
|
||
};
|
||
|
||
// 打开 WinINet 会话
|
||
HINTERNET hInternet = InternetOpenA("Mozilla/5.0", INTERNET_OPEN_TYPE_DIRECT, NULL, NULL, 0);
|
||
if (!hInternet) {
|
||
Mprintf("InternetOpen failed. cost %d ms.\n", clock() - t);
|
||
return "";
|
||
}
|
||
|
||
// 设置超时 (毫秒)
|
||
DWORD timeout = 3000; // 3 秒
|
||
InternetSetOptionA(hInternet, INTERNET_OPTION_CONNECT_TIMEOUT, &timeout, sizeof(timeout));
|
||
InternetSetOptionA(hInternet, INTERNET_OPTION_SEND_TIMEOUT, &timeout, sizeof(timeout));
|
||
InternetSetOptionA(hInternet, INTERNET_OPTION_RECEIVE_TIMEOUT, &timeout, sizeof(timeout));
|
||
|
||
std::string result;
|
||
char buffer[2048];
|
||
DWORD bytesRead = 0;
|
||
|
||
// 轮询不同 IP 查询源
|
||
for (const auto& url : urls) {
|
||
HINTERNET hConnect = InternetOpenUrlA(
|
||
hInternet, url.c_str(), NULL, 0,
|
||
INTERNET_FLAG_RELOAD | INTERNET_FLAG_SECURE | INTERNET_FLAG_NO_CACHE_WRITE,
|
||
0
|
||
);
|
||
|
||
if (!hConnect) {
|
||
continue; // 当前源失败，尝试下一个
|
||
}
|
||
|
||
memset(buffer, 0, sizeof(buffer));
|
||
if (InternetReadFile(hConnect, buffer, sizeof(buffer) - 1, &bytesRead) && bytesRead > 0) {
|
||
result.assign(buffer, bytesRead);
|
||
|
||
// 去除换行符和空格
|
||
while (!result.empty() && (result.back() == '\n' || result.back() == '\r' || result.back() == ' '))
|
||
result.pop_back();
|
||
|
||
InternetCloseHandle(hConnect);
|
||
break; // 成功获取，停止尝试
|
||
}
|
||
|
||
InternetCloseHandle(hConnect);
|
||
}
|
||
|
||
InternetCloseHandle(hInternet);
|
||
|
||
Mprintf("getPublicIP %s cost %d ms.\n", result.empty() ? "failed" : "succeed", clock() - t);
|
||
|
||
return result;
|
||
}
|
||
};
|
||
|
||
enum IPDatabaseType {
|
||
NoneDB = 0,
|
||
QQWry = 1,
|
||
Ip2Region = 2,
|
||
};
|
||
|
||
IPConverter* LoadFileQQWry(const char* datPath);
|
||
|
||
IPConverter* LoadFileIp2Region(const char* xdbPath);
|