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.8  2008/04/18 13:32:00  haraldkipp
00097  * Changed size parameter from u_short to int, which is easier to handle
00098  * for 32-bit targets. You need to recompile your ARM code. No impact on
00099  * AVR expected
00100  * I changed u_int to int at some places to avoid some warnings during
00101  * compilation of Nut/Net.
00102  * libs.
00103  *
00104  * Revision 1.7  2005/08/02 17:47:03  haraldkipp
00105  * Major API documentation update.
00106  *
00107  * Revision 1.6  2005/04/08 15:20:51  olereinhardt
00108  * added <sys/types.h> (__APPLE__) and <netinet/in.h> (__linux__)
00109  * for htons and simmilar.
00110  *
00111  * Revision 1.5  2003/11/26 12:55:12  drsung
00112  * Portability issues ... again
00113  *
00114  * Revision 1.4  2003/11/24 21:01:04  drsung
00115  * Packet queue added for UDP sockets.
00116  *
00117  * Revision 1.3  2003/08/14 15:07:18  haraldkipp
00118  * Optimization
00119  *
00120  * Revision 1.2  2003/07/20 16:01:01  haraldkipp
00121  * Bugfix: Send crashed on routing problems.
00122  *
00123  * Revision 1.1.1.1  2003/05/09 14:41:46  haraldkipp
00124  * Initial using 3.2.1
00125  *
00126  * Revision 1.13  2003/05/06 18:21:52  harald
00127  * Memory hole fixed
00128  *
00129  * Revision 1.12  2003/04/21 17:04:57  harald
00130  * Bugfix: Return port number in host byte order
00131  *
00132  * Revision 1.11  2003/02/04 18:14:57  harald
00133  * Version 3 released
00134  *
00135  * Revision 1.10  2002/06/26 17:29:36  harald
00136  * First pre-release with 2.4 stack
00137  *
00138  */
00139 
00140 #include <string.h>
00141 
00142 #include <sys/heap.h>
00143 #include <sys/event.h>
00144 #include <sys/types.h>
00145 #include <netinet/ip.h>
00146 #include <netinet/udp.h>
00147 #include <netinet/in.h>
00148 #include <sys/socket.h>
00149 
00150 
00155 
00156 
00157 UDPSOCKET *udpSocketList;       
00158 static u_short last_local_port = 4096;  /* Unassigned local port. */
00159 
00171 UDPSOCKET *NutUdpCreateSocket(u_short port)
00172 {
00173     UDPSOCKET *sock;
00174 
00175     if (port == 0) {
00176         do {
00177             if (++last_local_port == 0)
00178                 last_local_port = 4096;
00179             port = htons(last_local_port);
00180             sock = udpSocketList;
00181             while (sock) {
00182                 if (sock->so_local_port == port)
00183                     break;
00184                 sock = sock->so_next;
00185             }
00186         } while (sock);
00187         port = last_local_port;
00188     }
00189     if ((sock = NutHeapAllocClear(sizeof(UDPSOCKET))) != 0) {
00190         sock->so_local_port = htons(port);
00191         sock->so_next = udpSocketList;
00192         udpSocketList = sock;
00193     }
00194     return sock;
00195 }
00196 
00209 int NutUdpSendTo(UDPSOCKET * sock, u_long addr, u_short port, void *data, int len)
00210 {
00211     int rc;
00212     NETBUF *nb;
00213 
00214     if ((nb = NutNetBufAlloc(0, NBAF_APPLICATION, len)) == 0) {
00215         return -1;
00216     }
00217     memcpy(nb->nb_ap.vp, data, len);
00218 
00219     /* Bugfix by Ralph Mason. We should not free the NETBUF in case of an error. */
00220     if ((rc = NutUdpOutput(sock, addr, port, nb)) == 0)
00221         NutNetBufFree(nb);
00222 
00223     return rc;
00224 }
00225 
00242  /* @@@ 2003-10-24: modified by OS for udp packet queue */
00243 int NutUdpReceiveFrom(UDPSOCKET * sock, u_long * addr, u_short * port, void *data, int size, u_long timeout)
00244 {
00245     IPHDR *ip;
00246     UDPHDR *uh;
00247     NETBUF *nb;
00248 
00249     if (sock->so_rx_nb == 0)
00250         NutEventWait(&sock->so_rx_rdy, timeout);
00251 
00252     if ((nb = sock->so_rx_nb) == 0)
00253         return 0;
00254 
00255     /* forward the queue's head to the next packet */
00256     sock->so_rx_nb = nb->nb_next;
00257 
00258     ip = nb->nb_nw.vp;
00259     *addr = ip->ip_src;
00260 
00261     uh = nb->nb_tp.vp;
00262     *port = htons(uh->uh_sport);
00263 
00264     if (size > nb->nb_ap.sz)
00265         size = nb->nb_ap.sz;
00266 
00267     sock->so_rx_cnt -= nb->nb_ap.sz;    /* decrement input buffer count */
00268 
00269     memcpy(data, nb->nb_ap.vp, size);
00270     NutNetBufFree(nb);
00271 
00272     return size;
00273 }
00274 
00287  /* @@@ 2003-10-24: modified by OS for udp packet queue */
00288 int NutUdpDestroySocket(UDPSOCKET * sock)
00289 {
00290     UDPSOCKET *sp;
00291     UDPSOCKET **spp;
00292     int rc = -1;
00293     NETBUF *nb;
00294 
00295     spp = &udpSocketList;
00296     sp = udpSocketList;
00297 
00298     while (sp) {
00299         if (sp == sock) {
00300             *spp = sock->so_next;
00301             /* packets may have arrived that the application 
00302                did not read before closing the socket. */
00303             while ((nb = sock->so_rx_nb) != 0) {
00304                 sock->so_rx_nb = nb->nb_next;
00305                 NutNetBufFree(nb);
00306             }
00307             NutHeapFree(sock);
00308             rc = 0;
00309             break;
00310         }
00311         spp = &sp->so_next;
00312         sp = sp->so_next;
00313     }
00314     return rc;
00315 }
00316 
00328 UDPSOCKET *NutUdpFindSocket(u_short port)
00329 {
00330     UDPSOCKET *sp;
00331     UDPSOCKET *sock = 0;
00332 
00333     for (sp = udpSocketList; sp; sp = sp->so_next) {
00334         if (sp->so_local_port == port) {
00335             sock = sp;
00336             break;
00337         }
00338     }
00339     return sock;
00340 }
00341 
00356 int NutUdpSetSockOpt(UDPSOCKET * sock, int optname, CONST void *optval, int optlen)
00357 {
00358     int rc = -1;
00359 
00360     if (sock == 0)
00361         return -1;
00362     switch (optname) {
00363 
00364     case SO_RCVBUF:
00365         if (optval != 0 && optlen == sizeof(u_short)) {
00366             sock->so_rx_bsz = *((u_short *) optval);
00367             rc = 0;
00368         }
00369         break;
00370 
00371     default:
00372         /* sock->so_last_error = ENOPROTOOPT; */
00373         break;
00374     }
00375     return rc;
00376 }
00377 
00393 int NutUdpGetSockOpt(UDPSOCKET * sock, int optname, void *optval, int optlen)
00394 {
00395     int rc = -1;
00396 
00397     if (sock == 0)
00398         return -1;
00399     switch (optname) {
00400 
00401     case SO_RCVBUF:
00402         if (optval != 0 && optlen == sizeof(u_short)) {
00403             *((u_short *) optval) = sock->so_rx_bsz;
00404             rc = 0;
00405         }
00406         break;
00407 
00408     default:
00409         /* sock->so_last_error = ENOPROTOOPT; */
00410         break;
00411     }
00412     return rc;
00413 }
00414 
00415 

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