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 ASSERT_FALSE(directory_exists(subdir(profiles_dir, "does-not-exist"))); 56 #else 57 ASSERT_TRUE(directory_exists("/proc")); 58 ASSERT_FALSE(directory_exists("/proc/self")); // Symbolic link. 59 ASSERT_FALSE(directory_exists("/proc/does-not-exist")); 60 #endif 61 } 62 63 #if defined(_WIN32) 64 TEST(adb_utils, directory_exists_win32_symlink_junction) { 65 char profiles_dir[MAX_PATH]; 66 DWORD cch = arraysize(profiles_dir); 67 68 // On typical Windows 7, returns C:\Users 69 ASSERT_TRUE(GetProfilesDirectoryA(profiles_dir, &cch)); 70 71 // On modern (English?) Windows, this is a directory symbolic link to 72 // C:\ProgramData. Symbolic links are rare on Windows and the user requires 73 // a special permission (by default granted to Administrative users) to 74 // create symbolic links. 75 EXPECT_FALSE(directory_exists(subdir(profiles_dir, "All Users"))); 76 77 // On modern (English?) Windows, this is a directory junction to 78 // C:\Users\Default. Junctions are used throughout user profile directories 79 // for backwards compatibility and they don't require any special permissions 80 // to create. 81 EXPECT_FALSE(directory_exists(subdir(profiles_dir, "Default User"))); 82 } 83 #endif 84 85 TEST(adb_utils, escape_arg) { 86 ASSERT_EQ(R"('')", escape_arg("")); 87 88 ASSERT_EQ(R"('abc')", escape_arg("abc")); 89 90 ASSERT_EQ(R"(' abc')", escape_arg(" abc")); 91 ASSERT_EQ(R"(''\''abc')", escape_arg("'abc")); 92 ASSERT_EQ(R"('"abc')", escape_arg("\"abc")); 93 ASSERT_EQ(R"('\abc')", escape_arg("\\abc")); 94 ASSERT_EQ(R"('(abc')", escape_arg("(abc")); 95 ASSERT_EQ(R"(')abc')", escape_arg(")abc")); 96 97 ASSERT_EQ(R"('abc abc')", escape_arg("abc abc")); 98 ASSERT_EQ(R"('abc'\''abc')", escape_arg("abc'abc")); 99 ASSERT_EQ(R"('abc"abc')", escape_arg("abc\"abc")); 100 ASSERT_EQ(R"('abc\abc')", escape_arg("abc\\abc")); 101 ASSERT_EQ(R"('abc(abc')", escape_arg("abc(abc")); 102 ASSERT_EQ(R"('abc)abc')", escape_arg("abc)abc")); 103 104 ASSERT_EQ(R"('abc ')", escape_arg("abc ")); 105 ASSERT_EQ(R"('abc'\''')", escape_arg("abc'")); 106 ASSERT_EQ(R"('abc"')", escape_arg("abc\"")); 107 ASSERT_EQ(R"('abc\')", escape_arg("abc\\")); 108 ASSERT_EQ(R"('abc(')", escape_arg("abc(")); 109 ASSERT_EQ(R"('abc)')", escape_arg("abc)")); 110 } 111 112 TEST(adb_utils, adb_basename) { 113 EXPECT_EQ("sh", adb_basename("/system/bin/sh")); 114 EXPECT_EQ("sh", adb_basename("sh")); 115 EXPECT_EQ("sh", adb_basename("/system/bin/sh/")); 116 } 117 118 TEST(adb_utils, adb_dirname) { 119 EXPECT_EQ("/system/bin", adb_dirname("/system/bin/sh")); 120 EXPECT_EQ(".", adb_dirname("sh")); 121 EXPECT_EQ("/system/bin", adb_dirname("/system/bin/sh/")); 122 } 123 124 void test_mkdirs(const std::string basepath) { 125 // Test creating a directory hierarchy. 126 ASSERT_TRUE(mkdirs(basepath)); 127 // Test finding an existing directory hierarchy. 128 ASSERT_TRUE(mkdirs(basepath)); 129 // Test mkdirs on an existing hierarchy with a trailing slash. 130 ASSERT_TRUE(mkdirs(basepath + '/')); 131 #if defined(_WIN32) 132 ASSERT_TRUE(mkdirs(basepath + '\\')); 133 #endif 134 135 const std::string filepath = basepath + "/file"; 136 // Verify that the hierarchy was created by trying to create a file in it. 137 ASSERT_NE(-1, adb_creat(filepath.c_str(), 0600)); 138 // If a file exists where we want a directory, the operation should fail. 139 ASSERT_FALSE(mkdirs(filepath)); 140 } 141 142 TEST(adb_utils, mkdirs) { 143 TemporaryDir td; 144 145 // Absolute paths. 146 test_mkdirs(std::string(td.path) + "/dir/subdir"); 147 148 // Relative paths. 149 ASSERT_EQ(0, chdir(td.path)) << strerror(errno); 150 test_mkdirs(std::string("relative/subrel")); 151 } 152 153 #if !defined(_WIN32) 154 TEST(adb_utils, set_file_block_mode) { 155 int fd = adb_open("/dev/null", O_RDWR | O_APPEND); 156 ASSERT_GE(fd, 0); 157 int flags = fcntl(fd, F_GETFL, 0); 158 ASSERT_EQ(O_RDWR | O_APPEND, (flags & (O_RDWR | O_APPEND))); 159 ASSERT_TRUE(set_file_block_mode(fd, false)); 160 int new_flags = fcntl(fd, F_GETFL, 0); 161 ASSERT_EQ(flags | O_NONBLOCK, new_flags); 162 ASSERT_TRUE(set_file_block_mode(fd, true)); 163 new_flags = fcntl(fd, F_GETFL, 0); 164 ASSERT_EQ(flags, new_flags); 165 ASSERT_EQ(0, adb_close(fd)); 166 } 167 #endif 168