| /* |
| * Copyright (c) 2017 JingPiao Chen <chenjingpiao@gmail.com> |
| * Copyright (c) 2017 The strace developers. |
| * All rights reserved. |
| * |
| * Redistribution and use in source and binary forms, with or without |
| * modification, are permitted provided that the following conditions |
| * are met: |
| * 1. Redistributions of source code must retain the above copyright |
| * notice, this list of conditions and the following disclaimer. |
| * 2. Redistributions in binary form must reproduce the above copyright |
| * notice, this list of conditions and the following disclaimer in the |
| * documentation and/or other materials provided with the distribution. |
| * 3. The name of the author may not be used to endorse or promote products |
| * derived from this software without specific prior written permission. |
| * |
| * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR |
| * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES |
| * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. |
| * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, |
| * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT |
| * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
| * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
| * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
| * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF |
| * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| */ |
| |
| #include "tests.h" |
| |
| #include <stdio.h> |
| #include <string.h> |
| #include <sys/socket.h> |
| #include <arpa/inet.h> |
| #include <net/if.h> |
| #include <netinet/tcp.h> |
| #include "test_nlattr.h" |
| #include <linux/inet_diag.h> |
| #include <linux/rtnetlink.h> |
| #include <linux/sock_diag.h> |
| |
| static const char address[] = "10.11.12.13"; |
| static const unsigned int hdrlen = sizeof(struct inet_diag_req_v2); |
| static void *nlh0; |
| static char pattern[4096]; |
| |
| static void |
| init_inet_diag_req_v2(struct nlmsghdr *const nlh, const unsigned int msg_len) |
| { |
| SET_STRUCT(struct nlmsghdr, nlh, |
| .nlmsg_len = msg_len, |
| .nlmsg_type = SOCK_DIAG_BY_FAMILY, |
| .nlmsg_flags = NLM_F_REQUEST |
| ); |
| |
| struct inet_diag_req_v2 *const req = NLMSG_DATA(nlh); |
| SET_STRUCT(struct inet_diag_req_v2, req, |
| .sdiag_family = AF_INET, |
| .idiag_ext = 1 << (INET_DIAG_CONG - 1), |
| .sdiag_protocol = IPPROTO_TCP, |
| .idiag_states = 1 << TCP_CLOSE, |
| .id.idiag_if = ifindex_lo() |
| ); |
| |
| if (!inet_pton(AF_INET, address, req->id.idiag_src) || |
| !inet_pton(AF_INET, address, req->id.idiag_dst)) |
| perror_msg_and_skip("inet_pton"); |
| } |
| |
| static void |
| print_inet_diag_req_v2(const unsigned int msg_len) |
| { |
| printf("{len=%u, type=SOCK_DIAG_BY_FAMILY" |
| ", flags=NLM_F_REQUEST, seq=0, pid=0}" |
| ", {sdiag_family=AF_INET, sdiag_protocol=IPPROTO_TCP" |
| ", idiag_ext=1<<(INET_DIAG_CONG-1)" |
| ", idiag_states=1<<TCP_CLOSE" |
| ", id={idiag_sport=htons(0), idiag_dport=htons(0)" |
| ", idiag_src=inet_addr(\"%s\")" |
| ", idiag_dst=inet_addr(\"%s\")" |
| ", idiag_if=" IFINDEX_LO_STR |
| ", idiag_cookie=[0, 0]}}", |
| msg_len, address, address); |
| } |
| |
| static void |
| test_inet_diag_bc_op(const int fd) |
| { |
| static const struct inet_diag_bc_op op = { |
| .code = INET_DIAG_BC_S_COND, |
| .yes = 0xaf, |
| .no = 0xafcd |
| }; |
| TEST_NLATTR_OBJECT(fd, nlh0, hdrlen, |
| init_inet_diag_req_v2, print_inet_diag_req_v2, |
| INET_DIAG_REQ_BYTECODE, pattern, op, |
| printf("{code=INET_DIAG_BC_S_COND"); |
| PRINT_FIELD_U(", ", op, yes); |
| PRINT_FIELD_U(", ", op, no); |
| printf("}")); |
| } |
| |
| static void |
| print_inet_diag_bc_op(const char *const code) |
| { |
| printf("{{code=%s, yes=0, no=0}, ", code); |
| } |
| |
| static void |
| test_inet_diag_bc_s_cond(const int fd) |
| { |
| static const struct inet_diag_bc_op op = { |
| .code = INET_DIAG_BC_S_COND, |
| }; |
| static const struct inet_diag_hostcond cond = { |
| .family = AF_UNSPEC, |
| .prefix_len = 0xad, |
| .port = 0xadfa |
| }; |
| char buf[sizeof(op) + sizeof(cond)]; |
| memcpy(buf, &op, sizeof(op)); |
| |
| const unsigned int plen = sizeof(cond) - 1 > DEFAULT_STRLEN ? |
| sizeof(op) + DEFAULT_STRLEN : sizeof(buf) - 1; |
| memcpy(buf + sizeof(op), &pattern, sizeof(cond)); |
| TEST_NLATTR(fd, nlh0, hdrlen, |
| init_inet_diag_req_v2, print_inet_diag_req_v2, |
| INET_DIAG_REQ_BYTECODE, |
| plen, buf, plen, |
| print_inet_diag_bc_op("INET_DIAG_BC_S_COND"); |
| print_quoted_hex(buf + sizeof(op), plen - sizeof(op)); |
| printf("}")); |
| |
| TEST_NLATTR(fd, nlh0, hdrlen, |
| init_inet_diag_req_v2, print_inet_diag_req_v2, |
| INET_DIAG_REQ_BYTECODE, |
| sizeof(buf), buf, sizeof(buf) - 1, |
| print_inet_diag_bc_op("INET_DIAG_BC_S_COND"); |
| printf("%p}", RTA_DATA(TEST_NLATTR_nla) + sizeof(op))); |
| |
| memcpy(buf + sizeof(op), &cond, sizeof(cond)); |
| TEST_NLATTR(fd, nlh0, hdrlen, |
| init_inet_diag_req_v2, print_inet_diag_req_v2, |
| INET_DIAG_REQ_BYTECODE, |
| sizeof(buf), buf, sizeof(buf), |
| print_inet_diag_bc_op("INET_DIAG_BC_S_COND"); |
| printf("{family=AF_UNSPEC"); |
| PRINT_FIELD_U(", ", cond, prefix_len); |
| PRINT_FIELD_U(", ", cond, port); |
| printf("}}")); |
| } |
| |
| static void |
| print_inet_diag_hostcond(const char *const family) |
| { |
| printf("{family=%s, prefix_len=0, port=0, ", family); |
| } |
| |
| static void |
| test_in_addr(const int fd) |
| { |
| static const struct inet_diag_bc_op op = { |
| .code = INET_DIAG_BC_S_COND, |
| }; |
| static const struct inet_diag_hostcond cond = { |
| .family = AF_INET, |
| }; |
| struct in_addr addr; |
| if (!inet_pton(AF_INET, address, &addr)) |
| perror_msg_and_skip("inet_pton"); |
| |
| char buf[sizeof(op) + sizeof(cond) + sizeof(addr)]; |
| memcpy(buf, &op, sizeof(op)); |
| memcpy(buf + sizeof(op), &cond, sizeof(cond)); |
| |
| const unsigned int plen = sizeof(addr) - 1 > DEFAULT_STRLEN ? |
| sizeof(cond) + sizeof(cond) + DEFAULT_STRLEN : sizeof(buf) - 1; |
| memcpy(buf + sizeof(op) + sizeof(cond), &pattern, sizeof(addr)); |
| TEST_NLATTR(fd, nlh0, hdrlen, |
| init_inet_diag_req_v2, print_inet_diag_req_v2, |
| INET_DIAG_REQ_BYTECODE, |
| plen, buf, plen, |
| print_inet_diag_bc_op("INET_DIAG_BC_S_COND"); |
| print_inet_diag_hostcond("AF_INET"); |
| printf("addr="); |
| print_quoted_hex(pattern, plen - sizeof(op) - sizeof(cond)); |
| printf("}}")); |
| |
| TEST_NLATTR(fd, nlh0, hdrlen, |
| init_inet_diag_req_v2, print_inet_diag_req_v2, |
| INET_DIAG_REQ_BYTECODE, |
| sizeof(buf), buf, sizeof(buf) - 1, |
| print_inet_diag_bc_op("INET_DIAG_BC_S_COND"); |
| print_inet_diag_hostcond("AF_INET"); |
| printf("addr=%p}}", |
| RTA_DATA(TEST_NLATTR_nla) |
| + sizeof(op) + sizeof(cond))); |
| |
| memcpy(buf + sizeof(op) + sizeof(cond), &addr, sizeof(addr)); |
| TEST_NLATTR(fd, nlh0, hdrlen, |
| init_inet_diag_req_v2, print_inet_diag_req_v2, |
| INET_DIAG_REQ_BYTECODE, |
| sizeof(buf), buf, sizeof(buf), |
| print_inet_diag_bc_op("INET_DIAG_BC_S_COND"); |
| print_inet_diag_hostcond("AF_INET"); |
| printf("addr=inet_addr(\"%s\")}}", address)); |
| } |
| |
| static void |
| test_in6_addr(const int fd) |
| { |
| const char address6[] = "12:34:56:78:90:ab:cd:ef"; |
| static const struct inet_diag_bc_op op = { |
| .code = INET_DIAG_BC_S_COND, |
| }; |
| static const struct inet_diag_hostcond cond = { |
| .family = AF_INET6, |
| }; |
| struct in6_addr addr; |
| if (!inet_pton(AF_INET6, address6, &addr)) |
| perror_msg_and_skip("inet_pton"); |
| |
| char buf[sizeof(op) + sizeof(cond) + sizeof(addr)]; |
| memcpy(buf, &op, sizeof(op)); |
| memcpy(buf + sizeof(op), &cond, sizeof(cond)); |
| |
| const unsigned int plen = sizeof(addr) - 1 > DEFAULT_STRLEN ? |
| sizeof(cond) + sizeof(cond) + DEFAULT_STRLEN : sizeof(buf) - 1; |
| memcpy(buf + sizeof(op) + sizeof(cond), &pattern, sizeof(addr)); |
| TEST_NLATTR(fd, nlh0, hdrlen, |
| init_inet_diag_req_v2, print_inet_diag_req_v2, |
| INET_DIAG_REQ_BYTECODE, |
| plen, buf, plen, |
| print_inet_diag_bc_op("INET_DIAG_BC_S_COND"); |
| print_inet_diag_hostcond("AF_INET6"); |
| printf("addr="); |
| print_quoted_hex(pattern, plen - sizeof(op) - sizeof(cond)); |
| printf("}}")); |
| |
| TEST_NLATTR(fd, nlh0, hdrlen, |
| init_inet_diag_req_v2, print_inet_diag_req_v2, |
| INET_DIAG_REQ_BYTECODE, |
| sizeof(buf), buf, sizeof(buf) - 1, |
| print_inet_diag_bc_op("INET_DIAG_BC_S_COND"); |
| print_inet_diag_hostcond("AF_INET6"); |
| printf("addr=%p}}", |
| RTA_DATA(TEST_NLATTR_nla) |
| + sizeof(op) + sizeof(cond))); |
| |
| memcpy(buf + sizeof(op) + sizeof(cond), &addr, sizeof(addr)); |
| TEST_NLATTR(fd, nlh0, hdrlen, |
| init_inet_diag_req_v2, print_inet_diag_req_v2, |
| INET_DIAG_REQ_BYTECODE, |
| sizeof(buf), buf, sizeof(buf), |
| print_inet_diag_bc_op("INET_DIAG_BC_S_COND"); |
| print_inet_diag_hostcond("AF_INET6"); |
| printf("inet_pton(AF_INET6, \"%s\", &addr)}}", address6)); |
| } |
| |
| static void |
| test_inet_diag_bc_dev_cond(const int fd) |
| { |
| static const struct inet_diag_bc_op op = { |
| .code = INET_DIAG_BC_DEV_COND, |
| }; |
| const uint32_t ifindex = ifindex_lo(); |
| char buf[sizeof(op) + sizeof(ifindex)]; |
| memcpy(buf, &op, sizeof(op)); |
| memcpy(buf + sizeof(op), pattern, sizeof(ifindex)); |
| |
| TEST_NLATTR(fd, nlh0, hdrlen, |
| init_inet_diag_req_v2, print_inet_diag_req_v2, |
| INET_DIAG_REQ_BYTECODE, |
| sizeof(buf) - 1, buf, sizeof(buf) - 1, |
| print_inet_diag_bc_op("INET_DIAG_BC_DEV_COND"); |
| print_quoted_hex(pattern, sizeof(ifindex) - 1); |
| printf("}")); |
| |
| TEST_NLATTR(fd, nlh0, hdrlen, |
| init_inet_diag_req_v2, print_inet_diag_req_v2, |
| INET_DIAG_REQ_BYTECODE, |
| sizeof(buf), buf, sizeof(buf) - 1, |
| print_inet_diag_bc_op("INET_DIAG_BC_DEV_COND"); |
| printf("%p}", RTA_DATA(TEST_NLATTR_nla) + sizeof(op))); |
| |
| memcpy(buf + sizeof(op), &ifindex, sizeof(ifindex)); |
| TEST_NLATTR(fd, nlh0, hdrlen, |
| init_inet_diag_req_v2, print_inet_diag_req_v2, |
| INET_DIAG_REQ_BYTECODE, |
| sizeof(buf), buf, sizeof(buf), |
| print_inet_diag_bc_op("INET_DIAG_BC_DEV_COND"); |
| printf(IFINDEX_LO_STR "}")); |
| } |
| |
| static void |
| test_inet_diag_bc_s_le(const int fd) |
| { |
| static const struct inet_diag_bc_op op[] = { |
| { |
| .code = INET_DIAG_BC_S_LE, |
| }, |
| { |
| .code = INET_DIAG_BC_DEV_COND, |
| .yes = 0xaf, |
| .no = 0xafcd |
| } |
| }; |
| |
| char buf[sizeof(op)]; |
| memcpy(buf, op, sizeof(op[0])); |
| memcpy(buf + sizeof(op[0]), pattern, sizeof(op[1])); |
| |
| const unsigned int plen = sizeof(op[1]) - 1 > DEFAULT_STRLEN ? |
| sizeof(op[0]) + DEFAULT_STRLEN : sizeof(buf) - 1; |
| TEST_NLATTR(fd, nlh0, hdrlen, |
| init_inet_diag_req_v2, print_inet_diag_req_v2, |
| INET_DIAG_REQ_BYTECODE, |
| plen, buf, plen, |
| print_inet_diag_bc_op("INET_DIAG_BC_S_LE"); |
| print_quoted_hex(buf + sizeof(op[0]), plen - sizeof(op[0])); |
| printf("}")); |
| |
| TEST_NLATTR(fd, nlh0, hdrlen, |
| init_inet_diag_req_v2, print_inet_diag_req_v2, |
| INET_DIAG_REQ_BYTECODE, |
| sizeof(buf), buf, sizeof(buf) - 1, |
| print_inet_diag_bc_op("INET_DIAG_BC_S_LE"); |
| printf("%p}", RTA_DATA(TEST_NLATTR_nla) + sizeof(op[0]))); |
| |
| memcpy(buf + sizeof(op[0]), &op[1], sizeof(op[1])); |
| TEST_NLATTR(fd, nlh0, hdrlen, |
| init_inet_diag_req_v2, print_inet_diag_req_v2, |
| INET_DIAG_REQ_BYTECODE, |
| sizeof(buf), buf, sizeof(buf), |
| print_inet_diag_bc_op("INET_DIAG_BC_S_LE"); |
| printf("{code=INET_DIAG_BC_DEV_COND"); |
| PRINT_FIELD_U(", ", op[1], yes); |
| PRINT_FIELD_U(", ", op[1], no); |
| printf("}}")); |
| }; |
| |
| static void |
| test_inet_diag_bc_mark_cond(const int fd) |
| { |
| static const struct inet_diag_bc_op op = { |
| .code = INET_DIAG_BC_MARK_COND, |
| }; |
| static const struct inet_diag_markcond markcond = { |
| .mark = 0xafbcafcd, |
| .mask = 0xbafaacda |
| }; |
| char buf[sizeof(op) + sizeof(markcond)]; |
| memcpy(buf, &op, sizeof(op)); |
| memcpy(buf + sizeof(op), pattern, sizeof(markcond)); |
| |
| const unsigned int plen = sizeof(markcond) - 1 > DEFAULT_STRLEN ? |
| sizeof(markcond) + DEFAULT_STRLEN : sizeof(buf) - 1; |
| TEST_NLATTR(fd, nlh0, hdrlen, |
| init_inet_diag_req_v2, print_inet_diag_req_v2, |
| INET_DIAG_REQ_BYTECODE, |
| plen, buf, plen, |
| print_inet_diag_bc_op("INET_DIAG_BC_MARK_COND"); |
| print_quoted_hex(buf + sizeof(op), plen - sizeof(op)); |
| printf("}")); |
| |
| TEST_NLATTR(fd, nlh0, hdrlen, |
| init_inet_diag_req_v2, print_inet_diag_req_v2, |
| INET_DIAG_REQ_BYTECODE, |
| sizeof(buf), buf, sizeof(buf) - 1, |
| print_inet_diag_bc_op("INET_DIAG_BC_MARK_COND"); |
| printf("%p}", RTA_DATA(TEST_NLATTR_nla) + sizeof(op))); |
| |
| memcpy(buf + sizeof(op), &markcond, sizeof(markcond)); |
| TEST_NLATTR(fd, nlh0, hdrlen, |
| init_inet_diag_req_v2, print_inet_diag_req_v2, |
| INET_DIAG_REQ_BYTECODE, |
| sizeof(buf), buf, sizeof(buf), |
| print_inet_diag_bc_op("INET_DIAG_BC_MARK_COND"); |
| PRINT_FIELD_U("{", markcond, mark); |
| PRINT_FIELD_U(", ", markcond, mask); |
| printf("}}")); |
| } |
| |
| static void |
| test_inet_diag_bc_nop(const int fd) |
| { |
| static const struct inet_diag_bc_op op = { |
| .code = INET_DIAG_BC_AUTO, |
| }; |
| char buf[sizeof(op) + 4]; |
| memcpy(buf, &op, sizeof(op)); |
| memcpy(buf + sizeof(op), pattern, 4); |
| |
| TEST_NLATTR(fd, nlh0, hdrlen, |
| init_inet_diag_req_v2, print_inet_diag_req_v2, |
| INET_DIAG_REQ_BYTECODE, |
| sizeof(buf), buf, sizeof(buf), |
| print_inet_diag_bc_op("INET_DIAG_BC_AUTO"); |
| print_quoted_hex(buf + sizeof(op), |
| sizeof(buf) - sizeof(op)); |
| printf("}")); |
| } |
| |
| int |
| main(void) |
| { |
| skip_if_unavailable("/proc/self/fd/"); |
| |
| int fd = create_nl_socket(NETLINK_SOCK_DIAG); |
| nlh0 = tail_alloc(NLMSG_SPACE(hdrlen)); |
| fill_memory_ex(pattern, sizeof(pattern), 'a', 'z' - 'a' + 1); |
| |
| test_inet_diag_bc_op(fd); |
| test_inet_diag_bc_s_cond(fd); |
| test_in_addr(fd); |
| test_in6_addr(fd); |
| test_inet_diag_bc_dev_cond(fd); |
| test_inet_diag_bc_s_le(fd); |
| test_inet_diag_bc_mark_cond(fd); |
| test_inet_diag_bc_nop(fd); |
| |
| printf("+++ exited with 0 +++\n"); |
| return 0; |
| } |