FBI FRATER / Mbed 2 deprecated InetTest

Library for Bert van Dam's book "ARM MICROCONTROLLERS" For all chapters with internet.

Dependencies:   mbed

NetServices/if/lwip/lwipNetUdpSocket.cpp@0:4edb816d21e1, 2015-10-16 (annotated)

Committer:
ICTFBI
Date:
Fri Oct 16 14:28:26 2015 +0000
Revision:
0:4edb816d21e1
Pre-update 16-10-15

Who changed what in which revision?

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