VirtualBox

source: vbox/trunk/src/VBox/GuestHost/SharedClipboard/clipboard-transfers.cpp@ 100540

Last change on this file since 100540 was 100540, checked in by vboxsync, 19 months ago

Shared Clipboard: Make a copy of SHCLTRANSFERTHREADCTX in shClTransferThreadWorker() to avoid going the pointer out of scope as soon as we signalled the parent. bugref:9437
  • Property svn:eol-style set to native
  • Property svn:keywords set to Author Date Id Revision
File size: 101.3 KB
Line 
1/* $Id: clipboard-transfers.cpp 100540 2023-07-12 10:37:48Z vboxsync $ */
2/** @file
3 * Shared Clipboard: Common clipboard transfer handling code.
4 */
5
6/*
7 * Copyright (C) 2019-2023 Oracle and/or its affiliates.
8 *
9 * This file is part of VirtualBox base platform packages, as
10 * available from https://www.virtualbox.org.
11 *
12 * This program is free software; you can redistribute it and/or
13 * modify it under the terms of the GNU General Public License
14 * as published by the Free Software Foundation, in version 3 of the
15 * License.
16 *
17 * This program is distributed in the hope that it will be useful, but
18 * WITHOUT ANY WARRANTY; without even the implied warranty of
19 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
20 * General Public License for more details.
21 *
22 * You should have received a copy of the GNU General Public License
23 * along with this program; if not, see <https://www.gnu.org/licenses>.
24 *
25 * SPDX-License-Identifier: GPL-3.0-only
26 */
27
28#define LOG_GROUP LOG_GROUP_SHARED_CLIPBOARD
29#include <VBox/log.h>
30
31#include <iprt/dir.h>
32#include <iprt/file.h>
33#include <iprt/list.h>
34#include <iprt/path.h>
35#include <iprt/rand.h>
36#include <iprt/semaphore.h>
37#include <iprt/uri.h>
38#include <iprt/utf16.h>
39
40#include <VBox/err.h>
41#include <VBox/HostServices/VBoxClipboardSvc.h>
42#include <VBox/GuestHost/clipboard-helper.h>
43#include <VBox/GuestHost/SharedClipboard-transfers.h>
44
45
46
47/*********************************************************************************************************************************
48 * Prototypes *
49 ********************************************************************************************************************************/
50
51static void shClTransferCopyCallbacks(PSHCLTRANSFERCALLBACKS pCallbacksDst, PSHCLTRANSFERCALLBACKS pCallbacksSrc);
52DECLINLINE(void) shClTransferLock(PSHCLTRANSFER pTransfer);
53DECLINLINE(void) shClTransferUnlock(PSHCLTRANSFER pTransfer);
54static void shClTransferSetCallbacks(PSHCLTRANSFER pTransfer, PSHCLTRANSFERCALLBACKS pCallbacks);
55static int shClTransferSetStatus(PSHCLTRANSFER pTransfer, SHCLTRANSFERSTATUS enmStatus);
56static int shClTransferThreadCreate(PSHCLTRANSFER pTransfer, PFNSHCLTRANSFERTHREAD pfnThreadFunc, void *pvUser);
57static int shClTransferThreadDestroy(PSHCLTRANSFER pTransfer, RTMSINTERVAL uTimeoutMs);
58
59static void shclTransferCtxTransferRemoveAndUnregister(PSHCLTRANSFERCTX pTransferCtx, PSHCLTRANSFER pTransfer);
60static PSHCLTRANSFER shClTransferCtxGetTransferByIdInternal(PSHCLTRANSFERCTX pTransferCtx, SHCLTRANSFERID uId);
61static PSHCLTRANSFER shClTransferCtxGetTransferByIndexInternal(PSHCLTRANSFERCTX pTransferCtx, uint32_t uIdx);
62
63
64/*********************************************************************************************************************************
65 * Transfer List *
66 ********************************************************************************************************************************/
67
68/**
69 * Initializes a transfer list.
70 *
71 * @param pList Transfer list to initialize.
72 */
73void ShClTransferListInit(PSHCLLIST pList)
74{
75 RT_ZERO(pList->Hdr);
76 RTListInit(&pList->lstEntries);
77}
78
79/**
80 * Destroys a transfer list.
81 *
82 * @param pList Transfer list to destroy.
83 */
84void ShClTransferListDestroy(PSHCLLIST pList)
85{
86 if (!pList)
87 return;
88
89 PSHCLLISTENTRY pEntry, pEntryNext;
90 RTListForEachSafe(&pList->lstEntries, pEntry, pEntryNext, SHCLLISTENTRY, Node)
91 {
92 RTListNodeRemove(&pEntry->Node);
93 ShClTransferListEntryDestroy(pEntry);
94 RTMemFree(pEntry);
95 }
96
97 RT_ZERO(pList->Hdr);
98}
99
100/**
101 * Adds a list entry to a transfer list.
102 *
103 * @returns VBox status code.
104 * @param pList Transfer list to add entry to.
105 * @param pEntry Entry to add.
106 * @param fAppend \c true to append to a list, or \c false to prepend.
107 */
108int ShClTransferListAddEntry(PSHCLLIST pList, PSHCLLISTENTRY pEntry, bool fAppend)
109{
110 AssertReturn(ShClTransferListEntryIsValid(pEntry), VERR_INVALID_PARAMETER);
111
112 if (fAppend)
113 RTListAppend(&pList->lstEntries, &pEntry->Node);
114 else
115 RTListPrepend(&pList->lstEntries, &pEntry->Node);
116 pList->Hdr.cEntries++;
117
118 LogFlowFunc(("%p: '%s' (%RU32 bytes) + %RU32 bytes info -> now %RU32 entries\n",
119 pList, pEntry->pszName, pEntry->cbName, pEntry->cbInfo, pList->Hdr.cEntries));
120
121 return VINF_SUCCESS;
122}
123
124/**
125 * Allocates a new transfer list.
126 *
127 * @returns Allocated transfer list on success, or NULL on failure.
128 */
129PSHCLLIST ShClTransferListAlloc(void)
130{
131 PSHCLLIST pList = (PSHCLLIST)RTMemAllocZ(sizeof(SHCLLIST));
132 if (pList)
133 {
134 ShClTransferListInit(pList);
135 return pList;
136 }
137
138 return NULL;
139}
140
141/**
142 * Frees a transfer list.
143 *
144 * @param pList Transfer list to free. The pointer will be
145 * invalid after returning from this function.
146 */
147void ShClTransferListFree(PSHCLLIST pList)
148{
149 if (!pList)
150 return;
151
152 ShClTransferListDestroy(pList);
153
154 RTMemFree(pList);
155 pList = NULL;
156}
157
158/**
159 * Returns a specific list entry of a transfer list.
160 *
161 * @returns Pointer to list entry if found, or NULL if not found.
162 * @param pList Clipboard transfer list to get list entry from.
163 * @param uIdx Index of list entry to return.
164 */
165DECLINLINE(PSHCLLISTENTRY) shClTransferListGetEntryById(PSHCLLIST pList, uint32_t uIdx)
166{
167 if (uIdx >= pList->Hdr.cEntries)
168 return NULL;
169
170 Assert(!RTListIsEmpty(&pList->lstEntries));
171
172 PSHCLLISTENTRY pIt = RTListGetFirst(&pList->lstEntries, SHCLLISTENTRY, Node);
173 while (uIdx) /** @todo Slow, but works for now. */
174 {
175 pIt = RTListGetNext(&pList->lstEntries, pIt, SHCLLISTENTRY, Node);
176 uIdx--;
177 }
178
179 return pIt;
180}
181
182/**
183 * Initializes an list handle info structure.
184 *
185 * @returns VBox status code.
186 * @param pInfo List handle info structure to initialize.
187 */
188int ShClTransferListHandleInfoInit(PSHCLLISTHANDLEINFO pInfo)
189{
190 AssertPtrReturn(pInfo, VERR_INVALID_POINTER);
191
192 pInfo->hList = NIL_SHCLLISTHANDLE;
193 pInfo->enmType = SHCLOBJTYPE_INVALID;
194
195 pInfo->pszPathLocalAbs = NULL;
196
197 RT_ZERO(pInfo->u);
198
199 return VINF_SUCCESS;
200}
201
202/**
203 * Destroys a list handle info structure.
204 *
205 * @param pInfo List handle info structure to destroy.
206 */
207void ShClTransferListHandleInfoDestroy(PSHCLLISTHANDLEINFO pInfo)
208{
209 if (!pInfo)
210 return;
211
212 if (pInfo->pszPathLocalAbs)
213 {
214 RTStrFree(pInfo->pszPathLocalAbs);
215 pInfo->pszPathLocalAbs = NULL;
216 }
217}
218
219/**
220 * Allocates a transfer list header structure.
221 *
222 * @returns VBox status code.
223 * @param ppListHdr Where to store the allocated transfer list header structure on success.
224 */
225int ShClTransferListHdrAlloc(PSHCLLISTHDR *ppListHdr)
226{
227 int rc;
228
229 PSHCLLISTHDR pListHdr = (PSHCLLISTHDR)RTMemAllocZ(sizeof(SHCLLISTHDR));
230 if (pListHdr)
231 {
232 *ppListHdr = pListHdr;
233 rc = VINF_SUCCESS;
234 }
235 else
236 rc = VERR_NO_MEMORY;
237
238 LogFlowFuncLeaveRC(rc);
239 return rc;
240}
241
242/**
243 * Frees a transfer list header structure.
244 *
245 * @param pListEntry Transfer list header structure to free.
246 * The pointer will be invalid on return.
247 */
248void ShClTransferListHdrFree(PSHCLLISTHDR pListHdr)
249{
250 if (!pListHdr)
251 return;
252
253 LogFlowFuncEnter();
254
255 ShClTransferListHdrDestroy(pListHdr);
256
257 RTMemFree(pListHdr);
258 pListHdr = NULL;
259}
260
261/**
262 * Duplicates (allocates) a transfer list header structure.
263 *
264 * @returns Duplicated transfer list header structure on success.
265 * @param pListHdr Transfer list header to duplicate.
266 */
267PSHCLLISTHDR ShClTransferListHdrDup(PSHCLLISTHDR pListHdr)
268{
269 AssertPtrReturn(pListHdr, NULL);
270
271 PSHCLLISTHDR pListHdrDup = (PSHCLLISTHDR)RTMemAlloc(sizeof(SHCLLISTHDR));
272 if (pListHdrDup)
273 *pListHdrDup = *pListHdr;
274
275 return pListHdrDup;
276}
277
278/**
279 * Initializes a transfer list header structure.
280 *
281 * @returns VBox status code.
282 * @param pListHdr Transfer list header struct to initialize.
283 */
284int ShClTransferListHdrInit(PSHCLLISTHDR pListHdr)
285{
286 AssertPtrReturn(pListHdr, VERR_INVALID_POINTER);
287
288 LogFlowFuncEnter();
289
290 ShClTransferListHdrReset(pListHdr);
291
292 return VINF_SUCCESS;
293}
294
295/**
296 * Destroys a transfer list header structure.
297 *
298 * @param pListHdr Transfer list header struct to destroy.
299 */
300void ShClTransferListHdrDestroy(PSHCLLISTHDR pListHdr)
301{
302 if (!pListHdr)
303 return;
304
305 LogFlowFuncEnter();
306}
307
308/**
309 * Resets a transfer list header structure.
310 *
311 * @returns VBox status code.
312 * @param pListHdr Transfer list header struct to reset.
313 */
314void ShClTransferListHdrReset(PSHCLLISTHDR pListHdr)
315{
316 AssertPtrReturnVoid(pListHdr);
317
318 LogFlowFuncEnter();
319
320 RT_BZERO(pListHdr, sizeof(SHCLLISTHDR));
321}
322
323/**
324 * Returns whether a given transfer list header is valid or not.
325 *
326 * @returns \c true if valid, \c false if not.
327 * @param pListHdr Transfer list header to validate.
328 */
329bool ShClTransferListHdrIsValid(PSHCLLISTHDR pListHdr)
330{
331 RT_NOREF(pListHdr);
332 return true; /** @todo Implement this. */
333}
334
335/**
336 * (Deep-)Copies a transfer list open parameters structure from one into another.
337 *
338 * @returns VBox status code.
339 * @param pDst Destination parameters to copy to.
340 * @param pSrc Source parameters to copy from.
341 */
342int ShClTransferListOpenParmsCopy(PSHCLLISTOPENPARMS pDst, PSHCLLISTOPENPARMS pSrc)
343{
344 AssertPtrReturn(pDst, VERR_INVALID_POINTER);
345 AssertPtrReturn(pSrc, VERR_INVALID_POINTER);
346
347 int rc = VINF_SUCCESS;
348
349 if (pSrc->pszFilter)
350 {
351 pDst->pszFilter = RTStrDup(pSrc->pszFilter);
352 if (!pDst->pszFilter)
353 rc = VERR_NO_MEMORY;
354 }
355
356 if ( RT_SUCCESS(rc)
357 && pSrc->pszPath)
358 {
359 pDst->pszPath = RTStrDup(pSrc->pszPath);
360 if (!pDst->pszPath)
361 rc = VERR_NO_MEMORY;
362 }
363
364 if (RT_SUCCESS(rc))
365 {
366 pDst->fList = pDst->fList;
367 pDst->cbFilter = pSrc->cbFilter;
368 pDst->cbPath = pSrc->cbPath;
369 }
370
371 return rc;
372}
373
374/**
375 * Duplicates a transfer list open parameters structure.
376 *
377 * @returns Duplicated transfer list open parameters structure on success, or NULL on failure.
378 * @param pParms Transfer list open parameters structure to duplicate.
379 */
380PSHCLLISTOPENPARMS ShClTransferListOpenParmsDup(PSHCLLISTOPENPARMS pParms)
381{
382 AssertPtrReturn(pParms, NULL);
383
384 PSHCLLISTOPENPARMS pParmsDup = (PSHCLLISTOPENPARMS)RTMemAllocZ(sizeof(SHCLLISTOPENPARMS));
385 if (!pParmsDup)
386 return NULL;
387
388 int rc = ShClTransferListOpenParmsCopy(pParmsDup, pParms);
389 if (RT_FAILURE(rc))
390 {
391 ShClTransferListOpenParmsDestroy(pParmsDup);
392
393 RTMemFree(pParmsDup);
394 pParmsDup = NULL;
395 }
396
397 return pParmsDup;
398}
399
400/**
401 * Initializes a transfer list open parameters structure.
402 *
403 * @returns VBox status code.
404 * @param pParms Transfer list open parameters structure to initialize.
405 */
406int ShClTransferListOpenParmsInit(PSHCLLISTOPENPARMS pParms)
407{
408 AssertPtrReturn(pParms, VERR_INVALID_POINTER);
409
410 RT_BZERO(pParms, sizeof(SHCLLISTOPENPARMS));
411
412 pParms->cbFilter = SHCL_TRANSFER_PATH_MAX; /** @todo Make this dynamic. */
413 pParms->pszFilter = RTStrAlloc(pParms->cbFilter);
414
415 pParms->cbPath = SHCL_TRANSFER_PATH_MAX; /** @todo Make this dynamic. */
416 pParms->pszPath = RTStrAlloc(pParms->cbPath);
417
418 LogFlowFuncLeave();
419 return VINF_SUCCESS;
420}
421
422/**
423 * Destroys a transfer list open parameters structure.
424 *
425 * @param pParms Transfer list open parameters structure to destroy.
426 */
427void ShClTransferListOpenParmsDestroy(PSHCLLISTOPENPARMS pParms)
428{
429 if (!pParms)
430 return;
431
432 if (pParms->pszFilter)
433 {
434 RTStrFree(pParms->pszFilter);
435 pParms->pszFilter = NULL;
436 }
437
438 if (pParms->pszPath)
439 {
440 RTStrFree(pParms->pszPath);
441 pParms->pszPath = NULL;
442 }
443}
444
445/**
446 * Creates (allocates) and initializes a clipboard list entry structure.
447 *
448 * @returns VBox status code.
449 * @param ppListEntry Where to return the created clipboard list entry structure on success.
450 * Must be free'd with ShClTransferListEntryFree().
451 */
452int ShClTransferListEntryAlloc(PSHCLLISTENTRY *ppListEntry)
453{
454 AssertPtrReturn(ppListEntry, VERR_INVALID_POINTER);
455
456 PSHCLLISTENTRY pListEntry = (PSHCLLISTENTRY)RTMemAllocZ(sizeof(SHCLLISTENTRY));
457 if (!pListEntry)
458 return VERR_NO_MEMORY;
459
460 *ppListEntry = pListEntry;
461
462 return VINF_SUCCESS;
463}
464
465/**
466 * Frees a clipboard list entry structure.
467 *
468 * @param pEntry Clipboard list entry structure to free.
469 * The pointer will be invalid on return.
470 */
471void ShClTransferListEntryFree(PSHCLLISTENTRY pEntry)
472{
473 if (!pEntry)
474 return;
475
476 /* Make sure to destroy the entry properly, in case the caller forgot this. */
477 ShClTransferListEntryDestroy(pEntry);
478
479 RTMemFree(pEntry);
480 pEntry = NULL;
481}
482
483/**
484 * (Deep-)Copies a clipboard list entry structure.
485 *
486 * @returns VBox status code.
487 * @param pDst Destination list entry to copy to.
488 * @param pSrc Source list entry to copy from.
489 */
490int ShClTransferListEntryCopy(PSHCLLISTENTRY pDst, PSHCLLISTENTRY pSrc)
491{
492 AssertPtrReturn(pDst, VERR_INVALID_POINTER);
493 AssertPtrReturn(pSrc, VERR_INVALID_POINTER);
494
495 int rc = VINF_SUCCESS;
496
497 *pDst = *pSrc;
498
499 if (pSrc->pszName)
500 {
501 pDst->pszName = RTStrDup(pSrc->pszName);
502 if (!pDst->pszName)
503 rc = VERR_NO_MEMORY;
504 }
505
506 if ( RT_SUCCESS(rc)
507 && pSrc->pvInfo)
508 {
509 pDst->pvInfo = RTMemDup(pSrc->pvInfo, pSrc->cbInfo);
510 if (pDst->pvInfo)
511 {
512 pDst->cbInfo = pSrc->cbInfo;
513 }
514 else
515 rc = VERR_NO_MEMORY;
516 }
517
518 if (RT_FAILURE(rc))
519 {
520 if (pDst->pvInfo)
521 {
522 RTMemFree(pDst->pvInfo);
523 pDst->pvInfo = NULL;
524 pDst->cbInfo = 0;
525 }
526 }
527
528 return rc;
529}
530
531/**
532 * Duplicates (allocates) a clipboard list entry structure.
533 *
534 * @returns Duplicated clipboard list entry structure on success.
535 * @param pEntry Clipboard list entry to duplicate.
536 */
537PSHCLLISTENTRY ShClTransferListEntryDup(PSHCLLISTENTRY pEntry)
538{
539 AssertPtrReturn(pEntry, NULL);
540
541 int rc = VINF_SUCCESS;
542
543 PSHCLLISTENTRY pListEntryDup = (PSHCLLISTENTRY)RTMemAllocZ(sizeof(SHCLLISTENTRY));
544 if (pListEntryDup)
545 rc = ShClTransferListEntryCopy(pListEntryDup, pEntry);
546
547 if (RT_FAILURE(rc))
548 {
549 ShClTransferListEntryDestroy(pListEntryDup);
550
551 RTMemFree(pListEntryDup);
552 pListEntryDup = NULL;
553 }
554
555 return pListEntryDup;
556}
557
558/**
559 * Returns whether a given list entry name is valid or not.
560 *
561 * @returns \c true if valid, or \c false if not.
562 * @param pszName Name to check.
563 * @param cbName Size (in bytes) of \a pszName to check.
564 * Includes terminator.
565 */
566static bool shclTransferListEntryNameIsValid(const char *pszName, size_t cbName)
567{
568 if (!pszName)
569 return false;
570
571 size_t const cchLen = strlen(pszName);
572
573 if ( !cbName
574 || cchLen == 0
575 || cchLen > cbName /* Includes zero termination */ - 1
576 || cchLen > SHCLLISTENTRY_MAX_NAME /* Ditto */ - 1)
577 {
578 return false;
579 }
580
581 int rc = ShClTransferValidatePath(pszName, false /* fMustExist */);
582 if (RT_FAILURE(rc))
583 return false;
584
585 return true;
586}
587
588/**
589 * Initializes a clipboard list entry structure, extended version.
590 *
591 * @returns VBox status code.
592 * @param pListEntry Clipboard list entry structure to initialize.
593 * @param fInfo Info flags (of type VBOX_SHCL_INFO_F_XXX).
594 * @param pszName Name (e.g. filename) to use. Can be NULL if not being used.
595 * Up to SHCLLISTENTRY_MAX_NAME characters.
596 * @param pvInfo Pointer to info data to assign. Must match \a fInfo.
597 * The list entry takes the ownership of the data on success.
598 * @param cbInfo Size (in bytes) of \a pvInfo data to assign.
599 */
600int ShClTransferListEntryInitEx(PSHCLLISTENTRY pListEntry, uint32_t fInfo, const char *pszName, void *pvInfo, uint32_t cbInfo)
601{
602 AssertPtrReturn(pListEntry, VERR_INVALID_POINTER);
603 AssertReturn ( pszName == NULL
604 || shclTransferListEntryNameIsValid(pszName, strlen(pszName) + 1), VERR_INVALID_PARAMETER);
605 /* pvInfo + cbInfo depend on fInfo. See below. */
606
607 RT_BZERO(pListEntry, sizeof(SHCLLISTENTRY));
608
609 if (pszName)
610 {
611 pListEntry->pszName = RTStrDupN(pszName, SHCLLISTENTRY_MAX_NAME);
612 AssertPtrReturn(pListEntry->pszName, VERR_NO_MEMORY);
613 pListEntry->cbName = (uint32_t)strlen(pListEntry->pszName) + 1 /* Include terminator */;
614 }
615
616 pListEntry->pvInfo = pvInfo;
617 pListEntry->cbInfo = cbInfo;
618 pListEntry->fInfo = fInfo;
619
620 return VINF_SUCCESS;
621}
622
623/**
624 * Initializes a clipboard list entry structure (as empty / invalid).
625 *
626 * @returns VBox status code.
627 * @param pListEntry Clipboard list entry structure to initialize.
628 */
629int ShClTransferListEntryInit(PSHCLLISTENTRY pListEntry)
630{
631 return ShClTransferListEntryInitEx(pListEntry, VBOX_SHCL_INFO_F_NONE, NULL /* pszName */, NULL /* pvInfo */, 0 /* cbInfo */);
632}
633
634/**
635 * Destroys a clipboard list entry structure.
636 *
637 * @param pListEntry Clipboard list entry structure to destroy.
638 */
639void ShClTransferListEntryDestroy(PSHCLLISTENTRY pListEntry)
640{
641 if (!pListEntry)
642 return;
643
644 if (pListEntry->pszName)
645 {
646 RTStrFree(pListEntry->pszName);
647
648 pListEntry->pszName = NULL;
649 pListEntry->cbName = 0;
650 }
651
652 if (pListEntry->pvInfo)
653 {
654 RTMemFree(pListEntry->pvInfo);
655 pListEntry->pvInfo = NULL;
656 pListEntry->cbInfo = 0;
657 }
658}
659
660/**
661 * Returns whether a given clipboard list entry is valid or not.
662 *
663 * @returns \c true if valid, \c false if not.
664 * @param pListEntry Clipboard list entry to validate.
665 */
666bool ShClTransferListEntryIsValid(PSHCLLISTENTRY pListEntry)
667{
668 AssertPtrReturn(pListEntry, false);
669
670 bool fValid = false;
671
672 if (shclTransferListEntryNameIsValid(pListEntry->pszName, pListEntry->cbName))
673 {
674 fValid = pListEntry->cbInfo == 0 /* cbInfo / pvInfo is optional. */
675 || pListEntry->pvInfo != NULL;
676 }
677
678 if (!fValid)
679 LogRel2(("Shared Clipboard: List entry '%s' is invalid\n", pListEntry->pszName));
680
681 return fValid;
682}
683
684
685/*********************************************************************************************************************************
686 * Transfer Object *
687 ********************************************************************************************************************************/
688
689/**
690 * Initializes a transfer object context.
691 *
692 * @returns VBox status code.
693 * @param pObjCtx Transfer object context to initialize.
694 */
695int ShClTransferObjCtxInit(PSHCLCLIENTTRANSFEROBJCTX pObjCtx)
696{
697 AssertPtrReturn(pObjCtx, VERR_INVALID_POINTER);
698
699 LogFlowFuncEnter();
700
701 pObjCtx->uHandle = NIL_SHCLOBJHANDLE;
702
703 return VINF_SUCCESS;
704}
705
706/**
707 * Destroys a transfer object context.
708 *
709 * @param pObjCtx Transfer object context to destroy.
710 */
711void ShClTransferObjCtxDestroy(PSHCLCLIENTTRANSFEROBJCTX pObjCtx)
712{
713 AssertPtrReturnVoid(pObjCtx);
714
715 LogFlowFuncEnter();
716}
717
718/**
719 * Returns if a transfer object context is valid or not.
720 *
721 * @returns \c true if valid, \c false if not.
722 * @param pObjCtx Transfer object context to check.
723 */
724bool ShClTransferObjCtxIsValid(PSHCLCLIENTTRANSFEROBJCTX pObjCtx)
725{
726 return ( pObjCtx
727 && pObjCtx->uHandle != NIL_SHCLOBJHANDLE);
728}
729
730/**
731 * Initializes a transfer object structure.
732 *
733 * @returns VBox status code.
734 * @param pObj Object structure to initialize.
735 */
736int ShClTransferObjInit(PSHCLTRANSFEROBJ pObj)
737{
738 AssertPtrReturn(pObj, VERR_INVALID_POINTER);
739
740 pObj->hObj = NIL_SHCLOBJHANDLE;
741 pObj->enmType = SHCLOBJTYPE_INVALID;
742
743 pObj->pszPathLocalAbs = NULL;
744
745 RT_ZERO(pObj->u);
746
747 return VINF_SUCCESS;
748}
749
750/**
751 * Destroys a transfer object structure.
752 *
753 * @param pObj Object structure to destroy.
754 */
755void ShClTransferObjDestroy(PSHCLTRANSFEROBJ pObj)
756{
757 if (!pObj)
758 return;
759
760 if (pObj->pszPathLocalAbs)
761 {
762 RTStrFree(pObj->pszPathLocalAbs);
763 pObj->pszPathLocalAbs = NULL;
764 }
765}
766
767/**
768 * Initializes a transfer object open parameters structure.
769 *
770 * @returns VBox status code.
771 * @param pParms Transfer object open parameters structure to initialize.
772 */
773int ShClTransferObjOpenParmsInit(PSHCLOBJOPENCREATEPARMS pParms)
774{
775 AssertPtrReturn(pParms, VERR_INVALID_POINTER);
776
777 int rc;
778
779 RT_BZERO(pParms, sizeof(SHCLOBJOPENCREATEPARMS));
780
781 pParms->cbPath = RTPATH_MAX; /** @todo Make this dynamic. */
782 pParms->pszPath = RTStrAlloc(pParms->cbPath);
783 if (pParms->pszPath)
784 {
785 rc = VINF_SUCCESS;
786 }
787 else
788 rc = VERR_NO_MEMORY;
789
790 LogFlowFuncLeaveRC(rc);
791 return rc;
792}
793
794/**
795 * Copies a transfer object open parameters structure from source to destination.
796 *
797 * @returns VBox status code.
798 * @param pParmsDst Where to copy the source transfer object open parameters to.
799 * @param pParmsSrc Which source transfer object open parameters to copy.
800 */
801int ShClTransferObjOpenParmsCopy(PSHCLOBJOPENCREATEPARMS pParmsDst, PSHCLOBJOPENCREATEPARMS pParmsSrc)
802{
803 int rc;
804
805 *pParmsDst = *pParmsSrc;
806
807 if (pParmsSrc->pszPath)
808 {
809 Assert(pParmsSrc->cbPath);
810 pParmsDst->pszPath = RTStrDup(pParmsSrc->pszPath);
811 if (pParmsDst->pszPath)
812 {
813 rc = VINF_SUCCESS;
814 }
815 else
816 rc = VERR_NO_MEMORY;
817 }
818 else
819 rc = VINF_SUCCESS;
820
821 LogFlowFuncLeaveRC(rc);
822 return rc;
823}
824
825/**
826 * Destroys a transfer object open parameters structure.
827 *
828 * @param pParms Transfer object open parameters structure to destroy.
829 */
830void ShClTransferObjOpenParmsDestroy(PSHCLOBJOPENCREATEPARMS pParms)
831{
832 if (!pParms)
833 return;
834
835 if (pParms->pszPath)
836 {
837 RTStrFree(pParms->pszPath);
838 pParms->pszPath = NULL;
839 }
840}
841
842/**
843 * Returns a specific transfer object of a transfer.
844 *
845 * @returns Pointer to transfer object if found, or NULL if not found.
846 * @param pTransfer Clipboard transfer to get transfer object from.
847 * @param hObj Object handle of the object to get handle info for.
848 */
849PSHCLTRANSFEROBJ ShClTransferObjGet(PSHCLTRANSFER pTransfer, SHCLOBJHANDLE hObj)
850{
851 PSHCLTRANSFEROBJ pIt;
852 RTListForEach(&pTransfer->lstObj, pIt, SHCLTRANSFEROBJ, Node) /** @todo Slooow ...but works for now. */
853 {
854 if (pIt->hObj == hObj)
855 return pIt;
856 }
857
858 return NULL;
859}
860
861/**
862 * Opens a transfer object.
863 *
864 * @returns VBox status code.
865 * @param pTransfer Clipboard transfer to open the object for.
866 * @param pOpenCreateParms Open / create parameters of transfer object to open / create.
867 * @param phObj Where to store the handle of transfer object opened on success.
868 */
869int ShClTransferObjOpen(PSHCLTRANSFER pTransfer, PSHCLOBJOPENCREATEPARMS pOpenCreateParms, PSHCLOBJHANDLE phObj)
870{
871 AssertPtrReturn(pTransfer, VERR_INVALID_POINTER);
872 AssertPtrReturn(pOpenCreateParms, VERR_INVALID_POINTER);
873 AssertPtrReturn(phObj, VERR_INVALID_POINTER);
874 AssertMsgReturn(pTransfer->pszPathRootAbs, ("Transfer has no root path set\n"), VERR_INVALID_PARAMETER);
875 /** @todo Check pOpenCreateParms->fCreate flags. */
876 AssertMsgReturn(pOpenCreateParms->pszPath, ("No path in open/create params set\n"), VERR_INVALID_PARAMETER);
877
878 if (pTransfer->cObjHandles >= pTransfer->cMaxObjHandles)
879 return VERR_SHCLPB_MAX_OBJECTS_REACHED;
880
881 LogFlowFunc(("pszPath=%s, fCreate=0x%x\n", pOpenCreateParms->pszPath, pOpenCreateParms->fCreate));
882
883 int rc;
884 if (pTransfer->ProviderIface.pfnObjOpen)
885 rc = pTransfer->ProviderIface.pfnObjOpen(&pTransfer->ProviderCtx, pOpenCreateParms, phObj);
886 else
887 rc = VERR_NOT_SUPPORTED;
888
889 if (RT_FAILURE(rc))
890 LogRel(("Shared Clipboard: Opening object '%s' (flags %#x) failed with %Rrc\n",
891 pOpenCreateParms->pszPath, pOpenCreateParms->fCreate, rc));
892
893 LogFlowFuncLeaveRC(rc);
894 return rc;
895}
896
897/**
898 * Closes a transfer object.
899 *
900 * @returns VBox status code.
901 * @param pTransfer Clipboard transfer that contains the object to close.
902 * @param hObj Handle of transfer object to close.
903 */
904int ShClTransferObjClose(PSHCLTRANSFER pTransfer, SHCLOBJHANDLE hObj)
905{
906 AssertPtrReturn(pTransfer, VERR_INVALID_POINTER);
907
908 int rc;
909 if (pTransfer->ProviderIface.pfnObjClose)
910 rc = pTransfer->ProviderIface.pfnObjClose(&pTransfer->ProviderCtx, hObj);
911 else
912 rc = VERR_NOT_SUPPORTED;
913
914 if (RT_FAILURE(rc))
915 LogRel(("Shared Clipboard: Reading object 0x%x failed with %Rrc\n", hObj, rc));
916
917 LogFlowFuncLeaveRC(rc);
918 return rc;
919}
920
921/**
922 * Reads from a transfer object.
923 *
924 * @returns VBox status code.
925 * @param pTransfer Clipboard transfer that contains the object to read from.
926 * @param hObj Handle of transfer object to read from.
927 * @param pvBuf Buffer for where to store the read data.
928 * @param cbBuf Size (in bytes) of buffer.
929 * @param fFlags Read flags. Optional.
930 * @param pcbRead Where to return how much bytes were read on success. Optional.
931 */
932int ShClTransferObjRead(PSHCLTRANSFER pTransfer,
933 SHCLOBJHANDLE hObj, void *pvBuf, uint32_t cbBuf, uint32_t fFlags, uint32_t *pcbRead)
934{
935 AssertPtrReturn(pTransfer, VERR_INVALID_POINTER);
936 AssertPtrReturn(pvBuf, VERR_INVALID_POINTER);
937 AssertReturn (cbBuf, VERR_INVALID_PARAMETER);
938 /* pcbRead is optional. */
939 /** @todo Validate fFlags. */
940
941 int rc;
942 if (pTransfer->ProviderIface.pfnObjRead)
943 rc = pTransfer->ProviderIface.pfnObjRead(&pTransfer->ProviderCtx, hObj, pvBuf, cbBuf, fFlags, pcbRead);
944 else
945 rc = VERR_NOT_SUPPORTED;
946
947 if (RT_FAILURE(rc))
948 LogRel(("Shared Clipboard: Reading object 0x%x failed with %Rrc\n", hObj, rc));
949
950 LogFlowFuncLeaveRC(rc);
951 return rc;
952}
953
954/**
955 * Writes to a transfer object.
956 *
957 * @returns VBox status code.
958 * @param pTransfer Clipboard transfer that contains the object to write to.
959 * @param hObj Handle of transfer object to write to.
960 * @param pvBuf Buffer of data to write.
961 * @param cbBuf Size (in bytes) of buffer to write.
962 * @param fFlags Write flags. Optional.
963 * @param pcbWritten How much bytes were writtenon success. Optional.
964 */
965int ShClTransferObjWrite(PSHCLTRANSFER pTransfer,
966 SHCLOBJHANDLE hObj, void *pvBuf, uint32_t cbBuf, uint32_t fFlags, uint32_t *pcbWritten)
967{
968 AssertPtrReturn(pTransfer, VERR_INVALID_POINTER);
969 AssertPtrReturn(pvBuf, VERR_INVALID_POINTER);
970 AssertReturn (cbBuf, VERR_INVALID_PARAMETER);
971 /* pcbWritten is optional. */
972
973 int rc;
974 if (pTransfer->ProviderIface.pfnObjWrite)
975 rc = pTransfer->ProviderIface.pfnObjWrite(&pTransfer->ProviderCtx, hObj, pvBuf, cbBuf, fFlags, pcbWritten);
976 else
977 rc = VERR_NOT_SUPPORTED;
978
979 if (RT_FAILURE(rc))
980 LogRel(("Shared Clipboard: Writing object 0x%x failed with %Rrc\n", hObj, rc));
981
982 LogFlowFuncLeaveRC(rc);
983 return rc;
984}
985
986/**
987 * Duplicates a transfer object data chunk.
988 *
989 * @returns Duplicated object data chunk on success, or NULL on failure.
990 * @param pDataChunk Transfer object data chunk to duplicate.
991 */
992PSHCLOBJDATACHUNK ShClTransferObjDataChunkDup(PSHCLOBJDATACHUNK pDataChunk)
993{
994 AssertPtrReturn(pDataChunk, NULL);
995
996 PSHCLOBJDATACHUNK pDataChunkDup = (PSHCLOBJDATACHUNK)RTMemAllocZ(sizeof(SHCLOBJDATACHUNK));
997 if (!pDataChunkDup)
998 return NULL;
999
1000 if (pDataChunk->pvData)
1001 {
1002 Assert(pDataChunk->cbData);
1003
1004 pDataChunkDup->uHandle = pDataChunk->uHandle;
1005 pDataChunkDup->pvData = RTMemDup(pDataChunk->pvData, pDataChunk->cbData);
1006 AssertPtrReturn(pDataChunkDup->pvData, NULL);
1007 pDataChunkDup->cbData = pDataChunk->cbData;
1008 }
1009
1010 return pDataChunkDup;
1011}
1012
1013/**
1014 * Destroys a transfer object data chunk.
1015 *
1016 * @param pDataChunk Transfer object data chunk to destroy.
1017 */
1018void ShClTransferObjDataChunkDestroy(PSHCLOBJDATACHUNK pDataChunk)
1019{
1020 if (!pDataChunk)
1021 return;
1022
1023 if (pDataChunk->pvData)
1024 {
1025 Assert(pDataChunk->cbData);
1026
1027 RTMemFree(pDataChunk->pvData);
1028
1029 pDataChunk->pvData = NULL;
1030 pDataChunk->cbData = 0;
1031 }
1032
1033 pDataChunk->uHandle = 0;
1034}
1035
1036/**
1037 * Frees a transfer object data chunk.
1038 *
1039 * @param pDataChunk Transfer object data chunk to free.
1040 * The pointer will be invalid on return.
1041 */
1042void ShClTransferObjDataChunkFree(PSHCLOBJDATACHUNK pDataChunk)
1043{
1044 if (!pDataChunk)
1045 return;
1046
1047 ShClTransferObjDataChunkDestroy(pDataChunk);
1048
1049 RTMemFree(pDataChunk);
1050 pDataChunk = NULL;
1051}
1052
1053
1054/*********************************************************************************************************************************
1055 * Transfer *
1056 ********************************************************************************************************************************/
1057
1058/**
1059 * Creates a clipboard transfer, extended version.
1060 *
1061 * @returns VBox status code.
1062 * @param enmDir Specifies the transfer direction of this transfer.
1063 * @param enmSource Specifies the data source of the transfer.
1064 * @param pCallbacks Callback table to use. Optional and can be NULL.
1065 * @param cbMaxChunkSize Maximum transfer chunk size (in bytes) to use.
1066 * @param cMaxListHandles Maximum list entries the transfer can have.
1067 * @param cMaxObjHandles Maximum transfer objects the transfer can have.
1068 * @param ppTransfer Where to return the created clipboard transfer struct.
1069 * Must be destroyed by ShClTransferDestroy().
1070 */
1071int ShClTransferCreateEx(SHCLTRANSFERDIR enmDir, SHCLSOURCE enmSource, PSHCLTRANSFERCALLBACKS pCallbacks,
1072 uint32_t cbMaxChunkSize, uint32_t cMaxListHandles, uint32_t cMaxObjHandles, PSHCLTRANSFER *ppTransfer)
1073{
1074 AssertPtrReturn(ppTransfer, VERR_INVALID_POINTER);
1075 /* pCallbacks can be NULL. */
1076
1077 LogFlowFuncEnter();
1078
1079 PSHCLTRANSFER pTransfer = (PSHCLTRANSFER)RTMemAllocZ(sizeof(SHCLTRANSFER));
1080 AssertPtrReturn(pTransfer, VERR_NO_MEMORY);
1081
1082 pTransfer->State.uID = NIL_SHCLTRANSFERID;
1083 pTransfer->State.enmStatus = SHCLTRANSFERSTATUS_NONE;
1084 pTransfer->State.enmDir = enmDir;
1085 pTransfer->State.enmSource = enmSource;
1086
1087 pTransfer->Thread.hThread = NIL_RTTHREAD;
1088 pTransfer->Thread.fCancelled = false;
1089 pTransfer->Thread.fStarted = false;
1090 pTransfer->Thread.fStop = false;
1091
1092 pTransfer->pszPathRootAbs = NULL;
1093
1094 pTransfer->uTimeoutMs = SHCL_TIMEOUT_DEFAULT_MS;
1095 pTransfer->cbMaxChunkSize = cbMaxChunkSize;
1096 pTransfer->cMaxListHandles = cMaxListHandles;
1097 pTransfer->cMaxObjHandles = cMaxObjHandles;
1098
1099 pTransfer->pvUser = NULL;
1100 pTransfer->cbUser = 0;
1101
1102 RTListInit(&pTransfer->lstHandles);
1103 RTListInit(&pTransfer->lstObj);
1104
1105 /* The provider context + interface is NULL by default. */
1106 RT_ZERO(pTransfer->ProviderCtx);
1107 RT_ZERO(pTransfer->ProviderIface);
1108
1109 /* Make sure to set the callbacks before calling pfnOnCreate below. */
1110 shClTransferSetCallbacks(pTransfer, pCallbacks);
1111
1112 ShClTransferListInit(&pTransfer->lstRoots);
1113
1114 int rc = RTCritSectInit(&pTransfer->CritSect);
1115 AssertRCReturn(rc, rc);
1116
1117 rc = RTSemEventCreate(&pTransfer->StatusChangeEvent);
1118 AssertRCReturn(rc, rc);
1119
1120 rc = ShClEventSourceCreate(&pTransfer->Events, 0 /* uID */);
1121 if (RT_SUCCESS(rc))
1122 {
1123 if (pTransfer->Callbacks.pfnOnCreated)
1124 pTransfer->Callbacks.pfnOnCreated(&pTransfer->CallbackCtx);
1125
1126 *ppTransfer = pTransfer;
1127 }
1128 else
1129 {
1130 if (pTransfer)
1131 {
1132 ShClTransferDestroy(pTransfer);
1133 RTMemFree(pTransfer);
1134 }
1135 }
1136
1137 LogFlowFuncLeaveRC(rc);
1138 return rc;
1139}
1140
1141/**
1142 * Creates a clipboard transfer with default settings.
1143 *
1144 * @returns VBox status code.
1145 * @param enmDir Specifies the transfer direction of this transfer.
1146 * @param enmSource Specifies the data source of the transfer.
1147 * @param pCallbacks Callback table to use. Optional and can be NULL.
1148 * @param ppTransfer Where to return the created clipboard transfer struct.
1149 * Must be destroyed by ShClTransferDestroy().
1150 */
1151int ShClTransferCreate(SHCLTRANSFERDIR enmDir, SHCLSOURCE enmSource, PSHCLTRANSFERCALLBACKS pCallbacks, PSHCLTRANSFER *ppTransfer)
1152{
1153 return ShClTransferCreateEx(enmDir, enmSource, pCallbacks,
1154 SHCL_TRANSFER_DEFAULT_MAX_CHUNK_SIZE,
1155 SHCL_TRANSFER_DEFAULT_MAX_LIST_HANDLES,
1156 SHCL_TRANSFER_DEFAULT_MAX_OBJ_HANDLES,
1157 ppTransfer);
1158}
1159
1160/**
1161 * Destroys a clipboard transfer.
1162 *
1163 * @returns VBox status code.
1164 * @param pTransferCtx Clipboard transfer to destroy.
1165 */
1166int ShClTransferDestroy(PSHCLTRANSFER pTransfer)
1167{
1168 if (!pTransfer)
1169 return VINF_SUCCESS;
1170
1171 /* Must come before the refcount check below, as the callback might release a reference. */
1172 if (pTransfer->Callbacks.pfnOnDestroy)
1173 pTransfer->Callbacks.pfnOnDestroy(&pTransfer->CallbackCtx);
1174
1175 AssertMsgReturn(ASMAtomicReadU32(&pTransfer->cRefs) == 0,
1176 ("Number of references > 0 (%RU32)\n", pTransfer->cRefs), VERR_WRONG_ORDER);
1177
1178 LogFlowFuncEnter();
1179
1180 int rc = shClTransferThreadDestroy(pTransfer, SHCL_TIMEOUT_DEFAULT_MS);
1181 if (RT_FAILURE(rc))
1182 return rc;
1183
1184 ShClTransferReset(pTransfer);
1185
1186 if (RTCritSectIsInitialized(&pTransfer->CritSect))
1187 RTCritSectDelete(&pTransfer->CritSect);
1188
1189 rc = RTSemEventDestroy(pTransfer->StatusChangeEvent);
1190 AssertRCReturn(rc, rc);
1191 pTransfer->StatusChangeEvent = NIL_RTSEMEVENT;
1192
1193 ShClEventSourceDestroy(&pTransfer->Events);
1194
1195 LogFlowFuncLeave();
1196 return VINF_SUCCESS;
1197}
1198
1199/**
1200 * Initializes a clipboard transfer.
1201 *
1202 * @returns VBox status code.
1203 * @param pTransfer Transfer to initialize.
1204 */
1205int ShClTransferInit(PSHCLTRANSFER pTransfer)
1206{
1207 shClTransferLock(pTransfer);
1208
1209 AssertMsgReturnStmt(pTransfer->State.enmStatus < SHCLTRANSFERSTATUS_INITIALIZED,
1210 ("Wrong status (currently is %s)\n", ShClTransferStatusToStr(pTransfer->State.enmStatus)),
1211 shClTransferUnlock(pTransfer), VERR_WRONG_ORDER);
1212
1213 pTransfer->cRefs = 0;
1214
1215 LogFlowFunc(("uID=%RU32, enmDir=%RU32, enmSource=%RU32\n",
1216 pTransfer->State.uID, pTransfer->State.enmDir, pTransfer->State.enmSource));
1217
1218 pTransfer->cListHandles = 0;
1219 pTransfer->uListHandleNext = 1;
1220
1221 pTransfer->cObjHandles = 0;
1222 pTransfer->uObjHandleNext = 1;
1223
1224 int rc = shClTransferSetStatus(pTransfer, SHCLTRANSFERSTATUS_INITIALIZED);
1225
1226 shClTransferUnlock(pTransfer);
1227
1228 if (RT_SUCCESS(rc))
1229 {
1230 if (pTransfer->Callbacks.pfnOnInitialized)
1231 pTransfer->Callbacks.pfnOnInitialized(&pTransfer->CallbackCtx);
1232 }
1233
1234 LogFlowFuncLeaveRC(rc);
1235 return rc;
1236}
1237
1238/**
1239 * Locks a transfer.
1240 *
1241 * @param pTransfer Transfer to lock.
1242 */
1243DECLINLINE(void) shClTransferLock(PSHCLTRANSFER pTransfer)
1244{
1245 int rc2 = RTCritSectEnter(&pTransfer->CritSect);
1246 AssertRC(rc2);
1247}
1248
1249/**
1250 * Unlocks a transfer.
1251 *
1252 * @param pTransfer Transfer to unlock.
1253 */
1254DECLINLINE(void) shClTransferUnlock(PSHCLTRANSFER pTransfer)
1255{
1256 int rc2 = RTCritSectLeave(&pTransfer->CritSect);
1257 AssertRC(rc2);
1258}
1259
1260/**
1261 * Acquires a reference to this transfer.
1262 *
1263 * @returns New reference count.
1264 * @param pTransfer Transfer to acquire reference for.
1265 */
1266uint32_t ShClTransferAcquire(PSHCLTRANSFER pTransfer)
1267{
1268 return ASMAtomicIncU32(&pTransfer->cRefs);
1269}
1270
1271/**
1272 * Releases a reference to this transfer.
1273 *
1274 * @returns New reference count.
1275 * @param pTransfer Transfer to release reference for.
1276 */
1277uint32_t ShClTransferRelease(PSHCLTRANSFER pTransfer)
1278{
1279 const uint32_t cRefs = ASMAtomicDecU32(&pTransfer->cRefs);
1280 Assert(pTransfer->cRefs <= VBOX_SHCL_MAX_TRANSFERS); /* Not perfect, but better than nothing. */
1281 return cRefs;
1282}
1283
1284/**
1285 * Opens a transfer list.
1286 *
1287 * @returns VBox status code.
1288 * @param pTransfer Clipboard transfer to handle.
1289 * @param pOpenParms List open parameters to use for opening.
1290 * @param phList Where to store the List handle of opened list on success.
1291 */
1292int ShClTransferListOpen(PSHCLTRANSFER pTransfer, PSHCLLISTOPENPARMS pOpenParms,
1293 PSHCLLISTHANDLE phList)
1294{
1295 AssertPtrReturn(pTransfer, VERR_INVALID_POINTER);
1296 AssertPtrReturn(pOpenParms, VERR_INVALID_POINTER);
1297 AssertPtrReturn(phList, VERR_INVALID_POINTER);
1298
1299 if (pTransfer->cListHandles == pTransfer->cMaxListHandles)
1300 return VERR_SHCLPB_MAX_LISTS_REACHED;
1301
1302 int rc;
1303 if (pTransfer->ProviderIface.pfnListOpen)
1304 rc = pTransfer->ProviderIface.pfnListOpen(&pTransfer->ProviderCtx, pOpenParms, phList);
1305 else
1306 rc = VERR_NOT_SUPPORTED;
1307
1308 if (RT_FAILURE(rc))
1309 LogRel(("Shared Clipboard: Opening list '%s' (fiter '%s', flags %#x) failed with %Rrc\n",
1310 pOpenParms->pszPath, pOpenParms->pszFilter, pOpenParms->fList, rc));
1311
1312 LogFlowFuncLeaveRC(rc);
1313 return rc;
1314}
1315
1316/**
1317 * Closes a transfer list.
1318 *
1319 * @returns VBox status code.
1320 * @param pTransfer Clipboard transfer to handle.
1321 * @param hList Handle of list to close.
1322 */
1323int ShClTransferListClose(PSHCLTRANSFER pTransfer, SHCLLISTHANDLE hList)
1324{
1325 AssertPtrReturn(pTransfer, VERR_INVALID_POINTER);
1326
1327 if (hList == NIL_SHCLLISTHANDLE)
1328 return VINF_SUCCESS;
1329
1330 int rc;
1331 if (pTransfer->ProviderIface.pfnListClose)
1332 rc = pTransfer->ProviderIface.pfnListClose(&pTransfer->ProviderCtx, hList);
1333 else
1334 rc = VERR_NOT_SUPPORTED;
1335
1336 if (RT_FAILURE(rc))
1337 LogRel(("Shared Clipboard: Closing list 0x%x entry failed with %Rrc\n", hList, rc));
1338
1339 LogFlowFuncLeaveRC(rc);
1340 return rc;
1341}
1342
1343/**
1344 * Retrieves the header of a transfer list.
1345 *
1346 * @returns VBox status code.
1347 * @param pTransfer Clipboard transfer to handle.
1348 * @param hList Handle of list to get header for.
1349 * @param pHdr Where to store the returned list header information.
1350 */
1351int ShClTransferListGetHeader(PSHCLTRANSFER pTransfer, SHCLLISTHANDLE hList,
1352 PSHCLLISTHDR pHdr)
1353{
1354 AssertPtrReturn(pTransfer, VERR_INVALID_POINTER);
1355 AssertPtrReturn(pHdr, VERR_INVALID_POINTER);
1356
1357 LogFlowFunc(("hList=%RU64\n", hList));
1358
1359 int rc;
1360 if (pTransfer->ProviderIface.pfnListHdrRead)
1361 rc = pTransfer->ProviderIface.pfnListHdrRead(&pTransfer->ProviderCtx, hList, pHdr);
1362 else
1363 rc = VERR_NOT_SUPPORTED;
1364
1365 if (RT_FAILURE(rc))
1366 LogRel(("Shared Clipboard: Reading list header list 0x%x entry failed with %Rrc\n", hList, rc));
1367
1368 LogFlowFuncLeaveRC(rc);
1369 return rc;
1370}
1371
1372/**
1373 * Returns a specific list handle info of a clipboard transfer.
1374 *
1375 * @returns Pointer to list handle info if found, or NULL if not found.
1376 * @param pTransfer Clipboard transfer to get list handle info from.
1377 * @param hList List handle of the list to get handle info for.
1378 */
1379PSHCLLISTHANDLEINFO ShClTransferListGetByHandle(PSHCLTRANSFER pTransfer, SHCLLISTHANDLE hList)
1380{
1381 PSHCLLISTHANDLEINFO pIt;
1382 RTListForEach(&pTransfer->lstHandles, pIt, SHCLLISTHANDLEINFO, Node) /** @todo Sloooow ... improve this. */
1383 {
1384 if (pIt->hList == hList)
1385 return pIt;
1386 }
1387
1388 return NULL;
1389}
1390
1391/**
1392 * Returns the a transfer object of a transfer list.
1393 *
1394 * Currently not implemented and wil return NULL.
1395 *
1396 * @returns Pointer to transfer object, or NULL if not found / invalid.
1397 * @param pTransfer Clipboard transfer to return transfer object for.
1398 * @param hList Handle of clipboard transfer list to get object for.
1399 * @param uIdx Index of object to get.
1400 */
1401PSHCLTRANSFEROBJ ShClTransferListGetObj(PSHCLTRANSFER pTransfer,
1402 SHCLLISTHANDLE hList, uint64_t uIdx)
1403{
1404 AssertPtrReturn(pTransfer, NULL);
1405
1406 RT_NOREF(hList, uIdx);
1407
1408 LogFlowFunc(("hList=%RU64\n", hList));
1409
1410 return NULL;
1411}
1412
1413/**
1414 * Reads a single transfer list entry.
1415 *
1416 * @returns VBox status code or VERR_NO_MORE_FILES if the end of the list has been reached.
1417 * @param pTransfer Clipboard transfer to handle.
1418 * @param hList List handle of list to read from.
1419 * @param pEntry Where to store the read information.
1420 */
1421int ShClTransferListRead(PSHCLTRANSFER pTransfer, SHCLLISTHANDLE hList,
1422 PSHCLLISTENTRY pEntry)
1423{
1424 AssertPtrReturn(pTransfer, VERR_INVALID_POINTER);
1425 AssertPtrReturn(pEntry, VERR_INVALID_POINTER);
1426
1427 LogFlowFunc(("hList=%RU64\n", hList));
1428
1429 int rc;
1430 if (pTransfer->ProviderIface.pfnListEntryRead)
1431 rc = pTransfer->ProviderIface.pfnListEntryRead(&pTransfer->ProviderCtx, hList, pEntry);
1432 else
1433 rc = VERR_NOT_SUPPORTED;
1434
1435 if (RT_FAILURE(rc))
1436 LogRel(("Shared Clipboard: Reading list for list 0x%x entry failed with %Rrc\n", hList, rc));
1437
1438 LogFlowFuncLeaveRC(rc);
1439 return rc;
1440}
1441
1442/**
1443 * Writes a single transfer list entry.
1444 *
1445 * @returns VBox status code.
1446 * @param pTransfer Clipboard transfer to handle.
1447 * @param hList List handle of list to write to.
1448 * @param pEntry Entry information to write.
1449 */
1450int ShClTransferListWrite(PSHCLTRANSFER pTransfer, SHCLLISTHANDLE hList,
1451 PSHCLLISTENTRY pEntry)
1452{
1453 RT_NOREF(pTransfer, hList, pEntry);
1454
1455 int rc = VINF_SUCCESS;
1456
1457#if 0
1458 if (pTransfer->ProviderIface.pfnListEntryWrite)
1459 rc = pTransfer->ProviderIface.pfnListEntryWrite(&pTransfer->ProviderCtx, hList, pEntry);
1460#endif
1461
1462 if (RT_FAILURE(rc))
1463 LogRel(("Shared Clipboard: Writing list entry to list 0x%x failed with %Rrc\n", hList, rc));
1464
1465 LogFlowFuncLeaveRC(rc);
1466 return rc;
1467}
1468
1469/**
1470 * Returns whether a given transfer list handle is valid or not.
1471 *
1472 * @returns \c true if list handle is valid, \c false if not.
1473 * @param pTransfer Clipboard transfer to handle.
1474 * @param hList List handle to check.
1475 */
1476bool ShClTransferListHandleIsValid(PSHCLTRANSFER pTransfer, SHCLLISTHANDLE hList)
1477{
1478 bool fIsValid = false;
1479
1480 if (pTransfer->State.enmSource == SHCLSOURCE_LOCAL)
1481 {
1482 fIsValid = ShClTransferListGetByHandle(pTransfer, hList) != NULL;
1483 }
1484 else if (pTransfer->State.enmSource == SHCLSOURCE_REMOTE)
1485 {
1486 AssertFailed(); /** @todo Implement. */
1487 }
1488 else
1489 AssertFailedStmt(fIsValid = false);
1490
1491 return fIsValid;
1492}
1493
1494/**
1495 * Copies a transfer callback table from source to destination.
1496 *
1497 * @param pCallbacksDst Callback destination.
1498 * @param pCallbacksSrc Callback source. If set to NULL, the
1499 * destination callback table will be unset.
1500 */
1501static void shClTransferCopyCallbacks(PSHCLTRANSFERCALLBACKS pCallbacksDst, PSHCLTRANSFERCALLBACKS pCallbacksSrc)
1502{
1503 AssertPtrReturnVoid(pCallbacksDst);
1504 /* pCallbacksSrc can be NULL */
1505
1506 if (pCallbacksSrc) /* Set */
1507 {
1508#define SET_CALLBACK(a_pfnCallback) \
1509 if (pCallbacksSrc->a_pfnCallback) \
1510 pCallbacksDst->a_pfnCallback = pCallbacksSrc->a_pfnCallback
1511
1512 SET_CALLBACK(pfnOnCreated);
1513 SET_CALLBACK(pfnOnInitialized);
1514 SET_CALLBACK(pfnOnDestroy);
1515 SET_CALLBACK(pfnOnStarted);
1516 SET_CALLBACK(pfnOnCompleted);
1517 SET_CALLBACK(pfnOnError);
1518 SET_CALLBACK(pfnOnRegistered);
1519 SET_CALLBACK(pfnOnUnregistered);
1520
1521#undef SET_CALLBACK
1522
1523 pCallbacksDst->pvUser = pCallbacksSrc->pvUser;
1524 pCallbacksDst->cbUser = pCallbacksSrc->cbUser;
1525 }
1526 else /* Unset */
1527 RT_BZERO(pCallbacksDst, sizeof(SHCLTRANSFERCALLBACKS));
1528}
1529
1530/**
1531 * Sets or unsets the callback table to be used for a clipboard transfer.
1532 *
1533 * @returns VBox status code.
1534 * @param pTransfer Clipboard transfer to set callbacks for.
1535 * @param pCallbacks Pointer to callback table to set. If set to NULL,
1536 * existing callbacks for this transfer will be unset.
1537 *
1538 * @note Must come before initializing the transfer via ShClTransferInit().
1539 */
1540static void shClTransferSetCallbacks(PSHCLTRANSFER pTransfer, PSHCLTRANSFERCALLBACKS pCallbacks)
1541{
1542 AssertPtrReturnVoid(pTransfer);
1543 /* pCallbacks can be NULL. */
1544
1545 shClTransferCopyCallbacks(&pTransfer->Callbacks, pCallbacks);
1546
1547 /* Make sure that the callback context has all values set according to the callback table.
1548 * This only needs to be done once, so do this here. */
1549 pTransfer->CallbackCtx.pTransfer = pTransfer;
1550 pTransfer->CallbackCtx.pvUser = pTransfer->Callbacks.pvUser;
1551 pTransfer->CallbackCtx.cbUser = pTransfer->Callbacks.cbUser;
1552}
1553
1554/**
1555 * Sets the transfer provider for a given transfer.
1556 *
1557 * @returns VBox status code.
1558 * @param pTransfer Transfer to create transfer provider for.
1559 * @param pProvider Provider to use.
1560 */
1561int ShClTransferSetProvider(PSHCLTRANSFER pTransfer, PSHCLTXPROVIDER pProvider)
1562{
1563 AssertPtrReturn(pTransfer, VERR_INVALID_POINTER);
1564 AssertPtrReturn(pProvider, VERR_INVALID_POINTER);
1565
1566 LogFlowFuncEnter();
1567
1568 int rc = VINF_SUCCESS;
1569
1570 pTransfer->ProviderIface = pProvider->Interface;
1571 pTransfer->ProviderCtx.pTransfer = pTransfer;
1572 pTransfer->ProviderCtx.pvUser = pProvider->pvUser;
1573 pTransfer->ProviderCtx.cbUser = pProvider->cbUser;
1574
1575 LogFlowFuncLeaveRC(rc);
1576 return rc;
1577}
1578
1579/**
1580 * Sets the current status.
1581 *
1582 * @returns VBox status code.
1583 * @param pTransfer Clipboard transfer to set status for.
1584 * @param enmStatus Status to set.
1585 *
1586 * @note Caller needs to take critical section.
1587 */
1588static int shClTransferSetStatus(PSHCLTRANSFER pTransfer, SHCLTRANSFERSTATUS enmStatus)
1589{
1590 Assert(RTCritSectIsOwner(&pTransfer->CritSect));
1591#if 0
1592 AssertMsgReturn(pTransfer->State.enmStatus != enmStatus,
1593 ("Setting the same status twice in a row (%#x), please report this!\n", enmStatus), VERR_WRONG_ORDER);
1594#endif
1595 pTransfer->State.enmStatus = enmStatus;
1596
1597 LogFlowFunc(("enmStatus=%s\n", ShClTransferStatusToStr(pTransfer->State.enmStatus)));
1598
1599 return RTSemEventSignal(pTransfer->StatusChangeEvent);
1600}
1601
1602/**
1603 * Returns the number of transfer root list entries.
1604 *
1605 * @returns Root list entry count.
1606 * @param pTransfer Clipboard transfer to return root entry count for.
1607 */
1608uint64_t ShClTransferRootsCount(PSHCLTRANSFER pTransfer)
1609{
1610 AssertPtrReturn(pTransfer, 0);
1611
1612 shClTransferLock(pTransfer);
1613
1614 uint32_t const cRoots = pTransfer->lstRoots.Hdr.cEntries;
1615
1616 shClTransferUnlock(pTransfer);
1617
1618 return cRoots;
1619}
1620
1621/**
1622 * Resets the root list of a clipboard transfer.
1623 *
1624 * @param pTransfer Transfer to clear transfer root list for.
1625 *
1626 * @note Caller needs to take critical section.
1627 */
1628static void shClTransferRootsReset(PSHCLTRANSFER pTransfer)
1629{
1630 AssertPtrReturnVoid(pTransfer);
1631 Assert(RTCritSectIsOwner(&pTransfer->CritSect));
1632
1633 if (pTransfer->pszPathRootAbs)
1634 {
1635 RTStrFree(pTransfer->pszPathRootAbs);
1636 pTransfer->pszPathRootAbs = NULL;
1637 }
1638
1639 ShClTransferListDestroy(&pTransfer->lstRoots);
1640}
1641
1642/**
1643 * Resets a clipboard transfer.
1644 *
1645 * @param pTransfer Clipboard transfer to reset.
1646 */
1647void ShClTransferReset(PSHCLTRANSFER pTransfer)
1648{
1649 AssertPtrReturnVoid(pTransfer);
1650
1651 LogFlowFuncEnter();
1652
1653 shClTransferLock(pTransfer);
1654
1655 shClTransferRootsReset(pTransfer);
1656
1657 PSHCLLISTHANDLEINFO pItList, pItListNext;
1658 RTListForEachSafe(&pTransfer->lstHandles, pItList, pItListNext, SHCLLISTHANDLEINFO, Node)
1659 {
1660 ShClTransferListHandleInfoDestroy(pItList);
1661
1662 RTListNodeRemove(&pItList->Node);
1663
1664 RTMemFree(pItList);
1665 }
1666
1667 PSHCLTRANSFEROBJ pItObj, pItObjNext;
1668 RTListForEachSafe(&pTransfer->lstObj, pItObj, pItObjNext, SHCLTRANSFEROBJ, Node)
1669 {
1670 ShClTransferObjDestroy(pItObj);
1671
1672 RTListNodeRemove(&pItObj->Node);
1673
1674 RTMemFree(pItObj);
1675 }
1676
1677 shClTransferUnlock(pTransfer);
1678}
1679
1680/**
1681 * Get a specific root list entry.
1682 *
1683 * @returns Const pointer to root list entry if found, or NULL if not found..
1684 * @param pTransfer Clipboard transfer to get root list entry of.
1685 * @param uIndex Index (zero-based) of entry to get.
1686 */
1687PCSHCLLISTENTRY ShClTransferRootsEntryGet(PSHCLTRANSFER pTransfer, uint64_t uIndex)
1688{
1689 AssertPtrReturn(pTransfer, NULL);
1690
1691 shClTransferLock(pTransfer);
1692
1693 if (uIndex >= pTransfer->lstRoots.Hdr.cEntries)
1694 {
1695 shClTransferUnlock(pTransfer);
1696 return NULL;
1697 }
1698
1699 PCSHCLLISTENTRY pEntry = shClTransferListGetEntryById(&pTransfer->lstRoots, uIndex);
1700
1701 shClTransferUnlock(pTransfer);
1702
1703 return pEntry;
1704}
1705
1706/**
1707 * Reads the root entries of a clipboard transfer.
1708 *
1709 * This gives the provider interface the chance of reading root entries information.
1710 *
1711 * @returns VBox status code.
1712 * @param pTransfer Clipboard transfer to read root list for.
1713 * Must be in STARTED state.
1714 */
1715int ShClTransferRootListRead(PSHCLTRANSFER pTransfer)
1716{
1717 AssertPtrReturn(pTransfer, VERR_INVALID_POINTER);
1718
1719 LogFlowFuncEnter();
1720
1721#ifdef DEBUG
1722 shClTransferLock(pTransfer);
1723 AssertMsgReturn( pTransfer->State.enmStatus == SHCLTRANSFERSTATUS_INITIALIZED
1724 || pTransfer->State.enmStatus == SHCLTRANSFERSTATUS_STARTED,
1725 ("Cannot read root list in status %s\n", ShClTransferStatusToStr(pTransfer->State.enmStatus)),
1726 VERR_WRONG_ORDER);
1727 shClTransferUnlock(pTransfer);
1728#endif
1729
1730 int rc;
1731 if (pTransfer->ProviderIface.pfnRootListRead)
1732 rc = pTransfer->ProviderIface.pfnRootListRead(&pTransfer->ProviderCtx);
1733 else
1734 rc = VERR_NOT_SUPPORTED;
1735
1736 shClTransferLock(pTransfer);
1737
1738 /* Make sure that we have at least an empty root path set. */
1739 if ( RT_SUCCESS(rc)
1740 && !pTransfer->pszPathRootAbs)
1741 {
1742 if (RTStrAPrintf(&pTransfer->pszPathRootAbs, "") < 0)
1743 rc = VERR_NO_MEMORY;
1744 }
1745
1746 shClTransferUnlock(pTransfer);
1747
1748 LogFlowFuncLeaveRC(rc);
1749 return rc;
1750}
1751
1752/**
1753 * Initializes the root list entries for a given clipboard transfer.
1754 *
1755 * @returns VBox status code.
1756 * @param pTransfer Transfer to set transfer list entries for.
1757 * @param pszRoots String list (separated by CRLF) of root entries to set.
1758 * All entries must have the same root path.
1759 * @param cbRoots Size (in bytes) of string list. Includes zero terminator.
1760 *
1761 * @note Accepts local paths or URI string lists (absolute only).
1762 */
1763int ShClTransferRootsInitFromStringList(PSHCLTRANSFER pTransfer, const char *pszRoots, size_t cbRoots)
1764{
1765 AssertPtrReturn(pTransfer, VERR_INVALID_POINTER);
1766 AssertPtrReturn(pszRoots, VERR_INVALID_POINTER);
1767 AssertReturn(cbRoots, VERR_INVALID_PARAMETER);
1768
1769#ifdef DEBUG_andy
1770 LogFlowFunc(("Data:\n%.*Rhxd\n", cbRoots, pszRoots));
1771#endif
1772
1773 if (!RTStrIsValidEncoding(pszRoots))
1774 return VERR_INVALID_UTF8_ENCODING;
1775
1776 int rc = VINF_SUCCESS;
1777
1778 shClTransferLock(pTransfer);
1779
1780 shClTransferRootsReset(pTransfer);
1781
1782 PSHCLLIST pLstRoots = &pTransfer->lstRoots;
1783 char *pszPathRootAbs = NULL;
1784
1785 RTCList<RTCString> lstRootEntries = RTCString(pszRoots, cbRoots).split(SHCL_TRANSFER_URI_LIST_SEP_STR);
1786 if (!lstRootEntries.size())
1787 {
1788 shClTransferUnlock(pTransfer);
1789 return VINF_SUCCESS;
1790 }
1791
1792 for (size_t i = 0; i < lstRootEntries.size(); ++i)
1793 {
1794 char *pszPathCur = NULL;
1795
1796 char *pszPath = NULL;
1797 rc = RTUriFilePathEx(lstRootEntries.at(i).c_str(),
1798 RTPATH_STR_F_STYLE_UNIX, &pszPath, 0 /*cbPath*/, NULL /*pcchPath*/);
1799 if (RT_SUCCESS(rc))
1800 {
1801 pszPathCur = pszPath;
1802 pszPath = NULL; /* pszPath has ownership now. */
1803 }
1804 else if (rc == VERR_URI_NOT_FILE_SCHEME) /* Local file path? */
1805 {
1806 pszPathCur = RTStrDup(lstRootEntries.at(i).c_str());
1807 rc = VINF_SUCCESS;
1808 }
1809
1810 LogFlowFunc(("pszPathCur=%s\n", pszPathCur));
1811
1812 rc = ShClTransferValidatePath(pszPathCur, false /* fMustExist */);
1813 if (RT_FAILURE(rc))
1814 {
1815 RT_BREAKPOINT();
1816 break;
1817 }
1818
1819 /* No root path determined yet? */
1820 if (!pszPathRootAbs)
1821 {
1822 pszPathRootAbs = RTStrDup(pszPathCur);
1823 if (pszPathRootAbs)
1824 {
1825 RTPathStripFilename(pszPathRootAbs);
1826
1827 LogFlowFunc(("pszPathRootAbs=%s\n", pszPathRootAbs));
1828
1829 /* We don't want to have a relative directory here. */
1830 if (RTPathStartsWithRoot(pszPathRootAbs))
1831 {
1832 rc = ShClTransferValidatePath(pszPathRootAbs, true /* Path must exist */);
1833 }
1834 else
1835 rc = VERR_PATH_IS_RELATIVE;
1836 }
1837 else
1838 rc = VERR_NO_MEMORY;
1839 }
1840
1841 if (RT_SUCCESS(rc))
1842 {
1843 PSHCLLISTENTRY pEntry;
1844 rc = ShClTransferListEntryAlloc(&pEntry);
1845 if (RT_SUCCESS(rc))
1846 {
1847 PSHCLFSOBJINFO pFsObjInfo = (PSHCLFSOBJINFO)RTMemAllocZ(sizeof(SHCLFSOBJINFO));
1848 if (pFsObjInfo)
1849 {
1850 if (pTransfer->State.enmDir == SHCLTRANSFERDIR_TO_REMOTE)
1851 rc = ShClFsObjInfoQueryLocal(pszPathCur, pFsObjInfo);
1852 if (RT_SUCCESS(rc))
1853 {
1854 /* Calculate the relative path within the root path. */
1855 const char *pszPathRelToRoot = &pszPathCur[strlen(pszPathRootAbs) + 1 /* Skip terminator or (back)slash. */];
1856 if ( pszPathRelToRoot
1857 && *pszPathRelToRoot != '\0')
1858 {
1859 LogFlowFunc(("pszPathRelToRoot=%s\n", pszPathRelToRoot));
1860
1861 rc = ShClTransferListEntryInitEx(pEntry, VBOX_SHCL_INFO_F_FSOBJINFO, pszPathRelToRoot,
1862 pFsObjInfo, sizeof(SHCLFSOBJINFO));
1863 if (RT_SUCCESS(rc))
1864 {
1865 rc = ShClTransferListAddEntry(pLstRoots, pEntry, true /* fAppend */);
1866 if (RT_SUCCESS(rc))
1867 {
1868 pFsObjInfo = NULL; /* pEntry has ownership now. */
1869 }
1870 }
1871 }
1872 else
1873 LogRel(("Shared Clipboard: Unable to construct relative path for '%s' (root is '%s')\n",
1874 pszPathCur, pszPathRootAbs));
1875 }
1876
1877 if (pFsObjInfo)
1878 {
1879 RTMemFree(pFsObjInfo);
1880 pFsObjInfo = NULL;
1881 }
1882 }
1883 else
1884 rc = VERR_NO_MEMORY;
1885
1886 if (RT_FAILURE(rc))
1887 ShClTransferListEntryFree(pEntry);
1888 }
1889 }
1890
1891 RTStrFree(pszPathCur);
1892 }
1893
1894 /* No (valid) root directory found? Bail out early. */
1895 if (!pszPathRootAbs)
1896 rc = VERR_PATH_DOES_NOT_START_WITH_ROOT;
1897
1898 if (RT_SUCCESS(rc))
1899 {
1900 pTransfer->pszPathRootAbs = pszPathRootAbs;
1901 LogFlowFunc(("pszPathRootAbs=%s, cRoots=%zu\n", pTransfer->pszPathRootAbs, pTransfer->lstRoots.Hdr.cEntries));
1902
1903 LogRel2(("Shared Clipboard: Transfer uses root '%s'\n", pTransfer->pszPathRootAbs));
1904 }
1905 else
1906 {
1907 LogRel(("Shared Clipboard: Unable to set roots for transfer, rc=%Rrc\n", rc));
1908 ShClTransferListDestroy(pLstRoots);
1909 RTStrFree(pszPathRootAbs);
1910 }
1911
1912 shClTransferUnlock(pTransfer);
1913
1914 LogFlowFuncLeaveRC(rc);
1915 return rc;
1916}
1917
1918/**
1919 * Initializes the root list entries for a given clipboard transfer, UTF-16 (Unicode) version.
1920 *
1921 * @returns VBox status code.
1922 * @param pTransfer Transfer to set transfer list entries for.
1923 * @param pwszRoots Unicode string list (separated by CRLF) of root entries to set.
1924 * All entries must have the same root path.
1925 * @param cbRoots Size (in bytes) of string list. Includes zero terminator.
1926 *
1927 * @note Accepts local paths or URI string lists (absolute only).
1928 */
1929int ShClTransferRootsInitFromStringListUnicode(PSHCLTRANSFER pTransfer, PRTUTF16 pwszRoots, size_t cbRoots)
1930{
1931 AssertPtrReturn(pwszRoots, VERR_INVALID_POINTER);
1932 AssertReturn(cbRoots, VERR_INVALID_PARAMETER);
1933 AssertReturn(cbRoots % sizeof(RTUTF16) == 0, VERR_INVALID_PARAMETER);
1934
1935 size_t cwcRoots = cbRoots / sizeof(RTUTF16);
1936
1937 /* This may slightly overestimate the space needed. */
1938 size_t chDst = 0;
1939 int rc = ShClUtf16LenUtf8(pwszRoots, cwcRoots, &chDst);
1940 if (RT_SUCCESS(rc))
1941 {
1942 chDst++; /* Add space for terminator. */
1943
1944 char *pszDst = (char *)RTStrAlloc(chDst);
1945 if (pszDst)
1946 {
1947 size_t cbActual = 0;
1948 rc = ShClConvUtf16CRLFToUtf8LF(pwszRoots, cwcRoots, pszDst, chDst, &cbActual);
1949 if (RT_SUCCESS(rc))
1950 rc = ShClTransferRootsInitFromStringList(pTransfer, pszDst, cbActual + 1 /* Include terminator */);
1951
1952 RTStrFree(pszDst);
1953 }
1954 else
1955 rc = VERR_NO_MEMORY;
1956 }
1957
1958 return rc;
1959}
1960
1961/**
1962 * Initializes a single file as a transfer root.
1963 *
1964 * @returns VBox status code.
1965 * @param pTransfer Transfer to set transfer list entries for.
1966 * @param pszFile File to use as transfer root.
1967 *
1968 * @note Convenience function, uses ShClTransferRootsSet() internally.
1969 */
1970int ShClTransferRootsInitFromFile(PSHCLTRANSFER pTransfer, const char *pszFile)
1971{
1972 char *pszRoots = NULL;
1973
1974 LogFlowFuncEnter();
1975
1976 int rc = RTStrAAppend(&pszRoots, pszFile);
1977 AssertRCReturn(rc, rc);
1978 rc = RTStrAAppend(&pszRoots, "\r\n");
1979 AssertRCReturn(rc, rc);
1980 rc = ShClTransferRootsInitFromStringList(pTransfer, pszRoots, strlen(pszRoots) + 1 /* Include terminator */);
1981 RTStrFree(pszRoots);
1982 return rc;
1983}
1984
1985/**
1986 * Returns the clipboard transfer's ID.
1987 *
1988 * @returns The transfer's ID.
1989 * @param pTransfer Clipboard transfer to return ID for.
1990 */
1991SHCLTRANSFERID ShClTransferGetID(PSHCLTRANSFER pTransfer)
1992{
1993 AssertPtrReturn(pTransfer, 0);
1994
1995 shClTransferLock(pTransfer);
1996
1997 SHCLTRANSFERID const uID = pTransfer->State.uID;
1998
1999 shClTransferUnlock(pTransfer);
2000
2001 return uID;
2002}
2003
2004/**
2005 * Returns the clipboard transfer's direction.
2006 *
2007 * @returns The transfer's direction.
2008 * @param pTransfer Clipboard transfer to return direction for.
2009 */
2010SHCLTRANSFERDIR ShClTransferGetDir(PSHCLTRANSFER pTransfer)
2011{
2012 AssertPtrReturn(pTransfer, SHCLTRANSFERDIR_UNKNOWN);
2013
2014 shClTransferLock(pTransfer);
2015
2016 SHCLTRANSFERDIR const enmDir = pTransfer->State.enmDir;
2017
2018 shClTransferUnlock(pTransfer);
2019
2020 return enmDir;
2021}
2022
2023/**
2024 * Returns the absolute root path of a transfer.
2025 *
2026 * @returns VBox status code.
2027 * @param pTransfer Clipboard transfer to return absolute root path for.
2028 * @param pszPath Where to store the returned path.
2029 * @param cbPath Size (in bytes) of \a pszPath.
2030 */
2031int ShClTransferGetRootPathAbs(PSHCLTRANSFER pTransfer, char *pszPath, size_t cbPath)
2032{
2033 AssertPtrReturn(pTransfer, VERR_INVALID_POINTER);
2034
2035 shClTransferLock(pTransfer);
2036
2037 AssertMsgReturn(pTransfer->pszPathRootAbs, ("Transfer has no root path set (yet)\n"), VERR_WRONG_ORDER);
2038
2039 int const rc = RTStrCopy(pszPath, cbPath, pTransfer->pszPathRootAbs);
2040
2041 shClTransferUnlock(pTransfer);
2042
2043 return rc;
2044}
2045
2046/**
2047 * Returns the transfer's source.
2048 *
2049 * @returns The transfer's source.
2050 * @param pTransfer Clipboard transfer to return source for.
2051 */
2052SHCLSOURCE ShClTransferGetSource(PSHCLTRANSFER pTransfer)
2053{
2054 AssertPtrReturn(pTransfer, SHCLSOURCE_INVALID);
2055
2056 shClTransferLock(pTransfer);
2057
2058 SHCLSOURCE const enmSource = pTransfer->State.enmSource;
2059
2060 shClTransferUnlock(pTransfer);
2061
2062 return enmSource;
2063}
2064
2065/**
2066 * Returns the current transfer status.
2067 *
2068 * @returns Current transfer status.
2069 * @param pTransfer Clipboard transfer to return status for.
2070 *
2071 * @note Caller needs to take critical section.
2072 */
2073DECLINLINE(SHCLTRANSFERSTATUS) shClTransferGetStatusLocked(PSHCLTRANSFER pTransfer)
2074{
2075 Assert(RTCritSectIsOwner(&pTransfer->CritSect));
2076
2077 shClTransferLock(pTransfer);
2078
2079 SHCLTRANSFERSTATUS const enmStatus = pTransfer->State.enmStatus;
2080
2081 shClTransferUnlock(pTransfer);
2082
2083 return enmStatus;
2084}
2085
2086/**
2087 * Returns the current transfer status.
2088 *
2089 * @returns Current transfer status.
2090 * @param pTransfer Clipboard transfer to return status for.
2091 */
2092SHCLTRANSFERSTATUS ShClTransferGetStatus(PSHCLTRANSFER pTransfer)
2093{
2094 AssertPtrReturn(pTransfer, SHCLTRANSFERSTATUS_NONE);
2095
2096 shClTransferLock(pTransfer);
2097
2098 SHCLTRANSFERSTATUS const enmSts = shClTransferGetStatusLocked(pTransfer);
2099
2100 shClTransferUnlock(pTransfer);
2101
2102 return enmSts;
2103}
2104
2105/**
2106 * Runs a started clipboard transfer in a dedicated thread.
2107 *
2108 * @returns VBox status code.
2109 * @param pTransfer Clipboard transfer to run.
2110 * @param pfnThreadFunc Pointer to thread function to use.
2111 * @param pvUser Pointer to user-provided data. Optional.
2112 */
2113int ShClTransferRun(PSHCLTRANSFER pTransfer, PFNSHCLTRANSFERTHREAD pfnThreadFunc, void *pvUser)
2114{
2115 AssertPtrReturn(pTransfer, VERR_INVALID_POINTER);
2116 AssertPtrReturn(pfnThreadFunc, VERR_INVALID_POINTER);
2117 /* pvUser is optional. */
2118
2119 AssertMsgReturn(pTransfer->State.enmStatus == SHCLTRANSFERSTATUS_STARTED,
2120 ("Wrong status (currently is %s)\n", ShClTransferStatusToStr(pTransfer->State.enmStatus)),
2121 VERR_WRONG_ORDER);
2122
2123 int rc = shClTransferThreadCreate(pTransfer, pfnThreadFunc, pvUser);
2124
2125 LogFlowFuncLeaveRC(rc);
2126 return rc;
2127}
2128
2129/**
2130 * Starts an initialized transfer.
2131 *
2132 * @returns VBox status code.
2133 * @param pTransfer Clipboard transfer to start.
2134 */
2135int ShClTransferStart(PSHCLTRANSFER pTransfer)
2136{
2137 AssertPtrReturn(pTransfer, VERR_INVALID_POINTER);
2138
2139 LogFlowFuncEnter();
2140
2141 shClTransferLock(pTransfer);
2142
2143 /* Ready to start? */
2144 AssertMsgReturnStmt(pTransfer->ProviderIface.pfnRootListRead != NULL,
2145 ("No provider interface set (yet)\n"),
2146 shClTransferUnlock(pTransfer), VERR_WRONG_ORDER);
2147 AssertMsgReturnStmt(pTransfer->State.enmStatus == SHCLTRANSFERSTATUS_INITIALIZED,
2148 ("Wrong status (currently is %s)\n", ShClTransferStatusToStr(pTransfer->State.enmStatus)),
2149 shClTransferUnlock(pTransfer), VERR_WRONG_ORDER);
2150
2151 int rc = shClTransferSetStatus(pTransfer, SHCLTRANSFERSTATUS_STARTED);
2152
2153 shClTransferUnlock(pTransfer);
2154
2155 if (pTransfer->Callbacks.pfnOnStarted)
2156 pTransfer->Callbacks.pfnOnStarted(&pTransfer->CallbackCtx);
2157
2158 LogFlowFuncLeaveRC(rc);
2159 return rc;
2160}
2161
2162/**
2163 * Internal struct for keeping a transfer thread context.
2164 */
2165typedef struct _SHCLTRANSFERTHREADCTX
2166{
2167 /** Pointer to transfer. */
2168 PSHCLTRANSFER pTransfer;
2169 /** User-supplied context data. Can be NULL if not being used. */
2170 void *pvUser;
2171 /** Pointer to thread function to use. */
2172 PFNSHCLTRANSFERTHREAD pfnThread;
2173} SHCLTRANSFERTHREADCTX;
2174/** Pointer to internal struct for keeping a transfer thread context. */
2175typedef SHCLTRANSFERTHREADCTX *PSHCLTRANSFERTHREADCTX;
2176
2177/**
2178 * Worker thread for a transfer.
2179 *
2180 * @returns VBox status code.
2181 * @param ThreadSelf Thread self handle. Not being used.
2182 * @param pvUser Context data of type PSHCLTRANSFERTHREADCTX.
2183 */
2184static DECLCALLBACK(int) shClTransferThreadWorker(RTTHREAD ThreadSelf, void *pvUser)
2185{
2186 RT_NOREF(ThreadSelf);
2187
2188 LogFlowFuncEnter();
2189
2190 SHCLTRANSFERTHREADCTX Ctx;
2191 memcpy(&Ctx, pvUser, sizeof(SHCLTRANSFERTHREADCTX));
2192
2193 LogFlowFunc(("pfnThread=%p, pTransfer=%p, pvUser=%p\n", Ctx.pfnThread, Ctx.pTransfer, Ctx.pvUser));
2194
2195 PSHCLTRANSFER pTransfer = Ctx.pTransfer;
2196
2197 shClTransferLock(pTransfer);
2198
2199 pTransfer->Thread.fStarted = true;
2200 pTransfer->Thread.fStop = false;
2201
2202 shClTransferUnlock(pTransfer);
2203
2204 RTThreadUserSignal(RTThreadSelf());
2205
2206 int rc = Ctx.pfnThread(pTransfer, Ctx.pvUser);
2207
2208 if (pTransfer->Callbacks.pfnOnCompleted)
2209 pTransfer->Callbacks.pfnOnCompleted(&pTransfer->CallbackCtx, rc);
2210
2211 LogFlowFuncLeaveRC(rc);
2212 return rc;
2213}
2214
2215/**
2216 * Creates a thread for a clipboard transfer.
2217 *
2218 * @returns VBox status code.
2219 * @param pTransfer Clipboard transfer to create thread for.
2220 * @param pfnThreadFunc Thread function to use for this transfer.
2221 * @param pvUser Pointer to user-provided data.
2222 */
2223static int shClTransferThreadCreate(PSHCLTRANSFER pTransfer, PFNSHCLTRANSFERTHREAD pfnThreadFunc, void *pvUser)
2224
2225{
2226 AssertPtrReturn(pTransfer, VERR_INVALID_POINTER);
2227
2228 shClTransferLock(pTransfer);
2229
2230 /* Already marked for stopping? */
2231 AssertMsgReturn(pTransfer->Thread.fStop == false,
2232 ("Transfer thread already marked for stopping"), VERR_WRONG_ORDER);
2233 /* Already started? */
2234 AssertMsgReturn(pTransfer->Thread.fStarted == false,
2235 ("Transfer thread already started"), VERR_WRONG_ORDER);
2236
2237 SHCLTRANSFERTHREADCTX Ctx = { pTransfer, pvUser, pfnThreadFunc };
2238
2239 /* Spawn a worker thread, so that we don't block the window thread for too long. */
2240 int rc = RTThreadCreate(&pTransfer->Thread.hThread, shClTransferThreadWorker,
2241 &Ctx, 0, RTTHREADTYPE_DEFAULT, RTTHREADFLAGS_WAITABLE,
2242 "shcltx");
2243 if (RT_SUCCESS(rc))
2244 {
2245 shClTransferUnlock(pTransfer); /* Leave lock while waiting. */
2246
2247 int rc2 = RTThreadUserWait(pTransfer->Thread.hThread, SHCL_TIMEOUT_DEFAULT_MS);
2248 AssertRC(rc2);
2249
2250 shClTransferLock(pTransfer);
2251
2252 if (pTransfer->Thread.fStarted) /* Did the thread indicate that it started correctly? */
2253 {
2254 /* Nothing to do in here. */
2255 }
2256 else
2257 rc = VERR_GENERAL_FAILURE; /** @todo Find a better rc. */
2258 }
2259
2260 shClTransferUnlock(pTransfer);
2261
2262 LogFlowFuncLeaveRC(rc);
2263 return rc;
2264}
2265
2266/**
2267 * Destroys the thread of a clipboard transfer.
2268 *
2269 * @returns VBox status code.
2270 * @param pTransfer Clipboard transfer to destroy thread for.
2271 * @param uTimeoutMs Timeout (in ms) to wait for thread destruction.
2272 */
2273static int shClTransferThreadDestroy(PSHCLTRANSFER pTransfer, RTMSINTERVAL uTimeoutMs)
2274{
2275 AssertPtrReturn(pTransfer, VERR_INVALID_POINTER);
2276
2277 shClTransferLock(pTransfer);
2278
2279 if (pTransfer->Thread.hThread == NIL_RTTHREAD)
2280 {
2281 shClTransferUnlock(pTransfer);
2282 return VINF_SUCCESS;
2283 }
2284
2285 LogFlowFuncEnter();
2286
2287 /* Set stop indicator. */
2288 pTransfer->Thread.fStop = true;
2289
2290 shClTransferUnlock(pTransfer); /* Leave lock while waiting. */
2291
2292 int rcThread = VERR_IPE_UNINITIALIZED_STATUS;
2293 int rc = RTThreadWait(pTransfer->Thread.hThread, uTimeoutMs, &rcThread);
2294
2295 LogFlowFunc(("Waiting for thread resulted in %Rrc (thread exited with %Rrc)\n", rc, rcThread));
2296
2297 return rc;
2298}
2299
2300/**
2301 * Waits for the transfer status to change, internal version.
2302 *
2303 * @returns VBox status code.
2304 * @param pTransfer Clipboard transfer to wait for.
2305 * @param msTimeout Timeout (in ms) to wait.
2306 * @param penmStatus Where to return the new (current) transfer status on success.
2307 * Optional and can be NULL.
2308 */
2309static int shClTransferWaitForStatusChangeInternal(PSHCLTRANSFER pTransfer, RTMSINTERVAL msTimeout, SHCLTRANSFERSTATUS *penmStatus)
2310{
2311 LogFlowFunc(("Waiting for status change (%RU32 timeout) ...\n", msTimeout));
2312
2313 int rc = RTSemEventWait(pTransfer->StatusChangeEvent, msTimeout);
2314 if (RT_SUCCESS(rc))
2315 {
2316 if (penmStatus)
2317 {
2318 shClTransferLock(pTransfer);
2319
2320 *penmStatus = pTransfer->State.enmStatus;
2321
2322 shClTransferUnlock(pTransfer);
2323 }
2324 }
2325
2326 return rc;
2327}
2328
2329/**
2330 * Waits for the transfer status to change.
2331 *
2332 * @returns VBox status code.
2333 * @param pTransfer Clipboard transfer to wait for.
2334 * @param msTimeout Timeout (in ms) to wait.
2335 * @param penmStatus Where to return the new (current) transfer status on success.
2336 * Optional and can be NULL.
2337 */
2338int ShClTransferWaitForStatusChange(PSHCLTRANSFER pTransfer, RTMSINTERVAL msTimeout, SHCLTRANSFERSTATUS *penmStatus)
2339{
2340 AssertPtrReturn(pTransfer, VERR_INVALID_POINTER);
2341
2342 int rc = shClTransferWaitForStatusChangeInternal(pTransfer, msTimeout, penmStatus);
2343
2344 LogFlowFuncLeaveRC(rc);
2345 return rc;
2346}
2347
2348/**
2349 * Waits for a specific transfer status.
2350 *
2351 * @returns VBox status code.
2352 * @param pTransfer Clipboard transfer to wait for.
2353 * @param msTimeout Timeout (in ms) to wait.
2354 * @param enmStatus Transfer status to wait for.
2355 */
2356int ShClTransferWaitForStatus(PSHCLTRANSFER pTransfer, RTMSINTERVAL msTimeout, SHCLTRANSFERSTATUS enmStatus)
2357{
2358 AssertPtrReturn(pTransfer, VERR_INVALID_POINTER);
2359
2360 int rc = VINF_SUCCESS;
2361
2362 uint64_t const tsStartMs = RTTimeMilliTS();
2363 uint64_t msLeft = msTimeout;
2364 for (;;)
2365 {
2366 SHCLTRANSFERSTATUS enmCurStatus;
2367 rc = shClTransferWaitForStatusChangeInternal(pTransfer, msLeft, &enmCurStatus);
2368 if (RT_FAILURE(rc))
2369 break;
2370
2371 if (enmCurStatus == enmStatus)
2372 break;
2373
2374 msLeft -= RT_MIN(msLeft, RTTimeMilliTS() - tsStartMs);
2375 if (msLeft == 0)
2376 {
2377 rc = VERR_TIMEOUT;
2378 break;
2379 }
2380 }
2381
2382 LogFlowFuncLeaveRC(rc);
2383 return rc;
2384}
2385
2386
2387/*********************************************************************************************************************************
2388 * Transfer Context *
2389 ********************************************************************************************************************************/
2390
2391/**
2392 * Locks a transfer context.
2393 *
2394 * @param pTransferCtx Transfer context to lock.
2395 */
2396DECLINLINE(void) shClTransferCtxLock(PSHCLTRANSFERCTX pTransferCtx)
2397{
2398 int rc2 = RTCritSectEnter(&pTransferCtx->CritSect);
2399 AssertRC(rc2);
2400}
2401
2402/**
2403 * Unlocks a transfer context.
2404 *
2405 * @param pTransferCtx Transfer context to unlock.
2406 */
2407DECLINLINE(void) shClTransferCtxUnlock(PSHCLTRANSFERCTX pTransferCtx)
2408{
2409 int rc2 = RTCritSectLeave(&pTransferCtx->CritSect);
2410 AssertRC(rc2);
2411}
2412
2413/**
2414 * Initializes a clipboard transfer context.
2415 *
2416 * @returns VBox status code.
2417 * @param pTransferCtx Transfer context to initialize.
2418 */
2419int ShClTransferCtxInit(PSHCLTRANSFERCTX pTransferCtx)
2420{
2421 AssertPtrReturn(pTransferCtx, VERR_INVALID_POINTER);
2422
2423 LogFlowFunc(("pTransferCtx=%p\n", pTransferCtx));
2424
2425 int rc = RTCritSectInit(&pTransferCtx->CritSect);
2426 if (RT_SUCCESS(rc))
2427 {
2428 rc = RTSemEventCreate(&pTransferCtx->ChangedEvent);
2429 if (RT_SUCCESS(rc))
2430 {
2431 RT_ZERO(pTransferCtx->ChangedEventData);
2432
2433 RTListInit(&pTransferCtx->List);
2434
2435 pTransferCtx->cTransfers = 0;
2436 pTransferCtx->cRunning = 0;
2437 pTransferCtx->cMaxRunning = 64; /** @todo Make this configurable? */
2438
2439 RT_ZERO(pTransferCtx->bmTransferIds);
2440
2441 ShClTransferCtxReset(pTransferCtx);
2442 }
2443 }
2444
2445 return VINF_SUCCESS;
2446}
2447
2448/**
2449 * Destroys a clipboard transfer context.
2450 *
2451 * @param pTransferCtx Transfer context to destroy.
2452 */
2453void ShClTransferCtxDestroy(PSHCLTRANSFERCTX pTransferCtx)
2454{
2455 if (!pTransferCtx)
2456 return;
2457
2458 LogFlowFunc(("pTransferCtx=%p\n", pTransferCtx));
2459
2460 shClTransferCtxLock(pTransferCtx);
2461
2462 PSHCLTRANSFER pTransfer, pTransferNext;
2463 RTListForEachSafe(&pTransferCtx->List, pTransfer, pTransferNext, SHCLTRANSFER, Node)
2464 {
2465 ShClTransferDestroy(pTransfer);
2466
2467 shclTransferCtxTransferRemoveAndUnregister(pTransferCtx, pTransfer);
2468
2469 RTMemFree(pTransfer);
2470 pTransfer = NULL;
2471 }
2472
2473 pTransferCtx->cRunning = 0;
2474 pTransferCtx->cTransfers = 0;
2475
2476 shClTransferCtxUnlock(pTransferCtx);
2477
2478 RTSemEventDestroy(pTransferCtx->ChangedEvent);
2479 pTransferCtx->ChangedEvent = NIL_RTSEMEVENT;
2480
2481 if (RTCritSectIsInitialized(&pTransferCtx->CritSect))
2482 RTCritSectDelete(&pTransferCtx->CritSect);
2483}
2484
2485/**
2486 * Resets a clipboard transfer context.
2487 *
2488 * @param pTransferCtx Transfer context to reset.
2489 */
2490void ShClTransferCtxReset(PSHCLTRANSFERCTX pTransferCtx)
2491{
2492 AssertPtrReturnVoid(pTransferCtx);
2493
2494 shClTransferCtxLock(pTransferCtx);
2495
2496 LogFlowFuncEnter();
2497
2498 PSHCLTRANSFER pTransfer;
2499 RTListForEach(&pTransferCtx->List, pTransfer, SHCLTRANSFER, Node)
2500 ShClTransferReset(pTransfer);
2501
2502#ifdef VBOX_WITH_SHARED_CLIPBOARD_TRANSFERS_HTTP
2503 /** @todo Anything to do here? */
2504#endif
2505
2506 shClTransferCtxUnlock(pTransferCtx);
2507}
2508
2509/**
2510 * Signals a change event.
2511 *
2512 * @returns VBox status code.
2513 * @param pTransferCtx Transfer context to return transfer for.
2514 * @param fRegistered Whether a transfer got registered or unregistered.
2515 * @param pTransfer Transfer bound to the event.
2516 */
2517static int shClTransferCtxSignal(PSHCLTRANSFERCTX pTransferCtx, bool fRegistered, PSHCLTRANSFER pTransfer)
2518{
2519 Assert(RTCritSectIsOwner(&pTransferCtx->CritSect));
2520
2521 LogFlowFunc(("fRegistered=%RTbool, pTransfer=%p\n", fRegistered, pTransfer));
2522
2523 pTransferCtx->ChangedEventData.fRegistered = fRegistered;
2524 pTransferCtx->ChangedEventData.pTransfer = pTransfer;
2525
2526 return RTSemEventSignal(pTransferCtx->ChangedEvent);
2527}
2528
2529/**
2530 * Returns a specific clipboard transfer, internal version.
2531 *
2532 * @returns Clipboard transfer found, or NULL if not found.
2533 * @param pTransferCtx Transfer context to return transfer for.
2534 * @param idTransfer ID of the transfer to return.
2535 *
2536 * @note Caller needs to take critical section.
2537 */
2538static PSHCLTRANSFER shClTransferCtxGetTransferByIdInternal(PSHCLTRANSFERCTX pTransferCtx, SHCLTRANSFERID idTransfer)
2539{
2540 Assert(RTCritSectIsOwner(&pTransferCtx->CritSect));
2541
2542 PSHCLTRANSFER pTransfer;
2543 RTListForEach(&pTransferCtx->List, pTransfer, SHCLTRANSFER, Node) /** @todo Slow, but works for now. */
2544 {
2545 if (pTransfer->State.uID == idTransfer)
2546 return pTransfer;
2547 }
2548
2549 return NULL;
2550}
2551
2552/**
2553 * Returns a specific clipboard transfer by index, internal version.
2554 *
2555 * @returns Clipboard transfer found, or NULL if not found.
2556 * @param pTransferCtx Transfer context to return transfer for.
2557 * @param uIdx Index of the transfer to return.
2558 *
2559 * @note Caller needs to take critical section.
2560 */
2561static PSHCLTRANSFER shClTransferCtxGetTransferByIndexInternal(PSHCLTRANSFERCTX pTransferCtx, uint32_t uIdx)
2562{
2563 Assert(RTCritSectIsOwner(&pTransferCtx->CritSect));
2564
2565 uint32_t idx = 0;
2566
2567 PSHCLTRANSFER pTransfer;
2568 RTListForEach(&pTransferCtx->List, pTransfer, SHCLTRANSFER, Node) /** @todo Slow, but works for now. */
2569 {
2570 if (uIdx == idx)
2571 return pTransfer;
2572 idx++;
2573 }
2574
2575 return NULL;
2576}
2577
2578/**
2579 * Returns a clipboard transfer for a specific transfer ID.
2580 *
2581 * @returns Clipboard transfer found, or NULL if not found.
2582 * @param pTransferCtx Transfer context to return transfer for.
2583 * @param uID ID of the transfer to return.
2584 */
2585PSHCLTRANSFER ShClTransferCtxGetTransferById(PSHCLTRANSFERCTX pTransferCtx, uint32_t uID)
2586{
2587 shClTransferCtxLock(pTransferCtx);
2588
2589 PSHCLTRANSFER const pTransfer = shClTransferCtxGetTransferByIdInternal(pTransferCtx, uID);
2590
2591 shClTransferCtxUnlock(pTransferCtx);
2592
2593 return pTransfer;
2594}
2595
2596/**
2597 * Returns a clipboard transfer for a specific list index.
2598 *
2599 * @returns Clipboard transfer found, or NULL if not found.
2600 * @param pTransferCtx Transfer context to return transfer for.
2601 * @param uIdx List index of the transfer to return.
2602 */
2603PSHCLTRANSFER ShClTransferCtxGetTransferByIndex(PSHCLTRANSFERCTX pTransferCtx, uint32_t uIdx)
2604{
2605 shClTransferCtxLock(pTransferCtx);
2606
2607 PSHCLTRANSFER const pTransfer = shClTransferCtxGetTransferByIndexInternal(pTransferCtx, uIdx);
2608
2609 shClTransferCtxUnlock(pTransferCtx);
2610
2611 return pTransfer;
2612}
2613
2614
2615/**
2616 * Returns the last clipboard transfer registered.
2617 *
2618 * @returns Clipboard transfer found, or NULL if not found.
2619 * @param pTransferCtx Transfer context to return transfer for.
2620 */
2621PSHCLTRANSFER ShClTransferCtxGetTransferLast(PSHCLTRANSFERCTX pTransferCtx)
2622{
2623 shClTransferCtxLock(pTransferCtx);
2624
2625 PSHCLTRANSFER const pTransfer = RTListGetLast(&pTransferCtx->List, SHCLTRANSFER, Node);
2626
2627 shClTransferCtxUnlock(pTransferCtx);
2628
2629 return pTransfer;
2630}
2631
2632/**
2633 * Returns the number of running clipboard transfers for a given transfer context.
2634 *
2635 * @returns Number of running transfers.
2636 * @param pTransferCtx Transfer context to return number for.
2637 */
2638uint32_t ShClTransferCtxGetRunningTransfers(PSHCLTRANSFERCTX pTransferCtx)
2639{
2640 AssertPtrReturn(pTransferCtx, 0);
2641
2642 shClTransferCtxLock(pTransferCtx);
2643
2644 uint32_t const cRunning = pTransferCtx->cRunning;
2645
2646 shClTransferCtxUnlock(pTransferCtx);
2647
2648 return cRunning;
2649}
2650
2651/**
2652 * Returns the number of total clipboard transfers for a given transfer context.
2653 *
2654 * @returns Number of total transfers.
2655 * @param pTransferCtx Transfer context to return number for.
2656 */
2657uint32_t ShClTransferCtxGetTotalTransfers(PSHCLTRANSFERCTX pTransferCtx)
2658{
2659 AssertPtrReturn(pTransferCtx, 0);
2660
2661 shClTransferCtxLock(pTransferCtx);
2662
2663 uint32_t const cTransfers = pTransferCtx->cTransfers;
2664
2665 shClTransferCtxUnlock(pTransferCtx);
2666
2667 return cTransfers;
2668}
2669
2670static int shClTransferCreateIDInternal(PSHCLTRANSFERCTX pTransferCtx, SHCLTRANSFERID *pidTransfer)
2671{
2672 /*
2673 * Pick a random bit as starting point. If it's in use, search forward
2674 * for a free one, wrapping around. We've reserved both the zero'th and
2675 * max-1 IDs.
2676 */
2677 SHCLTRANSFERID idTransfer = RTRandU32Ex(1, VBOX_SHCL_MAX_TRANSFERS - 2);
2678
2679 if (!ASMBitTestAndSet(&pTransferCtx->bmTransferIds[0], idTransfer))
2680 { /* likely */ }
2681 else if (pTransferCtx->cTransfers < VBOX_SHCL_MAX_TRANSFERS - 2 /* First and last are not used */)
2682 {
2683 /* Forward search. */
2684 int iHit = ASMBitNextClear(&pTransferCtx->bmTransferIds[0], VBOX_SHCL_MAX_TRANSFERS, idTransfer);
2685 if (iHit < 0)
2686 iHit = ASMBitFirstClear(&pTransferCtx->bmTransferIds[0], VBOX_SHCL_MAX_TRANSFERS);
2687 AssertLogRelMsgReturn(iHit >= 0, ("Transfer count: %RU16\n", pTransferCtx->cTransfers), VERR_SHCLPB_MAX_TRANSFERS_REACHED);
2688 idTransfer = iHit;
2689 AssertLogRelMsgReturn(!ASMBitTestAndSet(&pTransferCtx->bmTransferIds[0], idTransfer), ("idObject=%#x\n", idTransfer), VERR_INTERNAL_ERROR_2);
2690 }
2691 else
2692 {
2693 LogFunc(("Maximum number of transfers reached (%RU16 transfers)\n", pTransferCtx->cTransfers));
2694 return VERR_SHCLPB_MAX_TRANSFERS_REACHED;
2695 }
2696
2697 *pidTransfer = idTransfer;
2698
2699 return VINF_SUCCESS;
2700}
2701
2702/**
2703 * Creates a new transfer ID for a given transfer context.
2704 *
2705 * @returns VBox status code.
2706 * @retval VERR_SHCLPB_MAX_TRANSFERS_REACHED if the maximum of concurrent transfers is reached.
2707 * @param pTransferCtx Transfer context to create transfer ID for.
2708 * @param pidTransfer Where to return the transfer ID on success.
2709 */
2710int ShClTransferCtxCreateId(PSHCLTRANSFERCTX pTransferCtx, PSHCLTRANSFERID pidTransfer)
2711{
2712 AssertPtrReturn(pTransferCtx, VERR_INVALID_POINTER);
2713 AssertPtrReturn(pidTransfer, VERR_INVALID_POINTER);
2714
2715 shClTransferCtxLock(pTransferCtx);
2716
2717 int rc = shClTransferCreateIDInternal(pTransferCtx, pidTransfer);
2718
2719 shClTransferCtxUnlock(pTransferCtx);
2720
2721 return rc;
2722}
2723
2724/**
2725 * Registers a clipboard transfer with a new transfer ID.
2726 *
2727 * @return VBox status code.
2728 * @retval VERR_SHCLPB_MAX_TRANSFERS_REACHED if the maximum of concurrent transfers is reached.
2729 * @param pTransferCtx Transfer context to register transfer to.
2730 * @param pTransfer Transfer to register. The context takes ownership of the transfer on success.
2731 * @param idTransfer Transfer ID to use for registering the given transfer.
2732 */
2733static int shClTransferCtxTransferRegisterExInternal(PSHCLTRANSFERCTX pTransferCtx, PSHCLTRANSFER pTransfer, SHCLTRANSFERID idTransfer)
2734{
2735 AssertPtrReturn(pTransferCtx, VERR_INVALID_POINTER);
2736 AssertPtrReturn(pTransfer, VERR_INVALID_POINTER);
2737 Assert(idTransfer != NIL_SHCLTRANSFERID);
2738
2739 shClTransferCtxLock(pTransferCtx);
2740
2741 pTransfer->State.uID = idTransfer;
2742
2743 RTListAppend(&pTransferCtx->List, &pTransfer->Node);
2744
2745 pTransferCtx->cTransfers++;
2746
2747 Log2Func(("pTransfer=%p, idTransfer=%RU32 -- now %RU16 transfer(s)\n", pTransfer, idTransfer, pTransferCtx->cTransfers));
2748
2749 shClTransferCtxUnlock(pTransferCtx);
2750
2751 if (pTransfer->Callbacks.pfnOnRegistered)
2752 pTransfer->Callbacks.pfnOnRegistered(&pTransfer->CallbackCtx, pTransferCtx);
2753
2754 LogFlowFuncLeaveRC(VINF_SUCCESS);
2755 return VINF_SUCCESS;
2756}
2757
2758/**
2759 * Registers a clipboard transfer with a transfer context, i.e. allocates a transfer ID.
2760 *
2761 * @return VBox status code.
2762 * @retval VERR_SHCLPB_MAX_TRANSFERS_REACHED if the maximum of concurrent transfers for this context has been reached.
2763 * @param pTransferCtx Transfer context to register transfer to.
2764 * @param pTransfer Transfer to register. The context takes ownership of the transfer on success.
2765 * @param pidTransfer Where to return the transfer ID on success. Optional.
2766 */
2767int ShClTransferCtxRegister(PSHCLTRANSFERCTX pTransferCtx, PSHCLTRANSFER pTransfer, PSHCLTRANSFERID pidTransfer)
2768{
2769 AssertPtrReturn(pTransferCtx, VERR_INVALID_POINTER);
2770 AssertPtrReturn(pTransfer, VERR_INVALID_POINTER);
2771
2772 shClTransferCtxLock(pTransferCtx);
2773
2774 SHCLTRANSFERID idTransfer;
2775 int rc = shClTransferCreateIDInternal(pTransferCtx, &idTransfer);
2776 if (RT_SUCCESS(rc))
2777 {
2778 rc = shClTransferCtxTransferRegisterExInternal(pTransferCtx, pTransfer, idTransfer);
2779 if (RT_SUCCESS(rc))
2780 {
2781 if (pidTransfer)
2782 *pidTransfer = idTransfer;
2783 }
2784 }
2785
2786 shClTransferCtxUnlock(pTransferCtx);
2787
2788 return rc;
2789}
2790
2791/**
2792 * Registers a clipboard transfer with a transfer context by specifying an ID for the transfer.
2793 *
2794 * @return VBox status code.
2795 * @retval VERR_ALREADY_EXISTS if a transfer with the given ID already exists.
2796 * @retval VERR_SHCLPB_MAX_TRANSFERS_REACHED if the maximum of concurrent transfers for this context has been reached.
2797 * @param pTransferCtx Transfer context to register transfer to.
2798 * @param pTransfer Transfer to register.
2799 * @param idTransfer Transfer ID to use for registration.
2800 *
2801 * @note This function ASSUMES you have created \a idTransfer with ShClTransferCtxCreateId().
2802 */
2803int ShClTransferCtxRegisterById(PSHCLTRANSFERCTX pTransferCtx, PSHCLTRANSFER pTransfer, SHCLTRANSFERID idTransfer)
2804{
2805 shClTransferCtxLock(pTransferCtx);
2806
2807 if (pTransferCtx->cTransfers < VBOX_SHCL_MAX_TRANSFERS - 2 /* First and last are not used */)
2808 {
2809 RTListAppend(&pTransferCtx->List, &pTransfer->Node);
2810
2811 shClTransferLock(pTransfer);
2812
2813 pTransfer->State.uID = idTransfer;
2814
2815 shClTransferUnlock(pTransfer);
2816
2817 int rc = shClTransferCtxSignal(pTransferCtx, true /* fRegistered */, pTransfer);
2818
2819 pTransferCtx->cTransfers++;
2820
2821 LogFunc(("Registered transfer ID %RU16 -- now %RU16 transfers total\n", idTransfer, pTransferCtx->cTransfers));
2822
2823 shClTransferCtxUnlock(pTransferCtx);
2824
2825 if (pTransfer->Callbacks.pfnOnRegistered)
2826 pTransfer->Callbacks.pfnOnRegistered(&pTransfer->CallbackCtx, pTransferCtx);
2827
2828 return rc;
2829 }
2830
2831 LogFunc(("Maximum number of transfers reached (%RU16 transfers)\n", pTransferCtx->cTransfers));
2832
2833 shClTransferCtxUnlock(pTransferCtx);
2834
2835 return VERR_SHCLPB_MAX_TRANSFERS_REACHED;
2836}
2837
2838/**
2839 * Removes and unregisters a transfer from a transfer context.
2840 *
2841 * @param pTransferCtx Transfer context to remove transfer from.
2842 * @param pTransfer Transfer to remove.
2843 *
2844 * @note Caller needs to take critical section.
2845 */
2846static void shclTransferCtxTransferRemoveAndUnregister(PSHCLTRANSFERCTX pTransferCtx, PSHCLTRANSFER pTransfer)
2847{
2848 Assert(RTCritSectIsOwner(&pTransferCtx->CritSect));
2849
2850 RTListNodeRemove(&pTransfer->Node);
2851
2852 Assert(pTransferCtx->cTransfers);
2853 pTransferCtx->cTransfers--;
2854
2855 Assert(pTransferCtx->cTransfers >= pTransferCtx->cRunning);
2856
2857 shClTransferCtxUnlock(pTransferCtx);
2858
2859 if (pTransfer->Callbacks.pfnOnUnregistered)
2860 pTransfer->Callbacks.pfnOnUnregistered(&pTransfer->CallbackCtx, pTransferCtx);
2861
2862 shClTransferCtxLock(pTransferCtx);
2863
2864 LogFlowFunc(("Now %RU32 transfers left\n", pTransferCtx->cTransfers));
2865}
2866
2867/**
2868 * Unregisters a transfer from an transfer context, given by its ID.
2869 *
2870 * @retval VINF_SUCCESS on success.
2871 * @retval VERR_NOT_FOUND if the transfer ID was not found.
2872 * @param pTransferCtx Transfer context to unregister transfer from.
2873 * @param idTransfer Transfer ID to unregister.
2874 */
2875int ShClTransferCtxUnregisterById(PSHCLTRANSFERCTX pTransferCtx, SHCLTRANSFERID idTransfer)
2876{
2877 AssertPtrReturn(pTransferCtx, VERR_INVALID_POINTER);
2878 AssertReturn(idTransfer, VERR_INVALID_PARAMETER);
2879
2880 shClTransferCtxLock(pTransferCtx);
2881
2882 int rc = VINF_SUCCESS;
2883
2884 LogFlowFunc(("idTransfer=%RU32\n", idTransfer));
2885
2886 if (ASMBitTestAndClear(&pTransferCtx->bmTransferIds, idTransfer), ("idTransfer=%#x\n", idTransfer))
2887 {
2888 PSHCLTRANSFER pTransfer = shClTransferCtxGetTransferByIdInternal(pTransferCtx, idTransfer);
2889 AssertPtr(pTransfer);
2890 if (pTransfer)
2891 {
2892 shclTransferCtxTransferRemoveAndUnregister(pTransferCtx, pTransfer);
2893
2894 rc = shClTransferCtxSignal(pTransferCtx, false /* fRegistered */, pTransfer);
2895 }
2896 }
2897 else
2898 rc = VERR_NOT_FOUND;
2899
2900 shClTransferCtxUnlock(pTransferCtx);
2901
2902 LogFlowFuncLeaveRC(rc);
2903 return rc;
2904}
2905
2906/**
2907 * Waits for a transfer context event.
2908 *
2909 * @returns VBox status code.
2910 * @param pTransferCtx Transfer context to wait for.
2911 * @param msTimeout Timeout (in ms) to wait.
2912 * @param pEvent Where to return the event data on success.
2913 */
2914static int shClTransferCtxWaitInternal(PSHCLTRANSFERCTX pTransferCtx, RTMSINTERVAL msTimeout, PSHCLTRANSFERCTXEVENT pEvent)
2915{
2916 LogFlowFunc(("Waiting for transfer context change (%RU32 timeout) ...\n", msTimeout));
2917
2918 int rc = RTSemEventWait(pTransferCtx->ChangedEvent, msTimeout);
2919 if (RT_SUCCESS(rc))
2920 {
2921 shClTransferCtxLock(pTransferCtx);
2922
2923 memcpy(pEvent, &pTransferCtx->ChangedEventData, sizeof(SHCLTRANSFERCTXEVENT));
2924
2925 shClTransferCtxUnlock(pTransferCtx);
2926 }
2927
2928 LogFlowFuncLeaveRC(rc);
2929 return rc;
2930}
2931
2932/**
2933 * Waits for transfer to be (un-)registered.
2934 *
2935 * @returns VBox status code.
2936 * @param pTransferCtx Transfer context to wait for.
2937 * @param msTimeout Timeout (in ms) to wait.
2938 * @param fRegister Pass \c true for registering, or \c false for unregistering a transfer.
2939 * @param idTransfer Transfer ID to wait for.
2940 * Pass NIL_SHCLTRANSFERID for any transfer.
2941 * @param ppTransfer Where to return the transfer being (un-)registered. Optional and can be NULL.
2942 */
2943int ShClTransferCtxWait(PSHCLTRANSFERCTX pTransferCtx, RTMSINTERVAL msTimeout, bool fRegister, SHCLTRANSFERID idTransfer,
2944 PSHCLTRANSFER *ppTransfer)
2945{
2946 AssertPtrReturn(pTransferCtx, VERR_INVALID_POINTER);
2947
2948 int rc = VERR_TIMEOUT;
2949
2950 uint64_t const tsStartMs = RTTimeMilliTS();
2951 uint64_t msLeft = msTimeout;
2952 for (;;)
2953 {
2954 SHCLTRANSFERCTXEVENT Event;
2955 rc = shClTransferCtxWaitInternal(pTransferCtx, msLeft, &Event);
2956 if (RT_FAILURE(rc))
2957 break;
2958
2959 shClTransferCtxLock(pTransferCtx);
2960
2961 if (Event.fRegistered == fRegister)
2962 {
2963 if ( idTransfer != NIL_SHCLTRANSFERID
2964 && Event.pTransfer
2965 && ShClTransferGetID(Event.pTransfer) == idTransfer)
2966 {
2967 if (ppTransfer)
2968 *ppTransfer = Event.pTransfer;
2969 rc = VINF_SUCCESS;
2970 }
2971 }
2972
2973 shClTransferCtxUnlock(pTransferCtx);
2974
2975 if (RT_SUCCESS(rc))
2976 break;
2977
2978 msLeft -= RT_MIN(msLeft, RTTimeMilliTS() - tsStartMs);
2979 if (msLeft == 0)
2980 break;
2981 }
2982
2983 LogFlowFuncLeaveRC(rc);
2984 return rc;
2985}
2986
2987/**
2988 * Cleans up all associated transfers which are not needed (anymore).
2989 * This can be due to transfers which only have been announced but not / never being run.
2990 *
2991 * @param pTransferCtx Transfer context to cleanup transfers for.
2992 */
2993void ShClTransferCtxCleanup(PSHCLTRANSFERCTX pTransferCtx)
2994{
2995 AssertPtrReturnVoid(pTransferCtx);
2996
2997 shClTransferCtxLock(pTransferCtx);
2998
2999 LogFlowFunc(("pTransferCtx=%p, cTransfers=%RU16 cRunning=%RU16\n",
3000 pTransferCtx, pTransferCtx->cTransfers, pTransferCtx->cRunning));
3001
3002 if (pTransferCtx->cTransfers == 0)
3003 {
3004 shClTransferCtxUnlock(pTransferCtx);
3005 return;
3006 }
3007
3008 /* Remove all transfers which are not in a running state (e.g. only announced). */
3009 PSHCLTRANSFER pTransfer, pTransferNext;
3010 RTListForEachSafe(&pTransferCtx->List, pTransfer, pTransferNext, SHCLTRANSFER, Node)
3011 {
3012 shClTransferLock(pTransfer);
3013
3014 SHCLTRANSFERSTATUS const enmStatus = shClTransferGetStatusLocked(pTransfer);
3015 LogFlowFunc(("\tTransfer #%RU16: %s\n", pTransfer->State.uID, ShClTransferStatusToStr(enmStatus)));
3016
3017 if (enmStatus != SHCLTRANSFERSTATUS_STARTED)
3018 {
3019 shClTransferUnlock(pTransfer);
3020
3021 shclTransferCtxTransferRemoveAndUnregister(pTransferCtx, pTransfer);
3022
3023 ShClTransferDestroy(pTransfer);
3024
3025 RTMemFree(pTransfer);
3026 pTransfer = NULL;
3027 }
3028 else
3029 shClTransferUnlock(pTransfer);
3030 }
3031
3032 shClTransferCtxUnlock(pTransferCtx);
3033}
3034
3035/**
3036 * Returns whether the maximum of concurrent transfers of a specific transfer contexthas been reached or not.
3037 *
3038 * @returns \c if maximum has been reached, \c false if not.
3039 * @param pTransferCtx Transfer context to determine value for.
3040 */
3041bool ShClTransferCtxIsMaximumReached(PSHCLTRANSFERCTX pTransferCtx)
3042{
3043 AssertPtrReturn(pTransferCtx, true);
3044
3045 shClTransferCtxLock(pTransferCtx);
3046
3047 LogFlowFunc(("cRunning=%RU32, cMaxRunning=%RU32\n", pTransferCtx->cRunning, pTransferCtx->cMaxRunning));
3048
3049 Assert(pTransferCtx->cRunning <= pTransferCtx->cMaxRunning);
3050 bool const fMaximumReached = pTransferCtx->cRunning == pTransferCtx->cMaxRunning;
3051
3052 shClTransferCtxUnlock(pTransferCtx);
3053
3054 return fMaximumReached;
3055}
3056
3057/**
3058 * Copies file system objinfo from IPRT to Shared Clipboard format.
3059 *
3060 * @return VBox status code.
3061 * @param pDst The Shared Clipboard structure to convert data to.
3062 * @param pSrc The IPRT structure to convert data from.
3063 */
3064int ShClFsObjInfoFromIPRT(PSHCLFSOBJINFO pDst, PCRTFSOBJINFO pSrc)
3065{
3066 AssertPtrReturn(pDst, VERR_INVALID_POINTER);
3067 AssertPtrReturn(pSrc, VERR_INVALID_POINTER);
3068
3069 pDst->cbObject = pSrc->cbObject;
3070 pDst->cbAllocated = pSrc->cbAllocated;
3071 pDst->AccessTime = pSrc->AccessTime;
3072 pDst->ModificationTime = pSrc->ModificationTime;
3073 pDst->ChangeTime = pSrc->ChangeTime;
3074 pDst->BirthTime = pSrc->BirthTime;
3075 pDst->Attr.fMode = pSrc->Attr.fMode;
3076 /* Clear bits which we don't pass through for security reasons. */
3077 pDst->Attr.fMode &= ~(RTFS_UNIX_ISUID | RTFS_UNIX_ISGID | RTFS_UNIX_ISTXT);
3078 RT_ZERO(pDst->Attr.u);
3079 switch (pSrc->Attr.enmAdditional)
3080 {
3081 default:
3082 RT_FALL_THROUGH();
3083 case RTFSOBJATTRADD_NOTHING:
3084 pDst->Attr.enmAdditional = SHCLFSOBJATTRADD_NOTHING;
3085 break;
3086
3087 case RTFSOBJATTRADD_UNIX:
3088 pDst->Attr.enmAdditional = SHCLFSOBJATTRADD_UNIX;
3089 pDst->Attr.u.Unix.uid = pSrc->Attr.u.Unix.uid;
3090 pDst->Attr.u.Unix.gid = pSrc->Attr.u.Unix.gid;
3091 pDst->Attr.u.Unix.cHardlinks = pSrc->Attr.u.Unix.cHardlinks;
3092 pDst->Attr.u.Unix.INodeIdDevice = pSrc->Attr.u.Unix.INodeIdDevice;
3093 pDst->Attr.u.Unix.INodeId = pSrc->Attr.u.Unix.INodeId;
3094 pDst->Attr.u.Unix.fFlags = pSrc->Attr.u.Unix.fFlags;
3095 pDst->Attr.u.Unix.GenerationId = pSrc->Attr.u.Unix.GenerationId;
3096 pDst->Attr.u.Unix.Device = pSrc->Attr.u.Unix.Device;
3097 break;
3098
3099 case RTFSOBJATTRADD_EASIZE:
3100 pDst->Attr.enmAdditional = SHCLFSOBJATTRADD_EASIZE;
3101 pDst->Attr.u.EASize.cb = pSrc->Attr.u.EASize.cb;
3102 break;
3103 }
3104
3105 return VINF_SUCCESS;
3106}
3107
3108/**
3109 * Queries local file system information from a given path.
3110 *
3111 * @returns VBox status code.
3112 * @param pszPath Path to query file system information for.
3113 * @param pObjInfo Where to return the queried file system information on success.
3114 */
3115int ShClFsObjInfoQueryLocal(const char *pszPath, PSHCLFSOBJINFO pObjInfo)
3116{
3117 RTFSOBJINFO objInfo;
3118 int rc = RTPathQueryInfo(pszPath, &objInfo,
3119#if defined(RT_OS_WINDOWS) || defined(RT_OS_OS2)
3120 RTFSOBJATTRADD_NOTHING
3121#else
3122 RTFSOBJATTRADD_UNIX
3123#endif
3124 );
3125 if (RT_SUCCESS(rc))
3126 rc = ShClFsObjInfoFromIPRT(pObjInfo, &objInfo);
3127
3128 return rc;
3129}
3130
3131/**
3132 * Translates a clipboard transfer status (SHCLTRANSFERSTATUS_XXX) into a string.
3133 *
3134 * @returns Transfer status string name.
3135 * @param enmStatus The transfer status to translate.
3136 */
3137const char *ShClTransferStatusToStr(SHCLTRANSFERSTATUS enmStatus)
3138{
3139 switch (enmStatus)
3140 {
3141 RT_CASE_RET_STR(SHCLTRANSFERSTATUS_NONE);
3142 RT_CASE_RET_STR(SHCLTRANSFERSTATUS_REQUESTED);
3143 RT_CASE_RET_STR(SHCLTRANSFERSTATUS_INITIALIZED);
3144 RT_CASE_RET_STR(SHCLTRANSFERSTATUS_UNINITIALIZED);
3145 RT_CASE_RET_STR(SHCLTRANSFERSTATUS_STARTED);
3146 RT_CASE_RET_STR(SHCLTRANSFERSTATUS_STOPPED);
3147 RT_CASE_RET_STR(SHCLTRANSFERSTATUS_CANCELED);
3148 RT_CASE_RET_STR(SHCLTRANSFERSTATUS_KILLED);
3149 RT_CASE_RET_STR(SHCLTRANSFERSTATUS_ERROR);
3150 }
3151 return "Unknown";
3152}
3153
3154/**
3155 * Validates whether a given path matches our set of rules or not.
3156 *
3157 * @returns VBox status code.
3158 * @param pcszPath Path to validate.
3159 * @param fMustExist Whether the path to validate also must exist.
3160 */
3161int ShClTransferValidatePath(const char *pcszPath, bool fMustExist)
3162{
3163 int rc = VINF_SUCCESS;
3164
3165 if (!strlen(pcszPath))
3166 rc = VERR_INVALID_PARAMETER;
3167
3168 if ( RT_SUCCESS(rc)
3169 && !RTStrIsValidEncoding(pcszPath))
3170 {
3171 rc = VERR_INVALID_UTF8_ENCODING;
3172 }
3173
3174 if ( RT_SUCCESS(rc)
3175 && RTStrStr(pcszPath, ".."))
3176 {
3177 rc = VERR_INVALID_PARAMETER;
3178 }
3179
3180 if ( RT_SUCCESS(rc)
3181 && fMustExist)
3182 {
3183 RTFSOBJINFO objInfo;
3184 rc = RTPathQueryInfo(pcszPath, &objInfo, RTFSOBJATTRADD_NOTHING);
3185 if (RT_SUCCESS(rc))
3186 {
3187 if (RTFS_IS_DIRECTORY(objInfo.Attr.fMode))
3188 {
3189 if (!RTDirExists(pcszPath)) /* Path must exist. */
3190 rc = VERR_PATH_NOT_FOUND;
3191 }
3192 else if (RTFS_IS_FILE(objInfo.Attr.fMode))
3193 {
3194 if (!RTFileExists(pcszPath)) /* File must exist. */
3195 rc = VERR_FILE_NOT_FOUND;
3196 }
3197 else /* Everything else (e.g. symbolic links) are not supported. */
3198 {
3199 LogRelMax(64, ("Shared Clipboard: Path '%s' contains a symbolic link or junction, which are not supported\n", pcszPath));
3200 rc = VERR_NOT_SUPPORTED;
3201 }
3202 }
3203 }
3204
3205 if (RT_FAILURE(rc))
3206 LogRelMax(64, ("Shared Clipboard: Validating path '%s' failed: %Rrc\n", pcszPath, rc));
3207
3208 LogFlowFuncLeaveRC(rc);
3209 return rc;
3210}
3211
3212/**
3213 * Resolves a relative path of a specific transfer to its absolute path.
3214 *
3215 * @returns VBox status code.
3216 * @param pTransfer Clipboard transfer to resolve path for.
3217 * @param pszPath Relative path to resolve.
3218 * Paths have to end with a (back-)slash, otherwise this is considered to be a file.
3219 * @param fFlags Resolve flags. Currently not used and must be 0.
3220 * @param ppszResolved Where to store the allocated resolved path. Must be free'd by the called using RTStrFree().
3221 */
3222int ShClTransferResolvePathAbs(PSHCLTRANSFER pTransfer, const char *pszPath, uint32_t fFlags, char **ppszResolved)
3223{
3224 AssertPtrReturn(pTransfer, VERR_INVALID_POINTER);
3225 AssertPtrReturn(pszPath, VERR_INVALID_POINTER);
3226 AssertReturn (fFlags == 0, VERR_INVALID_PARAMETER);
3227 AssertPtrReturn(ppszResolved, VERR_INVALID_POINTER);
3228
3229 LogFlowFunc(("pszPath=%s, fFlags=%#x (pszPathRootAbs=%s, cRootListEntries=%RU64)\n",
3230 pszPath, fFlags, pTransfer->pszPathRootAbs, pTransfer->lstRoots.Hdr.cEntries));
3231
3232 int rc = ShClTransferValidatePath(pszPath, false /* fMustExist */);
3233 if (RT_SUCCESS(rc))
3234 {
3235 rc = VERR_PATH_NOT_FOUND; /* Play safe by default. */
3236
3237 PSHCLLISTENTRY pEntry;
3238 RTListForEach(&pTransfer->lstRoots.lstEntries, pEntry, SHCLLISTENTRY, Node)
3239 {
3240 LogFlowFunc(("\tpEntry->pszName=%s\n", pEntry->pszName));
3241
3242 if (RTStrStartsWith(pszPath, pEntry->pszName)) /* Case-sensitive! */
3243 {
3244 rc = VINF_SUCCESS;
3245 break;
3246 }
3247 }
3248
3249 if (RT_SUCCESS(rc))
3250 {
3251 char *pszPathAbs = RTPathJoinA(pTransfer->pszPathRootAbs, pszPath);
3252 if (pszPathAbs)
3253 {
3254 char szResolved[RTPATH_MAX];
3255 size_t cbResolved = sizeof(szResolved);
3256 rc = RTPathAbsEx(pTransfer->pszPathRootAbs, pszPathAbs, RTPATH_STR_F_STYLE_HOST, szResolved, &cbResolved);
3257
3258 RTStrFree(pszPathAbs);
3259 pszPathAbs = NULL;
3260
3261 if (RT_SUCCESS(rc))
3262 {
3263 LogRel2(("Shared Clipboard: Resolved: '%s' -> '%s'\n", pszPath, szResolved));
3264
3265 *ppszResolved = RTStrDup(szResolved);
3266 }
3267 }
3268 else
3269 rc = VERR_NO_MEMORY;
3270 }
3271 }
3272
3273 if (RT_FAILURE(rc))
3274 LogRel(("Shared Clipboard: Resolving absolute path for '%s' failed, rc=%Rrc\n", pszPath, rc));
3275
3276 LogFlowFuncLeaveRC(rc);
3277 return rc;
3278}
3279
3280/**
3281 * Converts Shared Clipboard create flags (see SharedClipboard-transfers.h) into IPRT create flags.
3282 *
3283 * @returns IPRT status code.
3284 * @param fShClFlags Shared clipboard create flags.
3285 * @param[out] pfOpen Where to store the RTFILE_O_XXX flags for
3286 * RTFileOpen.
3287 *
3288 * @sa Initially taken from vbsfConvertFileOpenFlags().
3289 */
3290int ShClTransferConvertFileCreateFlags(uint32_t fShClFlags, uint64_t *pfOpen)
3291{
3292 AssertMsgReturnStmt(!(fShClFlags & ~SHCL_OBJ_CF_VALID_MASK), ("%#x4\n", fShClFlags), *pfOpen = 0, VERR_INVALID_FLAGS);
3293
3294 uint64_t fOpen = 0;
3295
3296 switch (fShClFlags & SHCL_OBJ_CF_ACCESS_MASK_RW)
3297 {
3298 case SHCL_OBJ_CF_ACCESS_NONE:
3299 {
3300#ifdef RT_OS_WINDOWS
3301 if ((fShClFlags & SHCL_OBJ_CF_ACCESS_MASK_ATTR) != SHCL_OBJ_CF_ACCESS_ATTR_NONE)
3302 fOpen |= RTFILE_O_OPEN | RTFILE_O_ATTR_ONLY;
3303 else
3304#endif
3305 fOpen |= RTFILE_O_OPEN | RTFILE_O_READ;
3306 LogFlowFunc(("SHCL_OBJ_CF_ACCESS_NONE\n"));
3307 break;
3308 }
3309
3310 case SHCL_OBJ_CF_ACCESS_READ:
3311 {
3312 fOpen |= RTFILE_O_OPEN | RTFILE_O_READ;
3313 LogFlowFunc(("SHCL_OBJ_CF_ACCESS_READ\n"));
3314 break;
3315 }
3316
3317 default:
3318 AssertFailedReturn(VERR_IPE_NOT_REACHED_DEFAULT_CASE);
3319 }
3320
3321 switch (fShClFlags & SHCL_OBJ_CF_ACCESS_MASK_ATTR)
3322 {
3323 case SHCL_OBJ_CF_ACCESS_ATTR_NONE:
3324 {
3325 fOpen |= RTFILE_O_ACCESS_ATTR_DEFAULT;
3326 LogFlowFunc(("SHCL_OBJ_CF_ACCESS_ATTR_NONE\n"));
3327 break;
3328 }
3329
3330 case SHCL_OBJ_CF_ACCESS_ATTR_READ:
3331 {
3332 fOpen |= RTFILE_O_ACCESS_ATTR_READ;
3333 LogFlowFunc(("SHCL_OBJ_CF_ACCESS_ATTR_READ\n"));
3334 break;
3335 }
3336
3337 default:
3338 AssertFailedReturn(VERR_IPE_NOT_REACHED_DEFAULT_CASE);
3339 }
3340
3341 /* Sharing mask */
3342 switch (fShClFlags & SHCL_OBJ_CF_ACCESS_MASK_DENY)
3343 {
3344 case SHCL_OBJ_CF_ACCESS_DENYNONE:
3345 fOpen |= RTFILE_O_DENY_NONE;
3346 LogFlowFunc(("SHCL_OBJ_CF_ACCESS_DENYNONE\n"));
3347 break;
3348
3349 case SHCL_OBJ_CF_ACCESS_DENYWRITE:
3350 fOpen |= RTFILE_O_DENY_WRITE;
3351 LogFlowFunc(("SHCL_OBJ_CF_ACCESS_DENYWRITE\n"));
3352 break;
3353
3354 default:
3355 AssertFailedReturn(VERR_IPE_NOT_REACHED_DEFAULT_CASE);
3356 }
3357
3358 *pfOpen = fOpen;
3359
3360 LogFlowFuncLeaveRC(VINF_SUCCESS);
3361 return VINF_SUCCESS;
3362}
3363
Note: See TracBrowser for help on using the repository browser.

© 2024 Oracle Support Privacy / Do Not Sell My Info Terms of Use Trademark Policy Automated Access Etiquette