OLD | NEW |
---|---|
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 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 <errno.h> | 5 #include <errno.h> |
6 #include <fcntl.h> | 6 #include <fcntl.h> |
7 #include <string.h> | 7 #include <string.h> |
8 #include <sys/ioctl.h> | 8 #include <sys/ioctl.h> |
9 #include <sys/select.h> | 9 #include <sys/select.h> |
10 #include <sys/stat.h> | 10 #include <sys/stat.h> |
11 #include <sys/time.h> | 11 #include <sys/time.h> |
12 #include <string> | 12 #include <string> |
13 | 13 |
14 #include "dev_fs_for_testing.h" | 14 #include "dev_fs_for_testing.h" |
15 #include "gtest/gtest.h" | 15 #include "gtest/gtest.h" |
16 #include "nacl_io/devfs/dev_fs.h" | 16 #include "nacl_io/devfs/dev_fs.h" |
17 #include "nacl_io/filesystem.h" | 17 #include "nacl_io/filesystem.h" |
18 #include "nacl_io/ioctl.h" | 18 #include "nacl_io/ioctl.h" |
19 #include "nacl_io/kernel_intercept.h" | 19 #include "nacl_io/kernel_intercept.h" |
20 #include "nacl_io/kernel_proxy.h" | 20 #include "nacl_io/kernel_proxy.h" |
21 #include "nacl_io/osdirent.h" | 21 #include "nacl_io/osdirent.h" |
22 | 22 |
23 using namespace nacl_io; | 23 using namespace nacl_io; |
24 | 24 |
25 namespace { | 25 namespace { |
26 | 26 |
27 static int ki_ioctl_wrapper(int fd, int request, ...) { | |
28 va_list ap; | |
29 va_start(ap, request); | |
30 int rtn = ki_ioctl(fd, request, ap); | |
31 va_end(ap); | |
32 return rtn; | |
33 } | |
34 | |
27 class TtyNodeTest : public ::testing::Test { | 35 class TtyNodeTest : public ::testing::Test { |
28 public: | 36 public: |
29 TtyNodeTest() : fs_(&pepper_) {} | 37 TtyNodeTest() : fs_(&ppapi_) {} |
30 | 38 |
31 void SetUp() { | 39 void SetUp() { |
32 ASSERT_EQ(0, fs_.Access(Path("/tty"), R_OK | W_OK)); | 40 ASSERT_EQ(0, fs_.Access(Path("/tty"), R_OK | W_OK)); |
33 ASSERT_EQ(EACCES, fs_.Access(Path("/tty"), X_OK)); | 41 ASSERT_EQ(EACCES, fs_.Access(Path("/tty"), X_OK)); |
34 ASSERT_EQ(0, fs_.Open(Path("/tty"), O_RDWR, &dev_tty_)); | 42 ASSERT_EQ(0, fs_.Open(Path("/tty"), O_RDWR, &dev_tty_)); |
35 ASSERT_NE(NULL_NODE, dev_tty_.get()); | 43 ASSERT_NE(NULL_NODE, dev_tty_.get()); |
36 } | 44 } |
37 | 45 |
38 protected: | 46 protected: |
39 FakePepperInterface pepper_; | 47 FakePepperInterface ppapi_; |
40 DevFsForTesting fs_; | 48 DevFsForTesting fs_; |
41 ScopedNode dev_tty_; | 49 ScopedNode dev_tty_; |
42 }; | 50 }; |
43 | 51 |
44 class TtyTest : public ::testing::Test { | 52 class TtyTest : public ::testing::Test { |
45 public: | 53 public: |
46 void SetUp() { | 54 void SetUp() { |
47 ASSERT_EQ(0, ki_push_state_for_testing()); | 55 ASSERT_EQ(0, ki_push_state_for_testing()); |
48 ASSERT_EQ(0, ki_init(&kp_)); | 56 ASSERT_EQ(0, ki_init_interface(&kp_, &ppapi_)); |
57 | |
58 buffer_iface_ = ppapi_.GetVarArrayBufferInterface(); | |
59 var_iface_ = ppapi_.GetVarInterface(); | |
49 } | 60 } |
50 | 61 |
51 void TearDown() { | 62 void TearDown() { |
52 ki_uninit(); | 63 ki_uninit(); |
53 } | 64 } |
54 | 65 |
66 int TtyWrite(int fd, const char* string) { | |
67 size_t len = strlen(string); | |
68 PP_Var message_var = buffer_iface_->Create(strlen(string)); | |
69 | |
70 char* data = static_cast<char*>(buffer_iface_->Map(message_var)); | |
71 EXPECT_NE((char*)NULL, data); | |
72 | |
73 memcpy(data, string, len); | |
74 buffer_iface_->Unmap(message_var); | |
75 | |
76 int result = ki_ioctl_wrapper(fd, NACL_IOC_HANDLEMESSAGE, &message_var); | |
77 | |
78 var_iface_->Release(message_var); | |
79 return result; | |
80 } | |
81 | |
55 protected: | 82 protected: |
83 FakePepperInterface ppapi_; | |
56 KernelProxy kp_; | 84 KernelProxy kp_; |
85 VarArrayBufferInterface* buffer_iface_; | |
86 VarInterface* var_iface_; | |
57 }; | 87 }; |
58 | 88 |
59 TEST_F(TtyNodeTest, InvalidIoctl) { | 89 TEST_F(TtyNodeTest, InvalidIoctl) { |
60 // 123 is not a valid ioctl request. | 90 // 123 is not a valid ioctl request. |
61 EXPECT_EQ(EINVAL, dev_tty_->Ioctl(123)); | 91 EXPECT_EQ(EINVAL, dev_tty_->Ioctl(123)); |
62 } | 92 } |
63 | 93 |
64 TEST_F(TtyNodeTest, TtyInput) { | 94 TEST_F(TtyNodeTest, TtyInput) { |
65 // Now let's try sending some data over. | 95 // Now let's try sending some data over. |
66 // First we create the message. | 96 // First we create the message. |
67 std::string message("hello, how are you?\n"); | 97 std::string message("hello, how are you?\n"); |
68 struct tioc_nacl_input_string packaged_message; | 98 VarArrayBufferInterface* buffer_iface = ppapi_.GetVarArrayBufferInterface(); |
69 packaged_message.length = message.size(); | 99 VarInterface* var_iface = ppapi_.GetVarInterface(); |
70 packaged_message.buffer = message.data(); | 100 PP_Var message_var = buffer_iface->Create(message.size()); |
101 | |
102 char* data = static_cast<char*>(buffer_iface->Map(message_var)); | |
103 ASSERT_NE((char*)NULL, data); | |
104 | |
105 memcpy(data, message.data(), message.size()); | |
106 buffer_iface->Unmap(message_var); | |
71 | 107 |
72 // Now we make buffer we'll read into. | 108 // Now we make buffer we'll read into. |
73 // We fill the buffer and a backup buffer with arbitrary data | 109 // We fill the buffer and a backup buffer with arbitrary data |
74 // and compare them after reading to make sure read doesn't | 110 // and compare them after reading to make sure read doesn't |
75 // clobber parts of the buffer it shouldn't. | 111 // clobber parts of the buffer it shouldn't. |
76 int bytes_read; | 112 int bytes_read; |
77 char buffer[100]; | 113 char buffer[100]; |
78 char backup_buffer[100]; | 114 char backup_buffer[100]; |
79 memset(buffer, 'a', 100); | 115 memset(buffer, 'a', 100); |
80 memset(backup_buffer, 'a', 100); | 116 memset(backup_buffer, 'a', 100); |
81 | 117 |
82 // Now we actually send the data | 118 // Now we actually send the data |
83 EXPECT_EQ(0, dev_tty_->Ioctl(TIOCNACLINPUT, &packaged_message)); | 119 EXPECT_EQ(0, dev_tty_->Ioctl(NACL_IOC_HANDLEMESSAGE, &message_var)); |
120 | |
121 var_iface->Release(message_var); | |
84 | 122 |
85 // We read a small chunk first to ensure it doesn't give us | 123 // We read a small chunk first to ensure it doesn't give us |
86 // more than we ask for. | 124 // more than we ask for. |
87 HandleAttr attrs; | 125 HandleAttr attrs; |
88 EXPECT_EQ(0, dev_tty_->Read(attrs, buffer, 5, &bytes_read)); | 126 EXPECT_EQ(0, dev_tty_->Read(attrs, buffer, 5, &bytes_read)); |
89 EXPECT_EQ(5, bytes_read); | 127 EXPECT_EQ(5, bytes_read); |
90 EXPECT_EQ(0, memcmp(message.data(), buffer, 5)); | 128 EXPECT_EQ(0, memcmp(message.data(), buffer, 5)); |
91 EXPECT_EQ(0, memcmp(buffer + 5, backup_buffer + 5, 95)); | 129 EXPECT_EQ(0, memcmp(buffer + 5, backup_buffer + 5, 95)); |
92 | 130 |
93 // Now we ask for more data than is left in the tty, to ensure | 131 // Now we ask for more data than is left in the tty, to ensure |
(...skipping 36 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
130 handler.user_data = &user_data; | 168 handler.user_data = &user_data; |
131 | 169 |
132 EXPECT_EQ(0, dev_tty_->Ioctl(TIOCNACLOUTPUT, &handler)); | 170 EXPECT_EQ(0, dev_tty_->Ioctl(TIOCNACLOUTPUT, &handler)); |
133 | 171 |
134 EXPECT_EQ(0, dev_tty_->Write(attrs, message, message_len, &bytes_written)); | 172 EXPECT_EQ(0, dev_tty_->Write(attrs, message, message_len, &bytes_written)); |
135 EXPECT_EQ(message_len, bytes_written); | 173 EXPECT_EQ(message_len, bytes_written); |
136 EXPECT_EQ(message_len, user_data.output_count); | 174 EXPECT_EQ(message_len, user_data.output_count); |
137 EXPECT_EQ(0, strncmp(user_data.output_buf, message, message_len)); | 175 EXPECT_EQ(0, strncmp(user_data.output_buf, message, message_len)); |
138 } | 176 } |
139 | 177 |
140 static int ki_ioctl_wrapper(int fd, int request, ...) { | |
141 va_list ap; | |
142 va_start(ap, request); | |
143 int rtn = ki_ioctl(fd, request, ap); | |
144 va_end(ap); | |
145 return rtn; | |
146 } | |
147 | |
148 static int TtyWrite(int fd, const char* string) { | |
149 struct tioc_nacl_input_string input; | |
150 input.buffer = string; | |
151 input.length = strlen(input.buffer); | |
152 return ki_ioctl_wrapper(fd, TIOCNACLINPUT, &input); | |
153 } | |
154 | |
155 // Returns: | 178 // Returns: |
156 // 0 -> Not readable | 179 // 0 -> Not readable |
157 // 1 -> Readable | 180 // 1 -> Readable |
158 // -1 -> Error occured | 181 // -1 -> Error occured |
159 static int IsReadable(int fd) { | 182 static int IsReadable(int fd) { |
160 struct timeval timeout = {0, 0}; | 183 struct timeval timeout = {0, 0}; |
161 fd_set readfds; | 184 fd_set readfds; |
162 fd_set errorfds; | 185 fd_set errorfds; |
163 FD_ZERO(&readfds); | 186 FD_ZERO(&readfds); |
164 FD_ZERO(&errorfds); | 187 FD_ZERO(&errorfds); |
165 FD_SET(fd, &readfds); | 188 FD_SET(fd, &readfds); |
166 FD_SET(fd, &errorfds); | 189 FD_SET(fd, &errorfds); |
167 int rtn = ki_select(fd + 1, &readfds, NULL, &errorfds, &timeout); | 190 int rtn = ki_select(fd + 1, &readfds, NULL, &errorfds, &timeout); |
168 if (rtn == 0) | 191 if (rtn == 0) |
169 return 0; // not readable | 192 return 0; // not readable |
170 if (rtn != 1) | 193 if (rtn != 1) |
171 return -1; // error | 194 return -1; // error |
172 if (FD_ISSET(fd, &errorfds)) | 195 if (FD_ISSET(fd, &errorfds)) |
173 return -2; // error | 196 return -2; // error |
174 if (!FD_ISSET(fd, &readfds)) | 197 if (!FD_ISSET(fd, &readfds)) |
175 return -3; // error | 198 return -3; // error |
176 return 1; // readable | 199 return 1; // readable |
177 } | 200 } |
178 | 201 |
202 TEST_F(TtyTest, Empty) { | |
203 } | |
Sam Clegg
2014/06/17 22:29:29
?
binji
2014/06/17 22:55:23
Done.
| |
204 | |
179 TEST_F(TtyTest, TtySelect) { | 205 TEST_F(TtyTest, TtySelect) { |
180 struct timeval timeout; | 206 struct timeval timeout; |
181 fd_set readfds; | 207 fd_set readfds; |
182 fd_set writefds; | 208 fd_set writefds; |
183 fd_set errorfds; | 209 fd_set errorfds; |
184 | 210 |
185 int tty_fd = ki_open("/dev/tty", O_RDONLY); | 211 int tty_fd = ki_open("/dev/tty", O_RDONLY); |
186 ASSERT_GT(tty_fd, 0) << "tty open failed: " << errno; | 212 ASSERT_GT(tty_fd, 0) << "tty open failed: " << errno; |
187 | 213 |
188 FD_ZERO(&readfds); | 214 FD_ZERO(&readfds); |
(...skipping 141 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
330 pthread_join(resize_thread, NULL); | 356 pthread_join(resize_thread, NULL); |
331 ASSERT_EQ(-1, rtn); | 357 ASSERT_EQ(-1, rtn); |
332 ASSERT_EQ(EINTR, errno); | 358 ASSERT_EQ(EINTR, errno); |
333 ASSERT_EQ(0, IsReadable(tty_fd)); | 359 ASSERT_EQ(0, IsReadable(tty_fd)); |
334 } | 360 } |
335 | 361 |
336 /* | 362 /* |
337 * Sleep for 50ms then send some input to the /dev/tty. | 363 * Sleep for 50ms then send some input to the /dev/tty. |
338 */ | 364 */ |
339 static void* input_thread_main(void* arg) { | 365 static void* input_thread_main(void* arg) { |
366 TtyTest* thiz = static_cast<TtyTest*>(arg); | |
367 | |
340 usleep(50 * 1000); | 368 usleep(50 * 1000); |
341 | 369 |
342 int fd = ki_open("/dev/tty", O_RDONLY); | 370 int fd = ki_open("/dev/tty", O_RDONLY); |
343 TtyWrite(fd, "test\n"); | 371 thiz->TtyWrite(fd, "test\n"); |
344 return NULL; | 372 return NULL; |
345 } | 373 } |
346 | 374 |
347 TEST_F(TtyTest, InputDuringSelect) { | 375 TEST_F(TtyTest, InputDuringSelect) { |
348 // Test that input which occurs while in select causes | 376 // Test that input which occurs while in select causes |
349 // select to return. | 377 // select to return. |
350 int tty_fd = ki_open("/dev/tty", O_RDONLY); | 378 int tty_fd = ki_open("/dev/tty", O_RDONLY); |
351 | 379 |
352 fd_set readfds; | 380 fd_set readfds; |
353 fd_set errorfds; | 381 fd_set errorfds; |
354 FD_ZERO(&readfds); | 382 FD_ZERO(&readfds); |
355 FD_ZERO(&errorfds); | 383 FD_ZERO(&errorfds); |
356 FD_SET(tty_fd, &readfds); | 384 FD_SET(tty_fd, &readfds); |
357 FD_SET(tty_fd, &errorfds); | 385 FD_SET(tty_fd, &errorfds); |
358 | 386 |
359 pthread_t resize_thread; | 387 pthread_t resize_thread; |
360 pthread_create(&resize_thread, NULL, input_thread_main, NULL); | 388 pthread_create(&resize_thread, NULL, input_thread_main, this); |
361 | 389 |
362 struct timeval timeout; | 390 struct timeval timeout; |
363 timeout.tv_sec = 20; | 391 timeout.tv_sec = 20; |
364 timeout.tv_usec = 0; | 392 timeout.tv_usec = 0; |
365 | 393 |
366 int rtn = ki_select(tty_fd + 1, &readfds, NULL, &errorfds, &timeout); | 394 int rtn = ki_select(tty_fd + 1, &readfds, NULL, &errorfds, &timeout); |
367 pthread_join(resize_thread, NULL); | 395 pthread_join(resize_thread, NULL); |
368 | 396 |
369 ASSERT_EQ(1, rtn); | 397 ASSERT_EQ(1, rtn); |
370 } | 398 } |
371 } | 399 |
400 } // namespace | |
OLD | NEW |