479 lines
13 KiB
C++
479 lines
13 KiB
C++
// ConMsg.cpp : Defines the entry point for the application.
|
|
//
|
|
|
|
#include "stdafx.h"
|
|
#include "resource.h"
|
|
#include "stdio.h"
|
|
#include "tchar.h"
|
|
#include "wtsapi32.h"
|
|
#include "winsta.h"
|
|
|
|
#define MAX_LOADSTRING 100
|
|
|
|
// Global Variables:
|
|
HINSTANCE hInst; // current instance
|
|
TCHAR szTitle[MAX_LOADSTRING]; // The title bar text
|
|
TCHAR szWindowClass[MAX_LOADSTRING]; // The title bar text
|
|
|
|
// Foward declarations of functions included in this code module:
|
|
ATOM MyRegisterClass(HINSTANCE hInstance);
|
|
BOOL InitInstance(HINSTANCE, int);
|
|
LRESULT CALLBACK WndProc(HWND, UINT, WPARAM, LPARAM);
|
|
LRESULT CALLBACK About(HWND, UINT, WPARAM, LPARAM);
|
|
|
|
int APIENTRY WinMain(HINSTANCE hInstance,
|
|
HINSTANCE /* hPrevInstance*/,
|
|
LPSTR /* lpCmdLine */,
|
|
int nCmdShow)
|
|
{
|
|
// TODO: Place code here.
|
|
MSG msg;
|
|
HACCEL hAccelTable;
|
|
|
|
// Initialize global strings
|
|
LoadString(hInstance, IDS_APP_TITLE, szTitle, MAX_LOADSTRING);
|
|
LoadString(hInstance, IDC_POWERMSG, szWindowClass, MAX_LOADSTRING);
|
|
MyRegisterClass(hInstance);
|
|
|
|
// Perform application initialization:
|
|
if (!InitInstance (hInstance, nCmdShow))
|
|
{
|
|
return FALSE;
|
|
}
|
|
|
|
hAccelTable = LoadAccelerators(hInstance, (LPCTSTR)IDC_POWERMSG);
|
|
|
|
// Main message loop:
|
|
while (GetMessage(&msg, NULL, 0, 0))
|
|
{
|
|
if (!TranslateAccelerator(msg.hwnd, hAccelTable, &msg))
|
|
{
|
|
TranslateMessage(&msg);
|
|
DispatchMessage(&msg);
|
|
}
|
|
}
|
|
|
|
return msg.wParam;
|
|
}
|
|
|
|
|
|
|
|
//
|
|
// FUNCTION: MyRegisterClass()
|
|
//
|
|
// PURPOSE: Registers the window class.
|
|
//
|
|
// COMMENTS:
|
|
//
|
|
// This function and its usage is only necessary if you want this code
|
|
// to be compatible with Win32 systems prior to the 'RegisterClassEx'
|
|
// function that was added to Windows 95. It is important to call this function
|
|
// so that the application will get 'well formed' small icons associated
|
|
// with it.
|
|
//
|
|
ATOM MyRegisterClass(HINSTANCE hInstance)
|
|
{
|
|
WNDCLASSEX wcex;
|
|
|
|
wcex.cbSize = sizeof(WNDCLASSEX);
|
|
|
|
wcex.style = CS_HREDRAW | CS_VREDRAW;
|
|
wcex.lpfnWndProc = (WNDPROC)WndProc;
|
|
wcex.cbClsExtra = 0;
|
|
wcex.cbWndExtra = 0;
|
|
wcex.hInstance = hInstance;
|
|
wcex.hIcon = LoadIcon(hInstance, (LPCTSTR)IDI_POWERMSG);
|
|
wcex.hCursor = LoadCursor(NULL, IDC_ARROW);
|
|
wcex.hbrBackground = (HBRUSH)(COLOR_WINDOW+1);
|
|
wcex.lpszMenuName = (LPCSTR)IDC_POWERMSG;
|
|
wcex.lpszClassName = szWindowClass;
|
|
wcex.hIconSm = LoadIcon(wcex.hInstance, (LPCTSTR)IDI_SMALL);
|
|
|
|
return RegisterClassEx(&wcex);
|
|
}
|
|
|
|
//
|
|
// FUNCTION: InitInstance(HANDLE, int)
|
|
//
|
|
// PURPOSE: Saves instance handle and creates main window
|
|
//
|
|
// COMMENTS:
|
|
//
|
|
// In this function, we save the instance handle in a global variable and
|
|
// create and display the main program window.
|
|
//
|
|
BOOL InitInstance(HINSTANCE hInstance, int nCmdShow)
|
|
{
|
|
|
|
HWND hWnd;
|
|
|
|
hInst = hInstance; // Store instance handle in our global variable
|
|
|
|
hWnd = CreateWindow(szWindowClass, szTitle, WS_OVERLAPPEDWINDOW,
|
|
CW_USEDEFAULT, 0, CW_USEDEFAULT, 0, NULL, NULL, hInstance, NULL);
|
|
|
|
|
|
if (!hWnd)
|
|
{
|
|
return FALSE;
|
|
}
|
|
|
|
|
|
ShowWindow(hWnd, nCmdShow);
|
|
UpdateWindow(hWnd);
|
|
|
|
// if (!SetTimer(hWnd, 23, 1500, NULL))
|
|
// MessageBox(NULL, _T("Failed"), _T("Failed"), MB_OK);
|
|
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
//
|
|
// FUNCTION: WndProc(HWND, unsigned, WORD, LONG)
|
|
//
|
|
// PURPOSE: Processes messages for the main window.
|
|
//
|
|
// WM_COMMAND - process the application menu
|
|
// WM_PAINT - Paint the main window
|
|
// WM_DESTROY - post a quit message and return
|
|
//
|
|
//
|
|
LRESULT CALLBACK WndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
|
|
{
|
|
int wmId, wmEvent;
|
|
static HWND hChildWnd;
|
|
LRESULT lResult;
|
|
TCHAR szString[512];
|
|
int iCount;
|
|
static int iState = 0;
|
|
DWORD dwSessionId;
|
|
// int iVal;
|
|
// HWND hWnd2;
|
|
|
|
switch (message)
|
|
{
|
|
case WM_CREATE:
|
|
lResult = DefWindowProc(hWnd, message, wParam, lParam);
|
|
RECT rt;
|
|
GetClientRect(hWnd, &rt);
|
|
hChildWnd = CreateWindow("LISTBOX", szTitle, WS_CHILD | WS_VISIBLE,
|
|
0, 0, rt.right - rt.left, rt.bottom - rt.top, hWnd, NULL, hInst, NULL);
|
|
|
|
srand( 25 );
|
|
if (!SetTimer(hWnd, 23, 1500, NULL))
|
|
MessageBox(NULL, _T("Failed"), _T("Failed"), MB_OK);
|
|
|
|
|
|
|
|
return lResult;
|
|
break;
|
|
|
|
/*
|
|
case WM_TIMER:
|
|
SendMessage(hChildWnd, LB_ADDSTRING, 0, LPARAM(_T("Got Timer")));
|
|
// hWnd2 = CreateWindow(szWindowClass, szTitle, WS_OVERLAPPEDWINDOW,
|
|
// CW_USEDEFAULT, 0, CW_USEDEFAULT, 0, NULL, NULL, hInst, NULL);
|
|
// ShowWindow(hWnd2, SW_SHOWDEFAULT);
|
|
// UpdateWindow(hWnd2);
|
|
|
|
|
|
|
|
iVal = rand() % 3;
|
|
switch (iVal)
|
|
{
|
|
case 0:
|
|
PostMessage(hWnd, WM_COMMAND, ID_FILE_REGISTERALLSESSIONS, 0);
|
|
break;
|
|
case 1:
|
|
PostMessage(hWnd, WM_COMMAND, ID_FILE_REGISTERFORTHISSESSION, 0);
|
|
break;
|
|
case 2:
|
|
PostMessage(hWnd, WM_COMMAND, ID_FILE_UNREGISTER, 0);
|
|
break;
|
|
}
|
|
break;
|
|
|
|
*/
|
|
|
|
case WM_POWERBROADCAST:
|
|
{
|
|
switch (wParam)
|
|
{
|
|
case PBT_APMBATTERYLOW:
|
|
SendMessage(hChildWnd, LB_ADDSTRING, 0, LPARAM(_T("PBT_APMBATTERYLOW")));
|
|
break;
|
|
case PBT_APMOEMEVENT:
|
|
SendMessage(hChildWnd, LB_ADDSTRING, 0, LPARAM(_T("PBT_APMOEMEVENT")));
|
|
break;
|
|
case PBT_APMPOWERSTATUSCHANGE:
|
|
SendMessage(hChildWnd, LB_ADDSTRING, 0, LPARAM(_T("PBT_APMPOWERSTATUSCHANGE")));
|
|
break;
|
|
case PBT_APMQUERYSUSPEND:
|
|
SendMessage(hChildWnd, LB_ADDSTRING, 0, LPARAM(_T("PBT_APMQUERYSUSPEND")));
|
|
break;
|
|
case PBT_APMQUERYSUSPENDFAILED:
|
|
SendMessage(hChildWnd, LB_ADDSTRING, 0, LPARAM(_T("PBT_APMQUERYSUSPENDFAILED")));
|
|
break;
|
|
case PBT_APMRESUMEAUTOMATIC:
|
|
SendMessage(hChildWnd, LB_ADDSTRING, 0, LPARAM(_T("PBT_APMRESUMEAUTOMATIC")));
|
|
break;
|
|
case PBT_APMRESUMECRITICAL:
|
|
SendMessage(hChildWnd, LB_ADDSTRING, 0, LPARAM(_T("PBT_APMRESUMECRITICAL")));
|
|
break;
|
|
case PBT_APMRESUMESUSPEND:
|
|
SendMessage(hChildWnd, LB_ADDSTRING, 0, LPARAM(_T("PBT_APMRESUMESUSPEND")));
|
|
break;
|
|
case PBT_APMSUSPEND:
|
|
SendMessage(hChildWnd, LB_ADDSTRING, 0, LPARAM(_T("PBT_APMSUSPEND")));
|
|
break;
|
|
}
|
|
}
|
|
break;
|
|
|
|
case WM_COMMAND:
|
|
wmId = LOWORD(wParam);
|
|
wmEvent = HIWORD(wParam);
|
|
// Parse the menu selections:
|
|
switch (wmId)
|
|
{
|
|
case ID_FILE_REGISTERALLSESSIONS:
|
|
OutputDebugString(TEXT("* ConMsg:calling WTSRegisterSessionNotification\n"));
|
|
if (WTSRegisterSessionNotification(hWnd, NOTIFY_FOR_ALL_SESSIONS))
|
|
{
|
|
iState = 2;
|
|
}
|
|
else
|
|
{
|
|
SendMessage(hChildWnd, LB_ADDSTRING, 0, LPARAM(_T("ERROR:Registering for all sessions **failed**.")));
|
|
}
|
|
|
|
OutputDebugString(TEXT("* ConMsg:done with WTSRegisterSessionNotification\n"));
|
|
break;
|
|
|
|
case ID_FILE_REGISTERFORTHISSESSION:
|
|
OutputDebugString(TEXT("* ConMsg:calling WTSRegisterSessionNotification\n"));
|
|
if (WTSRegisterSessionNotification(hWnd, NOTIFY_FOR_THIS_SESSION))
|
|
{
|
|
iState = 1;
|
|
}
|
|
else
|
|
{
|
|
SendMessage(hChildWnd, LB_ADDSTRING, 0, LPARAM(_T("ERROR:Registered for this session **failed**.")));
|
|
}
|
|
|
|
OutputDebugString(TEXT("* ConMsg:done with WTSRegisterSessionNotification\n"));
|
|
break;
|
|
|
|
case ID_FILE_UNREGISTER:
|
|
OutputDebugString(TEXT("* ConMsg:calling WTSUnRegisterSessionNotification\n"));
|
|
if (WTSUnRegisterSessionNotification(hWnd))
|
|
{
|
|
iState = 0;
|
|
}
|
|
else
|
|
{
|
|
SendMessage(hChildWnd, LB_ADDSTRING, 0, LPARAM(_T("UNRegistered for this sessions **failed**.")));
|
|
}
|
|
OutputDebugString(TEXT("* ConMsg:done with WTSUnRegisterConsoleNotification\n"));
|
|
break;
|
|
|
|
case ID_FILE_CLEANLOG:
|
|
iCount = SendMessage(hChildWnd, LB_GETCOUNT , 0, 0);
|
|
while (iCount != LB_ERR && iCount > 0)
|
|
{
|
|
SendMessage(hChildWnd, LB_DELETESTRING , 0, 0);
|
|
iCount = SendMessage(hChildWnd, LB_GETCOUNT , 0, 0);
|
|
}
|
|
break;
|
|
|
|
|
|
case IDM_EXIT:
|
|
DestroyWindow(hWnd);
|
|
iState = 3;
|
|
break;
|
|
|
|
case IDM_GET_SESSIONSTATE:
|
|
{
|
|
BOOL bLockedState;
|
|
DWORD ReturnLength;
|
|
if (!WinStationQueryInformation( SERVERNAME_CURRENT,
|
|
LOGONID_CURRENT,
|
|
WinStationLockedState,
|
|
(PVOID)&bLockedState,
|
|
sizeof(bLockedState),
|
|
&ReturnLength))
|
|
{
|
|
_stprintf(szString, _T("WinStationQueryInformationfailed, LastError = %d"), GetLastError());
|
|
SendMessage(hChildWnd, LB_ADDSTRING, 0, LPARAM(szString));
|
|
}
|
|
else
|
|
{
|
|
_stprintf(szString, _T("WinStationQueryInformation Succeeded"));
|
|
SendMessage(hChildWnd, LB_ADDSTRING, 0, LPARAM(szString));
|
|
_stprintf(szString, _T(" LockedState = %s"), bLockedState ? _T("On") : _T("Off"));
|
|
SendMessage(hChildWnd, LB_ADDSTRING, 0, LPARAM(szString));
|
|
}
|
|
|
|
}
|
|
break;
|
|
|
|
case IDM_SET_WELCOME_ON:
|
|
{
|
|
BOOL bWelcomeOn = 1;
|
|
if (!WinStationSetInformation( SERVERNAME_CURRENT,
|
|
LOGONID_CURRENT,
|
|
WinStationLockedState,
|
|
&bWelcomeOn, sizeof(bWelcomeOn)))
|
|
|
|
{
|
|
|
|
_stprintf(szString, _T("WinStationSetInformation failed, LastError = %d"), GetLastError());
|
|
SendMessage(hChildWnd, LB_ADDSTRING, 0, LPARAM(szString));
|
|
}
|
|
else
|
|
{
|
|
_stprintf(szString, _T("WinStationSetInformation succeeded"));
|
|
SendMessage(hChildWnd, LB_ADDSTRING, 0, LPARAM(szString));
|
|
}
|
|
|
|
}
|
|
break;
|
|
|
|
case IDM_SET_WELCOME_OFF:
|
|
{
|
|
BOOL bWelcomeOff = 0;
|
|
if (!WinStationSetInformation( SERVERNAME_CURRENT,
|
|
LOGONID_CURRENT,
|
|
WinStationLockedState,
|
|
&bWelcomeOff, sizeof(bWelcomeOff)))
|
|
|
|
{
|
|
|
|
_stprintf(szString, _T("WinStationSetInformation failed, LastError = %d"), GetLastError());
|
|
SendMessage(hChildWnd, LB_ADDSTRING, 0, LPARAM(szString));
|
|
}
|
|
else
|
|
{
|
|
_stprintf(szString, _T("WinStationSetInformation succeeded"));
|
|
SendMessage(hChildWnd, LB_ADDSTRING, 0, LPARAM(szString));
|
|
}
|
|
}
|
|
|
|
break;
|
|
|
|
default:
|
|
return DefWindowProc(hWnd, message, wParam, lParam);
|
|
}
|
|
|
|
switch (iState)
|
|
{
|
|
case 0:
|
|
_tcscpy(szString, _T("UNRegistered for this sessions."));
|
|
break;
|
|
case 1:
|
|
_tcscpy(szString, _T("Registered for this sessions."));
|
|
break;
|
|
|
|
case 2:
|
|
_tcscpy(szString, _T("Registered for all sessions."));
|
|
break;
|
|
|
|
case 3:
|
|
default:
|
|
iState = 3;
|
|
}
|
|
|
|
if (iState != 3)
|
|
{
|
|
SendMessage(hChildWnd, LB_ADDSTRING, 0, LPARAM(szString));
|
|
}
|
|
|
|
break;
|
|
|
|
case WM_DESTROY:
|
|
PostQuitMessage(0);
|
|
break;
|
|
|
|
|
|
case WM_WTSSESSION_CHANGE:
|
|
DWORD GetSessionId(LPARAM lParam);
|
|
dwSessionId = GetSessionId(lParam);
|
|
|
|
switch (wParam)
|
|
{
|
|
case WTS_CONSOLE_CONNECT:
|
|
_stprintf(szString, _T("WTS_CONSOLE_CONNECT, Session Affected was %d"), dwSessionId);
|
|
break;
|
|
|
|
case WTS_CONSOLE_DISCONNECT:
|
|
_stprintf(szString, _T("WTS_CONSOLE_DISCONNECT, Session Affected was %d"), dwSessionId);
|
|
break;
|
|
|
|
case WTS_REMOTE_CONNECT:
|
|
_stprintf(szString, _T("WTS_REMOTE_CONNECT, Session Affected was %d"), dwSessionId);
|
|
break;
|
|
|
|
case WTS_REMOTE_DISCONNECT:
|
|
_stprintf(szString, _T("WTS_REMOTE_DISCONNECT, Session Affected was %d"), dwSessionId);
|
|
break;
|
|
|
|
case WTS_SESSION_LOGON:
|
|
_stprintf(szString, _T("WTS_SESSION_LOGON, Session Affected was %d"), dwSessionId);
|
|
break;
|
|
|
|
case WTS_SESSION_LOGOFF:
|
|
_stprintf(szString, _T("WTS_SESSION_LOGOFF, Session Affected was %d"), dwSessionId);
|
|
break;
|
|
|
|
case WTS_SESSION_LOCK:
|
|
_stprintf(szString, _T("WTS_SESSION_LOCK, Session Affected was %d"), dwSessionId);
|
|
break;
|
|
|
|
case WTS_SESSION_UNLOCK:
|
|
_stprintf(szString, _T("WTS_SESSION_UNLOCK, Session Affected was %d"), dwSessionId);
|
|
break;
|
|
|
|
default:
|
|
_stprintf(szString, _T("Unknown Notification Code!!!!, Session Affected was %d"), dwSessionId);
|
|
|
|
break;
|
|
|
|
}
|
|
SendMessage(hChildWnd, LB_ADDSTRING, 0, LPARAM(szString));
|
|
break;
|
|
|
|
|
|
default:
|
|
return DefWindowProc(hWnd, message, wParam, lParam);
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
// Mesage handler for about box.
|
|
LRESULT CALLBACK About(HWND hDlg, UINT message, WPARAM wParam, LPARAM /* lParam */)
|
|
{
|
|
switch (message)
|
|
{
|
|
case WM_INITDIALOG:
|
|
return TRUE;
|
|
|
|
case WM_COMMAND:
|
|
if (LOWORD(wParam) == IDOK || LOWORD(wParam) == IDCANCEL)
|
|
{
|
|
EndDialog(hDlg, LOWORD(wParam));
|
|
return TRUE;
|
|
}
|
|
break;
|
|
}
|
|
return FALSE;
|
|
}
|
|
|
|
DWORD GetSessionId(LPARAM lParam)
|
|
{
|
|
return lParam;
|
|
|
|
}
|
|
|
|
|