248 lines
10 KiB
C++
248 lines
10 KiB
C++
//
|
|
// Copyright (C) 2013 The Android Open Source Project
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
//
|
|
|
|
#include "shill/connection_info_reader.h"
|
|
|
|
#include <netinet/in.h>
|
|
|
|
#include <base/files/file_util.h>
|
|
#include <base/files/scoped_temp_dir.h>
|
|
#include <base/strings/stringprintf.h>
|
|
#include <gmock/gmock.h>
|
|
#include <gtest/gtest.h>
|
|
|
|
using base::FilePath;
|
|
using base::ScopedTempDir;
|
|
using base::StringPrintf;
|
|
using std::string;
|
|
using std::vector;
|
|
using testing::Return;
|
|
|
|
namespace shill {
|
|
|
|
namespace {
|
|
|
|
// TODO(benchan): Test IPv6 addresses.
|
|
|
|
const char* kConnectionInfoLines[] = {
|
|
"udp 17 30 src=192.168.1.1 dst=192.168.1.2 sport=9000 dport=53 "
|
|
"[UNREPLIED] src=192.168.1.2 dst=192.168.1.1 sport=53 dport=9000 use=2",
|
|
"tcp 6 299 ESTABLISHED src=192.168.2.1 dst=192.168.2.3 sport=8000 "
|
|
"dport=7000 src=192.168.2.3 dst=192.168.2.1 sport=7000 dport=8000 [ASSURED] "
|
|
"use=2",
|
|
};
|
|
|
|
} // namespace
|
|
|
|
class ConnectionInfoReaderUnderTest : public ConnectionInfoReader {
|
|
public:
|
|
// Mock out GetConnectionInfoFilePath to use a temporary created connection
|
|
// info file instead of the actual path in procfs (i.e.
|
|
// /proc/net/ip_conntrack).
|
|
MOCK_CONST_METHOD0(GetConnectionInfoFilePath, FilePath());
|
|
};
|
|
|
|
class ConnectionInfoReaderTest : public testing::Test {
|
|
protected:
|
|
IPAddress StringToIPv4Address(const string& address_string) {
|
|
IPAddress ip_address(IPAddress::kFamilyIPv4);
|
|
EXPECT_TRUE(ip_address.SetAddressFromString(address_string));
|
|
return ip_address;
|
|
}
|
|
|
|
IPAddress StringToIPv6Address(const string& address_string) {
|
|
IPAddress ip_address(IPAddress::kFamilyIPv6);
|
|
EXPECT_TRUE(ip_address.SetAddressFromString(address_string));
|
|
return ip_address;
|
|
}
|
|
|
|
void CreateConnectionInfoFile(const char** lines, size_t num_lines,
|
|
const FilePath& dir_path, FilePath* file_path) {
|
|
ASSERT_TRUE(base::CreateTemporaryFileInDir(dir_path, file_path));
|
|
for (size_t i = 0; i < num_lines; ++i) {
|
|
string line = lines[i];
|
|
line += '\n';
|
|
ASSERT_TRUE(base::AppendToFile(*file_path, line.data(), line.size()));
|
|
}
|
|
}
|
|
|
|
void ExpectConnectionInfoEqual(const ConnectionInfo& info1,
|
|
const ConnectionInfo& info2) {
|
|
EXPECT_EQ(info1.protocol(), info2.protocol());
|
|
EXPECT_EQ(info1.time_to_expire_seconds(), info2.time_to_expire_seconds());
|
|
EXPECT_EQ(info1.is_unreplied(), info2.is_unreplied());
|
|
EXPECT_TRUE(info1.original_source_ip_address()
|
|
.Equals(info2.original_source_ip_address()));
|
|
EXPECT_EQ(info1.original_source_port(), info2.original_source_port());
|
|
EXPECT_TRUE(info1.original_destination_ip_address()
|
|
.Equals(info2.original_destination_ip_address()));
|
|
EXPECT_EQ(info1.original_destination_port(),
|
|
info2.original_destination_port());
|
|
EXPECT_TRUE(info1.reply_source_ip_address()
|
|
.Equals(info2.reply_source_ip_address()));
|
|
EXPECT_EQ(info1.reply_source_port(), info2.reply_source_port());
|
|
EXPECT_TRUE(info1.reply_destination_ip_address()
|
|
.Equals(info2.reply_destination_ip_address()));
|
|
EXPECT_EQ(info1.reply_destination_port(), info2.reply_destination_port());
|
|
}
|
|
|
|
ConnectionInfoReaderUnderTest reader_;
|
|
};
|
|
|
|
TEST_F(ConnectionInfoReaderTest, LoadConnectionInfo) {
|
|
vector<ConnectionInfo> info_list;
|
|
ScopedTempDir temp_dir;
|
|
ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
|
|
|
|
// Loading a non-existent file should fail.
|
|
FilePath info_file("/non-existent-file");
|
|
EXPECT_CALL(reader_, GetConnectionInfoFilePath()).WillOnce(Return(info_file));
|
|
EXPECT_FALSE(reader_.LoadConnectionInfo(&info_list));
|
|
|
|
// Loading an empty file should succeed.
|
|
CreateConnectionInfoFile(kConnectionInfoLines, 0, temp_dir.path(),
|
|
&info_file);
|
|
EXPECT_CALL(reader_, GetConnectionInfoFilePath()).WillOnce(Return(info_file));
|
|
EXPECT_TRUE(reader_.LoadConnectionInfo(&info_list));
|
|
EXPECT_TRUE(info_list.empty());
|
|
|
|
// Loading a non-empty file should succeed.
|
|
CreateConnectionInfoFile(kConnectionInfoLines,
|
|
arraysize(kConnectionInfoLines),
|
|
temp_dir.path(),
|
|
&info_file);
|
|
EXPECT_CALL(reader_, GetConnectionInfoFilePath()).WillOnce(Return(info_file));
|
|
EXPECT_TRUE(reader_.LoadConnectionInfo(&info_list));
|
|
EXPECT_EQ(arraysize(kConnectionInfoLines), info_list.size());
|
|
|
|
ExpectConnectionInfoEqual(ConnectionInfo(IPPROTO_UDP,
|
|
30,
|
|
true,
|
|
StringToIPv4Address("192.168.1.1"),
|
|
9000,
|
|
StringToIPv4Address("192.168.1.2"),
|
|
53,
|
|
StringToIPv4Address("192.168.1.2"),
|
|
53,
|
|
StringToIPv4Address("192.168.1.1"),
|
|
9000),
|
|
info_list[0]);
|
|
ExpectConnectionInfoEqual(ConnectionInfo(IPPROTO_TCP,
|
|
299,
|
|
false,
|
|
StringToIPv4Address("192.168.2.1"),
|
|
8000,
|
|
StringToIPv4Address("192.168.2.3"),
|
|
7000,
|
|
StringToIPv4Address("192.168.2.3"),
|
|
7000,
|
|
StringToIPv4Address("192.168.2.1"),
|
|
8000),
|
|
info_list[1]);
|
|
}
|
|
|
|
TEST_F(ConnectionInfoReaderTest, ParseConnectionInfo) {
|
|
ConnectionInfo info;
|
|
|
|
EXPECT_FALSE(reader_.ParseConnectionInfo("", &info));
|
|
|
|
EXPECT_TRUE(reader_.ParseConnectionInfo(kConnectionInfoLines[0], &info));
|
|
ExpectConnectionInfoEqual(ConnectionInfo(IPPROTO_UDP,
|
|
30,
|
|
true,
|
|
StringToIPv4Address("192.168.1.1"),
|
|
9000,
|
|
StringToIPv4Address("192.168.1.2"),
|
|
53,
|
|
StringToIPv4Address("192.168.1.2"),
|
|
53,
|
|
StringToIPv4Address("192.168.1.1"),
|
|
9000),
|
|
info);
|
|
}
|
|
|
|
TEST_F(ConnectionInfoReaderTest, ParseProtocol) {
|
|
int protocol = 0;
|
|
|
|
EXPECT_FALSE(reader_.ParseProtocol("", &protocol));
|
|
EXPECT_FALSE(reader_.ParseProtocol("a", &protocol));
|
|
EXPECT_FALSE(reader_.ParseProtocol("-1", &protocol));
|
|
EXPECT_FALSE(reader_.ParseProtocol(StringPrintf("%d", IPPROTO_MAX),
|
|
&protocol));
|
|
|
|
for (int i = 0; i < IPPROTO_MAX; ++i) {
|
|
EXPECT_TRUE(reader_.ParseProtocol(StringPrintf("%d", i), &protocol));
|
|
EXPECT_EQ(i, protocol);
|
|
}
|
|
}
|
|
|
|
TEST_F(ConnectionInfoReaderTest, ParseTimeToExpireSeconds) {
|
|
int64_t time_to_expire = 0;
|
|
|
|
EXPECT_FALSE(reader_.ParseTimeToExpireSeconds("", &time_to_expire));
|
|
EXPECT_FALSE(reader_.ParseTimeToExpireSeconds("a", &time_to_expire));
|
|
EXPECT_FALSE(reader_.ParseTimeToExpireSeconds("-1", &time_to_expire));
|
|
|
|
EXPECT_TRUE(reader_.ParseTimeToExpireSeconds("100", &time_to_expire));
|
|
EXPECT_EQ(100, time_to_expire);
|
|
}
|
|
|
|
TEST_F(ConnectionInfoReaderTest, ParseIPAddress) {
|
|
IPAddress ip_address(IPAddress::kFamilyUnknown);
|
|
bool is_source = false;
|
|
|
|
EXPECT_FALSE(reader_.ParseIPAddress("", &ip_address, &is_source));
|
|
EXPECT_FALSE(reader_.ParseIPAddress("abc", &ip_address, &is_source));
|
|
EXPECT_FALSE(reader_.ParseIPAddress("src=", &ip_address, &is_source));
|
|
EXPECT_FALSE(reader_.ParseIPAddress("src=abc", &ip_address, &is_source));
|
|
EXPECT_FALSE(reader_.ParseIPAddress("dst=", &ip_address, &is_source));
|
|
EXPECT_FALSE(reader_.ParseIPAddress("dst=abc", &ip_address, &is_source));
|
|
|
|
EXPECT_TRUE(reader_.ParseIPAddress("src=192.168.1.1",
|
|
&ip_address, &is_source));
|
|
EXPECT_TRUE(ip_address.Equals(StringToIPv4Address("192.168.1.1")));
|
|
EXPECT_TRUE(is_source);
|
|
EXPECT_TRUE(reader_.ParseIPAddress("dst=192.168.1.2",
|
|
&ip_address, &is_source));
|
|
EXPECT_TRUE(ip_address.Equals(StringToIPv4Address("192.168.1.2")));
|
|
EXPECT_FALSE(is_source);
|
|
}
|
|
|
|
TEST_F(ConnectionInfoReaderTest, ParsePort) {
|
|
uint16_t port = 0;
|
|
bool is_source = false;
|
|
|
|
EXPECT_FALSE(reader_.ParsePort("", &port, &is_source));
|
|
EXPECT_FALSE(reader_.ParsePort("a", &port, &is_source));
|
|
EXPECT_FALSE(reader_.ParsePort("0", &port, &is_source));
|
|
EXPECT_FALSE(reader_.ParsePort("sport=", &port, &is_source));
|
|
EXPECT_FALSE(reader_.ParsePort("sport=a", &port, &is_source));
|
|
EXPECT_FALSE(reader_.ParsePort("sport=-1", &port, &is_source));
|
|
EXPECT_FALSE(reader_.ParsePort("sport=65536", &port, &is_source));
|
|
EXPECT_FALSE(reader_.ParsePort("dport=", &port, &is_source));
|
|
EXPECT_FALSE(reader_.ParsePort("dport=a", &port, &is_source));
|
|
EXPECT_FALSE(reader_.ParsePort("dport=-1", &port, &is_source));
|
|
EXPECT_FALSE(reader_.ParsePort("dport=65536", &port, &is_source));
|
|
|
|
EXPECT_TRUE(reader_.ParsePort("sport=53", &port, &is_source));
|
|
EXPECT_EQ(53, port);
|
|
EXPECT_TRUE(is_source);
|
|
EXPECT_TRUE(reader_.ParsePort("dport=80", &port, &is_source));
|
|
EXPECT_EQ(80, port);
|
|
EXPECT_FALSE(is_source);
|
|
}
|
|
|
|
} // namespace shill
|