C++ to obtain IP subnet mask gateway DNS and other local network parameters

  • 2020-04-02 02:29:59
  • OfStack

This paper introduces the method of C++ to obtain IP, subnet mask, gateway, DNS and other local network parameters in the form of a complete example, for your reference, the specific complete example is as follows:


#pragma comment(lib,"Ws2_32.lib")
#include <Iphlpapi.h>
#pragma comment(lib, "Iphlpapi.lib")
using namespace std;
typedef struct tagNetworkCfg 
{
 char szIP[18];
 char szNetmask[18];
 char szGateway[18];
 char szDns1[18];
 char szDns2[18];
}NetworkCfg;
bool GetNetworkCfg(NetworkCfg *cfg)
{
 log_printf("Get network config");
 //Gets network card name network card name, network card alias
 string strAdapterName,strAdapterAlias;
 HKEY hKey, hSubKey, hNdiIntKey;
 if(RegOpenKeyEx(HKEY_LOCAL_MACHINE,
 "System\CurrentControlSet\Control\Class\{4d36e972-e325-11ce-bfc1-08002be10318}",
 0,
 KEY_READ,
 &hKey) != ERROR_SUCCESS)
 return FALSE;

 DWORD dwIndex = 0;
 DWORD dwBufSize = 256;
 DWORD dwDataType;
 char szSubKey[256];
 unsigned char szData[256];

 while(RegEnumKeyEx(hKey, dwIndex++, szSubKey, &dwBufSize, NULL, NULL, NULL, NULL) == ERROR_SUCCESS)
 {
 if(RegOpenKeyEx(hKey, szSubKey, 0, KEY_READ, &hSubKey) == ERROR_SUCCESS)
 {    
  if(RegOpenKeyEx(hSubKey, "Ndi\Interfaces", 0, KEY_READ, &hNdiIntKey) == ERROR_SUCCESS)
  {
  dwBufSize = 256;
  if(RegQueryValueEx(hNdiIntKey, "LowerRange", 0, &dwDataType, szData, &dwBufSize) == ERROR_SUCCESS)
  {
   if(strstr((char*)szData, "ethernet") != NULL)//Determine if it's an Ethernet card
   {
   dwBufSize = 256;
   if(RegQueryValueEx(hSubKey, "DriverDesc", 0, &dwDataType, szData, &dwBufSize) == ERROR_SUCCESS)
   {
    strAdapterName = (LPCTSTR)szData;
    dwBufSize = 256;
    if(RegQueryValueEx(hSubKey, "NetCfgInstanceID", 0, &dwDataType, szData, &dwBufSize) == ERROR_SUCCESS)
    {
    strAdapterAlias = (LPCTSTR)szData;
    break;
    }
   }
   }
  }
  RegCloseKey(hNdiIntKey);
  }
  RegCloseKey(hSubKey);
 }

 dwBufSize = 256;
 }  

 RegCloseKey(hKey);
 if (strAdapterName.empty() || strAdapterAlias.empty())
 {
 log_printf("failed to get network config");
 return false;
 }
 string strKeyName = "SYSTEM\CurrentControlSet\Services\Tcpip\Parameters\Interfaces\";
 strKeyName += strAdapterAlias;
 if(RegOpenKeyEx(HKEY_LOCAL_MACHINE,
 strKeyName.c_str(),
 0,
 KEY_READ,
 &hKey) != ERROR_SUCCESS)
 return FALSE;
 dwBufSize = 256;
 if(RegQueryValueEx(hKey, "DhcpIPAddress", 0,&dwDataType, szData, &dwBufSize) == ERROR_SUCCESS)
 strcpy(cfg->szIP,(LPCTSTR)szData);
 else{
 if(RegQueryValueEx(hKey, "IPAddress", 0, &dwDataType, szData, &dwBufSize) == ERROR_SUCCESS)
 strcpy(cfg->szIP,(LPCTSTR)szData);
 }
  
 dwBufSize = 256;
 if(RegQueryValueEx(hKey, "DhcpSubnetMask", 0, &dwDataType, szData, &dwBufSize) == ERROR_SUCCESS)
 strcpy(cfg->szNetmask,(LPCTSTR)szData);
 else
 {
 if(RegQueryValueEx(hKey, "SubnetMask", 0, &dwDataType, szData, &dwBufSize) == ERROR_SUCCESS)
 trcpy(cfg->szNetmask,(LPCTSTR)szData);
 }
  
 dwBufSize = 256;
 if(RegQueryValueEx(hKey, "DhcpDefaultGateway", 0, &dwDataType, szData, &dwBufSize) == ERROR_SUCCESS)
 strcpy(cfg->szGateway,(LPCTSTR)szData);
 else
 {
 if(RegQueryValueEx(hKey, "DefaultGateway", 0, &dwDataType, szData, &dwBufSize) == ERROR_SUCCESS)
 strcpy(cfg->szGateway,(LPCSTR)szData);
 }
 RegCloseKey(hKey);

 //Get DNS server information
 FIXED_INFO *fi = (FIXED_INFO *)GlobalAlloc(GPTR,sizeof( FIXED_INFO));
 ULONG ulOutBufLen = sizeof(FIXED_INFO);
 DWORD ret = ::GetNetworkParams(fi, &ulOutBufLen);
 if(ret != ERROR_SUCCESS) 
 { 
 GlobalFree(fi);
 fi = (FIXED_INFO *) GlobalAlloc( GPTR, ulOutBufLen );
 ret = ::GetNetworkParams(fi, &ulOutBufLen);
 if(ret != ERROR_SUCCESS) 
 {
  log_printf("Get Dns server failed"); 
  return false; 
 }
 }
 strcpy(cfg->szDns1,fi->DnsServerList.IpAddress.String);
 IP_ADDR_STRING *pIPAddr = fi->DnsServerList.Next;
 if(pIPAddr != NULL) 
 strcpy(cfg->szDns2, pIPAddr->IpAddress.String);
 return false;
}

Related articles: