int wts_send_message(int mode, const char * title, const char * text,
int * perrcnt /* = 0 */)
{
if (perrcnt)
*perrcnt = 0;
// Get session list
WTS_SESSION_INFOA * sessions; DWORD count;
if (!WTSEnumerateSessionsA(WTS_CURRENT_SERVER_HANDLE, 0, 1, &sessions, &count))
return -1;
int msgcnt = 0;
for (unsigned i = 0; i < count; i++) {
if ( !strcmpi(sessions[i].pWinStationName, "Console")
|| (mode >= 1 && sessions[i].State == WTSActive)
|| (mode >= 2 && sessions[i].State == WTSConnected)) {
// Send Message, don't wait for OK button
DWORD result;
if (WTSSendMessageA(WTS_CURRENT_SERVER_HANDLE, sessions[i].SessionId,
const_cast<char *>(title), strlen(title),
const_cast<char *>(text), strlen(text),
MB_OK|MB_ICONEXCLAMATION, 0 /*Timeout*/,
&result, FALSE /*Wait*/))
msgcnt++;
else if (perrcnt)
(*perrcnt)++;
}
}
WTSFreeMemory(sessions);
return msgcnt;
}
int TestWtsApiQuerySessionInformation(int argc, char* argv[])
{
DWORD index, i;
DWORD count;
BOOL bSuccess;
HANDLE hServer;
LPSTR pBuffer;
DWORD sessionId;
DWORD bytesReturned;
PWTS_SESSION_INFOA pSessionInfo;
#ifndef _WIN32
if (!GetEnvironmentVariableA("WTSAPI_LIBRARY", NULL, 0))
{
printf("%s: No RDS environment detected, skipping test\n", __FUNCTION__);
return 0;
}
#endif
hServer = WTS_CURRENT_SERVER_HANDLE;
count = 0;
pSessionInfo = NULL;
bSuccess = WTSEnumerateSessionsA(hServer, 0, 1, &pSessionInfo, &count);
if (!bSuccess)
{
printf("WTSEnumerateSessions failed: %d\n", (int) GetLastError());
return 0;
}
printf("WTSEnumerateSessions count: %d\n", (int) count);
for (index = 0; index < count; index++)
{
char* Username;
char* Domain;
char* ClientName;
ULONG ClientBuildNumber;
USHORT ClientProductId;
ULONG ClientHardwareId;
USHORT ClientProtocolType;
PWTS_CLIENT_DISPLAY ClientDisplay;
PWTS_CLIENT_ADDRESS ClientAddress;
WTS_CONNECTSTATE_CLASS ConnectState;
pBuffer = NULL;
bytesReturned = 0;
sessionId = pSessionInfo[index].SessionId;
printf("[%u] SessionId: %u State: %s (%u) WinstationName: '%s'\n",
index,
pSessionInfo[index].SessionId,
WTSSessionStateToString(pSessionInfo[index].State),
pSessionInfo[index].State,
pSessionInfo[index].pWinStationName);
/* WTSUserName */
bSuccess = WTSQuerySessionInformationA(hServer, sessionId, WTSUserName, &pBuffer, &bytesReturned);
if (!bSuccess)
{
printf("WTSQuerySessionInformation WTSUserName failed: %d\n", (int) GetLastError());
return -1;
}
Username = (char*) pBuffer;
printf("\tWTSUserName: '******'\n", Username);
/* WTSDomainName */
bSuccess = WTSQuerySessionInformationA(hServer, sessionId, WTSDomainName, &pBuffer, &bytesReturned);
if (!bSuccess)
{
printf("WTSQuerySessionInformation WTSDomainName failed: %d\n", (int) GetLastError());
return -1;
}
Domain = (char*) pBuffer;
printf("\tWTSDomainName: '%s'\n", Domain);
/* WTSConnectState */
bSuccess = WTSQuerySessionInformationA(hServer, sessionId, WTSConnectState, &pBuffer, &bytesReturned);
if (!bSuccess)
{
printf("WTSQuerySessionInformation WTSConnectState failed: %d\n", (int) GetLastError());
return -1;
}
ConnectState = *((WTS_CONNECTSTATE_CLASS*) pBuffer);
printf("\tWTSConnectState: %u (%s)\n", ConnectState, WTSSessionStateToString(ConnectState));
/* WTSClientBuildNumber */
bSuccess = WTSQuerySessionInformationA(hServer, sessionId, WTSClientBuildNumber, &pBuffer, &bytesReturned);
if (!bSuccess)
{
printf("WTSQuerySessionInformation WTSClientBuildNumber failed: %d\n", (int) GetLastError());
return -1;
}
ClientBuildNumber = *((ULONG*) pBuffer);
printf("\tWTSClientBuildNumber: %u\n", ClientBuildNumber);
/* WTSClientName */
bSuccess = WTSQuerySessionInformationA(hServer, sessionId, WTSClientName, &pBuffer, &bytesReturned);
if (!bSuccess)
{
printf("WTSQuerySessionInformation WTSClientName failed: %d\n", (int) GetLastError());
return -1;
}
ClientName = (char*) pBuffer;
printf("\tWTSClientName: '%s'\n", ClientName);
/* WTSClientProductId */
bSuccess = WTSQuerySessionInformationA(hServer, sessionId, WTSClientProductId, &pBuffer, &bytesReturned);
if (!bSuccess)
{
printf("WTSQuerySessionInformation WTSClientProductId failed: %d\n", (int) GetLastError());
return -1;
}
ClientProductId = *((USHORT*) pBuffer);
printf("\tWTSClientProductId: %u\n", ClientProductId);
/* WTSClientHardwareId */
bSuccess = WTSQuerySessionInformationA(hServer, sessionId, WTSClientHardwareId, &pBuffer, &bytesReturned);
if (!bSuccess)
{
printf("WTSQuerySessionInformation WTSClientHardwareId failed: %d\n", (int) GetLastError());
return -1;
}
ClientHardwareId = *((ULONG*) pBuffer);
printf("\tWTSClientHardwareId: %u\n", ClientHardwareId);
/* WTSClientAddress */
bSuccess = WTSQuerySessionInformationA(hServer, sessionId, WTSClientAddress, &pBuffer, &bytesReturned);
if (!bSuccess)
{
printf("WTSQuerySessionInformation WTSClientAddress failed: %d\n", (int) GetLastError());
return -1;
}
ClientAddress = (PWTS_CLIENT_ADDRESS) pBuffer;
printf("\tWTSClientAddress: AddressFamily: %u Address: ",
ClientAddress->AddressFamily);
for (i = 0; i < sizeof(ClientAddress->Address); i++)
printf("%02X", ClientAddress->Address[i]);
printf("\n");
/* WTSClientDisplay */
bSuccess = WTSQuerySessionInformationA(hServer, sessionId, WTSClientDisplay, &pBuffer, &bytesReturned);
if (!bSuccess)
{
printf("WTSQuerySessionInformation WTSClientDisplay failed: %d\n", (int) GetLastError());
return -1;
}
ClientDisplay = (PWTS_CLIENT_DISPLAY) pBuffer;
printf("\tWTSClientDisplay: HorizontalResolution: %u VerticalResolution: %u ColorDepth: %u\n",
ClientDisplay->HorizontalResolution, ClientDisplay->VerticalResolution,
ClientDisplay->ColorDepth);
/* WTSClientProtocolType */
bSuccess = WTSQuerySessionInformationA(hServer, sessionId, WTSClientProtocolType, &pBuffer, &bytesReturned);
if (!bSuccess)
{
printf("WTSQuerySessionInformation WTSClientProtocolType failed: %d\n", (int) GetLastError());
return -1;
}
ClientProtocolType = *((USHORT*) pBuffer);
printf("\tWTSClientProtocolType: %u\n", ClientProtocolType);
}
WTSFreeMemory(pSessionInfo);
return 0;
}
int main(int argc, const char **argv)
{
int mode = 0, verbose = 0, status = 0, i;
const char * message = 0, * caption = "";
char msgbuf[1024];
WTS_SESSION_INFOA * sessions; DWORD count;
for (i = 1; i < argc && argv[i][0] == '-' && argv[i][1]; i++) {
int j;
for (j = 1; argv[i][j]; j++)
switch (argv[i][j]) {
case 'c': mode = 0; break;
case 'a': mode = 1; break;
case 's': mode = 2; break;
case 'v': verbose = 1; break;
default: return usage();
}
}
if (i < argc) {
if (i+1 < argc)
caption = argv[i++];
message = argv[i++];
if (i < argc)
return usage();
if (!strcmp(message, "-")) {
// Read message from stdin
i = fread(msgbuf, 1, sizeof(msgbuf)-1, stdin);
if (i < 0) {
perror("stdin");
return 1;
}
msgbuf[i] = 0;
message = msgbuf;
}
}
else {
if (!verbose)
return usage();
}
// Get session list
if (!WTSEnumerateSessionsA(WTS_CURRENT_SERVER_HANDLE, 0, 1, &sessions, &count)) {
fprintf(stderr, "WTSEnumerateSessions() failed\n");
return 1;
}
for (i = 0; i < (int)count; i++) {
if (verbose) {
printf("Session %d (\"%s\", State=%d)%s",
i, sessions[i].pWinStationName, sessions[i].State,
(!message ? "\n" : ": "));
if (!message)
continue; // List sessions only
fflush(stdout);
}
if ( !strcmpi(sessions[i].pWinStationName, "Console")
|| (mode >= 1 && sessions[i].State == WTSActive)
|| (mode >= 2 && sessions[i].State == WTSConnected)) {
// Send Message, don't wait for OK button
DWORD result;
if (WTSSendMessageA(WTS_CURRENT_SERVER_HANDLE, sessions[i].SessionId,
(char *)caption, strlen(caption),
(char *)message, strlen(message),
MB_OK|MB_ICONEXCLAMATION, 0 /*Timeout*/,
&result, FALSE /*!Wait*/)) {
if (verbose)
printf("message sent\n");
}
else {
status = 1;
if (verbose)
printf("WTSSendMessage() failed with error=%d\n", (int)GetLastError());
else
fprintf(stderr, "Session %d (\"%s\", State=%d): WTSSendMessage() failed with error=%d\n",
i, sessions[i].pWinStationName, sessions[i].State, (int)GetLastError());
}
}
else {
if (verbose)
printf("ignored\n");
}
}
WTSFreeMemory(sessions);
return status;
}
