mirror of
https://github.com/naksyn/PythonMemoryModule
synced 2026-06-06 16:24:25 +00:00
naksyn
db1893910c
This update include support to passing command line parameters to unmanaged exe via PEB stomping. This technique is not working with every executable since it depends on which functions are used to pass arguments. Generally, to get a universally working technique would be required to hook GetCommandlineA GetCommandlineW __getmainargs and __wgetmainargs since PEB stomping won't cover all cases, more details here: https://blog-30cm-tw.translate.goog/2020/08/windows-c-mainargc-argv.html?_x_tr_sl=auto&_x_tr_tl=en&_x_tr_hl=it&_x_tr_pto=wapp However, during my testing I found that mimikatz and several go binaries are working just by doing PEB stomping. On the other hand, cmdline passing via PEB stomping alone to mingw and VS compiled binaries won't likely work.
194 lines
8.3 KiB
Python
194 lines
8.3 KiB
Python
import ctypes
|
|
import windows.generated_def as gdef
|
|
from windows.pycompat import int_types
|
|
|
|
from ..apiproxy import ApiProxy, NeededParameter
|
|
from ..error import fail_on_zero
|
|
|
|
class DbgHelpProxy(ApiProxy):
|
|
APIDLL = "dbghelp"
|
|
default_error_check = staticmethod(fail_on_zero)
|
|
|
|
# We keep the simple definition where callback UserContext are PVOID
|
|
# Be we want to be able to pass arbitrary python object (list/dict)
|
|
# So ctypes magic to make the py_object->pvoid transformation
|
|
# !! this code loose a ref to obj.
|
|
# Should still work as our calling-caller method keep a ref
|
|
def transform_pyobject_to_pvoid(obj):
|
|
if obj is None or isinstance(obj, int_types):
|
|
return obj
|
|
return ctypes.POINTER(gdef.PVOID)(ctypes.py_object(obj))[0]
|
|
|
|
@DbgHelpProxy()
|
|
def SymInitialize(hProcess, UserSearchPath, fInvadeProcess):
|
|
return SymInitialize.ctypes_function(hProcess, UserSearchPath, fInvadeProcess)
|
|
|
|
@DbgHelpProxy()
|
|
def SymCleanup(hProcess):
|
|
return SymCleanup.ctypes_function(hProcess)
|
|
|
|
@DbgHelpProxy()
|
|
def SymLoadModuleExA(hProcess, hFile, ImageName, ModuleName, BaseOfDll, DllSize, Data, Flags):
|
|
return SymLoadModuleExA.ctypes_function(hProcess, hFile, ImageName, ModuleName, BaseOfDll, DllSize, Data, Flags)
|
|
|
|
@DbgHelpProxy()
|
|
def SymLoadModuleExW(hProcess, hFile, ImageName, ModuleName, BaseOfDll, DllSize, Data, Flags):
|
|
return SymLoadModuleExW.ctypes_function(hProcess, hFile, ImageName, ModuleName, BaseOfDll, DllSize, Data, Flags)
|
|
|
|
@DbgHelpProxy()
|
|
def SymUnloadModule64(hProcess, BaseOfDll):
|
|
return SymUnloadModule64.ctypes_function(hProcess, BaseOfDll)
|
|
|
|
@DbgHelpProxy()
|
|
def SymFromAddr(hProcess, Address, Displacement, Symbol):
|
|
return SymFromAddr.ctypes_function(hProcess, Address, Displacement, Symbol)
|
|
|
|
@DbgHelpProxy()
|
|
def SymGetModuleInfo64(hProcess, dwAddr, ModuleInfo):
|
|
return SymGetModuleInfo64.ctypes_function(hProcess, dwAddr, ModuleInfo)
|
|
|
|
@DbgHelpProxy()
|
|
def SymFromName(hProcess, Name, Symbol):
|
|
return SymFromName.ctypes_function(hProcess, Name, Symbol)
|
|
|
|
@DbgHelpProxy()
|
|
def SymLoadModuleEx(hProcess, hFile, ImageName, ModuleName, BaseOfDll, DllSize, Data, Flags):
|
|
return SymLoadModuleEx.ctypes_function(hProcess, hFile, ImageName, ModuleName, BaseOfDll, DllSize, Data, Flags)
|
|
|
|
@DbgHelpProxy(error_check=None)
|
|
def SymSetOptions(SymOptions):
|
|
return SymSetOptions.ctypes_function(SymOptions)
|
|
|
|
@DbgHelpProxy(error_check=None)
|
|
def SymGetOptions():
|
|
return SymGetOptions.ctypes_function()
|
|
|
|
@DbgHelpProxy()
|
|
def SymGetSearchPath(hProcess, SearchPath, SearchPathLength=None):
|
|
if SearchPath and SearchPathLength is None:
|
|
SearchPathLength = len(SearchPath)
|
|
return SymGetSearchPath.ctypes_function(hProcess, SearchPath, SearchPathLength)
|
|
|
|
@DbgHelpProxy()
|
|
def SymGetSearchPathW(hProcess, SearchPath, SearchPathLength=None):
|
|
if SearchPath and SearchPathLength is None:
|
|
SearchPathLength = len(SearchPath)
|
|
return SymGetSearchPathW.ctypes_function(hProcess, SearchPath, SearchPathLength)
|
|
|
|
@DbgHelpProxy()
|
|
def SymSetSearchPath(hProcess, SearchPath):
|
|
return SymSetSearchPath.ctypes_function(hProcess, SearchPath)
|
|
|
|
@DbgHelpProxy()
|
|
def SymSetSearchPathW(hProcess, SearchPath):
|
|
return SymSetSearchPathW.ctypes_function(hProcess, SearchPath)
|
|
|
|
|
|
@DbgHelpProxy()
|
|
def SymGetTypeInfo(hProcess, ModBase, TypeId, GetType, pInfo):
|
|
return SymGetTypeInfo.ctypes_function(hProcess, ModBase, TypeId, GetType, pInfo)
|
|
|
|
@DbgHelpProxy()
|
|
def SymEnumSymbols(hProcess, BaseOfDll, Mask, EnumSymbolsCallback, UserContext=None):
|
|
UserContext = transform_pyobject_to_pvoid(UserContext)
|
|
return SymEnumSymbols.ctypes_function(hProcess, BaseOfDll, Mask, EnumSymbolsCallback, UserContext)
|
|
|
|
@DbgHelpProxy()
|
|
def SymEnumSymbolsEx(hProcess, BaseOfDll, Mask, EnumSymbolsCallback, UserContext=None, Options=NeededParameter):
|
|
UserContext = transform_pyobject_to_pvoid(UserContext)
|
|
return SymEnumSymbolsEx.ctypes_function(hProcess, BaseOfDll, Mask, EnumSymbolsCallback, UserContext, Options)
|
|
|
|
@DbgHelpProxy()
|
|
def SymEnumSymbolsForAddr(hProcess, Address, EnumSymbolsCallback, UserContext=None):
|
|
UserContext = transform_pyobject_to_pvoid(UserContext)
|
|
return SymEnumSymbolsForAddr.ctypes_function(hProcess, Address, EnumSymbolsCallback, UserContext)
|
|
|
|
@DbgHelpProxy()
|
|
def SymEnumSymbolsForAddrW(hProcess, Address, EnumSymbolsCallback, UserContext=None):
|
|
UserContext = transform_pyobject_to_pvoid(UserContext)
|
|
return SymEnumSymbolsForAddrW.ctypes_function(hProcess, Address, EnumSymbolsCallback, UserContext)
|
|
|
|
@DbgHelpProxy()
|
|
def SymEnumTypes(hProcess, BaseOfDll, EnumSymbolsCallback, UserContext=None):
|
|
UserContext = transform_pyobject_to_pvoid(UserContext)
|
|
return SymEnumTypes.ctypes_function(hProcess, BaseOfDll, EnumSymbolsCallback, UserContext)
|
|
|
|
@DbgHelpProxy()
|
|
def SymEnumTypesByName(hProcess, BaseOfDll, mask, EnumSymbolsCallback, UserContext=None):
|
|
UserContext = transform_pyobject_to_pvoid(UserContext)
|
|
return SymEnumTypesByName.ctypes_function(hProcess, BaseOfDll, mask, EnumSymbolsCallback, UserContext)
|
|
|
|
@DbgHelpProxy()
|
|
def SymEnumerateModules64(hProcess, EnumModulesCallback, UserContext=None):
|
|
UserContext = transform_pyobject_to_pvoid(UserContext)
|
|
return SymEnumerateModules64.ctypes_function(hProcess, EnumModulesCallback, UserContext)
|
|
|
|
@DbgHelpProxy()
|
|
def SymGetTypeFromName(hProcess, BaseOfDll, Name, Symbol):
|
|
return SymGetTypeFromName.ctypes_function(hProcess, BaseOfDll, Name, Symbol)
|
|
|
|
@DbgHelpProxy()
|
|
def SymSearch(hProcess, BaseOfDll, Index, SymTag, Mask, Address, EnumSymbolsCallback, UserContext, Options):
|
|
UserContext = transform_pyobject_to_pvoid(UserContext)
|
|
return SymSearch.ctypes_function(hProcess, BaseOfDll, Index, SymTag, Mask, Address, EnumSymbolsCallback, UserContext, Options)
|
|
|
|
@DbgHelpProxy()
|
|
def SymSearchW(hProcess, BaseOfDll, Index, SymTag, Mask, Address, EnumSymbolsCallback, UserContext, Options):
|
|
UserContext = transform_pyobject_to_pvoid(UserContext)
|
|
return SymSearchW.ctypes_function(hProcess, BaseOfDll, Index, SymTag, Mask, Address, EnumSymbolsCallback, UserContext, Options)
|
|
|
|
@DbgHelpProxy()
|
|
def SymRefreshModuleList(hProcess):
|
|
return SymRefreshModuleList.ctypes_function(hProcess)
|
|
|
|
# Helpers
|
|
|
|
@DbgHelpProxy()
|
|
def SymFunctionTableAccess(hProcess, AddrBase):
|
|
return SymFunctionTableAccess.ctypes_function(hProcess, AddrBase)
|
|
|
|
@DbgHelpProxy()
|
|
def SymFunctionTableAccess64(hProcess, AddrBase):
|
|
return SymFunctionTableAccess64.ctypes_function(hProcess, AddrBase)
|
|
|
|
@DbgHelpProxy()
|
|
def SymGetModuleBase(hProcess, dwAddr):
|
|
return SymGetModuleBase.ctypes_function(hProcess, dwAddr)
|
|
|
|
@DbgHelpProxy()
|
|
def SymGetModuleBase64(hProcess, qwAddr):
|
|
return SymGetModuleBase64.ctypes_function(hProcess, qwAddr)
|
|
|
|
@DbgHelpProxy()
|
|
def SymEnumProcesses(EnumProcessesCallback, UserContext=None):
|
|
return SymEnumProcesses.ctypes_function(EnumProcessesCallback, UserContext)
|
|
|
|
## Sym callback
|
|
|
|
@DbgHelpProxy()
|
|
def SymRegisterCallback(hProcess, CallbackFunction, UserContext=None):
|
|
return SymRegisterCallback.ctypes_function(hProcess, CallbackFunction, UserContext)
|
|
|
|
@DbgHelpProxy()
|
|
def SymRegisterCallback64(hProcess, CallbackFunction, UserContext=0):
|
|
return SymRegisterCallback64.ctypes_function(hProcess, CallbackFunction, UserContext)
|
|
|
|
@DbgHelpProxy()
|
|
def SymRegisterCallbackW64(hProcess, CallbackFunction, UserContext=0):
|
|
return SymRegisterCallbackW64.ctypes_function(hProcess, CallbackFunction, UserContext)
|
|
|
|
|
|
# Stack walk
|
|
|
|
@DbgHelpProxy()
|
|
def StackWalk64(MachineType, hProcess, hThread, StackFrame, ContextRecord, ReadMemoryRoutine, FunctionTableAccessRoutine, GetModuleBaseRoutine, TranslateAddress):
|
|
return StackWalk64.ctypes_function(MachineType, hProcess, hThread, StackFrame, ContextRecord, ReadMemoryRoutine, FunctionTableAccessRoutine, GetModuleBaseRoutine, TranslateAddress)
|
|
|
|
@DbgHelpProxy()
|
|
def StackWalkEx(MachineType, hProcess, hThread, StackFrame, ContextRecord, ReadMemoryRoutine, FunctionTableAccessRoutine, GetModuleBaseRoutine, TranslateAddress, Flags):
|
|
return StackWalkEx.ctypes_function(MachineType, hProcess, hThread, StackFrame, ContextRecord, ReadMemoryRoutine, FunctionTableAccessRoutine, GetModuleBaseRoutine, TranslateAddress, Flags)
|
|
|
|
@DbgHelpProxy()
|
|
def StackWalk(MachineType, hProcess, hThread, StackFrame, ContextRecord, ReadMemoryRoutine, FunctionTableAccessRoutine, GetModuleBaseRoutine, TranslateAddress):
|
|
return StackWalk.ctypes_function(MachineType, hProcess, hThread, StackFrame, ContextRecord, ReadMemoryRoutine, FunctionTableAccessRoutine, GetModuleBaseRoutine, TranslateAddress)
|