mirror of
https://github.com/rbmm/SC
synced 2026-06-08 17:03:41 +00:00
rbmm
120 lines
2.5 KiB
C++
120 lines
2.5 KiB
C++
#include "stdafx.h"
|
|
|
|
#include "rtlframe.h"
|
|
|
|
ULONG GetLastErrorEx(ULONG dwError /*= GetLastError()*/)
|
|
{
|
|
NTSTATUS status = RtlGetLastNtStatus();
|
|
return RtlNtStatusToDosErrorNoTeb(status) == dwError ? HRESULT_FROM_NT(status) : dwError;
|
|
}
|
|
|
|
struct FICON
|
|
{
|
|
HICON hIcon = 0;
|
|
};
|
|
|
|
LRESULT CALLBACK CBTProc(int nCode, WPARAM wParam, LPARAM lParam)
|
|
{
|
|
if (nCode == HCBT_CREATEWND)
|
|
{
|
|
CBT_CREATEWND* pccw = reinterpret_cast<CBT_CREATEWND*>(lParam);
|
|
|
|
if (pccw->lpcs->lpszClass == WC_DIALOG)
|
|
{
|
|
if (FICON* p = RTL_FRAME<FICON>::get())
|
|
{
|
|
SendMessageW((HWND)wParam, WM_SETICON, ICON_SMALL, (LPARAM)p->hIcon);
|
|
}
|
|
}
|
|
}
|
|
|
|
return CallNextHookEx(0, nCode, wParam, lParam);
|
|
}
|
|
|
|
int CustomMessageBox(HWND hWnd, PCWSTR lpText, PCWSTR lpszCaption, UINT uType)
|
|
{
|
|
PCWSTR pszName = 0;
|
|
|
|
switch (uType & MB_ICONMASK)
|
|
{
|
|
case MB_ICONINFORMATION:
|
|
pszName = IDI_INFORMATION;
|
|
break;
|
|
case MB_ICONQUESTION:
|
|
pszName = IDI_QUESTION;
|
|
break;
|
|
case MB_ICONWARNING:
|
|
pszName = IDI_WARNING;
|
|
break;
|
|
case MB_ICONERROR:
|
|
pszName = IDI_ERROR;
|
|
break;
|
|
}
|
|
|
|
MSGBOXPARAMS mbp = {
|
|
sizeof(mbp),
|
|
IsWindowVisible(hWnd) ? hWnd : 0,
|
|
(HINSTANCE)&__ImageBase,
|
|
lpText,
|
|
lpszCaption,
|
|
(uType & ~MB_ICONMASK) | MB_USERICON,
|
|
MAKEINTRESOURCE(1)
|
|
};
|
|
|
|
HHOOK hhk = 0;
|
|
RTL_FRAME<FICON> frame;
|
|
|
|
if (pszName && 0 <= LoadIconWithScaleDown(0, pszName,
|
|
GetSystemMetrics(SM_CXSMICON), GetSystemMetrics(SM_CYSMICON), &frame.hIcon))
|
|
{
|
|
hhk = SetWindowsHookExW(WH_CBT, CBTProc, 0, GetCurrentThreadId());
|
|
}
|
|
|
|
int i = MessageBoxIndirect(&mbp);
|
|
|
|
if (hhk) UnhookWindowsHookEx(hhk);
|
|
if (frame.hIcon) DestroyIcon(frame.hIcon);
|
|
|
|
return i;
|
|
}
|
|
|
|
HMODULE GetNtMod()
|
|
{
|
|
static HMODULE s_hntmod;
|
|
if (!s_hntmod)
|
|
{
|
|
s_hntmod = GetModuleHandle(L"ntdll");
|
|
}
|
|
|
|
return s_hntmod;
|
|
}
|
|
|
|
int ShowErrorBox(HWND hwnd, HRESULT dwError, PCWSTR lpCaption)
|
|
{
|
|
int r = 0;
|
|
LPCVOID lpSource = 0;
|
|
ULONG dwFlags = FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_IGNORE_INSERTS;
|
|
|
|
if ((dwError & FACILITY_NT_BIT) || (0 > dwError && HRESULT_FACILITY(dwError) == FACILITY_NULL))
|
|
{
|
|
dwError &= ~FACILITY_NT_BIT;
|
|
__nt:
|
|
dwFlags = FORMAT_MESSAGE_FROM_HMODULE | FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_IGNORE_INSERTS;
|
|
|
|
lpSource = GetNtMod();
|
|
}
|
|
|
|
PWSTR lpText;
|
|
if (FormatMessageW(dwFlags, lpSource, dwError, 0, (PWSTR)&lpText, 0, 0))
|
|
{
|
|
r = CustomMessageBox(hwnd, lpText, lpCaption, dwError ? MB_ICONERROR : MB_ICONINFORMATION);
|
|
LocalFree(lpText);
|
|
}
|
|
else if (dwFlags & FORMAT_MESSAGE_FROM_SYSTEM)
|
|
{
|
|
goto __nt;
|
|
}
|
|
|
|
return r;
|
|
}
|