/* $Id: VirtualBoxImpl.cpp 55209 2015-04-13 14:26:04Z vboxsync $ */ /** @file * Implementation of IVirtualBox in VBoxSVC. */ /* * Copyright (C) 2006-2015 Oracle Corporation * * This file is part of VirtualBox Open Source Edition (OSE), as * available from http://www.virtualbox.org. This file is free software; * you can redistribute it and/or modify it under the terms of the GNU * General Public License (GPL) as published by the Free Software * Foundation, in version 2 as it comes in the "COPYING" file of the * VirtualBox OSE distribution. VirtualBox OSE is distributed in the * hope that it will be useful, but WITHOUT ANY WARRANTY of any kind. */ #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include "VBox/com/EventQueue.h" #include "VBox/com/MultiResult.h" #include #include #include #include #include #include #include #include #include // for auto_ptr #include "VirtualBoxImpl.h" #include "Global.h" #include "MachineImpl.h" #include "MediumImpl.h" #include "SharedFolderImpl.h" #include "ProgressImpl.h" #include "HostImpl.h" #include "USBControllerImpl.h" #include "SystemPropertiesImpl.h" #include "GuestOSTypeImpl.h" #include "NetworkServiceRunner.h" #include "DHCPServerImpl.h" #include "NATNetworkImpl.h" #ifdef VBOX_WITH_RESOURCE_USAGE_API # include "PerformanceImpl.h" #endif /* VBOX_WITH_RESOURCE_USAGE_API */ #include "EventImpl.h" #ifdef VBOX_WITH_EXTPACK # include "ExtPackManagerImpl.h" #endif #include "AutostartDb.h" #include "ClientWatcher.h" #include "AutoCaller.h" #include "Logging.h" #include #ifdef RT_OS_WINDOWS # include "win/svchlp.h" # include "win/VBoxComEvents.h" #endif //////////////////////////////////////////////////////////////////////////////// // // Definitions // //////////////////////////////////////////////////////////////////////////////// #define VBOX_GLOBAL_SETTINGS_FILE "VirtualBox.xml" //////////////////////////////////////////////////////////////////////////////// // // Global variables // //////////////////////////////////////////////////////////////////////////////// // static com::Utf8Str VirtualBox::sVersion; // static com::Utf8Str VirtualBox::sVersionNormalized; // static ULONG VirtualBox::sRevision; // static com::Utf8Str VirtualBox::sPackageType; // static com::Utf8Str VirtualBox::sAPIVersion; // static std::map VirtualBox::sNatNetworkNameToRefCount; // static leaked (todo: find better place to free it.) RWLockHandle *VirtualBox::spMtxNatNetworkNameToRefCountLock; //////////////////////////////////////////////////////////////////////////////// // // CallbackEvent class // //////////////////////////////////////////////////////////////////////////////// /** * Abstract callback event class to asynchronously call VirtualBox callbacks * on a dedicated event thread. Subclasses reimplement #handleCallback() * to call appropriate IVirtualBoxCallback methods depending on the event * to be dispatched. * * @note The VirtualBox instance passed to the constructor is strongly * referenced, so that the VirtualBox singleton won't be released until the * event gets handled by the event thread. */ class VirtualBox::CallbackEvent : public Event { public: CallbackEvent(VirtualBox *aVirtualBox, VBoxEventType_T aWhat) : mVirtualBox(aVirtualBox), mWhat(aWhat) { Assert(aVirtualBox); } void *handler(); virtual HRESULT prepareEventDesc(IEventSource* aSource, VBoxEventDesc& aEvDesc) = 0; private: /** * Note that this is a weak ref -- the CallbackEvent handler thread * is bound to the lifetime of the VirtualBox instance, so it's safe. */ VirtualBox *mVirtualBox; protected: VBoxEventType_T mWhat; }; //////////////////////////////////////////////////////////////////////////////// // // VirtualBox private member data definition // //////////////////////////////////////////////////////////////////////////////// typedef ObjectsList MediaOList; typedef ObjectsList GuestOSTypesOList; typedef ObjectsList SharedFoldersOList; typedef ObjectsList DHCPServersOList; typedef ObjectsList NATNetworksOList; typedef std::map > ProgressMap; typedef std::map > HardDiskMap; /** * Main VirtualBox data structure. * @note |const| members are persistent during lifetime so can be accessed * without locking. */ struct VirtualBox::Data { Data() : pMainConfigFile(NULL), uuidMediaRegistry("48024e5c-fdd9-470f-93af-ec29f7ea518c"), uRegistryNeedsSaving(0), lockMachines(LOCKCLASS_LISTOFMACHINES), allMachines(lockMachines), lockGuestOSTypes(LOCKCLASS_LISTOFOTHEROBJECTS), allGuestOSTypes(lockGuestOSTypes), lockMedia(LOCKCLASS_LISTOFMEDIA), allHardDisks(lockMedia), allDVDImages(lockMedia), allFloppyImages(lockMedia), lockSharedFolders(LOCKCLASS_LISTOFOTHEROBJECTS), allSharedFolders(lockSharedFolders), lockDHCPServers(LOCKCLASS_LISTOFOTHEROBJECTS), allDHCPServers(lockDHCPServers), lockNATNetworks(LOCKCLASS_LISTOFOTHEROBJECTS), allNATNetworks(lockNATNetworks), mtxProgressOperations(LOCKCLASS_PROGRESSLIST), pClientWatcher(NULL), threadAsyncEvent(NIL_RTTHREAD), pAsyncEventQ(NULL), pAutostartDb(NULL), fSettingsCipherKeySet(false) { } ~Data() { if (pMainConfigFile) { delete pMainConfigFile; pMainConfigFile = NULL; } }; // const data members not requiring locking const Utf8Str strHomeDir; // VirtualBox main settings file const Utf8Str strSettingsFilePath; settings::MainConfigFile *pMainConfigFile; // constant pseudo-machine ID for global media registry const Guid uuidMediaRegistry; // counter if global media registry needs saving, updated using atomic // operations, without requiring any locks uint64_t uRegistryNeedsSaving; // const objects not requiring locking const ComObjPtr pHost; const ComObjPtr pSystemProperties; #ifdef VBOX_WITH_RESOURCE_USAGE_API const ComObjPtr pPerformanceCollector; #endif /* VBOX_WITH_RESOURCE_USAGE_API */ // Each of the following lists use a particular lock handle that protects the // list as a whole. As opposed to version 3.1 and earlier, these lists no // longer need the main VirtualBox object lock, but only the respective list // lock. In each case, the locking order is defined that the list must be // requested before object locks of members of the lists (see the order definitions // in AutoLock.h; e.g. LOCKCLASS_LISTOFMACHINES before LOCKCLASS_MACHINEOBJECT). RWLockHandle lockMachines; MachinesOList allMachines; RWLockHandle lockGuestOSTypes; GuestOSTypesOList allGuestOSTypes; // All the media lists are protected by the following locking handle: RWLockHandle lockMedia; MediaOList allHardDisks, // base images only! allDVDImages, allFloppyImages; // the hard disks map is an additional map sorted by UUID for quick lookup // and contains ALL hard disks (base and differencing); it is protected by // the same lock as the other media lists above HardDiskMap mapHardDisks; // list of pending machine renames (also protected by media tree lock; // see VirtualBox::rememberMachineNameChangeForMedia()) struct PendingMachineRename { Utf8Str strConfigDirOld; Utf8Str strConfigDirNew; }; typedef std::list PendingMachineRenamesList; PendingMachineRenamesList llPendingMachineRenames; RWLockHandle lockSharedFolders; SharedFoldersOList allSharedFolders; RWLockHandle lockDHCPServers; DHCPServersOList allDHCPServers; RWLockHandle lockNATNetworks; NATNetworksOList allNATNetworks; RWLockHandle mtxProgressOperations; ProgressMap mapProgressOperations; ClientWatcher * const pClientWatcher; // the following are data for the async event thread const RTTHREAD threadAsyncEvent; EventQueue * const pAsyncEventQ; const ComObjPtr pEventSource; #ifdef VBOX_WITH_EXTPACK /** The extension pack manager object lives here. */ const ComObjPtr ptrExtPackManager; #endif /** The global autostart database for the user. */ AutostartDb * const pAutostartDb; /** Settings secret */ bool fSettingsCipherKeySet; uint8_t SettingsCipherKey[RTSHA512_HASH_SIZE]; }; // constructor / destructor ///////////////////////////////////////////////////////////////////////////// DEFINE_EMPTY_CTOR_DTOR(VirtualBox) HRESULT VirtualBox::FinalConstruct() { LogFlowThisFunc(("\n")); HRESULT rc = init(); BaseFinalConstruct(); return rc; } void VirtualBox::FinalRelease() { LogFlowThisFunc(("\n")); uninit(); BaseFinalRelease(); } // public initializer/uninitializer for internal purposes only ///////////////////////////////////////////////////////////////////////////// /** * Initializes the VirtualBox object. * * @return COM result code */ HRESULT VirtualBox::init() { /* Enclose the state transition NotReady->InInit->Ready */ AutoInitSpan autoInitSpan(this); AssertReturn(autoInitSpan.isOk(), E_FAIL); /* Locking this object for writing during init sounds a bit paradoxical, * but in the current locking mess this avoids that some code gets a * read lock and later calls code which wants the same write lock. */ AutoWriteLock lock(this COMMA_LOCKVAL_SRC_POS); // allocate our instance data m = new Data; LogFlow(("===========================================================\n")); LogFlowThisFuncEnter(); if (sVersion.isEmpty()) sVersion = RTBldCfgVersion(); if (sVersionNormalized.isEmpty()) { Utf8Str tmp(RTBldCfgVersion()); if (tmp.endsWith(VBOX_BUILD_PUBLISHER)) tmp = tmp.substr(0, tmp.length() - strlen(VBOX_BUILD_PUBLISHER)); sVersionNormalized = tmp; } sRevision = RTBldCfgRevision(); if (sPackageType.isEmpty()) sPackageType = VBOX_PACKAGE_STRING; if (sAPIVersion.isEmpty()) sAPIVersion = VBOX_API_VERSION_STRING; if (!spMtxNatNetworkNameToRefCountLock) spMtxNatNetworkNameToRefCountLock = new RWLockHandle(LOCKCLASS_VIRTUALBOXOBJECT); LogFlowThisFunc(("Version: %s, Package: %s, API Version: %s\n", sVersion.c_str(), sPackageType.c_str(), sAPIVersion.c_str())); /* Get the VirtualBox home directory. */ { char szHomeDir[RTPATH_MAX]; int vrc = com::GetVBoxUserHomeDirectory(szHomeDir, sizeof(szHomeDir)); if (RT_FAILURE(vrc)) return setError(E_FAIL, tr("Could not create the VirtualBox home directory '%s' (%Rrc)"), szHomeDir, vrc); unconst(m->strHomeDir) = szHomeDir; } /* compose the VirtualBox.xml file name */ unconst(m->strSettingsFilePath) = Utf8StrFmt("%s%c%s", m->strHomeDir.c_str(), RTPATH_DELIMITER, VBOX_GLOBAL_SETTINGS_FILE); HRESULT rc = S_OK; bool fCreate = false; try { // load and parse VirtualBox.xml; this will throw on XML or logic errors try { m->pMainConfigFile = new settings::MainConfigFile(&m->strSettingsFilePath); } catch (xml::EIPRTFailure &e) { // this is thrown by the XML backend if the RTOpen() call fails; // only if the main settings file does not exist, create it, // if there's something more serious, then do fail! if (e.rc() == VERR_FILE_NOT_FOUND) fCreate = true; else throw; } if (fCreate) m->pMainConfigFile = new settings::MainConfigFile(NULL); #ifdef VBOX_WITH_RESOURCE_USAGE_API /* create the performance collector object BEFORE host */ unconst(m->pPerformanceCollector).createObject(); rc = m->pPerformanceCollector->init(); ComAssertComRCThrowRC(rc); #endif /* VBOX_WITH_RESOURCE_USAGE_API */ /* create the host object early, machines will need it */ unconst(m->pHost).createObject(); rc = m->pHost->init(this); ComAssertComRCThrowRC(rc); rc = m->pHost->i_loadSettings(m->pMainConfigFile->host); if (FAILED(rc)) throw rc; /* * Create autostart database object early, because the system properties * might need it. */ unconst(m->pAutostartDb) = new AutostartDb; #ifdef VBOX_WITH_EXTPACK /* * Initialize extension pack manager before system properties because * it is required for the VD plugins. */ rc = unconst(m->ptrExtPackManager).createObject(); if (SUCCEEDED(rc)) rc = m->ptrExtPackManager->initExtPackManager(this, VBOXEXTPACKCTX_PER_USER_DAEMON); if (FAILED(rc)) throw rc; #endif /* create the system properties object, someone may need it too */ unconst(m->pSystemProperties).createObject(); rc = m->pSystemProperties->init(this); ComAssertComRCThrowRC(rc); rc = m->pSystemProperties->i_loadSettings(m->pMainConfigFile->systemProperties); if (FAILED(rc)) throw rc; /* guest OS type objects, needed by machines */ for (size_t i = 0; i < Global::cOSTypes; ++i) { ComObjPtr guestOSTypeObj; rc = guestOSTypeObj.createObject(); if (SUCCEEDED(rc)) { rc = guestOSTypeObj->init(Global::sOSTypes[i]); if (SUCCEEDED(rc)) m->allGuestOSTypes.addChild(guestOSTypeObj); } ComAssertComRCThrowRC(rc); } /* all registered media, needed by machines */ if (FAILED(rc = initMedia(m->uuidMediaRegistry, m->pMainConfigFile->mediaRegistry, Utf8Str::Empty))) // const Utf8Str &machineFolder throw rc; /* machines */ if (FAILED(rc = initMachines())) throw rc; #ifdef DEBUG LogFlowThisFunc(("Dumping media backreferences\n")); i_dumpAllBackRefs(); #endif /* net services - dhcp services */ for (settings::DHCPServersList::const_iterator it = m->pMainConfigFile->llDhcpServers.begin(); it != m->pMainConfigFile->llDhcpServers.end(); ++it) { const settings::DHCPServer &data = *it; ComObjPtr pDhcpServer; if (SUCCEEDED(rc = pDhcpServer.createObject())) rc = pDhcpServer->init(this, data); if (FAILED(rc)) throw rc; rc = i_registerDHCPServer(pDhcpServer, false /* aSaveRegistry */); if (FAILED(rc)) throw rc; } /* net services - nat networks */ for (settings::NATNetworksList::const_iterator it = m->pMainConfigFile->llNATNetworks.begin(); it != m->pMainConfigFile->llNATNetworks.end(); ++it) { const settings::NATNetwork &net = *it; ComObjPtr pNATNetwork; if (SUCCEEDED(rc = pNATNetwork.createObject())) { rc = pNATNetwork->init(this, net); AssertComRCReturnRC(rc); } rc = i_registerNATNetwork(pNATNetwork, false /* aSaveRegistry */); AssertComRCReturnRC(rc); } /* events */ if (SUCCEEDED(rc = unconst(m->pEventSource).createObject())) rc = m->pEventSource->init(); if (FAILED(rc)) throw rc; } catch (HRESULT err) { /* we assume that error info is set by the thrower */ rc = err; } catch (...) { rc = VirtualBoxBase::handleUnexpectedExceptions(this, RT_SRC_POS); } if (SUCCEEDED(rc)) { /* set up client monitoring */ try { unconst(m->pClientWatcher) = new ClientWatcher(this); if (!m->pClientWatcher->isReady()) { delete m->pClientWatcher; unconst(m->pClientWatcher) = NULL; rc = E_FAIL; } } catch (std::bad_alloc &) { rc = E_OUTOFMEMORY; } } if (SUCCEEDED(rc)) { try { /* start the async event handler thread */ int vrc = RTThreadCreate(&unconst(m->threadAsyncEvent), AsyncEventHandler, &unconst(m->pAsyncEventQ), 0, RTTHREADTYPE_MAIN_WORKER, RTTHREADFLAGS_WAITABLE, "EventHandler"); ComAssertRCThrow(vrc, E_FAIL); /* wait until the thread sets m->pAsyncEventQ */ RTThreadUserWait(m->threadAsyncEvent, RT_INDEFINITE_WAIT); ComAssertThrow(m->pAsyncEventQ, E_FAIL); } catch (HRESULT aRC) { rc = aRC; } } /* Confirm a successful initialization when it's the case */ if (SUCCEEDED(rc)) autoInitSpan.setSucceeded(); #ifdef VBOX_WITH_EXTPACK /* Let the extension packs have a go at things. */ if (SUCCEEDED(rc)) { lock.release(); m->ptrExtPackManager->i_callAllVirtualBoxReadyHooks(); } #endif LogFlowThisFunc(("rc=%08X\n", rc)); LogFlowThisFuncLeave(); LogFlow(("===========================================================\n")); return rc; } HRESULT VirtualBox::initMachines() { for (settings::MachinesRegistry::const_iterator it = m->pMainConfigFile->llMachines.begin(); it != m->pMainConfigFile->llMachines.end(); ++it) { HRESULT rc = S_OK; const settings::MachineRegistryEntry &xmlMachine = *it; Guid uuid = xmlMachine.uuid; /* Check if machine record has valid parameters. */ if (xmlMachine.strSettingsFile.isEmpty() || uuid.isZero()) { LogRel(("Skipped invalid machine record.\n")); continue; } ComObjPtr pMachine; if (SUCCEEDED(rc = pMachine.createObject())) { rc = pMachine->initFromSettings(this, xmlMachine.strSettingsFile, &uuid); if (SUCCEEDED(rc)) rc = i_registerMachine(pMachine); if (FAILED(rc)) return rc; } } return S_OK; } /** * Loads a media registry from XML and adds the media contained therein to * the global lists of known media. * * This now (4.0) gets called from two locations: * * -- VirtualBox::init(), to load the global media registry from VirtualBox.xml; * * -- Machine::loadMachineDataFromSettings(), to load the per-machine registry * from machine XML, for machines created with VirtualBox 4.0 or later. * * In both cases, the media found are added to the global lists so the * global arrays of media (including the GUI's virtual media manager) * continue to work as before. * * @param uuidMachineRegistry The UUID of the media registry. This is either the * transient UUID created at VirtualBox startup for the global registry or * a machine ID. * @param mediaRegistry The XML settings structure to load, either from VirtualBox.xml * or a machine XML. * @return */ HRESULT VirtualBox::initMedia(const Guid &uuidRegistry, const settings::MediaRegistry &mediaRegistry, const Utf8Str &strMachineFolder) { LogFlow(("VirtualBox::initMedia ENTERING, uuidRegistry=%s, strMachineFolder=%s\n", uuidRegistry.toString().c_str(), strMachineFolder.c_str())); AutoWriteLock treeLock(i_getMediaTreeLockHandle() COMMA_LOCKVAL_SRC_POS); HRESULT rc = S_OK; settings::MediaList::const_iterator it; for (it = mediaRegistry.llHardDisks.begin(); it != mediaRegistry.llHardDisks.end(); ++it) { const settings::Medium &xmlHD = *it; ComObjPtr pHardDisk; if (SUCCEEDED(rc = pHardDisk.createObject())) rc = pHardDisk->init(this, NULL, // parent DeviceType_HardDisk, uuidRegistry, xmlHD, // XML data; this recurses to processes the children strMachineFolder, treeLock); if (FAILED(rc)) return rc; rc = i_registerMedium(pHardDisk, &pHardDisk, treeLock); if (FAILED(rc)) return rc; } for (it = mediaRegistry.llDvdImages.begin(); it != mediaRegistry.llDvdImages.end(); ++it) { const settings::Medium &xmlDvd = *it; ComObjPtr pImage; if (SUCCEEDED(pImage.createObject())) rc = pImage->init(this, NULL, DeviceType_DVD, uuidRegistry, xmlDvd, strMachineFolder, treeLock); if (FAILED(rc)) return rc; rc = i_registerMedium(pImage, &pImage, treeLock); if (FAILED(rc)) return rc; } for (it = mediaRegistry.llFloppyImages.begin(); it != mediaRegistry.llFloppyImages.end(); ++it) { const settings::Medium &xmlFloppy = *it; ComObjPtr pImage; if (SUCCEEDED(pImage.createObject())) rc = pImage->init(this, NULL, DeviceType_Floppy, uuidRegistry, xmlFloppy, strMachineFolder, treeLock); if (FAILED(rc)) return rc; rc = i_registerMedium(pImage, &pImage, treeLock); if (FAILED(rc)) return rc; } LogFlow(("VirtualBox::initMedia LEAVING\n")); return S_OK; } void VirtualBox::uninit() { Assert(!m->uRegistryNeedsSaving); if (m->uRegistryNeedsSaving) i_saveSettings(); /* Enclose the state transition Ready->InUninit->NotReady */ AutoUninitSpan autoUninitSpan(this); if (autoUninitSpan.uninitDone()) return; LogFlow(("===========================================================\n")); LogFlowThisFuncEnter(); LogFlowThisFunc(("initFailed()=%d\n", autoUninitSpan.initFailed())); /* tell all our child objects we've been uninitialized */ LogFlowThisFunc(("Uninitializing machines (%d)...\n", m->allMachines.size())); if (m->pHost) { /* It is necessary to hold the VirtualBox and Host locks here because we may have to uninitialize SessionMachines. */ AutoMultiWriteLock2 multilock(this, m->pHost COMMA_LOCKVAL_SRC_POS); m->allMachines.uninitAll(); } else m->allMachines.uninitAll(); m->allFloppyImages.uninitAll(); m->allDVDImages.uninitAll(); m->allHardDisks.uninitAll(); m->allDHCPServers.uninitAll(); m->mapProgressOperations.clear(); m->allGuestOSTypes.uninitAll(); /* Note that we release singleton children after we've all other children. * In some cases this is important because these other children may use * some resources of the singletons which would prevent them from * uninitializing (as for example, mSystemProperties which owns * MediumFormat objects which Medium objects refer to) */ if (m->pSystemProperties) { m->pSystemProperties->uninit(); unconst(m->pSystemProperties).setNull(); } if (m->pHost) { m->pHost->uninit(); unconst(m->pHost).setNull(); } #ifdef VBOX_WITH_RESOURCE_USAGE_API if (m->pPerformanceCollector) { m->pPerformanceCollector->uninit(); unconst(m->pPerformanceCollector).setNull(); } #endif /* VBOX_WITH_RESOURCE_USAGE_API */ LogFlowThisFunc(("Terminating the async event handler...\n")); if (m->threadAsyncEvent != NIL_RTTHREAD) { /* signal to exit the event loop */ if (RT_SUCCESS(m->pAsyncEventQ->interruptEventQueueProcessing())) { /* * Wait for thread termination (only after we've successfully * interrupted the event queue processing!) */ int vrc = RTThreadWait(m->threadAsyncEvent, 60000, NULL); if (RT_FAILURE(vrc)) LogWarningFunc(("RTThreadWait(%RTthrd) -> %Rrc\n", m->threadAsyncEvent, vrc)); } else { AssertMsgFailed(("interruptEventQueueProcessing() failed\n")); RTThreadWait(m->threadAsyncEvent, 0, NULL); } unconst(m->threadAsyncEvent) = NIL_RTTHREAD; unconst(m->pAsyncEventQ) = NULL; } LogFlowThisFunc(("Releasing event source...\n")); if (m->pEventSource) { // Must uninit the event source here, because it makes no sense that // it survives longer than the base object. If someone gets an event // with such an event source then that's life and it has to be dealt // with appropriately on the API client side. m->pEventSource->uninit(); unconst(m->pEventSource).setNull(); } LogFlowThisFunc(("Terminating the client watcher...\n")); if (m->pClientWatcher) { delete m->pClientWatcher; unconst(m->pClientWatcher) = NULL; } delete m->pAutostartDb; // clean up our instance data delete m; /* Unload hard disk plugin backends. */ VDShutdown(); LogFlowThisFuncLeave(); LogFlow(("===========================================================\n")); } // Wrapped IVirtualBox properties ///////////////////////////////////////////////////////////////////////////// HRESULT VirtualBox::getVersion(com::Utf8Str &aVersion) { aVersion = sVersion; return S_OK; } HRESULT VirtualBox::getVersionNormalized(com::Utf8Str &aVersionNormalized) { aVersionNormalized = sVersionNormalized; return S_OK; } HRESULT VirtualBox::getRevision(ULONG *aRevision) { *aRevision = sRevision; return S_OK; } HRESULT VirtualBox::getPackageType(com::Utf8Str &aPackageType) { aPackageType = sPackageType; return S_OK; } HRESULT VirtualBox::getAPIVersion(com::Utf8Str &aAPIVersion) { aAPIVersion = sAPIVersion; return S_OK; } HRESULT VirtualBox::getHomeFolder(com::Utf8Str &aHomeFolder) { /* mHomeDir is const and doesn't need a lock */ aHomeFolder = m->strHomeDir; return S_OK; } HRESULT VirtualBox::getSettingsFilePath(com::Utf8Str &aSettingsFilePath) { /* mCfgFile.mName is const and doesn't need a lock */ aSettingsFilePath = m->strSettingsFilePath; return S_OK; } HRESULT VirtualBox::getHost(ComPtr &aHost) { /* mHost is const, no need to lock */ m->pHost.queryInterfaceTo(aHost.asOutParam()); return S_OK; } HRESULT VirtualBox::getSystemProperties(ComPtr &aSystemProperties) { /* mSystemProperties is const, no need to lock */ m->pSystemProperties.queryInterfaceTo(aSystemProperties.asOutParam()); return S_OK; } HRESULT VirtualBox::getMachines(std::vector > &aMachines) { AutoReadLock al(m->allMachines.getLockHandle() COMMA_LOCKVAL_SRC_POS); aMachines.resize(m->allMachines.size()); size_t i = 0; for (MachinesOList::const_iterator it= m->allMachines.begin(); it!= m->allMachines.end(); ++it, ++i) (*it).queryInterfaceTo(aMachines[i].asOutParam()); return S_OK; } HRESULT VirtualBox::getMachineGroups(std::vector &aMachineGroups) { std::list allGroups; /* get copy of all machine references, to avoid holding the list lock */ MachinesOList::MyList allMachines; { AutoReadLock al(m->allMachines.getLockHandle() COMMA_LOCKVAL_SRC_POS); allMachines = m->allMachines.getList(); } for (MachinesOList::MyList::const_iterator it = allMachines.begin(); it != allMachines.end(); ++it) { const ComObjPtr &pMachine = *it; AutoCaller autoMachineCaller(pMachine); if (FAILED(autoMachineCaller.rc())) continue; AutoReadLock mlock(pMachine COMMA_LOCKVAL_SRC_POS); if (pMachine->i_isAccessible()) { const StringsList &thisGroups = pMachine->i_getGroups(); for (StringsList::const_iterator it2 = thisGroups.begin(); it2 != thisGroups.end(); ++it2) allGroups.push_back(*it2); } } /* throw out any duplicates */ allGroups.sort(); allGroups.unique(); aMachineGroups.resize(allGroups.size()); size_t i = 0; for (std::list::const_iterator it = allGroups.begin(); it != allGroups.end(); ++it, ++i) aMachineGroups[i] = (*it); return S_OK; } HRESULT VirtualBox::getHardDisks(std::vector > &aHardDisks) { AutoReadLock al(m->allHardDisks.getLockHandle() COMMA_LOCKVAL_SRC_POS); aHardDisks.resize(m->allHardDisks.size()); size_t i = 0; for (MediaOList::const_iterator it = m->allHardDisks.begin(); it != m->allHardDisks.end(); ++it, ++i) (*it).queryInterfaceTo(aHardDisks[i].asOutParam()); return S_OK; } HRESULT VirtualBox::getDVDImages(std::vector > &aDVDImages) { AutoReadLock al(m->allDVDImages.getLockHandle() COMMA_LOCKVAL_SRC_POS); aDVDImages.resize(m->allDVDImages.size()); size_t i = 0; for (MediaOList::const_iterator it = m->allDVDImages.begin(); it!= m->allDVDImages.end(); ++it, ++i) (*it).queryInterfaceTo(aDVDImages[i].asOutParam()); return S_OK; } HRESULT VirtualBox::getFloppyImages(std::vector > &aFloppyImages) { AutoReadLock al(m->allFloppyImages.getLockHandle() COMMA_LOCKVAL_SRC_POS); aFloppyImages.resize(m->allFloppyImages.size()); size_t i = 0; for (MediaOList::const_iterator it = m->allFloppyImages.begin(); it != m->allFloppyImages.end(); ++it, ++i) (*it).queryInterfaceTo(aFloppyImages[i].asOutParam()); return S_OK; } HRESULT VirtualBox::getProgressOperations(std::vector > &aProgressOperations) { /* protect mProgressOperations */ AutoReadLock safeLock(m->mtxProgressOperations COMMA_LOCKVAL_SRC_POS); ProgressMap pmap(m->mapProgressOperations); aProgressOperations.resize(pmap.size()); size_t i = 0; for (ProgressMap::iterator it = pmap.begin(); it != pmap.end(); ++it, ++i) it->second.queryInterfaceTo(aProgressOperations[i].asOutParam()); return S_OK; } HRESULT VirtualBox::getGuestOSTypes(std::vector > &aGuestOSTypes) { AutoReadLock al(m->allGuestOSTypes.getLockHandle() COMMA_LOCKVAL_SRC_POS); aGuestOSTypes.resize(m->allGuestOSTypes.size()); size_t i = 0; for (GuestOSTypesOList::const_iterator it = m->allGuestOSTypes.begin(); it != m->allGuestOSTypes.end(); ++it, ++i) (*it).queryInterfaceTo(aGuestOSTypes[i].asOutParam()); return S_OK; } HRESULT VirtualBox::getSharedFolders(std::vector > &aSharedFolders) { #ifndef RT_OS_WINDOWS NOREF(aSharedFolders); #endif /* RT_OS_WINDOWS */ NOREF(aSharedFolders); return setError(E_NOTIMPL, "Not yet implemented"); } HRESULT VirtualBox::getPerformanceCollector(ComPtr &aPerformanceCollector) { #ifdef VBOX_WITH_RESOURCE_USAGE_API /* mPerformanceCollector is const, no need to lock */ m->pPerformanceCollector.queryInterfaceTo(aPerformanceCollector.asOutParam()); return S_OK; #else /* !VBOX_WITH_RESOURCE_USAGE_API */ NOREF(aPerformanceCollector); ReturnComNotImplemented(); #endif /* !VBOX_WITH_RESOURCE_USAGE_API */ } HRESULT VirtualBox::getDHCPServers(std::vector > &aDHCPServers) { AutoReadLock al(m->allDHCPServers.getLockHandle() COMMA_LOCKVAL_SRC_POS); aDHCPServers.resize(m->allDHCPServers.size()); size_t i = 0; for (DHCPServersOList::const_iterator it= m->allDHCPServers.begin(); it!= m->allDHCPServers.end(); ++it, ++i) (*it).queryInterfaceTo(aDHCPServers[i].asOutParam()); return S_OK; } HRESULT VirtualBox::getNATNetworks(std::vector > &aNATNetworks) { #ifdef VBOX_WITH_NAT_SERVICE AutoReadLock al(m->allNATNetworks.getLockHandle() COMMA_LOCKVAL_SRC_POS); aNATNetworks.resize(m->allNATNetworks.size()); size_t i = 0; for (NATNetworksOList::const_iterator it= m->allNATNetworks.begin(); it!= m->allNATNetworks.end(); ++it, ++i) (*it).queryInterfaceTo(aNATNetworks[i].asOutParam()); return S_OK; #else NOREF(aNATNetworks); # ifndef RT_OS_WINDOWS NOREF(aNATNetworks); # endif NOREF(aNATNetworks); return E_NOTIMPL; #endif } HRESULT VirtualBox::getEventSource(ComPtr &aEventSource) { /* event source is const, no need to lock */ m->pEventSource.queryInterfaceTo(aEventSource.asOutParam()); return S_OK; } HRESULT VirtualBox::getExtensionPackManager(ComPtr &aExtensionPackManager) { HRESULT hrc = S_OK; #ifdef VBOX_WITH_EXTPACK /* The extension pack manager is const, no need to lock. */ hrc = m->ptrExtPackManager.queryInterfaceTo(aExtensionPackManager.asOutParam()); #else hrc = E_NOTIMPL; NOREF(aExtensionPackManager); #endif return hrc; } HRESULT VirtualBox::getInternalNetworks(std::vector &aInternalNetworks) { std::list allInternalNetworks; /* get copy of all machine references, to avoid holding the list lock */ MachinesOList::MyList allMachines; { AutoReadLock al(m->allMachines.getLockHandle() COMMA_LOCKVAL_SRC_POS); allMachines = m->allMachines.getList(); } for (MachinesOList::MyList::const_iterator it = allMachines.begin(); it != allMachines.end(); ++it) { const ComObjPtr &pMachine = *it; AutoCaller autoMachineCaller(pMachine); if (FAILED(autoMachineCaller.rc())) continue; AutoReadLock mlock(pMachine COMMA_LOCKVAL_SRC_POS); if (pMachine->i_isAccessible()) { uint32_t cNetworkAdapters = Global::getMaxNetworkAdapters(pMachine->i_getChipsetType()); for (ULONG i = 0; i < cNetworkAdapters; i++) { ComPtr pNet; HRESULT rc = pMachine->GetNetworkAdapter(i, pNet.asOutParam()); if (FAILED(rc) || pNet.isNull()) continue; Bstr strInternalNetwork; rc = pNet->COMGETTER(InternalNetwork)(strInternalNetwork.asOutParam()); if (FAILED(rc) || strInternalNetwork.isEmpty()) continue; allInternalNetworks.push_back(Utf8Str(strInternalNetwork)); } } } /* throw out any duplicates */ allInternalNetworks.sort(); allInternalNetworks.unique(); size_t i = 0; aInternalNetworks.resize(allInternalNetworks.size()); for (std::list::const_iterator it = allInternalNetworks.begin(); it != allInternalNetworks.end(); ++it, ++i) aInternalNetworks[i] = *it; return S_OK; } HRESULT VirtualBox::getGenericNetworkDrivers(std::vector &aGenericNetworkDrivers) { std::list allGenericNetworkDrivers; /* get copy of all machine references, to avoid holding the list lock */ MachinesOList::MyList allMachines; { AutoReadLock al(m->allMachines.getLockHandle() COMMA_LOCKVAL_SRC_POS); allMachines = m->allMachines.getList(); } for (MachinesOList::MyList::const_iterator it = allMachines.begin(); it != allMachines.end(); ++it) { const ComObjPtr &pMachine = *it; AutoCaller autoMachineCaller(pMachine); if (FAILED(autoMachineCaller.rc())) continue; AutoReadLock mlock(pMachine COMMA_LOCKVAL_SRC_POS); if (pMachine->i_isAccessible()) { uint32_t cNetworkAdapters = Global::getMaxNetworkAdapters(pMachine->i_getChipsetType()); for (ULONG i = 0; i < cNetworkAdapters; i++) { ComPtr pNet; HRESULT rc = pMachine->GetNetworkAdapter(i, pNet.asOutParam()); if (FAILED(rc) || pNet.isNull()) continue; Bstr strGenericNetworkDriver; rc = pNet->COMGETTER(GenericDriver)(strGenericNetworkDriver.asOutParam()); if (FAILED(rc) || strGenericNetworkDriver.isEmpty()) continue; allGenericNetworkDrivers.push_back(Utf8Str(strGenericNetworkDriver).c_str()); } } } /* throw out any duplicates */ allGenericNetworkDrivers.sort(); allGenericNetworkDrivers.unique(); aGenericNetworkDrivers.resize(allGenericNetworkDrivers.size()); size_t i = 0; for (std::list::const_iterator it = allGenericNetworkDrivers.begin(); it != allGenericNetworkDrivers.end(); ++it, ++i) aGenericNetworkDrivers[i] = *it; return S_OK; } HRESULT VirtualBox::checkFirmwarePresent(FirmwareType_T aFirmwareType, const com::Utf8Str &aVersion, com::Utf8Str &aUrl, com::Utf8Str &aFile, BOOL *aResult) { NOREF(aVersion); static const struct { FirmwareType_T type; const char* fileName; const char* url; } firmwareDesc[] = { { /* compiled-in firmware */ FirmwareType_BIOS, NULL, NULL }, { FirmwareType_EFI32, "VBoxEFI32.fd", "http://virtualbox.org/firmware/VBoxEFI32.fd" }, { FirmwareType_EFI64, "VBoxEFI64.fd", "http://virtualbox.org/firmware/VBoxEFI64.fd" }, { FirmwareType_EFIDUAL, "VBoxEFIDual.fd", "http://virtualbox.org/firmware/VBoxEFIDual.fd" } }; for (size_t i = 0; i < sizeof(firmwareDesc) / sizeof(firmwareDesc[0]); i++) { if (aFirmwareType != firmwareDesc[i].type) continue; /* compiled-in firmware */ if (firmwareDesc[i].fileName == NULL) { *aResult = TRUE; break; } Utf8Str shortName, fullName; shortName = Utf8StrFmt("Firmware%c%s", RTPATH_DELIMITER, firmwareDesc[i].fileName); int rc = i_calculateFullPath(shortName, fullName); AssertRCReturn(rc, rc); if (RTFileExists(fullName.c_str())) { *aResult = TRUE; aFile = fullName; break; } char pszVBoxPath[RTPATH_MAX]; rc = RTPathExecDir(pszVBoxPath, RTPATH_MAX); AssertRCReturn(rc, rc); fullName = Utf8StrFmt("%s%c%s", pszVBoxPath, RTPATH_DELIMITER, firmwareDesc[i].fileName); if (RTFileExists(fullName.c_str())) { *aResult = TRUE; aFile = fullName; break; } /** @todo: account for version in the URL */ aUrl = firmwareDesc[i].url; *aResult = FALSE; /* Assume single record per firmware type */ break; } return S_OK; } // Wrapped IVirtualBox methods ///////////////////////////////////////////////////////////////////////////// /* Helper for VirtualBox::ComposeMachineFilename */ static void sanitiseMachineFilename(Utf8Str &aName); HRESULT VirtualBox::composeMachineFilename(const com::Utf8Str &aName, const com::Utf8Str &aGroup, const com::Utf8Str &aCreateFlags, const com::Utf8Str &aBaseFolder, com::Utf8Str &aFile) { LogFlowThisFuncEnter(); Utf8Str strBase = aBaseFolder; Utf8Str strName = aName; LogFlowThisFunc(("aName=\"%s\",aBaseFolder=\"%s\"\n", strName.c_str(), strBase.c_str())); Guid id; bool fDirectoryIncludesUUID = false; if (!aCreateFlags.isEmpty()) { size_t uPos = 0; do { com::Utf8Str strKey, strValue; uPos = aCreateFlags.parseKeyValue(strKey, strValue, uPos); if (strKey == "UUID") id = strValue.c_str(); else if (strKey == "directoryIncludesUUID") fDirectoryIncludesUUID = (strValue == "1"); } while(uPos != com::Utf8Str::npos); } if (id.isZero()) fDirectoryIncludesUUID = false; else if (!id.isValid()) { /* do something else */ return setError(E_INVALIDARG, tr("'%s' is not a valid Guid"), id.toStringCurly().c_str()); } Utf8Str strGroup(aGroup); if (strGroup.isEmpty()) strGroup = "/"; HRESULT rc = i_validateMachineGroup(strGroup, true); if (FAILED(rc)) return rc; /* Compose the settings file name using the following scheme: * * //.xml * * If a non-null and non-empty base folder is specified, the default * machine folder will be used as a base folder. * We sanitise the machine name to a safe white list of characters before * using it. */ Utf8Str strDirName(strName); if (fDirectoryIncludesUUID) strDirName += Utf8StrFmt(" (%RTuuid)", id.raw()); sanitiseMachineFilename(strName); sanitiseMachineFilename(strDirName); if (strBase.isEmpty()) /* we use the non-full folder value below to keep the path relative */ i_getDefaultMachineFolder(strBase); i_calculateFullPath(strBase, strBase); /* eliminate toplevel group to avoid // in the result */ if (strGroup == "/") strGroup.setNull(); aFile = com::Utf8StrFmt("%s%s%c%s%c%s.vbox", strBase.c_str(), strGroup.c_str(), RTPATH_DELIMITER, strDirName.c_str(), RTPATH_DELIMITER, strName.c_str()); return S_OK; } /** * Remove characters from a machine file name which can be problematic on * particular systems. * @param strName The file name to sanitise. */ void sanitiseMachineFilename(Utf8Str &strName) { /** Set of characters which should be safe for use in filenames: some basic * ASCII, Unicode from Latin-1 alphabetic to the end of Hangul. We try to * skip anything that could count as a control character in Windows or * *nix, or be otherwise difficult for shells to handle (I would have * preferred to remove the space and brackets too). We also remove all * characters which need UTF-16 surrogate pairs for Windows's benefit. */ RTUNICP aCpSet[] = { ' ', ' ', '(', ')', '-', '.', '0', '9', 'A', 'Z', 'a', 'z', '_', '_', 0xa0, 0xd7af, '\0' }; char *pszName = strName.mutableRaw(); ssize_t cReplacements = RTStrPurgeComplementSet(pszName, aCpSet, '_'); Assert(cReplacements >= 0); NOREF(cReplacements); /* No leading dot or dash. */ if (pszName[0] == '.' || pszName[0] == '-') pszName[0] = '_'; /* No trailing dot. */ if (pszName[strName.length() - 1] == '.') pszName[strName.length() - 1] = '_'; /* Mangle leading and trailing spaces. */ for (size_t i = 0; pszName[i] == ' '; ++i) pszName[i] = '_'; for (size_t i = strName.length() - 1; i && pszName[i] == ' '; --i) pszName[i] = '_'; } #ifdef DEBUG /** Simple unit test/operation examples for sanitiseMachineFilename(). */ static unsigned testSanitiseMachineFilename(void (*pfnPrintf)(const char *, ...)) { unsigned cErrors = 0; /** Expected results of sanitising given file names. */ static struct { /** The test file name to be sanitised (Utf-8). */ const char *pcszIn; /** The expected sanitised output (Utf-8). */ const char *pcszOutExpected; } aTest[] = { { "OS/2 2.1", "OS_2 2.1" }, { "-!My VM!-", "__My VM_-" }, { "\xF0\x90\x8C\xB0", "____" }, { " My VM ", "__My VM__" }, { ".My VM.", "_My VM_" }, { "My VM", "My VM" } }; for (unsigned i = 0; i < RT_ELEMENTS(aTest); ++i) { Utf8Str str(aTest[i].pcszIn); sanitiseMachineFilename(str); if (str.compare(aTest[i].pcszOutExpected)) { ++cErrors; pfnPrintf("%s: line %d, expected %s, actual %s\n", __PRETTY_FUNCTION__, i, aTest[i].pcszOutExpected, str.c_str()); } } return cErrors; } /** @todo Proper testcase. */ /** @todo Do we have a better method of doing init functions? */ namespace { class TestSanitiseMachineFilename { public: TestSanitiseMachineFilename(void) { Assert(!testSanitiseMachineFilename(RTAssertMsg2)); } }; TestSanitiseMachineFilename s_TestSanitiseMachineFilename; } #endif /** @note Locks mSystemProperties object for reading. */ HRESULT VirtualBox::createMachine(const com::Utf8Str &aSettingsFile, const com::Utf8Str &aName, const std::vector &aGroups, const com::Utf8Str &aOsTypeId, const com::Utf8Str &aFlags, ComPtr &aMachine) { LogFlowThisFuncEnter(); LogFlowThisFunc(("aSettingsFile=\"%s\", aName=\"%s\", aOsTypeId =\"%s\", aCreateFlags=\"%s\"\n", aSettingsFile.c_str(), aName.c_str(), aOsTypeId.c_str(), aFlags.c_str())); /** @todo tighten checks on aId? */ StringsList llGroups; HRESULT rc = i_convertMachineGroups(aGroups, &llGroups); if (FAILED(rc)) return rc; Utf8Str strCreateFlags(aFlags); Guid id; bool fForceOverwrite = false; bool fDirectoryIncludesUUID = false; if (!strCreateFlags.isEmpty()) { const char *pcszNext = strCreateFlags.c_str(); while (*pcszNext != '\0') { Utf8Str strFlag; const char *pcszComma = RTStrStr(pcszNext, ","); if (!pcszComma) strFlag = pcszNext; else strFlag = Utf8Str(pcszNext, pcszComma - pcszNext); const char *pcszEqual = RTStrStr(strFlag.c_str(), "="); /* skip over everything which doesn't contain '=' */ if (pcszEqual && pcszEqual != strFlag.c_str()) { Utf8Str strKey(strFlag.c_str(), pcszEqual - strFlag.c_str()); Utf8Str strValue(strFlag.c_str() + (pcszEqual - strFlag.c_str() + 1)); if (strKey == "UUID") id = strValue.c_str(); else if (strKey == "forceOverwrite") fForceOverwrite = (strValue == "1"); else if (strKey == "directoryIncludesUUID") fDirectoryIncludesUUID = (strValue == "1"); } if (!pcszComma) pcszNext += strFlag.length(); else pcszNext += strFlag.length() + 1; } } /* Create UUID if none was specified. */ if (id.isZero()) id.create(); else if (!id.isValid()) { /* do something else */ return setError(E_INVALIDARG, tr("'%s' is not a valid Guid"), id.toStringCurly().c_str()); } /* NULL settings file means compose automatically */ Bstr bstrSettingsFile(aSettingsFile); if (bstrSettingsFile.isEmpty()) { Utf8Str strNewCreateFlags(Utf8StrFmt("UUID=%RTuuid", id.raw())); if (fDirectoryIncludesUUID) strNewCreateFlags += ",directoryIncludesUUID=1"; com::Utf8Str blstr = ""; com::Utf8Str sf = aSettingsFile; rc = composeMachineFilename(aName, llGroups.front(), strNewCreateFlags, blstr /* aBaseFolder */, sf); if (FAILED(rc)) return rc; bstrSettingsFile = Bstr(sf).raw(); } /* create a new object */ ComObjPtr machine; rc = machine.createObject(); if (FAILED(rc)) return rc; GuestOSType *osType = NULL; rc = i_findGuestOSType(Bstr(aOsTypeId), osType); if (FAILED(rc)) return rc; /* initialize the machine object */ rc = machine->init(this, Utf8Str(bstrSettingsFile), Utf8Str(aName), llGroups, osType, id, fForceOverwrite, fDirectoryIncludesUUID); if (SUCCEEDED(rc)) { /* set the return value */ machine.queryInterfaceTo(aMachine.asOutParam()); AssertComRC(rc); #ifdef VBOX_WITH_EXTPACK /* call the extension pack hooks */ m->ptrExtPackManager->i_callAllVmCreatedHooks(machine); #endif } LogFlowThisFuncLeave(); return rc; } HRESULT VirtualBox::openMachine(const com::Utf8Str &aSettingsFile, ComPtr &aMachine) { HRESULT rc = E_FAIL; /* create a new object */ ComObjPtr machine; rc = machine.createObject(); if (SUCCEEDED(rc)) { /* initialize the machine object */ rc = machine->initFromSettings(this, aSettingsFile, NULL); /* const Guid *aId */ if (SUCCEEDED(rc)) { /* set the return value */ machine.queryInterfaceTo(aMachine.asOutParam()); ComAssertComRC(rc); } } return rc; } /** @note Locks objects! */ HRESULT VirtualBox::registerMachine(const ComPtr &aMachine) { HRESULT rc; Bstr name; rc = aMachine->COMGETTER(Name)(name.asOutParam()); if (FAILED(rc)) return rc; /* We can safely cast child to Machine * here because only Machine * implementations of IMachine can be among our children. */ IMachine *aM = aMachine; Machine *pMachine = static_cast(aM); AutoCaller machCaller(pMachine); ComAssertComRCRetRC(machCaller.rc()); rc = i_registerMachine(pMachine); /* fire an event */ if (SUCCEEDED(rc)) i_onMachineRegistered(pMachine->i_getId(), TRUE); return rc; } /** @note Locks this object for reading, then some machine objects for reading. */ HRESULT VirtualBox::findMachine(const com::Utf8Str &aSettingsFile, ComPtr &aMachine) { LogFlowThisFuncEnter(); LogFlowThisFunc(("aSettingsFile=\"%s\", aMachine={%p}\n", aSettingsFile.c_str(), &aMachine)); /* start with not found */ HRESULT rc = S_OK; ComObjPtr pMachineFound; Guid id(Bstr(aSettingsFile).raw()); Utf8Str strFile(aSettingsFile); if (id.isValid() && !id.isZero()) rc = i_findMachine(id, true /* fPermitInaccessible */, true /* setError */, &pMachineFound); // returns VBOX_E_OBJECT_NOT_FOUND if not found and sets error else { rc = i_findMachineByName(strFile, true /* setError */, &pMachineFound); // returns VBOX_E_OBJECT_NOT_FOUND if not found and sets error } /* this will set (*machine) to NULL if machineObj is null */ pMachineFound.queryInterfaceTo(aMachine.asOutParam()); LogFlowThisFunc(("aName=\"%s\", aMachine=%p, rc=%08X\n", aSettingsFile.c_str(), &aMachine, rc)); LogFlowThisFuncLeave(); return rc; } HRESULT VirtualBox::getMachinesByGroups(const std::vector &aGroups, std::vector > &aMachines) { StringsList llGroups; HRESULT rc = i_convertMachineGroups(aGroups, &llGroups); if (FAILED(rc)) return rc; /* we want to rely on sorted groups during compare, to save time */ llGroups.sort(); /* get copy of all machine references, to avoid holding the list lock */ MachinesOList::MyList allMachines; AutoReadLock al(m->allMachines.getLockHandle() COMMA_LOCKVAL_SRC_POS); allMachines = m->allMachines.getList(); std::vector > saMachines; saMachines.resize(0); for (MachinesOList::MyList::const_iterator it = allMachines.begin(); it != allMachines.end(); ++it) { const ComObjPtr &pMachine = *it; AutoCaller autoMachineCaller(pMachine); if (FAILED(autoMachineCaller.rc())) continue; AutoReadLock mlock(pMachine COMMA_LOCKVAL_SRC_POS); if (pMachine->i_isAccessible()) { const StringsList &thisGroups = pMachine->i_getGroups(); for (StringsList::const_iterator it2 = thisGroups.begin(); it2 != thisGroups.end(); ++it2) { const Utf8Str &group = *it2; bool fAppended = false; for (StringsList::const_iterator it3 = llGroups.begin(); it3 != llGroups.end(); ++it3) { int order = it3->compare(group); if (order == 0) { saMachines.push_back(static_cast(pMachine)); fAppended = true; break; } else if (order > 0) break; else continue; } /* avoid duplicates and save time */ if (fAppended) break; } } } aMachines.resize(saMachines.size()); size_t i = 0; for(i = 0; i < saMachines.size(); ++i) saMachines[i].queryInterfaceTo(aMachines[i].asOutParam()); return S_OK; } HRESULT VirtualBox::getMachineStates(const std::vector > &aMachines, std::vector &aStates) { com::SafeIfaceArray saMachines(aMachines); aStates.resize(aMachines.size()); for (size_t i = 0; i < saMachines.size(); i++) { ComPtr pMachine = saMachines[i]; MachineState_T state = MachineState_Null; if (!pMachine.isNull()) { HRESULT rc = pMachine->COMGETTER(State)(&state); if (rc == E_ACCESSDENIED) rc = S_OK; AssertComRC(rc); } aStates[i] = state; } return S_OK; } HRESULT VirtualBox::createMedium(const com::Utf8Str &aFormat, const com::Utf8Str &aLocation, AccessMode_T aAccessMode, DeviceType_T aDeviceType, ComPtr &aMedium) { NOREF(aAccessMode); /**< @todo r=klaus make use of access mode */ HRESULT rc = S_OK; ComObjPtr medium; medium.createObject(); com::Utf8Str format = aFormat; switch (aDeviceType) { case DeviceType_HardDisk: { /* we don't access non-const data members so no need to lock */ if (format.isEmpty()) i_getDefaultHardDiskFormat(format); rc = medium->init(this, format, aLocation, Guid::Empty /* media registry: none yet */, aDeviceType); } break; case DeviceType_DVD: case DeviceType_Floppy: { if (format.isEmpty()) return setError(E_INVALIDARG, "Format must be Valid Type%s", format.c_str()); // enforce read-only for DVDs even if caller specified ReadWrite if (aDeviceType == DeviceType_DVD) aAccessMode = AccessMode_ReadOnly; rc = medium->init(this, format, aLocation, Guid::Empty /* media registry: none yet */, aDeviceType); } break; default: return setError(E_INVALIDARG, "Device type must be HardDisk, DVD or Floppy %d", aDeviceType); } if (SUCCEEDED(rc)) medium.queryInterfaceTo(aMedium.asOutParam()); return rc; } HRESULT VirtualBox::openMedium(const com::Utf8Str &aLocation, DeviceType_T aDeviceType, AccessMode_T aAccessMode, BOOL aForceNewUuid, ComPtr &aMedium) { HRESULT rc = S_OK; Guid id(aLocation); ComObjPtr pMedium; // have to get write lock as the whole find/update sequence must be done // in one critical section, otherwise there are races which can lead to // multiple Medium objects with the same content AutoWriteLock treeLock(i_getMediaTreeLockHandle() COMMA_LOCKVAL_SRC_POS); // check if the device type is correct, and see if a medium for the // given path has already initialized; if so, return that switch (aDeviceType) { case DeviceType_HardDisk: if (id.isValid() && !id.isZero()) rc = i_findHardDiskById(id, false /* setError */, &pMedium); else rc = i_findHardDiskByLocation(aLocation, false, /* aSetError */ &pMedium); break; case DeviceType_Floppy: case DeviceType_DVD: if (id.isValid() && !id.isZero()) rc = i_findDVDOrFloppyImage(aDeviceType, &id, Utf8Str::Empty, false /* setError */, &pMedium); else rc = i_findDVDOrFloppyImage(aDeviceType, NULL, aLocation, false /* setError */, &pMedium); // enforce read-only for DVDs even if caller specified ReadWrite if (aDeviceType == DeviceType_DVD) aAccessMode = AccessMode_ReadOnly; break; default: return setError(E_INVALIDARG, "Device type must be HardDisk, DVD or Floppy %d", aDeviceType); } if (pMedium.isNull()) { pMedium.createObject(); treeLock.release(); rc = pMedium->init(this, aLocation, (aAccessMode == AccessMode_ReadWrite) ? Medium::OpenReadWrite : Medium::OpenReadOnly, !!aForceNewUuid, aDeviceType); treeLock.acquire(); if (SUCCEEDED(rc)) { rc = i_registerMedium(pMedium, &pMedium, treeLock); treeLock.release(); /* Note that it's important to call uninit() on failure to register * because the differencing hard disk would have been already associated * with the parent and this association needs to be broken. */ if (FAILED(rc)) { pMedium->uninit(); rc = VBOX_E_OBJECT_NOT_FOUND; } } else { if (rc != VBOX_E_INVALID_OBJECT_STATE) rc = VBOX_E_OBJECT_NOT_FOUND; } } if (SUCCEEDED(rc)) pMedium.queryInterfaceTo(aMedium.asOutParam()); return rc; } /** @note Locks this object for reading. */ HRESULT VirtualBox::getGuestOSType(const com::Utf8Str &aId, ComPtr &aType) { aType = NULL; AutoReadLock alock(m->allGuestOSTypes.getLockHandle() COMMA_LOCKVAL_SRC_POS); HRESULT rc = S_OK; for (GuestOSTypesOList::iterator it = m->allGuestOSTypes.begin(); it != m->allGuestOSTypes.end(); ++it) { const Bstr &typeId = (*it)->i_id(); AssertMsg(!typeId.isEmpty(), ("ID must not be NULL")); if (typeId.compare(aId, Bstr::CaseInsensitive) == 0) { (*it).queryInterfaceTo(aType.asOutParam()); break; } } return (aType) ? S_OK : setError(E_INVALIDARG, tr("'%s' is not a valid Guest OS type"), aId.c_str()); return rc; } HRESULT VirtualBox::createSharedFolder(const com::Utf8Str &aName, const com::Utf8Str &aHostPath, BOOL aWritable, BOOL aAutomount) { NOREF(aName); NOREF(aHostPath); NOREF(aWritable); NOREF(aAutomount); return setError(E_NOTIMPL, "Not yet implemented"); } HRESULT VirtualBox::removeSharedFolder(const com::Utf8Str &aName) { NOREF(aName); return setError(E_NOTIMPL, "Not yet implemented"); } /** * @note Locks this object for reading. */ HRESULT VirtualBox::getExtraDataKeys(std::vector &aKeys) { using namespace settings; AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS); aKeys.resize(m->pMainConfigFile->mapExtraDataItems.size()); size_t i = 0; for (StringsMap::const_iterator it = m->pMainConfigFile->mapExtraDataItems.begin(); it != m->pMainConfigFile->mapExtraDataItems.end(); ++it, ++i) aKeys[i] = it->first; return S_OK; } /** * @note Locks this object for reading. */ HRESULT VirtualBox::getExtraData(const com::Utf8Str &aKey, com::Utf8Str &aValue) { settings::StringsMap::const_iterator it = m->pMainConfigFile->mapExtraDataItems.find(aKey); if (it != m->pMainConfigFile->mapExtraDataItems.end()) // found: aValue = it->second; // source is a Utf8Str /* return the result to caller (may be empty) */ return S_OK; } /** * @note Locks this object for writing. */ HRESULT VirtualBox::setExtraData(const com::Utf8Str &aKey, const com::Utf8Str &aValue) { Utf8Str strKey(aKey); Utf8Str strValue(aValue); Utf8Str strOldValue; // empty HRESULT rc = S_OK; // locking note: we only hold the read lock briefly to look up the old value, // then release it and call the onExtraCanChange callbacks. There is a small // chance of a race insofar as the callback might be called twice if two callers // change the same key at the same time, but that's a much better solution // than the deadlock we had here before. The actual changing of the extradata // is then performed under the write lock and race-free. // look up the old value first; if nothing has changed then we need not do anything { AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS); // hold read lock only while looking up settings::StringsMap::const_iterator it = m->pMainConfigFile->mapExtraDataItems.find(strKey); if (it != m->pMainConfigFile->mapExtraDataItems.end()) strOldValue = it->second; } bool fChanged; if ((fChanged = (strOldValue != strValue))) { // ask for permission from all listeners outside the locks; // onExtraDataCanChange() only briefly requests the VirtualBox // lock to copy the list of callbacks to invoke Bstr error; if (!i_onExtraDataCanChange(Guid::Empty, Bstr(aKey).raw(), Bstr(aValue).raw(), error)) { const char *sep = error.isEmpty() ? "" : ": "; CBSTR err = error.raw(); LogWarningFunc(("Someone vetoed! Change refused%s%ls\n", sep, err)); return setError(E_ACCESSDENIED, tr("Could not set extra data because someone refused the requested change of '%s' to '%s'%s%ls"), strKey.c_str(), strValue.c_str(), sep, err); } // data is changing and change not vetoed: then write it out under the lock AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS); if (strValue.isEmpty()) m->pMainConfigFile->mapExtraDataItems.erase(strKey); else m->pMainConfigFile->mapExtraDataItems[strKey] = strValue; // creates a new key if needed /* save settings on success */ rc = i_saveSettings(); if (FAILED(rc)) return rc; } // fire notification outside the lock if (fChanged) i_onExtraDataChange(Guid::Empty, Bstr(aKey).raw(), Bstr(aValue).raw()); return rc; } /** * */ HRESULT VirtualBox::setSettingsSecret(const com::Utf8Str &aPassword) { i_storeSettingsKey(aPassword); i_decryptSettings(); return S_OK; } int VirtualBox::i_decryptMediumSettings(Medium *pMedium) { Bstr bstrCipher; HRESULT hrc = pMedium->GetProperty(Bstr("InitiatorSecretEncrypted").raw(), bstrCipher.asOutParam()); if (SUCCEEDED(hrc)) { Utf8Str strPlaintext; int rc = i_decryptSetting(&strPlaintext, bstrCipher); if (RT_SUCCESS(rc)) pMedium->i_setPropertyDirect("InitiatorSecret", strPlaintext); else return rc; } return VINF_SUCCESS; } /** * Decrypt all encrypted settings. * * So far we only have encrypted iSCSI initiator secrets so we just go through * all hard disk mediums and determine the plain 'InitiatorSecret' from * 'InitiatorSecretEncrypted. The latter is stored as Base64 because medium * properties need to be null-terminated strings. */ int VirtualBox::i_decryptSettings() { bool fFailure = false; AutoReadLock al(m->allHardDisks.getLockHandle() COMMA_LOCKVAL_SRC_POS); for (MediaList::const_iterator mt = m->allHardDisks.begin(); mt != m->allHardDisks.end(); ++mt) { ComObjPtr pMedium = *mt; AutoCaller medCaller(pMedium); if (FAILED(medCaller.rc())) continue; AutoWriteLock mlock(pMedium COMMA_LOCKVAL_SRC_POS); int vrc = i_decryptMediumSettings(pMedium); if (RT_FAILURE(vrc)) fFailure = true; } return fFailure ? VERR_INVALID_PARAMETER : VINF_SUCCESS; } /** * Encode. * * @param aPlaintext plaintext to be encrypted * @param aCiphertext resulting ciphertext (base64-encoded) */ int VirtualBox::i_encryptSetting(const Utf8Str &aPlaintext, Utf8Str *aCiphertext) { uint8_t abCiphertext[32]; char szCipherBase64[128]; size_t cchCipherBase64; int rc = i_encryptSettingBytes((uint8_t*)aPlaintext.c_str(), abCiphertext, aPlaintext.length()+1, sizeof(abCiphertext)); if (RT_SUCCESS(rc)) { rc = RTBase64Encode(abCiphertext, sizeof(abCiphertext), szCipherBase64, sizeof(szCipherBase64), &cchCipherBase64); if (RT_SUCCESS(rc)) *aCiphertext = szCipherBase64; } return rc; } /** * Decode. * * @param aPlaintext resulting plaintext * @param aCiphertext ciphertext (base64-encoded) to decrypt */ int VirtualBox::i_decryptSetting(Utf8Str *aPlaintext, const Utf8Str &aCiphertext) { uint8_t abPlaintext[64]; uint8_t abCiphertext[64]; size_t cbCiphertext; int rc = RTBase64Decode(aCiphertext.c_str(), abCiphertext, sizeof(abCiphertext), &cbCiphertext, NULL); if (RT_SUCCESS(rc)) { rc = i_decryptSettingBytes(abPlaintext, abCiphertext, cbCiphertext); if (RT_SUCCESS(rc)) { for (unsigned i = 0; i < cbCiphertext; i++) { /* sanity check: null-terminated string? */ if (abPlaintext[i] == '\0') { /* sanity check: valid UTF8 string? */ if (RTStrIsValidEncoding((const char*)abPlaintext)) { *aPlaintext = Utf8Str((const char*)abPlaintext); return VINF_SUCCESS; } } } rc = VERR_INVALID_MAGIC; } } return rc; } /** * Encrypt secret bytes. Use the m->SettingsCipherKey as key. * * @param aPlaintext clear text to be encrypted * @param aCiphertext resulting encrypted text * @param aPlaintextSize size of the plaintext * @param aCiphertextSize size of the ciphertext */ int VirtualBox::i_encryptSettingBytes(const uint8_t *aPlaintext, uint8_t *aCiphertext, size_t aPlaintextSize, size_t aCiphertextSize) const { unsigned i, j; uint8_t aBytes[64]; if (!m->fSettingsCipherKeySet) return VERR_INVALID_STATE; if (aCiphertextSize > sizeof(aBytes)) return VERR_BUFFER_OVERFLOW; if (aCiphertextSize < 32) return VERR_INVALID_PARAMETER; AssertCompile(sizeof(m->SettingsCipherKey) >= 32); /* store the first 8 bytes of the cipherkey for verification */ for (i = 0, j = 0; i < 8; i++, j++) aCiphertext[i] = m->SettingsCipherKey[j]; for (unsigned k = 0; k < aPlaintextSize && i < aCiphertextSize; i++, k++) { aCiphertext[i] = (aPlaintext[k] ^ m->SettingsCipherKey[j]); if (++j >= sizeof(m->SettingsCipherKey)) j = 0; } /* fill with random data to have a minimal length (salt) */ if (i < aCiphertextSize) { RTRandBytes(aBytes, aCiphertextSize - i); for (int k = 0; i < aCiphertextSize; i++, k++) { aCiphertext[i] = aBytes[k] ^ m->SettingsCipherKey[j]; if (++j >= sizeof(m->SettingsCipherKey)) j = 0; } } return VINF_SUCCESS; } /** * Decrypt secret bytes. Use the m->SettingsCipherKey as key. * * @param aPlaintext resulting plaintext * @param aCiphertext ciphertext to be decrypted * @param aCiphertextSize size of the ciphertext == size of the plaintext */ int VirtualBox::i_decryptSettingBytes(uint8_t *aPlaintext, const uint8_t *aCiphertext, size_t aCiphertextSize) const { unsigned i, j; if (!m->fSettingsCipherKeySet) return VERR_INVALID_STATE; if (aCiphertextSize < 32) return VERR_INVALID_PARAMETER; /* key verification */ for (i = 0, j = 0; i < 8; i++, j++) if (aCiphertext[i] != m->SettingsCipherKey[j]) return VERR_INVALID_MAGIC; /* poison */ memset(aPlaintext, 0xff, aCiphertextSize); for (int k = 0; i < aCiphertextSize; i++, k++) { aPlaintext[k] = aCiphertext[i] ^ m->SettingsCipherKey[j]; if (++j >= sizeof(m->SettingsCipherKey)) j = 0; } return VINF_SUCCESS; } /** * Store a settings key. * * @param aKey the key to store */ void VirtualBox::i_storeSettingsKey(const Utf8Str &aKey) { RTSha512(aKey.c_str(), aKey.length(), m->SettingsCipherKey); m->fSettingsCipherKeySet = true; } // public methods only for internal purposes ///////////////////////////////////////////////////////////////////////////// #ifdef DEBUG void VirtualBox::i_dumpAllBackRefs() { { AutoReadLock al(m->allHardDisks.getLockHandle() COMMA_LOCKVAL_SRC_POS); for (MediaList::const_iterator mt = m->allHardDisks.begin(); mt != m->allHardDisks.end(); ++mt) { ComObjPtr pMedium = *mt; pMedium->i_dumpBackRefs(); } } { AutoReadLock al(m->allDVDImages.getLockHandle() COMMA_LOCKVAL_SRC_POS); for (MediaList::const_iterator mt = m->allDVDImages.begin(); mt != m->allDVDImages.end(); ++mt) { ComObjPtr pMedium = *mt; pMedium->i_dumpBackRefs(); } } } #endif /** * Posts an event to the event queue that is processed asynchronously * on a dedicated thread. * * Posting events to the dedicated event queue is useful to perform secondary * actions outside any object locks -- for example, to iterate over a list * of callbacks and inform them about some change caused by some object's * method call. * * @param event event to post; must have been allocated using |new|, will * be deleted automatically by the event thread after processing * * @note Doesn't lock any object. */ HRESULT VirtualBox::i_postEvent(Event *event) { AssertReturn(event, E_FAIL); HRESULT rc; AutoCaller autoCaller(this); if (SUCCEEDED((rc = autoCaller.rc()))) { if (getObjectState().getState() != ObjectState::Ready) LogWarningFunc(("VirtualBox has been uninitialized (state=%d), the event is discarded!\n", getObjectState().getState())); // return S_OK else if ( (m->pAsyncEventQ) && (m->pAsyncEventQ->postEvent(event)) ) return S_OK; else rc = E_FAIL; } // in any event of failure, we must clean up here, or we'll leak; // the caller has allocated the object using new() delete event; return rc; } /** * Adds a progress to the global collection of pending operations. * Usually gets called upon progress object initialization. * * @param aProgress Operation to add to the collection. * * @note Doesn't lock objects. */ HRESULT VirtualBox::i_addProgress(IProgress *aProgress) { CheckComArgNotNull(aProgress); AutoCaller autoCaller(this); if (FAILED(autoCaller.rc())) return autoCaller.rc(); Bstr id; HRESULT rc = aProgress->COMGETTER(Id)(id.asOutParam()); AssertComRCReturnRC(rc); /* protect mProgressOperations */ AutoWriteLock safeLock(m->mtxProgressOperations COMMA_LOCKVAL_SRC_POS); m->mapProgressOperations.insert(ProgressMap::value_type(Guid(id), aProgress)); return S_OK; } /** * Removes the progress from the global collection of pending operations. * Usually gets called upon progress completion. * * @param aId UUID of the progress operation to remove * * @note Doesn't lock objects. */ HRESULT VirtualBox::i_removeProgress(IN_GUID aId) { AutoCaller autoCaller(this); if (FAILED(autoCaller.rc())) return autoCaller.rc(); ComPtr progress; /* protect mProgressOperations */ AutoWriteLock safeLock(m->mtxProgressOperations COMMA_LOCKVAL_SRC_POS); size_t cnt = m->mapProgressOperations.erase(aId); Assert(cnt == 1); NOREF(cnt); return S_OK; } #ifdef RT_OS_WINDOWS struct StartSVCHelperClientData { ComObjPtr that; ComObjPtr progress; bool privileged; VirtualBox::SVCHelperClientFunc func; void *user; }; /** * Helper method that starts a worker thread that: * - creates a pipe communication channel using SVCHlpClient; * - starts an SVC Helper process that will inherit this channel; * - executes the supplied function by passing it the created SVCHlpClient * and opened instance to communicate to the Helper process and the given * Progress object. * * The user function is supposed to communicate to the helper process * using the \a aClient argument to do the requested job and optionally expose * the progress through the \a aProgress object. The user function should never * call notifyComplete() on it: this will be done automatically using the * result code returned by the function. * * Before the user function is started, the communication channel passed to * the \a aClient argument is fully set up, the function should start using * its write() and read() methods directly. * * The \a aVrc parameter of the user function may be used to return an error * code if it is related to communication errors (for example, returned by * the SVCHlpClient members when they fail). In this case, the correct error * message using this value will be reported to the caller. Note that the * value of \a aVrc is inspected only if the user function itself returns * success. * * If a failure happens anywhere before the user function would be normally * called, it will be called anyway in special "cleanup only" mode indicated * by \a aClient, \a aProgress and \aVrc arguments set to NULL. In this mode, * all the function is supposed to do is to cleanup its aUser argument if * necessary (it's assumed that the ownership of this argument is passed to * the user function once #startSVCHelperClient() returns a success, thus * making it responsible for the cleanup). * * After the user function returns, the thread will send the SVCHlpMsg::Null * message to indicate a process termination. * * @param aPrivileged |true| to start the SVC Helper process as a privileged * user that can perform administrative tasks * @param aFunc user function to run * @param aUser argument to the user function * @param aProgress progress object that will track operation completion * * @note aPrivileged is currently ignored (due to some unsolved problems in * Vista) and the process will be started as a normal (unprivileged) * process. * * @note Doesn't lock anything. */ HRESULT VirtualBox::i_startSVCHelperClient(bool aPrivileged, SVCHelperClientFunc aFunc, void *aUser, Progress *aProgress) { AssertReturn(aFunc, E_POINTER); AssertReturn(aProgress, E_POINTER); AutoCaller autoCaller(this); if (FAILED(autoCaller.rc())) return autoCaller.rc(); /* create the SVCHelperClientThread() argument */ std::auto_ptr d(new StartSVCHelperClientData()); AssertReturn(d.get(), E_OUTOFMEMORY); d->that = this; d->progress = aProgress; d->privileged = aPrivileged; d->func = aFunc; d->user = aUser; RTTHREAD tid = NIL_RTTHREAD; int vrc = RTThreadCreate(&tid, SVCHelperClientThread, static_cast (d.get()), 0, RTTHREADTYPE_MAIN_WORKER, RTTHREADFLAGS_WAITABLE, "SVCHelper"); if (RT_FAILURE(vrc)) return setError(E_FAIL, "Could not create SVCHelper thread (%Rrc)", vrc); /* d is now owned by SVCHelperClientThread(), so release it */ d.release(); return S_OK; } /** * Worker thread for startSVCHelperClient(). */ /* static */ DECLCALLBACK(int) VirtualBox::SVCHelperClientThread(RTTHREAD aThread, void *aUser) { LogFlowFuncEnter(); std::auto_ptr d(static_cast(aUser)); HRESULT rc = S_OK; bool userFuncCalled = false; do { AssertBreakStmt(d.get(), rc = E_POINTER); AssertReturn(!d->progress.isNull(), E_POINTER); /* protect VirtualBox from uninitialization */ AutoCaller autoCaller(d->that); if (!autoCaller.isOk()) { /* it's too late */ rc = autoCaller.rc(); break; } int vrc = VINF_SUCCESS; Guid id; id.create(); SVCHlpClient client; vrc = client.create(Utf8StrFmt("VirtualBox\\SVCHelper\\{%RTuuid}", id.raw()).c_str()); if (RT_FAILURE(vrc)) { rc = d->that->setError(E_FAIL, tr("Could not create the communication channel (%Rrc)"), vrc); break; } /* get the path to the executable */ char exePathBuf[RTPATH_MAX]; char *exePath = RTProcGetExecutablePath(exePathBuf, RTPATH_MAX); if (!exePath) { rc = d->that->setError(E_FAIL, tr("Cannot get executable name")); break; } Utf8Str argsStr = Utf8StrFmt("/Helper %s", client.name().c_str()); LogFlowFunc(("Starting '\"%s\" %s'...\n", exePath, argsStr.c_str())); RTPROCESS pid = NIL_RTPROCESS; if (d->privileged) { /* Attempt to start a privileged process using the Run As dialog */ Bstr file = exePath; Bstr parameters = argsStr; SHELLEXECUTEINFO shExecInfo; shExecInfo.cbSize = sizeof(SHELLEXECUTEINFO); shExecInfo.fMask = NULL; shExecInfo.hwnd = NULL; shExecInfo.lpVerb = L"runas"; shExecInfo.lpFile = file.raw(); shExecInfo.lpParameters = parameters.raw(); shExecInfo.lpDirectory = NULL; shExecInfo.nShow = SW_NORMAL; shExecInfo.hInstApp = NULL; if (!ShellExecuteEx(&shExecInfo)) { int vrc2 = RTErrConvertFromWin32(GetLastError()); /* hide excessive details in case of a frequent error * (pressing the Cancel button to close the Run As dialog) */ if (vrc2 == VERR_CANCELLED) rc = d->that->setError(E_FAIL, tr("Operation canceled by the user")); else rc = d->that->setError(E_FAIL, tr("Could not launch a privileged process '%s' (%Rrc)"), exePath, vrc2); break; } } else { const char *args[] = { exePath, "/Helper", client.name().c_str(), 0 }; vrc = RTProcCreate(exePath, args, RTENV_DEFAULT, 0, &pid); if (RT_FAILURE(vrc)) { rc = d->that->setError(E_FAIL, tr("Could not launch a process '%s' (%Rrc)"), exePath, vrc); break; } } /* wait for the client to connect */ vrc = client.connect(); if (RT_SUCCESS(vrc)) { /* start the user supplied function */ rc = d->func(&client, d->progress, d->user, &vrc); userFuncCalled = true; } /* send the termination signal to the process anyway */ { int vrc2 = client.write(SVCHlpMsg::Null); if (RT_SUCCESS(vrc)) vrc = vrc2; } if (SUCCEEDED(rc) && RT_FAILURE(vrc)) { rc = d->that->setError(E_FAIL, tr("Could not operate the communication channel (%Rrc)"), vrc); break; } } while (0); if (FAILED(rc) && !userFuncCalled) { /* call the user function in the "cleanup only" mode * to let it free resources passed to in aUser */ d->func(NULL, NULL, d->user, NULL); } d->progress->i_notifyComplete(rc); LogFlowFuncLeave(); return 0; } #endif /* RT_OS_WINDOWS */ /** * Sends a signal to the client watcher to rescan the set of machines * that have open sessions. * * @note Doesn't lock anything. */ void VirtualBox::i_updateClientWatcher() { AutoCaller autoCaller(this); AssertComRCReturnVoid(autoCaller.rc()); AssertPtrReturnVoid(m->pClientWatcher); m->pClientWatcher->update(); } /** * Adds the given child process ID to the list of processes to be reaped. * This call should be followed by #updateClientWatcher() to take the effect. * * @note Doesn't lock anything. */ void VirtualBox::i_addProcessToReap(RTPROCESS pid) { AutoCaller autoCaller(this); AssertComRCReturnVoid(autoCaller.rc()); AssertPtrReturnVoid(m->pClientWatcher); m->pClientWatcher->addProcess(pid); } /** Event for onMachineStateChange(), onMachineDataChange(), onMachineRegistered() */ struct MachineEvent : public VirtualBox::CallbackEvent { MachineEvent(VirtualBox *aVB, VBoxEventType_T aWhat, const Guid &aId, BOOL aBool) : CallbackEvent(aVB, aWhat), id(aId.toUtf16()) , mBool(aBool) { } MachineEvent(VirtualBox *aVB, VBoxEventType_T aWhat, const Guid &aId, MachineState_T aState) : CallbackEvent(aVB, aWhat), id(aId.toUtf16()) , mState(aState) {} virtual HRESULT prepareEventDesc(IEventSource* aSource, VBoxEventDesc& aEvDesc) { switch (mWhat) { case VBoxEventType_OnMachineDataChanged: aEvDesc.init(aSource, mWhat, id.raw(), mBool); break; case VBoxEventType_OnMachineStateChanged: aEvDesc.init(aSource, mWhat, id.raw(), mState); break; case VBoxEventType_OnMachineRegistered: aEvDesc.init(aSource, mWhat, id.raw(), mBool); break; default: AssertFailedReturn(S_OK); } return S_OK; } Bstr id; MachineState_T mState; BOOL mBool; }; /** * VD plugin load */ int VirtualBox::i_loadVDPlugin(const char *pszPluginLibrary) { return m->pSystemProperties->i_loadVDPlugin(pszPluginLibrary); } /** * VD plugin unload */ int VirtualBox::i_unloadVDPlugin(const char *pszPluginLibrary) { return m->pSystemProperties->i_unloadVDPlugin(pszPluginLibrary); } /** * @note Doesn't lock any object. */ void VirtualBox::i_onMachineStateChange(const Guid &aId, MachineState_T aState) { i_postEvent(new MachineEvent(this, VBoxEventType_OnMachineStateChanged, aId, aState)); } /** * @note Doesn't lock any object. */ void VirtualBox::i_onMachineDataChange(const Guid &aId, BOOL aTemporary) { i_postEvent(new MachineEvent(this, VBoxEventType_OnMachineDataChanged, aId, aTemporary)); } /** * @note Locks this object for reading. */ BOOL VirtualBox::i_onExtraDataCanChange(const Guid &aId, IN_BSTR aKey, IN_BSTR aValue, Bstr &aError) { LogFlowThisFunc(("machine={%s} aKey={%ls} aValue={%ls}\n", aId.toString().c_str(), aKey, aValue)); AutoCaller autoCaller(this); AssertComRCReturn(autoCaller.rc(), autoCaller.rc()); BOOL allowChange = TRUE; Bstr id = aId.toUtf16(); VBoxEventDesc evDesc; evDesc.init(m->pEventSource, VBoxEventType_OnExtraDataCanChange, id.raw(), aKey, aValue); BOOL fDelivered = evDesc.fire(3000); /* Wait up to 3 secs for delivery */ //Assert(fDelivered); if (fDelivered) { ComPtr aEvent; evDesc.getEvent(aEvent.asOutParam()); ComPtr aCanChangeEvent = aEvent; Assert(aCanChangeEvent); BOOL fVetoed = FALSE; aCanChangeEvent->IsVetoed(&fVetoed); allowChange = !fVetoed; if (!allowChange) { SafeArray aVetos; aCanChangeEvent->GetVetos(ComSafeArrayAsOutParam(aVetos)); if (aVetos.size() > 0) aError = aVetos[0]; } } else allowChange = TRUE; LogFlowThisFunc(("allowChange=%RTbool\n", allowChange)); return allowChange; } /** Event for onExtraDataChange() */ struct ExtraDataEvent : public VirtualBox::CallbackEvent { ExtraDataEvent(VirtualBox *aVB, const Guid &aMachineId, IN_BSTR aKey, IN_BSTR aVal) : CallbackEvent(aVB, VBoxEventType_OnExtraDataChanged) , machineId(aMachineId.toUtf16()), key(aKey), val(aVal) {} virtual HRESULT prepareEventDesc(IEventSource* aSource, VBoxEventDesc& aEvDesc) { return aEvDesc.init(aSource, VBoxEventType_OnExtraDataChanged, machineId.raw(), key.raw(), val.raw()); } Bstr machineId, key, val; }; /** * @note Doesn't lock any object. */ void VirtualBox::i_onExtraDataChange(const Guid &aId, IN_BSTR aKey, IN_BSTR aValue) { i_postEvent(new ExtraDataEvent(this, aId, aKey, aValue)); } /** * @note Doesn't lock any object. */ void VirtualBox::i_onMachineRegistered(const Guid &aId, BOOL aRegistered) { i_postEvent(new MachineEvent(this, VBoxEventType_OnMachineRegistered, aId, aRegistered)); } /** Event for onSessionStateChange() */ struct SessionEvent : public VirtualBox::CallbackEvent { SessionEvent(VirtualBox *aVB, const Guid &aMachineId, SessionState_T aState) : CallbackEvent(aVB, VBoxEventType_OnSessionStateChanged) , machineId(aMachineId.toUtf16()), sessionState(aState) {} virtual HRESULT prepareEventDesc(IEventSource* aSource, VBoxEventDesc& aEvDesc) { return aEvDesc.init(aSource, VBoxEventType_OnSessionStateChanged, machineId.raw(), sessionState); } Bstr machineId; SessionState_T sessionState; }; /** * @note Doesn't lock any object. */ void VirtualBox::i_onSessionStateChange(const Guid &aId, SessionState_T aState) { i_postEvent(new SessionEvent(this, aId, aState)); } /** Event for onSnapshotTaken(), onSnapshotDeleted() and onSnapshotChange() */ struct SnapshotEvent : public VirtualBox::CallbackEvent { SnapshotEvent(VirtualBox *aVB, const Guid &aMachineId, const Guid &aSnapshotId, VBoxEventType_T aWhat) : CallbackEvent(aVB, aWhat) , machineId(aMachineId), snapshotId(aSnapshotId) {} virtual HRESULT prepareEventDesc(IEventSource* aSource, VBoxEventDesc& aEvDesc) { return aEvDesc.init(aSource, mWhat, machineId.toUtf16().raw(), snapshotId.toUtf16().raw()); } Guid machineId; Guid snapshotId; }; /** * @note Doesn't lock any object. */ void VirtualBox::i_onSnapshotTaken(const Guid &aMachineId, const Guid &aSnapshotId) { i_postEvent(new SnapshotEvent(this, aMachineId, aSnapshotId, VBoxEventType_OnSnapshotTaken)); } /** * @note Doesn't lock any object. */ void VirtualBox::i_onSnapshotDeleted(const Guid &aMachineId, const Guid &aSnapshotId) { i_postEvent(new SnapshotEvent(this, aMachineId, aSnapshotId, VBoxEventType_OnSnapshotDeleted)); } /** * @note Doesn't lock any object. */ void VirtualBox::i_onSnapshotChange(const Guid &aMachineId, const Guid &aSnapshotId) { i_postEvent(new SnapshotEvent(this, aMachineId, aSnapshotId, VBoxEventType_OnSnapshotChanged)); } /** Event for onGuestPropertyChange() */ struct GuestPropertyEvent : public VirtualBox::CallbackEvent { GuestPropertyEvent(VirtualBox *aVBox, const Guid &aMachineId, IN_BSTR aName, IN_BSTR aValue, IN_BSTR aFlags) : CallbackEvent(aVBox, VBoxEventType_OnGuestPropertyChanged), machineId(aMachineId), name(aName), value(aValue), flags(aFlags) {} virtual HRESULT prepareEventDesc(IEventSource* aSource, VBoxEventDesc& aEvDesc) { return aEvDesc.init(aSource, VBoxEventType_OnGuestPropertyChanged, machineId.toUtf16().raw(), name.raw(), value.raw(), flags.raw()); } Guid machineId; Bstr name, value, flags; }; /** * @note Doesn't lock any object. */ void VirtualBox::i_onGuestPropertyChange(const Guid &aMachineId, IN_BSTR aName, IN_BSTR aValue, IN_BSTR aFlags) { i_postEvent(new GuestPropertyEvent(this, aMachineId, aName, aValue, aFlags)); } /** * @note Doesn't lock any object. */ void VirtualBox::i_onNatRedirectChange(const Guid &aMachineId, ULONG ulSlot, bool fRemove, IN_BSTR aName, NATProtocol_T aProto, IN_BSTR aHostIp, uint16_t aHostPort, IN_BSTR aGuestIp, uint16_t aGuestPort) { fireNATRedirectEvent(m->pEventSource, aMachineId.toUtf16().raw(), ulSlot, fRemove, aName, aProto, aHostIp, aHostPort, aGuestIp, aGuestPort); } void VirtualBox::i_onNATNetworkChange(IN_BSTR aName) { fireNATNetworkChangedEvent(m->pEventSource, aName); } void VirtualBox::i_onNATNetworkStartStop(IN_BSTR aName, BOOL fStart) { fireNATNetworkStartStopEvent(m->pEventSource, aName, fStart); } void VirtualBox::i_onNATNetworkSetting(IN_BSTR aNetworkName, BOOL aEnabled, IN_BSTR aNetwork, IN_BSTR aGateway, BOOL aAdvertiseDefaultIpv6RouteEnabled, BOOL fNeedDhcpServer) { fireNATNetworkSettingEvent(m->pEventSource, aNetworkName, aEnabled, aNetwork, aGateway, aAdvertiseDefaultIpv6RouteEnabled, fNeedDhcpServer); } void VirtualBox::i_onNATNetworkPortForward(IN_BSTR aNetworkName, BOOL create, BOOL fIpv6, IN_BSTR aRuleName, NATProtocol_T proto, IN_BSTR aHostIp, LONG aHostPort, IN_BSTR aGuestIp, LONG aGuestPort) { fireNATNetworkPortForwardEvent(m->pEventSource, aNetworkName, create, fIpv6, aRuleName, proto, aHostIp, aHostPort, aGuestIp, aGuestPort); } void VirtualBox::i_onHostNameResolutionConfigurationChange() { if (m->pEventSource) fireHostNameResolutionConfigurationChangeEvent(m->pEventSource); } int VirtualBox::i_natNetworkRefInc(IN_BSTR aNetworkName) { AutoWriteLock safeLock(*spMtxNatNetworkNameToRefCountLock COMMA_LOCKVAL_SRC_POS); Bstr name(aNetworkName); if (!sNatNetworkNameToRefCount[name]) { ComPtr nat; HRESULT rc = FindNATNetworkByName(aNetworkName, nat.asOutParam()); if (FAILED(rc)) return -1; rc = nat->Start(Bstr("whatever").raw()); if (SUCCEEDED(rc)) LogRel(("Started NAT network '%ls'\n", aNetworkName)); else LogRel(("Error %Rhrc starting NAT network '%ls'\n", rc, aNetworkName)); AssertComRCReturn(rc, -1); } sNatNetworkNameToRefCount[name]++; return sNatNetworkNameToRefCount[name]; } int VirtualBox::i_natNetworkRefDec(IN_BSTR aNetworkName) { AutoWriteLock safeLock(*spMtxNatNetworkNameToRefCountLock COMMA_LOCKVAL_SRC_POS); Bstr name(aNetworkName); if (!sNatNetworkNameToRefCount[name]) return 0; sNatNetworkNameToRefCount[name]--; if (!sNatNetworkNameToRefCount[name]) { ComPtr nat; HRESULT rc = FindNATNetworkByName(aNetworkName, nat.asOutParam()); if (FAILED(rc)) return -1; rc = nat->Stop(); if (SUCCEEDED(rc)) LogRel(("Stopped NAT network '%ls'\n", aNetworkName)); else LogRel(("Error %Rhrc stopping NAT network '%ls'\n", rc, aNetworkName)); AssertComRCReturn(rc, -1); } return sNatNetworkNameToRefCount[name]; } /** * @note Locks the list of other objects for reading. */ ComObjPtr VirtualBox::i_getUnknownOSType() { ComObjPtr type; /* unknown type must always be the first */ ComAssertRet(m->allGuestOSTypes.size() > 0, type); return m->allGuestOSTypes.front(); } /** * Returns the list of opened machines (machines having direct sessions opened * by client processes) and optionally the list of direct session controls. * * @param aMachines Where to put opened machines (will be empty if none). * @param aControls Where to put direct session controls (optional). * * @note The returned lists contain smart pointers. So, clear it as soon as * it becomes no more necessary to release instances. * * @note It can be possible that a session machine from the list has been * already uninitialized, so do a usual AutoCaller/AutoReadLock sequence * when accessing unprotected data directly. * * @note Locks objects for reading. */ void VirtualBox::i_getOpenedMachines(SessionMachinesList &aMachines, InternalControlList *aControls /*= NULL*/) { AutoCaller autoCaller(this); AssertComRCReturnVoid(autoCaller.rc()); aMachines.clear(); if (aControls) aControls->clear(); AutoReadLock alock(m->allMachines.getLockHandle() COMMA_LOCKVAL_SRC_POS); for (MachinesOList::iterator it = m->allMachines.begin(); it != m->allMachines.end(); ++it) { ComObjPtr sm; ComPtr ctl; if ((*it)->i_isSessionOpen(sm, &ctl)) { aMachines.push_back(sm); if (aControls) aControls->push_back(ctl); } } } /** * Gets a reference to the machine list. This is the real thing, not a copy, * so bad things will happen if the caller doesn't hold the necessary lock. * * @returns reference to machine list * * @note Caller must hold the VirtualBox object lock at least for reading. */ VirtualBox::MachinesOList &VirtualBox::i_getMachinesList(void) { return m->allMachines; } /** * Searches for a machine object with the given ID in the collection * of registered machines. * * @param aId Machine UUID to look for. * @param aPermitInaccessible If true, inaccessible machines will be found; * if false, this will fail if the given machine is inaccessible. * @param aSetError If true, set errorinfo if the machine is not found. * @param aMachine Returned machine, if found. * @return */ HRESULT VirtualBox::i_findMachine(const Guid &aId, bool fPermitInaccessible, bool aSetError, ComObjPtr *aMachine /* = NULL */) { HRESULT rc = VBOX_E_OBJECT_NOT_FOUND; AutoCaller autoCaller(this); AssertComRCReturn(autoCaller.rc(), autoCaller.rc()); { AutoReadLock al(m->allMachines.getLockHandle() COMMA_LOCKVAL_SRC_POS); for (MachinesOList::iterator it = m->allMachines.begin(); it != m->allMachines.end(); ++it) { ComObjPtr pMachine = *it; if (!fPermitInaccessible) { // skip inaccessible machines AutoCaller machCaller(pMachine); if (FAILED(machCaller.rc())) continue; } if (pMachine->i_getId() == aId) { rc = S_OK; if (aMachine) *aMachine = pMachine; break; } } } if (aSetError && FAILED(rc)) rc = setError(rc, tr("Could not find a registered machine with UUID {%RTuuid}"), aId.raw()); return rc; } /** * Searches for a machine object with the given name or location in the * collection of registered machines. * * @param aName Machine name or location to look for. * @param aSetError If true, set errorinfo if the machine is not found. * @param aMachine Returned machine, if found. * @return */ HRESULT VirtualBox::i_findMachineByName(const Utf8Str &aName, bool aSetError, ComObjPtr *aMachine /* = NULL */) { HRESULT rc = VBOX_E_OBJECT_NOT_FOUND; AutoReadLock al(m->allMachines.getLockHandle() COMMA_LOCKVAL_SRC_POS); for (MachinesOList::iterator it = m->allMachines.begin(); it != m->allMachines.end(); ++it) { ComObjPtr &pMachine = *it; AutoCaller machCaller(pMachine); if (machCaller.rc()) continue; // we can't ask inaccessible machines for their names AutoReadLock machLock(pMachine COMMA_LOCKVAL_SRC_POS); if (pMachine->i_getName() == aName) { rc = S_OK; if (aMachine) *aMachine = pMachine; break; } if (!RTPathCompare(pMachine->i_getSettingsFileFull().c_str(), aName.c_str())) { rc = S_OK; if (aMachine) *aMachine = pMachine; break; } } if (aSetError && FAILED(rc)) rc = setError(rc, tr("Could not find a registered machine named '%s'"), aName.c_str()); return rc; } static HRESULT i_validateMachineGroupHelper(const Utf8Str &aGroup, bool fPrimary, VirtualBox *pVirtualBox) { /* empty strings are invalid */ if (aGroup.isEmpty()) return E_INVALIDARG; /* the toplevel group is valid */ if (aGroup == "/") return S_OK; /* any other strings of length 1 are invalid */ if (aGroup.length() == 1) return E_INVALIDARG; /* must start with a slash */ if (aGroup.c_str()[0] != '/') return E_INVALIDARG; /* must not end with a slash */ if (aGroup.c_str()[aGroup.length() - 1] == '/') return E_INVALIDARG; /* check the group components */ const char *pStr = aGroup.c_str() + 1; /* first char is /, skip it */ while (pStr) { char *pSlash = RTStrStr(pStr, "/"); if (pSlash) { /* no empty components (or // sequences in other words) */ if (pSlash == pStr) return E_INVALIDARG; /* check if the machine name rules are violated, because that means * the group components are too close to the limits. */ Utf8Str tmp((const char *)pStr, (size_t)(pSlash - pStr)); Utf8Str tmp2(tmp); sanitiseMachineFilename(tmp); if (tmp != tmp2) return E_INVALIDARG; if (fPrimary) { HRESULT rc = pVirtualBox->i_findMachineByName(tmp, false /* aSetError */); if (SUCCEEDED(rc)) return VBOX_E_VM_ERROR; } pStr = pSlash + 1; } else { /* check if the machine name rules are violated, because that means * the group components is too close to the limits. */ Utf8Str tmp(pStr); Utf8Str tmp2(tmp); sanitiseMachineFilename(tmp); if (tmp != tmp2) return E_INVALIDARG; pStr = NULL; } } return S_OK; } /** * Validates a machine group. * * @param aMachineGroup Machine group. * @param fPrimary Set if this is the primary group. * * @return S_OK or E_INVALIDARG */ HRESULT VirtualBox::i_validateMachineGroup(const Utf8Str &aGroup, bool fPrimary) { HRESULT rc = i_validateMachineGroupHelper(aGroup, fPrimary, this); if (FAILED(rc)) { if (rc == VBOX_E_VM_ERROR) rc = setError(E_INVALIDARG, tr("Machine group '%s' conflicts with a virtual machine name"), aGroup.c_str()); else rc = setError(rc, tr("Invalid machine group '%s'"), aGroup.c_str()); } return rc; } /** * Takes a list of machine groups, and sanitizes/validates it. * * @param aMachineGroups Array with the machine groups. * @param pllMachineGroups Pointer to list of strings for the result. * * @return S_OK or E_INVALIDARG */ HRESULT VirtualBox::i_convertMachineGroups(const std::vector aMachineGroups, StringsList *pllMachineGroups) { pllMachineGroups->clear(); if (aMachineGroups.size()) { for (size_t i = 0; i < aMachineGroups.size(); i++) { Utf8Str group(aMachineGroups[i]); if (group.length() == 0) group = "/"; HRESULT rc = i_validateMachineGroup(group, i == 0); if (FAILED(rc)) return rc; /* no duplicates please */ if ( find(pllMachineGroups->begin(), pllMachineGroups->end(), group) == pllMachineGroups->end()) pllMachineGroups->push_back(group); } if (pllMachineGroups->size() == 0) pllMachineGroups->push_back("/"); } else pllMachineGroups->push_back("/"); return S_OK; } /** * Searches for a Medium object with the given ID in the list of registered * hard disks. * * @param aId ID of the hard disk. Must not be empty. * @param aSetError If @c true , the appropriate error info is set in case * when the hard disk is not found. * @param aHardDisk Where to store the found hard disk object (can be NULL). * * @return S_OK, E_INVALIDARG or VBOX_E_OBJECT_NOT_FOUND when not found. * * @note Locks the media tree for reading. */ HRESULT VirtualBox::i_findHardDiskById(const Guid &id, bool aSetError, ComObjPtr *aHardDisk /*= NULL*/) { AssertReturn(!id.isZero(), E_INVALIDARG); // we use the hard disks map, but it is protected by the // hard disk _list_ lock handle AutoReadLock alock(m->allHardDisks.getLockHandle() COMMA_LOCKVAL_SRC_POS); HardDiskMap::const_iterator it = m->mapHardDisks.find(id); if (it != m->mapHardDisks.end()) { if (aHardDisk) *aHardDisk = (*it).second; return S_OK; } if (aSetError) return setError(VBOX_E_OBJECT_NOT_FOUND, tr("Could not find an open hard disk with UUID {%RTuuid}"), id.raw()); return VBOX_E_OBJECT_NOT_FOUND; } /** * Searches for a Medium object with the given ID or location in the list of * registered hard disks. If both ID and location are specified, the first * object that matches either of them (not necessarily both) is returned. * * @param aLocation Full location specification. Must not be empty. * @param aSetError If @c true , the appropriate error info is set in case * when the hard disk is not found. * @param aHardDisk Where to store the found hard disk object (can be NULL). * * @return S_OK, E_INVALIDARG or VBOX_E_OBJECT_NOT_FOUND when not found. * * @note Locks the media tree for reading. */ HRESULT VirtualBox::i_findHardDiskByLocation(const Utf8Str &strLocation, bool aSetError, ComObjPtr *aHardDisk /*= NULL*/) { AssertReturn(!strLocation.isEmpty(), E_INVALIDARG); // we use the hard disks map, but it is protected by the // hard disk _list_ lock handle AutoReadLock alock(m->allHardDisks.getLockHandle() COMMA_LOCKVAL_SRC_POS); for (HardDiskMap::const_iterator it = m->mapHardDisks.begin(); it != m->mapHardDisks.end(); ++it) { const ComObjPtr &pHD = (*it).second; AutoCaller autoCaller(pHD); if (FAILED(autoCaller.rc())) return autoCaller.rc(); AutoWriteLock mlock(pHD COMMA_LOCKVAL_SRC_POS); Utf8Str strLocationFull = pHD->i_getLocationFull(); if (0 == RTPathCompare(strLocationFull.c_str(), strLocation.c_str())) { if (aHardDisk) *aHardDisk = pHD; return S_OK; } } if (aSetError) return setError(VBOX_E_OBJECT_NOT_FOUND, tr("Could not find an open hard disk with location '%s'"), strLocation.c_str()); return VBOX_E_OBJECT_NOT_FOUND; } /** * Searches for a Medium object with the given ID or location in the list of * registered DVD or floppy images, depending on the @a mediumType argument. * If both ID and file path are specified, the first object that matches either * of them (not necessarily both) is returned. * * @param mediumType Must be either DeviceType_DVD or DeviceType_Floppy. * @param aId ID of the image file (unused when NULL). * @param aLocation Full path to the image file (unused when NULL). * @param aSetError If @c true, the appropriate error info is set in case when * the image is not found. * @param aImage Where to store the found image object (can be NULL). * * @return S_OK when found or E_INVALIDARG or VBOX_E_OBJECT_NOT_FOUND when not found. * * @note Locks the media tree for reading. */ HRESULT VirtualBox::i_findDVDOrFloppyImage(DeviceType_T mediumType, const Guid *aId, const Utf8Str &aLocation, bool aSetError, ComObjPtr *aImage /* = NULL */) { AssertReturn(aId || !aLocation.isEmpty(), E_INVALIDARG); Utf8Str location; if (!aLocation.isEmpty()) { int vrc = i_calculateFullPath(aLocation, location); if (RT_FAILURE(vrc)) return setError(VBOX_E_FILE_ERROR, tr("Invalid image file location '%s' (%Rrc)"), aLocation.c_str(), vrc); } MediaOList *pMediaList; switch (mediumType) { case DeviceType_DVD: pMediaList = &m->allDVDImages; break; case DeviceType_Floppy: pMediaList = &m->allFloppyImages; break; default: return E_INVALIDARG; } AutoReadLock alock(pMediaList->getLockHandle() COMMA_LOCKVAL_SRC_POS); bool found = false; for (MediaList::const_iterator it = pMediaList->begin(); it != pMediaList->end(); ++it) { // no AutoCaller, registered image life time is bound to this Medium *pMedium = *it; AutoReadLock imageLock(pMedium COMMA_LOCKVAL_SRC_POS); const Utf8Str &strLocationFull = pMedium->i_getLocationFull(); found = ( aId && pMedium->i_getId() == *aId) || ( !aLocation.isEmpty() && RTPathCompare(location.c_str(), strLocationFull.c_str()) == 0); if (found) { if (pMedium->i_getDeviceType() != mediumType) { if (mediumType == DeviceType_DVD) return setError(E_INVALIDARG, "Cannot mount DVD medium '%s' as floppy", strLocationFull.c_str()); else return setError(E_INVALIDARG, "Cannot mount floppy medium '%s' as DVD", strLocationFull.c_str()); } if (aImage) *aImage = pMedium; break; } } HRESULT rc = found ? S_OK : VBOX_E_OBJECT_NOT_FOUND; if (aSetError && !found) { if (aId) setError(rc, tr("Could not find an image file with UUID {%RTuuid} in the media registry ('%s')"), aId->raw(), m->strSettingsFilePath.c_str()); else setError(rc, tr("Could not find an image file with location '%s' in the media registry ('%s')"), aLocation.c_str(), m->strSettingsFilePath.c_str()); } return rc; } /** * Searches for an IMedium object that represents the given UUID. * * If the UUID is empty (indicating an empty drive), this sets pMedium * to NULL and returns S_OK. * * If the UUID refers to a host drive of the given device type, this * sets pMedium to the object from the list in IHost and returns S_OK. * * If the UUID is an image file, this sets pMedium to the object that * findDVDOrFloppyImage() returned. * * If none of the above apply, this returns VBOX_E_OBJECT_NOT_FOUND. * * @param mediumType Must be DeviceType_DVD or DeviceType_Floppy. * @param uuid UUID to search for; must refer to a host drive or an image file or be null. * @param fRefresh Whether to refresh the list of host drives in IHost (see Host::getDrives()) * @param pMedium out: IMedium object found. * @return */ HRESULT VirtualBox::i_findRemoveableMedium(DeviceType_T mediumType, const Guid &uuid, bool fRefresh, bool aSetError, ComObjPtr &pMedium) { if (uuid.isZero()) { // that's easy pMedium.setNull(); return S_OK; } else if (!uuid.isValid()) { /* handling of case invalid GUID */ return setError(VBOX_E_OBJECT_NOT_FOUND, tr("Guid '%s' is invalid"), uuid.toString().c_str()); } // first search for host drive with that UUID HRESULT rc = m->pHost->i_findHostDriveById(mediumType, uuid, fRefresh, pMedium); if (rc == VBOX_E_OBJECT_NOT_FOUND) // then search for an image with that UUID rc = i_findDVDOrFloppyImage(mediumType, &uuid, Utf8Str::Empty, aSetError, &pMedium); return rc; } HRESULT VirtualBox::i_findGuestOSType(const Bstr &bstrOSType, GuestOSType*& pGuestOSType) { /* Look for a GuestOSType object */ AssertMsg(m->allGuestOSTypes.size() != 0, ("Guest OS types array must be filled")); if (bstrOSType.isEmpty()) { pGuestOSType = NULL; return S_OK; } AutoReadLock alock(m->allGuestOSTypes.getLockHandle() COMMA_LOCKVAL_SRC_POS); for (GuestOSTypesOList::const_iterator it = m->allGuestOSTypes.begin(); it != m->allGuestOSTypes.end(); ++it) { if ((*it)->i_id() == bstrOSType) { pGuestOSType = *it; return S_OK; } } return setError(VBOX_E_OBJECT_NOT_FOUND, tr("Guest OS type '%ls' is invalid"), bstrOSType.raw()); } /** * Returns the constant pseudo-machine UUID that is used to identify the * global media registry. * * Starting with VirtualBox 4.0 each medium remembers in its instance data * in which media registry it is saved (if any): this can either be a machine * UUID, if it's in a per-machine media registry, or this global ID. * * This UUID is only used to identify the VirtualBox object while VirtualBox * is running. It is a compile-time constant and not saved anywhere. * * @return */ const Guid& VirtualBox::i_getGlobalRegistryId() const { return m->uuidMediaRegistry; } const ComObjPtr& VirtualBox::i_host() const { return m->pHost; } SystemProperties* VirtualBox::i_getSystemProperties() const { return m->pSystemProperties; } #ifdef VBOX_WITH_EXTPACK /** * Getter that SystemProperties and others can use to talk to the extension * pack manager. */ ExtPackManager* VirtualBox::i_getExtPackManager() const { return m->ptrExtPackManager; } #endif /** * Getter that machines can talk to the autostart database. */ AutostartDb* VirtualBox::i_getAutostartDb() const { return m->pAutostartDb; } #ifdef VBOX_WITH_RESOURCE_USAGE_API const ComObjPtr& VirtualBox::i_performanceCollector() const { return m->pPerformanceCollector; } #endif /* VBOX_WITH_RESOURCE_USAGE_API */ /** * Returns the default machine folder from the system properties * with proper locking. * @return */ void VirtualBox::i_getDefaultMachineFolder(Utf8Str &str) const { AutoReadLock propsLock(m->pSystemProperties COMMA_LOCKVAL_SRC_POS); str = m->pSystemProperties->m->strDefaultMachineFolder; } /** * Returns the default hard disk format from the system properties * with proper locking. * @return */ void VirtualBox::i_getDefaultHardDiskFormat(Utf8Str &str) const { AutoReadLock propsLock(m->pSystemProperties COMMA_LOCKVAL_SRC_POS); str = m->pSystemProperties->m->strDefaultHardDiskFormat; } const Utf8Str& VirtualBox::i_homeDir() const { return m->strHomeDir; } /** * Calculates the absolute path of the given path taking the VirtualBox home * directory as the current directory. * * @param aPath Path to calculate the absolute path for. * @param aResult Where to put the result (used only on success, can be the * same Utf8Str instance as passed in @a aPath). * @return IPRT result. * * @note Doesn't lock any object. */ int VirtualBox::i_calculateFullPath(const Utf8Str &strPath, Utf8Str &aResult) { AutoCaller autoCaller(this); AssertComRCReturn(autoCaller.rc(), VERR_GENERAL_FAILURE); /* no need to lock since mHomeDir is const */ char folder[RTPATH_MAX]; int vrc = RTPathAbsEx(m->strHomeDir.c_str(), strPath.c_str(), folder, sizeof(folder)); if (RT_SUCCESS(vrc)) aResult = folder; return vrc; } /** * Copies strSource to strTarget, making it relative to the VirtualBox config folder * if it is a subdirectory thereof, or simply copying it otherwise. * * @param strSource Path to evalue and copy. * @param strTarget Buffer to receive target path. */ void VirtualBox::i_copyPathRelativeToConfig(const Utf8Str &strSource, Utf8Str &strTarget) { AutoCaller autoCaller(this); AssertComRCReturnVoid(autoCaller.rc()); // no need to lock since mHomeDir is const // use strTarget as a temporary buffer to hold the machine settings dir strTarget = m->strHomeDir; if (RTPathStartsWith(strSource.c_str(), strTarget.c_str())) // is relative: then append what's left strTarget.append(strSource.c_str() + strTarget.length()); // include '/' else // is not relative: then overwrite strTarget = strSource; } // private methods ///////////////////////////////////////////////////////////////////////////// /** * Checks if there is a hard disk, DVD or floppy image with the given ID or * location already registered. * * On return, sets @a aConflict to the string describing the conflicting medium, * or sets it to @c Null if no conflicting media is found. Returns S_OK in * either case. A failure is unexpected. * * @param aId UUID to check. * @param aLocation Location to check. * @param aConflict Where to return parameters of the conflicting medium. * @param ppMedium Medium reference in case this is simply a duplicate. * * @note Locks the media tree and media objects for reading. */ HRESULT VirtualBox::i_checkMediaForConflicts(const Guid &aId, const Utf8Str &aLocation, Utf8Str &aConflict, ComObjPtr *ppMedium) { AssertReturn(!aId.isZero() && !aLocation.isEmpty(), E_FAIL); AssertReturn(ppMedium, E_INVALIDARG); aConflict.setNull(); ppMedium->setNull(); AutoReadLock alock(i_getMediaTreeLockHandle() COMMA_LOCKVAL_SRC_POS); HRESULT rc = S_OK; ComObjPtr pMediumFound; const char *pcszType = NULL; if (aId.isValid() && !aId.isZero()) rc = i_findHardDiskById(aId, false /* aSetError */, &pMediumFound); if (FAILED(rc) && !aLocation.isEmpty()) rc = i_findHardDiskByLocation(aLocation, false /* aSetError */, &pMediumFound); if (SUCCEEDED(rc)) pcszType = tr("hard disk"); if (!pcszType) { rc = i_findDVDOrFloppyImage(DeviceType_DVD, &aId, aLocation, false /* aSetError */, &pMediumFound); if (SUCCEEDED(rc)) pcszType = tr("CD/DVD image"); } if (!pcszType) { rc = i_findDVDOrFloppyImage(DeviceType_Floppy, &aId, aLocation, false /* aSetError */, &pMediumFound); if (SUCCEEDED(rc)) pcszType = tr("floppy image"); } if (pcszType && pMediumFound) { /* Note: no AutoCaller since bound to this */ AutoReadLock mlock(pMediumFound COMMA_LOCKVAL_SRC_POS); Utf8Str strLocFound = pMediumFound->i_getLocationFull(); Guid idFound = pMediumFound->i_getId(); if ( (RTPathCompare(strLocFound.c_str(), aLocation.c_str()) == 0) && (idFound == aId) ) *ppMedium = pMediumFound; aConflict = Utf8StrFmt(tr("%s '%s' with UUID {%RTuuid}"), pcszType, strLocFound.c_str(), idFound.raw()); } return S_OK; } /** * Checks whether the given UUID is already in use by one medium for the * given device type. * * @returns true if the UUID is already in use * fale otherwise * @param aId The UUID to check. * @param deviceType The device type the UUID is going to be checked for * conflicts. */ bool VirtualBox::i_isMediaUuidInUse(const Guid &aId, DeviceType_T deviceType) { /* A zero UUID is invalid here, always claim that it is already used. */ AssertReturn(!aId.isZero(), true); AutoReadLock alock(i_getMediaTreeLockHandle() COMMA_LOCKVAL_SRC_POS); HRESULT rc = S_OK; bool fInUse = false; ComObjPtr pMediumFound; switch (deviceType) { case DeviceType_HardDisk: rc = i_findHardDiskById(aId, false /* aSetError */, &pMediumFound); break; case DeviceType_DVD: rc = i_findDVDOrFloppyImage(DeviceType_DVD, &aId, Utf8Str::Empty, false /* aSetError */, &pMediumFound); break; case DeviceType_Floppy: rc = i_findDVDOrFloppyImage(DeviceType_Floppy, &aId, Utf8Str::Empty, false /* aSetError */, &pMediumFound); break; default: AssertMsgFailed(("Invalid device type %d\n", deviceType)); } if (SUCCEEDED(rc) && pMediumFound) fInUse = true; return fInUse; } /** * Called from Machine::prepareSaveSettings() when it has detected * that a machine has been renamed. Such renames will require * updating the global media registry during the * VirtualBox::saveSettings() that follows later. * * When a machine is renamed, there may well be media (in particular, * diff images for snapshots) in the global registry that will need * to have their paths updated. Before 3.2, Machine::saveSettings * used to call VirtualBox::saveSettings implicitly, which was both * unintuitive and caused locking order problems. Now, we remember * such pending name changes with this method so that * VirtualBox::saveSettings() can process them properly. */ void VirtualBox::i_rememberMachineNameChangeForMedia(const Utf8Str &strOldConfigDir, const Utf8Str &strNewConfigDir) { AutoWriteLock mediaLock(i_getMediaTreeLockHandle() COMMA_LOCKVAL_SRC_POS); Data::PendingMachineRename pmr; pmr.strConfigDirOld = strOldConfigDir; pmr.strConfigDirNew = strNewConfigDir; m->llPendingMachineRenames.push_back(pmr); } struct SaveMediaRegistriesDesc { MediaList llMedia; ComObjPtr pVirtualBox; }; static int fntSaveMediaRegistries(RTTHREAD ThreadSelf, void *pvUser) { NOREF(ThreadSelf); SaveMediaRegistriesDesc *pDesc = (SaveMediaRegistriesDesc *)pvUser; if (!pDesc) { LogRelFunc(("Thread for saving media registries lacks parameters\n")); return VERR_INVALID_PARAMETER; } for (MediaList::const_iterator it = pDesc->llMedia.begin(); it != pDesc->llMedia.end(); ++it) { Medium *pMedium = *it; pMedium->i_markRegistriesModified(); } pDesc->pVirtualBox->i_saveModifiedRegistries(); pDesc->llMedia.clear(); pDesc->pVirtualBox.setNull(); delete pDesc; return VINF_SUCCESS; } /** * Goes through all known media (hard disks, floppies and DVDs) and saves * those into the given settings::MediaRegistry structures whose registry * ID match the given UUID. * * Before actually writing to the structures, all media paths (not just the * ones for the given registry) are updated if machines have been renamed * since the last call. * * This gets called from two contexts: * * -- VirtualBox::saveSettings() with the UUID of the global registry * (VirtualBox::Data.uuidRegistry); this will save those media * which had been loaded from the global registry or have been * attached to a "legacy" machine which can't save its own registry; * * -- Machine::saveSettings() with the UUID of a machine, if a medium * has been attached to a machine created with VirtualBox 4.0 or later. * * Media which have only been temporarily opened without having been * attached to a machine have a NULL registry UUID and therefore don't * get saved. * * This locks the media tree. Throws HRESULT on errors! * * @param mediaRegistry Settings structure to fill. * @param uuidRegistry The UUID of the media registry; either a machine UUID * (if machine registry) or the UUID of the global registry. * @param strMachineFolder The machine folder for relative paths, if machine registry, or an empty string otherwise. */ void VirtualBox::i_saveMediaRegistry(settings::MediaRegistry &mediaRegistry, const Guid &uuidRegistry, const Utf8Str &strMachineFolder) { // lock all media for the following; use a write lock because we're // modifying the PendingMachineRenamesList, which is protected by this AutoWriteLock mediaLock(i_getMediaTreeLockHandle() COMMA_LOCKVAL_SRC_POS); // if a machine was renamed, then we'll need to refresh media paths if (m->llPendingMachineRenames.size()) { // make a single list from the three media lists so we don't need three loops MediaList llAllMedia; // with hard disks, we must use the map, not the list, because the list only has base images for (HardDiskMap::iterator it = m->mapHardDisks.begin(); it != m->mapHardDisks.end(); ++it) llAllMedia.push_back(it->second); for (MediaList::iterator it = m->allDVDImages.begin(); it != m->allDVDImages.end(); ++it) llAllMedia.push_back(*it); for (MediaList::iterator it = m->allFloppyImages.begin(); it != m->allFloppyImages.end(); ++it) llAllMedia.push_back(*it); SaveMediaRegistriesDesc *pDesc = new SaveMediaRegistriesDesc(); for (MediaList::iterator it = llAllMedia.begin(); it != llAllMedia.end(); ++it) { Medium *pMedium = *it; for (Data::PendingMachineRenamesList::iterator it2 = m->llPendingMachineRenames.begin(); it2 != m->llPendingMachineRenames.end(); ++it2) { const Data::PendingMachineRename &pmr = *it2; HRESULT rc = pMedium->i_updatePath(pmr.strConfigDirOld, pmr.strConfigDirNew); if (SUCCEEDED(rc)) { // Remember which medium objects has been changed, // to trigger saving their registries later. pDesc->llMedia.push_back(pMedium); } else if (rc == VBOX_E_FILE_ERROR) /* nothing */; else AssertComRC(rc); } } // done, don't do it again until we have more machine renames m->llPendingMachineRenames.clear(); if (pDesc->llMedia.size()) { // Handle the media registry saving in a separate thread, to // avoid giant locking problems and passing up the list many // levels up to whoever triggered saveSettings, as there are // lots of places which would need to handle saving more settings. pDesc->pVirtualBox = this; int vrc = RTThreadCreate(NULL, fntSaveMediaRegistries, (void *)pDesc, 0, // cbStack (default) RTTHREADTYPE_MAIN_WORKER, 0, // flags "SaveMediaReg"); ComAssertRC(vrc); // failure means that settings aren't saved, but there isn't // much we can do besides avoiding memory leaks if (RT_FAILURE(vrc)) { LogRelFunc(("Failed to create thread for saving media registries (%Rrc)\n", vrc)); delete pDesc; } } else delete pDesc; } struct { MediaOList &llSource; settings::MediaList &llTarget; } s[] = { // hard disks { m->allHardDisks, mediaRegistry.llHardDisks }, // CD/DVD images { m->allDVDImages, mediaRegistry.llDvdImages }, // floppy images { m->allFloppyImages, mediaRegistry.llFloppyImages } }; HRESULT rc; for (size_t i = 0; i < RT_ELEMENTS(s); ++i) { MediaOList &llSource = s[i].llSource; settings::MediaList &llTarget = s[i].llTarget; llTarget.clear(); for (MediaList::const_iterator it = llSource.begin(); it != llSource.end(); ++it) { Medium *pMedium = *it; AutoCaller autoCaller(pMedium); if (FAILED(autoCaller.rc())) throw autoCaller.rc(); AutoReadLock mlock(pMedium COMMA_LOCKVAL_SRC_POS); if (pMedium->i_isInRegistry(uuidRegistry)) { llTarget.push_back(settings::g_MediumEmpty); rc = pMedium->i_saveSettings(llTarget.back(), strMachineFolder); // this recurses into child hard disks if (FAILED(rc)) { llTarget.pop_back(); throw rc; } } } } } /** * Helper function which actually writes out VirtualBox.xml, the main configuration file. * Gets called from the public VirtualBox::SaveSettings() as well as from various other * places internally when settings need saving. * * @note Caller must have locked the VirtualBox object for writing and must not hold any * other locks since this locks all kinds of member objects and trees temporarily, * which could cause conflicts. */ HRESULT VirtualBox::i_saveSettings() { AutoCaller autoCaller(this); AssertComRCReturn(autoCaller.rc(), autoCaller.rc()); AssertReturn(isWriteLockOnCurrentThread(), E_FAIL); AssertReturn(!m->strSettingsFilePath.isEmpty(), E_FAIL); HRESULT rc = S_OK; try { // machines m->pMainConfigFile->llMachines.clear(); { AutoReadLock machinesLock(m->allMachines.getLockHandle() COMMA_LOCKVAL_SRC_POS); for (MachinesOList::iterator it = m->allMachines.begin(); it != m->allMachines.end(); ++it) { Machine *pMachine = *it; // save actual machine registry entry settings::MachineRegistryEntry mre; rc = pMachine->i_saveRegistryEntry(mre); m->pMainConfigFile->llMachines.push_back(mre); } } i_saveMediaRegistry(m->pMainConfigFile->mediaRegistry, m->uuidMediaRegistry, // global media registry ID Utf8Str::Empty); // strMachineFolder m->pMainConfigFile->llDhcpServers.clear(); { AutoReadLock dhcpLock(m->allDHCPServers.getLockHandle() COMMA_LOCKVAL_SRC_POS); for (DHCPServersOList::const_iterator it = m->allDHCPServers.begin(); it != m->allDHCPServers.end(); ++it) { settings::DHCPServer d; rc = (*it)->i_saveSettings(d); if (FAILED(rc)) throw rc; m->pMainConfigFile->llDhcpServers.push_back(d); } } #ifdef VBOX_WITH_NAT_SERVICE /* Saving NAT Network configuration */ m->pMainConfigFile->llNATNetworks.clear(); { AutoReadLock natNetworkLock(m->allNATNetworks.getLockHandle() COMMA_LOCKVAL_SRC_POS); for (NATNetworksOList::const_iterator it = m->allNATNetworks.begin(); it != m->allNATNetworks.end(); ++it) { settings::NATNetwork n; rc = (*it)->i_saveSettings(n); if (FAILED(rc)) throw rc; m->pMainConfigFile->llNATNetworks.push_back(n); } } #endif // leave extra data alone, it's still in the config file // host data (USB filters) rc = m->pHost->i_saveSettings(m->pMainConfigFile->host); if (FAILED(rc)) throw rc; rc = m->pSystemProperties->i_saveSettings(m->pMainConfigFile->systemProperties); if (FAILED(rc)) throw rc; // and write out the XML, still under the lock m->pMainConfigFile->write(m->strSettingsFilePath); } catch (HRESULT err) { /* we assume that error info is set by the thrower */ rc = err; } catch (...) { rc = VirtualBoxBase::handleUnexpectedExceptions(this, RT_SRC_POS); } return rc; } /** * Helper to register the machine. * * When called during VirtualBox startup, adds the given machine to the * collection of registered machines. Otherwise tries to mark the machine * as registered, and, if succeeded, adds it to the collection and * saves global settings. * * @note The caller must have added itself as a caller of the @a aMachine * object if calls this method not on VirtualBox startup. * * @param aMachine machine to register * * @note Locks objects! */ HRESULT VirtualBox::i_registerMachine(Machine *aMachine) { ComAssertRet(aMachine, E_INVALIDARG); AutoCaller autoCaller(this); if (FAILED(autoCaller.rc())) return autoCaller.rc(); HRESULT rc = S_OK; AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS); { ComObjPtr pMachine; rc = i_findMachine(aMachine->i_getId(), true /* fPermitInaccessible */, false /* aDoSetError */, &pMachine); if (SUCCEEDED(rc)) { /* sanity */ AutoLimitedCaller machCaller(pMachine); AssertComRC(machCaller.rc()); return setError(E_INVALIDARG, tr("Registered machine with UUID {%RTuuid} ('%s') already exists"), aMachine->i_getId().raw(), pMachine->i_getSettingsFileFull().c_str()); } ComAssertRet(rc == VBOX_E_OBJECT_NOT_FOUND, rc); rc = S_OK; } if (getObjectState().getState() != ObjectState::InInit) { rc = aMachine->i_prepareRegister(); if (FAILED(rc)) return rc; } /* add to the collection of registered machines */ m->allMachines.addChild(aMachine); if (getObjectState().getState() != ObjectState::InInit) rc = i_saveSettings(); return rc; } /** * Remembers the given medium object by storing it in either the global * medium registry or a machine one. * * @note Caller must hold the media tree lock for writing; in addition, this * locks @a pMedium for reading * * @param pMedium Medium object to remember. * @param ppMedium Actually stored medium object. Can be different if due * to an unavoidable race there was a duplicate Medium object * created. * @param mediaTreeLock Reference to the AutoWriteLock holding the media tree * lock, necessary to release it in the right spot. * @return */ HRESULT VirtualBox::i_registerMedium(const ComObjPtr &pMedium, ComObjPtr *ppMedium, AutoWriteLock &mediaTreeLock) { AssertReturn(pMedium != NULL, E_INVALIDARG); AssertReturn(ppMedium != NULL, E_INVALIDARG); // caller must hold the media tree write lock Assert(i_getMediaTreeLockHandle().isWriteLockOnCurrentThread()); AutoCaller autoCaller(this); AssertComRCReturn(autoCaller.rc(), autoCaller.rc()); AutoCaller mediumCaller(pMedium); AssertComRCReturn(mediumCaller.rc(), mediumCaller.rc()); const char *pszDevType = NULL; ObjectsList *pall = NULL; DeviceType_T devType; { AutoReadLock mediumLock(pMedium COMMA_LOCKVAL_SRC_POS); devType = pMedium->i_getDeviceType(); } switch (devType) { case DeviceType_HardDisk: pall = &m->allHardDisks; pszDevType = tr("hard disk"); break; case DeviceType_DVD: pszDevType = tr("DVD image"); pall = &m->allDVDImages; break; case DeviceType_Floppy: pszDevType = tr("floppy image"); pall = &m->allFloppyImages; break; default: AssertMsgFailedReturn(("invalid device type %d", devType), E_INVALIDARG); } Guid id; Utf8Str strLocationFull; ComObjPtr pParent; { AutoReadLock mediumLock(pMedium COMMA_LOCKVAL_SRC_POS); id = pMedium->i_getId(); strLocationFull = pMedium->i_getLocationFull(); pParent = pMedium->i_getParent(); } HRESULT rc; Utf8Str strConflict; ComObjPtr pDupMedium; rc = i_checkMediaForConflicts(id, strLocationFull, strConflict, &pDupMedium); if (FAILED(rc)) return rc; if (pDupMedium.isNull()) { if (strConflict.length()) return setError(E_INVALIDARG, tr("Cannot register the %s '%s' {%RTuuid} because a %s already exists"), pszDevType, strLocationFull.c_str(), id.raw(), strConflict.c_str(), m->strSettingsFilePath.c_str()); // add to the collection if it is a base medium if (pParent.isNull()) pall->getList().push_back(pMedium); // store all hard disks (even differencing images) in the map if (devType == DeviceType_HardDisk) m->mapHardDisks[id] = pMedium; mediumCaller.release(); mediaTreeLock.release(); *ppMedium = pMedium; } else { // pMedium may be the last reference to the Medium object, and the // caller may have specified the same ComObjPtr as the output parameter. // In this case the assignment will uninit the object, and we must not // have a caller pending. mediumCaller.release(); // release media tree lock, must not be held at uninit time. mediaTreeLock.release(); // must not hold the media tree write lock any more Assert(!i_getMediaTreeLockHandle().isWriteLockOnCurrentThread()); *ppMedium = pDupMedium; } // Restore the initial lock state, so that no unexpected lock changes are // done by this method, which would need adjustments everywhere. mediaTreeLock.acquire(); return rc; } /** * Removes the given medium from the respective registry. * * @param pMedium Hard disk object to remove. * * @note Caller must hold the media tree lock for writing; in addition, this locks @a pMedium for reading */ HRESULT VirtualBox::i_unregisterMedium(Medium *pMedium) { AssertReturn(pMedium != NULL, E_INVALIDARG); AutoCaller autoCaller(this); AssertComRCReturn(autoCaller.rc(), autoCaller.rc()); AutoCaller mediumCaller(pMedium); AssertComRCReturn(mediumCaller.rc(), mediumCaller.rc()); // caller must hold the media tree write lock Assert(i_getMediaTreeLockHandle().isWriteLockOnCurrentThread()); Guid id; ComObjPtr pParent; DeviceType_T devType; { AutoReadLock mediumLock(pMedium COMMA_LOCKVAL_SRC_POS); id = pMedium->i_getId(); pParent = pMedium->i_getParent(); devType = pMedium->i_getDeviceType(); } ObjectsList *pall = NULL; switch (devType) { case DeviceType_HardDisk: pall = &m->allHardDisks; break; case DeviceType_DVD: pall = &m->allDVDImages; break; case DeviceType_Floppy: pall = &m->allFloppyImages; break; default: AssertMsgFailedReturn(("invalid device type %d", devType), E_INVALIDARG); } // remove from the collection if it is a base medium if (pParent.isNull()) pall->getList().remove(pMedium); // remove all hard disks (even differencing images) from map if (devType == DeviceType_HardDisk) { size_t cnt = m->mapHardDisks.erase(id); Assert(cnt == 1); NOREF(cnt); } return S_OK; } /** * Little helper called from unregisterMachineMedia() to recursively add media to the given list, * with children appearing before their parents. * @param llMedia * @param pMedium */ void VirtualBox::i_pushMediumToListWithChildren(MediaList &llMedia, Medium *pMedium) { // recurse first, then add ourselves; this way children end up on the // list before their parents const MediaList &llChildren = pMedium->i_getChildren(); for (MediaList::const_iterator it = llChildren.begin(); it != llChildren.end(); ++it) { Medium *pChild = *it; i_pushMediumToListWithChildren(llMedia, pChild); } Log(("Pushing medium %RTuuid\n", pMedium->i_getId().raw())); llMedia.push_back(pMedium); } /** * Unregisters all Medium objects which belong to the given machine registry. * Gets called from Machine::uninit() just before the machine object dies * and must only be called with a machine UUID as the registry ID. * * Locks the media tree. * * @param uuidMachine Medium registry ID (always a machine UUID) * @return */ HRESULT VirtualBox::i_unregisterMachineMedia(const Guid &uuidMachine) { Assert(!uuidMachine.isZero() && uuidMachine.isValid()); LogFlowFuncEnter(); AutoCaller autoCaller(this); AssertComRCReturn(autoCaller.rc(), autoCaller.rc()); MediaList llMedia2Close; { AutoWriteLock tlock(i_getMediaTreeLockHandle() COMMA_LOCKVAL_SRC_POS); for (MediaOList::iterator it = m->allHardDisks.getList().begin(); it != m->allHardDisks.getList().end(); ++it) { ComObjPtr pMedium = *it; AutoCaller medCaller(pMedium); if (FAILED(medCaller.rc())) return medCaller.rc(); AutoReadLock medlock(pMedium COMMA_LOCKVAL_SRC_POS); if (pMedium->i_isInRegistry(uuidMachine)) // recursively with children first i_pushMediumToListWithChildren(llMedia2Close, pMedium); } } for (MediaList::iterator it = llMedia2Close.begin(); it != llMedia2Close.end(); ++it) { ComObjPtr pMedium = *it; Log(("Closing medium %RTuuid\n", pMedium->i_getId().raw())); AutoCaller mac(pMedium); pMedium->i_close(mac); } LogFlowFuncLeave(); return S_OK; } /** * Removes the given machine object from the internal list of registered machines. * Called from Machine::Unregister(). * @param pMachine * @param id UUID of the machine. Must be passed by caller because machine may be dead by this time. * @return */ HRESULT VirtualBox::i_unregisterMachine(Machine *pMachine, const Guid &id) { // remove from the collection of registered machines AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS); m->allMachines.removeChild(pMachine); // save the global registry HRESULT rc = i_saveSettings(); alock.release(); /* * Now go over all known media and checks if they were registered in the * media registry of the given machine. Each such medium is then moved to * a different media registry to make sure it doesn't get lost since its * media registry is about to go away. * * This fixes the following use case: Image A.vdi of machine A is also used * by machine B, but registered in the media registry of machine A. If machine * A is deleted, A.vdi must be moved to the registry of B, or else B will * become inaccessible. */ { AutoReadLock tlock(i_getMediaTreeLockHandle() COMMA_LOCKVAL_SRC_POS); // iterate over the list of *base* images for (MediaOList::iterator it = m->allHardDisks.getList().begin(); it != m->allHardDisks.getList().end(); ++it) { ComObjPtr &pMedium = *it; AutoCaller medCaller(pMedium); if (FAILED(medCaller.rc())) return medCaller.rc(); AutoWriteLock mlock(pMedium COMMA_LOCKVAL_SRC_POS); if (pMedium->i_removeRegistryRecursive(id)) { // machine ID was found in base medium's registry list: // move this base image and all its children to another registry then // 1) first, find a better registry to add things to const Guid *puuidBetter = pMedium->i_getAnyMachineBackref(); if (puuidBetter) { // 2) better registry found: then use that pMedium->i_addRegistryRecursive(*puuidBetter); // 3) and make sure the registry is saved below mlock.release(); tlock.release(); i_markRegistryModified(*puuidBetter); tlock.acquire(); mlock.acquire(); } } } } i_saveModifiedRegistries(); /* fire an event */ i_onMachineRegistered(id, FALSE); return rc; } /** * Marks the registry for @a uuid as modified, so that it's saved in a later * call to saveModifiedRegistries(). * * @param uuid */ void VirtualBox::i_markRegistryModified(const Guid &uuid) { if (uuid == i_getGlobalRegistryId()) ASMAtomicIncU64(&m->uRegistryNeedsSaving); else { ComObjPtr pMachine; HRESULT rc = i_findMachine(uuid, false /* fPermitInaccessible */, false /* aSetError */, &pMachine); if (SUCCEEDED(rc)) { AutoCaller machineCaller(pMachine); if (SUCCEEDED(machineCaller.rc())) ASMAtomicIncU64(&pMachine->uRegistryNeedsSaving); } } } /** * Marks the registry for @a uuid as unmodified, so that it's not saved in * a later call to saveModifiedRegistries(). * * @param uuid */ void VirtualBox::i_unmarkRegistryModified(const Guid &uuid) { uint64_t uOld; if (uuid == i_getGlobalRegistryId()) { for (;;) { uOld = ASMAtomicReadU64(&m->uRegistryNeedsSaving); if (!uOld) break; if (ASMAtomicCmpXchgU64(&m->uRegistryNeedsSaving, 0, uOld)) break; ASMNopPause(); } } else { ComObjPtr pMachine; HRESULT rc = i_findMachine(uuid, false /* fPermitInaccessible */, false /* aSetError */, &pMachine); if (SUCCEEDED(rc)) { AutoCaller machineCaller(pMachine); if (SUCCEEDED(machineCaller.rc())) { for (;;) { uOld = ASMAtomicReadU64(&pMachine->uRegistryNeedsSaving); if (!uOld) break; if (ASMAtomicCmpXchgU64(&pMachine->uRegistryNeedsSaving, 0, uOld)) break; ASMNopPause(); } } } } } /** * Saves all settings files according to the modified flags in the Machine * objects and in the VirtualBox object. * * This locks machines and the VirtualBox object as necessary, so better not * hold any locks before calling this. * * @return */ void VirtualBox::i_saveModifiedRegistries() { HRESULT rc = S_OK; bool fNeedsGlobalSettings = false; uint64_t uOld; { AutoReadLock alock(m->allMachines.getLockHandle() COMMA_LOCKVAL_SRC_POS); for (MachinesOList::iterator it = m->allMachines.begin(); it != m->allMachines.end(); ++it) { const ComObjPtr &pMachine = *it; for (;;) { uOld = ASMAtomicReadU64(&pMachine->uRegistryNeedsSaving); if (!uOld) break; if (ASMAtomicCmpXchgU64(&pMachine->uRegistryNeedsSaving, 0, uOld)) break; ASMNopPause(); } if (uOld) { AutoCaller autoCaller(pMachine); if (FAILED(autoCaller.rc())) continue; /* object is already dead, no point in saving settings */ if (getObjectState().getState() != ObjectState::Ready) continue; AutoWriteLock mlock(pMachine COMMA_LOCKVAL_SRC_POS); rc = pMachine->i_saveSettings(&fNeedsGlobalSettings, Machine::SaveS_Force); // caller said save, so stop arguing } } } for (;;) { uOld = ASMAtomicReadU64(&m->uRegistryNeedsSaving); if (!uOld) break; if (ASMAtomicCmpXchgU64(&m->uRegistryNeedsSaving, 0, uOld)) break; ASMNopPause(); } if (uOld || fNeedsGlobalSettings) { AutoWriteLock alock(this COMMA_LOCKVAL_SRC_POS); rc = i_saveSettings(); } NOREF(rc); /* XXX */ } /* static */ const com::Utf8Str &VirtualBox::i_getVersionNormalized() { return sVersionNormalized; } /** * Checks if the path to the specified file exists, according to the path * information present in the file name. Optionally the path is created. * * Note that the given file name must contain the full path otherwise the * extracted relative path will be created based on the current working * directory which is normally unknown. * * @param aFileName Full file name which path is checked/created. * @param aCreate Flag if the path should be created if it doesn't exist. * * @return Extended error information on failure to check/create the path. */ /* static */ HRESULT VirtualBox::i_ensureFilePathExists(const Utf8Str &strFileName, bool fCreate) { Utf8Str strDir(strFileName); strDir.stripFilename(); if (!RTDirExists(strDir.c_str())) { if (fCreate) { int vrc = RTDirCreateFullPath(strDir.c_str(), 0700); if (RT_FAILURE(vrc)) return i_setErrorStatic(VBOX_E_IPRT_ERROR, Utf8StrFmt(tr("Could not create the directory '%s' (%Rrc)"), strDir.c_str(), vrc)); } else return i_setErrorStatic(VBOX_E_IPRT_ERROR, Utf8StrFmt(tr("Directory '%s' does not exist"), strDir.c_str())); } return S_OK; } const Utf8Str& VirtualBox::i_settingsFilePath() { return m->strSettingsFilePath; } /** * Returns the lock handle which protects the machines list. As opposed * to version 3.1 and earlier, these lists are no longer protected by the * VirtualBox lock, but by this more specialized lock. Mind the locking * order: always request this lock after the VirtualBox object lock but * before the locks of any machine object. See AutoLock.h. */ RWLockHandle& VirtualBox::i_getMachinesListLockHandle() { return m->lockMachines; } /** * Returns the lock handle which protects the media trees (hard disks, * DVDs, floppies). As opposed to version 3.1 and earlier, these lists * are no longer protected by the VirtualBox lock, but by this more * specialized lock. Mind the locking order: always request this lock * after the VirtualBox object lock but before the locks of the media * objects contained in these lists. See AutoLock.h. */ RWLockHandle& VirtualBox::i_getMediaTreeLockHandle() { return m->lockMedia; } /** * Thread function that handles custom events posted using #postEvent(). */ // static DECLCALLBACK(int) VirtualBox::AsyncEventHandler(RTTHREAD thread, void *pvUser) { LogFlowFuncEnter(); AssertReturn(pvUser, VERR_INVALID_POINTER); HRESULT hr = com::Initialize(); if (FAILED(hr)) return VERR_COM_UNEXPECTED; int rc = VINF_SUCCESS; try { /* Create an event queue for the current thread. */ EventQueue *pEventQueue = new EventQueue(); AssertPtr(pEventQueue); /* Return the queue to the one who created this thread. */ *(static_cast (pvUser)) = pEventQueue; /* signal that we're ready. */ RTThreadUserSignal(thread); /* * In case of spurious wakeups causing VERR_TIMEOUTs and/or other return codes * we must not stop processing events and delete the pEventQueue object. This must * be done ONLY when we stop this loop via interruptEventQueueProcessing(). * See @bugref{5724}. */ for (;;) { rc = pEventQueue->processEventQueue(RT_INDEFINITE_WAIT); if (rc == VERR_INTERRUPTED) { LogFlow(("Event queue processing ended with rc=%Rrc\n", rc)); rc = VINF_SUCCESS; /* Set success when exiting. */ break; } } delete pEventQueue; } catch (std::bad_alloc &ba) { rc = VERR_NO_MEMORY; NOREF(ba); } com::Shutdown(); LogFlowFuncLeaveRC(rc); return rc; } //////////////////////////////////////////////////////////////////////////////// /** * Takes the current list of registered callbacks of the managed VirtualBox * instance, and calls #handleCallback() for every callback item from the * list, passing the item as an argument. * * @note Locks the managed VirtualBox object for reading but leaves the lock * before iterating over callbacks and calling their methods. */ void *VirtualBox::CallbackEvent::handler() { if (!mVirtualBox) return NULL; AutoCaller autoCaller(mVirtualBox); if (!autoCaller.isOk()) { LogWarningFunc(("VirtualBox has been uninitialized (state=%d), the callback event is discarded!\n", mVirtualBox->getObjectState().getState())); /* We don't need mVirtualBox any more, so release it */ mVirtualBox = NULL; return NULL; } { VBoxEventDesc evDesc; prepareEventDesc(mVirtualBox->m->pEventSource, evDesc); evDesc.fire(/* don't wait for delivery */0); } mVirtualBox = NULL; /* Not needed any longer. Still make sense to do this? */ return NULL; } //STDMETHODIMP VirtualBox::CreateDHCPServerForInterface(/*IHostNetworkInterface * aIinterface,*/ IDHCPServer ** aServer) //{ // return E_NOTIMPL; //} HRESULT VirtualBox::createDHCPServer(const com::Utf8Str &aName, ComPtr &aServer) { ComObjPtr dhcpServer; dhcpServer.createObject(); HRESULT rc = dhcpServer->init(this, Bstr(aName).raw()); if (FAILED(rc)) return rc; rc = i_registerDHCPServer(dhcpServer, true); if (FAILED(rc)) return rc; dhcpServer.queryInterfaceTo(aServer.asOutParam()); return rc; } HRESULT VirtualBox::findDHCPServerByNetworkName(const com::Utf8Str &aName, ComPtr &aServer) { HRESULT rc = S_OK; ComPtr found; AutoReadLock alock(m->allDHCPServers.getLockHandle() COMMA_LOCKVAL_SRC_POS); for (DHCPServersOList::const_iterator it = m->allDHCPServers.begin(); it != m->allDHCPServers.end(); ++it) { Bstr bstr; rc = (*it)->COMGETTER(NetworkName)(bstr.asOutParam()); if (FAILED(rc)) return rc; if (bstr == Bstr(aName).raw()) { found = *it; break; } } if (!found) return E_INVALIDARG; rc = found.queryInterfaceTo(aServer.asOutParam()); return rc; } HRESULT VirtualBox::removeDHCPServer(const ComPtr &aServer) { IDHCPServer *aP = aServer; HRESULT rc = i_unregisterDHCPServer(static_cast(aP)); return rc; } /** * Remembers the given DHCP server in the settings. * * @param aDHCPServer DHCP server object to remember. * @param aSaveSettings @c true to save settings to disk (default). * * When @a aSaveSettings is @c true, this operation may fail because of the * failed #saveSettings() method it calls. In this case, the dhcp server object * will not be remembered. It is therefore the responsibility of the caller to * call this method as the last step of some action that requires registration * in order to make sure that only fully functional dhcp server objects get * registered. * * @note Locks this object for writing and @a aDHCPServer for reading. */ HRESULT VirtualBox::i_registerDHCPServer(DHCPServer *aDHCPServer, bool aSaveSettings /*= true*/) { AssertReturn(aDHCPServer != NULL, E_INVALIDARG); AutoCaller autoCaller(this); AssertComRCReturn(autoCaller.rc(), autoCaller.rc()); // Acquire a lock on the VirtualBox object early to avoid lock order issues // when we call i_saveSettings() later on. AutoWriteLock vboxLock(this COMMA_LOCKVAL_SRC_POS); // need it below, in findDHCPServerByNetworkName (reading) and in // m->allDHCPServers.addChild, so need to get it here to avoid lock // order trouble with dhcpServerCaller AutoWriteLock alock(m->allDHCPServers.getLockHandle() COMMA_LOCKVAL_SRC_POS); AutoCaller dhcpServerCaller(aDHCPServer); AssertComRCReturn(dhcpServerCaller.rc(), dhcpServerCaller.rc()); Bstr name; com::Utf8Str uname; HRESULT rc = S_OK; rc = aDHCPServer->COMGETTER(NetworkName)(name.asOutParam()); if (FAILED(rc)) return rc; uname = Utf8Str(name); ComPtr existing; rc = findDHCPServerByNetworkName(uname, existing); if (SUCCEEDED(rc)) return E_INVALIDARG; rc = S_OK; m->allDHCPServers.addChild(aDHCPServer); // we need to release the list lock before we attempt to acquire locks // on other objects in i_saveSettings (see @bugref{7500}) alock.release(); if (aSaveSettings) { // we acquired the lock on 'this' earlier to avoid lock order issues rc = i_saveSettings(); if (FAILED(rc)) { alock.acquire(); m->allDHCPServers.removeChild(aDHCPServer); } } return rc; } /** * Removes the given DHCP server from the settings. * * @param aDHCPServer DHCP server object to remove. * * This operation may fail because of the failed #saveSettings() method it * calls. In this case, the DHCP server will NOT be removed from the settings * when this method returns. * * @note Locks this object for writing. */ HRESULT VirtualBox::i_unregisterDHCPServer(DHCPServer *aDHCPServer) { AssertReturn(aDHCPServer != NULL, E_INVALIDARG); AutoCaller autoCaller(this); AssertComRCReturn(autoCaller.rc(), autoCaller.rc()); AutoCaller dhcpServerCaller(aDHCPServer); AssertComRCReturn(dhcpServerCaller.rc(), dhcpServerCaller.rc()); AutoWriteLock vboxLock(this COMMA_LOCKVAL_SRC_POS); AutoWriteLock alock(m->allDHCPServers.getLockHandle() COMMA_LOCKVAL_SRC_POS); m->allDHCPServers.removeChild(aDHCPServer); // we need to release the list lock before we attempt to acquire locks // on other objects in i_saveSettings (see @bugref{7500}) alock.release(); HRESULT rc = i_saveSettings(); // undo the changes if we failed to save them if (FAILED(rc)) { alock.acquire(); m->allDHCPServers.addChild(aDHCPServer); } return rc; } /** * NAT Network */ HRESULT VirtualBox::createNATNetwork(const com::Utf8Str &aNetworkName, ComPtr &aNetwork) { #ifdef VBOX_WITH_NAT_SERVICE ComObjPtr natNetwork; natNetwork.createObject(); HRESULT rc = natNetwork->init(this, Bstr(aNetworkName).raw()); if (FAILED(rc)) return rc; rc = i_registerNATNetwork(natNetwork, true); if (FAILED(rc)) return rc; natNetwork.queryInterfaceTo(aNetwork.asOutParam()); fireNATNetworkCreationDeletionEvent(m->pEventSource, Bstr(aNetworkName).raw(), TRUE); return rc; #else NOREF(aName); NOREF(aNatNetwork); return E_NOTIMPL; #endif } HRESULT VirtualBox::findNATNetworkByName(const com::Utf8Str &aNetworkName, ComPtr &aNetwork) { #ifdef VBOX_WITH_NAT_SERVICE HRESULT rc = S_OK; ComPtr found; AutoReadLock alock(m->allNATNetworks.getLockHandle() COMMA_LOCKVAL_SRC_POS); for (NATNetworksOList::const_iterator it = m->allNATNetworks.begin(); it != m->allNATNetworks.end(); ++it) { Bstr bstr; rc = (*it)->COMGETTER(NetworkName)(bstr.asOutParam()); if (FAILED(rc)) return rc; if (bstr == Bstr(aNetworkName).raw()) { found = *it; break; } } if (!found) return E_INVALIDARG; found.queryInterfaceTo(aNetwork.asOutParam()); return rc; #else NOREF(aName); NOREF(aNetworkName); return E_NOTIMPL; #endif } HRESULT VirtualBox::removeNATNetwork(const ComPtr &aNetwork) { #ifdef VBOX_WITH_NAT_SERVICE Bstr name; HRESULT rc = S_OK; INATNetwork *iNw = aNetwork; NATNetwork *network = static_cast(iNw); rc = network->COMGETTER(NetworkName)(name.asOutParam()); rc = i_unregisterNATNetwork(network, true); fireNATNetworkCreationDeletionEvent(m->pEventSource, name.raw(), FALSE); return rc; #else NOREF(aNetwork); return E_NOTIMPL; #endif } /** * Remembers the given NAT network in the settings. * * @param aNATNetwork NAT Network object to remember. * @param aSaveSettings @c true to save settings to disk (default). * * * @note Locks this object for writing and @a aNATNetwork for reading. */ HRESULT VirtualBox::i_registerNATNetwork(NATNetwork *aNATNetwork, bool aSaveSettings /*= true*/) { #ifdef VBOX_WITH_NAT_SERVICE AssertReturn(aNATNetwork != NULL, E_INVALIDARG); AutoCaller autoCaller(this); AssertComRCReturnRC(autoCaller.rc()); AutoCaller natNetworkCaller(aNATNetwork); AssertComRCReturnRC(natNetworkCaller.rc()); Bstr name; HRESULT rc; rc = aNATNetwork->COMGETTER(NetworkName)(name.asOutParam()); AssertComRCReturnRC(rc); /* returned value isn't 0 and aSaveSettings is true * means that we create duplicate, otherwise we just load settings. */ if ( sNatNetworkNameToRefCount[name] && aSaveSettings) AssertComRCReturnRC(E_INVALIDARG); rc = S_OK; sNatNetworkNameToRefCount[name] = 0; m->allNATNetworks.addChild(aNATNetwork); if (aSaveSettings) { AutoWriteLock vboxLock(this COMMA_LOCKVAL_SRC_POS); rc = i_saveSettings(); vboxLock.release(); if (FAILED(rc)) i_unregisterNATNetwork(aNATNetwork, false /* aSaveSettings */); } return rc; #else NOREF(aNATNetwork); NOREF(aSaveSettings); /* No panic please (silently ignore) */ return S_OK; #endif } /** * Removes the given NAT network from the settings. * * @param aNATNetwork NAT network object to remove. * @param aSaveSettings @c true to save settings to disk (default). * * When @a aSaveSettings is @c true, this operation may fail because of the * failed #saveSettings() method it calls. In this case, the DHCP server * will NOT be removed from the settingsi when this method returns. * * @note Locks this object for writing. */ HRESULT VirtualBox::i_unregisterNATNetwork(NATNetwork *aNATNetwork, bool aSaveSettings /*= true*/) { #ifdef VBOX_WITH_NAT_SERVICE AssertReturn(aNATNetwork != NULL, E_INVALIDARG); AutoCaller autoCaller(this); AssertComRCReturn(autoCaller.rc(), autoCaller.rc()); AutoCaller natNetworkCaller(aNATNetwork); AssertComRCReturn(natNetworkCaller.rc(), natNetworkCaller.rc()); Bstr name; HRESULT rc = aNATNetwork->COMGETTER(NetworkName)(name.asOutParam()); /* Hm, there're still running clients. */ if (FAILED(rc) || sNatNetworkNameToRefCount[name]) AssertComRCReturnRC(E_INVALIDARG); m->allNATNetworks.removeChild(aNATNetwork); if (aSaveSettings) { AutoWriteLock vboxLock(this COMMA_LOCKVAL_SRC_POS); rc = i_saveSettings(); vboxLock.release(); if (FAILED(rc)) i_registerNATNetwork(aNATNetwork, false /* aSaveSettings */); } return rc; #else NOREF(aNATNetwork); NOREF(aSaveSettings); return E_NOTIMPL; #endif } /* vi: set tabstop=4 shiftwidth=4 expandtab: */