MASM32编程访问系统托盘区图标/07-09-29更新

;<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
; FileName: TrayIcon.asm
;   Function: Demo the way to enum the icons in system tray
;       Author: Purple Endurer | 紫郢剑侠㊣ (PurpleEndurer@163.com)
;     DevEnv: Win2000 pro SP4, MASM32 v8
;
;  log
; ----------------------------------------------------------------------------------
; 2007-09-29  Can run under Win XP
; 2007-09-22  Created!
;<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<

.386
.model  flat,  stdcall
option  casemap: none

include /masm32/ include/windows.inc

include /masm32/ include/kernel32.inc
includelib /masm32/ lib/kernel32.lib

include /masm32/ include/user32.inc
includelib /masm32/ lib/user32.lib

GetSysTrayToolBarHandle  proto
EnumSubCtl  proto :HWND, :LPARAM
GetSysTrayIconCount  proto
EnumSysTrayIcon  proto

.data
g_szAppName  db  "EnumSysTrayIcon", 0
g_szTaskBarCls  db  "Shell_TrayWnd", 0
g_szSysPagerCls  db  "SysPager", 0     ;WinXP need!
g_szTrayNotifyWndCls  db  "TrayNotifyWnd", 0
g_szToolbarWindow32Cls  db  "ToolbarWindow32", 0
g_szFailGetSysTray  db  "Fail to get system tray!", 0
g_hSysTray      HANDLE ?
g_dwTrayIconCount  dword ?
g_stTbButton TBBUTTON <>
g_szIconText  db MAX_PATH dup (?)

.code
start:
invoke GetSysTrayToolBarHandle
test    eax,  eax  ; .if  eax==NULL
.if ZERO?
     invoke MessageBox, NULL,  addr g_szFailGetSysTray,  addr g_szAppName, MB_ICONERROR
.else
     mov     g_hSysTray,  eax
     invoke EnumSysTrayIcon
.endif
invoke ExitProcess,NULL

;
; Function: Get the handle ToolbarWindow32 of in system tray
;           Shell_TrayWnd -> TrayNotifyWnd -> (WinXP:SysPager) -> ToolbarWindow32
;    Onput: if  fail eax=NULL, else eax = handle
;/
GetSysTrayToolBarHandle  proc
    ;--- Get the handle of task bar
    invoke FindWindow,  addr g_szTaskBarCls, NULL
    cmp      eax, NULL
    je         @GetSysTrayToolBarHandleRet  ; fail

      ; HWND FindWindowEx(
      ;     HWND hwndParent,    // handle to parent window
      ;     HWND hwndChildAfter,    // handle to a child window 
      ;     LPCTSTR lpszClass,    // pointer to class name
      ;     LPCTSTR lpszWindow    // pointer to window name
      ; );

     ;--- Get the handle of TrayNotifyWnd in task bar
     invoke FindWindowEx,  eax, NULL,   addr  g_szTrayNotifyWndCls,   NULL
     cmp     eax, NULL
     je         @GetSysTrayToolBarHandleRet

     ;--- (WinXP Only) Get the handle of g_szSysPager in TrayNotifyWnd 
     push  eax
     invoke FindWindowEx,  eax, NULL,   addr  g_szSysPagerCls, NULL
     .if    ( eax==NULL)
         pop  eax
     .else
         pop  edi
     .endif

     ;--- Get the handle of ToolbarWindow32 in TrayNotifyWnd
     invoke FindWindowEx,  eax, NULL,   addr  g_szToolbarWindow32Cls,   NULL

@GetSysTrayToolBarHandleRet:
     ret
GetSysTrayToolBarHandle  endp


;/
; Function: Enum the Child window in task bar
;/
EnumSubCtl  proc  proc  hWnd: HWND, lParam: LPARAM
     invoke GetClassName, hWnd,  addr g_szIconText, sizeof g_szIconText
     invoke MessageBox, NULL,  addr g_szIconText,  addr g_szIconText, MB_OK

     mov      eax,  TRUE
     ret
EnumSubCtl  endp


;/
; Function: Get the count of icon in system tray
;/
GetSysTrayIconCount  proc
     invoke SendMessage,g_hSysTray, TB_BUTTONCOUNT, 0, 0
     mov     g_dwTrayIconCount,  eax    
     ret
GetSysTrayIconCount  endp

;///
; Function: Enum the icon in system tray
;///
EnumSysTrayIcon  proc
     local dwProcID, dwReaded:  dword
     local hProcess: HANDLE
     local pMem:  dword

     invoke GetSysTrayIconCount

     invoke GetWindowThreadProcessId, g_hSysTray,  addr dwProcID
     invoke OpenProcess, PROCESS_VM_OPERATION  or PROCESS_VM_READ  or PROCESS_VM_WRITE,  FALSE, dwProcID
     mov    hProcess,  eax
     invoke VirtualAllocEx, hProcess,  NULL, 1024, MEM_RESERVE  or MEM_COMMIT,  PAGE_READWRITE
     mov    pMem,  eax

     xor       eax,  eax
     .while ( eax < g_dwTrayIconCount)
         push    eax

         invoke SendMessage, g_hSysTray, TB_GETBUTTON,  eax,  pMem
         invoke ReadProcessMemory, hProcess, pMem,  addr g_stTbButton, sizeof g_stTbButton,   addr dwReaded

         invoke SendMessage, g_hSysTray, TB_GETBUTTONTEXT, g_stTbButton.idCommand,  pMem

         inc        eax   ; If fail, the return value is  -1
         jz         @F    ; Fail, skip

         invoke ReadProcessMemory, hProcess,  pMem,    addr g_szIconText, sizeof g_szIconText,  addr dwReaded
         invoke MessageBox, NULL,  addr g_szIconText,  addr g_szAppName, NULL
@@:
         pop       eax
         inc        eax
     .endw

     invoke VirtualFreeEx, hProcess, pMem,  0,  MEM_RELEASE
     invoke CloseHandle, hProcess

     ret
EnumSysTrayIcon  endp

end start