Windows網絡通信流量和網速監控設計(二)

第一篇文章鏈接

結果視圖:

在這裏插入圖片描述
.h代碼:

#pragma once
#include "afxwin.h"


#define WMMY_NETWORKER_MESSAGE WM_USER+100


// CMyNetCheckSystemDlg 對話框
class CMyNetCheckSystemDlg : public CDialogEx
{
// 構造
public:
	CMyNetCheckSystemDlg(CWnd* pParent = NULL);	// 標準構造函數

// 對話框數據
	enum { IDD = IDD_MYNETCHECKSYSTEM_DIALOG };

	protected:
	virtual void DoDataExchange(CDataExchange* pDX);	// DDX/DDV 支持


// 實現
protected:
	HICON m_hIcon;

	afx_msg LRESULT MYMSGNETWORKER(WPARAM wParam, LPARAM lParam);

	// 生成的消息映射函數
	virtual BOOL OnInitDialog();
	afx_msg void OnPaint();
	afx_msg HCURSOR OnQueryDragIcon();
	DECLARE_MESSAGE_MAP()
public:
	afx_msg void OnBnClickedButton1();
	afx_msg void OnBnClickedButton5();
	CStatic m_downspeed;
	CStatic m_upspeed;
	afx_msg void OnTimer(UINT_PTR nIDEvent);
};

.cpp代碼:


// MyNetCheckSystemDlg.cpp : 實現文件
//

#include<strsafe.h>
#include<stdio.h>
#include<Windows.h>
#include<objbase.h>

#include<IPHlpApi.h>
#pragma comment(lib,"Iphlpapi.lib")

#include<SetupAPI.h>
#pragma comment(lib,"Setupapi.lib")

unsigned int _stdcall ThreadCheckSpeed(LPVOID lpArgs)
{
	CMyNetCheckSystemDlg *pDlg = (CMyNetCheckSystemDlg*)lpArgs;
	PMIB_IFTABLE m_pTable = NULL;

	DWORD m_dwAdapters = 0;
	ULONG uRetCode = GetIfTable(m_pTable,&m_dwAdapters,TRUE);

	if (uRetCode == ERROR_NOT_SUPPORTED)
	{
		//return (-1);
		exit(-1);
	}

	if (uRetCode == ERROR_INSUFFICIENT_BUFFER)
	{
		m_pTable = (PMIB_IFTABLE)new BYTE[65535];				//假設端口數不超過65535個
	}

	DWORD	dwLastIn = 0;						//上一秒鐘的接收字節數
	DWORD	dwLastOut = 0;						//上一秒鐘的發送字節數
	DWORD	dwBandIn = 0;						//下載速度
	DWORD	dwBandOut = 0;						//上傳速度

	while (1)
	{
		GetIfTable(m_pTable, &m_dwAdapters, TRUE);

		DWORD	dwInOctets = 0;
		DWORD	dwOutOctets = 0;

		//將所有端口的流量進行統計
		for (UINT i = 0; i < m_pTable->dwNumEntries; i++)
		{
			MIB_IFROW	Row = m_pTable->table[i];
			dwInOctets += Row.dwInOctets;
			dwOutOctets += Row.dwOutOctets;
		}

		dwBandIn = dwInOctets - dwLastIn;														//下載速度
		dwBandOut = dwOutOctets - dwLastOut;											//上牀速速
		if (dwLastIn <= 0)
		{
			dwBandIn = 0;
		}
		else
		{
			dwBandIn = dwBandIn / 1024;															//b轉換成kb
		}


		if (dwLastOut <= 0)
		{
			dwBandOut = 0;
		}
		else
		{
			dwBandOut = dwBandOut / 1024;													//b轉換成kb
		}

		dwLastIn = dwInOctets;
		dwLastOut = dwOutOctets;

		printf("收到字節: %u bytes\n", dwLastIn);
		printf("發送字節: %u bytes\n", dwLastOut);
		printf("下行速度: %u KB\n", dwBandIn);
		printf("上行速度: %u KB\n", dwBandOut);
		printf("--------------------------\n");

		PostMessage(pDlg->m_hWnd, WMMY_NETWORKER_MESSAGE, dwBandIn, dwBandOut);
		Sleep(1000);
		//休眠一秒鐘

	}


	delete[] m_pTable;
	return 0;

}



