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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Windows编程 > 系统编程 > 查看源码
SortListCtrl.cpp
资源名称：BehaviorProMon.rar　[点击查看]
上传用户：xuemeng126
上传日期：2022-07-05
资源大小：454k
文件大小：12k
源码类别：

系统编程

开发平台：

Visual C++

SortListCtrl.cpp：源码内容
  1. #include "stdafx.h"
  2. #include "SortListCtrl.h"
  4. #ifdef _DEBUG
  5. #define new DEBUG_NEW
  6. #undef THIS_FILE
  7. static char THIS_FILE[] = __FILE__;
  8. #endif
  11. LPCTSTR g_pszSection = _T("ListCtrls");
  14. struct ItemData
  15. {
  16. public:
  17. ItemData() : arrpsz( NULL ), dwData( NULL ) {}
  19. LPTSTR* arrpsz;
  20. DWORD dwData;
  22. private:
  23. // ban copying.
  24. ItemData( const ItemData& );
  25. ItemData& operator=( const ItemData& );
  26. };
  29. CSortListCtrl::CSortListCtrl()
  30. : m_iNumColumns( 0 )
  31. , m_iSortColumn( -1 )
  32. , m_bSortAscending( TRUE )
  33. {
  34. }
  37. CSortListCtrl::~CSortListCtrl()
  38. {
  39. }
  42. BEGIN_MESSAGE_MAP(CSortListCtrl, CListCtrl)
  43. //{{AFX_MSG_MAP(CSortListCtrl)
  44. ON_NOTIFY_REFLECT(LVN_COLUMNCLICK, OnColumnClick)
  45. ON_WM_DESTROY()
  46. ON_NOTIFY_REFLECT(NM_CUSTOMDRAW, OnCustomDraw)
  47. //}}AFX_MSG_MAP
  48. END_MESSAGE_MAP()
  50. /////////////////////////////////////////////////////////////////////////////
  51. // CSortListCtrl message handlers
  53. void CSortListCtrl::PreSubclassWindow()
  54. {
  55. // the list control must have the report style.
  56. ASSERT( GetStyle() & LVS_REPORT );
  58. CListCtrl::PreSubclassWindow();
  59. VERIFY( m_ctlHeader.SubclassWindow( GetHeaderCtrl()->GetSafeHwnd() ) );
  60. }
  63. BOOL CSortListCtrl::SetHeadings( UINT uiStringID )
  64. {
  65. CString strHeadings;
  66. VERIFY( strHeadings.LoadString( uiStringID ) );
  67. return SetHeadings( strHeadings );
  68. }
  71. // the heading text is in the format column 1 text,column 1 width;column 2 text,column 3 width;etc.
  72. BOOL CSortListCtrl::SetHeadings( const CString& strHeadings )
  73. {
  74. int iStart = 0;
  76. for( ;; )
  77. {
  78. const int iComma = strHeadings.Find( _T(','), iStart );
  80. if( iComma == -1 )
  81. break;
  83. const CString strHeading = strHeadings.Mid( iStart, iComma - iStart );
  85. iStart = iComma + 1;
  87. int iSemiColon = strHeadings.Find( _T(';'), iStart );
  89. if( iSemiColon == -1 )
  90. iSemiColon = strHeadings.GetLength();
  92. const int iWidth = atoi( strHeadings.Mid( iStart, iSemiColon - iStart ) );
  94. iStart = iSemiColon + 1;
  96. if( InsertColumn( m_iNumColumns++, strHeading, LVCFMT_LEFT, iWidth ) == -1 )
  97. return FALSE;
  98. }
  100. return TRUE;
  101. }
  104. int CSortListCtrl::AddItem( int j,LPCTSTR pszText, ... )
  105. {
  106. const int iIndex = InsertItem( GetItemCount(), pszText ,j);
  108. LPTSTR* arrpsz = new LPTSTR[ m_iNumColumns ];
  109. arrpsz[ 0 ] = new TCHAR[ lstrlen( pszText ) + 1 ];
  110. (void)lstrcpy( arrpsz[ 0 ], pszText );
  112.   va_list list;
  113. va_start( list, pszText );
  115. for( int iColumn = 1; iColumn < m_iNumColumns; iColumn++ )
  116. {
  117. pszText = va_arg( list, LPCTSTR );
  118. ASSERT_VALID_STRING( pszText );
  119. VERIFY( CListCtrl::SetItem( iIndex, iColumn, LVIF_TEXT, pszText, 0, 0, 0, 0 ) );
  121. arrpsz[ iColumn ] = new TCHAR[ lstrlen( pszText ) + 1 ];
  122. (void)lstrcpy( arrpsz[ iColumn ], pszText );
  123. }
  125. va_end( list );
  127. VERIFY( SetTextArray( iIndex, arrpsz ) );
  129. return iIndex;
  130. }
  133. void CSortListCtrl::FreeItemMemory( const int iItem )
  134. {
  135. ItemData* pid = reinterpret_cast<ItemData*>( CListCtrl::GetItemData( iItem ) );
  137. LPTSTR* arrpsz = pid->arrpsz;
  139. for( int i = 0; i < m_iNumColumns; i++ )
  140. delete[] arrpsz[ i ];
  142. delete[] arrpsz;
  143. delete pid;
  145. VERIFY( CListCtrl::SetItemData( iItem, NULL ) );
  146. }
  149. BOOL CSortListCtrl::DeleteItem( int iItem )
  150. {
  151. FreeItemMemory( iItem );
  152. return CListCtrl::DeleteItem( iItem );
  153. }
  156. BOOL CSortListCtrl::DeleteAllItems()
  157. {
  158. for( int iItem = 0; iItem < GetItemCount(); iItem ++ )
  159. FreeItemMemory( iItem );
  161. return CListCtrl::DeleteAllItems();
  162. }
  165. bool IsNumber( LPCTSTR pszText )
  166. {
  167. ASSERT_VALID_STRING( pszText );
  169. for( int i = 0; i < lstrlen( pszText ); i++ )
  170. if( !_istdigit( pszText[ i ] ) )
  171. return false;
  173. return true;
  174. }
  177. int NumberCompare( LPCTSTR pszNumber1, LPCTSTR pszNumber2 )
  178. {
  179. ASSERT_VALID_STRING( pszNumber1 );
  180. ASSERT_VALID_STRING( pszNumber2 );
  182. const int iNumber1 = atoi( pszNumber1 );
  183. const int iNumber2 = atoi( pszNumber2 );
  185. if( iNumber1 < iNumber2 )
  186. return -1;
  188. if( iNumber1 > iNumber2 )
  189. return 1;
  191. return 0;
  192. }
  195. bool IsDate( LPCTSTR pszText )
  196. {
  197. ASSERT_VALID_STRING( pszText );
  199. // format should be 99/99/9999.
  201. if( lstrlen( pszText ) != 10 )
  202. return false;
  204. return _istdigit( pszText[ 0 ] )
  205. && _istdigit( pszText[ 1 ] )
  206. && pszText[ 2 ] == _T('/')
  207. && _istdigit( pszText[ 3 ] )
  208. && _istdigit( pszText[ 4 ] )
  209. && pszText[ 5 ] == _T('/')
  210. && _istdigit( pszText[ 6 ] )
  211. && _istdigit( pszText[ 7 ] )
  212. && _istdigit( pszText[ 8 ] )
  213. && _istdigit( pszText[ 9 ] );
  214. }
  217. int DateCompare( const CString& strDate1, const CString& strDate2 )
  218. {
  219. const int iYear1 = atoi( strDate1.Mid( 6, 4 ) );
  220. const int iYear2 = atoi( strDate2.Mid( 6, 4 ) );
  222. if( iYear1 < iYear2 )
  223. return -1;
  225. if( iYear1 > iYear2 )
  226. return 1;
  228. const int iMonth1 = atoi( strDate1.Mid( 3, 2 ) );
  229. const int iMonth2 = atoi( strDate2.Mid( 3, 2 ) );
  231. if( iMonth1 < iMonth2 )
  232. return -1;
  234. if( iMonth1 > iMonth2 )
  235. return 1;
  237. const int iDay1 = atoi( strDate1.Mid( 0, 2 ) );
  238. const int iDay2 = atoi( strDate2.Mid( 0, 2 ) );
  240. if( iDay1 < iDay2 )
  241. return -1;
  243. if( iDay1 > iDay2 )
  244. return 1;
  246. return 0;
  247. }
  250. int CALLBACK CSortListCtrl::CompareFunction( LPARAM lParam1, LPARAM lParam2, LPARAM lParamData )
  251. {
  252. CSortListCtrl* pListCtrl = reinterpret_cast<CSortListCtrl*>( lParamData );
  253. ASSERT( pListCtrl->IsKindOf( RUNTIME_CLASS( CListCtrl ) ) );
  255. ItemData* pid1 = reinterpret_cast<ItemData*>( lParam1 );
  256. ItemData* pid2 = reinterpret_cast<ItemData*>( lParam2 );
  258. ASSERT( pid1 );
  259. ASSERT( pid2 );
  261. LPCTSTR pszText1 = pid1->arrpsz[ pListCtrl->m_iSortColumn ];
  262. LPCTSTR pszText2 = pid2->arrpsz[ pListCtrl->m_iSortColumn ];
  264. ASSERT_VALID_STRING( pszText1 );
  265. ASSERT_VALID_STRING( pszText2 );
  267. if( IsNumber( pszText1 ) )
  268. return pListCtrl->m_bSortAscending ? NumberCompare( pszText1, pszText2 ) : NumberCompare( pszText2, pszText1 );
  269. else if( IsDate( pszText1 ) )
  270. return pListCtrl->m_bSortAscending ? DateCompare( pszText1, pszText2 ) : DateCompare( pszText2, pszText1 );
  271. else
  272. // text.
  273. return pListCtrl->m_bSortAscending ? lstrcmp( pszText1, pszText2 ) : lstrcmp( pszText2, pszText1 );
  274. }
  277. void CSortListCtrl::OnColumnClick( NMHDR* pNMHDR, LRESULT* pResult )
  278. {
  279. NM_LISTVIEW* pNMListView = (NM_LISTVIEW*)pNMHDR;
  280. const int iColumn = pNMListView->iSubItem;
  282. // if it's a second click on the same column then reverse the sort order,
  283. // otherwise sort the new column in ascending order.
  284. Sort( iColumn, iColumn == m_iSortColumn ? !m_bSortAscending : TRUE );
  286. *pResult = 0;
  287. }
  290. void CSortListCtrl::Sort( int iColumn, BOOL bAscending )
  291. {
  292. m_iSortColumn = iColumn;
  293. m_bSortAscending = bAscending;
  295. // show the appropriate arrow in the header control.
  296. m_ctlHeader.SetSortArrow( m_iSortColumn, m_bSortAscending );
  298. VERIFY( SortItems( CompareFunction, reinterpret_cast<DWORD>( this ) ) );
  299. }
  302. void CSortListCtrl::LoadColumnInfo()
  303. {
  304. // you must call this after setting the column headings.
  305. ASSERT( m_iNumColumns > 0 );
  307. CString strKey;
  308. strKey.Format( _T("%d"), GetDlgCtrlID() );
  310. UINT nBytes = 0;
  311. BYTE* buf = NULL;
  312. if( AfxGetApp()->GetProfileBinary( g_pszSection, strKey, &buf, &nBytes ) )
  313. {
  314. if( nBytes > 0 )
  315. {
  316. CMemFile memFile( buf, nBytes );
  317. CArchive ar( &memFile, CArchive::load );
  318. m_ctlHeader.Serialize( ar );
  319. ar.Close();
  321. m_ctlHeader.Invalidate();
  322. }
  324. delete[] buf;
  325. }
  326. }
  329. void CSortListCtrl::SaveColumnInfo()
  330. {
  331. ASSERT( m_iNumColumns > 0 );
  333. CString strKey;
  334. strKey.Format( _T("%d"), GetDlgCtrlID() );
  336. CMemFile memFile;
  338. CArchive ar( &memFile, CArchive::store );
  339. m_ctlHeader.Serialize( ar );
  340. ar.Close();
  342. DWORD dwLen = memFile.GetLength();
  343. BYTE* buf = memFile.Detach();
  345. VERIFY( AfxGetApp()->WriteProfileBinary( g_pszSection, strKey, buf, dwLen ) );
  347. free( buf );
  348. }
  351. void CSortListCtrl::OnDestroy() 
  352. {
  353. for( int iItem = 0; iItem < GetItemCount(); iItem ++ )
  354. FreeItemMemory( iItem );
  356. CListCtrl::OnDestroy();
  357. }
  360. BOOL CSortListCtrl::SetItemText( int nItem, int nSubItem, LPCTSTR lpszText )
  361. {
  362. if( !CListCtrl::SetItemText( nItem, nSubItem, lpszText ) )
  363. return FALSE;
  365. LPTSTR* arrpsz = GetTextArray( nItem );
  366. LPTSTR pszText = arrpsz[ nSubItem ];
  367. delete[] pszText;
  368. pszText = new TCHAR[ lstrlen( lpszText ) + 1 ];
  369. (void)lstrcpy( pszText, lpszText );
  370. arrpsz[ nSubItem ] = pszText;
  372. return TRUE;
  373. }
  376. BOOL CSortListCtrl::SetItemData( int nItem, DWORD dwData )
  377. {
  378. if( nItem >= GetItemCount() )
  379. return FALSE;
  381. ItemData* pid = reinterpret_cast<ItemData*>( CListCtrl::GetItemData( nItem ) );
  382. ASSERT( pid );
  383. pid->dwData = dwData;
  385. return TRUE;
  386. }
  389. DWORD CSortListCtrl::GetItemData( int nItem ) const
  390. {
  391. ASSERT( nItem < GetItemCount() );
  393. ItemData* pid = reinterpret_cast<ItemData*>( CListCtrl::GetItemData( nItem ) );
  394. ASSERT( pid );
  395. return pid->dwData;
  396. }
  399. BOOL CSortListCtrl::SetTextArray( int iItem, LPTSTR* arrpsz )
  400. {
  401. ASSERT( CListCtrl::GetItemData( iItem ) == NULL );
  402. ItemData* pid = new ItemData;
  403. pid->arrpsz = arrpsz;
  404. return CListCtrl::SetItemData( iItem, reinterpret_cast<DWORD>( pid ) );
  405. }
  408. LPTSTR* CSortListCtrl::GetTextArray( int iItem ) const
  409. {
  410. ASSERT( iItem < GetItemCount() );
  412. ItemData* pid = reinterpret_cast<ItemData*>( CListCtrl::GetItemData( iItem ) );
  413. return pid->arrpsz;
  414. }
  415. //  hook_API_info[hook_num++].nativeAPIname="NtRestoreKey";
  416. //  hook_API_info[hook_num++].nativeAPIname="NtOpenKey";
  417. //  hook_API_info[hook_num++].nativeAPIname="NtCreateKey";
  418. //  hook_API_info[hook_num++].nativeAPIname="NtQueryValueKey";
  419. //  hook_API_info[hook_num++].nativeAPIname="NtQueryKey";
  420. //  hook_API_info[hook_num++].nativeAPIname="NtDeleteKey";
  421. //  hook_API_info[hook_num++].nativeAPIname="NtDeleteValueKey";
  422. //  hook_API_info[hook_num++].nativeAPIname="NtSetValueKey";
  423. //  hook_API_info[hook_num++].nativeAPIname="NtEnumerateKey";
  424. //  hook_API_info[hook_num++].nativeAPIname="NtEnumerateValueKey";
  425. //  hook_API_info[hook_num++].nativeAPIname="NtOpenFile";
  426. //  hook_API_info[hook_num++].nativeAPIname="NtWriteFile";
  427. //  hook_API_info[hook_num++].nativeAPIname="NtCreateFile";
  428. //  hook_API_info[hook_num++].nativeAPIname="NtCreateProcess";
  429. //  hook_API_info[hook_num++].nativeAPIname="NtCreateProcessEx";
  430. //  hook_API_info[hook_num++].nativeAPIname="NtCreateSection";
  431. //  hook_API_info[hook_num++].nativeAPIname="NtOpenThread";
  432. //  hook_API_info[hook_num++].nativeAPIname="NtCreateThread";
  433. //  hook_API_info[hook_num++].nativeAPIname="NtOpenProcess";
  434. //  hook_API_info[hook_num++].nativeAPIname="NtOpenSection";
  435. //  hook_API_info[hook_num++].nativeAPIname="NtCreateSymbolicLinkObject";
  436. //  hook_API_info[hook_num++].nativeAPIname="NtSetSystemTime";
  437. void GetRowColor(char * NativeAPIName,COLORREF *pclrBk,COLORREF *pclrTxt)
  438. {
  439. CString funname=CString(NativeAPIName);
  440. if(funname.Find("Key")==funname.GetLength()-3)
  441. {
  442. *pclrBk=RGB(255,255,255);
  443. *pclrTxt=RGB(0x00,0x33,0xff);
  444. return;
  445. }
  446. if(funname.Find("File")==funname.GetLength()-4)
  447. {
  448. *pclrBk=RGB(255,255,255);
  449. *pclrTxt=RGB(0xcc,0x00,0xff);
  450. return;
  451. }
  452. if(funname.Find("Thread")==funname.GetLength()-6 
  453. || funname.Find("Process")==funname.GetLength()-7 
  454. || funname.Find("ProcessEx")==funname.GetLength()-9)
  455. {
  456. *pclrBk=RGB(255,255,255);
  457. *pclrTxt=RGB(0xff,0x00,0x00);
  458. return;
  459. }
  460. if(funname.Find("Section")==funname.GetLength()-7)
  461. {
  462. *pclrBk=RGB(255,255,255);
  463. *pclrTxt=RGB(0xff,0x33,0x00);
  464. return;
  465. }
  466. *pclrBk=RGB(255,255,255);
  467. *pclrTxt=RGB(0,0,0);
  468. }
  470. void CSortListCtrl::OnCustomDraw(NMHDR   *pnotify,   LRESULT   *result)   
  471. {                 
  472. LPNMLVCUSTOMDRAW     lplvcd   =   (LPNMLVCUSTOMDRAW)pnotify;   
  473. char NativeAPIName[32];
  475. if(lplvcd->nmcd.dwDrawStage   ==   CDDS_PREPAINT)       
  476. *result   =     CDRF_NOTIFYITEMDRAW;     
  477. if(lplvcd->nmcd.dwDrawStage   ==   CDDS_ITEMPREPAINT){   
  478. int   nItem=(int)(lplvcd->nmcd.dwItemSpec);   
  479. COLORREF   clrBk,clrTxt;     
  480. this->GetItemText(nItem,3,NativeAPIName,32);
  482. GetRowColor(NativeAPIName,&clrBk,&clrTxt);
  484. lplvcd->clrText   =   clrTxt;   
  485. lplvcd->clrTextBk   =   clrBk;     
  486. *result   =   CDRF_DODEFAULT;                       
  487. }     
  488. }