king
/
v811_spc009
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Branches Tags
${ item.name }
Create tag ${ searchTerm }
Create branch ${ searchTerm }
from 'a2ed455462'
${ noResults }
v811_spc009/system/netd/server/XfrmControllerTest.cpp

801 lines
31 KiB
Raw Blame History

/*
* Copyright 2017 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.
*
* xfrm_ctrl_test.cpp - unit tests for xfrm controllers.
*/
#include <cerrno>
#include <cinttypes>
#include <cstdint>
#include <cstdlib>
#include <set>
#include <vector>
#include <arpa/inet.h>
#include <fcntl.h>
#include <gmock/gmock.h>
#include <ifaddrs.h>
#include <linux/if.h>
#include <linux/if_tun.h>
#include <linux/netfilter/nfnetlink.h>
#include <linux/netlink.h>
#include <netdb.h>
#include <netinet/in.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <android-base/macros.h>
#include <android-base/stringprintf.h>
#include <android-base/strings.h>
#include <android-base/unique_fd.h>
#include <gtest/gtest.h>
#include "Fwmark.h"
#include "NetdConstants.h"
#include "NetlinkCommands.h"
#include "Permission.h"
#include "XfrmController.h"
#include "android/net/INetd.h"
#include "binder/IServiceManager.h"
#include "netdutils/MockSyscalls.h"
#include "netdutils/Netlink.h"
#include "tun_interface.h"
using android::base::unique_fd;
using android::netdutils::Fd;
using android::netdutils::MockSyscalls;
using android::netdutils::Slice;
using android::netdutils::Status;
using ::testing::_;
using ::testing::DoAll;
using ::testing::Invoke;
using ::testing::Return;
using ::testing::SaveArg;
using ::testing::SetArgPointee;
using ::testing::Values;
using ::testing::WithArg;
/**
* This gMock action works like SetArgPointee, but for netdutils::Slice.
* It sets the memory which is pointed to by the N-th argument with the supplied value.
*/
ACTION_TEMPLATE(SetArgSlice, HAS_1_TEMPLATE_PARAMS(int, N), AND_1_VALUE_PARAMS(value)) {
Slice orig = ::testing::get<N>(args);
android::netdutils::copy(orig, value);
}
/**
* This gMock action works like SaveArg, but is specialized for vector<iovec>.
* It copies the memory pointed to by each of the iovecs into a single vector<uint8_t>.
*
* Flattening the iovec objects cannot be done later, since there is no guarantee that the memory
* they point to will still be valid after the mock method returns.
*/
ACTION_TEMPLATE(SaveFlattenedIovecs, HAS_1_TEMPLATE_PARAMS(int, N), AND_1_VALUE_PARAMS(resVec)) {
const std::vector<iovec>& iovs = ::testing::get<N>(args);
for (const iovec& iov : iovs) {
resVec->insert(resVec->end(), reinterpret_cast<uint8_t*>(iov.iov_base),
reinterpret_cast<uint8_t*>(iov.iov_base) + iov.iov_len);
}
}
namespace android {
namespace net {
static constexpr int DROID_SPI = 0xD1201D;
static constexpr size_t KEY_LENGTH = 32;
static constexpr int NLMSG_DEFAULTSIZE = 8192;
static constexpr uint32_t TEST_XFRM_UNDERLYING_NET = 0x512;
static constexpr uint32_t TEST_XFRM_IF_ID = 0x1234;
static constexpr uint32_t TEST_XFRM_MARK = 0x123;
static constexpr uint32_t TEST_XFRM_MASK = 0xFFFFFFFF;
static constexpr char LOCALHOST_V4[] = "127.0.0.1";
static constexpr char LOCALHOST_V6[] = "::1";
static constexpr char TEST_ADDR_V4[] = "8.8.8.8";
static constexpr char TEST_ADDR_V6[] = "2001:4860:4860::8888";
struct Policy {
xfrm_userpolicy_info info;
xfrm_user_tmpl tmpl;
};
struct NetlinkResponse {
nlmsghdr hdr;
char buf[NLMSG_DEFAULTSIZE];
};
void expectAddressEquals(int family, const std::string& expected, const xfrm_address_t& actual) {
char actualStr[INET6_ADDRSTRLEN];
const char* ret =
inet_ntop(family, reinterpret_cast<const void*>(&actual), actualStr, INET6_ADDRSTRLEN);
EXPECT_NE(nullptr, ret) << "Unable to convert xfrm_address_t to string";
EXPECT_EQ(expected, actualStr);
}
class XfrmControllerTest : public ::testing::Test {
public:
testing::StrictMock<netdutils::ScopedMockSyscalls> mockSyscalls;
};
/* Generate function to set value refered to by 3rd argument.
*
* This allows us to mock functions that pass in a pointer, expecting the result to be put into
* that location.
*/
ACTION_P(SetArg3IntValue, value) { *static_cast<int*>(arg3) = value; }
TEST_F(XfrmControllerTest, TestFchown) {
XfrmController ctrl;
unique_fd sockFd(socket(AF_INET, SOCK_DGRAM | SOCK_CLOEXEC, 0));
EXPECT_CALL(mockSyscalls, getsockopt(Fd(sockFd), IPPROTO_UDP, UDP_ENCAP, _, _))
.WillOnce(DoAll(SetArg3IntValue(UDP_ENCAP_ESPINUDP), Return(netdutils::status::ok)));
netdutils::Status res = ctrl.ipSecSetEncapSocketOwner(sockFd, 1001, getuid());
EXPECT_EQ(netdutils::status::ok, res);
struct stat info;
EXPECT_EQ(0, fstat(sockFd.get(), &info));
EXPECT_EQ(1001, (int)info.st_uid);
}
TEST_F(XfrmControllerTest, TestFchownInvalidFd) {
XfrmController ctrl;
netdutils::Status res = ctrl.ipSecSetEncapSocketOwner(unique_fd(), 1001, getuid());
EXPECT_EQ(netdutils::statusFromErrno(errno, "Failed to stat socket file descriptor"), res);
}
TEST_F(XfrmControllerTest, TestFchownIncorrectCallerUid) {
XfrmController ctrl;
unique_fd sockFd(socket(AF_INET, SOCK_DGRAM | SOCK_CLOEXEC, 0));
netdutils::Status res = ctrl.ipSecSetEncapSocketOwner(sockFd, 1001, 1001);
EXPECT_EQ(netdutils::statusFromErrno(EPERM, "fchown disabled for non-owner calls"), res);
}
TEST_F(XfrmControllerTest, TestFchownNonSocketFd) {
XfrmController ctrl;
unique_fd fd(open("/dev/null", O_CLOEXEC));
netdutils::Status res = ctrl.ipSecSetEncapSocketOwner(fd, 1001, getuid());
EXPECT_EQ(netdutils::statusFromErrno(EINVAL, "File descriptor was not a socket"), res);
}
TEST_F(XfrmControllerTest, TestFchownNonUdp) {
XfrmController ctrl;
unique_fd sockFd(socket(AF_INET, SOCK_STREAM | SOCK_CLOEXEC, 0));
EXPECT_CALL(mockSyscalls, getsockopt(Fd(sockFd), IPPROTO_UDP, UDP_ENCAP, _, _))
.WillOnce(DoAll(SetArg3IntValue(0), Return(netdutils::status::ok)));
netdutils::Status res = ctrl.ipSecSetEncapSocketOwner(sockFd, 1001, getuid());
EXPECT_EQ(netdutils::statusFromErrno(EINVAL, "Socket was not a UDP socket"), res);
}
TEST_F(XfrmControllerTest, TestFchownNonUdpEncap) {
XfrmController ctrl;
unique_fd sockFd(socket(AF_INET, SOCK_DGRAM | SOCK_CLOEXEC, 0));
EXPECT_CALL(mockSyscalls, getsockopt(Fd(sockFd), IPPROTO_UDP, UDP_ENCAP, _, _))
.WillOnce(DoAll(SetArg3IntValue(0), Return(netdutils::status::ok)));
netdutils::Status res = ctrl.ipSecSetEncapSocketOwner(sockFd, 1001, getuid());
EXPECT_EQ(netdutils::statusFromErrno(EINVAL, "Socket did not have UDP-encap sockopt set"), res);
}
struct testCaseParams {
const int version;
const bool xfrmInterfacesEnabled;
int getTunnelInterfaceNlAttrsLen() {
if (xfrmInterfacesEnabled) {
return NLMSG_ALIGN(sizeof(XfrmController::nlattr_xfrm_interface_id));
} else {
return NLMSG_ALIGN(sizeof(XfrmController::nlattr_xfrm_mark));
}
}
};
// Test class allowing IPv4/IPv6 parameterized tests.
class XfrmControllerParameterizedTest : public XfrmControllerTest,
public ::testing::WithParamInterface<testCaseParams> {};
// Helper to make generated test names readable.
std::string TestNameGenerator(::testing::TestParamInfo<testCaseParams> info) {
std::string name = "";
switch (info.param.version) {
case 4:
name += "IPv4";
break;
case 5:
name += "IPv64DualStack";
break;
case 6:
name += "IPv6";
break;
default:
name += android::base::StringPrintf("UNKNOWN family type: %d", info.param.version);
break;
}
name += "_";
if (info.param.xfrmInterfacesEnabled) {
name += "XFRMI";
} else {
name += "VTI";
}
return name;
}
// The TEST_P cases below will run with each of the following value parameters.
INSTANTIATE_TEST_CASE_P(ByFamily, XfrmControllerParameterizedTest,
Values(testCaseParams{4, false}, testCaseParams{4, true},
testCaseParams{5, false}, testCaseParams{5, true},
testCaseParams{6, false}, testCaseParams{6, true}),
TestNameGenerator);
TEST_P(XfrmControllerParameterizedTest, TestIpSecAllocateSpi) {
testCaseParams params = GetParam();
const int version = params.version;
const int family = (version == 6) ? AF_INET6 : AF_INET;
const std::string localAddr = (version == 6) ? LOCALHOST_V6 : LOCALHOST_V4;
const std::string remoteAddr = (version == 6) ? TEST_ADDR_V6 : TEST_ADDR_V4;
NetlinkResponse response{};
response.hdr.nlmsg_type = XFRM_MSG_ALLOCSPI;
Slice responseSlice = netdutils::makeSlice(response);
// No IF_ID expected for allocSPI.
size_t expectedMsgLength = NLMSG_HDRLEN + NLMSG_ALIGN(sizeof(xfrm_userspi_info));
// A vector to hold the flattened netlink message for nlMsgSlice
std::vector<uint8_t> nlMsgBuf;
EXPECT_CALL(mockSyscalls, writev(_, _))
.WillOnce(DoAll(SaveFlattenedIovecs<1>(&nlMsgBuf), Return(expectedMsgLength)));
EXPECT_CALL(mockSyscalls, read(_, _))
.WillOnce(DoAll(SetArgSlice<1>(responseSlice), Return(responseSlice)));
XfrmController ctrl(params.xfrmInterfacesEnabled);
int outSpi = 0;
Status res = ctrl.ipSecAllocateSpi(1 /* resourceId */, localAddr,
remoteAddr, DROID_SPI, &outSpi);
EXPECT_TRUE(isOk(res)) << res;
EXPECT_EQ(DROID_SPI, outSpi);
EXPECT_EQ(expectedMsgLength, nlMsgBuf.size());
Slice nlMsgSlice = netdutils::makeSlice(nlMsgBuf);
nlMsgSlice = drop(nlMsgSlice, NLMSG_HDRLEN);
xfrm_userspi_info userspi{};
netdutils::extract(nlMsgSlice, userspi);
EXPECT_EQ(AF_UNSPEC, userspi.info.sel.family);
expectAddressEquals(family, localAddr, userspi.info.saddr);
expectAddressEquals(family, remoteAddr, userspi.info.id.daddr);
EXPECT_EQ(DROID_SPI, static_cast<int>(userspi.min));
EXPECT_EQ(DROID_SPI, static_cast<int>(userspi.max));
}
void verifyXfrmiArguments(uint32_t mark, uint32_t mask, uint32_t ifId) {
// Check that correct arguments (and only those) are non-zero, and correct.
EXPECT_EQ(0U, mark);
EXPECT_EQ(0U, mask);
EXPECT_EQ(TEST_XFRM_IF_ID, ifId);
}
void verifyVtiArguments(uint32_t mark, uint32_t mask, uint32_t ifId) {
// Check that correct arguments (and only those) are non-zero, and correct.
EXPECT_EQ(TEST_XFRM_MARK, mark);
EXPECT_EQ(TEST_XFRM_MASK, mask);
EXPECT_EQ(0U, ifId);
}
void testIpSecAddSecurityAssociation(testCaseParams params, const MockSyscalls& mockSyscalls,
const XfrmMode& mode) {
const int version = params.version;
const int family = (version == 6) ? AF_INET6 : AF_INET;
const std::string localAddr = (version == 6) ? LOCALHOST_V6 : LOCALHOST_V4;
const std::string remoteAddr = (version == 6) ? TEST_ADDR_V6 : TEST_ADDR_V4;
NetlinkResponse response{};
response.hdr.nlmsg_type = XFRM_MSG_ALLOCSPI;
Slice responseSlice = netdutils::makeSlice(response);
std::vector<uint8_t> authKey(KEY_LENGTH, 0);
std::vector<uint8_t> cryptKey(KEY_LENGTH, 1);
// Calculate the length of the expected netlink message.
size_t expectedMsgLength =
NLMSG_HDRLEN + NLMSG_ALIGN(sizeof(xfrm_usersa_info)) +
NLA_ALIGN(offsetof(XfrmController::nlattr_algo_crypt, key) + KEY_LENGTH) +
NLA_ALIGN(offsetof(XfrmController::nlattr_algo_auth, key) + KEY_LENGTH);
uint32_t testIfId = 0;
uint32_t testMark = 0;
uint32_t testMarkMask = 0;
uint32_t testOutputNetid = 0;
if (mode == XfrmMode::TUNNEL) {
expectedMsgLength += params.getTunnelInterfaceNlAttrsLen();
expectedMsgLength += NLA_ALIGN(sizeof(XfrmController::nlattr_xfrm_output_mark));
testIfId = TEST_XFRM_IF_ID;
testMark = TEST_XFRM_MARK;
testMarkMask = TEST_XFRM_MASK;
testOutputNetid = TEST_XFRM_UNDERLYING_NET;
}
std::vector<uint8_t> nlMsgBuf;
EXPECT_CALL(mockSyscalls, writev(_, _))
.WillOnce(DoAll(SaveFlattenedIovecs<1>(&nlMsgBuf), Return(expectedMsgLength)));
EXPECT_CALL(mockSyscalls, read(_, _))
.WillOnce(DoAll(SetArgSlice<1>(responseSlice), Return(responseSlice)));
XfrmController ctrl(params.xfrmInterfacesEnabled);
Status res = ctrl.ipSecAddSecurityAssociation(
1 /* resourceId */, static_cast<int>(mode), localAddr, remoteAddr,
testOutputNetid /* underlying netid */, DROID_SPI, testMark /* mark */,
testMarkMask /* mask */, "hmac(sha256)" /* auth algo */, authKey,
128 /* auth trunc length */, "cbc(aes)" /* encryption algo */, cryptKey,
0 /* crypt trunc length? */, "" /* AEAD algo */, {}, 0,
static_cast<int>(XfrmEncapType::NONE), 0 /* local port */, 0 /* remote port */,
testIfId /* xfrm_if_id */);
EXPECT_TRUE(isOk(res)) << res;
EXPECT_EQ(expectedMsgLength, nlMsgBuf.size());
Slice nlMsgSlice = netdutils::makeSlice(nlMsgBuf);
nlMsgSlice = drop(nlMsgSlice, NLMSG_HDRLEN);
xfrm_usersa_info usersa{};
netdutils::extract(nlMsgSlice, usersa);
EXPECT_EQ(family, usersa.family);
EXPECT_EQ(1 /* Transform Id*/, static_cast<int>(usersa.reqid));
EXPECT_EQ(static_cast<int>(mode), usersa.mode);
if (mode == XfrmMode::TUNNEL) {
EXPECT_EQ(XFRM_STATE_AF_UNSPEC, usersa.flags);
} else {
EXPECT_EQ(0, usersa.flags);
}
EXPECT_EQ(htonl(DROID_SPI), usersa.id.spi);
EXPECT_EQ(IPPROTO_ESP, usersa.id.proto);
expectAddressEquals(family, localAddr, usersa.saddr);
expectAddressEquals(family, remoteAddr, usersa.id.daddr);
// Extract and check the encryption/authentication algorithms.
Slice attr_buf = drop(nlMsgSlice, NLA_ALIGN(sizeof(xfrm_usersa_info)));
// Extract and check the encryption/authentication algorithm
XfrmController::nlattr_algo_crypt encryptAlgo{};
XfrmController::nlattr_algo_auth authAlgo{};
XfrmController::nlattr_xfrm_mark mark{};
XfrmController::nlattr_xfrm_output_mark outputmark{};
XfrmController::nlattr_xfrm_interface_id xfrm_if_id{};
auto attrHandler = [&encryptAlgo, &authAlgo, &mark, &outputmark, &xfrm_if_id](
const nlattr& attr, const Slice& attr_payload) {
Slice buf = attr_payload;
if (attr.nla_type == XFRMA_ALG_CRYPT) {
encryptAlgo.hdr = attr;
netdutils::extract(buf, encryptAlgo.crypt);
buf = drop(buf, sizeof(xfrm_algo));
netdutils::extract(buf, encryptAlgo.key);
} else if (attr.nla_type == XFRMA_ALG_AUTH_TRUNC) {
authAlgo.hdr = attr;
netdutils::extract(buf, authAlgo.auth);
buf = drop(buf, sizeof(xfrm_algo_auth));
netdutils::extract(buf, authAlgo.key);
} else if (attr.nla_type == XFRMA_MARK) {
mark.hdr = attr;
netdutils::extract(buf, mark.mark);
} else if (attr.nla_type == XFRMA_OUTPUT_MARK) {
mark.hdr = attr;
netdutils::extract(buf, outputmark.outputMark);
} else if (attr.nla_type == XFRMA_IF_ID) {
xfrm_if_id.hdr = attr;
netdutils::extract(buf, xfrm_if_id.if_id);
} else {
FAIL() << "Unexpected nlattr type: " << attr.nla_type;
}
};
forEachNetlinkAttribute(attr_buf, attrHandler);
// TODO: Use ContainerEq or ElementsAreArray to get better test failure messages.
EXPECT_EQ(0, memcmp(reinterpret_cast<void*>(cryptKey.data()),
reinterpret_cast<void*>(&encryptAlgo.key), KEY_LENGTH));
EXPECT_EQ(0, memcmp(reinterpret_cast<void*>(authKey.data()),
reinterpret_cast<void*>(&authAlgo.key), KEY_LENGTH));
if (mode == XfrmMode::TUNNEL) {
if (params.xfrmInterfacesEnabled) {
verifyXfrmiArguments(mark.mark.v, mark.mark.m, xfrm_if_id.if_id);
} else {
verifyVtiArguments(mark.mark.v, mark.mark.m, xfrm_if_id.if_id);
}
Fwmark fwmark;
fwmark.intValue = outputmark.outputMark;
EXPECT_EQ(testOutputNetid, fwmark.netId);
EXPECT_EQ(PERMISSION_SYSTEM, fwmark.permission);
EXPECT_TRUE(fwmark.explicitlySelected);
EXPECT_TRUE(fwmark.protectedFromVpn);
} else {
EXPECT_EQ(0U, outputmark.outputMark);
EXPECT_EQ(0U, mark.mark.v);
EXPECT_EQ(0U, mark.mark.m);
EXPECT_EQ(0U, xfrm_if_id.if_id);
}
}
TEST_P(XfrmControllerParameterizedTest, TestTransportModeIpSecAddSecurityAssociation) {
testCaseParams params = GetParam();
testIpSecAddSecurityAssociation(params, mockSyscalls, XfrmMode::TRANSPORT);
}
TEST_P(XfrmControllerParameterizedTest, TestTunnelModeIpSecAddSecurityAssociation) {
testCaseParams params = GetParam();
testIpSecAddSecurityAssociation(params, mockSyscalls, XfrmMode::TUNNEL);
}
TEST_F(XfrmControllerTest, TestIpSecAddSecurityAssociationIPv4Encap) {
// TODO: Implement this test, which is nearly identical to
// TestIpSecAddSecurityAssociation.
}
// Test that input validation rejects IPv6 UDP encap.
TEST_F(XfrmControllerTest, TestIpSecAddSecurityAssociationIPv6Encap) {
EXPECT_CALL(mockSyscalls, writev(_, _)).Times(0);
XfrmController ctrl;
Status res = ctrl.ipSecAddSecurityAssociation(
1, static_cast<int>(XfrmMode::TRANSPORT), LOCALHOST_V6, TEST_ADDR_V6, 0, DROID_SPI, 0,
0, "hmac(sha256)", {}, 128, "cbc(aes)", {}, 0, "", {}, 0,
static_cast<int>(XfrmEncapType::ESPINUDP_NON_IKE), 0, 0, 0);
EXPECT_FALSE(isOk(res)) << "IPv6 UDP encap not rejected";
}
TEST_F(XfrmControllerTest, TestIpSecApplyTransportModeTransformChecksFamily) {
struct sockaddr socketaddr;
socketaddr.sa_family = AF_INET;
unique_fd sock(socket(AF_INET, SOCK_STREAM | SOCK_CLOEXEC, 0));
EXPECT_CALL(mockSyscalls, getsockname(Fd(sock), _, _))
.WillOnce(DoAll(SetArgPointee<1>(socketaddr), Return(netdutils::status::ok)));
XfrmController ctrl;
Status res = ctrl.ipSecApplyTransportModeTransform(sock, 1 /* resourceId */,
static_cast<int>(XfrmDirection::OUT),
LOCALHOST_V6, TEST_ADDR_V6, DROID_SPI);
EXPECT_EQ(res.code(), EINVAL);
}
TEST_P(XfrmControllerParameterizedTest, TestIpSecApplyTransportModeTransform) {
testCaseParams params = GetParam();
const int version = params.version;
const int sockFamily = (version == 4) ? AF_INET : AF_INET6;
const int xfrmFamily = (version == 6) ? AF_INET6: AF_INET;
const std::string localAddr = (version == 6) ? LOCALHOST_V6 : LOCALHOST_V4;
const std::string remoteAddr = (version == 6) ? TEST_ADDR_V6 : TEST_ADDR_V4;
size_t optlen = 0;
Policy policy{};
// Need to cast from void* in order to "SaveArg" policy. Easier to invoke a
// lambda than to write a gMock action.
auto SavePolicy = [&policy](const void* value) {
policy = *reinterpret_cast<const Policy*>(value);
};
struct sockaddr socketaddr;
socketaddr.sa_family = sockFamily;
unique_fd sock(socket(sockFamily, SOCK_STREAM | SOCK_CLOEXEC, 0));
EXPECT_CALL(mockSyscalls, getsockname(_, _, _))
.WillOnce(DoAll(SetArgPointee<1>(socketaddr), Return(netdutils::status::ok)));
EXPECT_CALL(mockSyscalls, setsockopt(_, _, _, _, _))
.WillOnce(DoAll(WithArg<3>(Invoke(SavePolicy)), SaveArg<4>(&optlen),
Return(netdutils::status::ok)));
XfrmController ctrl(params.xfrmInterfacesEnabled);
Status res = ctrl.ipSecApplyTransportModeTransform(sock, 1 /* resourceId */,
static_cast<int>(XfrmDirection::OUT),
localAddr, remoteAddr, DROID_SPI);
EXPECT_TRUE(isOk(res)) << res;
EXPECT_EQ(sizeof(Policy), optlen);
EXPECT_EQ(1 /* resourceId */, static_cast<int>(policy.tmpl.reqid));
EXPECT_EQ(htonl(DROID_SPI), policy.tmpl.id.spi);
expectAddressEquals(xfrmFamily, localAddr, policy.tmpl.saddr);
expectAddressEquals(xfrmFamily, remoteAddr, policy.tmpl.id.daddr);
}
TEST_P(XfrmControllerParameterizedTest, TestIpSecRemoveTransportModeTransform) {
testCaseParams params = GetParam();
const int version = params.version;
const int family = (version == 6) ? AF_INET6 : AF_INET;
const std::string localAddr = (version == 6) ? LOCALHOST_V6 : LOCALHOST_V4;
const std::string remoteAddr = (version == 6) ? TEST_ADDR_V6 : TEST_ADDR_V4;
socklen_t optlen;
const void* optval;
struct sockaddr socketaddr;
socketaddr.sa_family = family;
unique_fd sock(socket(family, SOCK_STREAM | SOCK_CLOEXEC, 0));
EXPECT_CALL(mockSyscalls, getsockname(_, _, _))
.WillOnce(DoAll(SetArgPointee<1>(socketaddr), Return(netdutils::status::ok)));
EXPECT_CALL(mockSyscalls, setsockopt(_, _, _, _, _))
.WillOnce(DoAll(SaveArg<3>(&optval), SaveArg<4>(&optlen),
Return(netdutils::status::ok)));
XfrmController ctrl(params.xfrmInterfacesEnabled);
Status res = ctrl.ipSecRemoveTransportModeTransform(sock);
EXPECT_TRUE(isOk(res)) << res;
EXPECT_EQ(nullptr, optval);
EXPECT_EQ(static_cast<socklen_t>(0), optlen);
}
void parseTunnelNetlinkAttrs(XfrmController::nlattr_xfrm_mark* mark,
XfrmController::nlattr_xfrm_interface_id* xfrm_if_id, Slice attr_buf) {
auto attrHandler = [mark, xfrm_if_id](const nlattr& attr, const Slice& attr_payload) {
Slice buf = attr_payload;
if (attr.nla_type == XFRMA_MARK) {
mark->hdr = attr;
netdutils::extract(buf, mark->mark);
} else if (attr.nla_type == XFRMA_IF_ID) {
xfrm_if_id->hdr = attr;
netdutils::extract(buf, xfrm_if_id->if_id);
} else {
FAIL() << "Unexpected nlattr type: " << attr.nla_type;
}
};
forEachNetlinkAttribute(attr_buf, attrHandler);
}
TEST_P(XfrmControllerParameterizedTest, TestIpSecDeleteSecurityAssociation) {
testCaseParams params = GetParam();
const int version = params.version;
const int family = (version == 6) ? AF_INET6 : AF_INET;
const std::string localAddr = (version == 6) ? LOCALHOST_V6 : LOCALHOST_V4;
const std::string remoteAddr = (version == 6) ? TEST_ADDR_V6 : TEST_ADDR_V4;
NetlinkResponse response{};
response.hdr.nlmsg_type = XFRM_MSG_DELSA;
Slice responseSlice = netdutils::makeSlice(response);
size_t expectedMsgLength = NLMSG_HDRLEN + NLMSG_ALIGN(sizeof(xfrm_usersa_id)) +
params.getTunnelInterfaceNlAttrsLen();
std::vector<uint8_t> nlMsgBuf;
EXPECT_CALL(mockSyscalls, writev(_, _))
.WillOnce(DoAll(SaveFlattenedIovecs<1>(&nlMsgBuf), Return(expectedMsgLength)));
EXPECT_CALL(mockSyscalls, read(_, _))
.WillOnce(DoAll(SetArgSlice<1>(responseSlice), Return(responseSlice)));
XfrmController ctrl(params.xfrmInterfacesEnabled);
Status res = ctrl.ipSecDeleteSecurityAssociation(1 /* resourceId */, localAddr, remoteAddr,
DROID_SPI, TEST_XFRM_MARK, TEST_XFRM_MASK,
TEST_XFRM_IF_ID);
EXPECT_TRUE(isOk(res)) << res;
EXPECT_EQ(expectedMsgLength, nlMsgBuf.size());
Slice nlMsgSlice = netdutils::makeSlice(nlMsgBuf);
nlMsgSlice = netdutils::drop(nlMsgSlice, NLMSG_HDRLEN);
// Extract and check the usersa_id
xfrm_usersa_id said{};
netdutils::extract(nlMsgSlice, said);
nlMsgSlice = drop(nlMsgSlice, sizeof(xfrm_usersa_id));
EXPECT_EQ(htonl(DROID_SPI), said.spi);
expectAddressEquals(family, remoteAddr, said.daddr);
// Extract and check the marks and xfrm_if_id
XfrmController::nlattr_xfrm_mark mark{};
XfrmController::nlattr_xfrm_interface_id xfrm_if_id{};
parseTunnelNetlinkAttrs(&mark, &xfrm_if_id, nlMsgSlice);
if (params.xfrmInterfacesEnabled) {
verifyXfrmiArguments(mark.mark.v, mark.mark.m, xfrm_if_id.if_id);
} else {
verifyVtiArguments(mark.mark.v, mark.mark.m, xfrm_if_id.if_id);
}
}
TEST_P(XfrmControllerParameterizedTest, TestIpSecAddSecurityPolicy) {
testCaseParams params = GetParam();
const int version = params.version;
const int family = (version == 6) ? AF_INET6 : AF_INET;
const std::string localAddr = (version == 6) ? LOCALHOST_V6 : LOCALHOST_V4;
const std::string remoteAddr = (version == 6) ? TEST_ADDR_V6 : TEST_ADDR_V4;
NetlinkResponse response{};
response.hdr.nlmsg_type = XFRM_MSG_NEWPOLICY;
Slice responseSlice = netdutils::makeSlice(response);
size_t expectedMsgLength = NLMSG_HDRLEN + NLMSG_ALIGN(sizeof(xfrm_userpolicy_info)) +
NLMSG_ALIGN(sizeof(XfrmController::nlattr_user_tmpl)) +
params.getTunnelInterfaceNlAttrsLen();
std::vector<uint8_t> nlMsgBuf;
EXPECT_CALL(mockSyscalls, writev(_, _))
.WillOnce(DoAll(SaveFlattenedIovecs<1>(&nlMsgBuf), Return(expectedMsgLength)));
EXPECT_CALL(mockSyscalls, read(_, _))
.WillOnce(DoAll(SetArgSlice<1>(responseSlice), Return(responseSlice)));
XfrmController ctrl(params.xfrmInterfacesEnabled);
Status res = ctrl.ipSecAddSecurityPolicy(
1 /* resourceId */, family, static_cast<int>(XfrmDirection::OUT), localAddr, remoteAddr,
0 /* SPI */, TEST_XFRM_MARK, TEST_XFRM_MASK, TEST_XFRM_IF_ID);
EXPECT_TRUE(isOk(res)) << res;
EXPECT_EQ(expectedMsgLength, nlMsgBuf.size());
Slice nlMsgSlice = netdutils::makeSlice(nlMsgBuf);
nlmsghdr hdr;
netdutils::extract(nlMsgSlice, hdr);
EXPECT_EQ(XFRM_MSG_NEWPOLICY, hdr.nlmsg_type);
nlMsgSlice = netdutils::drop(nlMsgSlice, NLMSG_HDRLEN);
xfrm_userpolicy_info userpolicy{};
netdutils::extract(nlMsgSlice, userpolicy);
EXPECT_EQ(static_cast<uint8_t>(XfrmDirection::OUT), userpolicy.dir);
// Drop the user policy info.
Slice attr_buf = drop(nlMsgSlice, NLA_ALIGN(sizeof(xfrm_userpolicy_info)));
// Extract and check the user tmpl and mark.
XfrmController::nlattr_user_tmpl usertmpl{};
XfrmController::nlattr_xfrm_mark mark{};
XfrmController::nlattr_xfrm_interface_id xfrm_if_id{};
auto attrHandler = [&usertmpl, &mark, &xfrm_if_id](const nlattr& attr,
const Slice& attr_payload) {
Slice buf = attr_payload;
if (attr.nla_type == XFRMA_TMPL) {
usertmpl.hdr = attr;
netdutils::extract(buf, usertmpl.tmpl);
} else if (attr.nla_type == XFRMA_MARK) {
mark.hdr = attr;
netdutils::extract(buf, mark.mark);
} else if (attr.nla_type == XFRMA_IF_ID) {
mark.hdr = attr;
netdutils::extract(buf, xfrm_if_id.if_id);
} else {
FAIL() << "Unexpected nlattr type: " << attr.nla_type;
}
};
forEachNetlinkAttribute(attr_buf, attrHandler);
expectAddressEquals(family, remoteAddr, usertmpl.tmpl.id.daddr);
if (params.xfrmInterfacesEnabled) {
verifyXfrmiArguments(mark.mark.v, mark.mark.m, xfrm_if_id.if_id);
} else {
verifyVtiArguments(mark.mark.v, mark.mark.m, xfrm_if_id.if_id);
}
}
TEST_P(XfrmControllerParameterizedTest, TestIpSecUpdateSecurityPolicy) {
testCaseParams params = GetParam();
const int version = params.version;
const int family = (version == 6) ? AF_INET6 : AF_INET;
const std::string localAddr = (version == 6) ? LOCALHOST_V6 : LOCALHOST_V4;
const std::string remoteAddr = (version == 6) ? TEST_ADDR_V6 : TEST_ADDR_V4;
NetlinkResponse response{};
response.hdr.nlmsg_type = XFRM_MSG_UPDPOLICY;
Slice responseSlice = netdutils::makeSlice(response);
size_t expectedMsgLength = NLMSG_HDRLEN + NLMSG_ALIGN(sizeof(xfrm_userpolicy_info)) +
NLMSG_ALIGN(sizeof(XfrmController::nlattr_user_tmpl)) +
params.getTunnelInterfaceNlAttrsLen();
std::vector<uint8_t> nlMsgBuf;
EXPECT_CALL(mockSyscalls, writev(_, _))
.WillOnce(DoAll(SaveFlattenedIovecs<1>(&nlMsgBuf), Return(expectedMsgLength)));
EXPECT_CALL(mockSyscalls, read(_, _))
.WillOnce(DoAll(SetArgSlice<1>(responseSlice), Return(responseSlice)));
XfrmController ctrl(params.xfrmInterfacesEnabled);
Status res = ctrl.ipSecUpdateSecurityPolicy(
1 /* resourceId */, family, static_cast<int>(XfrmDirection::OUT), localAddr, remoteAddr,
0 /* SPI */, TEST_XFRM_MARK /* Mark */, TEST_XFRM_MARK /* Mask */,
TEST_XFRM_IF_ID /* xfrm_if_id */);
EXPECT_TRUE(isOk(res)) << res;
EXPECT_EQ(expectedMsgLength, nlMsgBuf.size());
Slice nlMsgSlice = netdutils::makeSlice(nlMsgBuf);
nlmsghdr hdr;
netdutils::extract(nlMsgSlice, hdr);
EXPECT_EQ(XFRM_MSG_UPDPOLICY, hdr.nlmsg_type);
}
TEST_P(XfrmControllerParameterizedTest, TestIpSecDeleteSecurityPolicy) {
testCaseParams params = GetParam();
const int version = params.version;
const int family = (version == 6) ? AF_INET6 : AF_INET;
const std::string localAddr = (version == 6) ? LOCALHOST_V6 : LOCALHOST_V4;
const std::string remoteAddr = (version == 6) ? TEST_ADDR_V6 : TEST_ADDR_V4;
NetlinkResponse response{};
response.hdr.nlmsg_type = XFRM_MSG_DELPOLICY;
Slice responseSlice = netdutils::makeSlice(response);
size_t expectedMsgLength = NLMSG_HDRLEN + NLMSG_ALIGN(sizeof(xfrm_userpolicy_id)) +
params.getTunnelInterfaceNlAttrsLen();
std::vector<uint8_t> nlMsgBuf;
EXPECT_CALL(mockSyscalls, writev(_, _))
.WillOnce(DoAll(SaveFlattenedIovecs<1>(&nlMsgBuf), Return(expectedMsgLength)));
EXPECT_CALL(mockSyscalls, read(_, _))
.WillOnce(DoAll(SetArgSlice<1>(responseSlice), Return(responseSlice)));
XfrmController ctrl(params.xfrmInterfacesEnabled);
Status res = ctrl.ipSecDeleteSecurityPolicy(1 /* resourceId */, family,
static_cast<int>(XfrmDirection::OUT),
TEST_XFRM_MARK, TEST_XFRM_MASK, TEST_XFRM_IF_ID);
EXPECT_TRUE(isOk(res)) << res;
EXPECT_EQ(expectedMsgLength, nlMsgBuf.size());
Slice nlMsgSlice = netdutils::makeSlice(nlMsgBuf);
nlMsgSlice = netdutils::drop(nlMsgSlice, NLMSG_HDRLEN);
xfrm_userpolicy_id policyid{};
netdutils::extract(nlMsgSlice, policyid);
EXPECT_EQ(static_cast<uint8_t>(XfrmDirection::OUT), policyid.dir);
// Drop the user policy id.
nlMsgSlice = drop(nlMsgSlice, NLA_ALIGN(sizeof(xfrm_userpolicy_id)));
// Extract and check the marks and xfrm_if_id
XfrmController::nlattr_xfrm_mark mark{};
XfrmController::nlattr_xfrm_interface_id xfrm_if_id{};
parseTunnelNetlinkAttrs(&mark, &xfrm_if_id, nlMsgSlice);
if (params.xfrmInterfacesEnabled) {
verifyXfrmiArguments(mark.mark.v, mark.mark.m, xfrm_if_id.if_id);
} else {
verifyVtiArguments(mark.mark.v, mark.mark.m, xfrm_if_id.if_id);
}
}
// TODO: Add tests for VTIs, ensuring that we are sending the correct data over netlink.
} // namespace net
} // namespace android
Reference in new issue View Git Blame Copy Permalink