admin/BeneficialCode-WinArk
1
0
Fork 0
mirror of https://github.com/BeneficialCode/WinArk synced 2026-08-09 12:00:31 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
32754a45551ba99d66187539c1c81a0095c8b5e7
BeneficialCode-WinArk/WinArk/SystemConfigDlg.cpp
T

574 lines
20 KiB
C++
Raw Blame History

#include "stdafx.h"
#include "SystemConfigDlg.h"
#include <DriverHelper.h>
#include "FormatHelper.h"
#include "SymbolHelper.h"
using namespace WinSys;
LRESULT CSystemConfigDlg::OnInitDialog(UINT /*uMsg*/, WPARAM /*wParam*/, LPARAM /*lParam*/, BOOL& /*bHandled*/) {
//DlgResize_Init(true);
m_CheckImageLoad.Attach(GetDlgItem(IDC_INTERCEPT_DRIVER));
m_CheckDriverLoad.Attach(GetDlgItem(IDC_DISABLE_DRIVER_LOAD));
m_CheckLogHash.Attach(GetDlgItem(IDC_LOG_HASH));
m_List.Attach(GetDlgItem(IDC_DEBUGGER_LIST));
m_List.SetExtendedListViewStyle(LVS_EX_FULLROWSELECT | LVS_EX_INFOTIP | LVS_EX_GRIDLINES);
m_List.InsertColumn(0, L"Debugger", LVCFMT_LEFT, 80);
CRect rect;
m_List.GetClientRect(&rect);
m_List.InsertColumn(1, L"Path", LVCFMT_LEFT, rect.Width());
m_BasicSysInfo = SystemInformation::GetBasicSystemInfo();
GetDlgItem(IDC_REMOVE_CALLBACK).EnableWindow(FALSE);
CString text;
auto& ver = SystemInformation::GetWindowsVersion();
text.Format(L"%u.%u.%u", ver.Major, ver.Minor, ver.Build);
SetDlgItemText(IDC_WIN_VERSION, text);
text = FormatHelper::TimeToString(SystemInformation::GetBootTime());
SetDlgItemText(IDC_BOOT_TIME, text);
text = FormatHelper::FormatWithCommas(m_BasicSysInfo.TotalPhysicalInPages >> 8) + L" MB";
text.Format(L"%s (%u GB)", text, (ULONG)((m_BasicSysInfo.TotalPhysicalInPages + (1 << 17)) >> 18));
SetDlgItemText(IDC_USABLE_RAM, text);
SetDlgItemInt(IDC_PROCESSOR_COUNT, m_BasicSysInfo.NumberOfProcessors);
std::string brand = SystemInformation::GetCpuBrand();
SetDlgItemTextA(m_hWnd, IDC_PROCESSOR, brand.c_str());
GetFirmwareEnvironmentVariable(L"",
L"{00000000-0000-0000-0000-000000000000}",
nullptr, 0);
if (GetLastError() == ERROR_INVALID_FUNCTION) {
SetDlgItemText(IDC_BOOT_MODE, L"Legacy");
}
else {
SetDlgItemText(IDC_BOOT_MODE, L"UEFI");
}
return TRUE;
}
LRESULT CSystemConfigDlg::OnDestroy(UINT /*uMsg*/, WPARAM /*wParam*/, LPARAM /*lParam*/, BOOL& /*bHandled*/) {
bool enable = GetDlgItem(IDC_SET_CALLBACK).IsWindowEnabled();
if (!enable)
SendMessage(WM_COMMAND, IDC_REMOVE_CALLBACK);
if (m_enableDbgSys) {
DriverHelper::DisableDbgSys();
}
return TRUE;
}
LRESULT CSystemConfigDlg::OnSetCallback(WORD /*wNotifyCode*/, WORD /*wID*/, HWND /*hWndCtl*/, BOOL& /*bHandled*/) {
int count = 0;
bool success = false;
int checkCode = m_CheckImageLoad.GetCheck();
if (checkCode == BST_CHECKED) {
count += 1;
success = DriverHelper::SetImageLoadNotify();
if (success)
m_CheckImageLoad.EnableWindow(FALSE);
}
checkCode = m_CheckDriverLoad.GetCheck();
if (checkCode == BST_CHECKED) {
count += 1;
CiSymbols symbols;
success = InitCiSymbols(&symbols);
if(success)
success = DriverHelper::DisableDriverLoad(&symbols);
if (success)
m_CheckDriverLoad.EnableWindow(FALSE);
else {
AtlMessageBox(m_hWnd, L"Failed to disable driver load", IDS_TITLE, MB_ICONERROR);
return FALSE;
}
}
checkCode = m_CheckLogHash.GetCheck();
if (checkCode == BST_CHECKED) {
count += 1;
CiSymbols symbols;
success = InitCiSymbols(&symbols);
if (success)
success = DriverHelper::StartLogDriverHash(&symbols);
if (success)
m_CheckLogHash.EnableWindow(FALSE);
else {
AtlMessageBox(m_hWnd, L"Failed to enable log driver load hash", IDS_TITLE, MB_ICONERROR);
return FALSE;
}
}
if (count == 0) {
AtlMessageBox(m_hWnd, L"You should select one config", L"Error", MB_ICONERROR);
return FALSE;
}
GetDlgItem(IDC_SET_CALLBACK).EnableWindow(FALSE);
GetDlgItem(IDC_REMOVE_CALLBACK).EnableWindow();
return TRUE;
}
LRESULT CSystemConfigDlg::OnRemoveCallback(WORD /*wNotifyCode*/, WORD /*wID*/, HWND /*hWndCtl*/, BOOL& /*bHandled*/) {
bool success = false;
int checkCode = m_CheckImageLoad.GetCheck();
if (checkCode == BST_CHECKED) {
success = DriverHelper::RemoveImageLoadNotify();
if (success)
m_CheckImageLoad.EnableWindow(TRUE);
}
checkCode = m_CheckDriverLoad.GetCheck();
if (checkCode == BST_CHECKED) {
success = DriverHelper::EnableDriverLoad();
if (success)
m_CheckDriverLoad.EnableWindow(TRUE);
}
checkCode = m_CheckLogHash.GetCheck();
if (checkCode == BST_CHECKED) {
success = DriverHelper::StopLogDriverHash();
if (success)
m_CheckLogHash.EnableWindow(TRUE);
}
GetDlgItem(IDC_SET_CALLBACK).EnableWindow(TRUE);
GetDlgItem(IDC_REMOVE_CALLBACK).EnableWindow(FALSE);
return TRUE;
}
LRESULT CSystemConfigDlg::OnEnableDbgSys(WORD /*wNotifyCode*/, WORD /*wID*/, HWND /*hWndCtl*/, BOOL& /*bHandled*/) {
if (m_enableDbgSys) {
bool success = DriverHelper::DisableDbgSys();
if (success) {
SetDlgItemText(IDC_ENABLE_DBGSYS, L"Enable");
m_enableDbgSys = false;
}
else {
AtlMessageBox(m_hWnd, L"Disable failed");
}
}
else {
DbgSysCoreInfo info;
bool success = InitDbgSymbols(&info);
do
{
if (!success)
break;
success = DriverHelper::EnableDbgSys(&info);
} while (false);
if (success) {
SetDlgItemText(IDC_ENABLE_DBGSYS, L"Disable");
m_enableDbgSys = true;
}
else {
AtlMessageBox(m_hWnd, L"Enable Failed!");
}
}
return TRUE;
}
bool CSystemConfigDlg::InitDbgSymbols(DbgSysCoreInfo* pInfo) {
bool initSuccess = false;
initSuccess = InitRoutines(pInfo);
if (!initSuccess)
return initSuccess;
initSuccess = InitEprocessOffsets(pInfo);
if (!initSuccess)
return initSuccess;
initSuccess = InitEthreadOffsets(pInfo);
if (!initSuccess)
return initSuccess;
initSuccess = InitPebOffsets(pInfo);
if (!initSuccess)
return initSuccess;
initSuccess = InitKthreadOffsets(pInfo);
if (!initSuccess)
return initSuccess;
return true;
}
bool CSystemConfigDlg::InitRoutines(DbgSysCoreInfo* pInfo) {
pInfo->NtCreateDebugObjectAddress = (void*)SymbolHelper::GetKernelSymbolAddressFromName("NtCreateDebugObject");
if (!pInfo->NtCreateDebugObjectAddress)
return false;
pInfo->DbgkDebugObjectTypeAddress = (void*)SymbolHelper::GetKernelSymbolAddressFromName("DbgkDebugObjectType");
if (!pInfo->DbgkDebugObjectTypeAddress)
return false;
pInfo->ZwProtectVirtualMemory = (void*)SymbolHelper::GetKernelSymbolAddressFromName("ZwProtectVirtualMemory");
if (!pInfo->ZwProtectVirtualMemory)
return false;
pInfo->NtDebugActiveProcess = (void*)SymbolHelper::GetKernelSymbolAddressFromName("NtDebugActiveProcess");
if (!pInfo->NtDebugActiveProcess)
return false;
pInfo->DbgkpPostFakeProcessCreateMessages = (void*)SymbolHelper::GetKernelSymbolAddressFromName("DbgkpPostFakeProcessCreateMessages");
if (!pInfo->DbgkpPostFakeProcessCreateMessages)
return false;
pInfo->DbgkpSetProcessDebugObject = (void*)SymbolHelper::GetKernelSymbolAddressFromName("DbgkpSetProcessDebugObject");
if (!pInfo->DbgkpSetProcessDebugObject)
return false;
pInfo->DbgkpPostFakeThreadMessages = (void*)SymbolHelper::GetKernelSymbolAddressFromName("DbgkpPostFakeThreadMessages");
if (!pInfo->DbgkpPostFakeThreadMessages)
return false;
pInfo->DbgkpPostModuleMessages = (void*)SymbolHelper::GetKernelSymbolAddressFromName("DbgkpPostModuleMessages");
if (!pInfo->DbgkpPostModuleMessages)
return false;
pInfo->PsGetNextProcessThread = (void*)SymbolHelper::GetKernelSymbolAddressFromName("PsGetNextProcessThread");
if (!pInfo->PsGetNextProcessThread)
return false;
pInfo->DbgkpProcessDebugPortMutex = (void*)SymbolHelper::GetKernelSymbolAddressFromName("DbgkpProcessDebugPortMutex");
if (!pInfo->DbgkpProcessDebugPortMutex)
return false;
pInfo->DbgkpWakeTarget = (void*)SymbolHelper::GetKernelSymbolAddressFromName("DbgkpWakeTarget");
if (!pInfo->DbgkpWakeTarget)
return false;
pInfo->DbgkpMarkProcessPeb = (void*)SymbolHelper::GetKernelSymbolAddressFromName("DbgkpMarkProcessPeb");
if (!pInfo->DbgkpMarkProcessPeb)
return false;
pInfo->DbgkpMaxModuleMsgs = (void*)SymbolHelper::GetKernelSymbolAddressFromName("DbgkpMaxModuleMsgs");
if (!pInfo->DbgkpMaxModuleMsgs)
return false;
pInfo->MmGetFileNameForAddress = (void*)SymbolHelper::GetKernelSymbolAddressFromName("MmGetFileNameForAddress");
if (!pInfo->MmGetFileNameForAddress)
return false;
pInfo->DbgkpSendApiMessage = (void*)SymbolHelper::GetKernelSymbolAddressFromName("DbgkpSendApiMessage");
if (!pInfo->DbgkpSendApiMessage)
return false;
pInfo->DbgkpQueueMessage = (void*)SymbolHelper::GetKernelSymbolAddressFromName("DbgkpQueueMessage");
if (!pInfo->DbgkpQueueMessage)
return false;
pInfo->DbgkpMaxModuleMsgs = (void*)SymbolHelper::GetKernelSymbolAddressFromName("DbgkpMaxModuleMsgs");
if (!pInfo->DbgkpMaxModuleMsgs)
return false;
pInfo->KeThawAllThreads = (void*)SymbolHelper::GetKernelSymbolAddressFromName("KeThawAllThreads");
if (pInfo->KeThawAllThreads) {
}
pInfo->DbgkpSectionToFileHandle = (void*)SymbolHelper::GetKernelSymbolAddressFromName("DbgkpSectionToFileHandle");
if (!pInfo->DbgkpSectionToFileHandle)
return false;
pInfo->PsResumeThread = (void*)SymbolHelper::GetKernelSymbolAddressFromName("PsResumeThread");
if (!pInfo->PsResumeThread)
return false;
pInfo->DbgkSendSystemDllMessages = (void*)SymbolHelper::GetKernelSymbolAddressFromName("DbgkSendSystemDllMessages");
if (!pInfo->DbgkSendSystemDllMessages)
return false;
pInfo->PsSuspendThread = (void*)SymbolHelper::GetKernelSymbolAddressFromName("PsSuspendThread");
if (!pInfo->PsSuspendThread)
return false;
pInfo->PsQuerySystemDllInfo = (void*)SymbolHelper::GetKernelSymbolAddressFromName("PsQuerySystemDllInfo");
if (!pInfo->PsQuerySystemDllInfo)
return false;
pInfo->ObFastReferenceObject = (void*)SymbolHelper::GetKernelSymbolAddressFromName("ObFastReferenceObject");
if (!pInfo->ObFastReferenceObject)
return false;
pInfo->ExfAcquirePushLockShared = (void*)SymbolHelper::GetKernelSymbolAddressFromName("ExfAcquirePushLockShared");
if (!pInfo->ExfAcquirePushLockShared)
return false;
pInfo->ExfReleasePushLockShared = (void*)SymbolHelper::GetKernelSymbolAddressFromName("ExfReleasePushLockShared");
if (!pInfo->ExfReleasePushLockShared)
return false;
pInfo->ObFastReferenceObjectLocked = (void*)SymbolHelper::GetKernelSymbolAddressFromName("ObFastReferenceObjectLocked");
if (!pInfo->ObFastReferenceObjectLocked)
return false;
pInfo->PsGetNextProcess = (void*)SymbolHelper::GetKernelSymbolAddressFromName("PsGetNextProcess");
if (!pInfo->PsGetNextProcess)
return false;
pInfo->DbgkpConvertKernelToUserStateChange = (void*)SymbolHelper::GetKernelSymbolAddressFromName("DbgkpConvertKernelToUserStateChange");
if (!pInfo->DbgkpConvertKernelToUserStateChange)
return false;
pInfo->PsCaptureExceptionPort = (void*)SymbolHelper::GetKernelSymbolAddressFromName("PsCaptureExceptionPort");
if (!pInfo->PsCaptureExceptionPort)
return false;
pInfo->DbgkpSendErrorMessage = (void*)SymbolHelper::GetKernelSymbolAddressFromName("DbgkpSendErrorMessage");
if (!pInfo->DbgkpSendErrorMessage)
return false;
pInfo->DbgkpSendApiMessageLpc = (void*)SymbolHelper::GetKernelSymbolAddressFromName("DbgkpSendApiMessageLpc");
if (!pInfo->DbgkpSendApiMessageLpc)
return false;
pInfo->DbgkpOpenHandles = (void*)SymbolHelper::GetKernelSymbolAddressFromName("DbgkpOpenHandles");
if (!pInfo->DbgkpOpenHandles)
return false;
pInfo->DbgkpSuppressDbgMsg = (void*)SymbolHelper::GetKernelSymbolAddressFromName("DbgkpSuppressDbgMsg");
if (!pInfo->DbgkpSuppressDbgMsg)
return false;
pInfo->ObFastDereferenceObject = (void*)SymbolHelper::GetKernelSymbolAddressFromName("ObFastDereferenceObject");
if (!pInfo->ObFastDereferenceObject)
return false;
pInfo->DbgkCreateThread = (void*)SymbolHelper::GetKernelSymbolAddressFromName("DbgkCreateThread");
if (!pInfo->DbgkCreateThread)
return false;
pInfo->DbgkExitThread = (void*)SymbolHelper::GetKernelSymbolAddressFromName("DbgkExitThread");
if (!pInfo->DbgkExitThread)
return false;
pInfo->DbgkExitProcess = (void*)SymbolHelper::GetKernelSymbolAddressFromName("DbgkExitProcess");
if (!pInfo->DbgkExitProcess)
return false;
pInfo->DbgkMapViewOfSection = (void*)SymbolHelper::GetKernelSymbolAddressFromName("DbgkMapViewOfSection");
if (!pInfo->DbgkMapViewOfSection)
return false;
pInfo->DbgkUnMapViewOfSection = (void*)SymbolHelper::GetKernelSymbolAddressFromName("DbgkUnMapViewOfSection");
if (!pInfo->DbgkUnMapViewOfSection)
return false;
pInfo->NtWaitForDebugEvent = (void*)SymbolHelper::GetKernelSymbolAddressFromName("NtWaitForDebugEvent");
if (!pInfo->NtWaitForDebugEvent)
return false;
pInfo->NtDebugContinue = (void*)SymbolHelper::GetKernelSymbolAddressFromName("NtDebugContinue");
if (!pInfo->NtDebugContinue)
return false;
pInfo->NtRemoveProcessDebug = (void*)SymbolHelper::GetKernelSymbolAddressFromName("NtRemoveProcessDebug");
if (!pInfo->NtRemoveProcessDebug)
return false;
pInfo->DbgkForwardException = (void*)SymbolHelper::GetKernelSymbolAddressFromName("DbgkForwardException");
if (!pInfo->DbgkForwardException)
return false;
pInfo->DbgkCopyProcessDebugPort = (void*)SymbolHelper::GetKernelSymbolAddressFromName("DbgkCopyProcessDebugPort");
if (!pInfo->DbgkCopyProcessDebugPort)
return false;
pInfo->DbgkClearProcessDebugObject = (void*)SymbolHelper::GetKernelSymbolAddressFromName("DbgkClearProcessDebugObject");
if (!pInfo->DbgkClearProcessDebugObject)
return false;
pInfo->NtSetInformationDebugObject = (void*)SymbolHelper::GetKernelSymbolAddressFromName("NtSetInformationDebugObject");
if (!pInfo->NtSetInformationDebugObject)
return false;
pInfo->PsTerminateProcess = (void*)SymbolHelper::GetKernelSymbolAddressFromName("PsTerminateProcess");
if (!pInfo->PsTerminateProcess)
return false;
pInfo->PspNotifyEnableMask = (void*)SymbolHelper::GetKernelSymbolAddressFromName("PspNotifyEnableMask");
if (!pInfo->PspNotifyEnableMask)
return false;
pInfo->PspLoadImageNotifyRoutine = (void*)SymbolHelper::GetKernelSymbolAddressFromName("PspLoadImageNotifyRoutine");
if (!pInfo->PspLoadImageNotifyRoutine)
return false;
pInfo->MiSectionControlArea = (void*)SymbolHelper::GetKernelSymbolAddressFromName("MiSectionControlArea");
if (!pInfo->MiSectionControlArea)
return false;
pInfo->MiReferenceControlAreaFile = (void*)SymbolHelper::GetKernelSymbolAddressFromName("MiReferenceControlAreaFile");
if (!pInfo->MiReferenceControlAreaFile) {
// not exist on win10 1511
}
return true;
}
bool CSystemConfigDlg::InitEprocessOffsets(DbgSysCoreInfo* pInfo) {
pInfo->EprocessOffsets.RundownProtect = SymbolHelper::GetKernelStructMemberOffset("_EPROCESS", "RundownProtect");
if (pInfo->EprocessOffsets.RundownProtect == -1)
return false;
#ifdef _WIN64
pInfo->EprocessOffsets.WoW64Process = SymbolHelper::GetKernelStructMemberOffset("_EPROCESS", "WoW64Process");
if (pInfo->EprocessOffsets.WoW64Process == -1)
return false;
#endif // _WIN64
pInfo->EprocessOffsets.DebugPort = SymbolHelper::GetKernelStructMemberOffset("_EPROCESS", "DebugPort");
if (pInfo->EprocessOffsets.DebugPort == -1)
return false;
pInfo->EprocessOffsets.Peb = SymbolHelper::GetKernelStructMemberOffset("_EPROCESS", "Peb");
if (pInfo->EprocessOffsets.Peb == -1)
return false;
pInfo->EprocessOffsets.Flags = SymbolHelper::GetKernelStructMemberOffset("_EPROCESS", "Flags");
if (pInfo->EprocessOffsets.Flags == -1)
return false;
pInfo->EprocessOffsets.SectionBaseAddress = SymbolHelper::GetKernelStructMemberOffset("_EPROCESS", "SectionBaseAddress");
if (pInfo->EprocessOffsets.SectionBaseAddress == -1)
return false;
pInfo->EprocessOffsets.SectionObject = SymbolHelper::GetKernelStructMemberOffset("_EPROCESS", "SectionObject");
if (pInfo->EprocessOffsets.SectionObject == -1)
return false;
pInfo->EprocessOffsets.UniqueProcessId = SymbolHelper::GetKernelStructMemberOffset("_EPROCESS", "UniqueProcessId");
if (pInfo->EprocessOffsets.UniqueProcessId == -1)
return false;
pInfo->EprocessOffsets.ExitTime = SymbolHelper::GetKernelStructMemberOffset("_EPROCESS", "ExitTime");
if (pInfo->EprocessOffsets.ExitTime == -1)
return false;
return true;
}
bool CSystemConfigDlg::InitEthreadOffsets(DbgSysCoreInfo* pInfo) {
pInfo->EthreadOffsets.RundownProtect = SymbolHelper::GetKernelStructMemberOffset("_ETHREAD", "RundownProtect");
if (pInfo->EthreadOffsets.RundownProtect == -1)
return false;
pInfo->EthreadOffsets.CrossThreadFlags = SymbolHelper::GetKernelStructMemberOffset("_ETHREAD", "CrossThreadFlags");
if (pInfo->EthreadOffsets.CrossThreadFlags == -1)
return false;
pInfo->EthreadOffsets.ClonedThread = SymbolHelper::GetKernelStructMemberOffset("_ETHREAD", "ClonedThread");
if (pInfo->EthreadOffsets.ClonedThread != -1) {
ULONG pos = SymbolHelper::GetKernelBitFieldPos("_ETHREAD", "ClonedThread");
pInfo->EthreadOffsets.ClonedThreadBitField.Position = pos;
ULONG size = SymbolHelper::GetKernelStructMemberSize("_ETHREAD", "ClonedThread");
pInfo->EthreadOffsets.ClonedThreadBitField.Size = size;
}
else {
return false;
}
pInfo->EthreadOffsets.Cid = SymbolHelper::GetKernelStructMemberOffset("_ETHREAD", "Cid");
if (pInfo->EthreadOffsets.Cid == -1)
return false;
pInfo->EthreadOffsets.ThreadInserted = SymbolHelper::GetKernelStructMemberOffset("_ETHREAD", "ThreadInserted");
if (pInfo->EthreadOffsets.ThreadInserted != -1) {
ULONG pos = SymbolHelper::GetKernelBitFieldPos("_ETHREAD", "ThreadInserted");
pInfo->EthreadOffsets.ThreadInsertedBitField.Position = pos;
ULONG size = SymbolHelper::GetKernelStructMemberSize("_ETHREAD", "ThreadInserted");
pInfo->EthreadOffsets.ThreadInsertedBitField.Size = size;
}
else {
return false;
}
pInfo->EthreadOffsets.Tcb = SymbolHelper::GetKernelStructMemberOffset("_ETHREAD", "Tcb");
if (pInfo->EthreadOffsets.Tcb == -1)
return false;
pInfo->EthreadOffsets.StartAddress = SymbolHelper::GetKernelStructMemberOffset("_ETHREAD", "StartAddress");
if (pInfo->EthreadOffsets.StartAddress == -1)
return false;
pInfo->EthreadOffsets.Win32StartAddress = SymbolHelper::GetKernelStructMemberOffset("_ETHREAD", "Win32StartAddress");
if (pInfo->EthreadOffsets.Win32StartAddress == -1)
return false;
return true;
}
bool CSystemConfigDlg::InitPebOffsets(DbgSysCoreInfo* pInfo) {
pInfo->PebOffsets.Ldr = SymbolHelper::GetKernelStructMemberOffset("_PEB", "Ldr");
if (!pInfo->PebOffsets.Ldr)
return false;
return true;
}
bool CSystemConfigDlg::InitKthreadOffsets(DbgSysCoreInfo* pInfo) {
pInfo->KthreadOffsets.ApcState = SymbolHelper::GetKernelStructMemberOffset("_KTHREAD", "ApcState");
if (pInfo->KthreadOffsets.ApcState == -1)
return false;
pInfo->KthreadOffsets.ApcStateIndex = SymbolHelper::GetKernelStructMemberOffset("_KTHREAD", "ApcStateIndex");
if (pInfo->KthreadOffsets.ApcStateIndex == -1)
return false;
return true;
}
LRESULT CSystemConfigDlg::OnListViewContext(int /*idCtrl*/, LPNMHDR /*pnmh*/, BOOL& /*bHandled*/) {
CPoint pt;
::GetCursorPos(&pt);
CMenu menu;
menu.LoadMenu(IDR_DBG_CONTEXT);
auto hMenu = menu.GetSubMenu(0);
int index = m_List.GetSelectionMark();
if (index == -1) {
menu.EnableMenuItem(ID_DBG_REMOVE, MF_GRAYED);
}
TrackPopupMenu(hMenu, 0, pt.x, pt.y, 0, m_hWnd, nullptr);
return 0;
}
LRESULT CSystemConfigDlg::OnAddDebugger(WORD /*wNotifyCode*/, WORD /*wID*/, HWND /*hWndCtl*/, BOOL& /*bHandled*/) {
WCHAR path[MAX_PATH] = { 0 };
CFileDialog dlg(TRUE, L"Debugger", nullptr, OFN_FILEMUSTEXIST | OFN_PATHMUSTEXIST | OFN_EXPLORER, L"Executables (*.exe)\0", m_hWnd);
if (dlg.DoModal() == IDOK) {
int n = m_List.AddItem(m_List.GetItemCount(), 0, dlg.m_szFileTitle, 0);
m_List.SetItemText(n, 1, dlg.m_szFileName);
}
return 0;
}
LRESULT CSystemConfigDlg::OnRemoveDebugger(WORD /*wNotifyCode*/, WORD /*wID*/, HWND /*hWndCtl*/, BOOL& /*bHandled*/) {
ATLASSERT(m_List.GetSelectedCount() > 0);
int index = -1;
index = m_List.GetSelectionMark();
if (index == -1) {
return 0;
}
m_List.DeleteItem(index);
return 0;
}
bool CSystemConfigDlg::InitCiSymbols(CiSymbols* pSym) {
pSym->MinCryptIsFileRevoked = (void*)SymbolHelper::GetCiSymbolAddressFromName("MinCryptIsFileRevoked");
if (pSym->MinCryptIsFileRevoked == nullptr)
return false;
pSym->I_MinCryptHashSearchCompare = (void*)SymbolHelper::GetCiSymbolAddressFromName("I_MinCryptHashSearchCompare");
if (pSym->I_MinCryptHashSearchCompare == nullptr)
return false;
return true;
}
Reference in New Issue View Git Blame Copy Permalink