Changeset 25573 in vbox for trunk/src/VBox/Frontends/VBoxHeadless

Timestamp:

Dec 22, 2009 4:30:25 PM (15 years ago)

Author:

vboxsync

svn:sync-xref-src-repo-rev:

56251

Message:

VBoxHeadless: ExtraData option to terminate RDP connection if the user logs out of the guest OS (xTracker 4294)

File:

• trunk/src/VBox/Frontends/VBoxHeadless/VBoxHeadless.cpp (modified) (4 diffs)

trunk/src/VBox/Frontends/VBoxHeadless/VBoxHeadless.cpp

@@ -116 +116 @@
         refcnt = 0;
 #endif
+        mfNoLoggedInUsers = true;
     }
 
@@ -197 +198 @@
     STDMETHOD(OnGuestPropertyChange)(IN_BSTR machineId, IN_BSTR key, IN_BSTR value, IN_BSTR flags)
     {
+#ifdef VBOX_WITH_GUEST_PROPS
+        Utf8Str utf8Key = key;
+        if (utf8Key == "/VirtualBox/GuestInfo/OS/NoLoggedInUsers")
+        {
+            /* Check if the "disconnect on logout feature" is enabled. */
+            BOOL fDisconnectOnGuestLogout = FALSE;
+            ComPtr <IMachine> machine;
+            HRESULT hrc = S_OK;
+
+            if (gConsole)
+            {
+                hrc = gConsole->COMGETTER(Machine)(machine.asOutParam());
+                if (SUCCEEDED(hrc) && machine)
+                {
+                    Bstr value;
+                    hrc = machine->GetExtraData(Bstr("VRDP/DisconnectOnGuestLogout"), value.asOutParam());
+                    if (SUCCEEDED(hrc) && value == "1")
+                    {
+                        fDisconnectOnGuestLogout = TRUE;
+                    }
+                }
+            }
+
+            if (fDisconnectOnGuestLogout)
+            {
+                Utf8Str utf8Value = value;
+                if (utf8Value == "true")
+                {
+                    if (!mfNoLoggedInUsers) /* Only if the property really changes. */
+                    {
+                        mfNoLoggedInUsers = true;
+
+                        /* If there is a VRDP connection, drop it. */
+                        ComPtr<IRemoteDisplayInfo> info;
+                        hrc = gConsole->COMGETTER(RemoteDisplayInfo)(info.asOutParam());
+                        if (SUCCEEDED(hrc) && info)
+                        {
+                            ULONG cClients = 0;
+                            hrc = info->COMGETTER(NumberOfClients)(&cClients);
+                            if (SUCCEEDED(hrc) && cClients > 0)
+                            {
+                                ComPtr <IVRDPServer> vrdpServer;
+                                hrc = machine->COMGETTER(VRDPServer)(vrdpServer.asOutParam());
+                                if (SUCCEEDED(hrc) && vrdpServer)
+                                {
+                                    vrdpServer->COMSETTER(Enabled)(FALSE);
+                                    vrdpServer->COMSETTER(Enabled)(TRUE);
+                                }
+                            }
+                        }
+                    }
+                }
+                else
+                {
+                    mfNoLoggedInUsers = false;
+                }
+            }
+        }
+#endif /* VBOX_WITH_GUEST_PROPS */
         return S_OK;
     }
@@ -205 +265 @@
 #endif
 
+    bool mfNoLoggedInUsers;
 };
 
@@ -1077 +1138 @@
     while (0);
 
+    /* No more access to the 'console' object, which will be uninitialized by the next session->Close call. */
+    gConsole = NULL;
+
     if (fSessionOpened)
     {