Changeset 47232 in vbox for trunk/src/VBox/Additions/common/VBoxService

Timestamp:

Jul 18, 2013 12:00:49 PM (12 years ago)

Author:

vboxsync

Message:

Windows Guest Additions: Support for querying user idle times.

File:

• trunk/src/VBox/Additions/common/VBoxService/VBoxServiceVMInfo-win.cpp (modified) (5 diffs)

trunk/src/VBox/Additions/common/VBoxService/VBoxServiceVMInfo-win.cpp

@@ -31 +31 @@
 #include <iprt/assert.h>
 #include <iprt/ldr.h>
+#include <iprt/localipc.h>
 #include <iprt/mem.h>
 #include <iprt/thread.h>
@@ -40 +41 @@
 #include "VBoxServiceInternal.h"
 #include "VBoxServiceUtils.h"
+#include "../../WINNT/VBoxTray/VBoxTrayMsg.h" /* For IPC. */
 
 static uint32_t s_uGuestPropClientID = 0;
@@ -90 +92 @@
 int  VBoxServiceVMInfoWinProcessesEnumerate(PVBOXSERVICEVMINFOPROC *ppProc, DWORD *pdwCount);
 void VBoxServiceVMInfoWinProcessesFree(DWORD cProcs, PVBOXSERVICEVMINFOPROC paProcs);
+int vboxServiceVMInfoWinWriteLastInput(char *pszUser);
 
 typedef BOOL WINAPI FNQUERYFULLPROCESSIMAGENAME(HANDLE,  DWORD, LPTSTR, PDWORD);
@@ -805 +808 @@
 
 
+static int vboxServiceVMInfoWinWriteLastInput(char *pszUser)
+{
+    AssertPtrReturn(pszUser, VERR_INVALID_POINTER);
+
+    int rc = VINF_SUCCESS;
+
+    char szPipeName[255];
+    if (RTStrPrintf(szPipeName, sizeof(szPipeName), "%s%s",
+                    VBOXTRAY_IPC_PIPE_PREFIX, pszUser))
+    {
+        RTLOCALIPCSESSION hSession;
+        rc = RTLocalIpcSessionConnect(&hSession, szPipeName, 0 /* Flags */);
+        if (RT_SUCCESS(rc))
+        {
+            VBOXTRAYIPCHEADER ipcHdr = { 0 /* Header version */,
+                                         VBOXTRAYIPCMSGTYPE_USERLASTINPUT, 0 /* No msg */ };
+
+            rc = RTLocalIpcSessionWrite(hSession, &ipcHdr, sizeof(ipcHdr));
+
+            VBOXTRAYIPCRES_USERLASTINPUT ipcRes;
+            if (RT_SUCCESS(rc))
+                rc = RTLocalIpcSessionRead(hSession, &ipcRes, sizeof(ipcRes),
+                                           NULL /* Exact read */);
+            if (RT_SUCCESS(rc))
+            {
+                if (ipcRes.uLastInputMs)
+                    VBoxServiceVerbose(4, "User \"%s\" is idle for %RU32ms\n",
+                                       pszUser, ipcRes.uLastInputMs);
+            }
+#ifdef DEBUG
+            VBoxServiceVerbose(4, "Querying last input for user \"%s\" ended with rc=%Rrc\n",
+                               pszUser, rc);
+#endif
+            int rc2 = RTLocalIpcSessionClose(hSession);
+            if (RT_SUCCESS(rc))
+                rc = rc2;
+        }
+        else
+        {
+            switch (rc)
+            {
+                case VERR_FILE_NOT_FOUND:
+                    /* No VBoxTray (or too old version which does not support IPC) running
+                       for the given user. Not much we can do then. */
+                    VBoxServiceVerbose(4, "User \"%s\" not logged in, no last input available\n",
+                                       pszUser);
+                    break;
+
+                default:
+                    VBoxServiceVerbose(4, "Error querying last input for user \"%s\", rc=%Rrc\n",
+                                       pszUser, rc);
+                    break;
+            }
+
+            rc = VINF_SUCCESS;
+        }
+    }
+
+    return rc;
+}
+
+
 /**
  * Retrieves the currently logged in users and stores their names along with the
@@ -995 +1060 @@
                     *pcUsersInList += 1;
 
-                    char *pszTemp;
-                    int rc2 = RTUtf16ToUtf8(pUserInfo[i].wszUser, &pszTemp);
+                    char *pszUser;
+                    int rc2 = RTUtf16ToUtf8(pUserInfo[i].wszUser, &pszUser);
                     if (RT_SUCCESS(rc2))
                     {
-                        rc = RTStrAAppend(ppszUserList, pszTemp);
-                        RTMemFree(pszTemp);
+                        rc = RTStrAAppend(ppszUserList, pszUser);
+                        if (RT_SUCCESS(rc))
+                            rc = vboxServiceVMInfoWinWriteLastInput(pszUser);
+                        RTMemFree(pszUser);
                     }
                     else