2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

43747b1f0bc8302a238fb35e55857a5e9aa1933dvboxsync#define LOG_GROUP_MAIN_OVERRIDE LOG_GROUP_HGCM

43747b1f0bc8302a238fb35e55857a5e9aa1933dvboxsync#include "Logging.h"

43747b1f0bc8302a238fb35e55857a5e9aa1933dvboxsync

43747b1f0bc8302a238fb35e55857a5e9aa1933dvboxsync#include "hgcm/HGCM.h"

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync#include "hgcm/HGCMThread.h"

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync#include <VBox/err.h>

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync#include <VBox/hgcmsvc.h>

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync#include <iprt/alloc.h>

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync#include <iprt/alloca.h>

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync#include <iprt/avl.h>

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync#include <iprt/critsect.h>

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync#include <iprt/asm.h>

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync#include <iprt/ldr.h>

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync#include <iprt/param.h>

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync#include <iprt/path.h>

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync#include <iprt/string.h>

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync#include <iprt/semaphore.h>

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync#include <iprt/thread.h>

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync#include <VBox/VMMDev.h>

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

79baed6836ae36c5f15b182292387484dcf7a752vboxsync

79baed6836ae36c5f15b182292387484dcf7a752vboxsync

79baed6836ae36c5f15b182292387484dcf7a752vboxsync#define VBOX_HGCM_SVC_NAME_MAX_BYTES 1024

79baed6836ae36c5f15b182292387484dcf7a752vboxsync

79baed6836ae36c5f15b182292387484dcf7a752vboxsyncstruct _HGCMSVCEXTHANDLEDATA

79baed6836ae36c5f15b182292387484dcf7a752vboxsync{

79baed6836ae36c5f15b182292387484dcf7a752vboxsync char *pszServiceName;

79baed6836ae36c5f15b182292387484dcf7a752vboxsync /* The service name follows. */

79baed6836ae36c5f15b182292387484dcf7a752vboxsync};

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsyncclass HGCMService

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync{

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync private:

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync VBOXHGCMSVCHELPERS m_svcHelpers;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync static HGCMService *sm_pSvcListHead;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync static HGCMService *sm_pSvcListTail;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync static int sm_cServices;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync HGCMTHREADHANDLE m_thread;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync friend DECLCALLBACK(void) hgcmServiceThread (HGCMTHREADHANDLE ThreadHandle, void *pvUser);

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync uint32_t volatile m_u32RefCnt;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync HGCMService *m_pSvcNext;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync HGCMService *m_pSvcPrev;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync char *m_pszSvcName;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync char *m_pszSvcLibrary;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync RTLDRMOD m_hLdrMod;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync PFNVBOXHGCMSVCLOAD m_pfnLoad;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync VBOXHGCMSVCFNTABLE m_fntable;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync int m_cClients;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync int m_cClientsAllocated;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync uint32_t *m_paClientIds;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync#ifdef VBOX_WITH_CRHGSMI

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync uint32_t m_cHandleAcquires;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync#endif

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync HGCMSVCEXTHANDLE m_hExtension;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync int loadServiceDLL (void);

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync void unloadServiceDLL (void);

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync /*

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync int instanceCreate (const char *pszServiceLibrary, const char *pszServiceName);

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync void instanceDestroy (void);

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync int saveClientState(uint32_t u32ClientId, PSSMHANDLE pSSM);

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync int loadClientState(uint32_t u32ClientId, PSSMHANDLE pSSM);

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync HGCMService ();

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync ~HGCMService () {};

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync static DECLCALLBACK(void) svcHlpCallComplete (VBOXHGCMCALLHANDLE callHandle, int32_t rc);

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync static DECLCALLBACK(void) svcHlpDisconnectClient (void *pvInstance, uint32_t u32ClientId);

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync public:

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync /*

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync static int LoadService (const char *pszServiceLibrary, const char *pszServiceName);

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync void UnloadService (void);

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync static void UnloadAll (void);

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync static int ResolveService (HGCMService **ppsvc, const char *pszServiceName);

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync void ReferenceService (void);

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync void ReleaseService (void);

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync static void Reset (void);

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync static int SaveState (PSSMHANDLE pSSM);

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync static int LoadState (PSSMHANDLE pSSM);

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync int CreateAndConnectClient (uint32_t *pu32ClientIdOut, uint32_t u32ClientIdIn);

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync int DisconnectClient (uint32_t u32ClientId, bool fFromService);

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync int HostCall (uint32_t u32Function, uint32_t cParms, VBOXHGCMSVCPARM *paParms);

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync#ifdef VBOX_WITH_CRHGSMI

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync int HandleAcquired();

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync int HandleReleased();

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync int HostFastCallAsync (uint32_t u32Function, VBOXHGCMSVCPARM *pParm, PHGCMHOSTFASTCALLCB pfnCompletion, void *pvCompletion);

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync#endif

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync uint32_t SizeOfClient (void) { return m_fntable.cbClient; };

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync int RegisterExtension (HGCMSVCEXTHANDLE handle, PFNHGCMSVCEXT pfnExtension, void *pvExtension);

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync void UnregisterExtension (HGCMSVCEXTHANDLE handle);

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync /*

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync int GuestCall (PPDMIHGCMPORT pHGCMPort, PVBOXHGCMCMD pCmd, uint32_t u32ClientId, uint32_t u32Function, uint32_t cParms, VBOXHGCMSVCPARM aParms[]);

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync};

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsyncclass HGCMClient: public HGCMObject

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync{

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync public:

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync HGCMClient () : HGCMObject(HGCMOBJ_CLIENT), pService(NULL),

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync pvData(NULL) {};

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync ~HGCMClient ();

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync int Init (HGCMService *pSvc);

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync /** Service that the client is connected to. */

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync HGCMService *pService;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync /** Client specific data. */

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync void *pvData;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync};

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsyncHGCMClient::~HGCMClient ()

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync{

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync if (pService->SizeOfClient () > 0)

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync RTMemFree (pvData);

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync}

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsyncint HGCMClient::Init (HGCMService *pSvc)

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync{

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync pService = pSvc;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync if (pService->SizeOfClient () > 0)

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync {

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync pvData = RTMemAllocZ (pService->SizeOfClient ());

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync if (!pvData)

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync {

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync return VERR_NO_MEMORY;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync }

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync }

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync return VINF_SUCCESS;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync}

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync#define HGCM_CLIENT_DATA(pService, pClient) (pClient->pvData)

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsyncHGCMService *HGCMService::sm_pSvcListHead = NULL;

ad27e1d5e48ca41245120c331cc88b50464813cevboxsyncHGCMService *HGCMService::sm_pSvcListTail = NULL;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsyncint HGCMService::sm_cServices = 0;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsyncHGCMService::HGCMService ()

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync :

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync m_thread (0),

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync m_u32RefCnt (0),

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync m_pSvcNext (NULL),

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync m_pSvcPrev (NULL),

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync m_pszSvcName (NULL),

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync m_pszSvcLibrary (NULL),

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync m_hLdrMod (NIL_RTLDRMOD),

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync m_pfnLoad (NULL),

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync m_cClients (0),

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync m_cClientsAllocated (0),

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync m_paClientIds (NULL),

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync#ifdef VBOX_WITH_CRHGSMI

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync m_cHandleAcquires (0),

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync#endif

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync m_hExtension (NULL)

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync{

ad27e1d5e48ca41245120c331cc88b50464813cevboxsync memset (&m_fntable, 0, sizeof (m_fntable));

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync}

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsyncstatic bool g_fResetting = false;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsyncstatic bool g_fSaveState = false;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsyncint HGCMService::loadServiceDLL (void)

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync{

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync LogFlowFunc(("m_pszSvcLibrary = %s\n", m_pszSvcLibrary));

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync if (m_pszSvcLibrary == NULL)

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync {

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync return VERR_INVALID_PARAMETER;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync }

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync char szErr[8192];

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync szErr[0] = '\0';

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync int rc = SUPR3HardenedLdrLoadAppPriv (m_pszSvcLibrary, &m_hLdrMod, szErr, sizeof(szErr));

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync if (RT_SUCCESS(rc))

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync {

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync LogFlowFunc(("successfully loaded the library.\n"));

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync m_pfnLoad = NULL;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync rc = RTLdrGetSymbol (m_hLdrMod, VBOX_HGCM_SVCLOAD_NAME, (void**)&m_pfnLoad);

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync if (RT_FAILURE(rc) || !m_pfnLoad)

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync {

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync Log(("HGCMService::loadServiceDLL: Error resolving the service entry point %s, rc = %d, m_pfnLoad = %p\n", VBOX_HGCM_SVCLOAD_NAME, rc, m_pfnLoad));

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync if (RT_SUCCESS(rc))

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync {

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync /* m_pfnLoad was NULL */

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync rc = VERR_SYMBOL_NOT_FOUND;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync }

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync }

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync if (RT_SUCCESS(rc))

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync {

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync memset (&m_fntable, 0, sizeof (m_fntable));

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync m_fntable.cbSize = sizeof (m_fntable);

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync m_fntable.u32Version = VBOX_HGCM_SVC_VERSION;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync m_fntable.pHelpers = &m_svcHelpers;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync rc = m_pfnLoad (&m_fntable);

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync LogFlowFunc(("m_pfnLoad rc = %Rrc\n", rc));

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync if (RT_SUCCESS(rc))

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync {

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync if ( m_fntable.pfnUnload == NULL

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync || m_fntable.pfnConnect == NULL

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync || m_fntable.pfnDisconnect == NULL

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync || m_fntable.pfnCall == NULL

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync )

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync {

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync Log(("HGCMService::loadServiceDLL: at least one of function pointers is NULL\n"));

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync rc = VERR_INVALID_PARAMETER;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync if (m_fntable.pfnUnload)

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync {

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync m_fntable.pfnUnload (m_fntable.pvService);

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync }

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync }

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync }

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync }

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync }

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync else

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync {

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync LogRel(("HGCM: Failed to load the service library: [%s], rc = %Rrc - %s. The service will be not available.\n",

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync m_pszSvcLibrary, rc, szErr));

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync m_hLdrMod = NIL_RTLDRMOD;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync }

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync if (RT_FAILURE(rc))

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync {

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync unloadServiceDLL ();

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync }

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync return rc;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync}

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

