blob: e225f501d1486ffe041548e12979136ed1d3fa57 [file] [log] [blame]
// Copyright 2018 The Chromium OS Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "shill/connection.h"
#include <arpa/inet.h>
#include <linux/rtnetlink.h>
#include <memory>
#include <string>
#include <vector>
#include <gtest/gtest.h>
#include <gmock/gmock.h>
#include "shill/ipconfig.h"
#include "shill/mock_connection.h"
#include "shill/mock_control.h"
#include "shill/mock_device.h"
#include "shill/mock_device_info.h"
#include "shill/mock_resolver.h"
#include "shill/mock_routing_table.h"
#include "shill/net/mock_rtnl_handler.h"
#include "shill/routing_table_entry.h"
using std::string;
using std::vector;
using testing::_;
using testing::Mock;
using testing::Return;
using testing::ReturnRef;
using testing::StrictMock;
using testing::Test;
namespace shill {
namespace {
const char kTestDeviceName0[] = "netdev0";
const int kTestDeviceInterfaceIndex0 = 123;
const char kTestDeviceName1[] = "netdev1";
const int kTestDeviceInterfaceIndex1 = 321;
const char kIPAddress0[] = "192.168.1.1";
const char kGatewayAddress0[] = "192.168.1.254";
const char kBroadcastAddress0[] = "192.168.1.255";
const char kNameServer0[] = "8.8.8.8";
const char kNameServer1[] = "8.8.9.9";
const int32_t kPrefix0 = 24;
const int32_t kPrefix1 = 31;
const char kSearchDomain0[] = "chromium.org";
const char kSearchDomain1[] = "google.com";
const char kIPv6Address[] = "2001:db8::1";
const char kIPv6NameServer0[] = "2001:db9::1";
const char kIPv6NameServer1[] = "2001:db9::2";
MATCHER_P2(IsIPAddress, address, prefix, "") {
IPAddress match_address(address);
match_address.set_prefix(prefix);
return match_address.Equals(arg);
}
MATCHER_P(IsIPv6Address, address, "") {
IPAddress match_address(address);
return match_address.Equals(arg);
}
MATCHER(IsDefaultAddress, "") {
IPAddress match_address(arg);
return match_address.IsDefault();
}
MATCHER_P(IsValidRoutingTableEntry, dst, "") {
return dst.Equals(arg.dst);
}
MATCHER_P(IsValidThrowRoute, dst, "") {
return dst.Equals(arg.dst) && arg.type == RTN_THROW;
}
MATCHER_P2(IsValidRoutingRule, family, priority, "") {
return arg.family == family && arg.priority == priority;
}
MATCHER_P3(IsValidUidRule, family, priority, uid, "") {
return arg.family == family && arg.priority == priority && arg.has_uidrange &&
arg.uidrange_start == uid && arg.uidrange_end == uid;
}
MATCHER_P(IsLinkRouteTo, dst, "") {
return dst.HasSameAddressAs(arg.dst) &&
arg.dst.prefix() ==
IPAddress::GetMaxPrefixLength(IPAddress::kFamilyIPv4) &&
!arg.src.IsValid() && !arg.gateway.IsValid() &&
arg.scope == RT_SCOPE_LINK && !arg.from_rtnl;
}
} // namespace
class ConnectionTest : public Test {
public:
ConnectionTest()
: device_info_(new StrictMock<MockDeviceInfo>(
&control_, nullptr, nullptr, nullptr)),
connection_(new Connection(kTestDeviceInterfaceIndex0,
kTestDeviceName0,
false,
Technology::kUnknown,
device_info_.get(),
&control_)),
ipconfig_(new IPConfig(&control_, kTestDeviceName0)),
ip6config_(new IPConfig(&control_, kTestDeviceName0)),
local_address_(IPAddress::kFamilyIPv4),
broadcast_address_(IPAddress::kFamilyIPv4),
gateway_address_(IPAddress::kFamilyIPv4),
default_address_(IPAddress::kFamilyIPv4),
local_ipv6_address_(IPAddress::kFamilyIPv6) {}
void SetUp() override {
ReplaceSingletons(connection_);
properties_.address = kIPAddress0;
properties_.subnet_prefix = kPrefix0;
properties_.gateway = kGatewayAddress0;
properties_.broadcast_address = kBroadcastAddress0;
properties_.dns_servers.push_back(kNameServer0);
properties_.dns_servers.push_back(kNameServer1);
properties_.domain_search.push_back(kSearchDomain0);
properties_.domain_search.push_back(kSearchDomain1);
properties_.address_family = IPAddress::kFamilyIPv4;
UpdateProperties();
ipv6_properties_.address = kIPv6Address;
ipv6_properties_.dns_servers.push_back(kIPv6NameServer0);
ipv6_properties_.dns_servers.push_back(kIPv6NameServer1);
ipv6_properties_.address_family = IPAddress::kFamilyIPv6;
UpdateIPv6Properties();
EXPECT_TRUE(local_address_.SetAddressFromString(kIPAddress0));
EXPECT_TRUE(broadcast_address_.SetAddressFromString(kBroadcastAddress0));
EXPECT_TRUE(gateway_address_.SetAddressFromString(kGatewayAddress0));
EXPECT_TRUE(local_ipv6_address_.SetAddressFromString(kIPv6Address));
}
void TearDown() override {
AddDestructorExpectations();
connection_ = nullptr;
}
void ReplaceSingletons(ConnectionRefPtr connection) {
connection->resolver_ = &resolver_;
connection->routing_table_ = &routing_table_;
connection->rtnl_handler_ = &rtnl_handler_;
}
void UpdateProperties() {
ipconfig_->UpdateProperties(properties_, true);
}
void UpdateIPv6Properties() {
ip6config_->UpdateProperties(ipv6_properties_, true);
}
const IPAddress& GetLocalAddress(ConnectionRefPtr connection) {
return connection->local_;
}
const IPAddress& GetGatewayAddress(ConnectionRefPtr connection) {
return connection->gateway_;
}
bool GetHasBroadcastDomain(ConnectionRefPtr connection) {
return connection->has_broadcast_domain_;
}
void SetLocal(const IPAddress& local) {
connection_->local_ = local;
}
protected:
class DisconnectCallbackTarget {
public:
DisconnectCallbackTarget()
: callback_(base::Bind(&DisconnectCallbackTarget::CallTarget,
base::Unretained(this))) {}
MOCK_METHOD0(CallTarget, void());
const base::Closure& callback() const { return callback_; }
private:
base::Closure callback_;
};
void AddDestructorExpectations() {
EXPECT_CALL(routing_table_, FlushRoutes(kTestDeviceInterfaceIndex0));
EXPECT_CALL(routing_table_, FlushRoutesWithTag(kTestDeviceInterfaceIndex0));
EXPECT_CALL(*device_info_, FlushAddresses(kTestDeviceInterfaceIndex0));
EXPECT_CALL(routing_table_, FlushRules(kTestDeviceInterfaceIndex0));
EXPECT_CALL(routing_table_, FreeTableId(_));
}
void AddRoutingPolicyExpectations(int interface_index,
uint32_t priority) {
EXPECT_CALL(routing_table_, FlushRules(interface_index));
EXPECT_CALL(routing_table_,
AddRule(interface_index,
IsValidRoutingRule(IPAddress::kFamilyIPv4, priority)))
.WillOnce(Return(true));
EXPECT_CALL(routing_table_,
AddRule(interface_index,
IsValidRoutingRule(IPAddress::kFamilyIPv6, priority)))
.WillOnce(Return(true));
}
// Returns a new test connection object. The caller usually needs to call
// AddDestructorExpectations before destroying the object.
ConnectionRefPtr GetNewConnection() {
ConnectionRefPtr connection(new Connection(kTestDeviceInterfaceIndex0,
kTestDeviceName0,
false,
Technology::kUnknown,
device_info_.get(),
&control_));
ReplaceSingletons(connection);
return connection;
}
std::unique_ptr<StrictMock<MockDeviceInfo>> device_info_;
ConnectionRefPtr connection_;
MockControl control_;
IPConfigRefPtr ipconfig_;
IPConfigRefPtr ip6config_;
IPConfig::Properties properties_;
IPConfig::Properties ipv6_properties_;
IPAddress local_address_;
IPAddress broadcast_address_;
IPAddress gateway_address_;
IPAddress default_address_;
IPAddress local_ipv6_address_;
StrictMock<MockResolver> resolver_;
StrictMock<MockRoutingTable> routing_table_;
StrictMock<MockRTNLHandler> rtnl_handler_;
};
TEST_F(ConnectionTest, InitState) {
EXPECT_EQ(kTestDeviceInterfaceIndex0, connection_->interface_index_);
EXPECT_EQ(kTestDeviceName0, connection_->interface_name_);
EXPECT_FALSE(connection_->IsDefault());
EXPECT_FALSE(connection_->routing_request_count_);
}
TEST_F(ConnectionTest, AddConfig) {
EXPECT_CALL(*device_info_,
HasOtherAddress(kTestDeviceInterfaceIndex0,
IsIPAddress(local_address_, kPrefix0)))
.WillOnce(Return(false));
EXPECT_CALL(rtnl_handler_,
AddInterfaceAddress(kTestDeviceInterfaceIndex0,
IsIPAddress(local_address_, kPrefix0),
IsIPAddress(broadcast_address_, 0),
IsIPAddress(default_address_, 0)));
EXPECT_CALL(routing_table_,
SetDefaultRoute(kTestDeviceInterfaceIndex0,
IsIPAddress(gateway_address_, 0),
_, RT_TABLE_MAIN));
AddRoutingPolicyExpectations(kTestDeviceInterfaceIndex0,
Connection::kLowestPriorityMetric);
EXPECT_CALL(routing_table_,
ConfigureRoutes(kTestDeviceInterfaceIndex0,
ipconfig_,
Connection::kDefaultMetric,
RT_TABLE_MAIN));
EXPECT_CALL(rtnl_handler_, SetInterfaceMTU(kTestDeviceInterfaceIndex0,
IPConfig::kDefaultMTU));
connection_->UpdateFromIPConfig(ipconfig_);
IPAddress test_local_address(local_address_);
test_local_address.set_prefix(kPrefix0);
EXPECT_TRUE(test_local_address.Equals(GetLocalAddress(connection_)));
EXPECT_TRUE(gateway_address_.Equals(GetGatewayAddress(connection_)));
EXPECT_TRUE(GetHasBroadcastDomain(connection_));
EXPECT_FALSE(connection_->IsIPv6());
AddRoutingPolicyExpectations(kTestDeviceInterfaceIndex0,
Connection::kDefaultMetric);
EXPECT_CALL(routing_table_, SetDefaultMetric(kTestDeviceInterfaceIndex0,
Connection::kDefaultMetric));
EXPECT_CALL(resolver_, SetDNSFromLists(
ipconfig_->properties().dns_servers,
ipconfig_->properties().domain_search));
scoped_refptr<MockDevice> device(new StrictMock<MockDevice>(
&control_,
nullptr,
nullptr,
nullptr,
kTestDeviceName0,
string(),
kTestDeviceInterfaceIndex0));
EXPECT_CALL(*device_info_, GetDevice(kTestDeviceInterfaceIndex0))
.WillOnce(Return(device));
EXPECT_CALL(*device, RequestPortalDetection()).WillOnce(Return(true));
EXPECT_CALL(routing_table_, FlushCache())
.WillOnce(Return(true));
connection_->SetUseDNS(true);
connection_->SetMetric(Connection::kDefaultMetric, true);
Mock::VerifyAndClearExpectations(&routing_table_);
EXPECT_TRUE(connection_->IsDefault());
AddRoutingPolicyExpectations(kTestDeviceInterfaceIndex0,
Connection::kLowestPriorityMetric);
EXPECT_CALL(routing_table_,
SetDefaultMetric(kTestDeviceInterfaceIndex0, _));
EXPECT_CALL(routing_table_, FlushCache())
.WillOnce(Return(true));
connection_->SetUseDNS(false);
connection_->SetMetric(Connection::kLowestPriorityMetric, false);
EXPECT_FALSE(connection_->IsDefault());
}
TEST_F(ConnectionTest, AddConfigUserTrafficOnly) {
ConnectionRefPtr connection = GetNewConnection();
const std::string kExcludeAddress1 = "192.0.1.0/24";
const std::string kExcludeAddress2 = "192.0.2.0/24";
const uint8_t table_id = 8;
const uint32_t uid = 1000;
IPAddress address1(IPAddress::kFamilyIPv4);
IPAddress address2(IPAddress::kFamilyIPv4);
EXPECT_TRUE(address1.SetAddressAndPrefixFromString(kExcludeAddress1));
EXPECT_TRUE(address2.SetAddressAndPrefixFromString(kExcludeAddress2));
EXPECT_CALL(*device_info_,
HasOtherAddress(kTestDeviceInterfaceIndex0,
IsIPAddress(local_address_, kPrefix0)))
.WillOnce(Return(false));
EXPECT_CALL(rtnl_handler_,
AddInterfaceAddress(kTestDeviceInterfaceIndex0,
IsIPAddress(local_address_, kPrefix0),
IsIPAddress(broadcast_address_, 0),
IsIPAddress(default_address_, 0)));
EXPECT_CALL(
routing_table_,
ConfigureRoutes(
kTestDeviceInterfaceIndex0, ipconfig_, Connection::kDefaultMetric,
table_id));
EXPECT_CALL(rtnl_handler_, SetInterfaceMTU(kTestDeviceInterfaceIndex0,
IPConfig::kDefaultMTU));
// SetupExcludedRoutes should create RTN_THROW entries for both networks.
EXPECT_CALL(routing_table_, AddRoute(kTestDeviceInterfaceIndex0,
IsValidThrowRoute(address1)))
.WillOnce(Return(true));
EXPECT_CALL(routing_table_, AddRoute(kTestDeviceInterfaceIndex0,
IsValidThrowRoute(address2)))
.WillOnce(Return(true));
// UpdateRoutingPolicy should create rules for IPv4 and IPv6
EXPECT_CALL(routing_table_, FreeTableId(RT_TABLE_MAIN));
EXPECT_CALL(routing_table_, AllocTableId()).WillOnce(Return(table_id));
EXPECT_CALL(routing_table_, FlushRules(kTestDeviceInterfaceIndex0));
EXPECT_CALL(
routing_table_,
AddRule(kTestDeviceInterfaceIndex0,
IsValidUidRule(IPAddress::kFamilyIPv4,
Connection::kLowestPriorityMetric, uid)))
.WillOnce(Return(true));
EXPECT_CALL(
routing_table_,
AddRule(kTestDeviceInterfaceIndex0,
IsValidUidRule(IPAddress::kFamilyIPv6,
Connection::kLowestPriorityMetric, uid)))
.WillOnce(Return(true));
properties_.allowed_uids.push_back(uid);
properties_.default_route = false;
properties_.exclusion_list.push_back(kExcludeAddress1);
properties_.exclusion_list.push_back(kExcludeAddress2);
UpdateProperties();
connection->UpdateFromIPConfig(ipconfig_);
scoped_refptr<MockDevice> device1(
new MockDevice(&control_, nullptr, nullptr, nullptr, kTestDeviceName1,
string(), kTestDeviceInterfaceIndex1));
scoped_refptr<MockConnection> mock_connection(
new MockConnection(device_info_.get()));
ConnectionRefPtr device_connection = mock_connection.get();
EXPECT_CALL(*device_info_,
FlushAddresses(mock_connection->interface_index()));
const string kInterfaceName(kTestDeviceName1);
EXPECT_CALL(*mock_connection, interface_name())
.WillRepeatedly(ReturnRef(kInterfaceName));
EXPECT_CALL(*device1, connection())
.WillRepeatedly(testing::ReturnRef(device_connection));
IPAddress test_local_address(local_address_);
test_local_address.set_prefix(kPrefix0);
EXPECT_TRUE(test_local_address.Equals(GetLocalAddress(connection)));
EXPECT_TRUE(gateway_address_.Equals(GetGatewayAddress(connection)));
EXPECT_TRUE(GetHasBroadcastDomain(connection));
EXPECT_FALSE(connection->IsIPv6());
EXPECT_CALL(routing_table_, FlushRules(kTestDeviceInterfaceIndex0));
EXPECT_CALL(
routing_table_,
AddRule(kTestDeviceInterfaceIndex0,
IsValidUidRule(IPAddress::kFamilyIPv4,
Connection::kDefaultMetric,
uid)))
.WillOnce(Return(true));
EXPECT_CALL(
routing_table_,
AddRule(kTestDeviceInterfaceIndex0,
IsValidUidRule(IPAddress::kFamilyIPv6,
Connection::kDefaultMetric,
uid)))
.WillOnce(Return(true));
EXPECT_CALL(resolver_,
SetDNSFromLists(ipconfig_->properties().dns_servers,
ipconfig_->properties().domain_search));
scoped_refptr<MockDevice> device(new StrictMock<MockDevice>(
&control_, nullptr, nullptr, nullptr, kTestDeviceName0, string(),
kTestDeviceInterfaceIndex0));
EXPECT_CALL(*device_info_, GetDevice(kTestDeviceInterfaceIndex0))
.WillOnce(Return(device));
EXPECT_CALL(*device, RequestPortalDetection()).WillOnce(Return(true));
EXPECT_CALL(routing_table_, FlushCache()).WillOnce(Return(true));
connection->SetUseDNS(true);
connection->SetMetric(Connection::kDefaultMetric, true);
Mock::VerifyAndClearExpectations(&routing_table_);
EXPECT_TRUE(connection->IsDefault());
EXPECT_CALL(routing_table_, FlushRules(kTestDeviceInterfaceIndex0));
EXPECT_CALL(routing_table_,
AddRule(kTestDeviceInterfaceIndex0,
IsValidUidRule(IPAddress::kFamilyIPv4,
Connection::kLowestPriorityMetric, uid)))
.WillOnce(Return(true));
EXPECT_CALL(routing_table_,
AddRule(kTestDeviceInterfaceIndex0,
IsValidUidRule(IPAddress::kFamilyIPv6,
Connection::kLowestPriorityMetric, uid)))
.WillOnce(Return(true));
EXPECT_CALL(routing_table_, FlushCache()).WillOnce(Return(true));
connection->SetUseDNS(false);
connection->SetMetric(Connection::kLowestPriorityMetric, false);
EXPECT_FALSE(connection->IsDefault());
AddDestructorExpectations();
}
TEST_F(ConnectionTest, AddConfigIPv6) {
EXPECT_CALL(*device_info_,
HasOtherAddress(kTestDeviceInterfaceIndex0,
IsIPv6Address(local_ipv6_address_)))
.WillOnce(Return(false));
EXPECT_CALL(rtnl_handler_,
AddInterfaceAddress(kTestDeviceInterfaceIndex0,
IsIPv6Address(local_ipv6_address_),
IsDefaultAddress(),
_));
AddRoutingPolicyExpectations(kTestDeviceInterfaceIndex0,
Connection::kLowestPriorityMetric);
EXPECT_CALL(routing_table_,
ConfigureRoutes(kTestDeviceInterfaceIndex0,
ip6config_,
Connection::kDefaultMetric,
RT_TABLE_MAIN));
EXPECT_CALL(rtnl_handler_, SetInterfaceMTU(kTestDeviceInterfaceIndex0,
IPConfig::kDefaultMTU));
connection_->UpdateFromIPConfig(ip6config_);
IPAddress test_local_address(local_ipv6_address_);
EXPECT_TRUE(test_local_address.Equals(GetLocalAddress(connection_)));
EXPECT_TRUE(connection_->IsIPv6());
}
TEST_F(ConnectionTest, AddConfigWithPeer) {
const string kPeerAddress("192.168.1.222");
IPAddress peer_address(IPAddress::kFamilyIPv4);
EXPECT_TRUE(peer_address.SetAddressFromString(kPeerAddress));
properties_.peer_address = kPeerAddress;
properties_.gateway = string();
UpdateProperties();
EXPECT_CALL(*device_info_,
HasOtherAddress(kTestDeviceInterfaceIndex0,
IsIPAddress(local_address_, kPrefix0)))
.WillOnce(Return(false));
EXPECT_CALL(rtnl_handler_,
AddInterfaceAddress(kTestDeviceInterfaceIndex0,
IsIPAddress(local_address_, kPrefix0),
IsIPAddress(broadcast_address_, 0),
IsIPAddress(peer_address, 0)));
EXPECT_CALL(routing_table_, SetDefaultRoute(_, _, _, _)).Times(0);
AddRoutingPolicyExpectations(kTestDeviceInterfaceIndex0,
Connection::kLowestPriorityMetric);
EXPECT_CALL(routing_table_,
ConfigureRoutes(kTestDeviceInterfaceIndex0,
ipconfig_,
Connection::kDefaultMetric,
RT_TABLE_MAIN));
EXPECT_CALL(rtnl_handler_, SetInterfaceMTU(kTestDeviceInterfaceIndex0,
IPConfig::kDefaultMTU));
connection_->UpdateFromIPConfig(ipconfig_);
EXPECT_FALSE(GetHasBroadcastDomain(connection_));
}
TEST_F(ConnectionTest, AddConfigWithBrokenNetmask) {
// Assign a prefix that makes the gateway unreachable.
properties_.subnet_prefix = kPrefix1;
UpdateProperties();
// Connection should add a link route which will allow the
// gateway to be reachable.
IPAddress gateway_address(IPAddress::kFamilyIPv4);
EXPECT_TRUE(gateway_address.SetAddressFromString(kGatewayAddress0));
EXPECT_CALL(routing_table_, AddRoute(kTestDeviceInterfaceIndex0,
IsLinkRouteTo(gateway_address)))
.WillOnce(Return(true));
EXPECT_CALL(*device_info_,
HasOtherAddress(kTestDeviceInterfaceIndex0,
IsIPAddress(local_address_, kPrefix1)))
.WillOnce(Return(false));
EXPECT_CALL(rtnl_handler_,
AddInterfaceAddress(kTestDeviceInterfaceIndex0,
IsIPAddress(local_address_, kPrefix1),
IsIPAddress(broadcast_address_, 0),
IsIPAddress(default_address_, 0)));
EXPECT_CALL(routing_table_,
SetDefaultRoute(kTestDeviceInterfaceIndex0,
IsIPAddress(gateway_address_, 0),
_, RT_TABLE_MAIN));
AddRoutingPolicyExpectations(kTestDeviceInterfaceIndex0,
Connection::kLowestPriorityMetric);
EXPECT_CALL(routing_table_,
ConfigureRoutes(kTestDeviceInterfaceIndex0,
ipconfig_,
Connection::kDefaultMetric,
RT_TABLE_MAIN));
EXPECT_CALL(rtnl_handler_, SetInterfaceMTU(kTestDeviceInterfaceIndex0,
IPConfig::kDefaultMTU));
connection_->UpdateFromIPConfig(ipconfig_);
}
TEST_F(ConnectionTest, AddConfigReverse) {
AddRoutingPolicyExpectations(kTestDeviceInterfaceIndex0,
Connection::kDefaultMetric);
EXPECT_CALL(routing_table_, SetDefaultMetric(kTestDeviceInterfaceIndex0,
Connection::kDefaultMetric));
vector<string> empty_list;
EXPECT_CALL(resolver_, SetDNSFromLists(empty_list, empty_list));
scoped_refptr<MockDevice> device(new StrictMock<MockDevice>(
&control_,
nullptr,
nullptr,
nullptr,
kTestDeviceName0,
string(),
kTestDeviceInterfaceIndex0));
EXPECT_CALL(*device_info_, GetDevice(kTestDeviceInterfaceIndex0))
.WillOnce(Return(device));
EXPECT_CALL(*device, RequestPortalDetection()).WillOnce(Return(true));
EXPECT_CALL(routing_table_, FlushCache())
.WillOnce(Return(true));
connection_->SetUseDNS(true);
connection_->SetMetric(Connection::kDefaultMetric, true);
Mock::VerifyAndClearExpectations(&routing_table_);
EXPECT_CALL(*device_info_,
HasOtherAddress(kTestDeviceInterfaceIndex0,
IsIPAddress(local_address_, kPrefix0)))
.WillOnce(Return(false));
EXPECT_CALL(rtnl_handler_,
AddInterfaceAddress(kTestDeviceInterfaceIndex0,
IsIPAddress(local_address_, kPrefix0),
IsIPAddress(broadcast_address_, 0),
IsIPAddress(default_address_, 0)));
EXPECT_CALL(routing_table_, SetDefaultRoute(kTestDeviceInterfaceIndex0,
IsIPAddress(gateway_address_, 0),
Connection::kDefaultMetric,
RT_TABLE_MAIN));
AddRoutingPolicyExpectations(kTestDeviceInterfaceIndex0,
Connection::kDefaultMetric);
EXPECT_CALL(routing_table_,
ConfigureRoutes(kTestDeviceInterfaceIndex0,
ipconfig_,
Connection::kDefaultMetric,
RT_TABLE_MAIN));
EXPECT_CALL(resolver_,
SetDNSFromLists(ipconfig_->properties().dns_servers,
ipconfig_->properties().domain_search));
EXPECT_CALL(rtnl_handler_, SetInterfaceMTU(kTestDeviceInterfaceIndex0,
IPConfig::kDefaultMTU));
connection_->UpdateFromIPConfig(ipconfig_);
}
TEST_F(ConnectionTest, AddConfigWithDNSDomain) {
const string kDomainName("chromium.org");
properties_.domain_search.clear();
properties_.domain_name = kDomainName;
UpdateProperties();
EXPECT_CALL(*device_info_, HasOtherAddress(_, _))
.WillOnce(Return(false));
EXPECT_CALL(rtnl_handler_, AddInterfaceAddress(_, _, _, _));
EXPECT_CALL(routing_table_, SetDefaultRoute(_, _, _, _));
AddRoutingPolicyExpectations(kTestDeviceInterfaceIndex0,
Connection::kLowestPriorityMetric);
EXPECT_CALL(routing_table_, ConfigureRoutes(_, _, _, _));
EXPECT_CALL(rtnl_handler_, SetInterfaceMTU(_, _));
connection_->UpdateFromIPConfig(ipconfig_);
AddRoutingPolicyExpectations(kTestDeviceInterfaceIndex0,
Connection::kDefaultMetric);
EXPECT_CALL(routing_table_, SetDefaultMetric(_, _));
vector<string> domain_search_list;
domain_search_list.push_back(kDomainName + ".");
EXPECT_CALL(resolver_, SetDNSFromLists(_, domain_search_list));
DeviceRefPtr device;
EXPECT_CALL(*device_info_, GetDevice(_)).WillOnce(Return(device));
EXPECT_CALL(routing_table_, FlushCache()).WillOnce(Return(true));
connection_->SetUseDNS(true);
connection_->SetMetric(Connection::kDefaultMetric, true);
}
TEST_F(ConnectionTest, AddConfigWithFixedIpParams) {
ConnectionRefPtr connection(new Connection(kTestDeviceInterfaceIndex0,
kTestDeviceName0,
true,
Technology::kUnknown,
device_info_.get(),
&control_));
ReplaceSingletons(connection);
// Initial setup: routes but no IP configuration.
EXPECT_CALL(*device_info_, HasOtherAddress(_, _)).Times(0);
EXPECT_CALL(rtnl_handler_, AddInterfaceAddress(_, _, _, _)).Times(0);
EXPECT_CALL(routing_table_, SetDefaultRoute(_, _, _, _));
AddRoutingPolicyExpectations(kTestDeviceInterfaceIndex0,
Connection::kLowestPriorityMetric);
EXPECT_CALL(routing_table_, ConfigureRoutes(_, _, _, _));
EXPECT_CALL(rtnl_handler_, SetInterfaceMTU(_, _)).Times(0);
connection->UpdateFromIPConfig(ipconfig_);
Mock::VerifyAndClearExpectations(&routing_table_);
Mock::VerifyAndClearExpectations(&rtnl_handler_);
Mock::VerifyAndClearExpectations(device_info_.get());
// Change metric to make this the default service.
AddRoutingPolicyExpectations(kTestDeviceInterfaceIndex0,
Connection::kDefaultMetric);
EXPECT_CALL(routing_table_, SetDefaultMetric(_, _));
EXPECT_CALL(resolver_, SetDNSFromLists(_, _));
DeviceRefPtr device;
EXPECT_CALL(*device_info_, GetDevice(_)).WillOnce(Return(device));
EXPECT_CALL(routing_table_, FlushCache()).WillOnce(Return(true));
connection->SetUseDNS(true);
connection->SetMetric(Connection::kDefaultMetric, true);
// Destructor should flush routes + rules, but not addresses.
EXPECT_CALL(*device_info_, FlushAddresses(_)).Times(0);
EXPECT_CALL(routing_table_, FlushRoutes(kTestDeviceInterfaceIndex0));
EXPECT_CALL(routing_table_, FlushRoutesWithTag(kTestDeviceInterfaceIndex0));
EXPECT_CALL(routing_table_, FlushRules(kTestDeviceInterfaceIndex0));
EXPECT_CALL(routing_table_, FreeTableId(_));
}
TEST_F(ConnectionTest, HasOtherAddress) {
EXPECT_CALL(*device_info_,
HasOtherAddress(kTestDeviceInterfaceIndex0,
IsIPAddress(local_address_, kPrefix0)))
.WillOnce(Return(true));
EXPECT_CALL(routing_table_, FlushRoutes(kTestDeviceInterfaceIndex0));
EXPECT_CALL(*device_info_, FlushAddresses(kTestDeviceInterfaceIndex0));
EXPECT_CALL(rtnl_handler_,
AddInterfaceAddress(kTestDeviceInterfaceIndex0,
IsIPAddress(local_address_, kPrefix0),
IsIPAddress(broadcast_address_, 0),
IsIPAddress(default_address_, 0)));
EXPECT_CALL(routing_table_,
SetDefaultRoute(kTestDeviceInterfaceIndex0,
IsIPAddress(gateway_address_, 0),
_, RT_TABLE_MAIN));
AddRoutingPolicyExpectations(kTestDeviceInterfaceIndex0,
Connection::kLowestPriorityMetric);
EXPECT_CALL(routing_table_,
ConfigureRoutes(kTestDeviceInterfaceIndex0,
ipconfig_,
Connection::kDefaultMetric,
RT_TABLE_MAIN));
EXPECT_CALL(rtnl_handler_, SetInterfaceMTU(kTestDeviceInterfaceIndex0,
IPConfig::kDefaultMTU));
connection_->UpdateFromIPConfig(ipconfig_);
}
TEST_F(ConnectionTest, UpdateDNSServers) {
static const char* const kDnsServers[] = {"1.1.1.1", "1.1.1.2"};
vector<string> dns_servers(kDnsServers, std::end(kDnsServers));
// Non-default connection.
connection_->metric_ = Connection::kLowestPriorityMetric;
EXPECT_CALL(resolver_, SetDNSFromLists(_, _)).Times(0);
connection_->UpdateDNSServers(dns_servers);
Mock::VerifyAndClearExpectations(&resolver_);
// Default connection.
connection_->use_dns_ = true;
connection_->metric_ = Connection::kDefaultMetric;
EXPECT_CALL(resolver_, SetDNSFromLists(dns_servers, _));
connection_->UpdateDNSServers(dns_servers);
Mock::VerifyAndClearExpectations(&resolver_);
}
TEST_F(ConnectionTest, RouteRequest) {
ConnectionRefPtr connection = GetNewConnection();
scoped_refptr<MockDevice> device(new StrictMock<MockDevice>(
&control_,
nullptr,
nullptr,
nullptr,
kTestDeviceName0,
string(),
kTestDeviceInterfaceIndex0));
EXPECT_CALL(*device_info_, GetDevice(kTestDeviceInterfaceIndex0))
.WillRepeatedly(Return(device));
EXPECT_CALL(*device, SetLooseRouting(true)).Times(1);
connection->RequestRouting();
connection->RequestRouting();
// The first release should only decrement the reference counter.
connection->ReleaseRouting();
// Another release will re-enable reverse-path filter.
EXPECT_CALL(*device, SetLooseRouting(false));
EXPECT_CALL(routing_table_, FlushCache());
connection->ReleaseRouting();
// The destructor will remove the routes and addresses.
AddDestructorExpectations();
}
TEST_F(ConnectionTest, Destructor) {
ConnectionRefPtr connection(new Connection(kTestDeviceInterfaceIndex1,
kTestDeviceName1,
false,
Technology::kUnknown,
device_info_.get(),
&control_));
connection->resolver_ = &resolver_;
connection->routing_table_ = &routing_table_;
connection->rtnl_handler_ = &rtnl_handler_;
EXPECT_CALL(routing_table_, FlushRoutes(kTestDeviceInterfaceIndex1));
EXPECT_CALL(routing_table_, FlushRoutesWithTag(kTestDeviceInterfaceIndex1));
EXPECT_CALL(*device_info_, FlushAddresses(kTestDeviceInterfaceIndex1));
EXPECT_CALL(routing_table_, FlushRules(kTestDeviceInterfaceIndex1));
EXPECT_CALL(routing_table_, FreeTableId(RT_TABLE_MAIN));
connection = nullptr;
}
TEST_F(ConnectionTest, BlackholeIPv6) {
const uint8_t table_id = 9;
properties_.blackhole_ipv6 = true;
UpdateProperties();
EXPECT_CALL(*device_info_, HasOtherAddress(_, _))
.WillOnce(Return(false));
EXPECT_CALL(rtnl_handler_, AddInterfaceAddress(_, _, _, _));
EXPECT_CALL(routing_table_, SetDefaultRoute(_, _, _, _));
EXPECT_CALL(routing_table_, FreeTableId(RT_TABLE_MAIN));
EXPECT_CALL(routing_table_, AllocTableId()).WillOnce(Return(table_id));
EXPECT_CALL(routing_table_, FlushRules(_));
EXPECT_CALL(routing_table_, AddRule(_, _)).WillRepeatedly(Return(true));
EXPECT_CALL(routing_table_, ConfigureRoutes(_, _, _, _));
EXPECT_CALL(routing_table_,
CreateBlackholeRoute(kTestDeviceInterfaceIndex0,
IPAddress::kFamilyIPv6,
0,
table_id))
.WillOnce(Return(true));
EXPECT_CALL(rtnl_handler_, SetInterfaceMTU(kTestDeviceInterfaceIndex0,
IPConfig::kDefaultMTU));
connection_->UpdateFromIPConfig(ipconfig_);
}
TEST_F(ConnectionTest, FixGatewayReachability) {
ConnectionRefPtr connection = GetNewConnection();
static const char kLocal[] = "10.242.2.13";
IPAddress local(IPAddress::kFamilyIPv4);
ASSERT_TRUE(local.SetAddressFromString(kLocal));
const int kPrefix = 24;
local.set_prefix(kPrefix);
IPAddress gateway(IPAddress::kFamilyIPv4);
IPAddress peer(IPAddress::kFamilyIPv4);
IPAddress trusted_ip(IPAddress::kFamilyIPv4);
// Should fail because no gateway is set.
EXPECT_FALSE(connection->FixGatewayReachability(
local, &peer, &gateway, trusted_ip));
EXPECT_EQ(kPrefix, local.prefix());
EXPECT_FALSE(peer.IsValid());
EXPECT_FALSE(gateway.IsValid());
// Should succeed because with the given prefix, this gateway is reachable.
static const char kReachableGateway[] = "10.242.2.14";
ASSERT_TRUE(gateway.SetAddressFromString(kReachableGateway));
IPAddress gateway_backup(gateway);
peer = IPAddress(IPAddress::kFamilyIPv4);
EXPECT_TRUE(connection->FixGatewayReachability(
local, &peer, &gateway, trusted_ip));
// Prefix should remain unchanged.
EXPECT_EQ(kPrefix, local.prefix());
// Peer should remain unchanged.
EXPECT_FALSE(peer.IsValid());
// Gateway should remain unchanged.
EXPECT_TRUE(gateway_backup.Equals(gateway));
// Should succeed because we created a link route to the gateway.
static const char kRemoteGateway[] = "10.242.3.14";
ASSERT_TRUE(gateway.SetAddressFromString(kRemoteGateway));
gateway_backup = gateway;
peer = IPAddress(IPAddress::kFamilyIPv4);
EXPECT_CALL(routing_table_, AddRoute(kTestDeviceInterfaceIndex0,
IsLinkRouteTo(gateway)))
.WillOnce(Return(true));
EXPECT_TRUE(connection->FixGatewayReachability(
local, &peer, &gateway, trusted_ip));
// Peer should remain unchanged.
EXPECT_FALSE(peer.IsValid());
// Gateway should remain unchanged.
EXPECT_TRUE(gateway_backup.Equals(gateway));
// Should fail if AddRoute() fails.
EXPECT_CALL(routing_table_, AddRoute(kTestDeviceInterfaceIndex0,
IsLinkRouteTo(gateway)))
.WillOnce(Return(false));
EXPECT_FALSE(connection->FixGatewayReachability(
local, &peer, &gateway, trusted_ip));
// If this is a peer-to-peer interface and the peer matches the gateway,
// we should succeed.
local.set_prefix(kPrefix);
static const char kUnreachableGateway[] = "11.242.2.14";
ASSERT_TRUE(gateway.SetAddressFromString(kUnreachableGateway));
gateway_backup = gateway;
ASSERT_TRUE(peer.SetAddressFromString(kUnreachableGateway));
EXPECT_TRUE(connection->FixGatewayReachability(
local, &peer, &gateway, trusted_ip));
EXPECT_EQ(kPrefix, local.prefix());
EXPECT_TRUE(peer.Equals(gateway));
EXPECT_TRUE(gateway_backup.Equals(gateway));
// If there is a peer specified and it does not match the gateway (even
// if it was reachable via netmask), we should fail.
ASSERT_TRUE(gateway.SetAddressFromString(kReachableGateway));
EXPECT_FALSE(connection->FixGatewayReachability(
local, &peer, &gateway, trusted_ip));
EXPECT_EQ(kPrefix, local.prefix());
EXPECT_FALSE(peer.Equals(gateway));
// If this is a peer-to-peer interface and the peer matches the gateway,
// but it also matches the trusted IP address, the gateway and peer address
// should be modified to allow routing to work correctly.
ASSERT_TRUE(gateway.SetAddressFromString(kUnreachableGateway));
ASSERT_TRUE(peer.SetAddressFromString(kUnreachableGateway));
ASSERT_TRUE(trusted_ip.SetAddressAndPrefixFromString(
string(kUnreachableGateway) + "/32"));
EXPECT_TRUE(connection->FixGatewayReachability(
local, &peer, &gateway, trusted_ip));
EXPECT_TRUE(peer.IsDefault());
EXPECT_TRUE(gateway.IsDefault());
// The destructor will remove the routes and addresses.
AddDestructorExpectations();
}
TEST_F(ConnectionTest, Binders) {
EXPECT_TRUE(connection_->binders_.empty());
DisconnectCallbackTarget target0;
DisconnectCallbackTarget target1;
DisconnectCallbackTarget target2;
DisconnectCallbackTarget target3;
Connection::Binder binder0("binder0", target0.callback());
Connection::Binder binder1("binder1", target1.callback());
Connection::Binder binder2("binder2", target2.callback());
Connection::Binder binder3("binder3", target3.callback());
binder0.Attach(connection_);
binder1.Attach(connection_);
EXPECT_CALL(target1, CallTarget()).Times(0);
binder1.Attach(connection_);
binder3.Attach(connection_);
binder2.Attach(connection_);
EXPECT_CALL(target3, CallTarget()).Times(0);
binder3.Attach(nullptr);
ASSERT_EQ(3, connection_->binders_.size());
EXPECT_TRUE(connection_->binders_.at(0) == &binder0);
EXPECT_TRUE(connection_->binders_.at(1) == &binder1);
EXPECT_TRUE(connection_->binders_.at(2) == &binder2);
EXPECT_CALL(target0, CallTarget()).Times(1);
EXPECT_CALL(target1, CallTarget()).Times(1);
EXPECT_CALL(target2, CallTarget()).Times(1);
connection_->NotifyBindersOnDisconnect();
EXPECT_TRUE(connection_->binders_.empty());
// Should be a no-op.
connection_->NotifyBindersOnDisconnect();
}
TEST_F(ConnectionTest, Binder) {
DisconnectCallbackTarget target0;
Connection::Binder binder("empty_test", target0.callback());
EXPECT_FALSE(binder.client_disconnect_callback_.is_null());
EXPECT_FALSE(binder.IsBound());
ConnectionRefPtr connection1 = GetNewConnection();
EXPECT_TRUE(connection1->binders_.empty());
// Bind lower |connection1| and check if it's bound.
binder.Attach(connection1);
EXPECT_TRUE(binder.IsBound());
EXPECT_EQ(connection1.get(), binder.connection().get());
ASSERT_FALSE(connection1->binders_.empty());
EXPECT_TRUE(&binder == connection1->binders_.at(0));
// Unbind lower |connection1| and check if it's unbound.
binder.Attach(nullptr);
EXPECT_FALSE(binder.IsBound());
EXPECT_TRUE(connection1->binders_.empty());
AddDestructorExpectations();
connection1 = nullptr;
EXPECT_FALSE(binder.IsBound());
{
// Test the weak pointer to the bound Connection. This is not a case that
// should occur but the weak pointer should handle it gracefully.
DisconnectCallbackTarget target;
Connection::Binder binder("test_weak", target.callback());
ConnectionRefPtr connection = GetNewConnection();
binder.Attach(connection);
// Make sure the connection doesn't notify the binder on destruction.
connection->binders_.clear();
AddDestructorExpectations();
EXPECT_CALL(target, CallTarget()).Times(0);
connection = nullptr;
// Ensure no crash -- the weak pointer to connection should be nullptr.
EXPECT_FALSE(binder.connection());
binder.Attach(nullptr);
}
}
TEST_F(ConnectionTest, GetSubnetName) {
EXPECT_EQ("", connection_->GetSubnetName());
IPAddress local("1.2.3.4");
local.set_prefix(24);
SetLocal(local);
EXPECT_EQ("1.2.3.0/24", connection_->GetSubnetName());
}
TEST_F(ConnectionTest, SetMTU) {
testing::InSequence seq;
EXPECT_CALL(rtnl_handler_, SetInterfaceMTU(
kTestDeviceInterfaceIndex0, IPConfig::kDefaultMTU));
connection_->SetMTU(0);
EXPECT_CALL(rtnl_handler_, SetInterfaceMTU(
kTestDeviceInterfaceIndex0, IPConfig::kDefaultMTU));
connection_->SetMTU(IPConfig::kUndefinedMTU);
// Test IPv4 minimum MTU.
SetLocal(local_address_);
EXPECT_CALL(rtnl_handler_, SetInterfaceMTU(
kTestDeviceInterfaceIndex0, IPConfig::kMinIPv4MTU));
connection_->SetMTU(1);
EXPECT_CALL(rtnl_handler_, SetInterfaceMTU(
kTestDeviceInterfaceIndex0, IPConfig::kMinIPv4MTU));
connection_->SetMTU(IPConfig::kMinIPv4MTU - 1);
EXPECT_CALL(rtnl_handler_, SetInterfaceMTU(
kTestDeviceInterfaceIndex0, IPConfig::kMinIPv4MTU));
connection_->SetMTU(IPConfig::kMinIPv4MTU);
EXPECT_CALL(rtnl_handler_, SetInterfaceMTU(
kTestDeviceInterfaceIndex0, IPConfig::kMinIPv4MTU + 1));
connection_->SetMTU(IPConfig::kMinIPv4MTU + 1);
// Test IPv6 minimum MTU.
SetLocal(local_ipv6_address_);
EXPECT_CALL(rtnl_handler_, SetInterfaceMTU(
kTestDeviceInterfaceIndex0, IPConfig::kMinIPv6MTU));
connection_->SetMTU(1);
EXPECT_CALL(rtnl_handler_, SetInterfaceMTU(
kTestDeviceInterfaceIndex0, IPConfig::kMinIPv6MTU));
connection_->SetMTU(IPConfig::kMinIPv6MTU - 1);
EXPECT_CALL(rtnl_handler_, SetInterfaceMTU(
kTestDeviceInterfaceIndex0, IPConfig::kMinIPv6MTU));
connection_->SetMTU(IPConfig::kMinIPv6MTU);
EXPECT_CALL(rtnl_handler_, SetInterfaceMTU(
kTestDeviceInterfaceIndex0, IPConfig::kMinIPv6MTU + 1));
connection_->SetMTU(IPConfig::kMinIPv6MTU + 1);
}
} // namespace shill