Add unit test to check for broker FD leak

sandbox/linux/services/broker_process_unittest.cc

 // Copyright (c) 2012 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.
 
 #include "sandbox/linux/services/broker_process.h"
 
 #include <errno.h>
 #include <fcntl.h>
 #include <sys/stat.h>
 #include <sys/types.h>
 #include <sys/wait.h>
 #include <unistd.h>
 
 #include <string>
 #include <vector>
 
 #include "base/basictypes.h"
 #include "base/bind.h"
 #include "base/file_util.h"
 #include "base/files/scoped_file.h"
 #include "base/logging.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/posix/eintr_wrapper.h"
+#include "base/posix/unix_domain_socket_linux.h"
 #include "sandbox/linux/tests/test_utils.h"
 #include "sandbox/linux/tests/unit_tests.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace sandbox {
 
 namespace {
 
 // Creates and open a temporary file on creation and closes
 // and removes it on destruction.
@@ skipping 393 matching lines @@
   // Don't do anything here, so that ASSERT works in the subfunction as
   // expected.
 }
 
 TEST(BrokerProcess, OpenComplexFlagsNoClientCheck) {
   TestOpenComplexFlags(false /* fast_check_in_client */);
   // Don't do anything here, so that ASSERT works in the subfunction as
   // expected.
 }
 
+SANDBOX_DEATH_TEST(BrokerProcess, RecvMsgDescriptorLeak, DEATH_EXIT_CODE(0)) {
+  // Find the lowest available file descriptor currently.
+  const int min_fd = dup(STDIN_FILENO);
+  SANDBOX_ASSERT(min_fd >= 0);
+  SANDBOX_ASSERT(0 == IGNORE_EINTR(close(min_fd)));
+
+  // Lower our file descriptor limit to just above the current limit so we can
+  // test for descriptor leaks easier.
+  const unsigned kExtraFiles = 8;
+  const struct rlimit new_rlim = {min_fd + kExtraFiles, min_fd + kExtraFiles};
+  SANDBOX_ASSERT(0 == setrlimit(RLIMIT_NOFILE, &new_rlim));
+
+  const char kCpuInfo[] = "/proc/cpuinfo";
+  std::vector<std::string> read_whitelist;
+  read_whitelist.push_back(kCpuInfo);
+
+  BrokerProcess open_broker(EPERM, read_whitelist, std::vector<std::string>());
+  SANDBOX_ASSERT(open_broker.Init(base::Bind(&NoOpCallback)));
+
+  static const char kBogus[] = "not a pickle";
+  const std::vector<int> fds(1, STDIN_FILENO);
+
+  for (unsigned i = 0; i < kExtraFiles; ++i) {
+    SANDBOX_ASSERT(UnixDomainSocket::SendMsg(
+        open_broker.ipc_socketpair(), kBogus, sizeof(kBogus), fds));
+  }
+
+  const int fd = open_broker.Open(kCpuInfo, O_RDONLY);
+  SANDBOX_ASSERT(fd >= 0);
+  SANDBOX_ASSERT(0 == IGNORE_EINTR(close(fd)));
+  _exit(0);
+}
+
 }  // namespace sandbox