1b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell/* 2b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell * Copyright (C) 2015 The Android Open Source Project 3b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell * 4b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell * Licensed under the Apache License, Version 2.0 (the "License"); 5b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell * you may not use this file except in compliance with the License. 6b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell * You may obtain a copy of the License at 7b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell * 8b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell * http://www.apache.org/licenses/LICENSE-2.0 9b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell * 10b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell * Unless required by applicable law or agreed to in writing, software 11b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell * distributed under the License is distributed on an "AS IS" BASIS, 12b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell * See the License for the specific language governing permissions and 14b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell * limitations under the License. 15b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell */ 16b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell 17b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell#include "shell_service.h" 18b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell 19b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell#include <gtest/gtest.h> 20b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell 21b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell#include <signal.h> 22b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell#include <string.h> 23b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell 24b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell#include "sysdeps.h" 25b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell 26b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursellclass ShellProtocolTest : public ::testing::Test { 27b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell public: 28b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell static void SetUpTestCase() { 29b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell#if !defined(_WIN32) 30b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell // This is normally done in main.cpp. 31b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell saved_sigpipe_handler_ = signal(SIGPIPE, SIG_IGN); 32b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell#endif 33b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell } 34b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell 35b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell static void TearDownTestCase() { 36b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell#if !defined(_WIN32) 37b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell signal(SIGPIPE, saved_sigpipe_handler_); 38b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell#endif 39b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell } 40b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell 41b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell // Initializes the socketpair and ShellProtocols needed for testing. 42b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell void SetUp() { 43b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell int fds[2]; 44b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell ASSERT_EQ(0, adb_socketpair(fds)); 45b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell read_fd_ = fds[0]; 46b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell write_fd_ = fds[1]; 47b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell 48b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell write_protocol_ = new ShellProtocol(write_fd_); 49b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell ASSERT_TRUE(write_protocol_ != nullptr); 50b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell 51b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell read_protocol_ = new ShellProtocol(read_fd_); 52b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell ASSERT_TRUE(read_protocol_ != nullptr); 53b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell } 54b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell 55b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell // Cleans up FDs and ShellProtocols. If an FD is closed manually during a 56b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell // test, set it to -1 to prevent TearDown() trying to close it again. 57b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell void TearDown() { 58b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell for (int fd : {read_fd_, write_fd_}) { 59b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell if (fd >= 0) { 60b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell adb_close(fd); 61b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell } 62b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell } 63b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell for (ShellProtocol* protocol : {read_protocol_, write_protocol_}) { 64b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell if (protocol) { 65b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell delete protocol; 66b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell } 67b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell } 68b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell } 69b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell 70b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell // Fakes the buffer size so we can test filling buffers. 71b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell void SetReadDataCapacity(size_t size) { 72b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell read_protocol_->buffer_end_ = read_protocol_->data() + size; 73b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell } 74b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell 75b404dec85ab07043c0a1933671aa76a159d797acDavid Pursell#if !defined(_WIN32) 76b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell static sig_t saved_sigpipe_handler_; 77b404dec85ab07043c0a1933671aa76a159d797acDavid Pursell#endif 78b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell 79b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell int read_fd_ = -1, write_fd_ = -1; 80b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell ShellProtocol *read_protocol_ = nullptr, *write_protocol_ = nullptr; 81b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell}; 82b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell 83b404dec85ab07043c0a1933671aa76a159d797acDavid Pursell#if !defined(_WIN32) 84b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursellsig_t ShellProtocolTest::saved_sigpipe_handler_ = nullptr; 85b404dec85ab07043c0a1933671aa76a159d797acDavid Pursell#endif 86b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell 87b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursellnamespace { 88b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell 89b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell// Returns true if the packet contains the given values. 90b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursellbool PacketEquals(const ShellProtocol* protocol, ShellProtocol::Id id, 91b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell const void* data, size_t data_length) { 92b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell return (protocol->id() == id && 93b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell protocol->data_length() == data_length && 94b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell !memcmp(data, protocol->data(), data_length)); 95b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell} 96b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell 97b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell} // namespace 98b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell 99b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell// Tests data that can fit in a single packet. 100b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David PursellTEST_F(ShellProtocolTest, FullPacket) { 101b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell ShellProtocol::Id id = ShellProtocol::kIdStdout; 102b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell char data[] = "abc 123 \0\r\n"; 103b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell 104b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell memcpy(write_protocol_->data(), data, sizeof(data)); 105b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell ASSERT_TRUE(write_protocol_->Write(id, sizeof(data))); 106b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell 107b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell ASSERT_TRUE(read_protocol_->Read()); 108b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell ASSERT_TRUE(PacketEquals(read_protocol_, id, data, sizeof(data))); 109b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell} 110b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell 111b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell// Tests data that has to be read multiple times due to smaller read buffer. 112b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David PursellTEST_F(ShellProtocolTest, ReadBufferOverflow) { 113b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell ShellProtocol::Id id = ShellProtocol::kIdStdin; 114b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell 115b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell memcpy(write_protocol_->data(), "1234567890", 10); 116b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell ASSERT_TRUE(write_protocol_->Write(id, 10)); 117b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell 118b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell SetReadDataCapacity(4); 119b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell ASSERT_TRUE(read_protocol_->Read()); 120b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell ASSERT_TRUE(PacketEquals(read_protocol_, id, "1234", 4)); 121b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell ASSERT_TRUE(read_protocol_->Read()); 122b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell ASSERT_TRUE(PacketEquals(read_protocol_, id, "5678", 4)); 123b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell ASSERT_TRUE(read_protocol_->Read()); 124b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell ASSERT_TRUE(PacketEquals(read_protocol_, id, "90", 2)); 125b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell} 126b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell 127b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell// Tests a zero length packet. 128b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David PursellTEST_F(ShellProtocolTest, ZeroLengthPacket) { 129b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell ShellProtocol::Id id = ShellProtocol::kIdStderr; 130b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell 131b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell ASSERT_TRUE(write_protocol_->Write(id, 0)); 132b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell ASSERT_TRUE(read_protocol_->Read()); 133b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell ASSERT_TRUE(PacketEquals(read_protocol_, id, nullptr, 0)); 134b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell} 135b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell 136b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell// Tests exit code packets. 137b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David PursellTEST_F(ShellProtocolTest, ExitCodePacket) { 138b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell write_protocol_->data()[0] = 20; 139b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell ASSERT_TRUE(write_protocol_->Write(ShellProtocol::kIdExit, 1)); 140b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell 141b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell ASSERT_TRUE(read_protocol_->Read()); 142b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell ASSERT_EQ(ShellProtocol::kIdExit, read_protocol_->id()); 143b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell ASSERT_EQ(20, read_protocol_->data()[0]); 144b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell} 145b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell 146b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell// Tests writing to a closed pipe. 147b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David PursellTEST_F(ShellProtocolTest, WriteToClosedPipeFail) { 148b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell adb_close(read_fd_); 149b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell read_fd_ = -1; 150b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell 151b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell ASSERT_FALSE(write_protocol_->Write(ShellProtocol::kIdStdout, 0)); 152b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell} 153b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell 154b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell// Tests writing to a closed FD. 155b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David PursellTEST_F(ShellProtocolTest, WriteToClosedFdFail) { 156b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell adb_close(write_fd_); 157b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell write_fd_ = -1; 158b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell 159b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell ASSERT_FALSE(write_protocol_->Write(ShellProtocol::kIdStdout, 0)); 160b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell} 161b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell 162b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell// Tests reading from a closed pipe. 163b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David PursellTEST_F(ShellProtocolTest, ReadFromClosedPipeFail) { 164b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell adb_close(write_fd_); 165b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell write_fd_ = -1; 166b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell 167b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell ASSERT_FALSE(read_protocol_->Read()); 168b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell} 169b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell 170b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell// Tests reading from a closed FD. 171b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David PursellTEST_F(ShellProtocolTest, ReadFromClosedFdFail) { 172b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell adb_close(read_fd_); 173b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell read_fd_ = -1; 174b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell 175b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell ASSERT_FALSE(read_protocol_->Read()); 176b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell} 177b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell 178b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell// Tests reading from a closed pipe that has a packet waiting. This checks that 179b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell// even if the pipe closes before we can fully read its contents we will still 180b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell// be able to access the last packets. 181b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David PursellTEST_F(ShellProtocolTest, ReadPacketFromClosedPipe) { 182b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell ShellProtocol::Id id = ShellProtocol::kIdStdout; 183b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell char data[] = "foo bar"; 184b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell 185b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell memcpy(write_protocol_->data(), data, sizeof(data)); 186b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell ASSERT_TRUE(write_protocol_->Write(id, sizeof(data))); 187b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell adb_close(write_fd_); 188b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell write_fd_ = -1; 189b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell 190b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell // First read should grab the packet. 191b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell ASSERT_TRUE(read_protocol_->Read()); 192b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell ASSERT_TRUE(PacketEquals(read_protocol_, id, data, sizeof(data))); 193b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell 194b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell // Second read should fail. 195b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell ASSERT_FALSE(read_protocol_->Read()); 196b9e2e84e7e3cc5c9f30d63ef7d1337363437d2c7David Pursell} 197