Changeset 92676 in vbox

Timestamp:

Dec 1, 2021 2:59:59 PM (3 years ago)

Author:

vboxsync

Message:

iprt: Local Ipc Server: add Linux implementation for getting remote peer credentials, bugref:10134.

File:

• trunk/src/VBox/Runtime/r3/posix/localipc-posix.cpp (modified) (2 diffs)

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

@@ -1048 +1048 @@
 }
 
+/**
+ * Get IPC session socket peer credentials.
+ *
+ * @return IPRT status code.
+ * @param hSession      IPC session handle.
+ * @param pProcess      Output buffer to store remote peer PID (can be NULL).
+ * @param pUid          Output buffer to store remote peer UID (can be NULL).
+ * @param pGid          Output buffer to store remote peer GID (can be NULL).
+ */
+static int rtLocalIpcSessionQueryUcred(RTLOCALIPCSESSION hSession, PRTPROCESS pProcess, PRTUID pUid, PRTGID pGid)
+{
+#if defined(RT_OS_LINUX)
+    PRTLOCALIPCSESSIONINT pThis = hSession;
+
+    AssertPtrReturn(pThis, VERR_INVALID_HANDLE);
+    AssertReturn(pThis->u32Magic == RTLOCALIPCSESSION_MAGIC, VERR_INVALID_HANDLE);
+
+    struct ucred PeerCred;
+    socklen_t cbPeerCred = sizeof(PeerCred);
+    int iSock = RTSocketToNative(pThis->hSocket);
+
+    RT_ZERO(PeerCred);
+
+    if (getsockopt(iSock, SOL_SOCKET, SO_PEERCRED, &PeerCred, &cbPeerCred) == -1)
+        return RTErrConvertFromErrno(errno);
+
+    if (pProcess)
+        *pProcess = PeerCred.pid;
+    if (pUid)
+        *pUid = PeerCred.uid;
+    if (pGid)
+        *pGid = PeerCred.gid;
+
+    return VINF_SUCCESS;
+#else
+    return VERR_NOT_SUPPORTED;
+#endif
+}
 
 RTDECL(int) RTLocalIpcSessionQueryProcess(RTLOCALIPCSESSION hSession, PRTPROCESS pProcess)
 {
-    RT_NOREF_PV(hSession); RT_NOREF_PV(pProcess);
-    return VERR_NOT_SUPPORTED;
+    return rtLocalIpcSessionQueryUcred(hSession, pProcess, NULL, NULL);
 }
 
@@ -1058 +1095 @@
 RTDECL(int) RTLocalIpcSessionQueryUserId(RTLOCALIPCSESSION hSession, PRTUID pUid)
 {
-    RT_NOREF_PV(hSession); RT_NOREF_PV(pUid);
-    return VERR_NOT_SUPPORTED;
-}
-
+    return rtLocalIpcSessionQueryUcred(hSession, NULL, pUid, NULL);
+}
 
 RTDECL(int) RTLocalIpcSessionQueryGroupId(RTLOCALIPCSESSION hSession, PRTGID pGid)
 {
-    RT_NOREF_PV(hSession); RT_NOREF_PV(pGid);
-    return VERR_NOT_SUPPORTED;
-}
-
+    return rtLocalIpcSessionQueryUcred(hSession, NULL, NULL, pGid);
+}
+