granicus.if.org Git - strace/blob - tests/nlattr_smc_diag_msg.c

   1 /*
   2  * Copyright (c) 2017 JingPiao Chen <chenjingpiao@gmail.com>
   3  * Copyright (c) 2017 The strace developers.
   4  * All rights reserved.
   5  *
   6  * Redistribution and use in source and binary forms, with or without
   7  * modification, are permitted provided that the following conditions
   8  * are met:
   9  * 1. Redistributions of source code must retain the above copyright
  10  *    notice, this list of conditions and the following disclaimer.
  11  * 2. Redistributions in binary form must reproduce the above copyright
  12  *    notice, this list of conditions and the following disclaimer in the
  13  *    documentation and/or other materials provided with the distribution.
  14  * 3. The name of the author may not be used to endorse or promote products
  15  *    derived from this software without specific prior written permission.
  16  *
  17  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
  18  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  19  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
  20  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
  21  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  22  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  23  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  24  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  25  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  26  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  27  */
  28
  29 #include "tests.h"
  30 #include <sys/socket.h>
  31
  32 #ifdef AF_SMC
  33
  34 # include <stdio.h>
  35 # include <string.h>
  36 # include <stdint.h>
  37 # include <arpa/inet.h>
  38 # include "test_nlattr.h"
  39 # include <linux/rtnetlink.h>
  40 # include <linux/smc_diag.h>
  41 # include <linux/sock_diag.h>
  42
  43 # ifndef SMC_CLNT
  44 #  define SMC_CLNT 0
  45 # endif
  46 # ifndef SMC_ACTIVE
  47 #  define SMC_ACTIVE 1
  48 # endif
  49
  50 static const char address[] = "12.34.56.78";
  51
  52 static void
  53 init_smc_diag_msg(struct nlmsghdr *const nlh, const unsigned int msg_len)
  54 {
  55         SET_STRUCT(struct nlmsghdr, nlh,
  56                 .nlmsg_len = msg_len,
  57                 .nlmsg_type = SOCK_DIAG_BY_FAMILY,
  58                 .nlmsg_flags = NLM_F_DUMP
  59         );
  60
  61         struct smc_diag_msg *const msg = NLMSG_DATA(nlh);
  62         SET_STRUCT(struct smc_diag_msg, msg,
  63                 .diag_family = AF_SMC,
  64                 .diag_state = SMC_ACTIVE
  65         );
  66
  67         if (!inet_pton(AF_INET, address, msg->id.idiag_src) ||
  68             !inet_pton(AF_INET, address, msg->id.idiag_dst))
  69                 perror_msg_and_skip("inet_pton");
  70 }
  71
  72 static void
  73 print_smc_diag_msg(const unsigned int msg_len)
  74 {
  75         printf("{len=%u, type=SOCK_DIAG_BY_FAMILY"
  76                ", flags=NLM_F_DUMP, seq=0, pid=0}"
  77                ", {diag_family=AF_SMC, diag_state=SMC_ACTIVE"
  78                ", diag_fallback=0, diag_shutdown=0"
  79                ", id={idiag_sport=htons(0), idiag_dport=htons(0)"
  80                ", idiag_src=inet_addr(\"%s\")"
  81                ", idiag_dst=inet_addr(\"%s\")"
  82                ", idiag_if=0, idiag_cookie=[0, 0]}"
  83                ", diag_uid=0, diag_inode=0}",
  84                msg_len, address, address);
  85 }
  86
  87 #define PRINT_FIELD_SMC_DIAG_CURSOR(prefix_, where_, field_)            \
  88         do {                                                            \
  89                 printf("%s%s=", (prefix_), #field_);                    \
  90                 PRINT_FIELD_U("{", (where_).field_, reserved);          \
  91                 PRINT_FIELD_U(", ", (where_).field_, wrap);             \
  92                 PRINT_FIELD_U(", ", (where_).field_, count);            \
  93                 printf("}");                                            \
  94         } while (0)
  95
  96 int main(void)
  97 {
  98         skip_if_unavailable("/proc/self/fd/");
  99
 100         int fd = create_nl_socket(NETLINK_SOCK_DIAG);
 101         const unsigned int hdrlen = sizeof(struct smc_diag_msg);
 102         void *const nlh0 = tail_alloc(NLMSG_SPACE(hdrlen));
 103
 104         static char pattern[4096];
 105         fill_memory_ex(pattern, sizeof(pattern), 'a', 'z' - 'a' + 1);
 106
 107         static const struct smc_diag_conninfo cinfo = {
 108                 .token = 0xabcdefac,
 109                 .sndbuf_size = 0xbcdaefad,
 110                 .rmbe_size = 0xcdbaefab,
 111                 .peer_rmbe_size = 0xdbcdedaf,
 112                 .rx_prod = {
 113                         .reserved = 0xabc1,
 114                         .wrap = 0xbca1,
 115                         .count = 0xcdedbad1
 116                 },
 117                 .rx_cons = {
 118                         .reserved = 0xabc2,
 119                         .wrap = 0xbca2,
 120                         .count = 0xcdedbad2
 121                 },
 122                 .tx_prod = {
 123                         .reserved = 0xabc3,
 124                         .wrap = 0xbca3,
 125                         .count = 0xcdedbad3
 126                 },
 127                 .tx_cons = {
 128                         .reserved = 0xabc4,
 129                         .wrap = 0xbca4,
 130                         .count = 0xcdedbad4
 131                 },
 132                 .rx_prod_flags = 0xff,
 133                 .rx_conn_state_flags = 0xff,
 134                 .tx_prod_flags = 0xff,
 135                 .tx_conn_state_flags = 0xff,
 136                 .tx_prep = {
 137                         .reserved = 0xabc5,
 138                         .wrap = 0xbca5,
 139                         .count = 0xcdedbad5
 140                 },
 141                 .tx_sent = {
 142                         .reserved = 0xabc6,
 143                         .wrap = 0xbca6,
 144                         .count = 0xcdedbad6
 145                 },
 146                 .tx_fin = {
 147                         .reserved = 0xabc7,
 148                         .wrap = 0xbca7,
 149                         .count = 0xcdedbad7
 150                 }
 151         };
 152
 153         TEST_NLATTR_OBJECT(fd, nlh0, hdrlen,
 154                            init_smc_diag_msg, print_smc_diag_msg,
 155                            SMC_DIAG_CONNINFO, pattern, cinfo,
 156                            PRINT_FIELD_U("{", cinfo, token);
 157                            PRINT_FIELD_U(", ", cinfo, sndbuf_size);
 158                            PRINT_FIELD_U(", ", cinfo, rmbe_size);
 159                            PRINT_FIELD_U(", ", cinfo, peer_rmbe_size);
 160                            PRINT_FIELD_SMC_DIAG_CURSOR(", ", cinfo, rx_prod);
 161                            PRINT_FIELD_SMC_DIAG_CURSOR(", ", cinfo, rx_cons);
 162                            PRINT_FIELD_SMC_DIAG_CURSOR(", ", cinfo, tx_prod);
 163                            PRINT_FIELD_SMC_DIAG_CURSOR(", ", cinfo, tx_cons);
 164                            printf(", rx_prod_flags=0xff");
 165                            printf(", rx_conn_state_flags=0xff");
 166                            printf(", tx_prod_flags=0xff");
 167                            printf(", tx_conn_state_flags=0xff");
 168                            PRINT_FIELD_SMC_DIAG_CURSOR(", ", cinfo, tx_prep);
 169                            PRINT_FIELD_SMC_DIAG_CURSOR(", ", cinfo, tx_sent);
 170                            PRINT_FIELD_SMC_DIAG_CURSOR(", ", cinfo, tx_fin);
 171                            printf("}"));
 172
 173         static const struct smc_diag_lgrinfo linfo = {
 174                 .lnk[0] = {
 175                         .link_id = 0xaf,
 176                         .ibport = 0xfa,
 177                         .ibname = "123",
 178                         .gid = "456",
 179                         .peer_gid = "789"
 180                 },
 181                 .role = SMC_CLNT
 182         };
 183         TEST_NLATTR_OBJECT(fd, nlh0, hdrlen,
 184                            init_smc_diag_msg, print_smc_diag_msg,
 185                            SMC_DIAG_LGRINFO, pattern, linfo,
 186                            PRINT_FIELD_U("{lnk[0]={", linfo.lnk[0], link_id);
 187                            printf(", ibname=\"%s\"", linfo.lnk[0].ibname);
 188                            PRINT_FIELD_U(", ", linfo.lnk[0], ibport);
 189                            printf(", gid=\"%s\"", linfo.lnk[0].gid);
 190                            printf(", peer_gid=\"%s\"}", linfo.lnk[0].peer_gid);
 191                            printf(", role=SMC_CLNT}"));
 192
 193         printf("+++ exited with 0 +++\n");
 194         return 0;
 195 }
 196
 197 #else
 198
 199 SKIP_MAIN_UNDEFINED("AF_SMC")
 200
 201 #endif