granicus.if.org Git - strace/blob - tests/nlattr_tcmsg.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
  31 #include <stdio.h>
  32 #include <stddef.h>
  33 #include "test_nlattr.h"
  34 #include <linux/pkt_sched.h>
  35 #include <linux/rtnetlink.h>
  36
  37 static void
  38 init_tcmsg(struct nlmsghdr *const nlh, const unsigned int msg_len)
  39 {
  40         SET_STRUCT(struct nlmsghdr, nlh,
  41                 .nlmsg_len = msg_len,
  42                 .nlmsg_type = RTM_GETQDISC,
  43                 .nlmsg_flags = NLM_F_DUMP
  44         );
  45
  46         struct tcmsg *const msg = NLMSG_DATA(nlh);
  47         SET_STRUCT(struct tcmsg, msg,
  48                 .tcm_family = AF_UNIX,
  49                 .tcm_ifindex = ifindex_lo()
  50         );
  51
  52 }
  53
  54 static void
  55 print_tcmsg(const unsigned int msg_len)
  56 {
  57         printf("{len=%u, type=RTM_GETQDISC, flags=NLM_F_DUMP"
  58                ", seq=0, pid=0}, {tcm_family=AF_UNIX"
  59                ", tcm_ifindex=" IFINDEX_LO_STR
  60                ", tcm_handle=0, tcm_parent=0, tcm_info=0}",
  61                msg_len);
  62 }
  63
  64 int
  65 main(void)
  66 {
  67         skip_if_unavailable("/proc/self/fd/");
  68
  69         const int fd = create_nl_socket(NETLINK_ROUTE);
  70         const unsigned int hdrlen = sizeof(struct tcmsg);
  71         void *nlh0 = midtail_alloc(NLMSG_SPACE(hdrlen),
  72                                    NLA_HDRLEN + sizeof(struct tc_stats));
  73
  74         static char pattern[4096];
  75         fill_memory_ex(pattern, sizeof(pattern), 'a', 'z' - 'a' + 1);
  76
  77         const unsigned int nla_type = 0xffff & NLA_TYPE_MASK;
  78         char nla_type_str[256];
  79         sprintf(nla_type_str, "%#x /* TCA_??? */", nla_type);
  80         TEST_NLATTR_(fd, nlh0, hdrlen,
  81                      init_tcmsg, print_tcmsg,
  82                      nla_type, nla_type_str,
  83                      4, pattern, 4,
  84                      print_quoted_hex(pattern, 4));
  85
  86         static const struct tc_stats st = {
  87                 .bytes = 0xabcdcdbefeadefac,
  88                 .packets = 0xbcdeaefd,
  89                 .drops = 0xcdedafed,
  90                 .overlimits = 0xdcdbefad,
  91                 .bps = 0xefaebfad,
  92                 .pps = 0xfefbaedb,
  93                 .qlen = 0xabcdefab,
  94                 .backlog = 0xbdeabeab
  95         };
  96         char buf[offsetofend(struct tc_stats, backlog)];
  97         memcpy(buf, &st, sizeof(buf));
  98         TEST_NLATTR_OBJECT(fd, nlh0, hdrlen,
  99                            init_tcmsg, print_tcmsg,
 100                            TCA_STATS, pattern, buf,
 101                            PRINT_FIELD_U("{", st, bytes);
 102                            PRINT_FIELD_U(", ", st, packets);
 103                            PRINT_FIELD_U(", ", st, drops);
 104                            PRINT_FIELD_U(", ", st, overlimits);
 105                            PRINT_FIELD_U(", ", st, bps);
 106                            PRINT_FIELD_U(", ", st, pps);
 107                            PRINT_FIELD_U(", ", st, qlen);
 108                            PRINT_FIELD_U(", ", st, backlog);
 109                            printf("}"));
 110
 111         static const struct tc_estimator est = {
 112                 .interval = 0xcd,
 113                 .ewma_log = 0xab
 114         };
 115         TEST_NLATTR_OBJECT(fd, nlh0, hdrlen,
 116                            init_tcmsg, print_tcmsg,
 117                            TCA_RATE, pattern, est,
 118                            PRINT_FIELD_D("{", est, interval);
 119                            PRINT_FIELD_U(", ", est, ewma_log);
 120                            printf("}"));
 121
 122         puts("+++ exited with 0 +++");
 123         return 0;
 124 }