OLD | NEW |
(Empty) | |
| 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 |
| 14 namespace sandbox { |
| 15 |
| 16 TEST(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 |
| 27 TEST(ProcUtil, HasOpenDirectory) { |
| 28 // No open directory should exist at startup. |
| 29 EXPECT_FALSE(ProcUtil::HasOpenDirectory(-1)); |
| 30 { |
| 31 // Have a "/dev" file descriptor around. |
| 32 int dev_fd = open("/dev", O_RDONLY | O_DIRECTORY); |
| 33 base::ScopedFD dev_fd_closer(dev_fd); |
| 34 EXPECT_TRUE(ProcUtil::HasOpenDirectory(-1)); |
| 35 } |
| 36 EXPECT_FALSE(ProcUtil::HasOpenDirectory(-1)); |
| 37 } |
| 38 |
| 39 TEST(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(-1)); |
| 47 // Pass |proc_fd| and no open directory should be detected. |
| 48 EXPECT_FALSE(ProcUtil::HasOpenDirectory(proc_fd)); |
| 49 |
| 50 { |
| 51 // Have a "/dev" file descriptor around. |
| 52 int dev_fd = open("/dev", O_RDONLY | O_DIRECTORY); |
| 53 base::ScopedFD dev_fd_closer(dev_fd); |
| 54 EXPECT_TRUE(ProcUtil::HasOpenDirectory(proc_fd)); |
| 55 } |
| 56 |
| 57 // The "/dev" file descriptor should now be closed, |proc_fd| is the only |
| 58 // directory file descriptor open. |
| 59 EXPECT_FALSE(ProcUtil::HasOpenDirectory(proc_fd)); |
| 60 } |
| 61 |
| 62 } // namespace sandbox |
OLD | NEW |