Network Traffic Calculator using VC++ 6 on XP

An Introduction
This is a simple VC++ (made in 6) program that calculates the total number of bytes transfered in a network connection. Made in and tested for Windows XP.

main.cpp

/*
Network Traffic Calculator
(c) digitalpbk.blogspot.com
*/

#include <windows.h>
#include <wingdi.h>
#include <protocol.h>
#include <stdio.h>

#define REGULAR 0
#define BOLD 1
#define ITALIC 2
#define ULINE 3

void getInterfaceList(char list[][17],int max);
int ConnectSock(char sck[],int port,HWND hwnd);
void processData(char[]);

HWND txtW;

HFONT GetAFont(int fnFont);

typedef struct _DATA // our datatransfer structure.
{
 int mb;
 int kb;
 int bytes;
}BITDATA;

BITDATA recd,sent,total;
SOCKET gSokt;

BITDATA Add(BITDATA b1,BITDATA b2) // adds bytes to our data structure
{
 BITDATA ret;
 ret.bytes = b1.bytes + b2.bytes;
 ret.kb = b1.kb + b2.kb;
 ret.mb = b1.mb + b2.mb;
 
 if(ret.bytes >= 1024)
 {
  ret.bytes = 0;
  ret.kb ++;
 }
 if(ret.kb >=1024)
 {
  ret.mb ++;
  ret.kb =0;
 }
 return ret;
}
BITDATA Add(BITDATA b1,int bytes)
{
 BITDATA ret=b1;
 ret.bytes += bytes;
 if(ret.bytes >= 1024)
 {
  ret.bytes = 0;
  ret.kb ++;
 }
 if(ret.kb >=1024)
 {
  ret.mb ++;
  ret.kb =0;
 }
 return ret;
}

long FAR PASCAL WndProc( HWND hwnd,UINT msg,UINT wParam, LONG lParam )
{
 PAINTSTRUCT ps;
 RECT rc;
 HDC hdc;
 

 int x;
 static si=48;
 si++;
 char bfr[60]={si,0},num[6];

 long l;

 switch(msg)
 {

 case 0x401: // network event detected.
  if(lParam == FD_READ)
  {
   x=0;
   char *bufr=new char[10000];
   do
   {
    x=recv(wParam,bufr,10000,0);
    processData(bufr);
   }while(x==10000);
   delete bufr;
  }
 case WM_PAINT:

  hdc=BeginPaint(hwnd,&ps);

  strcpy(bfr ,"Data S/R: ");
  itoa(total.mb,num,10);
  strcat(bfr,num);
  strcat(bfr,".");
  
  itoa(total.kb,num,10);
  strcat(bfr,num);
  strcat(bfr,".");

  itoa(total.bytes,num,10);
  strcat(bfr,num);

  l=strlen(bfr);
  SetWindowText(txtW,bfr);

  EndPaint(hwnd,&ps);
  return 0;
 case WM_DESTROY:
  WSAAsyncSelect(gSokt,hwnd,0x401,0);
  closesocket(gSokt);
  if(WSAIsBlocking())WSACancelBlockingCall();
  WSACleanup();

  PostQuitMessage(0);
  return 0;
 }
 return DefWindowProc(hwnd,msg,wParam,lParam);
}

