1 /* Copyright (C) 2003 Art Yerkes
2 * A reimplementation of ifenum.c by Juan Lang
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.
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.
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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
18 * This file is implemented on the IOCTL_TCP_QUERY_INFORMATION_EX ioctl on
25 #include "iphlpapi_private.h"
29 #ifndef TCPS_ESTABLISHED
30 # define TCPS_ESTABLISHED TCP_ESTABLISHED
33 # define TCPS_SYN_SENT TCP_SYN_SENT
35 #ifndef TCPS_SYN_RECEIVED
36 # define TCPS_SYN_RECEIVED TCP_SYN_RECV
38 #ifndef TCPS_FIN_WAIT_1
39 # define TCPS_FIN_WAIT_1 TCP_FIN_WAIT1
41 #ifndef TCPS_FIN_WAIT_2
42 # define TCPS_FIN_WAIT_2 TCP_FIN_WAIT2
44 #ifndef TCPS_TIME_WAIT
45 # define TCPS_TIME_WAIT TCP_TIME_WAIT
48 # define TCPS_CLOSED TCP_CLOSE
50 #ifndef TCPS_CLOSE_WAIT
51 # define TCPS_CLOSE_WAIT TCP_CLOSE_WAIT
54 # define TCPS_LAST_ACK TCP_LAST_ACK
57 # define TCPS_LISTEN TCP_LISTEN
60 # define TCPS_CLOSING TCP_CLOSING
63 BOOL
isIpEntity( HANDLE tcpFile
, TDIEntityID
*ent
) {
64 DWORD entityType
, returnedLen
;
66 TCP_REQUEST_QUERY_INFORMATION_EX req
;
68 req
.ID
.toi_class
= INFO_CLASS_GENERIC
;
69 req
.ID
.toi_type
= INFO_TYPE_PROVIDER
;
70 req
.ID
.toi_id
= ENTITY_TYPE_ID
;
71 req
.ID
.toi_entity
= *ent
;
76 IOCTL_TCP_QUERY_INFORMATION_EX
,
84 DPRINT("Ent: %04x:d -> %04x\n",
85 ent
->tei_entity
, ent
->tei_instance
, entityType
);
87 return NT_SUCCESS(status
) && entityType
== CL_NL_IP
;
90 NTSTATUS
getNthIpEntity( HANDLE tcpFile
, DWORD index
, TDIEntityID
*ent
) {
91 DWORD numEntities
= 0;
93 TDIEntityID
*entitySet
= 0;
94 NTSTATUS status
= tdiGetEntityIDSet( tcpFile
, &entitySet
, &numEntities
);
97 if( !NT_SUCCESS(status
) )
100 for( i
= 0; i
< numEntities
; i
++ ) {
101 if( isIpEntity( tcpFile
, &entitySet
[i
] ) ) {
102 DPRINT("Entity %d is an IP Entity\n", i
);
103 if( numRoutes
== index
) break;
108 if( numRoutes
== index
&& i
< numEntities
) {
109 DPRINT("Index %d is entity #%d - %04x:%08x\n", index
, i
,
110 entitySet
[i
].tei_entity
, entitySet
[i
].tei_instance
);
111 memcpy( ent
, &entitySet
[i
], sizeof(*ent
) );
112 tdiFreeThingSet( entitySet
);
113 return STATUS_SUCCESS
;
115 tdiFreeThingSet( entitySet
);
116 return STATUS_UNSUCCESSFUL
;
120 NTSTATUS tdiGetMibForIpEntity
121 ( HANDLE tcpFile
, TDIEntityID
*ent
, IPSNMPInfo
*entry
) {
122 TCP_REQUEST_QUERY_INFORMATION_EX req
= TCP_REQUEST_QUERY_INFORMATION_INIT
;
123 NTSTATUS status
= STATUS_SUCCESS
;
126 memset( entry
, 0, sizeof( *entry
) );
128 DPRINT("TdiGetMibForIpEntity(tcpFile %x,entityId %x)\n",
129 (DWORD
)tcpFile
, ent
->tei_instance
);
131 req
.ID
.toi_class
= INFO_CLASS_PROTOCOL
;
132 req
.ID
.toi_type
= INFO_TYPE_PROVIDER
;
133 req
.ID
.toi_id
= IP_MIB_STATS_ID
;
134 req
.ID
.toi_entity
= *ent
;
136 status
= DeviceIoControl( tcpFile
,
137 IOCTL_TCP_QUERY_INFORMATION_EX
,
145 DPRINT("TdiGetMibForIpEntity() => {\n"
146 " ipsi_forwarding ............ %d\n"
147 " ipsi_defaultttl ............ %d\n"
148 " ipsi_inreceives ............ %d\n"
149 " ipsi_indelivers ............ %d\n"
150 " ipsi_outrequests ........... %d\n"
151 " ipsi_routingdiscards ....... %d\n"
152 " ipsi_outdiscards ........... %d\n"
153 " ipsi_outnoroutes ........... %d\n"
154 " ipsi_numif ................. %d\n"
155 " ipsi_numaddr ............... %d\n"
156 " ipsi_numroutes ............. %d\n"
158 entry
->ipsi_forwarding
,
159 entry
->ipsi_defaultttl
,
160 entry
->ipsi_inreceives
,
161 entry
->ipsi_indelivers
,
162 entry
->ipsi_outrequests
,
163 entry
->ipsi_routingdiscards
,
164 entry
->ipsi_outdiscards
,
165 entry
->ipsi_outnoroutes
,
168 entry
->ipsi_numroutes
,
174 NTSTATUS tdiGetRoutesForIpEntity
175 ( HANDLE tcpFile
, TDIEntityID
*ent
, IPRouteEntry
**routes
, PDWORD numRoutes
) {
176 NTSTATUS status
= STATUS_SUCCESS
;
178 DPRINT("TdiGetRoutesForIpEntity(tcpFile %x,entityId %x)\n",
179 (DWORD
)tcpFile
, ent
->tei_instance
);
181 status
= tdiGetSetOfThings( tcpFile
,
184 IP_MIB_ROUTETABLE_ENTRY_ID
,
187 sizeof(IPRouteEntry
),
194 NTSTATUS tdiGetIpAddrsForIpEntity
195 ( HANDLE tcpFile
, TDIEntityID
*ent
, IPAddrEntry
**addrs
, PDWORD numAddrs
) {
198 DPRINT("TdiGetIpAddrsForIpEntity(tcpFile %x,entityId %x)\n",
199 (DWORD
)tcpFile
, ent
->tei_instance
);
201 status
= tdiGetSetOfThings( tcpFile
,
204 IP_MIB_ADDRTABLE_ENTRY_ID
,
214 DWORD
getInterfaceStatsByName(const char *name
, PMIB_IFROW entry
)
217 return ERROR_INVALID_PARAMETER
;
219 return ERROR_INVALID_PARAMETER
;
224 DWORD
getInterfaceStatsByIndex(DWORD index
, PMIB_IFROW entry
)
226 return ERROR_INVALID_PARAMETER
;
229 DWORD
getICMPStats(MIB_ICMP
*stats
)
234 return ERROR_INVALID_PARAMETER
;
236 memset(stats
, 0, sizeof(MIB_ICMP
));
237 /* get most of these stats from /proc/net/snmp, no error if can't */
238 fp
= fopen("/proc/net/snmp", "r");
240 const char hdr
[] = "Icmp:";
241 char buf
[512] = { 0 }, *ptr
;
244 ptr
= fgets(buf
, sizeof(buf
), fp
);
245 } while (ptr
&& strncasecmp(buf
, hdr
, sizeof(hdr
) - 1));
247 /* last line was a header, get another */
248 ptr
= fgets(buf
, sizeof(buf
), fp
);
249 if (ptr
&& strncasecmp(buf
, hdr
, sizeof(hdr
) - 1) == 0) {
254 stats
->stats
.icmpInStats
.dwMsgs
= strtoul(ptr
, &endPtr
, 10);
258 stats
->stats
.icmpInStats
.dwErrors
= strtoul(ptr
, &endPtr
, 10);
262 stats
->stats
.icmpInStats
.dwDestUnreachs
= strtoul(ptr
, &endPtr
, 10);
266 stats
->stats
.icmpInStats
.dwTimeExcds
= strtoul(ptr
, &endPtr
, 10);
270 stats
->stats
.icmpInStats
.dwParmProbs
= strtoul(ptr
, &endPtr
, 10);
274 stats
->stats
.icmpInStats
.dwSrcQuenchs
= strtoul(ptr
, &endPtr
, 10);
278 stats
->stats
.icmpInStats
.dwRedirects
= strtoul(ptr
, &endPtr
, 10);
282 stats
->stats
.icmpInStats
.dwEchoReps
= strtoul(ptr
, &endPtr
, 10);
286 stats
->stats
.icmpInStats
.dwTimestamps
= strtoul(ptr
, &endPtr
, 10);
290 stats
->stats
.icmpInStats
.dwTimestampReps
= strtoul(ptr
, &endPtr
, 10);
294 stats
->stats
.icmpInStats
.dwAddrMasks
= strtoul(ptr
, &endPtr
, 10);
298 stats
->stats
.icmpInStats
.dwAddrMaskReps
= strtoul(ptr
, &endPtr
, 10);
302 stats
->stats
.icmpOutStats
.dwMsgs
= strtoul(ptr
, &endPtr
, 10);
306 stats
->stats
.icmpOutStats
.dwErrors
= strtoul(ptr
, &endPtr
, 10);
310 stats
->stats
.icmpOutStats
.dwDestUnreachs
= strtoul(ptr
, &endPtr
, 10);
314 stats
->stats
.icmpOutStats
.dwTimeExcds
= strtoul(ptr
, &endPtr
, 10);
318 stats
->stats
.icmpOutStats
.dwParmProbs
= strtoul(ptr
, &endPtr
, 10);
322 stats
->stats
.icmpOutStats
.dwSrcQuenchs
= strtoul(ptr
, &endPtr
, 10);
326 stats
->stats
.icmpOutStats
.dwRedirects
= strtoul(ptr
, &endPtr
, 10);
330 stats
->stats
.icmpOutStats
.dwEchoReps
= strtoul(ptr
, &endPtr
, 10);
334 stats
->stats
.icmpOutStats
.dwTimestamps
= strtoul(ptr
, &endPtr
, 10);
338 stats
->stats
.icmpOutStats
.dwTimestampReps
= strtoul(ptr
, &endPtr
, 10);
342 stats
->stats
.icmpOutStats
.dwAddrMasks
= strtoul(ptr
, &endPtr
, 10);
346 stats
->stats
.icmpOutStats
.dwAddrMaskReps
= strtoul(ptr
, &endPtr
, 10);
356 DWORD
getIPStats(PMIB_IPSTATS stats
, DWORD family
)
359 return ERROR_INVALID_PARAMETER
;
363 DWORD
getTCPStats(MIB_TCPSTATS
*stats
, DWORD family
)
366 return ERROR_INVALID_PARAMETER
;
370 DWORD
getUDPStats(MIB_UDPSTATS
*stats
, DWORD family
)
373 return ERROR_INVALID_PARAMETER
;
377 static DWORD
getNumWithOneHeader(const char *filename
)
382 DWORD
getNumRoutes(void)
384 DWORD numEntities
, numRoutes
= 0;
385 TDIEntityID
*entitySet
;
392 status
= openTcpFile( &tcpFile
);
394 if( !NT_SUCCESS(status
) ) {
395 TRACE("failure: %08x\n", (int)status
);
399 status
= tdiGetEntityIDSet( tcpFile
, &entitySet
, &numEntities
);
401 if( !NT_SUCCESS(status
) ) {
402 TRACE("failure: %08x\n", (int)status
);
406 for( i
= 0; i
< numEntities
; i
++ ) {
407 if( isIpEntity( tcpFile
, &entitySet
[i
] ) ) {
409 memset( &isnmp
, 0, sizeof( isnmp
) );
410 status
= tdiGetMibForIpEntity( tcpFile
, &entitySet
[i
], &isnmp
);
411 if( !NT_SUCCESS(status
) ) {
412 tdiFreeThingSet( entitySet
);
415 numRoutes
+= isnmp
.ipsi_numroutes
;
419 TRACE("numRoutes: %d\n", (int)numRoutes
);
421 closeTcpFile( tcpFile
);
426 VOID
HexDump( PCHAR Data
, DWORD Len
) {
429 for( i
= 0; i
< Len
; i
++ ) {
431 if( i
) fprintf(stderr
,"\n");
432 fprintf(stderr
,"%08x:", i
);
434 fprintf( stderr
, " %02x", Data
[i
] & 0xff );
436 fprintf(stderr
,"\n");
439 RouteTable
*getRouteTable(void)
441 RouteTable
*out_route_table
;
442 DWORD numRoutes
= getNumRoutes(), routesAdded
= 0;
446 NTSTATUS status
= openTcpFile( &tcpFile
);
449 if( !NT_SUCCESS(status
) )
452 DPRINT("GETTING ROUTE TABLE\n");
454 out_route_table
= HeapAlloc( GetProcessHeap(), 0,
456 (sizeof(RouteEntry
) * (numRoutes
- 1)) );
458 out_route_table
->numRoutes
= numRoutes
;
460 for( i
= 0; routesAdded
< numRoutes
; i
++ ) {
462 IPRouteEntry
*route_set
;
464 getNthIpEntity( tcpFile
, i
, &ent
);
465 tdiGetMibForIpEntity( tcpFile
, &ent
, &snmpInfo
);
467 DPRINT( "%d routes in instance %d\n", snmpInfo
.ipsi_numroutes
, i
);
469 tdiGetRoutesForIpEntity( tcpFile
, &ent
, &route_set
, &numRoutes
);
472 closeTcpFile( tcpFile
);
473 HeapFree( GetProcessHeap(), 0, out_route_table
);
477 DPRINT("Route set returned\n");
480 sizeof( IPRouteEntry
) *
481 snmpInfo
.ipsi_numroutes
);
484 for( j
= 0; j
< snmpInfo
.ipsi_numroutes
; j
++ ) {
485 int routeNum
= j
+ routesAdded
;
486 out_route_table
->routes
[routeNum
].dest
=
487 route_set
[j
].ire_dest
;
488 out_route_table
->routes
[routeNum
].mask
=
489 route_set
[j
].ire_mask
;
490 out_route_table
->routes
[routeNum
].gateway
=
492 out_route_table
->routes
[routeNum
].ifIndex
=
493 route_set
[j
].ire_index
;
494 out_route_table
->routes
[routeNum
].metric
=
495 route_set
[j
].ire_metric1
;
498 if( route_set
) tdiFreeThingSet( route_set
);
500 routesAdded
+= snmpInfo
.ipsi_numroutes
;
503 DPRINT("Return: %08x, %08x\n", status
, out_route_table
);
505 return out_route_table
;
508 DWORD
getNumArpEntries(void)
510 return getNumWithOneHeader("/proc/net/arp");
513 PMIB_IPNETTABLE
getArpTable(void)
518 DWORD
getNumUdpEntries(void)
520 return getNumWithOneHeader("/proc/net/udp");
523 PMIB_UDPTABLE
getUdpTable(void)
525 DWORD numEntries
= getNumUdpEntries();
528 ret
= (PMIB_UDPTABLE
)calloc(1, sizeof(MIB_UDPTABLE
) +
529 (numEntries
- 1) * sizeof(MIB_UDPROW
));
533 /* get from /proc/net/udp, no error if can't */
534 fp
= fopen("/proc/net/udp", "r");
536 char buf
[512] = { 0 }, *ptr
;
538 /* skip header line */
539 ptr
= fgets(buf
, sizeof(buf
), fp
);
540 while (ptr
&& ret
->dwNumEntries
< numEntries
) {
541 ptr
= fgets(buf
, sizeof(buf
), fp
);
546 strtoul(ptr
, &endPtr
, 16); /* skip */
551 ret
->table
[ret
->dwNumEntries
].dwLocalAddr
= strtoul(ptr
, &endPtr
,
557 ret
->table
[ret
->dwNumEntries
].dwLocalPort
= strtoul(ptr
, &endPtr
,
570 DWORD
getNumTcpEntries(void)
572 return getNumWithOneHeader("/proc/net/tcp");
575 PMIB_TCPTABLE
getTcpTable(void)