| /* |
| * 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 |