Changeset 106880 in vbox for trunk/src/VBox/Main/src-server

Timestamp:

Nov 8, 2024 8:36:27 AM (2 months ago)

Author:

vboxsync

Message:

Added implementations of getTrackedObject and getTrackedObjectIds. Reshuffle lit bit init() and uninit() for the correct build.

File:

• trunk/src/VBox/Main/src-server/VirtualBoxImpl.cpp (modified) (10 diffs)

trunk/src/VBox/Main/src-server/VirtualBoxImpl.cpp

@@ -109 +109 @@
 #include "VBoxEvents.h"
 
+#include "ObjectsTracker.h"
+
 #include <QMTranslator.h>
 
@@ -130 +132 @@
 //
 ////////////////////////////////////////////////////////////////////////////////
+
+extern TrackedObjectsCollector gTrackedObjectsCollector;
 
 // static
@@ -265 +269 @@
 typedef std::map<RTPROCESS, WatchedClientProcess *> WatchedClientProcessMap;
 #endif
-
 
 typedef ObjectsList<Medium> MediaOList;
@@ -331 +334 @@
         , cRefsCrypto(0)
         , pCryptoIf(NULL)
+        , objectTrackerTask(NULL)
     {
 #if defined(RT_OS_WINDOWS) && defined(VBOXSVC_WITH_CLIENT_WATCHER)
@@ -468 +472 @@
     RTCRITSECT                          CritSectModCrypto;
     /** @} */
+
+    /** The tracked object collector (better if it'll be a singleton) */
+    ObjectTracker *objectTrackerTask;
 };
 
@@ -561 +568 @@
 
     LogFlowThisFunc(("Version: %s, Package: %s, API Version: %s\n", sVersion.c_str(), sPackageType.c_str(), sAPIVersion.c_str()));
+
+    /* Try to start Object tracker thread as earlier as possible */
+    {
+        int vrc = 0;
+        if(gTrackedObjectsCollector.init())
+        {
+            LogRel(("Starting the Object tracker thread\n"));
+            try
+            {
+                m->objectTrackerTask = new ObjectTracker();
+                if (!m->objectTrackerTask->init()) // some init procedure
+                    vrc = E_FAIL;
+                else
+                    vrc = m->objectTrackerTask->createThread();
+            }
+            catch (...)
+            {
+                LogRel(("Exception during starting the Object tracker thread\n"));
+                if (m->objectTrackerTask)
+                {
+                    delete m->objectTrackerTask;
+                    m->objectTrackerTask = NULL;
+                }
+                vrc = E_FAIL;
+            }
+        }
+
+        if(RT_SUCCESS(vrc))
+            LogRel(("Successfully started the Object tracker thread\n"));
+        else
+            LogRel(("Failed to start the Object tracker thread (%Rrc)\n", vrc));
+    }
 
     /* Important: DO NOT USE any kind of "early return" (except the single
@@ -1075 +1114 @@
     else
         m->allMachines.uninitAll();
+
     m->allFloppyImages.uninitAll();
     m->allDVDImages.uninitAll();
     m->allHardDisks.uninitAll();
     m->allDHCPServers.uninitAll();
-
-    m->mapProgressOperations.clear();
-
     m->allGuestOSTypes.uninitAll();
 
@@ -1130 +1167 @@
 
     RTCritSectDelete(&m->CritSectModCrypto);
+
+    m->mapProgressOperations.clear();
+    /*
+     * Call gTrackedObjectsCollector uninitialization before ExtPackManager uninitialization!!!
+     * Otherwise, this results in an error when releasing resources (in ComPtr::cleanup).
+     */
+    if (m->objectTrackerTask)
+    {
+        LogRel(("BACKEND: Terminating the object tracker...\n"));
+        m->objectTrackerTask->finish();//set the termination flag in the thread
+        delete m->objectTrackerTask;//waiting the thread termination is going in the m_objectTrackerTask destructor
+        gTrackedObjectsCollector.uninit();
+    }
 
 #ifdef VBOX_WITH_EXTPACK
@@ -6389 +6439 @@
 }
 
-
 HRESULT VirtualBox::findProgressById(const com::Guid &aId,
                                      ComPtr<IProgress> &aProgressObject)
@@ -6410 +6459 @@
 }
 
+HRESULT VirtualBox::getTrackedObject (const com::Utf8Str& aTrObjId,
+                                      ComPtr<IUnknown> &aPIface)
+{
+    TrackedObjectData trObjData;
+    HRESULT hrc = gTrackedObjectsCollector.getObj(aTrObjId, trObjData);
+    if (SUCCEEDED(hrc))
+        trObjData.getInterface().queryInterfaceTo(aPIface.asOutParam());
+
+    return hrc;
+}
+
+
+std::map <com::Utf8Str, com::Utf8Str> lMapInterfaceNameToIID = {
+    {"IProgress", Guid(IID_IProgress).toString()},
+    {"ISession", Guid(IID_ISession).toString()},
+    {"IMedium", Guid(IID_IMedium).toString()},
+    {"IMachine", Guid(IID_IMachine).toString()}
+};
+
+/**
+ * Get the tracked object Ids list by the interface name.
+ *
+ * @param aName  Interface name (like "IProgress", "ISession", "IMedium" etc)
+ * @param aObjIdsList  The list of Ids of the found objects
+ *
+ * @return The list of the found objects Ids
+ */
+HRESULT VirtualBox::getTrackedObjectIds (const com::Utf8Str& aName,
+                                         std::vector<com::Utf8Str> &aObjIdsList)
+{
+    HRESULT hrc = S_OK;
+
+    try
+    {
+        /* Check the supported tracked classes to avoid "out of range" exception */
+        if (lMapInterfaceNameToIID.count(aName))
+        {
+            hrc = gTrackedObjectsCollector.getObjIdsByClassIID(lMapInterfaceNameToIID.at(aName), aObjIdsList);
+            if (FAILED(hrc))
+                hrc = setError(VBOX_E_OBJECT_NOT_FOUND, tr("No objects were found for the passed interface name '%s'."),
+                               aName.c_str());
+        }
+        else
+            hrc = setError(E_INVALIDARG, tr("The objects of the passed interface '%s' are not tracked at moment."), aName.c_str());
+    }
+    catch (...)
+    {
+            hrc = setError(E_FAIL, tr("The unknown exception in the VirtualBox::getTrackedObjectIds()."));
+    }
+
+    return hrc;
+}
 
 /**