TestWtsApiEnumerateProcesses.c

 
#include <winpr/crt.h>
#include <winpr/error.h>
#include <winpr/wtsapi.h>
#include <winpr/environment.h>
 
int TestWtsApiEnumerateProcesses(int argc, char* argv[])
{
  DWORD count = 0;
  BOOL bSuccess = 0;
  HANDLE hServer = NULL;
  PWTS_PROCESS_INFOA pProcessInfo = NULL;
 
  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;
  }
#endif
 
  hServer = WTS_CURRENT_SERVER_HANDLE;
 
  count = 0;
  pProcessInfo = NULL;
 
  bSuccess = WTSEnumerateProcessesA(hServer, 0, 1, &pProcessInfo, &count);
 
  if (!bSuccess)
  {
    printf("WTSEnumerateProcesses failed: %" PRIu32 "\n", GetLastError());
    return -1;
  }
 
  int rc = 0;
  {
    printf("WTSEnumerateProcesses enumerated %"PRIu32" process:\n", count);
    for (DWORD i = 0; i < count; i++)
    {
      const WTS_PROCESS_INFOA* cur = &pProcessInfo[i];
      if (!cur->pProcessName)
        rc = -1;
      printf("\t[%" PRIu32 "]: %s (%" PRIu32 ")\n", i, cur->pProcessName, cur->ProcessId);
    }
  }
 
  WTSFreeMemory(pProcessInfo);
 
  return rc;
}