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

Timestamp:

Feb 20, 2002, 12:57:31 PM (24 years ago)

Author:

sandervl

Message:

PF: Implemented GetAdaptersInfo & GetNetworkParams

File:

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

trunk/src/iphlpapi/iphlpapi.cpp

@@ -1 @@
-/* $Id: iphlpapi.cpp,v 1.5 2001-10-29 13:37:29 phaller Exp $ */
+/* $Id: iphlpapi.cpp,v 1.6 2002-02-20 11:57:31 sandervl Exp $ */
 /*
  *      IPHLPAPI library
@@ -10 +10 @@
  ****************************************************************************/
 
+
+#include <stdio.h>
 #include <odin.h>
 #include <odinwrap.h>
@@ -18 +20 @@
 
 #include <string.h>
-
 #include "iphlpapi.h"
+
+#define BSD_SELECT 1
+#define OS2 1
+
+#include <types.h>
+#include <sys\socket.h>
+#include <sys\ioctl.h>
+#include <net\route.h>
+#include <net\if.h>
+#include <net\if_arp.h>
+#ifdef TCPV40HDRS
+#include <netinet\in.h>
+#include <arpa\NAMESER.H>
+#endif
+#include <resolv.h>
+#include <unistd.h>
+
+#include "iphlwrap.h"
+
+#pragma pack(1)
+typedef struct
+{
+        unsigned long IPAddress;
+        unsigned short interfaceNum;
+        unsigned long netmask;
+        unsigned long broadcastAddress;
+} interfaceInformation;
+
+/* Incomplete but will do for us */
+
+#ifndef TCPV40HDRS
+struct ifnet {
+        char    *if_name;               /* name, e.g. ``en'' or ``lo'' */
+        short   if_unit;                /* sub-unit for lower level driver */
+        short   if_mtu;                 /* maximum transmission unit */
+};
+#endif
+
+struct ilist {
+  struct ilist *next;
+  unsigned long addr;
+};
+
+struct rlist {
+  struct rlist *next;
+  unsigned long gate;
+  unsigned long net;
+  unsigned long mask;
+  char done;
+};
+
+#pragma pack()
 
 ODINDEBUGCHANNEL(IPHLPAPI-IPHLPAPI)
@@ -28 +81 @@
  ****************************************************************************/
 
