William A. Kennington III | 6a92363 | 2022-11-06 18:17:33 -0800 | [diff] [blame] | 1 | #include "rtnetlink.hpp" |
Nagaraju Goruganti | 067ca2d | 2018-01-17 01:12:00 -0600 | [diff] [blame] | 2 | |
William A. Kennington III | 6a92363 | 2022-11-06 18:17:33 -0800 | [diff] [blame] | 3 | #include <linux/netlink.h> |
Gunnar Mills | 57d9c50 | 2018-09-14 14:42:34 -0500 | [diff] [blame] | 4 | #include <linux/rtnetlink.h> |
Nagaraju Goruganti | 067ca2d | 2018-01-17 01:12:00 -0600 | [diff] [blame] | 5 | |
William A. Kennington III | 6a92363 | 2022-11-06 18:17:33 -0800 | [diff] [blame] | 6 | #include <stdplus/raw.hpp> |
Nagaraju Goruganti | 067ca2d | 2018-01-17 01:12:00 -0600 | [diff] [blame] | 7 | |
Gunnar Mills | 57d9c50 | 2018-09-14 14:42:34 -0500 | [diff] [blame] | 8 | #include <gtest/gtest.h> |
Nagaraju Goruganti | 067ca2d | 2018-01-17 01:12:00 -0600 | [diff] [blame] | 9 | |
William A. Kennington III | 6a92363 | 2022-11-06 18:17:33 -0800 | [diff] [blame] | 10 | namespace phosphor::network::netlink |
Nagaraju Goruganti | 067ca2d | 2018-01-17 01:12:00 -0600 | [diff] [blame] | 11 | { |
| 12 | |
William A. Kennington III | 6a92363 | 2022-11-06 18:17:33 -0800 | [diff] [blame] | 13 | TEST(AddrFromRtm, MissingAddr) |
Nagaraju Goruganti | 067ca2d | 2018-01-17 01:12:00 -0600 | [diff] [blame] | 14 | { |
William A. Kennington III | 6a92363 | 2022-11-06 18:17:33 -0800 | [diff] [blame] | 15 | struct |
Gunnar Mills | 57d9c50 | 2018-09-14 14:42:34 -0500 | [diff] [blame] | 16 | { |
William A. Kennington III | 6a92363 | 2022-11-06 18:17:33 -0800 | [diff] [blame] | 17 | alignas(NLMSG_ALIGNTO) ifaddrmsg ifa = {}; |
| 18 | } msg; |
| 19 | EXPECT_THROW(addrFromRtm(stdplus::raw::asView<char>(msg)), |
| 20 | std::runtime_error); |
Nagaraju Goruganti | 067ca2d | 2018-01-17 01:12:00 -0600 | [diff] [blame] | 21 | } |
| 22 | |
William A. Kennington III | 6a92363 | 2022-11-06 18:17:33 -0800 | [diff] [blame] | 23 | TEST(AddrFromRtm, Regular) |
| 24 | { |
| 25 | struct |
| 26 | { |
| 27 | alignas(NLMSG_ALIGNTO) ifaddrmsg ifa; |
| 28 | alignas(NLMSG_ALIGNTO) rtattr addr_hdr; |
| 29 | alignas(NLMSG_ALIGNTO) uint8_t addr[4] = {192, 168, 1, 20}; |
| 30 | } msg; |
| 31 | msg.ifa.ifa_family = AF_INET; |
| 32 | msg.ifa.ifa_prefixlen = 28; |
| 33 | msg.ifa.ifa_flags = 4; |
| 34 | msg.ifa.ifa_scope = 3; |
| 35 | msg.ifa.ifa_index = 10; |
| 36 | msg.addr_hdr.rta_type = IFA_ADDRESS; |
| 37 | msg.addr_hdr.rta_len = RTA_LENGTH(sizeof(msg.addr)); |
| 38 | |
| 39 | auto ret = addrFromRtm(stdplus::raw::asView<char>(msg)); |
| 40 | EXPECT_EQ(msg.ifa.ifa_flags, ret.flags); |
| 41 | EXPECT_EQ(msg.ifa.ifa_scope, ret.scope); |
| 42 | EXPECT_EQ(msg.ifa.ifa_index, ret.ifidx); |
| 43 | EXPECT_EQ((IfAddr{in_addr{hton(0xc0a80114)}, 28}), ret.ifaddr); |
| 44 | } |
| 45 | |
| 46 | TEST(AddrFromRtm, ExtraFlags) |
| 47 | { |
| 48 | struct |
| 49 | { |
| 50 | alignas(NLMSG_ALIGNTO) ifaddrmsg ifa = {}; |
| 51 | alignas(NLMSG_ALIGNTO) rtattr flags_hdr; |
| 52 | alignas(NLMSG_ALIGNTO) uint32_t flags = 0xff00ff00; |
| 53 | alignas(NLMSG_ALIGNTO) rtattr addr_hdr; |
| 54 | alignas(NLMSG_ALIGNTO) uint8_t addr[16] = {}; |
| 55 | } msg; |
| 56 | msg.ifa.ifa_family = AF_INET6; |
| 57 | msg.flags_hdr.rta_type = IFA_FLAGS; |
| 58 | msg.flags_hdr.rta_len = RTA_LENGTH(sizeof(msg.flags)); |
| 59 | msg.addr_hdr.rta_type = IFA_ADDRESS; |
| 60 | msg.addr_hdr.rta_len = RTA_LENGTH(sizeof(msg.addr)); |
| 61 | |
| 62 | auto ret = addrFromRtm(stdplus::raw::asView<char>(msg)); |
| 63 | EXPECT_EQ(0xff00ff00, ret.flags); |
| 64 | } |
| 65 | |
William A. Kennington III | a842690 | 2022-11-07 15:37:41 -0800 | [diff] [blame] | 66 | TEST(NeighFromRtm, MissingAddr) |
| 67 | { |
| 68 | struct |
| 69 | { |
| 70 | alignas(NLMSG_ALIGNTO) ndmsg ndm = {}; |
| 71 | } msg; |
William A. Kennington III | d361514 | 2022-11-12 01:28:33 -0800 | [diff] [blame] | 72 | |
| 73 | EXPECT_EQ((NeighborInfo{}), neighFromRtm(stdplus::raw::asView<char>(msg))); |
William A. Kennington III | a842690 | 2022-11-07 15:37:41 -0800 | [diff] [blame] | 74 | } |
| 75 | |
| 76 | TEST(NeighFromRtm, NoMac) |
| 77 | { |
| 78 | struct |
| 79 | { |
| 80 | alignas(NLMSG_ALIGNTO) ndmsg ndm; |
| 81 | alignas(NLMSG_ALIGNTO) rtattr addr_hdr; |
| 82 | alignas(NLMSG_ALIGNTO) uint8_t addr[4] = {192, 168, 1, 20}; |
| 83 | } msg; |
| 84 | msg.ndm.ndm_family = AF_INET; |
| 85 | msg.ndm.ndm_state = 4; |
| 86 | msg.addr_hdr.rta_type = NDA_DST; |
| 87 | msg.addr_hdr.rta_len = RTA_LENGTH(sizeof(msg.addr)); |
| 88 | |
| 89 | auto ret = neighFromRtm(stdplus::raw::asView<char>(msg)); |
| 90 | EXPECT_EQ(msg.ndm.ndm_state, ret.state); |
| 91 | EXPECT_EQ((in_addr{hton(0xc0a80114)}), ret.addr); |
| 92 | EXPECT_FALSE(ret.mac); |
| 93 | } |
| 94 | |
| 95 | TEST(NeighFromRtm, Full) |
| 96 | { |
| 97 | struct |
| 98 | { |
| 99 | alignas(NLMSG_ALIGNTO) ndmsg ndm; |
| 100 | alignas(NLMSG_ALIGNTO) rtattr addr_hdr; |
| 101 | alignas(NLMSG_ALIGNTO) uint8_t addr[4] = {192, 168, 1, 20}; |
| 102 | alignas(NLMSG_ALIGNTO) rtattr mac_hdr; |
| 103 | alignas(NLMSG_ALIGNTO) uint8_t mac[6] = {1, 2, 3, 4, 5, 6}; |
| 104 | } msg; |
| 105 | msg.ndm.ndm_family = AF_INET; |
| 106 | msg.addr_hdr.rta_type = NDA_DST; |
| 107 | msg.addr_hdr.rta_len = RTA_LENGTH(sizeof(msg.addr)); |
| 108 | msg.mac_hdr.rta_type = NDA_LLADDR; |
| 109 | msg.mac_hdr.rta_len = RTA_LENGTH(sizeof(msg.mac)); |
| 110 | |
| 111 | auto ret = neighFromRtm(stdplus::raw::asView<char>(msg)); |
| 112 | EXPECT_EQ((in_addr{hton(0xc0a80114)}), ret.addr); |
| 113 | EXPECT_EQ((ether_addr{1, 2, 3, 4, 5, 6}), ret.mac); |
| 114 | } |
| 115 | |
William A. Kennington III | 6a92363 | 2022-11-06 18:17:33 -0800 | [diff] [blame] | 116 | } // namespace phosphor::network::netlink |