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

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

网络截获/分析

开发平台：

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
  319. // 得到当前的输入流量
  320. // m_cTrafficClass.SetTrafficType(IncomingTraffic);
  321. // m_cTrafficClass.SetTrafficType(OutGoingTraffic);
  322. //m_cTrafficClass.SetTrafficType(0);
  324. double traffic = m_cTrafficClass.GetTraffic(SelectedInterface);
  325. DWORD  totaltraffic = m_cTrafficClass.GetInterfaceTotalTraffic(SelectedInterface);
  327. double delta1;
  328. double divisor = (1000.0/(double)NETUPDATESPEED);
  329. delta1 = (double)(traffic * divisor) / 1024.0;
  330. CurrentTraffic.Format("当前流量: %.1f KB/sec",delta1);
  333. // Should we recalculate the local maximum per session or per display?
  334. if(useAdaptiveScale==TRUE)
  335. {
  336. MaxTrafficAmount = 0.0;
  337. }
  338. // Shift whole array 1 step to left and calculate local maximum
  339. for(DWORD x=0; x<TrafficEntries; x++)
  340. {
  341. TrafficStats[x].connected = TrafficStats[x+1].connected;
  342. TrafficStats[x].value = TrafficStats[x+1].value;
  343. if(TrafficStats[x].value > MaxTrafficAmount)
  344. MaxTrafficAmount = TrafficStats[x].value;
  345. }
  346. if(isOnline == TRUE)
  347. {
  348. TrafficStats[TrafficEntries].connected = TRUE;
  349. TrafficStats[TrafficEntries].value = traffic;
  350. if(TrafficStats[TrafficEntries].value > MaxTrafficAmount)
  351. MaxTrafficAmount = TrafficStats[TrafficEntries].value;
  352. }
  353. else
  354. {
  355. TrafficStats[TrafficEntries].connected = FALSE;
  356. TrafficStats[TrafficEntries].value = traffic;
  357. if(TrafficStats[TrafficEntries].value > MaxTrafficAmount)
  358. MaxTrafficAmount = TrafficStats[TrafficEntries].value;
  359. }
  361. double delta2;
  362. divisor = (1000.0/(double)NETUPDATESPEED);
  363. delta2 = (double)(MaxTrafficAmount * divisor) / 1024.0;
  364. MaximalTraffic.Format("最大流量:%.1f KB/sec",delta2);
  365. AllTraffic.Format("流量为: %.1f / %.1f KB/sec",delta1, delta2);
  367. }
  369. // 绘制
  370. Invalidate(FALSE);
  372. CButton::OnTimer(nIDEvent);
  373. }
  375. //创建右键按下后产生的一个pop-up菜单
  376. void MFTrafficButton::OnRButtonDown(UINT nFlags, CPoint point) 
  377. {
  378. // TODO: Add your message handler code here and/or call default
  379. #define STARTINDEX 2000
  380. #define ENDINDEX 2999
  381. #define SET_SCALING 3000
  383. HMENU hMenu , hPopup;
  384. long retId = 0;
  385. int InterfaceIndex = STARTINDEX;
  386. POINT pt;
  388. hMenu = CreatePopupMenu( );   
  389. if( hMenu )
  390. {
  391. //  If the menu was created, add the item text
  392. CString tmp, tmp2;
  393. double tottraff = (double)m_cTrafficClass.GetInterfaceTotalTraffic(SelectedInterface) / (1024.0*1024.0);
  394. m_cTrafficClass.GetNetworkInterfaceName(&tmp, SelectedInterface);
  395. tmp2.Format("接口名称:%s : %.1f MB", tmp, tottraff);
  397. AppendMenu( hMenu , MF_STRING , 1001 , tmp2 );
  398. AppendMenu( hMenu , MF_SEPARATOR , 1000 , NULL );
  399. // Submenu
  400. hPopup = CreatePopupMenu( );   //  Create a new popup menu
  401. if( hPopup )
  402. {
  403. CString tmp;
  404. int cnt;
  405. for(cnt=0; cnt<=m_cTrafficClass.GetNetworkInterfacesCount(); cnt++)
  406. {
  407. CString tmp2;
  408. double tottraff = (double)m_cTrafficClass.GetInterfaceTotalTraffic(cnt) / (1024.0*1024.0);
  409. m_cTrafficClass.GetNetworkInterfaceName(&tmp, cnt);
  411. tmp2.Format("%s : %.1f MB", tmp, tottraff);
  412. AppendMenu( hPopup , MF_STRING , InterfaceIndex++ , tmp2 );
  413. }
  414. AppendMenu( hMenu , MF_POPUP , ( UINT )hPopup , "选择一个接口" );
  415. AppendMenu( hMenu , MF_SEPARATOR , 1000 , NULL );
  416. AppendMenu( hMenu , MF_STRING , SET_SCALING , "锁定的缩放比例:" +  ToggleStatusText);
  417. }
  418. }
  420. GetCursorPos( & pt );
  421. //显示菜单
  422. retId = TrackPopupMenu( hMenu , TPM_LEFTALIGN | TPM_RIGHTBUTTON | TPM_RETURNCMD ,  pt.x , pt.y , 0 , this -> m_hWnd , NULL );  //  start the menu
  423. DestroyMenu( hMenu );
  424. if( retId == 0 )
  425. {
  426. return;
  427. }
  428. else
  429. {
  430. if(retId == SET_SCALING)
  431. {
  432. if(useAdaptiveScale == TRUE)
  433. {
  434. useAdaptiveScale = FALSE;
  435. ToggleStatusText = "全局最大值";
  436. }
  437. else
  438. {
  439. useAdaptiveScale = TRUE;
  440. ToggleStatusText = "局部最大值";
  441. }
  442. }
  443. if((retId>=STARTINDEX) && (retId<ENDINDEX) )
  444. {
  445. SelectedInterface = retId-STARTINDEX;
  446. InterfaceHasChanged();
  447. ReInit(SelectedInterface);
  448. }
  449. }
  451. CButton::OnRButtonDown(nFlags, point);
  452. }
  454. void MFTrafficButton::ReInit(RECT newSize)
  455. {
  456. KillTimer(GRIDTIMER);
  457. KillTimer(NETTIMER);
  459. delete [] TrafficStats;
  461. TGSize.cx = newSize.right - newSize.left;
  462. TGSize.cy = newSize.bottom - newSize.top;
  464. TrafficEntries = TGSize.cx / plotgranularity;
  465. TrafficStats = new TRAFFICENTRY[TrafficEntries+1];
  467. MaxTrafficAmount = 0.0;
  468. SelectedInterface = 0;
  470. for(DWORD x=0; x<TrafficEntries; x++)
  471. {
  472. TrafficStats[x].connected = TRUE;
  473. TrafficStats[x].value = 0.0;
  474. }
  475. m_cTrafficClass.GetTraffic(SelectedInterface);
  476. m_cTrafficClass.GetTraffic(SelectedInterface);
  477. MaxTrafficAmount = 0.0;
  478. SetTimer(GRIDTIMER, gridupdatespeed, 0);
  479. SetTimer(NETTIMER, netupdatespeed, 0);
  481. }
  483. //监视其他接口
  484. void MFTrafficButton::ReInit(int newInterface)
  485. {
  486. KillTimer(GRIDTIMER);
  487. KillTimer(NETTIMER);
  489. SelectedInterface = newInterface;
  491. for(DWORD x=0; x<TrafficEntries; x++)
  492. {
  493. TrafficStats[x].connected = TRUE;
  494. TrafficStats[x].value = 0.0;
  495. }
  496. m_cTrafficClass.GetTraffic(SelectedInterface);
  497. m_cTrafficClass.GetTraffic(SelectedInterface);
  498. MaxTrafficAmount = 0.0;
  499. SetTimer(GRIDTIMER, gridupdatespeed, 0);
  500. SetTimer(NETTIMER, netupdatespeed, 0);
  502. }
  504. //设置更新速度
  505. void MFTrafficButton::SetUpdateSpeed(UINT netspeed, UINT gridspeed)
  506. {
  507. gridupdatespeed = gridspeed;
  508. netupdatespeed = netspeed;
  509. KillTimer(GRIDTIMER);
  510. KillTimer(NETTIMER);
  511. SetTimer(GRIDTIMER, gridupdatespeed, 0);
  512. SetTimer(NETTIMER, netupdatespeed, 0);
  513. }
  516. //回调函数
  517. void MFTrafficButton::InterfaceHasChanged()
  518. {
  519. if(interfaceCallBack !=NULL)
  520. {
  521. interfaceCallBack(SelectedInterface);
  522. }
  523. }
  525. void MFTrafficButton::SetInterfaceNumberNotificationFunction(INTERFACECHANCEDPROC callfunct)
  526. {
  527. interfaceCallBack = callfunct;
  528. }
  530. void MFTrafficButton::SelectTrafficType(int trafficType)
  531. {
  532. switch(trafficType)
  533. {
  534. case Traffic_Incoming:  m_cTrafficClass.SetTrafficType(MFNetTraffic::IncomingTraffic); break;
  535. case Traffic_Outgoing: m_cTrafficClass.SetTrafficType(MFNetTraffic::OutGoingTraffic); break;
  536. case Traffic_Total: m_cTrafficClass.SetTrafficType(MFNetTraffic::AllTraffic); break;
  537. default: m_cTrafficClass.SetTrafficType(MFNetTraffic::AllTraffic);
  538. }
  539. }
  540. void MFTrafficButton::SetInterfaceNumber(int interfacenumber)
  541. {
  542. SelectedInterface = interfacenumber;
  543. ReInit(SelectedInterface);
  544. }
  546. void MFTrafficButton::SetAdaptiveScaling(BOOL adaptive)
  547. {
  548. useAdaptiveScale = adaptive;
  549. }