-static PIP_ADAPTER_INFO pipAdapters = NULL;
-
-
-//******************************************************************************
-//******************************************************************************
+static PIP_ADAPTER_INFO pipAdapter = NULL;
+
+
+//******************************************************************************
+//******************************************************************************
+
+void stringIPAddress(char* dst,u_long data)
+{
+    sprintf(dst, "%u.%u.%u.%u", 
+                        (char)data,
+                        (char)(*(((char*)&data) + 1)),
+                        (char)(*(((char*)&data) + 2)),
+                        (char)(*(((char*)&data) + 3)));
+}
+
+void stringNetmask(char* dst,u_long data)
+{
+   sprintf(dst,"%u.%u.%u.%u",
+                        (char)(*(((char*)&data) + 3)),
+                        (char)(*(((char*)&data) + 2)),
+                        (char)(*(((char*)&data) + 1)),
+                        (char)data);
+}
+
 static void i_initializeAdapterInformation(void)
 {
-  // @@@PH
-  // yet a fake to test some app
-  pipAdapters = (PIP_ADAPTER_INFO)malloc (sizeof (IP_ADAPTER_INFO) );
-  
-  memset(pipAdapters, 0, sizeof( IP_ADAPTER_INFO ));
-  pipAdapters->Next = NULL;
-  pipAdapters->ComboIndex = 1;
-  strcpy(pipAdapters->AdapterName, "ODIN IPHLPAPI Test Adapter");
-  strcpy(pipAdapters->Description, "ODIN IPHLPAPI Test Adapter (faked information)");
-  pipAdapters->AddressLength = 6; // MAX address
-  pipAdapters->Address[0] = 'V';
-  pipAdapters->Address[1] = 'P';
-  pipAdapters->Address[2] = 'C';
-  pipAdapters->Address[3] = 'O';
-  pipAdapters->Address[4] = 'S';
-  pipAdapters->Address[5] = '2';
-  pipAdapters->Index = 16777218;
-  pipAdapters->Type = 6;
-  pipAdapters->DhcpEnabled = 0;
-  
-  static IP_ADDR_STRING iasLocalhost;
-  iasLocalhost.Next = NULL;
-  strcpy((char*)&iasLocalhost.IpAddress,"127.0.0.1");
-  strcpy((char*)&iasLocalhost.IpMask, "255.0.0.0");
-  iasLocalhost.Context = 0;
-  
-  static IP_ADDR_STRING iasGateway;
-  iasGateway.Next = NULL;
-  strcpy((char*)&iasGateway.IpAddress,"192.168.1.1");
-  strcpy((char*)&iasGateway.IpMask, "255.255.255.0");
-  iasGateway.Context = 0;
-  
-  
-  memcpy((char*)&pipAdapters->IpAddressList, (char*)&iasLocalhost, sizeof(iasLocalhost));
-  pipAdapters->CurrentIpAddress = &pipAdapters->IpAddressList;
-  memcpy((char*)&pipAdapters->GatewayList,   (char*)&iasGateway, sizeof(iasGateway));
-  memset((char*)&pipAdapters->DhcpServer, 0, sizeof( IP_ADDR_STRING ) );
-  pipAdapters->HaveWins = 0;
-  memset((char*)&pipAdapters->PrimaryWinsServer, 0, sizeof( IP_ADDR_STRING ) );
-  memset((char*)&pipAdapters->SecondaryWinsServer, 0, sizeof( IP_ADDR_STRING ) );
-  pipAdapters->LeaseObtained = 0;
-  pipAdapters->LeaseExpires = 0;
-}
-
-// copy over the whole list and advance the target pointer
-static void i_copyIP_ADDRESS_STRING(PBYTE *ppTarget, PIP_ADDR_STRING pias)
-{
+  unsigned char *p;
+  char iShortName[6];
+  struct rlist * rnode = NULL, * rdef = NULL;
+  PIP_ADAPTER_INFO oldAdapter = NULL,topAdapter = NULL;
+  struct sockaddr_in * sin;
+  struct ifmib ifmibget;
+  int rc,i;
+  char *buffer,*buffer2;
+  struct ifnet *pifnet;
+  short int cInterfaces;
+#ifndef TCPV40HDRS
+  struct ortentry *r;
+#else
+  struct rtentry *r;
+#endif
+
+  // Init Subsystem and open a socket for ioctl() calls 
+  sock_init();
+
+  int clientSocket = socket(PF_INET, SOCK_STREAM, 0);
+  dprintf(("IPHLPAPI(Init): Opened socket %d\n", clientSocket));
+
+  // Whole buf minimum size is 65536 and memsets are really needed in other case 
+  // we will get garbage in adapter names.
+
+  buffer = (char*)malloc(64 * 1024);
+  memset(buffer, 0, 65536);
+  memset(&ifmibget,0, sizeof(struct ifmib));
+
+  rc = ioctl(clientSocket, SIOSTATIF, (char*)&ifmibget, sizeof(struct ifmib));
+  dprintf(("IPHLPAPI(SIOSTATIF) ioctl returned: %d\n", rc));
+  if (rc == -1)
+  {
+    free(buffer);
+    soclose(clientSocket);         
+    return; 
+  }        
+
+  rc = ioctl(clientSocket, SIOSTATAT, (char*)buffer, 65536);
+  dprintf(("IPHLPAPI(SIOSTATAT) ioctl returned: %d\n", rc));
+  if (rc == -1)
+  {
+    free(buffer);
+    soclose(clientSocket);         
+    return; 
+  }        
+  cInterfaces = *(short int*)buffer;
+  dprintf(("IPHLPAPI Call returned %d interfaces\n", cInterfaces));
+
+  for (int currentInterface = 0; currentInterface < cInterfaces; currentInterface++)
+  {
+    interfaceInformation *ifInfo = (interfaceInformation*)(buffer + 2 + (currentInterface * sizeof(interfaceInformation)));
+    // For now let's dump only lanX and pppX adapters, loopback (?)
+//    if (ifInfo->interfaceNum ==9) continue;
+
+    // Allocate and clear mem 
+    pipAdapter = (PIP_ADAPTER_INFO)malloc (sizeof (IP_ADAPTER_INFO));
+    memset(pipAdapter, 0, sizeof(IP_ADAPTER_INFO));
+    if (oldAdapter) 
+       oldAdapter->Next = pipAdapter;
+
+    pipAdapter->Next = NULL;
+    pipAdapter->ComboIndex = 1;
+    i =  ifInfo->interfaceNum;
+    // Gather some other stats 
+    dprintf(("IPHLPAPI: interface number: %u\n", ifInfo->interfaceNum));
+
+    if (ifInfo->interfaceNum>=0 && ifInfo->interfaceNum<9) // lanX
+       { strcpy(iShortName,"lan"); iShortName[3] = ifInfo->interfaceNum+48;
+         iShortName[4] = 0;}
+
+    // I do not like this very much, but seems there is no other documented
+    // way
+
+    if (strstr(ifmibget.iftable[i].ifDescr,"back")) // lo
+        strcpy(iShortName,"lo");       
+
+    if (strstr(ifmibget.iftable[i].ifDescr,"ace ppp")) // pppX
+        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"));       
+
+    if (strstr(ifmibget.iftable[i].ifDescr,"ace dod")) // dodX
+        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 
+    memcpy(pipAdapter->Address,ifmibget.iftable[i].ifPhysAddr,6);
+    pipAdapter->Index = ifmibget.iftable[i].ifIndex;
+    pipAdapter->Type = ifmibget.iftable[i].ifType; // Careful with this (?)
+    pipAdapter->DhcpEnabled = 0; // Also a question
+  
+    // TODO: Adapter may support multiple IP addrs
+    IP_ADDR_STRING iasAdapterIP;
+    iasAdapterIP.Next = NULL;
+    stringIPAddress((char*)&iasAdapterIP.IpAddress,ifInfo->IPAddress);
+    stringNetmask((char*)&iasAdapterIP.IpMask,ifInfo->netmask);
+    iasAdapterIP.Context = 0;
+ 
+    // 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) 
+    {
+      free(buffer);
+      free(buffer2);
+      soclose(clientSocket);       
+      // better return nothing than some trash, unwinding and freeing
+      for (topAdapter; pipAdapter; topAdapter = pipAdapter) {
+          pipAdapter = topAdapter -> Next;
+          free(topAdapter); 
+      }
+      return; 
+     }        
+         
+     rtentries *routeEntries = (rtentries*)buffer2;
+     p = (unsigned char *)&routeEntries->rttable[0];
+
+     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    
+#ifndef TCPV40HDRS
+         r = (struct ortentry *) (p);
+#else
+         r = (struct rtentry *) (p);
+#endif
+         if (strcmp((const char*)(p + sizeof(struct rtentry)),iShortName)==0)
+         {
+            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    
+               sin = (struct sockaddr_in *)(&r->rt_gateway);
+               strcpy(iasGateway.IpAddress.String,inet_ntoa(sin->sin_addr));
+               sin = (struct sockaddr_in *)&p[-4];
+               strcpy(iasGateway.IpMask.String,inet_ntoa(sin->sin_addr));
+               iasGateway.Context = 0;
+            }
+         }
+#ifndef TCPV40HDRS
+       p+=sizeof(struct ortentry);
+#else
+       p+=sizeof(struct rtentry);
+#endif
+       p+=strlen((char *)p)+1;
+    }
+  
+  memcpy((char*)&pipAdapter->IpAddressList, (char*)&iasAdapterIP, sizeof(iasAdapterIP));
+  pipAdapter->CurrentIpAddress = &pipAdapter->IpAddressList;
+  memcpy((char*)&pipAdapter->GatewayList,   (char*)&iasGateway, sizeof(iasGateway));
+  // what about OS/2 DHCP?
+  memset((char*)&pipAdapter->DhcpServer, 0, sizeof( IP_ADDR_STRING ) );
+  pipAdapter->HaveWins = 0;
+  memset((char*)&pipAdapter->PrimaryWinsServer, 0, sizeof( IP_ADDR_STRING ) );
+  memset((char*)&pipAdapter->SecondaryWinsServer, 0, sizeof( IP_ADDR_STRING ) );
+  pipAdapter->LeaseObtained = 0;
+  pipAdapter->LeaseExpires = 0;
+  if (!topAdapter) topAdapter = pipAdapter;
+  oldAdapter = pipAdapter;
+ }
+ pipAdapter = topAdapter;
+
+ // Cleanup
+ if (buffer) free(buffer);
+ if (buffer2) free(buffer2);
+ soclose(clientSocket); 
+}
+
+// copy over the whole list and advance the target pointer and correct new list
+static void i_copyIP_ADDRESS_STRING(PBYTE *ppTarget, PIP_ADDR_STRING pstruct,PIP_ADDR_STRING pias)
+{
+  PIP_ADDR_STRING dummy = NULL;
+  // We already have this copied
+  pias = pias -> Next;
   while (pias)
   {
     memcpy(*ppTarget, pias, sizeof( IP_ADDR_STRING ) );
+    pstruct->Next = (PIP_ADDR_STRING) *ppTarget;
     *ppTarget += sizeof ( IP_ADDR_STRING );
     pias = pias->Next;
-  }
+    pstruct = pstruct->Next;
+  }
+}
+
+static DWORD i_sizeOfFIXED_INFO(PFIXED_INFO pinfo)
+{
+ PIP_ADDR_STRING pias;
+
+ DWORD dwRequired = sizeof( FIXED_INFO );
+ 
+ // follow the IP_ADDR_STRING lists
+ pias = &pinfo->DnsServerList;
+ while( pias )
+ {
+   dwRequired += sizeof( IP_ADDR_STRING );
+   pias = pias->Next;
+ }
+ return dwRequired;
 }
 
