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

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/posix/process-posix.cpp (modified) (2 diffs)

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

@@ -38 +38 @@
 #include <sys/wait.h>
 #include <signal.h>
+#include <pwd.h>
 
 #include <iprt/process.h>
@@ -146 +147 @@
 }
 
+
+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;
+
+    int32_t cbPwdMax = sysconf(_SC_GETPW_R_SIZE_MAX);
+    if (cbPwdMax == -1)
+        return RTErrConvertFromErrno(errno);
+
+    char *pbBuf = (char *)RTMemAllocZ(cbPwdMax);
+    if (!pbBuf)
+        return VERR_NO_MEMORY;
+
+    struct passwd Pwd, *pPwd;
+    int rc = getpwuid_r(geteuid(), &Pwd, pbBuf, cbPwdMax, &pPwd);
+    if (!rc)
+    {
+        size_t cbPwdUser = strlen(pPwd->pw_name) + 1;
+
+        *pcbUser = cbPwdUser;
+
+        if (cbPwdUser > cbUser)
+            rc = VERR_BUFFER_OVERFLOW;
+        else
+        {
+            memcpy(pszUser, pPwd->pw_name, cbPwdUser);
+            rc = VINF_SUCCESS;
+        }
+    }
+    else
+        rc = RTErrConvertFromErrno(rc);
+
+    RTMemFree(pbBuf);
+    return rc;
+}
+
+