mirror of
https://github.com/HyperDbg/HyperDbg.git
synced 2026-08-23 07:25:29 +00:00
101 lines
2.1 KiB
C++
101 lines
2.1 KiB
C++
/**
|
|
* @file load.cpp
|
|
* @author Sina Karvandi (sina@rayanfam.com)
|
|
* @brief load command
|
|
* @details
|
|
* @version 0.1
|
|
* @date 2020-05-27
|
|
*
|
|
* @copyright This project is released under the GNU Public License v3.
|
|
*
|
|
*/
|
|
#include "pch.h"
|
|
|
|
//
|
|
// Global Variables
|
|
//
|
|
extern BOOLEAN g_IsConnectedToHyperDbgLocally;
|
|
extern BOOLEAN g_IsDebuggerModulesLoaded;
|
|
|
|
/**
|
|
* @brief help of load command
|
|
*
|
|
* @return VOID
|
|
*/
|
|
VOID CommandLoadHelp() {
|
|
ShowMessages("load : installs the drivers and load the modules.\n\n");
|
|
ShowMessages("syntax : \tload [module name]\n");
|
|
ShowMessages("\t\te.g : load vmm\n");
|
|
}
|
|
|
|
/**
|
|
* @brief load command handler
|
|
*
|
|
* @param SplittedCommand
|
|
* @return VOID
|
|
*/
|
|
VOID CommandLoad(vector<string> SplittedCommand) {
|
|
|
|
BOOL Status;
|
|
HANDLE hToken;
|
|
|
|
if (SplittedCommand.size() != 2) {
|
|
ShowMessages("incorrect use of 'load'\n\n");
|
|
CommandLoadHelp();
|
|
return;
|
|
}
|
|
if (!g_IsConnectedToHyperDbgLocally) {
|
|
ShowMessages("You're not connected to any instance of HyperDbg, did you "
|
|
"use '.connect'? \n");
|
|
return;
|
|
}
|
|
|
|
//
|
|
// Enable Debug privilege
|
|
//
|
|
Status =
|
|
OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES, &hToken);
|
|
if (!Status) {
|
|
ShowMessages("OpenProcessToken failed, error : %u \n");
|
|
return;
|
|
}
|
|
|
|
Status = SetPrivilege(hToken, SE_DEBUG_NAME, TRUE);
|
|
if (!Status) {
|
|
CloseHandle(hToken);
|
|
return;
|
|
}
|
|
|
|
//
|
|
// Check for the module
|
|
//
|
|
if (!SplittedCommand.at(1).compare("vmm")) {
|
|
|
|
//
|
|
// Load VMM Module
|
|
//
|
|
ShowMessages("try to install driver...\n");
|
|
if (HyperdbgInstallVmmDriver()) {
|
|
ShowMessages("Failed to install driver\n");
|
|
return;
|
|
}
|
|
|
|
ShowMessages("try to install load kernel modules...\n");
|
|
if (HyperdbgLoadVmm()) {
|
|
ShowMessages("Failed to load driver\n");
|
|
return;
|
|
}
|
|
|
|
//
|
|
// If we reach here so the module are loaded
|
|
//
|
|
g_IsDebuggerModulesLoaded = TRUE;
|
|
} else {
|
|
|
|
//
|
|
// Module not found
|
|
//
|
|
ShowMessages("module not found, currently, 'vmm' is the only available "
|
|
"module for HyperDbg.\n");
|
|
}
|
|
}
|