@@ -147 +376 @@
               PULONG,           pOutBufLen)
 {
-  dprintf(("GetAdaptersInfo incorrectly implemented"));
-  
+  dprintf(("GetAdaptersInfo API called"));
+  dprintf(("Address passed is %p",pAdapterInfo)); 
   if (NULL == pOutBufLen)
     return ERROR_INVALID_PARAMETER;
@@ -156 +385 @@
     return ERROR_INVALID_PARAMETER;
   
-  if (NULL == pipAdapters)
+  if (NULL == pipAdapter)
   {
     // gather the information and save it
     i_initializeAdapterInformation();
-    
-    // determine number of IP adapters (interfaces) in the system
-    
-    // os2_ioctl() SIOSTATIF42
-  }
-  
-  if (NULL == pipAdapters)
+  }
+  
+  if (NULL == pipAdapter)
     return ERROR_NO_DATA;
   
@@ -177 +402 @@
   DWORD dwRequiredBuffer = 0;
   
-  for( pip = pipAdapters ; pip ; pip = pip->Next )
+  for( pip = pipAdapter ; pip ; pip = pip->Next )
   {
     // check for sufficient space
@@ -183 +408 @@
   }
   
-  for( pip = pipAdapters ; pip ; pip = pip->Next )
+  for( pip = pipAdapter ; pip ; pip = pip->Next )
   {
     // check for sufficient space
@@ -195 +420 @@
       memcpy(pTarget, pip, sizeof( IP_ADAPTER_INFO ));
       pTarget += sizeof( IP_ADAPTER_INFO );
-      
-      // @@@PH shall point to somewhere within the current buffer
-      pip->CurrentIpAddress = (PIP_ADDR_STRING)pTarget;
-      
-      i_copyIP_ADDRESS_STRING(&pTarget, &pip->IpAddressList);
-      i_copyIP_ADDRESS_STRING(&pTarget, &pip->GatewayList);
-      i_copyIP_ADDRESS_STRING(&pTarget, &pip->DhcpServer);
-      i_copyIP_ADDRESS_STRING(&pTarget, &pip->PrimaryWinsServer);
-      i_copyIP_ADDRESS_STRING(&pTarget, &pip->SecondaryWinsServer);
+       
+      // point to currentIPAddress 
+      ((PIP_ADAPTER_INFO)(pTarget))->CurrentIpAddress = &((PIP_ADAPTER_INFO)(pTarget))->IpAddressList;
+      //@@PF Need to fix lots of stuff here
+//      i_copyIP_ADDRESS_STRING(&pTarget, &pip->IpAddressList);
+//      i_copyIP_ADDRESS_STRING(&pTarget, &pip->GatewayList);
+//      i_copyIP_ADDRESS_STRING(&pTarget, &pip->DhcpServer);
+//      i_copyIP_ADDRESS_STRING(&pTarget, &pip->PrimaryWinsServer);
+//      i_copyIP_ADDRESS_STRING(&pTarget, &pip->SecondaryWinsServer);
     }
     else
