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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > P2P编程 > 查看源码
PeersListCtrl.cpp
资源名称：GGBT.rar　[点击查看]
上传用户：lds876
上传日期：2013-05-25
资源大小：567k
文件大小：24k
源码类别：

P2P编程

开发平台：

Visual C++

PeersListCtrl.cpp：源码内容
  1. // PeersListCtrl.cpp : implementation file
  2. //
  4. #include "stdafx.h"
  5. #include "testbt.h"
  6. #include "PeersListCtrl.h"
  8. #include "FileDB.h"
  9. #include "download.h"
  10. #include "CategoryBarWnd.h"
  11. #include "FileBase.h"
  12. #include "SortClass.h"
  13. #include "MonitorWnd.h"
  15. #ifdef _DEBUG
  16. #define new DEBUG_NEW
  17. #undef THIS_FILE
  18. static char THIS_FILE[] = __FILE__;
  19. #endif
  22. #define ID_MENUITEM_PAUSE_PEERS 52834
  23. #define ID_MENUITEM_CONTINUE_PEERS ID_MENUITEM_PAUSE_PEERS+1
  24. #define ID_MENUITEM_STOP_PEERS ID_MENUITEM_PAUSE_PEERS+2
  25. #define ID_MENUITEM_SHOW_PEER_INFO ID_MENUITEM_PAUSE_PEERS+3
  27. /////////////////////////////////////////////////////////////////////////////
  28. // CPeersListCtrl
  30. CPeersListCtrl::CPeersListCtrl()
  31. :CListCtrlBaseEx("peerslistctrl", true)
  32. {
  33. m_pFileDBItem = 0;
  34. m_pSetupRegister = 0;
  35. m_pMonitorWnd = 0;
  37. //
  38. // column infos.
  39. //
  40. CString m_strArrToolTip[ePeersColumnCount];
  41. CString strArrPeersFieldNames[ePeersColumnCount];
  42. bool bColumnNumeric[ePeersColumnCount];
  44. m_strArrToolTip[eIcon] = "箭头向右:正在与该客户端建立连接rn箭头向上:正在向该客户端传送数据rn箭头向下:正在从该客户端下载数据rn";
  45. m_strArrToolTip[eIP] = "客户端的IP地址";
  46. m_strArrToolTip[eLocale] = "本地主动连接/远程主动连接";
  47. m_strArrToolTip[eDownRate] = "从客户端下载的速度";
  48. m_strArrToolTip[eUpRate] = "上传给客户端的速度";
  49. m_strArrToolTip[eDownCount] = "已经从客户端下载的数据量";
  50. m_strArrToolTip[eUpCount] = "已经上传给客户端的数据量";
  51. m_strArrToolTip[eHave] = "在客户端的机器上已经完成下载的比例";
  52. m_strArrToolTip[eInterested] = "你是否拥有客户端需要的数据";
  53. m_strArrToolTip[eChoking] = "你是否拒绝传输数据给客户端";
  54. m_strArrToolTip[eInteresting] = "客户端是否有你需要的数据";
  55. m_strArrToolTip[eChocked] = "客户端是否拒绝出数数据给你";
  57. strArrPeersFieldNames[eIcon] = "状态";
  58. strArrPeersFieldNames[eIP] = "IP";
  59. strArrPeersFieldNames[eLocale] = "连出/连入";
  60. strArrPeersFieldNames[eDownRate] = "下载速度";
  61. strArrPeersFieldNames[eUpRate] = "上传速度";
  62. strArrPeersFieldNames[eDownCount] = "已下载";
  63. strArrPeersFieldNames[eUpCount] = "已上传";
  64. strArrPeersFieldNames[eHave] = "对方已下载";
  65. strArrPeersFieldNames[eInterested] = "你是否有对方需要的数据";
  66. strArrPeersFieldNames[eChoking] = "你是否暂时拒绝传输数据给对方对方";
  67. strArrPeersFieldNames[eInteresting] = "对方是否有你需要的数据";
  68. strArrPeersFieldNames[eChocked] = "对方是否暂时拒绝传输数据给你";
  70. for (int i=0; i<ePeersColumnCount; i++)
  71. bColumnNumeric[i] = true;
  73. bColumnNumeric[eIcon] = false;
  74. bColumnNumeric[eIP] = false;
  75. bColumnNumeric[eDownRate] = true;
  76. bColumnNumeric[eUpRate] = true;
  77. bColumnNumeric[eDownCount] = true;
  78. bColumnNumeric[eUpCount] = true;
  79. bColumnNumeric[eHave] = false;
  81. for (i=0; i<ePeersColumnCount; i++)
  82. AddColumn(strArrPeersFieldNames[i], bColumnNumeric[i], 100, m_strArrToolTip[i]);
  84. // sort .
  86. m_iSortedCol = 0; // 
  87. // m_iSortedCol = -1;
  88. // m_bSortAscending = true;
  90. m_arrColType[eIcon] = CSortClass::dtSTRING;
  91. m_arrColType[eIP] = CSortClass::dtSTRING;
  92. m_arrColType[eLocale] = CSortClass::dtSTRING;
  93. m_arrColType[eDownRate] = CSortClass::dtSTRING;
  94. m_arrColType[eUpRate] = CSortClass::dtSTRING;
  95. m_arrColType[eDownCount] = CSortClass::dtSTRING;
  96. m_arrColType[eUpCount] = CSortClass::dtSTRING;
  97. m_arrColType[eHave] = CSortClass::dtINT;
  98. m_arrColType[eInterested] = CSortClass::dtSTRING;
  99. m_arrColType[eChoking] = CSortClass::dtSTRING;
  100. m_arrColType[eInteresting] = CSortClass::dtSTRING;
  101. m_arrColType[eChocked] = CSortClass::dtSTRING;
  104. }
  106. CPeersListCtrl::~CPeersListCtrl()
  107. {
  108. }
  110. // CListCtrlBaseEx
  111. BEGIN_MESSAGE_MAP(CPeersListCtrl, CListCtrlBaseEx)
  112. //{{AFX_MSG_MAP(CPeersListCtrl)
  113. ON_WM_CONTEXTMENU()
  114. ON_WM_CREATE()
  115. ON_NOTIFY_REFLECT(LVN_DELETEITEM, OnDeleteitem)
  116. ON_WM_SIZE()
  117. ON_WM_LBUTTONDBLCLK()
  118. ON_WM_LBUTTONDOWN()
  119. //}}AFX_MSG_MAP
  120. // ON_NOTIFY_REFLECT(LVN_COLUMNCLICK, OnColumnclick)
  121. ON_NOTIFY_REFLECT(NM_CUSTOMDRAW, OnCustomDraw)
  122. ON_COMMAND(ID_EDIT_COPY, OnMenuitemCopy)
  123. ON_COMMAND(ID_MENUITEM_SELECT_ALL, OnMenuitemSelectAll)
  124. ON_COMMAND_RANGE(ID_MENUITEM_PAUSE_PEERS, ID_MENUITEM_SHOW_PEER_INFO, OnMenuItemPeers)
  125. ON_COMMAND_RANGE(ID_MENUITEM_MONITOR_INFORMATION, ID_MENUITEM_MONITOR_LOGS, OnMenuitemMonitorSub)
  126. END_MESSAGE_MAP()
  128. /////////////////////////////////////////////////////////////////////////////
  129. // CPeersListCtrl message handlers
  131. BOOL CPeersListCtrl::CreateEx(CWnd *pParent, CRect rc, long lID, long lParam, CSetupRegister* pSetupRegister, CMonitorWnd* pMonitorWnd)
  132. {
  133. assert(pMonitorWnd && pSetupRegister);
  134. m_pSetupRegister = pSetupRegister;
  135. m_pMonitorWnd = pMonitorWnd;
  136. assert(pSetupRegister);
  137. return Create(WS_VISIBLE|WS_CHILD|WS_BORDER|LVS_REPORT, rc, pParent, lID);
  138. }
  140. int CPeersListCtrl::OnCreate(LPCREATESTRUCT lpCreateStruct) 
  141. {
  142. if (CListCtrlBaseEx::OnCreate(lpCreateStruct) == -1)
  143. return -1;
  145. //
  146. // ImageList.
  147. //
  148. HBITMAP hbm = (HBITMAP)::LoadImage(AfxGetInstanceHandle(),
  149. MAKEINTRESOURCE(IDB_BITMAP_IMAGELIST_PEER),
  150. IMAGE_BITMAP,
  151. 0,0, 
  152. LR_CREATEDIBSECTION);
  153. CBitmap bm;
  154. bm.Attach(hbm);
  156. m_ctlImage.Create(16, 17, ILC_COLOR8|ILC_MASK, 11, 4);
  157. m_ctlImage.SetBkColor(::GetSysColor(COLOR_WINDOW)); 
  158. m_ctlImage.Add(&bm, (COLORREF)RGB(255,0,255));
  159. // m_ctlImage.Create(16,17, ILC_COLOR8|ILC_MASK, 4, 4);
  160. // m_ctlImage.Add(AfxGetApp()->LoadIcon(IDI_ICON_ASTERISK1));
  162. SetImageList(&m_ctlImage, LVSIL_SMALL);
  163. SetExtendedStyle(LVS_EX_FULLROWSELECT|LVS_EX_GRIDLINES);
  165. SetSortColumn(1);
  168. return 0;
  169. }
  172. void CPeersListCtrl::OnSize(UINT nType, int cx, int cy) 
  173. {
  174. CListCtrlBaseEx::OnSize(nType, cx, cy);
  175. }
  177. void CPeersListCtrl::OnContextMenu(CWnd* pWnd, CPoint point) 
  178. {
  179. if (CListCtrlBaseEx::OnContextMenu(pWnd, point))
  180. return;
  182. bool bcopy = false, bContinue = false, bPause = false, bClose = false, bShowPeerInfo = false;
  184. if (GetSelectedCount() == 1)
  185. bShowPeerInfo = true;
  187. if (GetSelectedCount())
  188. bClose = bcopy = true;
  190. POSITION pos = GetFirstSelectedItemPosition();
  191. while (pos)
  192. {
  193. int nItem = GetNextSelectedItem(pos);
  194. CConncetionData* pData = (CConncetionData*)GetItemData(nItem);
  195. if (!pData)
  196. {
  197. assert(false);
  198. return;
  199. }
  201. if (pData->m_bPaused)
  202. bContinue = true;
  204. if (!pData->m_bPaused)
  205. bPause = true;
  206.  
  207. if (bPause && bContinue)
  208. break;
  209. }
  211. CMenu menu_bar;
  212. menu_bar.CreatePopupMenu();
  214. menu_bar.AppendMenu(MF_STRING, ID_MENUITEM_SHOW_PEER_INFO, "显示选中连接的详细信息(&S)t");
  215. menu_bar.AppendMenu(MF_STRING, ID_MENUITEM_CONTINUE_PEERS, "继续选中连接(&X)t");
  216. menu_bar.AppendMenu(MF_STRING, ID_MENUITEM_PAUSE_PEERS, "暂停选中连接(&P)t");
  217. menu_bar.AppendMenu(MF_STRING, ID_MENUITEM_STOP_PEERS, "关闭选中连接(&Q)t");
  218. menu_bar.AppendMenu(MF_SEPARATOR);
  219. menu_bar.AppendMenu(MF_STRING, ID_EDIT_COPY, "复制(&C)t");
  220. menu_bar.AppendMenu(MF_STRING, ID_MENUITEM_SELECT_ALL, "全选(&A)t");
  222. menu_bar.EnableMenuItem(ID_MENUITEM_CONTINUE_PEERS, MF_BYCOMMAND | 
  223. (bContinue ? MF_ENABLED : (MF_DISABLED | MF_GRAYED)));
  224. menu_bar.EnableMenuItem(ID_MENUITEM_PAUSE_PEERS, MF_BYCOMMAND | 
  225. (bPause ? MF_ENABLED : (MF_DISABLED | MF_GRAYED)));
  226. menu_bar.EnableMenuItem(ID_MENUITEM_STOP_PEERS, MF_BYCOMMAND | 
  227. (bClose ? MF_ENABLED : (MF_DISABLED | MF_GRAYED)));
  228. menu_bar.EnableMenuItem(ID_EDIT_COPY, MF_BYCOMMAND | 
  229. (bcopy? MF_ENABLED : (MF_DISABLED | MF_GRAYED)));
  230. menu_bar.EnableMenuItem(ID_MENUITEM_SHOW_PEER_INFO, MF_BYCOMMAND | 
  231. (bShowPeerInfo? MF_ENABLED : (MF_DISABLED | MF_GRAYED)));
  233. menu_bar.AppendMenu(MF_SEPARATOR);
  234. if (!m_pMonitorWnd)
  235. {
  236. assert(false);
  237. return;
  238. }
  239. m_pMonitorWnd->OnContextQuery(menu_bar);
  241. CPoint posMouse;
  242. GetCursorPos(&posMouse);
  243. menu_bar.TrackPopupMenu(TPM_LEFTALIGN |TPM_RIGHTBUTTON,posMouse.x,posMouse.y,this);
  244. }
  246. void CPeersListCtrl::OnMenuitemMonitorSub(UINT uID) 
  247. {
  248. if (!m_pMonitorWnd)
  249. {
  250. assert(false);
  251. return;
  252. }
  253. if (uID >= ID_MENUITEM_MONITOR_INFORMATION && 
  254. uID <= ID_MENUITEM_MONITOR_LOGS)
  255. {
  256. m_pMonitorWnd->OnShowNode(uID );
  257. }
  258. }
  260. void CPeersListCtrl::OnColumnSort(int iColumn)
  261. {
  262. assert(abs(iColumn) <= ePeersColumnCount && abs(iColumn) > 0);
  263. m_iSortedCol = iColumn;
  264. SortItems(Compare1, m_iSortedCol);
  267. /*
  268. assert(abs(iColumn) <= ePeersColumnCount && abs(iColumn) > 0);
  269. m_bSortAscending = iColumn > 0;
  270. m_iSortedCol = abs(iColumn) - 1;
  272. CSortClass csc(this, m_iSortedCol);
  273. csc.Sort(m_bSortAscending, (CSortClass::EDataType)m_arrColType[m_iSortedCol]);
  274. //*/
  275. }
  277. int CALLBACK CPeersListCtrl::Compare1(LPARAM lParam1, LPARAM lParam2, LPARAM 
  278.  lParamSort)
  279. {
  280. CConncetionData* pItem1 = (CConncetionData*) lParam1;
  281. CConncetionData* pItem2 = (CConncetionData*) lParam2;
  282. if (!pItem1 || !pItem2)
  283. {
  284. ASSERT(pItem1 && pItem2);
  285. return 0;
  286. }
  287. CConncetionData& conn1 = *pItem1;
  288. CConncetionData& conn2 = *pItem2;
  290. long sOrder = lParamSort < 0 ? -1 : 1; 
  291. eColumns eCol = (eColumns)((lParamSort * sOrder) - 1);
  293. ////////////////////////////////////////////////////////////
  294. // for dynamic data.
  296. CStatics statics1, statics2;
  297. int iStat1 = 0, iStat2 = 0;
  298. switch (eCol)
  299. {
  300. case eIcon:
  301. {
  302. int iState1 = 0, iState2 = 0;
  303. if (conn1.m_lDownRate > 0)
  304. iState1 = 2;
  305. else if (conn1.m_lUpRate > 0)
  306. iState1 = 1;
  308. if (conn2.m_lDownRate > 0)
  309. iState2= 2;
  310. else if (conn2.m_lUpRate > 0)
  311. iState2 = 1;
  312. return (iState1 - iState2) * sOrder;
  313. }
  314. break;
  315. case eIP:
  316. {
  317. /*
  318. in_addr inaddress;
  319. inaddress.S_un.S_addr = conn1.m_lAddr;
  320. CString str1 = inet_ntoa(inaddress);
  321. inaddress.S_un.S_addr = conn2.m_lAddr;
  322. CString str2 = inet_ntoa(inaddress);
  323. //*/
  324. ULONG l1 = ntohl(conn1.m_lAddr);
  325. ULONG l2 = ntohl(conn2.m_lAddr);
  326. if (l1 == l2) return 0;
  327. return ((l1 > l2) ? 1 : -1) * sOrder;
  328. }
  329. break;
  330. case eLocale:
  331. return (conn1.m_bLocallyInitiated- conn2.m_bLocallyInitiated) * sOrder;
  332. break;
  333. case eDownRate:
  334. return (conn1.m_lDownRate - conn2.m_lDownRate) * sOrder;
  335. break;
  336. case eUpRate:
  337. return (conn1.m_lUpRate - conn2.m_lUpRate) * sOrder;
  338. break;
  339. case eDownCount:
  340. return (conn1.m_lDownCount- conn2.m_lDownCount) * sOrder;
  341. break;
  342. case eUpCount:
  343. return (conn1.m_lUpCount - conn2.m_lUpCount) * sOrder;
  344. break;
  345. case eHave:
  346. return (conn1.m_fPeerHavePercent - conn2.m_fPeerHavePercent) * 100 * sOrder;
  347. break;
  348. case eInterested:
  349. return (conn1.m_bInterested - conn2.m_bInterested) * sOrder;
  350. break;
  351. case eChoking:
  352. return (conn1.m_bChoking - conn2.m_bChoking) * sOrder;
  353. break;
  354. case eInteresting:
  355. return (conn1.m_bInteresting - conn2.m_bInteresting) * sOrder;
  356. break;
  357. case eChocked:
  358. return (conn1.m_bChoked - conn2.m_bChoked) * sOrder;
  359. break;
  360. default:
  361. assert(false);
  362. break;
  363. }
  364. return 0;
  365. }
  366. /*
  367. void CPeersListCtrl::OnColumnclick(NMHDR* pNMHDR, LRESULT* pResult) 
  368. {
  369. NM_LISTVIEW* pNMListView = (NM_LISTVIEW*)pNMHDR;
  370. *pResult = 0;
  372. int iSubItem = GetLogicalIndex(pNMListView->iSubItem);
  373. if (iSubItem == -1)
  374. {
  375. assert(false);
  376. return;
  377. }
  379. if(iSubItem == m_iSortedCol)
  380. m_bSortAscending = !m_bSortAscending;
  381. else
  382. m_bSortAscending = TRUE;
  384. m_iSortedCol = iSubItem;
  386. CSortClass csc(this, m_iSortedCol);
  388. csc.Sort(m_bSortAscending, (CSortClass::EDataType)m_arrColType[m_iSortedCol]);
  389. }
  390. //*/
  392. void CPeersListCtrl::OnDeleteitem(NMHDR* pNMHDR, LRESULT* pResult) 
  393. {
  394. *pResult = 0;
  396. NM_LISTVIEW* pNMListView = (NM_LISTVIEW*)pNMHDR;
  397. CConncetionData* pData = (CConncetionData*)pNMListView->lParam;;
  398. if (!pData)
  399. {
  400. assert(false);
  401. return;
  402. }
  404. delete pData;
  405. }
  407. void CPeersListCtrl::Monitor(CFileDBItem* pFileDBItem)
  408. {
  409. m_pFileDBItem = pFileDBItem;
  410. DeleteAllItems();
  411. }
  413. void CPeersListCtrl::ShowAllPeers(bool bRefreshInfo)
  414. {
  415. SetRedraw(false);
  416. _ShowAllPeers(bRefreshInfo);
  417. SetRedraw(true);
  419. if (m_PeerInfoWnd.IsMonitoring())
  420. {
  421. if (m_pFileDBItem && m_pFileDBItem->m_pDownload && m_pFileDBItem->m_pDownload->IsDownloading())
  422. {
  423. bool bFind = false;
  424. vector<CConncetionData> vConnections = m_pFileDBItem->m_pDownload->GetConnections();
  425. for (int i=0; i<vConnections.size(); i++)
  426. {
  427. if (vConnections[i].m_lAddr == m_PeerInfoWnd.m_conData.m_lAddr && 
  428. vConnections[i].m_sPort == m_PeerInfoWnd.m_conData.m_sPort)
  429. {
  430. bFind = true;
  431. m_PeerInfoWnd.ShowPeer(vConnections[i]);
  432. break;
  433. }
  434. }
  435. if (!bFind)
  436. { 
  437. m_PeerInfoWnd.ClosePeer();
  438. }
  439. }
  440. }
  441. }
  443. void CPeersListCtrl::_ShowAllPeers(bool bRefreshInfo)
  444. {
  445. if (bRefreshInfo)
  446. DeleteAllItems();
  448. if (m_pFileDBItem && m_pFileDBItem->m_pDownload && m_pFileDBItem->m_pDownload->IsDownloading())
  449. {
  450. vector<CConncetionData> vConnections = m_pFileDBItem->m_pDownload->GetConnections();
  452. if (!bRefreshInfo)
  453. {
  454. vector<int> arrDelIndex;
  455. const long lCount = GetItemCount();
  457. for (int j=0; j<lCount; j++)
  458. {
  459. CConncetionData* pData = (CConncetionData*)GetItemData(j);
  460. if (!pData)
  461. {
  462. assert(false);
  463. for (int k=0; k<lCount; k++)
  464. {
  465. pData = (CConncetionData*)GetItemData(k);
  466. }
  467. return;
  468. }
  470. bool bFind = false;
  471. for (int i=0; i<vConnections.size(); i++)
  472. {
  473. if (vConnections[i].m_lAddr == pData->m_lAddr && 
  474. vConnections[i].m_sPort == pData->m_sPort)
  475. {
  476. if (*pData != vConnections[i])
  477. {
  478. RefreshPeerText(j, vConnections[i]);
  479. *pData = vConnections[i];
  480. // ShowPeerText(j);
  481. }
  483. vConnections.erase(vConnections.begin() + i);
  484. bFind = true;
  485. break;
  486. }
  487. }
  489. if (!bFind)
  490. arrDelIndex.push_back(j);
  491. }
  493. for (j=arrDelIndex.size() - 1; j >=0; j--)
  494. DeleteItem(arrDelIndex[j]);
  495. }
  498. char szText[1024] = {0};
  499. for (int i=0; i<vConnections.size(); i++)
  500. {
  501. long lInx = GetItemCount();
  502. InsertItem(lInx, "");
  504. CConncetionData* pData = new CConncetionData(vConnections[i]);
  505. SetItemData(lInx, (long)pData);
  507. ShowPeerText(lInx);
  508. }
  509. }
  511. }
  513. void CPeersListCtrl::OnShowColumn(int iItem)
  514. {
  515. ShowPeerText(iItem);
  516. }
  518. void CPeersListCtrl::RefreshPeerText(int iItem, CConncetionData& Conn)
  519. {
  520. CConncetionData* pOldData = (CConncetionData*)GetItemData(iItem);
  521. if (!pOldData)
  522. {
  523. assert(false);
  524. return;
  525. }
  526. CConncetionData& oldConn = *pOldData;
  528. int iStateOld = 0;
  529. if (oldConn.m_lDownRate > 0)
  530. iStateOld = 2;
  531. else if (oldConn.m_lUpRate > 0)
  532. iStateOld = 1;
  534. int iState = 0;
  535. if (Conn.m_lDownRate > 0)
  536. iState= 2;
  537. else if (Conn.m_lUpRate > 0)
  538. iState = 1;
  540. if (iStateOld != iState)
  541. {
  542. *pOldData = Conn;
  543. ShowPeerText(iItem);
  544. return;
  545. }
  549. char szText[1024] = {0};
  551. if (oldConn.m_bLocallyInitiated != Conn.m_bLocallyInitiated)
  552. {
  553. sprintf(szText, Conn.m_bLocallyInitiated ? "出" : "入");
  554. SetItemTextEx(iItem, eLocale, szText);
  555. }
  557. if (oldConn.m_bInterested != Conn.m_bInterested)
  558. SetItemTextEx(iItem, eInterested, Conn.m_bInterested ? "有" : "无");
  560. if (oldConn.m_bChoking != Conn.m_bChoking)
  561. SetItemTextEx(iItem, eChoking, Conn.m_bChoking ? "是" : "否");
  563. if (oldConn.m_bInteresting != Conn.m_bInteresting)
  564. SetItemTextEx(iItem, eInteresting, Conn.m_bInteresting? "有" : "无");
  566. if (oldConn.m_bChoked != Conn.m_bChoked)
  567. SetItemTextEx(iItem, eChocked, Conn.m_bChoked ? "是" : "否");
  568. if (oldConn.m_lUpRate != Conn.m_lUpRate)
  569. SetItemTextEx(iItem, eUpRate, FormatSizeK(Conn.m_lUpRate).data());
  570. if (oldConn.m_lDownRate != Conn.m_lDownRate)
  571. SetItemTextEx(iItem, eDownRate, FormatSizeK(Conn.m_lDownRate).data());
  572. if (oldConn.m_lDownCount != Conn.m_lDownCount)
  573. SetItemTextEx(iItem, eDownCount, FormatSizeM(Conn.m_lDownCount).data());
  574. if (oldConn.m_lUpCount != Conn.m_lUpCount)
  575. SetItemTextEx(iItem, eUpCount, FormatSizeM(Conn.m_lUpCount).data());
  577. if (oldConn.m_fPeerHavePercent != Conn.m_fPeerHavePercent)
  578. {
  579. sprintf(szText, "%.0f%%", Conn.m_fPeerHavePercent*100);
  580. SetItemTextEx(iItem, eHave, szText);
  581. }
  583. if (oldConn.m_bPaused != Conn.m_bPaused || oldConn.m_lDownRate != Conn.m_lDownRate)
  584. {
  585. if (Conn.m_bPaused)
  586. SetItem(iItem, 0, LVIF_IMAGE, 0, IID_PAUSE, 0, 0, 0);
  587. else
  588. {
  589. if (Conn.m_lDownRate <= 0)
  590. {
  591. if (Conn.m_lUpRate <= 0)
  592. {
  593. SetItem(iItem, 0, LVIF_IMAGE, 0, IID_CONNECTING, 0, 0, 0);
  594. }
  595. else
  596. {
  597. SetItem(iItem, 0, LVIF_IMAGE, 0, IID_UPLOAD, 0, 0, 0);
  598. }
  599. }
  600. else
  601. SetItem(iItem, 0, LVIF_IMAGE, 0, IID_DOWNLOAD, 0, 0, 0);
  602. }
  603. }
  604. }
  606. void CPeersListCtrl::ShowPeerText(int iItem)
  607. {
  608. CConncetionData* pData = (CConncetionData*)GetItemData(iItem);
  609. if (!pData)
  610. {
  611. assert(false);
  612. return;
  613. }
  615. CConncetionData& Conn = *pData;
  617. char szText[1024] = {0};
  619. in_addr inaddress;
  620. inaddress.S_un.S_addr = Conn.m_lAddr;
  621. sprintf(szText, "%s:%d", inet_ntoa(inaddress), ntohs(Conn.m_sPort));
  622. SetItemTextEx(iItem, eIP, szText);
  624. sprintf(szText, Conn.m_bLocallyInitiated ? "出" : "入");
  625. SetItemTextEx(iItem, eLocale, szText);
  627. SetItemTextEx(iItem, eIcon, "");
  628. SetItemTextEx(iItem, eInterested, Conn.m_bInterested ? "有" : "无");
  629. SetItemTextEx(iItem, eChoking, Conn.m_bChoking ? "是" : "否");
  630. SetItemTextEx(iItem, eInteresting, Conn.m_bInteresting? "有" : "无");
  631. SetItemTextEx(iItem, eChocked, Conn.m_bChoked ? "是" : "否");
  632. SetItemTextEx(iItem, eUpRate, FormatSizeK(Conn.m_lUpRate).data());
  633. SetItemTextEx(iItem, eDownRate, FormatSizeK(Conn.m_lDownRate).data());
  634. SetItemTextEx(iItem, eDownCount, FormatSizeM(Conn.m_lDownCount).data());
  635. SetItemTextEx(iItem, eUpCount, FormatSizeM(Conn.m_lUpCount).data());
  637. sprintf(szText, "%.0f%%", Conn.m_fPeerHavePercent*100);
  638. SetItemTextEx(iItem, eHave, szText);
  640. if (pData->m_bPaused)
  641. SetItem(iItem, 0, LVIF_IMAGE, 0, IID_PAUSE, 0, 0, 0);
  642. else
  643. {
  644. if (pData->m_lDownRate <= 0)
  645. {
  646. if (pData->m_lUpRate <= 0)
  647. {
  648. SetItem(iItem, 0, LVIF_IMAGE, 0, IID_CONNECTING, 0, 0, 0);
  649. }
  650. else
  651. {
  652. SetItem(iItem, 0, LVIF_IMAGE, 0, IID_UPLOAD, 0, 0, 0);
  653. }
  654. }
  655. else
  656. SetItem(iItem, 0, LVIF_IMAGE, 0, IID_DOWNLOAD, 0, 0, 0);
  657. }
  658. }
  662. void CPeersListCtrl::OnMenuItemPeers(UINT uID)
  663. {
  664. if (!m_pFileDBItem || !m_pFileDBItem->m_pDownload || !m_pFileDBItem->m_pDownload->IsDownloading())
  665. {
  666. assert(false);
  667. return;
  668. }
  670. switch (uID)
  671. {
  672. case ID_MENUITEM_CONTINUE_PEERS:
  673. {
  674. POSITION pos = GetFirstSelectedItemPosition();
  675. while (pos)
  676. {
  677. int nItem = GetNextSelectedItem(pos);
  678. CConncetionData* pData = (CConncetionData*)GetItemData(nItem);
  679. if (!pData)
  680. {
  681. assert(false);
  682. return;
  683. }
  685. if (pData->m_bPaused)
  686. {
  687. m_pFileDBItem->m_pDownload->PausePeer(CAddrPort(pData->m_lAddr, pData->m_sPort), false);
  688. pData->m_bPaused = false;
  689. ShowPeerText(nItem);
  690. }
  691. }
  692. }
  693. break;
  694. case ID_MENUITEM_PAUSE_PEERS:
  695. {
  696. POSITION pos = GetFirstSelectedItemPosition();
  697. while (pos)
  698. {
  699. int nItem = GetNextSelectedItem(pos);
  700. CConncetionData* pData = (CConncetionData*)GetItemData(nItem);
  701. if (!pData)
  702. {
  703. assert(false);
  704. return;
  705. }
  707. if (!pData->m_bPaused)
  708. {
  709. m_pFileDBItem->m_pDownload->PausePeer(CAddrPort(pData->m_lAddr, pData->m_sPort), true);
  710. pData->m_bPaused = true;
  711. ShowPeerText(nItem);
  712. }
  714. pData = (CConncetionData*)GetItemData(nItem);
  715. if (!pData)
  716. {
  717. assert(false);
  718. return;
  719. }
  720. }
  721. }
  722. break;
  723. case ID_MENUITEM_STOP_PEERS:
  724. {
  725. POSITION pos = GetFirstSelectedItemPosition();
  726. while (pos)
  727. {
  728. int nItem = GetNextSelectedItem(pos);
  729. CConncetionData* pData = (CConncetionData*)GetItemData(nItem);
  730. if (!pData)
  731. {
  732. assert(false);
  733. return;
  734. }
  736. m_pFileDBItem->m_pDownload->ClosePeer(CAddrPort(pData->m_lAddr, pData->m_sPort));
  737. }
  738. }
  739. break;
  740. case ID_MENUITEM_SHOW_PEER_INFO:
  741. {
  742. POSITION pos = GetFirstSelectedItemPosition();
  743. if (pos)
  744. {
  745. int nItem = GetNextSelectedItem(pos);
  746. CConncetionData* pData = (CConncetionData*)GetItemData(nItem);
  748. ShowWindow(false);
  749. m_PeerInfoWnd.ShowWindow(true);
  750. m_PeerInfoWnd.ShowPeer(*pData);
  751. m_PeerInfoWnd.SetFocus();
  752. }
  753. }
  754. break;
  755. default:
  756. assert(false);
  757. }
  758. }
  760. void CPeersListCtrl::OnMenuitemCopy()
  761. {
  762. CString strSource;
  764. int nColumnCount = GetHeaderCtrl()->GetItemCount();
  765. POSITION pos = GetFirstSelectedItemPosition();
  766. while (pos)
  767. {
  768. int index = GetNextSelectedItem(pos);
  769. for (int i=0; i<nColumnCount; i++)
  770. strSource += GetItemText(index, i) + "  ";
  772. strSource += "rn";
  773. }
  775. if (!CopyTextToClipboard(strSource))
  776. AfxMessageBox("复制失败");
  777. }
  779. void CPeersListCtrl::OnMenuitemSelectAll()
  780. {
  781. for (int i=0; i<GetItemCount(); i++)
  782. SetItemState(i, LVIS_SELECTED, LVIS_SELECTED);
  783. SetFocus();
  784. }
  787. void CPeersListCtrl::OnCustomDraw(NMHDR* pNMHDR, LRESULT* pResult)
  788. {
  789. LPNMLVCUSTOMDRAW  lplvcd = (LPNMLVCUSTOMDRAW)pNMHDR;
  791. switch(lplvcd->nmcd.dwDrawStage)
  792. {
  793. case CDDS_PREPAINT:
  794. *pResult = CDRF_NOTIFYSUBITEMDRAW;          // ask for subitem notifications.
  795. *pResult = CDRF_NOTIFYPOSTERASE;
  796. break;
  798. case CDDS_ITEMPREERASE:
  799. {
  800. int ik = 0;
  801. }
  802. break;
  803. case CDDS_ITEMPREPAINT:
  804. {
  805. *pResult = CDRF_DODEFAULT;
  807. if (!(lplvcd->nmcd.uItemState & CDIS_SELECTED ))
  808. {
  809. CConncetionData* pData = (CConncetionData*)lplvcd->nmcd.lItemlParam;
  810. if (!pData)
  811. {
  812. assert(false);
  813. return;
  814. }
  816. CConncetionData& Conn = *pData;
  817. if (Conn.m_lDownRate > 0)
  818. {
  819. lplvcd->clrTextBk = RGB(235, 242, 254);
  820. }
  821. else if (Conn.m_lUpRate > 0)
  822. {
  823. lplvcd->clrTextBk = RGB(255, 250, 237);
  824. }
  825. else
  826. {
  827. lplvcd->clrTextBk = RGB(255, 255, 255);
  828. }
  830. /*
  831. CSystemMsg* pData = (CSystemMsg*)lplvcd->nmcd.lItemlParam;
  832. CSystemMsg& msg = *pData;
  834. switch (msg.m_eMsgType)
  835. {
  836. case CSystemMsg::eMsgOut:
  837. lplvcd->clrTextBk = RGB(235, 242, 254);
  838. break;
  839. case CSystemMsg::eMsgIn:
  840. lplvcd->clrTextBk = RGB(255, 250, 237); //RGB(230, 230, 250); // 175, 238, 238
  841. break;
  842. case CSystemMsg::eCmd:
  843. lplvcd->clrTextBk = RGB(235, 242, 254);
  844. break;
  845. case CSystemMsg::eBad:
  846. lplvcd->clrTextBk = RGB(254, 238, 231);
  847. break;
  848. default:
  849. assert(false);
  850. return;
  851. }
  852. //*/
  853. *pResult = CDRF_NEWFONT;
  854. }
  856. //*pResult = CDRF_NOTIFYSUBITEMDRAW;          // ask for subitem notifications.
  857. }
  858. break;
  860. case CDDS_ITEMPREPAINT|CDDS_SUBITEM: // recd when CDRF_NOTIFYSUBITEMDRAW is returned in
  861. {                                    // response to CDDS_ITEMPREPAINT.
  862. *pResult = CDRF_DODEFAULT;
  863. int iCol = lplvcd->iSubItem;
  864. int iRow = lplvcd->nmcd.dwItemSpec;
  865. break;
  866. }
  868. default:// it wasn't a notification that was interesting to us.
  869. *pResult = CDRF_DODEFAULT;
  870. }
  871. }
  873. void CPeersListCtrl::OnLButtonDblClk(UINT nFlags, CPoint point) 
  874. {
  875. POSITION pos = GetFirstSelectedItemPosition();
  876. if (pos)
  877. {
  878. int nItem = GetNextSelectedItem(pos);
  879. CConncetionData* pData = (CConncetionData*)GetItemData(nItem);
  881. ShowWindow(false);
  882. m_PeerInfoWnd.ShowWindow(true);
  883. m_PeerInfoWnd.ShowPeer(*pData);
  884. m_PeerInfoWnd.SetFocus();
  885. }
  887. CListCtrlBaseEx::OnLButtonDblClk(nFlags, point);
  888. }
  890. void CPeersListCtrl::OnLButtonDown(UINT nFlags, CPoint point) 
  891. {
  892. CListCtrlBaseEx::OnLButtonDown(nFlags, point);
  893. }
  895. void CPeersListCtrl::OnMenuItemPeerInfo(const UINT uID, const CConncetionData& conData )
  896. {
  897. if (!m_pFileDBItem)
  898. {
  899. assert(false);
  900. return;
  901. }
  902. switch (uID)
  903. {
  904. case ID_MENUITEM_CONTINUE_PEERS:
  905. {
  906. m_pFileDBItem->m_pDownload->PausePeer(CAddrPort(conData.m_lAddr, conData.m_sPort), false);
  907. }
  908. break;
  909. case ID_MENUITEM_PAUSE_PEERS:
  910. {
  911. m_pFileDBItem->m_pDownload->PausePeer(CAddrPort(conData.m_lAddr, conData.m_sPort), true);
  912. }
  913. break;
  914. case ID_MENUITEM_STOP_PEERS:
  915. {
  916. m_pFileDBItem->m_pDownload->ClosePeer(CAddrPort(conData.m_lAddr, conData.m_sPort));
  917. }
  918. break;
  919. default:
  920. {
  921. assert(false);
  922. return;
  923. }
  924. }
  925. }