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 "adb_utils.h" 18 19 #ifdef _WIN32 20 #include <windows.h> 21 #include <userenv.h> 22 #endif 23 24 #include <string> 25 26 #include <gtest/gtest.h> 27 28 #include <stdlib.h> 29 #include <string.h> 30 31 #include "sysdeps.h" 32 33 #include <android-base/macros.h> 34 #include <android-base/test_utils.h> 35 36 #ifdef _WIN32 37 static std::string subdir(const char* parent, const char* child) { 38 std::string str(parent); 39 str += OS_PATH_SEPARATOR; 40 str += child; 41 return str; 42 } 43 #endif 44 45 TEST(adb_utils, directory_exists) { 46 #ifdef _WIN32 47 char profiles_dir[MAX_PATH]; 48 DWORD cch = arraysize(profiles_dir); 49 50 // On typical Windows 7, returns C:\Users 51 ASSERT_TRUE(GetProfilesDirectoryA(profiles_dir, &cch)); 52 53 ASSERT_TRUE(directory_exists(profiles_dir)); 54 55 // On modern (English?) Windows, this is a directory symbolic link to 56 // C:\ProgramData. Symbolic links are rare on Windows and the user requires 57 // a special permission (by default granted to Administrative users) to 58 // create symbolic links. 59 ASSERT_FALSE(directory_exists(subdir(profiles_dir, "All Users"))); 60 61 // On modern (English?) Windows, this is a directory junction to 62 // C:\Users\Default. Junctions are used throughout user profile directories 63 // for backwards compatibility and they don't require any special permissions 64 // to create. 65 ASSERT_FALSE(directory_exists(subdir(profiles_dir, "Default User"))); 66 67 ASSERT_FALSE(directory_exists(subdir(profiles_dir, "does-not-exist"))); 68 #else 69 ASSERT_TRUE(directory_exists("/proc")); 70 ASSERT_FALSE(directory_exists("/proc/self")); // Symbolic link. 71 ASSERT_FALSE(directory_exists("/proc/does-not-exist")); 72 #endif 73 } 74 75 TEST(adb_utils, escape_arg) { 76 ASSERT_EQ(R"('')", escape_arg("")); 77 78 ASSERT_EQ(R"('abc')", escape_arg("abc")); 79 80 ASSERT_EQ(R"(' abc')", escape_arg(" abc")); 81 ASSERT_EQ(R"(''\''abc')", escape_arg("'abc")); 82 ASSERT_EQ(R"('"abc')", escape_arg("\"abc")); 83 ASSERT_EQ(R"('\abc')", escape_arg("\\abc")); 84 ASSERT_EQ(R"('(abc')", escape_arg("(abc")); 85 ASSERT_EQ(R"(')abc')", escape_arg(")abc")); 86 87 ASSERT_EQ(R"('abc abc')", escape_arg("abc abc")); 88 ASSERT_EQ(R"('abc'\''abc')", escape_arg("abc'abc")); 89 ASSERT_EQ(R"('abc"abc')", escape_arg("abc\"abc")); 90 ASSERT_EQ(R"('abc\abc')", escape_arg("abc\\abc")); 91 ASSERT_EQ(R"('abc(abc')", escape_arg("abc(abc")); 92 ASSERT_EQ(R"('abc)abc')", escape_arg("abc)abc")); 93 94 ASSERT_EQ(R"('abc ')", escape_arg("abc ")); 95 ASSERT_EQ(R"('abc'\''')", escape_arg("abc'")); 96 ASSERT_EQ(R"('abc"')", escape_arg("abc\"")); 97 ASSERT_EQ(R"('abc\')", escape_arg("abc\\")); 98 ASSERT_EQ(R"('abc(')", escape_arg("abc(")); 99 ASSERT_EQ(R"('abc)')", escape_arg("abc)")); 100 } 101 102 TEST(adb_utils, adb_basename) { 103 EXPECT_EQ("sh", adb_basename("/system/bin/sh")); 104 EXPECT_EQ("sh", adb_basename("sh")); 105 EXPECT_EQ("sh", adb_basename("/system/bin/sh/")); 106 } 107 108 TEST(adb_utils, adb_dirname) { 109 EXPECT_EQ("/system/bin", adb_dirname("/system/bin/sh")); 110 EXPECT_EQ(".", adb_dirname("sh")); 111 EXPECT_EQ("/system/bin", adb_dirname("/system/bin/sh/")); 112 } 113 114 void test_mkdirs(const std::string basepath) { 115 // Test creating a directory hierarchy. 116 EXPECT_TRUE(mkdirs(basepath)); 117 // Test finding an existing directory hierarchy. 118 EXPECT_TRUE(mkdirs(basepath)); 119 const std::string filepath = basepath + "/file"; 120 // Verify that the hierarchy was created by trying to create a file in it. 121 EXPECT_NE(-1, adb_creat(filepath.c_str(), 0600)); 122 // If a file exists where we want a directory, the operation should fail. 123 EXPECT_FALSE(mkdirs(filepath)); 124 } 125 126 TEST(adb_utils, mkdirs) { 127 TemporaryDir td; 128 129 // Absolute paths. 130 test_mkdirs(std::string(td.path) + "/dir/subdir"); 131 132 // Relative paths. 133 ASSERT_EQ(0, chdir(td.path)) << strerror(errno); 134 test_mkdirs(std::string("relative/subrel")); 135 } 136 137 #if !defined(_WIN32) 138 TEST(adb_utils, set_file_block_mode) { 139 int fd = adb_open("/dev/null", O_RDWR | O_APPEND); 140 ASSERT_GE(fd, 0); 141 int flags = fcntl(fd, F_GETFL, 0); 142 ASSERT_EQ(O_RDWR | O_APPEND, (flags & (O_RDWR | O_APPEND))); 143 ASSERT_TRUE(set_file_block_mode(fd, false)); 144 int new_flags = fcntl(fd, F_GETFL, 0); 145 ASSERT_EQ(flags | O_NONBLOCK, new_flags); 146 ASSERT_TRUE(set_file_block_mode(fd, true)); 147 new_flags = fcntl(fd, F_GETFL, 0); 148 ASSERT_EQ(flags, new_flags); 149 ASSERT_EQ(0, adb_close(fd)); 150 } 151 #endif 152
