| /* |
| * Copyright 2022 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. |
| * |
| * TcUtilsTest.cpp - unit tests for TcUtils.cpp |
| */ |
| |
| #include <gtest/gtest.h> |
| |
| #include "kernelversion.h" |
| #include <tcutils/tcutils.h> |
| |
| #include <BpfSyscallWrappers.h> |
| #include <errno.h> |
| #include <linux/if_ether.h> |
| |
| namespace android { |
| |
| TEST(LibTcUtilsTest, IsEthernetOfNonExistingIf) { |
| bool result = false; |
| int error = isEthernet("not_existing_if", result); |
| ASSERT_FALSE(result); |
| ASSERT_EQ(-ENODEV, error); |
| } |
| |
| TEST(LibTcUtilsTest, IsEthernetOfLoopback) { |
| bool result = false; |
| int error = isEthernet("lo", result); |
| ASSERT_FALSE(result); |
| ASSERT_EQ(-EAFNOSUPPORT, error); |
| } |
| |
| // If wireless 'wlan0' interface exists it should be Ethernet. |
| // See also HardwareAddressTypeOfWireless. |
| TEST(LibTcUtilsTest, IsEthernetOfWireless) { |
| bool result = false; |
| int error = isEthernet("wlan0", result); |
| if (!result && error == -ENODEV) |
| return; |
| |
| ASSERT_EQ(0, error); |
| ASSERT_TRUE(result); |
| } |
| |
| // If cellular 'rmnet_data0' interface exists it should |
| // *probably* not be Ethernet and instead be RawIp. |
| // See also HardwareAddressTypeOfCellular. |
| TEST(LibTcUtilsTest, IsEthernetOfCellular) { |
| bool result = false; |
| int error = isEthernet("rmnet_data0", result); |
| if (!result && error == -ENODEV) |
| return; |
| |
| ASSERT_EQ(0, error); |
| ASSERT_FALSE(result); |
| } |
| |
| // See Linux kernel source in include/net/flow.h |
| static constexpr int LOOPBACK_IFINDEX = 1; |
| |
| TEST(LibTcUtilsTest, AttachReplaceDetachClsactLo) { |
| // This attaches and detaches a configuration-less and thus no-op clsact |
| // qdisc to loopback interface (and it takes fractions of a second) |
| EXPECT_EQ(0, tcAddQdiscClsact(LOOPBACK_IFINDEX)); |
| EXPECT_EQ(0, tcReplaceQdiscClsact(LOOPBACK_IFINDEX)); |
| EXPECT_EQ(0, tcDeleteQdiscClsact(LOOPBACK_IFINDEX)); |
| EXPECT_EQ(-EINVAL, tcDeleteQdiscClsact(LOOPBACK_IFINDEX)); |
| } |
| |
| TEST(LibTcUtilsTest, AddAndDeleteBpfFilter) { |
| // TODO: this should use bpf_shared.h rather than hardcoding the path |
| static constexpr char bpfProgPath[] = |
| "/sys/fs/bpf/tethering/prog_offload_schedcls_tether_downstream6_ether"; |
| const int errNOENT = isAtLeastKernelVersion(4, 19, 0) ? ENOENT : EINVAL; |
| |
| // static test values |
| static constexpr bool ingress = true; |
| static constexpr uint16_t prio = 17; |
| static constexpr uint16_t proto = ETH_P_ALL; |
| |
| // try to delete missing filter from missing qdisc |
| EXPECT_EQ(-EINVAL, tcDeleteFilter(LOOPBACK_IFINDEX, ingress, prio, proto)); |
| // try to attach bpf filter to missing qdisc |
| EXPECT_EQ(-EINVAL, tcAddBpfFilter(LOOPBACK_IFINDEX, ingress, prio, proto, |
| bpfProgPath)); |
| // add the clsact qdisc |
| EXPECT_EQ(0, tcAddQdiscClsact(LOOPBACK_IFINDEX)); |
| // try to delete missing filter when there is a qdisc attached |
| EXPECT_EQ(-errNOENT, tcDeleteFilter(LOOPBACK_IFINDEX, ingress, prio, proto)); |
| // add and delete a bpf filter |
| EXPECT_EQ( |
| 0, tcAddBpfFilter(LOOPBACK_IFINDEX, ingress, prio, proto, bpfProgPath)); |
| EXPECT_EQ(0, tcDeleteFilter(LOOPBACK_IFINDEX, ingress, prio, proto)); |
| // try to remove the same filter a second time |
| EXPECT_EQ(-errNOENT, tcDeleteFilter(LOOPBACK_IFINDEX, ingress, prio, proto)); |
| // remove the clsact qdisc |
| EXPECT_EQ(0, tcDeleteQdiscClsact(LOOPBACK_IFINDEX)); |
| // once again, try to delete missing filter from missing qdisc |
| EXPECT_EQ(-EINVAL, tcDeleteFilter(LOOPBACK_IFINDEX, ingress, prio, proto)); |
| } |
| |
| TEST(LibTcUtilsTest, AddAndDeleteIngressPoliceFilter) { |
| // TODO: this should use bpf_shared.h rather than hardcoding the path |
| static constexpr char bpfProgPath[] = |
| "/sys/fs/bpf/prog_netd_schedact_ingress_account"; |
| int fd = bpf::retrieveProgram(bpfProgPath); |
| if (fd == -1) { |
| // ingress policing is not supported. |
| return; |
| } |
| close(fd); |
| |
| const int errNOENT = isAtLeastKernelVersion(4, 19, 0) ? ENOENT : EINVAL; |
| |
| // static test values |
| static constexpr unsigned rateInBytesPerSec = |
| 1024 * 1024; // 8mbit/s => 1mbyte/s => 1024*1024 bytes/s. |
| static constexpr uint16_t prio = 17; |
| static constexpr uint16_t proto = ETH_P_ALL; |
| |
| // try to delete missing filter from missing qdisc |
| EXPECT_EQ(-EINVAL, |
| tcDeleteFilter(LOOPBACK_IFINDEX, true /*ingress*/, prio, proto)); |
| // try to attach bpf filter to missing qdisc |
| EXPECT_EQ(-EINVAL, tcAddIngressPoliceFilter(LOOPBACK_IFINDEX, prio, proto, |
| rateInBytesPerSec, bpfProgPath)); |
| // add the clsact qdisc |
| EXPECT_EQ(0, tcAddQdiscClsact(LOOPBACK_IFINDEX)); |
| // try to delete missing filter when there is a qdisc attached |
| EXPECT_EQ(-errNOENT, |
| tcDeleteFilter(LOOPBACK_IFINDEX, true /*ingress*/, prio, proto)); |
| // add and delete a bpf filter |
| EXPECT_EQ(0, tcAddIngressPoliceFilter(LOOPBACK_IFINDEX, prio, proto, |
| rateInBytesPerSec, bpfProgPath)); |
| EXPECT_EQ(0, tcDeleteFilter(LOOPBACK_IFINDEX, true /*ingress*/, prio, proto)); |
| // try to remove the same filter a second time |
| EXPECT_EQ(-errNOENT, |
| tcDeleteFilter(LOOPBACK_IFINDEX, true /*ingress*/, prio, proto)); |
| // remove the clsact qdisc |
| EXPECT_EQ(0, tcDeleteQdiscClsact(LOOPBACK_IFINDEX)); |
| // once again, try to delete missing filter from missing qdisc |
| EXPECT_EQ(-EINVAL, |
| tcDeleteFilter(LOOPBACK_IFINDEX, true /*ingress*/, prio, proto)); |
| } |
| |
| } // namespace android |