Windows 系统托盘图标

最新推荐文章于 2022-04-06 21:43:00 发布

kusey

最新推荐文章于 2022-04-06 21:43:00 发布

阅读量1.3k

点赞数

分类专栏： windows 文章标签： windows null path shell imagelist button

windows 专栏收录该内容

12 篇文章 0 订阅

订阅专栏

在 codeproject 上找到一篇关于获取系统托盘图标的文章 Shell Tray Info - Arrange your system tray icons，感觉很好，于是把他的代码的一部分分离出来，做了一个 dll 专门获取系统的托盘信息，其它的功能都没有加进去，代码如下：

1. dll 部分

ProcessData.h

#ifndef __ProcessData_h__ #define __ProcessData_h__ #include <Windows.h> typedef struct _process_data { HANDLE _hProcess; LPVOID _lpData; } PROCESSDATA, *PPROCESSDATA; #ifdef __cplusplus extern "C" { #endif int ProcessData_InitInstance ( PPROCESSDATA pProcessData, DWORD dwProcessId, SIZE_T nDataSize, DWORD dwDesiredAccess, DWORD dwAllocationType, DWORD dwProtect ); void ProcessData_UninitInstance ( PPROCESSDATA pProcessData ); int ProcessData_WriteData ( PPROCESSDATA pProcessData, const LPVOID pData, SIZE_T nSize ); int ProcessData_ReadData ( PPROCESSDATA pProcessData, LPVOID pData, SIZE_T nSize ); int ProcessData_GetData ( PPROCESSDATA pProcessData, const LPVOID lpcData, LPVOID pData, SIZE_T nSize ); #ifdef __cplusplus } #endif #endif // __ProcessData_h__

ProcessData.c

#include "ProcessData.h" int ProcessData_InitInstance( PPROCESSDATA pProcessData, DWORD dwProcessId, SIZE_T nDataSize, DWORD dwDesiredAccess, DWORD dwAllocationType, DWORD dwProtect ) { if( NULL == pProcessData || dwProcessId <= 0 || nDataSize <= 0 ) return 0; pProcessData->_hProcess = OpenProcess( dwDesiredAccess, FALSE, dwProcessId ); if( NULL == pProcessData->_hProcess ) return 0; pProcessData->_lpData = VirtualAllocEx( pProcessData->_hProcess, NULL, nDataSize, dwAllocationType, dwProtect ); if( NULL == pProcessData->_lpData ) return 0; return 1; } void ProcessData_UninitInstance( PPROCESSDATA pProcessData ) { if( NULL != pProcessData ) { if( NULL != pProcessData->_hProcess ) { if( NULL != pProcessData->_lpData ) VirtualFreeEx( pProcessData->_hProcess, pProcessData->_lpData, 0, MEM_RELEASE ); CloseHandle( pProcessData->_hProcess ); } } } int ProcessData_WriteData( PPROCESSDATA pProcessData, const LPVOID pData, SIZE_T nSize ) { if( NULL == pProcessData || NULL == pProcessData->_hProcess || / NULL == pProcessData->_lpData || nSize <= 0 ) return 0; return WriteProcessMemory( pProcessData->_hProcess, pProcessData->_lpData, pData, nSize, NULL ); } int ProcessData_ReadData( PPROCESSDATA pProcessData, LPVOID pData, SIZE_T nSize ) { if( NULL == pProcessData || NULL == pProcessData->_hProcess || / NULL == pProcessData->_lpData || nSize <= 0 ) return 0; return ReadProcessMemory( pProcessData->_hProcess, pProcessData->_lpData, pData, nSize, NULL ); } int ProcessData_GetData( PPROCESSDATA pProcessData, const LPVOID lpcData, LPVOID pData, SIZE_T nSize ) { if( NULL == pProcessData || NULL == lpcData || NULL == pData || nSize <= 0 ) return 0; return ReadProcessMemory( pProcessData->_hProcess, lpcData, pData, nSize, NULL ); }

TrayInfo.h

