-
Notifications
You must be signed in to change notification settings - Fork 12
/
ExtListView.ahk
279 lines (246 loc) · 15.4 KB
/
ExtListView.ahk
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
; ----------------------------------------------------------------------------------------------------------------------
; Name .........: ExtListView library
; Description ..: Collection of functions dealing with the ListViews of external processes.
; AHK Version ..: AHK_L 1.1.13.01 x32/64 ANSI/Unicode
; Author .......: Cyruz - http://ciroprincipe.info
; License ......: WTFPL - http://www.wtfpl.net/txt/copying/
; Changelog ....: May 18, 2013 - v0.1 - First revision.
; ..............: Jun. 28, 2013 - v0.2 - Added resizable buffer option.
; ..............: Feb. 04, 2014 - v0.3 - Unicode and x64 compatibility.
; ..............: Apr. 10, 2014 - v1.0 - Code refactoring. Added encoding option and simple error management.
; ..............: May 04, 2014 - v1.1 - Detached the handles and memory allocation code.
; ..............: May 05, 2014 - v1.2 - Created ExtListView_GetAllItems and ExtListView_ToggleSelection functions.
; ..............: May 06, 2014 - v2.0 - Complete rewrite of the library. Code more modular and updateable. Separated
; ..............: code for De/Initialization, GetNextItem and GetItemText.
; ----------------------------------------------------------------------------------------------------------------------
; ----------------------------------------------------------------------------------------------------------------------
; Function .....: ExtListView_GetSingleItem
; Description ..: Get the first item with the desired state.
; Parameters ...: objLV - External ListView initialized object.
; ..............: sState - Status of the searched item. Common statuses are:
; ..............: LVNI_ALL - 0x0000
; ..............: LVNI_FOCUSED - 0x0001
; ..............: LVNI_SELECTED - 0x0002
; ..............: LVNI_CUT - 0x0004
; ..............: LVNI_DROPHILITED - 0x0008
; ..............: nCol - Column of the desired item (0-based index).
; Info .........: For more info on the sState parameter have a look at the MSDN docs for the LVM_GETNEXTITEM message:
; ..............: http://msdn.microsoft.com/en-us/library/windows/desktop/bb761057%28v=vs.85%29.aspx
; Return .......: Single item as a string.
; ----------------------------------------------------------------------------------------------------------------------
ExtListView_GetSingleItem(ByRef objLV, sState, nCol) {
Try {
If ( (nRow := ExtListView_GetNextItem(objLV, -1, sState)) != 0xFFFFFFFF )
sItem := ExtListView_GetItemText(objLV, nRow, nCol)
} Catch e
Throw e
Return (sItem) ? sItem : 0
}
; ----------------------------------------------------------------------------------------------------------------------
; Function .....: ExtListView_GetAllItems
; Description ..: Get all items that share the same status on the target ListView.
; Parameters ...: objLV - External ListView initialized object.
; ..............: sState - Status of the searched item. Common statuses are:
; ..............: LVNI_ALL - 0x0000
; ..............: LVNI_FOCUSED - 0x0001
; ..............: LVNI_SELECTED - 0x0002
; ..............: LVNI_CUT - 0x0004
; ..............: LVNI_DROPHILITED - 0x0008
; Info .........: For more infor on the sState parameter have a look at the MSDN docs for the LVM_GETNEXTITEM message:
; ..............: http://msdn.microsoft.com/en-us/library/windows/desktop/bb761057%28v=vs.85%29.aspx
; Return .......: Multidimensional array containing ListView's items. Eg: array[row][column] := "SomeString".
; ----------------------------------------------------------------------------------------------------------------------
ExtListView_GetAllItems(ByRef objLV, sState:=0x0000) {
Try {
nRow := -1, objList := []
Loop
{
If ( (nRow := ExtListView_GetNextItem(objLV, nRow, sState)) == 0xFFFFFFFF )
Break
x := A_Index, objList[x] := []
Loop % objLV.cols
objList[x][A_Index] := ExtListView_GetItemText(objLV, nRow, A_Index-1)
}
} Catch e
Throw e
Return ( objList.MaxIndex() ) ? objList : 0
}
; ----------------------------------------------------------------------------------------------------------------------
; Function .....: ExtListView_ToggleSelection
; Description ..: Select/deselect items in the target ListView.
; Parameters ...: objLV - External ListView initialized object.
; ..............: bSelect - 1 for selection, 0 for deselection.
; ..............: nItem - -1 for all items or "n" (0-based) for a specific ListView item.
; ----------------------------------------------------------------------------------------------------------------------
ExtListView_ToggleSelection(ByRef objLV, bSelect:=1, nItem:=-1) {
VarSetCapacity( LVITEM, 20, 0 )
NumPut( 0x0008, LVITEM, 0 ) ; mask = LVIF_STATE = 0x0008.
NumPut( nItem, LVITEM, 4 ) ; iItem.
NumPut( 0, LVITEM, 8 ) ; iSubItem.
NumPut( (bSelect) ? 0x0002 : 0x0000, LVITEM, 12 ) ; state = LVIS_SELECTED = 0x0002 or 0x0000 (reset mask).
NumPut( 0x0002, LVITEM, 16 ) ; stateMask = LVIS_SELECTED = 0x0002.
If ( !DllCall( "WriteProcessMemory", Ptr,objLV.hproc, Ptr,objLV.pwritebuf, Ptr,&LVITEM, UInt,20, UInt,0 ) )
Throw Exception("objLV.pwritebuf: error writing memory", "WriteProcessMemory", "LastError: " A_LastError)
SendMessage, 0x102B, % nItem, % objLV.pwritebuf,, % "ahk_id " objLV.hlv
}
; ----------------------------------------------------------------------------------------------------------------------
; Function .....: ExtListView_GetNextItem
; Description ..: Get the next item in the target ListView.
; Parameters ...: objLV - External ListView initialized object.
; ..............: nRow - Row where to start the search for the next item (0-based index).
; ..............: lParam - Status of the searched item. Common statuses are:
; ..............: LVNI_ALL - 0x0000
; ..............: LVNI_FOCUSED - 0x0001
; ..............: LVNI_SELECTED - 0x0002
; ..............: LVNI_CUT - 0x0004
; ..............: LVNI_DROPHILITED - 0x0008
; Info .........: LVM_GETNEXTITEM - http://msdn.microsoft.com/en-us/library/windows/desktop/bb761057%28v=vs.85%29.aspx
; Return .......: Item content as a string.
; ----------------------------------------------------------------------------------------------------------------------
ExtListView_GetNextItem(ByRef objLV, nRow, lParam:=0x0000) {
; LVM_GETNEXTITEM = LVM_FIRST (0x1000) + 12 = 0x100C.
SendMessage, 0x100C, %nRow%, %lParam%,, % "ahk_id " objLV.hlv
Return ErrorLevel
}
; ----------------------------------------------------------------------------------------------------------------------
; Function .....: ExtListView_GetItemText
; Description ..: Get the text of the desired item.
; Parameters ...: objLV - External ListView initialized object.
; ..............: nRow - Row of the desired item (0-based index).
; ..............: nCol - Column of the desired item (0-based index).
; Return .......: Item content as a string.
; ----------------------------------------------------------------------------------------------------------------------
ExtListView_GetItemText(ByRef objLV, nRow, nCol) {
VarSetCapacity( LVITEM, objLV.szwritebuf, 0 )
NumPut( 0x0001, LVITEM, 0 ) ; mask = LVIF_TEXT = 0x0001.
NumPut( nRow, LVITEM, 4 ) ; iItem = Row to retrieve (0 = 1st row).
NumPut( nCol, LVITEM, 8 ) ; iSubItem = The column index of the item to retrieve.
NumPut( objLV.preadbuf, LVITEM, 20 ) ; pszText = Pointer to item text string.
NumPut( objLV.szreadbuf, LVITEM, (A_PtrSize == 4) ? 24 : 32 ) ; cchTextMax = Number of TCHARs in the buffer.
If ( !DllCall( "WriteProcessMemory", Ptr,objLV.hproc, Ptr,objLV.pwritebuf, Ptr,&LVITEM, UInt,objLV.szwritebuf
, UInt,0 ) )
Throw Exception("objLV.pwritebuf: error writing memory", "WriteProcessMemory", "LastError: " A_LastError)
; LVM_GETITEMTEXTA = LVM_FIRST (0x1000) + 45 = 0x102D.
; LVM_GETITEMTEXTW = LVM_FIRST (0x1000) + 115 = 0x1073.
LVM_GETITEMTEXT := (objLV.senc == "UTF-8" || objLV.senc == "UTF-16") ? 0x1073 : 0x102D
SendMessage, %LVM_GETITEMTEXT%, %nRow%, % objLV.pwritebuf,, % "ahk_id " objLV.hlv
VarSetCapacity(cRecvBuf, objLV.szreadbuf, 1)
If ( !DllCall( "ReadProcessMemory", Ptr,objLV.hproc, Ptr,objLV.preadbuf, Ptr,&cRecvBuf, UInt,objLV.szreadbuf
, Ptr,0 ) )
Throw Exception("objLV.preadbuf: error reading memory", "ReadProcessMemory", "LastError: " A_LastError)
Return StrGet(&cRecvBuf, objLV.szreadbuf, objLV.senc)
}
; ----------------------------------------------------------------------------------------------------------------------
; Function .....: ExtListView_Initialize
; Description ..: Initialize the object containing ListView's related data.
; Parameters ...: sWnd - Title of the window containing the ListView.
; ..............: szReadBuf - Size of the buffer used for reading the target process memory. It must be capable enough
; ..............: to hold the longest cell in the ListView.
; ..............: sEnc - Target ListView's encoding. "CP0" for ANSI, "UTF-8" or "UTF-16" for Unicode.
; Return .......: objLV - External ListView initialized object with the following keys:
; ..............: objLV.swnd - Title of the window owning the ListView.
; ..............: objLV.hwnd - Handle to the window owning the ListView.
; ..............: objLV.hproc - Handle to the process owning the ListView.
; ..............: objLV.hlv - Handle to the ListView.
; ..............: objLV.hhdr - Handle to the header of the ListView.
; ..............: objLV.rows - Number of rows in the ListView.
; ..............: objLV.cols - Number of columns in the ListView.
; ..............: objLV.senc - Encoding used by the process owning the ListView.
; ..............: objLV.pwritebuf - Address to the buffer used to write the LVITEM message to the target ListView.
; ..............: objLV.szwritebuf - Size of the write buffer.
; ..............: objLV.preadbuf - Address to the buffer used to read the answer to the message sent.
; ..............: objLV.szreadbuf - Size of the read buffer.
; ----------------------------------------------------------------------------------------------------------------------
ExtListView_Initialize(sWnd, szReadBuf:=1024, sEnc:="CP0") {
objLV := Object()
objLV.swnd := sWnd
objLV.hwnd := WinExist(sWnd)
objLV.szwritebuf := (A_PtrSize == 4) ? 28 : 36 ; Size of LVITEM
objLV.szreadbuf := szReadBuf
objLV.senc := sEnc
ControlGet, hLv, Hwnd,, SysListView321, % "ahk_id " objLV.hwnd
objLV.hlv := hLv
DllCall( "GetWindowThreadProcessId", Ptr,hLv, PtrP,dwProcessId )
; PROCESS_VM_OPERATION = 0x0008, PROCESS_VM_READ = 0x0010, PROCESS_VM_WRITE = 0x0020.
If ( !(objLV.hproc := DllCall( "OpenProcess", UInt,0x0008|0x0010|0x0020, Int,0, UInt,dwProcessId )) )
Throw Exception("objLV.hproc: error opening process", "OpenProcess", "LastError: " A_LastError)
; LVM_GETITEMCOUNT = LVM_FIRST (0x1000) + 4 = 0x1004.
SendMessage, 0x1004, 0, 0,, % "ahk_id " objLV.hlv
objLV.rows := ErrorLevel
; LVM_GETHEADER = LVM_FIRST (0x1000) + 31 = 0x101F.
SendMessage, 0x101F, 0, 0,, % "ahk_id " objLV.hlv
objLV.hhdr := ErrorLevel
; HDM_GETITEMCOUNT = HDM_FIRST (0x1200) + 0 = 0x1200.
SendMessage, 0x1200, 0, 0,, % "ahk_id " objLV.hhdr
objLV.cols := ErrorLevel
; Allocate memory on the target process before returning the object.
If ( !__ExtListView_AllocateMemory(objLV) )
Throw Exception("Error allocating memory", "__ExtListView_Initialize", "LastError: " A_LastError)
Return objLV
}
; ----------------------------------------------------------------------------------------------------------------------
; Function .....: ExtListView_DeInitialize
; Description ..: DeInitialize the object.
; Parameters ...: objLV - External ListView initialized object.
; ----------------------------------------------------------------------------------------------------------------------
ExtListView_DeInitialize(ByRef objLV) {
; Free the previously allocated memory on the target process.
If ( !__ExtListView_DeAllocateMemory(objLV) )
Throw Exception("Error deallocating memory", "__ExtListView_DeInitialize", "LastError: " A_LastError)
DllCall( "CloseHandle", Ptr,objLV.hproc )
objLV := ""
}
; ----------------------------------------------------------------------------------------------------------------------
; Function .....: ExtListView_CheckInitObject
; Description ..: Check if the object is still referring to a valid ListView.
; Parameters ...: objLV - External ListView initialized object.
; Return .......: 0 if false, handle of the window containing the ListView if true.
; ----------------------------------------------------------------------------------------------------------------------
ExtListView_CheckInitObject(ByRef objLV) {
Return WinExist("ahk_id " objLV.hwnd)
}
; ----------------------------------------------------------------------------------------------------------------------
; Function .....: __ExtListView_AllocateMemory
; Description ..: Allocates memory into the target process.
; Parameters ...: objLV - External ListView initialized object.
; ----------------------------------------------------------------------------------------------------------------------
__ExtListView_AllocateMemory(ByRef objLV) {
; MEM_COMMIT = 0x1000, PAGE_READWRITE = 0x4.
If ( !(objLV.pwritebuf := DllCall( "VirtualAllocEx", Ptr,objLV.hproc, Ptr,0, UInt,objLV.szwritebuf, UInt,0x1000
, UInt,0x4 )) )
Return 0
If ( !(objLV.preadbuf := DllCall( "VirtualAllocEx", Ptr,objLV.hproc, Ptr,0, UInt,objLV.szreadbuf, UInt,0x1000
, UInt,0x4 )) )
Return 0
Return 1
}
; ----------------------------------------------------------------------------------------------------------------------
; Function .....: __ExtListView_DeAllocateMemory
; Description ..: Frees previously allocated memory.
; Parameters ...: objLV - External ListView initialized object.
; ----------------------------------------------------------------------------------------------------------------------
__ExtListView_DeAllocateMemory(ByRef objLV) {
; MEM_RELEASE = 0x8000.
If ( !DllCall( "VirtualFreeEx", Ptr,objLV.hproc, Ptr,objLV.pwritebuf, UInt,0, UInt,0x8000 ) )
Return 0
objLV.Remove("pwritebuf")
If ( !DllCall( "VirtualFreeEx", Ptr,objLV.hproc, Ptr,objLV.preadbuf, UInt,0, UInt,0x8000 ) )
Return 0
objLV.Remove("preadbuf")
Return 1
}
/* EXAMPLE CODE:
OnExit, QUIT
WINTITLE = Active Directory Users and Computers ahk_class MMCMainFrame
objLV := ExtListView_Initialize(WINTITLE)
HotKey, IfWinActive, %WINTITLE%
HotKey, ^!g, TOGGLESEL
Return
TOGGLESEL:
( ExtListView_CheckInitObject(objLV) ) ? objLV := ExtListView_Initialize(WINTITLE)
ExtListView_ToggleSelection(objLV, 1, 0)
Return
QUIT:
ExtListView_DeInitialize(objLV)
ExitApp
*/