HyperDbg/hyperdbg/hprdbgctrl/dr.cpp

103 lines
2.6 KiB
C++

/**
* @file dr.cpp
* @author Sina Karvandi (sina@rayanfam.com)
* @brief !dr commands
* @details
* @version 0.1
* @date 2020-06-11
*
* @copyright This project is released under the GNU Public License v3.
*
*/
#include "pch.h"
/**
* @brief help of !dr command
*
* @return VOID
*/
VOID CommandDrHelp() {
ShowMessages("!dr : Monitors any access to debug registers.\n\n");
ShowMessages("syntax : \t!dr core [core index "
"(hex value)] pid [process id (hex value)] condition {[assembly "
"in hex]} code {[assembly in hex]} buffer [pre-require buffer - "
"(hex value)] \n");
ShowMessages("\t\te.g : !dr\n");
ShowMessages("\t\te.g : !dr pid 400\n");
ShowMessages("\t\te.g : !dr core 2 pid 400\n");
}
/**
* @brief !dr command handler
*
* @param SplittedCommand
* @return VOID
*/
VOID CommandDr(vector<string> SplittedCommand) {
PDEBUGGER_GENERAL_EVENT_DETAIL Event = NULL;
PDEBUGGER_GENERAL_ACTION ActionBreakToDebugger = NULL;
PDEBUGGER_GENERAL_ACTION ActionCustomCode = NULL;
PDEBUGGER_GENERAL_ACTION ActionScript = NULL;
UINT32 EventLength;
UINT32 ActionBreakToDebuggerLength = 0;
UINT32 ActionCustomCodeLength = 0;
UINT32 ActionScriptLength = 0;
//
// Interpret and fill the general event and action fields
//
//
if (!InterpretGeneralEventAndActionsFields(
&SplittedCommand, DEBUG_REGISTERS_ACCESSED, &Event, &EventLength,
&ActionBreakToDebugger, &ActionBreakToDebuggerLength,
&ActionCustomCode, &ActionCustomCodeLength, &ActionScript,
&ActionScriptLength)) {
CommandDrHelp();
return;
}
//
// Check for size
//
if (SplittedCommand.size() > 1) {
ShowMessages("incorrect use of '!dr'\n");
CommandDrHelp();
return;
}
//
// Send the ioctl to the kernel for event registeration
//
if (!SendEventToKernel(Event, EventLength)) {
//
// There was an error, probably the handle was not initialized
// we have to free the Action before exit, it is because, we
// already freed the Event and string buffers
//
if (ActionBreakToDebugger != NULL) {
free(ActionBreakToDebugger);
}
if (ActionCustomCode != NULL) {
free(ActionCustomCode);
}
if (ActionScript != NULL) {
free(ActionScript);
}
return;
}
//
// Add the event to the kernel
//
if (!RegisterActionToEvent(ActionBreakToDebugger, ActionBreakToDebuggerLength,
ActionCustomCode, ActionCustomCodeLength,
ActionScript, ActionScriptLength)) {
//
// There was an error
//
return;
}
}