diff --git a/tests/.gitignore b/tests/.gitignore index 318889ef..b2e42fce 100644 --- a/tests/.gitignore +++ b/tests/.gitignore @@ -213,6 +213,7 @@ newfstatat nlattr nlattr_inet_diag_msg nlattr_inet_diag_req_compat +nlattr_inet_diag_req_v2 nlattr_netlink_diag_msg nlattr_packet_diag_msg nlattr_smc_diag_msg diff --git a/tests/gen_tests.in b/tests/gen_tests.in index dfeec06c..3302cdfc 100644 --- a/tests/gen_tests.in +++ b/tests/gen_tests.in @@ -196,6 +196,7 @@ newfstatat -a32 -v -P stat.sample -P /dev/full nlattr +netlink_sock_diag.test nlattr_inet_diag_msg +netlink_sock_diag.test nlattr_inet_diag_req_compat +netlink_sock_diag.test +nlattr_inet_diag_req_v2 +netlink_sock_diag.test nlattr_netlink_diag_msg +netlink_sock_diag.test nlattr_packet_diag_msg +netlink_sock_diag-v.sh nlattr_smc_diag_msg +netlink_sock_diag.test diff --git a/tests/nlattr_inet_diag_req_v2.c b/tests/nlattr_inet_diag_req_v2.c new file mode 100644 index 00000000..2d98c5bb --- /dev/null +++ b/tests/nlattr_inet_diag_req_v2.c @@ -0,0 +1,429 @@ +/* + * Copyright (c) 2017 JingPiao Chen + * 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 +#include +#include +#include +#include +#include +#include "test_nlattr.h" +#include +#include +#include + +#ifdef HAVE_IF_INDEXTONAME +# define IFINDEX_LO (if_nametoindex("lo")) +#else +# define IFINDEX_LO 1 +#endif + +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< 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("if_nametoindex(\"lo\")}")); +} + +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; +} diff --git a/tests/pure_executables.list b/tests/pure_executables.list index 7a620d4d..221d9523 100755 --- a/tests/pure_executables.list +++ b/tests/pure_executables.list @@ -178,6 +178,7 @@ newfstatat nlattr nlattr_inet_diag_msg nlattr_inet_diag_req_compat +nlattr_inet_diag_req_v2 nlattr_netlink_diag_msg nlattr_packet_diag_msg nlattr_smc_diag_msg