mirror of
https://github.com/rapid7/metasploit-payloads
synced 2025-03-24 18:16:24 +01:00
Land #597, Fix Python EventLog commands
This commit is contained in:
commit
acee11b308
@ -1440,7 +1440,10 @@ def stdapi_sys_power_exitwindows(request, response):
|
||||
@register_function_if(has_windll)
|
||||
def stdapi_sys_eventlog_open(request, response):
|
||||
source_name = packet_get_tlv(request, TLV_TYPE_EVENT_SOURCENAME)['value']
|
||||
handle = ctypes.windll.advapi32.OpenEventLogW(None, source_name)
|
||||
OpenEventLogA = ctypes.windll.advapi32.OpenEventLogA
|
||||
OpenEventLogA.argtypes = [ctypes.c_char_p, ctypes.c_char_p]
|
||||
OpenEventLogA.restype = ctypes.c_void_p
|
||||
handle = OpenEventLogA(None, bytes(source_name, 'UTF-8'))
|
||||
if not handle:
|
||||
return error_result_windows(), response
|
||||
response += tlv_pack(TLV_TYPE_EVENT_HANDLE, handle)
|
||||
@ -1451,13 +1454,15 @@ def stdapi_sys_eventlog_read(request, response):
|
||||
handle = packet_get_tlv(request, TLV_TYPE_EVENT_HANDLE)['value']
|
||||
flags = packet_get_tlv(request, TLV_TYPE_EVENT_READFLAGS)['value']
|
||||
offset = packet_get_tlv(request, TLV_TYPE_EVENT_RECORDOFFSET)['value']
|
||||
adv32 = ctypes.windll.advapi32
|
||||
bytes_read = ctypes.c_ulong(0)
|
||||
bytes_needed = ctypes.c_ulong(0)
|
||||
if adv32.ReadEventLogW(handle, flags, offset, ctypes.byref(bytes_read), 0, ctypes.byref(bytes_read), ctypes.byref(bytes_needed)):
|
||||
bytes_read = ctypes.c_uint32(0)
|
||||
bytes_needed = ctypes.c_uint32(0)
|
||||
ReadEventLogA = ctypes.windll.advapi32.ReadEventLogA
|
||||
ReadEventLogA.argtypes = [ctypes.c_void_p, ctypes.c_uint32, ctypes.c_uint32, ctypes.c_void_p, ctypes.c_uint32, ctypes.POINTER(ctypes.c_uint32), ctypes.POINTER(ctypes.c_uint32)]
|
||||
ReadEventLogA.restype = ctypes.c_bool
|
||||
if ReadEventLogA(handle, flags, offset, ctypes.byref(bytes_read), 0, ctypes.byref(bytes_read), ctypes.byref(bytes_needed)):
|
||||
return error_result_windows(), response
|
||||
buf = ctypes.create_unicode_buffer(bytes_needed.value)
|
||||
if not adv32.ReadEventLogW(handle, flags, offset, buf, bytes_needed, ctypes.byref(bytes_read), ctypes.byref(bytes_needed)):
|
||||
buf = (ctypes.c_uint8 * bytes_needed.value)()
|
||||
if not ReadEventLogA(handle, flags, offset, buf, bytes_needed, ctypes.byref(bytes_read), ctypes.byref(bytes_needed)):
|
||||
return error_result_windows(), response
|
||||
record = ctstruct_unpack(EVENTLOGRECORD, buf)
|
||||
response += tlv_pack(TLV_TYPE_EVENT_RECORDNUMBER, record.RecordNumber)
|
||||
@ -1466,8 +1471,9 @@ def stdapi_sys_eventlog_read(request, response):
|
||||
response += tlv_pack(TLV_TYPE_EVENT_ID, record.EventID)
|
||||
response += tlv_pack(TLV_TYPE_EVENT_TYPE, record.EventType)
|
||||
response += tlv_pack(TLV_TYPE_EVENT_CATEGORY, record.EventCategory)
|
||||
response += tlv_pack(TLV_TYPE_EVENT_DATA, buf.raw[record.DataOffset:record.DataOffset + record.DataLength])
|
||||
event_strings = buf.raw[record.StringOffset:].split('\x00', record.NumStrings)
|
||||
response += tlv_pack(TLV_TYPE_EVENT_DATA, ctarray_to_bytes(buf[record.DataOffset:record.DataOffset + record.DataLength]))
|
||||
event_string_buf = (ctypes.c_uint8 * len(buf[record.StringOffset:]))(*buf[record.StringOffset:])
|
||||
event_strings = ctarray_to_bytes(event_string_buf).split(NULL_BYTE, record.NumStrings)[:record.NumStrings]
|
||||
for event_string in event_strings:
|
||||
response += tlv_pack(TLV_TYPE_EVENT_STRING, event_string)
|
||||
return ERROR_SUCCESS, response
|
||||
@ -1475,14 +1481,20 @@ def stdapi_sys_eventlog_read(request, response):
|
||||
@register_function_if(has_windll)
|
||||
def stdapi_sys_eventlog_clear(request, response):
|
||||
handle = packet_get_tlv(request, TLV_TYPE_EVENT_HANDLE)['value']
|
||||
if not ctypes.windll.advapi32.ClearEventLogW(handle, None):
|
||||
ClearEventLogA = ctypes.windll.advapi32.ClearEventLogA
|
||||
ClearEventLogA.argtypes = [ctypes.c_void_p, ctypes.c_char_p]
|
||||
ClearEventLogA.restype = ctypes.c_bool
|
||||
if not ClearEventLogA(handle, None):
|
||||
return error_result_windows(), response
|
||||
return ERROR_SUCCESS, response
|
||||
|
||||
@register_function_if(has_windll)
|
||||
def stdapi_sys_eventlog_numrecords(request, response):
|
||||
handle = packet_get_tlv(request, TLV_TYPE_EVENT_HANDLE)['value']
|
||||
total = ctypes.c_ulong(0)
|
||||
total = ctypes.c_uint32(0)
|
||||
GetNumberOfEventLogRecords = ctypes.windll.advapi32.GetNumberOfEventLogRecords
|
||||
GetNumberOfEventLogRecords.argtypes = [ctypes.c_void_p, ctypes.POINTER(ctypes.c_uint32)]
|
||||
GetNumberOfEventLogRecords.restype = ctypes.c_bool
|
||||
if not ctypes.windll.advapi32.GetNumberOfEventLogRecords(handle, ctypes.byref(total)):
|
||||
return error_result_windows(), response
|
||||
response += tlv_pack(TLV_TYPE_EVENT_NUMRECORDS, total.value)
|
||||
@ -1491,16 +1503,22 @@ def stdapi_sys_eventlog_numrecords(request, response):
|
||||
@register_function_if(has_windll)
|
||||
def stdapi_sys_eventlog_oldest(request, response):
|
||||
handle = packet_get_tlv(request, TLV_TYPE_EVENT_HANDLE)['value']
|
||||
oldest = ctypes.c_ulong(0)
|
||||
if not ctypes.windll.advapi32.GetOldestEventLogRecordW(handle, ctypes.byref(oldest)):
|
||||
GetOldestEventLogRecord = ctypes.windll.advapi32.GetOldestEventLogRecord
|
||||
GetOldestEventLogRecord.argtypes = [ctypes.c_void_p, ctypes.POINTER(ctypes.c_uint32)]
|
||||
GetOldestEventLogRecord.restype = ctypes.c_bool
|
||||
oldest = ctypes.c_uint32(0)
|
||||
if not GetOldestEventLogRecord(handle, ctypes.byref(oldest)):
|
||||
return error_result_windows(), response
|
||||
response += tlv_pack(TLV_TYPE_EVENT_RECORDNUMBER, oldest)
|
||||
response += tlv_pack(TLV_TYPE_EVENT_RECORDNUMBER, oldest.value)
|
||||
return ERROR_SUCCESS, response
|
||||
|
||||
@register_function_if(has_windll)
|
||||
def stdapi_sys_eventlog_close(request, response):
|
||||
handle = packet_get_tlv(request, TLV_TYPE_EVENT_HANDLE)['value']
|
||||
if not ctypes.windll.advapi32.CloseEventLogW(handle):
|
||||
CloseEventLog = ctypes.windll.advapi32.CloseEventLog
|
||||
CloseEventLog.argtypes = [ctypes.c_void_p]
|
||||
CloseEventLog.restype = ctypes.c_bool
|
||||
if not CloseEventLog(handle):
|
||||
return error_result_windows(), response
|
||||
return ERROR_SUCCESS, response
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user