Milestone 5: deliver embedded RDP sessions and lifecycle hardening
This commit is contained in:
Keith Smith
855
third_party/FreeRDP/server/shadow/Win/win_wds.c
vendored
Normal file
855
third_party/FreeRDP/server/shadow/Win/win_wds.c
vendored
Normal file
@@ -0,0 +1,855 @@
|
||||
/**
|
||||
* FreeRDP: A Remote Desktop Protocol Implementation
|
||||
*
|
||||
* Copyright 2014 Marc-Andre Moreau <marcandre.moreau@gmail.com>
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#include <freerdp/config.h>
|
||||
|
||||
#include <winpr/crt.h>
|
||||
#include <winpr/print.h>
|
||||
#include <freerdp/log.h>
|
||||
|
||||
#include "win_rdp.h"
|
||||
|
||||
#include "win_wds.h"
|
||||
|
||||
/**
|
||||
* Windows Desktop Sharing API:
|
||||
* http://blogs.msdn.com/b/rds/archive/2007/03/08/windows-desktop-sharing-api.aspx
|
||||
*
|
||||
* Windows Desktop Sharing Interfaces:
|
||||
* http://msdn.microsoft.com/en-us/library/aa373871%28v=vs.85%29.aspx
|
||||
*
|
||||
* Offer Remote Assistance Sample C:
|
||||
* http://msdn.microsoft.com/en-us/library/ms811079.aspx#remoteassistanceapi_topic2b
|
||||
*
|
||||
* Remote Assistance in XP: Programmatically establish an RDP session:
|
||||
* http://www.codeproject.com/Articles/29939/Remote-Assistance-in-XP-Programmatically-establish
|
||||
*/
|
||||
|
||||
#undef DEFINE_GUID
|
||||
#define INITGUID
|
||||
|
||||
#include <initguid.h>
|
||||
|
||||
#include <freerdp/assistance.h>
|
||||
|
||||
#define TAG SERVER_TAG("shadow.win")
|
||||
|
||||
DEFINE_GUID(CLSID_RDPSession, 0x9B78F0E6, 0x3E05, 0x4A5B, 0xB2, 0xE8, 0xE7, 0x43, 0xA8, 0x95, 0x6B,
|
||||
0x65);
|
||||
DEFINE_GUID(DIID__IRDPSessionEvents, 0x98a97042, 0x6698, 0x40e9, 0x8e, 0xfd, 0xb3, 0x20, 0x09, 0x90,
|
||||
0x00, 0x4b);
|
||||
DEFINE_GUID(IID_IRDPSRAPISharingSession, 0xeeb20886, 0xe470, 0x4cf6, 0x84, 0x2b, 0x27, 0x39, 0xc0,
|
||||
0xec, 0x5c, 0xfb);
|
||||
DEFINE_GUID(IID_IRDPSRAPIAttendee, 0xec0671b3, 0x1b78, 0x4b80, 0xa4, 0x64, 0x91, 0x32, 0x24, 0x75,
|
||||
0x43, 0xe3);
|
||||
DEFINE_GUID(IID_IRDPSRAPIAttendeeManager, 0xba3a37e8, 0x33da, 0x4749, 0x8d, 0xa0, 0x07, 0xfa, 0x34,
|
||||
0xda, 0x79, 0x44);
|
||||
DEFINE_GUID(IID_IRDPSRAPISessionProperties, 0x339b24f2, 0x9bc0, 0x4f16, 0x9a, 0xac, 0xf1, 0x65,
|
||||
0x43, 0x3d, 0x13, 0xd4);
|
||||
DEFINE_GUID(CLSID_RDPSRAPIApplicationFilter, 0xe35ace89, 0xc7e8, 0x427e, 0xa4, 0xf9, 0xb9, 0xda,
|
||||
0x07, 0x28, 0x26, 0xbd);
|
||||
DEFINE_GUID(CLSID_RDPSRAPIInvitationManager, 0x53d9c9db, 0x75ab, 0x4271, 0x94, 0x8a, 0x4c, 0x4e,
|
||||
0xb3, 0x6a, 0x8f, 0x2b);
|
||||
|
||||
static ULONG Shadow_IRDPSessionEvents_RefCount = 0;
|
||||
|
||||
const char* GetRDPSessionEventString(DISPID id)
|
||||
{
|
||||
switch (id)
|
||||
{
|
||||
case DISPID_RDPSRAPI_EVENT_ON_ATTENDEE_CONNECTED:
|
||||
return "OnAttendeeConnected";
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_ATTENDEE_DISCONNECTED:
|
||||
return "OnAttendeeDisconnected";
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_ATTENDEE_UPDATE:
|
||||
return "OnAttendeeUpdate";
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_ERROR:
|
||||
return "OnError";
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_VIEWER_CONNECTED:
|
||||
return "OnConnectionEstablished";
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_VIEWER_DISCONNECTED:
|
||||
return "OnConnectionTerminated";
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_VIEWER_AUTHENTICATED:
|
||||
return "OnConnectionAuthenticated";
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_VIEWER_CONNECTFAILED:
|
||||
return "OnConnectionFailed";
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_CTRLLEVEL_CHANGE_REQUEST:
|
||||
return "OnControlLevelChangeRequest";
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_GRAPHICS_STREAM_PAUSED:
|
||||
return "OnGraphicsStreamPaused";
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_GRAPHICS_STREAM_RESUMED:
|
||||
return "OnGraphicsStreamResumed";
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_VIRTUAL_CHANNEL_JOIN:
|
||||
return "OnChannelJoin";
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_VIRTUAL_CHANNEL_LEAVE:
|
||||
return "OnChannelLeave";
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_VIRTUAL_CHANNEL_DATARECEIVED:
|
||||
return "OnChannelDataReceived";
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_VIRTUAL_CHANNEL_SENDCOMPLETED:
|
||||
return "OnChannelDataSent";
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_APPLICATION_OPEN:
|
||||
return "OnApplicationOpen";
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_APPLICATION_CLOSE:
|
||||
return "OnApplicationClose";
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_APPLICATION_UPDATE:
|
||||
return "OnApplicationUpdate";
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_WINDOW_OPEN:
|
||||
return "OnWindowOpen";
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_WINDOW_CLOSE:
|
||||
return "OnWindowClose";
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_WINDOW_UPDATE:
|
||||
return "OnWindowUpdate";
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_APPFILTER_UPDATE:
|
||||
return "OnAppFilterUpdate";
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_SHARED_RECT_CHANGED:
|
||||
return "OnSharedRectChanged";
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_FOCUSRELEASED:
|
||||
return "OnFocusReleased";
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_SHARED_DESKTOP_SETTINGS_CHANGED:
|
||||
return "OnSharedDesktopSettingsChanged";
|
||||
break;
|
||||
|
||||
case DISPID_RDPAPI_EVENT_ON_BOUNDING_RECT_CHANGED:
|
||||
return "OnViewingSizeChanged";
|
||||
break;
|
||||
}
|
||||
|
||||
return "OnUnknown";
|
||||
}
|
||||
|
||||
static HRESULT STDMETHODCALLTYPE
|
||||
Shadow_IRDPSessionEvents_QueryInterface(__RPC__in _IRDPSessionEvents* This,
|
||||
/* [in] */ __RPC__in REFIID riid,
|
||||
/* [annotation][iid_is][out] */
|
||||
_COM_Outptr_ void** ppvObject)
|
||||
{
|
||||
*ppvObject = nullptr;
|
||||
|
||||
if (IsEqualIID(riid, &DIID__IRDPSessionEvents) || IsEqualIID(riid, &IID_IDispatch) ||
|
||||
IsEqualIID(riid, &IID_IUnknown))
|
||||
{
|
||||
*ppvObject = This;
|
||||
}
|
||||
|
||||
if (!(*ppvObject))
|
||||
return E_NOINTERFACE;
|
||||
|
||||
This->lpVtbl->AddRef(This);
|
||||
return S_OK;
|
||||
}
|
||||
|
||||
static ULONG STDMETHODCALLTYPE Shadow_IRDPSessionEvents_AddRef(__RPC__in _IRDPSessionEvents* This)
|
||||
{
|
||||
Shadow_IRDPSessionEvents_RefCount++;
|
||||
return Shadow_IRDPSessionEvents_RefCount;
|
||||
}
|
||||
|
||||
static ULONG STDMETHODCALLTYPE Shadow_IRDPSessionEvents_Release(__RPC__in _IRDPSessionEvents* This)
|
||||
{
|
||||
if (!Shadow_IRDPSessionEvents_RefCount)
|
||||
return 0;
|
||||
|
||||
Shadow_IRDPSessionEvents_RefCount--;
|
||||
return Shadow_IRDPSessionEvents_RefCount;
|
||||
}
|
||||
|
||||
static HRESULT STDMETHODCALLTYPE
|
||||
Shadow_IRDPSessionEvents_GetTypeInfoCount(__RPC__in _IRDPSessionEvents* This,
|
||||
/* [out] */ __RPC__out UINT* pctinfo)
|
||||
{
|
||||
WLog_INFO(TAG, "Shadow_IRDPSessionEvents_GetTypeInfoCount");
|
||||
*pctinfo = 1;
|
||||
return S_OK;
|
||||
}
|
||||
|
||||
static HRESULT STDMETHODCALLTYPE
|
||||
Shadow_IRDPSessionEvents_GetTypeInfo(__RPC__in _IRDPSessionEvents* This,
|
||||
/* [in] */ UINT iTInfo,
|
||||
/* [in] */ LCID lcid,
|
||||
/* [out] */ __RPC__deref_out_opt ITypeInfo** ppTInfo)
|
||||
{
|
||||
WLog_INFO(TAG, "Shadow_IRDPSessionEvents_GetTypeInfo");
|
||||
return E_NOTIMPL;
|
||||
}
|
||||
|
||||
static HRESULT STDMETHODCALLTYPE Shadow_IRDPSessionEvents_GetIDsOfNames(
|
||||
__RPC__in _IRDPSessionEvents* This,
|
||||
/* [in] */ __RPC__in REFIID riid,
|
||||
/* [size_is][in] */ __RPC__in_ecount_full(cNames) LPOLESTR* rgszNames,
|
||||
/* [range][in] */ __RPC__in_range(0, 16384) UINT cNames,
|
||||
/* [in] */ LCID lcid,
|
||||
/* [size_is][out] */ __RPC__out_ecount_full(cNames) DISPID* rgDispId)
|
||||
{
|
||||
WLog_INFO(TAG, "Shadow_IRDPSessionEvents_GetIDsOfNames");
|
||||
return E_NOTIMPL;
|
||||
}
|
||||
|
||||
static HRESULT STDMETHODCALLTYPE Shadow_IRDPSessionEvents_Invoke(_IRDPSessionEvents* This,
|
||||
/* [annotation][in] */
|
||||
_In_ DISPID dispIdMember,
|
||||
/* [annotation][in] */
|
||||
_In_ REFIID riid,
|
||||
/* [annotation][in] */
|
||||
_In_ LCID lcid,
|
||||
/* [annotation][in] */
|
||||
_In_ WORD wFlags,
|
||||
/* [annotation][out][in] */
|
||||
_In_ DISPPARAMS* pDispParams,
|
||||
/* [annotation][out] */
|
||||
_Out_opt_ VARIANT* pVarResult,
|
||||
/* [annotation][out] */
|
||||
_Out_opt_ EXCEPINFO* pExcepInfo,
|
||||
/* [annotation][out] */
|
||||
_Out_opt_ UINT* puArgErr)
|
||||
{
|
||||
HRESULT hr;
|
||||
VARIANT vr;
|
||||
UINT uArgErr;
|
||||
WLog_INFO(TAG, "%s (%ld)", GetRDPSessionEventString(dispIdMember), dispIdMember);
|
||||
|
||||
switch (dispIdMember)
|
||||
{
|
||||
case DISPID_RDPSRAPI_EVENT_ON_ATTENDEE_CONNECTED:
|
||||
{
|
||||
int level;
|
||||
IDispatch* pDispatch;
|
||||
IRDPSRAPIAttendee* pAttendee;
|
||||
vr.vt = VT_DISPATCH;
|
||||
vr.pdispVal = nullptr;
|
||||
hr = DispGetParam(pDispParams, 0, VT_DISPATCH, &vr, &uArgErr);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "%s DispGetParam(0, VT_DISPATCH) failure: 0x%08lX",
|
||||
GetRDPSessionEventString(dispIdMember), hr);
|
||||
return hr;
|
||||
}
|
||||
|
||||
pDispatch = vr.pdispVal;
|
||||
hr = pDispatch->lpVtbl->QueryInterface(pDispatch, &IID_IRDPSRAPIAttendee,
|
||||
(void**)&pAttendee);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_INFO(TAG, "%s IDispatch::QueryInterface(IRDPSRAPIAttendee) failure: 0x%08lX",
|
||||
GetRDPSessionEventString(dispIdMember), hr);
|
||||
return hr;
|
||||
}
|
||||
|
||||
level = CTRL_LEVEL_VIEW;
|
||||
// level = CTRL_LEVEL_INTERACTIVE;
|
||||
hr = pAttendee->lpVtbl->put_ControlLevel(pAttendee, level);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_INFO(TAG, "%s IRDPSRAPIAttendee::put_ControlLevel() failure: 0x%08lX",
|
||||
GetRDPSessionEventString(dispIdMember), hr);
|
||||
return hr;
|
||||
}
|
||||
|
||||
pAttendee->lpVtbl->Release(pAttendee);
|
||||
}
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_ATTENDEE_DISCONNECTED:
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_ATTENDEE_UPDATE:
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_ERROR:
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_VIEWER_CONNECTED:
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_VIEWER_DISCONNECTED:
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_VIEWER_AUTHENTICATED:
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_VIEWER_CONNECTFAILED:
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_CTRLLEVEL_CHANGE_REQUEST:
|
||||
{
|
||||
int level;
|
||||
IDispatch* pDispatch;
|
||||
IRDPSRAPIAttendee* pAttendee;
|
||||
vr.vt = VT_INT;
|
||||
vr.pdispVal = nullptr;
|
||||
hr = DispGetParam(pDispParams, 1, VT_INT, &vr, &uArgErr);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_INFO(TAG, "%s DispGetParam(1, VT_INT) failure: 0x%08lX",
|
||||
GetRDPSessionEventString(dispIdMember), hr);
|
||||
return hr;
|
||||
}
|
||||
|
||||
level = vr.intVal;
|
||||
vr.vt = VT_DISPATCH;
|
||||
vr.pdispVal = nullptr;
|
||||
hr = DispGetParam(pDispParams, 0, VT_DISPATCH, &vr, &uArgErr);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "%s DispGetParam(0, VT_DISPATCH) failure: 0x%08lX",
|
||||
GetRDPSessionEventString(dispIdMember), hr);
|
||||
return hr;
|
||||
}
|
||||
|
||||
pDispatch = vr.pdispVal;
|
||||
hr = pDispatch->lpVtbl->QueryInterface(pDispatch, &IID_IRDPSRAPIAttendee,
|
||||
(void**)&pAttendee);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_INFO(TAG, "%s IDispatch::QueryInterface(IRDPSRAPIAttendee) failure: 0x%08lX",
|
||||
GetRDPSessionEventString(dispIdMember), hr);
|
||||
return hr;
|
||||
}
|
||||
|
||||
hr = pAttendee->lpVtbl->put_ControlLevel(pAttendee, level);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_INFO(TAG, "%s IRDPSRAPIAttendee::put_ControlLevel() failure: 0x%08lX",
|
||||
GetRDPSessionEventString(dispIdMember), hr);
|
||||
return hr;
|
||||
}
|
||||
|
||||
pAttendee->lpVtbl->Release(pAttendee);
|
||||
}
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_GRAPHICS_STREAM_PAUSED:
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_GRAPHICS_STREAM_RESUMED:
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_VIRTUAL_CHANNEL_JOIN:
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_VIRTUAL_CHANNEL_LEAVE:
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_VIRTUAL_CHANNEL_DATARECEIVED:
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_VIRTUAL_CHANNEL_SENDCOMPLETED:
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_APPLICATION_OPEN:
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_APPLICATION_CLOSE:
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_APPLICATION_UPDATE:
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_WINDOW_OPEN:
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_WINDOW_CLOSE:
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_WINDOW_UPDATE:
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_APPFILTER_UPDATE:
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_SHARED_RECT_CHANGED:
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_FOCUSRELEASED:
|
||||
break;
|
||||
|
||||
case DISPID_RDPSRAPI_EVENT_ON_SHARED_DESKTOP_SETTINGS_CHANGED:
|
||||
break;
|
||||
|
||||
case DISPID_RDPAPI_EVENT_ON_BOUNDING_RECT_CHANGED:
|
||||
break;
|
||||
}
|
||||
|
||||
return S_OK;
|
||||
}
|
||||
|
||||
static _IRDPSessionEventsVtbl Shadow_IRDPSessionEventsVtbl = {
|
||||
/* IUnknown */
|
||||
Shadow_IRDPSessionEvents_QueryInterface, Shadow_IRDPSessionEvents_AddRef,
|
||||
Shadow_IRDPSessionEvents_Release,
|
||||
|
||||
/* IDispatch */
|
||||
Shadow_IRDPSessionEvents_GetTypeInfoCount, Shadow_IRDPSessionEvents_GetTypeInfo,
|
||||
Shadow_IRDPSessionEvents_GetIDsOfNames, Shadow_IRDPSessionEvents_Invoke
|
||||
};
|
||||
|
||||
static _IRDPSessionEvents Shadow_IRDPSessionEvents = { &Shadow_IRDPSessionEventsVtbl };
|
||||
|
||||
static LRESULT CALLBACK ShadowWndProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
|
||||
{
|
||||
switch (uMsg)
|
||||
{
|
||||
case WM_CLOSE:
|
||||
DestroyWindow(hwnd);
|
||||
break;
|
||||
|
||||
case WM_DESTROY:
|
||||
PostQuitMessage(0);
|
||||
break;
|
||||
|
||||
default:
|
||||
return DefWindowProc(hwnd, uMsg, wParam, lParam);
|
||||
break;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
int win_shadow_wds_wnd_init(winShadowSubsystem* subsystem)
|
||||
{
|
||||
HMODULE hModule;
|
||||
HINSTANCE hInstance;
|
||||
WNDCLASSEX wndClassEx = WINPR_C_ARRAY_INIT;
|
||||
hModule = GetModuleHandle(nullptr);
|
||||
|
||||
wndClassEx.cbSize = sizeof(WNDCLASSEX);
|
||||
wndClassEx.style = 0;
|
||||
wndClassEx.lpfnWndProc = ShadowWndProc;
|
||||
wndClassEx.cbClsExtra = 0;
|
||||
wndClassEx.cbWndExtra = 0;
|
||||
wndClassEx.hInstance = hModule;
|
||||
wndClassEx.hIcon = nullptr;
|
||||
wndClassEx.hCursor = nullptr;
|
||||
wndClassEx.hbrBackground = nullptr;
|
||||
wndClassEx.lpszMenuName = _T("ShadowWndMenu");
|
||||
wndClassEx.lpszClassName = _T("ShadowWndClass");
|
||||
wndClassEx.hIconSm = nullptr;
|
||||
|
||||
if (!RegisterClassEx(&wndClassEx))
|
||||
{
|
||||
WLog_ERR(TAG, "RegisterClassEx failure");
|
||||
return -1;
|
||||
}
|
||||
|
||||
hInstance = wndClassEx.hInstance;
|
||||
subsystem->hWnd = CreateWindowEx(0, wndClassEx.lpszClassName, 0, 0, 0, 0, 0, 0, HWND_MESSAGE, 0,
|
||||
hInstance, nullptr);
|
||||
|
||||
if (!subsystem->hWnd)
|
||||
{
|
||||
WLog_INFO(TAG, "CreateWindowEx failure");
|
||||
return -1;
|
||||
}
|
||||
|
||||
return 1;
|
||||
}
|
||||
|
||||
int win_shadow_wds_init(winShadowSubsystem* subsystem)
|
||||
{
|
||||
int status = -1;
|
||||
|
||||
long left = 0;
|
||||
long top = 0;
|
||||
long right = 0;
|
||||
long bottom = 0;
|
||||
BSTR bstrAuthString = nullptr;
|
||||
BSTR bstrGroupName = nullptr;
|
||||
BSTR bstrPassword = nullptr;
|
||||
BSTR bstrPropertyName = nullptr;
|
||||
VARIANT varPropertyValue;
|
||||
rdpAssistanceFile* file = nullptr;
|
||||
IConnectionPoint* pCP = nullptr;
|
||||
IConnectionPointContainer* pCPC = nullptr;
|
||||
|
||||
win_shadow_wds_wnd_init(subsystem);
|
||||
HRESULT hr = OleInitialize(nullptr);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "OleInitialize() failure");
|
||||
return -1;
|
||||
}
|
||||
|
||||
hr = CoInitializeEx(nullptr, COINIT_APARTMENTTHREADED);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "CoInitialize() failure");
|
||||
return -1;
|
||||
}
|
||||
|
||||
hr = CoCreateInstance(&CLSID_RDPSession, nullptr, CLSCTX_ALL, &IID_IRDPSRAPISharingSession,
|
||||
(void**)&(subsystem->pSharingSession));
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "CoCreateInstance(IRDPSRAPISharingSession) failure: 0x%08lX", hr);
|
||||
return -1;
|
||||
}
|
||||
|
||||
IUnknown* pUnknown = (IUnknown*)subsystem->pSharingSession;
|
||||
hr = pUnknown->lpVtbl->QueryInterface(pUnknown, &IID_IConnectionPointContainer, (void**)&pCPC);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "QueryInterface(IID_IConnectionPointContainer) failure: 0x%08lX", hr);
|
||||
return -1;
|
||||
}
|
||||
|
||||
pCPC->lpVtbl->FindConnectionPoint(pCPC, &DIID__IRDPSessionEvents, &pCP);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(
|
||||
TAG,
|
||||
"IConnectionPointContainer::FindConnectionPoint(_IRDPSessionEvents) failure: 0x%08lX",
|
||||
hr);
|
||||
return -1;
|
||||
}
|
||||
|
||||
DWORD dwCookie = 0;
|
||||
subsystem->pSessionEvents = &Shadow_IRDPSessionEvents;
|
||||
subsystem->pSessionEvents->lpVtbl->AddRef(subsystem->pSessionEvents);
|
||||
hr = pCP->lpVtbl->Advise(pCP, (IUnknown*)subsystem->pSessionEvents, &dwCookie);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "IConnectionPoint::Advise(Shadow_IRDPSessionEvents) failure: 0x%08lX", hr);
|
||||
return -1;
|
||||
}
|
||||
|
||||
hr = subsystem->pSharingSession->lpVtbl->put_ColorDepth(subsystem->pSharingSession, 32);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "IRDPSRAPISharingSession::put_ColorDepth() failure: 0x%08lX", hr);
|
||||
return -1;
|
||||
}
|
||||
|
||||
hr = subsystem->pSharingSession->lpVtbl->GetDesktopSharedRect(subsystem->pSharingSession, &left,
|
||||
&top, &right, &bottom);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "IRDPSRAPISharingSession::GetDesktopSharedRect() failure: 0x%08lX", hr);
|
||||
return -1;
|
||||
}
|
||||
|
||||
long width = right - left;
|
||||
long height = bottom - top;
|
||||
WLog_INFO(
|
||||
TAG,
|
||||
"GetDesktopSharedRect(): left: %ld top: %ld right: %ld bottom: %ld width: %ld height: %ld",
|
||||
left, top, right, bottom, width, height);
|
||||
hr = subsystem->pSharingSession->lpVtbl->get_VirtualChannelManager(
|
||||
subsystem->pSharingSession, &(subsystem->pVirtualChannelMgr));
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "IRDPSRAPISharingSession::get_VirtualChannelManager() failure: 0x%08lX", hr);
|
||||
return -1;
|
||||
}
|
||||
|
||||
hr = subsystem->pSharingSession->lpVtbl->get_ApplicationFilter(
|
||||
subsystem->pSharingSession, &(subsystem->pApplicationFilter));
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "IRDPSRAPISharingSession::get_ApplicationFilter() failure: 0x%08lX", hr);
|
||||
return -1;
|
||||
}
|
||||
|
||||
hr = subsystem->pSharingSession->lpVtbl->get_Attendees(subsystem->pSharingSession,
|
||||
&(subsystem->pAttendeeMgr));
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "IRDPSRAPISharingSession::get_Attendees() failure: 0x%08lX", hr);
|
||||
return -1;
|
||||
}
|
||||
|
||||
hr = subsystem->pSharingSession->lpVtbl->get_Properties(subsystem->pSharingSession,
|
||||
&(subsystem->pSessionProperties));
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "IRDPSRAPISharingSession::get_Properties() failure: 0x%08lX", hr);
|
||||
return -1;
|
||||
}
|
||||
|
||||
bstrPropertyName = SysAllocString(L"PortId");
|
||||
varPropertyValue.vt = VT_I4;
|
||||
varPropertyValue.intVal = 40000;
|
||||
hr = subsystem->pSessionProperties->lpVtbl->put_Property(subsystem->pSessionProperties,
|
||||
bstrPropertyName, varPropertyValue);
|
||||
SysFreeString(bstrPropertyName);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "IRDPSRAPISessionProperties::put_Property(PortId) failure: 0x%08lX", hr);
|
||||
return -1;
|
||||
}
|
||||
|
||||
bstrPropertyName = SysAllocString(L"DrvConAttach");
|
||||
varPropertyValue.vt = VT_BOOL;
|
||||
varPropertyValue.boolVal = VARIANT_TRUE;
|
||||
hr = subsystem->pSessionProperties->lpVtbl->put_Property(subsystem->pSessionProperties,
|
||||
bstrPropertyName, varPropertyValue);
|
||||
SysFreeString(bstrPropertyName);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "IRDPSRAPISessionProperties::put_Property(DrvConAttach) failure: 0x%08lX",
|
||||
hr);
|
||||
return -1;
|
||||
}
|
||||
|
||||
bstrPropertyName = SysAllocString(L"PortProtocol");
|
||||
varPropertyValue.vt = VT_I4;
|
||||
// varPropertyValue.intVal = 0; // AF_UNSPEC
|
||||
varPropertyValue.intVal = 2; // AF_INET
|
||||
// varPropertyValue.intVal = 23; // AF_INET6
|
||||
hr = subsystem->pSessionProperties->lpVtbl->put_Property(subsystem->pSessionProperties,
|
||||
bstrPropertyName, varPropertyValue);
|
||||
SysFreeString(bstrPropertyName);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "IRDPSRAPISessionProperties::put_Property(PortProtocol) failure: 0x%08lX",
|
||||
hr);
|
||||
return -1;
|
||||
}
|
||||
|
||||
hr = subsystem->pSharingSession->lpVtbl->Open(subsystem->pSharingSession);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "IRDPSRAPISharingSession::Open() failure: 0x%08lX", hr);
|
||||
return -1;
|
||||
}
|
||||
|
||||
hr = subsystem->pSharingSession->lpVtbl->get_Invitations(subsystem->pSharingSession,
|
||||
&(subsystem->pInvitationMgr));
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "IRDPSRAPISharingSession::get_Invitations() failure");
|
||||
return -1;
|
||||
}
|
||||
|
||||
bstrAuthString = SysAllocString(L"Shadow");
|
||||
bstrGroupName = SysAllocString(L"ShadowGroup");
|
||||
bstrPassword = SysAllocString(L"Shadow123!");
|
||||
hr = subsystem->pInvitationMgr->lpVtbl->CreateInvitation(
|
||||
subsystem->pInvitationMgr, bstrAuthString, bstrGroupName, bstrPassword, 5,
|
||||
&(subsystem->pInvitation));
|
||||
SysFreeString(bstrAuthString);
|
||||
SysFreeString(bstrGroupName);
|
||||
SysFreeString(bstrPassword);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "IRDPSRAPIInvitationManager::CreateInvitation() failure: 0x%08lX", hr);
|
||||
return -1;
|
||||
}
|
||||
|
||||
file = subsystem->pAssistanceFile = freerdp_assistance_file_new();
|
||||
|
||||
if (!file)
|
||||
{
|
||||
WLog_ERR(TAG, "freerdp_assistance_file_new() failed");
|
||||
return -1;
|
||||
}
|
||||
|
||||
{
|
||||
int status2 = -1;
|
||||
char* ConnectionString2;
|
||||
BSTR bstrConnectionString;
|
||||
hr = subsystem->pInvitation->lpVtbl->get_ConnectionString(subsystem->pInvitation,
|
||||
&bstrConnectionString);
|
||||
|
||||
if (FAILED(hr))
|
||||
{
|
||||
WLog_ERR(TAG, "IRDPSRAPIInvitation::get_ConnectionString() failure: 0x%08lX", hr);
|
||||
return -1;
|
||||
}
|
||||
|
||||
ConnectionString2 = ConvertWCharToUtf8Alloc(bstrConnectionString, nullptr);
|
||||
SysFreeString(bstrConnectionString);
|
||||
status2 = freerdp_assistance_set_connection_string2(file, ConnectionString2, "Shadow123!");
|
||||
free(ConnectionString2);
|
||||
|
||||
if ((!ConnectionString2) || (status2 < 1))
|
||||
{
|
||||
WLog_ERR(TAG, "failed to convert connection string");
|
||||
return -1;
|
||||
}
|
||||
}
|
||||
|
||||
freerdp_assistance_print_file(file, WLog_Get(TAG), WLOG_INFO);
|
||||
status = win_shadow_rdp_init(subsystem);
|
||||
|
||||
if (status < 0)
|
||||
{
|
||||
WLog_ERR(TAG, "win_shadow_rdp_init() failure: %d", status);
|
||||
return status;
|
||||
}
|
||||
|
||||
rdpSettings* settings = subsystem->shw->settings;
|
||||
if (!freerdp_assistance_populate_settings_from_assistance_file(file, settings))
|
||||
return -1;
|
||||
if (!freerdp_settings_set_string(settings, FreeRDP_Domain, "RDP"))
|
||||
return -1;
|
||||
if (!freerdp_settings_set_string(settings, FreeRDP_Username, "Shadow"))
|
||||
return -1;
|
||||
if (!freerdp_settings_set_bool(settings, FreeRDP_AutoLogonEnabled, TRUE))
|
||||
return -1;
|
||||
if (!freerdp_settings_set_uint32(settings, FreeRDP_DesktopWidth, width))
|
||||
return -1;
|
||||
if (!freerdp_settings_set_uint32(settings, FreeRDP_DesktopHeight, height))
|
||||
return -1;
|
||||
status = win_shadow_rdp_start(subsystem);
|
||||
|
||||
if (status < 0)
|
||||
{
|
||||
WLog_ERR(TAG, "win_shadow_rdp_start() failure: %d", status);
|
||||
return status;
|
||||
}
|
||||
|
||||
return 1;
|
||||
}
|
||||
|
||||
int win_shadow_wds_uninit(winShadowSubsystem* subsystem)
|
||||
{
|
||||
if (subsystem->pSharingSession)
|
||||
{
|
||||
subsystem->pSharingSession->lpVtbl->Close(subsystem->pSharingSession);
|
||||
subsystem->pSharingSession->lpVtbl->Release(subsystem->pSharingSession);
|
||||
subsystem->pSharingSession = nullptr;
|
||||
}
|
||||
|
||||
if (subsystem->pVirtualChannelMgr)
|
||||
{
|
||||
subsystem->pVirtualChannelMgr->lpVtbl->Release(subsystem->pVirtualChannelMgr);
|
||||
subsystem->pVirtualChannelMgr = nullptr;
|
||||
}
|
||||
|
||||
if (subsystem->pApplicationFilter)
|
||||
{
|
||||
subsystem->pApplicationFilter->lpVtbl->Release(subsystem->pApplicationFilter);
|
||||
subsystem->pApplicationFilter = nullptr;
|
||||
}
|
||||
|
||||
if (subsystem->pAttendeeMgr)
|
||||
{
|
||||
subsystem->pAttendeeMgr->lpVtbl->Release(subsystem->pAttendeeMgr);
|
||||
subsystem->pAttendeeMgr = nullptr;
|
||||
}
|
||||
|
||||
if (subsystem->pSessionProperties)
|
||||
{
|
||||
subsystem->pSessionProperties->lpVtbl->Release(subsystem->pSessionProperties);
|
||||
subsystem->pSessionProperties = nullptr;
|
||||
}
|
||||
|
||||
if (subsystem->pInvitationMgr)
|
||||
{
|
||||
subsystem->pInvitationMgr->lpVtbl->Release(subsystem->pInvitationMgr);
|
||||
subsystem->pInvitationMgr = nullptr;
|
||||
}
|
||||
|
||||
if (subsystem->pInvitation)
|
||||
{
|
||||
subsystem->pInvitation->lpVtbl->Release(subsystem->pInvitation);
|
||||
subsystem->pInvitation = nullptr;
|
||||
}
|
||||
|
||||
if (subsystem->pAssistanceFile)
|
||||
{
|
||||
freerdp_assistance_file_free(subsystem->pAssistanceFile);
|
||||
subsystem->pAssistanceFile = nullptr;
|
||||
}
|
||||
|
||||
if (subsystem->hWnd)
|
||||
{
|
||||
DestroyWindow(subsystem->hWnd);
|
||||
subsystem->hWnd = nullptr;
|
||||
}
|
||||
|
||||
if (subsystem->shw)
|
||||
{
|
||||
win_shadow_rdp_uninit(subsystem);
|
||||
subsystem->shw = nullptr;
|
||||
}
|
||||
|
||||
return 1;
|
||||
}
|
||||
Reference in New Issue
Block a user