Changeset 21411 for trunk/src/iphlpapi/iphlpapi.cpp

Timestamp:

Aug 12, 2010, 2:04:53 PM (15 years ago)

Author:

dmik

Message:

Expose the iphlpapi.dll interface at source level through iprtrmib.h/iphlpapi.h.

File:

• trunk/src/iphlpapi/iphlpapi.cpp (modified) (26 diffs)

trunk/src/iphlpapi/iphlpapi.cpp

@@ -20 +20 @@
 
 #include <string.h>
-#include "iphlpapi.h"
+#include <iprtrmib.h>
 #include <winnls.h>
 
@@ -41 +41 @@
 #include "iphlwrap.h"
 
+/* from ipexport.h */
+typedef ULONG  IPAddr;
+typedef ULONG  IPMask;
+typedef ULONG  IP_STATUS;
+
 #pragma pack(1)
 typedef struct
@@ -94 +99 @@
 void stringIPAddress(char* dst,u_long data)
 {
-    sprintf(dst, "%u.%u.%u.%u", 
+    sprintf(dst, "%u.%u.%u.%u",
                         (char)data,
                         (char)(*(((char*)&data) + 1)),
@@ -128 +133 @@
 #endif
 
-  // Init Subsystem and open a socket for ioctl() calls 
+  // Init Subsystem and open a socket for ioctl() calls
   sock_init();
 
@@ -134 +139 @@
   dprintf(("IPHLPAPI(Init): Opened socket %d\n", clientSocket));
 
-  // Whole buf minimum size is 65536 and memsets are really needed in other case 
+  // Whole buf minimum size is 65536 and memsets are really needed in other case
   // we will get garbage in adapter names.
 
@@ -146 +151 @@
   {
     free(buffer);
-    soclose(clientSocket);         
-    return; 
-  }        
+    soclose(clientSocket);
+    return;
+  }
 
   rc = ioctl(clientSocket, SIOSTATAT, (char*)buffer, 65536);
@@ -155 +160 @@
   {
     free(buffer);
-    soclose(clientSocket);         
-    return; 
-  }        
+    soclose(clientSocket);
+    return;
+  }
   cInterfaces = *(short int*)buffer;
   dprintf(("IPHLPAPI Call returned %d interfaces\n", cInterfaces));
@@ -175 +180 @@
 //    if (ifInfo->interfaceNum ==9) continue;
 
-    // Allocate and clear mem 
+    // Allocate and clear mem
     pipAdapter = (PIP_ADAPTER_INFO)malloc (sizeof (IP_ADAPTER_INFO));
     memset(pipAdapter, 0, sizeof(IP_ADAPTER_INFO));
-    if (oldAdapter) 
+    if (oldAdapter)
        oldAdapter->Next = pipAdapter;
 
@@ -184 +189 @@
     pipAdapter->ComboIndex = 1;
     i =  ifInfo->interfaceNum;
-    // Gather some other stats 
+    // Gather some other stats
     dprintf(("IPHLPAPI: interface number: %u\n", ifInfo->interfaceNum));
 
@@ -195 +200 @@
 
     if (strstr(ifmibget.iftable[i].ifDescr,"back")) // lo
-        strcpy(iShortName,"lo");       
+        strcpy(iShortName,"lo");
 
     if (strstr(ifmibget.iftable[i].ifDescr,"ace ppp")) // pppX
-        strcpy(iShortName,strstr(ifmibget.iftable[i].ifDescr,"ppp"));       
+        strcpy(iShortName,strstr(ifmibget.iftable[i].ifDescr,"ppp"));
 
     if (strstr(ifmibget.iftable[i].ifDescr,"ace sl")) // slX
-        strcpy(iShortName,strstr(ifmibget.iftable[i].ifDescr,"sl"));       
+        strcpy(iShortName,strstr(ifmibget.iftable[i].ifDescr,"sl"));
 
     if (strstr(ifmibget.iftable[i].ifDescr,"ace dod")) // dodX
-        strcpy(iShortName,strstr(ifmibget.iftable[i].ifDescr,"dod"));       
+        strcpy(iShortName,strstr(ifmibget.iftable[i].ifDescr,"dod"));
 
     dprintf(("IPHLPAPI: interface name[%s] : %s\n",iShortName, ifmibget.iftable[i].ifDescr));
     strcpy(pipAdapter->AdapterName, ifmibget.iftable[i].ifDescr);
     strcpy(pipAdapter->Description, ifmibget.iftable[i].ifDescr);
-         
-    pipAdapter->AddressLength = 6; // MAX address 
+
+    pipAdapter->AddressLength = 6; // MAX address
     memcpy(pipAdapter->Address,ifmibget.iftable[i].ifPhysAddr,6);
     pipAdapter->Index = ifmibget.iftable[i].ifIndex;
@@ -217 +222 @@
 
     MultiByteToWideChar(CP_ACP, 0, iShortName, strlen(iShortName),
-                        pmibTable->table[currentInterface].wszName, 
+                        pmibTable->table[currentInterface].wszName,
                         MAX_INTERFACE_NAME_LEN);
-  
+
     pmibTable->table[currentInterface].dwIndex = ifmibget.iftable[i].ifIndex;
     pmibTable->table[currentInterface].dwType  = ifmibget.iftable[i].ifType;         /* type of the interface   */
@@ -265 +270 @@
     stringNetmask((char*)&iasAdapterIP.IpMask,ifInfo->netmask);
     iasAdapterIP.Context = 0;
- 
-    // Now we are going to catch gateways for this interface  
-    buffer2 = (char*) malloc(64*1024);   
+
+    // Now we are going to catch gateways for this interface
+    buffer2 = (char*) malloc(64*1024);
     memset(buffer2, 0, 65536);
-    
+
     rc = ioctl(clientSocket, SIOSTATRT, (char*)buffer2, 65536);
     dprintf(("IPHLPAPI(SIOSTATRT):ioctl returned: %d\n", rc));
 
-    if (rc == -1) 
+    if (rc == -1)
     {
       free(buffer);
       free(buffer2);
-      soclose(clientSocket);       
+      soclose(clientSocket);
       // better return nothing than some trash, unwinding and freeing
       for (topAdapter; pipAdapter; topAdapter = pipAdapter) {
           pipAdapter = topAdapter -> Next;
-          free(topAdapter); 
+          free(topAdapter);
       }
-      return; 
-     }        
-         
+      return;
+     }
+
      rtentries *routeEntries = (rtentries*)buffer2;
      p = (unsigned char *)&routeEntries->rttable[0];
 
