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