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

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

SNMP编程

开发平台：

Visual C++

TrafficDlg.cpp：源码内容
  1. // TrafficDlg.cpp : implementation file
  2. //
  4. #include "stdafx.h"
  5. #include "Traffic.h"
  6. #include "TrafficDlg.h"
  7. #include "input.h"
  9. #ifdef _DEBUG
  10. #define new DEBUG_NEW
  11. #undef THIS_FILE
  12. static char THIS_FILE[] = __FILE__;
  13. #endif
  15. /////////////////////////////////////////////////////////////////////////////
  16. // CAboutDlg dialog used for App About
  18. class CAboutDlg : public CDialog
  19. {
  20. public:
  21. CAboutDlg();
  23. // Dialog Data
  24. //{{AFX_DATA(CAboutDlg)
  25. enum { IDD = IDD_ABOUTBOX };
  26. //}}AFX_DATA
  28. // ClassWizard generated virtual function overrides
  29. //{{AFX_VIRTUAL(CAboutDlg)
  30. protected:
  31. virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
  32. //}}AFX_VIRTUAL
  34. // Implementation
  35. protected:
  36. //{{AFX_MSG(CAboutDlg)
  37. //}}AFX_MSG
  38. DECLARE_MESSAGE_MAP()
  39. };
  41. CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
  42. {
  43. //{{AFX_DATA_INIT(CAboutDlg)
  44. //}}AFX_DATA_INIT
  45. }
  47. void CAboutDlg::DoDataExchange(CDataExchange* pDX)
  48. {
  49. CDialog::DoDataExchange(pDX);
  50. //{{AFX_DATA_MAP(CAboutDlg)
  51. //}}AFX_DATA_MAP
  52. }
  54. BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
  55. //{{AFX_MSG_MAP(CAboutDlg)
  56. // No message handlers
  57. //}}AFX_MSG_MAP
  58. END_MESSAGE_MAP()
  60. /////////////////////////////////////////////////////////////////////////////
  61. // CTrafficDlg dialog
  63. CTrafficDlg::CTrafficDlg(CWnd* pParent /*=NULL*/)
  64. : CDialog(CTrafficDlg::IDD, pParent)
  65. {
  66. //{{AFX_DATA_INIT(CTrafficDlg)
  67. //}}AFX_DATA_INIT
  68. // Note that LoadIcon does not require a subsequent DestroyIcon in Win32
  69. m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
  70. }
  72. void CTrafficDlg::DoDataExchange(CDataExchange* pDX)
  73. {
  74. CDialog::DoDataExchange(pDX);
  75. //{{AFX_DATA_MAP(CTrafficDlg)
  76. DDX_Control(pDX, IDC_EDIT2, m_community);
  77. DDX_Control(pDX, IDC_IPADDRESS1, m_ipadd);
  78. DDX_Control(pDX, IDC_LIST2, m_list1);
  79. //}}AFX_DATA_MAP
  80. }
  82. BEGIN_MESSAGE_MAP(CTrafficDlg, CDialog)
  83. //{{AFX_MSG_MAP(CTrafficDlg)
  84. ON_WM_SYSCOMMAND()
  85. ON_WM_PAINT()
  86. ON_WM_QUERYDRAGICON()
  87. ON_WM_DESTROY()
  88. ON_BN_CLICKED(IDC_BUTTON1, OnScan)
  89. ON_BN_CLICKED(IDC_BUTTON2, OnSure)
  90. ON_BN_CLICKED(IDC_BUTTON3, OnModify)
  91. ON_BN_CLICKED(IDC_BUTTON4, OnStart)
  92. ON_WM_TIMER()
  93. ON_BN_CLICKED(IDC_BUTTON5, OnEnd)
  94. ON_BN_CLICKED(IDC_BUTTON6, OnWorkData)
  95. ON_BN_CLICKED(IDC_BUTTON7, OnOpenFile)
  96. ON_BN_CLICKED(IDC_BUTTON8, OnToLeft)
  97. ON_BN_CLICKED(IDC_BUTTON9, OnToRight)
  98. ON_BN_CLICKED(IDC_CHECK1, OnChangeUse)
  99. //}}AFX_MSG_MAP
  100. END_MESSAGE_MAP()
  102. /////////////////////////////////////////////////////////////////////////////
  103. // CTrafficDlg message handlers
  105. BOOL CTrafficDlg::OnInitDialog()
  106. {
  107. CDialog::OnInitDialog();
  109. // Add "About..." menu item to system menu.
  111. // IDM_ABOUTBOX must be in the system command range.
  112. ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
  113. ASSERT(IDM_ABOUTBOX < 0xF000);
  115. CMenu* pSysMenu = GetSystemMenu(FALSE);
  116. if (pSysMenu != NULL)
  117. {
  118. CString strAboutMenu;
  119. strAboutMenu.LoadString(IDS_ABOUTBOX);
  120. if (!strAboutMenu.IsEmpty())
  121. {
  122. pSysMenu->AppendMenu(MF_SEPARATOR);
  123. pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
  124. }
  125. }
  127. // Set the icon for this dialog.  The framework does this automatically
  128. //  when the application's main window is not a dialog
  129. SetIcon(m_hIcon, TRUE); // Set big icon
  130. SetIcon(m_hIcon, FALSE); // Set small icon
  132. // TODO: Add extra initialization here
  133.     m_list1.InsertColumn(0,"  类型   ",LVCFMT_LEFT,80,0);
  134. m_list1.InsertColumn(1,"  名称（DLCI）",LVCFMT_LEFT,130,1);
  135.     m_list1.InsertColumn(2,"  最大速率",LVCFMT_LEFT,70,2);
  136.     m_list1.InsertColumn(3,"  状态  ",LVCFMT_LEFT,70,3);
  137. m_list1.InsertColumn(4,"  辅助信息  ",LVCFMT_LEFT,300,4);
  138. m_list1.SetExtendedStyle(LVS_EX_FULLROWSELECT|LVS_EX_GRIDLINES);
  139.     CRect cr(25,12,518,150);
  140. CSize c(388,150);
  141. pDay=NULL;
  142. DayNumber=288;
  143. pDay=new CLineChart(c,DayNumber);
  144. pDay->Create(NULL,WS_CHILD|WS_VISIBLE|SS_NOTIFY|SS_BLACKFRAME, cr,this,50000);
  145. pDay->SetTitle("日流量图");
  146. pDay->SetLegend("流入","流出");
  147.     CRect cr1(25,169,518,170);
  148. CSize c1(388,150);
  149. pWeek=NULL;
  150. pWeek=new CLineChart(c1,DayNumber);
  151. pWeek->Create(NULL,WS_CHILD|WS_VISIBLE|SS_NOTIFY|SS_BLACKFRAME, cr1,this,50001);
  152. pWeek->SetTitle("周流量图");
  153. pWeek->SetLegend("流入","流出");
  154. index="";
  155. SampleNumber=0;
  156. return TRUE;  // return TRUE  unless you set the focus to a control
  157. }
  159. void CTrafficDlg::OnSysCommand(UINT nID, LPARAM lParam)
  160. {
  161. if ((nID & 0xFFF0) == IDM_ABOUTBOX)
  162. {
  163. CAboutDlg dlgAbout;
  164. dlgAbout.DoModal();
  165. }
  166. else
  167. {
  168. CDialog::OnSysCommand(nID, lParam);
  169. }
  170. }
  172. // If you add a minimize button to your dialog, you will need the code below
  173. //  to draw the icon.  For MFC applications using the document/view model,
  174. //  this is automatically done for you by the framework.
  176. void CTrafficDlg::OnPaint() 
  177. {
  178. if (IsIconic())
  179. {
  180. CPaintDC dc(this); // device context for painting
  182. SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);
  184. // Center icon in client rectangle
  185. int cxIcon = GetSystemMetrics(SM_CXICON);
  186. int cyIcon = GetSystemMetrics(SM_CYICON);
  187. CRect rect;
  188. GetClientRect(&rect);
  189. int x = (rect.Width() - cxIcon + 1) / 2;
  190. int y = (rect.Height() - cyIcon + 1) / 2;
  192. // Draw the icon
  193. dc.DrawIcon(x, y, m_hIcon);
  194. }
  195. else
  196. {
  197. CDialog::OnPaint();
  198. }
  199. }
  201. // The system calls this to obtain the cursor to display while the user drags
  202. //  the minimized window.
  203. HCURSOR CTrafficDlg::OnQueryDragIcon()
  204. {
  205. return (HCURSOR) m_hIcon;
  206. }
  208. void CTrafficDlg::OnDestroy() 
  209. {
  210. CDialog::OnDestroy();
  212. if (pDay!=NULL) delete pDay;
  213. if (pWeek!=NULL) delete pWeek;
  214. }
  216. void CTrafficDlg::OnScan() 
  217. {
  218. CString BaseOid;
  219. m_ipadd.GetWindowText(ip);
  220. m_community.GetWindowText(community);
  221. if ((community.GetLength()==0)||(ip.GetLength()==0))
  222. {
  223. MessageBox("地址或口令错误!");
  224. return;
  225. }
  226.     m_snmp.clear();
  227. BaseOid="1.3.6.1.2.1.2.2.1.1";
  228. m_snmp.GetTree(ip,community,BaseOid);
  229. BaseOid="1.3.6.1.2.1.2.2.1.2";
  230. m_snmp.GetTree(ip,community,BaseOid);
  231. BaseOid="1.3.6.1.2.1.2.2.1.8";
  232. m_snmp.GetTree(ip,community,BaseOid);
  233. BaseOid="1.3.6.1.2.1.2.2.1.5";
  234. m_snmp.GetTree(ip,community,BaseOid);
  235. int row=m_snmp.GetCount()/4;
  236. for (int i=0;i<row;i++)
  237. {
  238. pLine=new Line;
  239.         pLine->Type="普通接口";
  240. pLine->Index=m_snmp.GetValue(i);
  241. pLine->Name=m_snmp.GetValue(i+row);
  242. pLine->Speed=m_snmp.GetValue(i+3*row);
  243. if (m_snmp.GetValue(i+2*row)=="1")
  244. pLine->State="UP";
  245. else 
  246. pLine->State="Down";
  247.       pLine->Descr="";
  248. pLine->Ip=ip;
  249. LineArray.Add(pLine);
  250. }
  251. m_snmp.clear();
  252. BaseOid="1.3.6.1.2.1.4.20.1.2";
  253. m_snmp.GetTree(ip,community,BaseOid);
  254. CString temp;
  255. for (i=0;i<m_snmp.GetCount();i++)
  256. {
  257. for (int j=0;j<LineArray.GetSize();j++)
  258. if (m_snmp.GetValue(i)==LineArray[j]->Index)
  259. {
  260. temp=m_snmp.GetOid(i);
  261. temp=temp.Right(temp.GetLength()-21);
  262. LineArray[j]->Descr="IP:"+temp;
  263. break;
  264. }
  266. }
  267. for (int j=0;j<LineArray.GetSize();j++)
  268. {
  269. BaseOid="1.3.6.1.2.1.31.1.1.1.18."+LineArray[j]->Index;
  270. m_snmp.GetOne(ip,community,BaseOid,temp);
  271. LineArray[j]->Descr+=" 备注："+temp;
  272. }
  273. //获取网络接口的信息并保存；
  274. m_snmp.clear();
  275. BaseOid="1.3.6.1.2.1.10.32.2.1.1";
  276. m_snmp.GetTree(ip,community,BaseOid);
  277. BaseOid="1.3.6.1.2.1.10.32.2.1.2";
  278. m_snmp.GetTree(ip,community,BaseOid);
  279. BaseOid="1.3.6.1.2.1.10.32.2.1.3";
  280. m_snmp.GetTree(ip,community,BaseOid);
  281. row=m_snmp.GetCount()/3;
  282. for (i=0;i<row;i++)
  283. {
  284. pLine=new Line;
  286. pLine->Type="帧中继PVC";
  287. pLine->Ip=ip;
  288. pLine->Speed="65535";
  289. pLine->Index=m_snmp.GetValue(i);
  290. pLine->Name=m_snmp.GetValue(i+row);
  291. if (m_snmp.GetValue(i+2*row)=="2")
  292. pLine->State="UP";
  293. else  pLine->State="Down";
  294. BaseOid="1.3.6.1.4.1.9.9.49.1.2.2.1.2."+pLine->Index+"."+pLine->Name;
  295. m_snmp.GetOne(ip,community,BaseOid,temp);
  296. if ("2"==temp)
  297. {
  298. CString rv;
  299. BaseOid="1.3.6.1.4.1.9.9.49.1.2.2.1.3."+pLine->Index+"."+pLine->Name;
  300. m_snmp.GetOne(ip,community,BaseOid,rv);
  301. BaseOid="1.3.6.1.2.1.31.1.1.1.18."+rv;
  302. m_snmp.GetOne(ip,community,BaseOid,rv);
  303. pLine->Descr=rv;
  304. }
  305. else
  306. {
  307. CString rv;
  308. BaseOid="1.3.6.1.4.1.9.9.49.1.3.1.1.3."+pLine->Index+"."+pLine->Name;
  309. m_snmp.GetOne(ip,community,BaseOid,rv);
  310. pLine->Descr=rv;
  311. }
  312. LineArray.Add(pLine);
  313.     
  314. }
  315.     m_list1.DeleteAllItems();
  316. for (i=0;i<LineArray.GetSize();i++)
  317. {
  318.      m_list1.InsertItem(i,LineArray[i]->Type);
  319.  m_list1.SetItemText(i,1,LineArray[i]->Name);
  320.      m_list1.SetItemText(i,2,LineArray[i]->Speed);
  321.      m_list1.SetItemText(i,3,LineArray[i]->State);
  322.      m_list1.SetItemText(i,4,LineArray[i]->Descr);
  323. }
  324. }
  326. void CTrafficDlg::OnSure() 
  327. {
  328. if (LineArray.GetSize()==0)
  329.     {
  330. MessageBox("无可用数据!");
  331. return;
  332. }
  333.   POSITION pos;
  334.   pos=m_list1.GetFirstSelectedItemPosition();
  335.   if (pos==NULL)
  336.   {
  337.   MessageBox("没有选择线路!");
  338.   return;
  339.   }
  340.   int n=m_list1.GetNextSelectedItem(pos);
  341.   if (LineArray[n]->Type=="帧中继PVC")
  342.   {
  343.     index=LineArray[n]->Index+"."+LineArray[n]->Name;
  344. oid="1.3.6.1.2.1.10.32.2.1";
  345.   }
  346.   else
  347.   {
  348. index=LineArray[n]->Index;
  349.     oid="1.3.6.1.2.1.2.2.1";
  350.   }
  351.   MaxSpeed=atoi(LineArray[n]->Speed);
  352.  for (int i=0;i<LineArray.GetSize();i++)
  353.  {
  354.  if (LineArray[i]!=NULL) delete LineArray[i];
  355.  }  
  356.  m_list1.DeleteAllItems();
  357. }
  359. void CTrafficDlg::OnModify() 
  360. {
  361. char ch[10];
  362. itoa(MaxSpeed,ch,10);
  363. Cinput dlg;
  364. dlg.m_old=ch;
  365. if (dlg.DoModal()==IDOK)
  366. {   
  367. if (dlg.m_new.GetLength()>0)
  368. MaxSpeed=atoi((LPCTSTR)dlg.m_new);
  369. MessageBox("速率更改成功!");
  370. }
  371. }
  373. void CTrafficDlg::OnStart() 
  374. {
  375. if (index.GetLength()==0)
  376. {
  377. MessageBox("OID错误!");
  378. return;
  379. }
  380. pDay->Init();
  381. pWeek->Init();
  382. CTime t=CTime::GetCurrentTime();
  383. fDayName=t.Format("%m-%d-%H");
  384. fDayName=fDayName+"Day.txt";
  385. DataFile.Open(fDayName,CFile::modeCreate);
  386. DataFile.Close();
  387. fWeekName=t.Format("%m-%d-%H");
  388. fWeekName=fWeekName+"Week.txt";
  389. DataFile.Open(fWeekName,CFile::modeCreate);
  390. DataFile.Close();
  391. SetTimer(1,300000,NULL);
  392. m_snmp.GetThree(ip,community,oid,index,last_in,last_out,last_time);
  393. CButton* m_pB=(CButton*)GetDlgItem(IDC_BUTTON4);
  394. m_pB->EnableWindow(FALSE);
  395. }
  397. void CTrafficDlg::OnTimer(UINT nIDEvent) 
  398. {
  399. unsigned long InBits,OutBits;
  400. unsigned long in_v,out_v,time_v,in_diff,out_diff,time_diff;
  401. CTime t=CTime::GetCurrentTime();
  402. CString time=t.Format("%H:%M");
  403. m_snmp.GetThree(ip,community,oid,index,in_v,out_v,time_v);
  404. in_diff=in_v-last_in;
  405. if (in_diff<0)
  406. in_diff+=4294967295;
  407. out_diff=out_v-last_out;
  408. if (out_diff<0)
  409. out_diff+=4294967295;
  410. time_diff=time_v-last_time;
  411. if (time_diff<0)
  412. time_diff+=4294967295;
  413. InBits=(in_diff*8)/(time_diff/100);
  414. OutBits=(out_diff*8)/(time_diff/100);
  415. int In_use=InBits*100/MaxSpeed;
  416. int Out_use=OutBits*100/MaxSpeed;
  417. pDay->Append(InBits,OutBits,time,In_use,Out_use);
  418. SampleNumber+=1;
  419. CString line,s;
  420. line="";
  421. if ((SampleNumber % 6)==0)
  422. {
  423. CString time1=t.Format("%a");
  424. pWeek->Append(pDay->GetXAverage(),pDay->GetYAverage(),time1,0,0);
  425. line+=time1;
  426. if(DataFile.Open(fWeekName,CFile::modeWrite))
  427. {
  428. s.Format("%d",pDay->GetXAverage());
  429. line=line+"@"+s;
  430. s.Format("%d",pDay->GetYAverage());
  431. line=line+"#"+s;
  432. s.Format("%d",0);
  433. line=line+"$"+s;
  434. s.Format("%d",0);
  435. line=line+"&"+s+"rn";
  436. DataFile.SeekToEnd();
  437. DataFile.WriteString(line);
  438. line="";
  439. DataFile.Close();
  440. }
  441. }
  442. last_in=in_v;
  443. last_out=out_v;
  444. last_time=time_v;
  445. line+=time;
  446. if(DataFile.Open(fDayName,CFile::modeWrite))
  447. {
  448. s.Format("%d",InBits);
  449. line=line+"@"+s;
  450. s.Format("%d",OutBits);
  451. line=line+"#"+s;
  452. s.Format("%d",In_use);
  453. line=line+"$"+s;
  454. s.Format("%d",Out_use);
  455. line=line+"&"+s+"rn";
  456. DataFile.SeekToEnd();
  457. DataFile.WriteString(line);
  458. DataFile.Close();
  459. }
  460. if (SampleNumber==DayNumber)
  461. {
  462. CTime t=CTime::GetCurrentTime();
  463. fDayName=t.Format("%m-%d-%H");
  464. fDayName=fDayName+".txt";
  465. DataFile.Open(fDayName,CFile::modeCreate);
  466. DataFile.Close();
  467. SampleNumber=0;
  468. }
  470. CDialog::OnTimer(nIDEvent);
  471. }
  473. void CTrafficDlg::OnEnd() 
  474. {
  475. KillTimer(1);
  476. CButton* m_pB=(CButton*)GetDlgItem(IDC_BUTTON4);
  477. m_pB->EnableWindow(TRUE);
  478. }
  480. void CTrafficDlg::OnWorkData() 
  481. {
  482. CButton* m_pB=(CButton*)GetDlgItem(IDC_BUTTON4);
  483. if (!(m_pB->IsWindowEnabled()))
  484. {
  485. MessageBox("先停止流量测试!");
  486. return;
  487. }
  488. pDay->Init();
  489. pWeek->Init();
  490. pDay->SetCross(true);
  491. }
  493. void CTrafficDlg::OnOpenFile() 
  494. {   
  495. bool show;
  496.     unsigned long x,y;
  497. int number,Usex,Usey;
  498.     CStdioFile dataFile;
  499. CString line,t;
  500. pDay->Init();
  501. CFileDialog openfile(TRUE,"*.txt",NULL,
  502.  OFN_HIDEREADONLY|OFN_OVERWRITEPROMPT,
  503.  "Text files(*.txt)|*.txt");
  504. if (openfile.DoModal()==IDOK)
  505. {
  506. if(dataFile.Open(openfile.GetPathName(),CFile::modeRead))
  507. {   
  508. number=0;
  509. while (dataFile.ReadString(line))
  510. {
  511. int n0=line.Find('@');
  512. CString str=line.Mid(0,n0);
  513. t=str;
  514.                 int n1=line.Find('#');
  515. str=line.Mid(n0+1,n1-n0-1);
  516. x=atoi((LPCTSTR)str);
  517. n0=line.Find('$');
  518. str=line.Mid(n1+1,n0-n1-1);
  519.                 y=atoi((LPCTSTR)str);
  520. n1=line.Find('&');
  521.                 str=line.Mid(n0+1,n1-n0-1);
  522.                 Usex=atoi((LPCTSTR)str);
  523. str=line.Mid(n1+1,10);
  524.                 Usey=atoi((LPCTSTR)str);
  525. if ((number % 36)==0)
  526. show=true;
  527. else 
  528. show=false;
  529. pDay->AddData(x,y,t,Usex,Usey,number,show);
  530.     number+=1;
  531. }
  533. }
  534.    dataFile.Close();
  535. }
  536. pDay->Invalidate();
  537. }
  539. void CTrafficDlg::OnToLeft() 
  540. {
  541. pDay->ToLeft();
  542. }
  544. void CTrafficDlg::OnToRight() 
  545. {
  546. pDay->ToRight();
  547. }
  549. void CTrafficDlg::OnChangeUse() 
  550. {
  551. CButton* m_pCheck=(CButton*)GetDlgItem(IDC_CHECK1);
  552. if(m_pCheck->GetCheck())
  553. pDay->SetUse(true);
  554. else
  555. pDay->SetUse(false);
  556. }