1 /* Copyright (C) 2003 Art Yerkes
2  * A reimplementation of ifenum.c by Juan Lang
3  *
4  * This library is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Lesser General Public
6  * License as published by the Free Software Foundation; either
7  * version 2.1 of the License, or (at your option) any later version.
8  *
9  * This library is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  * Lesser General Public License for more details.
13  *
14  * You should have received a copy of the GNU Lesser General Public
15  * License along with this library; if not, write to the Free Software
16  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
17  *
18  * This file is implemented on the IOCTL_TCP_QUERY_INFORMATION_EX ioctl on
19  * tcpip.sys
20  */
21 
22 #include "iphlpapi_private.h"
23 
24 WINE_DEFAULT_DEBUG_CHANNEL(iphlpapi);
25 
26 #ifndef TCPS_ESTABLISHED
27 # define TCPS_ESTABLISHED TCP_ESTABLISHED
28 #endif
29 #ifndef TCPS_SYN_SENT
30 # define TCPS_SYN_SENT TCP_SYN_SENT
31 #endif
32 #ifndef TCPS_SYN_RECEIVED
33 # define TCPS_SYN_RECEIVED TCP_SYN_RECV
34 #endif
35 #ifndef TCPS_FIN_WAIT_1
36 # define TCPS_FIN_WAIT_1 TCP_FIN_WAIT1
37 #endif
38 #ifndef TCPS_FIN_WAIT_2
39 # define TCPS_FIN_WAIT_2 TCP_FIN_WAIT2
40 #endif
41 #ifndef TCPS_TIME_WAIT
42 # define TCPS_TIME_WAIT TCP_TIME_WAIT
43 #endif
44 #ifndef TCPS_CLOSED
45 # define TCPS_CLOSED TCP_CLOSE
46 #endif
47 #ifndef TCPS_CLOSE_WAIT
48 # define TCPS_CLOSE_WAIT TCP_CLOSE_WAIT
49 #endif
50 #ifndef TCPS_LAST_ACK
51 # define TCPS_LAST_ACK TCP_LAST_ACK
52 #endif
53 #ifndef TCPS_LISTEN
54 # define TCPS_LISTEN TCP_LISTEN
55 #endif
56 #ifndef TCPS_CLOSING
57 # define TCPS_CLOSING TCP_CLOSING
58 #endif
59 
60 BOOL isIpEntity( HANDLE tcpFile, TDIEntityID *ent ) {
61     return (ent->tei_entity == CL_NL_ENTITY ||
62             ent->tei_entity == CO_NL_ENTITY);
63 }
64 
65 NTSTATUS getNthIpEntity( HANDLE tcpFile, DWORD index, TDIEntityID *ent ) {
66     DWORD numEntities = 0;
67     DWORD numRoutes = 0;
68     TDIEntityID *entitySet = 0;
69     NTSTATUS status = tdiGetEntityIDSet( tcpFile, &entitySet, &numEntities );
70     int i;
71 
72     if( !NT_SUCCESS(status) )
73         return status;
74 
75     for( i = 0; i < numEntities; i++ ) {
76         if( isIpEntity( tcpFile, &entitySet[i] ) ) {
77             TRACE("Entity %d is an IP Entity\n", i);
78             if( numRoutes == index ) break;
79             else numRoutes++;
80         }
81     }
82 
83     if( numRoutes == index && i < numEntities ) {
84         TRACE("Index %lu is entity #%d - %04x:%08x\n", index, i,
85               entitySet[i].tei_entity, entitySet[i].tei_instance);
86         memcpy( ent, &entitySet[i], sizeof(*ent) );
87         tdiFreeThingSet( entitySet );
88         return STATUS_SUCCESS;
89     } else {
90         tdiFreeThingSet( entitySet );
91         return STATUS_UNSUCCESSFUL;
92     }
93 }
94 
95 NTSTATUS tdiGetMibForIpEntity
96 ( HANDLE tcpFile, TDIEntityID *ent, IPSNMPInfo *entry ) {
97     TCP_REQUEST_QUERY_INFORMATION_EX req = TCP_REQUEST_QUERY_INFORMATION_INIT;
98     NTSTATUS status = STATUS_SUCCESS;
99     DWORD returnSize;
100 
101     memset( entry, 0, sizeof( *entry ) );
102 
103     TRACE("TdiGetMibForIpEntity(tcpFile 0x%p, entityId 0x%x)\n",
104           tcpFile, ent->tei_instance);
105 
106     req.ID.toi_class                = INFO_CLASS_PROTOCOL;
107     req.ID.toi_type                 = INFO_TYPE_PROVIDER;
108     req.ID.toi_id                   = IP_MIB_STATS_ID;
109     req.ID.toi_entity               = *ent;
110 
111     status = DeviceIoControl( tcpFile,
112                               IOCTL_TCP_QUERY_INFORMATION_EX,
113                               &req,
114                               sizeof(req),
115                               entry,
116                               sizeof(*entry),
117                               &returnSize,
118                               NULL );
119 
120     TRACE("TdiGetMibForIpEntity() => status = 0x%08lx, entry = {\n"
121            "  ipsi_forwarding ............ %lu\n"
122            "  ipsi_defaultttl ............ %lu\n"
123            "  ipsi_inreceives ............ %lu\n"
124            "  ipsi_indelivers ............ %lu\n"
125            "  ipsi_outrequests ........... %lu\n"
126            "  ipsi_routingdiscards ....... %lu\n"
127            "  ipsi_outdiscards ........... %lu\n"
128            "  ipsi_outnoroutes ........... %lu\n"
129            "  ipsi_numif ................. %lu\n"
130            "  ipsi_numaddr ............... %lu\n"
131            "  ipsi_numroutes ............. %lu\n"
132            "}\n",
133           status,
134           entry->ipsi_forwarding,
135           entry->ipsi_defaultttl,
136           entry->ipsi_inreceives,
137           entry->ipsi_indelivers,
138           entry->ipsi_outrequests,
139           entry->ipsi_routingdiscards,
140           entry->ipsi_outdiscards,
141           entry->ipsi_outnoroutes,
142           entry->ipsi_numif,
143           entry->ipsi_numaddr,
144           entry->ipsi_numroutes);
145 
146     return status;
147 }
148 
149 NTSTATUS tdiGetRoutesForIpEntity
150 ( HANDLE tcpFile, TDIEntityID *ent, IPRouteEntry **routes, PDWORD numRoutes ) {
151     NTSTATUS status = STATUS_SUCCESS;
152 
153     TRACE("TdiGetRoutesForIpEntity(tcpFile 0x%p, entityId 0x%x)\n",
154           tcpFile, ent->tei_instance);
155 
156     status = tdiGetSetOfThings( tcpFile,
157                                 INFO_CLASS_PROTOCOL,
158                                 INFO_TYPE_PROVIDER,
159                                 IP_MIB_ARPTABLE_ENTRY_ID,
160                                 CL_NL_ENTITY,
161 				ent->tei_instance,
162                                 0,
163                                 sizeof(IPRouteEntry),
164                                 (PVOID *)routes,
165                                 numRoutes);
166 
167     return status;
168 }
169 
170 NTSTATUS tdiGetIpAddrsForIpEntity
171 ( HANDLE tcpFile, TDIEntityID *ent, IPAddrEntry **addrs, PDWORD numAddrs ) {
172     NTSTATUS status;
173 
174     TRACE("TdiGetIpAddrsForIpEntity(tcpFile 0x%p, entityId 0x%x)\n",
175           tcpFile, ent->tei_instance);
176 
177     status = tdiGetSetOfThings( tcpFile,
178                                 INFO_CLASS_PROTOCOL,
179                                 INFO_TYPE_PROVIDER,
180                                 IP_MIB_ADDRTABLE_ENTRY_ID,
181                                 CL_NL_ENTITY,
182                                 ent->tei_instance,
183                                 0,
184                                 sizeof(IPAddrEntry),
185                                 (PVOID *)addrs,
186                                 numAddrs );
187 
188     return status;
189 }
190 
191 DWORD getInterfaceStatsByName(const char *name, PMIB_IFROW entry)
192 {
193   if (!name)
194     return ERROR_INVALID_PARAMETER;
195   if (!entry)
196     return ERROR_INVALID_PARAMETER;
197 
198   return NO_ERROR;
199 }
200 
201 DWORD getInterfaceStatsByIndex(DWORD index, PMIB_IFROW entry)
202 {
203     return ERROR_INVALID_PARAMETER;
204 }
205 
206 DWORD getICMPStats(MIB_ICMP *stats)
207 {
208   FILE *fp;
209 
210   if (!stats)
211     return ERROR_INVALID_PARAMETER;
212 
213   memset(stats, 0, sizeof(MIB_ICMP));
214   /* get most of these stats from /proc/net/snmp, no error if can't */
215   fp = fopen("/proc/net/snmp", "r");
216   if (fp) {
217     const char hdr[] = "Icmp:";
218     char buf[512] = { 0 }, *ptr;
219 
220     do {
221       ptr = fgets(buf, sizeof(buf), fp);
222     } while (ptr && strncasecmp(buf, hdr, sizeof(hdr) - 1));
223     if (ptr) {
224       /* last line was a header, get another */
225       ptr = fgets(buf, sizeof(buf), fp);
226       if (ptr && strncasecmp(buf, hdr, sizeof(hdr) - 1) == 0) {
227         char *endPtr;
228 
229         ptr += sizeof(hdr);
230         if (ptr && *ptr) {
231           stats->stats.icmpInStats.dwMsgs = strtoul(ptr, &endPtr, 10);
232           ptr = endPtr;
233         }
234         if (ptr && *ptr) {
235           stats->stats.icmpInStats.dwErrors = strtoul(ptr, &endPtr, 10);
236           ptr = endPtr;
237         }
238         if (ptr && *ptr) {
239           stats->stats.icmpInStats.dwDestUnreachs = strtoul(ptr, &endPtr, 10);
240           ptr = endPtr;
241         }
242         if (ptr && *ptr) {
243           stats->stats.icmpInStats.dwTimeExcds = strtoul(ptr, &endPtr, 10);
244           ptr = endPtr;
245         }
246         if (ptr && *ptr) {
247           stats->stats.icmpInStats.dwParmProbs = strtoul(ptr, &endPtr, 10);
248           ptr = endPtr;
249         }
250         if (ptr && *ptr) {
251           stats->stats.icmpInStats.dwSrcQuenchs = strtoul(ptr, &endPtr, 10);
252           ptr = endPtr;
253         }
254         if (ptr && *ptr) {
255           stats->stats.icmpInStats.dwRedirects = strtoul(ptr, &endPtr, 10);
256           ptr = endPtr;
257         }
258         if (ptr && *ptr) {
259           stats->stats.icmpInStats.dwEchoReps = strtoul(ptr, &endPtr, 10);
260           ptr = endPtr;
261         }
262         if (ptr && *ptr) {
263           stats->stats.icmpInStats.dwTimestamps = strtoul(ptr, &endPtr, 10);
264           ptr = endPtr;
265         }
266         if (ptr && *ptr) {
267           stats->stats.icmpInStats.dwTimestampReps = strtoul(ptr, &endPtr, 10);
268           ptr = endPtr;
269         }
270         if (ptr && *ptr) {
271           stats->stats.icmpInStats.dwAddrMasks = strtoul(ptr, &endPtr, 10);
272           ptr = endPtr;
273         }
274         if (ptr && *ptr) {
275           stats->stats.icmpInStats.dwAddrMaskReps = strtoul(ptr, &endPtr, 10);
276           ptr = endPtr;
277         }
278         if (ptr && *ptr) {
279           stats->stats.icmpOutStats.dwMsgs = strtoul(ptr, &endPtr, 10);
280           ptr = endPtr;
281         }
282         if (ptr && *ptr) {
283           stats->stats.icmpOutStats.dwErrors = strtoul(ptr, &endPtr, 10);
284           ptr = endPtr;
285         }
286         if (ptr && *ptr) {
287           stats->stats.icmpOutStats.dwDestUnreachs = strtoul(ptr, &endPtr, 10);
288           ptr = endPtr;
289         }
290         if (ptr && *ptr) {
291           stats->stats.icmpOutStats.dwTimeExcds = strtoul(ptr, &endPtr, 10);
292           ptr = endPtr;
293         }
294         if (ptr && *ptr) {
295           stats->stats.icmpOutStats.dwParmProbs = strtoul(ptr, &endPtr, 10);
296           ptr = endPtr;
297         }
298         if (ptr && *ptr) {
299           stats->stats.icmpOutStats.dwSrcQuenchs = strtoul(ptr, &endPtr, 10);
300           ptr = endPtr;
301         }
302         if (ptr && *ptr) {
303           stats->stats.icmpOutStats.dwRedirects = strtoul(ptr, &endPtr, 10);
304           ptr = endPtr;
305         }
306         if (ptr && *ptr) {
307           stats->stats.icmpOutStats.dwEchoReps = strtoul(ptr, &endPtr, 10);
308           ptr = endPtr;
309         }
310         if (ptr && *ptr) {
311           stats->stats.icmpOutStats.dwTimestamps = strtoul(ptr, &endPtr, 10);
312           ptr = endPtr;
313         }
314         if (ptr && *ptr) {
315           stats->stats.icmpOutStats.dwTimestampReps = strtoul(ptr, &endPtr, 10);
316           ptr = endPtr;
317         }
318         if (ptr && *ptr) {
319           stats->stats.icmpOutStats.dwAddrMasks = strtoul(ptr, &endPtr, 10);
320           ptr = endPtr;
321         }
322         if (ptr && *ptr) {
323           stats->stats.icmpOutStats.dwAddrMaskReps = strtoul(ptr, &endPtr, 10);
324           ptr = endPtr;
325         }
326       }
327     }
328     fclose(fp);
329   }
330   return NO_ERROR;
331 }
332 
333 DWORD getIPStats(PMIB_IPSTATS stats, DWORD family)
334 {
335   if (!stats)
336     return ERROR_INVALID_PARAMETER;
337 
338   if (family != AF_INET && family != AF_INET6)
339     return ERROR_INVALID_PARAMETER;
340 
341   return NO_ERROR;
342 }
343 
344 DWORD getTCPStats(MIB_TCPSTATS *stats, DWORD family)
345 {
346   if (!stats)
347     return ERROR_INVALID_PARAMETER;
348 
349   if (family != AF_INET && family != AF_INET6)
350     return ERROR_INVALID_PARAMETER;
351 
352   return NO_ERROR;
353 }
354 
355 DWORD getUDPStats(MIB_UDPSTATS *stats, DWORD family)
356 {
357   if (!stats)
358     return ERROR_INVALID_PARAMETER;
359 
360   if (family != AF_INET && family != AF_INET6)
361     return ERROR_INVALID_PARAMETER;
362 
363   return NO_ERROR;
364 }
365 
366 DWORD getNumRoutes(void)
367 {
368     DWORD numEntities, numRoutes = 0;
369     TDIEntityID *entitySet;
370     HANDLE tcpFile;
371     int i;
372     NTSTATUS status;
373 
374     TRACE("called.\n");
375 
376     status = openTcpFile( &tcpFile, FILE_READ_DATA );
377     if( !NT_SUCCESS(status) ) {
378         ERR("openTcpFile returned 0x%08lx\n", status);
379         return 0;
380     }
381 
382     status = tdiGetEntityIDSet( tcpFile, &entitySet, &numEntities );
383     if( !NT_SUCCESS(status) ) {
384         ERR("tdiGetEntityIDSet returned 0x%08lx\n", status);
385         closeTcpFile( tcpFile );
386         return 0;
387     }
388 
389     for( i = 0; i < numEntities; i++ ) {
390         if( isIpEntity( tcpFile, &entitySet[i] ) ) {
391             IPSNMPInfo isnmp;
392             memset( &isnmp, 0, sizeof( isnmp ) );
393             status = tdiGetMibForIpEntity( tcpFile, &entitySet[i], &isnmp );
394             if( !NT_SUCCESS(status) ) {
395                 ERR("tdiGetMibForIpEntity returned 0x%08lx, for i = %d\n", status, i);
396                 numRoutes = 0;
397                 break;
398             }
399             numRoutes += isnmp.ipsi_numroutes;
400         }
401     }
402 
403     TRACE("numRoutes = %lu\n", numRoutes);
404 
405     tdiFreeThingSet( entitySet );
406     closeTcpFile( tcpFile );
407 
408     return numRoutes;
409 }
410 
411 VOID HexDump( PCHAR Data, DWORD Len ) {
412     int i;
413 
414     for( i = 0; i < Len; i++ ) {
415         if( !(i & 0xf) ) {
416             if( i ) fprintf(stderr,"\n");
417             fprintf(stderr,"%08x:", i);
418         }
419         fprintf( stderr, " %02x", Data[i] & 0xff );
420     }
421     fprintf(stderr,"\n");
422 }
423 
424 RouteTable *getRouteTable(void)
425 {
426     RouteTable *out_route_table;
427     DWORD numRoutes = getNumRoutes(), routesAdded = 0;
428     TDIEntityID ent;
429     HANDLE tcpFile;
430     NTSTATUS status = openTcpFile( &tcpFile, FILE_READ_DATA );
431     int i;
432 
433     if( !NT_SUCCESS(status) )
434         return 0;
435 
436     TRACE("GETTING ROUTE TABLE\n");
437 
438     out_route_table = HeapAlloc( GetProcessHeap(), 0,
439                                  sizeof(RouteTable) +
440                                  (sizeof(RouteEntry) * (numRoutes - 1)) );
441     if (!out_route_table) {
442         closeTcpFile(tcpFile);
443         return NULL;
444     }
445 
446     out_route_table->numRoutes = numRoutes;
447 
448     for( i = 0; routesAdded < out_route_table->numRoutes; i++ ) {
449         int j;
450         IPRouteEntry *route_set;
451 
452         getNthIpEntity( tcpFile, i, &ent );
453 
454         tdiGetRoutesForIpEntity( tcpFile, &ent, &route_set, &numRoutes );
455 
456         if( !route_set ) {
457             closeTcpFile( tcpFile );
458             HeapFree( GetProcessHeap(), 0, out_route_table );
459             return 0;
460         }
461 
462         TRACE("%lu routes in instance %d\n", numRoutes, i);
463 #if 0
464         HexDump( route_set,
465                  sizeof( IPRouteEntry ) *
466                  snmpInfo.ipsi_numroutes );
467 #endif
468 
469         for( j = 0; j < numRoutes; j++ ) {
470             int routeNum = j + routesAdded;
471             out_route_table->routes[routeNum].dest =
472                 route_set[j].ire_dest;
473             out_route_table->routes[routeNum].mask =
474                 route_set[j].ire_mask;
475             out_route_table->routes[routeNum].gateway =
476                 route_set[j].ire_gw;
477             out_route_table->routes[routeNum].ifIndex =
478                 route_set[j].ire_index;
479             out_route_table->routes[routeNum].metric =
480                 route_set[j].ire_metric1;
481         }
482 
483         if( route_set ) tdiFreeThingSet( route_set );
484 
485         routesAdded += numRoutes;
486     }
487 
488     closeTcpFile( tcpFile );
489 
490     TRACE("status = 0x%08lx, out_route_table = 0x%p\n", status, out_route_table);
491 
492     return out_route_table;
493 }
494 
495 DWORD getNumArpEntries(void)
496 {
497     DWORD numEntities;
498     TDIEntityID *entitySet = NULL;
499     HANDLE tcpFile;
500     int i, totalNumber = 0;
501     NTSTATUS status;
502     PMIB_IPNETROW IpArpTable = NULL;
503     DWORD returnSize;
504 
505     TRACE("called.\n");
506 
507     status = openTcpFile( &tcpFile, FILE_READ_DATA );
508     if( !NT_SUCCESS(status) ) {
509         ERR("openTcpFile returned 0x%08lx\n", status);
510         return 0;
511     }
512 
513     status = tdiGetEntityIDSet( tcpFile, &entitySet, &numEntities );
514 
515     for( i = 0; i < numEntities; i++ ) {
516         if( isInterface( &entitySet[i] ) &&
517 	    hasArp( tcpFile, &entitySet[i] ) ) {
518 
519 	    status = tdiGetSetOfThings( tcpFile,
520 					INFO_CLASS_PROTOCOL,
521 					INFO_TYPE_PROVIDER,
522 					IP_MIB_ARPTABLE_ENTRY_ID,
523 					AT_ENTITY,
524 					entitySet[i].tei_instance,
525 					0,
526 					sizeof(MIB_IPNETROW),
527 					(PVOID *)&IpArpTable,
528 					&returnSize );
529 
530 	    if( status == STATUS_SUCCESS ) totalNumber += returnSize;
531 		if( IpArpTable ) {
532 			tdiFreeThingSet( IpArpTable );
533 			IpArpTable = NULL;
534 		}
535 	}
536     }
537 
538     closeTcpFile( tcpFile );
539     if( entitySet ) tdiFreeThingSet( entitySet );
540     return totalNumber;
541 }
542 
543 PMIB_IPNETTABLE getArpTable(void)
544 {
545     DWORD numEntities, returnSize;
546     TDIEntityID *entitySet;
547     HANDLE tcpFile;
548     int i, totalNumber, TmpIdx, CurrIdx = 0;
549     NTSTATUS status;
550     PMIB_IPNETTABLE IpArpTable = NULL;
551     PMIB_IPNETROW AdapterArpTable = NULL;
552 
553     TRACE("called.\n");
554 
555     totalNumber = getNumArpEntries();
556 
557     status = openTcpFile( &tcpFile, FILE_READ_DATA );
558     if( !NT_SUCCESS(status) ) {
559         ERR("openTcpFile returned 0x%08lx\n", status);
560         return 0;
561     }
562 
563     IpArpTable = HeapAlloc
564 	( GetProcessHeap(), 0,
565 	  sizeof(DWORD) + (sizeof(MIB_IPNETROW) * totalNumber) );
566     if (!IpArpTable) {
567         closeTcpFile(tcpFile);
568         return NULL;
569     }
570 
571     status = tdiGetEntityIDSet( tcpFile, &entitySet, &numEntities );
572 
573     for( i = 0; i < numEntities; i++ ) {
574         if( isInterface( &entitySet[i] ) &&
575 	    hasArp( tcpFile, &entitySet[i] ) ) {
576 
577 	    status = tdiGetSetOfThings( tcpFile,
578 					INFO_CLASS_PROTOCOL,
579 					INFO_TYPE_PROVIDER,
580 					IP_MIB_ARPTABLE_ENTRY_ID,
581 					AT_ENTITY,
582 					entitySet[i].tei_instance,
583 					0,
584 					sizeof(MIB_IPNETROW),
585 					(PVOID *)&AdapterArpTable,
586 					&returnSize );
587 
588             if( status == STATUS_SUCCESS ) {
589                 for( TmpIdx = 0; TmpIdx < returnSize; TmpIdx++, CurrIdx++ )
590                     IpArpTable->table[CurrIdx] = AdapterArpTable[TmpIdx];
591                 tdiFreeThingSet( AdapterArpTable );
592             }
593         }
594     }
595 
596     closeTcpFile( tcpFile );
597 
598     tdiFreeThingSet( entitySet );
599     IpArpTable->dwNumEntries = CurrIdx;
600 
601     return IpArpTable;
602 }
603 
604 struct _TABLE_CALL
605 {
606     DWORD TOIID;
607     SIZE_T UdpSize;
608     SIZE_T TcpSize;
609     SIZE_T UdpOffset;
610     SIZE_T TcpOffset;
611 } UdpTcpTableCall[] = {
612     {IP_MIB_ARPTABLE_ENTRY_ID, sizeof(MIB_UDPROW), sizeof(MIB_TCPROW), FIELD_OFFSET(MIB_UDPTABLE, table), FIELD_OFFSET(MIB_TCPTABLE, table)},
613     {IP_MIB_ADDRTABLE_ENTRY_ID, sizeof(MIB_UDPROW_OWNER_PID), sizeof(MIB_TCPROW_OWNER_PID), FIELD_OFFSET(MIB_UDPTABLE_OWNER_PID, table), FIELD_OFFSET(MIB_TCPTABLE_OWNER_PID, table)},
614     {IP_SPECIFIC_MODULE_ENTRY_ID, sizeof(MIB_UDPROW_OWNER_MODULE), sizeof(MIB_TCPROW_OWNER_MODULE), FIELD_OFFSET(MIB_UDPTABLE_OWNER_MODULE, table), FIELD_OFFSET(MIB_TCPTABLE_OWNER_MODULE, table)},
615 };
616 
617 #define Add2Ptr(PTR, INC) (PVOID)((ULONG_PTR)(PTR) + (INC))
618 
619 DWORD getNumUdpEntries(void)
620 {
621     DWORD numEntities;
622     TDIEntityID *entitySet = NULL;
623     HANDLE tcpFile;
624     int i, totalNumber = 0;
625     NTSTATUS status;
626     PMIB_UDPROW IpUdpTable = NULL;
627     DWORD returnSize;
628 
629     TRACE("called.\n");
630 
631     status = openTcpFile( &tcpFile, FILE_READ_DATA );
632     if( !NT_SUCCESS(status) ) {
633         ERR("openTcpFile returned 0x%08lx\n", status);
634         return 0;
635     }
636 
637     status = tdiGetEntityIDSet( tcpFile, &entitySet, &numEntities );
638 
639     for( i = 0; i < numEntities; i++ ) {
640         if( entitySet[i].tei_entity == CL_TL_ENTITY &&
641 	    hasArp( tcpFile, &entitySet[i] ) ) {
642 
643 	    status = tdiGetSetOfThings( tcpFile,
644 					INFO_CLASS_PROTOCOL,
645 					INFO_TYPE_PROVIDER,
646 					IP_MIB_ARPTABLE_ENTRY_ID,
647 					CL_TL_ENTITY,
648 					entitySet[i].tei_instance,
649 					0,
650 					sizeof(MIB_UDPROW),
651 					(PVOID *)&IpUdpTable,
652 					&returnSize );
653 
654 	    if( status == STATUS_SUCCESS ) totalNumber += returnSize;
655 		if( IpUdpTable ) {
656 			tdiFreeThingSet( IpUdpTable );
657 			IpUdpTable = NULL;
658 		}
659 	}
660     }
661 
662     closeTcpFile( tcpFile );
663     if( entitySet ) tdiFreeThingSet( entitySet );
664     return totalNumber;
665 }
666 
667 PVOID getUdpTable(CLASS_TABLE Class)
668 {
669     DWORD numEntities, returnSize;
670     TDIEntityID *entitySet;
671     HANDLE tcpFile;
672     int i, totalNumber, TmpIdx, CurrIdx = 0;
673     NTSTATUS status;
674     PMIB_UDPTABLE IpUdpTable = NULL;
675     PVOID AdapterUdpTable = NULL;
676 
677     TRACE("called.\n");
678 
679     totalNumber = getNumUdpEntries();
680 
681     status = openTcpFile( &tcpFile, FILE_READ_DATA );
682     if( !NT_SUCCESS(status) ) {
683         ERR("openTcpFile returned 0x%08lx\n", status);
684         return 0;
685     }
686 
687     IpUdpTable = HeapAlloc
688 	( GetProcessHeap(), 0,
689 	  UdpTcpTableCall[Class].UdpOffset + (UdpTcpTableCall[Class].UdpSize * totalNumber) );
690     if (!IpUdpTable) {
691         closeTcpFile(tcpFile);
692         return NULL;
693     }
694 
695     status = tdiGetEntityIDSet( tcpFile, &entitySet, &numEntities );
696 
697     for( i = 0; i < numEntities; i++ ) {
698         if( entitySet[i].tei_entity == CL_TL_ENTITY &&
699 	    hasArp( tcpFile, &entitySet[i] ) ) {
700 
701 	    status = tdiGetSetOfThings( tcpFile,
702 					INFO_CLASS_PROTOCOL,
703 					INFO_TYPE_PROVIDER,
704 					UdpTcpTableCall[Class].TOIID,
705 					CL_TL_ENTITY,
706 					entitySet[i].tei_instance,
707 					0,
708 					UdpTcpTableCall[Class].UdpSize,
709 					&AdapterUdpTable,
710 					&returnSize );
711 
712             if( status == STATUS_SUCCESS ) {
713                 for( TmpIdx = 0; TmpIdx < returnSize; TmpIdx++, CurrIdx++ )
714                     CopyMemory(Add2Ptr(IpUdpTable, UdpTcpTableCall[Class].UdpOffset + UdpTcpTableCall[Class].UdpSize * CurrIdx),
715                                Add2Ptr(AdapterUdpTable, UdpTcpTableCall[Class].UdpSize * TmpIdx),
716                                UdpTcpTableCall[Class].UdpSize);
717                 tdiFreeThingSet( AdapterUdpTable );
718             }
719         }
720     }
721 
722     closeTcpFile( tcpFile );
723 
724     tdiFreeThingSet( entitySet );
725     IpUdpTable->dwNumEntries = CurrIdx;
726 
727     return IpUdpTable;
728 }
729 
730 DWORD getNumTcpEntries(void)
731 {
732     DWORD numEntities;
733     TDIEntityID *entitySet = NULL;
734     HANDLE tcpFile;
735     int i, totalNumber = 0;
736     NTSTATUS status;
737     PMIB_TCPROW IpTcpTable = NULL;
738     DWORD returnSize;
739 
740     TRACE("called.\n");
741 
742     status = openTcpFile( &tcpFile, FILE_READ_DATA );
743     if( !NT_SUCCESS(status) ) {
744         ERR("openTcpFile returned 0x%08lx\n", status);
745         return 0;
746     }
747 
748     status = tdiGetEntityIDSet( tcpFile, &entitySet, &numEntities );
749 
750     for( i = 0; i < numEntities; i++ ) {
751         if( entitySet[i].tei_entity == CO_TL_ENTITY &&
752 	    hasArp( tcpFile, &entitySet[i] ) ) {
753 
754 	    status = tdiGetSetOfThings( tcpFile,
755 					INFO_CLASS_PROTOCOL,
756 					INFO_TYPE_PROVIDER,
757 					IP_MIB_ARPTABLE_ENTRY_ID,
758 					CO_TL_ENTITY,
759 					entitySet[i].tei_instance,
760 					0,
761 					sizeof(MIB_TCPROW),
762 					(PVOID *)&IpTcpTable,
763 					&returnSize );
764 
765 	    if( status == STATUS_SUCCESS ) totalNumber += returnSize;
766 		if( IpTcpTable ) {
767 			tdiFreeThingSet( IpTcpTable );
768 			IpTcpTable = NULL;
769 		}
770 	}
771     }
772 
773     closeTcpFile( tcpFile );
774     if( entitySet ) tdiFreeThingSet( entitySet );
775     return totalNumber;
776 }
777 
778 PVOID getTcpTable(CLASS_TABLE Class)
779 {
780     DWORD numEntities, returnSize;
781     TDIEntityID *entitySet;
782     HANDLE tcpFile;
783     int i, totalNumber, TmpIdx, CurrIdx = 0;
784     NTSTATUS status;
785     PMIB_TCPTABLE IpTcpTable = NULL;
786     PVOID AdapterTcpTable = NULL;
787 
788     TRACE("called.\n");
789 
790     totalNumber = getNumTcpEntries();
791 
792     status = openTcpFile( &tcpFile, FILE_READ_DATA );
793     if( !NT_SUCCESS(status) ) {
794         ERR("openTcpFile returned 0x%08lx\n", status);
795         return 0;
796     }
797 
798     IpTcpTable = HeapAlloc
799 	( GetProcessHeap(), 0,
800 	  UdpTcpTableCall[Class].TcpOffset + (UdpTcpTableCall[Class].TcpSize * totalNumber) );
801     if (!IpTcpTable) {
802         closeTcpFile(tcpFile);
803         return NULL;
804     }
805 
806     status = tdiGetEntityIDSet( tcpFile, &entitySet, &numEntities );
807 
808     for( i = 0; i < numEntities; i++ ) {
809         if( entitySet[i].tei_entity == CO_TL_ENTITY &&
810 	    hasArp( tcpFile, &entitySet[i] ) ) {
811 
812 	    status = tdiGetSetOfThings( tcpFile,
813 					INFO_CLASS_PROTOCOL,
814 					INFO_TYPE_PROVIDER,
815 					UdpTcpTableCall[Class].TOIID,
816 					CO_TL_ENTITY,
817 					entitySet[i].tei_instance,
818 					0,
819 					UdpTcpTableCall[Class].TcpSize,
820 					&AdapterTcpTable,
821 					&returnSize );
822 
823             if( status == STATUS_SUCCESS ) {
824                 for( TmpIdx = 0; TmpIdx < returnSize; TmpIdx++, CurrIdx++ )
825                     CopyMemory(Add2Ptr(IpTcpTable, UdpTcpTableCall[Class].TcpOffset + UdpTcpTableCall[Class].TcpSize * CurrIdx),
826                                Add2Ptr(AdapterTcpTable, UdpTcpTableCall[Class].TcpSize * TmpIdx),
827                                UdpTcpTableCall[Class].TcpSize);
828                 tdiFreeThingSet( AdapterTcpTable );
829             }
830         }
831     }
832 
833     closeTcpFile( tcpFile );
834 
835     tdiFreeThingSet( entitySet );
836     IpTcpTable->dwNumEntries = CurrIdx;
837 
838     return IpTcpTable;
839 }
840