1: /*
2: * Copyright (c) 1998-2001
3: * University of Southern California/Information Sciences Institute.
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. Neither the name of the project nor the names of its contributors
15: * may be used to endorse or promote products derived from this software
16: * without specific prior written permission.
17: *
18: * THIS SOFTWARE IS PROVIDED BY THE PROJECT AND CONTRIBUTORS ``AS IS'' AND
19: * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
20: * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
21: * ARE DISCLAIMED. IN NO EVENT SHALL THE PROJECT OR CONTRIBUTORS BE LIABLE
22: * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
23: * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
24: * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
25: * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
26: * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
27: * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
28: * SUCH DAMAGE.
29: */
30: /*
31: * Part of this program has been derived from mrouted.
32: * The mrouted program is covered by the license in the accompanying file
33: * named "LICENSE.mrouted".
34: *
35: * The mrouted program is COPYRIGHT 1989 by The Board of Trustees of
36: * Leland Stanford Junior University.
37: *
38: */
39:
40: #include "defs.h"
41:
42: #ifdef RAW_OUTPUT_IS_RAW
43: int curttl = 0;
44: #endif
45:
46: /*
47: * XXX: in Some BSD's there is only MRT_ASSERT, but in Linux there are
48: * both MRT_ASSERT and MRT_PIM
49: */
50: #ifndef MRT_PIM
51: #define MRT_PIM MRT_ASSERT
52: #endif
53:
54: #ifdef __linux__ /* Currently only available on Linux */
55: # ifndef MRT_TABLE
56: # define MRT_TABLE (MRT_BASE + 9)
57: # endif
58: #endif
59:
60: /*
61: * Open/init the multicast routing in the kernel and sets the
62: * MRT_PIM (aka MRT_ASSERT) flag in the kernel.
63: */
64: void k_init_pim(int socket)
65: {
66: int v = 1;
67:
68: #ifdef MRT_TABLE /* Currently only available on Linux */
69: if (mrt_table_id != 0) {
70: logit(LOG_INFO, 0, "Initializing multicast routing table id %u", mrt_table_id);
71: if (setsockopt(socket, IPPROTO_IP, MRT_TABLE, &mrt_table_id, sizeof(mrt_table_id)) < 0) {
72: logit(LOG_WARNING, errno, "Cannot set multicast routing table id");
73: logit(LOG_ERR, 0, "Make sure your kernel has CONFIG_IP_MROUTE_MULTIPLE_TABLES=y");
74: }
75: }
76: #endif
77:
78: if (setsockopt(socket, IPPROTO_IP, MRT_INIT, (char *)&v, sizeof(int)) < 0) {
79: if (errno == EADDRINUSE)
80: logit(LOG_ERR, 0, "Another multicast routing application is already running.");
81: else
82: logit(LOG_ERR, errno, "Cannot enable multicast routing in kernel");
83: }
84:
85: if (setsockopt(socket, IPPROTO_IP, MRT_PIM, (char *)&v, sizeof(int)) < 0)
86: logit(LOG_ERR, errno, "Cannot set PIM flag in kernel");
87: }
88:
89:
90: /*
91: * Stops the multicast routing in the kernel and resets the
92: * MRT_PIM (aka MRT_ASSERT) flag in the kernel.
93: */
94: void k_stop_pim(int socket)
95: {
96: int v = 0;
97:
98: if (setsockopt(socket, IPPROTO_IP, MRT_PIM, (char *)&v, sizeof(int)) < 0)
99: logit(LOG_ERR, errno, "Cannot reset PIM flag in kernel");
100:
101: if (setsockopt(socket, IPPROTO_IP, MRT_DONE, (char *)NULL, 0) < 0)
102: logit(LOG_ERR, errno, "Cannot disable multicast routing in kernel");
103: }
104:
105:
106: /*
107: * Set the socket sending buffer. `bufsize` is the preferred size,
108: * `minsize` is the smallest acceptable size.
109: */
110: void k_set_sndbuf(int socket, int bufsize, int minsize)
111: {
112: int delta = bufsize / 2;
113: int iter = 0;
114:
115: /*
116: * Set the socket buffer. If we can't set it as large as we
117: * want, search around to try to find the highest acceptable
118: * value. The highest acceptable value being smaller than
119: * minsize is a fatal error.
120: */
121: if (setsockopt(socket, SOL_SOCKET, SO_SNDBUF, (char *)&bufsize, sizeof(bufsize)) < 0) {
122: bufsize -= delta;
123: while (1) {
124: iter++;
125: if (delta > 1)
126: delta /= 2;
127:
128: if (setsockopt(socket, SOL_SOCKET, SO_SNDBUF, (char *)&bufsize, sizeof(bufsize)) < 0) {
129: bufsize -= delta;
130: } else {
131: if (delta < 1024)
132: break;
133: bufsize += delta;
134: }
135: }
136: if (bufsize < minsize) {
137: logit(LOG_ERR, 0, "OS-allowed send buffer size %u < app min %u",
138: bufsize, minsize);
139: /*NOTREACHED*/
140: }
141: }
142:
143: IF_DEBUG(DEBUG_KERN) {
144: logit(LOG_DEBUG, 0, "Got %d byte send buffer size in %d iterations",
145: bufsize, iter);
146: }
147: }
148:
149:
150: /*
151: * Set the socket receiving buffer. `bufsize` is the preferred size,
152: * `minsize` is the smallest acceptable size.
153: */
154: void k_set_rcvbuf(int socket, int bufsize, int minsize)
155: {
156: int delta = bufsize / 2;
157: int iter = 0;
158:
159: /*
160: * Set the socket buffer. If we can't set it as large as we
161: * want, search around to try to find the highest acceptable
162: * value. The highest acceptable value being smaller than
163: * minsize is a fatal error.
164: */
165: if (setsockopt(socket, SOL_SOCKET, SO_RCVBUF, (char *)&bufsize, sizeof(bufsize)) < 0) {
166: bufsize -= delta;
167: while (1) {
168: iter++;
169: if (delta > 1)
170: delta /= 2;
171:
172: if (setsockopt(socket, SOL_SOCKET, SO_RCVBUF, (char *)&bufsize, sizeof(bufsize)) < 0) {
173: bufsize -= delta;
174: } else {
175: if (delta < 1024)
176: break;
177: bufsize += delta;
178: }
179: }
180:
181: if (bufsize < minsize) {
182: logit(LOG_ERR, 0, "OS-allowed recv buffer size %u < app min %u", bufsize, minsize);
183: /*NOTREACHED*/
184: }
185: }
186:
187: IF_DEBUG(DEBUG_KERN) {
188: logit(LOG_DEBUG, 0, "Got %d byte recv buffer size in %d iterations",
189: bufsize, iter);
190: }
191: }
192:
193:
194: /*
195: * Set/reset the IP_HDRINCL option. My guess is we don't need it for raw
196: * sockets, but having it here won't hurt. Well, unless you are running
197: * an older version of FreeBSD (older than 2.2.2). If the multicast
198: * raw packet is bigger than 208 bytes, then IP_HDRINCL triggers a bug
199: * in the kernel and "panic". The kernel patch for netinet/ip_raw.c
200: * coming with this distribution fixes it.
201: */
202: void k_hdr_include(int socket, int val)
203: {
204: #ifdef IP_HDRINCL
205: if (setsockopt(socket, IPPROTO_IP, IP_HDRINCL, (char *)&val, sizeof(val)) < 0)
206: logit(LOG_ERR, errno, "Failed %s IP_HDRINCL on socket %d",
207: ENABLINGSTR(val), socket);
208: #endif
209: }
210:
211:
212: /*
213: * Set the default TTL for the multicast packets outgoing from this
214: * socket.
215: * TODO: Does it affect the unicast packets?
216: */
217: void k_set_ttl(int socket __attribute__((unused)), int t)
218: {
219: #ifdef RAW_OUTPUT_IS_RAW
220: curttl = t;
221: #else
222: uint8_t ttl;
223:
224: ttl = t;
225: if (setsockopt(socket, IPPROTO_IP, IP_MULTICAST_TTL, (char *)&ttl, sizeof(ttl)) < 0)
226: logit(LOG_ERR, errno, "Failed setting IP_MULTICAST_TTL %u on socket %d", ttl, socket);
227: #endif
228: }
229:
230:
231: /*
232: * Set/reset the IP_MULTICAST_LOOP. Set/reset is specified by "flag".
233: */
234: void k_set_loop(int socket, int flag)
235: {
236: uint8_t loop;
237:
238: loop = flag;
239: if (setsockopt(socket, IPPROTO_IP, IP_MULTICAST_LOOP, (char *)&loop, sizeof(loop)) < 0)
240: logit(LOG_ERR, errno, "Failed %s IP_MULTICAST_LOOP on socket %d",
241: ENABLINGSTR(flag), socket);
242: }
243:
244:
245: /*
246: * Set the IP_MULTICAST_IF option on local interface ifa.
247: */
248: void k_set_if(int socket, uint32_t ifa)
249: {
250: struct in_addr adr;
251:
252: adr.s_addr = ifa;
253: if (setsockopt(socket, IPPROTO_IP, IP_MULTICAST_IF, (char *)&adr, sizeof(adr)) < 0) {
254: if (errno == EADDRNOTAVAIL || errno == EINVAL)
255: return;
256:
257: logit(LOG_ERR, errno, "Failed setting IP_MULTICAST_IF option on %s",
258: inet_fmt(adr.s_addr, s1, sizeof(s1)));
259: }
260: }
261:
262:
263: /*
264: * Set Router Alert IP option, RFC2113
265: */
266: void k_set_router_alert(int socket)
267: {
268: char router_alert[4]; /* Router Alert IP Option */
269:
270: router_alert[0] = IPOPT_RA; /* Router Alert */
271: router_alert[1] = 4; /* 4 bytes */
272: router_alert[2] = 0;
273: router_alert[3] = 0;
274:
275: if (setsockopt(socket, IPPROTO_IP, IP_OPTIONS, router_alert, sizeof(router_alert) )< 0)
276: logit(LOG_ERR, errno, "setsockopt IP_OPTIONS IPOPT_RA");
277: }
278:
279:
280: /*
281: * Join a multicast group on virtual interface 'v'.
282: */
283: void k_join(int socket, uint32_t grp, struct uvif *v)
284: {
285: #ifdef __linux__
286: struct ip_mreqn mreq;
287: #else
288: struct ip_mreq mreq;
289: #endif /* __linux__ */
290:
291: #ifdef __linux__
292: mreq.imr_ifindex = v->uv_ifindex;
293: mreq.imr_address.s_addr = v->uv_lcl_addr;
294: #else
295: mreq.imr_interface.s_addr = v->uv_lcl_addr;
296: #endif /* __linux__ */
297: mreq.imr_multiaddr.s_addr = grp;
298:
299: if (setsockopt(socket, IPPROTO_IP, IP_ADD_MEMBERSHIP,
300: (char *)&mreq, sizeof(mreq)) < 0) {
301: #ifdef __linux__
302: logit(LOG_WARNING, errno,
303: "Cannot join group %s on interface %s (ifindex %d)",
304: inet_fmt(grp, s1, sizeof(s1)), inet_fmt(v->uv_lcl_addr, s2, sizeof(s2)), v->uv_ifindex);
305: #else
306: logit(LOG_WARNING, errno,
307: "Cannot join group %s on interface %s",
308: inet_fmt(grp, s1, sizeof(s1)), inet_fmt(v->uv_lcl_addr, s2, sizeof(s2)));
309: #endif /* __linux__ */
310: }
311: }
312:
313:
314: /*
315: * Leave a multicast group on virtual interface 'v'.
316: */
317: void k_leave(int socket, uint32_t grp, struct uvif *v)
318: {
319: #ifdef __linux__
320: struct ip_mreqn mreq;
321: #else
322: struct ip_mreq mreq;
323: #endif /* __linux__ */
324:
325: #ifdef __linux__
326: mreq.imr_ifindex = v->uv_ifindex;
327: mreq.imr_address.s_addr = v->uv_lcl_addr;
328: #else
329: mreq.imr_interface.s_addr = v->uv_lcl_addr;
330: #endif /* __linux__ */
331: mreq.imr_multiaddr.s_addr = grp;
332:
333: if (setsockopt(socket, IPPROTO_IP, IP_DROP_MEMBERSHIP, (char *)&mreq, sizeof(mreq)) < 0) {
334: #ifdef __linux__
335: logit(LOG_WARNING, errno,
336: "Cannot leave group %s on interface %s (ifindex %d)",
337: inet_fmt(grp, s1, sizeof(s1)), inet_fmt(v->uv_lcl_addr, s2, sizeof(s2)), v->uv_ifindex);
338: #else
339: logit(LOG_WARNING, errno,
340: "Cannot leave group %s on interface %s",
341: inet_fmt(grp, s1, sizeof(s1)), inet_fmt(v->uv_lcl_addr, s2, sizeof(s2)));
342: #endif /* __linux__ */
343: }
344: }
345:
346: /*
347: * Fill struct vifctl using corresponding fields from struct uvif.
348: */
349: static void uvif_to_vifctl(struct vifctl *vc, struct uvif *v)
350: {
351: /* XXX: we don't support VIFF_TUNNEL; VIFF_SRCRT is obsolete */
352: vc->vifc_flags = 0;
353: if (v->uv_flags & VIFF_REGISTER)
354: vc->vifc_flags |= VIFF_REGISTER;
355: vc->vifc_threshold = v->uv_threshold;
356: vc->vifc_rate_limit = v->uv_rate_limit;
357: vc->vifc_lcl_addr.s_addr = v->uv_lcl_addr;
358: vc->vifc_rmt_addr.s_addr = v->uv_rmt_addr;
359: }
360:
361: /*
362: * Add a virtual interface in the kernel.
363: */
364: void k_add_vif(int socket, vifi_t vifi, struct uvif *v)
365: {
366: struct vifctl vc;
367:
368: vc.vifc_vifi = vifi;
369: uvif_to_vifctl(&vc, v);
370: if (setsockopt(socket, IPPROTO_IP, MRT_ADD_VIF, (char *)&vc, sizeof(vc)) < 0)
371: logit(LOG_ERR, errno, "Failed adding VIF %d (MRT_ADD_VIF) for iface %s",
372: vifi, v->uv_name);
373: }
374:
375:
376: /*
377: * Delete a virtual interface in the kernel.
378: */
379: void k_del_vif(int socket, vifi_t vifi, struct uvif *v __attribute__((unused)))
380: {
381: /*
382: * Unfortunately Linux MRT_DEL_VIF API differs a bit from the *BSD one. It
383: * expects to receive a pointer to struct vifctl that corresponds to the VIF
384: * we're going to delete. *BSD systems on the other hand exepect only the
385: * index of that VIF.
386: */
387: #ifdef __linux__
388: struct vifctl vc;
389:
390: vc.vifc_vifi = vifi;
391: uvif_to_vifctl(&vc, v); /* 'v' is used only on Linux systems. */
392:
393: if (setsockopt(socket, IPPROTO_IP, MRT_DEL_VIF, (char *)&vc, sizeof(vc)) < 0)
394: #else /* *BSD et al. */
395: if (setsockopt(socket, IPPROTO_IP, MRT_DEL_VIF, (char *)&vifi, sizeof(vifi)) < 0)
396: #endif /* !__linux__ */
397: {
398: if (errno == EADDRNOTAVAIL || errno == EINVAL)
399: return;
400:
401: logit(LOG_ERR, errno, "Failed removing VIF %d (MRT_DEL_VIF)", vifi);
402: }
403: }
404:
405:
406: /*
407: * Delete all MFC entries for particular routing entry from the kernel.
408: */
409: int k_del_mfc(int socket, uint32_t source, uint32_t group)
410: {
411: struct mfcctl mc;
412:
413: memset(&mc, 0, sizeof(mc));
414: mc.mfcc_origin.s_addr = source;
415: mc.mfcc_mcastgrp.s_addr = group;
416:
417: if (setsockopt(socket, IPPROTO_IP, MRT_DEL_MFC, (char *)&mc, sizeof(mc)) < 0) {
418: logit(LOG_WARNING, errno, "Failed removing MFC entry src %s, grp %s",
419: inet_fmt(mc.mfcc_origin.s_addr, s1, sizeof(s1)),
420: inet_fmt(mc.mfcc_mcastgrp.s_addr, s2, sizeof(s2)));
421:
422: return FALSE;
423: }
424:
425: logit(LOG_INFO, 0, "Removed MFC entry src %s, grp %s",
426: inet_fmt(mc.mfcc_origin.s_addr, s1, sizeof(s1)),
427: inet_fmt(mc.mfcc_mcastgrp.s_addr, s2, sizeof(s2)));
428:
429: return TRUE;
430: }
431:
432:
433: /*
434: * Install/modify a MFC entry in the kernel
435: */
436: int k_chg_mfc(int socket, uint32_t source, uint32_t group, vifi_t iif, vifbitmap_t oifs, uint32_t rp_addr __attribute__((unused)))
437: {
438: char input[IFNAMSIZ], output[MAXVIFS * (IFNAMSIZ + 2)] = "";
439: vifi_t vifi;
440: struct uvif *v;
441: struct mfcctl mc;
442:
443: memset(&mc, 0, sizeof(mc));
444: mc.mfcc_origin.s_addr = source;
445: mc.mfcc_mcastgrp.s_addr = group;
446: mc.mfcc_parent = iif;
447: /*
448: * draft-ietf-pim-sm-v2-new-05.txt section 4.2 mentions iif is removed
449: * at the packet forwarding phase
450: */
451: VIFM_CLR(mc.mfcc_parent, oifs);
452:
453: for (vifi = 0, v = uvifs; vifi < numvifs; vifi++, v++) {
454: if (VIFM_ISSET(vifi, oifs)) {
455: mc.mfcc_ttls[vifi] = v->uv_threshold;
456: if (output[0] != 0)
457: strlcat(output, ", ", sizeof(output));
458: strlcat(output, v->uv_name, sizeof(output));
459: } else {
460: mc.mfcc_ttls[vifi] = 0;
461: }
462: }
463: strlcpy(input, uvifs[iif].uv_name, sizeof(input));
464:
465: #ifdef PIM_REG_KERNEL_ENCAP
466: mc.mfcc_rp_addr.s_addr = rp_addr;
467: #endif
468: if (setsockopt(socket, IPPROTO_IP, MRT_ADD_MFC, (char *)&mc, sizeof(mc)) < 0) {
469: logit(LOG_WARNING, errno, "Failed adding MFC entry src %s grp %s from %s to %s",
470: inet_fmt(mc.mfcc_origin.s_addr, s1, sizeof(s1)),
471: inet_fmt(mc.mfcc_mcastgrp.s_addr, s2, sizeof(s2)),
472: input, output);
473:
474: return FALSE;
475: }
476:
477: logit(LOG_INFO, 0, "Added kernel MFC entry src %s grp %s from %s to %s",
478: inet_fmt(mc.mfcc_origin.s_addr, s1, sizeof(s1)),
479: inet_fmt(mc.mfcc_mcastgrp.s_addr, s2, sizeof(s2)),
480: input, output);
481:
482: return TRUE;
483: }
484:
485:
486: /*
487: * Get packet counters for particular interface
488: * XXX: TODO: currently not used, but keep just in case we need it later.
489: */
490: int k_get_vif_count(vifi_t vifi, struct vif_count *retval)
491: {
492: struct sioc_vif_req vreq;
493:
494: memset(&vreq, 0, sizeof(vreq));
495: vreq.vifi = vifi;
496: if (ioctl(udp_socket, SIOCGETVIFCNT, (char *)&vreq) < 0) {
497: logit(LOG_WARNING, errno, "Failed reading kernel packet count (SIOCGETVIFCNT) on vif %d", vifi);
498:
499: retval->icount =
500: retval->ocount =
501: retval->ibytes =
502: retval->obytes = 0xffffffff;
503:
504: return 1;
505: }
506:
507: retval->icount = vreq.icount;
508: retval->ocount = vreq.ocount;
509: retval->ibytes = vreq.ibytes;
510: retval->obytes = vreq.obytes;
511:
512: return 0;
513: }
514:
515:
516: /*
517: * Gets the number of packets, bytes, and number op packets arrived
518: * on wrong if in the kernel for particular (S,G) entry.
519: */
520: int k_get_sg_cnt(int socket, uint32_t source, uint32_t group, struct sg_count *retval)
521: {
522: struct sioc_sg_req sgreq;
523:
524: memset(&sgreq, 0, sizeof(sgreq));
525: sgreq.src.s_addr = source;
526: sgreq.grp.s_addr = group;
527: if ((ioctl(socket, SIOCGETSGCNT, (char *)&sgreq) < 0) || (sgreq.wrong_if == 0xffffffff)) {
528: /* XXX: ipmulti-3.5 has bug in ip_mroute.c, get_sg_cnt():
529: * the return code is always 0, so this is why we need to check
530: * the wrong_if value.
531: */
532: logit(LOG_WARNING, errno, "Failed reading kernel count (SIOCGETSGCNT) for (S,G) on (%s, %s)",
533: inet_fmt(source, s1, sizeof(s1)), inet_fmt(group, s2, sizeof(s2)));
534: retval->pktcnt = retval->bytecnt = retval->wrong_if = ~0;
535:
536: return 1;
537: }
538:
539: retval->pktcnt = sgreq.pktcnt;
540: retval->bytecnt = sgreq.bytecnt;
541: retval->wrong_if = sgreq.wrong_if;
542:
543: return 0;
544: }
545:
546: /**
547: * Local Variables:
548: * version-control: t
549: * indent-tabs-mode: t
550: * c-file-style: "ellemtel"
551: * c-basic-offset: 4
552: * End:
553: */
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>
![[BACK]](/icons/cvsweb/back.gif){kind=icon}
Return to kern.c CVS log ![[TXT]](/icons/cvsweb/text.gif){kind=icon}
![[DIR]](/icons/cvsweb/dir.gif){kind=icon}
Up to [ELWIX - Embedded LightWeight unIX -] / embedaddon / pimd