可以将文章内容翻译成中文,广告屏蔽插件可能会导致该功能失效(如失效，请关闭广告屏蔽插件后再试):
问题:
I'm trying to send the TB_GETBUTTON message to get info about the buttons inside this Toolbar control marked in red color:
( The System tray notification area )
The problem is that when I send the message, the Explorer refreshes itself, is very annonying because all the desktop refreshes, and also I'm not getting the proper values with the TBBUTTON structure definition that I'm using, I tested three different definitions, those with unions from pinvoke.net, and the one published here by @David Heffernan.
I'm running the code below in a 64-Bit Windows 10 and with the x64 config set in my project properties.
How can I fix the struct and the annonying system's refresh?.
These are the relevant definitions I'm using:
Const WM_USER As Integer = &H400
Const TB_BUTTONCOUNT As Integer = (WM_USER + 24)
Const TB_GETBUTTON As Integer = (WM_USER + 23)
' Toolbar values are defined in "CommCtrl.h" Windows SDK header files.
<StructLayout(LayoutKind.Sequential)>
Public Structure TBBUTTON64
Public iBitmap As Integer
Public idCommand As Integer
Public fsState As Byte
Public fsStyle As Byte
<MarshalAsAttribute(UnmanagedType.ByValArray, SizeConst:=6)> ' 6 on x64
Public bReserved As Byte()
Public dwData As UIntPtr
Public iString As IntPtr
End Structure
<DllImport("User32.dll", SetLastError:=True)>
Public Shared Function SendMessage(ByVal hwnd As IntPtr,
ByVal msg As Integer,
ByVal wParam As IntPtr,
ByVal lParam As IntPtr
) As IntPtr
End Function
<SuppressUnmanagedCodeSecurity>
<DllImport("User32.dll", SetLastError:=True, CharSet:=CharSet.Auto, BestFitMapping:=False, ThrowOnUnmappableChar:=True)>
Public Shared Function FindWindow(ByVal lpClassName As String,
ByVal lpWindowName As String
) As IntPtr
End Function
<SuppressUnmanagedCodeSecurity>
<DllImport("User32.dll", SetLastError:=True, CharSet:=CharSet.Auto, BestFitMapping:=False, ThrowOnUnmappableChar:=True)>
Public Shared Function FindWindowEx(ByVal hwndParent As IntPtr,
ByVal hwndChildAfter As IntPtr,
ByVal strClassName As String,
ByVal strWindowName As String
) As IntPtr
End Function
And this is the code to test them:
Dim tskBarHwnd As IntPtr =
NativeMethods.FindWindow("Shell_TrayWnd", Nothing)
Dim systrayBarHwnd As IntPtr =
NativeMethods.FindWindowEx(tskBarHwnd, IntPtr.Zero, "TrayNotifyWnd", Nothing)
Dim sysPagerHwnd As IntPtr =
NativeMethods.FindWindowEx(systrayBarHwnd, IntPtr.Zero, "SysPager", Nothing)
Dim ntfyBarHwnd As IntPtr =
NativeMethods.FindWindowEx(sysPagerHwnd, IntPtr.Zero, "ToolbarWindow32", Nothing)
Dim buttonCount As Integer =
NativeMethods.SendMessage(ntfyBarHwnd, TB_BUTTONCOUNT, IntPtr.Zero, IntPtr.Zero).ToInt32()
For index As Integer = 0 To (buttonCount - 1)
Dim btInfo As New TBBUTTON64
Dim alloc As IntPtr = Marshal.AllocHGlobal(Marshal.SizeOf(GetType(TBBUTTON64)))
Marshal.StructureToPtr(btInfo, alloc, fDeleteOld:=True)
NativeMethods.SendMessage(ntfyBarHwnd, TB_GETBUTTON, New IntPtr(index), alloc)
Marshal.PtrToStructure(Of TBBUTTON64)(alloc)
Marshal.FreeHGlobal(alloc)
' This line always prints "00000"
Console.WriteLine(btInfo.iBitmap &
btInfo.fsState &
btInfo.fsStyle &
btInfo.idCommand &
btInfo.iString.ToInt32())
Next index
UPDATE (25-MARCH-2019)
I came back on this need because now I need to hide a sys-tray icon of a external app. So I started investigating again these days...
Please note the commentary did by @Remy Lebeau:
TB_GETBUTTON can be sent to another process. You just have to give it
the address of a TBBUTTON that exists in the target process's address
space. Use VirtualAllocEx() to allocate it, then send the message,
then use ReadProcessMemory() to read its content.
I'm not sure at all how to reproduce the steps that he gave, but after investigating a lot I found a code that apparently does that, it seem to read the process memory to retrieve the icon text:
- Get ToolTip Text from Icon in System Tray
However, it is written in C#, using unsafe and fixed keywords and I'm not sure how to translate it entirely in a proper way. Also, and just as a personal opinion, I feel that code is not simplified in any way, and I see bad design practices with var nomenclatures like "b", "b2" and "b4" which I don't get their purpose at all...
And, if helpful, I also found this in C/C++:
- Shell Tray Info - Arrange your system tray icons
In resume, what I'm asking for is to reproduce in VB.NET code the solution pointed by @Remy Lebeau, or to translate and simplify the C# code that I mentioned.
This is the best I can do by the moment with the help of a code converter, note that this code IS NOT working (it is broken / not fully converted to VB.NET):
Private Function GetTBButton(ByVal hToolbar As IntPtr, ByVal i As Integer, ByRef tbButton As ToolBarButton64, ByRef text As String, ByRef ipWindowHandle As IntPtr) As Boolean
' One page
Const BUFFER_SIZE As Integer = &H1000
Dim localBuffer(BUFFER_SIZE - 1) As Byte
Dim processId As Integer = 0
Dim threadId As Integer = NativeMethods.GetWindowThreadProcessId(hToolbar, processId)
Dim hProcess As IntPtr = NativeMethods.OpenProcess(ProcessAccessRights.AllAccess, False, processId)
If hProcess = IntPtr.Zero Then
Debug.Assert(False)
Return False
End If
Dim ipRemoteBuffer As UIntPtr = NativeMethods.VirtualAllocEx(hProcess, IntPtr.Zero, New UIntPtr(BUFFER_SIZE), MemoryAllocationType.Commit, MemoryProtectionOptions.ReadWrite)
If ipRemoteBuffer = UIntPtr.Zero Then
Debug.Assert(False)
Return False
End If
' TBButton
'INSTANT VB TODO TASK: There is no equivalent to a 'fixed' block in VB:
' fixed (TBBUTTON* pTBButton = &tbButton)
Dim ipTBButton As New IntPtr(pTBButton)
Dim b As Integer = CInt(Math.Truncate(NativeMethods.SendMessage(hToolbar, TB.GETBUTTON, CType(i, IntPtr), ipRemoteBuffer)))
If b = 0 Then
Debug.Assert(False)
Return False
End If
' this is fixed
Dim dwBytesRead As Int32 = 0
Dim ipBytesRead As New IntPtr(& dwBytesRead)
'INSTANT VB TODO TASK: There is no VB equivalent to 'sizeof':
Dim b2 As Boolean = NativeMethods.ReadProcessMemory(hProcess, ipRemoteBuffer, ipTBButton, New UIntPtr(CUInt(Math.Truncate(Marshal.SizeOf(tbButton)))), ipBytesRead)
If Not b2 Then
Debug.Assert(False)
Return False
End If
'INSTANT VB NOTE: End of the original C# 'fixed' block.
' button text
'INSTANT VB TODO TASK: There is no equivalent to a 'fixed' block in VB:
' fixed (byte* pLocalBuffer = localBuffer)
Dim ipLocalBuffer As New IntPtr(pLocalBuffer)
Dim chars As Integer = CInt(Math.Truncate(NativeMethods.SendMessage(hToolbar, TB.GETBUTTONTEXTW, CType(tbButton.idCommand, IntPtr), ipRemoteBuffer)))
If chars = -1 Then
Debug.Assert(False)
Return False
End If
' this is fixed
Dim dwBytesRead As Integer = 0
Dim ipBytesRead As New IntPtr(& dwBytesRead)
Dim b4 As Boolean = NativeMethods.ReadProcessMemory(hProcess, ipRemoteBuffer, ipLocalBuffer, New UIntPtr(BUFFER_SIZE), ipBytesRead)
If Not b4 Then
Debug.Assert(False)
Return False
End If
text = Marshal.PtrToStringUni(ipLocalBuffer, chars)
If text = " " Then
text = String.Empty
End If
'INSTANT VB NOTE: End of the original C# 'fixed' block.
NativeMethods.VirtualFreeEx(hProcess, ipRemoteBuffer, UIntPtr.Zero, MemoryFreeType.Release)
NativeMethods.CloseHandle(hProcess)
Return True
End Function
Which in theory it would be called like this:
Dim sysTrayHwnd As IntPtr = NotificationAreaUtil.Hwnd
Dim btIndex As Integer = 0
Dim tbButton As New ToolBarButton64() ' TBBUTTON struct for a x64 process
Dim text As String
Dim ipHwnd As IntPtr
GetTBButton(sysTrayHwnd, btIndex , tbButton, text, ipHwnd)
UPDATE (13-APRIL-2019)
I tried to convert the C/C++ solution provided by @RbMm in this answer, however, I'm getting a related memory error through a System.AccessViolationException when I try to marshal the TBBUTTON structure at this line:
...
Dim ptbi As ToolBarButtonInfo = Marshal.PtrToStructure(Of ToolBarButtonInfo)(remoteBaseAddress)
...
Note that to ensure that the root of the problem was not a bad TBBUTTONINFOW definition from my side, instead of using Marshal.PtrToStructure() I just used the Marshal.ReadInt32() function to try read a single field on the specific offset, and I get the same kind of error.
Probably I'm doing something wrong because I don't manage C/C++. Here is my attempt of code conversion in VB.NET:
( I will omit to share the P/Invoke definitions to simplify the code example )
Dim sysTray As IntPtr = NotificationAreaUtil.Hwnd
Dim pid As Integer
If (NativeMethods.GetWindowThreadProcessId(sysTray, pid) <> 0) Then
Dim hProcess As IntPtr = NativeMethods.OpenProcess(ProcessAccessRights.VirtualMemoryOperation, False, pid)
If (hProcess <> IntPtr.Zero) Then
Dim hSection As IntPtr
Dim pageSize As ULong = 81920 ' LARGE_INTEGER
Dim viewSize As IntPtr ' SIZE_T
Dim baseAddress As IntPtr ' PVOID
Dim remoteBaseAddress As IntPtr ' PVOID
If (NativeMethods.NtCreateSection(hSection, SectionAccessRights.AllAccess,
IntPtr.Zero, pageSize,
MemoryProtectionOptions.ReadWrite,
SectionAttributes.Commit,
IntPtr.Zero) = NTStatus.SUCCESS) Then
If (NativeMethods.NtMapViewOfSection(hSection, NativeMethods.GetCurrentProcess(), baseAddress,
IntPtr.Zero, IntPtr.Zero, IntPtr.Zero, viewSize,
ViewOfSectionInherit.ViewUnmap,
MemoryAllocationType.Default,
MemoryProtectionOptions.ReadWrite) = NTStatus.SUCCESS) Then
If (NativeMethods.NtMapViewOfSection(hSection, hProcess, remoteBaseAddress,
IntPtr.Zero, IntPtr.Zero, IntPtr.Zero, viewSize,
ViewOfSectionInherit.ViewUnmap,
MemoryAllocationType.Default,
MemoryProtectionOptions.ReadWrite) = NTStatus.SUCCESS) Then
Dim btIndex As Integer = 3 ' Button index from which I'll try to retrieve a valid TBBUTTONINFOW struct.
' Const TBIF_BYINDEX As Integer = &H80000000
' Const TBIF_TEXT As Integer = &H2
If (NativeMethods.SendMessage(sysTray, ToolbarMessages.GetButtonInfoUnicode, New IntPtr(btIndex), remoteBaseAddress) <> IntPtr.Zero) Then
' AT THIS LINE THROWS THE ACCESSVIOLATIONEXCEPTION.
Dim ptbi As ToolBarButtonInfo = Marshal.PtrToStructure(Of ToolBarButtonInfo)(remoteBaseAddress)
Console.WriteLine(ptbi.CommandId)
Console.WriteLine(Marshal.PtrToStringUni(ptbi.Text))
Else
Throw New Win32Exception(Marshal.GetLastWin32Error())
End If
NativeMethods.NtUnmapViewOfSection(hProcess, remoteBaseAddress)
End If
NativeMethods.NtUnmapViewOfSection(NativeMethods.GetCurrentProcess(), baseAddress)
End If
NativeMethods.NtClose(hSection)
End If
NativeMethods.CloseHandle(hProcess)
End If
End If
And here is a code conversion(on-the-fly and not tested) to C# of the code above:
IntPtr sysTray = NotificationAreaUtil.Hwnd;
int pid = 0;
if (NativeMethods.GetWindowThreadProcessId(sysTray, pid) != 0)
{
IntPtr hProcess = NativeMethods.OpenProcess(ProcessAccessRights.VirtualMemoryOperation, false, pid);
if (hProcess != IntPtr.Zero)
{
IntPtr hSection = System.IntPtr.Zero;
ulong pageSize = 81920; // LARGE_INTEGER
IntPtr viewSize = System.IntPtr.Zero; // SIZE_T
IntPtr baseAddress = System.IntPtr.Zero; // PVOID
IntPtr remoteBaseAddress = System.IntPtr.Zero; // PVOID
if (NativeMethods.NtCreateSection(hSection, SectionAccessRights.AllAccess, IntPtr.Zero, pageSize, MemoryProtectionOptions.ReadWrite, SectionAttributes.Commit, IntPtr.Zero) == NTStatus.SUCCESS)
{
if (NativeMethods.NtMapViewOfSection(hSection, NativeMethods.GetCurrentProcess(), baseAddress, IntPtr.Zero, IntPtr.Zero, IntPtr.Zero, viewSize, ViewOfSectionInherit.ViewUnmap, MemoryAllocationType.Default, MemoryProtectionOptions.ReadWrite) == NTStatus.SUCCESS)
{
if (NativeMethods.NtMapViewOfSection(hSection, hProcess, remoteBaseAddress, IntPtr.Zero, IntPtr.Zero, IntPtr.Zero, viewSize, ViewOfSectionInherit.ViewUnmap, MemoryAllocationType.Default, MemoryProtectionOptions.ReadWrite) == NTStatus.SUCCESS)
{
int btIndex = 3; // Button index from which I'll try to retrieve a valid TBBUTTONINFOW struct.
if (NativeMethods.SendMessage(sysTray, ToolbarMessages.GetButtonInfoUnicode, new IntPtr(btIndex), remoteBaseAddress) != IntPtr.Zero)
{
// AT THIS LINE THROWS THE ACCESSVIOLATIONEXCEPTION.
ToolBarButtonInfo ptbi = Marshal.PtrToStructure<ToolBarButtonInfo>(remoteBaseAddress);
Console.WriteLine(ptbi.CommandId);
Console.WriteLine(Marshal.PtrToStringUni(ptbi.Text));
}
else
{
throw new Win32Exception(Marshal.GetLastWin32Error());
}
NativeMethods.NtUnmapViewOfSection(hProcess, remoteBaseAddress);
}
NativeMethods.NtUnmapViewOfSection(NativeMethods.GetCurrentProcess(), baseAddress);
}
NativeMethods.NtClose(hSection);
}
NativeMethods.CloseHandle(hProcess);
}
}
回答1:
Edit: The originally posted code was in error in assuming that Intptr could be coersed to work across the bitness boundary. That mistake has been corrected.
I have also expanded the tray button data retrieval to retrieve a the button icon if possible. The Code Project article: Shell Tray Info - Arrange your system tray icons was used as the basis for writing this .Net implementation.
Do note that retrieved Icon instances do not own their respective handles as those are still owned by the OS.
The TBBUTTON structure is a bit of pain as fields after fsStyle vary their size based on the OS bitness (32/64). The following works on my system Win 10 (64 bit) for both x86 and x64 compilations. I apologize for the length and apparent formatting (I use 2 char tab stops and SO really messes with the formatting of multiple tabs), but I wish to present all code used in the example.
First is my TBBUTTON declaration. It is defined as base class and a class for 32-bit and class for 64-bit OS. The base class has a factory method (TBBUTTON.CreateForOS) to return the proper implementation. I deal with the varying field sizes by declaring byte place-holders to receive the marshaled structure and re-assemble when needed.
Imports System.Runtime.InteropServices
' Ref: https://docs.microsoft.com/en-us/windows/desktop/api/commctrl/ns-commctrl-tbbutton
' For info on native type size: Windows Data Types
' https://docs.microsoft.com/en-us/windows/desktop/WinProg/windows-data-types
' typedef struct _TBBUTTON {
' int iBitmap;
' int idCommand;
' Byte fsState;
' Byte fsStyle;
' #If ...
' Byte bReserved[6]; - 64 bit
' #Else
' BYTE bReserved[2]; - 32 bit
' #End If
' DWORD_PTR dwData; ' DWORD_PTR = ULONG_PTR 32/64 bits OS=> 4/8 bytes
' INT_PTR iString; ' 32/64 bits OS => 4/8 bytes
' ref: How to Display Tooltips for Buttons
' https://docs.microsoft.com/en-us/windows/desktop/Controls/display-tooltips-for-buttons
' Set the tooltip text as the iString member of the TBBUTTON structure for each button.
' so iString is a pointer to the Tooltip text
' } TBBUTTON, *PTBBUTTON, *LPTBBUTTON;
<StructLayout(LayoutKind.Sequential)>
Friend MustInherit Class TBBUTTON
Public iBitmap As Int32
Public idCommand As Int32
Public fsState As NativeMethods.ToolBars.TBSTATE
Public fsStyle As Byte
Protected bReserved0 As Byte
Protected bReserved1 As Byte
Public Shared ReadOnly Property Is64Bit As Boolean
Get
Return Environment.Is64BitOperatingSystem
End Get
End Property
Public Shared Function CreateForOS() As TBBUTTON
Dim ret As TBBUTTON = Nothing
If AppBitnessMatchesOS() Then
If Environment.Is64BitOperatingSystem Then
ret = New TBBUTTON64
Else
ret = New TBBUTTON32
End If
Else
Throw New Exception($"Application is {If(Environment.Is64BitProcess, 64, 32)} bits and OS is {If(Environment.Is64BitOperatingSystem, 64, 32)}. Bitnesses much match.")
End If
Return ret
End Function
Private Shared Function AppBitnessMatchesOS() As Boolean
Return Environment.Is64BitProcess.Equals(Environment.Is64BitOperatingSystem)
End Function
Public ReadOnly Property MarshalSize As IntPtr
Get
Return New IntPtr(Marshal.SizeOf(Me))
End Get
End Property
Public MustOverride ReadOnly Property Reserved As Byte()
Public MustOverride ReadOnly Property DwData As IntPtr
Public MustOverride ReadOnly Property IString As IntPtr
End Class
<StructLayout(LayoutKind.Sequential)>
Friend NotInheritable Class TBBUTTON32 : Inherits TBBUTTON
Private _dwData As IntPtr
Private _iString As IntPtr
Public Overrides ReadOnly Property Reserved As Byte()
Get
Return New Byte() {bReserved0, bReserved1}
End Get
End Property
Public Overrides ReadOnly Property DwData As IntPtr
Get
Return _dwData
End Get
End Property
Public Overrides ReadOnly Property IString As IntPtr
Get
Return _iString
End Get
End Property
End Class
<StructLayout(LayoutKind.Sequential)>
Friend NotInheritable Class TBBUTTON64 : Inherits TBBUTTON
Protected bReserved2 As Byte
Protected bReserved3 As Byte
Protected bReserved4 As Byte
Protected bReserved5 As Byte
Private _dwData As IntPtr
Private _iString As IntPtr
Public Overrides ReadOnly Property Reserved As Byte()
Get
Return New Byte() {bReserved0, bReserved1, bReserved2, bReserved3, bReserved4, bReserved5}
End Get
End Property
Public Overrides ReadOnly Property DwData As IntPtr
Get
Return _dwData
End Get
End Property
Public Overrides ReadOnly Property IString As IntPtr
Get
Return _iString
End Get
End Property
End Class
Next is my native methods class. This class declares various function overloads to let the interop-marshaling system perform the necessary allocations/conversions.
Imports System.Diagnostics.CodeAnalysis
Imports System.Runtime.ConstrainedExecution
Imports System.Runtime.InteropServices
Imports System.Security
Friend Class NativeMethods
Public Const WM_User As Int32 = &H400
Public Shared Sub FreeHGlobal(ptr As IntPtr)
If ptr <> IntPtr.Zero Then
Marshal.FreeHGlobal(ptr)
End If
End Sub
Public Class ToolBars
#Region "Constants"
' values from CommCtrl.h
''' <summary>Retrieves a count of the buttons currently in the toolbar. </summary>
Public Const TB_BUTTONCOUNT As Int32 = WM_User + 24
Public Const TB_GETBUTTON As Int32 = WM_User + 23
Public Const TB_DELETEBUTTON As Int32 = WM_User + 22
Private Const TB_GETBUTTONINFOW As Int32 = WM_User + 63
Private Const TB_SETBUTTONINFOW As Int32 = WM_User + 64
Private Const TB_GETBUTTONINFOA As Int32 = WM_User + 65
Private Const TB_SETBUTTONINFOA As Int32 = WM_User + 66
''' <summary> The cbSize and dwMask members of this structure must be filled in prior to sending this message.</summary>
Public Const TB_GETBUTTONINFO As Int32 = TB_GETBUTTONINFOW
''' <summary> The cbSize and dwMask members of this structure must be filled in prior to sending this message.</summary>
Public Const TB_SETBUTTONINFO As Int32 = TB_SETBUTTONINFOW
Public Const TB_GETBUTTONTEXTA As Int32 = WM_User + 45
Public Const TB_GETBUTTONTEXTW As Int32 = WM_User + 75
Public Const TB_GETBUTTONTEXT As Int32 = TB_GETBUTTONTEXTW
Public Const TB_GETSTRINGW As Int32 = WM_User + 91
Public Const TB_GETSTRINGA As Int32 = WM_User + 92
''' <summary>This message returns the specified string from the toolbar's string pool. It does not necessarily correspond to the text string currently being displayed by a button.</summary>
Public Const TB_GETSTRING As Int32 = TB_GETSTRINGW
''' <summary>wParam and lParam must be zero. returns handle to the image list, or NULL if no image list is set.</summary>
Public Const TB_GETIMAGELIST As Int32 = WM_User + 49
''' <summary>Retrieves the index of the bitmap associated with a button in a toolbar.
''' wParam=Command identifier of the button whose bitmap index is to be retrieved.</summary>
Public Const TB_GETBITMAP As Int32 = WM_User + 44
<DllImport("comctl32.dll", SetLastError:=True)>
Public Shared Function ImageList_GetIcon(himl As IntPtr, imageIndex As Int32, flags As UInt32) As IntPtr
End Function
<DllImport("comctl32.dll", SetLastError:=True)>
Public Shared Function ImageList_GetImageCount(himl As IntPtr) As Int32
End Function
#End Region
Public Enum TBSTATE As Byte
CHECKED = &H1
PRESSED = &H2
ENABLED = &H4
HIDDEN = &H8
INDETERMINATE = &H10
WRAP = &H20
ELLIPSES = &H40
MARKED = &H80
End Enum
End Class
Public Class User32
#Region "Utility Methods"
Public Shared Function GetNotificationAreaToolBarHandle() As IntPtr
Dim hWndTray As IntPtr = FindWindow("Shell_TrayWnd", Nothing)
If hWndTray <> IntPtr.Zero Then
hWndTray = FindWindowEx(hWndTray, IntPtr.Zero, "TrayNotifyWnd", Nothing)
If hWndTray <> IntPtr.Zero Then
hWndTray = FindWindowEx(hWndTray, IntPtr.Zero, "SysPager", Nothing)
If hWndTray <> IntPtr.Zero Then
hWndTray = FindWindowEx(hWndTray, IntPtr.Zero, "ToolbarWindow32", Nothing)
Return hWndTray
End If
End If
End If
Return IntPtr.Zero
End Function
Public Shared Function GetTaskBarHandle() As IntPtr
Dim hWndTray As IntPtr = FindWindow("Shell_TrayWnd", Nothing)
If hWndTray <> IntPtr.Zero Then
hWndTray = FindWindowEx(hWndTray, IntPtr.Zero, "TrayNotifyWnd", Nothing)
If hWndTray <> IntPtr.Zero Then
'hWndTray = FindWindowEx(hWndTray, IntPtr.Zero, "SysPager", Nothing)
If hWndTray <> IntPtr.Zero Then
'hWndTray = FindWindowEx(hWndTray, IntPtr.Zero, "ToolbarWindow32", Nothing)
End If
End If
End If
Return hWndTray
End Function
#End Region
<DllImport("user32.dll", SetLastError:=True, CharSet:=CharSet.Unicode)>
Public Shared Function FindWindow(ByVal lpClassName As String, ByVal lpWindowName As String) As IntPtr
End Function
<DllImport("user32.dll", SetLastError:=True, CharSet:=CharSet.Unicode)>
Public Shared Function FindWindowEx(ByVal hwndParent As IntPtr, ByVal hwndChildAfter As IntPtr, ByVal lpszClass As String, ByVal lpszWindow As String) As IntPtr
End Function
<DllImport("User32.dll", SetLastError:=True, CharSet:=CharSet.Unicode)>
Public Shared Function GetWindowThreadProcessId(ByVal hWnd As IntPtr, <System.Runtime.InteropServices.Out()> ByRef lpdwProcessId As Int32) As Int32
End Function
#Region "SendMessage Overloads"
<DllImport("User32.dll", CharSet:=CharSet.Unicode, EntryPoint:="SendMessage")>
Public Shared Function SendMessage(ByVal hWnd As IntPtr, ByVal msg As Int32, ByVal wParam As IntPtr, ByVal lParam As System.Text.StringBuilder) As Int32
End Function
<DllImport("User32.dll", CharSet:=CharSet.Unicode, EntryPoint:="SendMessage")>
Public Shared Function SendMessage(ByVal hWnd As IntPtr, ByVal msg As Int32, ByVal wParam As IntPtr, ByVal lParam As IntPtr) As IntPtr
End Function
<DllImport("User32.dll", CharSet:=CharSet.Unicode)>
Public Shared Function SendMessage(ByVal hWnd As IntPtr, ByVal msg As Int32, ByVal wParam As Int32, ByVal lParam As Int32) As Int32
End Function
<DllImport("User32.dll", CharSet:=CharSet.Unicode)>
Public Shared Function SendMessage(ByVal hWnd As IntPtr, ByVal msg As Int32, ByVal wParam As Int32, ByVal lParam As IntPtr) As Boolean
End Function
#End Region
<StructLayout(LayoutKind.Sequential)>
Friend Structure ICONINFO
''' <summary>Specifies whether this structure defines an icon or a cursor. A value of TRUE specifies an icon; FALSE specifies a cursor.</summary>
Public fIcon As Boolean
''' <summary>Specifies the x-coordinate of a cursor's hot spot. If this structure defines an icon, the hot spot is always in the center of the icon, and this member is ignored.</summary>
Public xHotspot As Int32
''' <summary>Specifies the y-coordinate of the cursor's hot spot. If this structure defines an icon, the hot spot is always in the center of the icon, and this member is ignored.</summary>
Public yHotspot As Int32 '
''' <summary>(HBITMAP) Specifies the icon bitmask bitmap. If this structure defines a black and white icon, this bitmask is formatted so that the upper half is the icon AND bitmask and the lower half is the icon XOR bitmask. Under this condition, the height should be an even multiple of two. If this structure defines a color icon, this mask only defines the AND bitmask of the icon.</summary>
Public hbmMask As IntPtr '
''' <summary>(HBITMAP) Handle to the icon color bitmap. This member can be optional if this structure defines a black and white icon. The AND bitmask of hbmMask is applied with the SRCAND flag to the destination; subsequently, the color bitmap is applied (using XOR) to the estination by using the SRCINVERT flag.</summary>
Public hbmColor As IntPtr
End Structure
<DllImport("user32.dll")>
Shared Function GetIconInfo(ByVal hIcon As IntPtr, ByRef piconinfo As ICONINFO) As Boolean
End Function
End Class
Public Class Kernel32
<DllImport("kernel32.dll", SetLastError:=True)>
<ReliabilityContract(Consistency.WillNotCorruptState, Cer.Success)>
<SuppressUnmanagedCodeSecurity>
Public Shared Function CloseHandle(ByVal hObject As IntPtr) As <MarshalAs(UnmanagedType.Bool)> Boolean
End Function
#Region "OpenProcess"
<DllImport("kernel32.dll", SetLastError:=True)>
Public Shared Function OpenProcess(ByVal processAccess As ProcessAccessFlags, ByVal bInheritHandle As Boolean, ByVal processId As Int32) As IntPtr
End Function
<Flags>
Public Enum ProcessAccessFlags As UInt32
All = &H1F0FFF
Terminate = &H1
CreateThread = &H2
VirtualMemoryOperation = &H8
VirtualMemoryRead = &H10
VirtualMemoryWrite = &H20
DuplicateHandle = &H40
CreateProcess = &H80
SetQuota = &H100
SetInformation = &H200
QueryInformation = &H400
QueryLimitedInformation = &H1000
Synchronize = &H100000
End Enum
#End Region
#Region "ReadProcessMemory Overloads"
<DllImport("kernel32.dll", SetLastError:=True)>
Public Shared Function ReadProcessMemory(
ByVal hProcess As IntPtr,
ByVal lpBaseAddress As IntPtr,
ByVal lpBuffer As IntPtr,
<MarshalAs(UnmanagedType.SysInt)> ByVal iSize As IntPtr,
<MarshalAs(UnmanagedType.SysInt)> ByRef lpNumberOfBytesRead As IntPtr) As Boolean
End Function
<DllImport("kernel32.dll", SetLastError:=True, EntryPoint:="ReadProcessMemory")>
Public Shared Function ReadProcessMemory(
ByVal hProcess As IntPtr,
ByVal lpBaseAddress As IntPtr,
ByRef lpBuffer As Int32,
<MarshalAs(UnmanagedType.SysInt)> ByVal iSize As IntPtr,
<MarshalAs(UnmanagedType.SysInt)> ByRef lpNumberOfBytesRead As IntPtr) As Boolean
End Function
<DllImport("kernel32.dll", SetLastError:=True, EntryPoint:="ReadProcessMemory")>
Public Shared Function ReadProcessMemory(
ByVal hProcess As IntPtr,
ByVal lpBaseAddress As IntPtr,
ByRef lpBuffer As Int64,
<MarshalAs(UnmanagedType.SysInt)> ByVal iSize As IntPtr,
<MarshalAs(UnmanagedType.SysInt)> ByRef lpNumberOfBytesRead As IntPtr) As Boolean
End Function
<DllImport("kernel32.dll", SetLastError:=True, EntryPoint:="ReadProcessMemory")>
Public Shared Function ReadProcessMemory(
ByVal hProcess As IntPtr,
ByVal lpBaseAddress As IntPtr,
ByVal lpBuffer As TBBUTTON,
<MarshalAs(UnmanagedType.SysInt)> ByVal iSize As IntPtr,
<MarshalAs(UnmanagedType.SysInt)> ByRef lpNumberOfBytesRead As IntPtr) As Boolean
End Function
<DllImport("kernel32.dll", SetLastError:=True, EntryPoint:="ReadProcessMemory")>
Public Shared Function ReadProcessMemory(
ByVal hProcess As IntPtr,
ByVal lpBaseAddress As IntPtr,
ByVal lpBuffer As TrayData,
<MarshalAs(UnmanagedType.SysInt)> ByVal iSize As IntPtr,
<MarshalAs(UnmanagedType.SysInt)> ByRef lpNumberOfBytesRead As IntPtr) As Boolean
End Function
#End Region
#Region "VirtualAllocEx"
<DllImport("kernel32.dll", SetLastError:=True, ExactSpelling:=True)>
Shared Function VirtualAllocEx(ByVal hProcess As IntPtr,
ByVal lpAddress As IntPtr,
<MarshalAs(UnmanagedType.SysInt)> ByVal dwSize As IntPtr,
<MarshalAs(UnmanagedType.U4)> ByVal flAllocationType As AllocationType,
ByVal flProtect As MemoryProtection) As IntPtr
End Function
<Flags>
Public Enum AllocationType As UInt32
Commit = &H1000
Reserve = &H2000
Decommit = &H4000
Release = &H8000
Reset = &H80000
Physical = &H400000
TopDown = &H100000
WriteWatch = &H200000
LargePages = &H20000000
End Enum
<Flags>
Public Enum MemoryProtection As UInt32
Execute = &H10
ExecuteRead = &H20
ExecuteReadWrite = &H40
ExecuteWriteCopy = &H80
NoAccess = &H1
[ReadOnly] = &H2
ReadWrite = &H4
WriteCopy = &H8
GuardModifierflag = &H100
NoCacheModifierflag = &H200
WriteCombineModifierflag = &H400
End Enum
#End Region
#Region "VirtualFreeEx"
<DllImport("kernel32.dll")>
Public Shared Function VirtualFreeEx(ByVal hProcess As IntPtr,
ByVal lpAddress As IntPtr,
<MarshalAs(UnmanagedType.SysInt)> ByVal dwSize As IntPtr,
ByVal dwFreeType As FreeType) As Boolean
End Function
''' <summary>helper method to release memory allocated with VirtualAllocEx</summary>
''' <param name="lpAddress">ptr received from VirtualAllocEx</param>
''' <param name="hProcess">ptr to process received from OpenProcess</param>
Public Shared Sub ReleaseVirtualAlloc(ByRef lpAddress As IntPtr, hProcess As IntPtr)
If lpAddress <> IntPtr.Zero Then
VirtualFreeEx(hProcess, lpAddress, Nothing, FreeType.RELEASE)
lpAddress = IntPtr.Zero
End If
End Sub
<Flags()>
Public Enum FreeType As UInt32
DECOMMIT = &H4000
RELEASE = &H8000
End Enum
#End Region
End Class
End Class
There are two additional support classes defined below.
Friend Class TrayButtonInfo
Public Property Icon As Icon
Public Property Index As Int32
Public Property CommandID As Int32
Public Property DisplayText As String = String.Empty
Public Property ProcessFound As Boolean = False
Public Property ProcessName As String = String.Empty
Public Property ProcessID As Int32
Public Property State As NativeMethods.ToolBars.TBSTATE
End Class
...
Imports System.Runtime.InteropServices
<StructLayout(LayoutKind.Sequential)>
Friend Class TrayData
Public hWnd As IntPtr
Public uID As Int32
Public uCallbackMessage As UInt32
Public Reservered0 As UInt32
Public Reservered1 As UInt32
Public hIcon As IntPtr
Public ReadOnly Property OwningProcess As Process
Get
Dim ret As Process = Nothing
If hWnd <> IntPtr.Zero Then
Dim processIDOfButton As Int32
Dim threadId As Int32 = NativeMethods.User32.GetWindowThreadProcessId(hWnd, processIDOfButton)
Try ' Process.GetProcessById can throw an exception if the id is not found
ret = Process.GetProcessById(processIDOfButton)
Catch ex As Exception
' eat it
End Try
End If
Return ret
End Get
End Property
End Class
Now for the actual example code. This code prepare a list of TrayButtonInfo instances that can be searched to find a matching button. It also shows an example of this search and how to delete the button. I tried to explain the code in the comments, but feel free to ask about anything that is not clear.
Imports System.Runtime.InteropServices
Public Class Form1
Private Sub Button1_Click(sender As Object, e As EventArgs) Handles Button1.Click
Dim retrievedNotificationButtons As List(Of TrayButtonInfo) = PrepareButtonInfoList()
' Now we have filled retrievedNotificationButtons with the information for all the buttons
' "Realtek HD Audio Manager" is just for testing on my machine.
' apply criteria for finding the button you want to delete
' note that the button can be restored by restarting Explorer
Dim targetBtn As TrayButtonInfo = retrievedNotificationButtons.FirstOrDefault(Function(info) info.DisplayText.StartsWith("Realtek HD Audio Manager"))
If targetBtn IsNot Nothing Then ' delete the button
' the following statement will delete the button if uncommented
'Dim toolBarHwnd As IntPtr = NativeMethods.User32.GetNotificationAreaToolBarHandle()
'NativeMethods.User32.SendMessage(toolBarHwnd, NativeMethods.ToolBars.TB_DELETEBUTTON, targetBtn.Index, 0)
End If
DataGridView1.DataSource = retrievedNotificationButtons
End Sub
Private Function PrepareButtonInfoList() As List(Of TrayButtonInfo)
Dim ret As List(Of TrayButtonInfo)
' create a TBButon structure appropriate for the OS bitness
Dim btn As TBBUTTON = TBBUTTON.CreateForOS
Dim toolBarProcessHandle As IntPtr
Dim toolBarButtonProcessMemoryPtr As IntPtr
' obtain window handle of the notification area toolbar
Dim toolBarHwnd As IntPtr = NativeMethods.User32.GetNotificationAreaToolBarHandle()
' obtain id of process that owns the notification area toolbar, threadId is of no consequence
Dim proccessIDOwningToolBar As Int32
Dim threadId As Int32 = NativeMethods.User32.GetWindowThreadProcessId(toolBarHwnd, proccessIDOwningToolBar)
Try
' obtain handle to the toolbar process that will allow allocating,reading and writing
toolBarProcessHandle = NativeMethods.Kernel32.OpenProcess(NativeMethods.Kernel32.ProcessAccessFlags.VirtualMemoryOperation Or
NativeMethods.Kernel32.ProcessAccessFlags.VirtualMemoryRead Or
NativeMethods.Kernel32.ProcessAccessFlags.VirtualMemoryWrite,
False, proccessIDOwningToolBar)
' allocate memory in the toolbar process to hold the TBButton
' need ReadWrite access due to TB_GETBUTTON writing to the allocated memory
toolBarButtonProcessMemoryPtr = NativeMethods.Kernel32.VirtualAllocEx(toolBarProcessHandle,
Nothing,
btn.MarshalSize,
NativeMethods.Kernel32.AllocationType.Commit,
NativeMethods.Kernel32.MemoryProtection.ReadWrite)
' now we can request the toolbar to fill the allocated memory with a TBButton structure
' for each button in the notifiaction area.
' determine how many toolbar buttons are visible notification area contains
Dim buttonCount As Int32 = NativeMethods.User32.SendMessage(toolBarHwnd, NativeMethods.ToolBars.TB_BUTTONCOUNT, 0, 0)
ret = New List(Of TrayButtonInfo)(buttonCount)
For btnIndex As Int32 = 0 To buttonCount - 1
Dim btnInfo As New TrayButtonInfo With {.Index = btnIndex}
ret.Add(btnInfo)
If NativeMethods.User32.SendMessage(toolBarHwnd, NativeMethods.ToolBars.TB_GETBUTTON, btnIndex, toolBarButtonProcessMemoryPtr) Then
' the toolbar owning process has successfully filled toolBarButtonProcessMemoryPtr
' use a customize ReadProcessMemory that takes a TBButtonBase instance as the destination buffer
If NativeMethods.Kernel32.ReadProcessMemory(toolBarProcessHandle, toolBarButtonProcessMemoryPtr, btn, btn.MarshalSize, Nothing) Then
' btn has been loaded, get the data
btnInfo.CommandID = btn.idCommand
btnInfo.State = btn.fsState
' Note that per the documentation, TBBUTTON.iString can contain a pointer the Tooltip Text
' In testing it does, but I have not found out how to determine the length of the string
' without the length, a guess on the size of process memory to read must be made and that
' seems unwise when accessing memory.
' GetButtonText use the TB_GETBUTTONTEXT message. This message's documentation indicates
' that the retrieved text may differ from the Tooltip text. As Tooltip text can be provided
' via several mechanisms, this makes sense.
btnInfo.DisplayText = GetButtonText(btnInfo.CommandID, toolBarHwnd, toolBarProcessHandle)
' get the process pointed to by dwData
' according to: Code Project article: A tool to order the window buttons in your taskbar
' https://www.codeproject.com/Articles/10497/A-tool-to-order-the-window-buttons-in-your-taskbar
' this is a pointer to the window handle of the process that owns the button
' while I can find no documentation that this is true, it appears to work
GetButtonData(btn, toolBarProcessHandle, btnInfo)
End If ' ReadProcessMemoryToTBButton
End If
Next
Finally ' cleanup handles
If toolBarProcessHandle <> IntPtr.Zero Then
NativeMethods.Kernel32.ReleaseVirtualAlloc(toolBarButtonProcessMemoryPtr, toolBarProcessHandle)
End If
If toolBarProcessHandle <> IntPtr.Zero Then
NativeMethods.Kernel32.CloseHandle(toolBarProcessHandle)
toolBarProcessHandle = IntPtr.Zero
End If
End Try
Return ret
End Function
Private Function GetButtonText(CommandID As Int32, toolBarWindowHandle As IntPtr, toolBarProcessHandle As IntPtr) As String
Dim ret As String = String.Empty
'1st determine the number of characters to retrieve
Dim lenText As Int32 = NativeMethods.User32.SendMessage(toolBarWindowHandle, NativeMethods.ToolBars.TB_GETBUTTONTEXT, New IntPtr(CommandID), IntPtr.Zero).ToInt32
If lenText > 0 Then
Dim ptrToText As IntPtr
Dim localBuffer As IntPtr
Try
Dim numBytes As New IntPtr((lenText * 2) + 1) ' Unicode 2 bytes per character + 1 for null terminator
'need to allocate the string in the process space
ptrToText = NativeMethods.Kernel32.VirtualAllocEx(toolBarProcessHandle,
Nothing,
numBytes,
NativeMethods.Kernel32.AllocationType.Commit,
NativeMethods.Kernel32.MemoryProtection.ReadWrite)
Dim receivedLen As Int32 = NativeMethods.User32.SendMessage(toolBarWindowHandle, NativeMethods.ToolBars.TB_GETBUTTONTEXT, New IntPtr(CommandID), ptrToText).ToInt32
localBuffer = Marshal.AllocHGlobal(numBytes) ' allocate local buffer to receive bytes from process space
If NativeMethods.Kernel32.ReadProcessMemory(toolBarProcessHandle, ptrToText, localBuffer, numBytes, Nothing) Then
ret = Marshal.PtrToStringUni(localBuffer)
End If
Finally ' release handles to unmanaged memory
NativeMethods.Kernel32.ReleaseVirtualAlloc(ptrToText, toolBarProcessHandle)
NativeMethods.FreeHGlobal(localBuffer)
End Try
End If
Return ret
End Function
Private Sub GetButtonData(btn As TBBUTTON, toolBarProcessHandle As IntPtr, btnInfo As TrayButtonInfo)
Dim data As TrayData
Try
data = New TrayData()
Dim dataMarshalSize As New IntPtr(Marshal.SizeOf(data))
If NativeMethods.Kernel32.ReadProcessMemory(toolBarProcessHandle, btn.DwData, data, dataMarshalSize, Nothing) Then
' use GetIconInfo to validate icon handle
Dim iconInfo As New NativeMethods.User32.ICONINFO
If NativeMethods.User32.GetIconInfo(data.hIcon, iconInfo) Then
btnInfo.Icon = Icon.FromHandle(data.hIcon)
End If
Using p As Process = data.OwningProcess
If p IsNot Nothing Then
btnInfo.ProcessFound = True
btnInfo.ProcessID = p.Id
btnInfo.ProcessName = p.ProcessName
End If
End Using
End If
Catch ex As Exception
Debug.Print(ex.Message)
End Try
End Sub
End Class
One thing that I observed in limited testing is that intermitantly, addition characters are post-pended onto the retrieved button text. I tried zeroing the memory but that did not help. Hence the use of the StartsWith when searching for a target button to delete.
回答2:
we can send TB_GETBUTTONINFOW for get the button text. if we know what text must be for button - we can detect it by text and delete it via TB_DELETEBUTTON. the pointer to TBBUTTONINFOW of course must be valid in context of process, which own the hwnd. the pszText inside TBBUTTONINFOW also of course must be valid in context of target process.
for implement this - the best create section and map it in self and target process. this let easy manipulate with memory.
BOOL IsTargetButton(PCWSTR pszText);
HWND GetNotificationWnd()
{
HWND hwnd;
(hwnd = FindWindow(L"Shell_TrayWnd", 0)) &&
(hwnd = FindWindowExW(hwnd, 0, L"TrayNotifyWnd", 0)) &&
(hwnd = FindWindowExW(hwnd, 0, L"SysPager", 0)) &&
(hwnd = FindWindowExW(hwnd, 0, L"ToolbarWindow32", 0));
return hwnd;
}
void deleteInTray()
{
if (HWND hwnd = GetNotificationWnd())
{
if (ULONG iIndex = SendMessageW(hwnd, TB_BUTTONCOUNT, 0, 0))
{
ULONG dwProcessId;
if (GetWindowThreadProcessId(hwnd, &dwProcessId))
{
if (HANDLE hProcess = OpenProcess(PROCESS_VM_OPERATION, FALSE, dwProcessId))
{
HANDLE hSection;
LARGE_INTEGER Size = { 1 };
if (0 <= ZwCreateSection(&hSection, SECTION_ALL_ACCESS, 0, &Size, PAGE_READWRITE, SEC_COMMIT, 0))
{
SIZE_T ViewSize = 0;
union {
PVOID RemoteBaseAddress;
PBYTE pb2;
};
union {
PVOID BaseAddress;
TBBUTTONINFO* ptbi;
PBYTE pb;
};
if (0 <= ZwMapViewOfSection(hSection, NtCurrentProcess(), &(BaseAddress = 0),
0, 0, 0, &ViewSize, ViewUnmap, 0, PAGE_READWRITE))
{
if (0 <= ZwMapViewOfSection(hSection, hProcess, &(RemoteBaseAddress = 0),
0, 0, 0, &ViewSize, ViewUnmap, 0, PAGE_READWRITE))
{
ptbi->cbSize = sizeof(TBBUTTONINFO);
ptbi->dwMask = TBIF_BYINDEX|TBIF_TEXT;
ptbi->pszText = (PWSTR)(pb2 + sizeof(TBBUTTONINFO));
ptbi->cchText = ((int)ViewSize - sizeof(TBBUTTONINFO)) / sizeof(WCHAR);
PWSTR pszText = (PWSTR)(pb + sizeof(TBBUTTONINFO));
do
{
if (0 <= SendMessageW(hwnd, TB_GETBUTTONINFOW, --iIndex, (LPARAM)RemoteBaseAddress))
{
DbgPrint("%u: %S\n", iIndex, pszText);
if (IsTargetButton(pszText))
{
SendMessageW(hwnd, TB_DELETEBUTTON, iIndex, 0);
}
}
} while (iIndex);
ZwUnmapViewOfSection(hProcess, RemoteBaseAddress);
}
ZwUnmapViewOfSection(NtCurrentProcess(), BaseAddress);
}
CloseHandle(hSection);
}
CloseHandle(hProcess);
}
}
}
}
}
回答3:
You can't send the TB_GETBUTTON message to windows in another process and get valid results, because TB_GETBUTTON needs to manipulate a pointer to a struct, and window messages that do that have to marshal the struct. Very few windows messages do that. TB_GETBUTTONCOUNT works because it doesn't need to be marshaled.
{var%20f='http://v.t.sina.com.cn/share/share.php?appkey=1515056452',u=z||d.location,p=['&url=',e(u),'&title=',e(t||d.title),'&source=',e(r),'&sourceUrl=',e(l),'&content=',c||'gb2312','&pic=',e(p||'')].join('');function%20a(){if(!window.open([f,p].join(''),'mb',['toolbar=0,status=0,resizable=1,width=440,height=430,left=',(s.width-440)/2,',top=',(s.height-430)/2].join('')))u.href=[f,p].join('');};if(/Firefox/.test(navigator.userAgent))setTimeout(a,0);else%20a();})(screen,document,encodeURIComponent,'','','https://i.stack.imgur.com/lwe5c.png', '推荐 走好不送 的文章《TBBUTTON struct not working with SendMessage》','https://www.manongdao.com/article-957505.html','页面编码gb2312|utf-8默认gb2312'));){kind=link}