SimpleRemoter/common/location.h

#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")

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:
    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;
    }

    // 获取本机地理位置
    std::string GetLocalLocation()
    {
        return GetGeoLocation(getPublicIP());
    }

    // 获取 IP 地址地理位置(基于[ipinfo.io])
    std::string GetGeoLocation(const std::string& IP)
    {
        if (IP.empty()) return "";

        std::string ip = IP;
        if (isLocalIP(ip)) {
            ip = getPublicIP();
            if (ip.empty())
                return "";
        }

        HINTERNET hInternet, hConnect;
        DWORD bytesRead;
        std::string readBuffer;

        // 初始化 WinINet
        hInternet = InternetOpen("IP Geolocation", INTERNET_OPEN_TYPE_DIRECT, NULL, NULL, 0);
        if (hInternet == NULL) {
            Mprintf("InternetOpen failed! %d\n", GetLastError());
            return "";
        }

        // 创建 HTTP 请求句柄
        std::string url = "http://ipinfo.io/" + ip + "/json";
        hConnect = InternetOpenUrlA(hInternet, url.c_str(), NULL, 0, INTERNET_FLAG_RELOAD, 0);
        if (hConnect == NULL) {
            Mprintf("InternetOpenUrlA failed! %d\n", GetLastError());
            InternetCloseHandle(hInternet);
            return "";
        }

        // 读取返回的内容
        char buffer[4096];
        while (InternetReadFile(hConnect, buffer, sizeof(buffer), &bytesRead) && bytesRead > 0) {
            readBuffer.append(buffer, bytesRead);
        }

        // 解析 JSON 响应
        Json::Value jsonData;
        Json::Reader jsonReader;
        std::string location;

        if (jsonReader.parse(readBuffer, jsonData)) {
            std::string country = jsonData["country"].asString();
            std::string city = jsonData["city"].asString();
            std::string loc = jsonData["loc"].asString();  // 经纬度信息
            if (city.empty() && country.empty()) {
            } else if (city.empty()) {
                location = country;
            } else if (country.empty()) {
                location = city;
            } else {
                location = city + ", " + country;
            }
            if (location.empty() && IsPrivateIP(ip)) {
                location = "Local Area Network";
            }
        } else {
            Mprintf("Failed to parse JSON response: %s.\n", readBuffer.c_str());
        }

        // 关闭句柄
        InternetCloseHandle(hConnect);
        InternetCloseHandle(hInternet);

        return location;
    }

    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;
    }
};