|
|
// 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;
}
|
