vrr/HyperDbg
2
0
Fork 0
mirror of https://github.com/HyperDbg/HyperDbg.git synced 2026-07-10 01:29:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add multi-line mechanism

Browse source
This commit is contained in:
SinaKarvandi 2021-10-09 02:14:36 +03:30
parent 417deee4fc
commit c94caa789d
3 changed files with 134 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

129
hyperdbg/hyperdbg-cli/hyperdbg-cli.cpp
View file

@ -21,6 +21,13 @@
using namespace std;
//
// Global variables
//
bool IsOnString = false;
bool IsPreviousCharacterABackSlash = false;
int CountOfOpenCurlyBrackets = 0;
//
// Header file of HPRDBGCTRL
// Imports
@ -37,6 +44,84 @@ __declspec(dllimport) bool HyperdbgContinuePreviousCommand();
__declspec(dllimport) void HyperdbgSetTextMessageCallback(Callback handler);
}
/**
* @brief check for multi-line commands
*
* @param CurrentCommand
* @return bool return true if the command needs extra input, otherwise
* return false
*/
bool
CheckMultilineCommand(std::string CurrentCommand)
{
for (size_t i = 0; i < CurrentCommand.length(); i++)
{
switch (CurrentCommand.at(i))
{
case '"':
if (IsPreviousCharacterABackSlash)
{
IsPreviousCharacterABackSlash = false;
break; // it's an escaped \" double-quote
}
if (IsOnString)
IsOnString = false;
else
IsOnString = true;
break;
case '{':
if (!IsOnString)
CountOfOpenCurlyBrackets++;
break;
case '}':
if (!IsOnString && CountOfOpenCurlyBrackets > 0)
CountOfOpenCurlyBrackets--;
break;
case '\\':
if (IsPreviousCharacterABackSlash)
IsPreviousCharacterABackSlash = false; // it's not a escape character (two backslashes \\ )
else
IsPreviousCharacterABackSlash = true;
break;
default:
if (IsPreviousCharacterABackSlash)
IsPreviousCharacterABackSlash = false;
break;
}
}
if (IsOnString == FALSE && CountOfOpenCurlyBrackets == 0)
{
//
// either the command is finished or it's a single
// line command
//
return false;
}
else
{
//
// There still other lines, this command is incomplete
//
return true;
}
}
/**
* @brief CLI main function
*
@ -81,9 +166,16 @@ main(int argc, char * argv[])
{
HyperdbgShowSignature();
string CurrentCommand;
string CurrentCommand = "";
IsOnString = false;
IsPreviousCharacterABackSlash = false;
CountOfOpenCurlyBrackets = 0;
getline(cin, CurrentCommand);
GetMultiLinecCommand:
string TempCommand = "";
getline(cin, TempCommand);
if (cin.fail() || cin.eof())
{
@ -96,6 +188,39 @@ main(int argc, char * argv[])
continue;
}
//
// Check for multi-line commands
//
if (CheckMultilineCommand(TempCommand) == true)
{
//
// It's a multi-line command
//
//
// Save the command with a space separator
//
CurrentCommand += TempCommand + " ";
//
// Show a small signature
//
printf("> ");
//
// Get next command
//
goto GetMultiLinecCommand;
}
else
{
//
// Either the multi-line is finished or it's a
// single line command
//
CurrentCommand += TempCommand;
}
if (!CurrentCommand.compare("") &&
HyperdbgContinuePreviousCommand())
{