blob: 2654a70e3c321203167be8504e5e3bcb7fe5b095 [file] [log] [blame]
mukesh agrawal6bb9e7c2012-01-30 14:57:54 -08001// Copyright (c) 2012 The Chromium OS Authors. All rights reserved.
Darin Petkove7cb7f82011-06-03 13:21:51 -07002// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
Chris Masone8a7b8be2011-07-22 12:43:37 -07005#include "shill/dhcp_config.h"
6
Eric Shienbrood3e20a232012-02-16 11:35:56 -05007#include <base/bind.h>
Darin Petkov92c43902011-06-09 20:46:06 -07008#include <base/file_util.h>
Eric Shienbrood3e20a232012-02-16 11:35:56 -05009#include <base/scoped_temp_dir.h>
Darin Petkov92c43902011-06-09 20:46:06 -070010#include <base/stringprintf.h>
Chris Masone43b48a12011-07-01 13:37:07 -070011#include <chromeos/dbus/service_constants.h>
Chris Masone8a7b8be2011-07-22 12:43:37 -070012#include <gtest/gtest.h>
Darin Petkov92c43902011-06-09 20:46:06 -070013
Chris Masone43b48a12011-07-01 13:37:07 -070014#include "shill/dbus_adaptor.h"
Darin Petkovf7897bc2011-06-08 17:13:36 -070015#include "shill/dhcp_provider.h"
Paul Stewart26b327e2011-10-19 11:38:09 -070016#include "shill/event_dispatcher.h"
Chris Masone19e30402011-07-19 15:48:47 -070017#include "shill/mock_control.h"
Darin Petkov98dd6a02011-06-10 15:12:57 -070018#include "shill/mock_dhcp_proxy.h"
Darin Petkovf7897bc2011-06-08 17:13:36 -070019#include "shill/mock_glib.h"
mukesh agrawal6c6655d2012-12-06 14:49:50 -080020#include "shill/mock_log.h"
Jorge Lucangeli Obesad43cc62012-04-11 16:25:43 -070021#include "shill/mock_minijail.h"
Chris Masone43b48a12011-07-01 13:37:07 -070022#include "shill/property_store_unittest.h"
Darin Petkova7b89492011-07-27 12:48:17 -070023#include "shill/proxy_factory.h"
Darin Petkove7cb7f82011-06-03 13:21:51 -070024
Eric Shienbrood3e20a232012-02-16 11:35:56 -050025using base::Bind;
26using base::Unretained;
Darin Petkove7cb7f82011-06-03 13:21:51 -070027using std::string;
28using std::vector;
Darin Petkovf7897bc2011-06-08 17:13:36 -070029using testing::_;
30using testing::Return;
31using testing::SetArgumentPointee;
Darin Petkove7cb7f82011-06-03 13:21:51 -070032using testing::Test;
33
34namespace shill {
35
Darin Petkov92c43902011-06-09 20:46:06 -070036namespace {
Darin Petkova7b89492011-07-27 12:48:17 -070037const char kDeviceName[] = "eth0";
Paul Stewartd32f4842012-01-11 16:08:13 -080038const char kHostName[] = "hostname";
Paul Stewartd408fdf2012-05-07 17:15:57 -070039const char kLeaseFileSuffix[] = "leasefilesuffix";
40const bool kArpGateway = true;
Darin Petkov92c43902011-06-09 20:46:06 -070041} // namespace {}
42
Chris Masone43b48a12011-07-01 13:37:07 -070043class DHCPConfigTest : public PropertyStoreTest {
Darin Petkove7cb7f82011-06-03 13:21:51 -070044 public:
Darin Petkovf7897bc2011-06-08 17:13:36 -070045 DHCPConfigTest()
Darin Petkovf65e9282011-06-21 14:29:56 -070046 : proxy_(new MockDHCPProxy()),
Darin Petkova7b89492011-07-27 12:48:17 -070047 proxy_factory_(this),
Jorge Lucangeli Obesad43cc62012-04-11 16:25:43 -070048 minijail_(new MockMinijail()),
Chris Masone19e30402011-07-19 15:48:47 -070049 config_(new DHCPConfig(&control_,
Chris Masone2176a882011-09-14 22:29:15 -070050 dispatcher(),
Chris Masone19e30402011-07-19 15:48:47 -070051 DHCPProvider::GetInstance(),
Darin Petkov92c43902011-06-09 20:46:06 -070052 kDeviceName,
Paul Stewartd32f4842012-01-11 16:08:13 -080053 kHostName,
Paul Stewartd408fdf2012-05-07 17:15:57 -070054 kLeaseFileSuffix,
55 kArpGateway,
Chris Masone2176a882011-09-14 22:29:15 -070056 glib())) {}
Darin Petkova7b89492011-07-27 12:48:17 -070057
58 virtual void SetUp() {
Darin Petkovab565bb2011-10-06 02:55:51 -070059 config_->proxy_factory_ = &proxy_factory_;
Jorge Lucangeli Obesad43cc62012-04-11 16:25:43 -070060 config_->minijail_ = minijail_.get();
Darin Petkova7b89492011-07-27 12:48:17 -070061 }
62
63 virtual void TearDown() {
Darin Petkovab565bb2011-10-06 02:55:51 -070064 config_->proxy_factory_ = NULL;
Jorge Lucangeli Obesad43cc62012-04-11 16:25:43 -070065 config_->minijail_ = NULL;
Darin Petkov98dd6a02011-06-10 15:12:57 -070066 }
Darin Petkove7cb7f82011-06-03 13:21:51 -070067
Jorge Lucangeli Obesad43cc62012-04-11 16:25:43 -070068 DHCPConfigRefPtr CreateMockMinijailConfig(const string &hostname,
69 const string &lease_suffix,
70 bool arp_gateway);
Paul Stewartd408fdf2012-05-07 17:15:57 -070071 DHCPConfigRefPtr CreateRunningConfig(const string &hostname,
72 const string &lease_suffix,
73 bool arp_gateway);
74 void StopRunningConfigAndExpect(DHCPConfigRefPtr config,
75 bool lease_file_exists);
76
Darin Petkove7cb7f82011-06-03 13:21:51 -070077 protected:
Darin Petkova7b89492011-07-27 12:48:17 -070078 class TestProxyFactory : public ProxyFactory {
79 public:
Paul Stewart7355ce12011-09-02 10:47:01 -070080 explicit TestProxyFactory(DHCPConfigTest *test) : test_(test) {}
Darin Petkova7b89492011-07-27 12:48:17 -070081
mukesh agrawal1830fa12011-09-26 14:31:40 -070082 virtual DHCPProxyInterface *CreateDHCPProxy(const string &/*service*/) {
Darin Petkova7b89492011-07-27 12:48:17 -070083 return test_->proxy_.release();
84 }
85
86 private:
87 DHCPConfigTest *test_;
88 };
89
Paul Stewartd408fdf2012-05-07 17:15:57 -070090 static const int kPID;
91 static const unsigned int kTag;
92
93 FilePath lease_file_;
94 FilePath pid_file_;
95 ScopedTempDir temp_dir_;
Darin Petkova7b89492011-07-27 12:48:17 -070096 scoped_ptr<MockDHCPProxy> proxy_;
97 TestProxyFactory proxy_factory_;
Chris Masone19e30402011-07-19 15:48:47 -070098 MockControl control_;
Jorge Lucangeli Obesad43cc62012-04-11 16:25:43 -070099 scoped_ptr<MockMinijail> minijail_;
Darin Petkovf7897bc2011-06-08 17:13:36 -0700100 DHCPConfigRefPtr config_;
Darin Petkove7cb7f82011-06-03 13:21:51 -0700101};
102
Paul Stewartd408fdf2012-05-07 17:15:57 -0700103const int DHCPConfigTest::kPID = 123456;
104const unsigned int DHCPConfigTest::kTag = 77;
105
Jorge Lucangeli Obesad43cc62012-04-11 16:25:43 -0700106DHCPConfigRefPtr DHCPConfigTest::CreateMockMinijailConfig(
107 const string &hostname,
108 const string &lease_suffix,
109 bool arp_gateway) {
110 DHCPConfigRefPtr config(new DHCPConfig(&control_,
111 dispatcher(),
112 DHCPProvider::GetInstance(),
113 kDeviceName,
114 hostname,
115 lease_suffix,
116 arp_gateway,
117 glib()));
118 config->minijail_ = minijail_.get();
119 EXPECT_CALL(*minijail_, RunAndDestroy(_, _, _)).WillOnce(Return(false));
120
121 return config;
122}
123
Paul Stewartd408fdf2012-05-07 17:15:57 -0700124DHCPConfigRefPtr DHCPConfigTest::CreateRunningConfig(const string &hostname,
125 const string &lease_suffix,
126 bool arp_gateway) {
127 DHCPConfigRefPtr config(new DHCPConfig(&control_,
128 dispatcher(),
129 DHCPProvider::GetInstance(),
130 kDeviceName,
131 hostname,
132 lease_suffix,
133 arp_gateway,
134 glib()));
Jorge Lucangeli Obesad43cc62012-04-11 16:25:43 -0700135 config->minijail_ = minijail_.get();
136 EXPECT_CALL(*minijail_, RunAndDestroy(_, _, _))
137 .WillOnce(DoAll(SetArgumentPointee<2>(kPID), Return(true)));
Paul Stewartd408fdf2012-05-07 17:15:57 -0700138 EXPECT_CALL(*glib(), ChildWatchAdd(kPID, _, _)).WillOnce(Return(kTag));
139 EXPECT_TRUE(config->Start());
140 EXPECT_EQ(kPID, config->pid_);
141 EXPECT_EQ(config.get(), DHCPProvider::GetInstance()->GetConfig(kPID).get());
142 EXPECT_EQ(kTag, config->child_watch_tag_);
143
144 EXPECT_TRUE(temp_dir_.CreateUniqueTempDir());
145 config->root_ = temp_dir_.path();
Jorge Lucangeli Obes2f3169d2012-04-25 11:38:25 -0700146 FilePath varrun = temp_dir_.path().Append("var/run/dhcpcd");
Paul Stewartd408fdf2012-05-07 17:15:57 -0700147 EXPECT_TRUE(file_util::CreateDirectory(varrun));
148 pid_file_ = varrun.Append(base::StringPrintf("dhcpcd-%s.pid", kDeviceName));
149 FilePath varlib = temp_dir_.path().Append("var/lib/dhcpcd");
150 EXPECT_TRUE(file_util::CreateDirectory(varlib));
151 lease_file_ =
152 varlib.Append(base::StringPrintf("dhcpcd-%s.lease", kDeviceName));
153 EXPECT_EQ(0, file_util::WriteFile(pid_file_, "", 0));
154 EXPECT_EQ(0, file_util::WriteFile(lease_file_, "", 0));
155 EXPECT_TRUE(file_util::PathExists(pid_file_));
156 EXPECT_TRUE(file_util::PathExists(lease_file_));
157
158 return config;
159}
160
161void DHCPConfigTest::StopRunningConfigAndExpect(DHCPConfigRefPtr config,
162 bool lease_file_exists) {
mukesh agrawal6c6655d2012-12-06 14:49:50 -0800163 ScopedMockLog log;
164 // We use a non-zero exit status so that we get the log message.
165 EXPECT_CALL(log, Log(_, _, ::testing::EndsWith("status 10")));
166 DHCPConfig::ChildWatchCallback(kPID, 10, config.get());
Paul Stewartd408fdf2012-05-07 17:15:57 -0700167 EXPECT_EQ(NULL, DHCPProvider::GetInstance()->GetConfig(kPID).get());
168
169 EXPECT_FALSE(file_util::PathExists(pid_file_));
170 EXPECT_EQ(lease_file_exists, file_util::PathExists(lease_file_));
171}
172
Darin Petkove7cb7f82011-06-03 13:21:51 -0700173TEST_F(DHCPConfigTest, GetIPv4AddressString) {
Darin Petkovf7897bc2011-06-08 17:13:36 -0700174 EXPECT_EQ("255.255.255.255", config_->GetIPv4AddressString(0xffffffff));
175 EXPECT_EQ("0.0.0.0", config_->GetIPv4AddressString(0));
176 EXPECT_EQ("1.2.3.4", config_->GetIPv4AddressString(0x04030201));
Darin Petkove7cb7f82011-06-03 13:21:51 -0700177}
178
Darin Petkova7b89492011-07-27 12:48:17 -0700179TEST_F(DHCPConfigTest, InitProxy) {
180 static const char kService[] = ":1.200";
Darin Petkova7b89492011-07-27 12:48:17 -0700181 EXPECT_TRUE(proxy_.get());
182 EXPECT_FALSE(config_->proxy_.get());
Eric Shienbrood9a245532012-03-07 14:20:39 -0500183 config_->InitProxy(kService);
Darin Petkova7b89492011-07-27 12:48:17 -0700184 EXPECT_FALSE(proxy_.get());
185 EXPECT_TRUE(config_->proxy_.get());
186
187 config_->InitProxy(kService);
Darin Petkova7b89492011-07-27 12:48:17 -0700188}
189
Paul Stewart65bcd082012-11-16 09:37:14 -0800190TEST_F(DHCPConfigTest, ParseClasslessStaticRoutes) {
191 const string kDefaultAddress = "0.0.0.0";
192 const string kDefaultDestination = kDefaultAddress + "/0";
193 const string kRouter0 = "10.0.0.254";
194 const string kAddress1 = "192.168.1.0";
195 const string kDestination1 = kAddress1 + "/24";
196 // Last gateway missing, leaving an odd number of parameters.
197 const string kBrokenClasslessRoutes0 = kDefaultDestination + " " + kRouter0 +
198 " " + kDestination1;
199 IPConfig::Properties properties;
200 EXPECT_FALSE(DHCPConfig::ParseClasslessStaticRoutes(kBrokenClasslessRoutes0,
201 &properties));
202 EXPECT_TRUE(properties.routes.empty());
203 EXPECT_TRUE(properties.gateway.empty());
204
205 // Gateway argument for the second route is malformed, but we were able
206 // to salvage a default gateway.
207 const string kBrokenRouter1 = "10.0.0";
208 const string kBrokenClasslessRoutes1 = kBrokenClasslessRoutes0 + " " +
209 kBrokenRouter1;
210 EXPECT_FALSE(DHCPConfig::ParseClasslessStaticRoutes(kBrokenClasslessRoutes1,
211 &properties));
212 EXPECT_TRUE(properties.routes.empty());
213 EXPECT_EQ(kRouter0, properties.gateway);
214
215 const string kRouter1 = "10.0.0.253";
216 const string kRouter2 = "10.0.0.252";
217 const string kClasslessRoutes0 = kDefaultDestination + " " + kRouter2 + " " +
218 kDestination1 + " " + kRouter1;
219 EXPECT_TRUE(DHCPConfig::ParseClasslessStaticRoutes(kClasslessRoutes0,
220 &properties));
221 // The old default route is preserved.
222 EXPECT_EQ(kRouter0, properties.gateway);
223
224 // The two routes (including the one which would have otherwise been
225 // classified as a default route) are added to the routing table.
226 EXPECT_EQ(2, properties.routes.size());
227 const IPConfig::Route &route0 = properties.routes[0];
228 EXPECT_EQ(kDefaultAddress, route0.host);
229 EXPECT_EQ("0.0.0.0", route0.netmask);
230 EXPECT_EQ(kRouter2, route0.gateway);
231
232 const IPConfig::Route &route1 = properties.routes[1];
233 EXPECT_EQ(kAddress1, route1.host);
234 EXPECT_EQ("255.255.255.0", route1.netmask);
235 EXPECT_EQ(kRouter1, route1.gateway);
236
237 // A malformed routing table should not affect the current table.
238 EXPECT_FALSE(DHCPConfig::ParseClasslessStaticRoutes(kBrokenClasslessRoutes1,
239 &properties));
240 EXPECT_EQ(2, properties.routes.size());
241 EXPECT_EQ(kRouter0, properties.gateway);
242}
243
Darin Petkove7cb7f82011-06-03 13:21:51 -0700244TEST_F(DHCPConfigTest, ParseConfiguration) {
245 DHCPConfig::Configuration conf;
246 conf[DHCPConfig::kConfigurationKeyIPAddress].writer().append_uint32(
247 0x01020304);
248 conf[DHCPConfig::kConfigurationKeySubnetCIDR].writer().append_byte(
249 16);
250 conf[DHCPConfig::kConfigurationKeyBroadcastAddress].writer().append_uint32(
251 0x10203040);
252 {
Darin Petkove7cb7f82011-06-03 13:21:51 -0700253 vector<unsigned int> routers;
254 routers.push_back(0x02040608);
255 routers.push_back(0x03050709);
Darin Petkovf7897bc2011-06-08 17:13:36 -0700256 DBus::MessageIter writer =
257 conf[DHCPConfig::kConfigurationKeyRouters].writer();
Darin Petkove7cb7f82011-06-03 13:21:51 -0700258 writer << routers;
Darin Petkove7cb7f82011-06-03 13:21:51 -0700259 }
260 {
Darin Petkove7cb7f82011-06-03 13:21:51 -0700261 vector<unsigned int> dns;
262 dns.push_back(0x09070503);
263 dns.push_back(0x08060402);
Darin Petkovf7897bc2011-06-08 17:13:36 -0700264 DBus::MessageIter writer = conf[DHCPConfig::kConfigurationKeyDNS].writer();
Darin Petkove7cb7f82011-06-03 13:21:51 -0700265 writer << dns;
Darin Petkove7cb7f82011-06-03 13:21:51 -0700266 }
267 conf[DHCPConfig::kConfigurationKeyDomainName].writer().append_string(
268 "domain-name");
269 {
Darin Petkove7cb7f82011-06-03 13:21:51 -0700270 vector<string> search;
271 search.push_back("foo.com");
272 search.push_back("bar.com");
Darin Petkovf7897bc2011-06-08 17:13:36 -0700273 DBus::MessageIter writer =
274 conf[DHCPConfig::kConfigurationKeyDomainSearch].writer();
Darin Petkove7cb7f82011-06-03 13:21:51 -0700275 writer << search;
Darin Petkove7cb7f82011-06-03 13:21:51 -0700276 }
277 conf[DHCPConfig::kConfigurationKeyMTU].writer().append_uint16(600);
278 conf["UnknownKey"] = DBus::Variant();
279
Darin Petkove7cb7f82011-06-03 13:21:51 -0700280 IPConfig::Properties properties;
Darin Petkovf7897bc2011-06-08 17:13:36 -0700281 ASSERT_TRUE(config_->ParseConfiguration(conf, &properties));
Darin Petkove7cb7f82011-06-03 13:21:51 -0700282 EXPECT_EQ("4.3.2.1", properties.address);
Paul Stewart48100b02012-03-19 07:53:52 -0700283 EXPECT_EQ(16, properties.subnet_prefix);
Darin Petkove7cb7f82011-06-03 13:21:51 -0700284 EXPECT_EQ("64.48.32.16", properties.broadcast_address);
285 EXPECT_EQ("8.6.4.2", properties.gateway);
286 ASSERT_EQ(2, properties.dns_servers.size());
287 EXPECT_EQ("3.5.7.9", properties.dns_servers[0]);
288 EXPECT_EQ("2.4.6.8", properties.dns_servers[1]);
289 EXPECT_EQ("domain-name", properties.domain_name);
290 ASSERT_EQ(2, properties.domain_search.size());
291 EXPECT_EQ("foo.com", properties.domain_search[0]);
292 EXPECT_EQ("bar.com", properties.domain_search[1]);
293 EXPECT_EQ(600, properties.mtu);
294}
295
Darin Petkov92c43902011-06-09 20:46:06 -0700296TEST_F(DHCPConfigTest, StartFail) {
Jorge Lucangeli Obesad43cc62012-04-11 16:25:43 -0700297 EXPECT_CALL(*minijail_, RunAndDestroy(_, _, _)).WillOnce(Return(false));
Chris Masone2176a882011-09-14 22:29:15 -0700298 EXPECT_CALL(*glib(), ChildWatchAdd(_, _, _)).Times(0);
Darin Petkovf7897bc2011-06-08 17:13:36 -0700299 EXPECT_FALSE(config_->Start());
300 EXPECT_EQ(0, config_->pid_);
Darin Petkov92c43902011-06-09 20:46:06 -0700301}
Darin Petkovf7897bc2011-06-08 17:13:36 -0700302
Paul Stewartd408fdf2012-05-07 17:15:57 -0700303MATCHER_P3(IsDHCPCDArgs, has_hostname, has_arp_gateway, has_lease_suffix, "") {
Paul Stewartd32f4842012-01-11 16:08:13 -0800304 if (string(arg[0]) != "/sbin/dhcpcd" ||
Paul Stewartd408fdf2012-05-07 17:15:57 -0700305 string(arg[1]) != "-B") {
Paul Stewartd32f4842012-01-11 16:08:13 -0800306 return false;
307 }
308
Paul Stewartd408fdf2012-05-07 17:15:57 -0700309 int end_offset = 2;
Paul Stewartd32f4842012-01-11 16:08:13 -0800310 if (has_hostname) {
Paul Stewartd408fdf2012-05-07 17:15:57 -0700311 if (string(arg[end_offset]) != "-h" ||
312 string(arg[end_offset + 1]) != kHostName) {
Paul Stewartd32f4842012-01-11 16:08:13 -0800313 return false;
314 }
Paul Stewartd408fdf2012-05-07 17:15:57 -0700315 end_offset += 2;
Paul Stewartd32f4842012-01-11 16:08:13 -0800316 }
317
Paul Stewartd408fdf2012-05-07 17:15:57 -0700318 if (has_arp_gateway) {
319 if (string(arg[end_offset]) != "-R")
320 return false;
321 ++end_offset;
322 }
323
324 string device_arg = has_lease_suffix ?
325 string(kDeviceName) + "=" + string(kLeaseFileSuffix) : kDeviceName;
326 return string(arg[end_offset]) == device_arg && arg[end_offset + 1] == NULL;
Paul Stewartd32f4842012-01-11 16:08:13 -0800327}
328
329TEST_F(DHCPConfigTest, StartWithHostname) {
Jorge Lucangeli Obesad43cc62012-04-11 16:25:43 -0700330 EXPECT_CALL(*minijail_, RunAndDestroy(_, _, _)).WillOnce(Return(false));
Paul Stewartd32f4842012-01-11 16:08:13 -0800331 EXPECT_FALSE(config_->Start());
332}
333
334TEST_F(DHCPConfigTest, StartWithoutHostname) {
Jorge Lucangeli Obesad43cc62012-04-11 16:25:43 -0700335 DHCPConfigRefPtr config = CreateMockMinijailConfig("",
336 kLeaseFileSuffix,
337 kArpGateway);
Paul Stewartd408fdf2012-05-07 17:15:57 -0700338 EXPECT_FALSE(config->Start());
339}
340
341TEST_F(DHCPConfigTest, StartWithoutArpGateway) {
Jorge Lucangeli Obesad43cc62012-04-11 16:25:43 -0700342 DHCPConfigRefPtr config = CreateMockMinijailConfig(kHostName,
343 kLeaseFileSuffix,
344 false);
Paul Stewartd408fdf2012-05-07 17:15:57 -0700345 EXPECT_FALSE(config->Start());
346}
347
348TEST_F(DHCPConfigTest, StartWithoutLeaseSuffix) {
Jorge Lucangeli Obesad43cc62012-04-11 16:25:43 -0700349 DHCPConfigRefPtr config = CreateMockMinijailConfig(kHostName,
350 kDeviceName,
351 kArpGateway);
Paul Stewartd32f4842012-01-11 16:08:13 -0800352 EXPECT_FALSE(config->Start());
353}
354
Darin Petkovf9b0ca82011-06-20 12:10:23 -0700355namespace {
356
357class UpdateCallbackTest {
358 public:
359 UpdateCallbackTest(const string &message,
Chris Masone2b105542011-06-22 10:58:09 -0700360 const IPConfigRefPtr &ipconfig,
Darin Petkovf9b0ca82011-06-20 12:10:23 -0700361 bool success)
362 : message_(message),
363 ipconfig_(ipconfig),
364 success_(success),
365 called_(false) {}
366
Chris Masone2b105542011-06-22 10:58:09 -0700367 void Callback(const IPConfigRefPtr &ipconfig, bool success) {
Darin Petkovf9b0ca82011-06-20 12:10:23 -0700368 called_ = true;
369 EXPECT_EQ(ipconfig_.get(), ipconfig.get()) << message_;
370 EXPECT_EQ(success_, success) << message_;
371 }
372
373 bool called() const { return called_; }
374
375 private:
376 const string message_;
377 IPConfigRefPtr ipconfig_;
378 bool success_;
379 bool called_;
380};
381
mukesh agrawalcc0fded2012-05-09 13:40:58 -0700382void DoNothing() {}
383
Darin Petkovf9b0ca82011-06-20 12:10:23 -0700384} // namespace {}
385
386TEST_F(DHCPConfigTest, ProcessEventSignalFail) {
387 DHCPConfig::Configuration conf;
388 conf[DHCPConfig::kConfigurationKeyIPAddress].writer().append_uint32(
389 0x01020304);
390 UpdateCallbackTest callback_test(DHCPConfig::kReasonFail, config_, false);
391 config_->RegisterUpdateCallback(
Eric Shienbrood3e20a232012-02-16 11:35:56 -0500392 Bind(&UpdateCallbackTest::Callback, Unretained(&callback_test)));
mukesh agrawalcc0fded2012-05-09 13:40:58 -0700393 config_->lease_acquisition_timeout_callback_.Reset(base::Bind(&DoNothing));
Darin Petkovf9b0ca82011-06-20 12:10:23 -0700394 config_->ProcessEventSignal(DHCPConfig::kReasonFail, conf);
395 EXPECT_TRUE(callback_test.called());
396 EXPECT_TRUE(config_->properties().address.empty());
mukesh agrawalcc0fded2012-05-09 13:40:58 -0700397 EXPECT_TRUE(config_->lease_acquisition_timeout_callback_.IsCancelled());
Darin Petkovf9b0ca82011-06-20 12:10:23 -0700398}
399
400TEST_F(DHCPConfigTest, ProcessEventSignalSuccess) {
401 static const char * const kReasons[] = {
402 DHCPConfig::kReasonBound,
403 DHCPConfig::kReasonRebind,
404 DHCPConfig::kReasonReboot,
405 DHCPConfig::kReasonRenew
406 };
407 for (size_t r = 0; r < arraysize(kReasons); r++) {
408 DHCPConfig::Configuration conf;
409 string message = string(kReasons[r]) + " failed";
410 conf[DHCPConfig::kConfigurationKeyIPAddress].writer().append_uint32(r);
411 UpdateCallbackTest callback_test(message, config_, true);
412 config_->RegisterUpdateCallback(
Eric Shienbrood3e20a232012-02-16 11:35:56 -0500413 Bind(&UpdateCallbackTest::Callback, Unretained(&callback_test)));
mukesh agrawalcc0fded2012-05-09 13:40:58 -0700414 config_->lease_acquisition_timeout_callback_.Reset(base::Bind(&DoNothing));
Darin Petkovf9b0ca82011-06-20 12:10:23 -0700415 config_->ProcessEventSignal(kReasons[r], conf);
416 EXPECT_TRUE(callback_test.called()) << message;
Paul Stewart83224402011-11-30 14:52:30 -0800417 EXPECT_EQ(base::StringPrintf("%zu.0.0.0", r), config_->properties().address)
Darin Petkovf9b0ca82011-06-20 12:10:23 -0700418 << message;
mukesh agrawalcc0fded2012-05-09 13:40:58 -0700419 EXPECT_TRUE(config_->lease_acquisition_timeout_callback_.IsCancelled());
Darin Petkovf9b0ca82011-06-20 12:10:23 -0700420 }
421}
422
423TEST_F(DHCPConfigTest, ProcessEventSignalUnknown) {
424 DHCPConfig::Configuration conf;
425 conf[DHCPConfig::kConfigurationKeyIPAddress].writer().append_uint32(
426 0x01020304);
427 static const char kReasonUnknown[] = "UNKNOWN_REASON";
428 UpdateCallbackTest callback_test(kReasonUnknown, config_, false);
429 config_->RegisterUpdateCallback(
Eric Shienbrood3e20a232012-02-16 11:35:56 -0500430 Bind(&UpdateCallbackTest::Callback, Unretained(&callback_test)));
mukesh agrawalcc0fded2012-05-09 13:40:58 -0700431 config_->lease_acquisition_timeout_callback_.Reset(base::Bind(&DoNothing));
Darin Petkovf9b0ca82011-06-20 12:10:23 -0700432 config_->ProcessEventSignal(kReasonUnknown, conf);
433 EXPECT_FALSE(callback_test.called());
434 EXPECT_TRUE(config_->properties().address.empty());
mukesh agrawalcc0fded2012-05-09 13:40:58 -0700435 EXPECT_FALSE(config_->lease_acquisition_timeout_callback_.IsCancelled());
Darin Petkovf9b0ca82011-06-20 12:10:23 -0700436}
437
438
Darin Petkov98dd6a02011-06-10 15:12:57 -0700439TEST_F(DHCPConfigTest, ReleaseIP) {
440 config_->pid_ = 1 << 18; // Ensure unknown positive PID.
Paul Stewarta02ee492012-05-16 10:04:53 -0700441 config_->arp_gateway_ = false;
Darin Petkovaceede32011-07-18 15:32:38 -0700442 EXPECT_CALL(*proxy_, Release(kDeviceName)).Times(1);
Darin Petkova7b89492011-07-27 12:48:17 -0700443 config_->proxy_.reset(proxy_.release());
Darin Petkov98dd6a02011-06-10 15:12:57 -0700444 EXPECT_TRUE(config_->ReleaseIP());
445 config_->pid_ = 0;
446}
447
Paul Stewarta02ee492012-05-16 10:04:53 -0700448TEST_F(DHCPConfigTest, ReleaseIPArpGW) {
449 config_->pid_ = 1 << 18; // Ensure unknown positive PID.
450 config_->arp_gateway_ = true;
451 EXPECT_CALL(*proxy_, Release(kDeviceName)).Times(0);
452 config_->proxy_.reset(proxy_.release());
453 EXPECT_TRUE(config_->ReleaseIP());
454 config_->pid_ = 0;
455}
456
Darin Petkov98dd6a02011-06-10 15:12:57 -0700457TEST_F(DHCPConfigTest, RenewIP) {
mukesh agrawalcc0fded2012-05-09 13:40:58 -0700458 EXPECT_TRUE(config_->lease_acquisition_timeout_callback_.IsCancelled());
Darin Petkov98dd6a02011-06-10 15:12:57 -0700459 config_->pid_ = 456;
Paul Stewartc02344a2012-08-17 16:57:37 -0700460 EXPECT_FALSE(config_->RenewIP()); // Expect no crash with NULL proxy.
Darin Petkovaceede32011-07-18 15:32:38 -0700461 EXPECT_CALL(*proxy_, Rebind(kDeviceName)).Times(1);
Darin Petkova7b89492011-07-27 12:48:17 -0700462 config_->proxy_.reset(proxy_.release());
Darin Petkov98dd6a02011-06-10 15:12:57 -0700463 EXPECT_TRUE(config_->RenewIP());
mukesh agrawalcc0fded2012-05-09 13:40:58 -0700464 EXPECT_FALSE(config_->lease_acquisition_timeout_callback_.IsCancelled());
Darin Petkov98dd6a02011-06-10 15:12:57 -0700465 config_->pid_ = 0;
466}
467
468TEST_F(DHCPConfigTest, RequestIP) {
mukesh agrawalcc0fded2012-05-09 13:40:58 -0700469 EXPECT_TRUE(config_->lease_acquisition_timeout_callback_.IsCancelled());
Darin Petkov98dd6a02011-06-10 15:12:57 -0700470 config_->pid_ = 567;
Darin Petkovaceede32011-07-18 15:32:38 -0700471 EXPECT_CALL(*proxy_, Rebind(kDeviceName)).Times(1);
Darin Petkova7b89492011-07-27 12:48:17 -0700472 config_->proxy_.reset(proxy_.release());
Darin Petkov98dd6a02011-06-10 15:12:57 -0700473 EXPECT_TRUE(config_->RenewIP());
mukesh agrawalcc0fded2012-05-09 13:40:58 -0700474 EXPECT_FALSE(config_->lease_acquisition_timeout_callback_.IsCancelled());
475 config_->pid_ = 0;
476}
477
478TEST_F(DHCPConfigTest, RequestIPTimeout) {
479 UpdateCallbackTest callback_test(DHCPConfig::kReasonFail, config_, false);
480 config_->RegisterUpdateCallback(
481 Bind(&UpdateCallbackTest::Callback, Unretained(&callback_test)));
482 config_->lease_acquisition_timeout_seconds_ = 0;
483 config_->pid_ = 567;
484 EXPECT_CALL(*proxy_, Rebind(kDeviceName)).Times(1);
485 config_->proxy_.reset(proxy_.release());
486 config_->RenewIP();
487 config_->dispatcher_->DispatchPendingEvents();
488 EXPECT_TRUE(callback_test.called());
Darin Petkov98dd6a02011-06-10 15:12:57 -0700489 config_->pid_ = 0;
490}
491
492TEST_F(DHCPConfigTest, Restart) {
493 const int kPID1 = 1 << 17; // Ensure unknown positive PID.
494 const int kPID2 = 987;
495 const unsigned int kTag1 = 11;
496 const unsigned int kTag2 = 22;
497 config_->pid_ = kPID1;
498 config_->child_watch_tag_ = kTag1;
499 DHCPProvider::GetInstance()->BindPID(kPID1, config_);
Chris Masone2176a882011-09-14 22:29:15 -0700500 EXPECT_CALL(*glib(), SourceRemove(kTag1)).WillOnce(Return(true));
Jorge Lucangeli Obesad43cc62012-04-11 16:25:43 -0700501 EXPECT_CALL(*minijail_, RunAndDestroy(_, _, _)).WillOnce(
502 DoAll(SetArgumentPointee<2>(kPID2), Return(true)));
Chris Masone2176a882011-09-14 22:29:15 -0700503 EXPECT_CALL(*glib(), ChildWatchAdd(kPID2, _, _)).WillOnce(Return(kTag2));
Darin Petkov98dd6a02011-06-10 15:12:57 -0700504 EXPECT_TRUE(config_->Restart());
505 EXPECT_EQ(kPID2, config_->pid_);
506 EXPECT_EQ(config_.get(), DHCPProvider::GetInstance()->GetConfig(kPID2).get());
507 EXPECT_EQ(kTag2, config_->child_watch_tag_);
508 DHCPProvider::GetInstance()->UnbindPID(kPID2);
509 config_->pid_ = 0;
510 config_->child_watch_tag_ = 0;
511}
512
513TEST_F(DHCPConfigTest, RestartNoClient) {
514 const int kPID = 777;
515 const unsigned int kTag = 66;
Chris Masone2176a882011-09-14 22:29:15 -0700516 EXPECT_CALL(*glib(), SourceRemove(_)).Times(0);
Jorge Lucangeli Obesad43cc62012-04-11 16:25:43 -0700517 EXPECT_CALL(*minijail_, RunAndDestroy(_, _, _)).WillOnce(
518 DoAll(SetArgumentPointee<2>(kPID), Return(true)));
Chris Masone2176a882011-09-14 22:29:15 -0700519 EXPECT_CALL(*glib(), ChildWatchAdd(kPID, _, _)).WillOnce(Return(kTag));
Darin Petkov98dd6a02011-06-10 15:12:57 -0700520 EXPECT_TRUE(config_->Restart());
521 EXPECT_EQ(kPID, config_->pid_);
522 EXPECT_EQ(config_.get(), DHCPProvider::GetInstance()->GetConfig(kPID).get());
523 EXPECT_EQ(kTag, config_->child_watch_tag_);
524 DHCPProvider::GetInstance()->UnbindPID(kPID);
525 config_->pid_ = 0;
526 config_->child_watch_tag_ = 0;
527}
528
Paul Stewartd408fdf2012-05-07 17:15:57 -0700529TEST_F(DHCPConfigTest, StartSuccessEphemeral) {
530 DHCPConfigRefPtr config =
531 CreateRunningConfig(kHostName, kDeviceName, kArpGateway);
532 StopRunningConfigAndExpect(config, false);
533}
Darin Petkov92c43902011-06-09 20:46:06 -0700534
Paul Stewartd408fdf2012-05-07 17:15:57 -0700535TEST_F(DHCPConfigTest, StartSuccessPersistent) {
536 DHCPConfigRefPtr config =
537 CreateRunningConfig(kHostName, kLeaseFileSuffix, kArpGateway);
538 StopRunningConfigAndExpect(config, true);
Darin Petkovf7897bc2011-06-08 17:13:36 -0700539}
540
mukesh agrawalcc0fded2012-05-09 13:40:58 -0700541TEST_F(DHCPConfigTest, StartTimeout) {
542 UpdateCallbackTest callback_test(DHCPConfig::kReasonFail, config_, false);
543 config_->RegisterUpdateCallback(
544 Bind(&UpdateCallbackTest::Callback, Unretained(&callback_test)));
545 config_->lease_acquisition_timeout_seconds_ = 0;
546 config_->proxy_.reset(proxy_.release());
Jorge Lucangeli Obesad43cc62012-04-11 16:25:43 -0700547 EXPECT_CALL(*minijail_, RunAndDestroy(_, _, _)).WillOnce(Return(true));
mukesh agrawalcc0fded2012-05-09 13:40:58 -0700548 config_->Start();
549 config_->dispatcher_->DispatchPendingEvents();
550 EXPECT_TRUE(callback_test.called());
551}
552
Darin Petkov98dd6a02011-06-10 15:12:57 -0700553TEST_F(DHCPConfigTest, Stop) {
554 // Ensure no crashes.
555 const int kPID = 1 << 17; // Ensure unknown positive PID.
Darin Petkov98dd6a02011-06-10 15:12:57 -0700556 config_->pid_ = kPID;
557 config_->Stop();
mukesh agrawalcc0fded2012-05-09 13:40:58 -0700558 EXPECT_TRUE(config_->lease_acquisition_timeout_callback_.IsCancelled());
559}
560
561TEST_F(DHCPConfigTest, StopDuringRequestIP) {
562 config_->pid_ = 567;
563 EXPECT_CALL(*proxy_, Rebind(kDeviceName)).Times(1);
564 config_->proxy_.reset(proxy_.release());
565 EXPECT_TRUE(config_->RenewIP());
566 EXPECT_FALSE(config_->lease_acquisition_timeout_callback_.IsCancelled());
567 config_->pid_ = 0; // Keep Stop from killing a real process.
568 config_->Stop();
569 EXPECT_TRUE(config_->lease_acquisition_timeout_callback_.IsCancelled());
Darin Petkov98dd6a02011-06-10 15:12:57 -0700570}
571
mukesh agrawal6bb9e7c2012-01-30 14:57:54 -0800572TEST_F(DHCPConfigTest, SetProperty) {
Chris Masone43b48a12011-07-01 13:37:07 -0700573 ::DBus::Error error;
574 // Ensure that an attempt to write a R/O property returns InvalidArgs error.
mukesh agrawal6bb9e7c2012-01-30 14:57:54 -0800575 EXPECT_FALSE(DBusAdaptor::SetProperty(config_->mutable_store(),
576 flimflam::kAddressProperty,
577 PropertyStoreTest::kStringV,
578 &error));
Chris Masone9d779932011-08-25 16:33:41 -0700579 EXPECT_EQ(invalid_args(), error.name());
Chris Masone43b48a12011-07-01 13:37:07 -0700580}
581
Darin Petkove7cb7f82011-06-03 13:21:51 -0700582} // namespace shill