Nino Der Kinderen / Mbed 2 deprecated InetTest

Embedded C project:18/12/2014

Dependencies:   DS1307 TextLCD mbed

NetServices/if/lwip/lwipNetUdpSocket.cpp@0:8d87bc453349, 2014-12-18 (annotated)

Committer:
ninoderkinderen
Date:
Thu Dec 18 09:35:49 2014 +0000
Revision:
0:8d87bc453349
Programma embedded C

Who changed what in which revision?

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