开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > SNMP编程 > 查看源码
LanFlowDlg.cpp
资源名称：VC_SNMP_NetManage.rar　[点击查看]
上传用户：hbytqc8
上传日期：2014-07-31
资源大小：527k
文件大小：9k
源码类别：

SNMP编程

开发平台：

Visual C++

LanFlowDlg.cpp：源码内容
  1. // LanFlowDlg.cpp : implementation file
  2. //
  4. #include "stdafx.h"
  5. #include "LanFlow.h"
  6. #include "LanFlowDlg.h"
  8. #ifdef _DEBUG
  9. #define new DEBUG_NEW
  10. #undef THIS_FILE
  11. static char THIS_FILE[] = __FILE__;
  12. #endif
  14. /////////////////////////////////////////////////////////////////////////////
  15. // CAboutDlg dialog used for App About
  17. class CAboutDlg : public CDialog
  18. {
  19. public:
  20. CAboutDlg();
  22. // Dialog Data
  23. //{{AFX_DATA(CAboutDlg)
  24. enum { IDD = IDD_ABOUTBOX };
  25. //}}AFX_DATA
  27. // ClassWizard generated virtual function overrides
  28. //{{AFX_VIRTUAL(CAboutDlg)
  29. protected:
  30. virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
  31. //}}AFX_VIRTUAL
  33. // Implementation
  34. protected:
  35. //{{AFX_MSG(CAboutDlg)
  36. //}}AFX_MSG
  37. DECLARE_MESSAGE_MAP()
  38. };
  40. CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
  41. {
  42. //{{AFX_DATA_INIT(CAboutDlg)
  43. //}}AFX_DATA_INIT
  44. }
  46. void CAboutDlg::DoDataExchange(CDataExchange* pDX)
  47. {
  48. CDialog::DoDataExchange(pDX);
  49. //{{AFX_DATA_MAP(CAboutDlg)
  50. //}}AFX_DATA_MAP
  51. }
  53. BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
  54. //{{AFX_MSG_MAP(CAboutDlg)
  55. // No message handlers
  56. //}}AFX_MSG_MAP
  57. END_MESSAGE_MAP()
  59. /////////////////////////////////////////////////////////////////////////////
  60. // CLanFlowDlg dialog
  62. CLanFlowDlg::CLanFlowDlg(CWnd* pParent /*=NULL*/)
  63. : CDialog(CLanFlowDlg::IDD, pParent)
  64. {
  65. //{{AFX_DATA_INIT(CLanFlowDlg)
  66. //}}AFX_DATA_INIT
  67. // Note that LoadIcon does not require a subsequent DestroyIcon in Win32
  68. m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
  69. }
  71. void CLanFlowDlg::DoDataExchange(CDataExchange* pDX)
  72. {
  73. CDialog::DoDataExchange(pDX);
  74. //{{AFX_DATA_MAP(CLanFlowDlg)
  75. DDX_Control(pDX, IDC_EDIT2, m_community);
  76. DDX_Control(pDX, IDC_EDIT1, m_text);
  77. DDX_Control(pDX, IDC_IPADDRESS1, m_ipadd);
  78. DDX_Control(pDX, IDC_COMBO1, m_port);
  79. //}}AFX_DATA_MAP
  80. }
  82. BEGIN_MESSAGE_MAP(CLanFlowDlg, CDialog)
  83. //{{AFX_MSG_MAP(CLanFlowDlg)
  84. ON_WM_SYSCOMMAND()
  85. ON_WM_PAINT()
  86. ON_WM_QUERYDRAGICON()
  87. ON_WM_DESTROY()
  88. ON_BN_CLICKED(IDC_BUTTON1, OnSure)
  89. ON_BN_CLICKED(IDC_BUTTON2, OnMAC)
  90. ON_BN_CLICKED(IDC_BUTTON3, OnIP)
  91. ON_BN_CLICKED(IDC_LEGEND, OnLegend)
  92. ON_BN_CLICKED(IDC_PERCENT, OnPercent)
  93. ON_BN_CLICKED(IDC_LEFT, OnLeft)
  94. ON_BN_CLICKED(IDC_RIGHT, OnRight)
  95. //}}AFX_MSG_MAP
  96. END_MESSAGE_MAP()
  98. /////////////////////////////////////////////////////////////////////////////
  99. // CLanFlowDlg message handlers
  101. BOOL CLanFlowDlg::OnInitDialog()
  102. {
  103. CDialog::OnInitDialog();
  104. // IDM_ABOUTBOX must be in the system command range.
  105. ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
  106. ASSERT(IDM_ABOUTBOX < 0xF000);
  108. CMenu* pSysMenu = GetSystemMenu(FALSE);
  109. if (pSysMenu != NULL)
  110. {
  111. CString strAboutMenu;
  112. strAboutMenu.LoadString(IDS_ABOUTBOX);
  113. if (!strAboutMenu.IsEmpty())
  114. {
  115. pSysMenu->AppendMenu(MF_SEPARATOR);
  116. pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
  117. }
  118. }
  119. SetIcon(m_hIcon, TRUE); // Set big icon
  120. SetIcon(m_hIcon, FALSE); // Set small icon
  121. // TODO: Add extra initialization here
  122. CRect cr(13,110,550,240);
  123. CSize c(550,240);
  124.     pChart=NULL;
  125. pChart=new CLineChart(c);
  126. pChart->Create(NULL,WS_CHILD|WS_VISIBLE|SS_NOTIFY|SS_BLACKFRAME, cr,this,50000);
  127. pChart->SetTitle("基于地址流量统计图");
  128. pChart->SetLegend("流入","流出");
  129. m_community.SetWindowText("public");
  130. m_ipadd.SetWindowText("127.0.0.1");
  131.     CButton* m_pB=(CButton*)GetDlgItem(IDC_BUTTON2);
  132.     m_pB->EnableWindow(FALSE);
  133.     m_pB=(CButton*)GetDlgItem(IDC_BUTTON3);
  134.     m_pB->EnableWindow(FALSE);
  135. return TRUE;  // return TRUE  unless you set the focus to a control
  136. }
  138. void CLanFlowDlg::OnSysCommand(UINT nID, LPARAM lParam)
  139. {
  140. if ((nID & 0xFFF0) == IDM_ABOUTBOX)
  141. {
  142. CAboutDlg dlgAbout;
  143. dlgAbout.DoModal();
  144. }
  145. else
  146. {
  147. CDialog::OnSysCommand(nID, lParam);
  148. }
  149. }
  151. // If you add a minimize button to your dialog, you will need the code below
  152. //  to draw the icon.  For MFC applications using the document/view model,
  153. //  this is automatically done for you by the framework.
  155. void CLanFlowDlg::OnPaint() 
  156. {
  157. if (IsIconic())
  158. {
  159. CPaintDC dc(this); // device context for painting
  161. SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);
  163. // Center icon in client rectangle
  164. int cxIcon = GetSystemMetrics(SM_CXICON);
  165. int cyIcon = GetSystemMetrics(SM_CYICON);
  166. CRect rect;
  167. GetClientRect(&rect);
  168. int x = (rect.Width() - cxIcon + 1) / 2;
  169. int y = (rect.Height() - cyIcon + 1) / 2;
  171. // Draw the icon
  172. dc.DrawIcon(x, y, m_hIcon);
  173. }
  174. else
  175. {
  176. CDialog::OnPaint();
  177. }
  178. }
  180. // The system calls this to obtain the cursor to display while the user drags
  181. //  the minimized window.
  182. HCURSOR CLanFlowDlg::OnQueryDragIcon()
  183. {
  184. return (HCURSOR) m_hIcon;
  185. }
  186. CString CLanFlowDlg::itoh(int m)
  187. {
  188.  char ch[2];
  189.  CString Hex;
  190.  if ((m>=0)&&(m<10))
  191.  {
  192.    itoa(m,ch,10);
  193.    Hex=ch;
  194.  }
  195.  else
  196.   switch (m)
  197.       {
  198.        case 10:
  199.   Hex="A";
  200.   break;
  201.       case 11:
  202.   Hex="B";
  203.   break;
  204.       case 12:
  205.   Hex="C";
  206.   break;
  207.       case 13:
  208.   Hex="D";
  209.   break;
  210.       case 14:
  211.   Hex="E";
  212.   break;
  213.       case 15:
  214.   Hex="F";
  215.   break;
  216.       }
  217.  return Hex;
  218. }
  219. CString CLanFlowDlg::stoh(CString s)
  220. {
  221. if (s.GetLength()==0)
  222. return "";
  223. s=s+".";
  224. int n,m,x,y;
  225. CString H,hex,temp;
  226. H="";
  227. hex="";
  228. while (s.GetLength()>0)
  229. {
  230.   n=s.Find('.');
  231.   temp=s.Left(n);
  232.   s=s.Mid(n+1,s.GetLength());
  233.   m=atoi(temp);
  234.   x=m/16;
  235.   y=m %16;
  236.   hex=itoh(x);
  237.   hex=hex+itoh(y);
  238.   H+=hex+" ";
  239. }
  240. H=H.Left(17);
  241. return H;
  242. }
  243. void CLanFlowDlg::OnDestroy() 
  244. {
  245. CDialog::OnDestroy();
  246. delete pChart;
  247. }
  249. void CLanFlowDlg::OnSure() 
  250. {
  251. CString oid,mac;
  252. m_ipadd.GetWindowText(ip);
  253. m_community.GetWindowText(community);
  254. m_snmp.clear();
  255. oid="1.3.6.1.2.1.2.2.1.1";
  256. m_snmp.GetTree(ip,community,oid);
  257. oid="1.3.6.1.2.1.2.2.1.2";
  258. m_snmp.GetTree(ip,community,oid);
  259. int row=m_snmp.GetCount()/2;
  260. for (int i=0;i<row;i++)
  261. {
  262. NameToIndex.SetAt(m_snmp.GetValue(i+row),m_snmp.GetValue(i));
  263. m_port.AddString(m_snmp.GetValue(i+row));
  264. }
  265. m_snmp.clear();
  266. oid="1.3.6.1.2.1.3.1.1.2";
  267. m_snmp.GetTree(ip,community,oid);
  268. //物理地址；
  269. oid="1.3.6.1.2.1.3.1.1.3";
  270. m_snmp.GetTree(ip,community,oid);
  271. //网络地址；
  272. row=m_snmp.GetCount()/2;
  273. for (i=0;i<row;i++)
  274. {
  275.         mac=m_snmp.GetValue(i);
  276. mac=mac.Left(17);
  277. mac.TrimLeft();
  278. mac.TrimRight();
  279. MacToIp.SetAt(mac,m_snmp.GetValue(i+row));
  280. }
  281. CString str="               OK!           rn";
  282. str+="如是基于MAC地址统计流量， rn";  
  283. str+="请先选择一个以太网络端口；rn" ;
  284. str+="如是基于IP地址统计流量，rn";
  285. str+="则不需要选择端口。";    
  286. MessageBox(str);
  287. CButton* m_pB=(CButton*)GetDlgItem(IDC_BUTTON2);
  288. m_pB->EnableWindow(TRUE);
  289. m_pB=(CButton*)GetDlgItem(IDC_BUTTON3);
  290. m_pB->EnableWindow(TRUE);
  291. }
  293. void CLanFlowDlg::OnMAC() 
  294. {   
  295. CString temp="";
  296. m_port.GetWindowText(temp);
  297. if (temp.GetLength()==0)
  298. {
  299. MessageBox("请选择网络接口!");
  300. return;
  301. }
  302. NameToIndex.Lookup(temp,index);
  303. CString oid;
  304. oid="1.3.6.1.4.1.9.9.84.1.2.1.1.4."+index+".1";
  305. m_snmp.clear();
  306. m_snmp.GetTree(ip,community,oid);
  307. Mac mac;
  308. for(int i=0;i<m_snmp.GetCount();i++)
  309. {
  310. mac.x=atoi(m_snmp.GetValue(i));
  311. mac.y=0;
  312. temp=m_snmp.GetOid(i);
  313. int n=temp.GetLength()-oid.GetLength()-1;
  314. temp=temp.Right(n);
  315. mac.str=stoh(temp);
  316. ArrayMac.Add(mac);
  317. }
  318. int count=ArrayMac.GetSize();
  319. m_snmp.clear();
  320. oid="1.3.6.1.4.1.9.9.84.1.2.1.1.4."+index+".2";
  321. m_snmp.GetTree(ip,community,oid);
  322. for(i=0;i<m_snmp.GetCount();i++)
  323. {
  324. temp=m_snmp.GetOid(i);
  325. int n=temp.GetLength()-oid.GetLength()-1;
  326. temp=temp.Right(n);
  327. temp=stoh(temp);
  328. for (int j=0;j<count;j++)
  329. {
  330. if (temp==ArrayMac[j].str)
  331. {
  332. ArrayMac[j].y=atoi(m_snmp.GetValue(i));
  333. break;
  334. }
  335. }
  336. if (j==count)
  337. {
  338. mac.x=0;
  339. mac.y=atoi(m_snmp.GetValue(i));
  340. mac.str=temp;
  341. ArrayMac.Add(mac);
  342. }
  343. }
  344. CString ipStr;
  345. pChart->Init();
  346. for (i=0;i<ArrayMac.GetSize();i++)
  347. {
  348. ipStr="*.*.*.*";
  349.         if (MacToIp.IsEmpty()==0)
  350. MacToIp.Lookup(ArrayMac[i].str,ipStr);
  351. pChart->Append(ArrayMac[i].x,ArrayMac[i].y,ArrayMac[i].str,ipStr);
  352.         
  353. }
  354. pChart->WorkPercent();
  355. ArrayMac.RemoveAll();
  356. CString txt="";
  357. txt=pChart->GetText();
  358. m_text.SetWindowText(txt);
  359. }
  361. void CLanFlowDlg::OnIP() 
  362. {
  363. CString oid,cp1,cp2;
  364. m_snmp.clear();
  365. oid="1.3.6.1.4.1.9.2.4.11.0";
  366.     m_snmp.GetOne(ip,community,oid,cp1);
  367. m_snmp.Set(ip,community,oid,cp1,0);
  368.     m_snmp.GetOne(ip,community,oid,cp2);
  369.     if (atoi(cp2)==atoi(cp1)+1)
  370. {
  371. oid="1.3.6.1.4.1.9.2.4.9.1.1";
  372. m_snmp.GetTree(ip,community,oid);
  373. oid="1.3.6.1.4.1.9.2.4.9.1.2";
  374. m_snmp.GetTree(ip,community,oid);
  375. oid="1.3.6.1.4.1.9.2.4.9.1.4";
  376. m_snmp.GetTree(ip,community,oid);
  377. int row=m_snmp.GetCount()/3;
  378. pChart->Init();
  379. for (int i=0;i<row;i++)
  380. {
  381. int data=atoi(m_snmp.GetValue(i+2*row));
  382. pChart->Append(data,0,m_snmp.GetValue(i),m_snmp.GetValue(i+row));
  383. }
  384. pChart->WorkPercent();
  385. CString txt="";
  386. txt=pChart->GetText();
  387. m_text.SetWindowText(txt);
  388. }
  389. else
  390. {
  391. MessageBox("出现错误,请检查配置!");
  392. return;
  393. }
  395. }
  397. void CLanFlowDlg::OnLegend() 
  398. {
  399. pChart->SetMark();
  400. }
  402. void CLanFlowDlg::OnPercent() 
  403. {
  404. pChart->SetPercent();
  405. }
  407. void CLanFlowDlg::OnLeft() 
  408. {
  409. pChart->SetFirst(false);
  410. }
  412. void CLanFlowDlg::OnRight() 
  413. {
  414. pChart->SetFirst(true);
  415. }