blob: c30d914e5a37042a0b209f172e24beff48b4f25d [file] [log] [blame]
// Copyright (c) 2012 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 <arpa/inet.h>
#include <linux/rtnetlink.h>
#include <string>
#include <vector>
#include <base/memory/scoped_ptr.h>
#include <gtest/gtest.h>
#include <gmock/gmock.h>
#include "shill/connection.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/mock_rtnl_handler.h"
#include "shill/routing_table_entry.h"
using std::string;
using std::vector;
using testing::_;
using testing::Mock;
using testing::NiceMock;
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 kGatewayAddress1[] = "192.168.2.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";
} // namespace
class ConnectionTest : public Test {
public:
ConnectionTest()
: device_info_(new StrictMock<MockDeviceInfo>(
&control_,
static_cast<EventDispatcher*>(NULL),
static_cast<Metrics*>(NULL),
static_cast<Manager*>(NULL))),
connection_(new Connection(
kTestDeviceInterfaceIndex0,
kTestDeviceName0,
Technology::kUnknown,
device_info_.get())),
ipconfig_(new IPConfig(&control_, kTestDeviceName0)),
local_address_(IPAddress::kFamilyIPv4),
broadcast_address_(IPAddress::kFamilyIPv4),
gateway_address_(IPAddress::kFamilyIPv4),
default_address_(IPAddress::kFamilyIPv4) {}
virtual void SetUp() {
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();
EXPECT_TRUE(local_address_.SetAddressFromString(kIPAddress0));
EXPECT_TRUE(broadcast_address_.SetAddressFromString(kBroadcastAddress0));
EXPECT_TRUE(gateway_address_.SetAddressFromString(kGatewayAddress0));
}
virtual void TearDown() {
AddDestructorExpectations();
connection_ = NULL;
}
void ReplaceSingletons(ConnectionRefPtr connection) {
connection->resolver_ = &resolver_;
connection->routing_table_ = &routing_table_;
connection->rtnl_handler_ = &rtnl_handler_;
}
void UpdateProperties() {
ipconfig_->UpdateProperties(properties_);
}
bool PinHostRoute(ConnectionRefPtr connection,
const IPAddress trusted_ip,
const IPAddress gateway) {
return connection->PinHostRoute(trusted_ip, gateway);
}
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_;
}
uint32_t GetDefaultMetric() {
return Connection::kDefaultMetric;
}
uint32_t GetNonDefaultMetricBase() {
return Connection::kNonDefaultMetricBase;
}
protected:
class DisconnectCallbackTarget {
public:
DisconnectCallbackTarget()
: callback_(base::Bind(&DisconnectCallbackTarget::CallTarget,
base::Unretained(this))) {}
MOCK_METHOD0(CallTarget, void());
const base::Closure &callback() { return callback_; }
private:
base::Closure callback_;
};
void AddDestructorExpectations() {
EXPECT_CALL(routing_table_, FlushRoutes(kTestDeviceInterfaceIndex0));
EXPECT_CALL(routing_table_, FlushRoutesWithTag(kTestDeviceInterfaceIndex0));
EXPECT_CALL(*device_info_.get(),
FlushAddresses(kTestDeviceInterfaceIndex0));
}
// 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,
Technology::kUnknown,
device_info_.get()));
ReplaceSingletons(connection);
return connection;
}
scoped_ptr<StrictMock<MockDeviceInfo> > device_info_;
ConnectionRefPtr connection_;
MockControl control_;
IPConfigRefPtr ipconfig_;
IPConfig::Properties properties_;
IPAddress local_address_;
IPAddress broadcast_address_;
IPAddress gateway_address_;
IPAddress default_address_;
StrictMock<MockResolver> resolver_;
StrictMock<MockRoutingTable> routing_table_;
StrictMock<MockRTNLHandler> rtnl_handler_;
};
namespace {
MATCHER_P2(IsIPAddress, address, prefix, "") {
IPAddress match_address(address);
match_address.set_prefix(prefix);
return match_address.Equals(arg);
}
MATCHER(IsNonNullCallback, "") {
return !arg.is_null();
}
} // namespace
TEST_F(ConnectionTest, InitState) {
EXPECT_EQ(kTestDeviceInterfaceIndex0, connection_->interface_index_);
EXPECT_EQ(kTestDeviceName0, connection_->interface_name_);
EXPECT_FALSE(connection_->is_default());
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),
GetNonDefaultMetricBase() +
kTestDeviceInterfaceIndex0));
EXPECT_CALL(routing_table_,
ConfigureRoutes(kTestDeviceInterfaceIndex0,
ipconfig_,
GetDefaultMetric()));
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_CALL(routing_table_,
CreateLinkRoute(kTestDeviceInterfaceIndex0,
IsIPAddress(local_address_, kPrefix0),
IsIPAddress(gateway_address_, 0)))
.WillOnce(Return(true))
.WillOnce(Return(false));
EXPECT_TRUE(connection_->CreateGatewayRoute());
EXPECT_FALSE(connection_->CreateGatewayRoute());
connection_->has_broadcast_domain_ = false;
EXPECT_FALSE(connection_->CreateGatewayRoute());
EXPECT_CALL(routing_table_, SetDefaultMetric(kTestDeviceInterfaceIndex0,
GetDefaultMetric()));
EXPECT_CALL(resolver_, SetDNSFromLists(
ipconfig_->properties().dns_servers,
ipconfig_->properties().domain_search));
scoped_refptr<MockDevice> device(new StrictMock<MockDevice>(
&control_,
reinterpret_cast<EventDispatcher *>(NULL),
reinterpret_cast<Metrics *>(NULL),
reinterpret_cast<Manager *>(NULL),
kTestDeviceName0,
string(),
kTestDeviceInterfaceIndex0));
EXPECT_CALL(*device_info_, GetDevice(kTestDeviceInterfaceIndex0))
.WillOnce(Return(device));
EXPECT_CALL(*device.get(), RequestPortalDetection())
.WillOnce(Return(true));
EXPECT_CALL(routing_table_, FlushCache())
.WillOnce(Return(true));
connection_->SetIsDefault(true);
Mock::VerifyAndClearExpectations(&routing_table_);
EXPECT_TRUE(connection_->is_default());
EXPECT_CALL(routing_table_,
SetDefaultMetric(kTestDeviceInterfaceIndex0,
GetNonDefaultMetricBase() +
kTestDeviceInterfaceIndex0));
EXPECT_CALL(routing_table_, FlushCache())
.WillOnce(Return(true));
connection_->SetIsDefault(false);
EXPECT_FALSE(connection_->is_default());
}
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);
EXPECT_CALL(routing_table_,
ConfigureRoutes(kTestDeviceInterfaceIndex0,
ipconfig_,
GetDefaultMetric()));
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 override with a prefix which will allow the
// gateway to be reachable.
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),
GetNonDefaultMetricBase() +
kTestDeviceInterfaceIndex0));
EXPECT_CALL(routing_table_,
ConfigureRoutes(kTestDeviceInterfaceIndex0,
ipconfig_,
GetDefaultMetric()));
connection_->UpdateFromIPConfig(ipconfig_);
// Assign a gateway address that violates the minimum plausible prefix
// the Connection can assign.
properties_.gateway = kGatewayAddress1;
UpdateProperties();
IPAddress gateway_address1(IPAddress::kFamilyIPv4);
EXPECT_TRUE(gateway_address1.SetAddressFromString(kGatewayAddress1));
// Connection cannot override this prefix, so it will switch to a
// model where the peer address is set to the value of the gateway
// address.
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(gateway_address1, 0)));
EXPECT_CALL(routing_table_,
SetDefaultRoute(kTestDeviceInterfaceIndex0, _, _));
EXPECT_CALL(routing_table_,
ConfigureRoutes(kTestDeviceInterfaceIndex0, _, _));
connection_->UpdateFromIPConfig(ipconfig_);
}
TEST_F(ConnectionTest, AddConfigReverse) {
EXPECT_CALL(routing_table_, SetDefaultMetric(kTestDeviceInterfaceIndex0,
GetDefaultMetric()));
vector<string> empty_list;
EXPECT_CALL(resolver_, SetDNSFromLists(empty_list, empty_list));
scoped_refptr<MockDevice> device(new StrictMock<MockDevice>(
&control_,
reinterpret_cast<EventDispatcher *>(NULL),
reinterpret_cast<Metrics *>(NULL),
reinterpret_cast<Manager *>(NULL),
kTestDeviceName0,
string(),
kTestDeviceInterfaceIndex0));
EXPECT_CALL(*device_info_, GetDevice(kTestDeviceInterfaceIndex0))
.WillOnce(Return(device));
EXPECT_CALL(*device.get(), RequestPortalDetection())
.WillOnce(Return(true));
EXPECT_CALL(routing_table_, FlushCache())
.WillOnce(Return(true));
connection_->SetIsDefault(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),
GetDefaultMetric()));
EXPECT_CALL(routing_table_,
ConfigureRoutes(kTestDeviceInterfaceIndex0,
ipconfig_,
GetDefaultMetric()));
EXPECT_CALL(resolver_,
SetDNSFromLists(ipconfig_->properties().dns_servers,
ipconfig_->properties().domain_search));
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(_, _, _));
EXPECT_CALL(routing_table_, ConfigureRoutes(_, _, _));
connection_->UpdateFromIPConfig(ipconfig_);
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_->SetIsDefault(true);
}
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),
GetNonDefaultMetricBase() +
kTestDeviceInterfaceIndex0));
EXPECT_CALL(routing_table_,
ConfigureRoutes(kTestDeviceInterfaceIndex0,
ipconfig_,
GetDefaultMetric()));
connection_->UpdateFromIPConfig(ipconfig_);
}
TEST_F(ConnectionTest, UpdateDNSServers) {
const char* kDnsServers[] = {"1.1.1.1", "1.1.1.2"};
vector<string> dns_servers(kDnsServers, std::end(kDnsServers));
// Non-default connection.
connection_->is_default_ = false;
EXPECT_CALL(resolver_, SetDNSFromLists(_, _)).Times(0);
connection_->UpdateDNSServers(dns_servers);
Mock::VerifyAndClearExpectations(&resolver_);
// Default connection.
connection_->is_default_ = true;
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_,
reinterpret_cast<EventDispatcher *>(NULL),
reinterpret_cast<Metrics *>(NULL),
reinterpret_cast<Manager *>(NULL),
kTestDeviceName0,
string(),
kTestDeviceInterfaceIndex0));
EXPECT_CALL(*device_info_, GetDevice(kTestDeviceInterfaceIndex0))
.WillRepeatedly(Return(device));
EXPECT_CALL(*device.get(), DisableReversePathFilter()).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.get(), EnableReversePathFilter());
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,
Technology::kUnknown,
device_info_.get()));
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));
connection = NULL;
}
TEST_F(ConnectionTest, RequestHostRoute) {
ConnectionRefPtr connection = GetNewConnection();
IPAddress address(IPAddress::kFamilyIPv4);
ASSERT_TRUE(address.SetAddressFromString(kIPAddress0));
size_t prefix_len = address.GetLength() * 8;
EXPECT_CALL(routing_table_,
RequestRouteToHost(IsIPAddress(address, prefix_len),
-1,
kTestDeviceInterfaceIndex0,
IsNonNullCallback()))
.WillOnce(Return(true));
EXPECT_TRUE(connection->RequestHostRoute(address));
// The destructor will remove the routes and addresses.
AddDestructorExpectations();
}
TEST_F(ConnectionTest, BlackholeIPv6) {
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_, ConfigureRoutes(_, _, _));
EXPECT_CALL(routing_table_,
CreateBlackholeRoute(kTestDeviceInterfaceIndex0,
IPAddress::kFamilyIPv6,
Connection::kDefaultMetric))
.WillOnce(Return(true));
connection_->UpdateFromIPConfig(ipconfig_);
}
TEST_F(ConnectionTest, PinHostRoute) {
ConnectionRefPtr connection = GetNewConnection();
IPAddress gateway(IPAddress::kFamilyIPv4);
IPAddress trusted_ip(IPAddress::kFamilyIPv4);
// Should fail because neither IP address is set.
EXPECT_FALSE(PinHostRoute(connection, trusted_ip, gateway));
static const char kGateway[] = "10.242.2.13";
ASSERT_TRUE(gateway.SetAddressFromString(kGateway));
// Should fail because trusted IP is not set.
EXPECT_FALSE(PinHostRoute(connection, trusted_ip, gateway));
static const char kTrustedIP[] = "10.0.1.1";
ASSERT_TRUE(trusted_ip.SetAddressFromString(kTrustedIP));
// Should pass without calling RequestRouteToHost since if the gateway
// is not set, there is no work to be done.
EXPECT_CALL(routing_table_, RequestRouteToHost(_, _, _, _)).Times(0);
EXPECT_TRUE(PinHostRoute(connection, trusted_ip,
IPAddress(gateway.family())));
Mock::VerifyAndClearExpectations(&routing_table_);
size_t prefix_len = IPAddress::GetMaxPrefixLength(trusted_ip.family());
EXPECT_CALL(routing_table_, RequestRouteToHost(
IsIPAddress(trusted_ip, prefix_len), -1, kTestDeviceInterfaceIndex0, _))
.WillOnce(Return(false));
EXPECT_FALSE(PinHostRoute(connection, trusted_ip, gateway));
EXPECT_CALL(routing_table_, RequestRouteToHost(
IsIPAddress(trusted_ip, prefix_len), -1, kTestDeviceInterfaceIndex0, _))
.WillOnce(Return(true));
EXPECT_TRUE(PinHostRoute(connection, trusted_ip, gateway));
// The destructor will remove the routes and addresses.
AddDestructorExpectations();
}
TEST_F(ConnectionTest, FixGatewayReachability) {
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 modified the prefix to match the gateway.
static const char kExpandableGateway[] = "10.242.3.14";
ASSERT_TRUE(gateway.SetAddressFromString(kExpandableGateway));
gateway_backup = gateway;
peer = IPAddress(IPAddress::kFamilyIPv4);
EXPECT_TRUE(Connection::FixGatewayReachability(
&local, &peer, &gateway, trusted_ip));
// Prefix should have opened up by 1 bit.
EXPECT_EQ(kPrefix - 1, local.prefix());
// Peer should remain unchanged.
EXPECT_FALSE(peer.IsValid());
// Gateway should remain unchanged.
EXPECT_TRUE(gateway_backup.Equals(gateway));
// Should change models to assuming point-to-point because we cannot
// plausibly expand the prefix past 8.
local.set_prefix(kPrefix);
static const char kUnreachableGateway[] = "11.242.2.14";
ASSERT_TRUE(gateway.SetAddressFromString(kUnreachableGateway));
gateway_backup = gateway;
peer = IPAddress(IPAddress::kFamilyIPv4);
EXPECT_TRUE(Connection::FixGatewayReachability(
&local, &peer, &gateway, trusted_ip));
// Prefix should not have changed.
EXPECT_EQ(kPrefix, local.prefix());
// Peer address should be set to the gateway address.
EXPECT_TRUE(peer.Equals(gateway));
// Gateway should remain unchanged.
EXPECT_TRUE(gateway_backup.Equals(gateway));
// Should also use point-to-point model if the netmask is set to the
// "all-ones" addresss, even if this address could have been made
// accessible by plausibly changing the prefix.
const int kIPv4MaxPrefix =
IPAddress::GetMaxPrefixLength(IPAddress::kFamilyIPv4);
local.set_prefix(kIPv4MaxPrefix);
ASSERT_TRUE(gateway.SetAddressFromString(kExpandableGateway));
gateway_backup = gateway;
peer = IPAddress(IPAddress::kFamilyIPv4);
EXPECT_TRUE(Connection::FixGatewayReachability(
&local, &peer, &gateway, trusted_ip));
// Prefix should not have changed.
EXPECT_EQ(kIPv4MaxPrefix, local.prefix());
// Peer address should be set to the gateway address.
EXPECT_TRUE(peer.Equals(gateway));
// Gateway should remain unchanged.
EXPECT_TRUE(gateway_backup.Equals(gateway));
// If this is a peer-to-peer interface and the peer matches the gateway,
// we should succeed.
local.set_prefix(kPrefix);
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.SetAddressFromString(kUnreachableGateway));
EXPECT_TRUE(Connection::FixGatewayReachability(
&local, &peer, &gateway, trusted_ip));
EXPECT_TRUE(peer.IsDefault());
EXPECT_TRUE(gateway.IsDefault());
}
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(NULL);
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) {
// No connection should be bound initially.
Connection::Binder *binder = &connection_->lower_binder_;
EXPECT_EQ(connection_->interface_name(), binder->name_);
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(NULL);
EXPECT_FALSE(binder->IsBound());
EXPECT_TRUE(connection1->binders_.empty());
ConnectionRefPtr connection2 = GetNewConnection();
// Bind lower |connection1| to upper |connection2| and destroy the upper
// |connection2|. Make sure lower |connection1| is unbound (i.e., the
// disconnect callback is deregistered).
connection2->lower_binder_.Attach(connection1);
EXPECT_FALSE(connection1->binders_.empty());
AddDestructorExpectations();
connection2 = NULL;
EXPECT_TRUE(connection1->binders_.empty());
// Bind lower |connection1| to upper |connection_| and destroy lower
// |connection1|. Make sure lower |connection1| is unbound from upper
// |connection_| and upper |connection_|'s registered disconnect callbacks are
// run.
binder->Attach(connection1);
DisconnectCallbackTarget target;
Connection::Binder test_binder("from_test", target.callback());
test_binder.Attach(connection_);
EXPECT_CALL(target, CallTarget()).Times(1);
ASSERT_FALSE(connection_->binders_.empty());
AddDestructorExpectations();
connection1 = NULL;
EXPECT_FALSE(binder->IsBound());
EXPECT_FALSE(test_binder.IsBound());
EXPECT_TRUE(connection_->binders_.empty());
{
// Binding a connection to itself should be safe.
ConnectionRefPtr connection = GetNewConnection();
connection->lower_binder_.Attach(connection);
EXPECT_FALSE(connection->binders_.empty());
DisconnectCallbackTarget target;
Connection::Binder binder("test", target.callback());
binder.Attach(connection);
AddDestructorExpectations();
EXPECT_CALL(target, CallTarget()).Times(1);
connection = NULL;
}
{
// Circular binding of multiple connections should be safe.
ConnectionRefPtr connection_a = GetNewConnection();
ConnectionRefPtr connection_b = GetNewConnection();
connection_a->lower_binder_.Attach(connection_b);
connection_b->lower_binder_.Attach(connection_a);
EXPECT_FALSE(connection_a->binders_.empty());
EXPECT_FALSE(connection_b->binders_.empty());
DisconnectCallbackTarget target_a;
DisconnectCallbackTarget target_b;
Connection::Binder binder_a("test_a", target_a.callback());
Connection::Binder binder_b("test_b", target_b.callback());
binder_a.Attach(connection_a);
binder_b.Attach(connection_b);
AddDestructorExpectations();
EXPECT_CALL(target_a, CallTarget()).Times(1);
EXPECT_CALL(target_b, CallTarget()).Times(1);
connection_b = NULL;
EXPECT_TRUE(connection_a->binders_.empty());
AddDestructorExpectations();
connection_a = NULL;
}
{
// 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 = NULL;
// Ensure no crash -- the weak pointer to connection should be NULL.
EXPECT_FALSE(binder.connection());
binder.Attach(NULL);
}
}
TEST_F(ConnectionTest, OnRouteQueryResponse) {
Connection::Binder *binder = &connection_->lower_binder_;
ConnectionRefPtr connection = GetNewConnection();
scoped_refptr<MockDevice> device(new StrictMock<MockDevice>(
&control_,
reinterpret_cast<EventDispatcher *>(NULL),
reinterpret_cast<Metrics *>(NULL),
reinterpret_cast<Manager *>(NULL),
kTestDeviceName1,
string(),
kTestDeviceInterfaceIndex1));
// Make sure we unbind the old lower connection even if we can't lookup the
// lower connection device.
binder->Attach(connection);
scoped_refptr<MockDevice> null_device;
EXPECT_CALL(*device_info_, GetDevice(kTestDeviceInterfaceIndex1))
.WillOnce(Return(null_device));
connection_->OnRouteQueryResponse(
kTestDeviceInterfaceIndex1, RoutingTableEntry());
EXPECT_FALSE(binder->IsBound());
// Check for graceful handling of a device with no connection.
EXPECT_CALL(*device_info_, GetDevice(kTestDeviceInterfaceIndex1))
.WillOnce(Return(device));
connection_->OnRouteQueryResponse(
kTestDeviceInterfaceIndex1, RoutingTableEntry());
EXPECT_FALSE(binder->IsBound());
// Create a mock connection that will be used for binding.
scoped_refptr<MockConnection> mock_connection(
new StrictMock<MockConnection>(device_info_.get()));
EXPECT_CALL(*device_info_.get(),
FlushAddresses(mock_connection->interface_index()));
const string kInterfaceName(kTestDeviceName0);
EXPECT_CALL(*mock_connection, interface_name())
.WillRepeatedly(ReturnRef(kInterfaceName));
device->connection_ = mock_connection;
EXPECT_CALL(*device_info_, GetDevice(kTestDeviceInterfaceIndex1))
.WillOnce(Return(device));
// Check that the binding process completes, causing its upper
// connection to create a gateway route.
EXPECT_CALL(*mock_connection, CreateGatewayRoute())
.WillOnce(Return(true));
// Ensure that the Device is notified of the change to the connection.
EXPECT_CALL(*device, OnConnectionUpdated()).Times(1);
connection_->OnRouteQueryResponse(
kTestDeviceInterfaceIndex1, RoutingTableEntry());
// Check that the upper connection is bound to the lower connection.
EXPECT_TRUE(binder->IsBound());
EXPECT_EQ(mock_connection.get(), binder->connection().get());
device->connection_ = NULL;
AddDestructorExpectations();
connection = NULL;
}
TEST_F(ConnectionTest, GetCarrierConnection) {
EXPECT_EQ(connection_.get(), connection_->GetCarrierConnection().get());
ConnectionRefPtr connection1 = GetNewConnection();
ConnectionRefPtr connection2 = GetNewConnection();
ConnectionRefPtr connection3 = GetNewConnection();
connection_->lower_binder_.Attach(connection1);
EXPECT_EQ(connection1.get(), connection_->GetCarrierConnection().get());
connection1->lower_binder_.Attach(connection2);
EXPECT_EQ(connection2.get(), connection_->GetCarrierConnection().get());
connection2->lower_binder_.Attach(connection3);
EXPECT_EQ(connection3.get(), connection_->GetCarrierConnection().get());
// Create a cycle back to |connection1|.
connection3->lower_binder_.Attach(connection1);
EXPECT_EQ(NULL, connection_->GetCarrierConnection().get());
AddDestructorExpectations();
connection3 = NULL;
AddDestructorExpectations();
connection2 = NULL;
AddDestructorExpectations();
connection1 = NULL;
}
} // namespace shill