LRESULT CMyNetCheckSystemDlg::MYMSGNETWORKER(WPARAM wParam, LPARAM lParam)
{
	// 數據的處理
	CString dwBandIn;
	dwBandIn.Format(_T("下行速度:%u KB"),wParam);
	m_downspeed.SetWindowText(dwBandIn);


	CString dwBandOut;
	dwBandOut.Format(_T("上行速度:%u KB"), lParam);
	m_upspeed.SetWindowText(dwBandOut);

	return 0;
}

//------------------------------------------------------------------------------
int DisableLANFunc(void)
{
	HDEVINFO  hDevInfo = INVALID_HANDLE_VALUE;

	// 獲得一個返回值
	hDevInfo = SetupDiGetClassDevs(NULL, NULL, NULL, DIGCF_PRESENT | DIGCF_ALLCLASSES);

	if (INVALID_HANDLE_VALUE==hDevInfo)
	{
		return GetLastError();
	}

	SP_DEVINFO_DATA DeviceInfoData = { sizeof(SP_DEVINFO_DATA) };
	LPOLESTR guid;
	TCHAR devName[128];
	TCHAR instanceId[128];
	for (int i = 0;SetupDiEnumDeviceInfo(hDevInfo,i,&DeviceInfoData) ; i++)
	{
		// 將CLSID結構轉換爲字符串方式
		StringFromCLSID(DeviceInfoData.ClassGuid, &guid);
		SetupDiClassNameFromGuid(&DeviceInfoData.ClassGuid, devName, 128, NULL);

		if (!strcmp(devName,_T("Net")))
		{
			SetupDiGetDeviceInstanceId(hDevInfo, &DeviceInfoData, instanceId, 128, NULL);
			if (!strncmp(instanceId, _T("PCI"), 3))
			{
				SP_PROPCHANGE_PARAMS params = { sizeof(SP_CLASSINSTALL_HEADER) };
				params.ClassInstallHeader.InstallFunction = DIF_PROPERTYCHANGE;
				params.Scope = DICS_FLAG_CONFIGSPECIFIC;
				params.StateChange = DICS_DISABLE;
				params.HwProfile = 0;
				SetupDiSetClassInstallParams(hDevInfo, &DeviceInfoData, (SP_CLASSINSTALL_HEADER*)&params, sizeof(SP_PROPCHANGE_PARAMS));
				SetupDiChangeState(hDevInfo, &DeviceInfoData);
			}
		}
		CoTaskMemFree(guid);

	}
	SetupDiDestroyDeviceInfoList(hDevInfo);


	return 0;
}

int EnableLANFunc(void)
{
	HDEVINFO  hDevInfo = INVALID_HANDLE_VALUE;

	hDevInfo = SetupDiGetClassDevs(NULL, NULL, NULL, DIGCF_PRESENT | DIGCF_ALLCLASSES);

	if (INVALID_HANDLE_VALUE == hDevInfo)
		return GetLastError();

	SP_DEVINFO_DATA DeviceInfoData = { sizeof(SP_DEVINFO_DATA) };
	LPOLESTR guid;
	TCHAR devName[128];
	TCHAR instanceId[128];
	for (int i = 0; SetupDiEnumDeviceInfo(hDevInfo, i, &DeviceInfoData); i++)
	{
		StringFromCLSID(DeviceInfoData.ClassGuid, &guid);
		SetupDiClassNameFromGuid(&DeviceInfoData.ClassGuid, devName, 128, NULL);

		if (!strcmp(devName, _T("Net")))
		{
			SetupDiGetDeviceInstanceId(hDevInfo, &DeviceInfoData, instanceId, 128, NULL);
			if (!strncmp(instanceId, _T("PCI"), 3))
			{
				SP_PROPCHANGE_PARAMS params = { sizeof(SP_CLASSINSTALL_HEADER) };
				params.ClassInstallHeader.InstallFunction = DIF_PROPERTYCHANGE;
				params.Scope = DICS_FLAG_CONFIGSPECIFIC;
				params.StateChange = CDCS_ENABLED;
				params.HwProfile = 0;
				SetupDiSetClassInstallParams(hDevInfo, &DeviceInfoData, (SP_CLASSINSTALL_HEADER*)&params, sizeof(SP_PROPCHANGE_PARAMS));
				SetupDiChangeState(hDevInfo, &DeviceInfoData);
			}
		}
		CoTaskMemFree(guid);
	}
	SetupDiDestroyDeviceInfoList(hDevInfo);

	return 0;
}


