adb/adb_utils_test.cpp

   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 <base/macros.h>
  34 #include <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 TEST(adb_utils, parse_host_and_port) {
 115   std::string canonical_address;
 116   std::string host;
 117   int port;
 118   std::string error;
 119
 120   // Name, default port.
 121   port = 123;
 122   ASSERT_TRUE(parse_host_and_port("www.google.com", &canonical_address, &host, &port, &error));
 123   ASSERT_EQ("www.google.com:123", canonical_address);
 124   ASSERT_EQ("www.google.com", host);
 125   ASSERT_EQ(123, port);
 126
 127   // Name, explicit port.
 128   ASSERT_TRUE(parse_host_and_port("www.google.com:666", &canonical_address, &host, &port, &error));
 129   ASSERT_EQ("www.google.com:666", canonical_address);
 130   ASSERT_EQ("www.google.com", host);
 131   ASSERT_EQ(666, port);
 132
 133   // IPv4, default port.
 134   port = 123;
 135   ASSERT_TRUE(parse_host_and_port("1.2.3.4", &canonical_address, &host, &port, &error));
 136   ASSERT_EQ("1.2.3.4:123", canonical_address);
 137   ASSERT_EQ("1.2.3.4", host);
 138   ASSERT_EQ(123, port);
 139
 140   // IPv4, explicit port.
 141   ASSERT_TRUE(parse_host_and_port("1.2.3.4:666", &canonical_address, &host, &port, &error));
 142   ASSERT_EQ("1.2.3.4:666", canonical_address);
 143   ASSERT_EQ("1.2.3.4", host);
 144   ASSERT_EQ(666, port);
 145
 146   // Simple IPv6, default port.
 147   port = 123;
 148   ASSERT_TRUE(parse_host_and_port("::1", &canonical_address, &host, &port, &error));
 149   ASSERT_EQ("[::1]:123", canonical_address);
 150   ASSERT_EQ("::1", host);
 151   ASSERT_EQ(123, port);
 152
 153   // Simple IPv6, explicit port.
 154   ASSERT_TRUE(parse_host_and_port("[::1]:666", &canonical_address, &host, &port, &error));
 155   ASSERT_EQ("[::1]:666", canonical_address);
 156   ASSERT_EQ("::1", host);
 157   ASSERT_EQ(666, port);
 158
 159   // Hairy IPv6, default port.
 160   port = 123;
 161   ASSERT_TRUE(parse_host_and_port("fe80::200:5aee:feaa:20a2", &canonical_address, &host, &port, &error));
 162   ASSERT_EQ("[fe80::200:5aee:feaa:20a2]:123", canonical_address);
 163   ASSERT_EQ("fe80::200:5aee:feaa:20a2", host);
 164   ASSERT_EQ(123, port);
 165
 166   // Simple IPv6, explicit port.
 167   ASSERT_TRUE(parse_host_and_port("[fe80::200:5aee:feaa:20a2]:666", &canonical_address, &host, &port, &error));
 168   ASSERT_EQ("[fe80::200:5aee:feaa:20a2]:666", canonical_address);
 169   ASSERT_EQ("fe80::200:5aee:feaa:20a2", host);
 170   ASSERT_EQ(666, port);
 171
 172   // Invalid IPv4.
 173   EXPECT_FALSE(parse_host_and_port("1.2.3.4:", &canonical_address, &host, &port, &error));
 174   EXPECT_FALSE(parse_host_and_port("1.2.3.4::", &canonical_address, &host, &port, &error));
 175   EXPECT_FALSE(parse_host_and_port("1.2.3.4:hello", &canonical_address, &host, &port, &error));
 176   EXPECT_FALSE(parse_host_and_port(":123", &canonical_address, &host, &port, &error));
 177
 178   // Invalid IPv6.
 179   EXPECT_FALSE(parse_host_and_port(":1", &canonical_address, &host, &port, &error));
 180   EXPECT_FALSE(parse_host_and_port("::::::::1", &canonical_address, &host, &port, &error));
 181   EXPECT_FALSE(parse_host_and_port("[::1", &canonical_address, &host, &port, &error));
 182   EXPECT_FALSE(parse_host_and_port("[::1]", &canonical_address, &host, &port, &error));
 183   EXPECT_FALSE(parse_host_and_port("[::1]:", &canonical_address, &host, &port, &error));
 184   EXPECT_FALSE(parse_host_and_port("[::1]::", &canonical_address, &host, &port, &error));
 185   EXPECT_FALSE(parse_host_and_port("[::1]:hello", &canonical_address, &host, &port, &error));
 186
 187   // Invalid ports.
 188   EXPECT_FALSE(parse_host_and_port("[::1]:-1", &canonical_address, &host, &port, &error));
 189   EXPECT_FALSE(parse_host_and_port("[::1]:0", &canonical_address, &host, &port, &error));
 190   EXPECT_FALSE(parse_host_and_port("[::1]:65536", &canonical_address, &host, &port, &error));
 191   EXPECT_FALSE(parse_host_and_port("1.2.3.4:-1", &canonical_address, &host, &port, &error));
 192   EXPECT_FALSE(parse_host_and_port("1.2.3.4:0", &canonical_address, &host, &port, &error));
 193   EXPECT_FALSE(parse_host_and_port("1.2.3.4:65536", &canonical_address, &host, &port, &error));
 194 }
 195
 196 void test_mkdirs(const std::string basepath) {
 197   EXPECT_TRUE(mkdirs(basepath));
 198   EXPECT_NE(-1, adb_creat(basepath.c_str(), 0600));
 199   EXPECT_FALSE(mkdirs(basepath + "/subdir/"));
 200 }
 201
 202 TEST(adb_utils, mkdirs) {
 203   TemporaryDir td;
 204
 205   // Absolute paths.
 206   test_mkdirs(std::string(td.path) + "/dir/subdir/file");
 207
 208   // Relative paths.
 209   ASSERT_EQ(0, chdir(td.path)) << strerror(errno);
 210   test_mkdirs(std::string("relative/subrel/file"));
 211 }
 212
 213 #if !defined(_WIN32)
 214 TEST(adb_utils, set_file_block_mode) {
 215   int fd = adb_open("/dev/null", O_RDWR | O_APPEND);
 216   ASSERT_GE(fd, 0);
 217   int flags = fcntl(fd, F_GETFL, 0);
 218   ASSERT_EQ(O_RDWR | O_APPEND, (flags & (O_RDWR | O_APPEND)));
 219   ASSERT_TRUE(set_file_block_mode(fd, false));
 220   int new_flags = fcntl(fd, F_GETFL, 0);
 221   ASSERT_EQ(flags | O_NONBLOCK, new_flags);
 222   ASSERT_TRUE(set_file_block_mode(fd, true));
 223   new_flags = fcntl(fd, F_GETFL, 0);
 224   ASSERT_EQ(flags, new_flags);
 225   ASSERT_EQ(0, adb_close(fd));
 226 }
 227 #endif