metasploit-payloads/c/meterpreter/source/client/remote_dispatch.c

#include "metcli.h"


/*
 * Extension callback for printing out notifications for channels opening
 */
DWORD ex_remote_response_core_channel_open(Remote *remote, Packet *packet)
{
	DWORD channelId = packet_get_tlv_value_uint(packet, TLV_TYPE_CHANNEL_ID);

	if (channelId)
	{
		console_write_output(
				"\n"
				INBOUND_PREFIX " CHANNEL: New remote channel allocated: %lu.\n", 
				channelId);

		console_write_prompt();
	}

	return ERROR_SUCCESS;
}

/*
 * Extension callback for printing out notifications for when the remote
 * endpoint is telling us to close a channel
 */
DWORD ex_remote_request_core_channel_close(Remote *remote, Packet *packet)
{
	DWORD channelId = packet_get_tlv_value_uint(packet, TLV_TYPE_CHANNEL_ID);

	if (channelId)
	{
		// If an interactive channel is closing, reset it
		if (channelId == console_get_interactive_channel_id())
			console_set_interactive_channel(remote, NULL);
	}

	return ERROR_SUCCESS;
}


/****************************
 * Custom dispatch routines *
 ****************************/

// Dispatch table
Command custom_commands[] = 
{
	{ "core_channel_open",
	  { EMPTY_DISPATCH_HANDLER                              },
	  { ex_remote_response_core_channel_open,      { 0 }, 0 },
	},
	{ "core_channel_close",
	  { ex_remote_request_core_channel_close,      { 0 }, 0 },
	  { EMPTY_DISPATCH_HANDLER                              },
	},

	// Terminator
	{ NULL,
	  { EMPTY_DISPATCH_HANDLER                              },
	  { EMPTY_DISPATCH_HANDLER                              },
	},
};


/*
 * Registers custom command handlers
 */
VOID remote_register_core_dispatch_routines()
{
	DWORD index;

	for (index = 0;
	     custom_commands[index].method;
	     index++)
		command_register(&custom_commands[index]);
}

/*
 * Deregisters previously registered custom commands
 */
VOID remote_deregister_core_dispatch_routines()
{
	DWORD index;

	for (index = 0;
	     custom_commands[index].method;
	     index++)
		command_deregister(&custom_commands[index]);
}