//------------------------------------------------------------------------------


BEGIN_MESSAGE_MAP(CMyNetCheckSystemDlg, CDialogEx)
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()


	ON_MESSAGE(WMMY_NETWORKER_MESSAGE, MYMSGNETWORKER)


	ON_BN_CLICKED(IDC_BUTTON1, &CMyNetCheckSystemDlg::OnBnClickedButton1)
	ON_BN_CLICKED(IDC_BUTTON5, &CMyNetCheckSystemDlg::OnBnClickedButton5)
	ON_WM_TIMER()
END_MESSAGE_MAP()


// CMyNetCheckSystemDlg 消息處理程序

BOOL CMyNetCheckSystemDlg::OnInitDialog()
{
	CDialogEx::OnInitDialog();

	// 設置此對話框的圖標。  當應用程序主窗口不是對話框時,框架將自動
	//  執行此操作
	SetIcon(m_hIcon, TRUE);			// 設置大圖標
	SetIcon(m_hIcon, FALSE);		// 設置小圖標

	// TODO:  在此添加額外的初始化代碼

	// 添加定時器
	SetTimer(1, 1000, 0);

	return TRUE;  // 除非將焦點設置到控件,否則返回 TRUE
}

// 如果向對話框添加最小化按鈕,則需要下面的代碼
//  來繪製該圖標。  對於使用文檔/視圖模型的 MFC 應用程序,
//  這將由框架自動完成。



void CMyNetCheckSystemDlg::OnBnClickedButton1()
{
	// TODO:  在此添加控件通知處理程序代碼
	HKEY hKey;
	LPCSTR data_set = _T("SYSTEM\\CurrentControlSet\\Control\\Class\\{4d36e972-e325-11ce-bfc1-08002be10318}\\0001");
	if (ERROR_SUCCESS == RegOpenKeyEx(HKEY_LOCAL_MACHINE, data_set, 0, KEY_ALL_ACCESS, &hKey))
	{
		CString itemValue = _T("4");  // 定義字符串常量 
		DWORD size = itemValue.GetLength();	// 獲取字符串長度

		// 設置項值
		if (ERROR_SUCCESS == RegSetValueEx(hKey, _T("*CNetworkCheckSystemDlg"), 0, REG_SZ, (unsigned char*)itemValue.GetBuffer(0), size))
		{
			DisableLANFunc();
			EnableLANFunc();
		}
		else
		{
			exit(1);
		}
	}
	else
	{
		exit(1);
	}
	RegCloseKey(hKey);	//關閉鍵句柄
}


void CMyNetCheckSystemDlg::OnBnClickedButton5()
{
	// TODO:  在此添加控件通知處理程序代碼

	HKEY hKEY;
	LPCTSTR data_Set = _T("SYSTEM\\CurrentControlSet\\Control\\Class\\{4d36e972-e325-11ce-bfc1-08002be10318}\\0001");
	if (ERROR_SUCCESS == RegOpenKeyEx(HKEY_LOCAL_MACHINE, data_Set, 0, KEY_ALL_ACCESS, &hKEY))
	{
		CString itemvalue = _T("6");//定義字符串變量
		DWORD size = itemvalue.GetLength();//獲取字符串長度
		//設置項值
		if (ERROR_SUCCESS == RegSetValueEx(hKEY, _T("*CNetworkMonitorDlg"), 0, REG_SZ, (unsigned char*)itemvalue.GetBuffer(0), size))
		{
			DisableLANFunc();
			EnableLANFunc();
		}
		else
		{
			exit(1);
		}
	}
	else
	{
		exit(1);
	}

	RegCloseKey(hKEY);//關閉鍵句柄
}


void CMyNetCheckSystemDlg::OnTimer(UINT_PTR nIDEvent)
{
	// TODO:  在此添加消息處理程序代碼和/或調用默認值
	switch (nIDEvent)
	{

	case 1:
	{
		HANDLE m_hThread_speed = (HANDLE)_beginthreadex(NULL, 0, ThreadCheckSpeed, this, 0, NULL);
		break;
	}

	default:
		break;
	}


	CDialogEx::OnTimer(nIDEvent);
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章