Annotation of embedaddon/quagga/pimd/pim_sock.c, revision 1.1
1.1 ! misho 1: /*
! 2: PIM for Quagga
! 3: Copyright (C) 2008 Everton da Silva Marques
! 4:
! 5: This program is free software; you can redistribute it and/or modify
! 6: it under the terms of the GNU General Public License as published by
! 7: the Free Software Foundation; either version 2 of the License, or
! 8: (at your option) any later version.
! 9:
! 10: This program is distributed in the hope that it will be useful, but
! 11: WITHOUT ANY WARRANTY; without even the implied warranty of
! 12: MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
! 13: General Public License for more details.
! 14:
! 15: You should have received a copy of the GNU General Public License
! 16: along with this program; see the file COPYING; if not, write to the
! 17: Free Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston,
! 18: MA 02110-1301 USA
! 19:
! 20: $QuaggaId: $Format:%an, %ai, %h$ $
! 21: */
! 22:
! 23: #include <zebra.h>
! 24: #include "pim_mroute.h"
! 25:
! 26: #include <sys/types.h>
! 27: #include <sys/socket.h>
! 28: #include <netinet/in.h>
! 29: #include <netinet/igmp.h>
! 30: #include <arpa/inet.h>
! 31: #include <unistd.h>
! 32: #include <netdb.h>
! 33: #include <errno.h>
! 34:
! 35: #include "log.h"
! 36: #include "privs.h"
! 37:
! 38: #include "pimd.h"
! 39: #include "pim_sock.h"
! 40: #include "pim_str.h"
! 41: #include "pim_igmp_join.h"
! 42:
! 43: /* GLOBAL VARS */
! 44: extern struct zebra_privs_t pimd_privs;
! 45:
! 46: int pim_socket_raw(int protocol)
! 47: {
! 48: int fd;
! 49:
! 50: if ( pimd_privs.change (ZPRIVS_RAISE) )
! 51: zlog_err ("pim_sockek_raw: could not raise privs, %s",
! 52: safe_strerror (errno) );
! 53:
! 54: fd = socket(AF_INET, SOCK_RAW, protocol);
! 55:
! 56: if ( pimd_privs.change (ZPRIVS_LOWER) )
! 57: zlog_err ("pim_socket_raw: could not lower privs, %s",
! 58: safe_strerror (errno) );
! 59:
! 60: if (fd < 0) {
! 61: zlog_warn("Could not create raw socket: errno=%d: %s",
! 62: errno, safe_strerror(errno));
! 63: return PIM_SOCK_ERR_SOCKET;
! 64: }
! 65:
! 66: return fd;
! 67: }
! 68:
! 69: int pim_socket_mcast(int protocol, struct in_addr ifaddr, int loop)
! 70: {
! 71: int fd;
! 72:
! 73: fd = pim_socket_raw(protocol);
! 74: if (fd < 0) {
! 75: zlog_warn("Could not create multicast socket: errno=%d: %s",
! 76: errno, safe_strerror(errno));
! 77: return PIM_SOCK_ERR_SOCKET;
! 78: }
! 79:
! 80: /* Needed to obtain destination address from recvmsg() */
! 81: {
! 82: #if defined(HAVE_IP_PKTINFO)
! 83: /* Linux and Solaris IP_PKTINFO */
! 84: int opt = 1;
! 85: if (setsockopt(fd, IPPROTO_IP, IP_PKTINFO, &opt, sizeof(opt))) {
! 86: zlog_warn("Could not set IP_PKTINFO on socket fd=%d: errno=%d: %s",
! 87: fd, errno, safe_strerror(errno));
! 88: }
! 89: #elif defined(HAVE_IP_RECVDSTADDR)
! 90: /* BSD IP_RECVDSTADDR */
! 91: int opt = 1;
! 92: if (setsockopt(fd, IPPROTO_IP, IP_RECVDSTADDR, &opt, sizeof(opt))) {
! 93: zlog_warn("Could not set IP_RECVDSTADDR on socket fd=%d: errno=%d: %s",
! 94: fd, errno, safe_strerror(errno));
! 95: }
! 96: #else
! 97: zlog_err("%s %s: Missing IP_PKTINFO and IP_RECVDSTADDR: unable to get dst addr from recvmsg()",
! 98: __FILE__, __PRETTY_FUNCTION__);
! 99: close(fd);
! 100: return PIM_SOCK_ERR_DSTADDR;
! 101: #endif
! 102: }
! 103:
! 104:
! 105: /* Set router alert (RFC 2113) for all IGMP messages (RFC 3376 4. Message Formats)*/
! 106: if (protocol == IPPROTO_IGMP) {
! 107: char ra[4];
! 108: ra[0] = 148;
! 109: ra[1] = 4;
! 110: ra[2] = 0;
! 111: ra[3] = 0;
! 112: if (setsockopt(fd, IPPROTO_IP, IP_OPTIONS, ra, 4)) {
! 113: zlog_warn("Could not set Router Alert Option on socket fd=%d: errno=%d: %s",
! 114: fd, errno, safe_strerror(errno));
! 115: close(fd);
! 116: return PIM_SOCK_ERR_RA;
! 117: }
! 118: }
! 119:
! 120: {
! 121: int reuse = 1;
! 122: if (setsockopt(fd, SOL_SOCKET, SO_REUSEADDR,
! 123: (void *) &reuse, sizeof(reuse))) {
! 124: zlog_warn("Could not set Reuse Address Option on socket fd=%d: errno=%d: %s",
! 125: fd, errno, safe_strerror(errno));
! 126: close(fd);
! 127: return PIM_SOCK_ERR_REUSE;
! 128: }
! 129: }
! 130:
! 131: {
! 132: const int MTTL = 1;
! 133: int ttl = MTTL;
! 134: if (setsockopt(fd, IPPROTO_IP, IP_MULTICAST_TTL,
! 135: (void *) &ttl, sizeof(ttl))) {
! 136: zlog_warn("Could not set multicast TTL=%d on socket fd=%d: errno=%d: %s",
! 137: MTTL, fd, errno, safe_strerror(errno));
! 138: close(fd);
! 139: return PIM_SOCK_ERR_TTL;
! 140: }
! 141: }
! 142:
! 143: if (setsockopt(fd, IPPROTO_IP, IP_MULTICAST_LOOP,
! 144: (void *) &loop, sizeof(loop))) {
! 145: zlog_warn("Could not %s Multicast Loopback Option on socket fd=%d: errno=%d: %s",
! 146: loop ? "enable" : "disable",
! 147: fd, errno, safe_strerror(errno));
! 148: close(fd);
! 149: return PIM_SOCK_ERR_LOOP;
! 150: }
! 151:
! 152: if (setsockopt(fd, IPPROTO_IP, IP_MULTICAST_IF,
! 153: (void *) &ifaddr, sizeof(ifaddr))) {
! 154: zlog_warn("Could not set Outgoing Interface Option on socket fd=%d: errno=%d: %s",
! 155: fd, errno, safe_strerror(errno));
! 156: close(fd);
! 157: return PIM_SOCK_ERR_IFACE;
! 158: }
! 159:
! 160: {
! 161: long flags;
! 162:
! 163: flags = fcntl(fd, F_GETFL, 0);
! 164: if (flags < 0) {
! 165: zlog_warn("Could not get fcntl(F_GETFL,O_NONBLOCK) on socket fd=%d: errno=%d: %s",
! 166: fd, errno, safe_strerror(errno));
! 167: close(fd);
! 168: return PIM_SOCK_ERR_NONBLOCK_GETFL;
! 169: }
! 170:
! 171: if (fcntl(fd, F_SETFL, flags | O_NONBLOCK)) {
! 172: zlog_warn("Could not set fcntl(F_SETFL,O_NONBLOCK) on socket fd=%d: errno=%d: %s",
! 173: fd, errno, safe_strerror(errno));
! 174: close(fd);
! 175: return PIM_SOCK_ERR_NONBLOCK_SETFL;
! 176: }
! 177: }
! 178:
! 179: return fd;
! 180: }
! 181:
! 182: int pim_socket_join(int fd, struct in_addr group,
! 183: struct in_addr ifaddr, ifindex_t ifindex)
! 184: {
! 185: int ret;
! 186:
! 187: #ifdef HAVE_STRUCT_IP_MREQN_IMR_IFINDEX
! 188: struct ip_mreqn opt;
! 189: #else
! 190: struct ip_mreq opt;
! 191: #endif
! 192:
! 193: opt.imr_multiaddr = group;
! 194:
! 195: #ifdef HAVE_STRUCT_IP_MREQN_IMR_IFINDEX
! 196: opt.imr_address = ifaddr;
! 197: opt.imr_ifindex = ifindex;
! 198: #else
! 199: opt.imr_interface = ifaddr;
! 200: #endif
! 201:
! 202: ret = setsockopt(fd, IPPROTO_IP, IP_ADD_MEMBERSHIP, &opt, sizeof(opt));
! 203: if (ret) {
! 204: char group_str[100];
! 205: char ifaddr_str[100];
! 206: if (!inet_ntop(AF_INET, &group, group_str , sizeof(group_str)))
! 207: sprintf(group_str, "<group?>");
! 208: if (!inet_ntop(AF_INET, &ifaddr, ifaddr_str , sizeof(ifaddr_str)))
! 209: sprintf(ifaddr_str, "<ifaddr?>");
! 210:
! 211: zlog_err("Failure socket joining fd=%d group %s on interface address %s: errno=%d: %s",
! 212: fd, group_str, ifaddr_str, errno, safe_strerror(errno));
! 213: return ret;
! 214: }
! 215:
! 216: if (PIM_DEBUG_TRACE) {
! 217: char group_str[100];
! 218: char ifaddr_str[100];
! 219: if (!inet_ntop(AF_INET, &group, group_str , sizeof(group_str)))
! 220: sprintf(group_str, "<group?>");
! 221: if (!inet_ntop(AF_INET, &ifaddr, ifaddr_str , sizeof(ifaddr_str)))
! 222: sprintf(ifaddr_str, "<ifaddr?>");
! 223:
! 224: zlog_debug("Socket fd=%d joined group %s on interface address %s",
! 225: fd, group_str, ifaddr_str);
! 226: }
! 227:
! 228: return ret;
! 229: }
! 230:
! 231: int pim_socket_join_source(int fd, ifindex_t ifindex,
! 232: struct in_addr group_addr,
! 233: struct in_addr source_addr,
! 234: const char *ifname)
! 235: {
! 236: if (pim_igmp_join_source(fd, ifindex, group_addr, source_addr)) {
! 237: int e = errno;
! 238: char group_str[100];
! 239: char source_str[100];
! 240: pim_inet4_dump("<grp?>", group_addr, group_str, sizeof(group_str));
! 241: pim_inet4_dump("<src?>", source_addr, source_str, sizeof(source_str));
! 242: zlog_warn("%s: setsockopt(fd=%d) failure for IGMP group %s source %s ifindex %d on interface %s: errno=%d: %s",
! 243: __PRETTY_FUNCTION__,
! 244: fd, group_str, source_str, ifindex, ifname,
! 245: e, safe_strerror(e));
! 246: return -1;
! 247: }
! 248:
! 249: return 0;
! 250: }
! 251:
! 252: int pim_socket_recvfromto(int fd, uint8_t *buf, size_t len,
! 253: struct sockaddr_in *from, socklen_t *fromlen,
! 254: struct sockaddr_in *to, socklen_t *tolen,
! 255: ifindex_t *ifindex)
! 256: {
! 257: struct msghdr msgh;
! 258: struct cmsghdr *cmsg;
! 259: struct iovec iov;
! 260: char cbuf[1000];
! 261: int err;
! 262:
! 263: /*
! 264: * IP_PKTINFO / IP_RECVDSTADDR don't yield sin_port.
! 265: * Use getsockname() to get sin_port.
! 266: */
! 267: if (to) {
! 268: struct sockaddr_in si;
! 269: socklen_t si_len = sizeof(si);
! 270:
! 271: ((struct sockaddr_in *) to)->sin_family = AF_INET;
! 272:
! 273: if (pim_socket_getsockname(fd, (struct sockaddr *) &si, &si_len)) {
! 274: ((struct sockaddr_in *) to)->sin_port = ntohs(0);
! 275: ((struct sockaddr_in *) to)->sin_addr.s_addr = ntohl(0);
! 276: }
! 277: else {
! 278: ((struct sockaddr_in *) to)->sin_port = si.sin_port;
! 279: ((struct sockaddr_in *) to)->sin_addr = si.sin_addr;
! 280: }
! 281:
! 282: if (tolen)
! 283: *tolen = sizeof(si);
! 284: }
! 285:
! 286: memset(&msgh, 0, sizeof(struct msghdr));
! 287: iov.iov_base = buf;
! 288: iov.iov_len = len;
! 289: msgh.msg_control = cbuf;
! 290: msgh.msg_controllen = sizeof(cbuf);
! 291: msgh.msg_name = from;
! 292: msgh.msg_namelen = fromlen ? *fromlen : 0;
! 293: msgh.msg_iov = &iov;
! 294: msgh.msg_iovlen = 1;
! 295: msgh.msg_flags = 0;
! 296:
! 297: err = recvmsg(fd, &msgh, 0);
! 298: if (err < 0)
! 299: return err;
! 300:
! 301: if (fromlen)
! 302: *fromlen = msgh.msg_namelen;
! 303:
! 304: for (cmsg = CMSG_FIRSTHDR(&msgh);
! 305: cmsg != NULL;
! 306: cmsg = CMSG_NXTHDR(&msgh,cmsg)) {
! 307:
! 308: #ifdef HAVE_IP_PKTINFO
! 309: if ((cmsg->cmsg_level == IPPROTO_IP) && (cmsg->cmsg_type == IP_PKTINFO)) {
! 310: struct in_pktinfo *i = (struct in_pktinfo *) CMSG_DATA(cmsg);
! 311: if (to)
! 312: ((struct sockaddr_in *) to)->sin_addr = i->ipi_addr;
! 313: if (tolen)
! 314: *tolen = sizeof(struct sockaddr_in);
! 315: if (ifindex)
! 316: *ifindex = i->ipi_ifindex;
! 317:
! 318: if (to && PIM_DEBUG_PACKETS) {
! 319: char to_str[100];
! 320: pim_inet4_dump("<to?>", to->sin_addr, to_str, sizeof(to_str));
! 321: zlog_debug("%s: HAVE_IP_PKTINFO to=%s,%d",
! 322: __PRETTY_FUNCTION__,
! 323: to_str, ntohs(to->sin_port));
! 324: }
! 325:
! 326: break;
! 327: }
! 328: #endif
! 329:
! 330: #ifdef HAVE_IP_RECVDSTADDR
! 331: if ((cmsg->cmsg_level == IPPROTO_IP) && (cmsg->cmsg_type == IP_RECVDSTADDR)) {
! 332: struct in_addr *i = (struct in_addr *) CMSG_DATA(cmsg);
! 333: if (to)
! 334: ((struct sockaddr_in *) to)->sin_addr = *i;
! 335: if (tolen)
! 336: *tolen = sizeof(struct sockaddr_in);
! 337:
! 338: if (to && PIM_DEBUG_PACKETS) {
! 339: char to_str[100];
! 340: pim_inet4_dump("<to?>", to->sin_addr, to_str, sizeof(to_str));
! 341: zlog_debug("%s: HAVE_IP_RECVDSTADDR to=%s,%d",
! 342: __PRETTY_FUNCTION__,
! 343: to_str, ntohs(to->sin_port));
! 344: }
! 345:
! 346: break;
! 347: }
! 348: #endif
! 349:
! 350: #if defined(HAVE_IP_RECVIF) && defined(CMSG_IFINDEX)
! 351: if (cmsg->cmsg_type == IP_RECVIF)
! 352: if (ifindex)
! 353: *ifindex = CMSG_IFINDEX(cmsg);
! 354: #endif
! 355:
! 356: } /* for (cmsg) */
! 357:
! 358: return err; /* len */
! 359: }
! 360:
! 361: int pim_socket_mcastloop_get(int fd)
! 362: {
! 363: int loop;
! 364: socklen_t loop_len = sizeof(loop);
! 365:
! 366: if (getsockopt(fd, IPPROTO_IP, IP_MULTICAST_LOOP,
! 367: &loop, &loop_len)) {
! 368: int e = errno;
! 369: zlog_warn("Could not get Multicast Loopback Option on socket fd=%d: errno=%d: %s",
! 370: fd, errno, safe_strerror(errno));
! 371: errno = e;
! 372: return PIM_SOCK_ERR_LOOP;
! 373: }
! 374:
! 375: return loop;
! 376: }
! 377:
! 378: int pim_socket_getsockname(int fd, struct sockaddr *name, socklen_t *namelen)
! 379: {
! 380: if (getsockname(fd, name, namelen)) {
! 381: int e = errno;
! 382: zlog_warn("Could not get Socket Name for socket fd=%d: errno=%d: %s",
! 383: fd, errno, safe_strerror(errno));
! 384: errno = e;
! 385: return PIM_SOCK_ERR_NAME;
! 386: }
! 387:
! 388: return PIM_SOCK_ERR_NONE;
! 389: }
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>
![[BACK]](/icons/cvsweb/back.gif){kind=icon}
Return to pim_sock.c CVS log ![[TXT]](/icons/cvsweb/text.gif){kind=icon}
![[DIR]](/icons/cvsweb/dir.gif){kind=icon}
Up to [ELWIX - Embedded LightWeight unIX -] / embedaddon / quagga / pimd