572 lines
12 KiB
C++
572 lines
12 KiB
C++
|
/*++
|
||
|
|
|
||
|
|
Copyright (c) 1999-2000 Microsoft Corporation
|
||
|
|
|
||
|
|
Module Name:
|
||
|
|
|
||
|
|
RDPRemoteDesktopSession
|
||
|
|
|
||
|
|
Abstract:
|
||
|
|
|
||
|
|
The CRemoteDesktopSession class is the parent
|
||
|
|
class for the Remote Desktop class hierarchy on the server-side.
|
||
|
|
It helps the CRemoteDesktopServerHost class to implement
|
||
|
|
the ISAFRemoteDesktopSession interface.
|
||
|
|
|
||
|
|
The Remote Desktop class hierarchy provides a pluggable C++ interface
|
||
|
|
for remote desktop access, by abstracting the implementation
|
||
|
|
specific details of remote desktop access for the server-side.
|
||
|
|
|
||
|
|
Author:
|
||
|
|
|
||
|
|
Tad Brockway 02/00
|
||
|
|
|
||
|
|
Revision History:
|
||
|
|
|
||
|
|
--*/
|
||
|
|
|
||
|
|
#include <RemoteDesktop.h>
|
||
|
|
#include "stdafx.h"
|
||
|
|
|
||
|
|
#ifdef TRC_FILE
|
||
|
|
#undef TRC_FILE
|
||
|
|
#endif
|
||
|
|
|
||
|
|
#define TRC_FILE "_srdses"
|
||
|
|
|
||
|
|
#include "RDSHost.h"
|
||
|
|
#include "RemoteDesktopSession.h"
|
||
|
|
#include "RemoteDesktopServerHost.h"
|
||
|
|
#include <sessmgr_i.c>
|
||
|
|
#include <objidl.h>
|
||
|
|
#include <objbase.h>
|
||
|
|
|
||
|
|
|
||
|
|
///////////////////////////////////////////////////////
|
||
|
|
//
|
||
|
|
// CRemoteDesktopSession Methods
|
||
|
|
//
|
||
|
|
|
||
|
|
HRESULT
|
||
|
|
CRemoteDesktopSession::FinalConstruct()
|
||
|
|
/*++
|
||
|
|
|
||
|
|
Routine Description:
|
||
|
|
|
||
|
|
Final Constructor
|
||
|
|
|
||
|
|
Arguments:
|
||
|
|
|
||
|
|
Return Value:
|
||
|
|
|
||
|
|
S_OK on success. Otherwise, an error code is returned.
|
||
|
|
|
||
|
|
--*/
|
||
|
|
{
|
||
|
|
DC_BEGIN_FN("CRemoteDesktopSession::FinalConstruct");
|
||
|
|
|
||
|
|
DC_END_FN();
|
||
|
|
|
||
|
|
return S_OK;
|
||
|
|
}
|
||
|
|
|
||
|
|
CRemoteDesktopSession::~CRemoteDesktopSession()
|
||
|
|
/*++
|
||
|
|
|
||
|
|
Routine Description:
|
||
|
|
|
||
|
|
Destructor
|
||
|
|
|
||
|
|
Arguments:
|
||
|
|
|
||
|
|
Return Value:
|
||
|
|
|
||
|
|
--*/
|
||
|
|
{
|
||
|
|
DC_BEGIN_FN("CRemoteDesktopSession::~CRemoteDesktopSession");
|
||
|
|
|
||
|
|
Shutdown();
|
||
|
|
|
||
|
|
//
|
||
|
|
// Release any lingering outgoing interfaces. Need to catch
|
||
|
|
// exceptions here in case the outgoing interface application
|
||
|
|
// has already gone away.
|
||
|
|
//
|
||
|
|
try {
|
||
|
|
if (m_OnConnected != NULL) {
|
||
|
|
m_OnConnected->Release();
|
||
|
|
}
|
||
|
|
if (m_OnDisconnected != NULL) {
|
||
|
|
m_OnDisconnected->Release();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
catch (...) {
|
||
|
|
TRC_ALT((TB, L"Exception caught in outgoing interface release."));
|
||
|
|
}
|
||
|
|
|
||
|
|
DC_END_FN();
|
||
|
|
}
|
||
|
|
|
||
|
|
HRESULT
|
||
|
|
CRemoteDesktopSession::Initialize(
|
||
|
|
BSTR connectParms,
|
||
|
|
CRemoteDesktopServerHost *hostObject,
|
||
|
|
REMOTE_DESKTOP_SHARING_CLASS sharingClass,
|
||
|
|
BOOL bEnableCallback,
|
||
|
|
DWORD timeOut,
|
||
|
|
BSTR userHelpCreateBlob,
|
||
|
|
LONG tsSessionID,
|
||
|
|
BSTR userSid
|
||
|
|
)
|
||
|
|
/*++
|
||
|
|
|
||
|
|
Routine Description:
|
||
|
|
|
||
|
|
The Initialize method prepares the COM object for connection by
|
||
|
|
the client-side Remote Desktop Host ActiveX Control.
|
||
|
|
|
||
|
|
Arguments:
|
||
|
|
|
||
|
|
connectParms - If parms are non-NULL, then the session already exists.
|
||
|
|
Otherwise, a new session should be created.
|
||
|
|
hostObject - Back pointer to containing RDS Host object.
|
||
|
|
sharingClass - Level of desktop sharing for a new session.
|
||
|
|
bEnableCallback - TRUE to instruct sessmgr to call session resolver, FALSE otherwise.
|
||
|
|
timeOut - Help session timeout value. 0, if not specified.
|
||
|
|
userHelpCreateBlob - user specified help session creation blob.
|
||
|
|
tsSessionID - Terminal Services Session ID or -1 if
|
||
|
|
undefined.
|
||
|
|
userSid - User SID or "" if undefined.
|
||
|
|
|
||
|
|
Return Value:
|
||
|
|
|
||
|
|
S_OK on success. Otherwise, an error code is returned.
|
||
|
|
|
||
|
|
--*/
|
||
|
|
{
|
||
|
|
DC_BEGIN_FN("CRemoteDesktopSession::Initialize");
|
||
|
|
|
||
|
|
HRESULT hr;
|
||
|
|
DWORD ret;
|
||
|
|
DWORD protocolType;
|
||
|
|
CComBSTR parmsMachineName;
|
||
|
|
CComBSTR parmsAssistantAccount;
|
||
|
|
CComBSTR parmsAssistantAccountPwd;
|
||
|
|
CComBSTR parmsHelpSessionName;
|
||
|
|
CComBSTR parmsHelpSessionPwd;
|
||
|
|
CComBSTR parmsProtocolSpecificParms;
|
||
|
|
CComBSTR helpSessionName;
|
||
|
|
CComBSTR sessionDescr;
|
||
|
|
DWORD dwVersion;
|
||
|
|
|
||
|
|
ASSERT(IsValid());
|
||
|
|
if (!IsValid()) {
|
||
|
|
return E_FAIL;
|
||
|
|
}
|
||
|
|
|
||
|
|
TRC_NRM((TB, L"***Ref count is: %ld", m_dwRef));
|
||
|
|
|
||
|
|
//
|
||
|
|
// Keep a back pointer to the RDS host object.
|
||
|
|
//
|
||
|
|
m_RDSHost = hostObject;
|
||
|
|
|
||
|
|
//
|
||
|
|
// Open an instance of the Remote Desktop Help Session Manager service.
|
||
|
|
//
|
||
|
|
ASSERT(m_HelpSessionManager == NULL);
|
||
|
|
hr = m_HelpSessionManager.CoCreateInstance(CLSID_RemoteDesktopHelpSessionMgr);
|
||
|
|
if (!SUCCEEDED(hr)) {
|
||
|
|
TRC_ERR((TB, TEXT("Can't create help session manager: %08X"), hr));
|
||
|
|
goto CLEANUPANDEXIT;
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// Set the security level to impersonate. This is required by
|
||
|
|
// the session manager.
|
||
|
|
//
|
||
|
|
hr = CoSetProxyBlanket(
|
||
|
|
(IUnknown *)m_HelpSessionManager,
|
||
|
|
RPC_C_AUTHN_DEFAULT,
|
||
|
|
RPC_C_AUTHZ_DEFAULT,
|
||
|
|
NULL,
|
||
|
|
RPC_C_AUTHN_LEVEL_DEFAULT,
|
||
|
|
RPC_C_IMP_LEVEL_IMPERSONATE,
|
||
|
|
NULL,
|
||
|
|
EOAC_NONE
|
||
|
|
);
|
||
|
|
if (!SUCCEEDED(hr)) {
|
||
|
|
TRC_ERR((TB, TEXT("CoSetProxyBlanket: %08X"), hr));
|
||
|
|
goto CLEANUPANDEXIT;
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// Create a new help session if we don't already have connection
|
||
|
|
// parms.
|
||
|
|
//
|
||
|
|
if (connectParms == NULL) {
|
||
|
|
TRC_NRM((TB, L"Creating new help session."));
|
||
|
|
GetSessionName(helpSessionName);
|
||
|
|
GetSessionDescription(sessionDescr);
|
||
|
|
|
||
|
|
hr = m_HelpSessionManager->CreateHelpSessionEx(
|
||
|
|
sharingClass,
|
||
|
|
bEnableCallback,
|
||
|
|
timeOut,
|
||
|
|
tsSessionID,
|
||
|
|
userSid,
|
||
|
|
userHelpCreateBlob,
|
||
|
|
&m_HelpSession
|
||
|
|
);
|
||
|
|
if (!SUCCEEDED(hr)) {
|
||
|
|
TRC_ERR((TB, L"CreateHelpSession: %08X", hr));
|
||
|
|
goto CLEANUPANDEXIT;
|
||
|
|
}
|
||
|
|
|
||
|
|
hr = m_HelpSession->get_HelpSessionId(&m_HelpSessionID);
|
||
|
|
if (!SUCCEEDED(hr)) {
|
||
|
|
TRC_ERR((TB, L"get_HelpSessionId: %08X", hr));
|
||
|
|
goto CLEANUPANDEXIT;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else {
|
||
|
|
|
||
|
|
//
|
||
|
|
// Parse the connection parms to get the help
|
||
|
|
// session ID.
|
||
|
|
//
|
||
|
|
ret = ParseConnectParmsString(
|
||
|
|
connectParms,
|
||
|
|
&dwVersion,
|
||
|
|
&protocolType,
|
||
|
|
parmsMachineName,
|
||
|
|
parmsAssistantAccount,
|
||
|
|
parmsAssistantAccountPwd,
|
||
|
|
m_HelpSessionID,
|
||
|
|
parmsHelpSessionName,
|
||
|
|
parmsHelpSessionPwd,
|
||
|
|
parmsProtocolSpecificParms
|
||
|
|
);
|
||
|
|
if (ret != ERROR_SUCCESS) {
|
||
|
|
hr = HRESULT_FROM_WIN32(ret);
|
||
|
|
goto CLEANUPANDEXIT;
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// Open the help session interface.
|
||
|
|
//
|
||
|
|
hr = m_HelpSessionManager->RetrieveHelpSession(
|
||
|
|
m_HelpSessionID,
|
||
|
|
&m_HelpSession
|
||
|
|
);
|
||
|
|
if (!SUCCEEDED(hr)) {
|
||
|
|
TRC_ERR((TB, L"Failed to open existing help session %s: %08X.",
|
||
|
|
m_HelpSessionID, hr));
|
||
|
|
goto CLEANUPANDEXIT;
|
||
|
|
}
|
||
|
|
else {
|
||
|
|
TRC_NRM((TB, L"Opened existing help session %s.", m_HelpSessionID));
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
CLEANUPANDEXIT:
|
||
|
|
|
||
|
|
DC_END_FN();
|
||
|
|
return hr;
|
||
|
|
}
|
||
|
|
|
||
|
|
void
|
||
|
|
CRemoteDesktopSession::Shutdown()
|
||
|
|
/*++
|
||
|
|
|
||
|
|
Routine Description:
|
||
|
|
|
||
|
|
The Shutdown method causes the COM object to no longer be
|
||
|
|
prepared for connection by the client-side Remote Desktop Host
|
||
|
|
ActiveX Control.
|
||
|
|
|
||
|
|
Arguments:
|
||
|
|
|
||
|
|
Return Value:
|
||
|
|
|
||
|
|
S_OK on success. Otherwise, an error code is returned.
|
||
|
|
|
||
|
|
--*/
|
||
|
|
{
|
||
|
|
DC_BEGIN_FN("CRemoteDesktopSession::Shutdown");
|
||
|
|
|
||
|
|
//
|
||
|
|
// Delete the help session created via the call to StartListening
|
||
|
|
// and delete the instance of the Remote Desktop Help Session Manager
|
||
|
|
// service.
|
||
|
|
//
|
||
|
|
if (m_HelpSessionManager != NULL) {
|
||
|
|
m_HelpSessionManager->DeleteHelpSession(m_HelpSessionID);
|
||
|
|
m_HelpSessionManager = NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
CLEANUPANDEXIT:
|
||
|
|
|
||
|
|
DC_END_FN();
|
||
|
|
}
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
CRemoteDesktopSession::put_SharingClass(
|
||
|
|
REMOTE_DESKTOP_SHARING_CLASS sharingClass
|
||
|
|
)
|
||
|
|
/*++
|
||
|
|
|
||
|
|
Routine Description:
|
||
|
|
|
||
|
|
Set the desktop sharing level.
|
||
|
|
|
||
|
|
Arguments:
|
||
|
|
|
||
|
|
Return Value:
|
||
|
|
|
||
|
|
--*/
|
||
|
|
{
|
||
|
|
DC_BEGIN_FN("CRemoteDesktopSession::put_SharingClass");
|
||
|
|
|
||
|
|
HRESULT hr = m_HelpSession->put_UserHelpSessionRemoteDesktopSharingSetting(
|
||
|
|
sharingClass
|
||
|
|
);
|
||
|
|
DC_END_FN();
|
||
|
|
return hr;
|
||
|
|
}
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
CRemoteDesktopSession::get_SharingClass(
|
||
|
|
REMOTE_DESKTOP_SHARING_CLASS *sharingClass
|
||
|
|
)
|
||
|
|
/*++
|
||
|
|
|
||
|
|
Routine Description:
|
||
|
|
|
||
|
|
Get the desktop sharing level.
|
||
|
|
|
||
|
|
Arguments:
|
||
|
|
|
||
|
|
Return Value:
|
||
|
|
|
||
|
|
--*/
|
||
|
|
{
|
||
|
|
DC_BEGIN_FN("CRemoteDesktopSession::get_SharingClass");
|
||
|
|
|
||
|
|
HRESULT hr = m_HelpSession->get_UserHelpSessionRemoteDesktopSharingSetting(
|
||
|
|
sharingClass
|
||
|
|
);
|
||
|
|
|
||
|
|
DC_END_FN();
|
||
|
|
return hr;
|
||
|
|
}
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
CRemoteDesktopSession::put_OnConnected(
|
||
|
|
IDispatch *iDisp
|
||
|
|
)
|
||
|
|
/*++
|
||
|
|
|
||
|
|
Routine Description:
|
||
|
|
|
||
|
|
Assign the outgoing interface for 'connected' events.
|
||
|
|
Only one interface can be assigned at a time.
|
||
|
|
|
||
|
|
Arguments:
|
||
|
|
|
||
|
|
Return Value:
|
||
|
|
|
||
|
|
--*/
|
||
|
|
{
|
||
|
|
DC_BEGIN_FN("CRemoteDesktopSession::put_OnConnected");
|
||
|
|
|
||
|
|
HRESULT hr = S_OK;
|
||
|
|
|
||
|
|
if (m_OnConnected != NULL) {
|
||
|
|
//
|
||
|
|
// The client proc may have gone away, so we need
|
||
|
|
// to catch exceptions on the release.
|
||
|
|
//
|
||
|
|
try {
|
||
|
|
m_OnConnected->Release();
|
||
|
|
}
|
||
|
|
catch (...) {
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
m_OnConnected = iDisp;
|
||
|
|
if (m_OnConnected != NULL) {
|
||
|
|
try {
|
||
|
|
m_OnConnected->AddRef();
|
||
|
|
}
|
||
|
|
catch (...) {
|
||
|
|
m_OnConnected = NULL;
|
||
|
|
TRC_ERR((TB, L"Exception caught in AddRef"));
|
||
|
|
hr = E_FAIL;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
DC_END_FN();
|
||
|
|
return hr;
|
||
|
|
}
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
CRemoteDesktopSession::put_OnDisconnected(
|
||
|
|
IDispatch *iDisp
|
||
|
|
)
|
||
|
|
/*++
|
||
|
|
|
||
|
|
Routine Description:
|
||
|
|
|
||
|
|
Assign the outgoing interface for 'disconnected' events.
|
||
|
|
Only one interface can be assigned at a time.
|
||
|
|
|
||
|
|
Arguments:
|
||
|
|
|
||
|
|
Return Value:
|
||
|
|
|
||
|
|
--*/
|
||
|
|
{
|
||
|
|
DC_BEGIN_FN("CRemoteDesktopSession::put_OnDisconnected(");
|
||
|
|
|
||
|
|
HRESULT hr = S_OK;
|
||
|
|
if (m_OnDisconnected != NULL) {
|
||
|
|
//
|
||
|
|
// The client proc may have gone away, so we need
|
||
|
|
// to catch exceptions on the release.
|
||
|
|
//
|
||
|
|
try {
|
||
|
|
m_OnDisconnected->Release();
|
||
|
|
}
|
||
|
|
catch (...) {
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
m_OnDisconnected = iDisp;
|
||
|
|
if (m_OnDisconnected != NULL) {
|
||
|
|
try {
|
||
|
|
m_OnDisconnected->AddRef();
|
||
|
|
}
|
||
|
|
catch (...) {
|
||
|
|
m_OnDisconnected = NULL;
|
||
|
|
TRC_ERR((TB, L"Exception caught in AddRef"));
|
||
|
|
hr = E_FAIL;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
DC_END_FN();
|
||
|
|
return hr;
|
||
|
|
}
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
CRemoteDesktopSession::CloseRemoteDesktopSession()
|
||
|
|
/*++
|
||
|
|
|
||
|
|
Routine Description:
|
||
|
|
|
||
|
|
Remove RDS session from the containing host object. Note that
|
||
|
|
this function does not dereference the ISAFRemoteDesktopSession
|
||
|
|
interface.
|
||
|
|
|
||
|
|
Arguments:
|
||
|
|
|
||
|
|
Return Value:
|
||
|
|
|
||
|
|
--*/
|
||
|
|
{
|
||
|
|
DC_BEGIN_FN("CRemoteDesktopSession::CloseRemoteDesktopSession");
|
||
|
|
|
||
|
|
HRESULT hr = m_RDSHost->CloseRemoteDesktopSession(this);
|
||
|
|
|
||
|
|
DC_END_FN();
|
||
|
|
return hr;
|
||
|
|
}
|
||
|
|
|
||
|
|
VOID
|
||
|
|
CRemoteDesktopSession::ClientConnected()
|
||
|
|
/*++
|
||
|
|
|
||
|
|
Routine Description:
|
||
|
|
|
||
|
|
Called when a connection to the client has been established.
|
||
|
|
|
||
|
|
Arguments:
|
||
|
|
|
||
|
|
Return Value:
|
||
|
|
|
||
|
|
--*/
|
||
|
|
{
|
||
|
|
DC_BEGIN_FN("CRemoteDesktopSession::Connected");
|
||
|
|
|
||
|
|
ASSERT(IsValid());
|
||
|
|
|
||
|
|
//
|
||
|
|
// We will catch and ignore exceptions here. The interface may
|
||
|
|
// have been implemented in a client application that has 'gone
|
||
|
|
// away.'
|
||
|
|
//
|
||
|
|
try {
|
||
|
|
Fire_ClientConnected(m_OnConnected);
|
||
|
|
}
|
||
|
|
catch (...) {
|
||
|
|
TRC_ALT((TB, L"Exception caught."));
|
||
|
|
}
|
||
|
|
|
||
|
|
DC_END_FN();
|
||
|
|
}
|
||
|
|
|
||
|
|
VOID
|
||
|
|
CRemoteDesktopSession::ClientDisconnected()
|
||
|
|
/*++
|
||
|
|
|
||
|
|
Routine Description:
|
||
|
|
|
||
|
|
Called when a connection to the client has been terminated.
|
||
|
|
cd
|
||
|
|
Arguments:
|
||
|
|
|
||
|
|
Return Value:
|
||
|
|
|
||
|
|
--*/
|
||
|
|
{
|
||
|
|
DC_BEGIN_FN("CRemoteDesktopSession::Disconnected");
|
||
|
|
|
||
|
|
ASSERT(IsValid());
|
||
|
|
|
||
|
|
//
|
||
|
|
// We will catch and ignore exceptions here. The interface may
|
||
|
|
// have been implemented in a client application that has 'gone
|
||
|
|
// away.'
|
||
|
|
//
|
||
|
|
try {
|
||
|
|
Fire_ClientDisconnected(m_OnDisconnected);
|
||
|
|
}
|
||
|
|
catch (...) {
|
||
|
|
TRC_ALT((TB, L"Exception caught."));
|
||
|
|
}
|
||
|
|
|
||
|
|
DC_END_FN();
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|