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

   1 /*
   2  * Copyright (c) 2015-2016 Dmitry V. Levin <ldv@altlinux.org>
   3  * All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in the
  12  *    documentation and/or other materials provided with the distribution.
  13  * 3. The name of the author may not be used to endorse or promote products
  14  *    derived from this software without specific prior written permission.
  15  *
  16  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
  17  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  18  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
  19  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
  20  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  21  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  22  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  23  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  24  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  25  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  26  */
  27
  28 #include "tests.h"
  29 #include <netinet/in.h>
  30
  31 #if defined IP_ADD_MEMBERSHIP && defined IPV6_ADD_MEMBERSHIP \
  32  && defined IPV6_JOIN_ANYCAST && defined HAVE_IF_INDEXTONAME
  33
  34 # include <stdio.h>
  35 # include <unistd.h>
  36 # include <sys/socket.h>
  37 # include <arpa/inet.h>
  38 # include <net/if.h>
  39
  40 int
  41 main(void)
  42 {
  43         static const char multi4addr[] = "224.0.0.3";
  44         static const char multi6addr[] = "ff01::c";
  45         static const char interface[] = "127.0.0.1";
  46
  47         TAIL_ALLOC_OBJECT_CONST_PTR(struct ip_mreq, m4);
  48         TAIL_ALLOC_OBJECT_CONST_PTR(struct ipv6_mreq, m6);
  49         unsigned int i;
  50         int rc;
  51
  52         inet_pton(AF_INET, multi4addr, &m4->imr_multiaddr);
  53         inet_pton(AF_INET, interface, &m4->imr_interface);
  54         inet_pton(AF_INET6, multi6addr, &m6->ipv6mr_multiaddr);
  55
  56         m6->ipv6mr_interface = if_nametoindex("lo");
  57         if (!m6->ipv6mr_interface)
  58                 perror_msg_and_skip("lo");
  59
  60         (void) close(0);
  61         if (socket(AF_INET, SOCK_DGRAM, 0))
  62                 perror_msg_and_skip("socket");
  63
  64         struct {
  65                 int level;
  66                 const char *str_level;
  67                 int optname;
  68                 const char *str_optname;
  69                 void *optval;
  70                 unsigned int optsize;
  71         } short_any[] = {
  72                 {
  73                         ARG_STR(SOL_IP), ARG_STR(IP_ADD_MEMBERSHIP),
  74                         m4, sizeof(*m4)
  75                 },
  76                 {
  77                         ARG_STR(SOL_IP), ARG_STR(IP_DROP_MEMBERSHIP),
  78                         m4, sizeof(*m4)
  79                 },
  80                 {
  81                         ARG_STR(SOL_IPV6), ARG_STR(IPV6_ADD_MEMBERSHIP),
  82                         m6, sizeof(*m6)
  83                 },
  84                 {
  85                         ARG_STR(SOL_IPV6), ARG_STR(IPV6_DROP_MEMBERSHIP),
  86                         m6, sizeof(*m6)
  87                 },
  88                 {
  89                         ARG_STR(SOL_IPV6), ARG_STR(IPV6_JOIN_ANYCAST),
  90                         m6, sizeof(*m6)
  91                 },
  92                 {
  93                         ARG_STR(SOL_IPV6), ARG_STR(IPV6_LEAVE_ANYCAST),
  94                         m6, sizeof(*m6)
  95                 }
  96         };
  97
  98         for (i = 0; i < ARRAY_SIZE(short_any); ++i) {
  99                 rc = setsockopt(0, short_any[i].level, short_any[i].optname,
 100                                 short_any[i].optval, 1);
 101                 printf("setsockopt(0, %s, %s, \"\\%hho\", 1) = %s\n",
 102                        short_any[i].str_level, short_any[i].str_optname,
 103                        * (unsigned char *) short_any[i].optval,
 104                        sprintrc(rc));
 105
 106                 rc = setsockopt(0, short_any[i].level, short_any[i].optname,
 107                                 short_any[i].optval + 1, short_any[i].optsize);
 108                 printf("setsockopt(0, %s, %s, %p, %u) = %s\n",
 109                        short_any[i].str_level, short_any[i].str_optname,
 110                        short_any[i].optval + 1, short_any[i].optsize,
 111                        sprintrc(rc));
 112         }
 113
 114         struct {
 115                 int optname;
 116                 const char *str_optname;
 117         } long_ip[] = {
 118                 { ARG_STR(IP_ADD_MEMBERSHIP) },
 119                 { ARG_STR(IP_DROP_MEMBERSHIP) }
 120         }, long_ipv6[] = {
 121                 { ARG_STR(IPV6_ADD_MEMBERSHIP) },
 122                 { ARG_STR(IPV6_DROP_MEMBERSHIP) },
 123                 { ARG_STR(IPV6_JOIN_ANYCAST) },
 124                 { ARG_STR(IPV6_LEAVE_ANYCAST) }
 125         };
 126
 127         for (i = 0; i < ARRAY_SIZE(long_ip); ++i) {
 128                 rc = setsockopt(0, SOL_IP, long_ip[i].optname,
 129                                 m4, sizeof(*m4));
 130                 printf("setsockopt(0, SOL_IP, %s"
 131                        ", {imr_multiaddr=inet_addr(\"%s\")"
 132                        ", imr_interface=inet_addr(\"%s\")}, %u) = %s\n",
 133                        long_ip[i].str_optname, multi4addr,
 134                        interface, (unsigned) sizeof(*m4), sprintrc(rc));
 135         }
 136
 137         for (i = 0; i < ARRAY_SIZE(long_ipv6); ++i) {
 138                 rc = setsockopt(0, SOL_IPV6, long_ipv6[i].optname,
 139                                 m6, sizeof(*m6));
 140                 printf("setsockopt(0, SOL_IPV6, %s"
 141                        ", {ipv6mr_multiaddr=inet_pton(\"%s\")"
 142                        ", ipv6mr_interface=if_nametoindex(\"lo\")}"
 143                        ", %u) = %s\n",
 144                        long_ipv6[i].str_optname, multi6addr,
 145                        (unsigned) sizeof(*m6), sprintrc(rc));
 146         }
 147
 148         puts("+++ exited with 0 +++");
 149         return 0;
 150 }
 151
 152 #else
 153
 154 SKIP_MAIN_UNDEFINED("IP_ADD_MEMBERSHIP && IPV6_ADD_MEMBERSHIP"
 155                     " && IPV6_JOIN_ANYCAST && HAVE_IF_INDEXTONAME")
 156
 157 #endif