/* * Copyright (c) 2017 JingPiao Chen * Copyright (c) 2017-2018 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 #ifdef AF_SMC # include # include # include # include # include "test_nlattr.h" # include # include # include # ifndef SMC_CLNT # define SMC_CLNT 0 # endif # ifndef SMC_ACTIVE # define SMC_ACTIVE 1 # endif static const char address[] = "12.34.56.78"; static void init_smc_diag_msg(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_DUMP ); struct smc_diag_msg *const msg = NLMSG_DATA(nlh); SET_STRUCT(struct smc_diag_msg, msg, .diag_family = AF_SMC, .diag_state = SMC_ACTIVE ); if (!inet_pton(AF_INET, address, msg->id.idiag_src) || !inet_pton(AF_INET, address, msg->id.idiag_dst)) perror_msg_and_skip("inet_pton"); } static void print_smc_diag_msg(const unsigned int msg_len) { printf("{len=%u, type=SOCK_DIAG_BY_FAMILY" ", flags=NLM_F_DUMP, seq=0, pid=0}" ", {diag_family=AF_SMC, diag_state=SMC_ACTIVE" ", diag_fallback=0, diag_shutdown=0" ", id={idiag_sport=htons(0), idiag_dport=htons(0)" ", idiag_src=inet_addr(\"%s\")" ", idiag_dst=inet_addr(\"%s\")" ", idiag_if=0, idiag_cookie=[0, 0]}" ", diag_uid=0, diag_inode=0}", msg_len, address, address); } #define PRINT_FIELD_SMC_DIAG_CURSOR(prefix_, where_, field_) \ do { \ printf("%s%s=", (prefix_), #field_); \ PRINT_FIELD_U("{", (where_).field_, reserved); \ PRINT_FIELD_U(", ", (where_).field_, wrap); \ PRINT_FIELD_U(", ", (where_).field_, count); \ printf("}"); \ } while (0) int main(void) { skip_if_unavailable("/proc/self/fd/"); static const struct smc_diag_conninfo cinfo = { .token = 0xabcdefac, .sndbuf_size = 0xbcdaefad, .rmbe_size = 0xcdbaefab, .peer_rmbe_size = 0xdbcdedaf, .rx_prod = { .reserved = 0xabc1, .wrap = 0xbca1, .count = 0xcdedbad1 }, .rx_cons = { .reserved = 0xabc2, .wrap = 0xbca2, .count = 0xcdedbad2 }, .tx_prod = { .reserved = 0xabc3, .wrap = 0xbca3, .count = 0xcdedbad3 }, .tx_cons = { .reserved = 0xabc4, .wrap = 0xbca4, .count = 0xcdedbad4 }, .rx_prod_flags = 0xff, .rx_conn_state_flags = 0xff, .tx_prod_flags = 0xff, .tx_conn_state_flags = 0xff, .tx_prep = { .reserved = 0xabc5, .wrap = 0xbca5, .count = 0xcdedbad5 }, .tx_sent = { .reserved = 0xabc6, .wrap = 0xbca6, .count = 0xcdedbad6 }, .tx_fin = { .reserved = 0xabc7, .wrap = 0xbca7, .count = 0xcdedbad7 } }; static const struct smc_diag_lgrinfo linfo = { .lnk[0] = { .link_id = 0xaf, .ibport = 0xfa, .ibname = "123", .gid = "456", .peer_gid = "789" }, .role = SMC_CLNT }; int fd = create_nl_socket(NETLINK_SOCK_DIAG); const unsigned int hdrlen = sizeof(struct smc_diag_msg); void *const nlh0 = midtail_alloc(NLMSG_SPACE(hdrlen), NLA_HDRLEN + MAX(sizeof(cinfo), sizeof(linfo))); static char pattern[4096]; fill_memory_ex(pattern, sizeof(pattern), 'a', 'z' - 'a' + 1); TEST_NLATTR_OBJECT(fd, nlh0, hdrlen, init_smc_diag_msg, print_smc_diag_msg, SMC_DIAG_CONNINFO, pattern, cinfo, PRINT_FIELD_U("{", cinfo, token); PRINT_FIELD_U(", ", cinfo, sndbuf_size); PRINT_FIELD_U(", ", cinfo, rmbe_size); PRINT_FIELD_U(", ", cinfo, peer_rmbe_size); PRINT_FIELD_SMC_DIAG_CURSOR(", ", cinfo, rx_prod); PRINT_FIELD_SMC_DIAG_CURSOR(", ", cinfo, rx_cons); PRINT_FIELD_SMC_DIAG_CURSOR(", ", cinfo, tx_prod); PRINT_FIELD_SMC_DIAG_CURSOR(", ", cinfo, tx_cons); printf(", rx_prod_flags=0xff"); printf(", rx_conn_state_flags=0xff"); printf(", tx_prod_flags=0xff"); printf(", tx_conn_state_flags=0xff"); PRINT_FIELD_SMC_DIAG_CURSOR(", ", cinfo, tx_prep); PRINT_FIELD_SMC_DIAG_CURSOR(", ", cinfo, tx_sent); PRINT_FIELD_SMC_DIAG_CURSOR(", ", cinfo, tx_fin); printf("}")); TEST_NLATTR_OBJECT(fd, nlh0, hdrlen, init_smc_diag_msg, print_smc_diag_msg, SMC_DIAG_LGRINFO, pattern, linfo, PRINT_FIELD_U("{lnk[0]={", linfo.lnk[0], link_id); printf(", ibname=\"%s\"", linfo.lnk[0].ibname); PRINT_FIELD_U(", ", linfo.lnk[0], ibport); printf(", gid=\"%s\"", linfo.lnk[0].gid); printf(", peer_gid=\"%s\"}", linfo.lnk[0].peer_gid); printf(", role=SMC_CLNT}")); printf("+++ exited with 0 +++\n"); return 0; } #else SKIP_MAIN_UNDEFINED("AF_SMC") #endif