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.10  2008/08/20 06:57:00  haraldkipp
00097  * Implemented IP demultiplexer.
00098  *
00099  * Revision 1.9  2008/08/11 07:00:33  haraldkipp
00100  * BSD types replaced by stdint types (feature request #1282721).
00101  *
00102  * Revision 1.8  2008/04/18 13:32:00  haraldkipp
00103  * Changed size parameter from u_short to int, which is easier to handle
00104  * for 32-bit targets. You need to recompile your ARM code. No impact on
00105  * AVR expected
00106  * I changed u_int to int at some places to avoid some warnings during
00107  * compilation of Nut/Net.
00108  * libs.
00109  *
00110  * Revision 1.7  2005/08/02 17:47:03  haraldkipp
00111  * Major API documentation update.
00112  *
00113  * Revision 1.6  2005/04/08 15:20:51  olereinhardt
00114  * added <sys/types.h> (__APPLE__) and <netinet/in.h> (__linux__)
00115  * for htons and simmilar.
00116  *
00117  * Revision 1.5  2003/11/26 12:55:12  drsung
00118  * Portability issues ... again
00119  *
00120  * Revision 1.4  2003/11/24 21:01:04  drsung
00121  * Packet queue added for UDP sockets.
00122  *
00123  * Revision 1.3  2003/08/14 15:07:18  haraldkipp
00124  * Optimization
00125  *
00126  * Revision 1.2  2003/07/20 16:01:01  haraldkipp
00127  * Bugfix: Send crashed on routing problems.
00128  *
00129  * Revision 1.1.1.1  2003/05/09 14:41:46  haraldkipp
00130  * Initial using 3.2.1
00131  *
00132  * Revision 1.13  2003/05/06 18:21:52  harald
00133  * Memory hole fixed
00134  *
00135  * Revision 1.12  2003/04/21 17:04:57  harald
00136  * Bugfix: Return port number in host byte order
00137  *
00138  * Revision 1.11  2003/02/04 18:14:57  harald
00139  * Version 3 released
00140  *
00141  * Revision 1.10  2002/06/26 17:29:36  harald
00142  * First pre-release with 2.4 stack
00143  *
00144  */
00145 
00146 #include <string.h>
00147 
00148 #include <sys/heap.h>
00149 #include <sys/event.h>
00150 #include <sys/types.h>
00151 #include <netinet/ip.h>
00152 #include <netinet/udp.h>
00153 #include <netinet/in.h>
00154 #include <sys/socket.h>
00155 
00156 
00161 
00162 
00163 UDPSOCKET *udpSocketList;       
00164 static uint16_t last_local_port = 4096;  /* Unassigned local port. */
00165 static uint_fast8_t registered;
00166 
00178 UDPSOCKET *NutUdpCreateSocket(uint16_t port)
00179 {
00180     UDPSOCKET *sock;
00181 
00182     if (!registered) {
00183         if (NutRegisterIpHandler(IPPROTO_UDP, NutUdpInput)) {
00184             return NULL;
00185         }
00186         registered = 1;
00187     }
00188     if (port == 0) {
00189         do {
00190             if (++last_local_port == 0)
00191                 last_local_port = 4096;
00192             port = htons(last_local_port);
00193             sock = udpSocketList;
00194             while (sock) {
00195                 if (sock->so_local_port == port)
00196                     break;
00197                 sock = sock->so_next;
00198             }
00199         } while (sock);
00200         port = last_local_port;
00201     }
00202     if ((sock = NutHeapAllocClear(sizeof(UDPSOCKET))) != 0) {
00203         sock->so_local_port = htons(port);
00204         sock->so_next = udpSocketList;
00205         udpSocketList = sock;
00206     }
00207     return sock;
00208 }
00209 
00222 int NutUdpSendTo(UDPSOCKET * sock, uint32_t addr, uint16_t port, void *data, int len)
00223 {
00224     int rc;
00225     NETBUF *nb;
00226 
00227     if ((nb = NutNetBufAlloc(0, NBAF_APPLICATION, len)) == 0) {
00228         return -1;
00229     }
00230     memcpy(nb->nb_ap.vp, data, len);
00231 
00232     /* Bugfix by Ralph Mason. We should not free the NETBUF in case of an error. */
00233     if ((rc = NutUdpOutput(sock, addr, port, nb)) == 0)
00234         NutNetBufFree(nb);
00235 
00236     return rc;
00237 }
00238 
00255  /* @@@ 2003-10-24: modified by OS for udp packet queue */
00256 int NutUdpReceiveFrom(UDPSOCKET * sock, uint32_t * addr, uint16_t * port, void *data, int size, uint32_t timeout)
00257 {
00258     IPHDR *ip;
00259     UDPHDR *uh;
00260     NETBUF *nb;
00261 
00262     if (sock->so_rx_nb == 0)
00263         NutEventWait(&sock->so_rx_rdy, timeout);
00264 
00265     if ((nb = sock->so_rx_nb) == 0)
00266         return 0;
00267 
00268     /* forward the queue's head to the next packet */
00269     sock->so_rx_nb = nb->nb_next;
00270 
00271     ip = nb->nb_nw.vp;
00272     *addr = ip->ip_src;
00273 
00274     uh = nb->nb_tp.vp;
00275     *port = htons(uh->uh_sport);
00276 
00277     if (size > nb->nb_ap.sz)
00278         size = nb->nb_ap.sz;
00279 
00280     sock->so_rx_cnt -= nb->nb_ap.sz;    /* decrement input buffer count */
00281 
00282     memcpy(data, nb->nb_ap.vp, size);
00283     NutNetBufFree(nb);
00284 
00285     return size;
00286 }
00287 
00300  /* @@@ 2003-10-24: modified by OS for udp packet queue */
00301 int NutUdpDestroySocket(UDPSOCKET * sock)
00302 {
00303     UDPSOCKET *sp;
00304     UDPSOCKET **spp;
00305     int rc = -1;
00306     NETBUF *nb;
00307 
00308     spp = &udpSocketList;
00309     sp = udpSocketList;
00310 
00311     while (sp) {
00312         if (sp == sock) {
00313             *spp = sock->so_next;
00314             /* packets may have arrived that the application 
00315                did not read before closing the socket. */
00316             while ((nb = sock->so_rx_nb) != 0) {
00317                 sock->so_rx_nb = nb->nb_next;
00318                 NutNetBufFree(nb);
00319             }
00320             NutHeapFree(sock);
00321             rc = 0;
00322             break;
00323         }
00324         spp = &sp->so_next;
00325         sp = sp->so_next;
00326     }
00327     return rc;
00328 }
00329 
00341 UDPSOCKET *NutUdpFindSocket(uint16_t port)
00342 {
00343     UDPSOCKET *sp;
00344     UDPSOCKET *sock = 0;
00345 
00346     for (sp = udpSocketList; sp; sp = sp->so_next) {
00347         if (sp->so_local_port == port) {
00348             sock = sp;
00349             break;
00350         }
00351     }
00352     return sock;
00353 }
00354 
00369 int NutUdpSetSockOpt(UDPSOCKET * sock, int optname, CONST void *optval, int optlen)
00370 {
00371     int rc = -1;
00372 
00373     if (sock == 0)
00374         return -1;
00375     switch (optname) {
00376 
00377     case SO_RCVBUF:
00378         if (optval != 0 && optlen == sizeof(uint16_t)) {
00379             sock->so_rx_bsz = *((uint16_t *) optval);
00380             rc = 0;
00381         }
00382         break;
00383 
00384     default:
00385         /* sock->so_last_error = ENOPROTOOPT; */
00386         break;
00387     }
00388     return rc;
00389 }
00390 
00406 int NutUdpGetSockOpt(UDPSOCKET * sock, int optname, void *optval, int optlen)
00407 {
00408     int rc = -1;
00409 
00410     if (sock == 0)
00411         return -1;
00412     switch (optname) {
00413 
00414     case SO_RCVBUF:
00415         if (optval != 0 && optlen == sizeof(uint16_t)) {
00416             *((uint16_t *) optval) = sock->so_rx_bsz;
00417             rc = 0;
00418         }
00419         break;
00420 
00421     default:
00422         /* sock->so_last_error = ENOPROTOOPT; */
00423         break;
00424     }
00425     return rc;
00426 }
00427 
00428 

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