Extend UnixDomainSocket::RecvMsg to return sender's PID

base/posix/unix_domain_socket_linux.h

 // Copyright (c) 2011 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #ifndef BASE_POSIX_UNIX_DOMAIN_SOCKET_LINUX_H_
 #define BASE_POSIX_UNIX_DOMAIN_SOCKET_LINUX_H_
 
 #include <stdint.h>
 #include <sys/types.h>
 #include <vector>
 
 #include "base/base_export.h"
+#include "base/process/process_handle.h"
 
 class Pickle;
 
 class BASE_EXPORT UnixDomainSocket {
  public:
   // Maximum number of file descriptors that can be read by RecvMsg().
   static const size_t kMaxFileDescriptors;
 
+  // Use to enable receiving pids in RecvMsg.  Returns true if successful.
+  static bool EnableReceiveProcessId(int fd);

[jln (very slow on Chromium), 2014/04/24 18:32:58]

Perhaps make it clear that this should be used on the receiving end?

[mdempsky, 2014/04/24 19:00:34]

Done.

+
   // Use sendmsg to write the given msg and include a vector of file
   // descriptors. Returns true if successful.
   static bool SendMsg(int fd,
                       const void* msg,
                       size_t length,
                       const std::vector<int>& fds);
 
   // Use recvmsg to read a message and an array of file descriptors. Returns
   // -1 on failure. Note: will read, at most, |kMaxFileDescriptors| descriptors.
   static ssize_t RecvMsg(int fd,
                          void* msg,
                          size_t length,
                          std::vector<int>* fds);
 
+  // Same as RecvMsg above, but also returns the sender's PID (as seen from
+  // the caller's namespace).  However, before using this function to receive
+  // PIDs, EnableReceiveProcessId() should be called on the socket.
+  static ssize_t RecvMsgWithPid(int fd,
+                                void* msg,
+                                size_t length,
+                                std::vector<int>* fds,
+                                base::ProcessId* pid);
+
   // Perform a sendmsg/recvmsg pair.
   //   1. This process creates a UNIX SEQPACKET socketpair. Using
   //      connection-oriented sockets (SEQPACKET or STREAM) is critical here,
   //      because if one of the ends closes the other one must be notified.
   //   2. This process writes a request to |fd| with an SCM_RIGHTS control
   //      message containing on end of the fresh socket pair.
   //   3. This process blocks reading from the other end of the fresh
   //      socketpair.
   //   4. The target process receives the request, processes it and writes the
   //      reply to the end of the socketpair contained in the request.
@@ skipping 18 matching lines @@
                                       unsigned reply_len,
                                       int recvmsg_flags,
                                       int* result_fd,
                                       const Pickle& request);
  private:
   // Similar to RecvMsg, but allows to specify |flags| for recvmsg(2).
   static ssize_t RecvMsgWithFlags(int fd,
                                   void* msg,
                                   size_t length,
                                   int flags,
-                                  std::vector<int>* fds);
+                                  std::vector<int>* fds,
+                                  base::ProcessId* pid);

[jln (very slow on Chromium), 2014/04/24 18:32:58]

You didn't have to modify any caller? I was pretty sure RecvMsgWithFlags was
used (I think I added it).

[mdempsky, 2014/04/24 19:00:34]

It has 'private' visibility, so it's only used within
unix_domain_socket_linux.cc, and I fixed up the one caller in there (in
SendRecvMsgWithFlags).

 };
 
 #endif  // BASE_POSIX_UNIX_DOMAIN_SOCKET_LINUX_H_