m_pPropertiesDlg = NULL;

ZeroMemory(&m_SystemTray, sizeof(m_SystemTray));

ResetData();

m_dwStartTime = GetTickCount();

KillTimer(TIMER_ID_WINPROC);

if (m_SystemTray.hWnd != NULL)

Shell_NotifyIcon(NIM_DELETE, &m_SystemTray);

CWnd::OnClose();

if (!snmp.Init())

PostQuitMessage(0);

else {

SetTimer(TIMER_ID_WINPROC, g_nSampleRate, NULL);

// 1.1 init using NIF_ICON

HICON hIcon;

if (g_IconStyle == ICON_HISTOGRAM)

hIcon = (HICON)LoadImage(AfxGetInstanceHandle(), MAKEINTRESOURCE(IDI_HISTOGRAM), IMAGE_ICON, 16, 16, LR_DEFAULTCOLOR);

else

hIcon = (HICON)LoadImage(AfxGetInstanceHandle(), MAKEINTRESOURCE(IDI_BARGRAPH), IMAGE_ICON, 16, 16, LR_DEFAULTCOLOR);

m_SystemTray.cbSize = sizeof(NOTIFYICONDATA);

m_SystemTray.hWnd = GetSafeHwnd();

m_SystemTray.uID = 1;

m_SystemTray.hIcon = hIcon;

m_SystemTray.uFlags = NIF_MESSAGE | NIF_ICON;

m_SystemTray.uCallbackMessage = TaskbarCallbackMsg;

if (!Shell_NotifyIcon(NIM_ADD, &m_SystemTray))

AfxMessageBox("System tray error.");

}

ASSERT(g_nAveragingWindow <= MAX_SAMPLES);

// Write first few fields into array head element

pStats[0].Bps = dwBps;

pStats[0].total = dbTotal;

pStats[0].time = dwTime; // Current time interval (ms)

// The index in the array for calculating averages

int start = g_nAveragingWindow;

while (start > 0 && (start >= MAX_SAMPLES || pStats[start].total == 0))

start--;

// Total bytes received/sent in our sampling window

double dbSampleTotal = dbTotal - pStats[start].total;

// Elapsed time (ms)

DWORD dwElapsed = pStats[0].time - pStats[start].time;

// Calculate average

pStats[0].ave = dwElapsed > 0 ? (DWORD)(dbSampleTotal * 1000.0 / dwElapsed + 0.5) : 0;

ASSERT(num <= MAX_SAMPLES);

DWORD result = 1; // To prevent division by zero when using the maximum for scaling

if (type == 0) {

for (int i = 0; i < num; i++)

result = max(stats[i].Bps, result);

} else if (type == 1) {

for (int i = 0; i < num; i++)

result = max(stats[i].ave, result);

} else

ASSERT(false);

return result;

// Retrieve the total number of bytes received and sent by all network adapters, modulo 2^32.

// Each network adapter counts from its own epoch, which is the last time the cable or wireless LAN was connected.

DWORD r = 0;

DWORD s = 0;

bool ok = snmp.GetReceivedAndSentOctets(r, s);

DWORD curRecv = 0;

DWORD curSent = 0;

if (ok) {

if (m_PrevBytesOk) {

curRecv = r - m_PrevBytesRecv; // With 32-bit unsigned wraparound

curSent = s - m_PrevBytesSent;

}

m_PrevBytesRecv = r;

m_PrevBytesSent = s;

}

m_TotalBytesRecv += curRecv;

m_TotalBytesSent += curSent;

m_PrevBytesOk = ok;

// Don't depend upon timing of WM_TIMER messages. Get the true number of milliseconds elapsed

DWORD dwTime = GetTickCount();

DWORD elapsed = dwTime - m_dwStartTime;

m_dwStartTime = dwTime;

// Calculate bits per second

DWORD dwRecv_bps = 0;

DWORD dwSent_bps = 0;

if (elapsed > 0) {

dwRecv_bps = (DWORD)(curRecv * 1000.0 / elapsed + 0.5);

dwSent_bps = (DWORD)(curSent * 1000.0 / elapsed + 0.5);

}

// Shift over previous data

for (int i = MAX_SAMPLES - 1; i >= 1; i--) {

RecvStats[i] = RecvStats[i - 1];

SentStats[i] = SentStats[i - 1];

}

// Calc the average bps and add new entry to array

