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 "transport.h"
18
19 #include <gtest/gtest.h>
20
21 #include "adb.h"
22
23 class TransportSetup {
24 public:
TransportSetup()25 TransportSetup() {
26 #ifdef _WIN32
27 // Use extern instead of including sysdeps.h which brings in various macros
28 // that conflict with APIs used in this file.
29 extern void adb_sysdeps_init(void);
30 adb_sysdeps_init();
31 #else
32 // adb_sysdeps_init() is an inline function that we cannot link against.
33 #endif
34 }
35 };
36
37 // Static initializer will call adb_sysdeps_init() before main() to initialize
38 // the transport mutex before it is used in the tests. Alternatives would be to
39 // use __attribute__((constructor)) here or to use that or a static initializer
40 // for adb_sysdeps_init() itself in sysdeps_win32.cpp (caveats of unclear
41 // init order), or to use a test fixture whose SetUp() could do the init once.
42 static TransportSetup g_TransportSetup;
43
TEST(transport,kick_transport)44 TEST(transport, kick_transport) {
45 atransport t;
46 static size_t kick_count;
47 kick_count = 0;
48 // Mutate some member so we can test that the function is run.
49 t.SetKickFunction([](atransport* trans) { kick_count++; });
50 ASSERT_FALSE(t.IsKicked());
51 t.Kick();
52 ASSERT_TRUE(t.IsKicked());
53 ASSERT_EQ(1u, kick_count);
54 // A transport can only be kicked once.
55 t.Kick();
56 ASSERT_TRUE(t.IsKicked());
57 ASSERT_EQ(1u, kick_count);
58 }
59
DisconnectFunc(void * arg,atransport *)60 static void DisconnectFunc(void* arg, atransport*) {
61 int* count = reinterpret_cast<int*>(arg);
62 ++*count;
63 }
64
TEST(transport,RunDisconnects)65 TEST(transport, RunDisconnects) {
66 atransport t;
67 // RunDisconnects() can be called with an empty atransport.
68 t.RunDisconnects();
69
70 int count = 0;
71 adisconnect disconnect;
72 disconnect.func = DisconnectFunc;
73 disconnect.opaque = &count;
74 t.AddDisconnect(&disconnect);
75 t.RunDisconnects();
76 ASSERT_EQ(1, count);
77
78 // disconnect should have been removed automatically.
79 t.RunDisconnects();
80 ASSERT_EQ(1, count);
81
82 count = 0;
83 t.AddDisconnect(&disconnect);
84 t.RemoveDisconnect(&disconnect);
85 t.RunDisconnects();
86 ASSERT_EQ(0, count);
87 }
88
TEST(transport,SetFeatures)89 TEST(transport, SetFeatures) {
90 atransport t;
91 ASSERT_EQ(0U, t.features().size());
92
93 t.SetFeatures(FeatureSetToString(FeatureSet{"foo"}));
94 ASSERT_EQ(1U, t.features().size());
95 ASSERT_TRUE(t.has_feature("foo"));
96
97 t.SetFeatures(FeatureSetToString(FeatureSet{"foo", "bar"}));
98 ASSERT_EQ(2U, t.features().size());
99 ASSERT_TRUE(t.has_feature("foo"));
100 ASSERT_TRUE(t.has_feature("bar"));
101
102 t.SetFeatures(FeatureSetToString(FeatureSet{"foo", "bar", "foo"}));
103 ASSERT_EQ(2U, t.features().size());
104 ASSERT_TRUE(t.has_feature("foo"));
105 ASSERT_TRUE(t.has_feature("bar"));
106
107 t.SetFeatures(FeatureSetToString(FeatureSet{"bar", "baz"}));
108 ASSERT_EQ(2U, t.features().size());
109 ASSERT_FALSE(t.has_feature("foo"));
110 ASSERT_TRUE(t.has_feature("bar"));
111 ASSERT_TRUE(t.has_feature("baz"));
112
113 t.SetFeatures("");
114 ASSERT_EQ(0U, t.features().size());
115 }
116
TEST(transport,parse_banner_no_features)117 TEST(transport, parse_banner_no_features) {
118 atransport t;
119
120 parse_banner("host::", &t);
121
122 ASSERT_EQ(0U, t.features().size());
123 ASSERT_EQ(kCsHost, t.connection_state);
124
125 ASSERT_EQ(nullptr, t.product);
126 ASSERT_EQ(nullptr, t.model);
127 ASSERT_EQ(nullptr, t.device);
128 }
129
TEST(transport,parse_banner_product_features)130 TEST(transport, parse_banner_product_features) {
131 atransport t;
132
133 const char banner[] =
134 "host::ro.product.name=foo;ro.product.model=bar;ro.product.device=baz;";
135 parse_banner(banner, &t);
136
137 ASSERT_EQ(kCsHost, t.connection_state);
138
139 ASSERT_EQ(0U, t.features().size());
140
141 ASSERT_EQ(std::string("foo"), t.product);
142 ASSERT_EQ(std::string("bar"), t.model);
143 ASSERT_EQ(std::string("baz"), t.device);
144 }
145
TEST(transport,parse_banner_features)146 TEST(transport, parse_banner_features) {
147 atransport t;
148
149 const char banner[] =
150 "host::ro.product.name=foo;ro.product.model=bar;ro.product.device=baz;"
151 "features=woodly,doodly";
152 parse_banner(banner, &t);
153
154 ASSERT_EQ(kCsHost, t.connection_state);
155
156 ASSERT_EQ(2U, t.features().size());
157 ASSERT_TRUE(t.has_feature("woodly"));
158 ASSERT_TRUE(t.has_feature("doodly"));
159
160 ASSERT_EQ(std::string("foo"), t.product);
161 ASSERT_EQ(std::string("bar"), t.model);
162 ASSERT_EQ(std::string("baz"), t.device);
163 }
164
TEST(transport,test_matches_target)165 TEST(transport, test_matches_target) {
166 std::string serial = "foo";
167 std::string devpath = "/path/to/bar";
168 std::string product = "test_product";
169 std::string model = "test_model";
170 std::string device = "test_device";
171
172 atransport t;
173 t.serial = &serial[0];
174 t.devpath = &devpath[0];
175 t.product = &product[0];
176 t.model = &model[0];
177 t.device = &device[0];
178
179 // These tests should not be affected by the transport type.
180 for (TransportType type : {kTransportAny, kTransportLocal}) {
181 t.type = type;
182
183 EXPECT_TRUE(t.MatchesTarget(serial));
184 EXPECT_TRUE(t.MatchesTarget(devpath));
185 EXPECT_TRUE(t.MatchesTarget("product:" + product));
186 EXPECT_TRUE(t.MatchesTarget("model:" + model));
187 EXPECT_TRUE(t.MatchesTarget("device:" + device));
188
189 // Product, model, and device don't match without the prefix.
190 EXPECT_FALSE(t.MatchesTarget(product));
191 EXPECT_FALSE(t.MatchesTarget(model));
192 EXPECT_FALSE(t.MatchesTarget(device));
193 }
194 }
195
TEST(transport,test_matches_target_local)196 TEST(transport, test_matches_target_local) {
197 std::string serial = "100.100.100.100:5555";
198
199 atransport t;
200 t.serial = &serial[0];
201
202 // Network address matching should only be used for local transports.
203 for (TransportType type : {kTransportAny, kTransportLocal}) {
204 t.type = type;
205 bool should_match = (type == kTransportLocal);
206
207 EXPECT_EQ(should_match, t.MatchesTarget("100.100.100.100"));
208 EXPECT_EQ(should_match, t.MatchesTarget("tcp:100.100.100.100"));
209 EXPECT_EQ(should_match, t.MatchesTarget("tcp:100.100.100.100:5555"));
210 EXPECT_EQ(should_match, t.MatchesTarget("udp:100.100.100.100"));
211 EXPECT_EQ(should_match, t.MatchesTarget("udp:100.100.100.100:5555"));
212
213 // Wrong protocol, hostname, or port should never match.
214 EXPECT_FALSE(t.MatchesTarget("100.100.100"));
215 EXPECT_FALSE(t.MatchesTarget("100.100.100.100:"));
216 EXPECT_FALSE(t.MatchesTarget("100.100.100.100:-1"));
217 EXPECT_FALSE(t.MatchesTarget("100.100.100.100:5554"));
218 EXPECT_FALSE(t.MatchesTarget("abc:100.100.100.100"));
219 }
220 }
221