File:  [ELWIX - Embedded LightWeight unIX -] / embedaddon / dhcp / common / icmp.c
Revision 1.1.1.1 (vendor branch): download - view: text, annotated - select for diffs - revision graph
Tue Oct 9 09:06:54 2012 UTC (11 years, 8 months ago) by misho
Branches: dhcp, MAIN
CVS tags: v4_1_R7p0, v4_1_R7, v4_1_R4, HEAD
dhcp 4.1 r7

    1: /* dhcp.c
    2: 
    3:    ICMP Protocol engine - for sending out pings and receiving
    4:    responses. */
    5: 
    6: /*
    7:  * Copyright (c) 2011 by Internet Systems Consortium, Inc. ("ISC")
    8:  * Copyright (c) 2004,2007,2009 by Internet Systems Consortium, Inc. ("ISC")
    9:  * Copyright (c) 1996-2003 by Internet Software Consortium
   10:  *
   11:  * Permission to use, copy, modify, and distribute this software for any
   12:  * purpose with or without fee is hereby granted, provided that the above
   13:  * copyright notice and this permission notice appear in all copies.
   14:  *
   15:  * THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES
   16:  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
   17:  * MERCHANTABILITY AND FITNESS.  IN NO EVENT SHALL ISC BE LIABLE FOR
   18:  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
   19:  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
   20:  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
   21:  * OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
   22:  *
   23:  *   Internet Systems Consortium, Inc.
   24:  *   950 Charter Street
   25:  *   Redwood City, CA 94063
   26:  *   <info@isc.org>
   27:  *   https://www.isc.org/
   28:  *
   29:  * This software has been written for Internet Systems Consortium
   30:  * by Ted Lemon in cooperation with Vixie Enterprises and Nominum, Inc.
   31:  * To learn more about Internet Systems Consortium, see
   32:  * ``https://www.isc.org/''.  To learn more about Vixie Enterprises,
   33:  * see ``http://www.vix.com''.   To learn more about Nominum, Inc., see
   34:  * ``http://www.nominum.com''.
   35:  */
   36: 
   37: #include "dhcpd.h"
   38: #include "netinet/ip.h"
   39: #include "netinet/ip_icmp.h"
   40: 
   41: struct icmp_state *icmp_state;
   42: static omapi_object_type_t *dhcp_type_icmp;
   43: static int no_icmp;
   44: 
   45: OMAPI_OBJECT_ALLOC (icmp_state, struct icmp_state, dhcp_type_icmp)
   46: 
   47: #if defined (TRACING)
   48: trace_type_t *trace_icmp_input;
   49: trace_type_t *trace_icmp_output;
   50: #endif
   51: 
   52: /* Initialize the ICMP protocol. */
   53: 
   54: void icmp_startup (routep, handler)
   55: 	int routep;
   56: 	void (*handler) (struct iaddr, u_int8_t *, int);
   57: {
   58: 	struct protoent *proto;
   59: 	int protocol = 1;
   60: 	int state;
   61: 	isc_result_t result;
   62: 
   63: 	/* Only initialize icmp once. */
   64: 	if (dhcp_type_icmp)
   65: 		log_fatal ("attempted to reinitialize icmp protocol");
   66: 
   67: 	result = omapi_object_type_register (&dhcp_type_icmp, "icmp",
   68: 					     0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
   69: 					     sizeof (struct icmp_state),
   70: 					     0, RC_MISC);
   71: 
   72: 	if (result != ISC_R_SUCCESS)
   73: 		log_fatal ("Can't register icmp object type: %s",
   74: 			   isc_result_totext (result));
   75: 
   76: 	icmp_state_allocate (&icmp_state, MDL);
   77: 	icmp_state -> icmp_handler = handler;
   78: 
   79: #if defined (TRACING)
   80: 	trace_icmp_input = trace_type_register ("icmp-input", (void *)0,
   81: 						trace_icmp_input_input,
   82: 						trace_icmp_input_stop, MDL);
   83: 	trace_icmp_output = trace_type_register ("icmp-output", (void *)0,
   84: 						 trace_icmp_output_input,
   85: 						 trace_icmp_output_stop, MDL);
   86: 
   87: 	/* If we're playing back a trace file, don't create the socket
   88: 	   or set up the callback. */
   89: 	if (!trace_playback ()) {
   90: #endif
   91: 		/* Get the protocol number (should be 1). */
   92: 		proto = getprotobyname ("icmp");
   93: 		if (proto)
   94: 			protocol = proto -> p_proto;
   95: 		
   96: 		/* Get a raw socket for the ICMP protocol. */
   97: 		icmp_state -> socket = socket (AF_INET, SOCK_RAW, protocol);
   98: 		if (icmp_state -> socket < 0) {
   99: 			no_icmp = 1;
  100: 			log_error ("unable to create icmp socket: %m");
  101: 			return;
  102: 		}
  103: 
  104: #if defined (HAVE_SETFD)
  105: 		if (fcntl (icmp_state -> socket, F_SETFD, 1) < 0)
  106: 			log_error ("Can't set close-on-exec on icmp: %m");
  107: #endif
  108: 
  109: 		/* Make sure it does routing... */
  110: 		state = 0;
  111: 		if (setsockopt (icmp_state -> socket, SOL_SOCKET, SO_DONTROUTE,
  112: 				(char *)&state, sizeof state) < 0)
  113: 			log_fatal ("Can't disable SO_DONTROUTE on ICMP: %m");
  114: 
  115: 		result = (omapi_register_io_object
  116: 			  ((omapi_object_t *)icmp_state,
  117: 			   icmp_readsocket, 0, icmp_echoreply, 0, 0));
  118: 		if (result != ISC_R_SUCCESS)
  119: 			log_fatal ("Can't register icmp handle: %s",
  120: 				   isc_result_totext (result));
  121: #if defined (TRACING)
  122: 	}
  123: #endif
  124: }
  125: 
  126: int icmp_readsocket (h)
  127: 	omapi_object_t *h;
  128: {
  129: 	struct icmp_state *state;
  130: 
  131: 	state = (struct icmp_state *)h;
  132: 	return state -> socket;
  133: }
  134: 
  135: int icmp_echorequest (addr)
  136: 	struct iaddr *addr;
  137: {
  138: 	struct sockaddr_in to;
  139: 	struct icmp icmp;
  140: 	int status;
  141: #if defined (TRACING)
  142: 	trace_iov_t iov [2];
  143: #endif
  144: 
  145: 	if (no_icmp)
  146: 		return 1;
  147: 	if (!icmp_state)
  148: 		log_fatal ("ICMP protocol used before initialization.");
  149: 
  150: 	memset (&to, 0, sizeof(to));
  151: #ifdef HAVE_SA_LEN
  152: 	to.sin_len = sizeof to;
  153: #endif
  154: 	to.sin_family = AF_INET;
  155: 	to.sin_port = 0; /* unused. */
  156: 	memcpy (&to.sin_addr, addr -> iabuf, sizeof to.sin_addr); /* XXX */
  157: 
  158: 	icmp.icmp_type = ICMP_ECHO;
  159: 	icmp.icmp_code = 0;
  160: 	icmp.icmp_cksum = 0;
  161: 	icmp.icmp_seq = 0;
  162: #if SIZEOF_STRUCT_IADDR_P == 8
  163: 	icmp.icmp_id = (((u_int32_t)(u_int64_t)addr) ^
  164:   			(u_int32_t)(((u_int64_t)addr) >> 32));
  165: #else
  166: 	icmp.icmp_id = (u_int32_t)addr;
  167: #endif
  168: 	memset (&icmp.icmp_dun, 0, sizeof icmp.icmp_dun);
  169: 
  170: 	icmp.icmp_cksum = wrapsum (checksum ((unsigned char *)&icmp,
  171: 					     sizeof icmp, 0));
  172: 
  173: #if defined (TRACING)
  174: 	if (trace_playback ()) {
  175: 		char *buf = (char *)0;
  176: 		unsigned buflen = 0;
  177: 
  178: 		/* Consume the ICMP event. */
  179: 		status = trace_get_packet (&trace_icmp_output, &buflen, &buf);
  180: 		if (status != ISC_R_SUCCESS)
  181: 			log_error ("icmp_echorequest: %s",
  182: 				   isc_result_totext (status));
  183: 		if (buf)
  184: 			dfree (buf, MDL);
  185: 	} else {
  186: 		if (trace_record ()) {
  187: 			iov [0].buf = (char *)addr;
  188: 			iov [0].len = sizeof *addr;
  189: 			iov [1].buf = (char *)&icmp;
  190: 			iov [1].len = sizeof icmp;
  191: 			trace_write_packet_iov (trace_icmp_output,
  192: 						2, iov, MDL);
  193: 		}
  194: #endif
  195: 		/* Send the ICMP packet... */
  196: 		status = sendto (icmp_state -> socket,
  197: 				 (char *)&icmp, sizeof icmp, 0,
  198: 				 (struct sockaddr *)&to, sizeof to);
  199: 		if (status < 0)
  200: 			log_error ("icmp_echorequest %s: %m",
  201: 				   inet_ntoa(to.sin_addr));
  202: 
  203: 		if (status != sizeof icmp)
  204: 			return 0;
  205: #if defined (TRACING)
  206: 	}
  207: #endif
  208: 	return 1;
  209: }
  210: 
  211: isc_result_t icmp_echoreply (h)
  212: 	omapi_object_t *h;
  213: {
  214: 	struct icmp *icfrom;
  215: 	struct ip *ip;
  216: 	struct sockaddr_in from;
  217: 	u_int8_t icbuf [1500];
  218: 	int status;
  219: 	SOCKLEN_T sl;
  220: 	int hlen, len;
  221: 	struct iaddr ia;
  222: 	struct icmp_state *state;
  223: #if defined (TRACING)
  224: 	trace_iov_t iov [2];
  225: #endif
  226: 
  227: 	state = (struct icmp_state *)h;
  228: 
  229: 	sl = sizeof from;
  230: 	status = recvfrom (state -> socket, (char *)icbuf, sizeof icbuf, 0,
  231: 			  (struct sockaddr *)&from, &sl);
  232: 	if (status < 0) {
  233: 		log_error ("icmp_echoreply: %m");
  234: 		return ISC_R_UNEXPECTED;
  235: 	}
  236: 
  237: 	/* Find the IP header length... */
  238: 	ip = (struct ip *)icbuf;
  239: 	hlen = IP_HL (ip);
  240: 
  241: 	/* Short packet? */
  242: 	if (status < hlen + (sizeof *icfrom)) {
  243: 		return ISC_R_SUCCESS;
  244: 	}
  245: 
  246: 	len = status - hlen;
  247: 	icfrom = (struct icmp *)(icbuf + hlen);
  248: 
  249: 	/* Silently discard ICMP packets that aren't echoreplies. */
  250: 	if (icfrom -> icmp_type != ICMP_ECHOREPLY) {
  251: 		return ISC_R_SUCCESS;
  252: 	}
  253: 
  254: 	/* If we were given a second-stage handler, call it. */
  255: 	if (state -> icmp_handler) {
  256: 		memcpy (ia.iabuf, &from.sin_addr, sizeof from.sin_addr);
  257: 		ia.len = sizeof from.sin_addr;
  258: 
  259: #if defined (TRACING)
  260: 		if (trace_record ()) {
  261: 			ia.len = htonl(ia.len);
  262: 			iov [0].buf = (char *)&ia;
  263: 			iov [0].len = sizeof ia;
  264: 			iov [1].buf = (char *)icbuf;
  265: 			iov [1].len = len;
  266: 			trace_write_packet_iov (trace_icmp_input, 2, iov, MDL);
  267: 			ia.len = ntohl(ia.len);
  268: 		}
  269: #endif
  270: 		(*state -> icmp_handler) (ia, icbuf, len);
  271: 	}
  272: 	return ISC_R_SUCCESS;
  273: }
  274: 
  275: #if defined (TRACING)
  276: void trace_icmp_input_input (trace_type_t *ttype, unsigned length, char *buf)
  277: {
  278: 	struct iaddr *ia;
  279: 	u_int8_t *icbuf;
  280: 	ia = (struct iaddr *)buf;
  281: 	ia->len = ntohl(ia->len);
  282: 	icbuf = (u_int8_t *)(ia + 1);
  283: 	if (icmp_state -> icmp_handler)
  284: 		(*icmp_state -> icmp_handler) (*ia, icbuf,
  285: 					       (int)(length - sizeof ia));
  286: }
  287: 
  288: void trace_icmp_input_stop (trace_type_t *ttype) { }
  289: 
  290: void trace_icmp_output_input (trace_type_t *ttype, unsigned length, char *buf)
  291: {
  292: 	struct icmp *icmp;
  293: 	struct iaddr ia;
  294: 
  295: 	if (length != (sizeof (*icmp) + (sizeof ia))) {
  296: 		log_error ("trace_icmp_output_input: data size mismatch %d:%d",
  297: 			   length, (int)((sizeof (*icmp)) + (sizeof ia)));
  298: 		return;
  299: 	}
  300: 	ia.len = 4;
  301: 	memcpy (ia.iabuf, buf, 4);
  302: 	icmp = (struct icmp *)(buf + 1);
  303: 
  304: 	log_error ("trace_icmp_output_input: unsent ping to %s", piaddr (ia));
  305: }
  306: 
  307: void trace_icmp_output_stop (trace_type_t *ttype) { }
  308: #endif /* TRACING */

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