| /* |
| * Copyright 2016 The Android Open Source Project |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| #define LOG_TAG "MtpFfsHandle_test.cpp" |
| |
| #include <android-base/unique_fd.h> |
| #include <android-base/test_utils.h> |
| #include <fcntl.h> |
| #include <gtest/gtest.h> |
| #include <memory> |
| #include <random> |
| #include <string> |
| #include <unistd.h> |
| #include <log/log.h> |
| |
| #include "MtpDescriptors.h" |
| #include "MtpFfsHandle.h" |
| #include "MtpFfsCompatHandle.h" |
| |
| namespace android { |
| |
| constexpr int TEST_PACKET_SIZE = 500; |
| constexpr int SMALL_MULT = 30; |
| constexpr int MED_MULT = 510; |
| |
| static const std::string dummyDataStr = |
| "/*\n * Copyright 2015 The Android Open Source Project\n *\n * Licensed un" |
| "der the Apache License, Version 2.0 (the \"License\");\n * you may not us" |
| "e this file except in compliance with the License.\n * You may obtain a c" |
| "opy of the License at\n *\n * http://www.apache.org/licenses/LICENSE" |
| "-2.0\n *\n * Unless required by applicable law or agreed to in writing, s" |
| "oftware\n * distributed under the License is distributed on an \"AS IS\" " |
| "BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express o" |
| "r im"; |
| |
| /** |
| * Functional tests for the MtpFfsHandle class. Ensures header and data integrity |
| * by mocking ffs endpoints as pipes to capture input / output. |
| */ |
| template <class T> |
| class MtpFfsHandleTest : public ::testing::Test { |
| protected: |
| std::unique_ptr<MtpFfsHandle> handle; |
| |
| // Pipes for reading endpoint data |
| android::base::unique_fd control; |
| android::base::unique_fd bulk_in; |
| android::base::unique_fd bulk_out; |
| android::base::unique_fd intr; |
| |
| TemporaryFile dummy_file; |
| |
| MtpFfsHandleTest() { |
| int fd[2]; |
| handle = std::make_unique<T>(-1); |
| |
| EXPECT_EQ(pipe(fd), 0); |
| control.reset(fd[0]); |
| handle->mControl.reset(fd[1]); |
| |
| EXPECT_EQ(pipe(fd), 0); |
| EXPECT_EQ(fcntl(fd[0], F_SETPIPE_SZ, 1048576), 1048576); |
| bulk_in.reset(fd[0]); |
| handle->mBulkIn.reset(fd[1]); |
| |
| EXPECT_EQ(pipe(fd), 0); |
| EXPECT_EQ(fcntl(fd[0], F_SETPIPE_SZ, 1048576), 1048576); |
| bulk_out.reset(fd[1]); |
| handle->mBulkOut.reset(fd[0]); |
| |
| EXPECT_EQ(pipe(fd), 0); |
| intr.reset(fd[0]); |
| handle->mIntr.reset(fd[1]); |
| |
| EXPECT_EQ(handle->start(false), 0); |
| } |
| |
| ~MtpFfsHandleTest() { |
| handle->close(); |
| } |
| }; |
| |
| typedef ::testing::Types<MtpFfsHandle, MtpFfsCompatHandle> mtpHandles; |
| TYPED_TEST_CASE(MtpFfsHandleTest, mtpHandles); |
| |
| TYPED_TEST(MtpFfsHandleTest, testMtpControl) { |
| EXPECT_TRUE(this->handle->writeDescriptors(false)); |
| struct desc_v2 desc; |
| struct functionfs_strings strings; |
| EXPECT_EQ(read(this->control, &desc, sizeof(desc)), (long)sizeof(desc)); |
| EXPECT_EQ(read(this->control, &strings, sizeof(strings)), (long)sizeof(strings)); |
| EXPECT_TRUE(std::memcmp(&desc, &mtp_desc_v2, sizeof(desc)) == 0); |
| EXPECT_TRUE(std::memcmp(&strings, &mtp_strings, sizeof(strings)) == 0); |
| } |
| |
| TYPED_TEST(MtpFfsHandleTest, testPtpControl) { |
| EXPECT_TRUE(this->handle->writeDescriptors(true)); |
| struct desc_v2 desc; |
| struct functionfs_strings strings; |
| EXPECT_EQ(read(this->control, &desc, sizeof(desc)), (long)sizeof(desc)); |
| EXPECT_EQ(read(this->control, &strings, sizeof(strings)), (long)sizeof(strings)); |
| EXPECT_TRUE(std::memcmp(&desc, &ptp_desc_v2, sizeof(desc)) == 0); |
| EXPECT_TRUE(std::memcmp(&strings, &mtp_strings, sizeof(strings)) == 0); |
| } |
| |
| TYPED_TEST(MtpFfsHandleTest, testRead) { |
| EXPECT_EQ(write(this->bulk_out, dummyDataStr.c_str(), TEST_PACKET_SIZE), TEST_PACKET_SIZE); |
| char buf[TEST_PACKET_SIZE + 1]; |
| buf[TEST_PACKET_SIZE] = '\0'; |
| EXPECT_EQ(this->handle->read(buf, TEST_PACKET_SIZE), TEST_PACKET_SIZE); |
| EXPECT_STREQ(buf, dummyDataStr.c_str()); |
| } |
| |
| TYPED_TEST(MtpFfsHandleTest, testReadLarge) { |
| std::stringstream ss; |
| int size = TEST_PACKET_SIZE * MED_MULT; |
| char buf[size + 1]; |
| buf[size] = '\0'; |
| |
| for (int i = 0; i < MED_MULT; i++) |
| ss << dummyDataStr; |
| |
| EXPECT_EQ(write(this->bulk_out, ss.str().c_str(), size), size); |
| EXPECT_EQ(this->handle->read(buf, size), size); |
| |
| EXPECT_STREQ(buf, ss.str().c_str()); |
| } |
| |
| TYPED_TEST(MtpFfsHandleTest, testWrite) { |
| char buf[TEST_PACKET_SIZE + 1]; |
| buf[TEST_PACKET_SIZE] = '\0'; |
| EXPECT_EQ(this->handle->write(dummyDataStr.c_str(), TEST_PACKET_SIZE), TEST_PACKET_SIZE); |
| EXPECT_EQ(read(this->bulk_in, buf, TEST_PACKET_SIZE), TEST_PACKET_SIZE); |
| EXPECT_STREQ(buf, dummyDataStr.c_str()); |
| } |
| |
| TYPED_TEST(MtpFfsHandleTest, testWriteLarge) { |
| std::stringstream ss; |
| int size = TEST_PACKET_SIZE * MED_MULT; |
| char buf[size + 1]; |
| buf[size] = '\0'; |
| |
| for (int i = 0; i < MED_MULT; i++) |
| ss << dummyDataStr; |
| |
| EXPECT_EQ(this->handle->write(ss.str().c_str(), size), size); |
| EXPECT_EQ(read(this->bulk_in, buf, size), size); |
| |
| EXPECT_STREQ(buf, ss.str().c_str()); |
| } |
| |
| TYPED_TEST(MtpFfsHandleTest, testReceiveFileEmpty) { |
| std::stringstream ss; |
| mtp_file_range mfr; |
| int size = 0; |
| char buf[size + 1]; |
| buf[size] = '\0'; |
| |
| mfr.offset = 0; |
| mfr.length = size; |
| mfr.fd = this->dummy_file.fd; |
| |
| EXPECT_EQ(write(this->bulk_out, ss.str().c_str(), size), size); |
| EXPECT_EQ(this->handle->receiveFile(mfr, false), 0); |
| |
| EXPECT_EQ(read(this->dummy_file.fd, buf, size), size); |
| } |
| |
| TYPED_TEST(MtpFfsHandleTest, testReceiveFileSmall) { |
| std::stringstream ss; |
| mtp_file_range mfr; |
| int size = TEST_PACKET_SIZE * SMALL_MULT; |
| char buf[size + 1]; |
| buf[size] = '\0'; |
| |
| mfr.offset = 0; |
| mfr.length = size; |
| mfr.fd = this->dummy_file.fd; |
| for (int i = 0; i < SMALL_MULT; i++) |
| ss << dummyDataStr; |
| |
| EXPECT_EQ(write(this->bulk_out, ss.str().c_str(), size), size); |
| EXPECT_EQ(this->handle->receiveFile(mfr, false), 0); |
| |
| EXPECT_EQ(read(this->dummy_file.fd, buf, size), size); |
| |
| EXPECT_STREQ(buf, ss.str().c_str()); |
| } |
| |
| TYPED_TEST(MtpFfsHandleTest, testReceiveFileMed) { |
| std::stringstream ss; |
| mtp_file_range mfr; |
| int size = TEST_PACKET_SIZE * MED_MULT; |
| char buf[size + 1]; |
| buf[size] = '\0'; |
| |
| mfr.offset = 0; |
| mfr.length = size; |
| mfr.fd = this->dummy_file.fd; |
| for (int i = 0; i < MED_MULT; i++) |
| ss << dummyDataStr; |
| |
| EXPECT_EQ(write(this->bulk_out, ss.str().c_str(), size), size); |
| EXPECT_EQ(this->handle->receiveFile(mfr, false), 0); |
| |
| EXPECT_EQ(read(this->dummy_file.fd, buf, size), size); |
| |
| EXPECT_STREQ(buf, ss.str().c_str()); |
| } |
| |
| TYPED_TEST(MtpFfsHandleTest, testReceiveFileMedPartial) { |
| std::stringstream ss; |
| mtp_file_range mfr; |
| int size = TEST_PACKET_SIZE * MED_MULT; |
| char buf[size + 1]; |
| buf[size] = '\0'; |
| |
| mfr.fd = this->dummy_file.fd; |
| for (int i = 0; i < MED_MULT; i++) |
| ss << dummyDataStr; |
| |
| EXPECT_EQ(write(this->bulk_out, ss.str().c_str(), size), size); |
| |
| std::random_device rd; |
| std::mt19937 gen(rd()); |
| std::uniform_int_distribution<> dis(1, TEST_PACKET_SIZE); |
| int offset = 0; |
| while (offset != size) { |
| mfr.offset = offset; |
| int length = std::min(size - offset, dis(gen)); |
| mfr.length = length; |
| |
| EXPECT_EQ(this->handle->receiveFile(mfr, false), 0); |
| offset += length; |
| } |
| |
| EXPECT_EQ(read(this->dummy_file.fd, buf, size), size); |
| |
| EXPECT_STREQ(buf, ss.str().c_str()); |
| } |
| |
| TYPED_TEST(MtpFfsHandleTest, testSendFileSmall) { |
| std::stringstream ss; |
| mtp_file_range mfr; |
| mfr.command = 42; |
| mfr.transaction_id = 1337; |
| mfr.offset = 0; |
| int size = TEST_PACKET_SIZE * SMALL_MULT; |
| char buf[size + sizeof(mtp_data_header) + 1]; |
| buf[size + sizeof(mtp_data_header)] = '\0'; |
| |
| mfr.length = size; |
| mfr.fd = this->dummy_file.fd; |
| for (int i = 0; i < SMALL_MULT; i++) |
| ss << dummyDataStr; |
| |
| EXPECT_EQ(write(this->dummy_file.fd, ss.str().c_str(), size), size); |
| EXPECT_EQ(this->handle->sendFile(mfr), 0); |
| |
| EXPECT_EQ(read(this->bulk_in, buf, size + sizeof(mtp_data_header)), |
| static_cast<long>(size + sizeof(mtp_data_header))); |
| |
| struct mtp_data_header *header = reinterpret_cast<struct mtp_data_header*>(buf); |
| EXPECT_STREQ(buf + sizeof(mtp_data_header), ss.str().c_str()); |
| EXPECT_EQ(header->length, static_cast<unsigned int>(size + sizeof(mtp_data_header))); |
| EXPECT_EQ(header->type, static_cast<unsigned int>(2)); |
| EXPECT_EQ(header->command, static_cast<unsigned int>(42)); |
| EXPECT_EQ(header->transaction_id, static_cast<unsigned int>(1337)); |
| } |
| |
| TYPED_TEST(MtpFfsHandleTest, testSendFileMed) { |
| std::stringstream ss; |
| mtp_file_range mfr; |
| mfr.command = 42; |
| mfr.transaction_id = 1337; |
| mfr.offset = 0; |
| int size = TEST_PACKET_SIZE * MED_MULT; |
| char buf[size + sizeof(mtp_data_header) + 1]; |
| buf[size + sizeof(mtp_data_header)] = '\0'; |
| |
| mfr.length = size; |
| mfr.fd = this->dummy_file.fd; |
| for (int i = 0; i < MED_MULT; i++) |
| ss << dummyDataStr; |
| |
| EXPECT_EQ(write(this->dummy_file.fd, ss.str().c_str(), size), size); |
| EXPECT_EQ(this->handle->sendFile(mfr), 0); |
| |
| EXPECT_EQ(read(this->bulk_in, buf, size + sizeof(mtp_data_header)), |
| static_cast<long>(size + sizeof(mtp_data_header))); |
| |
| struct mtp_data_header *header = reinterpret_cast<struct mtp_data_header*>(buf); |
| EXPECT_STREQ(buf + sizeof(mtp_data_header), ss.str().c_str()); |
| EXPECT_EQ(header->length, static_cast<unsigned int>(size + sizeof(mtp_data_header))); |
| EXPECT_EQ(header->type, static_cast<unsigned int>(2)); |
| EXPECT_EQ(header->command, static_cast<unsigned int>(42)); |
| EXPECT_EQ(header->transaction_id, static_cast<unsigned int>(1337)); |
| } |
| |
| TYPED_TEST(MtpFfsHandleTest, testSendFileMedPartial) { |
| std::stringstream ss; |
| mtp_file_range mfr; |
| mfr.fd = this->dummy_file.fd; |
| mfr.command = 42; |
| mfr.transaction_id = 1337; |
| int size = TEST_PACKET_SIZE * MED_MULT; |
| char buf[size + 1]; |
| buf[size] = '\0'; |
| |
| for (int i = 0; i < MED_MULT; i++) |
| ss << dummyDataStr; |
| |
| EXPECT_EQ(write(this->dummy_file.fd, ss.str().c_str(), size), size); |
| |
| std::random_device rd; |
| std::mt19937 gen(rd()); |
| std::uniform_int_distribution<> dis(1, TEST_PACKET_SIZE); |
| int offset = 0; |
| while (offset != size) { |
| mfr.offset = offset; |
| int length = std::min(size - offset, dis(gen)); |
| mfr.length = length; |
| char temp_buf[length + sizeof(mtp_data_header)]; |
| EXPECT_EQ(this->handle->sendFile(mfr), 0); |
| |
| EXPECT_EQ(read(this->bulk_in, temp_buf, length + sizeof(mtp_data_header)), |
| static_cast<long>(length + sizeof(mtp_data_header))); |
| |
| struct mtp_data_header *header = reinterpret_cast<struct mtp_data_header*>(temp_buf); |
| EXPECT_EQ(header->length, static_cast<unsigned int>(length + sizeof(mtp_data_header))); |
| EXPECT_EQ(header->type, static_cast<unsigned int>(2)); |
| EXPECT_EQ(header->command, static_cast<unsigned int>(42)); |
| EXPECT_EQ(header->transaction_id, static_cast<unsigned int>(1337)); |
| memcpy(buf + offset, temp_buf + sizeof(mtp_data_header), length); |
| offset += length; |
| } |
| EXPECT_STREQ(buf, ss.str().c_str()); |
| } |
| |
| TYPED_TEST(MtpFfsHandleTest, testSendFileEmpty) { |
| mtp_file_range mfr; |
| mfr.command = 42; |
| mfr.transaction_id = 1337; |
| mfr.offset = 0; |
| int size = 0; |
| char buf[size + sizeof(mtp_data_header) + 1]; |
| buf[size + sizeof(mtp_data_header)] = '\0'; |
| |
| mfr.length = size; |
| mfr.fd = this->dummy_file.fd; |
| |
| EXPECT_EQ(this->handle->sendFile(mfr), 0); |
| |
| EXPECT_EQ(read(this->bulk_in, buf, size + sizeof(mtp_data_header)), |
| static_cast<long>(size + sizeof(mtp_data_header))); |
| |
| struct mtp_data_header *header = reinterpret_cast<struct mtp_data_header*>(buf); |
| EXPECT_EQ(header->length, static_cast<unsigned int>(size + sizeof(mtp_data_header))); |
| EXPECT_EQ(header->type, static_cast<unsigned int>(2)); |
| EXPECT_EQ(header->command, static_cast<unsigned int>(42)); |
| EXPECT_EQ(header->transaction_id, static_cast<unsigned int>(1337)); |
| } |
| |
| TYPED_TEST(MtpFfsHandleTest, testSendEvent) { |
| struct mtp_event event; |
| event.length = TEST_PACKET_SIZE; |
| event.data = const_cast<char*>(dummyDataStr.c_str()); |
| char buf[TEST_PACKET_SIZE + 1]; |
| buf[TEST_PACKET_SIZE] = '\0'; |
| |
| this->handle->sendEvent(event); |
| read(this->intr, buf, TEST_PACKET_SIZE); |
| EXPECT_STREQ(buf, dummyDataStr.c_str()); |
| } |
| |
| } // namespace android |