RshDlg.cpp
上传用户:hbytqc8
上传日期:2014-07-31
资源大小:527k
文件大小:9k
源码类别:

SNMP编程

开发平台:

Visual C++

  1. // RshDlg.cpp : implementation file
  2. //
  3. #include "stdafx.h"
  4. #include "Rsh.h"
  5. #include "RshDlg.h"
  6. #ifdef _DEBUG
  7. #define new DEBUG_NEW
  8. #undef THIS_FILE
  9. static char THIS_FILE[] = __FILE__;
  10. #endif
  11. /////////////////////////////////////////////////////////////////////////////
  12. // CAboutDlg dialog used for App About
  13. class CAboutDlg : public CDialog
  14. {
  15. public:
  16. CAboutDlg();
  17. // Dialog Data
  18. //{{AFX_DATA(CAboutDlg)
  19. enum { IDD = IDD_ABOUTBOX };
  20. //}}AFX_DATA
  21. // ClassWizard generated virtual function overrides
  22. //{{AFX_VIRTUAL(CAboutDlg)
  23. protected:
  24. virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
  25. //}}AFX_VIRTUAL
  26. // Implementation
  27. protected:
  28. //{{AFX_MSG(CAboutDlg)
  29. //}}AFX_MSG
  30. DECLARE_MESSAGE_MAP()
  31. };
  32. CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
  33. {
  34. //{{AFX_DATA_INIT(CAboutDlg)
  35. //}}AFX_DATA_INIT
  36. }
  37. void CAboutDlg::DoDataExchange(CDataExchange* pDX)
  38. {
  39. CDialog::DoDataExchange(pDX);
  40. //{{AFX_DATA_MAP(CAboutDlg)
  41. //}}AFX_DATA_MAP
  42. }
  43. BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
  44. //{{AFX_MSG_MAP(CAboutDlg)
  45. // No message handlers
  46. //}}AFX_MSG_MAP
  47. END_MESSAGE_MAP()
  48. /////////////////////////////////////////////////////////////////////////////
  49. // CRshDlg dialog
  50. CRshDlg::CRshDlg(CWnd* pParent /*=NULL*/)
  51. : CDialog(CRshDlg::IDD, pParent)
  52. {
  53. //{{AFX_DATA_INIT(CRshDlg)
  54. //}}AFX_DATA_INIT
  55. // Note that LoadIcon does not require a subsequent DestroyIcon in Win32
  56. m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
  57. }
  58. void CRshDlg::DoDataExchange(CDataExchange* pDX)
  59. {
  60. CDialog::DoDataExchange(pDX);
  61. //{{AFX_DATA_MAP(CRshDlg)
  62. DDX_Control(pDX, IDC_COMBO1, m_oid1);
  63. DDX_Control(pDX, IDC_COMBO2, m_type);
  64. DDX_Control(pDX, IDC_COMBO3, m_oid2);
  65. DDX_Control(pDX, IDC_EDIT2, m_interval);
  66. DDX_Control(pDX, IDC_EDIT1, m_community);
  67. DDX_Control(pDX, IDC_IPADDRESS1, m_ipadd);
  68. //}}AFX_DATA_MAP
  69. }
  70. BEGIN_MESSAGE_MAP(CRshDlg, CDialog)
  71. //{{AFX_MSG_MAP(CRshDlg)
  72. ON_WM_SYSCOMMAND()
  73. ON_WM_PAINT()
  74. ON_WM_QUERYDRAGICON()
  75. ON_BN_CLICKED(IDC_BUTTON1, OnScan)
  76. ON_WM_DESTROY()
  77. ON_BN_CLICKED(IDC_BUTTON3, OnStart)
  78. ON_BN_CLICKED(IDC_BUTTON4, OnEnd)
  79. ON_BN_CLICKED(IDC_BUTTON2, OnOK)
  80. ON_WM_TIMER()
  81. //}}AFX_MSG_MAP
  82. END_MESSAGE_MAP()
  83. /////////////////////////////////////////////////////////////////////////////
  84. // CRshDlg message handlers
  85. BOOL CRshDlg::OnInitDialog()
  86. {
  87. CDialog::OnInitDialog();
  88. // Add "About..." menu item to system menu.
  89. // IDM_ABOUTBOX must be in the system command range.
  90. ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
  91. ASSERT(IDM_ABOUTBOX < 0xF000);
  92. CMenu* pSysMenu = GetSystemMenu(FALSE);
  93. if (pSysMenu != NULL)
  94. {
  95. CString strAboutMenu;
  96. strAboutMenu.LoadString(IDS_ABOUTBOX);
  97. if (!strAboutMenu.IsEmpty())
  98. {
  99. pSysMenu->AppendMenu(MF_SEPARATOR);
  100. pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
  101. }
  102. }
  103. // Set the icon for this dialog.  The framework does this automatically
  104. //  when the application's main window is not a dialog
  105. SetIcon(m_hIcon, TRUE); // Set big icon
  106. SetIcon(m_hIcon, FALSE); // Set small icon
  107. // TODO: Add extra initialization here
  108.     CRect cr(22,34,518,150);
  109. CSize c(388,150);
  110. pDay=NULL;
  111. pDay=new CLineChart(c,288);
  112. pDay->Create(NULL,WS_CHILD|WS_VISIBLE|SS_NOTIFY|SS_BLACKFRAME, cr,this,50000);
  113. pDay->SetTitle("实时数据曲线图");
  114. pDay->SetLegend("内存","CPU");
  115.     CButton* m_pB=(CButton*)GetDlgItem(IDC_BUTTON3);
  116.     m_pB->EnableWindow(FALSE);
  117.     m_pB=(CButton*)GetDlgItem(IDC_BUTTON4);
  118.     m_pB->EnableWindow(FALSE);
  119. return TRUE;  // return TRUE  unless you set the focus to a control
  120. }
  121. void CRshDlg::OnSysCommand(UINT nID, LPARAM lParam)
  122. {
  123. if ((nID & 0xFFF0) == IDM_ABOUTBOX)
  124. {
  125. CAboutDlg dlgAbout;
  126. dlgAbout.DoModal();
  127. }
  128. else
  129. {
  130. CDialog::OnSysCommand(nID, lParam);
  131. }
  132. }
  133. // If you add a minimize button to your dialog, you will need the code below
  134. //  to draw the icon.  For MFC applications using the document/view model,
  135. //  this is automatically done for you by the framework.
  136. void CRshDlg::OnPaint() 
  137. {
  138. if (IsIconic())
  139. {
  140. CPaintDC dc(this); // device context for painting
  141. SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);
  142. // Center icon in client rectangle
  143. int cxIcon = GetSystemMetrics(SM_CXICON);
  144. int cyIcon = GetSystemMetrics(SM_CYICON);
  145. CRect rect;
  146. GetClientRect(&rect);
  147. int x = (rect.Width() - cxIcon + 1) / 2;
  148. int y = (rect.Height() - cyIcon + 1) / 2;
  149. // Draw the icon
  150. dc.DrawIcon(x, y, m_hIcon);
  151. }
  152. else
  153. {
  154. CDialog::OnPaint();
  155. }
  156. }
  157. // The system calls this to obtain the cursor to display while the user drags
  158. //  the minimized window.
  159. HCURSOR CRshDlg::OnQueryDragIcon()
  160. {
  161. return (HCURSOR) m_hIcon;
  162. }
  163. void CRshDlg::OnScan() 
  164. {
  165. CString oid;
  166. m_ipadd.GetWindowText(ip);
  167. m_community.GetWindowText(community);
  168. CString type,rv;
  169. m_type.GetWindowText(type);
  170. m_oid1.ResetContent();
  171. m_oid2.ResetContent();
  172. NameToOid.RemoveAll();
  173. if (type=="主机")
  174. oid="1.3.6.1.4.1.9600.1.1.5.1.5.1.48";
  175. m_snmp.GetOne(ip,community,oid,rv);
  176. if (rv.GetLength()>0)
  177. {
  178. m_oid1.AddString("CPU利用率");
  179. NameToOid.SetAt("CPU利用率",oid);
  180. }
  181. oid="1.3.6.1.4.1.9600.1.1.2.1.0";
  182. m_snmp.GetOne(ip,community,oid,rv);
  183. if (rv.GetLength()>0)
  184. {
  185. m_oid2.AddString("可用物理内存(Bytes)");
  186. NameToOid.SetAt("可用物理内存(Bytes)",oid);
  187. }
  188. oid="1.3.6.1.4.1.9600.1.1.2.2.0";
  189. m_snmp.GetOne(ip,community,oid,rv);
  190. if (rv.GetLength()>0)
  191. {
  192. m_oid2.AddString("可用物理内存(KBytes)");
  193. NameToOid.SetAt("可用物理内存(KBytes)",oid);
  194. }
  195. oid="1.3.6.1.4.1.9600.1.1.2.3.0";
  196. m_snmp.GetOne(ip,community,oid,rv);
  197. if (rv.GetLength()>0)
  198. {
  199. m_oid2.AddString("可用物理内存(MBytes)");
  200. NameToOid.SetAt("可用物理内存(MBytes)",oid);
  201. }
  202. }
  203. else
  204. {
  205. oid="1.3.6.1.4.1.9.2.1.56.0";
  206. m_snmp.GetOne(ip,community,oid,rv);
  207. if (rv.GetLength()>0)
  208. {
  209. m_oid1.AddString("CPU 5秒内平均利用率");
  210. NameToOid.SetAt("CPU 5秒内平均利用率",oid);
  211. }
  212. oid="1.3.6.1.4.1.9.2.1.57.0";
  213. m_snmp.GetOne(ip,community,oid,rv);
  214. if (rv.GetLength()>0)
  215. {
  216. m_oid1.AddString("CPU 1分钟内平均利用率");
  217. NameToOid.SetAt("CPU 1分钟内平均利用率",oid);
  218. }
  219. oid="1.3.6.1.4.1.9.2.1.58.0";
  220. m_snmp.GetOne(ip,community,oid,rv);
  221. if (rv.GetLength()>0)
  222. {
  223. m_oid1.AddString("CPU 5分钟内平均利用率");
  224. NameToOid.SetAt("CPU 5分钟内平均利用率",oid);
  225. }
  226. oid="1.3.6.1.4.1.9.9.48.1.1.1.5.2";
  227. m_snmp.GetOne(ip,community,oid,rv);
  228. if (rv.GetLength()>0)
  229. {
  230. m_oid2.AddString("I/O内存(已经使用字节KByts)");
  231. NameToOid.SetAt("I/O内存(已经使用字节KByts)",oid);
  232. }
  233. oid="1.3.6.1.4.1.9.9.48.1.1.1.6.2";
  234. m_snmp.GetOne(ip,community,oid,rv);
  235. if (rv.GetLength()>0)
  236. {
  237. m_oid2.AddString("I/O(内存空闲字节KByts)");
  238. NameToOid.SetAt("I/O(内存空闲字节KByts)",oid);
  239. }
  240. oid="1.3.6.1.4.1.9.9.48.1.1.1.5.4";
  241. m_snmp.GetOne(ip,community,oid,rv);
  242. if (rv.GetLength()>0)
  243. {
  244. m_oid2.AddString("FAST内存(已经使用字节KByts)");
  245. NameToOid.SetAt("FAST内存(已经使用字节KByts)",oid);
  246. }
  247. oid="1.3.6.1.4.1.9.9.48.1.1.1.6.4";
  248. m_snmp.GetOne(ip,community,oid,rv);
  249. if (rv.GetLength()>0)
  250. {
  251. m_oid2.AddString("FAST内存(空闲字节KByts)");
  252. NameToOid.SetAt("FAST内存(空闲字节KByts)",oid);
  253. }
  254. }
  255. MessageBox("操作完成!");
  256. }
  257. void CRshDlg::OnDestroy() 
  258. {
  259. CDialog::OnDestroy();
  260. delete pDay;
  261. }
  262. void CRshDlg::OnOK() 
  263. {
  264. CString temp,t;
  265. m_interval.GetWindowText(t);
  266. if (t.GetLength()==0)
  267. {
  268. MessageBox("采样时间错误!");
  269. return;
  270. }
  271. interval=atoi((LPCTSTR)t);
  272. if (interval<1)
  273.      {
  274. MessageBox("采样时间要大于或等于1分钟!");
  275. return;
  276. }
  277. interval=interval*60000;
  278. m_oid1.GetWindowText(oid1);
  279. m_oid2.GetWindowText(oid2);
  280. if (oid1.Find("1.3.6.1")==-1)
  281. {
  282. NameToOid.Lookup(oid1,temp);
  283. oid1=temp;
  284. }
  285. if (oid2.Find("1.3.6.1")==-1)
  286. {
  287. NameToOid.Lookup(oid2,temp);
  288. oid2=temp;
  289. }
  290.    CButton *m_pB=(CButton*)GetDlgItem(IDC_BUTTON3);
  291.    m_pB->EnableWindow(TRUE); 
  292.    MessageBox("点“开始”按钮开始测试!");
  293. }
  294. void CRshDlg::OnStart() 
  295. {   
  296. pDay->Init();
  297. unsigned long Memory;
  298.     unsigned long CPUUse; 
  299. CTime t=CTime::GetCurrentTime();
  300. CString time=t.Format("%H:%M");
  301. fName=t.Format("%m-%d-%H");
  302. fName=fName+"CPUMemory.txt";
  303. CString s,line;
  304. line="";
  305. DataFile.Open(fName,CFile::modeWrite|CFile::modeCreate);
  306. if (m_snmp.GetTwo(ip,community,oid1,oid2,CPUUse,Memory))
  307. {   
  308. Memory=Memory/1024;
  309. pDay->Append(Memory,CPUUse,time,0,0);
  310. s.Format("%d",CPUUse);
  311. line=line+"@"+s;
  312. s.Format("%d",Memory);
  313. line=line+"#"+s;
  314. line=line+"$0&0rn";
  315. DataFile.SeekToEnd();
  316. DataFile.WriteString(line);
  317. }
  318. else
  319. MessageBox("读取数据失败,请检查OID或网络是否有错误!");
  320. DataFile.Close();
  321. if (interval>=60000)
  322. SetTimer(1,interval,NULL);
  323.  CButton *m_pB=(CButton*)GetDlgItem(IDC_BUTTON4);
  324.      m_pB->EnableWindow(TRUE); 
  325. }
  326. void CRshDlg::OnEnd() 
  327. {
  328. KillTimer(1);
  329. MessageBox("测试结束!");
  330. }
  331. void CRshDlg::OnTimer(UINT nIDEvent) 
  332. {
  333. // TODO: Add your message handler code here and/or call default
  334. unsigned long Memory;
  335.     unsigned long CPUUse;
  336. CTime t=CTime::GetCurrentTime();
  337. CString time=t.Format("%H:%M");
  338. CString s,line;
  339. line="";
  340. DataFile.Open(fName,CFile::modeWrite);
  341. if (m_snmp.GetTwo(ip,community,oid1,oid2,CPUUse,Memory))
  342. {  
  343.         Memory=Memory/1024;
  344. pDay->Append(Memory,CPUUse,time,0,0);
  345. s.Format("%d",CPUUse);
  346. line=line+"@"+s;
  347. s.Format("%d",Memory);
  348. line=line+"#"+s;
  349. line=line+"$0&0rn";
  350. DataFile.SeekToEnd();
  351. DataFile.WriteString(line);
  352. }
  353. DataFile.Close();
  354. CDialog::OnTimer(nIDEvent);
  355. }