@@ -212 +437 @@
     }
   }
-  
   return ERROR_SUCCESS;
 }
@@ -223 +447 @@
               PULONG,      pOutBufLen)
 {
-  dprintf(("GetNetworkParams not implemented"));
-  return ERROR_NOT_SUPPORTED; //NT returns this 
-}
-//******************************************************************************
-//******************************************************************************
-
+  struct sockaddr_in * sin;
+  FIXED_INFO fi; 
+  DWORD memNeeded;
+  PIP_ADDR_STRING dnslist = NULL, pdnslist = NULL;
+
+  // This is dynamically updated info
+  res_init();
+  memset(&fi,0,sizeof(FIXED_INFO));
+
+  dprintf(("GetNetworkParams pFixedInfo:%x pOutBufLen:%d",pFixedInfo,*pOutBufLen));
+
+  gethostname(fi.HostName,128);
+  strcpy(fi.DomainName,_res.defdname);
+
+  // For VPC DNS Servers are pretty much enough for now
+  fi.CurrentDnsServer = &fi.DnsServerList;   
+  dnslist = &fi.DnsServerList;
+  pdnslist = dnslist;
+
+  for (int i = 0; i<_res.nscount; i++)
+  {
+      if (!dnslist)    
+      {
+          dnslist = (PIP_ADDR_STRING) malloc(sizeof(IP_ADDR_STRING));
+          memset(dnslist,0,sizeof(IP_ADDR_STRING));
+          pdnslist->Next = dnslist;
+      }
+      sin = (struct sockaddr_in *)&_res.nsaddr_list[i];               
+      strcpy(dnslist->IpAddress.String,inet_ntoa(sin->sin_addr));
+      dnslist->Context = 0;
+      dprintf(("IPHLPAPI: GetNetworkParams Adding DNS Server %s",inet_ntoa(sin->sin_addr)));
+      pdnslist = dnslist;
+      dnslist = dnslist->Next;
+  }     
+  fi.EnableDns = 1;
+  // NodeType (?)
+  memNeeded = i_sizeOfFIXED_INFO(&fi);
+  LONG buflen = *pOutBufLen;
+  PBYTE pTarget        = (PBYTE)pFixedInfo;
+  if (((LONG)(buflen - memNeeded)) > 0)
+  {
+    // copy over the whole structure hierarchy
+    memcpy(pTarget, &fi, sizeof( FIXED_INFO ));
+    pTarget += sizeof( FIXED_INFO );
+    i_copyIP_ADDRESS_STRING(&pTarget, &pFixedInfo->DnsServerList,&fi.DnsServerList);
+  }
+  else
+  {
+   // return overall size of required buffer
+   *pOutBufLen = memNeeded;
+   return ERROR_BUFFER_OVERFLOW;
+  }
+
+ //cleanup
+ pdnslist = fi.DnsServerList.Next;
+ while( pdnslist )
+ {
+   dnslist = pdnslist->Next;
+   free(pdnslist);
+   pdnslist = dnslist;
+ } 
+
+ return ERROR_SUCCESS;
+}
+//******************************************************************************
+//******************************************************************************
 
 DWORD AddIPAddress(IPAddr Address, // IP address to add