137 lines
		
	
	
		
			4.0 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			137 lines
		
	
	
		
			4.0 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
/*	$NetBSD: udp.c,v 1.11 2011/05/11 16:23:40 zoltan Exp $	*/
 | 
						|
 | 
						|
/*
 | 
						|
 * Copyright (c) 1992 Regents of the University of California.
 | 
						|
 * All rights reserved.
 | 
						|
 *
 | 
						|
 * This software was developed by the Computer Systems Engineering group
 | 
						|
 * at Lawrence Berkeley Laboratory under DARPA contract BG 91-66 and
 | 
						|
 * contributed to Berkeley.
 | 
						|
 *
 | 
						|
 * Redistribution and use in source and binary forms, with or without
 | 
						|
 * modification, are permitted provided that the following conditions
 | 
						|
 * are met:
 | 
						|
 * 1. Redistributions of source code must retain the above copyright
 | 
						|
 *    notice, this list of conditions and the following disclaimer.
 | 
						|
 * 2. Redistributions in binary form must reproduce the above copyright
 | 
						|
 *    notice, this list of conditions and the following disclaimer in the
 | 
						|
 *    documentation and/or other materials provided with the distribution.
 | 
						|
 * 3. All advertising materials mentioning features or use of this software
 | 
						|
 *    must display the following acknowledgement:
 | 
						|
 *	This product includes software developed by the University of
 | 
						|
 *	California, Lawrence Berkeley Laboratory and its contributors.
 | 
						|
 * 4. Neither the name of the University nor the names of its contributors
 | 
						|
 *    may be used to endorse or promote products derived from this software
 | 
						|
 *    without specific prior written permission.
 | 
						|
 *
 | 
						|
 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
 | 
						|
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 | 
						|
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 | 
						|
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
 | 
						|
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 | 
						|
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 | 
						|
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 | 
						|
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 | 
						|
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 | 
						|
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 | 
						|
 * SUCH DAMAGE.
 | 
						|
 *
 | 
						|
 * @(#) Header: net.c,v 1.9 93/08/06 19:32:15 leres Exp  (LBL)
 | 
						|
 */
 | 
						|
 | 
						|
#include <sys/param.h>
 | 
						|
#include <sys/socket.h>
 | 
						|
 | 
						|
#ifdef _STANDALONE
 | 
						|
#include <lib/libkern/libkern.h>
 | 
						|
#else
 | 
						|
#include <string.h>
 | 
						|
#endif
 | 
						|
 | 
						|
#include <net/if.h>
 | 
						|
#include <net/if_ether.h>
 | 
						|
 | 
						|
#include <netinet/in.h>
 | 
						|
#include <netinet/in_systm.h>
 | 
						|
#include <netinet/ip.h>
 | 
						|
#include <netinet/ip_var.h>
 | 
						|
#include <netinet/udp.h>
 | 
						|
#include <netinet/udp_var.h>
 | 
						|
 | 
						|
#include "stand.h"
 | 
						|
#include "net.h"
 | 
						|
 | 
						|
 | 
						|
/* Caller must leave room for ethernet, ip and udp headers in front!! */
 | 
						|
ssize_t
 | 
						|
sendudp(struct iodesc *d, void *pkt, size_t len)
 | 
						|
{
 | 
						|
	ssize_t cc;
 | 
						|
	struct udphdr *uh;
 | 
						|
 | 
						|
#ifdef NET_DEBUG
 | 
						|
 	if (debug) {
 | 
						|
		printf("sendudp: d=%lx called.\n", (long)d);
 | 
						|
		if (d) {
 | 
						|
			printf("saddr: %s:%d",
 | 
						|
			    inet_ntoa(d->myip), ntohs(d->myport));
 | 
						|
			printf(" daddr: %s:%d\n",
 | 
						|
			    inet_ntoa(d->destip), ntohs(d->destport));
 | 
						|
		}
 | 
						|
	}
 | 
						|
#endif
 | 
						|
 | 
						|
	uh = (struct udphdr *)pkt - 1;
 | 
						|
	len += sizeof(*uh);
 | 
						|
 | 
						|
	(void)memset(uh, 0, sizeof(*uh));
 | 
						|
 | 
						|
	uh->uh_sport = d->myport;
 | 
						|
	uh->uh_dport = d->destport;
 | 
						|
	uh->uh_ulen = htons(len);
 | 
						|
 | 
						|
	cc = sendip(d, uh, len, IPPROTO_UDP);
 | 
						|
	if (cc == -1)
 | 
						|
		return -1;
 | 
						|
	if ((size_t)cc != len)
 | 
						|
		panic("sendudp: bad write (%zd != %zu)", cc, len);
 | 
						|
	return (cc - sizeof(*uh));
 | 
						|
}
 | 
						|
 | 
						|
/*
 | 
						|
 * Receive a UDP packet and validate it is for us.
 | 
						|
 * Caller leaves room for the headers (Ether, IP, UDP)
 | 
						|
 */
 | 
						|
ssize_t
 | 
						|
readudp(struct iodesc *d, void *pkt, size_t len, saseconds_t tleft)
 | 
						|
{
 | 
						|
	ssize_t n;
 | 
						|
	struct udphdr *uh;
 | 
						|
 | 
						|
	uh = (struct udphdr *)pkt - 1;
 | 
						|
	n = readip(d, uh, len + sizeof(*uh), tleft, IPPROTO_UDP);
 | 
						|
	if (n == -1 || (size_t)n < sizeof(*uh))
 | 
						|
		return -1;
 | 
						|
 | 
						|
	if (uh->uh_dport != d->myport) {
 | 
						|
#ifdef NET_DEBUG
 | 
						|
		if (debug)
 | 
						|
			printf("readudp: bad dport %d != %d\n",
 | 
						|
				d->myport, ntohs(uh->uh_dport));
 | 
						|
#endif
 | 
						|
		return -1;
 | 
						|
	}
 | 
						|
 | 
						|
	if (ntohs(uh->uh_ulen) < sizeof(*uh)) {
 | 
						|
#ifdef NET_DEBUG
 | 
						|
		if (debug)
 | 
						|
			printf("readudp: bad udp len %d < %d\n",
 | 
						|
				ntohs(uh->uh_ulen), (int)sizeof(*uh));
 | 
						|
#endif
 | 
						|
		return -1;
 | 
						|
	}
 | 
						|
 | 
						|
	n -= sizeof(*uh);
 | 
						|
	return n;
 | 
						|
}
 |