ospf6d/ospf6_network.c

   1 /*
   2  * Copyright (C) 2003 Yasuhiro Ohara
   3  *
   4  * This file is part of GNU Zebra.
   5  *
   6  * GNU Zebra is free software; you can redistribute it and/or modify it
   7  * under the terms of the GNU General Public License as published by the
   8  * Free Software Foundation; either version 2, or (at your option) any
   9  * later version.
  10  *
  11  * GNU Zebra is distributed in the hope that it will be useful, but
  12  * WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with GNU Zebra; see the file COPYING.  If not, write to the
  18  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  19  * Boston, MA 02111-1307, USA.
  20  */
  21
  22 #include <zebra.h>
  23
  24 #include "log.h"
  25 #include "memory.h"
  26 #include "sockunion.h"
  27 #include "sockopt.h"
  28 #include "privs.h"
  29
  30 #include "libospf.h"
  31 #include "ospf6_proto.h"
  32 #include "ospf6_network.h"
  33
  34 extern struct zebra_privs_t ospf6d_privs;
  35
  36 int  ospf6_sock;
  37 struct in6_addr allspfrouters6;
  38 struct in6_addr alldrouters6;
  39
  40 /* setsockopt MulticastLoop to off */
  41 static void
  42 ospf6_reset_mcastloop (void)
  43 {
  44   u_int off = 0;
  45   if (setsockopt (ospf6_sock, IPPROTO_IPV6, IPV6_MULTICAST_LOOP,
  46                   &off, sizeof (u_int)) < 0)
  47     zlog_warn ("Network: reset IPV6_MULTICAST_LOOP failed: %s",
  48                safe_strerror (errno));
  49 }
  50
  51 static void
  52 ospf6_set_pktinfo (void)
  53 {
  54   setsockopt_ipv6_pktinfo (ospf6_sock, 1);
  55 }
  56
  57 static void
  58 ospf6_set_transport_class (void)
  59 {
  60 #ifdef IPTOS_PREC_INTERNETCONTROL
  61   setsockopt_ipv6_tclass (ospf6_sock, IPTOS_PREC_INTERNETCONTROL);
  62 #endif
  63 }
  64
  65 static void
  66 ospf6_set_checksum (void)
  67 {
  68   int offset = 12;
  69 #ifndef DISABLE_IPV6_CHECKSUM
  70   if (setsockopt (ospf6_sock, IPPROTO_IPV6, IPV6_CHECKSUM,
  71                   &offset, sizeof (offset)) < 0)
  72     zlog_warn ("Network: set IPV6_CHECKSUM failed: %s", safe_strerror (errno));
  73 #else
  74   zlog_warn ("Network: Don't set IPV6_CHECKSUM");
  75 #endif /* DISABLE_IPV6_CHECKSUM */
  76 }
  77
  78 /* Make ospf6d's server socket. */
  79 int
  80 ospf6_serv_sock (void)
  81 {
  82   if (ospf6d_privs.change (ZPRIVS_RAISE))
  83     zlog_err ("ospf6_serv_sock: could not raise privs");
  84
  85   ospf6_sock = socket (AF_INET6, SOCK_RAW, IPPROTO_OSPFIGP);
  86   if (ospf6_sock < 0)
  87     {
  88       zlog_warn ("Network: can't create OSPF6 socket.");
  89       if (ospf6d_privs.change (ZPRIVS_LOWER))
  90         zlog_err ("ospf_sock_init: could not lower privs");
  91       return -1;
  92     }
  93   if (ospf6d_privs.change (ZPRIVS_LOWER))
  94       zlog_err ("ospf_sock_init: could not lower privs");
  95
  96   /* set socket options */
  97 #if 1
  98   sockopt_reuseaddr (ospf6_sock);
  99 #else
 100   ospf6_set_reuseaddr ();
 101 #endif /*1*/
 102   ospf6_reset_mcastloop ();
 103   ospf6_set_pktinfo ();
 104   ospf6_set_transport_class ();
 105   ospf6_set_checksum ();
 106
 107   /* setup global in6_addr, allspf6 and alldr6 for later use */
 108   inet_pton (AF_INET6, ALLSPFROUTERS6, &allspfrouters6);
 109   inet_pton (AF_INET6, ALLDROUTERS6, &alldrouters6);
 110
 111   return 0;
 112 }
 113
 114 /* ospf6 set socket option */
 115 int
 116 ospf6_sso (ifindex_t ifindex, struct in6_addr *group, int option)
 117 {
 118   struct ipv6_mreq mreq6;
 119   int ret;
 120   int bufsize = (8 * 1024 * 1024);
 121
 122   assert (ifindex);
 123   mreq6.ipv6mr_interface = ifindex;
 124   memcpy (&mreq6.ipv6mr_multiaddr, group, sizeof (struct in6_addr));
 125
 126   ret = setsockopt (ospf6_sock, IPPROTO_IPV6, option,
 127                     &mreq6, sizeof (mreq6));
 128   if (ret < 0)
 129     {
 130       zlog_err ("Network: setsockopt (%d) on ifindex %d failed: %s",
 131                 option, ifindex, safe_strerror (errno));
 132       return ret;
 133     }
 134
 135   setsockopt_so_sendbuf (ospf6_sock, bufsize);
 136   setsockopt_so_recvbuf (ospf6_sock, bufsize);
 137
 138   return 0;
 139 }
 140
 141 static int
 142 iov_count (struct iovec *iov)
 143 {
 144   int i;
 145   for (i = 0; iov[i].iov_base; i++)
 146     ;
 147   return i;
 148 }
 149
 150 static int
 151 iov_totallen (struct iovec *iov)
 152 {
 153   int i;
 154   int totallen = 0;
 155   for (i = 0; iov[i].iov_base; i++)
 156     totallen += iov[i].iov_len;
 157   return totallen;
 158 }
 159
 160 int
 161 ospf6_sendmsg (struct in6_addr *src, struct in6_addr *dst,
 162                ifindex_t *ifindex, struct iovec *message)
 163 {
 164   int retval;
 165   struct msghdr smsghdr;
 166   struct cmsghdr *scmsgp;
 167   union
 168   {
 169     struct cmsghdr hdr;
 170     u_char buf[CMSG_SPACE (sizeof (struct in6_pktinfo))];
 171   } cmsgbuf;
 172   struct in6_pktinfo *pktinfo;
 173   struct sockaddr_in6 dst_sin6;
 174
 175   assert (dst);
 176   assert (*ifindex);
 177
 178   scmsgp = (struct cmsghdr *)&cmsgbuf;
 179   pktinfo = (struct in6_pktinfo *)(CMSG_DATA(scmsgp));
 180   memset (&dst_sin6, 0, sizeof (struct sockaddr_in6));
 181
 182   /* source address */
 183   pktinfo->ipi6_ifindex = *ifindex;
 184   if (src)
 185     memcpy (&pktinfo->ipi6_addr, src, sizeof (struct in6_addr));
 186   else
 187     memset (&pktinfo->ipi6_addr, 0, sizeof (struct in6_addr));
 188
 189   /* destination address */
 190   dst_sin6.sin6_family = AF_INET6;
 191 #ifdef SIN6_LEN
 192   dst_sin6.sin6_len = sizeof (struct sockaddr_in6);
 193 #endif /*SIN6_LEN*/
 194   memcpy (&dst_sin6.sin6_addr, dst, sizeof (struct in6_addr));
 195   dst_sin6.sin6_scope_id = *ifindex;
 196
 197   /* send control msg */
 198   scmsgp->cmsg_level = IPPROTO_IPV6;
 199   scmsgp->cmsg_type = IPV6_PKTINFO;
 200   scmsgp->cmsg_len = CMSG_LEN (sizeof (struct in6_pktinfo));
 201   /* scmsgp = CMSG_NXTHDR (&smsghdr, scmsgp); */
 202
 203   /* send msg hdr */
 204   memset (&smsghdr, 0, sizeof (smsghdr));
 205   smsghdr.msg_iov = message;
 206   smsghdr.msg_iovlen = iov_count (message);
 207   smsghdr.msg_name = (caddr_t) &dst_sin6;
 208   smsghdr.msg_namelen = sizeof (struct sockaddr_in6);
 209   smsghdr.msg_control = (caddr_t) &cmsgbuf.buf;
 210   smsghdr.msg_controllen = sizeof(cmsgbuf.buf);
 211
 212   retval = sendmsg (ospf6_sock, &smsghdr, 0);
 213   if (retval != iov_totallen (message))
 214     zlog_warn ("sendmsg failed: ifindex: %d: %s (%d)",
 215                *ifindex, safe_strerror (errno), errno);
 216
 217   return retval;
 218 }
 219
 220 int
 221 ospf6_recvmsg (struct in6_addr *src, struct in6_addr *dst,
 222                ifindex_t *ifindex, struct iovec *message)
 223 {
 224   int retval;
 225   struct msghdr rmsghdr;
 226   struct cmsghdr *rcmsgp;
 227   u_char cmsgbuf[CMSG_SPACE(sizeof (struct in6_pktinfo))];
 228   struct in6_pktinfo *pktinfo;
 229   struct sockaddr_in6 src_sin6;
 230
 231   rcmsgp = (struct cmsghdr *)cmsgbuf;
 232   pktinfo = (struct in6_pktinfo *)(CMSG_DATA(rcmsgp));
 233   memset (&src_sin6, 0, sizeof (struct sockaddr_in6));
 234
 235   /* receive control msg */
 236   rcmsgp->cmsg_level = IPPROTO_IPV6;
 237   rcmsgp->cmsg_type = IPV6_PKTINFO;
 238   rcmsgp->cmsg_len = CMSG_LEN (sizeof (struct in6_pktinfo));
 239   /* rcmsgp = CMSG_NXTHDR (&rmsghdr, rcmsgp); */
 240
 241   /* receive msg hdr */
 242   memset (&rmsghdr, 0, sizeof (rmsghdr));
 243   rmsghdr.msg_iov = message;
 244   rmsghdr.msg_iovlen = iov_count (message);
 245   rmsghdr.msg_name = (caddr_t) &src_sin6;
 246   rmsghdr.msg_namelen = sizeof (struct sockaddr_in6);
 247   rmsghdr.msg_control = (caddr_t) cmsgbuf;
 248   rmsghdr.msg_controllen = sizeof (cmsgbuf);
 249
 250   retval = recvmsg (ospf6_sock, &rmsghdr, 0);
 251   if (retval < 0)
 252     zlog_warn ("recvmsg failed: %s", safe_strerror (errno));
 253   else if (retval == iov_totallen (message))
 254     zlog_warn ("recvmsg read full buffer size: %d", retval);
 255
 256   /* source address */
 257   assert (src);
 258   memcpy (src, &src_sin6.sin6_addr, sizeof (struct in6_addr));
 259
 260   /* destination address */
 261   if (ifindex)
 262     *ifindex = pktinfo->ipi6_ifindex;
 263   if (dst)
 264     memcpy (dst, &pktinfo->ipi6_addr, sizeof (struct in6_addr));
 265
 266   return retval;
 267 }
 268
 269