CalcAverages(m_TotalBytesRecv, dwTime, dwRecv_bps, RecvStats);

CalcAverages(m_TotalBytesSent, dwTime, dwSent_bps, SentStats);

// Get the icon for the system tray

HICON hIcon = theApp.m_Icons.GetIcon(RecvStats, SentStats, g_IconStyle);

UpdateTrayIcon(hIcon);

DestroyIcon(hIcon);

if (m_pPropertiesDlg != NULL)

m_pPropertiesDlg->SetForegroundWindow();

else {

// Fake out MFC in order to receive the "minimize all windows" syscommand message

// The window is restored in initdialog

theApp.m_pMainWnd = NULL;

m_pPropertiesDlg = new DlgPropSheet(SZ_APPNAME, NULL);

m_pPropertiesDlg->m_psh.dwFlags |= PSH_NOAPPLYNOW | PSH_MODELESS;

if (m_pPropertiesDlg->DoModal() == IDOK)

SaveSettings();

else

ReadSettings();

delete m_pPropertiesDlg;

m_pPropertiesDlg = NULL;

SetTimer(TIMER_ID_WINPROC, g_nSampleRate, NULL);

}

UNREFERENCED_PARAMETER(wParam);

switch (lParam) {

case WM_MOUSEMOVE:

{

CString s, sRecvBPS, sRecvAVE;

FormatBytes(RecvStats[0].Bps, sRecvBPS, true);

FormatBytes(RecvStats[0].ave, sRecvAVE, true);

s.Format("Current: %s Average: %s", sRecvBPS, sRecvAVE);

m_SystemTray.cbSize = sizeof(NOTIFYICONDATA);

m_SystemTray.hWnd = GetSafeHwnd();

m_SystemTray.uID = 1;

m_SystemTray.uFlags = NIF_TIP;

strcpy_s(m_SystemTray.szTip, s);

Shell_NotifyIcon(NIM_MODIFY, &m_SystemTray);

}

break;

case WM_LBUTTONDBLCLK:

ShowPropertiesDlg();

break;

case WM_RBUTTONUP:

{

CMenu menu;

POINT pt;

GetCursorPos(&pt);

menu.LoadMenu(IDR_MENU1);

menu.SetDefaultItem(0, TRUE);

CMenu &pMenu = *menu.GetSubMenu(0);

pMenu.SetDefaultItem(0, TRUE);

// See Q135788 "PRB: Menus for Notification Icons Do Not Work Correctly"

SetForegroundWindow();

int cmd = pMenu.TrackPopupMenu(TPM_RETURNCMD | TPM_LEFTALIGN | TPM_NONOTIFY, pt.x, pt.y, this);

PostMessage(WM_NULL, 0, 0);

if (cmd == IDCLOSE) {

// Save any settings if the user closes the tray icon while the dialog is open

if (m_pPropertiesDlg != NULL) {

SaveSettings();

m_pPropertiesDlg->SendMessage(WM_CLOSE);

}

theApp.m_wnd.PostMessage(WM_CLOSE);

} else if (cmd == ID_PROPERTIES)

ShowPropertiesDlg();

}

break;

}

return 0;

ASSERT(hIcon != 0);

if (m_SystemTray.hWnd != NULL && hIcon != NULL) {

m_SystemTray.cbSize = sizeof(NOTIFYICONDATA);

m_SystemTray.hWnd = GetSafeHwnd();

m_SystemTray.uID = 1;

m_SystemTray.hIcon = hIcon;

m_SystemTray.uFlags = NIF_ICON;

m_SystemTray.uCallbackMessage = TaskbarCallbackMsg;

Shell_NotifyIcon(NIM_MODIFY, &m_SystemTray);

}

ZeroMemory(RecvStats, sizeof(RecvStats));

ZeroMemory(SentStats, sizeof(SentStats));

m_PrevBytesOk = false;

m_TotalBytesRecv = 0;

m_TotalBytesSent = 0;

m_PrevBytesRecv = 0;

m_PrevBytesSent = 0;

if (m_pPropertiesDlg != NULL) {

switch (m_pPropertiesDlg->GetActiveIndex()) {

case 0: CWnd::WinHelp(IDH_connection_tab); return;

case 1: CWnd::WinHelp(IDH_options_tab ); return;

case 2: CWnd::WinHelp(IDH_colors_tab ); return;

}

}

CWnd::WinHelp(0, HELP_CONTENTS);