File:  [ELWIX - Embedded LightWeight unIX -] / embedaddon / pimdd / kern.c
Revision 1.1.1.1 (vendor branch): download - view: text, annotated - select for diffs - revision graph
Mon Jun 12 07:58:55 2017 UTC (7 years ago) by misho
Branches: pimdd, MAIN
CVS tags: v0_2_1p0, v0_2_1, HEAD
pimdd-dense 0.2.1.0_2

    1: /*
    2:  *  Copyright (c) 1998 by the University of Southern California.
    3:  *  All rights reserved.
    4:  *
    5:  *  Permission to use, copy, modify, and distribute this software and
    6:  *  its documentation in source and binary forms for lawful
    7:  *  purposes and without fee is hereby granted, provided
    8:  *  that the above copyright notice appear in all copies and that both
    9:  *  the copyright notice and this permission notice appear in supporting
   10:  *  documentation, and that any documentation, advertising materials,
   11:  *  and other materials related to such distribution and use acknowledge
   12:  *  that the software was developed by the University of Southern
   13:  *  California and/or Information Sciences Institute.
   14:  *  The name of the University of Southern California may not
   15:  *  be used to endorse or promote products derived from this software
   16:  *  without specific prior written permission.
   17:  *
   18:  *  THE UNIVERSITY OF SOUTHERN CALIFORNIA DOES NOT MAKE ANY REPRESENTATIONS
   19:  *  ABOUT THE SUITABILITY OF THIS SOFTWARE FOR ANY PURPOSE.  THIS SOFTWARE IS
   20:  *  PROVIDED "AS IS" AND WITHOUT ANY EXPRESS OR IMPLIED WARRANTIES,
   21:  *  INCLUDING, WITHOUT LIMITATION, THE IMPLIED WARRANTIES OF
   22:  *  MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE, TITLE, AND 
   23:  *  NON-INFRINGEMENT.
   24:  *
   25:  *  IN NO EVENT SHALL USC, OR ANY OTHER CONTRIBUTOR BE LIABLE FOR ANY
   26:  *  SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES, WHETHER IN CONTRACT,
   27:  *  TORT, OR OTHER FORM OF ACTION, ARISING OUT OF OR IN CONNECTION WITH,
   28:  *  THE USE OR PERFORMANCE OF THIS SOFTWARE.
   29:  *
   30:  *  Other copyrights might apply to parts of this software and are so
   31:  *  noted when applicable.
   32:  */
   33: /*
   34:  *  Questions concerning this software should be directed to 
   35:  *  Pavlin Ivanov Radoslavov (pavlin@catarina.usc.edu)
   36:  *
   37:  *  $Id: kern.c,v 1.1.1.1 2017/06/12 07:58:55 misho Exp $
   38:  */
   39: /*
   40:  * Part of this program has been derived from mrouted.
   41:  * The mrouted program is covered by the license in the accompanying file
   42:  * named "LICENSE.mrouted".
   43:  *
   44:  * The mrouted program is COPYRIGHT 1989 by The Board of Trustees of
   45:  * Leland Stanford Junior University.
   46:  *
   47:  */
   48: 
   49: #include "defs.h"
   50: 
   51: #ifdef RAW_OUTPUT_IS_RAW
   52: int curttl = 0;
   53: #endif
   54: 
   55: 
   56: /*
   57:  * Open/init the multicast routing in the kernel and sets the MRT_ASSERT
   58:  * flag in the kernel.
   59:  * 
   60:  */
   61: void
   62: k_init_pim(socket)
   63:   int socket;
   64: {
   65:     int v = 1;
   66:     
   67:     if (setsockopt(socket, IPPROTO_IP, MRT_INIT, (char *)&v, sizeof(int)) < 0)
   68: 	log(LOG_ERR, errno, "cannot enable multicast routing in kernel");
   69:     
   70:     if(setsockopt(socket, IPPROTO_IP, MRT_ASSERT, (char *)&v, sizeof(int)) < 0)
   71: 	log(LOG_ERR, errno, "cannot set ASSERT flag in kernel");
   72: }
   73: 
   74: 
   75: /*
   76:  * Stops the multicast routing in the kernel and resets the MRT_ASSERT
   77:  * flag in the kernel.
   78:  */
   79: void
   80: k_stop_pim(socket)
   81:     int socket;
   82: {
   83:     int v = 0;
   84:     
   85:     if (setsockopt(socket, IPPROTO_IP, MRT_DONE, (char *)NULL, 0) < 0)
   86: 	log(LOG_ERR, errno, "cannot disable multicast routing in kernel");
   87:     
   88:     if(setsockopt(socket, IPPROTO_IP, MRT_ASSERT, (char *)&v, sizeof(int)) < 0)
   89: 	log(LOG_ERR, errno, "cannot reset ASSERT flag in kernel");
   90: }
   91: 
   92: 
   93: /*
   94:  * Set the socket receiving buffer. `bufsize` is the preferred size,
   95:  * `minsize` is the smallest acceptable size.
   96:  */
   97: void k_set_rcvbuf(socket, bufsize, minsize)
   98:     int socket;
   99:     int bufsize;
  100:     int minsize;
  101: {
  102:     int delta = bufsize / 2;
  103:     int iter = 0;
  104:     
  105:     /*
  106:      * Set the socket buffer.  If we can't set it as large as we
  107:      * want, search around to try to find the highest acceptable
  108:      * value.  The highest acceptable value being smaller than
  109:      * minsize is a fatal error.
  110:      */
  111:     if (setsockopt(socket, SOL_SOCKET, SO_RCVBUF,
  112: 		   (char *)&bufsize, sizeof(bufsize)) < 0) {
  113: 	bufsize -= delta;
  114: 	while (1) {
  115: 	    iter++;
  116: 	    if (delta > 1)
  117: 	      delta /= 2;
  118: 	    
  119: 	    if (setsockopt(socket, SOL_SOCKET, SO_RCVBUF,
  120: 			   (char *)&bufsize, sizeof(bufsize)) < 0) {
  121: 		bufsize -= delta;
  122: 	    } else {
  123: 		if (delta < 1024)
  124: 		    break;
  125: 		bufsize += delta;
  126: 	    }
  127: 	}
  128: 	if (bufsize < minsize) {
  129: 	    log(LOG_ERR, 0, "OS-allowed buffer size %u < app min %u",
  130: 		bufsize, minsize);
  131: 	    /*NOTREACHED*/
  132: 	}
  133:     }
  134:     IF_DEBUG(DEBUG_KERN)
  135: 	log(LOG_DEBUG, 0, "Got %d byte buffer size in %d iterations",
  136: 	    bufsize, iter);
  137: }
  138: 
  139: 
  140: /*
  141:  * Set/reset the IP_HDRINCL option. My guess is we don't need it for raw
  142:  * sockets, but having it here won't hurt. Well, unless you are running
  143:  * an older version of FreeBSD (older than 2.2.2). If the multicast
  144:  * raw packet is bigger than 208 bytes, then IP_HDRINCL triggers a bug
  145:  * in the kernel and "panic". The kernel patch for netinet/ip_raw.c
  146:  * coming with this distribution fixes it.
  147:  */
  148: void k_hdr_include(socket, bool)
  149:     int socket;
  150:     int bool;
  151: {
  152: #ifdef IP_HDRINCL
  153:     if (setsockopt(socket, IPPROTO_IP, IP_HDRINCL,
  154: 		   (char *)&bool, sizeof(bool)) < 0)
  155: 	log(LOG_ERR, errno, "setsockopt IP_HDRINCL %u", bool);
  156: #endif
  157: }
  158: 
  159: 
  160: /*
  161:  * Set the default TTL for the multicast packets outgoing from this
  162:  * socket.
  163:  * TODO: Does it affect the unicast packets?
  164:  */
  165: void k_set_ttl(socket, t)
  166:     int socket;
  167:     int t;
  168: {
  169: #ifdef RAW_OUTPUT_IS_RAW
  170:     curttl = t;
  171: #else
  172:     u_char ttl;
  173:     
  174:     ttl = t;
  175:     if (setsockopt(socket, IPPROTO_IP, IP_MULTICAST_TTL,
  176: 		   (char *)&ttl, sizeof(ttl)) < 0)
  177: 	log(LOG_ERR, errno, "setsockopt IP_MULTICAST_TTL %u", ttl);
  178: #endif
  179: }
  180: 
  181: 
  182: /*
  183:  * Set/reset the IP_MULTICAST_LOOP. Set/reset is specified by "flag".
  184:  */
  185: void k_set_loop(socket, flag)
  186:     int socket;
  187:     int flag;
  188: {
  189:     u_char loop;
  190: 
  191:     loop = flag;
  192:     if (setsockopt(socket, IPPROTO_IP, IP_MULTICAST_LOOP,
  193: 		   (char *)&loop, sizeof(loop)) < 0)
  194: 	log(LOG_ERR, errno, "setsockopt IP_MULTICAST_LOOP %u", loop);
  195: }
  196: 
  197: 
  198: /*
  199:  * Set the IP_MULTICAST_IF option on local interface ifa.
  200:  */
  201: void k_set_if(socket, ifa)
  202:     int socket;
  203:     u_int32 ifa;
  204: {
  205:     struct in_addr adr;
  206: 
  207:     adr.s_addr = ifa;
  208:     if (setsockopt(socket, IPPROTO_IP, IP_MULTICAST_IF,
  209: 		   (char *)&adr, sizeof(adr)) < 0)
  210: 	log(LOG_ERR, errno, "setsockopt IP_MULTICAST_IF %s",
  211: 	    inet_fmt(ifa, s1));
  212: }
  213: 
  214: 
  215: /*
  216:  * Join a multicast grp group on local interface ifa.
  217:  */
  218: void k_join(socket, grp, ifa)
  219:     int socket;
  220:     u_int32 grp;
  221:     u_int32 ifa;
  222: {
  223:     struct ip_mreq mreq;
  224:     
  225:     mreq.imr_multiaddr.s_addr = grp;
  226:     mreq.imr_interface.s_addr = ifa;
  227: 
  228:     if (setsockopt(socket, IPPROTO_IP, IP_ADD_MEMBERSHIP,
  229: 		   (char *)&mreq, sizeof(mreq)) < 0)
  230: 	log(LOG_WARNING, errno, "cannot join group %s on interface %s",
  231: 	    inet_fmt(grp, s1), inet_fmt(ifa, s2));
  232: }
  233: 
  234: 
  235: /*
  236:  * Leave a multicats grp group on local interface ifa.
  237:  */
  238: void k_leave(socket, grp, ifa)
  239:     int socket;
  240:     u_int32 grp;
  241:     u_int32 ifa;
  242: {
  243:     struct ip_mreq mreq;
  244: 
  245:     mreq.imr_multiaddr.s_addr = grp;
  246:     mreq.imr_interface.s_addr = ifa;
  247:     
  248:     if (setsockopt(socket, IPPROTO_IP, IP_DROP_MEMBERSHIP,
  249: 		   (char *)&mreq, sizeof(mreq)) < 0)
  250: 	log(LOG_WARNING, errno, "cannot leave group %s on interface %s",
  251: 	    inet_fmt(grp, s1), inet_fmt(ifa, s2));
  252: }
  253: 
  254: 
  255: /*
  256:  * Add a virtual interface in the kernel.
  257:  */
  258: void k_add_vif(socket, vifi, v)
  259:     int socket;
  260:     vifi_t vifi;
  261:     struct uvif *v;
  262: {
  263:     struct vifctl vc;
  264: 
  265:     vc.vifc_vifi            = vifi;
  266:     /* TODO: only for DVMRP tunnels?
  267:     vc.vifc_flags           = v->uv_flags & VIFF_KERNEL_FLAGS;
  268:     */
  269:     vc.vifc_flags           = v->uv_flags;
  270:     vc.vifc_threshold       = v->uv_threshold;
  271:     vc.vifc_rate_limit	    = v->uv_rate_limit;
  272:     vc.vifc_lcl_addr.s_addr = v->uv_lcl_addr;
  273:     vc.vifc_rmt_addr.s_addr = v->uv_rmt_addr;
  274: 
  275:     if (setsockopt(socket, IPPROTO_IP, MRT_ADD_VIF,
  276: 		   (char *)&vc, sizeof(vc)) < 0)
  277: 	log(LOG_ERR, errno, "setsockopt MRT_ADD_VIF on vif %d", vifi);
  278: }
  279: 
  280: 
  281: /*
  282:  * Delete a virtual interface in the kernel.
  283:  */
  284: void k_del_vif(socket, vifi)
  285:     int socket;
  286:     vifi_t vifi;
  287: {
  288:     if (setsockopt(socket, IPPROTO_IP, MRT_DEL_VIF,
  289: 		   (char *)&vifi, sizeof(vifi)) < 0)
  290: 	log(LOG_ERR, errno, "setsockopt MRT_DEL_VIF on vif %d", vifi);
  291: }
  292: 
  293: 
  294: /*
  295:  * Delete all MFC entries for particular routing entry from the kernel.
  296:  */
  297: int
  298: k_del_mfc(socket, source, group)
  299:     int socket;
  300:     u_int32 source;
  301:     u_int32 group;
  302: {
  303:     struct mfcctl mc;
  304: 
  305:     mc.mfcc_origin.s_addr   = source;
  306:     mc.mfcc_mcastgrp.s_addr = group;
  307: 	
  308:     if (setsockopt(socket, IPPROTO_IP, MRT_DEL_MFC, (char *)&mc,
  309: 		   sizeof(mc)) < 0) {
  310: 	log(LOG_WARNING, errno, "setsockopt k_del_mfc");
  311: 	return FALSE;
  312:     }
  313: 	
  314:     IF_DEBUG(DEBUG_MFC)
  315: 	log(LOG_DEBUG, 0, "Deleted MFC entry: src %s, grp %s",
  316: 	    inet_fmt(mc.mfcc_origin.s_addr, s1),
  317: 	    inet_fmt(mc.mfcc_mcastgrp.s_addr, s2));
  318: 
  319:     return(TRUE);
  320: }
  321: 
  322: 
  323: /*
  324:  * Install/modify a MFC entry in the kernel
  325:  */
  326: int
  327: k_chg_mfc(socket, source, group, iif, oifs)
  328:     int socket;
  329:     u_int32 source;
  330:     u_int32 group;
  331:     vifi_t iif;
  332:     vifbitmap_t oifs;
  333: {
  334:     struct mfcctl mc;
  335:     vifi_t vifi;
  336:     struct uvif *v;
  337: 
  338:     mc.mfcc_origin.s_addr = source;
  339: #ifdef OLD_KERNEL
  340:     mc.mfcc_originmas.s_addr = 0xffffffff;    /* Got it from mrouted-3.9 */
  341: #endif /* OLD_KERNEL */
  342:     mc.mfcc_mcastgrp.s_addr = group;
  343:     mc.mfcc_parent = iif;
  344:     for (vifi = 0, v = uvifs; vifi < numvifs; vifi++, v++) {
  345: 	if (VIFM_ISSET(vifi, oifs))
  346: 	    mc.mfcc_ttls[vifi] = v->uv_threshold;
  347: 	else
  348: 	    mc.mfcc_ttls[vifi] = 0;
  349:     }
  350:     
  351:     if (setsockopt(socket, IPPROTO_IP, MRT_ADD_MFC, (char *)&mc,
  352:                    sizeof(mc)) < 0) {
  353:         log(LOG_WARNING, errno,
  354: 	    "setsockopt MRT_ADD_MFC for source %s and group %s",
  355: 	    inet_fmt(source, s1), inet_fmt(group, s2));
  356:         return(FALSE);
  357:     }
  358:     return(TRUE);
  359: }
  360: 
  361: 
  362: /*
  363:  * Get packet counters for particular interface
  364:  */
  365: /*
  366:  * XXX: TODO: currently not used, but keep just in case we need it later.
  367:  */
  368: int k_get_vif_count(vifi, retval)
  369:     vifi_t vifi;
  370:     struct vif_count *retval;
  371: {
  372:     struct sioc_vif_req vreq;
  373:     
  374:     vreq.vifi = vifi;
  375:     if (ioctl(udp_socket, SIOCGETVIFCNT, (char *)&vreq) < 0) {
  376: 	log(LOG_WARNING, errno, "SIOCGETVIFCNT on vif %d", vifi);
  377: 	retval->icount = retval->ocount = retval->ibytes =
  378: 	    retval->obytes = 0xffffffff;
  379: 	return (1);
  380:     }
  381:     retval->icount = vreq.icount;
  382:     retval->ocount = vreq.ocount;
  383:     retval->ibytes = vreq.ibytes;
  384:     retval->obytes = vreq.obytes;
  385:     return (0);
  386: }
  387: 
  388: 
  389: /*
  390:  * Gets the number of packets, bytes, and number op packets arrived
  391:  * on wrong if in the kernel for particular (S,G) entry.
  392:  */
  393: int
  394: k_get_sg_cnt(socket, source, group, retval)
  395:     int socket;    /* udp_socket */
  396:     u_int32 source;
  397:     u_int32 group;
  398:     struct sg_count *retval;
  399: {
  400:     struct sioc_sg_req sgreq;
  401:     
  402:     sgreq.src.s_addr = source;
  403:     sgreq.grp.s_addr = group;
  404:     if ((ioctl(socket, SIOCGETSGCNT, (char *)&sgreq) < 0)
  405: 	|| (sgreq.wrong_if == 0xffffffff)) {
  406: 	/* XXX: ipmulti-3.5 has bug in ip_mroute.c, get_sg_cnt():
  407: 	 * the return code is always 0, so this is why we need to check
  408: 	 * the wrong_if value.
  409: 	 */
  410: 	log(LOG_WARNING, errno, "SIOCGETSGCNT on (%s %s)",
  411: 	    inet_fmt(source, s1), inet_fmt(group, s2));
  412: 	retval->pktcnt = retval->bytecnt = retval->wrong_if = ~0;
  413: 	return(1);
  414:     }
  415:     retval->pktcnt = sgreq.pktcnt;
  416:     retval->bytecnt = sgreq.bytecnt;
  417:     retval->wrong_if = sgreq.wrong_if;
  418:     return(0);
  419: }
  420: 
  421: 
  422: 

FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>