blob: aabc5d73385484baaef781be945d83ce7cd827c8 [file] [log] [blame]
Elliott Hughes58305772015-04-17 13:57:15 -07001/*
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
Spencer Lowcf168a82015-05-24 15:36:28 -070019#ifdef _WIN32
20#include <windows.h>
21#include <userenv.h>
22#endif
23
24#include <string>
25
Elliott Hughes58305772015-04-17 13:57:15 -070026#include <gtest/gtest.h>
27
Alex Vallée47d67c92015-05-06 16:26:00 -040028#include <stdlib.h>
29#include <string.h>
30
31#include "sysdeps.h"
32
Elliott Hughes4f713192015-12-04 22:00:26 -080033#include <android-base/macros.h>
34#include <android-base/test_utils.h>
Alex Vallée47d67c92015-05-06 16:26:00 -040035
Spencer Lowcf168a82015-05-24 15:36:28 -070036#ifdef _WIN32
37static 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
Elliott Hughes58305772015-04-17 13:57:15 -070045TEST(adb_utils, directory_exists) {
Spencer Lowcf168a82015-05-24 15:36:28 -070046#ifdef _WIN32
47 char profiles_dir[MAX_PATH];
48 DWORD cch = arraysize(profiles_dir);
49
50 // On typical Windows 7, returns C:\Users
Spencer Lowcbf26b72015-08-03 20:43:24 -070051 ASSERT_TRUE(GetProfilesDirectoryA(profiles_dir, &cch));
Spencer Lowcf168a82015-05-24 15:36:28 -070052
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
Elliott Hughes58305772015-04-17 13:57:15 -070069 ASSERT_TRUE(directory_exists("/proc"));
70 ASSERT_FALSE(directory_exists("/proc/self")); // Symbolic link.
71 ASSERT_FALSE(directory_exists("/proc/does-not-exist"));
Spencer Lowcf168a82015-05-24 15:36:28 -070072#endif
Elliott Hughes58305772015-04-17 13:57:15 -070073}
74
75TEST(adb_utils, escape_arg) {
Elliott Hughes5498ade2015-04-17 20:50:11 -070076 ASSERT_EQ(R"('')", escape_arg(""));
Elliott Hughes58305772015-04-17 13:57:15 -070077
Elliott Hughes5498ade2015-04-17 20:50:11 -070078 ASSERT_EQ(R"('abc')", escape_arg("abc"));
Elliott Hughes58305772015-04-17 13:57:15 -070079
Elliott Hughes5498ade2015-04-17 20:50:11 -070080 ASSERT_EQ(R"(' abc')", escape_arg(" abc"));
Elliott Hughes84b0bf22015-05-15 12:06:00 -070081 ASSERT_EQ(R"(''\''abc')", escape_arg("'abc"));
Elliott Hughes5498ade2015-04-17 20:50:11 -070082 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"));
Elliott Hughes58305772015-04-17 13:57:15 -070086
Elliott Hughes5498ade2015-04-17 20:50:11 -070087 ASSERT_EQ(R"('abc abc')", escape_arg("abc abc"));
Elliott Hughes84b0bf22015-05-15 12:06:00 -070088 ASSERT_EQ(R"('abc'\''abc')", escape_arg("abc'abc"));
Elliott Hughes5498ade2015-04-17 20:50:11 -070089 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"));
Elliott Hughes58305772015-04-17 13:57:15 -070093
Elliott Hughes5498ade2015-04-17 20:50:11 -070094 ASSERT_EQ(R"('abc ')", escape_arg("abc "));
Elliott Hughes84b0bf22015-05-15 12:06:00 -070095 ASSERT_EQ(R"('abc'\''')", escape_arg("abc'"));
Elliott Hughes5498ade2015-04-17 20:50:11 -070096 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)"));
Elliott Hughes58305772015-04-17 13:57:15 -0700100}
Elliott Hughes3d5f60d2015-07-18 12:21:30 -0700101
Elliott Hughes5c742702015-07-30 17:42:01 -0700102TEST(adb_utils, adb_basename) {
103 EXPECT_EQ("sh", adb_basename("/system/bin/sh"));
104 EXPECT_EQ("sh", adb_basename("sh"));
Josh Gao787f3442015-11-03 18:52:35 -0800105 EXPECT_EQ("sh", adb_basename("/system/bin/sh/"));
106}
107
108TEST(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/"));
Elliott Hughes5c742702015-07-30 17:42:01 -0700112}
113
Spencer Low22191c32015-08-01 17:29:23 -0700114void test_mkdirs(const std::string basepath) {
Spencer Low85c45bd2016-02-10 15:03:50 -0800115 // 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));
Spencer Low22191c32015-08-01 17:29:23 -0700124}
125
Alex Vallée47d67c92015-05-06 16:26:00 -0400126TEST(adb_utils, mkdirs) {
127 TemporaryDir td;
Spencer Low22191c32015-08-01 17:29:23 -0700128
129 // Absolute paths.
Spencer Low85c45bd2016-02-10 15:03:50 -0800130 test_mkdirs(std::string(td.path) + "/dir/subdir");
Spencer Low22191c32015-08-01 17:29:23 -0700131
132 // Relative paths.
133 ASSERT_EQ(0, chdir(td.path)) << strerror(errno);
Spencer Low85c45bd2016-02-10 15:03:50 -0800134 test_mkdirs(std::string("relative/subrel"));
Alex Vallée47d67c92015-05-06 16:26:00 -0400135}
Yabin Cui6dfef252015-10-06 15:10:05 -0700136
137#if !defined(_WIN32)
138TEST(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
David Purselleaae97e2016-04-07 11:25:48 -0700152
153TEST(adb_utils, test_forward_targets_are_valid) {
154 std::string error;
155
156 // Source port can be >= 0.
157 EXPECT_FALSE(forward_targets_are_valid("tcp:-1", "tcp:9000", &error));
158 EXPECT_TRUE(forward_targets_are_valid("tcp:0", "tcp:9000", &error));
159 EXPECT_TRUE(forward_targets_are_valid("tcp:8000", "tcp:9000", &error));
160
161 // Destination port must be >0.
162 EXPECT_FALSE(forward_targets_are_valid("tcp:8000", "tcp:-1", &error));
163 EXPECT_FALSE(forward_targets_are_valid("tcp:8000", "tcp:0", &error));
164
165 // Port must be a number.
166 EXPECT_FALSE(forward_targets_are_valid("tcp:", "tcp:9000", &error));
167 EXPECT_FALSE(forward_targets_are_valid("tcp:a", "tcp:9000", &error));
168 EXPECT_FALSE(forward_targets_are_valid("tcp:22x", "tcp:9000", &error));
169 EXPECT_FALSE(forward_targets_are_valid("tcp:8000", "tcp:", &error));
170 EXPECT_FALSE(forward_targets_are_valid("tcp:8000", "tcp:a", &error));
171 EXPECT_FALSE(forward_targets_are_valid("tcp:8000", "tcp:22x", &error));
172}