|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
#include <winpr/crt.h>
|
|
Packit |
1fb8d4 |
#include <winpr/error.h>
|
|
Packit |
1fb8d4 |
#include <winpr/wtsapi.h>
|
|
Packit |
1fb8d4 |
#include <winpr/environment.h>
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
int TestWtsApiQuerySessionInformation(int argc, char* argv[])
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
DWORD index, i;
|
|
Packit |
1fb8d4 |
DWORD count;
|
|
Packit |
1fb8d4 |
BOOL bSuccess;
|
|
Packit |
1fb8d4 |
HANDLE hServer;
|
|
Packit |
1fb8d4 |
LPSTR pBuffer;
|
|
Packit |
1fb8d4 |
DWORD sessionId;
|
|
Packit |
1fb8d4 |
DWORD bytesReturned;
|
|
Packit |
1fb8d4 |
PWTS_SESSION_INFOA pSessionInfo;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
#ifndef _WIN32
|
|
Packit |
1fb8d4 |
if (!GetEnvironmentVariableA("WTSAPI_LIBRARY", NULL, 0))
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
printf("%s: No RDS environment detected, skipping test\n", __FUNCTION__);
|
|
Packit |
1fb8d4 |
return 0;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
#endif
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
hServer = WTS_CURRENT_SERVER_HANDLE;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
count = 0;
|
|
Packit |
1fb8d4 |
pSessionInfo = NULL;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
bSuccess = WTSEnumerateSessionsA(hServer, 0, 1, &pSessionInfo, &count);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!bSuccess)
|
|
Packit |
1fb8d4 |
{
|
|
Packit Service |
5a9772 |
printf("WTSEnumerateSessions failed: %" PRIu32 "\n", GetLastError());
|
|
Packit |
1fb8d4 |
return 0;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit Service |
5a9772 |
printf("WTSEnumerateSessions count: %" PRIu32 "\n", count);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
for (index = 0; index < count; index++)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
char* Username;
|
|
Packit |
1fb8d4 |
char* Domain;
|
|
Packit |
1fb8d4 |
char* ClientName;
|
|
Packit |
1fb8d4 |
ULONG ClientBuildNumber;
|
|
Packit |
1fb8d4 |
USHORT ClientProductId;
|
|
Packit |
1fb8d4 |
ULONG ClientHardwareId;
|
|
Packit |
1fb8d4 |
USHORT ClientProtocolType;
|
|
Packit |
1fb8d4 |
PWTS_CLIENT_DISPLAY ClientDisplay;
|
|
Packit |
1fb8d4 |
PWTS_CLIENT_ADDRESS ClientAddress;
|
|
Packit |
1fb8d4 |
WTS_CONNECTSTATE_CLASS ConnectState;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
pBuffer = NULL;
|
|
Packit |
1fb8d4 |
bytesReturned = 0;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
sessionId = pSessionInfo[index].SessionId;
|
|
Packit |
1fb8d4 |
|
|
Packit Service |
5a9772 |
printf("[%" PRIu32 "] SessionId: %" PRIu32 " State: %s (%u) WinstationName: '%s'\n", index,
|
|
Packit Service |
5a9772 |
pSessionInfo[index].SessionId, WTSSessionStateToString(pSessionInfo[index].State),
|
|
Packit Service |
5a9772 |
pSessionInfo[index].State, pSessionInfo[index].pWinStationName);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
/* WTSUserName */
|
|
Packit |
1fb8d4 |
|
|
Packit Service |
5a9772 |
bSuccess =
|
|
Packit Service |
5a9772 |
WTSQuerySessionInformationA(hServer, sessionId, WTSUserName, &pBuffer, &bytesReturned);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!bSuccess)
|
|
Packit |
1fb8d4 |
{
|
|
Packit Service |
5a9772 |
printf("WTSQuerySessionInformation WTSUserName failed: %" PRIu32 "\n", GetLastError());
|
|
Packit |
1fb8d4 |
return -1;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit Service |
5a9772 |
Username = (char*)pBuffer;
|
|
Packit |
1fb8d4 |
printf("\tWTSUserName: '%s'\n", Username);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
/* WTSDomainName */
|
|
Packit |
1fb8d4 |
|
|
Packit Service |
5a9772 |
bSuccess = WTSQuerySessionInformationA(hServer, sessionId, WTSDomainName, &pBuffer,
|
|
Packit Service |
5a9772 |
&bytesReturned);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!bSuccess)
|
|
Packit |
1fb8d4 |
{
|
|
Packit Service |
5a9772 |
printf("WTSQuerySessionInformation WTSDomainName failed: %" PRIu32 "\n",
|
|
Packit Service |
5a9772 |
GetLastError());
|
|
Packit |
1fb8d4 |
return -1;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit Service |
5a9772 |
Domain = (char*)pBuffer;
|
|
Packit |
1fb8d4 |
printf("\tWTSDomainName: '%s'\n", Domain);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
/* WTSConnectState */
|
|
Packit |
1fb8d4 |
|
|
Packit Service |
5a9772 |
bSuccess = WTSQuerySessionInformationA(hServer, sessionId, WTSConnectState, &pBuffer,
|
|
Packit Service |
5a9772 |
&bytesReturned);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!bSuccess)
|
|
Packit |
1fb8d4 |
{
|
|
Packit Service |
5a9772 |
printf("WTSQuerySessionInformation WTSConnectState failed: %" PRIu32 "\n",
|
|
Packit Service |
5a9772 |
GetLastError());
|
|
Packit |
1fb8d4 |
return -1;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit Service |
5a9772 |
ConnectState = *((WTS_CONNECTSTATE_CLASS*)pBuffer);
|
|
Packit |
1fb8d4 |
printf("\tWTSConnectState: %u (%s)\n", ConnectState, WTSSessionStateToString(ConnectState));
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
/* WTSClientBuildNumber */
|
|
Packit |
1fb8d4 |
|
|
Packit Service |
5a9772 |
bSuccess = WTSQuerySessionInformationA(hServer, sessionId, WTSClientBuildNumber, &pBuffer,
|
|
Packit Service |
5a9772 |
&bytesReturned);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!bSuccess)
|
|
Packit |
1fb8d4 |
{
|
|
Packit Service |
5a9772 |
printf("WTSQuerySessionInformation WTSClientBuildNumber failed: %" PRIu32 "\n",
|
|
Packit Service |
5a9772 |
GetLastError());
|
|
Packit |
1fb8d4 |
return -1;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit Service |
5a9772 |
ClientBuildNumber = *((ULONG*)pBuffer);
|
|
Packit Service |
5a9772 |
printf("\tWTSClientBuildNumber: %" PRIu32 "\n", ClientBuildNumber);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
/* WTSClientName */
|
|
Packit |
1fb8d4 |
|
|
Packit Service |
5a9772 |
bSuccess = WTSQuerySessionInformationA(hServer, sessionId, WTSClientName, &pBuffer,
|
|
Packit Service |
5a9772 |
&bytesReturned);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!bSuccess)
|
|
Packit |
1fb8d4 |
{
|
|
Packit Service |
5a9772 |
printf("WTSQuerySessionInformation WTSClientName failed: %" PRIu32 "\n",
|
|
Packit Service |
5a9772 |
GetLastError());
|
|
Packit |
1fb8d4 |
return -1;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit Service |
5a9772 |
ClientName = (char*)pBuffer;
|
|
Packit |
1fb8d4 |
printf("\tWTSClientName: '%s'\n", ClientName);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
/* WTSClientProductId */
|
|
Packit |
1fb8d4 |
|
|
Packit Service |
5a9772 |
bSuccess = WTSQuerySessionInformationA(hServer, sessionId, WTSClientProductId, &pBuffer,
|
|
Packit Service |
5a9772 |
&bytesReturned);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!bSuccess)
|
|
Packit |
1fb8d4 |
{
|
|
Packit Service |
5a9772 |
printf("WTSQuerySessionInformation WTSClientProductId failed: %" PRIu32 "\n",
|
|
Packit Service |
5a9772 |
GetLastError());
|
|
Packit |
1fb8d4 |
return -1;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit Service |
5a9772 |
ClientProductId = *((USHORT*)pBuffer);
|
|
Packit Service |
5a9772 |
printf("\tWTSClientProductId: %" PRIu16 "\n", ClientProductId);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
/* WTSClientHardwareId */
|
|
Packit |
1fb8d4 |
|
|
Packit Service |
5a9772 |
bSuccess = WTSQuerySessionInformationA(hServer, sessionId, WTSClientHardwareId, &pBuffer,
|
|
Packit Service |
5a9772 |
&bytesReturned);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!bSuccess)
|
|
Packit |
1fb8d4 |
{
|
|
Packit Service |
5a9772 |
printf("WTSQuerySessionInformation WTSClientHardwareId failed: %" PRIu32 "\n",
|
|
Packit Service |
5a9772 |
GetLastError());
|
|
Packit |
1fb8d4 |
return -1;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit Service |
5a9772 |
ClientHardwareId = *((ULONG*)pBuffer);
|
|
Packit Service |
5a9772 |
printf("\tWTSClientHardwareId: %" PRIu32 "\n", ClientHardwareId);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
/* WTSClientAddress */
|
|
Packit |
1fb8d4 |
|
|
Packit Service |
5a9772 |
bSuccess = WTSQuerySessionInformationA(hServer, sessionId, WTSClientAddress, &pBuffer,
|
|
Packit Service |
5a9772 |
&bytesReturned);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!bSuccess)
|
|
Packit |
1fb8d4 |
{
|
|
Packit Service |
5a9772 |
printf("WTSQuerySessionInformation WTSClientAddress failed: %" PRIu32 "\n",
|
|
Packit Service |
5a9772 |
GetLastError());
|
|
Packit |
1fb8d4 |
return -1;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit Service |
5a9772 |
ClientAddress = (PWTS_CLIENT_ADDRESS)pBuffer;
|
|
Packit Service |
5a9772 |
printf("\tWTSClientAddress: AddressFamily: %" PRIu32 " Address: ",
|
|
Packit Service |
5a9772 |
ClientAddress->AddressFamily);
|
|
Packit |
1fb8d4 |
for (i = 0; i < sizeof(ClientAddress->Address); i++)
|
|
Packit Service |
5a9772 |
printf("%02" PRIX8 "", ClientAddress->Address[i]);
|
|
Packit |
1fb8d4 |
printf("\n");
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
/* WTSClientDisplay */
|
|
Packit |
1fb8d4 |
|
|
Packit Service |
5a9772 |
bSuccess = WTSQuerySessionInformationA(hServer, sessionId, WTSClientDisplay, &pBuffer,
|
|
Packit Service |
5a9772 |
&bytesReturned);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!bSuccess)
|
|
Packit |
1fb8d4 |
{
|
|
Packit Service |
5a9772 |
printf("WTSQuerySessionInformation WTSClientDisplay failed: %" PRIu32 "\n",
|
|
Packit Service |
5a9772 |
GetLastError());
|
|
Packit |
1fb8d4 |
return -1;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit Service |
5a9772 |
ClientDisplay = (PWTS_CLIENT_DISPLAY)pBuffer;
|
|
Packit Service |
5a9772 |
printf("\tWTSClientDisplay: HorizontalResolution: %" PRIu32 " VerticalResolution: %" PRIu32
|
|
Packit Service |
5a9772 |
" ColorDepth: %" PRIu32 "\n",
|
|
Packit Service |
5a9772 |
ClientDisplay->HorizontalResolution, ClientDisplay->VerticalResolution,
|
|
Packit Service |
5a9772 |
ClientDisplay->ColorDepth);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
/* WTSClientProtocolType */
|
|
Packit |
1fb8d4 |
|
|
Packit Service |
5a9772 |
bSuccess = WTSQuerySessionInformationA(hServer, sessionId, WTSClientProtocolType, &pBuffer,
|
|
Packit Service |
5a9772 |
&bytesReturned);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!bSuccess)
|
|
Packit |
1fb8d4 |
{
|
|
Packit Service |
5a9772 |
printf("WTSQuerySessionInformation WTSClientProtocolType failed: %" PRIu32 "\n",
|
|
Packit Service |
5a9772 |
GetLastError());
|
|
Packit |
1fb8d4 |
return -1;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit Service |
5a9772 |
ClientProtocolType = *((USHORT*)pBuffer);
|
|
Packit Service |
5a9772 |
printf("\tWTSClientProtocolType: %" PRIu16 "\n", ClientProtocolType);
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
WTSFreeMemory(pSessionInfo);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
return 0;
|
|
Packit |
1fb8d4 |
}
|