152 lines
4.0 KiB
C++
152 lines
4.0 KiB
C++
#pragma once
|
|
#include <winsock2.h>
|
|
#include <ws2tcpip.h>
|
|
#include<iostream>
|
|
#include<Windows.h>
|
|
#include<ipmib.h>
|
|
#include<winerror.h>
|
|
#include<iphlpapi.h>
|
|
#include<string>
|
|
|
|
#pragma comment(lib,"ws2_32.lib")
|
|
#pragma comment(lib, "iphlpapi.lib")
|
|
|
|
class LocalAddr {
|
|
|
|
private:
|
|
void FreeIpForwardTable(PMIB_IPFORWARDTABLE pIpRouteTab)
|
|
{
|
|
if (pIpRouteTab != NULL)
|
|
{
|
|
::GlobalFree(pIpRouteTab);
|
|
pIpRouteTab = NULL;
|
|
}
|
|
}
|
|
|
|
PMIB_IPFORWARDTABLE GetIpForwardTable(BOOL bOrder)
|
|
{
|
|
PMIB_IPFORWARDTABLE pIpRouteTab = NULL;
|
|
DWORD dwActualSize = 0;
|
|
|
|
// 查询所需缓冲区的大小
|
|
if (::GetIpForwardTable(pIpRouteTab, &dwActualSize, bOrder) == ERROR_INSUFFICIENT_BUFFER)
|
|
{
|
|
// 为MIB_IPFORWARDTABLE结构申请内存
|
|
pIpRouteTab = (PMIB_IPFORWARDTABLE)::GlobalAlloc(GPTR, dwActualSize);
|
|
// 获取路由表
|
|
if (::GetIpForwardTable(pIpRouteTab, &dwActualSize, bOrder) == NO_ERROR)
|
|
return pIpRouteTab;
|
|
::GlobalFree(pIpRouteTab);
|
|
}
|
|
return NULL;
|
|
}
|
|
|
|
public:
|
|
std::string GetSystemIpAddress()
|
|
{
|
|
std::string strLoalhostIp;
|
|
PMIB_IPFORWARDTABLE pIpRouteTable = GetIpForwardTable(TRUE);
|
|
if (pIpRouteTable != NULL)
|
|
{
|
|
DWORD dwCurrIndex;
|
|
struct in_addr inadDest;
|
|
struct in_addr inadMask;
|
|
struct in_addr inadGateway;
|
|
char szDestIp[128] = { 0 };
|
|
char szMaskIp[128] = { 0 };
|
|
char szGatewayIp[128] = { 0 };
|
|
DWORD IfIndex = 0;
|
|
DWORD ForwardMetric1 = 0;
|
|
if (pIpRouteTable->dwNumEntries > 0)
|
|
{
|
|
char ip[INET_ADDRSTRLEN] = {0};
|
|
for (DWORD i = 0; i < pIpRouteTable->dwNumEntries; i++){
|
|
dwCurrIndex = pIpRouteTable->table[i].dwForwardIfIndex;
|
|
// 目的地址
|
|
inadDest.s_addr = pIpRouteTable->table[i].dwForwardDest;
|
|
strcpy_s(szDestIp, sizeof(szDestIp), inet_ntop(AF_INET, &inadDest, ip, INET_ADDRSTRLEN));
|
|
// 子网掩码
|
|
inadMask.s_addr = pIpRouteTable->table[i].dwForwardMask;
|
|
strcpy_s(szMaskIp, sizeof(szDestIp), inet_ntop(AF_INET, &inadMask, ip, INET_ADDRSTRLEN));
|
|
// 网关地址
|
|
inadGateway.s_addr = pIpRouteTable->table[i].dwForwardNextHop;
|
|
strcpy_s(szGatewayIp, sizeof(szDestIp), inet_ntop(AF_INET, &inadGateway, ip, INET_ADDRSTRLEN));
|
|
if ((strcmp(szDestIp, "0.0.0.0") == 0) && (strcmp(szMaskIp, "0.0.0.0") == 0))
|
|
{
|
|
|
|
if (i == 0)
|
|
{
|
|
ForwardMetric1 = pIpRouteTable->table[i].dwForwardMetric1;
|
|
IfIndex = pIpRouteTable->table[i].dwForwardIfIndex;
|
|
struct in_addr inadDest;
|
|
inadDest.s_addr = pIpRouteTable->table[i].dwForwardDest;
|
|
}
|
|
else if (ForwardMetric1 > pIpRouteTable->table[i].dwForwardMetric1)
|
|
{
|
|
ForwardMetric1 = pIpRouteTable->table[i].dwForwardMetric1;
|
|
IfIndex = pIpRouteTable->table[i].dwForwardIfIndex;
|
|
struct in_addr inadDest;
|
|
inadDest.s_addr = pIpRouteTable->table[i].dwForwardDest;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
FreeIpForwardTable(pIpRouteTable);
|
|
return "";
|
|
}
|
|
FreeIpForwardTable(pIpRouteTable);
|
|
|
|
|
|
if (IfIndex > 0)
|
|
{
|
|
DWORD ipdwSize = 0;
|
|
PBYTE m_pBuffer = new BYTE[MAX_PATH];
|
|
ULONG m_ulSize = MAX_PATH;
|
|
DWORD m_dwResult;
|
|
PMIB_IPADDRTABLE pAddrTable;
|
|
PMIB_IPADDRROW pAddrRow;
|
|
in_addr ia;
|
|
GetIpAddrTable((PMIB_IPADDRTABLE)m_pBuffer, &m_ulSize, TRUE);
|
|
delete[] m_pBuffer;
|
|
m_pBuffer = new BYTE[m_ulSize];
|
|
if (NULL != m_pBuffer)
|
|
{
|
|
m_dwResult = GetIpAddrTable((PMIB_IPADDRTABLE)m_pBuffer, &m_ulSize, TRUE);
|
|
if (m_dwResult == NO_ERROR)
|
|
{
|
|
pAddrTable = (PMIB_IPADDRTABLE)m_pBuffer;
|
|
|
|
for (DWORD x = 0; x < pAddrTable->dwNumEntries; ++x)
|
|
{
|
|
pAddrRow = (PMIB_IPADDRROW) & (pAddrTable->table[x]);
|
|
|
|
ia.S_un.S_addr = pAddrRow->dwAddr;
|
|
char IPMsg[100] = { 0 };
|
|
if (IfIndex == pAddrRow->dwIndex)
|
|
{
|
|
char ip[INET_ADDRSTRLEN] = {0};
|
|
LPCSTR psz = inet_ntop(AF_INET, &ia, ip, INET_ADDRSTRLEN);
|
|
if (psz)
|
|
{
|
|
strLoalhostIp = psz;
|
|
}
|
|
delete[] m_pBuffer;
|
|
return strLoalhostIp;
|
|
}
|
|
}
|
|
}
|
|
delete[] m_pBuffer;
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
FreeIpForwardTable(pIpRouteTable);
|
|
return "";
|
|
}
|
|
return "";
|
|
}
|
|
|
|
}; |