#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);
};

// 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);
}