| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2006-2009 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 // This test is POSIX only. | |
| 6 | |
| 7 #include <unistd.h> | |
| 8 #include <fcntl.h> | |
| 9 | |
| 10 #include "base/basictypes.h" | |
| 11 #include "base/eintr_wrapper.h" | |
| 12 #include "chrome/common/file_descriptor_set_posix.h" | |
| 13 #include "testing/gtest/include/gtest/gtest.h" | |
| 14 | |
| 15 namespace { | |
| 16 | |
| 17 // Get a safe file descriptor for test purposes. | |
| 18 int GetSafeFd() { | |
| 19 return open("/dev/null", O_RDONLY); | |
| 20 } | |
| 21 | |
| 22 // Returns true if fd was already closed. Closes fd if not closed. | |
| 23 bool VerifyClosed(int fd) { | |
| 24 const int duped = dup(fd); | |
| 25 if (duped != -1) { | |
| 26 HANDLE_EINTR(close(duped)); | |
| 27 HANDLE_EINTR(close(fd)); | |
| 28 return false; | |
| 29 } | |
| 30 return true; | |
| 31 } | |
| 32 | |
| 33 // The FileDescriptorSet will try and close some of the descriptor numbers | |
| 34 // which we given it. This is the base descriptor value. It's great enough such | |
| 35 // that no real descriptor will accidently be closed. | |
| 36 static const int kFDBase = 50000; | |
| 37 | |
| 38 TEST(FileDescriptorSet, BasicAdd) { | |
| 39 scoped_refptr<FileDescriptorSet> set = new FileDescriptorSet; | |
| 40 | |
| 41 ASSERT_EQ(set->size(), 0u); | |
| 42 ASSERT_TRUE(set->empty()); | |
| 43 ASSERT_TRUE(set->Add(kFDBase)); | |
| 44 ASSERT_EQ(set->size(), 1u); | |
| 45 ASSERT_TRUE(!set->empty()); | |
| 46 | |
| 47 // Empties the set and stops a warning about deleting a set with unconsumed | |
| 48 // descriptors | |
| 49 set->CommitAll(); | |
| 50 } | |
| 51 | |
| 52 TEST(FileDescriptorSet, BasicAddAndClose) { | |
| 53 scoped_refptr<FileDescriptorSet> set = new FileDescriptorSet; | |
| 54 | |
| 55 ASSERT_EQ(set->size(), 0u); | |
| 56 ASSERT_TRUE(set->empty()); | |
| 57 const int fd = GetSafeFd(); | |
| 58 ASSERT_TRUE(set->AddAndAutoClose(fd)); | |
| 59 ASSERT_EQ(set->size(), 1u); | |
| 60 ASSERT_TRUE(!set->empty()); | |
| 61 | |
| 62 set->CommitAll(); | |
| 63 | |
| 64 ASSERT_TRUE(VerifyClosed(fd)); | |
| 65 } | |
| 66 TEST(FileDescriptorSet, MaxSize) { | |
| 67 scoped_refptr<FileDescriptorSet> set = new FileDescriptorSet; | |
| 68 | |
| 69 for (unsigned i = 0; | |
| 70 i < FileDescriptorSet::MAX_DESCRIPTORS_PER_MESSAGE; ++i) { | |
| 71 ASSERT_TRUE(set->Add(kFDBase + 1 + i)); | |
| 72 } | |
| 73 | |
| 74 ASSERT_TRUE(!set->Add(kFDBase)); | |
| 75 | |
| 76 set->CommitAll(); | |
| 77 } | |
| 78 | |
| 79 TEST(FileDescriptorSet, SetDescriptors) { | |
| 80 scoped_refptr<FileDescriptorSet> set = new FileDescriptorSet; | |
| 81 | |
| 82 ASSERT_TRUE(set->empty()); | |
| 83 set->SetDescriptors(NULL, 0); | |
| 84 ASSERT_TRUE(set->empty()); | |
| 85 | |
| 86 const int fd = GetSafeFd(); | |
| 87 static const int fds[] = {fd}; | |
| 88 set->SetDescriptors(fds, 1); | |
| 89 ASSERT_TRUE(!set->empty()); | |
| 90 ASSERT_EQ(set->size(), 1u); | |
| 91 | |
| 92 set->CommitAll(); | |
| 93 | |
| 94 ASSERT_TRUE(VerifyClosed(fd)); | |
| 95 } | |
| 96 | |
| 97 TEST(FileDescriptorSet, GetDescriptors) { | |
| 98 scoped_refptr<FileDescriptorSet> set = new FileDescriptorSet; | |
| 99 | |
| 100 set->GetDescriptors(NULL); | |
| 101 ASSERT_TRUE(set->Add(kFDBase)); | |
| 102 | |
| 103 int fds[1]; | |
| 104 fds[0] = 0; | |
| 105 set->GetDescriptors(fds); | |
| 106 ASSERT_EQ(fds[0], kFDBase); | |
| 107 set->CommitAll(); | |
| 108 ASSERT_TRUE(set->empty()); | |
| 109 } | |
| 110 | |
| 111 TEST(FileDescriptorSet, WalkInOrder) { | |
| 112 scoped_refptr<FileDescriptorSet> set = new FileDescriptorSet; | |
| 113 | |
| 114 ASSERT_TRUE(set->Add(kFDBase)); | |
| 115 ASSERT_TRUE(set->Add(kFDBase + 1)); | |
| 116 ASSERT_TRUE(set->Add(kFDBase + 2)); | |
| 117 | |
| 118 ASSERT_EQ(set->GetDescriptorAt(0), kFDBase); | |
| 119 ASSERT_EQ(set->GetDescriptorAt(1), kFDBase + 1); | |
| 120 ASSERT_EQ(set->GetDescriptorAt(2), kFDBase + 2); | |
| 121 | |
| 122 set->CommitAll(); | |
| 123 } | |
| 124 | |
| 125 TEST(FileDescriptorSet, WalkWrongOrder) { | |
| 126 scoped_refptr<FileDescriptorSet> set = new FileDescriptorSet; | |
| 127 | |
| 128 ASSERT_TRUE(set->Add(kFDBase)); | |
| 129 ASSERT_TRUE(set->Add(kFDBase + 1)); | |
| 130 ASSERT_TRUE(set->Add(kFDBase + 2)); | |
| 131 | |
| 132 ASSERT_EQ(set->GetDescriptorAt(0), kFDBase); | |
| 133 ASSERT_EQ(set->GetDescriptorAt(2), -1); | |
| 134 | |
| 135 set->CommitAll(); | |
| 136 } | |
| 137 | |
| 138 TEST(FileDescriptorSet, WalkCycle) { | |
| 139 scoped_refptr<FileDescriptorSet> set = new FileDescriptorSet; | |
| 140 | |
| 141 ASSERT_TRUE(set->Add(kFDBase)); | |
| 142 ASSERT_TRUE(set->Add(kFDBase + 1)); | |
| 143 ASSERT_TRUE(set->Add(kFDBase + 2)); | |
| 144 | |
| 145 ASSERT_EQ(set->GetDescriptorAt(0), kFDBase); | |
| 146 ASSERT_EQ(set->GetDescriptorAt(1), kFDBase + 1); | |
| 147 ASSERT_EQ(set->GetDescriptorAt(2), kFDBase + 2); | |
| 148 ASSERT_EQ(set->GetDescriptorAt(0), kFDBase); | |
| 149 ASSERT_EQ(set->GetDescriptorAt(1), kFDBase + 1); | |
| 150 ASSERT_EQ(set->GetDescriptorAt(2), kFDBase + 2); | |
| 151 ASSERT_EQ(set->GetDescriptorAt(0), kFDBase); | |
| 152 ASSERT_EQ(set->GetDescriptorAt(1), kFDBase + 1); | |
| 153 ASSERT_EQ(set->GetDescriptorAt(2), kFDBase + 2); | |
| 154 | |
| 155 set->CommitAll(); | |
| 156 } | |
| 157 | |
| 158 TEST(FileDescriptorSet, DontClose) { | |
| 159 scoped_refptr<FileDescriptorSet> set = new FileDescriptorSet; | |
| 160 | |
| 161 const int fd = GetSafeFd(); | |
| 162 ASSERT_TRUE(set->Add(fd)); | |
| 163 set->CommitAll(); | |
| 164 | |
| 165 ASSERT_FALSE(VerifyClosed(fd)); | |
| 166 } | |
| 167 | |
| 168 TEST(FileDescriptorSet, DoClose) { | |
| 169 scoped_refptr<FileDescriptorSet> set = new FileDescriptorSet; | |
| 170 | |
| 171 const int fd = GetSafeFd(); | |
| 172 ASSERT_TRUE(set->AddAndAutoClose(fd)); | |
| 173 set->CommitAll(); | |
| 174 | |
| 175 ASSERT_TRUE(VerifyClosed(fd)); | |
| 176 } | |
| 177 | |
| 178 } // namespace | |
| OLD | NEW |