mirror of
https://github.com/HyperDbg/HyperDbg.git
synced 2026-08-11 01:25:35 +00:00
130 lines
2.4 KiB
C++
130 lines
2.4 KiB
C++
/**
|
|
* @file script.cpp
|
|
* @author Sina Karvandi (sina@rayanfam.com)
|
|
* @brief .script command
|
|
* @details
|
|
* @version 0.1
|
|
* @date 2020-07-03
|
|
*
|
|
* @copyright This project is released under the GNU Public License v3.
|
|
*
|
|
*/
|
|
#include "..\hprdbgctrl\pch.h"
|
|
|
|
using namespace std;
|
|
|
|
//
|
|
// Global Variables
|
|
//
|
|
extern BOOLEAN g_ExecutingScript;
|
|
|
|
/**
|
|
* @brief help of .script command
|
|
*
|
|
* @return VOID
|
|
*/
|
|
VOID
|
|
CommandScriptHelp()
|
|
{
|
|
ShowMessages(".script : run a HyperDbg script.\n\n");
|
|
ShowMessages("syntax : \.script [FilePath]\n");
|
|
}
|
|
|
|
/**
|
|
* @brief .script command handler
|
|
*
|
|
* @param SplittedCommand
|
|
* @param Command
|
|
* @return VOID
|
|
*/
|
|
VOID
|
|
CommandScript(vector<string> SplittedCommand, string Command)
|
|
{
|
|
std::string Line;
|
|
BOOLEAN IsOpened = FALSE;
|
|
string NewPath;
|
|
int CommandExecutionResult = 0;
|
|
|
|
if (SplittedCommand.size() == 1)
|
|
{
|
|
ShowMessages("please specify a file\n");
|
|
CommandScriptHelp();
|
|
return;
|
|
}
|
|
|
|
//
|
|
// Trim the command
|
|
//
|
|
Trim(Command);
|
|
|
|
//
|
|
// Remove .script from it
|
|
//
|
|
Command.erase(0, 7);
|
|
|
|
//
|
|
// Trim it again
|
|
//
|
|
Trim(Command);
|
|
|
|
//
|
|
// Parse the script file
|
|
//
|
|
ifstream File(Command);
|
|
|
|
if (File.is_open())
|
|
{
|
|
IsOpened = TRUE;
|
|
|
|
//
|
|
// Indicate that it's a script
|
|
//
|
|
g_ExecutingScript = TRUE;
|
|
|
|
while (std::getline(File, Line))
|
|
{
|
|
char * LineContent = (char *)Line.c_str();
|
|
//
|
|
// Check if the it's a command or not
|
|
//
|
|
InterpreterRemoveComments(LineContent);
|
|
|
|
if (IsEmptyString(LineContent))
|
|
{
|
|
continue;
|
|
}
|
|
|
|
//
|
|
// Show current running command
|
|
//
|
|
ShowMessages("\n%s\n", LineContent);
|
|
|
|
CommandExecutionResult = HyperdbgInterpreter(LineContent);
|
|
|
|
ShowMessages("\n");
|
|
|
|
//
|
|
// if the debugger encounters an exit state then the return will be 1
|
|
//
|
|
if (CommandExecutionResult == 1)
|
|
{
|
|
//
|
|
// Exit from the debugger
|
|
//
|
|
exit(0);
|
|
}
|
|
}
|
|
|
|
//
|
|
// Indicate that script is finished
|
|
//
|
|
g_ExecutingScript = FALSE;
|
|
|
|
File.close();
|
|
}
|
|
|
|
if (!IsOpened)
|
|
{
|
|
ShowMessages("err, invalid file specified for .script command\n");
|
|
}
|
|
}
|