#ifndef __TrayInfo_h__ #define __TrayInfo_h__ #ifdef DLL_EXPORT #define DLLAPI __declspec(dllexport) #else #define DLLAPI __declspec(dllimport) #endif #include <Windows.h> #include <tchar.h> typedef struct _tray_data { HWND _hWnd; UINT _uId; UINT _uCallbackMsg; DWORD _dwReserved[2]; HICON _hIcon; } TRAYDATA, *PTRAYDATA; typedef struct _tray_item_info { HWND _hWnd; UINT _uId; UINT _uCallbackMsg; TCHAR _lpszTip[1024]; TCHAR _lpszProcessPath[MAX_PATH]; BOOL _bVisible; } TRAYITEMINFO, *PTRAYITEMINFO; #ifdef __cplusplus extern "C" { #endif int DLLAPI GetTrayItems( PTRAYITEMINFO pTrayItemInfo, HICON *phIcon, int nMaxCount ); #ifdef __cplusplus } #endif #endif // __TrayInfo_h__

TrayInfo.c

#define DLL_EXPORT #include "TrayInfo.h" #include "ProcessData.h" #include <CommCtrl.h> #include <string.h> #include <Psapi.h> #pragma comment( lib, "Psapi.lib" ) // HWND FindTrayWnd (); TCHAR GetDriveLetter ( TCHAR *lpszDevicePath ); TCHAR* GetFilePathFromPid ( DWORD dwPid ); // int GetTrayItems( PTRAYITEMINFO pTrayItemInfo, HICON *phIcon, int nMaxCount ) { HWND hWndTray = NULL; DWORD dwTrayPid = 0; DWORD dwProcessId = 0; int nCount = 0; int i = 0; int j = 0; TCHAR cTip = 0; TCHAR *pTip = NULL; TCHAR szTip[1024] = { 0 }; PROCESSDATA process_data = { 0 }; TBBUTTON tb_button = { 0 }; TRAYDATA tray_data = { 0 }; TRAYITEMINFO tray_item_info = { 0 }; if( NULL == pTrayItemInfo || NULL == phIcon || nMaxCount <= 0 ) return 0; hWndTray = FindTrayWnd(); if( NULL == hWndTray ) return 0; GetWindowThreadProcessId( hWndTray, &dwTrayPid ); nCount = (int)SendMessage( hWndTray, TB_BUTTONCOUNT, 0, 0 ); nCount = nCount < nMaxCount ? nCount : nMaxCount; if( 0 == ProcessData_InitInstance(&process_data, dwTrayPid, sizeof(TBBUTTON), PROCESS_ALL_ACCESS, MEM_COMMIT, PAGE_READWRITE) ) return 0; for( i = 0; i < nCount; i++ ) { memset( &tb_button, 0, sizeof(TBBUTTON) ); memset( &tray_data, 0, sizeof(TRAYDATA) ); memset( &tray_item_info, 0, sizeof(TRAYITEMINFO) ); SendMessage( hWndTray, TB_GETBUTTON, i, (LPARAM)(process_data._lpData) ); ProcessData_ReadData( &process_data, (LPVOID)&tb_button, sizeof(TBBUTTON) ); ProcessData_GetData( &process_data, (LPCVOID)tb_button.dwData, &tray_data, sizeof(TRAYDATA) ); GetWindowThreadProcessId( tray_data._hWnd, &dwProcessId ); _tcscpy_s( tray_item_info._lpszProcessPath, MAX_PATH, GetFilePathFromPid(dwProcessId) ); memset( szTip, 0, sizeof(TCHAR) * 1024 ); pTip = (TCHAR*)tb_button.iString; if( !(tb_button.fsState & TBSTATE_HIDDEN) ) { tray_item_info._bVisible = TRUE; j = 0; ProcessData_GetData( &process_data, (LPVOID)pTip++, (LPVOID)&cTip, sizeof(TCHAR) ); while( _T('/0') != cTip ) { if( 1023 == j ) break; szTip[j++] = cTip; ProcessData_GetData( &process_data, (LPVOID)pTip++, (LPVOID)&cTip, sizeof(TCHAR) ); } } else { tray_item_info._bVisible = FALSE; _tcscpy_s( szTip, 1024, _T("[Hidden Icon]") ); } _tcscpy( tray_item_info._lpszTip, szTip ); tray_item_info._hWnd = tray_data._hWnd; tray_item_info._uId = tray_data._uId; tray_item_info._uCallbackMsg = tray_data._uCallbackMsg; pTrayItemInfo[i] = tray_item_info; phIcon[i] = tray_data._hIcon; } ProcessData_UninitInstance( &process_data ); return nCount; } // HWND FindTrayWnd() { HWND hwnd_Shell_TrayWnd = NULL; HWND hwnd_TrayNotifyWnd = NULL; HWND hwnd_SysPager = NULL; hwnd_Shell_TrayWnd = FindWindow( _T("Shell_TrayWnd"), NULL ); if( NULL == hwnd_Shell_TrayWnd ) return NULL; hwnd_TrayNotifyWnd = FindWindowEx( hwnd_Shell_TrayWnd, NULL, _T("TrayNotifyWnd"), NULL ); if( NULL == hwnd_TrayNotifyWnd ) return NULL; hwnd_SysPager = FindWindowEx( hwnd_TrayNotifyWnd, NULL, _T("SysPager"), NULL ); if( NULL == hwnd_SysPager ) return NULL; return FindWindowEx( hwnd_SysPager, NULL, _T("ToolbarWindow32"), NULL ); } TCHAR GetDriveLetter( TCHAR *lpszDevicePath ) { TCHAR c = _T('A'); TCHAR szDeviceName[3] = { 0 }; TCHAR szTarget[512] = { 0 }; while( c <= _T('Z') ) { szDeviceName[0] = c; szDeviceName[1] = _T(':'); szDeviceName[2] = _T('/0'); memset( szTarget, 0, sizeof(TCHAR) * 512 ); if( 0 != QueryDosDevice(szDeviceName, szTarget, 512) ) { if( 0 == _tcscmp(szTarget, lpszDevicePath) ) return c; } c++; } return _T('/0'); } TCHAR* GetFilePathFromPid( DWORD dwPid ) { static TCHAR strRet[MAX_PATH] = { 0 }; TCHAR lpszImageFileName[MAX_PATH] = { 0 }; TCHAR lpszSearch[] = _T("//Device//HarddiskVolume"); TCHAR lpszDrive[MAX_PATH] = { 0 }; TCHAR *p = NULL; TCHAR cDriveLetter = 0; HANDLE hProcess = NULL; int i = 0; if( dwPid <= 0 ) return NULL; hProcess = OpenProcess( PROCESS_QUERY_INFORMATION, FALSE, dwPid ); if( NULL == hProcess ) return NULL; GetProcessImageFileName( hProcess, lpszImageFileName, MAX_PATH ); CloseHandle( hProcess ); p = _tcsstr( lpszImageFileName, lpszSearch ); if( NULL == p ) return NULL; p = _tcschr( p + _tcslen(lpszSearch), _T('//') ); if( NULL == p ) return NULL; _tcscpy_s( lpszDrive, MAX_PATH, lpszImageFileName ); lpszDrive[p - lpszImageFileName] = _T('/0'); cDriveLetter = GetDriveLetter( lpszDrive ); memset( strRet, 0, sizeof(TCHAR) * MAX_PATH ); strRet[0] = cDriveLetter; strRet[1] = _T(':'); _tcscat( strRet, p ); return strRet; }

