Yabin Cui | c1b1f6f | 2015-09-15 16:27:09 -0700 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (C) 2015 The Android Open Source Project |
| 3 | * |
| 4 | * Licensed under the Apache License, Version 2.0 (the "License"); |
| 5 | * you may not use this file except in compliance with the License. |
| 6 | * You may obtain a copy of the License at |
| 7 | * |
| 8 | * http://www.apache.org/licenses/LICENSE-2.0 |
| 9 | * |
| 10 | * Unless required by applicable law or agreed to in writing, software |
| 11 | * distributed under the License is distributed on an "AS IS" BASIS, |
| 12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 13 | * See the License for the specific language governing permissions and |
| 14 | * limitations under the License. |
| 15 | */ |
| 16 | |
| 17 | #include "fdevent.h" |
| 18 | |
| 19 | #include <gtest/gtest.h> |
| 20 | |
Josh Gao | 022d447 | 2016-02-10 14:49:00 -0800 | [diff] [blame] | 21 | #include <array> |
Yabin Cui | c1b1f6f | 2015-09-15 16:27:09 -0700 | [diff] [blame] | 22 | #include <limits> |
| 23 | #include <queue> |
| 24 | #include <string> |
Elliott Hughes | dbe91ee | 2016-11-15 12:37:32 -0800 | [diff] [blame] | 25 | #include <thread> |
Yabin Cui | c1b1f6f | 2015-09-15 16:27:09 -0700 | [diff] [blame] | 26 | #include <vector> |
| 27 | |
Yabin Cui | c1b1f6f | 2015-09-15 16:27:09 -0700 | [diff] [blame] | 28 | #include <unistd.h> |
| 29 | |
| 30 | #include "adb.h" |
| 31 | #include "adb_io.h" |
Josh Gao | 022d447 | 2016-02-10 14:49:00 -0800 | [diff] [blame] | 32 | #include "fdevent_test.h" |
Yabin Cui | c1b1f6f | 2015-09-15 16:27:09 -0700 | [diff] [blame] | 33 | #include "socket.h" |
| 34 | #include "sysdeps.h" |
Josh Gao | 4602adb | 2016-11-15 18:55:47 -0800 | [diff] [blame] | 35 | #include "sysdeps/chrono.h" |
Yabin Cui | c1b1f6f | 2015-09-15 16:27:09 -0700 | [diff] [blame] | 36 | |
Yabin Cui | c1b1f6f | 2015-09-15 16:27:09 -0700 | [diff] [blame] | 37 | struct ThreadArg { |
| 38 | int first_read_fd; |
| 39 | int last_write_fd; |
| 40 | size_t middle_pipe_count; |
| 41 | }; |
| 42 | |
Josh Gao | 022d447 | 2016-02-10 14:49:00 -0800 | [diff] [blame] | 43 | class LocalSocketTest : public FdeventTest {}; |
Yabin Cui | c1b1f6f | 2015-09-15 16:27:09 -0700 | [diff] [blame] | 44 | |
Josh Gao | 4602adb | 2016-11-15 18:55:47 -0800 | [diff] [blame] | 45 | constexpr auto SLEEP_FOR_FDEVENT = 100ms; |
Yabin Cui | 2407d7c | 2016-04-25 19:48:19 -0700 | [diff] [blame] | 46 | |
Yabin Cui | c1b1f6f | 2015-09-15 16:27:09 -0700 | [diff] [blame] | 47 | TEST_F(LocalSocketTest, smoke) { |
Josh Gao | 022d447 | 2016-02-10 14:49:00 -0800 | [diff] [blame] | 48 | // Join two socketpairs with a chain of intermediate socketpairs. |
| 49 | int first[2]; |
| 50 | std::vector<std::array<int, 2>> intermediates; |
| 51 | int last[2]; |
| 52 | |
| 53 | constexpr size_t INTERMEDIATE_COUNT = 50; |
| 54 | constexpr size_t MESSAGE_LOOP_COUNT = 100; |
Yabin Cui | c1b1f6f | 2015-09-15 16:27:09 -0700 | [diff] [blame] | 55 | const std::string MESSAGE = "socket_test"; |
Yabin Cui | c1b1f6f | 2015-09-15 16:27:09 -0700 | [diff] [blame] | 56 | |
Josh Gao | 022d447 | 2016-02-10 14:49:00 -0800 | [diff] [blame] | 57 | intermediates.resize(INTERMEDIATE_COUNT); |
| 58 | ASSERT_EQ(0, adb_socketpair(first)) << strerror(errno); |
| 59 | ASSERT_EQ(0, adb_socketpair(last)) << strerror(errno); |
| 60 | asocket* prev_tail = create_local_socket(first[1]); |
| 61 | ASSERT_NE(nullptr, prev_tail); |
Yabin Cui | c1b1f6f | 2015-09-15 16:27:09 -0700 | [diff] [blame] | 62 | |
Josh Gao | 022d447 | 2016-02-10 14:49:00 -0800 | [diff] [blame] | 63 | auto connect = [](asocket* tail, asocket* head) { |
| 64 | tail->peer = head; |
| 65 | head->peer = tail; |
| 66 | tail->ready(tail); |
| 67 | }; |
| 68 | |
| 69 | for (auto& intermediate : intermediates) { |
| 70 | ASSERT_EQ(0, adb_socketpair(intermediate.data())) << strerror(errno); |
| 71 | |
| 72 | asocket* head = create_local_socket(intermediate[0]); |
| 73 | ASSERT_NE(nullptr, head); |
| 74 | |
| 75 | asocket* tail = create_local_socket(intermediate[1]); |
| 76 | ASSERT_NE(nullptr, tail); |
| 77 | |
| 78 | connect(prev_tail, head); |
| 79 | prev_tail = tail; |
| 80 | } |
| 81 | |
| 82 | asocket* end = create_local_socket(last[0]); |
| 83 | ASSERT_NE(nullptr, end); |
| 84 | connect(prev_tail, end); |
| 85 | |
| 86 | PrepareThread(); |
Josh Gao | e1dacfc | 2017-04-12 17:00:49 -0700 | [diff] [blame] | 87 | std::thread thread(fdevent_loop); |
Josh Gao | 022d447 | 2016-02-10 14:49:00 -0800 | [diff] [blame] | 88 | |
Yabin Cui | c1b1f6f | 2015-09-15 16:27:09 -0700 | [diff] [blame] | 89 | for (size_t i = 0; i < MESSAGE_LOOP_COUNT; ++i) { |
| 90 | std::string read_buffer = MESSAGE; |
| 91 | std::string write_buffer(MESSAGE.size(), 'a'); |
Josh Gao | 022d447 | 2016-02-10 14:49:00 -0800 | [diff] [blame] | 92 | ASSERT_TRUE(WriteFdExactly(first[0], &read_buffer[0], read_buffer.size())); |
| 93 | ASSERT_TRUE(ReadFdExactly(last[1], &write_buffer[0], write_buffer.size())); |
Yabin Cui | c1b1f6f | 2015-09-15 16:27:09 -0700 | [diff] [blame] | 94 | ASSERT_EQ(read_buffer, write_buffer); |
| 95 | } |
Yabin Cui | c1b1f6f | 2015-09-15 16:27:09 -0700 | [diff] [blame] | 96 | |
Josh Gao | 022d447 | 2016-02-10 14:49:00 -0800 | [diff] [blame] | 97 | ASSERT_EQ(0, adb_close(first[0])); |
| 98 | ASSERT_EQ(0, adb_close(last[1])); |
| 99 | |
| 100 | // Wait until the local sockets are closed. |
Elliott Hughes | dbe91ee | 2016-11-15 12:37:32 -0800 | [diff] [blame] | 101 | std::this_thread::sleep_for(SLEEP_FOR_FDEVENT); |
Yabin Cui | 2407d7c | 2016-04-25 19:48:19 -0700 | [diff] [blame] | 102 | ASSERT_EQ(GetAdditionalLocalSocketCount(), fdevent_installed_count()); |
Josh Gao | 022d447 | 2016-02-10 14:49:00 -0800 | [diff] [blame] | 103 | TerminateThread(thread); |
Yabin Cui | c1b1f6f | 2015-09-15 16:27:09 -0700 | [diff] [blame] | 104 | } |
| 105 | |
| 106 | struct CloseWithPacketArg { |
| 107 | int socket_fd; |
| 108 | size_t bytes_written; |
| 109 | int cause_close_fd; |
| 110 | }; |
| 111 | |
| 112 | static void CloseWithPacketThreadFunc(CloseWithPacketArg* arg) { |
| 113 | asocket* s = create_local_socket(arg->socket_fd); |
| 114 | ASSERT_TRUE(s != nullptr); |
| 115 | arg->bytes_written = 0; |
| 116 | while (true) { |
Josh Gao | 27cb7dc | 2018-02-01 13:17:50 -0800 | [diff] [blame] | 117 | std::string data; |
| 118 | data.resize(MAX_PAYLOAD); |
| 119 | arg->bytes_written += data.size(); |
| 120 | int ret = s->enqueue(s, std::move(data)); |
Yabin Cui | c1b1f6f | 2015-09-15 16:27:09 -0700 | [diff] [blame] | 121 | if (ret == 1) { |
| 122 | // The writer has one packet waiting to send. |
| 123 | break; |
| 124 | } |
| 125 | } |
| 126 | |
| 127 | asocket* cause_close_s = create_local_socket(arg->cause_close_fd); |
| 128 | ASSERT_TRUE(cause_close_s != nullptr); |
| 129 | cause_close_s->peer = s; |
| 130 | s->peer = cause_close_s; |
| 131 | cause_close_s->ready(cause_close_s); |
| 132 | |
Yabin Cui | c1b1f6f | 2015-09-15 16:27:09 -0700 | [diff] [blame] | 133 | fdevent_loop(); |
| 134 | } |
| 135 | |
| 136 | // This test checks if we can close local socket in the following situation: |
| 137 | // The socket is closing but having some packets, so it is not closed. Then |
| 138 | // some write error happens in the socket's file handler, e.g., the file |
| 139 | // handler is closed. |
Yabin Cui | aa77e22 | 2015-09-29 12:25:33 -0700 | [diff] [blame] | 140 | TEST_F(LocalSocketTest, close_socket_with_packet) { |
Yabin Cui | c1b1f6f | 2015-09-15 16:27:09 -0700 | [diff] [blame] | 141 | int socket_fd[2]; |
| 142 | ASSERT_EQ(0, adb_socketpair(socket_fd)); |
| 143 | int cause_close_fd[2]; |
| 144 | ASSERT_EQ(0, adb_socketpair(cause_close_fd)); |
| 145 | CloseWithPacketArg arg; |
| 146 | arg.socket_fd = socket_fd[1]; |
| 147 | arg.cause_close_fd = cause_close_fd[1]; |
Yabin Cui | c1b1f6f | 2015-09-15 16:27:09 -0700 | [diff] [blame] | 148 | |
Josh Gao | 022d447 | 2016-02-10 14:49:00 -0800 | [diff] [blame] | 149 | PrepareThread(); |
Josh Gao | e1dacfc | 2017-04-12 17:00:49 -0700 | [diff] [blame] | 150 | std::thread thread(CloseWithPacketThreadFunc, &arg); |
Josh Gao | 022d447 | 2016-02-10 14:49:00 -0800 | [diff] [blame] | 151 | // Wait until the fdevent_loop() starts. |
Elliott Hughes | dbe91ee | 2016-11-15 12:37:32 -0800 | [diff] [blame] | 152 | std::this_thread::sleep_for(SLEEP_FOR_FDEVENT); |
Josh Gao | 022d447 | 2016-02-10 14:49:00 -0800 | [diff] [blame] | 153 | ASSERT_EQ(0, adb_close(cause_close_fd[0])); |
Elliott Hughes | dbe91ee | 2016-11-15 12:37:32 -0800 | [diff] [blame] | 154 | std::this_thread::sleep_for(SLEEP_FOR_FDEVENT); |
Yabin Cui | 2407d7c | 2016-04-25 19:48:19 -0700 | [diff] [blame] | 155 | EXPECT_EQ(1u + GetAdditionalLocalSocketCount(), fdevent_installed_count()); |
Josh Gao | 022d447 | 2016-02-10 14:49:00 -0800 | [diff] [blame] | 156 | ASSERT_EQ(0, adb_close(socket_fd[0])); |
Elliott Hughes | dbe91ee | 2016-11-15 12:37:32 -0800 | [diff] [blame] | 157 | std::this_thread::sleep_for(SLEEP_FOR_FDEVENT); |
Yabin Cui | 2407d7c | 2016-04-25 19:48:19 -0700 | [diff] [blame] | 158 | ASSERT_EQ(GetAdditionalLocalSocketCount(), fdevent_installed_count()); |
Josh Gao | 022d447 | 2016-02-10 14:49:00 -0800 | [diff] [blame] | 159 | TerminateThread(thread); |
Yabin Cui | c1b1f6f | 2015-09-15 16:27:09 -0700 | [diff] [blame] | 160 | } |
| 161 | |
Yabin Cui | c1b1f6f | 2015-09-15 16:27:09 -0700 | [diff] [blame] | 162 | // This test checks if we can read packets from a closing local socket. |
Yabin Cui | aa77e22 | 2015-09-29 12:25:33 -0700 | [diff] [blame] | 163 | TEST_F(LocalSocketTest, read_from_closing_socket) { |
Yabin Cui | c1b1f6f | 2015-09-15 16:27:09 -0700 | [diff] [blame] | 164 | int socket_fd[2]; |
| 165 | ASSERT_EQ(0, adb_socketpair(socket_fd)); |
| 166 | int cause_close_fd[2]; |
| 167 | ASSERT_EQ(0, adb_socketpair(cause_close_fd)); |
| 168 | CloseWithPacketArg arg; |
| 169 | arg.socket_fd = socket_fd[1]; |
| 170 | arg.cause_close_fd = cause_close_fd[1]; |
| 171 | |
Josh Gao | 022d447 | 2016-02-10 14:49:00 -0800 | [diff] [blame] | 172 | PrepareThread(); |
Josh Gao | e1dacfc | 2017-04-12 17:00:49 -0700 | [diff] [blame] | 173 | std::thread thread(CloseWithPacketThreadFunc, &arg); |
Yabin Cui | c1b1f6f | 2015-09-15 16:27:09 -0700 | [diff] [blame] | 174 | // Wait until the fdevent_loop() starts. |
Elliott Hughes | dbe91ee | 2016-11-15 12:37:32 -0800 | [diff] [blame] | 175 | std::this_thread::sleep_for(SLEEP_FOR_FDEVENT); |
Yabin Cui | c1b1f6f | 2015-09-15 16:27:09 -0700 | [diff] [blame] | 176 | ASSERT_EQ(0, adb_close(cause_close_fd[0])); |
Elliott Hughes | dbe91ee | 2016-11-15 12:37:32 -0800 | [diff] [blame] | 177 | std::this_thread::sleep_for(SLEEP_FOR_FDEVENT); |
Yabin Cui | 2407d7c | 2016-04-25 19:48:19 -0700 | [diff] [blame] | 178 | EXPECT_EQ(1u + GetAdditionalLocalSocketCount(), fdevent_installed_count()); |
Yabin Cui | c1b1f6f | 2015-09-15 16:27:09 -0700 | [diff] [blame] | 179 | |
| 180 | // Verify if we can read successfully. |
| 181 | std::vector<char> buf(arg.bytes_written); |
Josh Gao | 022d447 | 2016-02-10 14:49:00 -0800 | [diff] [blame] | 182 | ASSERT_NE(0u, arg.bytes_written); |
Yabin Cui | c1b1f6f | 2015-09-15 16:27:09 -0700 | [diff] [blame] | 183 | ASSERT_EQ(true, ReadFdExactly(socket_fd[0], buf.data(), buf.size())); |
| 184 | ASSERT_EQ(0, adb_close(socket_fd[0])); |
| 185 | |
Elliott Hughes | dbe91ee | 2016-11-15 12:37:32 -0800 | [diff] [blame] | 186 | std::this_thread::sleep_for(SLEEP_FOR_FDEVENT); |
Yabin Cui | 2407d7c | 2016-04-25 19:48:19 -0700 | [diff] [blame] | 187 | ASSERT_EQ(GetAdditionalLocalSocketCount(), fdevent_installed_count()); |
Josh Gao | 022d447 | 2016-02-10 14:49:00 -0800 | [diff] [blame] | 188 | TerminateThread(thread); |
Yabin Cui | c1b1f6f | 2015-09-15 16:27:09 -0700 | [diff] [blame] | 189 | } |
| 190 | |
| 191 | // This test checks if we can close local socket in the following situation: |
| 192 | // The socket is not closed and has some packets. When it fails to write to |
| 193 | // the socket's file handler because the other end is closed, we check if the |
| 194 | // socket is closed. |
| 195 | TEST_F(LocalSocketTest, write_error_when_having_packets) { |
| 196 | int socket_fd[2]; |
| 197 | ASSERT_EQ(0, adb_socketpair(socket_fd)); |
| 198 | int cause_close_fd[2]; |
| 199 | ASSERT_EQ(0, adb_socketpair(cause_close_fd)); |
| 200 | CloseWithPacketArg arg; |
| 201 | arg.socket_fd = socket_fd[1]; |
| 202 | arg.cause_close_fd = cause_close_fd[1]; |
| 203 | |
Josh Gao | 022d447 | 2016-02-10 14:49:00 -0800 | [diff] [blame] | 204 | PrepareThread(); |
Josh Gao | e1dacfc | 2017-04-12 17:00:49 -0700 | [diff] [blame] | 205 | std::thread thread(CloseWithPacketThreadFunc, &arg); |
Yabin Cui | c1b1f6f | 2015-09-15 16:27:09 -0700 | [diff] [blame] | 206 | // Wait until the fdevent_loop() starts. |
Elliott Hughes | dbe91ee | 2016-11-15 12:37:32 -0800 | [diff] [blame] | 207 | std::this_thread::sleep_for(SLEEP_FOR_FDEVENT); |
Yabin Cui | 2407d7c | 2016-04-25 19:48:19 -0700 | [diff] [blame] | 208 | EXPECT_EQ(2u + GetAdditionalLocalSocketCount(), fdevent_installed_count()); |
Yabin Cui | c1b1f6f | 2015-09-15 16:27:09 -0700 | [diff] [blame] | 209 | ASSERT_EQ(0, adb_close(socket_fd[0])); |
| 210 | |
Elliott Hughes | dbe91ee | 2016-11-15 12:37:32 -0800 | [diff] [blame] | 211 | std::this_thread::sleep_for(SLEEP_FOR_FDEVENT); |
Yabin Cui | 2407d7c | 2016-04-25 19:48:19 -0700 | [diff] [blame] | 212 | ASSERT_EQ(GetAdditionalLocalSocketCount(), fdevent_installed_count()); |
Josh Gao | 022d447 | 2016-02-10 14:49:00 -0800 | [diff] [blame] | 213 | TerminateThread(thread); |
Yabin Cui | c1b1f6f | 2015-09-15 16:27:09 -0700 | [diff] [blame] | 214 | } |
| 215 | |
Josh Gao | 1430d39 | 2018-03-15 15:28:55 -0700 | [diff] [blame^] | 216 | #if 0 |
| 217 | // Ensure that if we fail to write output to an fd, we will still flush data coming from it. |
| 218 | TEST_F(LocalSocketTest, flush_after_shutdown) { |
| 219 | int head_fd[2]; |
| 220 | int tail_fd[2]; |
| 221 | ASSERT_EQ(0, adb_socketpair(head_fd)); |
| 222 | ASSERT_EQ(0, adb_socketpair(tail_fd)); |
| 223 | |
| 224 | asocket* head = create_local_socket(head_fd[1]); |
| 225 | asocket* tail = create_local_socket(tail_fd[1]); |
| 226 | |
| 227 | head->peer = tail; |
| 228 | head->ready(head); |
| 229 | |
| 230 | tail->peer = head; |
| 231 | tail->ready(tail); |
| 232 | |
| 233 | PrepareThread(); |
| 234 | std::thread thread(fdevent_loop); |
| 235 | |
| 236 | ASSERT_TRUE(WriteFdExactly(head_fd[0], "foo", 3)); |
| 237 | ASSERT_EQ(0, adb_shutdown(head_fd[0], SHUT_RD)); |
| 238 | const char* str = "write succeeds, but local_socket will fail to write"; |
| 239 | ASSERT_TRUE(WriteFdExactly(tail_fd[0], str, strlen(str))); |
| 240 | ASSERT_TRUE(WriteFdExactly(head_fd[0], "bar", 3)); |
| 241 | char buf[6]; |
| 242 | ASSERT_TRUE(ReadFdExactly(tail_fd[0], buf, 6)); |
| 243 | |
| 244 | ASSERT_EQ(0, memcmp(buf, "foobar", 6)); |
| 245 | |
| 246 | adb_close(head_fd[0]); |
| 247 | adb_close(tail_fd[0]); |
| 248 | |
| 249 | // Wait until the local sockets are closed. |
| 250 | std::this_thread::sleep_for(SLEEP_FOR_FDEVENT); |
| 251 | ASSERT_EQ(GetAdditionalLocalSocketCount(), fdevent_installed_count()); |
| 252 | TerminateThread(thread); |
| 253 | } |
| 254 | #endif |
| 255 | |
Yabin Cui | aa77e22 | 2015-09-29 12:25:33 -0700 | [diff] [blame] | 256 | #if defined(__linux__) |
| 257 | |
| 258 | static void ClientThreadFunc() { |
| 259 | std::string error; |
| 260 | int fd = network_loopback_client(5038, SOCK_STREAM, &error); |
| 261 | ASSERT_GE(fd, 0) << error; |
Josh Gao | 4602adb | 2016-11-15 18:55:47 -0800 | [diff] [blame] | 262 | std::this_thread::sleep_for(200ms); |
Yabin Cui | aa77e22 | 2015-09-29 12:25:33 -0700 | [diff] [blame] | 263 | ASSERT_EQ(0, adb_close(fd)); |
| 264 | } |
| 265 | |
| 266 | struct CloseRdHupSocketArg { |
Josh Gao | 022d447 | 2016-02-10 14:49:00 -0800 | [diff] [blame] | 267 | int socket_fd; |
Yabin Cui | c1b1f6f | 2015-09-15 16:27:09 -0700 | [diff] [blame] | 268 | }; |
| 269 | |
Yabin Cui | aa77e22 | 2015-09-29 12:25:33 -0700 | [diff] [blame] | 270 | static void CloseRdHupSocketThreadFunc(CloseRdHupSocketArg* arg) { |
Josh Gao | 022d447 | 2016-02-10 14:49:00 -0800 | [diff] [blame] | 271 | asocket* s = create_local_socket(arg->socket_fd); |
| 272 | ASSERT_TRUE(s != nullptr); |
Yabin Cui | c1b1f6f | 2015-09-15 16:27:09 -0700 | [diff] [blame] | 273 | |
Josh Gao | 022d447 | 2016-02-10 14:49:00 -0800 | [diff] [blame] | 274 | fdevent_loop(); |
Yabin Cui | c1b1f6f | 2015-09-15 16:27:09 -0700 | [diff] [blame] | 275 | } |
| 276 | |
Yabin Cui | aa77e22 | 2015-09-29 12:25:33 -0700 | [diff] [blame] | 277 | // This test checks if we can close sockets in CLOSE_WAIT state. |
| 278 | TEST_F(LocalSocketTest, close_socket_in_CLOSE_WAIT_state) { |
Josh Gao | 022d447 | 2016-02-10 14:49:00 -0800 | [diff] [blame] | 279 | std::string error; |
| 280 | int listen_fd = network_inaddr_any_server(5038, SOCK_STREAM, &error); |
| 281 | ASSERT_GE(listen_fd, 0); |
Yabin Cui | c1b1f6f | 2015-09-15 16:27:09 -0700 | [diff] [blame] | 282 | |
Josh Gao | e1dacfc | 2017-04-12 17:00:49 -0700 | [diff] [blame] | 283 | std::thread client_thread(ClientThreadFunc); |
Josh Gao | 022d447 | 2016-02-10 14:49:00 -0800 | [diff] [blame] | 284 | |
Elliott Hughes | 3dcfa3f | 2016-08-23 12:50:00 -0700 | [diff] [blame] | 285 | int accept_fd = adb_socket_accept(listen_fd, nullptr, nullptr); |
Josh Gao | 022d447 | 2016-02-10 14:49:00 -0800 | [diff] [blame] | 286 | ASSERT_GE(accept_fd, 0); |
| 287 | CloseRdHupSocketArg arg; |
| 288 | arg.socket_fd = accept_fd; |
| 289 | |
| 290 | PrepareThread(); |
Josh Gao | e1dacfc | 2017-04-12 17:00:49 -0700 | [diff] [blame] | 291 | std::thread thread(CloseRdHupSocketThreadFunc, &arg); |
Josh Gao | 022d447 | 2016-02-10 14:49:00 -0800 | [diff] [blame] | 292 | |
| 293 | // Wait until the fdevent_loop() starts. |
Elliott Hughes | dbe91ee | 2016-11-15 12:37:32 -0800 | [diff] [blame] | 294 | std::this_thread::sleep_for(SLEEP_FOR_FDEVENT); |
Yabin Cui | 2407d7c | 2016-04-25 19:48:19 -0700 | [diff] [blame] | 295 | EXPECT_EQ(1u + GetAdditionalLocalSocketCount(), fdevent_installed_count()); |
Josh Gao | 022d447 | 2016-02-10 14:49:00 -0800 | [diff] [blame] | 296 | |
| 297 | // Wait until the client closes its socket. |
Josh Gao | e1dacfc | 2017-04-12 17:00:49 -0700 | [diff] [blame] | 298 | client_thread.join(); |
Josh Gao | 022d447 | 2016-02-10 14:49:00 -0800 | [diff] [blame] | 299 | |
Elliott Hughes | dbe91ee | 2016-11-15 12:37:32 -0800 | [diff] [blame] | 300 | std::this_thread::sleep_for(SLEEP_FOR_FDEVENT); |
Yabin Cui | 2407d7c | 2016-04-25 19:48:19 -0700 | [diff] [blame] | 301 | ASSERT_EQ(GetAdditionalLocalSocketCount(), fdevent_installed_count()); |
Josh Gao | 022d447 | 2016-02-10 14:49:00 -0800 | [diff] [blame] | 302 | TerminateThread(thread); |
Yabin Cui | c1b1f6f | 2015-09-15 16:27:09 -0700 | [diff] [blame] | 303 | } |
Yabin Cui | aa77e22 | 2015-09-29 12:25:33 -0700 | [diff] [blame] | 304 | |
| 305 | #endif // defined(__linux__) |
David Pursell | 3f902aa | 2016-03-01 08:58:26 -0800 | [diff] [blame] | 306 | |
| 307 | #if ADB_HOST |
| 308 | |
| 309 | // Checks that skip_host_serial(serial) returns a pointer to the part of |serial| which matches |
| 310 | // |expected|, otherwise logs the failure to gtest. |
Dan Austin | b4cff49 | 2016-03-28 15:32:37 -0700 | [diff] [blame] | 311 | void VerifySkipHostSerial(std::string serial, const char* expected) { |
| 312 | char* result = internal::skip_host_serial(&serial[0]); |
David Pursell | 3f902aa | 2016-03-01 08:58:26 -0800 | [diff] [blame] | 313 | if (expected == nullptr) { |
| 314 | EXPECT_EQ(nullptr, result); |
| 315 | } else { |
| 316 | EXPECT_STREQ(expected, result); |
| 317 | } |
| 318 | } |
| 319 | |
| 320 | // Check [tcp:|udp:]<serial>[:<port>]:<command> format. |
| 321 | TEST(socket_test, test_skip_host_serial) { |
| 322 | for (const std::string& protocol : {"", "tcp:", "udp:"}) { |
| 323 | VerifySkipHostSerial(protocol, nullptr); |
| 324 | VerifySkipHostSerial(protocol + "foo", nullptr); |
| 325 | |
| 326 | VerifySkipHostSerial(protocol + "foo:bar", ":bar"); |
| 327 | VerifySkipHostSerial(protocol + "foo:bar:baz", ":bar:baz"); |
| 328 | |
| 329 | VerifySkipHostSerial(protocol + "foo:123:bar", ":bar"); |
| 330 | VerifySkipHostSerial(protocol + "foo:123:456", ":456"); |
| 331 | VerifySkipHostSerial(protocol + "foo:123:bar:baz", ":bar:baz"); |
| 332 | |
| 333 | // Don't register a port unless it's all numbers and ends with ':'. |
| 334 | VerifySkipHostSerial(protocol + "foo:123", ":123"); |
| 335 | VerifySkipHostSerial(protocol + "foo:123bar:baz", ":123bar:baz"); |
David Pursell | 73d55aa | 2016-09-21 12:08:37 -0700 | [diff] [blame] | 336 | |
| 337 | VerifySkipHostSerial(protocol + "100.100.100.100:5555:foo", ":foo"); |
| 338 | VerifySkipHostSerial(protocol + "[0123:4567:89ab:CDEF:0:9:a:f]:5555:foo", ":foo"); |
| 339 | VerifySkipHostSerial(protocol + "[::1]:5555:foo", ":foo"); |
| 340 | |
| 341 | // If we can't find both [] then treat it as a normal serial with [ in it. |
| 342 | VerifySkipHostSerial(protocol + "[0123:foo", ":foo"); |
| 343 | |
| 344 | // Don't be fooled by random IPv6 addresses in the command string. |
| 345 | VerifySkipHostSerial(protocol + "foo:ping [0123:4567:89ab:CDEF:0:9:a:f]:5555", |
| 346 | ":ping [0123:4567:89ab:CDEF:0:9:a:f]:5555"); |
David Pursell | 3f902aa | 2016-03-01 08:58:26 -0800 | [diff] [blame] | 347 | } |
| 348 | } |
| 349 | |
| 350 | // Check <prefix>:<serial>:<command> format. |
| 351 | TEST(socket_test, test_skip_host_serial_prefix) { |
| 352 | for (const std::string& prefix : {"usb:", "product:", "model:", "device:"}) { |
| 353 | VerifySkipHostSerial(prefix, nullptr); |
| 354 | VerifySkipHostSerial(prefix + "foo", nullptr); |
| 355 | |
| 356 | VerifySkipHostSerial(prefix + "foo:bar", ":bar"); |
| 357 | VerifySkipHostSerial(prefix + "foo:bar:baz", ":bar:baz"); |
| 358 | VerifySkipHostSerial(prefix + "foo:123:bar", ":123:bar"); |
| 359 | } |
| 360 | } |
| 361 | |
| 362 | #endif // ADB_HOST |