You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
423 lines
11 KiB
423 lines
11 KiB
/*++
|
|
|
|
Copyright (c) Microsoft Corporation. All rights reserved.
|
|
|
|
Module Name:
|
|
|
|
Main.cpp
|
|
|
|
Abstract:
|
|
|
|
Implements the entry point and message
|
|
pump for the application.
|
|
|
|
Notes:
|
|
|
|
Unicode only.
|
|
|
|
History:
|
|
|
|
05/04/2001 rparsons Created
|
|
01/11/2002 rparsons Cleaned up
|
|
|
|
--*/
|
|
#include "precomp.h"
|
|
|
|
APPINFO g_ai;
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Application entry point.
|
|
|
|
Arguments:
|
|
|
|
hInstance - App instance handle.
|
|
hPrevInstance - Always NULL.
|
|
lpCmdLine - Pointer to the command line.
|
|
nCmdShow - Window show flag.
|
|
|
|
Return Value:
|
|
|
|
The wParam of the message or 0 on failure.
|
|
|
|
--*/
|
|
int
|
|
APIENTRY
|
|
WinMain(
|
|
IN HINSTANCE hInstance,
|
|
IN HINSTANCE hPrevInstance,
|
|
IN LPSTR lpCmdLine,
|
|
IN int nCmdShow
|
|
)
|
|
{
|
|
MSG msg;
|
|
WNDCLASS wndclass;
|
|
WCHAR wszError[MAX_PATH];
|
|
RECT rcDialog;
|
|
RECT rcDesktopWorkArea;
|
|
INITCOMMONCONTROLSEX icex;
|
|
POINT pt;
|
|
HANDLE hMutex;
|
|
int nWidth = 0, nHeight = 0;
|
|
int nTaskbarHeight = 0;
|
|
OSVERSIONINFOEX osvi = {0};
|
|
|
|
UNREFERENCED_PARAMETER(lpCmdLine);
|
|
UNREFERENCED_PARAMETER(hPrevInstance);
|
|
|
|
g_ai.hInstance = hInstance;
|
|
|
|
//
|
|
// Make sure we're the only instance running.
|
|
//
|
|
hMutex = CreateMutex(NULL, FALSE, L"ShimViewer");
|
|
|
|
if (ERROR_ALREADY_EXISTS == GetLastError()) {
|
|
return 0;
|
|
}
|
|
|
|
osvi.dwOSVersionInfoSize = sizeof(OSVERSIONINFOEX);
|
|
|
|
if (!GetVersionEx((OSVERSIONINFO*)&osvi)) {
|
|
MessageBox(
|
|
NULL,
|
|
L"Failed to get the OS version info",
|
|
L"Error",
|
|
MB_ICONERROR);
|
|
|
|
return -1;
|
|
}
|
|
|
|
if (osvi.dwPlatformId == VER_PLATFORM_WIN32_NT &&
|
|
osvi.dwMajorVersion == 5 &&
|
|
osvi.dwMinorVersion >= 2) {
|
|
|
|
g_ai.bUsingNewShimEng = TRUE;
|
|
|
|
//
|
|
// If we are using shimeng from NT 5.2 or newer we need
|
|
// to create the debug objects to get debug spew from
|
|
// OutputDebugString.
|
|
//
|
|
if (!CreateDebugObjects()) {
|
|
MessageBox(
|
|
NULL,
|
|
L"Failed to create the necessary objects to get debug spew "
|
|
L"from OutputDebugString",
|
|
L"Error",
|
|
MB_ICONERROR);
|
|
return 0;
|
|
}
|
|
|
|
} else {
|
|
|
|
g_ai.bUsingNewShimEng = FALSE;
|
|
}
|
|
|
|
wndclass.style = CS_HREDRAW | CS_VREDRAW;
|
|
wndclass.lpfnWndProc = (WNDPROC)MainWndProc;
|
|
wndclass.cbClsExtra = 0;
|
|
wndclass.cbWndExtra = DLGWINDOWEXTRA;
|
|
wndclass.hInstance = hInstance;
|
|
wndclass.hIcon = NULL;
|
|
wndclass.hCursor = LoadCursor(NULL, IDC_ARROW);
|
|
wndclass.hbrBackground = (HBRUSH)(COLOR_BTNFACE + 1);
|
|
wndclass.lpszMenuName = NULL;
|
|
wndclass.lpszClassName = APP_CLASS;
|
|
|
|
if (!RegisterClass(&wndclass)) {
|
|
LoadString(hInstance, IDS_NO_CLASS, wszError, ARRAYSIZE(wszError));
|
|
MessageBox(NULL, wszError, APP_NAME, MB_ICONERROR);
|
|
return 0;
|
|
}
|
|
|
|
//
|
|
// Set up the common controls.
|
|
//
|
|
icex.dwSize = sizeof(INITCOMMONCONTROLSEX);
|
|
icex.dwICC = ICC_LISTVIEW_CLASSES;
|
|
|
|
if (!InitCommonControlsEx(&icex)) {
|
|
InitCommonControls();
|
|
}
|
|
|
|
//
|
|
// Get application settings from the registry, if there are any.
|
|
//
|
|
GetSaveSettings(FALSE);
|
|
|
|
g_ai.hMainDlg = CreateDialog(hInstance,
|
|
(LPCWSTR)IDD_MAIN,
|
|
NULL,
|
|
MainWndProc);
|
|
|
|
if (!g_ai.hMainDlg) {
|
|
LoadString(hInstance, IDS_NO_MAIN_DLG, wszError, ARRAYSIZE(wszError));
|
|
MessageBox(NULL, wszError, APP_NAME, MB_ICONERROR);
|
|
return 0;
|
|
}
|
|
|
|
//
|
|
// Get the window position info from the registry, if it's there.
|
|
//
|
|
GetSavePositionInfo(FALSE, &pt);
|
|
|
|
//
|
|
// If previous settings were retrieved from the registry, use them.
|
|
//
|
|
if (pt.x != 0) {
|
|
SetWindowPos(g_ai.hMainDlg,
|
|
g_ai.fOnTop ? HWND_TOPMOST : HWND_NOTOPMOST,
|
|
pt.x,
|
|
pt.y,
|
|
0,
|
|
0,
|
|
SWP_NOSIZE | SWP_SHOWWINDOW);
|
|
|
|
} else {
|
|
//
|
|
// Get the coords of the desktop window and place the dialog.
|
|
// We put it in the bottom-right corner of the desktop above
|
|
// the taskbar.
|
|
//
|
|
SystemParametersInfo(SPI_GETWORKAREA, 0, &rcDesktopWorkArea, 0);
|
|
|
|
GetWindowRect(g_ai.hMainDlg, &rcDialog);
|
|
|
|
nWidth = rcDialog.right - rcDialog.left;
|
|
nHeight = rcDialog.bottom - rcDialog.top;
|
|
|
|
SetWindowPos(g_ai.hMainDlg,
|
|
g_ai.fOnTop ? HWND_TOPMOST : HWND_NOTOPMOST,
|
|
rcDesktopWorkArea.right - nWidth,
|
|
rcDesktopWorkArea.bottom - nHeight,
|
|
0,
|
|
0,
|
|
SWP_NOSIZE | SWP_SHOWWINDOW);
|
|
}
|
|
|
|
ShowWindow(g_ai.hMainDlg, g_ai.fMinimize ? SW_MINIMIZE : SW_SHOWNORMAL);
|
|
|
|
while (GetMessage(&msg, (HWND)NULL, 0, 0)) {
|
|
if (!IsDialogMessage(g_ai.hMainDlg, &msg)) {
|
|
TranslateMessage(&msg);
|
|
DispatchMessage(&msg);
|
|
}
|
|
}
|
|
|
|
return (int)msg.wParam;
|
|
}
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Runs the message loop for the app.
|
|
|
|
Arguments:
|
|
|
|
hWnd - Handle to the window.
|
|
uMsg - Windows message.
|
|
wParam - Additional message info.
|
|
lParam - Additional message info.
|
|
|
|
Return Value:
|
|
|
|
TRUE if handled, FALSE otherwise.
|
|
|
|
--*/
|
|
INT_PTR
|
|
CALLBACK
|
|
MainWndProc(
|
|
IN HWND hWnd,
|
|
IN UINT uMsg,
|
|
IN WPARAM wParam,
|
|
IN LPARAM lParam
|
|
)
|
|
{
|
|
switch (uMsg) {
|
|
case WM_INITDIALOG:
|
|
{
|
|
WCHAR wszError[MAX_PATH];
|
|
HICON hIcon;
|
|
|
|
//
|
|
// Initialize the list view, menu items, and then create our thread.
|
|
//
|
|
g_ai.hWndList = GetDlgItem(hWnd, IDC_LIST);
|
|
InitListViewColumn();
|
|
|
|
hIcon = LoadIcon(g_ai.hInstance, MAKEINTRESOURCE(IDI_APPICON));
|
|
|
|
SetClassLongPtr(hWnd, GCLP_HICON, (LONG_PTR)hIcon);
|
|
|
|
CheckMenuItem(GetMenu(hWnd),
|
|
IDM_ON_TOP,
|
|
g_ai.fOnTop ? MF_CHECKED : MF_UNCHECKED);
|
|
|
|
CheckMenuItem(GetMenu(hWnd),
|
|
IDM_START_SMALL,
|
|
g_ai.fMinimize ? MF_CHECKED : MF_UNCHECKED);
|
|
|
|
CheckMenuItem(GetMenu(hWnd),
|
|
IDM_MONITOR,
|
|
g_ai.fMonitor ? MF_CHECKED : MF_UNCHECKED);
|
|
|
|
if (g_ai.fMonitor) {
|
|
if (!CreateReceiveThread()) {
|
|
LoadString(g_ai.hInstance,
|
|
IDS_CREATE_FAILED,
|
|
wszError,
|
|
ARRAYSIZE(wszError));
|
|
MessageBox(hWnd, wszError, APP_NAME, MB_ICONERROR);
|
|
|
|
g_ai.fMonitor = FALSE;
|
|
} else {
|
|
SetPriorityClass(GetCurrentProcess(), ABOVE_NORMAL_PRIORITY_CLASS);
|
|
}
|
|
}
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
case WM_CLOSE:
|
|
{
|
|
RECT rc;
|
|
|
|
GetWindowRect(hWnd, &rc);
|
|
GetSavePositionInfo(TRUE, (LPPOINT)&rc);
|
|
GetSaveSettings(TRUE);
|
|
RemoveFromTray(hWnd);
|
|
EndDialog(hWnd, 0);
|
|
PostQuitMessage(0);
|
|
break;
|
|
}
|
|
|
|
case WM_SIZE:
|
|
{
|
|
HICON hIcon;
|
|
|
|
if (SIZE_MINIMIZED == wParam) {
|
|
ShowWindow(hWnd, SW_HIDE);
|
|
|
|
hIcon = (HICON)LoadImage(g_ai.hInstance,
|
|
MAKEINTRESOURCE(IDI_APPICON),
|
|
IMAGE_ICON,
|
|
16,
|
|
16,
|
|
0);
|
|
|
|
AddIconToTray(hWnd, hIcon, APP_NAME);
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
MoveWindow(g_ai.hWndList, 0, 0, LOWORD(lParam), HIWORD(lParam), TRUE);
|
|
|
|
break;
|
|
}
|
|
|
|
case WM_NOTIFYICON:
|
|
switch (lParam) {
|
|
case WM_RBUTTONUP:
|
|
|
|
DisplayMenu(hWnd);
|
|
break;
|
|
|
|
case WM_LBUTTONDBLCLK:
|
|
|
|
RemoveFromTray(hWnd);
|
|
ShowWindow(hWnd, SW_SHOWNORMAL);
|
|
break;
|
|
|
|
default:
|
|
break;
|
|
}
|
|
|
|
case WM_COMMAND:
|
|
switch (LOWORD(wParam)) {
|
|
case IDM_EXIT:
|
|
|
|
PostMessage(hWnd, WM_CLOSE, 0, 0);
|
|
break;
|
|
|
|
case IDM_RESTORE:
|
|
|
|
ShowWindow(hWnd, SW_SHOWNORMAL);
|
|
SetWindowPos(hWnd,
|
|
g_ai.fOnTop ? HWND_TOPMOST : HWND_NOTOPMOST,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
SWP_NOSIZE | SWP_NOMOVE);
|
|
RemoveFromTray(hWnd);
|
|
break;
|
|
|
|
case IDM_ABOUT:
|
|
|
|
ShellAbout(hWnd,
|
|
APP_NAME,
|
|
WRITTEN_BY,
|
|
LoadIcon(g_ai.hInstance, MAKEINTRESOURCE(IDI_APPICON)));
|
|
break;
|
|
|
|
case IDM_MONITOR:
|
|
|
|
CheckMenuItem(GetMenu(hWnd),
|
|
IDM_MONITOR,
|
|
g_ai.fMonitor ? MF_UNCHECKED : MF_CHECKED);
|
|
|
|
g_ai.fMonitor = g_ai.fMonitor ? FALSE : TRUE;
|
|
|
|
if (g_ai.fMonitor) {
|
|
CreateReceiveThread();
|
|
}
|
|
|
|
break;
|
|
|
|
case IDM_ON_TOP:
|
|
|
|
CheckMenuItem(GetMenu(hWnd),
|
|
IDM_ON_TOP,
|
|
g_ai.fOnTop ? MF_UNCHECKED : MF_CHECKED);
|
|
|
|
SetWindowPos(hWnd,
|
|
g_ai.fOnTop ? HWND_NOTOPMOST : HWND_TOPMOST,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
SWP_NOSIZE | SWP_NOMOVE);
|
|
|
|
g_ai.fOnTop = g_ai.fOnTop ? FALSE : TRUE;
|
|
break;
|
|
|
|
case IDM_START_SMALL:
|
|
|
|
CheckMenuItem(GetMenu(hWnd),
|
|
IDM_START_SMALL,
|
|
g_ai.fMinimize ? MF_UNCHECKED : MF_CHECKED);
|
|
|
|
g_ai.fMinimize = g_ai.fMinimize ? FALSE : TRUE;
|
|
break;
|
|
|
|
case IDM_CLEAR:
|
|
|
|
ListView_DeleteAllItems(g_ai.hWndList);
|
|
break;
|
|
|
|
default:
|
|
break;
|
|
|
|
}
|
|
|
|
default:
|
|
break;
|
|
}
|
|
|
|
return FALSE;
|
|
}
|