iphlpapi: Simplify parsing of UDP stats. Only try to open /proc on Linux.
This commit is contained in:
parent
3a60d69b47
commit
4c7216fc77
|
@ -612,23 +612,48 @@ DWORD WINAPI GetTcpStatistics(PMIB_TCPSTATS stats)
|
|||
*/
|
||||
DWORD WINAPI GetUdpStatistics(PMIB_UDPSTATS stats)
|
||||
{
|
||||
#if defined(HAVE_SYS_SYSCTL_H) && defined(UDPCTL_STATS)
|
||||
DWORD ret = ERROR_NOT_SUPPORTED;
|
||||
|
||||
if (!stats) return ERROR_INVALID_PARAMETER;
|
||||
memset( stats, 0, sizeof(*stats) );
|
||||
|
||||
#ifdef __linux__
|
||||
{
|
||||
FILE *fp;
|
||||
|
||||
if ((fp = fopen("/proc/net/snmp", "r")))
|
||||
{
|
||||
static const char hdr[] = "Udp:";
|
||||
char buf[512], *ptr;
|
||||
|
||||
while ((ptr = fgets(buf, sizeof(buf), fp)))
|
||||
{
|
||||
if (strncasecmp(buf, hdr, sizeof(hdr) - 1)) continue;
|
||||
/* last line was a header, get another */
|
||||
if (!(ptr = fgets(buf, sizeof(buf), fp))) break;
|
||||
if (!strncasecmp(buf, hdr, sizeof(hdr) - 1))
|
||||
{
|
||||
ptr += sizeof(hdr);
|
||||
sscanf( ptr, "%u %u %u %u %u",
|
||||
&stats->dwInDatagrams, &stats->dwNoPorts,
|
||||
&stats->dwInErrors, &stats->dwOutDatagrams, &stats->dwNumAddrs );
|
||||
break;
|
||||
}
|
||||
}
|
||||
fclose(fp);
|
||||
ret = NO_ERROR;
|
||||
}
|
||||
}
|
||||
#elif defined(HAVE_SYS_SYSCTL_H) && defined(UDPCTL_STATS)
|
||||
{
|
||||
int mib[] = {CTL_NET, PF_INET, IPPROTO_UDP, UDPCTL_STATS};
|
||||
#define MIB_LEN (sizeof(mib) / sizeof(mib[0]))
|
||||
struct udpstat udp_stat;
|
||||
MIB_UDPTABLE *udp_table;
|
||||
size_t needed;
|
||||
if (!stats)
|
||||
return ERROR_INVALID_PARAMETER;
|
||||
size_t needed = sizeof(udp_stat);
|
||||
|
||||
needed = sizeof(udp_stat);
|
||||
|
||||
if(sysctl(mib, MIB_LEN, &udp_stat, &needed, NULL, 0) == -1)
|
||||
if(sysctl(mib, MIB_LEN, &udp_stat, &needed, NULL, 0) != -1)
|
||||
{
|
||||
ERR ("failed to get udpstat\n");
|
||||
return ERROR_NOT_SUPPORTED;
|
||||
}
|
||||
|
||||
stats->dwInDatagrams = udp_stat.udps_ipackets;
|
||||
stats->dwOutDatagrams = udp_stat.udps_opackets;
|
||||
stats->dwNoPorts = udp_stat.udps_noport;
|
||||
|
@ -638,66 +663,14 @@ DWORD WINAPI GetUdpStatistics(PMIB_UDPSTATS stats)
|
|||
stats->dwNumAddrs = udp_table->dwNumEntries;
|
||||
HeapFree( GetProcessHeap(), 0, udp_table );
|
||||
}
|
||||
else stats->dwNumAddrs = 0;
|
||||
|
||||
return NO_ERROR;
|
||||
ret = NO_ERROR;
|
||||
}
|
||||
else ERR ("failed to get udpstat\n");
|
||||
}
|
||||
#else
|
||||
FILE *fp;
|
||||
|
||||
if (!stats)
|
||||
return ERROR_INVALID_PARAMETER;
|
||||
|
||||
memset(stats, 0, sizeof(MIB_UDPSTATS));
|
||||
|
||||
/* get from /proc/net/snmp, no error if can't */
|
||||
fp = fopen("/proc/net/snmp", "r");
|
||||
if (fp) {
|
||||
static const char hdr[] = "Udp:";
|
||||
char buf[512] = { 0 }, *ptr;
|
||||
|
||||
|
||||
do {
|
||||
ptr = fgets(buf, sizeof(buf), fp);
|
||||
} while (ptr && strncasecmp(buf, hdr, sizeof(hdr) - 1));
|
||||
if (ptr) {
|
||||
/* last line was a header, get another */
|
||||
ptr = fgets(buf, sizeof(buf), fp);
|
||||
if (ptr && strncasecmp(buf, hdr, sizeof(hdr) - 1) == 0) {
|
||||
char *endPtr;
|
||||
|
||||
ptr += sizeof(hdr);
|
||||
if (ptr && *ptr) {
|
||||
stats->dwInDatagrams = strtoul(ptr, &endPtr, 10);
|
||||
ptr = endPtr;
|
||||
}
|
||||
if (ptr && *ptr) {
|
||||
stats->dwNoPorts = strtoul(ptr, &endPtr, 10);
|
||||
ptr = endPtr;
|
||||
}
|
||||
if (ptr && *ptr) {
|
||||
stats->dwInErrors = strtoul(ptr, &endPtr, 10);
|
||||
ptr = endPtr;
|
||||
}
|
||||
if (ptr && *ptr) {
|
||||
stats->dwOutDatagrams = strtoul(ptr, &endPtr, 10);
|
||||
ptr = endPtr;
|
||||
}
|
||||
if (ptr && *ptr) {
|
||||
stats->dwNumAddrs = strtoul(ptr, &endPtr, 10);
|
||||
ptr = endPtr;
|
||||
}
|
||||
}
|
||||
}
|
||||
fclose(fp);
|
||||
}
|
||||
else
|
||||
{
|
||||
ERR ("unimplemented!\n");
|
||||
return ERROR_NOT_SUPPORTED;
|
||||
}
|
||||
|
||||
return NO_ERROR;
|
||||
FIXME( "unimplemented\n" );
|
||||
#endif
|
||||
return ret;
|
||||
}
|
||||
|
||||
|
||||
|
|
Loading…
Reference in New Issue