Rik Van Dyck / Mbed 2 deprecated Project_Embedded_C

Important changes to repositories hosted on mbed.com

Mbed hosted mercurial repositories are deprecated and are due to be permanently deleted in July 2026.

To keep a copy of this software download the repository Zip archive or clone locally using Mercurial.

It is also possible to export all your personal repositories from the account settings page.

Dependencies:   mbed DS1307 Servo TextLCD

NetServices/if/lwip/lwipNetUdpSocket.cpp@0:e1edd52b1ee2, 2014-12-04 (annotated)

Committer:
rikvandyck
Date:
Thu Dec 04 10:44:07 2014 +0000
Revision:
0:e1edd52b1ee2
test

Who changed what in which revision?

UserRevisionLine numberNew contents of line
rikvandyck 0:e1edd52b1ee2 1 #pragma diag_remark 1464
rikvandyck 0:e1edd52b1ee2 2 /*
rikvandyck 0:e1edd52b1ee2 3 Copyright (c) 2010 Donatien Garnier (donatiengar [at] gmail [dot] com)
rikvandyck 0:e1edd52b1ee2 4
rikvandyck 0:e1edd52b1ee2 5 Permission is hereby granted, free of charge, to any person obtaining a copy
rikvandyck 0:e1edd52b1ee2 6 of this software and associated documentation files (the "Software"), to deal
rikvandyck 0:e1edd52b1ee2 7 in the Software without restriction, including without limitation the rights
rikvandyck 0:e1edd52b1ee2 8 to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
rikvandyck 0:e1edd52b1ee2 9 copies of the Software, and to permit persons to whom the Software is
rikvandyck 0:e1edd52b1ee2 10 furnished to do so, subject to the following conditions:
rikvandyck 0:e1edd52b1ee2 11
rikvandyck 0:e1edd52b1ee2 12 The above copyright notice and this permission notice shall be included in
rikvandyck 0:e1edd52b1ee2 13 all copies or substantial portions of the Software.
rikvandyck 0:e1edd52b1ee2 14
rikvandyck 0:e1edd52b1ee2 15 THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
rikvandyck 0:e1edd52b1ee2 16 IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
rikvandyck 0:e1edd52b1ee2 17 FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
rikvandyck 0:e1edd52b1ee2 18 AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
rikvandyck 0:e1edd52b1ee2 19 LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
rikvandyck 0:e1edd52b1ee2 20 OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
rikvandyck 0:e1edd52b1ee2 21 THE SOFTWARE.
rikvandyck 0:e1edd52b1ee2 22 */
rikvandyck 0:e1edd52b1ee2 23
rikvandyck 0:e1edd52b1ee2 24 #include "lwip/ip_addr.h"
rikvandyck 0:e1edd52b1ee2 25 #include "lwipNetUdpSocket.h"
rikvandyck 0:e1edd52b1ee2 26 #include "lwip/udp.h"
rikvandyck 0:e1edd52b1ee2 27 #include "lwip/igmp.h"
rikvandyck 0:e1edd52b1ee2 28
rikvandyck 0:e1edd52b1ee2 29
rikvandyck 0:e1edd52b1ee2 30 //#define __DEBUG
rikvandyck 0:e1edd52b1ee2 31 #include "dbg/dbg.h"
rikvandyck 0:e1edd52b1ee2 32
rikvandyck 0:e1edd52b1ee2 33 #include "netCfg.h"
rikvandyck 0:e1edd52b1ee2 34 #if NET_LWIP_STACK
rikvandyck 0:e1edd52b1ee2 35
rikvandyck 0:e1edd52b1ee2 36 LwipNetUdpSocket::LwipNetUdpSocket(udp_pcb* pPcb /*= NULL*/) : NetUdpSocket(), m_pPcb(pPcb), m_lInPkt(), m_multicastGroup() //Passes a pcb if already created (by an accept req for instance), in that case transfers ownership
rikvandyck 0:e1edd52b1ee2 37 {
rikvandyck 0:e1edd52b1ee2 38 DBG("New LwipNetUdpSocket %p (pPCb=%p)\n", (void*)this, (void*) pPcb);
rikvandyck 0:e1edd52b1ee2 39 if(!m_pPcb)
rikvandyck 0:e1edd52b1ee2 40 m_pPcb = udp_new();
rikvandyck 0:e1edd52b1ee2 41 if(m_pPcb)
rikvandyck 0:e1edd52b1ee2 42 {
rikvandyck 0:e1edd52b1ee2 43 //Setup callback
rikvandyck 0:e1edd52b1ee2 44 udp_recv( (udp_pcb*) m_pPcb, LwipNetUdpSocket::sRecvCb, (void*) this );
rikvandyck 0:e1edd52b1ee2 45 }
rikvandyck 0:e1edd52b1ee2 46 }
rikvandyck 0:e1edd52b1ee2 47
rikvandyck 0:e1edd52b1ee2 48 LwipNetUdpSocket::~LwipNetUdpSocket()
rikvandyck 0:e1edd52b1ee2 49 {
rikvandyck 0:e1edd52b1ee2 50 close();
rikvandyck 0:e1edd52b1ee2 51 }
rikvandyck 0:e1edd52b1ee2 52
rikvandyck 0:e1edd52b1ee2 53 NetUdpSocketErr LwipNetUdpSocket::bind(const Host& me)
rikvandyck 0:e1edd52b1ee2 54 {
rikvandyck 0:e1edd52b1ee2 55 err_t err;
rikvandyck 0:e1edd52b1ee2 56
rikvandyck 0:e1edd52b1ee2 57 if(!m_pPcb)
rikvandyck 0:e1edd52b1ee2 58 return NETUDPSOCKET_MEM; //NetUdpSocket was not properly initialised, should destroy it & retry
rikvandyck 0:e1edd52b1ee2 59
rikvandyck 0:e1edd52b1ee2 60 #if LWIP_IGMP //Multicast support enabled
rikvandyck 0:e1edd52b1ee2 61 if(me.getIp().isMulticast())
rikvandyck 0:e1edd52b1ee2 62 {
rikvandyck 0:e1edd52b1ee2 63 DBG("This is a multicast addr, joining multicast group\n");
rikvandyck 0:e1edd52b1ee2 64 m_multicastGroup = me.getIp();
rikvandyck 0:e1edd52b1ee2 65 err = igmp_joingroup(IP_ADDR_ANY, &(m_multicastGroup.getStruct()));
rikvandyck 0:e1edd52b1ee2 66 if(err)
rikvandyck 0:e1edd52b1ee2 67 return NETUDPSOCKET_IF; //Could not find or create group
rikvandyck 0:e1edd52b1ee2 68 }
rikvandyck 0:e1edd52b1ee2 69 #endif
rikvandyck 0:e1edd52b1ee2 70
rikvandyck 0:e1edd52b1ee2 71 err = udp_bind( (udp_pcb*) m_pPcb, IP_ADDR_ANY, me.getPort()); //IP_ADDR_ANY : Bind the connection to all local addresses
rikvandyck 0:e1edd52b1ee2 72 if(err)
rikvandyck 0:e1edd52b1ee2 73 return NETUDPSOCKET_INUSE;
rikvandyck 0:e1edd52b1ee2 74
rikvandyck 0:e1edd52b1ee2 75 //Setup callback
rikvandyck 0:e1edd52b1ee2 76 udp_recv( (udp_pcb*) m_pPcb, LwipNetUdpSocket::sRecvCb, (void*) this );
rikvandyck 0:e1edd52b1ee2 77
rikvandyck 0:e1edd52b1ee2 78 return NETUDPSOCKET_OK;
rikvandyck 0:e1edd52b1ee2 79 }
rikvandyck 0:e1edd52b1ee2 80
rikvandyck 0:e1edd52b1ee2 81 #define MAX(a,b) ((a>b)?a:b)
rikvandyck 0:e1edd52b1ee2 82 #define MIN(a,b) ((a<b)?a:b)
rikvandyck 0:e1edd52b1ee2 83
rikvandyck 0:e1edd52b1ee2 84 int /*if < 0 : NetUdpSocketErr*/ LwipNetUdpSocket::sendto(const char* buf, int len, Host* pHost)
rikvandyck 0:e1edd52b1ee2 85 {
rikvandyck 0:e1edd52b1ee2 86 if( !m_pPcb ) //Pcb doesn't exist (anymore)
rikvandyck 0:e1edd52b1ee2 87 return NETUDPSOCKET_MEM;
rikvandyck 0:e1edd52b1ee2 88 pbuf* p = pbuf_alloc(PBUF_TRANSPORT, len, PBUF_POOL);
rikvandyck 0:e1edd52b1ee2 89 if( !p )
rikvandyck 0:e1edd52b1ee2 90 return NETUDPSOCKET_MEM;
rikvandyck 0:e1edd52b1ee2 91 char* pBuf = (char*) buf;
rikvandyck 0:e1edd52b1ee2 92 pbuf* q = p;
rikvandyck 0:e1edd52b1ee2 93 do
rikvandyck 0:e1edd52b1ee2 94 {
rikvandyck 0:e1edd52b1ee2 95 memcpy (q->payload, (void*)pBuf, q->len);
rikvandyck 0:e1edd52b1ee2 96 pBuf += q->len;
rikvandyck 0:e1edd52b1ee2 97 q = q->next;
rikvandyck 0:e1edd52b1ee2 98 } while(q != NULL);
rikvandyck 0:e1edd52b1ee2 99
rikvandyck 0:e1edd52b1ee2 100 err_t err = udp_sendto( (udp_pcb*) m_pPcb, p, &(pHost->getIp().getStruct()), pHost->getPort() );
rikvandyck 0:e1edd52b1ee2 101 pbuf_free( p );
rikvandyck 0:e1edd52b1ee2 102 if(err)
rikvandyck 0:e1edd52b1ee2 103 return NETUDPSOCKET_SETUP; //Connection problem
rikvandyck 0:e1edd52b1ee2 104 DBG("%d bytes sent in UDP Socket.\n", len);
rikvandyck 0:e1edd52b1ee2 105 return len;
rikvandyck 0:e1edd52b1ee2 106 }
rikvandyck 0:e1edd52b1ee2 107
rikvandyck 0:e1edd52b1ee2 108 int /*if < 0 : NetUdpSocketErr*/ LwipNetUdpSocket::recvfrom(char* buf, int len, Host* pHost)
rikvandyck 0:e1edd52b1ee2 109 {
rikvandyck 0:e1edd52b1ee2 110 if( !m_pPcb ) //Pcb doesn't exist (anymore)
rikvandyck 0:e1edd52b1ee2 111 return NETUDPSOCKET_MEM;
rikvandyck 0:e1edd52b1ee2 112 int inLen = 0;
rikvandyck 0:e1edd52b1ee2 113 int cpyLen = 0;
rikvandyck 0:e1edd52b1ee2 114
rikvandyck 0:e1edd52b1ee2 115 static int rmgLen = 0;
rikvandyck 0:e1edd52b1ee2 116 //Contains the remaining len in this pbuf
rikvandyck 0:e1edd52b1ee2 117
rikvandyck 0:e1edd52b1ee2 118 if( m_lInPkt.empty() )
rikvandyck 0:e1edd52b1ee2 119 return 0;
rikvandyck 0:e1edd52b1ee2 120
rikvandyck 0:e1edd52b1ee2 121 pbuf* pBuf = (pbuf*) m_lInPkt.front().pBuf;
rikvandyck 0:e1edd52b1ee2 122
rikvandyck 0:e1edd52b1ee2 123 if(pHost)
rikvandyck 0:e1edd52b1ee2 124 *pHost = Host( IpAddr(&m_lInPkt.front().addr), m_lInPkt.front().port );
rikvandyck 0:e1edd52b1ee2 125
rikvandyck 0:e1edd52b1ee2 126 if( !pBuf )
rikvandyck 0:e1edd52b1ee2 127 {
rikvandyck 0:e1edd52b1ee2 128 rmgLen = 0;
rikvandyck 0:e1edd52b1ee2 129 return 0;
rikvandyck 0:e1edd52b1ee2 130 }
rikvandyck 0:e1edd52b1ee2 131
rikvandyck 0:e1edd52b1ee2 132 if ( !rmgLen ) //We did not know m_pReadPbuf->len last time we called this fn
rikvandyck 0:e1edd52b1ee2 133 {
rikvandyck 0:e1edd52b1ee2 134 rmgLen = pBuf->len;
rikvandyck 0:e1edd52b1ee2 135 }
rikvandyck 0:e1edd52b1ee2 136
rikvandyck 0:e1edd52b1ee2 137 while ( inLen < len )
rikvandyck 0:e1edd52b1ee2 138 {
rikvandyck 0:e1edd52b1ee2 139 cpyLen = MIN( (len - inLen), rmgLen ); //Remaining len to copy, remaining len in THIS pbuf
rikvandyck 0:e1edd52b1ee2 140 memcpy((void*)buf, (void*)((char*)(pBuf->payload) + (pBuf->len - rmgLen)), cpyLen);
rikvandyck 0:e1edd52b1ee2 141 inLen += cpyLen;
rikvandyck 0:e1edd52b1ee2 142 buf += cpyLen;
rikvandyck 0:e1edd52b1ee2 143
rikvandyck 0:e1edd52b1ee2 144 rmgLen = rmgLen - cpyLen; //Update rmgLen
rikvandyck 0:e1edd52b1ee2 145
rikvandyck 0:e1edd52b1ee2 146 if( rmgLen > 0 )
rikvandyck 0:e1edd52b1ee2 147 {
rikvandyck 0:e1edd52b1ee2 148 //We did not read this pbuf completely, so let's save it's pos & return
rikvandyck 0:e1edd52b1ee2 149 break;
rikvandyck 0:e1edd52b1ee2 150 }
rikvandyck 0:e1edd52b1ee2 151
rikvandyck 0:e1edd52b1ee2 152 if(pBuf->next)
rikvandyck 0:e1edd52b1ee2 153 {
rikvandyck 0:e1edd52b1ee2 154 pbuf* pNextPBuf = pBuf->next;
rikvandyck 0:e1edd52b1ee2 155 pBuf->next = NULL; //So that it is not freed as well
rikvandyck 0:e1edd52b1ee2 156 //We get the reference to pNextPBuf from m_pReadPbuf
rikvandyck 0:e1edd52b1ee2 157 pbuf_free((pbuf*)pBuf);
rikvandyck 0:e1edd52b1ee2 158 pBuf = pNextPBuf;
rikvandyck 0:e1edd52b1ee2 159 rmgLen = pBuf->len;
rikvandyck 0:e1edd52b1ee2 160 }
rikvandyck 0:e1edd52b1ee2 161 else
rikvandyck 0:e1edd52b1ee2 162 {
rikvandyck 0:e1edd52b1ee2 163 pbuf_free((pbuf*)pBuf);
rikvandyck 0:e1edd52b1ee2 164 pBuf = NULL;
rikvandyck 0:e1edd52b1ee2 165 rmgLen = 0;
rikvandyck 0:e1edd52b1ee2 166 m_lInPkt.pop_front();
rikvandyck 0:e1edd52b1ee2 167 break; //No more data to read
rikvandyck 0:e1edd52b1ee2 168 }
rikvandyck 0:e1edd52b1ee2 169 }
rikvandyck 0:e1edd52b1ee2 170
rikvandyck 0:e1edd52b1ee2 171 return inLen;
rikvandyck 0:e1edd52b1ee2 172 }
rikvandyck 0:e1edd52b1ee2 173
rikvandyck 0:e1edd52b1ee2 174 NetUdpSocketErr LwipNetUdpSocket::close()
rikvandyck 0:e1edd52b1ee2 175 {
rikvandyck 0:e1edd52b1ee2 176 DBG("LwipNetUdpSocket::close() : Closing...\n");
rikvandyck 0:e1edd52b1ee2 177
rikvandyck 0:e1edd52b1ee2 178 if(m_closed)
rikvandyck 0:e1edd52b1ee2 179 return NETUDPSOCKET_OK; //Already being closed
rikvandyck 0:e1edd52b1ee2 180 m_closed = true;
rikvandyck 0:e1edd52b1ee2 181
rikvandyck 0:e1edd52b1ee2 182 if( !m_pPcb ) //Pcb doesn't exist (anymore)
rikvandyck 0:e1edd52b1ee2 183 return NETUDPSOCKET_MEM;
rikvandyck 0:e1edd52b1ee2 184
rikvandyck 0:e1edd52b1ee2 185 DBG("LwipNetUdpSocket::close() : Cleanup...\n");
rikvandyck 0:e1edd52b1ee2 186
rikvandyck 0:e1edd52b1ee2 187 //Cleanup incoming data
rikvandyck 0:e1edd52b1ee2 188 cleanUp();
rikvandyck 0:e1edd52b1ee2 189
rikvandyck 0:e1edd52b1ee2 190
rikvandyck 0:e1edd52b1ee2 191 DBG("LwipNetUdpSocket::close() : removing m_pPcb...\n");
rikvandyck 0:e1edd52b1ee2 192 udp_remove( (udp_pcb*) m_pPcb);
rikvandyck 0:e1edd52b1ee2 193
rikvandyck 0:e1edd52b1ee2 194 m_pPcb = NULL;
rikvandyck 0:e1edd52b1ee2 195 return NETUDPSOCKET_OK;
rikvandyck 0:e1edd52b1ee2 196 }
rikvandyck 0:e1edd52b1ee2 197
rikvandyck 0:e1edd52b1ee2 198 NetUdpSocketErr LwipNetUdpSocket::poll()
rikvandyck 0:e1edd52b1ee2 199 {
rikvandyck 0:e1edd52b1ee2 200 NetUdpSocket::flushEvents();
rikvandyck 0:e1edd52b1ee2 201 return NETUDPSOCKET_OK;
rikvandyck 0:e1edd52b1ee2 202 }
rikvandyck 0:e1edd52b1ee2 203
rikvandyck 0:e1edd52b1ee2 204 // Callbacks events
rikvandyck 0:e1edd52b1ee2 205
rikvandyck 0:e1edd52b1ee2 206 void LwipNetUdpSocket::recvCb(udp_pcb* pcb, struct pbuf* p, ip_addr_t* addr, u16_t port)
rikvandyck 0:e1edd52b1ee2 207 {
rikvandyck 0:e1edd52b1ee2 208 DBG(" Packet of length %d arrived in UDP Socket.\n", p->tot_len);
rikvandyck 0:e1edd52b1ee2 209 list<InPacket>::iterator it;
rikvandyck 0:e1edd52b1ee2 210 for ( it = m_lInPkt.begin(); it != m_lInPkt.end(); it++ )
rikvandyck 0:e1edd52b1ee2 211 {
rikvandyck 0:e1edd52b1ee2 212 if( ip_addr_cmp((&((*it).addr)), addr) && ((*it).port == port) )
rikvandyck 0:e1edd52b1ee2 213 {
rikvandyck 0:e1edd52b1ee2 214 //Let's tail this packet to the previous one
rikvandyck 0:e1edd52b1ee2 215 pbuf_cat((pbuf*)((*it).pBuf), p);
rikvandyck 0:e1edd52b1ee2 216 //No need to queue an event in that case since the read buf has not been processed yet
rikvandyck 0:e1edd52b1ee2 217 return;
rikvandyck 0:e1edd52b1ee2 218 }
rikvandyck 0:e1edd52b1ee2 219 }
rikvandyck 0:e1edd52b1ee2 220
rikvandyck 0:e1edd52b1ee2 221 //New host, add a packet to the queue
rikvandyck 0:e1edd52b1ee2 222 InPacket pkt;
rikvandyck 0:e1edd52b1ee2 223 pkt.pBuf = p;
rikvandyck 0:e1edd52b1ee2 224 pkt.addr = *addr;
rikvandyck 0:e1edd52b1ee2 225 pkt.port = port;
rikvandyck 0:e1edd52b1ee2 226 m_lInPkt.push_back(pkt);
rikvandyck 0:e1edd52b1ee2 227
rikvandyck 0:e1edd52b1ee2 228 queueEvent(NETUDPSOCKET_READABLE);
rikvandyck 0:e1edd52b1ee2 229 }
rikvandyck 0:e1edd52b1ee2 230
rikvandyck 0:e1edd52b1ee2 231 void LwipNetUdpSocket::cleanUp() //Flush input buffer
rikvandyck 0:e1edd52b1ee2 232 {
rikvandyck 0:e1edd52b1ee2 233 //Ensure that further error won't be followed to this inst (which can be destroyed)
rikvandyck 0:e1edd52b1ee2 234 if( m_pPcb )
rikvandyck 0:e1edd52b1ee2 235 {
rikvandyck 0:e1edd52b1ee2 236 udp_recv( (udp_pcb*) m_pPcb, NULL, (void*) NULL );
rikvandyck 0:e1edd52b1ee2 237 }
rikvandyck 0:e1edd52b1ee2 238
rikvandyck 0:e1edd52b1ee2 239 //Leaving multicast group(Ok because LwIP has a refscount for multicast group)
rikvandyck 0:e1edd52b1ee2 240 #if LWIP_IGMP //Multicast support enabled
rikvandyck 0:e1edd52b1ee2 241 if(m_multicastGroup.isMulticast())
rikvandyck 0:e1edd52b1ee2 242 {
rikvandyck 0:e1edd52b1ee2 243 igmp_leavegroup(IP_ADDR_ANY, &(m_multicastGroup.getStruct()));
rikvandyck 0:e1edd52b1ee2 244 m_multicastGroup = IpAddr();
rikvandyck 0:e1edd52b1ee2 245 }
rikvandyck 0:e1edd52b1ee2 246 #endif
rikvandyck 0:e1edd52b1ee2 247
rikvandyck 0:e1edd52b1ee2 248 list<InPacket>::iterator it;
rikvandyck 0:e1edd52b1ee2 249 for ( it = m_lInPkt.begin(); it != m_lInPkt.end(); it++ )
rikvandyck 0:e1edd52b1ee2 250 {
rikvandyck 0:e1edd52b1ee2 251 //Free buf
rikvandyck 0:e1edd52b1ee2 252 pbuf_free((pbuf*)((*it).pBuf));
rikvandyck 0:e1edd52b1ee2 253 }
rikvandyck 0:e1edd52b1ee2 254 m_lInPkt.clear();
rikvandyck 0:e1edd52b1ee2 255 }
rikvandyck 0:e1edd52b1ee2 256
rikvandyck 0:e1edd52b1ee2 257 // Static callback from LwIp
rikvandyck 0:e1edd52b1ee2 258
rikvandyck 0:e1edd52b1ee2 259 void LwipNetUdpSocket::sRecvCb(void *arg, struct udp_pcb *pcb, struct pbuf *p, ip_addr_t *addr, u16_t port)
rikvandyck 0:e1edd52b1ee2 260 {
rikvandyck 0:e1edd52b1ee2 261 LwipNetUdpSocket* pMe = (LwipNetUdpSocket*) arg;
rikvandyck 0:e1edd52b1ee2 262 return pMe->recvCb( pcb, p, addr, port );
rikvandyck 0:e1edd52b1ee2 263 }
rikvandyck 0:e1edd52b1ee2 264
rikvandyck 0:e1edd52b1ee2 265 #endif