58ab0ad45444e80bdd970ddeb468d0872dbbbb47vboxsyncvoid HGCMService::unloadServiceDLL (void)

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync{

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync if (m_hLdrMod)

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync {

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync RTLdrClose (m_hLdrMod);

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync }

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync memset (&m_fntable, 0, sizeof (m_fntable));

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync m_pfnLoad = NULL;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync m_hLdrMod = NIL_RTLDRMOD;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync}

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync#define SVC_MSG_LOAD (0) /* Load the service library and call VBOX_HGCM_SVCLOAD_NAME entry point. */

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync#define SVC_MSG_UNLOAD (1) /* call pfnUnload and unload the service library. */

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync#define SVC_MSG_CONNECT (2) /* pfnConnect */

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync#define SVC_MSG_DISCONNECT (3) /* pfnDisconnect */

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync#define SVC_MSG_GUESTCALL (4) /* pfnGuestCall */

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync#define SVC_MSG_HOSTCALL (5) /* pfnHostCall */

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync#define SVC_MSG_LOADSTATE (6) /* pfnLoadState. */

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync#define SVC_MSG_SAVESTATE (7) /* pfnSaveState. */

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync#define SVC_MSG_QUIT (8) /* Terminate the thread. */

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync#define SVC_MSG_REGEXT (9) /* pfnRegisterExtension */

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync#define SVC_MSG_UNREGEXT (10) /* pfnRegisterExtension */

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync#ifdef VBOX_WITH_CRHGSMI

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync# define SVC_MSG_HOSTFASTCALLASYNC (21) /* pfnHostCall */

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync#endif

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsyncclass HGCMMsgSvcLoad: public HGCMMsgCore

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync{

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync};

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsyncclass HGCMMsgSvcUnload: public HGCMMsgCore

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync{

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync};

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsyncclass HGCMMsgSvcConnect: public HGCMMsgCore

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync{

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync public:

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync /* client identifier */

58ab0ad45444e80bdd970ddeb468d0872dbbbb47vboxsync uint32_t u32ClientId;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync};

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsyncclass HGCMMsgSvcDisconnect: public HGCMMsgCore

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync{

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync public:

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync /* client identifier */

58ab0ad45444e80bdd970ddeb468d0872dbbbb47vboxsync uint32_t u32ClientId;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync};

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsyncclass HGCMMsgHeader: public HGCMMsgCore

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync{

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync public:

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync HGCMMsgHeader () : pCmd (NULL), pHGCMPort (NULL) {};

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync /* Command pointer/identifier. */

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync PVBOXHGCMCMD pCmd;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync /* Port to be informed on message completion. */

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync PPDMIHGCMPORT pHGCMPort;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync};

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsyncclass HGCMMsgCall: public HGCMMsgHeader

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync{

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync public:

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync /* client identifier */

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync uint32_t u32ClientId;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync /* function number */

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync uint32_t u32Function;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync /* number of parameters */

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync uint32_t cParms;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync VBOXHGCMSVCPARM *paParms;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync};

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsyncclass HGCMMsgLoadSaveStateClient: public HGCMMsgCore

