mirror of
https://github.com/rapid7/metasploit-payloads
synced 2025-03-24 18:16:24 +01:00
c93ba9608c
* Make sure POSIX has the new extension command enumeration function. * Add support for deinit of extensions. * Make sure extensions are tracked like they in Windows. * Fix up a few export definitions. * Stop using strncpy_s in POSIX code.
105 lines
2.4 KiB
C
105 lines
2.4 KiB
C
#include "metsrv.h"
|
|
|
|
#ifdef _WIN32
|
|
// see ReflectiveLoader.c...
|
|
extern HINSTANCE hAppInstance;
|
|
#endif
|
|
|
|
PLIST gExtensionList = NULL;
|
|
|
|
DWORD request_core_enumextcmd(Remote* pRemote, Packet* pPacket);
|
|
|
|
// Dispatch table
|
|
Command customCommands[] =
|
|
{
|
|
COMMAND_REQ("core_loadlib", request_core_loadlib),
|
|
COMMAND_REQ("core_enumextcmd", request_core_enumextcmd),
|
|
COMMAND_TERMINATOR
|
|
};
|
|
|
|
typedef struct _EnumExtensions
|
|
{
|
|
Packet* pResponse;
|
|
char* lpExtensionName;
|
|
} EnumExtensions, * PEnumExtensions;
|
|
|
|
BOOL ext_cmd_callback(LPVOID pState, LPVOID pData)
|
|
{
|
|
PEnumExtensions pEnum = (PEnumExtensions)pState;
|
|
Command* command = NULL;
|
|
|
|
if (pEnum != NULL && pEnum->pResponse != NULL && pData != NULL)
|
|
{
|
|
PEXTENSION pExt = (PEXTENSION)pData;
|
|
if (pExt->name[0] != '\0' && pEnum->lpExtensionName != NULL && strcmp(pExt->name, pEnum->lpExtensionName) == 0)
|
|
{
|
|
dprintf("[LISTEXT] Found extension: %s", pExt->name);
|
|
for (command = pExt->start; command != pExt->end; command = command->next)
|
|
{
|
|
packet_add_tlv_string(pEnum->pResponse, TLV_TYPE_STRING, command->method);
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
}
|
|
return FALSE;
|
|
}
|
|
|
|
DWORD request_core_enumextcmd(Remote* pRemote, Packet* pPacket)
|
|
{
|
|
BOOL bResult = FALSE;
|
|
Packet* pResponse = packet_create_response(pPacket);
|
|
|
|
if (pResponse != NULL)
|
|
{
|
|
EnumExtensions enumExt;
|
|
enumExt.pResponse = pResponse;
|
|
enumExt.lpExtensionName = packet_get_tlv_value_string(pPacket, TLV_TYPE_STRING);
|
|
|
|
dprintf("[LISTEXTCMD] Listing extension commands for %s ...", enumExt.lpExtensionName);
|
|
// Start by enumerating the names of the extensions
|
|
bResult = list_enumerate(gExtensionList, ext_cmd_callback, &enumExt);
|
|
|
|
packet_add_tlv_uint(pResponse, TLV_TYPE_RESULT, ERROR_SUCCESS);
|
|
packet_transmit(pRemote, pResponse, NULL);
|
|
}
|
|
|
|
return ERROR_SUCCESS;
|
|
}
|
|
|
|
/*
|
|
* Registers custom command handlers
|
|
*/
|
|
VOID register_dispatch_routines()
|
|
{
|
|
gExtensionList = list_create();
|
|
|
|
command_register_all(customCommands);
|
|
}
|
|
|
|
/*
|
|
* Deregisters previously registered custom commands and loaded extensions.
|
|
*/
|
|
VOID deregister_dispatch_routines(Remote * remote)
|
|
{
|
|
while (TRUE)
|
|
{
|
|
PEXTENSION extension = list_pop(gExtensionList);
|
|
if (!extension)
|
|
{
|
|
break;
|
|
}
|
|
|
|
if (extension->deinit)
|
|
{
|
|
extension->deinit(remote);
|
|
}
|
|
|
|
free(extension);
|
|
}
|
|
|
|
command_deregister_all(customCommands);
|
|
|
|
list_destroy(gExtensionList);
|
|
}
|