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.
88 lines
2.7 KiB
Python
88 lines
2.7 KiB
Python
import ctypes
|
|
|
|
import windows
|
|
from windows import winproxy
|
|
import windows.generated_def as gdef
|
|
from windows.generated_def.winstructs import *
|
|
|
|
from windows.utils import AutoHandle
|
|
|
|
import sys
|
|
|
|
|
|
class LogicalDrive(AutoHandle):
|
|
DRIVE_TYPE = gdef.FlagMapper(DRIVE_UNKNOWN, DRIVE_NO_ROOT_DIR, DRIVE_REMOVABLE,
|
|
DRIVE_FIXED, DRIVE_REMOTE, DRIVE_CDROM, DRIVE_RAMDISK)
|
|
|
|
def __init__(self, name):
|
|
self.name = name
|
|
|
|
@property
|
|
def type(self):
|
|
"""The type of drive, values are:
|
|
|
|
* DRIVE_UNKNOWN(0x0L)
|
|
* DRIVE_NO_ROOT_DIR(0x1L)
|
|
* DRIVE_REMOVABLE(0x2L)
|
|
* DRIVE_FIXED(0x3L)
|
|
* DRIVE_REMOTE(0x4L)
|
|
* DRIVE_CDROM(0x5L)
|
|
* DRIVE_RAMDISK(0x6L)
|
|
|
|
:type: :class:`long` or :class:`int` (or subclass)
|
|
"""
|
|
t = winproxy.GetDriveTypeA(self.name)
|
|
return self.DRIVE_TYPE.get(t,t)
|
|
|
|
@property
|
|
def path(self):
|
|
"""The target path of the device
|
|
|
|
:type: :class:`str`"""
|
|
res = query_dos_device(self.name.strip("\\"))
|
|
if len(res) != 1:
|
|
raise ValueError("[Unexpected result] query_dos_device(logicaldrive) returned multiple path")
|
|
return res[0]
|
|
|
|
def query_info(self, info):
|
|
return windows.utils.query_volume_information(self.handle, info)
|
|
|
|
@property
|
|
def volume_info(self):
|
|
return self.query_info(gdef.FileFsVolumeInformation)
|
|
|
|
@property
|
|
def serial(self):
|
|
return self.volume_info.VolumeSerialNumber
|
|
|
|
def _get_handle(self):
|
|
nt_name = windows.utils.dospath_to_ntpath(self.name)
|
|
handle = windows.winproxy.CreateFileA(nt_name, gdef.GENERIC_READ,
|
|
gdef.FILE_SHARE_READ, None, gdef.OPEN_EXISTING, gdef.FILE_FLAG_BACKUP_SEMANTICS , None)
|
|
return handle
|
|
|
|
def __repr__(self):
|
|
return """<{0} "{1}" ({2})>""".format(type(self).__name__, self.name, self.type.name)
|
|
|
|
def enum_logical_drive():
|
|
return [LogicalDrive(name) for name in get_logical_drive_names()]
|
|
|
|
def get_logical_drive_names():
|
|
size = 0x100
|
|
buffer = ctypes.c_buffer(size)
|
|
rsize = winproxy.GetLogicalDriveStringsA(0x1000, buffer)
|
|
return buffer[:rsize].rstrip(b"\x00").split(b"\x00")
|
|
|
|
def get_info(drivename):
|
|
size = 0x1000
|
|
volume_name = ctypes.c_buffer(size)
|
|
fs_name = ctypes.c_buffer(size)
|
|
flags = DWORD()
|
|
winproxy.GetVolumeInformationA(drivename, volume_name, size, None, None, ctypes.byref(flags), fs_name, size)
|
|
return volume_name[:10], fs_name[:10]
|
|
|
|
def query_dos_device(name):
|
|
size = 0x1000
|
|
buffer = ctypes.c_buffer(size)
|
|
rsize = winproxy.QueryDosDeviceA(name, buffer, size)
|
|
return buffer[:rsize].rstrip(b"\x00").split(b"\x00") |