-     IP_ADDR_STRING iasGateway;    
+     IP_ADDR_STRING iasGateway;
      memset(&iasGateway,0,sizeof(iasGateway));
 
      for (int currentRoute = 0; currentRoute < (routeEntries->hostcount+routeEntries->netcount); currentRoute++)
      {
-     // First check if this route is for our interface    
+     // First check if this route is for our interface
 #ifndef TCPV40HDRS
          r = (struct ortentry *) (p);
@@ -304 +309 @@
             sin = (struct sockaddr_in *)(&r->rt_dst);
             if (strcmp(inet_ntoa(sin->sin_addr),"0.0.0.0")==0)
-            {  
+            {
                iasGateway.Next = NULL;
-               // TODO : Some systems may have many gateways    
+               // TODO : Some systems may have many gateways
                sin = (struct sockaddr_in *)(&r->rt_gateway);
                strcpy(iasGateway.IpAddress.String,inet_ntoa(sin->sin_addr));
@@ -321 +326 @@
        p+=strlen((char *)p)+1;
     }
-  
+
   memcpy((char*)&pipAdapter->IpAddressList, (char*)&iasAdapterIP, sizeof(iasAdapterIP));
   pipAdapter->CurrentIpAddress = &pipAdapter->IpAddressList;
@@ -340 +345 @@
  if (buffer) free(buffer);
  if (buffer2) free(buffer2);
- soclose(clientSocket); 
+ soclose(clientSocket);
 }
 
@@ -362 +367 @@
 {
   PIP_ADDR_STRING pias;
-  
+
   // check for sufficient space
   DWORD dwRequired = sizeof( IP_ADAPTER_INFO );
-    
+
   // follow the IP_ADDR_STRING lists
   pias = &piai->IpAddressList;
@@ -387 +392 @@
     pias = pias->Next;
   }
-    
+
   pias = &piai->PrimaryWinsServer;
   while( pias )
@@ -401 +406 @@
     pias = pias->Next;
   }
