www.pudn.com > NetTraffic.rar > MFNetTraffic.cpp
/*******************************************
MFTrafficButton
Version: 1.0
Date: 31.10.2001
Author: Michael Fatzi
Mail: Michael_Fatzi@hotmail.com
Copyright 1996-1997, Keith Rule
You may freely use or modify this code provided this
Copyright is included in all derived versions.
History: 10.2001 Startup
Handy little button control to display current
nettraffic as graph in a button.
********************************************/
// MFNetTraffic.cpp: implementation of the MFNetTraffic class.
//
//////////////////////////////////////////////////////////////////////
#include "stdafx.h"
#include "MFNetTraffic.h"
#include "float.h"
#ifdef _DEBUG
#undef THIS_FILE
static char THIS_FILE[]=__FILE__;
#define new DEBUG_NEW
#endif
#include "winperf.h"
//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////
MFNetTraffic::MFNetTraffic()
{
lasttraffic = 0.0;
CurrentInterface = -1;
CurrentTrafficType = AllTraffic;
GetInterfaces();
}
MFNetTraffic::~MFNetTraffic()
{
}
// Little helper functions
// Found them on CodeGuru, but do not know who has written them originally
static PERF_OBJECT_TYPE *FirstObject(PERF_DATA_BLOCK *dataBlock)
{
return (PERF_OBJECT_TYPE *) ((BYTE *)dataBlock + dataBlock->HeaderLength);
}
static PERF_OBJECT_TYPE *NextObject(PERF_OBJECT_TYPE *act)
{
return (PERF_OBJECT_TYPE *) ((BYTE *)act + act->TotalByteLength);
}
static PERF_COUNTER_DEFINITION *FirstCounter(PERF_OBJECT_TYPE *perfObject)
{
return (PERF_COUNTER_DEFINITION *) ((BYTE *) perfObject + perfObject->HeaderLength);
}
static PERF_COUNTER_DEFINITION *NextCounter(PERF_COUNTER_DEFINITION *perfCounter)
{
return (PERF_COUNTER_DEFINITION *) ((BYTE *) perfCounter + perfCounter->ByteLength);
}
static PERF_COUNTER_BLOCK *GetCounterBlock(PERF_INSTANCE_DEFINITION *pInstance)
{
return (PERF_COUNTER_BLOCK *) ((BYTE *)pInstance + pInstance->ByteLength);
}
static PERF_INSTANCE_DEFINITION *FirstInstance (PERF_OBJECT_TYPE *pObject)
{
return (PERF_INSTANCE_DEFINITION *) ((BYTE *) pObject + pObject->DefinitionLength);
}
static PERF_INSTANCE_DEFINITION *NextInstance (PERF_INSTANCE_DEFINITION *pInstance)
{
// next instance is after
// this instance + this instances counter data
PERF_COUNTER_BLOCK *pCtrBlk = GetCounterBlock(pInstance);
return (PERF_INSTANCE_DEFINITION *) ((BYTE *)pInstance + pInstance->ByteLength + pCtrBlk->ByteLength);
}
static char *WideToMulti(wchar_t *source, char *dest, int size)
{
int nLen = WideCharToMultiByte(CP_ACP, 0, source, -1, dest, size, 0, 0);
dest[nLen] = '\0';
return dest;
}
/*
returns the traffic of given interface
*/
double MFNetTraffic::GetTraffic(int interfaceNumber)
{
try
{
#define DEFAULT_BUFFER_SIZE 40960L
POSITION pos;
CString InterfaceName;
pos = Interfaces.FindIndex(interfaceNumber);
if(pos==NULL)
return 0.0;
InterfaceName = Interfaces.GetAt(pos);
// buffer for performance data
unsigned char *data = new unsigned char [DEFAULT_BUFFER_SIZE];
// return value from RegQueryValueEx: ignored for this application
DWORD type;
// Buffer size
DWORD size = DEFAULT_BUFFER_SIZE;
// return value of RegQueryValueEx
DWORD ret;
// request performance data from network object (index 510)
while((ret = RegQueryValueEx(HKEY_PERFORMANCE_DATA, "510", 0, &type, data, &size)) != ERROR_SUCCESS) {
if(ret == ERROR_MORE_DATA)
{
// buffer size was too small, increase allocation size
size += DEFAULT_BUFFER_SIZE;
delete [] data;
data = new unsigned char [size];
}
else
{
// some unspecified error has occured
return 1;
}
}
PERF_DATA_BLOCK *dataBlockPtr = (PERF_DATA_BLOCK *)data;
// enumerate first object of list
PERF_OBJECT_TYPE *objectPtr = FirstObject(dataBlockPtr);
// trespassing the list
for(int a=0 ; a<(int)dataBlockPtr->NumObjectTypes ; a++)
{
char nameBuffer[255];
// did we receive a network object?
if(objectPtr->ObjectNameTitleIndex == 510)
{
// Calculate the offset
DWORD processIdOffset = ULONG_MAX;
// find first counter
PERF_COUNTER_DEFINITION *counterPtr = FirstCounter(objectPtr);
// walking trough the list of objects
for(int b=0 ; b<(int)objectPtr->NumCounters ; b++)
{
// Check if we received datatype wished
if((int)counterPtr->CounterNameTitleIndex == CurrentTrafficType)
{
processIdOffset = counterPtr->CounterOffset;
break;
}
// watch next counter
counterPtr = NextCounter(counterPtr);
}
if(processIdOffset == ULONG_MAX) {
delete [] data;
return 1;
}
// Find first instance
PERF_INSTANCE_DEFINITION *instancePtr = FirstInstance(objectPtr);
DWORD fullTraffic;
DWORD traffic;
for(b=0 ; bNumInstances ; b++)
{
// evaluate pointer to name
wchar_t *namePtr = (wchar_t *) ((BYTE *)instancePtr + instancePtr->NameOffset);
// get PERF_COUNTER_BLOCK of this instance
PERF_COUNTER_BLOCK *counterBlockPtr = GetCounterBlock(instancePtr);
// now we have the interface name
char *pName = WideToMulti(namePtr, nameBuffer, sizeof(nameBuffer));
CString iName("");
iName.Format("%s",pName);
if( iName == "" ) continue;
POSITION pos = TotalTraffics.FindIndex(b);
if(pos!=NULL)
{
fullTraffic = *((DWORD *) ((BYTE *)counterBlockPtr + processIdOffset));
TotalTraffics.SetAt(pos,fullTraffic);
}
// If the interface the currently selected interface?
if(InterfaceName == iName)
{
traffic = *((DWORD *) ((BYTE *)counterBlockPtr + processIdOffset));
double acttraffic = (double)traffic;
double trafficdelta;
// Do we handle a new interface (e.g. due a change of the interface number
if(CurrentInterface != interfaceNumber)
{
lasttraffic = acttraffic;
trafficdelta = 0.0;
CurrentInterface = interfaceNumber;
}
else
{
trafficdelta = acttraffic - lasttraffic;
lasttraffic = acttraffic;
}
delete [] data;
return(trafficdelta);
}
// next instance
instancePtr = NextInstance(instancePtr);
}
}
// next object in list
objectPtr = NextObject(objectPtr);
}
delete [] data;
return 0;
}
catch(...)
{
return 0;
}
}
/*
Enumerate installed interfaces.
See comments above
*/
BOOL MFNetTraffic::GetInterfaces()
{
try
{
#define DEFAULT_BUFFER_SIZE 40960L
Interfaces.RemoveAll();
unsigned char *data = (unsigned char*)malloc(DEFAULT_BUFFER_SIZE);
DWORD type;
DWORD size = DEFAULT_BUFFER_SIZE;
DWORD ret;
char s_key[4096];
sprintf( s_key , "%d" , 510 );
while((ret = RegQueryValueEx(HKEY_PERFORMANCE_DATA, s_key, 0, &type, data, &size)) != ERROR_SUCCESS) {
while(ret == ERROR_MORE_DATA)
{
size += DEFAULT_BUFFER_SIZE;
data = (unsigned char*) realloc(data, size);
}
if(ret != ERROR_SUCCESS)
{
return FALSE;
}
}
PERF_DATA_BLOCK *dataBlockPtr = (PERF_DATA_BLOCK *)data;
PERF_OBJECT_TYPE *objectPtr = FirstObject(dataBlockPtr);
for(int a=0 ; a<(int)dataBlockPtr->NumObjectTypes ; a++)
{
char nameBuffer[255];
if(objectPtr->ObjectNameTitleIndex == 510)
{
DWORD processIdOffset = ULONG_MAX;
PERF_COUNTER_DEFINITION *counterPtr = FirstCounter(objectPtr);
for(int b=0 ; b<(int)objectPtr->NumCounters ; b++)
{
if(counterPtr->CounterNameTitleIndex == 520)
processIdOffset = counterPtr->CounterOffset;
counterPtr = NextCounter(counterPtr);
}
if(processIdOffset == ULONG_MAX) {
free(data);
return 1;
}
PERF_INSTANCE_DEFINITION *instancePtr = FirstInstance(objectPtr);
for(b=0 ; bNumInstances ; b++)
{
wchar_t *namePtr = (wchar_t *) ((BYTE *)instancePtr + instancePtr->NameOffset);
PERF_COUNTER_BLOCK *counterBlockPtr = GetCounterBlock(instancePtr);
char *pName = WideToMulti(namePtr, nameBuffer, sizeof(nameBuffer));
if( strcmp(pName, "") == 0 ) continue;
DWORD bandwith = *((DWORD *) ((BYTE *)counterBlockPtr + processIdOffset));
DWORD tottraff = 0;
Interfaces.AddTail(CString(pName));
Bandwidths.AddTail(bandwith);
TotalTraffics.AddTail(tottraff); // initial 0, just for creating the list
instancePtr = NextInstance(instancePtr);
}
}
objectPtr = NextObject(objectPtr);
}
free(data);
return TRUE;
}
catch(...)
{
return FALSE;
}
}
/*
Returns the count of installed interfaces
*/
int MFNetTraffic::GetNetworkInterfacesCount()
{
return Interfaces.GetCount()-1;
}
/*
Returns the name of the given interface (-number)
*/
BOOL MFNetTraffic::GetNetworkInterfaceName(CString *InterfaceName, int index)
{
POSITION pos = Interfaces.FindIndex(index);
if(pos==NULL)
return FALSE;
InterfaceName->Format("%s",Interfaces.GetAt(pos));
return TRUE;
}
/*
Returns bandwith of interface e.g. 100000 for 100MBit
*/
DWORD MFNetTraffic::GetInterfaceBandwidth(int index)
{
POSITION pos = Bandwidths.FindIndex(index);
if(pos==NULL)
return 0;
else
return Bandwidths.GetAt(pos) / 8;
}
/*
Sometime it is nice to know, how much traffic has a specific interface sent and received
*/
DWORD MFNetTraffic::GetInterfaceTotalTraffic(int index)
{
DWORD totaltraffic = 0;
POSITION pos;
pos= TotalTraffics.FindIndex(index);
if(pos!=NULL)
{
totaltraffic = TotalTraffics.GetAt(pos);
if(totaltraffic == 0.0)
{
GetTraffic(index);
pos= TotalTraffics.FindIndex(index);
if(pos!=NULL)
{
totaltraffic = TotalTraffics.GetAt(pos);
}
}
}
return(totaltraffic);
}
/*
To prevent direct manipulation of member variables....
*/
void MFNetTraffic::SetTrafficType(int trafficType)
{
CurrentTrafficType = trafficType;
}