Changeset 41919 in vbox for trunk/src/VBox/Runtime/r3/win

Timestamp:

Jun 27, 2012 8:06:27 AM (13 years ago)

Author:

vboxsync

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

78783

Message:

Runtime/process: Add RTProcQueryUsername to query the username of the current process

File:

• trunk/src/VBox/Runtime/r3/win/process-win.cpp (modified) (2 diffs)

trunk/src/VBox/Runtime/r3/win/process-win.cpp

@@ -38 +38 @@
 #include <errno.h>
 #include <Strsafe.h>
+#include <Lmcons.h>
 
 #include <iprt/process.h>
@@ -1424 +1425 @@
 }
 
+
+RTR3DECL(int) RTProcQueryUsername(RTPROCESS hProcess, char *pszUser, size_t cbUser,
+                                  size_t *pcbUser)
+{
+    AssertPtrReturn(pszUser, VERR_INVALID_POINTER);
+    AssertReturn(cbUser > 0, VERR_INVALID_PARAMETER);
+    AssertPtrReturn(pcbUser, VERR_INVALID_POINTER);
+
+    if (hProcess != RTProcSelf())
+        return VERR_NOT_SUPPORTED;
+
+    RTUTF16 awszUserName[UNLEN + 1];
+    DWORD   cchUserName = UNLEN + 1;
+
+    if (!GetUserNameW(&awszUserName[0], &cchUserName))
+        return RTErrConvertFromWin32(GetLastError());
+
+    char *pszUserName = NULL;
+    int rc = RTUtf16ToUtf8(awszUserName, &pszUserName);
+    if (RT_SUCCESS(rc))
+    {
+        size_t cbUserName = strlen(pszUserName) + 1;
+
+        *pcbUser = cbUserName;
+
+        if (cbUserName > cbUser)
+            rc = VERR_BUFFER_OVERFLOW;
+        else
+        {
+            memcpy(pszUser, pszUserName, cbUserName);
+            rc = VINF_SUCCESS;
+        }
+
+        RTStrFree(pszUserName);
+    }
+
+    return rc;
+}
+