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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > 网络截获/分析 > 查看源码
mftrafficbutton.cpp
资源名称：jiankong1.rar　[点击查看]
上传用户：fangbo2004
上传日期：2022-05-14
资源大小：2257k
文件大小：14k
源码类别：

网络截获/分析

开发平台：

Visual C++

mftrafficbutton.cpp：源码内容
  1. #include "stdafx.h"
  2. #include "MFTrafficButton.h"
  3. #include "MemDC.h"
  4. #include <math.h>
  6. #ifndef _WIN32_IE
  7. #define _WIN32_IE 0x0500
  8. #endif
  12. #ifdef _DEBUG
  13. #define new DEBUG_NEW
  14. #undef THIS_FILE
  15. static char THIS_FILE[] = __FILE__;
  16. #endif
  19. /////////////////////////////////////////////////////////////////////////////
  20. // MFTrafficButton
  22. MFTrafficButton::MFTrafficButton()
  23. {
  24. brushInitalized = FALSE;
  25. interfaceCallBack = NULL;
  26. useAdaptiveScale = FALSE;
  27. gridxstartpos = 0;
  28. gridystartpos = 0;
  29. gridxresolution = GRIDXRESOLUTION;
  30. gridyresolution = GRIDYRESOLUTION;
  31. gridscrollxspeed = GRIDSCROLLXSPEED;
  32. gridscrollyspeed = GRIDSCROLLYSPEED; 
  33. plotgranularity = PLOTGRANULATRITY;
  34. netupdatespeed = NETUPDATESPEED;
  35. gridupdatespeed = GRIDUPDATESPEED;
  36. ToggleStatusText = "局部最大值";
  37. }
  39. MFTrafficButton::~MFTrafficButton()
  40. {
  42. }
  45. BEGIN_MESSAGE_MAP(MFTrafficButton, CButton)
  46. //{{AFX_MSG_MAP(MFTrafficButton)
  47. ON_WM_TIMER()
  48. ON_WM_RBUTTONDOWN()
  49. //}}AFX_MSG_MAP
  50. END_MESSAGE_MAP()
  52. /////////////////////////////////////////////////////////////////////////////
  53. // MFTrafficButton message handlers
  57. void MFTrafficButton::PreSubclassWindow() 
  58. {
  60. CButton::PreSubclassWindow();
  61. // 这个函数确保这个button是用来绘制的而不是接收按下的消息的
  62. ModifyStyle( 0 , BS_OWNERDRAW );
  64. this->GetWindowRect(&TrafficDrawRectangle);
  65. this->GetWindowRect(&TrafficDrawUpdateRectangle);
  66. ScreenToClient(&TrafficDrawUpdateRectangle);
  67. ScreenToClient(&TrafficDrawRectangle);
  69. TGSize.cx = TrafficDrawRectangle.right - TrafficDrawRectangle.left;
  70. TGSize.cy = TrafficDrawRectangle.bottom - TrafficDrawRectangle.top;
  72. initalized = FALSE;
  73. MaxTrafficAmount = 0.0;
  74. SelectedInterface = 0;
  76. smallFont.CreateFont(-10,0,0,0,FW_THIN, FALSE, FALSE, FALSE, DEFAULT_CHARSET,OUT_DEFAULT_PRECIS, CLIP_DEFAULT_PRECIS, DEFAULT_QUALITY, VARIABLE_PITCH, "Times New Roman");
  78. red = RGB(255,125,125);
  79. green = RGB(125,255,125);
  80. cyan = RGB(0,255,255);
  81. darkblue= RGB(0,0,75);
  82. // darkblue= RGB(255,255,255);
  83. darkgray= RGB(50,50,50);
  84. white = RGB(255,255,255);
  85. black = RGB(0,0,0);
  86. lightgreen = RGB(156,255,156);
  87. darkgreen = RGB(32,64,32);
  88. // darkgreen    =0;
  89. greenbrush.CreateSolidBrush(green);
  90. redbrush.CreateSolidBrush(red);
  91. GridPen.CreatePen(PS_SOLID ,1 , darkgreen);
  92. GetWindowRect(&TrafficDrawRectangle);
  93. ScreenToClient(&TrafficDrawRectangle);
  96. TGSize.cx = TrafficDrawRectangle.right - TrafficDrawRectangle.left;
  97. TGSize.cy = TrafficDrawRectangle.bottom - TrafficDrawRectangle.top;
  100. TrafficEntries = TGSize.cx / plotgranularity;
  101. TrafficStats = new TRAFFICENTRY[TrafficEntries+1];
  103. for(DWORD x=0; x<TrafficEntries; x++)
  104. {
  105. TrafficStats[x].connected = TRUE;//FALSE;
  106. TrafficStats[x].value = 0.0;//(float)x * (float)TrafficEntries*0.05;//;10.0 + 10.0*(sin(5.0*(float)x*3.14/180.0));
  107. }
  109. MaxTrafficAmount = 0.0;
  110. SelectedInterface = 0;
  112. CRgn rectRgn, ellRgn, finalRgn;
  113. rectRgn.CreateRectRgn(0,0,TGSize.cx, TGSize.cy);
  114. ShapeWNDRegion.CreateRectRgn(0,0,TGSize.cx, TGSize.cy);;
  115. ShapeDCRegion.CreateRectRgn(0,0,TGSize.cx, TGSize.cy);;
  117. int x1,x2,y1,y2,xe,ye,xs,ys;
  118. int xof, yof;
  119. int r;
  120. xs = TGSize.cx;
  121. ys = TGSize.cy;
  122. x1 = 0;
  123. y1 = 0;
  124. x2 = xs;
  125. y2 = ys;
  126. xe = 5;//Radius of edge
  127. ye = 5;//Radius of edge
  128. xof = (int)( (float)xs*0.0);
  129. yof = (int)( (float)ys*0.0);
  130. r = ellRgn.CreateRoundRectRgn(x1,y1,x2,y2,xe,ye);
  131. r = ellRgn.OffsetRgn(-xof, -yof);
  132. r = ShapeWNDRegion.CombineRgn(&rectRgn, &ellRgn,RGN_AND );
  133. r = ShapeDCRegion.CombineRgn(&rectRgn, &ellRgn,RGN_AND );
  135. this->SetWindowRgn((HRGN)ShapeWNDRegion, TRUE);
  136. initalized = TRUE;
  138. SetTimer(GRIDTIMER, gridupdatespeed, 0);
  139. SetTimer(NETTIMER, netupdatespeed, 0);
  141. }
  144. void MFTrafficButton::DrawItem(LPDRAWITEMSTRUCT lpDrawItemStruct) 
  145. {
  146. // TODO: Add your message handler code here and/or call default
  147. // CButton::OnDrawItem(nIDCtl, lpDrawItemStruct);
  149. ASSERT( lpDrawItemStruct != NULL );
  151. CDC * pDC   = CDC::FromHandle( lpDrawItemStruct -> hDC );
  152. int erg = pDC->SelectClipRgn(&ShapeDCRegion);
  154. CRect rect = lpDrawItemStruct -> rcItem;
  155. UINT state = lpDrawItemStruct -> itemState;
  156. UINT nStyle = GetStyle( );
  158. int nSavedDC = pDC -> SaveDC( );
  160. // Create the brush for the color bar
  161. if(brushInitalized == FALSE)
  162. {
  163. CBitmap bmp;
  164. CMemDC *memDC = new CMemDC(pDC);
  166. RECT clipRect;
  167. memDC->GetClipBox(&clipRect);
  169. if(clipRect.right - clipRect.left > 1)
  170. {
  171. bmp.CreateCompatibleBitmap(memDC,plotgranularity, TGSize.cy);
  172. CBitmap *pOld = memDC->SelectObject(&bmp);
  174. CSize bmps = bmp.GetBitmapDimension();
  176. double factor = 255.0 / (float)TGSize.cy;
  177. BYTE r,g,b;
  178. for(int x = 0; x<TGSize.cy; x++)
  179. {
  180. g = (BYTE)(255-factor*x);
  181. r = (BYTE)(factor*x);
  182. b = (BYTE)64;
  183. memDC->SetPixelV(0,x,RGB(r,g,b));
  184. memDC->SetPixelV(1,x,RGB(r,g,b));
  185. }
  186. memDC->SelectObject(pOld);
  188. colorbrush.CreatePatternBrush(&bmp);
  189. brushInitalized = TRUE;
  190. }
  191. }
  192. if(initalized == TRUE)
  193. {
  194. COLORREF backcolor = GetSysColor(COLOR_BTNFACE);
  196. CBrush brush;
  197. CMemDC *memDC = new CMemDC(pDC);
  199. RECT clipRect;
  200. memDC->GetClipBox(&clipRect);
  201. memDC->FillSolidRect(&clipRect,backcolor);
  203. CFont *oldFont;
  204. int xp, yp, xx, yy;
  205. orgBrushOrigin = memDC->GetBrushOrg();
  207. oldFont = memDC->SelectObject(&smallFont);
  209. double scale = (double)TGSize.cy / (double)MaxTrafficAmount;
  211. yp = TrafficDrawRectangle.bottom;
  212. xp = TrafficDrawRectangle.left;
  214. RECT fillrect;
  216. CString tmp;
  218. // 填充背景
  219. back = memDC->GetBkColor();
  220. brush.CreateSolidBrush(darkblue);
  221. memDC->FillRect(&TrafficDrawRectangle, &brush);
  223. // 画网格
  224. int xgridlines, ygridlines;
  226. xgridlines = TGSize.cx / gridxresolution;
  227. ygridlines = TGSize.cy / gridyresolution;
  228. CPen* oldPen = memDC->SelectObject(&GridPen);
  229. // 创建垂直线
  230. for (int x=0; x<= xgridlines; x++)
  231. {
  232. memDC->MoveTo(x*gridxresolution + gridxstartpos, 0 );
  233. memDC->LineTo(x*gridxresolution + gridxstartpos, TGSize.cy );
  234. }
  235. // 添加水平线
  236. for (int y=0; y<= ygridlines; y++)
  237. {
  238. memDC->MoveTo(0 , gridystartpos + TGSize.cy - y*gridyresolution - 2);
  239. memDC->LineTo(TGSize.cx , gridystartpos + TGSize.cy - y*gridyresolution - 2);
  240. }
  242. gridxstartpos += gridscrollxspeed;
  243. gridystartpos += gridscrollyspeed;
  244. if(gridxstartpos < 0 ) gridxstartpos = gridxresolution;
  245. if(gridxstartpos > gridxresolution ) gridxstartpos = 0;
  246. if(gridystartpos < 0 ) gridystartpos = gridyresolution;
  247. if(gridystartpos > gridyresolution ) gridystartpos = 0;
  250. memDC->SelectObject(oldPen );
  252. for(DWORD cnt=0; cnt<TrafficEntries; cnt++)
  253. {
  254. xx = xp + cnt*plotgranularity;
  255. double traffic = (double)TrafficStats[cnt].value; 
  256. yy = yp - (int)((double)TrafficStats[cnt].value * scale);
  258. // 网络处在连接状态才绘制
  259. if(TrafficStats[cnt].connected == TRUE)
  260. {
  261. fillrect.bottom = yp;
  262. fillrect.top = yy;
  263. fillrect.left = xx;
  264. fillrect.right = xx+plotgranularity;
  265. memDC->SetBrushOrg(xx,yp);
  266. if(TrafficStats[cnt].value > 0.0) 
  267. {
  268. memDC->FillRect(&fillrect, &colorbrush);
  269. memDC->SetPixelV(xx, yy, cyan);
  270. }
  271. }
  272. }
  274. tmp.Format("%8.1f",TrafficStats[TrafficEntries-1].value);
  275. COLORREF textcolor = memDC->GetTextColor();
  276. int bkmode = memDC->GetBkMode();
  277. memDC->SetBkMode(TRANSPARENT);
  278. memDC->SetTextColor(darkblue);
  279. memDC->TextOut(6,5,AllTraffic);
  280. memDC->SetTextColor(cyan);
  281. // memDC->SetTextColor(darkblue);
  282. memDC->TextOut(5,5,AllTraffic); 
  283. memDC->SetTextColor(textcolor);
  284. memDC->SetBkMode(bkmode);
  286. memDC->SelectObject(oldFont);
  287. memDC->SetBrushOrg(orgBrushOrigin.x, orgBrushOrigin.y);
  289. delete memDC;
  290. }
  291. pDC -> RestoreDC( nSavedDC );
  293. }
  295. void MFTrafficButton::OnTimer(UINT nIDEvent) 
  296. {
  297. // TODO: Add your message handler code here and/or call default
  298. if(nIDEvent == NETTIMER)
  299. {
  301. #ifdef _I_HAVE_PLATFORM_SDK_INSTALLED_
  303. DWORD flag, reserved;
  304. BOOL erg;
  305. flag = 0;//INTERNET_CONNECTION_OFFLINE ;
  306. reserved = 0;
  307. TCHAR connectionname[1024];
  308. erg =  InternetGetConnectedStateEx(
  309. &flag, //OUT LPDWORD lpdwFlags,
  310. (LPTSTR)&connectionname,//OUT LPTSTR lpszConnectionName,
  311. 1024,//IN DWORD dwNameLen,
  312. 0//IN DWORD dwReserved
  313. );
  315. isOnline = erg;
  316. #else
  317. isOnline = TRUE;
  318. #endif
  320. // 得到当前的流量
  321. double traffic = m_cTrafficClass.GetTraffic(SelectedInterface);
  322. DWORD  totaltraffic = m_cTrafficClass.GetInterfaceTotalTraffic(SelectedInterface);
  324. double delta1;
  325. double divisor = (1000.0/(double)NETUPDATESPEED);
  326. delta1 = (double)(traffic * divisor) / 1024.0;
  327. CurrentTraffic.Format("当前流量: %.1f KB/sec",delta1);
  330. // Should we recalculate the local maximum per session or per display?
  331. if(useAdaptiveScale==TRUE)
  332. {
  333. MaxTrafficAmount = 0.0;
  334. }
  335. // Shift whole array 1 step to left and calculate local maximum
  336. for(DWORD x=0; x<TrafficEntries; x++)
  337. {
  338. TrafficStats[x].connected = TrafficStats[x+1].connected;
  339. TrafficStats[x].value = TrafficStats[x+1].value;
  340. if(TrafficStats[x].value > MaxTrafficAmount)
  341. MaxTrafficAmount = TrafficStats[x].value;
  342. }
  343. if(isOnline == TRUE)
  344. {
  345. TrafficStats[TrafficEntries].connected = TRUE;
  346. TrafficStats[TrafficEntries].value = traffic;
  347. if(TrafficStats[TrafficEntries].value > MaxTrafficAmount)
  348. MaxTrafficAmount = TrafficStats[TrafficEntries].value;
  349. }
  350. else
  351. {
  352. TrafficStats[TrafficEntries].connected = FALSE;
  353. TrafficStats[TrafficEntries].value = traffic;
  354. if(TrafficStats[TrafficEntries].value > MaxTrafficAmount)
  355. MaxTrafficAmount = TrafficStats[TrafficEntries].value;
  356. }
  358. double delta2;
  359. divisor = (1000.0/(double)NETUPDATESPEED);
  360. delta2 = (double)(MaxTrafficAmount * divisor) / 1024.0;
  361. MaximalTraffic.Format("最大流量:%.1f KB/sec",delta2);
  362. AllTraffic.Format("总流量为: %.1f / %.1f KB/sec",delta1, delta2);
  364. }
  366. // 绘制
  367. Invalidate(FALSE);
  369. CButton::OnTimer(nIDEvent);
  370. }
  372. //创建右键按下后产生的一个pop-up菜单
  373. void MFTrafficButton::OnRButtonDown(UINT nFlags, CPoint point) 
  374. {
  375. // TODO: Add your message handler code here and/or call default
  376. #define STARTINDEX 2000
  377. #define ENDINDEX 2999
  378. #define SET_SCALING 3000
  380. HMENU hMenu , hPopup;
  381. long retId = 0;
  382. int InterfaceIndex = STARTINDEX;
  383. POINT pt;
  385. hMenu = CreatePopupMenu( );   
  386. if( hMenu )
  387. {
  388. //  If the menu was created, add the item text
  389. CString tmp, tmp2;
  390. double tottraff = (double)m_cTrafficClass.GetInterfaceTotalTraffic(SelectedInterface) / (1024.0*1024.0);
  391. m_cTrafficClass.GetNetworkInterfaceName(&tmp, SelectedInterface);
  392. tmp2.Format("接口名称:%s : %.1f MB", tmp, tottraff);
  394. AppendMenu( hMenu , MF_STRING , 1001 , tmp2 );
  395. AppendMenu( hMenu , MF_SEPARATOR , 1000 , NULL );
  396. // Submenu
  397. hPopup = CreatePopupMenu( );   //  Create a new popup menu
  398. if( hPopup )
  399. {
  400. CString tmp;
  401. int cnt;
  402. for(cnt=0; cnt<=m_cTrafficClass.GetNetworkInterfacesCount(); cnt++)
  403. {
  404. CString tmp2;
  405. double tottraff = (double)m_cTrafficClass.GetInterfaceTotalTraffic(cnt) / (1024.0*1024.0);
  406. m_cTrafficClass.GetNetworkInterfaceName(&tmp, cnt);
  408. tmp2.Format("%s : %.1f MB", tmp, tottraff);
  409. AppendMenu( hPopup , MF_STRING , InterfaceIndex++ , tmp2 );
  410. }
  411. AppendMenu( hMenu , MF_POPUP , ( UINT )hPopup , "选择一个接口" );
  412. AppendMenu( hMenu , MF_SEPARATOR , 1000 , NULL );
  413. AppendMenu( hMenu , MF_STRING , SET_SCALING , "锁定的缩放比例:" +  ToggleStatusText);
  414. }
  415. }
  417. GetCursorPos( & pt );
  418. //显示菜单
  419. retId = TrackPopupMenu( hMenu , TPM_LEFTALIGN | TPM_RIGHTBUTTON | TPM_RETURNCMD ,  pt.x , pt.y , 0 , this -> m_hWnd , NULL );  //  start the menu
  420. DestroyMenu( hMenu );
  421. if( retId == 0 )
  422. {
  423. return;
  424. }
  425. else
  426. {
  427. if(retId == SET_SCALING)
  428. {
  429. if(useAdaptiveScale == TRUE)
  430. {
  431. useAdaptiveScale = FALSE;
  432. ToggleStatusText = "全局最大值";
  433. }
  434. else
  435. {
  436. useAdaptiveScale = TRUE;
  437. ToggleStatusText = "局部最大值";
  438. }
  439. }
  440. if((retId>=STARTINDEX) && (retId<ENDINDEX) )
  441. {
  442. SelectedInterface = retId-STARTINDEX;
  443. InterfaceHasChanged();
  444. ReInit(SelectedInterface);
  445. }
  446. }
  448. CButton::OnRButtonDown(nFlags, point);
  449. }
  451. void MFTrafficButton::ReInit(RECT newSize)
  452. {
  453. KillTimer(GRIDTIMER);
  454. KillTimer(NETTIMER);
  456. delete [] TrafficStats;
  458. TGSize.cx = newSize.right - newSize.left;
  459. TGSize.cy = newSize.bottom - newSize.top;
  461. TrafficEntries = TGSize.cx / plotgranularity;
  462. TrafficStats = new TRAFFICENTRY[TrafficEntries+1];
  464. MaxTrafficAmount = 0.0;
  465. SelectedInterface = 0;
  467. for(DWORD x=0; x<TrafficEntries; x++)
  468. {
  469. TrafficStats[x].connected = TRUE;
  470. TrafficStats[x].value = 0.0;
  471. }
  472. m_cTrafficClass.GetTraffic(SelectedInterface);
  473. m_cTrafficClass.GetTraffic(SelectedInterface);
  474. MaxTrafficAmount = 0.0;
  475. SetTimer(GRIDTIMER, gridupdatespeed, 0);
  476. SetTimer(NETTIMER, netupdatespeed, 0);
  478. }
  480. //监视其他接口
  481. void MFTrafficButton::ReInit(int newInterface)
  482. {
  483. KillTimer(GRIDTIMER);
  484. KillTimer(NETTIMER);
  486. SelectedInterface = newInterface;
  488. for(DWORD x=0; x<TrafficEntries; x++)
  489. {
  490. TrafficStats[x].connected = TRUE;
  491. TrafficStats[x].value = 0.0;
  492. }
  493. m_cTrafficClass.GetTraffic(SelectedInterface);
  494. m_cTrafficClass.GetTraffic(SelectedInterface);
  495. MaxTrafficAmount = 0.0;
  496. SetTimer(GRIDTIMER, gridupdatespeed, 0);
  497. SetTimer(NETTIMER, netupdatespeed, 0);
  499. }
  501. //设置更新速度
  502. void MFTrafficButton::SetUpdateSpeed(UINT netspeed, UINT gridspeed)
  503. {
  504. gridupdatespeed = gridspeed;
  505. netupdatespeed = netspeed;
  506. KillTimer(GRIDTIMER);
  507. KillTimer(NETTIMER);
  508. SetTimer(GRIDTIMER, gridupdatespeed, 0);
  509. SetTimer(NETTIMER, netupdatespeed, 0);
  510. }
  513. //回调函数
  514. void MFTrafficButton::InterfaceHasChanged()
  515. {
  516. if(interfaceCallBack !=NULL)
  517. {
  518. interfaceCallBack(SelectedInterface);
  519. }
  520. }
  522. void MFTrafficButton::SetInterfaceNumberNotificationFunction(INTERFACECHANCEDPROC callfunct)
  523. {
  524. interfaceCallBack = callfunct;
  525. }
  527. void MFTrafficButton::SelectTrafficType(int trafficType)
  528. {
  529. switch(trafficType)
  530. {
  531. case Traffic_Incoming:  m_cTrafficClass.SetTrafficType(MFNetTraffic::IncomingTraffic); break;
  532. case Traffic_Outgoing: m_cTrafficClass.SetTrafficType(MFNetTraffic::OutGoingTraffic); break;
  533. case Traffic_Total: m_cTrafficClass.SetTrafficType(MFNetTraffic::AllTraffic); break;
  534. default: m_cTrafficClass.SetTrafficType(MFNetTraffic::AllTraffic);
  535. }
  536. }
  537. void MFTrafficButton::SetInterfaceNumber(int interfacenumber)
  538. {
  539. SelectedInterface = interfacenumber;
  540. ReInit(SelectedInterface);
  541. }
  543. void MFTrafficButton::SetAdaptiveScaling(BOOL adaptive)
  544. {
  545. useAdaptiveScale = adaptive;
  546. }