#include "stdafx.h"
#include "MainDlg.h"
#include "DumpHandler.h"
#include "DriverExtract.h"
#include
///
/// Crash dump notify callback
///
/// Dump file path
/// User context
/// Exception info
/// if false - crash dump file was not saved
/// status
int DumpNotifier( const wchar_t* path, void* context, EXCEPTION_POINTERS* expt, bool success )
{
Message::ShowError( NULL, L"Program has crashed. Dump file saved at '" + std::wstring( path ) + L"'" );
return 0;
}
///
/// Associate profile file extension
///
void AssociateExtension()
{
wchar_t tmp[255] = { 0 };
GetModuleFileNameW( NULL, tmp, sizeof( tmp ) );
#ifdef USE64
std::wstring ext = L".xpr64";
std::wstring alias = L"XenosProfile64";
std::wstring desc = L"Xenos 64-bit injection profile";
#else
std::wstring ext = L".xpr";
std::wstring alias = L"XenosProfile";
std::wstring desc = L"Xenos injection profile";
#endif
std::wstring editWith = std::wstring( tmp ) + L" --load %1";
std::wstring runWith = std::wstring( tmp ) + L" --run %1";
std::wstring icon = std::wstring( tmp ) + L",-" + std::to_wstring( IDI_ICON1 );
auto AddKey = []( const std::wstring& subkey, const std::wstring& value, const wchar_t* regValue ) {
SHSetValue( HKEY_CLASSES_ROOT, subkey.c_str(), regValue, REG_SZ, value.c_str(), (DWORD)(value.size() * sizeof( wchar_t )) );
};
SHDeleteKeyW( HKEY_CLASSES_ROOT, alias.c_str() );
AddKey( ext, alias, nullptr );
AddKey( ext, L"Application/xml", L"Content Type" );
AddKey( alias, desc, nullptr );
AddKey( alias + L"\\shell", L"Run", nullptr );
AddKey( alias + L"\\shell\\Edit\\command", editWith, nullptr );
AddKey( alias + L"\\shell\\Run\\command", runWith, nullptr );
AddKey( alias + L"\\DefaultIcon", icon, nullptr );
}
///
/// Log major OS information
///
void LogOSInfo()
{
SYSTEM_INFO info = { 0 };
char* osArch = "x64";
auto pPeb = (blackbone::PEB_T*)NtCurrentTeb()->ProcessEnvironmentBlock;
GetNativeSystemInfo( &info );
if (info.wProcessorArchitecture == PROCESSOR_ARCHITECTURE_INTEL)
osArch = "x86";
xlog::Normal(
"Started on Windows %d.%d.%d.%d %s. Driver status: 0x%X",
pPeb->OSMajorVersion,
pPeb->OSMinorVersion,
(pPeb->OSCSDVersion >> 8) & 0xFF,
pPeb->OSBuildNumber,
osArch,
blackbone::Driver().status()
);
}
///
/// Parse command line string
///
/// Resulting param
/// Profile action
MainDlg::StartAction ParseCmdLine( std::wstring& param )
{
int argc = 0;
auto pCmdLine = GetCommandLineW();
auto argv = CommandLineToArgvW( pCmdLine, &argc );
for (int i = 1; i < argc; i++)
{
if (_wcsicmp( argv[i], L"--load" ) == 0 && i + 1 < argc)
{
param = argv[i + 1];
return MainDlg::LoadProfile;
}
if (_wcsicmp( argv[i], L"--run" ) == 0 && i + 1 < argc)
{
param = argv[i + 1];
return MainDlg::RunProfile;
}
}
return MainDlg::Nothing;
}
int APIENTRY wWinMain( HINSTANCE /*hInstance*/, HINSTANCE /*hPrevInstance*/, LPWSTR /*lpCmdLine*/, int /*nCmdShow*/ )
{
// Setup dump generation
dump::DumpHandler::Instance().CreateWatchdog( blackbone::Utils::GetExeDirectory(), dump::CreateFullDump, &DumpNotifier );
AssociateExtension();
std::wstring param;
auto action = ParseCmdLine( param );
MainDlg mainDlg( action, param );
LogOSInfo();
if (action != MainDlg::RunProfile)
return (int)mainDlg.RunModeless( NULL, IDR_ACCELERATOR1 );
else
return mainDlg.LoadAndInject();
}