TestWtsApiSessionNotification.c

 
#include <winpr/crt.h>
#include <winpr/error.h>
#include <winpr/wtsapi.h>
#include <winpr/environment.h>
 
int TestWtsApiSessionNotification(int argc, char* argv[])
{
  HWND hWnd = NULL;
  BOOL bSuccess = 0;
  DWORD dwFlags = 0;
 
  WINPR_UNUSED(argc);
  WINPR_UNUSED(argv);
 
#ifndef _WIN32
  if (!GetEnvironmentVariableA("WTSAPI_LIBRARY", NULL, 0))
  {
    printf("%s: No RDS environment detected, skipping test\n", __func__);
    return 0;
  }
#else
  /* We create a message-only window and use the predefined class name "STATIC" for simplicity */
  hWnd = CreateWindowA("STATIC", "TestWtsApiSessionNotification", 0, 0, 0, 0, 0, HWND_MESSAGE,
                       NULL, NULL, NULL);
  if (!hWnd)
  {
    printf("%s: error creating message-only window: %" PRIu32 "\n", __func__, GetLastError());
    return -1;
  }
#endif
 
  dwFlags = NOTIFY_FOR_ALL_SESSIONS;
 
  bSuccess = WTSRegisterSessionNotification(hWnd, dwFlags);
 
  if (!bSuccess)
  {
    printf("%s: WTSRegisterSessionNotification failed: %" PRIu32 "\n", __func__,
           GetLastError());
    return -1;
  }
 
  bSuccess = WTSUnRegisterSessionNotification(hWnd);
 
#ifdef _WIN32
  if (hWnd)
  {
    DestroyWindow(hWnd);
    hWnd = NULL;
  }
#endif
 
  if (!bSuccess)
  {
    printf("%s: WTSUnRegisterSessionNotification failed: %" PRIu32 "\n", __func__,
           GetLastError());
    return -1;
  }
 
  return 0;
}