This repository has been archived by the owner on Nov 11, 2024. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 21
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
20 changed files
with
6,472 additions
and
0 deletions.
There are no files selected for viewing
Large diffs are not rendered by default.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,25 @@ | ||
|
||
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;; | ||
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;; | ||
|
||
Func _MemoryWriteGCD($iv_Address, $ah_Handle, $sv_Type, $subj) | ||
If Not IsArray($ah_Handle) Then | ||
SetError(1) | ||
Return 0 | ||
EndIf | ||
Local $v_buffer = DllStructCreate($sv_Type) | ||
If @error Then | ||
SetError(@error + 1) | ||
Return 0 | ||
EndIf | ||
DllStructSetData($v_buffer, 1, $subj) | ||
DllCall($ah_Handle[0], "int", "WriteProcessMemory", "int", $ah_Handle[1], "int", $iv_Address, "ptr", DllStructGetPtr($v_buffer), "int", DllStructGetSize($v_buffer), "int", "") | ||
If Not @error Then | ||
Return -1 | ||
Else | ||
SetError(6) | ||
Return 0 | ||
EndIf | ||
EndFunc | ||
|
||
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,281 @@ | ||
#include-once | ||
#region _Memory | ||
;================================================================================== | ||
; AutoIt Version: 3.1.127 (beta) | ||
; Language: English | ||
; Platform: All Windows | ||
; Author: Nomad | ||
; Requirements: These functions will only work with beta. | ||
;================================================================================== | ||
; Credits: wOuter - These functions are based on his original _Mem() functions. | ||
; But they are easier to comprehend and more reliable. These | ||
; functions are in no way a direct copy of his functions. His | ||
; functions only provided a foundation from which these evolved. | ||
;================================================================================== | ||
; | ||
; Functions: | ||
; | ||
;================================================================================== | ||
; Function: _MemoryOpen($iv_Pid[, $iv_DesiredAccess[, $iv_InheritHandle]]) | ||
; Description: Opens a process and enables all possible access rights to the | ||
; process. The Process ID of the process is used to specify which | ||
; process to open. You must call this function before calling | ||
; _MemoryClose(), _MemoryRead(), or _MemoryWrite(). | ||
; Parameter(s): $iv_Pid - The Process ID of the program you want to open. | ||
; $iv_DesiredAccess - (optional) Set to 0x1F0FFF by default, which | ||
; enables all possible access rights to the | ||
; process specified by the Process ID. | ||
; $iv_InheritHandle - (optional) If this value is TRUE, all processes | ||
; created by this process will inherit the access | ||
; handle. Set to 1 (TRUE) by default. Set to 0 | ||
; if you want it FALSE. | ||
; Requirement(s): None. | ||
; Return Value(s): On Success - Returns an array containing the Dll handle and an | ||
; open handle to the specified process. | ||
; On Failure - Returns 0 | ||
; @Error - 0 = No error. | ||
; 1 = Invalid $iv_Pid. | ||
; 2 = Failed to open Kernel32.dll. | ||
; 3 = Failed to open the specified process. | ||
; Author(s): Nomad | ||
; Note(s): | ||
;================================================================================== | ||
Func _MemoryOpen($iv_Pid, $iv_DesiredAccess = 0x1F0FFF, $iv_InheritHandle = 1) | ||
|
||
If Not ProcessExists($iv_Pid) Then | ||
SetError(1) | ||
Return 0 | ||
EndIf | ||
|
||
Local $ah_Handle[2] = [DllOpen('kernel32.dll')] | ||
|
||
If @Error Then | ||
SetError(2) | ||
Return 0 | ||
EndIf | ||
|
||
Local $av_OpenProcess = DllCall($ah_Handle[0], 'int', 'OpenProcess', 'int', $iv_DesiredAccess, 'int', $iv_InheritHandle, 'int', $iv_Pid) | ||
|
||
If @Error Then | ||
DllClose($ah_Handle[0]) | ||
SetError(3) | ||
Return 0 | ||
EndIf | ||
|
||
$ah_Handle[1] = $av_OpenProcess[0] | ||
|
||
Return $ah_Handle | ||
|
||
EndFunc | ||
|
||
;================================================================================== | ||
; Function: _MemoryRead($iv_Address, $ah_Handle[, $sv_Type]) | ||
; Description: Reads the value located in the memory address specified. | ||
; Parameter(s): $iv_Address - The memory address you want to read from. It must | ||
; be in hex format (0x00000000). | ||
; $ah_Handle - An array containing the Dll handle and the handle | ||
; of the open process as returned by _MemoryOpen(). | ||
; $sv_Type - (optional) The "Type" of value you intend to read. | ||
; This is set to 'dword'(32bit(4byte) signed integer) | ||
; by default. See the help file for DllStructCreate | ||
; for all types. An example: If you want to read a | ||
; word that is 15 characters in length, you would use | ||
; 'char[16]' since a 'char' is 8 bits (1 byte) in size. | ||
; Return Value(s): On Success - Returns the value located at the specified address. | ||
; On Failure - Returns 0 | ||
; @Error - 0 = No error. | ||
; 1 = Invalid $ah_Handle. | ||
; 2 = $sv_Type was not a string. | ||
; 3 = $sv_Type is an unknown data type. | ||
; 4 = Failed to allocate the memory needed for the DllStructure. | ||
; 5 = Error allocating memory for $sv_Type. | ||
; 6 = Failed to read from the specified process. | ||
; Author(s): Nomad | ||
; Note(s): Values returned are in Decimal format, unless specified as a | ||
; 'char' type, then they are returned in ASCII format. Also note | ||
; that size ('char[size]') for all 'char' types should be 1 | ||
; greater than the actual size. | ||
;================================================================================== | ||
Func _MemoryRead($iv_Address, $ah_Handle, $sv_Type = 'dword') | ||
|
||
If Not IsArray($ah_Handle) Then | ||
SetError(1) | ||
Return 0 | ||
EndIf | ||
|
||
Local $v_Buffer = DllStructCreate($sv_Type) | ||
|
||
If @Error Then | ||
SetError(@Error + 1) | ||
Return 0 | ||
EndIf | ||
|
||
DllCall($ah_Handle[0], 'int', 'ReadProcessMemory', 'int', $ah_Handle[1], 'int', $iv_Address, 'ptr', DllStructGetPtr($v_Buffer), 'int', DllStructGetSize($v_Buffer), 'int', '') | ||
|
||
If Not @Error Then | ||
Local $v_Value = DllStructGetData($v_Buffer, 1) | ||
Return $v_Value | ||
Else | ||
SetError(6) | ||
Return 0 | ||
EndIf | ||
|
||
EndFunc | ||
|
||
;================================================================================== | ||
; Function: _MemoryWrite($iv_Address, $ah_Handle, $v_Data[, $sv_Type]) | ||
; Description: Writes data to the specified memory address. | ||
; Parameter(s): $iv_Address - The memory address which you want to write to. | ||
; It must be in hex format (0x00000000). | ||
; $ah_Handle - An array containing the Dll handle and the handle | ||
; of the open process as returned by _MemoryOpen(). | ||
; $v_Data - The data to be written. | ||
; $sv_Type - (optional) The "Type" of value you intend to write. | ||
; This is set to 'dword'(32bit(4byte) signed integer) | ||
; by default. See the help file for DllStructCreate | ||
; for all types. An example: If you want to write a | ||
; word that is 15 characters in length, you would use | ||
; 'char[16]' since a 'char' is 8 bits (1 byte) in size. | ||
; Return Value(s): On Success - Returns 1 | ||
; On Failure - Returns 0 | ||
; @Error - 0 = No error. | ||
; 1 = Invalid $ah_Handle. | ||
; 2 = $sv_Type was not a string. | ||
; 3 = $sv_Type is an unknown data type. | ||
; 4 = Failed to allocate the memory needed for the DllStructure. | ||
; 5 = Error allocating memory for $sv_Type. | ||
; 6 = $v_Data is not in the proper format to be used with the | ||
; "Type" selected for $sv_Type, or it is out of range. | ||
; 7 = Failed to write to the specified process. | ||
; Author(s): Nomad | ||
; Note(s): Values sent must be in Decimal format, unless specified as a | ||
; 'char' type, then they must be in ASCII format. Also note | ||
; that size ('char[size]') for all 'char' types should be 1 | ||
; greater than the actual size. | ||
;================================================================================== | ||
Func _MemoryWrite($iv_Address, $ah_Handle, $v_Data, $sv_Type = 'dword') | ||
|
||
If Not IsArray($ah_Handle) Then | ||
SetError(1) | ||
Return 0 | ||
EndIf | ||
|
||
Local $v_Buffer = DllStructCreate($sv_Type) | ||
|
||
If @Error Then | ||
SetError(@Error + 1) | ||
Return 0 | ||
Else | ||
DllStructSetData($v_Buffer, 1, $v_Data) | ||
If @Error Then | ||
SetError(6) | ||
Return 0 | ||
EndIf | ||
EndIf | ||
|
||
DllCall($ah_Handle[0], 'int', 'WriteProcessMemory', 'int', $ah_Handle[1], 'int', $iv_Address, 'ptr', DllStructGetPtr($v_Buffer), 'int', DllStructGetSize($v_Buffer), 'int', '') | ||
|
||
If Not @Error Then | ||
Return 1 | ||
Else | ||
SetError(7) | ||
Return 0 | ||
EndIf | ||
|
||
EndFunc | ||
|
||
;================================================================================== | ||
; Function: _MemoryClose($ah_Handle) | ||
; Description: Closes the process handle opened by using _MemoryOpen(). | ||
; Parameter(s): $ah_Handle - An array containing the Dll handle and the handle | ||
; of the open process as returned by _MemoryOpen(). | ||
; Return Value(s): On Success - Returns 1 | ||
; On Failure - Returns 0 | ||
; @Error - 0 = No error. | ||
; 1 = Invalid $ah_Handle. | ||
; 2 = Unable to close the process handle. | ||
; Author(s): Nomad | ||
; Note(s): | ||
;================================================================================== | ||
Func _MemoryClose($ah_Handle) | ||
|
||
If Not IsArray($ah_Handle) Then | ||
SetError(1) | ||
Return 0 | ||
EndIf | ||
|
||
DllCall($ah_Handle[0], 'int', 'CloseHandle', 'int', $ah_Handle[1]) | ||
If Not @Error Then | ||
DllClose($ah_Handle[0]) | ||
Return 1 | ||
Else | ||
DllClose($ah_Handle[0]) | ||
SetError(2) | ||
Return 0 | ||
EndIf | ||
|
||
EndFunc | ||
|
||
;================================================================================== | ||
; Function: SetPrivilege( $privilege, $bEnable ) | ||
; Description: Enables (or disables) the $privilege on the current process | ||
; (Probably) requires administrator privileges to run | ||
; | ||
; Author(s): Larry (from autoitscript.com's Forum) | ||
; Notes(s): | ||
; http://www.autoitscript.com/forum/index.php?s=&showtopic=31248&view=findpost&p=223999 | ||
;================================================================================== | ||
|
||
Func SetPrivilege( $privilege, $bEnable ) | ||
Const $MY_TOKEN_ADJUST_PRIVILEGES = 0x0020 | ||
Const $MY_TOKEN_QUERY = 0x0008 | ||
Const $MY_SE_PRIVILEGE_ENABLED = 0x0002 | ||
Local $hToken, $SP_auxret, $SP_ret, $hCurrProcess, $nTokens, $nTokenIndex, $priv | ||
$nTokens = 1 | ||
$LUID = DLLStructCreate("dword;int") | ||
If IsArray($privilege) Then $nTokens = UBound($privilege) | ||
$TOKEN_PRIVILEGES = DLLStructCreate("dword;dword[" & (3 * $nTokens) & "]") | ||
$NEWTOKEN_PRIVILEGES = DLLStructCreate("dword;dword[" & (3 * $nTokens) & "]") | ||
$hCurrProcess = DLLCall("kernel32.dll","hwnd","GetCurrentProcess") | ||
$SP_auxret = DLLCall("advapi32.dll","int","OpenProcessToken","hwnd",$hCurrProcess[0], _ | ||
"int",BitOR($MY_TOKEN_ADJUST_PRIVILEGES,$MY_TOKEN_QUERY),"int*",0) | ||
If $SP_auxret[0] Then | ||
$hToken = $SP_auxret[3] | ||
DLLStructSetData($TOKEN_PRIVILEGES,1,1) | ||
$nTokenIndex = 1 | ||
While $nTokenIndex <= $nTokens | ||
If IsArray($privilege) Then | ||
$priv = $privilege[$nTokenIndex-1] | ||
Else | ||
$priv = $privilege | ||
EndIf | ||
$ret = DLLCall("advapi32.dll","int","LookupPrivilegeValue","str","","str",$priv, _ | ||
"ptr",DLLStructGetPtr($LUID)) | ||
If $ret[0] Then | ||
If $bEnable Then | ||
DLLStructSetData($TOKEN_PRIVILEGES,2,$MY_SE_PRIVILEGE_ENABLED,(3 * $nTokenIndex)) | ||
Else | ||
DLLStructSetData($TOKEN_PRIVILEGES,2,0,(3 * $nTokenIndex)) | ||
EndIf | ||
DLLStructSetData($TOKEN_PRIVILEGES,2,DllStructGetData($LUID,1),(3 * ($nTokenIndex-1)) + 1) | ||
DLLStructSetData($TOKEN_PRIVILEGES,2,DllStructGetData($LUID,2),(3 * ($nTokenIndex-1)) + 2) | ||
DLLStructSetData($LUID,1,0) | ||
DLLStructSetData($LUID,2,0) | ||
EndIf | ||
$nTokenIndex += 1 | ||
WEnd | ||
$ret = DLLCall("advapi32.dll","int","AdjustTokenPrivileges","hwnd",$hToken,"int",0, _ | ||
"ptr",DllStructGetPtr($TOKEN_PRIVILEGES),"int",DllStructGetSize($NEWTOKEN_PRIVILEGES), _ | ||
"ptr",DllStructGetPtr($NEWTOKEN_PRIVILEGES),"int*",0) | ||
$f = DLLCall("kernel32.dll","int","GetLastError") | ||
EndIf | ||
$NEWTOKEN_PRIVILEGES=0 | ||
$TOKEN_PRIVILEGES=0 | ||
$LUID=0 | ||
If $SP_auxret[0] = 0 Then Return 0 | ||
$SP_auxret = DLLCall("kernel32.dll","int","CloseHandle","hwnd",$hToken) | ||
If Not $ret[0] And Not $SP_auxret[0] Then Return 0 | ||
return $ret[0] | ||
EndFunc ;==>SetPrivilege | ||
|
||
#endregion |
Oops, something went wrong.