int WINAPI WinMain(HINSTANCE hInstance,HINSTANCE hPrevInstance,LPSTR  lpszCmdParam,int nCmdShow) //Windows Main
{
 /*standard win32 application init*/
 static char szAppName[] = "Traffic Analyzer";
 HWND        hwnd;
 MSG   msg;
 WNDCLASS    wc;

 if(!hPrevInstance)
 {
  wc.style = CS_HREDRAW | CS_VREDRAW;
  wc.lpfnWndProc = WndProc;
  wc.cbClsExtra  = 0;
  wc.cbWndExtra  = 0;
  wc.hInstance = hInstance;
  wc.hIcon = LoadIcon(NULL,IDI_WARNING);
  wc.hCursor = LoadCursor(NULL,IDC_ARROW);
  wc.hbrBackground = (HBRUSH)GetStockObject(BLACK_BRUSH);
  wc.lpszMenuName = NULL;
  wc.lpszClassName = szAppName;
  RegisterClass(&amb;wc);
 }
 /*gets network addresses*/
 char localaddr[5][17];
 if(getInterfaceList(localaddr,5))
  return 1;
 char szTitle[100]="Traffic Calc on ";
 strcat(szTitle,localaddr[0]);
 /*window creation
 hwnd = CreateWindowEx(WS_EX_TOOLWINDOW|WS_EX_TOPMOST,szAppName,szTitle,WS_SYSMENU,CW_USEDEFAULT,CW_USEDEFAULT,240,40,NULL,NULL,hInstance,NULL);
 
 HWND hwndTxtOut=CreateWindow("EDIT","Usage Stats loading....",WS_CHILD|WS_VISIBLE,0,0,240,20,hwnd,NULL,hInstance ,NULL);
 txtW = hwndTxtOut;
 
 gSokt = ConnectSock(localaddr[0],7000,hwnd);

 ShowWindow(hwnd,nCmdShow);
 UpdateWindow( hwnd );

 /*Window Procedure Loop
 while(GetMessage(&msg,NULL,0,0))
 {
  TranslateMessage(&msg);
  DispatchMessage(&msg);
 }

 return msg.wParam;
}

void processData(char data[])
{
 IP_HEADER iphead;
 memcpy(&iphead,data,sizeof(iphead)); //gets the Length of each packet
 total = Add(total,ntohs(iphead.length)); // adds with previous results
}

winsocks.cpp

/*winsocks.cpp
(c) digitalpbk.blogspot.com
*/

#include <winsock2.h>
#include <WS2TCPIP.H>

int getInterfaceList(char list[][17],int max) // gets all network connections
{
 WORD wVs = MAKEWORD(2,2);
 WSADATA wsaData;
 

 int err=WSAStartup(wVs,&wsaData);
 if(err!=0)
  return err;
  
 long lnRes;
 SOCKET Sock;
 unsigned long lnBytesReturned;
 INTERFACE_INFO buffer[7];
 //start up winsock
 
 Sock = socket(AF_INET,SOCK_STREAM,IPPROTO_IP);
 
 lnRes = WSAIoctl(Sock, SIO_GET_INTERFACE_LIST,0, 0,&buffer, 1024, &lnBytesReturned,0, 0);
 
 if(lnRes)
  return WSAGetLastError();

 for(int i=0u;i<lnBytesReturned/76;i++)
 {
  if(i<max)strcpy(list[i],inet_ntoa(buffer[i].iiAddress.AddressIn.sin_addr));
 }
 closesocket(Sock);
 return 0;
}
int ConnectSock(char sck[],int port,HWND hwnd)
{
 WORD wVs = MAKEWORD(2,2);
 WSADATA wsaData;

 int err=WSAStartup(wVs,&wsaData);
 if(err!=0)
  return INVALID_SOCKET;
 
 SOCKADDR_IN sin;
 
 sin.sin_family = AF_INET;
 sin.sin_port = htons(port);
 
 if(sin.sin_port == INVALID_SOCKET)
  return INVALID_SOCKET;
 unsigned long iret = inet_addr(sck);
 memcpy(&sin.sin_addr,&iret,4);

 SOCKET s=socket(AF_INET,SOCK_RAW,IPPROTO_IP);
 if(s<0)return INVALID_SOCKET;
 long TIMEO = 5000;
 err = setsockopt(s,SOL_SOCKET,SO_RCVTIMEO,(char *)&TIMEO,4);
 if(err)
 {
  if(s>0)closesocket(s);
  return INVALID_SOCKET;
 }
 err = bind(s,(sockaddr *)&sin,sizeof(sin));
 if(err)
 {
  if(s>0)closesocket(s);
  return INVALID_SOCKET;
 }
 unsigned long retd;
 long lnInpBufr=1,lnOutBufr=0;

 err = WSAIoctl(s,0x98000001,&lnInpBufr,sizeof(long),&lnOutBufr,sizeof(long),&retd,NULL,NULL);
 if(err)
 {
  if(s>0)closesocket(s);
  return INVALID_SOCKET;
 }
 err = WSAAsyncSelect(s,hwnd,0x401,FD_READ);
 if(err)
 {
  if(s>0)closesocket(s);
  return INVALID_SOCKET;
 }
 return s;
}

Dont know what to do with the code?
Take VC++ 6,
File > New, In Projects tab, new Win32 Application, give a name also like "trafficcalc",
Choose empty project.
Next File > New, In Files tab, take C++ source file, give filename "main.cpp"
repeat the step for filename "winsocks.cpp"
Copy the sources to the files.
Save and Execute!

Download the full source with Workspace files from planet-source-code.com

Anything else?
leave a comment...