-  
+
   return dwRequired;
 }
@@ -417 +422 @@
 {
   dprintf(("GetAdaptersInfo API called"));
-  dprintf(("Address passed is %p",pAdapterInfo)); 
+  dprintf(("Address passed is %p",pAdapterInfo));
   if (NULL == pOutBufLen)
     return ERROR_INVALID_PARAMETER;
-  
+
   // verify first block of memory to write to
   if (IsBadWritePtr(pAdapterInfo, 4))
     return ERROR_INVALID_PARAMETER;
-  
+
   if (NULL == pipAdapter)
   {
@@ -430 +435 @@
     i_initializeAdapterInformation();
   }
-  
+
   if (NULL == pipAdapter)
     return ERROR_NO_DATA;
-  
+
   // OK, just copy over the information as far as possible
   LONG  lSpaceLeft     = *pOutBufLen;
   PBYTE pTarget        = (PBYTE)pAdapterInfo;
   PIP_ADAPTER_INFO pip;
-  
+
   // calculate overall required buffer size
   DWORD dwRequiredBuffer = 0;
-  
+
   for( pip = pipAdapter ; pip ; pip = pip->Next )
   {
@@ -447 +452 @@
     dwRequiredBuffer += i_sizeOfIP_ADAPTER_INFO(pip);
   }
-  
+
   for( pip = pipAdapter ; pip ; pip = pip->Next )
   {
@@ -455 +460 @@
     lSpaceLeft -= dwRequired;
     if (lSpaceLeft >= 0)
-    {     
+    {
       // @PF revised - this thing works because we currently do not support
       // multi-ip, multi-gateway or multi-DHCP servers lists
       // TODO - add lists support
       memcpy(pTarget, pip, sizeof( IP_ADAPTER_INFO ));
-      // point to currentIPAddress 
+      // point to currentIPAddress
       ((PIP_ADAPTER_INFO)(pTarget))->CurrentIpAddress = &((PIP_ADAPTER_INFO)(pTarget))->IpAddressList;
       pTarget += sizeof( IP_ADAPTER_INFO );
-       
+
 //      i_copyIP_ADDRESS_STRING(&pTarget, &pip->IpAddressList);
 //      i_copyIP_ADDRESS_STRING(&pTarget, &pip->GatewayList);
@@ -488 +493 @@
 {
   struct sockaddr_in * sin;
-  PFIXED_INFO fi = pFixedInfo; 
+  PFIXED_INFO fi = pFixedInfo;
   DWORD memNeeded;
   PIP_ADDR_STRING dnslist = NULL, pdnslist = NULL;
@@ -495 +500 @@
   res_init();
 
-  // Check how much mem we will need 
+  // Check how much mem we will need
   memNeeded = sizeof(FIXED_INFO)+_res.nscount*sizeof(IP_ADDR_STRING);
 
@@ -504 +509 @@
    return ERROR_BUFFER_OVERFLOW;
   }
-  
+
   // This is dynamically updated info
   memset(pFixedInfo,0,memNeeded);
-  
+
   ODIN_gethostname(fi->HostName,128);
   strcpy(fi->DomainName,_res.defdname);
 
-  // Fill in DNS Servers 
-  fi->CurrentDnsServer = &fi->DnsServerList;   
+  // Fill in DNS Servers
+  fi->CurrentDnsServer = &fi->DnsServerList;
   dnslist = &fi->DnsServerList;
-   
+
   for (int i = 0; i<_res.nscount; i++)
   {
       if (pdnslist) pdnslist->Next = dnslist;
-      sin = (struct sockaddr_in *)&_res.nsaddr_list[i];               
+      sin = (struct sockaddr_in *)&_res.nsaddr_list[i];
       strcpy(dnslist->IpAddress.String,inet_ntoa(sin->sin_addr));
       dprintf(("IPHLPAPI: GetNetworkParams Adding DNS Server %s",inet_ntoa(sin->sin_addr)));
@@ -524 +529 @@
       if ( pdnslist == &fi->DnsServerList) dnslist = (PIP_ADDR_STRING)(fi + 1);
       else dnslist += 1;
-  }     
+  }
   fi->EnableDns = 1;
   return ERROR_SUCCESS;