1// Copyright 2014 The Chromium Authors. All rights reserved. 2// Use of this source code is governed by a BSD-style license that can be 3// found in the LICENSE file. 4 5#include "sandbox/linux/services/proc_util.h" 6 7#include <fcntl.h> 8#include <unistd.h> 9 10#include "base/files/scoped_file.h" 11#include "base/posix/eintr_wrapper.h" 12#include "testing/gtest/include/gtest/gtest.h" 13 14namespace sandbox { 15 16TEST(ProcUtil, CountOpenFds) { 17 base::ScopedFD proc_fd(open("/proc/", O_RDONLY | O_DIRECTORY)); 18 ASSERT_TRUE(proc_fd.is_valid()); 19 int fd_count = ProcUtil::CountOpenFds(proc_fd.get()); 20 int fd = open("/dev/null", O_RDONLY); 21 ASSERT_LE(0, fd); 22 EXPECT_EQ(fd_count + 1, ProcUtil::CountOpenFds(proc_fd.get())); 23 ASSERT_EQ(0, IGNORE_EINTR(close(fd))); 24 EXPECT_EQ(fd_count, ProcUtil::CountOpenFds(proc_fd.get())); 25} 26 27TEST(ProcUtil, HasOpenDirectory) { 28 // No open directory should exist at startup. 29 EXPECT_FALSE(ProcUtil::HasOpenDirectory()); 30 { 31 // Have a "/proc" file descriptor around. 32 int proc_fd = open("/proc/", O_RDONLY | O_DIRECTORY); 33 base::ScopedFD proc_fd_closer(proc_fd); 34 EXPECT_TRUE(ProcUtil::HasOpenDirectory()); 35 } 36 EXPECT_FALSE(ProcUtil::HasOpenDirectory()); 37} 38 39TEST(ProcUtil, HasOpenDirectoryWithFD) { 40 int proc_fd = open("/proc/", O_RDONLY | O_DIRECTORY); 41 base::ScopedFD proc_fd_closer(proc_fd); 42 ASSERT_LE(0, proc_fd); 43 44 // Don't pass |proc_fd|, an open directory (proc_fd) should 45 // be detected. 46 EXPECT_TRUE(ProcUtil::HasOpenDirectory()); 47 // Pass |proc_fd| and no open directory should be detected. 48 EXPECT_FALSE(ProcUtil::HasOpenDirectory(proc_fd)); 49 50 { 51 // Have a directory file descriptor around. 52 int open_directory_fd = open("/proc/self/", O_RDONLY | O_DIRECTORY); 53 base::ScopedFD open_directory_fd_closer(open_directory_fd); 54 EXPECT_TRUE(ProcUtil::HasOpenDirectory(proc_fd)); 55 } 56 57 // The "/proc/" file descriptor should now be closed, |proc_fd| is the 58 // only directory file descriptor open. 59 EXPECT_FALSE(ProcUtil::HasOpenDirectory(proc_fd)); 60} 61 62} // namespace sandbox 63