Changeset 42067 in vbox for trunk

Timestamp:

Jul 9, 2012 5:49:58 PM (13 years ago)

Author:

vboxsync

Message:

Runtime: New variant for RTProcQueryUsername which allocates the required string

Location:

trunk

Files:

• include/iprt/mangling.h (modified) (1 diff)
• include/iprt/process.h (modified) (1 diff)
• src/VBox/Runtime/Makefile.kmk (modified) (1 diff)
• src/VBox/Runtime/generic/RTProcessQueryUsernameA-generic.cpp (added)
• src/VBox/Runtime/testcase/tstRTProcQueryUsername.cpp (modified) (1 diff)

trunk/include/iprt/mangling.h

@@ -964 +964 @@
 # define RTProcIsRunningByName                          RT_MANGLER(RTProcIsRunningByName)
 # define RTProcQueryUsername                            RT_MANGLER(RTProcQueryUsername)
+# define RTProcQueryUsernameA                           RT_MANGLER(RTProcQueryUsernameA)
 # define RTProcSelf                                     RT_MANGLER(RTProcSelf)
 # define RTProcSetPriority                              RT_MANGLER(RTProcSetPriority)

trunk/include/iprt/process.h

@@ -379 +379 @@
                                     size_t *pcbUser);
 
+/**
+ * Query the username of the given process allocating the string for the username.
+ *
+ * @returns IPRT status code.
+ * @param   hProcess     The process handle to query the username for.
+ * @param   ppszUser     Where to store the pointer to the string containing
+ *                       the username on success. Free with RTStrFree().
+ */
+RTR3DECL(int)   RTProcQueryUsernameA(RTPROCESS hProcess, char **ppszUser);
+
 #endif /* IN_RING3 */
 

trunk/src/VBox/Runtime/Makefile.kmk

@@ -453 +453 @@
         generic/RTLogWriteStdOut-generic.cpp \
         generic/RTLogWriteUser-generic.cpp \
+        generic/RTProcessQUeryUsernameA-generic.cpp \
         generic/RTTimerLRCreate-generic.cpp \
         generic/mempool-generic.cpp \

trunk/src/VBox/Runtime/testcase/tstRTProcQueryUsername.cpp

@@ -39 +39 @@
     char abUser[1024];
     size_t cbUser;
+    char *pszUser = NULL;
+
     RTTestISub("Basics");
 
     memset(abUser, 0, sizeof(abUser));
-    RTTESTI_CHECK_RC(RTProcQueryUsername(RTProcSelf(), NULL, 8, &cbUser), VERR_INVALID_POINTER);
+    RTTESTI_CHECK_RC(RTProcQueryUsername(RTProcSelf(), NULL, 8, &cbUser), VERR_INVALID_PARAMETER);
     RTTESTI_CHECK_RC(RTProcQueryUsername(RTProcSelf(), abUser, 0, &cbUser), VERR_INVALID_PARAMETER);
-    RTTESTI_CHECK_RC(RTProcQueryUsername(RTProcSelf(), abUser, sizeof(abUser), NULL), VERR_INVALID_POINTER);
+    RTTESTI_CHECK_RC(RTProcQueryUsername(RTProcSelf(), NULL, 0, NULL), VERR_BUFFER_OVERFLOW);
+    RTTESTI_CHECK_RC(RTProcQueryUsername(RTProcSelf(), NULL, 0, &cbUser), VERR_BUFFER_OVERFLOW);
 
     RTTESTI_CHECK_RC(RTProcQueryUsername(RTProcSelf(), abUser, sizeof(abUser), &cbUser), VINF_SUCCESS);
     RTTestPrintf(NULL, RTTESTLVL_ALWAYS, "Username: %s\n", abUser);
     RTTESTI_CHECK_RC(RTProcQueryUsername(RTProcSelf(), abUser, cbUser - 1, &cbUser), VERR_BUFFER_OVERFLOW);
+
+    RTTESTI_CHECK_RC(RTProcQueryUsernameA(RTProcSelf(), NULL), VERR_INVALID_POINTER);
+    RTTESTI_CHECK_RC(RTProcQueryUsernameA(RTProcSelf(), &pszUser), VINF_SUCCESS);
+    RTTestPrintf(NULL, RTTESTLVL_ALWAYS, "Username: %s\n", pszUser);
+    RTStrFree(pszUser);
 }