OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "mojo/edk/embedder/platform_channel_pair.h" | 5 #include "mojo/edk/embedder/platform_channel_pair.h" |
6 | 6 |
7 #include <errno.h> | 7 #include <errno.h> |
8 #include <poll.h> | 8 #include <poll.h> |
9 #include <signal.h> | 9 #include <signal.h> |
10 #include <stdio.h> | 10 #include <stdio.h> |
11 #include <sys/socket.h> | 11 #include <sys/socket.h> |
12 #include <sys/types.h> | 12 #include <sys/types.h> |
13 #include <sys/uio.h> | 13 #include <sys/uio.h> |
14 #include <unistd.h> | 14 #include <unistd.h> |
15 | 15 |
16 #include <deque> | 16 #include <deque> |
17 | 17 |
18 #include "base/logging.h" | 18 #include "base/logging.h" |
19 #include "build/build_config.h" | 19 #include "build/build_config.h" |
20 #include "mojo/edk/embedder/platform_channel_utils_posix.h" | 20 #include "mojo/edk/embedder/platform_channel_utils_posix.h" |
21 #include "mojo/edk/embedder/platform_handle.h" | 21 #include "mojo/edk/embedder/platform_handle.h" |
22 #include "mojo/edk/embedder/platform_handle_vector.h" | 22 #include "mojo/edk/embedder/platform_handle_vector.h" |
23 #include "mojo/edk/embedder/scoped_platform_handle.h" | 23 #include "mojo/edk/embedder/scoped_platform_handle.h" |
| 24 #include "mojo/edk/test/scoped_test_dir.h" |
24 #include "mojo/edk/test/test_utils.h" | 25 #include "mojo/edk/test/test_utils.h" |
25 #include "mojo/edk/util/scoped_file.h" | 26 #include "mojo/edk/util/scoped_file.h" |
26 #include "mojo/public/cpp/system/macros.h" | 27 #include "mojo/public/cpp/system/macros.h" |
27 #include "testing/gtest/include/gtest/gtest.h" | 28 #include "testing/gtest/include/gtest/gtest.h" |
28 | 29 |
29 #if defined(OS_ANDROID) | |
30 #include "base/android/path_utils.h" | |
31 #include "base/files/file_path.h" | |
32 #endif | |
33 | |
34 namespace mojo { | 30 namespace mojo { |
35 namespace embedder { | 31 namespace embedder { |
36 namespace { | 32 namespace { |
37 | 33 |
38 void WaitReadable(PlatformHandle h) { | 34 void WaitReadable(PlatformHandle h) { |
39 struct pollfd pfds = {}; | 35 struct pollfd pfds = {}; |
40 pfds.fd = h.fd; | 36 pfds.fd = h.fd; |
41 pfds.events = POLLIN; | 37 pfds.events = POLLIN; |
42 CHECK_EQ(poll(&pfds, 1, -1), 1); | 38 CHECK_EQ(poll(&pfds, 1, -1), 1); |
43 } | 39 } |
44 | 40 |
45 FILE* NewTmpFile() { | |
46 #if defined(OS_ANDROID) | |
47 base::FilePath tmpdir; | |
48 if (!base::android::GetCacheDirectory(&tmpdir)) | |
49 return nullptr; | |
50 std::string templ = tmpdir.Append("XXXXXXXX").value(); | |
51 int fd = mkstemp(const_cast<char*>(templ.c_str())); | |
52 if (fd == -1) | |
53 return nullptr; | |
54 CHECK(unlink(templ.c_str()) == 0); | |
55 return fdopen(fd, "w+"); | |
56 #else | |
57 return tmpfile(); | |
58 #endif | |
59 } | |
60 | |
61 class PlatformChannelPairPosixTest : public testing::Test { | 41 class PlatformChannelPairPosixTest : public testing::Test { |
62 public: | 42 public: |
63 PlatformChannelPairPosixTest() {} | 43 PlatformChannelPairPosixTest() {} |
64 ~PlatformChannelPairPosixTest() override {} | 44 ~PlatformChannelPairPosixTest() override {} |
65 | 45 |
66 void SetUp() override { | 46 void SetUp() override { |
67 // Make sure |SIGPIPE| isn't being ignored. | 47 // Make sure |SIGPIPE| isn't being ignored. |
68 struct sigaction action = {}; | 48 struct sigaction action = {}; |
69 action.sa_handler = SIG_DFL; | 49 action.sa_handler = SIG_DFL; |
70 ASSERT_EQ(0, sigaction(SIGPIPE, &action, &old_action_)); | 50 ASSERT_EQ(0, sigaction(SIGPIPE, &action, &old_action_)); |
(...skipping 69 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
140 std::deque<PlatformHandle> received_handles; | 120 std::deque<PlatformHandle> received_handles; |
141 ssize_t result = PlatformChannelRecvmsg(client_handle.get(), buf, | 121 ssize_t result = PlatformChannelRecvmsg(client_handle.get(), buf, |
142 sizeof(buf), &received_handles); | 122 sizeof(buf), &received_handles); |
143 EXPECT_EQ(static_cast<ssize_t>(send_string.size()), result); | 123 EXPECT_EQ(static_cast<ssize_t>(send_string.size()), result); |
144 EXPECT_EQ(send_string, std::string(buf, static_cast<size_t>(result))); | 124 EXPECT_EQ(send_string, std::string(buf, static_cast<size_t>(result))); |
145 EXPECT_TRUE(received_handles.empty()); | 125 EXPECT_TRUE(received_handles.empty()); |
146 } | 126 } |
147 } | 127 } |
148 | 128 |
149 TEST_F(PlatformChannelPairPosixTest, SendReceiveFDs) { | 129 TEST_F(PlatformChannelPairPosixTest, SendReceiveFDs) { |
| 130 mojo::test::ScopedTestDir test_dir; |
| 131 |
150 static const char kHello[] = "hello"; | 132 static const char kHello[] = "hello"; |
151 | 133 |
152 PlatformChannelPair channel_pair; | 134 PlatformChannelPair channel_pair; |
153 ScopedPlatformHandle server_handle = channel_pair.PassServerHandle().Pass(); | 135 ScopedPlatformHandle server_handle = channel_pair.PassServerHandle().Pass(); |
154 ScopedPlatformHandle client_handle = channel_pair.PassClientHandle().Pass(); | 136 ScopedPlatformHandle client_handle = channel_pair.PassClientHandle().Pass(); |
155 | 137 |
156 // Reduce the number of FDs opened on OS X to avoid test flake. | 138 // Reduce the number of FDs opened on OS X to avoid test flake. |
157 #if defined(OS_MACOSX) | 139 #if defined(OS_MACOSX) |
158 const size_t kNumHandlesToSend = kPlatformChannelMaxNumHandles / 2; | 140 const size_t kNumHandlesToSend = kPlatformChannelMaxNumHandles / 2; |
159 #else | 141 #else |
160 const size_t kNumHandlesToSend = kPlatformChannelMaxNumHandles; | 142 const size_t kNumHandlesToSend = kPlatformChannelMaxNumHandles; |
161 #endif | 143 #endif |
162 | 144 |
163 for (size_t i = 1; i < kNumHandlesToSend; i++) { | 145 for (size_t i = 1; i < kNumHandlesToSend; i++) { |
164 // Make |i| files, with the j-th file consisting of j copies of the digit | 146 // Make |i| files, with the j-th file consisting of j copies of the digit |
165 // |c|. | 147 // |c|. |
166 const char c = '0' + (i % 10); | 148 const char c = '0' + (i % 10); |
167 ScopedPlatformHandleVectorPtr platform_handles(new PlatformHandleVector); | 149 ScopedPlatformHandleVectorPtr platform_handles(new PlatformHandleVector); |
168 for (size_t j = 1; j <= i; j++) { | 150 for (size_t j = 1; j <= i; j++) { |
169 util::ScopedFILE fp(NewTmpFile()); | 151 util::ScopedFILE fp(test_dir.CreateFile()); |
170 ASSERT_TRUE(fp); | 152 ASSERT_TRUE(fp); |
171 ASSERT_EQ(j, fwrite(std::string(j, c).data(), 1, j, fp.get())); | 153 ASSERT_EQ(j, fwrite(std::string(j, c).data(), 1, j, fp.get())); |
172 platform_handles->push_back( | 154 platform_handles->push_back( |
173 test::PlatformHandleFromFILE(fp.Pass()).release()); | 155 test::PlatformHandleFromFILE(fp.Pass()).release()); |
174 ASSERT_TRUE(platform_handles->back().is_valid()); | 156 ASSERT_TRUE(platform_handles->back().is_valid()); |
175 } | 157 } |
176 | 158 |
177 // Send the FDs (+ "hello"). | 159 // Send the FDs (+ "hello"). |
178 struct iovec iov = {const_cast<char*>(kHello), sizeof(kHello)}; | 160 struct iovec iov = {const_cast<char*>(kHello), sizeof(kHello)}; |
179 // We assume that the |sendmsg()| actually sends all the data. | 161 // We assume that the |sendmsg()| actually sends all the data. |
(...skipping 21 matching lines...) Expand all Loading... |
201 rewind(fp.get()); | 183 rewind(fp.get()); |
202 char read_buf[kNumHandlesToSend]; | 184 char read_buf[kNumHandlesToSend]; |
203 size_t bytes_read = fread(read_buf, 1, sizeof(read_buf), fp.get()); | 185 size_t bytes_read = fread(read_buf, 1, sizeof(read_buf), fp.get()); |
204 EXPECT_EQ(j + 1, bytes_read); | 186 EXPECT_EQ(j + 1, bytes_read); |
205 EXPECT_EQ(std::string(j + 1, c), std::string(read_buf, bytes_read)); | 187 EXPECT_EQ(std::string(j + 1, c), std::string(read_buf, bytes_read)); |
206 } | 188 } |
207 } | 189 } |
208 } | 190 } |
209 | 191 |
210 TEST_F(PlatformChannelPairPosixTest, AppendReceivedFDs) { | 192 TEST_F(PlatformChannelPairPosixTest, AppendReceivedFDs) { |
| 193 mojo::test::ScopedTestDir test_dir; |
| 194 |
211 static const char kHello[] = "hello"; | 195 static const char kHello[] = "hello"; |
212 | 196 |
213 PlatformChannelPair channel_pair; | 197 PlatformChannelPair channel_pair; |
214 ScopedPlatformHandle server_handle = channel_pair.PassServerHandle().Pass(); | 198 ScopedPlatformHandle server_handle = channel_pair.PassServerHandle().Pass(); |
215 ScopedPlatformHandle client_handle = channel_pair.PassClientHandle().Pass(); | 199 ScopedPlatformHandle client_handle = channel_pair.PassClientHandle().Pass(); |
216 | 200 |
217 const std::string file_contents("hello world"); | 201 const std::string file_contents("hello world"); |
218 | 202 |
219 { | 203 { |
220 util::ScopedFILE fp(NewTmpFile()); | 204 util::ScopedFILE fp(test_dir.CreateFile()); |
221 ASSERT_TRUE(fp); | 205 ASSERT_TRUE(fp); |
222 ASSERT_EQ(file_contents.size(), | 206 ASSERT_EQ(file_contents.size(), |
223 fwrite(file_contents.data(), 1, file_contents.size(), fp.get())); | 207 fwrite(file_contents.data(), 1, file_contents.size(), fp.get())); |
224 ScopedPlatformHandleVectorPtr platform_handles(new PlatformHandleVector); | 208 ScopedPlatformHandleVectorPtr platform_handles(new PlatformHandleVector); |
225 platform_handles->push_back( | 209 platform_handles->push_back( |
226 test::PlatformHandleFromFILE(fp.Pass()).release()); | 210 test::PlatformHandleFromFILE(fp.Pass()).release()); |
227 ASSERT_TRUE(platform_handles->back().is_valid()); | 211 ASSERT_TRUE(platform_handles->back().is_valid()); |
228 | 212 |
229 // Send the FD (+ "hello"). | 213 // Send the FD (+ "hello"). |
230 struct iovec iov = {const_cast<char*>(kHello), sizeof(kHello)}; | 214 struct iovec iov = {const_cast<char*>(kHello), sizeof(kHello)}; |
(...skipping 29 matching lines...) Expand all Loading... |
260 char read_buf[100]; | 244 char read_buf[100]; |
261 size_t bytes_read = fread(read_buf, 1, sizeof(read_buf), fp.get()); | 245 size_t bytes_read = fread(read_buf, 1, sizeof(read_buf), fp.get()); |
262 EXPECT_EQ(file_contents.size(), bytes_read); | 246 EXPECT_EQ(file_contents.size(), bytes_read); |
263 EXPECT_EQ(file_contents, std::string(read_buf, bytes_read)); | 247 EXPECT_EQ(file_contents, std::string(read_buf, bytes_read)); |
264 } | 248 } |
265 } | 249 } |
266 | 250 |
267 } // namespace | 251 } // namespace |
268 } // namespace embedder | 252 } // namespace embedder |
269 } // namespace mojo | 253 } // namespace mojo |
OLD | NEW |