Changeset 36446 in vbox for trunk/src/VBox/Additions/WINNT

Timestamp:

Mar 28, 2011 9:42:12 AM (14 years ago)

Author:

vboxsync

Message:

VBoxGINA: Implemented support for handling/ignoring remote (desktop) sessions.

Location:

trunk/src/VBox/Additions/WINNT/VBoxGINA

Files:

• Helper.cpp (modified) (6 diffs)
• Helper.h (modified) (2 diffs)
• VBoxGINA.cpp (modified) (2 diffs)

trunk/src/VBox/Additions/WINNT/VBoxGINA/Helper.cpp

@@ -5 +5 @@
 
 /*
- * Copyright (C) 2006-2010 Oracle Corporation
+ * Copyright (C) 2006-2011 Oracle Corporation
  *
  * This file is part of VirtualBox Open Source Edition (OSE), as
@@ -24 +24 @@
 #include <VBox/VMMDev.h>
 #include <iprt/string.h>
+
+/* remote session handling */
+DWORD g_dwHandleRemoteSessions = 0;
 
 /* the credentials */
@@ -49 +52 @@
 }
 
+/**
+ * Detects whether our process is running in a remote session or not.
+ *
+ * @return  bool        true if running in a remote session, false if not.
+ */
+bool isRemoteSession(void)
+{
+    return (0 != GetSystemMetrics(SM_REMOTESESSION)) ? true : false;
+}
+
+/**
+ * Loads the global configuration from registry.
+ *
+ * @return  DWORD       Windows error code.
+ */
+DWORD loadConfiguration(void)
+{
+    HKEY hKey;
+    /** @todo Add some registry wrapper function(s) as soon as we got more values to retrieve. */
+    DWORD dwRet = RegOpenKeyEx(HKEY_LOCAL_MACHINE, L"SOFTWARE\\Oracle\\VirtualBox Guest Additions\\AutoLogon",
+                               0L, KEY_QUERY_VALUE, &hKey);
+    if (dwRet == ERROR_SUCCESS)
+    {
+        DWORD dwValue;
+        DWORD dwType = REG_DWORD;
+        DWORD dwSize = sizeof(DWORD);
+
+        dwRet = RegQueryValueEx(hKey, L"HandleRemoteSessions", NULL, &dwType, (LPBYTE)&dwValue, &dwSize);
+        if (   dwRet  == ERROR_SUCCESS
+            && dwType == REG_DWORD
+            && dwSize == sizeof(DWORD))
+        {
+            g_dwHandleRemoteSessions = dwValue;
+        }
+        RegCloseKey(hKey);
+    }
+    return dwRet;
+}
+
+/**
+ * Determines whether we should handle the current session or not.
+ *
+ * @return  bool        true if we should handle this session, false if not.
+ */
+bool handleCurrentSession(void)
+{
+    bool fHandle = false;
+    if (isRemoteSession())
+    {
+        if (g_dwHandleRemoteSessions) /* Force remote session handling. */
+            fHandle = true;
+    }
+    else /* No remote session. */
+        fHandle = true;
+    return fHandle;
+}
+
 void credentialsReset(void)
 {
@@ -58 +118 @@
 bool credentialsAvailable(void)
 {
+    if (!handleCurrentSession())
+        return false;
+
     HANDLE vboxDriver = getVBoxDriver();
     if (vboxDriver ==  INVALID_HANDLE_VALUE)
@@ -80 +143 @@
 bool credentialsRetrieve(void)
 {
+    if (!handleCurrentSession())
+        return false;
+
     Log(("VBoxGINA::credentialsRetrieve\n"));
 
@@ -161 +227 @@
 bool credentialsPollerCreate(void)
 {
+    if (!handleCurrentSession())
+        return false;
+
     Log(("VBoxGINA::credentialsPollerCreate\n"));
 

trunk/src/VBox/Additions/WINNT/VBoxGINA/Helper.h

@@ -4 +4 @@
 
 /*
- * Copyright (C) 2006-2010 Oracle Corporation
+ * Copyright (C) 2006-2011 Oracle Corporation
  *
  * This file is part of VirtualBox Open Source Edition (OSE), as
@@ -26 +26 @@
 extern wchar_t g_Domain[];
 
+DWORD loadConfiguration();
+bool handleCurrentSession(void);
+
 void credentialsReset(void);
 bool credentialsAvailable(void);

trunk/src/VBox/Additions/WINNT/VBoxGINA/VBoxGINA.cpp

@@ -112 +112 @@
 #endif
 
-    Log(("VBoxGINA::WlxNegotiate: dwWinlogonVersion: %d\n", dwWinlogonVersion));
-
-    /* load the standard Microsoft GINA DLL */
+    Log(("VBoxGINA::WlxNegotiate: dwWinlogonVersion: %ld\n", dwWinlogonVersion));
+
+    /* Load global configuration from registry. */
+    DWORD dwRet = loadConfiguration();
+    if (ERROR_SUCCESS != dwRet)
+        Log(("VBoxGINA: Error loading global configuration, error=%ld\n", dwRet));
+
+    /* If we have a remote session (that is, a connection via remote desktop /
+     * terminal services) deny it if not specified explicitly. */
+    if (!handleCurrentSession())
+        Log(("VBoxGINA: Handling of remote desktop sessions is disabled.\n"));
+
+    /* Load the standard Microsoft GINA DLL. */
     if (!(hDll = LoadLibrary(TEXT("MSGINA.DLL"))))
     {
-        Log(("VBoxGINA::WlxNegotiate: failed loading MSGINA! last error = %d\n", GetLastError()));
+        Log(("VBoxGINA::WlxNegotiate: failed loading MSGINA! Last error=%ld\n", GetLastError()));
         return FALSE;
     }
@@ -444 +454 @@
  * GINA 1.3 entry points
  */
-BOOL WINAPI WlxNetworkProviderLoad (PVOID pWlxContext, PWLX_MPR_NOTIFY_INFO pNprNotifyInfo)
+BOOL WINAPI WlxNetworkProviderLoad(PVOID pWlxContext, PWLX_MPR_NOTIFY_INFO pNprNotifyInfo)
 {
     Log(("VBoxGINA::WlxNetworkProviderLoad\n"));