FreeRDP/client/Windows/wf_cliprdr.c
2019-01-29 10:46:21 +01:00

2584 lines
61 KiB
C

/**
* FreeRDP: A Remote Desktop Protocol Implementation
* Windows Clipboard Redirection
*
* Copyright 2012 Jason Champion
* Copyright 2014 Marc-Andre Moreau <marcandre.moreau@gmail.com>
* Copyright 2015 Thincast Technologies GmbH
* Copyright 2015 DI (FH) Martin Haimberger <martin.haimberger@thincast.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.
*/
#ifdef HAVE_CONFIG_H
#include "config.h"
#endif
#define CINTERFACE
#define COBJMACROS
#include <ole2.h>
#include <shlobj.h>
#include <windows.h>
#include <winuser.h>
#include <assert.h>
#include <winpr/crt.h>
#include <winpr/tchar.h>
#include <winpr/stream.h>
#include <freerdp/log.h>
#include <freerdp/client/cliprdr.h>
#include <strsafe.h>
#include "wf_cliprdr.h"
#define TAG CLIENT_TAG("windows")
#ifdef WITH_DEBUG_CLIPRDR
#define DEBUG_CLIPRDR(...) WLog_DBG(TAG, __VA_ARGS__)
#else
#define DEBUG_CLIPRDR(...) do { } while (0)
#endif
typedef BOOL (WINAPI* fnAddClipboardFormatListener)(HWND hwnd);
typedef BOOL (WINAPI* fnRemoveClipboardFormatListener)(HWND hwnd);
typedef BOOL (WINAPI* fnGetUpdatedClipboardFormats)(PUINT lpuiFormats,
UINT cFormats, PUINT pcFormatsOut);
struct format_mapping
{
UINT32 remote_format_id;
UINT32 local_format_id;
WCHAR* name;
};
typedef struct format_mapping formatMapping;
struct _CliprdrEnumFORMATETC
{
IEnumFORMATETC iEnumFORMATETC;
LONG m_lRefCount;
LONG m_nIndex;
LONG m_nNumFormats;
FORMATETC* m_pFormatEtc;
};
typedef struct _CliprdrEnumFORMATETC CliprdrEnumFORMATETC;
struct _CliprdrStream
{
IStream iStream;
LONG m_lRefCount;
ULONG m_lIndex;
ULARGE_INTEGER m_lSize;
ULARGE_INTEGER m_lOffset;
FILEDESCRIPTORW m_Dsc;
void* m_pData;
};
typedef struct _CliprdrStream CliprdrStream;
struct _CliprdrDataObject
{
IDataObject iDataObject;
LONG m_lRefCount;
FORMATETC* m_pFormatEtc;
STGMEDIUM* m_pStgMedium;
ULONG m_nNumFormats;
ULONG m_nStreams;
IStream** m_pStream;
void* m_pData;
};
typedef struct _CliprdrDataObject CliprdrDataObject;
struct wf_clipboard
{
wfContext* wfc;
rdpChannels* channels;
CliprdrClientContext* context;
BOOL sync;
UINT32 capabilities;
size_t map_size;
size_t map_capacity;
formatMapping* format_mappings;
UINT32 requestedFormatId;
HWND hwnd;
HANDLE hmem;
HANDLE thread;
HANDLE response_data_event;
LPDATAOBJECT data_obj;
ULONG req_fsize;
char* req_fdata;
HANDLE req_fevent;
size_t nFiles;
size_t file_array_size;
WCHAR** file_names;
FILEDESCRIPTORW** fileDescriptor;
BOOL legacyApi;
HMODULE hUser32;
HWND hWndNextViewer;
fnAddClipboardFormatListener AddClipboardFormatListener;
fnRemoveClipboardFormatListener RemoveClipboardFormatListener;
fnGetUpdatedClipboardFormats GetUpdatedClipboardFormats;
};
typedef struct wf_clipboard wfClipboard;
#define WM_CLIPRDR_MESSAGE (WM_USER + 156)
#define OLE_SETCLIPBOARD 1
static BOOL wf_create_file_obj(wfClipboard* cliprdrrdr,
IDataObject** ppDataObject);
static void wf_destroy_file_obj(IDataObject* instance);
static UINT32 get_remote_format_id(wfClipboard* clipboard, UINT32 local_format);
static UINT cliprdr_send_data_request(wfClipboard* clipboard, UINT32 format);
static UINT cliprdr_send_lock(wfClipboard* clipboard);
static UINT cliprdr_send_unlock(wfClipboard* clipboard);
static UINT cliprdr_send_request_filecontents(wfClipboard* clipboard,
const void* streamid,
ULONG index, UINT32 flag, DWORD positionhigh,
DWORD positionlow, ULONG request);
static void CliprdrDataObject_Delete(CliprdrDataObject* instance);
static CliprdrEnumFORMATETC* CliprdrEnumFORMATETC_New(ULONG nFormats,
FORMATETC* pFormatEtc);
static void CliprdrEnumFORMATETC_Delete(CliprdrEnumFORMATETC* instance);
static void CliprdrStream_Delete(CliprdrStream* instance);
/**
* IStream
*/
static HRESULT STDMETHODCALLTYPE CliprdrStream_QueryInterface(IStream* This,
REFIID riid, void** ppvObject)
{
if (IsEqualIID(riid, &IID_IStream) || IsEqualIID(riid, &IID_IUnknown))
{
IStream_AddRef(This);
*ppvObject = This;
return S_OK;
}
else
{
*ppvObject = 0;
return E_NOINTERFACE;
}
}
static ULONG STDMETHODCALLTYPE CliprdrStream_AddRef(IStream* This)
{
CliprdrStream* instance = (CliprdrStream*) This;
if (!instance)
return 0;
return InterlockedIncrement(&instance->m_lRefCount);
}
static ULONG STDMETHODCALLTYPE CliprdrStream_Release(IStream* This)
{
LONG count;
CliprdrStream* instance = (CliprdrStream*) This;
if (!instance)
return 0;
count = InterlockedDecrement(&instance->m_lRefCount);
if (count == 0)
{
CliprdrStream_Delete(instance);
return 0;
}
else
{
return count;
}
}
static HRESULT STDMETHODCALLTYPE CliprdrStream_Read(IStream* This, void* pv,
ULONG cb, ULONG* pcbRead)
{
int ret;
CliprdrStream* instance = (CliprdrStream*) This;
wfClipboard* clipboard;
if (!pv || !pcbRead || !instance)
return E_INVALIDARG;
clipboard = (wfClipboard*) instance->m_pData;
*pcbRead = 0;
if (instance->m_lOffset.QuadPart >= instance->m_lSize.QuadPart)
return S_FALSE;
ret = cliprdr_send_request_filecontents(clipboard, (void*) This,
instance->m_lIndex, FILECONTENTS_RANGE,
instance->m_lOffset.HighPart, instance->m_lOffset.LowPart, cb);
if (ret < 0)
return E_FAIL;
if (clipboard->req_fdata)
{
CopyMemory(pv, clipboard->req_fdata, clipboard->req_fsize);
free(clipboard->req_fdata);
}
*pcbRead = clipboard->req_fsize;
instance->m_lOffset.QuadPart += clipboard->req_fsize;
if (clipboard->req_fsize < cb)
return S_FALSE;
return S_OK;
}
static HRESULT STDMETHODCALLTYPE CliprdrStream_Write(IStream* This,
const void* pv,
ULONG cb, ULONG* pcbWritten)
{
(void)This;
(void)pv;
(void)cb;
(void)pcbWritten;
return STG_E_ACCESSDENIED;
}
static HRESULT STDMETHODCALLTYPE CliprdrStream_Seek(IStream* This,
LARGE_INTEGER dlibMove,
DWORD dwOrigin, ULARGE_INTEGER* plibNewPosition)
{
ULONGLONG newoffset;
CliprdrStream* instance = (CliprdrStream*) This;
if (!instance)
return E_INVALIDARG;
newoffset = instance->m_lOffset.QuadPart;
switch (dwOrigin)
{
case STREAM_SEEK_SET:
newoffset = dlibMove.QuadPart;
break;
case STREAM_SEEK_CUR:
newoffset += dlibMove.QuadPart;
break;
case STREAM_SEEK_END:
newoffset = instance->m_lSize.QuadPart + dlibMove.QuadPart;
break;
default:
return E_INVALIDARG;
}
if (newoffset < 0 || newoffset >= instance->m_lSize.QuadPart)
return E_FAIL;
instance->m_lOffset.QuadPart = newoffset;
if (plibNewPosition)
plibNewPosition->QuadPart = instance->m_lOffset.QuadPart;
return S_OK;
}
static HRESULT STDMETHODCALLTYPE CliprdrStream_SetSize(IStream* This,
ULARGE_INTEGER libNewSize)
{
(void)This;
(void)libNewSize;
return E_NOTIMPL;
}
static HRESULT STDMETHODCALLTYPE CliprdrStream_CopyTo(IStream* This,
IStream* pstm,
ULARGE_INTEGER cb, ULARGE_INTEGER* pcbRead,
ULARGE_INTEGER* pcbWritten)
{
(void)This;
(void)pstm;
(void)cb;
(void)pcbRead;
(void)pcbWritten;
return E_NOTIMPL;
}
static HRESULT STDMETHODCALLTYPE CliprdrStream_Commit(IStream* This,
DWORD grfCommitFlags)
{
(void)This;
(void)grfCommitFlags;
return E_NOTIMPL;
}
static HRESULT STDMETHODCALLTYPE CliprdrStream_Revert(IStream* This)
{
(void)This;
return E_NOTIMPL;
}
static HRESULT STDMETHODCALLTYPE CliprdrStream_LockRegion(IStream* This,
ULARGE_INTEGER libOffset,
ULARGE_INTEGER cb, DWORD dwLockType)
{
(void)This;
(void)libOffset;
(void)cb;
(void)dwLockType;
return E_NOTIMPL;
}
static HRESULT STDMETHODCALLTYPE CliprdrStream_UnlockRegion(IStream* This,
ULARGE_INTEGER libOffset,
ULARGE_INTEGER cb, DWORD dwLockType)
{
(void)This;
(void)libOffset;
(void)cb;
(void)dwLockType;
return E_NOTIMPL;
}
static HRESULT STDMETHODCALLTYPE CliprdrStream_Stat(IStream* This,
STATSTG* pstatstg, DWORD grfStatFlag)
{
CliprdrStream* instance = (CliprdrStream*) This;
if (!instance)
return E_INVALIDARG;
if (pstatstg == NULL)
return STG_E_INVALIDPOINTER;
ZeroMemory(pstatstg, sizeof(STATSTG));
switch (grfStatFlag)
{
case STATFLAG_DEFAULT:
return STG_E_INSUFFICIENTMEMORY;
case STATFLAG_NONAME:
pstatstg->cbSize.QuadPart = instance->m_lSize.QuadPart;
pstatstg->grfLocksSupported = LOCK_EXCLUSIVE;
pstatstg->grfMode = GENERIC_READ;
pstatstg->grfStateBits = 0;
pstatstg->type = STGTY_STREAM;
break;
case STATFLAG_NOOPEN:
return STG_E_INVALIDFLAG;
default:
return STG_E_INVALIDFLAG;
}
return S_OK;
}
static HRESULT STDMETHODCALLTYPE CliprdrStream_Clone(IStream* This,
IStream** ppstm)
{
(void)This;
(void)ppstm;
return E_NOTIMPL;
}
static CliprdrStream* CliprdrStream_New(ULONG index, void* pData,
const FILEDESCRIPTORW* dsc)
{
IStream* iStream;
BOOL success = FALSE;
BOOL isDir = FALSE;
CliprdrStream* instance;
wfClipboard* clipboard = (wfClipboard*) pData;
instance = (CliprdrStream*) calloc(1, sizeof(CliprdrStream));
if (instance)
{
instance->m_Dsc = *dsc;
iStream = &instance->iStream;
iStream->lpVtbl = (IStreamVtbl*) calloc(1, sizeof(IStreamVtbl));
if (iStream->lpVtbl)
{
iStream->lpVtbl->QueryInterface = CliprdrStream_QueryInterface;
iStream->lpVtbl->AddRef = CliprdrStream_AddRef;
iStream->lpVtbl->Release = CliprdrStream_Release;
iStream->lpVtbl->Read = CliprdrStream_Read;
iStream->lpVtbl->Write = CliprdrStream_Write;
iStream->lpVtbl->Seek = CliprdrStream_Seek;
iStream->lpVtbl->SetSize = CliprdrStream_SetSize;
iStream->lpVtbl->CopyTo = CliprdrStream_CopyTo;
iStream->lpVtbl->Commit = CliprdrStream_Commit;
iStream->lpVtbl->Revert = CliprdrStream_Revert;
iStream->lpVtbl->LockRegion = CliprdrStream_LockRegion;
iStream->lpVtbl->UnlockRegion = CliprdrStream_UnlockRegion;
iStream->lpVtbl->Stat = CliprdrStream_Stat;
iStream->lpVtbl->Clone = CliprdrStream_Clone;
instance->m_lRefCount = 1;
instance->m_lIndex = index;
instance->m_pData = pData;
instance->m_lOffset.QuadPart = 0;
if (instance->m_Dsc.dwFlags & FD_ATTRIBUTES)
{
if (instance->m_Dsc.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
isDir = TRUE;
}
if (((instance->m_Dsc.dwFlags & FD_FILESIZE) == 0) && !isDir)
{
/* get content size of this stream */
if (cliprdr_send_request_filecontents(clipboard, (void*) instance,
instance->m_lIndex,
FILECONTENTS_SIZE, 0, 0, 8) == CHANNEL_RC_OK)
{
success = TRUE;
}
instance->m_lSize.QuadPart = *((LONGLONG*) clipboard->req_fdata);
free(clipboard->req_fdata);
}
else
success = TRUE;
}
}
if (!success)
{
CliprdrStream_Delete(instance);
instance = NULL;
}
return instance;
}
void CliprdrStream_Delete(CliprdrStream* instance)
{
if (instance)
{
free(instance->iStream.lpVtbl);
free(instance);
}
}
/**
* IDataObject
*/
static LONG cliprdr_lookup_format(CliprdrDataObject* instance,
FORMATETC* pFormatEtc)
{
ULONG i;
if (!instance || !pFormatEtc)
return -1;
for (i = 0; i < instance->m_nNumFormats; i++)
{
if ((pFormatEtc->tymed & instance->m_pFormatEtc[i].tymed) &&
pFormatEtc->cfFormat == instance->m_pFormatEtc[i].cfFormat &&
pFormatEtc->dwAspect & instance->m_pFormatEtc[i].dwAspect)
{
return (LONG)i;
}
}
return -1;
}
static HRESULT STDMETHODCALLTYPE CliprdrDataObject_QueryInterface(
IDataObject* This, REFIID riid, void** ppvObject)
{
(void)This;
if (!ppvObject)
return E_INVALIDARG;
if (IsEqualIID(riid, &IID_IDataObject) || IsEqualIID(riid, &IID_IUnknown))
{
IDataObject_AddRef(This);
*ppvObject = This;
return S_OK;
}
else
{
*ppvObject = 0;
return E_NOINTERFACE;
}
}
static ULONG STDMETHODCALLTYPE CliprdrDataObject_AddRef(IDataObject* This)
{
CliprdrDataObject* instance = (CliprdrDataObject*) This;
if (!instance)
return E_INVALIDARG;
return InterlockedIncrement(&instance->m_lRefCount);
}
static ULONG STDMETHODCALLTYPE CliprdrDataObject_Release(IDataObject* This)
{
LONG count;
CliprdrDataObject* instance = (CliprdrDataObject*) This;
if (!instance)
return E_INVALIDARG;
count = InterlockedDecrement(&instance->m_lRefCount);
if (count == 0)
{
CliprdrDataObject_Delete(instance);
return 0;
}
else
return count;
}
static HRESULT STDMETHODCALLTYPE CliprdrDataObject_GetData(
IDataObject* This, FORMATETC* pFormatEtc, STGMEDIUM* pMedium)
{
ULONG i;
LONG idx;
CliprdrDataObject* instance = (CliprdrDataObject*) This;
wfClipboard* clipboard;
if (!pFormatEtc || !pMedium || !instance)
return E_INVALIDARG;
clipboard = (wfClipboard*) instance->m_pData;
if (!clipboard)
return E_INVALIDARG;
if ((idx = cliprdr_lookup_format(instance, pFormatEtc)) == -1)
return DV_E_FORMATETC;
pMedium->tymed = instance->m_pFormatEtc[idx].tymed;
pMedium->pUnkForRelease = 0;
if (instance->m_pFormatEtc[idx].cfFormat == RegisterClipboardFormat(
CFSTR_FILEDESCRIPTORW))
{
FILEGROUPDESCRIPTOR* dsc;
DWORD remote = get_remote_format_id(clipboard,
instance->m_pFormatEtc[idx].cfFormat);
if (cliprdr_send_data_request(clipboard, remote) != 0)
return E_UNEXPECTED;
pMedium->hGlobal =
clipboard->hmem; /* points to a FILEGROUPDESCRIPTOR structure */
/* GlobalLock returns a pointer to the first byte of the memory block,
* in which is a FILEGROUPDESCRIPTOR structure, whose first UINT member
* is the number of FILEDESCRIPTOR's */
dsc = (FILEGROUPDESCRIPTOR*) GlobalLock(clipboard->hmem);
instance->m_nStreams = dsc->cItems;
GlobalUnlock(clipboard->hmem);
if (instance->m_nStreams > 0)
{
if (!instance->m_pStream)
{
instance->m_pStream = (LPSTREAM*) calloc(instance->m_nStreams,
sizeof(LPSTREAM));
if (instance->m_pStream)
{
for (i = 0; i < instance->m_nStreams; i++)
{
instance->m_pStream[i] = (IStream*) CliprdrStream_New(i, clipboard,
&dsc->fgd[i]);
if (!instance->m_pStream[i])
return E_OUTOFMEMORY;
}
}
}
}
if (!instance->m_pStream)
{
if (clipboard->hmem)
{
GlobalFree(clipboard->hmem);
clipboard->hmem = NULL;
}
pMedium->hGlobal = NULL;
return E_OUTOFMEMORY;
}
}
else if (instance->m_pFormatEtc[idx].cfFormat == RegisterClipboardFormat(
CFSTR_FILECONTENTS))
{
if (pFormatEtc->lindex < instance->m_nStreams)
{
pMedium->pstm = instance->m_pStream[pFormatEtc->lindex];
IDataObject_AddRef(instance->m_pStream[pFormatEtc->lindex]);
}
else
return E_INVALIDARG;
}
else
return E_UNEXPECTED;
return S_OK;
}
static HRESULT STDMETHODCALLTYPE CliprdrDataObject_GetDataHere(
IDataObject* This, FORMATETC* pformatetc, STGMEDIUM* pmedium)
{
(void)This;
(void)pformatetc;
(void)pmedium;
return E_NOTIMPL;
}
static HRESULT STDMETHODCALLTYPE CliprdrDataObject_QueryGetData(
IDataObject* This, FORMATETC* pformatetc)
{
CliprdrDataObject* instance = (CliprdrDataObject*) This;
if (!pformatetc)
return E_INVALIDARG;
if (cliprdr_lookup_format(instance, pformatetc) == -1)
return DV_E_FORMATETC;
return S_OK;
}
static HRESULT STDMETHODCALLTYPE CliprdrDataObject_GetCanonicalFormatEtc(
IDataObject* This, FORMATETC* pformatectIn, FORMATETC* pformatetcOut)
{
(void)This;
(void)pformatectIn;
if (!pformatetcOut)
return E_INVALIDARG;
pformatetcOut->ptd = NULL;
return E_NOTIMPL;
}
static HRESULT STDMETHODCALLTYPE CliprdrDataObject_SetData(
IDataObject* This, FORMATETC* pformatetc, STGMEDIUM* pmedium, BOOL fRelease)
{
(void)This;
(void)pformatetc;
(void)pmedium;
(void)fRelease;
return E_NOTIMPL;
}
static HRESULT STDMETHODCALLTYPE CliprdrDataObject_EnumFormatEtc(
IDataObject* This, DWORD dwDirection, IEnumFORMATETC** ppenumFormatEtc)
{
CliprdrDataObject* instance = (CliprdrDataObject*) This;
if (!instance || !ppenumFormatEtc)
return E_INVALIDARG;
if (dwDirection == DATADIR_GET)
{
*ppenumFormatEtc = (IEnumFORMATETC*) CliprdrEnumFORMATETC_New(
instance->m_nNumFormats, instance->m_pFormatEtc);
return (*ppenumFormatEtc) ? S_OK : E_OUTOFMEMORY;
}
else
{
return E_NOTIMPL;
}
}
static HRESULT STDMETHODCALLTYPE CliprdrDataObject_DAdvise(
IDataObject* This, FORMATETC* pformatetc, DWORD advf,
IAdviseSink* pAdvSink, DWORD* pdwConnection)
{
(void)This;
(void)pformatetc;
(void)advf;
(void)pAdvSink;
(void)pdwConnection;
return OLE_E_ADVISENOTSUPPORTED;
}
static HRESULT STDMETHODCALLTYPE CliprdrDataObject_DUnadvise(
IDataObject* This, DWORD dwConnection)
{
(void)This;
(void)dwConnection;
return OLE_E_ADVISENOTSUPPORTED;
}
static HRESULT STDMETHODCALLTYPE CliprdrDataObject_EnumDAdvise(
IDataObject* This, IEnumSTATDATA** ppenumAdvise)
{
(void)This;
(void)ppenumAdvise;
return OLE_E_ADVISENOTSUPPORTED;
}
static CliprdrDataObject* CliprdrDataObject_New(
FORMATETC* fmtetc, STGMEDIUM* stgmed, ULONG count, void* data)
{
int i;
CliprdrDataObject* instance;
IDataObject* iDataObject;
instance = (CliprdrDataObject*) calloc(1, sizeof(CliprdrDataObject));
if (!instance)
goto error;
iDataObject = &instance->iDataObject;
iDataObject->lpVtbl = (IDataObjectVtbl*) calloc(1, sizeof(IDataObjectVtbl));
if (!iDataObject->lpVtbl)
goto error;
iDataObject->lpVtbl->QueryInterface = CliprdrDataObject_QueryInterface;
iDataObject->lpVtbl->AddRef = CliprdrDataObject_AddRef;
iDataObject->lpVtbl->Release = CliprdrDataObject_Release;
iDataObject->lpVtbl->GetData = CliprdrDataObject_GetData;
iDataObject->lpVtbl->GetDataHere = CliprdrDataObject_GetDataHere;
iDataObject->lpVtbl->QueryGetData = CliprdrDataObject_QueryGetData;
iDataObject->lpVtbl->GetCanonicalFormatEtc =
CliprdrDataObject_GetCanonicalFormatEtc;
iDataObject->lpVtbl->SetData = CliprdrDataObject_SetData;
iDataObject->lpVtbl->EnumFormatEtc = CliprdrDataObject_EnumFormatEtc;
iDataObject->lpVtbl->DAdvise = CliprdrDataObject_DAdvise;
iDataObject->lpVtbl->DUnadvise = CliprdrDataObject_DUnadvise;
iDataObject->lpVtbl->EnumDAdvise = CliprdrDataObject_EnumDAdvise;
instance->m_lRefCount = 1;
instance->m_nNumFormats = count;
instance->m_pData = data;
instance->m_nStreams = 0;
instance->m_pStream = NULL;
if (count > 0)
{
instance->m_pFormatEtc = (FORMATETC*) calloc(count, sizeof(FORMATETC));
if (!instance->m_pFormatEtc)
goto error;
instance->m_pStgMedium = (STGMEDIUM*) calloc(count, sizeof(STGMEDIUM));
if (!instance->m_pStgMedium)
goto error;
for (i = 0; i < count; i++)
{
instance->m_pFormatEtc[i] = fmtetc[i];
instance->m_pStgMedium[i] = stgmed[i];
}
}
return instance;
error:
CliprdrDataObject_Delete(instance);
return NULL;
}
void CliprdrDataObject_Delete(CliprdrDataObject* instance)
{
if (instance)
{
free(instance->iDataObject.lpVtbl);
free(instance->m_pFormatEtc);
free(instance->m_pStgMedium);
if (instance->m_pStream)
{
ULONG i;
for (i = 0; i < instance->m_nStreams; i++)
CliprdrStream_Release(instance->m_pStream[i]);
free(instance->m_pStream);
}
free(instance);
}
}
static BOOL wf_create_file_obj(wfClipboard* clipboard,
IDataObject** ppDataObject)
{
FORMATETC fmtetc[2];
STGMEDIUM stgmeds[2];
if (!ppDataObject)
return FALSE;
fmtetc[0].cfFormat = RegisterClipboardFormat(CFSTR_FILEDESCRIPTORW);
fmtetc[0].dwAspect = DVASPECT_CONTENT;
fmtetc[0].lindex = 0;
fmtetc[0].ptd = NULL;
fmtetc[0].tymed = TYMED_HGLOBAL;
stgmeds[0].tymed = TYMED_HGLOBAL;
stgmeds[0].hGlobal = NULL;
stgmeds[0].pUnkForRelease = NULL;
fmtetc[1].cfFormat = RegisterClipboardFormat(CFSTR_FILECONTENTS);
fmtetc[1].dwAspect = DVASPECT_CONTENT;
fmtetc[1].lindex = 0;
fmtetc[1].ptd = NULL;
fmtetc[1].tymed = TYMED_ISTREAM;
stgmeds[1].tymed = TYMED_ISTREAM;
stgmeds[1].pstm = NULL;
stgmeds[1].pUnkForRelease = NULL;
*ppDataObject = (IDataObject*) CliprdrDataObject_New(fmtetc, stgmeds, 2,
clipboard);
return (*ppDataObject) ? TRUE : FALSE;
}
static void wf_destroy_file_obj(IDataObject* instance)
{
if (instance)
IDataObject_Release(instance);
}
/**
* IEnumFORMATETC
*/
static void cliprdr_format_deep_copy(FORMATETC* dest, FORMATETC* source)
{
*dest = *source;
if (source->ptd)
{
dest->ptd = (DVTARGETDEVICE*) CoTaskMemAlloc(sizeof(DVTARGETDEVICE));
if (dest->ptd)
*(dest->ptd) = *(source->ptd);
}
}
static HRESULT STDMETHODCALLTYPE CliprdrEnumFORMATETC_QueryInterface(
IEnumFORMATETC* This, REFIID riid, void** ppvObject)
{
(void)This;
if (IsEqualIID(riid, &IID_IEnumFORMATETC) || IsEqualIID(riid, &IID_IUnknown))
{
IEnumFORMATETC_AddRef(This);
*ppvObject = This;
return S_OK;
}
else
{
*ppvObject = 0;
return E_NOINTERFACE;
}
}
static ULONG STDMETHODCALLTYPE CliprdrEnumFORMATETC_AddRef(IEnumFORMATETC* This)
{
CliprdrEnumFORMATETC* instance = (CliprdrEnumFORMATETC*) This;
if (!instance)
return 0;
return InterlockedIncrement(&instance->m_lRefCount);
}
static ULONG STDMETHODCALLTYPE CliprdrEnumFORMATETC_Release(
IEnumFORMATETC* This)
{
LONG count;
CliprdrEnumFORMATETC* instance = (CliprdrEnumFORMATETC*) This;
if (!instance)
return 0;
count = InterlockedDecrement(&instance->m_lRefCount);
if (count == 0)
{
CliprdrEnumFORMATETC_Delete(instance);
return 0;
}
else
{
return count;
}
}
static HRESULT STDMETHODCALLTYPE CliprdrEnumFORMATETC_Next(IEnumFORMATETC* This,
ULONG celt,
FORMATETC* rgelt, ULONG* pceltFetched)
{
ULONG copied = 0;
CliprdrEnumFORMATETC* instance = (CliprdrEnumFORMATETC*) This;
if (!instance || !celt || !rgelt)
return E_INVALIDARG;
while ((instance->m_nIndex < instance->m_nNumFormats) && (copied < celt))
{
cliprdr_format_deep_copy(&rgelt[copied++],
&instance->m_pFormatEtc[instance->m_nIndex++]);
}
if (pceltFetched != 0)
*pceltFetched = copied;
return (copied == celt) ? S_OK : E_FAIL;
}
static HRESULT STDMETHODCALLTYPE CliprdrEnumFORMATETC_Skip(IEnumFORMATETC* This,
ULONG celt)
{
CliprdrEnumFORMATETC* instance = (CliprdrEnumFORMATETC*) This;
if (!instance)
return E_INVALIDARG;
if (instance->m_nIndex + (LONG) celt > instance->m_nNumFormats)
return E_FAIL;
instance->m_nIndex += celt;
return S_OK;
}
static HRESULT STDMETHODCALLTYPE CliprdrEnumFORMATETC_Reset(
IEnumFORMATETC* This)
{
CliprdrEnumFORMATETC* instance = (CliprdrEnumFORMATETC*) This;
if (!instance)
return E_INVALIDARG;
instance->m_nIndex = 0;
return S_OK;
}
static HRESULT STDMETHODCALLTYPE CliprdrEnumFORMATETC_Clone(
IEnumFORMATETC* This, IEnumFORMATETC** ppEnum)
{
CliprdrEnumFORMATETC* instance = (CliprdrEnumFORMATETC*) This;
if (!instance || !ppEnum)
return E_INVALIDARG;
*ppEnum = (IEnumFORMATETC*) CliprdrEnumFORMATETC_New(instance->m_nNumFormats,
instance->m_pFormatEtc);
if (!*ppEnum)
return E_OUTOFMEMORY;
((CliprdrEnumFORMATETC*) *ppEnum)->m_nIndex = instance->m_nIndex;
return S_OK;
}
CliprdrEnumFORMATETC* CliprdrEnumFORMATETC_New(ULONG nFormats,
FORMATETC* pFormatEtc)
{
ULONG i;
CliprdrEnumFORMATETC* instance;
IEnumFORMATETC* iEnumFORMATETC;
if ((nFormats != 0) && !pFormatEtc)
return NULL;
instance = (CliprdrEnumFORMATETC*) calloc(1, sizeof(CliprdrEnumFORMATETC));
if (!instance)
goto error;
iEnumFORMATETC = &instance->iEnumFORMATETC;
iEnumFORMATETC->lpVtbl = (IEnumFORMATETCVtbl*) calloc(1,
sizeof(IEnumFORMATETCVtbl));
if (!iEnumFORMATETC->lpVtbl)
goto error;
iEnumFORMATETC->lpVtbl->QueryInterface = CliprdrEnumFORMATETC_QueryInterface;
iEnumFORMATETC->lpVtbl->AddRef = CliprdrEnumFORMATETC_AddRef;
iEnumFORMATETC->lpVtbl->Release = CliprdrEnumFORMATETC_Release;
iEnumFORMATETC->lpVtbl->Next = CliprdrEnumFORMATETC_Next;
iEnumFORMATETC->lpVtbl->Skip = CliprdrEnumFORMATETC_Skip;
iEnumFORMATETC->lpVtbl->Reset = CliprdrEnumFORMATETC_Reset;
iEnumFORMATETC->lpVtbl->Clone = CliprdrEnumFORMATETC_Clone;
instance->m_lRefCount = 1;
instance->m_nIndex = 0;
instance->m_nNumFormats = nFormats;
if (nFormats > 0)
{
instance->m_pFormatEtc = (FORMATETC*) calloc(nFormats, sizeof(FORMATETC));
if (!instance->m_pFormatEtc)
goto error;
for (i = 0; i < nFormats; i++)
cliprdr_format_deep_copy(&instance->m_pFormatEtc[i], &pFormatEtc[i]);
}
return instance;
error:
CliprdrEnumFORMATETC_Delete(instance);
return NULL;
}
void CliprdrEnumFORMATETC_Delete(CliprdrEnumFORMATETC* instance)
{
LONG i;
if (instance)
{
free(instance->iEnumFORMATETC.lpVtbl);
if (instance->m_pFormatEtc)
{
for (i = 0; i < instance->m_nNumFormats; i++)
{
if (instance->m_pFormatEtc[i].ptd)
CoTaskMemFree(instance->m_pFormatEtc[i].ptd);
}
free(instance->m_pFormatEtc);
}
free(instance);
}
}
/***********************************************************************************/
static UINT32 get_local_format_id_by_name(wfClipboard* clipboard,
const TCHAR* format_name)
{
size_t i;
formatMapping* map;
WCHAR* unicode_name;
#if !defined(UNICODE)
size_t size;
#endif
if (!clipboard || !format_name)
return 0;
#if defined(UNICODE)
unicode_name = _wcsdup(format_name);
#else
size = _tcslen(format_name);
unicode_name = calloc(size + 1, sizeof(WCHAR));
if (!unicode_name)
return 0;
MultiByteToWideChar(CP_OEMCP, 0, format_name, strlen(format_name), unicode_name,
size);
#endif
if (!unicode_name)
return 0;
for (i = 0; i < clipboard->map_size; i++)
{
map = &clipboard->format_mappings[i];
if (map->name)
{
if (wcscmp(map->name, unicode_name) == 0)
{
free(unicode_name);
return map->local_format_id;
}
}
}
free(unicode_name);
return 0;
}
static INLINE BOOL file_transferring(wfClipboard* clipboard)
{
return get_local_format_id_by_name(clipboard,
CFSTR_FILEDESCRIPTORW) ? TRUE : FALSE;
}
static UINT32 get_remote_format_id(wfClipboard* clipboard, UINT32 local_format)
{
UINT32 i;
formatMapping* map;
if (!clipboard)
return 0;
for (i = 0; i < clipboard->map_size; i++)
{
map = &clipboard->format_mappings[i];
if (map->local_format_id == local_format)
return map->remote_format_id;
}
return local_format;
}
static void map_ensure_capacity(wfClipboard* clipboard)
{
if (!clipboard)
return;
if (clipboard->map_size >= clipboard->map_capacity)
{
size_t new_size;
formatMapping* new_map;
new_size = clipboard->map_capacity * 2;
new_map = (formatMapping*) realloc(clipboard->format_mappings,
sizeof(formatMapping) * new_size);
if (!new_map)
return;
clipboard->format_mappings = new_map;
clipboard->map_capacity = new_size;
}
}
static BOOL clear_format_map(wfClipboard* clipboard)
{
size_t i;
formatMapping* map;
if (!clipboard)
return FALSE;
if (clipboard->format_mappings)
{
for (i = 0; i < clipboard->map_capacity; i++)
{
map = &clipboard->format_mappings[i];
map->remote_format_id = 0;
map->local_format_id = 0;
free(map->name);
map->name = NULL;
}
}
clipboard->map_size = 0;
return TRUE;
}
static UINT cliprdr_send_tempdir(wfClipboard* clipboard)
{
CLIPRDR_TEMP_DIRECTORY tempDirectory;
if (!clipboard)
return -1;
if (GetEnvironmentVariableA("TEMP", tempDirectory.szTempDir,
sizeof(tempDirectory.szTempDir)) == 0)
return -1;
return clipboard->context->TempDirectory(clipboard->context, &tempDirectory);
}
static BOOL cliprdr_GetUpdatedClipboardFormats(wfClipboard* clipboard,
PUINT lpuiFormats, UINT cFormats, PUINT pcFormatsOut)
{
UINT index = 0;
UINT format = 0;
BOOL clipboardOpen = FALSE;
if (!clipboard->legacyApi)
return clipboard->GetUpdatedClipboardFormats(lpuiFormats, cFormats,
pcFormatsOut);
clipboardOpen = OpenClipboard(clipboard->hwnd);
if (!clipboardOpen)
return FALSE;
while (index < cFormats)
{
format = EnumClipboardFormats(format);
if (!format)
break;
lpuiFormats[index] = format;
index++;
}
*pcFormatsOut = index;
CloseClipboard();
return TRUE;
}
static UINT cliprdr_send_format_list(wfClipboard* clipboard)
{
UINT rc;
int count;
UINT32 index;
UINT32 numFormats;
UINT32 formatId = 0;
char formatName[1024];
CLIPRDR_FORMAT* formats;
CLIPRDR_FORMAT_LIST formatList;
if (!clipboard)
return ERROR_INTERNAL_ERROR;
ZeroMemory(&formatList, sizeof(CLIPRDR_FORMAT_LIST));
if (!OpenClipboard(clipboard->hwnd))
return ERROR_INTERNAL_ERROR;
count = CountClipboardFormats();
numFormats = (UINT32) count;
formats = (CLIPRDR_FORMAT*) calloc(numFormats, sizeof(CLIPRDR_FORMAT));
if (!formats)
{
CloseClipboard();
return CHANNEL_RC_NO_MEMORY;
}
index = 0;
if (IsClipboardFormatAvailable(CF_HDROP))
{
formats[index++].formatId = RegisterClipboardFormat(CFSTR_FILEDESCRIPTORW);
formats[index++].formatId = RegisterClipboardFormat(CFSTR_FILECONTENTS);
}
else
{
while (formatId = EnumClipboardFormats(formatId))
formats[index++].formatId = formatId;
}
numFormats = index;
if (!CloseClipboard())
{
free(formats);
return ERROR_INTERNAL_ERROR;
}
for (index = 0; index < numFormats; index++)
{
if (GetClipboardFormatNameA(formats[index].formatId, formatName,
sizeof(formatName)))
{
formats[index].formatName = _strdup(formatName);
}
}
formatList.numFormats = numFormats;
formatList.formats = formats;
rc = clipboard->context->ClientFormatList(clipboard->context, &formatList);
for (index = 0; index < numFormats; index++)
free(formats[index].formatName);
free(formats);
return rc;
}
static UINT cliprdr_send_data_request(wfClipboard* clipboard, UINT32 formatId)
{
UINT rc;
CLIPRDR_FORMAT_DATA_REQUEST formatDataRequest;
if (!clipboard || !clipboard->context ||
!clipboard->context->ClientFormatDataRequest)
return ERROR_INTERNAL_ERROR;
formatDataRequest.requestedFormatId = formatId;
clipboard->requestedFormatId = formatId;
rc = clipboard->context->ClientFormatDataRequest(clipboard->context,
&formatDataRequest);
if (WaitForSingleObject(clipboard->response_data_event,
INFINITE) != WAIT_OBJECT_0)
rc = ERROR_INTERNAL_ERROR;
else if (!ResetEvent(clipboard->response_data_event))
rc = ERROR_INTERNAL_ERROR;
return rc;
}
UINT cliprdr_send_request_filecontents(wfClipboard* clipboard,
const void* streamid,
ULONG index, UINT32 flag, DWORD positionhigh,
DWORD positionlow, ULONG nreq)
{
UINT rc;
CLIPRDR_FILE_CONTENTS_REQUEST fileContentsRequest;
if (!clipboard || !clipboard->context ||
!clipboard->context->ClientFileContentsRequest)
return ERROR_INTERNAL_ERROR;
fileContentsRequest.streamId = (UINT32) streamid;
fileContentsRequest.listIndex = index;
fileContentsRequest.dwFlags = flag;
fileContentsRequest.nPositionLow = positionlow;
fileContentsRequest.nPositionHigh = positionhigh;
fileContentsRequest.cbRequested = nreq;
fileContentsRequest.clipDataId = 0;
fileContentsRequest.msgFlags = 0;
rc = clipboard->context->ClientFileContentsRequest(clipboard->context,
&fileContentsRequest);
if (WaitForSingleObject(clipboard->req_fevent, INFINITE) != WAIT_OBJECT_0)
rc = ERROR_INTERNAL_ERROR;
else if (!ResetEvent(clipboard->req_fevent))
rc = ERROR_INTERNAL_ERROR;
return rc;
}
static UINT cliprdr_send_response_filecontents(wfClipboard* clipboard,
UINT32 streamId, UINT32 size, BYTE* data)
{
CLIPRDR_FILE_CONTENTS_RESPONSE fileContentsResponse;
if (!clipboard || !clipboard->context ||
!clipboard->context->ClientFileContentsResponse)
return ERROR_INTERNAL_ERROR;
fileContentsResponse.streamId = streamId;
fileContentsResponse.cbRequested = size;
fileContentsResponse.requestedData = data;
fileContentsResponse.msgFlags = CB_RESPONSE_OK;
return clipboard->context->ClientFileContentsResponse(clipboard->context,
&fileContentsResponse);
}
static LRESULT CALLBACK cliprdr_proc(HWND hWnd, UINT Msg, WPARAM wParam,
LPARAM lParam)
{
static wfClipboard* clipboard = NULL;
switch (Msg)
{
case WM_CREATE:
DEBUG_CLIPRDR("info: WM_CREATE");
clipboard = (wfClipboard*)((CREATESTRUCT*) lParam)->lpCreateParams;
clipboard->hwnd = hWnd;
if (!clipboard->legacyApi)
clipboard->AddClipboardFormatListener(hWnd);
else
clipboard->hWndNextViewer = SetClipboardViewer(hWnd);
break;
case WM_CLOSE:
DEBUG_CLIPRDR("info: WM_CLOSE");
if (!clipboard->legacyApi)
clipboard->RemoveClipboardFormatListener(hWnd);
break;
case WM_DESTROY:
if (clipboard->legacyApi)
ChangeClipboardChain(hWnd, clipboard->hWndNextViewer);
break;
case WM_CLIPBOARDUPDATE:
DEBUG_CLIPRDR("info: WM_CLIPBOARDUPDATE");
if (clipboard->sync)
{
if ((GetClipboardOwner() != clipboard->hwnd) &&
(S_FALSE == OleIsCurrentClipboard(clipboard->data_obj)))
{
if (clipboard->hmem)
{
GlobalFree(clipboard->hmem);
clipboard->hmem = NULL;
}
cliprdr_send_format_list(clipboard);
}
}
break;
case WM_RENDERALLFORMATS:
DEBUG_CLIPRDR("info: WM_RENDERALLFORMATS");
/* discard all contexts in clipboard */
if (!OpenClipboard(clipboard->hwnd))
{
DEBUG_CLIPRDR("OpenClipboard failed with 0x%x", GetLastError());
break;
}
EmptyClipboard();
CloseClipboard();
break;
case WM_RENDERFORMAT:
DEBUG_CLIPRDR("info: WM_RENDERFORMAT");
if (cliprdr_send_data_request(clipboard, (UINT32) wParam) != 0)
{
DEBUG_CLIPRDR("error: cliprdr_send_data_request failed.");
break;
}
if (!SetClipboardData((UINT) wParam, clipboard->hmem))
{
DEBUG_CLIPRDR("SetClipboardData failed with 0x%x", GetLastError());
if (clipboard->hmem)
{
GlobalFree(clipboard->hmem);
clipboard->hmem = NULL;
}
}
/* Note: GlobalFree() is not needed when success */
break;
case WM_DRAWCLIPBOARD:
if (clipboard->legacyApi)
{
if ((GetClipboardOwner() != clipboard->hwnd) &&
(S_FALSE == OleIsCurrentClipboard(clipboard->data_obj)))
{
cliprdr_send_format_list(clipboard);
}
SendMessage(clipboard->hWndNextViewer, Msg, wParam, lParam);
}
break;
case WM_CHANGECBCHAIN:
if (clipboard->legacyApi)
{
HWND hWndCurrViewer = (HWND) wParam;
HWND hWndNextViewer = (HWND) lParam;
if (hWndCurrViewer == clipboard->hWndNextViewer)
clipboard->hWndNextViewer = hWndNextViewer;
else if (clipboard->hWndNextViewer)
SendMessage(clipboard->hWndNextViewer, Msg, wParam, lParam);
}
break;
case WM_CLIPRDR_MESSAGE:
DEBUG_CLIPRDR("info: WM_CLIPRDR_MESSAGE");
switch (wParam)
{
case OLE_SETCLIPBOARD:
DEBUG_CLIPRDR("info: OLE_SETCLIPBOARD");
if (S_FALSE == OleIsCurrentClipboard(clipboard->data_obj))
{
if (wf_create_file_obj(clipboard, &clipboard->data_obj))
{
if (OleSetClipboard(clipboard->data_obj) != S_OK)
{
wf_destroy_file_obj(clipboard->data_obj);
clipboard->data_obj = NULL;
}
}
}
break;
default:
break;
}
break;
case WM_DESTROYCLIPBOARD:
case WM_ASKCBFORMATNAME:
case WM_HSCROLLCLIPBOARD:
case WM_PAINTCLIPBOARD:
case WM_SIZECLIPBOARD:
case WM_VSCROLLCLIPBOARD:
default:
return DefWindowProc(hWnd, Msg, wParam, lParam);
}
return 0;
}
static int create_cliprdr_window(wfClipboard* clipboard)
{
WNDCLASSEX wnd_cls;
ZeroMemory(&wnd_cls, sizeof(WNDCLASSEX));
wnd_cls.cbSize = sizeof(WNDCLASSEX);
wnd_cls.style = CS_OWNDC;
wnd_cls.lpfnWndProc = cliprdr_proc;
wnd_cls.cbClsExtra = 0;
wnd_cls.cbWndExtra = 0;
wnd_cls.hIcon = NULL;
wnd_cls.hCursor = NULL;
wnd_cls.hbrBackground = NULL;
wnd_cls.lpszMenuName = NULL;
wnd_cls.lpszClassName = _T("ClipboardHiddenMessageProcessor");
wnd_cls.hInstance = GetModuleHandle(NULL);
wnd_cls.hIconSm = NULL;
RegisterClassEx(&wnd_cls);
clipboard->hwnd = CreateWindowEx(WS_EX_LEFT,
_T("ClipboardHiddenMessageProcessor"),
_T("rdpclip"), 0, 0, 0, 0, 0, HWND_MESSAGE, NULL, GetModuleHandle(NULL),
clipboard);
if (!clipboard->hwnd)
{
DEBUG_CLIPRDR("error: CreateWindowEx failed with %x.", GetLastError());
return -1;
}
return 0;
}
static DWORD WINAPI cliprdr_thread_func(LPVOID arg)
{
int ret;
MSG msg;
BOOL mcode;
wfClipboard* clipboard = (wfClipboard*) arg;
OleInitialize(0);
if ((ret = create_cliprdr_window(clipboard)) != 0)
{
DEBUG_CLIPRDR("error: create clipboard window failed.");
return 0;
}
while ((mcode = GetMessage(&msg, 0, 0, 0)) != 0)
{
if (mcode == -1)
{
DEBUG_CLIPRDR("error: clipboard thread GetMessage failed.");
break;
}
else
{
TranslateMessage(&msg);
DispatchMessage(&msg);
}
}
OleUninitialize();
return 0;
}
static void clear_file_array(wfClipboard* clipboard)
{
size_t i;
if (!clipboard)
return;
/* clear file_names array */
if (clipboard->file_names)
{
for (i = 0; i < clipboard->nFiles; i++)
{
free(clipboard->file_names[i]);
clipboard->file_names[i] = NULL;
}
free(clipboard->file_names);
clipboard->file_names = NULL;
}
/* clear fileDescriptor array */
if (clipboard->fileDescriptor)
{
for (i = 0; i < clipboard->nFiles; i++)
{
free(clipboard->fileDescriptor[i]);
clipboard->fileDescriptor[i] = NULL;
}
free(clipboard->fileDescriptor);
clipboard->fileDescriptor = NULL;
}
clipboard->file_array_size = 0;
clipboard->nFiles = 0;
}
static BOOL wf_cliprdr_get_file_contents(WCHAR* file_name, BYTE* buffer,
LONG positionLow, LONG positionHigh,
DWORD nRequested, DWORD* puSize)
{
BOOL res = FALSE;
HANDLE hFile;
DWORD nGet, rc;
if (!file_name || !buffer || !puSize)
{
WLog_ERR(TAG, "get file contents Invalid Arguments.");
return FALSE;
}
hFile = CreateFileW(file_name, GENERIC_READ, FILE_SHARE_READ, NULL,
OPEN_EXISTING,
FILE_ATTRIBUTE_NORMAL | FILE_FLAG_BACKUP_SEMANTICS, NULL);
if (hFile == INVALID_HANDLE_VALUE)
return FALSE;
rc = SetFilePointer(hFile, positionLow, &positionHigh, FILE_BEGIN);
if (rc == INVALID_SET_FILE_POINTER)
goto error;
if (!ReadFile(hFile, buffer, nRequested, &nGet, NULL))
{
DEBUG_CLIPRDR("ReadFile failed with 0x%08lX.", GetLastError());
goto error;
}
res = TRUE;
error:
if (!CloseHandle(hFile))
res = FALSE;
if (res)
*puSize = nGet;
return res;
}
/* path_name has a '\' at the end. e.g. c:\newfolder\, file_name is c:\newfolder\new.txt */
static FILEDESCRIPTORW* wf_cliprdr_get_file_descriptor(WCHAR* file_name,
size_t pathLen)
{
HANDLE hFile;
FILEDESCRIPTORW* fd;
fd = (FILEDESCRIPTORW*) calloc(1, sizeof(FILEDESCRIPTORW));
if (!fd)
return NULL;
hFile = CreateFileW(file_name, GENERIC_READ, FILE_SHARE_READ,
NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL | FILE_FLAG_BACKUP_SEMANTICS, NULL);
if (hFile == INVALID_HANDLE_VALUE)
{
free(fd);
return NULL;
}
fd->dwFlags = FD_ATTRIBUTES | FD_FILESIZE | FD_WRITESTIME | FD_PROGRESSUI;
fd->dwFileAttributes = GetFileAttributes(file_name);
if (!GetFileTime(hFile, NULL, NULL, &fd->ftLastWriteTime))
{
fd->dwFlags &= ~FD_WRITESTIME;
}
fd->nFileSizeLow = GetFileSize(hFile, &fd->nFileSizeHigh);
wcscpy_s(fd->cFileName, sizeof(fd->cFileName) / 2, file_name + pathLen);
CloseHandle(hFile);
return fd;
}
static BOOL wf_cliprdr_array_ensure_capacity(wfClipboard* clipboard)
{
if (!clipboard)
return FALSE;
if (clipboard->nFiles == clipboard->file_array_size)
{
size_t new_size;
FILEDESCRIPTORW** new_fd;
WCHAR** new_name;
new_size = (clipboard->file_array_size + 1) * 2;
new_fd = (FILEDESCRIPTORW**) realloc(clipboard->fileDescriptor,
new_size * sizeof(FILEDESCRIPTORW*));
if (new_fd)
clipboard->fileDescriptor = new_fd;
new_name = (WCHAR**) realloc(clipboard->file_names, new_size * sizeof(WCHAR*));
if (new_name)
clipboard->file_names = new_name;
if (!new_fd || !new_name)
return FALSE;
clipboard->file_array_size = new_size;
}
return TRUE;
}
static BOOL wf_cliprdr_add_to_file_arrays(wfClipboard* clipboard,
WCHAR* full_file_name, size_t pathLen)
{
if (!wf_cliprdr_array_ensure_capacity(clipboard))
return FALSE;
/* add to name array */
clipboard->file_names[clipboard->nFiles] = (LPWSTR) malloc(MAX_PATH * 2);
if (!clipboard->file_names[clipboard->nFiles])
return FALSE;
wcscpy_s(clipboard->file_names[clipboard->nFiles], MAX_PATH, full_file_name);
/* add to descriptor array */
clipboard->fileDescriptor[clipboard->nFiles] =
wf_cliprdr_get_file_descriptor(full_file_name, pathLen);
if (!clipboard->fileDescriptor[clipboard->nFiles])
{
free(clipboard->file_names[clipboard->nFiles]);
return FALSE;
}
clipboard->nFiles++;
return TRUE;
}
static BOOL wf_cliprdr_traverse_directory(wfClipboard* clipboard,
WCHAR* Dir, size_t pathLen)
{
HANDLE hFind;
WCHAR DirSpec[MAX_PATH];
WIN32_FIND_DATA FindFileData;
if (!clipboard || !Dir)
return FALSE;
StringCchCopy(DirSpec, MAX_PATH, Dir);
StringCchCat(DirSpec, MAX_PATH, TEXT("\\*"));
hFind = FindFirstFile(DirSpec, &FindFileData);
if (hFind == INVALID_HANDLE_VALUE)
{
DEBUG_CLIPRDR("FindFirstFile failed with 0x%x.", GetLastError());
return FALSE;
}
while (FindNextFile(hFind, &FindFileData))
{
if ((FindFileData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) != 0
&& wcscmp(FindFileData.cFileName, _T(".")) == 0
|| wcscmp(FindFileData.cFileName, _T("..")) == 0)
{
continue;
}
if ((FindFileData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) != 0)
{
WCHAR DirAdd[MAX_PATH];
StringCchCopy(DirAdd, MAX_PATH, Dir);
StringCchCat(DirAdd, MAX_PATH, _T("\\"));
StringCchCat(DirAdd, MAX_PATH, FindFileData.cFileName);
if (!wf_cliprdr_add_to_file_arrays(clipboard, DirAdd, pathLen))
return FALSE;
if (!wf_cliprdr_traverse_directory(clipboard, DirAdd, pathLen))
return FALSE;
}
else
{
WCHAR fileName[MAX_PATH];
StringCchCopy(fileName, MAX_PATH, Dir);
StringCchCat(fileName, MAX_PATH, _T("\\"));
StringCchCat(fileName, MAX_PATH, FindFileData.cFileName);
if (!wf_cliprdr_add_to_file_arrays(clipboard, fileName, pathLen))
return FALSE;
}
}
FindClose(hFind);
return TRUE;
}
static UINT wf_cliprdr_send_client_capabilities(wfClipboard* clipboard)
{
CLIPRDR_CAPABILITIES capabilities;
CLIPRDR_GENERAL_CAPABILITY_SET generalCapabilitySet;
if (!clipboard || !clipboard->context ||
!clipboard->context->ClientCapabilities)
return ERROR_INTERNAL_ERROR;
capabilities.cCapabilitiesSets = 1;
capabilities.capabilitySets = (CLIPRDR_CAPABILITY_SET*) &
(generalCapabilitySet);
generalCapabilitySet.capabilitySetType = CB_CAPSTYPE_GENERAL;
generalCapabilitySet.capabilitySetLength = 12;
generalCapabilitySet.version = CB_CAPS_VERSION_2;
generalCapabilitySet.generalFlags =
CB_USE_LONG_FORMAT_NAMES |
CB_STREAM_FILECLIP_ENABLED |
CB_FILECLIP_NO_FILE_PATHS;
return clipboard->context->ClientCapabilities(clipboard->context,
&capabilities);
}
/**
* Function description
*
* @return 0 on success, otherwise a Win32 error code
*/
static UINT wf_cliprdr_monitor_ready(CliprdrClientContext* context,
const CLIPRDR_MONITOR_READY* monitorReady)
{
UINT rc;
wfClipboard* clipboard = (wfClipboard*) context->custom;
if (!context || !monitorReady)
return ERROR_INTERNAL_ERROR;
clipboard->sync = TRUE;
rc = wf_cliprdr_send_client_capabilities(clipboard);
if (rc != CHANNEL_RC_OK)
return rc;
return cliprdr_send_format_list(clipboard);
}
/**
* Function description
*
* @return 0 on success, otherwise a Win32 error code
*/
static UINT wf_cliprdr_server_capabilities(CliprdrClientContext* context,
const CLIPRDR_CAPABILITIES* capabilities)
{
UINT32 index;
CLIPRDR_CAPABILITY_SET* capabilitySet;
wfClipboard* clipboard = (wfClipboard*) context->custom;
if (!context || !capabilities)
return ERROR_INTERNAL_ERROR;
for (index = 0; index < capabilities->cCapabilitiesSets; index++)
{
capabilitySet = &(capabilities->capabilitySets[index]);
if ((capabilitySet->capabilitySetType == CB_CAPSTYPE_GENERAL) &&
(capabilitySet->capabilitySetLength >= CB_CAPSTYPE_GENERAL_LEN))
{
CLIPRDR_GENERAL_CAPABILITY_SET* generalCapabilitySet
= (CLIPRDR_GENERAL_CAPABILITY_SET*) capabilitySet;
clipboard->capabilities = generalCapabilitySet->generalFlags;
break;
}
}
return CHANNEL_RC_OK;
}
/**
* Function description
*
* @return 0 on success, otherwise a Win32 error code
*/
static UINT wf_cliprdr_server_format_list(CliprdrClientContext* context,
const CLIPRDR_FORMAT_LIST* formatList)
{
UINT rc = ERROR_INTERNAL_ERROR;
UINT32 i;
formatMapping* mapping;
CLIPRDR_FORMAT* format;
wfClipboard* clipboard = (wfClipboard*) context->custom;
if (!clear_format_map(clipboard))
return ERROR_INTERNAL_ERROR;
for (i = 0; i < formatList->numFormats; i++)
{
format = &(formatList->formats[i]);
mapping = &(clipboard->format_mappings[i]);
mapping->remote_format_id = format->formatId;
if (format->formatName)
{
int size = MultiByteToWideChar(CP_UTF8, 0, format->formatName,
strlen(format->formatName),
NULL, 0);
mapping->name = calloc(size + 1, sizeof(WCHAR));
if (mapping->name)
{
MultiByteToWideChar(CP_UTF8, 0, format->formatName, strlen(format->formatName),
mapping->name, size);
mapping->local_format_id = RegisterClipboardFormatW((LPWSTR) mapping->name);
}
}
else
{
mapping->name = NULL;
mapping->local_format_id = mapping->remote_format_id;
}
clipboard->map_size++;
map_ensure_capacity(clipboard);
}
if (file_transferring(clipboard))
{
if (PostMessage(clipboard->hwnd, WM_CLIPRDR_MESSAGE, OLE_SETCLIPBOARD, 0))
rc = CHANNEL_RC_OK;
}
else
{
if (!OpenClipboard(clipboard->hwnd))
return ERROR_INTERNAL_ERROR;
if (EmptyClipboard())
{
for (i = 0; i < (UINT32) clipboard->map_size; i++)
SetClipboardData(clipboard->format_mappings[i].local_format_id, NULL);
rc = CHANNEL_RC_OK;
}
if (!CloseClipboard() && GetLastError())
return ERROR_INTERNAL_ERROR;
}
return rc;
}
/**
* Function description
*
* @return 0 on success, otherwise a Win32 error code
*/
static UINT wf_cliprdr_server_format_list_response(CliprdrClientContext*
context,
const CLIPRDR_FORMAT_LIST_RESPONSE* formatListResponse)
{
(void)context;
(void)formatListResponse;
if (formatListResponse->msgFlags != CB_RESPONSE_OK)
return E_FAIL;
return CHANNEL_RC_OK;
}
/**
* Function description
*
* @return 0 on success, otherwise a Win32 error code
*/
static UINT wf_cliprdr_server_lock_clipboard_data(CliprdrClientContext* context,
const CLIPRDR_LOCK_CLIPBOARD_DATA* lockClipboardData)
{
(void)context;
(void)lockClipboardData;
return CHANNEL_RC_OK;
}
/**
* Function description
*
* @return 0 on success, otherwise a Win32 error code
*/
static UINT wf_cliprdr_server_unlock_clipboard_data(CliprdrClientContext*
context,
const CLIPRDR_UNLOCK_CLIPBOARD_DATA* unlockClipboardData)
{
(void)context;
(void)unlockClipboardData;
return CHANNEL_RC_OK;
}
static BOOL wf_cliprdr_process_filename(wfClipboard* clipboard,
WCHAR* wFileName, size_t str_len)
{
size_t pathLen;
size_t offset = str_len;
if (!clipboard || !wFileName)
return FALSE;
/* find the last '\' in full file name */
while (offset > 0)
{
if (wFileName[offset] == L'\\')
break;
else
offset--;
}
pathLen = offset + 1;
if (!wf_cliprdr_add_to_file_arrays(clipboard, wFileName, pathLen))
return FALSE;
if ((clipboard->fileDescriptor[clipboard->nFiles - 1]->dwFileAttributes&
FILE_ATTRIBUTE_DIRECTORY) != 0)
{
/* this is a directory */
if (!wf_cliprdr_traverse_directory(clipboard, wFileName, pathLen))
return FALSE;
}
return TRUE;
}
/**
* Function description
*
* @return 0 on success, otherwise a Win32 error code
*/
static UINT wf_cliprdr_server_format_data_request(CliprdrClientContext* context,
const CLIPRDR_FORMAT_DATA_REQUEST* formatDataRequest)
{
UINT rc;
size_t size = 0;
void* buff = NULL;
char* globlemem = NULL;
HANDLE hClipdata = NULL;
UINT32 requestedFormatId;
CLIPRDR_FORMAT_DATA_RESPONSE response;
wfClipboard* clipboard;
if (!context || !formatDataRequest)
return ERROR_INTERNAL_ERROR;
clipboard = (wfClipboard*) context->custom;
if (!clipboard)
return ERROR_INTERNAL_ERROR;
requestedFormatId = formatDataRequest->requestedFormatId;
if (requestedFormatId == RegisterClipboardFormat(CFSTR_FILEDESCRIPTORW))
{
size_t len;
size_t i;
WCHAR* wFileName;
HRESULT result;
LPDATAOBJECT dataObj;
FORMATETC format_etc;
STGMEDIUM stg_medium;
DROPFILES* dropFiles;
FILEGROUPDESCRIPTORW* groupDsc;
result = OleGetClipboard(&dataObj);
if (FAILED(result))
return ERROR_INTERNAL_ERROR;
ZeroMemory(&format_etc, sizeof(FORMATETC));
ZeroMemory(&stg_medium, sizeof(STGMEDIUM));
/* get DROPFILES struct from OLE */
format_etc.cfFormat = CF_HDROP;
format_etc.tymed = TYMED_HGLOBAL;
format_etc.dwAspect = 1;
format_etc.lindex = -1;
result = IDataObject_GetData(dataObj, &format_etc, &stg_medium);
if (FAILED(result))
{
DEBUG_CLIPRDR("dataObj->GetData failed.");
goto exit;
}
dropFiles = (DROPFILES*) GlobalLock(stg_medium.hGlobal);
if (!dropFiles)
{
GlobalUnlock(stg_medium.hGlobal);
ReleaseStgMedium(&stg_medium);
clipboard->nFiles = 0;
goto exit;
}
clear_file_array(clipboard);
if (dropFiles->fWide)
{
/* dropFiles contains file names */
for (wFileName = (WCHAR*)((char*)dropFiles + dropFiles->pFiles);
(len = wcslen(wFileName)) > 0; wFileName += len + 1)
{
wf_cliprdr_process_filename(clipboard, wFileName, wcslen(wFileName));
}
}
else
{
char* p;
for (p = (char*)((char*)dropFiles + dropFiles->pFiles);
(len = strlen(p)) > 0; p += len + 1, clipboard->nFiles++)
{
int cchWideChar;
WCHAR* wFileName;
cchWideChar = MultiByteToWideChar(CP_ACP, MB_COMPOSITE, p, len, NULL, 0);
wFileName = (LPWSTR) calloc(cchWideChar, sizeof(WCHAR));
MultiByteToWideChar(CP_ACP, MB_COMPOSITE, p, len, wFileName, cchWideChar);
wf_cliprdr_process_filename(clipboard, wFileName, cchWideChar);
}
}
GlobalUnlock(stg_medium.hGlobal);
ReleaseStgMedium(&stg_medium);
exit:
size = 4 + clipboard->nFiles * sizeof(FILEDESCRIPTORW);
groupDsc = (FILEGROUPDESCRIPTORW*) malloc(size);
if (groupDsc)
{
groupDsc->cItems = clipboard->nFiles;
for (i = 0; i < clipboard->nFiles; i++)
{
if (clipboard->fileDescriptor[i])
groupDsc->fgd[i] = *clipboard->fileDescriptor[i];
}
buff = groupDsc;
}
IDataObject_Release(dataObj);
}
else
{
if (!OpenClipboard(clipboard->hwnd))
return ERROR_INTERNAL_ERROR;
hClipdata = GetClipboardData(requestedFormatId);
if (!hClipdata)
{
CloseClipboard();
return ERROR_INTERNAL_ERROR;
}
globlemem = (char*) GlobalLock(hClipdata);
size = (int) GlobalSize(hClipdata);
buff = malloc(size);
CopyMemory(buff, globlemem, size);
GlobalUnlock(hClipdata);
CloseClipboard();
}
response.msgFlags = CB_RESPONSE_OK;
response.dataLen = size;
response.requestedFormatData = (BYTE*) buff;
rc = clipboard->context->ClientFormatDataResponse(clipboard->context,
&response);
free(buff);
return rc;
}
/**
* Function description
*
* @return 0 on success, otherwise a Win32 error code
*/
static UINT wf_cliprdr_server_format_data_response(CliprdrClientContext*
context,
const CLIPRDR_FORMAT_DATA_RESPONSE* formatDataResponse)
{
BYTE* data;
HANDLE hMem;
wfClipboard* clipboard;
if (!context || !formatDataResponse)
return ERROR_INTERNAL_ERROR;
if (formatDataResponse->msgFlags != CB_RESPONSE_OK)
return E_FAIL;
clipboard = (wfClipboard*) context->custom;
if (!clipboard)
return ERROR_INTERNAL_ERROR;
hMem = GlobalAlloc(GMEM_MOVEABLE, formatDataResponse->dataLen);
if (!hMem)
return ERROR_INTERNAL_ERROR;
data = (BYTE*) GlobalLock(hMem);
if (!data)
{
GlobalFree(hMem);
return ERROR_INTERNAL_ERROR;
}
CopyMemory(data, formatDataResponse->requestedFormatData,
formatDataResponse->dataLen);
if (!GlobalUnlock(hMem) && GetLastError())
{
GlobalFree(hMem);
return ERROR_INTERNAL_ERROR;
}
clipboard->hmem = hMem;
if (!SetEvent(clipboard->response_data_event))
return ERROR_INTERNAL_ERROR;
return CHANNEL_RC_OK;
}
/**
* Function description
*
* @return 0 on success, otherwise a Win32 error code
*/
static UINT wf_cliprdr_server_file_contents_request(CliprdrClientContext*
context,
const CLIPRDR_FILE_CONTENTS_REQUEST* fileContentsRequest)
{
DWORD uSize = 0;
BYTE* pData = NULL;
HRESULT hRet = S_OK;
FORMATETC vFormatEtc;
LPDATAOBJECT pDataObj = NULL;
STGMEDIUM vStgMedium;
BOOL bIsStreamFile = TRUE;
static LPSTREAM pStreamStc = NULL;
static UINT32 uStreamIdStc = 0;
wfClipboard* clipboard;
UINT rc = ERROR_INTERNAL_ERROR;
UINT sRc;
UINT32 cbRequested;
if (!context || !fileContentsRequest)
return ERROR_INTERNAL_ERROR;
clipboard = (wfClipboard*) context->custom;
if (!clipboard)
return ERROR_INTERNAL_ERROR;
cbRequested = fileContentsRequest->cbRequested;
if (fileContentsRequest->dwFlags == FILECONTENTS_SIZE)
cbRequested = sizeof(UINT64);
pData = (BYTE*) calloc(1, cbRequested);
if (!pData)
goto error;
hRet = OleGetClipboard(&pDataObj);
if (FAILED(hRet))
{
WLog_ERR(TAG, "filecontents: get ole clipboard failed.");
goto error;
}
ZeroMemory(&vFormatEtc, sizeof(FORMATETC));
ZeroMemory(&vStgMedium, sizeof(STGMEDIUM));
vFormatEtc.cfFormat = RegisterClipboardFormat(CFSTR_FILECONTENTS);
vFormatEtc.tymed = TYMED_ISTREAM;
vFormatEtc.dwAspect = 1;
vFormatEtc.lindex = fileContentsRequest->listIndex;
vFormatEtc.ptd = NULL;
if ((uStreamIdStc != fileContentsRequest->streamId) || !pStreamStc)
{
LPENUMFORMATETC pEnumFormatEtc;
ULONG CeltFetched;
FORMATETC vFormatEtc2;
if (pStreamStc)
{
IStream_Release(pStreamStc);
pStreamStc = NULL;
}
bIsStreamFile = FALSE;
hRet = IDataObject_EnumFormatEtc(pDataObj, DATADIR_GET, &pEnumFormatEtc);
if (hRet == S_OK)
{
do
{
hRet = IEnumFORMATETC_Next(pEnumFormatEtc, 1, &vFormatEtc2, &CeltFetched);
if (hRet == S_OK)
{
if (vFormatEtc2.cfFormat == RegisterClipboardFormat(CFSTR_FILECONTENTS))
{
hRet = IDataObject_GetData(pDataObj, &vFormatEtc, &vStgMedium);
if (hRet == S_OK)
{
pStreamStc = vStgMedium.pstm;
uStreamIdStc = fileContentsRequest->streamId;
bIsStreamFile = TRUE;
}
break;
}
}
}
while (hRet == S_OK);
}
}
if (bIsStreamFile == TRUE)
{
if (fileContentsRequest->dwFlags == FILECONTENTS_SIZE)
{
STATSTG vStatStg;
ZeroMemory(&vStatStg, sizeof(STATSTG));
hRet = IStream_Stat(pStreamStc, &vStatStg, STATFLAG_NONAME);
if (hRet == S_OK)
{
*((UINT32*) &pData[0]) = vStatStg.cbSize.LowPart;
*((UINT32*) &pData[4]) = vStatStg.cbSize.HighPart;
uSize = cbRequested;
}
}
else if (fileContentsRequest->dwFlags == FILECONTENTS_RANGE)
{
LARGE_INTEGER dlibMove;
ULARGE_INTEGER dlibNewPosition;
dlibMove.HighPart = fileContentsRequest->nPositionHigh;
dlibMove.LowPart = fileContentsRequest->nPositionLow;
hRet = IStream_Seek(pStreamStc, dlibMove, STREAM_SEEK_SET, &dlibNewPosition);
if (SUCCEEDED(hRet))
hRet = IStream_Read(pStreamStc, pData, cbRequested,
(PULONG) &uSize);
}
}
else
{
if (fileContentsRequest->dwFlags == FILECONTENTS_SIZE)
{
*((UINT32*) &pData[0]) =
clipboard->fileDescriptor[fileContentsRequest->listIndex]->nFileSizeLow;
*((UINT32*) &pData[4]) =
clipboard->fileDescriptor[fileContentsRequest->listIndex]->nFileSizeHigh;
uSize = cbRequested;
}
else if (fileContentsRequest->dwFlags == FILECONTENTS_RANGE)
{
BOOL bRet;
bRet = wf_cliprdr_get_file_contents(
clipboard->file_names[fileContentsRequest->listIndex], pData,
fileContentsRequest->nPositionLow, fileContentsRequest->nPositionHigh,
cbRequested, &uSize);
if (bRet == FALSE)
{
WLog_ERR(TAG, "get file contents failed.");
uSize = 0;
goto error;
}
}
}
rc = CHANNEL_RC_OK;
error:
if (pDataObj)
IDataObject_Release(pDataObj);
if (uSize == 0)
{
free(pData);
pData = NULL;
}
sRc = cliprdr_send_response_filecontents(clipboard,
fileContentsRequest->streamId,
uSize, pData);
free(pData);
if (sRc != CHANNEL_RC_OK)
return sRc;
return rc;
}
/**
* Function description
*
* @return 0 on success, otherwise a Win32 error code
*/
static UINT wf_cliprdr_server_file_contents_response(CliprdrClientContext*
context,
const CLIPRDR_FILE_CONTENTS_RESPONSE* fileContentsResponse)
{
wfClipboard* clipboard;
if (!context || !fileContentsResponse)
return ERROR_INTERNAL_ERROR;
if (fileContentsResponse->msgFlags != CB_RESPONSE_OK)
return E_FAIL;
clipboard = (wfClipboard*) context->custom;
if (!clipboard)
return ERROR_INTERNAL_ERROR;
clipboard->req_fsize = fileContentsResponse->cbRequested;
clipboard->req_fdata = (char*) malloc(fileContentsResponse->cbRequested);
if (!clipboard->req_fdata)
return ERROR_INTERNAL_ERROR;
CopyMemory(clipboard->req_fdata, fileContentsResponse->requestedData,
fileContentsResponse->cbRequested);
if (!SetEvent(clipboard->req_fevent))
{
free(clipboard->req_fdata);
return ERROR_INTERNAL_ERROR;
}
return CHANNEL_RC_OK;
}
BOOL wf_cliprdr_init(wfContext* wfc, CliprdrClientContext* cliprdr)
{
wfClipboard* clipboard;
rdpContext* context = (rdpContext*) wfc;
if (!context || !cliprdr)
return FALSE;
wfc->clipboard = (wfClipboard*) calloc(1, sizeof(wfClipboard));
if (!wfc->clipboard)
return FALSE;
clipboard = wfc->clipboard;
clipboard->wfc = wfc;
clipboard->context = cliprdr;
clipboard->channels = context->channels;
clipboard->sync = FALSE;
clipboard->map_capacity = 32;
clipboard->map_size = 0;
clipboard->hUser32 = LoadLibraryA("user32.dll");
if (clipboard->hUser32)
{
clipboard->AddClipboardFormatListener = (fnAddClipboardFormatListener)
GetProcAddress(clipboard->hUser32, "AddClipboardFormatListener");
clipboard->RemoveClipboardFormatListener = (fnRemoveClipboardFormatListener)
GetProcAddress(clipboard->hUser32, "RemoveClipboardFormatListener");
clipboard->GetUpdatedClipboardFormats = (fnGetUpdatedClipboardFormats)
GetProcAddress(clipboard->hUser32, "GetUpdatedClipboardFormats");
}
if (!(clipboard->hUser32 && clipboard->AddClipboardFormatListener
&& clipboard->RemoveClipboardFormatListener
&& clipboard->GetUpdatedClipboardFormats))
clipboard->legacyApi = TRUE;
if (!(clipboard->format_mappings = (formatMapping*) calloc(clipboard->map_capacity,
sizeof(formatMapping))))
goto error;
if (!(clipboard->response_data_event = CreateEvent(NULL, TRUE, FALSE,
_T("response_data_event"))))
goto error;
if (!(clipboard->req_fevent = CreateEvent(NULL, TRUE, FALSE,
_T("request_filecontents_event"))))
goto error;
if (!(clipboard->thread = CreateThread(NULL, 0, cliprdr_thread_func, clipboard, 0, NULL)))
goto error;
cliprdr->MonitorReady = wf_cliprdr_monitor_ready;
cliprdr->ServerCapabilities = wf_cliprdr_server_capabilities;
cliprdr->ServerFormatList = wf_cliprdr_server_format_list;
cliprdr->ServerFormatListResponse = wf_cliprdr_server_format_list_response;
cliprdr->ServerLockClipboardData = wf_cliprdr_server_lock_clipboard_data;
cliprdr->ServerUnlockClipboardData = wf_cliprdr_server_unlock_clipboard_data;
cliprdr->ServerFormatDataRequest = wf_cliprdr_server_format_data_request;
cliprdr->ServerFormatDataResponse = wf_cliprdr_server_format_data_response;
cliprdr->ServerFileContentsRequest = wf_cliprdr_server_file_contents_request;
cliprdr->ServerFileContentsResponse = wf_cliprdr_server_file_contents_response;
cliprdr->custom = (void*) wfc->clipboard;
return TRUE;
error:
wf_cliprdr_uninit(wfc, cliprdr);
return FALSE;
}
BOOL wf_cliprdr_uninit(wfContext* wfc, CliprdrClientContext* cliprdr)
{
wfClipboard* clipboard;
if (!wfc || !cliprdr)
return FALSE;
clipboard = wfc->clipboard;
if (!clipboard)
return FALSE;
cliprdr->custom = NULL;
if (clipboard->hwnd)
PostMessage(clipboard->hwnd, WM_QUIT, 0, 0);
if (clipboard->thread)
{
WaitForSingleObject(clipboard->thread, INFINITE);
CloseHandle(clipboard->thread);
}
if (clipboard->response_data_event)
CloseHandle(clipboard->response_data_event);
if (clipboard->req_fevent)
CloseHandle(clipboard->req_fevent);
clear_file_array(clipboard);
clear_format_map(clipboard);
free(clipboard->format_mappings);
free(clipboard);
return TRUE;
}