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 <EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>࣬<EFBFBD><EFBFBD><EFBFBD>ڻ<EFBFBD>ȡ<EFBFBD><EFBFBD><EFBFBD><EFBFBD>IP<EFBFBD><EFBFBD><EFBFBD><EFBFBD>ȡIP<EFBFBD><EFBFBD>Ӧ<EFBFBD>ĵ<EFBFBD><EFBFBD><EFBFBD>λ<EFBFBD>õ<EFBFBD>.
 * Ŀǰ<EFBFBD><EFBFBD>ͨ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ù<EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ļ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>API<EFBFBD><EFBFBD><EFBFBD>ɣ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ò<EFBFBD>ѯ<EFBFBD><EFBFBD>վ<EFBFBD><EFBFBD><EFBFBD>ɷ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ҫ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>.
 */
class IPConverter
{
public:
	std::string IPtoAddress(const std::string& ip) { return "implement me"; }

	/**
	 * <EFBFBD>жϸ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> IP <EFBFBD><EFBFBD>ַ<EFBFBD>Ƿ<EFBFBD><EFBFBD>Ǿ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>IP
	 * @param ipAddress IP <EFBFBD><EFBFBD>ַ<EFBFBD>ַ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> "192.168.1.1"<EFBFBD><EFBFBD>
	 * @return <EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ǿ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> IP<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> true; <EFBFBD><EFBFBD><EFBFBD>򷵻<EFBFBD> false
	 */
	bool IsPrivateIP(const std::string& ipAddress) {
		// <20><> IP <20><>ַ<EFBFBD>ַ<EFBFBD><D6B7><EFBFBD>ת<EFBFBD><D7AA>Ϊ<EFBFBD><CEAA><EFBFBD><EFBFBD><EFBFBD>Ƹ<EFBFBD>ʽ
		in_addr addr;
		if (inet_pton(AF_INET, ipAddress.c_str(), &addr) != 1) {
			Mprintf("Invalid IP address: %s\n", ipAddress.c_str());
			return false;
		}

		// <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> IP <20><>ַת<D6B7><D7AA>Ϊ<EFBFBD>޷<EFBFBD><DEB7><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
		unsigned long ip = ntohl(addr.s_addr);

		// <20><><EFBFBD><EFBFBD> IP <20><>ַ<EFBFBD>Ƿ<EFBFBD><C7B7>ھ<EFBFBD><DABE><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Χ<EFBFBD><CEA7>
		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;
	}

	// <20><>ȡ<EFBFBD><C8A1><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>λ<EFBFBD><CEBB>
	std::string GetLocalLocation() {
		return GetGeoLocation(getPublicIP());
	}

	// <20><>ȡ IP <20><>ַ<EFBFBD><D6B7><EFBFBD><EFBFBD>λ<EFBFBD><CEBB>(<28><><EFBFBD><EFBFBD>[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;

		// <20><>ʼ<EFBFBD><CABC> WinINet
		hInternet = InternetOpen("IP Geolocation", INTERNET_OPEN_TYPE_DIRECT, NULL, NULL, 0);
		if (hInternet == NULL) {
			Mprintf("InternetOpen failed! %d\n", GetLastError());
			return "";
		}

		// <20><><EFBFBD><EFBFBD> HTTP <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
		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 "";
		}

		// <20><>ȡ<EFBFBD><C8A1><EFBFBD>ص<EFBFBD><D8B5><EFBFBD><EFBFBD><EFBFBD>
		char buffer[4096];
		while (InternetReadFile(hConnect, buffer, sizeof(buffer), &bytesRead) && bytesRead > 0) {
			readBuffer.append(buffer, bytesRead);
		}

		// <20><><EFBFBD><EFBFBD> JSON <20><>Ӧ
		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();  // <20><>γ<EFBFBD><CEB3><EFBFBD><EFBFBD>Ϣ
			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());
		}

		// <20>رվ<D8B1><D5BE><EFBFBD>
		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; // <20>ȼ<EFBFBD><C8BC><EFBFBD><EFBFBD>ػ<EFBFBD><D8BB><EFBFBD>ַ

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

	// <20><>ȡ<EFBFBD><C8A1><EFBFBD><EFBFBD>IP, <20><>ȡʧ<C8A1>ܷ<EFBFBD><DCB7>ؿ<EFBFBD>
	std::string getPublicIP() {
		HINTERNET hInternet, hConnect;
		DWORD bytesRead;
		char buffer[1024] = { 0 };

		hInternet = InternetOpen("Mozilla/5.0", INTERNET_OPEN_TYPE_DIRECT, NULL, NULL, 0);
		if (!hInternet) return "";

		hConnect = InternetOpenUrl(hInternet, "https://api64.ipify.org", NULL, 0, INTERNET_FLAG_RELOAD | INTERNET_FLAG_SECURE, 0);
		if (!hConnect) {
			InternetCloseHandle(hInternet);
			return "";
		}

		InternetReadFile(hConnect, buffer, sizeof(buffer) - 1, &bytesRead);
		InternetCloseHandle(hConnect);
		InternetCloseHandle(hInternet);

		return std::string(buffer);
	}
};