2. 测试代码

_imageList.Create( 16, 16, ILC_COLOR32, 16, 16 ); _ctrlList.SetImageList( &_imageList, LVSIL_SMALL ); void CShellTrayInfoTestDlg::FillListCtrl() { _ctrlList.DeleteAllItems(); int nCount = _imageList.GetImageCount(); for( int i = 0; i < nCount; i++ ) _imageList.Remove( i ); TRAYITEMINFO tii[128] = { 0 }; HICON hIcon[128] = { 0 }; nCount = GetTrayItems( tii, hIcon, 128 ); if( 0 == nCount ) return; for( int i = 0; i < nCount; i++ ) { int iconIndex = 0; ICONINFO iconInfo = { 0 }; if( 0 != GetIconInfo(hIcon[i], &iconInfo) ) iconIndex = _imageList.Add( hIcon[i] ); LVITEM lvItem = { 0 }; lvItem.iItem = i; lvItem.iSubItem = 0; lvItem.mask = LVIF_IMAGE; lvItem.iImage = iconIndex; _ctrlList.InsertItem( &lvItem ); CString strHwnd; strHwnd.Format( _T("0x%08x"), tii[i]._hWnd ); _ctrlList.SetItemText( i, 1, strHwnd ); _ctrlList.SetItemText( i, 2, tii[i]._lpszProcessPath ); _ctrlList.SetItemText( i, 3, tii[i]._lpszTip ); } }

3. 效果显示