278 lines
7.5 KiB
C++
278 lines
7.5 KiB
C++
|
#include "nt.h"
|
||
|
|
#include "ntrtl.h"
|
||
|
|
#include "nturtl.h"
|
||
|
|
#include "stdafx.h"
|
||
|
|
#include <windows.h>
|
||
|
|
#include <winsta.h>
|
||
|
|
#include <rdpsndp.h>
|
||
|
|
#include <rdpstrm.h>
|
||
|
|
#include <stdio.h>
|
||
|
|
#include <strstrea.h>
|
||
|
|
|
||
|
|
extern ostrstream szMoreInfo;
|
||
|
|
BOOL
|
||
|
|
IsTSRedirectingAudio( VOID )
|
||
|
|
{
|
||
|
|
BOOL rv;
|
||
|
|
static BOOL bChecked = FALSE;
|
||
|
|
static BOOL RemoteConsoleAudio = 0; // Allow audio play at
|
||
|
|
// the console
|
||
|
|
// when console session is
|
||
|
|
// remoted
|
||
|
|
// is not consle ?
|
||
|
|
//
|
||
|
|
rv = !(USER_SHARED_DATA->ActiveConsoleId ==
|
||
|
|
NtCurrentPeb()->SessionId);
|
||
|
|
|
||
|
|
// console, we don't redirect
|
||
|
|
if ( !rv )
|
||
|
|
{
|
||
|
|
szMoreInfo << "Running on the console" << endl;
|
||
|
|
goto exitpt;
|
||
|
|
} else
|
||
|
|
szMoreInfo << "Running in a session" << endl;
|
||
|
|
|
||
|
|
// check if audio is redirected on PTS
|
||
|
|
//
|
||
|
|
if ( !bChecked )
|
||
|
|
{
|
||
|
|
WINSTATIONCLIENT ClientData;
|
||
|
|
ULONG Length;
|
||
|
|
|
||
|
|
if (WinStationQueryInformation(SERVERNAME_CURRENT,
|
||
|
|
LOGONID_CURRENT,
|
||
|
|
WinStationClient,
|
||
|
|
&ClientData,
|
||
|
|
sizeof(ClientData),
|
||
|
|
&Length ))
|
||
|
|
{
|
||
|
|
RemoteConsoleAudio = ClientData.fRemoteConsoleAudio;
|
||
|
|
}
|
||
|
|
else {
|
||
|
|
szMoreInfo << "WinStatinQueryInformation failed=" <<
|
||
|
|
GetLastError() << endl;
|
||
|
|
RemoteConsoleAudio = 0;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
rv = !RemoteConsoleAudio;
|
||
|
|
|
||
|
|
exitpt:
|
||
|
|
return rv;
|
||
|
|
}
|
||
|
|
|
||
|
|
BOOL
|
||
|
|
IsTSAudioDriverEnabled( VOID )
|
||
|
|
{
|
||
|
|
BOOL rv = FALSE;
|
||
|
|
WINSTATIONCONFIG config;
|
||
|
|
BOOLEAN fSuccess;
|
||
|
|
DWORD returnedLength;
|
||
|
|
|
||
|
|
fSuccess = WinStationQueryInformation(NULL, LOGONID_CURRENT,
|
||
|
|
WinStationConfiguration, &config,
|
||
|
|
sizeof(config), &returnedLength);
|
||
|
|
if (!fSuccess)
|
||
|
|
{
|
||
|
|
szMoreInfo << "WinStatinQueryInformation failed=" <<
|
||
|
|
GetLastError() <<endl;
|
||
|
|
goto exitpt;
|
||
|
|
}
|
||
|
|
|
||
|
|
rv = !config.User.fDisableCam;
|
||
|
|
|
||
|
|
exitpt:
|
||
|
|
return rv;
|
||
|
|
}
|
||
|
|
|
||
|
|
BOOL
|
||
|
|
DumpStreamInfo( VOID )
|
||
|
|
{
|
||
|
|
BOOL rv = TRUE;
|
||
|
|
HANDLE hEv = NULL;
|
||
|
|
HANDLE hStream = NULL;
|
||
|
|
PSNDSTREAM Stream = NULL;
|
||
|
|
DWORD dw;
|
||
|
|
|
||
|
|
hEv = OpenEvent(EVENT_ALL_ACCESS,
|
||
|
|
FALSE,
|
||
|
|
TSSND_DATAREADYEVENT);
|
||
|
|
|
||
|
|
if ( NULL == hEv )
|
||
|
|
{
|
||
|
|
szMoreInfo << "ERROR: Can't open DataReady event=" <<
|
||
|
|
GetLastError() <<endl;
|
||
|
|
rv = FALSE;
|
||
|
|
} else {
|
||
|
|
szMoreInfo << "OK: DataReady event exist" << endl;
|
||
|
|
dw = WaitForSingleObject( hEv, 0 );
|
||
|
|
if ( WAIT_OBJECT_0 == dw )
|
||
|
|
{
|
||
|
|
szMoreInfo << "WARNING: DataReady event is signaled (playing sound ?)" <<endl;
|
||
|
|
SetEvent( hEv );
|
||
|
|
rv = FALSE;
|
||
|
|
} else
|
||
|
|
szMoreInfo << "OK: DataReady event is not signaled" <<endl;
|
||
|
|
CloseHandle( hEv );
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
hEv = OpenEvent(EVENT_ALL_ACCESS,
|
||
|
|
FALSE,
|
||
|
|
TSSND_STREAMISEMPTYEVENT);
|
||
|
|
if ( NULL == hEv )
|
||
|
|
{
|
||
|
|
szMoreInfo << "ERROR: Can't open StreamEmpty event=" <<
|
||
|
|
GetLastError() <<endl;
|
||
|
|
rv = FALSE;
|
||
|
|
} else {
|
||
|
|
szMoreInfo << "OK: StreamEmpty event exist" <<endl;
|
||
|
|
dw = WaitForSingleObject( hEv, 0 );
|
||
|
|
if ( WAIT_OBJECT_0 == dw )
|
||
|
|
{
|
||
|
|
szMoreInfo << "WARNING: StreamEmpty is signaled (playing sound ?)" <<endl;
|
||
|
|
rv = FALSE;
|
||
|
|
} else {
|
||
|
|
szMoreInfo << "OK: StreamEmpty event is not signaled" <<endl;
|
||
|
|
SetEvent( hEv );
|
||
|
|
}
|
||
|
|
CloseHandle( hEv );
|
||
|
|
}
|
||
|
|
|
||
|
|
hEv = OpenEvent(EVENT_ALL_ACCESS,
|
||
|
|
FALSE,
|
||
|
|
TSSND_WAITTOINIT);
|
||
|
|
if ( NULL == hEv )
|
||
|
|
{
|
||
|
|
szMoreInfo << "ERROR: Can't open WaitToInit event=" <<
|
||
|
|
GetLastError() <<endl;
|
||
|
|
rv = FALSE;
|
||
|
|
}
|
||
|
|
else {
|
||
|
|
szMoreInfo << "OK: WaitToInit event exist" <<endl;
|
||
|
|
dw = WaitForSingleObject( hEv, 0 );
|
||
|
|
if ( WAIT_OBJECT_0 == dw )
|
||
|
|
{
|
||
|
|
szMoreInfo << "OK: WaitToInit is signaled" <<endl;
|
||
|
|
} else {
|
||
|
|
szMoreInfo << "WARNING: WaitToInit is NOT signaled" <<endl;
|
||
|
|
rv = FALSE;
|
||
|
|
}
|
||
|
|
CloseHandle( hEv );
|
||
|
|
}
|
||
|
|
|
||
|
|
hEv = OpenMutex(SYNCHRONIZE,
|
||
|
|
FALSE,
|
||
|
|
TSSND_STREAMMUTEX);
|
||
|
|
if ( NULL == hEv )
|
||
|
|
{
|
||
|
|
szMoreInfo << "ERROR: Can't open Stream mutex=" <<
|
||
|
|
GetLastError() <<endl;
|
||
|
|
rv = FALSE;
|
||
|
|
} else {
|
||
|
|
szMoreInfo << "OK: Stream mutex exist" <<endl;
|
||
|
|
dw = WaitForSingleObject( hEv, 3000 );
|
||
|
|
if ( WAIT_OBJECT_0 != dw )
|
||
|
|
{
|
||
|
|
szMoreInfo << "WARNING: Can't acquire Stream mutext after more than 3 seconds" <<endl;
|
||
|
|
rv = FALSE;
|
||
|
|
} else {
|
||
|
|
szMoreInfo << "OK: Stream mutext acquired" <<endl;
|
||
|
|
ReleaseMutex( hEv );
|
||
|
|
}
|
||
|
|
CloseHandle( hEv );
|
||
|
|
}
|
||
|
|
|
||
|
|
hStream = OpenFileMapping(
|
||
|
|
FILE_MAP_ALL_ACCESS,
|
||
|
|
FALSE,
|
||
|
|
TSSND_STREAMNAME
|
||
|
|
);
|
||
|
|
|
||
|
|
if (NULL == hStream)
|
||
|
|
{
|
||
|
|
szMoreInfo << "ERROR: Can't open the stream mapping" <<
|
||
|
|
GetLastError() <<endl;
|
||
|
|
rv = FALSE;
|
||
|
|
goto exitpt;
|
||
|
|
}
|
||
|
|
|
||
|
|
Stream = (PSNDSTREAM)MapViewOfFile(
|
||
|
|
hStream,
|
||
|
|
FILE_MAP_ALL_ACCESS,
|
||
|
|
0, 0, // offset
|
||
|
|
sizeof(*Stream)
|
||
|
|
);
|
||
|
|
|
||
|
|
if (NULL == Stream)
|
||
|
|
{
|
||
|
|
szMoreInfo << "ERROR: Can't map the stream view: " <<
|
||
|
|
GetLastError() <<endl;
|
||
|
|
rv = FALSE;
|
||
|
|
goto exitpt;
|
||
|
|
}
|
||
|
|
|
||
|
|
szMoreInfo << "SNDSTREAM:" <<endl;
|
||
|
|
szMoreInfo << "bNewVolume: 0x" << hex << Stream->bNewVolume <<endl;
|
||
|
|
szMoreInfo << "bNewPitch: 0x" << hex << Stream->bNewPitch <<endl;
|
||
|
|
|
||
|
|
szMoreInfo << "dwSoundCaps: 0x" << hex << Stream->dwSoundCaps <<endl;
|
||
|
|
if ( 0 != (TSSNDCAPS_ALIVE & Stream->dwSoundCaps) )
|
||
|
|
szMoreInfo << "\tTSSNDCAPS_ALIVE" <<endl;
|
||
|
|
if ( 0 != (TSSNDCAPS_VOLUME & Stream->dwSoundCaps) )
|
||
|
|
szMoreInfo << "\tTSSNDCAPS_VOLUME" << endl;
|
||
|
|
if ( 0 != (TSSNDCAPS_PITCH & Stream->dwSoundCaps) )
|
||
|
|
szMoreInfo << "\tTSSNDCAPS_PITCH" <<endl;
|
||
|
|
|
||
|
|
szMoreInfo << "dwVolume: 0x" << hex << Stream->dwVolume <<endl;
|
||
|
|
szMoreInfo << "dwPitch: 0x" << hex << Stream->dwPitch <<endl;
|
||
|
|
|
||
|
|
dw = Stream->cLastBlockQueued;
|
||
|
|
szMoreInfo << "cLastBlockQueued: 0x" << hex << dw <<endl;
|
||
|
|
dw = Stream->cLastBlockSent;
|
||
|
|
szMoreInfo << "cLastBlockSent: 0x" << hex << dw <<endl;
|
||
|
|
dw = Stream->cLastBlockConfirmed;
|
||
|
|
szMoreInfo << "cLastBlockConfirmed: 0x" << hex << dw <<endl;
|
||
|
|
|
||
|
|
exitpt:
|
||
|
|
if ( rv )
|
||
|
|
szMoreInfo << "OK: Stream seems good" <<endl;
|
||
|
|
else
|
||
|
|
szMoreInfo << "ERROR: Stream doesn't seem very healthy" <<endl;
|
||
|
|
|
||
|
|
if ( NULL != Stream )
|
||
|
|
UnmapViewOfFile( Stream );
|
||
|
|
|
||
|
|
if ( NULL != hStream )
|
||
|
|
CloseHandle( hStream );
|
||
|
|
|
||
|
|
return rv;
|
||
|
|
}
|
||
|
|
|
||
|
|
BOOL
|
||
|
|
IsAudioOk(
|
||
|
|
VOID
|
||
|
|
)
|
||
|
|
{
|
||
|
|
INT rv = TRUE;
|
||
|
|
|
||
|
|
if ( !IsTSRedirectingAudio() )
|
||
|
|
{
|
||
|
|
szMoreInfo << "Audio is not redirected" <<endl;
|
||
|
|
goto exitpt;
|
||
|
|
}
|
||
|
|
|
||
|
|
if ( !IsTSAudioDriverEnabled() )
|
||
|
|
{
|
||
|
|
szMoreInfo << "TS Auidio driver is disabled" <<endl;
|
||
|
|
goto exitpt;
|
||
|
|
}
|
||
|
|
|
||
|
|
szMoreInfo << "Audio redirection is enabled. Dumping stream info" <<endl;
|
||
|
|
|
||
|
|
rv = DumpStreamInfo();
|
||
|
|
|
||
|
|
exitpt:
|
||
|
|
return rv;
|
||
|
|
}
|