58ab0ad45444e80bdd970ddeb468d0872dbbbb47vboxsync{

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync public:

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync uint32_t u32ClientId;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync PSSMHANDLE pSSM;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync};

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsyncclass HGCMMsgHostCallSvc: public HGCMMsgCore

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync{

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync public:

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync /* function number */

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync uint32_t u32Function;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync /* number of parameters */

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync uint32_t cParms;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync VBOXHGCMSVCPARM *paParms;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync};

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsyncclass HGCMMsgSvcRegisterExtension: public HGCMMsgCore

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync{

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync public:

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync /* Handle of the extension to be registered. */

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync HGCMSVCEXTHANDLE handle;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync /* The extension entry point. */

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync PFNHGCMSVCEXT pfnExtension;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync /* The extension pointer. */

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync void *pvExtension;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync};

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsyncclass HGCMMsgSvcUnregisterExtension: public HGCMMsgCore

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync{

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync public:

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync /* Handle of the registered extension. */

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync HGCMSVCEXTHANDLE handle;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync};

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync#ifdef VBOX_WITH_CRHGSMI

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsyncclass HGCMMsgHostFastCallAsyncSvc: public HGCMMsgCore

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync{

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync public:

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync /* function number */

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync uint32_t u32Function;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync /* parameter */

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync VBOXHGCMSVCPARM Param;

11d3005e2935c925665896fa26fde09b3e656d70vboxsync /* completion info */

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync PHGCMHOSTFASTCALLCB pfnCompletion;

11d3005e2935c925665896fa26fde09b3e656d70vboxsync void *pvCompletion;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync};

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync#endif

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsyncstatic HGCMMsgCore *hgcmMessageAllocSvc (uint32_t u32MsgId)

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync{

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync switch (u32MsgId)

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync {

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync#ifdef VBOX_WITH_CRHGSMI

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync case SVC_MSG_HOSTFASTCALLASYNC: return new HGCMMsgHostFastCallAsyncSvc ();

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync#endif

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync case SVC_MSG_LOAD: return new HGCMMsgSvcLoad ();

befced03fd84a13590b8ce8be8c2480e9bc568c6vboxsync case SVC_MSG_UNLOAD: return new HGCMMsgSvcUnload ();

befced03fd84a13590b8ce8be8c2480e9bc568c6vboxsync case SVC_MSG_CONNECT: return new HGCMMsgSvcConnect ();

befced03fd84a13590b8ce8be8c2480e9bc568c6vboxsync case SVC_MSG_DISCONNECT: return new HGCMMsgSvcDisconnect ();

befced03fd84a13590b8ce8be8c2480e9bc568c6vboxsync case SVC_MSG_HOSTCALL: return new HGCMMsgHostCallSvc ();

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync case SVC_MSG_GUESTCALL: return new HGCMMsgCall ();

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync case SVC_MSG_LOADSTATE:

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync case SVC_MSG_SAVESTATE: return new HGCMMsgLoadSaveStateClient ();

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync case SVC_MSG_REGEXT: return new HGCMMsgSvcRegisterExtension ();

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync case SVC_MSG_UNREGEXT: return new HGCMMsgSvcUnregisterExtension ();

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync default:

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync AssertReleaseMsgFailed(("Msg id = %08X\n", u32MsgId));

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync }

befced03fd84a13590b8ce8be8c2480e9bc568c6vboxsync

e6a3673a23c632af5208dcc8a37f45d7a20b0554vboxsync return NULL;

befced03fd84a13590b8ce8be8c2480e9bc568c6vboxsync}

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsyncDECLCALLBACK(void) hgcmServiceThread (HGCMTHREADHANDLE ThreadHandle, void *pvUser)

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync{

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync HGCMService *pSvc = (HGCMService *)pvUser;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync AssertRelease(pSvc != NULL);

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync bool fQuit = false;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync while (!fQuit)

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync {

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync HGCMMsgCore *pMsgCore;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync int rc = hgcmMsgGet (ThreadHandle, &pMsgCore);

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync if (RT_FAILURE(rc))

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync {

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync /* The error means some serious unrecoverable problem in the hgcmMsg/hgcmThread layer. */

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync AssertMsgFailed (("%Rrc\n", rc));

befced03fd84a13590b8ce8be8c2480e9bc568c6vboxsync break;

befced03fd84a13590b8ce8be8c2480e9bc568c6vboxsync }

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync /* Cache required information to avoid unnecessary pMsgCore access. */

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync uint32_t u32MsgId = pMsgCore->MsgId ();

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync switch (u32MsgId)

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync {

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync#ifdef VBOX_WITH_CRHGSMI

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync case SVC_MSG_HOSTFASTCALLASYNC:

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync {

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync HGCMMsgHostFastCallAsyncSvc *pMsg = (HGCMMsgHostFastCallAsyncSvc *)pMsgCore;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync LogFlowFunc(("SVC_MSG_HOSTFASTCALLASYNC u32Function = %d, pParm = %p\n", pMsg->u32Function, &pMsg->Param));

16d723d9d597f4872dd4c2c960af9cbca4ed63bdvboxsync

16d723d9d597f4872dd4c2c960af9cbca4ed63bdvboxsync rc = pSvc->m_fntable.pfnHostCall (pSvc->m_fntable.pvService, pMsg->u32Function, 1, &pMsg->Param);

16d723d9d597f4872dd4c2c960af9cbca4ed63bdvboxsync } break;

16d723d9d597f4872dd4c2c960af9cbca4ed63bdvboxsync#endif

16d723d9d597f4872dd4c2c960af9cbca4ed63bdvboxsync case SVC_MSG_LOAD:

16d723d9d597f4872dd4c2c960af9cbca4ed63bdvboxsync {

16d723d9d597f4872dd4c2c960af9cbca4ed63bdvboxsync LogFlowFunc(("SVC_MSG_LOAD\n"));

16d723d9d597f4872dd4c2c960af9cbca4ed63bdvboxsync rc = pSvc->loadServiceDLL ();

16d723d9d597f4872dd4c2c960af9cbca4ed63bdvboxsync } break;

16d723d9d597f4872dd4c2c960af9cbca4ed63bdvboxsync

16d723d9d597f4872dd4c2c960af9cbca4ed63bdvboxsync case SVC_MSG_UNLOAD:

16d723d9d597f4872dd4c2c960af9cbca4ed63bdvboxsync {

16d723d9d597f4872dd4c2c960af9cbca4ed63bdvboxsync LogFlowFunc(("SVC_MSG_UNLOAD\n"));

16d723d9d597f4872dd4c2c960af9cbca4ed63bdvboxsync if (pSvc->m_fntable.pfnUnload)

16d723d9d597f4872dd4c2c960af9cbca4ed63bdvboxsync {

16d723d9d597f4872dd4c2c960af9cbca4ed63bdvboxsync pSvc->m_fntable.pfnUnload (pSvc->m_fntable.pvService);

16d723d9d597f4872dd4c2c960af9cbca4ed63bdvboxsync }

16d723d9d597f4872dd4c2c960af9cbca4ed63bdvboxsync

16d723d9d597f4872dd4c2c960af9cbca4ed63bdvboxsync pSvc->unloadServiceDLL ();

79baed6836ae36c5f15b182292387484dcf7a752vboxsync fQuit = true;

79baed6836ae36c5f15b182292387484dcf7a752vboxsync } break;

79baed6836ae36c5f15b182292387484dcf7a752vboxsync

79baed6836ae36c5f15b182292387484dcf7a752vboxsync case SVC_MSG_CONNECT:

79baed6836ae36c5f15b182292387484dcf7a752vboxsync {

79baed6836ae36c5f15b182292387484dcf7a752vboxsync HGCMMsgSvcConnect *pMsg = (HGCMMsgSvcConnect *)pMsgCore;

79baed6836ae36c5f15b182292387484dcf7a752vboxsync

79baed6836ae36c5f15b182292387484dcf7a752vboxsync LogFlowFunc(("SVC_MSG_CONNECT u32ClientId = %d\n", pMsg->u32ClientId));

79baed6836ae36c5f15b182292387484dcf7a752vboxsync

79baed6836ae36c5f15b182292387484dcf7a752vboxsync HGCMClient *pClient = (HGCMClient *)hgcmObjReference (pMsg->u32ClientId, HGCMOBJ_CLIENT);

79baed6836ae36c5f15b182292387484dcf7a752vboxsync

79baed6836ae36c5f15b182292387484dcf7a752vboxsync if (pClient)

79baed6836ae36c5f15b182292387484dcf7a752vboxsync {

79baed6836ae36c5f15b182292387484dcf7a752vboxsync rc = pSvc->m_fntable.pfnConnect (pSvc->m_fntable.pvService, pMsg->u32ClientId, HGCM_CLIENT_DATA(pSvc, pClient));

79baed6836ae36c5f15b182292387484dcf7a752vboxsync

79baed6836ae36c5f15b182292387484dcf7a752vboxsync hgcmObjDereference (pClient);

79baed6836ae36c5f15b182292387484dcf7a752vboxsync }

79baed6836ae36c5f15b182292387484dcf7a752vboxsync else

79baed6836ae36c5f15b182292387484dcf7a752vboxsync {

79baed6836ae36c5f15b182292387484dcf7a752vboxsync rc = VERR_HGCM_INVALID_CLIENT_ID;

79baed6836ae36c5f15b182292387484dcf7a752vboxsync }

79baed6836ae36c5f15b182292387484dcf7a752vboxsync } break;

79baed6836ae36c5f15b182292387484dcf7a752vboxsync

79baed6836ae36c5f15b182292387484dcf7a752vboxsync case SVC_MSG_DISCONNECT:

79baed6836ae36c5f15b182292387484dcf7a752vboxsync {

79baed6836ae36c5f15b182292387484dcf7a752vboxsync HGCMMsgSvcDisconnect *pMsg = (HGCMMsgSvcDisconnect *)pMsgCore;

79baed6836ae36c5f15b182292387484dcf7a752vboxsync

79baed6836ae36c5f15b182292387484dcf7a752vboxsync LogFlowFunc(("SVC_MSG_DISCONNECT u32ClientId = %d\n", pMsg->u32ClientId));

79baed6836ae36c5f15b182292387484dcf7a752vboxsync

79baed6836ae36c5f15b182292387484dcf7a752vboxsync HGCMClient *pClient = (HGCMClient *)hgcmObjReference (pMsg->u32ClientId, HGCMOBJ_CLIENT);

79baed6836ae36c5f15b182292387484dcf7a752vboxsync

79baed6836ae36c5f15b182292387484dcf7a752vboxsync if (pClient)

79baed6836ae36c5f15b182292387484dcf7a752vboxsync {

79baed6836ae36c5f15b182292387484dcf7a752vboxsync rc = pSvc->m_fntable.pfnDisconnect (pSvc->m_fntable.pvService, pMsg->u32ClientId, HGCM_CLIENT_DATA(pSvc, pClient));

79baed6836ae36c5f15b182292387484dcf7a752vboxsync

79baed6836ae36c5f15b182292387484dcf7a752vboxsync hgcmObjDereference (pClient);

79baed6836ae36c5f15b182292387484dcf7a752vboxsync }

79baed6836ae36c5f15b182292387484dcf7a752vboxsync else

79baed6836ae36c5f15b182292387484dcf7a752vboxsync {

79baed6836ae36c5f15b182292387484dcf7a752vboxsync rc = VERR_HGCM_INVALID_CLIENT_ID;

79baed6836ae36c5f15b182292387484dcf7a752vboxsync }

79baed6836ae36c5f15b182292387484dcf7a752vboxsync } break;

79baed6836ae36c5f15b182292387484dcf7a752vboxsync

79baed6836ae36c5f15b182292387484dcf7a752vboxsync case SVC_MSG_GUESTCALL:

79baed6836ae36c5f15b182292387484dcf7a752vboxsync {

79baed6836ae36c5f15b182292387484dcf7a752vboxsync HGCMMsgCall *pMsg = (HGCMMsgCall *)pMsgCore;

79baed6836ae36c5f15b182292387484dcf7a752vboxsync

79baed6836ae36c5f15b182292387484dcf7a752vboxsync LogFlowFunc(("SVC_MSG_GUESTCALL u32ClientId = %d, u32Function = %d, cParms = %d, paParms = %p\n",

79baed6836ae36c5f15b182292387484dcf7a752vboxsync pMsg->u32ClientId, pMsg->u32Function, pMsg->cParms, pMsg->paParms));

79baed6836ae36c5f15b182292387484dcf7a752vboxsync

79baed6836ae36c5f15b182292387484dcf7a752vboxsync HGCMClient *pClient = (HGCMClient *)hgcmObjReference (pMsg->u32ClientId, HGCMOBJ_CLIENT);

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync if (pClient)

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync {

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync pSvc->m_fntable.pfnCall (pSvc->m_fntable.pvService, (VBOXHGCMCALLHANDLE)pMsg, pMsg->u32ClientId, HGCM_CLIENT_DATA(pSvc, pClient), pMsg->u32Function, pMsg->cParms, pMsg->paParms);

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync hgcmObjDereference (pClient);

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync }

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync else

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync {

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync rc = VERR_HGCM_INVALID_CLIENT_ID;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync }

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync } break;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

11d3005e2935c925665896fa26fde09b3e656d70vboxsync case SVC_MSG_HOSTCALL:

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync {

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync HGCMMsgHostCallSvc *pMsg = (HGCMMsgHostCallSvc *)pMsgCore;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync LogFlowFunc(("SVC_MSG_HOSTCALL u32Function = %d, cParms = %d, paParms = %p\n", pMsg->u32Function, pMsg->cParms, pMsg->paParms));

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync rc = pSvc->m_fntable.pfnHostCall (pSvc->m_fntable.pvService, pMsg->u32Function, pMsg->cParms, pMsg->paParms);

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync } break;

79baed6836ae36c5f15b182292387484dcf7a752vboxsync

79baed6836ae36c5f15b182292387484dcf7a752vboxsync case SVC_MSG_LOADSTATE:

79baed6836ae36c5f15b182292387484dcf7a752vboxsync {

79baed6836ae36c5f15b182292387484dcf7a752vboxsync HGCMMsgLoadSaveStateClient *pMsg = (HGCMMsgLoadSaveStateClient *)pMsgCore;

79baed6836ae36c5f15b182292387484dcf7a752vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync LogFlowFunc(("SVC_MSG_LOADSTATE\n"));

69c6eff4fcb6dc027e94a3e7908926c4e8ef5aefvboxsync

69c6eff4fcb6dc027e94a3e7908926c4e8ef5aefvboxsync HGCMClient *pClient = (HGCMClient *)hgcmObjReference (pMsg->u32ClientId, HGCMOBJ_CLIENT);

69c6eff4fcb6dc027e94a3e7908926c4e8ef5aefvboxsync

69c6eff4fcb6dc027e94a3e7908926c4e8ef5aefvboxsync if (pClient)

69c6eff4fcb6dc027e94a3e7908926c4e8ef5aefvboxsync {

69c6eff4fcb6dc027e94a3e7908926c4e8ef5aefvboxsync if (pSvc->m_fntable.pfnLoadState)

79baed6836ae36c5f15b182292387484dcf7a752vboxsync {

79baed6836ae36c5f15b182292387484dcf7a752vboxsync rc = pSvc->m_fntable.pfnLoadState (pSvc->m_fntable.pvService, pMsg->u32ClientId, HGCM_CLIENT_DATA(pSvc, pClient), pMsg->pSSM);

79baed6836ae36c5f15b182292387484dcf7a752vboxsync }

79baed6836ae36c5f15b182292387484dcf7a752vboxsync

79baed6836ae36c5f15b182292387484dcf7a752vboxsync hgcmObjDereference (pClient);

79baed6836ae36c5f15b182292387484dcf7a752vboxsync }

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync else

79baed6836ae36c5f15b182292387484dcf7a752vboxsync {

79baed6836ae36c5f15b182292387484dcf7a752vboxsync rc = VERR_HGCM_INVALID_CLIENT_ID;

79baed6836ae36c5f15b182292387484dcf7a752vboxsync }

79baed6836ae36c5f15b182292387484dcf7a752vboxsync } break;

79baed6836ae36c5f15b182292387484dcf7a752vboxsync

79baed6836ae36c5f15b182292387484dcf7a752vboxsync case SVC_MSG_SAVESTATE:

79baed6836ae36c5f15b182292387484dcf7a752vboxsync {

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync HGCMMsgLoadSaveStateClient *pMsg = (HGCMMsgLoadSaveStateClient *)pMsgCore;

79baed6836ae36c5f15b182292387484dcf7a752vboxsync

79baed6836ae36c5f15b182292387484dcf7a752vboxsync LogFlowFunc(("SVC_MSG_SAVESTATE\n"));

79baed6836ae36c5f15b182292387484dcf7a752vboxsync

79baed6836ae36c5f15b182292387484dcf7a752vboxsync HGCMClient *pClient = (HGCMClient *)hgcmObjReference (pMsg->u32ClientId, HGCMOBJ_CLIENT);

79baed6836ae36c5f15b182292387484dcf7a752vboxsync

79baed6836ae36c5f15b182292387484dcf7a752vboxsync rc = VINF_SUCCESS;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

79baed6836ae36c5f15b182292387484dcf7a752vboxsync if (pClient)

79baed6836ae36c5f15b182292387484dcf7a752vboxsync {

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync if (pSvc->m_fntable.pfnSaveState)

79baed6836ae36c5f15b182292387484dcf7a752vboxsync {

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync g_fSaveState = true;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync rc = pSvc->m_fntable.pfnSaveState (pSvc->m_fntable.pvService, pMsg->u32ClientId, HGCM_CLIENT_DATA(pSvc, pClient), pMsg->pSSM);

79baed6836ae36c5f15b182292387484dcf7a752vboxsync g_fSaveState = false;

79baed6836ae36c5f15b182292387484dcf7a752vboxsync }

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync hgcmObjDereference (pClient);

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync }

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync else

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync {

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync rc = VERR_HGCM_INVALID_CLIENT_ID;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync }

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync } break;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync case SVC_MSG_REGEXT:

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync {

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync HGCMMsgSvcRegisterExtension *pMsg = (HGCMMsgSvcRegisterExtension *)pMsgCore;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync LogFlowFunc(("SVC_MSG_REGEXT handle = %p, pfn = %p\n", pMsg->handle, pMsg->pfnExtension));

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync if (pSvc->m_hExtension)

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync {

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync rc = VERR_NOT_SUPPORTED;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync }

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync else

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync {

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync if (pSvc->m_fntable.pfnRegisterExtension)

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync {

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync rc = pSvc->m_fntable.pfnRegisterExtension (pSvc->m_fntable.pvService, pMsg->pfnExtension, pMsg->pvExtension);

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync }

58461e707998a927c19da46b98748ee2b79f4190vboxsync else

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync {

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync rc = VERR_NOT_SUPPORTED;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync }

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync if (RT_SUCCESS(rc))

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync {

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync pSvc->m_hExtension = pMsg->handle;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync }

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync }

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync } break;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync case SVC_MSG_UNREGEXT:

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync {

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync HGCMMsgSvcUnregisterExtension *pMsg = (HGCMMsgSvcUnregisterExtension *)pMsgCore;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync LogFlowFunc(("SVC_MSG_UNREGEXT handle = %p\n", pMsg->handle));

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync if (pSvc->m_hExtension != pMsg->handle)

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync {

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync rc = VERR_NOT_SUPPORTED;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync }

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync else

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync {

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync if (pSvc->m_fntable.pfnRegisterExtension)

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync {

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync rc = pSvc->m_fntable.pfnRegisterExtension (pSvc->m_fntable.pvService, NULL, NULL);

11d3005e2935c925665896fa26fde09b3e656d70vboxsync }

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync else

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync {

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync rc = VERR_NOT_SUPPORTED;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync }

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync pSvc->m_hExtension = NULL;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync }

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync } break;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync default:

79baed6836ae36c5f15b182292387484dcf7a752vboxsync {

79baed6836ae36c5f15b182292387484dcf7a752vboxsync AssertMsgFailed(("hgcmServiceThread::Unsupported message number %08X\n", u32MsgId));

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync rc = VERR_NOT_SUPPORTED;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync } break;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync }

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync if (u32MsgId != SVC_MSG_GUESTCALL)

27152389a84c6dec057fba6fc21241991e079006vboxsync {

27152389a84c6dec057fba6fc21241991e079006vboxsync /* For SVC_MSG_GUESTCALL the service calls the completion helper.

27152389a84c6dec057fba6fc21241991e079006vboxsync hgcmMsgComplete (pMsgCore, rc);

27152389a84c6dec057fba6fc21241991e079006vboxsync }

27152389a84c6dec057fba6fc21241991e079006vboxsync }

27152389a84c6dec057fba6fc21241991e079006vboxsync}

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync/* static */ DECLCALLBACK(void) HGCMService::svcHlpCallComplete (VBOXHGCMCALLHANDLE callHandle, int32_t rc)

27152389a84c6dec057fba6fc21241991e079006vboxsync{

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync HGCMMsgCore *pMsgCore = (HGCMMsgCore *)callHandle;

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync if (pMsgCore->MsgId () == SVC_MSG_GUESTCALL)

27152389a84c6dec057fba6fc21241991e079006vboxsync {

27152389a84c6dec057fba6fc21241991e079006vboxsync /* Only call the completion for these messages. The helper

27152389a84c6dec057fba6fc21241991e079006vboxsync hgcmMsgComplete (pMsgCore, rc);

27152389a84c6dec057fba6fc21241991e079006vboxsync }

27152389a84c6dec057fba6fc21241991e079006vboxsync else

27152389a84c6dec057fba6fc21241991e079006vboxsync {

27152389a84c6dec057fba6fc21241991e079006vboxsync AssertFailed ();

27152389a84c6dec057fba6fc21241991e079006vboxsync }

27152389a84c6dec057fba6fc21241991e079006vboxsync}

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync/* static */ DECLCALLBACK(void) HGCMService::svcHlpDisconnectClient (void *pvInstance, uint32_t u32ClientId)

27152389a84c6dec057fba6fc21241991e079006vboxsync{

27152389a84c6dec057fba6fc21241991e079006vboxsync HGCMService *pService = static_cast <HGCMService *> (pvInstance);

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync if (pService)

27152389a84c6dec057fba6fc21241991e079006vboxsync {

27152389a84c6dec057fba6fc21241991e079006vboxsync pService->DisconnectClient (u32ClientId, true);

27152389a84c6dec057fba6fc21241991e079006vboxsync }

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync}

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsyncstatic DECLCALLBACK(void) hgcmMsgCompletionCallback (int32_t result, HGCMMsgCore *pMsgCore)

27152389a84c6dec057fba6fc21241991e079006vboxsync{

27152389a84c6dec057fba6fc21241991e079006vboxsync /* Call the VMMDev port interface to issue IRQ notification. */

27152389a84c6dec057fba6fc21241991e079006vboxsync HGCMMsgHeader *pMsgHdr = (HGCMMsgHeader *)pMsgCore;

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync LogFlow(("MAIN::hgcmMsgCompletionCallback: message %p\n", pMsgCore));

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync if (pMsgHdr->pHGCMPort && !g_fResetting)

27152389a84c6dec057fba6fc21241991e079006vboxsync {

27152389a84c6dec057fba6fc21241991e079006vboxsync pMsgHdr->pHGCMPort->pfnCompleted (pMsgHdr->pHGCMPort, g_fSaveState? VINF_HGCM_SAVE_STATE: result, pMsgHdr->pCmd);

27152389a84c6dec057fba6fc21241991e079006vboxsync }

27152389a84c6dec057fba6fc21241991e079006vboxsync}

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsyncint HGCMService::instanceCreate (const char *pszServiceLibrary, const char *pszServiceName)

27152389a84c6dec057fba6fc21241991e079006vboxsync{

27152389a84c6dec057fba6fc21241991e079006vboxsync LogFlowFunc(("name %s, lib %s\n", pszServiceName, pszServiceLibrary));

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync /* The maximum length of the thread name, allowed by the RT is 15. */

27152389a84c6dec057fba6fc21241991e079006vboxsync char achThreadName[16];

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync strncpy (achThreadName, pszServiceName, 15);

27152389a84c6dec057fba6fc21241991e079006vboxsync achThreadName[15] = 0;

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync int rc = hgcmThreadCreate (&m_thread, achThreadName, hgcmServiceThread, this);

27152389a84c6dec057fba6fc21241991e079006vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync if (RT_SUCCESS(rc))

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync {

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync m_pszSvcName = RTStrDup (pszServiceName);

27152389a84c6dec057fba6fc21241991e079006vboxsync m_pszSvcLibrary = RTStrDup (pszServiceLibrary);

27152389a84c6dec057fba6fc21241991e079006vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync if (!m_pszSvcName || !m_pszSvcLibrary)

27152389a84c6dec057fba6fc21241991e079006vboxsync {

27152389a84c6dec057fba6fc21241991e079006vboxsync RTStrFree (m_pszSvcLibrary);

27152389a84c6dec057fba6fc21241991e079006vboxsync m_pszSvcLibrary = NULL;

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync RTStrFree (m_pszSvcName);

27152389a84c6dec057fba6fc21241991e079006vboxsync m_pszSvcName = NULL;

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync rc = VERR_NO_MEMORY;

27152389a84c6dec057fba6fc21241991e079006vboxsync }

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync else

27152389a84c6dec057fba6fc21241991e079006vboxsync {

27152389a84c6dec057fba6fc21241991e079006vboxsync /* Initialize service helpers table. */

27152389a84c6dec057fba6fc21241991e079006vboxsync m_svcHelpers.pfnCallComplete = svcHlpCallComplete;

27152389a84c6dec057fba6fc21241991e079006vboxsync m_svcHelpers.pvInstance = this;

27152389a84c6dec057fba6fc21241991e079006vboxsync m_svcHelpers.pfnDisconnectClient = svcHlpDisconnectClient;

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync /* Execute the load request on the service thread. */

27152389a84c6dec057fba6fc21241991e079006vboxsync HGCMMSGHANDLE hMsg;

27152389a84c6dec057fba6fc21241991e079006vboxsync rc = hgcmMsgAlloc (m_thread, &hMsg, SVC_MSG_LOAD, hgcmMessageAllocSvc);

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync if (RT_SUCCESS(rc))

27152389a84c6dec057fba6fc21241991e079006vboxsync {

27152389a84c6dec057fba6fc21241991e079006vboxsync rc = hgcmMsgSend (hMsg);

27152389a84c6dec057fba6fc21241991e079006vboxsync }

27152389a84c6dec057fba6fc21241991e079006vboxsync }

27152389a84c6dec057fba6fc21241991e079006vboxsync }

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync if (RT_FAILURE(rc))

27152389a84c6dec057fba6fc21241991e079006vboxsync {

27152389a84c6dec057fba6fc21241991e079006vboxsync instanceDestroy ();

27152389a84c6dec057fba6fc21241991e079006vboxsync }

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync LogFlowFunc(("rc = %Rrc\n", rc));

27152389a84c6dec057fba6fc21241991e079006vboxsync return rc;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync}

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsyncvoid HGCMService::instanceDestroy (void)

27152389a84c6dec057fba6fc21241991e079006vboxsync{

27152389a84c6dec057fba6fc21241991e079006vboxsync LogFlowFunc(("%s\n", m_pszSvcName));

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync HGCMMSGHANDLE hMsg;

27152389a84c6dec057fba6fc21241991e079006vboxsync int rc = hgcmMsgAlloc (m_thread, &hMsg, SVC_MSG_UNLOAD, hgcmMessageAllocSvc);

27152389a84c6dec057fba6fc21241991e079006vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync if (RT_SUCCESS(rc))

27152389a84c6dec057fba6fc21241991e079006vboxsync {

27152389a84c6dec057fba6fc21241991e079006vboxsync rc = hgcmMsgSend (hMsg);

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync if (RT_SUCCESS(rc))

27152389a84c6dec057fba6fc21241991e079006vboxsync {

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync hgcmThreadWait (m_thread);

27152389a84c6dec057fba6fc21241991e079006vboxsync }

27152389a84c6dec057fba6fc21241991e079006vboxsync }

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync RTStrFree (m_pszSvcLibrary);

27152389a84c6dec057fba6fc21241991e079006vboxsync m_pszSvcLibrary = NULL;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync RTStrFree (m_pszSvcName);

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync m_pszSvcName = NULL;

27152389a84c6dec057fba6fc21241991e079006vboxsync}

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsyncint HGCMService::saveClientState(uint32_t u32ClientId, PSSMHANDLE pSSM)

27152389a84c6dec057fba6fc21241991e079006vboxsync{

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync LogFlowFunc(("%s\n", m_pszSvcName));

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync HGCMMSGHANDLE hMsg;

27152389a84c6dec057fba6fc21241991e079006vboxsync int rc = hgcmMsgAlloc (m_thread, &hMsg, SVC_MSG_SAVESTATE, hgcmMessageAllocSvc);

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync if (RT_SUCCESS(rc))

27152389a84c6dec057fba6fc21241991e079006vboxsync {

27152389a84c6dec057fba6fc21241991e079006vboxsync HGCMMsgLoadSaveStateClient *pMsg = (HGCMMsgLoadSaveStateClient *)hgcmObjReference (hMsg, HGCMOBJ_MSG);

27152389a84c6dec057fba6fc21241991e079006vboxsync AssertRelease(pMsg);

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync pMsg->u32ClientId = u32ClientId;

27152389a84c6dec057fba6fc21241991e079006vboxsync pMsg->pSSM = pSSM;

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync hgcmObjDereference (pMsg);

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync rc = hgcmMsgSend (hMsg);

27152389a84c6dec057fba6fc21241991e079006vboxsync }

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync LogFlowFunc(("rc = %Rrc\n", rc));

27152389a84c6dec057fba6fc21241991e079006vboxsync return rc;

27152389a84c6dec057fba6fc21241991e079006vboxsync}

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsyncint HGCMService::loadClientState (uint32_t u32ClientId, PSSMHANDLE pSSM)

27152389a84c6dec057fba6fc21241991e079006vboxsync{

27152389a84c6dec057fba6fc21241991e079006vboxsync LogFlowFunc(("%s\n", m_pszSvcName));

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync HGCMMSGHANDLE hMsg;

27152389a84c6dec057fba6fc21241991e079006vboxsync int rc = hgcmMsgAlloc (m_thread, &hMsg, SVC_MSG_LOADSTATE, hgcmMessageAllocSvc);

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync if (RT_SUCCESS(rc))

27152389a84c6dec057fba6fc21241991e079006vboxsync {

27152389a84c6dec057fba6fc21241991e079006vboxsync HGCMMsgLoadSaveStateClient *pMsg = (HGCMMsgLoadSaveStateClient *)hgcmObjReference (hMsg, HGCMOBJ_MSG);

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync AssertRelease(pMsg);

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync pMsg->u32ClientId = u32ClientId;

27152389a84c6dec057fba6fc21241991e079006vboxsync pMsg->pSSM = pSSM;

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync hgcmObjDereference (pMsg);

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync rc = hgcmMsgSend (hMsg);

27152389a84c6dec057fba6fc21241991e079006vboxsync }

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync LogFlowFunc(("rc = %Rrc\n", rc));

27152389a84c6dec057fba6fc21241991e079006vboxsync return rc;

27152389a84c6dec057fba6fc21241991e079006vboxsync}

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync/* static */ int HGCMService::LoadService (const char *pszServiceLibrary, const char *pszServiceName)

27152389a84c6dec057fba6fc21241991e079006vboxsync{

27152389a84c6dec057fba6fc21241991e079006vboxsync LogFlowFunc(("lib %s, name = %s\n", pszServiceLibrary, pszServiceName));

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync /* Look at already loaded services to avoid double loading. */

27152389a84c6dec057fba6fc21241991e079006vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync HGCMService *pSvc;

27152389a84c6dec057fba6fc21241991e079006vboxsync int rc = HGCMService::ResolveService (&pSvc, pszServiceName);

27152389a84c6dec057fba6fc21241991e079006vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync if (RT_SUCCESS(rc))

27152389a84c6dec057fba6fc21241991e079006vboxsync {

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync /* The service is already loaded. */

27152389a84c6dec057fba6fc21241991e079006vboxsync pSvc->ReleaseService ();

79baed6836ae36c5f15b182292387484dcf7a752vboxsync rc = VERR_HGCM_SERVICE_EXISTS;

27152389a84c6dec057fba6fc21241991e079006vboxsync }

27152389a84c6dec057fba6fc21241991e079006vboxsync else

27152389a84c6dec057fba6fc21241991e079006vboxsync {

27152389a84c6dec057fba6fc21241991e079006vboxsync /* Create the new service. */

27152389a84c6dec057fba6fc21241991e079006vboxsync pSvc = new HGCMService ();

27152389a84c6dec057fba6fc21241991e079006vboxsync

79baed6836ae36c5f15b182292387484dcf7a752vboxsync if (!pSvc)

79baed6836ae36c5f15b182292387484dcf7a752vboxsync {

27152389a84c6dec057fba6fc21241991e079006vboxsync rc = VERR_NO_MEMORY;

27152389a84c6dec057fba6fc21241991e079006vboxsync }

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync else

27152389a84c6dec057fba6fc21241991e079006vboxsync {

27152389a84c6dec057fba6fc21241991e079006vboxsync /* Load the library and call the initialization entry point. */

27152389a84c6dec057fba6fc21241991e079006vboxsync rc = pSvc->instanceCreate (pszServiceLibrary, pszServiceName);

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync if (RT_SUCCESS(rc))

27152389a84c6dec057fba6fc21241991e079006vboxsync {

27152389a84c6dec057fba6fc21241991e079006vboxsync /* Insert the just created service to list for future references. */

27152389a84c6dec057fba6fc21241991e079006vboxsync pSvc->m_pSvcNext = sm_pSvcListHead;

27152389a84c6dec057fba6fc21241991e079006vboxsync pSvc->m_pSvcPrev = NULL;

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync if (sm_pSvcListHead)

27152389a84c6dec057fba6fc21241991e079006vboxsync {

27152389a84c6dec057fba6fc21241991e079006vboxsync sm_pSvcListHead->m_pSvcPrev = pSvc;

27152389a84c6dec057fba6fc21241991e079006vboxsync }

27152389a84c6dec057fba6fc21241991e079006vboxsync else

27152389a84c6dec057fba6fc21241991e079006vboxsync {

27152389a84c6dec057fba6fc21241991e079006vboxsync sm_pSvcListTail = pSvc;

27152389a84c6dec057fba6fc21241991e079006vboxsync }

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync sm_pSvcListHead = pSvc;

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync sm_cServices++;

27152389a84c6dec057fba6fc21241991e079006vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync /* Reference the service (for first time) until it is unloaded on HGCM termination. */

27152389a84c6dec057fba6fc21241991e079006vboxsync AssertRelease (pSvc->m_u32RefCnt == 0);

27152389a84c6dec057fba6fc21241991e079006vboxsync pSvc->ReferenceService ();

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync LogFlowFunc(("service %p\n", pSvc));

27152389a84c6dec057fba6fc21241991e079006vboxsync }

27152389a84c6dec057fba6fc21241991e079006vboxsync }

27152389a84c6dec057fba6fc21241991e079006vboxsync }

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync LogFlowFunc(("rc = %Rrc\n", rc));

27152389a84c6dec057fba6fc21241991e079006vboxsync return rc;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync}

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsyncvoid HGCMService::UnloadService (void)

27152389a84c6dec057fba6fc21241991e079006vboxsync{

27152389a84c6dec057fba6fc21241991e079006vboxsync LogFlowFunc(("name = %s\n", m_pszSvcName));

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync /* Remove the service from the list. */

27152389a84c6dec057fba6fc21241991e079006vboxsync if (m_pSvcNext)

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync {

27152389a84c6dec057fba6fc21241991e079006vboxsync m_pSvcNext->m_pSvcPrev = m_pSvcPrev;

27152389a84c6dec057fba6fc21241991e079006vboxsync }

27152389a84c6dec057fba6fc21241991e079006vboxsync else

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync {

27152389a84c6dec057fba6fc21241991e079006vboxsync sm_pSvcListTail = m_pSvcPrev;

27152389a84c6dec057fba6fc21241991e079006vboxsync }

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync if (m_pSvcPrev)

27152389a84c6dec057fba6fc21241991e079006vboxsync {

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync m_pSvcPrev->m_pSvcNext = m_pSvcNext;

27152389a84c6dec057fba6fc21241991e079006vboxsync }

27152389a84c6dec057fba6fc21241991e079006vboxsync else

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync {

27152389a84c6dec057fba6fc21241991e079006vboxsync sm_pSvcListHead = m_pSvcNext;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync }

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync sm_cServices--;

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync /* The service must be unloaded only if all clients were disconnected. */

27152389a84c6dec057fba6fc21241991e079006vboxsync LogFlowFunc(("m_u32RefCnt = %d\n", m_u32RefCnt));

27152389a84c6dec057fba6fc21241991e079006vboxsync AssertRelease (m_u32RefCnt == 1);

27152389a84c6dec057fba6fc21241991e079006vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync /* Now the service can be released. */

27152389a84c6dec057fba6fc21241991e079006vboxsync ReleaseService ();

27152389a84c6dec057fba6fc21241991e079006vboxsync}

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync/* static */ void HGCMService::UnloadAll (void)

79baed6836ae36c5f15b182292387484dcf7a752vboxsync{

27152389a84c6dec057fba6fc21241991e079006vboxsync while (sm_pSvcListHead)

27152389a84c6dec057fba6fc21241991e079006vboxsync {

27152389a84c6dec057fba6fc21241991e079006vboxsync sm_pSvcListHead->UnloadService ();

79baed6836ae36c5f15b182292387484dcf7a752vboxsync }

27152389a84c6dec057fba6fc21241991e079006vboxsync}

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync/* static */ int HGCMService::ResolveService (HGCMService **ppSvc, const char *pszServiceName)

27152389a84c6dec057fba6fc21241991e079006vboxsync{

79baed6836ae36c5f15b182292387484dcf7a752vboxsync LogFlowFunc(("ppSvc = %p name = %s\n",

79baed6836ae36c5f15b182292387484dcf7a752vboxsync ppSvc, pszServiceName));

79baed6836ae36c5f15b182292387484dcf7a752vboxsync

79baed6836ae36c5f15b182292387484dcf7a752vboxsync if (!ppSvc || !pszServiceName)

27152389a84c6dec057fba6fc21241991e079006vboxsync {

79baed6836ae36c5f15b182292387484dcf7a752vboxsync return VERR_INVALID_PARAMETER;

27152389a84c6dec057fba6fc21241991e079006vboxsync }

79baed6836ae36c5f15b182292387484dcf7a752vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync HGCMService *pSvc = sm_pSvcListHead;

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync while (pSvc)

27152389a84c6dec057fba6fc21241991e079006vboxsync {

27152389a84c6dec057fba6fc21241991e079006vboxsync if (strcmp (pSvc->m_pszSvcName, pszServiceName) == 0)

79baed6836ae36c5f15b182292387484dcf7a752vboxsync {

27152389a84c6dec057fba6fc21241991e079006vboxsync break;

79baed6836ae36c5f15b182292387484dcf7a752vboxsync }

27152389a84c6dec057fba6fc21241991e079006vboxsync

79baed6836ae36c5f15b182292387484dcf7a752vboxsync pSvc = pSvc->m_pSvcNext;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync }

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync LogFlowFunc(("lookup in the list is %p\n", pSvc));

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync if (pSvc == NULL)

27152389a84c6dec057fba6fc21241991e079006vboxsync {

79baed6836ae36c5f15b182292387484dcf7a752vboxsync *ppSvc = NULL;

27152389a84c6dec057fba6fc21241991e079006vboxsync return VERR_HGCM_SERVICE_NOT_FOUND;

79baed6836ae36c5f15b182292387484dcf7a752vboxsync }

27152389a84c6dec057fba6fc21241991e079006vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync pSvc->ReferenceService ();

27152389a84c6dec057fba6fc21241991e079006vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync *ppSvc = pSvc;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync return VINF_SUCCESS;

895d68f6e6c15446ca98b8f6ec96ce19d7ea356fvboxsync}

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsyncvoid HGCMService::ReferenceService (void)

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync{

27152389a84c6dec057fba6fc21241991e079006vboxsync ASMAtomicIncU32 (&m_u32RefCnt);

27152389a84c6dec057fba6fc21241991e079006vboxsync LogFlowFunc(("[%s] m_u32RefCnt = %d\n", m_pszSvcName, m_u32RefCnt));

27152389a84c6dec057fba6fc21241991e079006vboxsync}

27152389a84c6dec057fba6fc21241991e079006vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsyncvoid HGCMService::ReleaseService (void)

27152389a84c6dec057fba6fc21241991e079006vboxsync{

27152389a84c6dec057fba6fc21241991e079006vboxsync LogFlowFunc(("m_u32RefCnt = %d\n", m_u32RefCnt));

27152389a84c6dec057fba6fc21241991e079006vboxsync uint32_t u32RefCnt = ASMAtomicDecU32 (&m_u32RefCnt);

27152389a84c6dec057fba6fc21241991e079006vboxsync AssertRelease(u32RefCnt != ~0U);

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync LogFlowFunc(("u32RefCnt = %d, name %s\n", u32RefCnt, m_pszSvcName));

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync if (u32RefCnt == 0)

27152389a84c6dec057fba6fc21241991e079006vboxsync {

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync instanceDestroy ();

27152389a84c6dec057fba6fc21241991e079006vboxsync delete this;

27152389a84c6dec057fba6fc21241991e079006vboxsync }

27152389a84c6dec057fba6fc21241991e079006vboxsync}

27152389a84c6dec057fba6fc21241991e079006vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync/* static */ void HGCMService::Reset (void)

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync{

27152389a84c6dec057fba6fc21241991e079006vboxsync g_fResetting = true;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync HGCMService *pSvc = sm_pSvcListHead;

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync while (pSvc)

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync {

27152389a84c6dec057fba6fc21241991e079006vboxsync while (pSvc->m_cClients && pSvc->m_paClientIds)

27152389a84c6dec057fba6fc21241991e079006vboxsync {

27152389a84c6dec057fba6fc21241991e079006vboxsync LogFlowFunc(("handle %d\n", pSvc->m_paClientIds[0]));

27152389a84c6dec057fba6fc21241991e079006vboxsync pSvc->DisconnectClient (pSvc->m_paClientIds[0], false);

27152389a84c6dec057fba6fc21241991e079006vboxsync }

27152389a84c6dec057fba6fc21241991e079006vboxsync

e66911cff9c18da76d9bddb0505ad329a9ccd795vboxsync#ifdef VBOX_WITH_CRHGSMI

27152389a84c6dec057fba6fc21241991e079006vboxsync /* @todo: could this actually happen that the service is destroyed on ReleaseService? */

27152389a84c6dec057fba6fc21241991e079006vboxsync HGCMService *pNextSvc = pSvc->m_pSvcNext;

27152389a84c6dec057fba6fc21241991e079006vboxsync while (pSvc->m_cHandleAcquires)

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync {

27152389a84c6dec057fba6fc21241991e079006vboxsync pSvc->HandleReleased ();

27152389a84c6dec057fba6fc21241991e079006vboxsync pSvc->ReleaseService ();

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync }

27152389a84c6dec057fba6fc21241991e079006vboxsync pSvc = pNextSvc;

27152389a84c6dec057fba6fc21241991e079006vboxsync#else

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync pSvc = pSvc->m_pSvcNext;

27152389a84c6dec057fba6fc21241991e079006vboxsync#endif

27152389a84c6dec057fba6fc21241991e079006vboxsync }

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync g_fResetting = false;

27152389a84c6dec057fba6fc21241991e079006vboxsync}

27152389a84c6dec057fba6fc21241991e079006vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync/* static */ int HGCMService::SaveState (PSSMHANDLE pSSM)

27152389a84c6dec057fba6fc21241991e079006vboxsync{

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync /* Save the current handle count and restore afterwards to avoid client id conflicts. */

27152389a84c6dec057fba6fc21241991e079006vboxsync int rc = SSMR3PutU32(pSSM, hgcmObjQueryHandleCount());

27152389a84c6dec057fba6fc21241991e079006vboxsync AssertRCReturn(rc, rc);

27152389a84c6dec057fba6fc21241991e079006vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync LogFlowFunc(("%d services to be saved:\n", sm_cServices));

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync /* Save number of services. */

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync rc = SSMR3PutU32(pSSM, sm_cServices);

27152389a84c6dec057fba6fc21241991e079006vboxsync AssertRCReturn(rc, rc);

27152389a84c6dec057fba6fc21241991e079006vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync /* Save every service. */

27152389a84c6dec057fba6fc21241991e079006vboxsync HGCMService *pSvc = sm_pSvcListHead;

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync while (pSvc)

27152389a84c6dec057fba6fc21241991e079006vboxsync {

27152389a84c6dec057fba6fc21241991e079006vboxsync LogFlowFunc(("Saving service [%s]\n", pSvc->m_pszSvcName));

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync /* Save the length of the service name. */

27152389a84c6dec057fba6fc21241991e079006vboxsync rc = SSMR3PutU32(pSSM, (uint32_t) strlen(pSvc->m_pszSvcName) + 1);

27152389a84c6dec057fba6fc21241991e079006vboxsync AssertRCReturn(rc, rc);

27152389a84c6dec057fba6fc21241991e079006vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync /* Save the name of the service. */

27152389a84c6dec057fba6fc21241991e079006vboxsync rc = SSMR3PutStrZ(pSSM, pSvc->m_pszSvcName);

27152389a84c6dec057fba6fc21241991e079006vboxsync AssertRCReturn(rc, rc);

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync /* Save the number of clients. */

27152389a84c6dec057fba6fc21241991e079006vboxsync rc = SSMR3PutU32(pSSM, pSvc->m_cClients);

27152389a84c6dec057fba6fc21241991e079006vboxsync AssertRCReturn(rc, rc);

895d68f6e6c15446ca98b8f6ec96ce19d7ea356fvboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync /* Call the service for every client. Normally a service must not have

27152389a84c6dec057fba6fc21241991e079006vboxsync int i;

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync for (i = 0; i < pSvc->m_cClients; i++)

27152389a84c6dec057fba6fc21241991e079006vboxsync {

27152389a84c6dec057fba6fc21241991e079006vboxsync uint32_t u32ClientId = pSvc->m_paClientIds[i];

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync Log(("client id 0x%08X\n", u32ClientId));

27152389a84c6dec057fba6fc21241991e079006vboxsync

895d68f6e6c15446ca98b8f6ec96ce19d7ea356fvboxsync /* Save the client id. */

27152389a84c6dec057fba6fc21241991e079006vboxsync rc = SSMR3PutU32(pSSM, u32ClientId);

27152389a84c6dec057fba6fc21241991e079006vboxsync AssertRCReturn(rc, rc);

895d68f6e6c15446ca98b8f6ec96ce19d7ea356fvboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync /* Call the service, so the operation is executed by the service thread. */

27152389a84c6dec057fba6fc21241991e079006vboxsync rc = pSvc->saveClientState (u32ClientId, pSSM);

27152389a84c6dec057fba6fc21241991e079006vboxsync AssertRCReturn(rc, rc);

27152389a84c6dec057fba6fc21241991e079006vboxsync }

27152389a84c6dec057fba6fc21241991e079006vboxsync

c9a51cded51b8f691f24de730e190ef573344bfcvboxsync pSvc = pSvc->m_pSvcNext;

27152389a84c6dec057fba6fc21241991e079006vboxsync }

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync return VINF_SUCCESS;

27152389a84c6dec057fba6fc21241991e079006vboxsync}

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync/* static */ int HGCMService::LoadState (PSSMHANDLE pSSM)

27152389a84c6dec057fba6fc21241991e079006vboxsync{

27152389a84c6dec057fba6fc21241991e079006vboxsync /* Restore handle count to avoid client id conflicts. */

27152389a84c6dec057fba6fc21241991e079006vboxsync uint32_t u32;

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync int rc = SSMR3GetU32(pSSM, &u32);

27152389a84c6dec057fba6fc21241991e079006vboxsync AssertRCReturn(rc, rc);

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync hgcmObjSetHandleCount(u32);

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync /* Get the number of services. */

27152389a84c6dec057fba6fc21241991e079006vboxsync uint32_t cServices;

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync rc = SSMR3GetU32(pSSM, &cServices);

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync AssertRCReturn(rc, rc);

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync LogFlowFunc(("%d services to be restored:\n", cServices));

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync while (cServices--)

ce9015004a47e95eb75e047ba42f2d3200d2d222vboxsync {

27152389a84c6dec057fba6fc21241991e079006vboxsync /* Get the length of the service name. */

27152389a84c6dec057fba6fc21241991e079006vboxsync rc = SSMR3GetU32(pSSM, &u32);

27152389a84c6dec057fba6fc21241991e079006vboxsync AssertRCReturn(rc, rc);

ce9015004a47e95eb75e047ba42f2d3200d2d222vboxsync AssertReturn(u32 <= VBOX_HGCM_SVC_NAME_MAX_BYTES, VERR_SSM_UNEXPECTED_DATA);

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync char *pszServiceName = (char *)alloca (u32);

895d68f6e6c15446ca98b8f6ec96ce19d7ea356fvboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync /* Get the service name. */

ce9015004a47e95eb75e047ba42f2d3200d2d222vboxsync rc = SSMR3GetStrZ(pSSM, pszServiceName, u32);

27152389a84c6dec057fba6fc21241991e079006vboxsync AssertRCReturn(rc, rc);

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync LogRel(("HGCM: restoring [%s]\n", pszServiceName));

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync /* Resolve the service instance. */

27152389a84c6dec057fba6fc21241991e079006vboxsync HGCMService *pSvc;

27152389a84c6dec057fba6fc21241991e079006vboxsync rc = ResolveService (&pSvc, pszServiceName);

27152389a84c6dec057fba6fc21241991e079006vboxsync AssertLogRelMsgReturn(pSvc, ("rc=%Rrc, %s\n", rc, pszServiceName), VERR_SSM_UNEXPECTED_DATA);

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync /* Get the number of clients. */

27152389a84c6dec057fba6fc21241991e079006vboxsync uint32_t cClients;

27152389a84c6dec057fba6fc21241991e079006vboxsync rc = SSMR3GetU32(pSSM, &cClients);

27152389a84c6dec057fba6fc21241991e079006vboxsync if (RT_FAILURE(rc))

27152389a84c6dec057fba6fc21241991e079006vboxsync {

27152389a84c6dec057fba6fc21241991e079006vboxsync pSvc->ReleaseService ();

27152389a84c6dec057fba6fc21241991e079006vboxsync AssertFailed();

27152389a84c6dec057fba6fc21241991e079006vboxsync return rc;

895d68f6e6c15446ca98b8f6ec96ce19d7ea356fvboxsync }

27152389a84c6dec057fba6fc21241991e079006vboxsync

895d68f6e6c15446ca98b8f6ec96ce19d7ea356fvboxsync while (cClients--)

27152389a84c6dec057fba6fc21241991e079006vboxsync {

27152389a84c6dec057fba6fc21241991e079006vboxsync /* Get the client id. */

27152389a84c6dec057fba6fc21241991e079006vboxsync uint32_t u32ClientId;

27152389a84c6dec057fba6fc21241991e079006vboxsync rc = SSMR3GetU32(pSSM, &u32ClientId);

27152389a84c6dec057fba6fc21241991e079006vboxsync if (RT_FAILURE(rc))

27152389a84c6dec057fba6fc21241991e079006vboxsync {

27152389a84c6dec057fba6fc21241991e079006vboxsync pSvc->ReleaseService ();

27152389a84c6dec057fba6fc21241991e079006vboxsync AssertFailed();

895d68f6e6c15446ca98b8f6ec96ce19d7ea356fvboxsync return rc;

27152389a84c6dec057fba6fc21241991e079006vboxsync }

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync /* Connect the client. */

27152389a84c6dec057fba6fc21241991e079006vboxsync rc = pSvc->CreateAndConnectClient (NULL, u32ClientId);

27152389a84c6dec057fba6fc21241991e079006vboxsync if (RT_FAILURE(rc))

27152389a84c6dec057fba6fc21241991e079006vboxsync {

27152389a84c6dec057fba6fc21241991e079006vboxsync pSvc->ReleaseService ();

27152389a84c6dec057fba6fc21241991e079006vboxsync AssertLogRelMsgFailed(("rc=%Rrc %s\n", rc, pszServiceName));

27152389a84c6dec057fba6fc21241991e079006vboxsync return rc;

27152389a84c6dec057fba6fc21241991e079006vboxsync }

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync /* Call the service, so the operation is executed by the service thread. */

27152389a84c6dec057fba6fc21241991e079006vboxsync rc = pSvc->loadClientState (u32ClientId, pSSM);

27152389a84c6dec057fba6fc21241991e079006vboxsync if (RT_FAILURE(rc))

27152389a84c6dec057fba6fc21241991e079006vboxsync {

27152389a84c6dec057fba6fc21241991e079006vboxsync pSvc->ReleaseService ();

27152389a84c6dec057fba6fc21241991e079006vboxsync AssertLogRelMsgFailed(("rc=%Rrc %s\n", rc, pszServiceName));

27152389a84c6dec057fba6fc21241991e079006vboxsync return rc;

895d68f6e6c15446ca98b8f6ec96ce19d7ea356fvboxsync }

27152389a84c6dec057fba6fc21241991e079006vboxsync }

27152389a84c6dec057fba6fc21241991e079006vboxsync

c9a51cded51b8f691f24de730e190ef573344bfcvboxsync pSvc->ReleaseService ();

27152389a84c6dec057fba6fc21241991e079006vboxsync }

27152389a84c6dec057fba6fc21241991e079006vboxsync

c9a51cded51b8f691f24de730e190ef573344bfcvboxsync return VINF_SUCCESS;

27152389a84c6dec057fba6fc21241991e079006vboxsync}

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsyncint HGCMService::CreateAndConnectClient (uint32_t *pu32ClientIdOut, uint32_t u32ClientIdIn)

27152389a84c6dec057fba6fc21241991e079006vboxsync{

27152389a84c6dec057fba6fc21241991e079006vboxsync LogFlowFunc(("pu32ClientIdOut = %p, u32ClientIdIn = %d\n", pu32ClientIdOut, u32ClientIdIn));

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync /* Allocate a client information structure. */

27152389a84c6dec057fba6fc21241991e079006vboxsync HGCMClient *pClient = new HGCMClient ();

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync if (!pClient)

27152389a84c6dec057fba6fc21241991e079006vboxsync {

27152389a84c6dec057fba6fc21241991e079006vboxsync LogWarningFunc(("Could not allocate HGCMClient!!!\n"));

27152389a84c6dec057fba6fc21241991e079006vboxsync return VERR_NO_MEMORY;

27152389a84c6dec057fba6fc21241991e079006vboxsync }

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync uint32_t handle;

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync if (pu32ClientIdOut != NULL)

27152389a84c6dec057fba6fc21241991e079006vboxsync {

27152389a84c6dec057fba6fc21241991e079006vboxsync handle = hgcmObjGenerateHandle (pClient);

27152389a84c6dec057fba6fc21241991e079006vboxsync }

27152389a84c6dec057fba6fc21241991e079006vboxsync else

27152389a84c6dec057fba6fc21241991e079006vboxsync {

5558d00238c7e019ef8c0803358fae6edeba394evboxsync handle = hgcmObjAssignHandle (pClient, u32ClientIdIn);

5558d00238c7e019ef8c0803358fae6edeba394evboxsync }

5558d00238c7e019ef8c0803358fae6edeba394evboxsync

5558d00238c7e019ef8c0803358fae6edeba394evboxsync LogFlowFunc(("client id = %d\n", handle));

5558d00238c7e019ef8c0803358fae6edeba394evboxsync

5558d00238c7e019ef8c0803358fae6edeba394evboxsync AssertRelease(handle);

5558d00238c7e019ef8c0803358fae6edeba394evboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync /* Initialize the HGCM part of the client. */

27152389a84c6dec057fba6fc21241991e079006vboxsync int rc = pClient->Init (this);

27152389a84c6dec057fba6fc21241991e079006vboxsync

5558d00238c7e019ef8c0803358fae6edeba394evboxsync if (RT_SUCCESS(rc))

27152389a84c6dec057fba6fc21241991e079006vboxsync {

ce9015004a47e95eb75e047ba42f2d3200d2d222vboxsync /* Call the service. */

27152389a84c6dec057fba6fc21241991e079006vboxsync HGCMMSGHANDLE hMsg;

27152389a84c6dec057fba6fc21241991e079006vboxsync

ce9015004a47e95eb75e047ba42f2d3200d2d222vboxsync rc = hgcmMsgAlloc (m_thread, &hMsg, SVC_MSG_CONNECT, hgcmMessageAllocSvc);

ce9015004a47e95eb75e047ba42f2d3200d2d222vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync if (RT_SUCCESS(rc))

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync {

27152389a84c6dec057fba6fc21241991e079006vboxsync HGCMMsgSvcConnect *pMsg = (HGCMMsgSvcConnect *)hgcmObjReference (hMsg, HGCMOBJ_MSG);

27152389a84c6dec057fba6fc21241991e079006vboxsync AssertRelease(pMsg);

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync pMsg->u32ClientId = handle;

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync hgcmObjDereference (pMsg);

27152389a84c6dec057fba6fc21241991e079006vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync rc = hgcmMsgSend (hMsg);

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync if (RT_SUCCESS(rc))

27152389a84c6dec057fba6fc21241991e079006vboxsync {

27152389a84c6dec057fba6fc21241991e079006vboxsync /* Add the client Id to the array. */

27152389a84c6dec057fba6fc21241991e079006vboxsync if (m_cClients == m_cClientsAllocated)

27152389a84c6dec057fba6fc21241991e079006vboxsync {

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync m_paClientIds = (uint32_t *)RTMemRealloc (m_paClientIds, (m_cClientsAllocated + 64) * sizeof (m_paClientIds[0]));

27152389a84c6dec057fba6fc21241991e079006vboxsync Assert(m_paClientIds);

27152389a84c6dec057fba6fc21241991e079006vboxsync m_cClientsAllocated += 64;

895d68f6e6c15446ca98b8f6ec96ce19d7ea356fvboxsync }

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync m_paClientIds[m_cClients] = handle;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync m_cClients++;

27152389a84c6dec057fba6fc21241991e079006vboxsync }

27152389a84c6dec057fba6fc21241991e079006vboxsync }

27152389a84c6dec057fba6fc21241991e079006vboxsync }

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync if (RT_FAILURE(rc))

27152389a84c6dec057fba6fc21241991e079006vboxsync {

27152389a84c6dec057fba6fc21241991e079006vboxsync hgcmObjDeleteHandle (handle);

895d68f6e6c15446ca98b8f6ec96ce19d7ea356fvboxsync }

27152389a84c6dec057fba6fc21241991e079006vboxsync else

27152389a84c6dec057fba6fc21241991e079006vboxsync {

27152389a84c6dec057fba6fc21241991e079006vboxsync if (pu32ClientIdOut != NULL)

27152389a84c6dec057fba6fc21241991e079006vboxsync {

895d68f6e6c15446ca98b8f6ec96ce19d7ea356fvboxsync *pu32ClientIdOut = handle;

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync }

27152389a84c6dec057fba6fc21241991e079006vboxsync

c9a51cded51b8f691f24de730e190ef573344bfcvboxsync ReferenceService ();

c9a51cded51b8f691f24de730e190ef573344bfcvboxsync }

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

feef019e08ac3cead66d03feb9094cac75d9bc0dvboxsync LogFlowFunc(("rc = %Rrc\n", rc));

feef019e08ac3cead66d03feb9094cac75d9bc0dvboxsync return rc;

27152389a84c6dec057fba6fc21241991e079006vboxsync}

feef019e08ac3cead66d03feb9094cac75d9bc0dvboxsync

feef019e08ac3cead66d03feb9094cac75d9bc0dvboxsyncint HGCMService::DisconnectClient (uint32_t u32ClientId, bool fFromService)

27152389a84c6dec057fba6fc21241991e079006vboxsync{

27152389a84c6dec057fba6fc21241991e079006vboxsync int rc = VINF_SUCCESS;

feef019e08ac3cead66d03feb9094cac75d9bc0dvboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync LogFlowFunc(("client id = %d, fFromService = %d\n", u32ClientId, fFromService));

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync if (!fFromService)

f5b7890acb8ed84648facc56aa99ef3c6dbf90c7vboxsync {

27152389a84c6dec057fba6fc21241991e079006vboxsync /* Call the service. */

27152389a84c6dec057fba6fc21241991e079006vboxsync HGCMMSGHANDLE hMsg;

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync rc = hgcmMsgAlloc (m_thread, &hMsg, SVC_MSG_DISCONNECT, hgcmMessageAllocSvc);

27152389a84c6dec057fba6fc21241991e079006vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync if (RT_SUCCESS(rc))

27152389a84c6dec057fba6fc21241991e079006vboxsync {

27152389a84c6dec057fba6fc21241991e079006vboxsync HGCMMsgSvcDisconnect *pMsg = (HGCMMsgSvcDisconnect *)hgcmObjReference (hMsg, HGCMOBJ_MSG);

27152389a84c6dec057fba6fc21241991e079006vboxsync AssertRelease(pMsg);

27152389a84c6dec057fba6fc21241991e079006vboxsync

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync pMsg->u32ClientId = u32ClientId;

27152389a84c6dec057fba6fc21241991e079006vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync hgcmObjDereference (pMsg);

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync

27152389a84c6dec057fba6fc21241991e079006vboxsync rc = hgcmMsgSend (hMsg);

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync }

27152389a84c6dec057fba6fc21241991e079006vboxsync else

27152389a84c6dec057fba6fc21241991e079006vboxsync {

27152389a84c6dec057fba6fc21241991e079006vboxsync LogRel(("(%d, %d) [%s] hgcmMsgAlloc(%p, SVC_MSG_DISCONNECT) failed %Rrc\n",

2ca380caf80f0dacc65f8c996077e827318f1c69vboxsync u32ClientId, fFromService, RT_VALID_PTR(m_pszSvcName)? m_pszSvcName: "", m_thread, rc));