github/metasploit-payloads
1
Fork 0
mirror of https://github.com/rapid7/metasploit-payloads synced 2025-02-16 00:24:29 +01:00
Code Releases Activity

fix minor changes from @smcintyre-r7 reviews

Browse Source
This commit is contained in:
Alex Romero 2023-01-29 13:33:54 -05:00
parent 90eacefad5
commit 3fbabbca7d
No known key found for this signature in database
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
python/meterpreter/ext_server_stdapi.py
View File

@ -2830,7 +2830,7 @@ def stdapi_ui_desktop_enum(request, response):
ProcessIdToSessionId = ctypes.windll.kernel32.ProcessIdToSessionId
ProcessIdToSessionId.argtypes = [ctypes.c_ulong, ctypes.POINTER(ctypes.c_ulong)]
ProcessIdToSessionId.restype = ctypes.c_bool
if not ProcessIdToSessionId(ctypes.c_ulong(pid), ctypes.byref(dwSessionId)):
dwSessionId = ctypes.c_ulong(-1)
@ -2878,7 +2878,6 @@ def stdapi_ui_desktop_enum(request, response):
@register_function_if(has_windll)
def stdapi_ui_desktop_get(request, response):
UOI_NAME = 2
GetCurrentProcessId = ctypes.windll.kernel32.GetCurrentProcessId
@ -2906,12 +2905,12 @@ def stdapi_ui_desktop_get(request, response):
if not ProcessIdToSessionId(GetCurrentProcessId(), ctypes.byref(dwSessionId)):
return error_result_windows(), response
station_name = ctypes.create_string_buffer(bytes(""), 256)
station_name = ctypes.create_string_buffer(bytes(), 256)
success = GetUserObjectInformationA(GetProcessWindowStation(), UOI_NAME, ctypes.byref(station_name), 256, None)
if not success:
return error_result_windows(), response
desktop_name = ctypes.create_string_buffer(bytes(""), 256)
desktop_name = ctypes.create_string_buffer(bytes(), 256)
success = GetUserObjectInformationA(GetThreadDesktop(GetCurrentThreadId()), UOI_NAME, ctypes.byref(desktop_name), 256, None)
if not success:
return error_result_windows(), response