udpsock.c

Go to the documentation of this file.
00001 /*
00002  * Copyright (C) 2001-2003 by egnite Software GmbH. All rights reserved.
00003  *
00004  * Redistribution and use in source and binary forms, with or without
00005  * modification, are permitted provided that the following conditions
00006  * are met:
00007  *
00008  * 1. Redistributions of source code must retain the above copyright
00009  *    notice, this list of conditions and the following disclaimer.
00010  * 2. Redistributions in binary form must reproduce the above copyright
00011  *    notice, this list of conditions and the following disclaimer in the
00012  *    documentation and/or other materials provided with the distribution.
00013  * 3. Neither the name of the copyright holders nor the names of
00014  *    contributors may be used to endorse or promote products derived
00015  *    from this software without specific prior written permission.
00016  *
00017  * THIS SOFTWARE IS PROVIDED BY EGNITE SOFTWARE GMBH AND CONTRIBUTORS
00018  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
00019  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
00020  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL EGNITE
00021  * SOFTWARE GMBH OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
00022  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
00023  * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS
00024  * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
00025  * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
00026  * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF
00027  * THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
00028  * SUCH DAMAGE.
00029  *
00030  * For additional information see http://www.ethernut.de/
00031  *
00032  * -
00033  * Portions Copyright (C) 2000 David J. Hudson <dave@humbug.demon.co.uk>
00034  *
00035  * This file is distributed in the hope that it will be useful, but WITHOUT
00036  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
00037  * FITNESS FOR A PARTICULAR PURPOSE.
00038  *
00039  * You can redistribute this file and/or modify it under the terms of the GNU
00040  * General Public License (GPL) as published by the Free Software Foundation;
00041  * either version 2 of the License, or (at your discretion) any later version.
00042  * See the accompanying file "copying-gpl.txt" for more details.
00043  *
00044  * As a special exception to the GPL, permission is granted for additional
00045  * uses of the text contained in this file.  See the accompanying file
00046  * "copying-liquorice.txt" for details.
00047  * -
00048  * Portions Copyright (c) 1983, 1993 by
00049  *  The Regents of the University of California.  All rights reserved.
00050  *
00051  * Redistribution and use in source and binary forms, with or without
00052  * modification, are permitted provided that the following conditions
00053  * are met:
00054  * 1. Redistributions of source code must retain the above copyright
00055  *    notice, this list of conditions and the following disclaimer.
00056  * 2. Redistributions in binary form must reproduce the above copyright
00057  *    notice, this list of conditions and the following disclaimer in the
00058  *    documentation and/or other materials provided with the distribution.
00059  * 3. Neither the name of the University nor the names of its contributors
00060  *    may be used to endorse or promote products derived from this software
00061  *    without specific prior written permission.
00062  *
00063  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
00064  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
00065  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
00066  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
00067  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
00068  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
00069  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
00070  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
00071  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
00072  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
00073  * SUCH DAMAGE.
00074  * -
00075  * Portions Copyright (c) 1993 by Digital Equipment Corporation.
00076  *
00077  * Permission to use, copy, modify, and distribute this software for any
00078  * purpose with or without fee is hereby granted, provided that the above
00079  * copyright notice and this permission notice appear in all copies, and that
00080  * the name of Digital Equipment Corporation not be used in advertising or
00081  * publicity pertaining to distribution of the document or software without
00082  * specific, written prior permission.
00083  * 
00084  * THE SOFTWARE IS PROVIDED "AS IS" AND DIGITAL EQUIPMENT CORP. DISCLAIMS ALL
00085  * WARRANTIES WITH REGARD TO THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES
00086  * OF MERCHANTABILITY AND FITNESS.   IN NO EVENT SHALL DIGITAL EQUIPMENT
00087  * CORPORATION BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL
00088  * DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR
00089  * PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS
00090  * ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS
00091  * SOFTWARE.
00092  */
00093 
00094 /*
00095  * $Log: udpsock.c,v $
00096  * Revision 1.12  2009/02/13 14:52:05  haraldkipp
00097  * Include memdebug.h for heap management debugging support.
00098  *
00099  * Revision 1.11  2009/02/06 15:40:29  haraldkipp
00100  * Using newly available strdup() and calloc().
00101  * Replaced NutHeap routines by standard malloc/free.
00102  * Replaced pointer value 0 by NULL.
00103  *
00104  * Revision 1.10  2008/08/20 06:57:00  haraldkipp
00105  * Implemented IP demultiplexer.
00106  *
00107  * Revision 1.9  2008/08/11 07:00:33  haraldkipp
00108  * BSD types replaced by stdint types (feature request #1282721).
00109  *
00110  * Revision 1.8  2008/04/18 13:32:00  haraldkipp
00111  * Changed size parameter from u_short to int, which is easier to handle
00112  * for 32-bit targets. You need to recompile your ARM code. No impact on
00113  * AVR expected
00114  * I changed u_int to int at some places to avoid some warnings during
00115  * compilation of Nut/Net.
00116  * libs.
00117  *
00118  * Revision 1.7  2005/08/02 17:47:03  haraldkipp
00119  * Major API documentation update.
00120  *
00121  * Revision 1.6  2005/04/08 15:20:51  olereinhardt
00122  * added <sys/types.h> (__APPLE__) and <netinet/in.h> (__linux__)
00123  * for htons and simmilar.
00124  *
00125  * Revision 1.5  2003/11/26 12:55:12  drsung
00126  * Portability issues ... again
00127  *
00128  * Revision 1.4  2003/11/24 21:01:04  drsung
00129  * Packet queue added for UDP sockets.
00130  *
00131  * Revision 1.3  2003/08/14 15:07:18  haraldkipp
00132  * Optimization
00133  *
00134  * Revision 1.2  2003/07/20 16:01:01  haraldkipp
00135  * Bugfix: Send crashed on routing problems.
00136  *
00137  * Revision 1.1.1.1  2003/05/09 14:41:46  haraldkipp
00138  * Initial using 3.2.1
00139  *
00140  * Revision 1.13  2003/05/06 18:21:52  harald
00141  * Memory hole fixed
00142  *
00143  * Revision 1.12  2003/04/21 17:04:57  harald
00144  * Bugfix: Return port number in host byte order
00145  *
00146  * Revision 1.11  2003/02/04 18:14:57  harald
00147  * Version 3 released
00148  *
00149  * Revision 1.10  2002/06/26 17:29:36  harald
00150  * First pre-release with 2.4 stack
00151  *
00152  */
00153 
00154 #include <sys/heap.h>
00155 #include <sys/event.h>
00156 #include <sys/types.h>
00157 #include <netinet/ip.h>
00158 #include <netinet/udp.h>
00159 #include <netinet/in.h>
00160 #include <sys/socket.h>
00161 
00162 #include <stdlib.h>
00163 #include <string.h>
00164 #include <memdebug.h>
00165 
00170 
00171 
00172 UDPSOCKET *udpSocketList;       
00173 static uint16_t last_local_port = 4096;  /* Unassigned local port. */
00174 static uint_fast8_t registered;
00175 
00187 UDPSOCKET *NutUdpCreateSocket(uint16_t port)
00188 {
00189     UDPSOCKET *sock;
00190 
00191     if (!registered) {
00192         if (NutRegisterIpHandler(IPPROTO_UDP, NutUdpInput)) {
00193             return NULL;
00194         }
00195         registered = 1;
00196     }
00197     if (port == 0) {
00198         do {
00199             if (++last_local_port == 0)
00200                 last_local_port = 4096;
00201             port = htons(last_local_port);
00202             sock = udpSocketList;
00203             while (sock) {
00204                 if (sock->so_local_port == port)
00205                     break;
00206                 sock = sock->so_next;
00207             }
00208         } while (sock);
00209         port = last_local_port;
00210     }
00211     if ((sock = calloc(1, sizeof(UDPSOCKET))) != 0) {
00212         sock->so_local_port = htons(port);
00213         sock->so_next = udpSocketList;
00214         udpSocketList = sock;
00215     }
00216     return sock;
00217 }
00218 
00231 int NutUdpSendTo(UDPSOCKET * sock, uint32_t addr, uint16_t port, void *data, int len)
00232 {
00233     int rc;
00234     NETBUF *nb;
00235 
00236     if ((nb = NutNetBufAlloc(0, NBAF_APPLICATION, len)) == 0) {
00237         return -1;
00238     }
00239     memcpy(nb->nb_ap.vp, data, len);
00240 
00241     /* Bugfix by Ralph Mason. We should not free the NETBUF in case of an error. */
00242     if ((rc = NutUdpOutput(sock, addr, port, nb)) == 0)
00243         NutNetBufFree(nb);
00244 
00245     return rc;
00246 }
00247 
00264  /* @@@ 2003-10-24: modified by OS for udp packet queue */
00265 int NutUdpReceiveFrom(UDPSOCKET * sock, uint32_t * addr, uint16_t * port, void *data, int size, uint32_t timeout)
00266 {
00267     IPHDR *ip;
00268     UDPHDR *uh;
00269     NETBUF *nb;
00270 
00271     if (sock->so_rx_nb == 0)
00272         NutEventWait(&sock->so_rx_rdy, timeout);
00273 
00274     if ((nb = sock->so_rx_nb) == 0)
00275         return 0;
00276 
00277     /* forward the queue's head to the next packet */
00278     sock->so_rx_nb = nb->nb_next;
00279 
00280     ip = nb->nb_nw.vp;
00281     *addr = ip->ip_src;
00282 
00283     uh = nb->nb_tp.vp;
00284     *port = htons(uh->uh_sport);
00285 
00286     if (size > nb->nb_ap.sz)
00287         size = nb->nb_ap.sz;
00288 
00289     sock->so_rx_cnt -= nb->nb_ap.sz;    /* decrement input buffer count */
00290 
00291     memcpy(data, nb->nb_ap.vp, size);
00292     NutNetBufFree(nb);
00293 
00294     return size;
00295 }
00296 
00309  /* @@@ 2003-10-24: modified by OS for udp packet queue */
00310 int NutUdpDestroySocket(UDPSOCKET * sock)
00311 {
00312     UDPSOCKET *sp;
00313     UDPSOCKET **spp;
00314     int rc = -1;
00315     NETBUF *nb;
00316 
00317     spp = &udpSocketList;
00318     sp = udpSocketList;
00319 
00320     while (sp) {
00321         if (sp == sock) {
00322             *spp = sock->so_next;
00323             /* packets may have arrived that the application 
00324                did not read before closing the socket. */
00325             while ((nb = sock->so_rx_nb) != 0) {
00326                 sock->so_rx_nb = nb->nb_next;
00327                 NutNetBufFree(nb);
00328             }
00329             free(sock);
00330             rc = 0;
00331             break;
00332         }
00333         spp = &sp->so_next;
00334         sp = sp->so_next;
00335     }
00336     return rc;
00337 }
00338 
00350 UDPSOCKET *NutUdpFindSocket(uint16_t port)
00351 {
00352     UDPSOCKET *sp;
00353     UDPSOCKET *sock = 0;
00354 
00355     for (sp = udpSocketList; sp; sp = sp->so_next) {
00356         if (sp->so_local_port == port) {
00357             sock = sp;
00358             break;
00359         }
00360     }
00361     return sock;
00362 }
00363 
00378 int NutUdpSetSockOpt(UDPSOCKET * sock, int optname, CONST void *optval, int optlen)
00379 {
00380     int rc = -1;
00381 
00382     if (sock == 0)
00383         return -1;
00384     switch (optname) {
00385 
00386     case SO_RCVBUF:
00387         if (optval != 0 && optlen == sizeof(uint16_t)) {
00388             sock->so_rx_bsz = *((uint16_t *) optval);
00389             rc = 0;
00390         }
00391         break;
00392 
00393     default:
00394         /* sock->so_last_error = ENOPROTOOPT; */
00395         break;
00396     }
00397     return rc;
00398 }
00399 
00415 int NutUdpGetSockOpt(UDPSOCKET * sock, int optname, void *optval, int optlen)
00416 {
00417     int rc = -1;
00418 
00419     if (sock == 0)
00420         return -1;
00421     switch (optname) {
00422 
00423     case SO_RCVBUF:
00424         if (optval != 0 && optlen == sizeof(uint16_t)) {
00425             *((uint16_t *) optval) = sock->so_rx_bsz;
00426             rc = 0;
00427         }
00428         break;
00429 
00430     default:
00431         /* sock->so_last_error = ENOPROTOOPT; */
00432         break;
00433     }
00434     return rc;
00435 }
00436 
00437 

© 2000-2007 by egnite Software GmbH - visit http://www.ethernut.de/