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/lwipNetTcpSocket.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
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 "lwipNetTcpSocket.h"
ICTFBI 0:4edb816d21e1 25 #include "lwip/tcp.h"
ICTFBI 0:4edb816d21e1 26
ICTFBI 0:4edb816d21e1 27 //#define __DEBUG
ICTFBI 0:4edb816d21e1 28 #include "dbg/dbg.h"
ICTFBI 0:4edb816d21e1 29
ICTFBI 0:4edb816d21e1 30 #include "netCfg.h"
ICTFBI 0:4edb816d21e1 31 #if NET_LWIP_STACK
ICTFBI 0:4edb816d21e1 32
ICTFBI 0:4edb816d21e1 33 LwipNetTcpSocket::LwipNetTcpSocket(tcp_pcb* pPcb /*= NULL*/) : NetTcpSocket(), m_pPcb(pPcb), m_lpInNetTcpSocket(), //Passes a pcb if already created (by an accept req for instance), in that case transfers ownership
ICTFBI 0:4edb816d21e1 34 m_pReadPbuf(NULL)
ICTFBI 0:4edb816d21e1 35 {
ICTFBI 0:4edb816d21e1 36 DBG("New NetTcpSocket %p\n", (void*)this);
ICTFBI 0:4edb816d21e1 37 if(!m_pPcb)
ICTFBI 0:4edb816d21e1 38 {
ICTFBI 0:4edb816d21e1 39 m_pPcb = tcp_new();
ICTFBI 0:4edb816d21e1 40 DBG("Creating new PCB %p\n", m_pPcb);
ICTFBI 0:4edb816d21e1 41 }
ICTFBI 0:4edb816d21e1 42 if(m_pPcb)
ICTFBI 0:4edb816d21e1 43 {
ICTFBI 0:4edb816d21e1 44 //Setup callbacks
ICTFBI 0:4edb816d21e1 45 tcp_arg( (tcp_pcb*) m_pPcb, (void*) this ); //this will be passed to each static callback
ICTFBI 0:4edb816d21e1 46
ICTFBI 0:4edb816d21e1 47 tcp_recv( (tcp_pcb*) m_pPcb, LwipNetTcpSocket::sRecvCb );
ICTFBI 0:4edb816d21e1 48 tcp_sent((tcp_pcb*) m_pPcb, LwipNetTcpSocket::sSentCb );
ICTFBI 0:4edb816d21e1 49 tcp_err( (tcp_pcb*) m_pPcb, LwipNetTcpSocket::sErrCb );
ICTFBI 0:4edb816d21e1 50 //Connected callback is defined in connect()
ICTFBI 0:4edb816d21e1 51 //Accept callback is defined in listen()
ICTFBI 0:4edb816d21e1 52 DBG("NetTcpSocket created.\n");
ICTFBI 0:4edb816d21e1 53 }
ICTFBI 0:4edb816d21e1 54 }
ICTFBI 0:4edb816d21e1 55
ICTFBI 0:4edb816d21e1 56 LwipNetTcpSocket::~LwipNetTcpSocket()
ICTFBI 0:4edb816d21e1 57 {
ICTFBI 0:4edb816d21e1 58 /* if(m_pPcb)
ICTFBI 0:4edb816d21e1 59 tcp_close( (tcp_pcb*) m_pPcb); //Disconnect & free pcb*/
ICTFBI 0:4edb816d21e1 60 close();
ICTFBI 0:4edb816d21e1 61 }
ICTFBI 0:4edb816d21e1 62
ICTFBI 0:4edb816d21e1 63 NetTcpSocketErr LwipNetTcpSocket::bind(const Host& me)
ICTFBI 0:4edb816d21e1 64 {
ICTFBI 0:4edb816d21e1 65 if(!m_pPcb)
ICTFBI 0:4edb816d21e1 66 return NETTCPSOCKET_MEM; //NetTcpSocket was not properly initialised, should destroy it & retry
ICTFBI 0:4edb816d21e1 67
ICTFBI 0:4edb816d21e1 68 err_t err = tcp_bind( (tcp_pcb*) m_pPcb, IP_ADDR_ANY, me.getPort()); //IP_ADDR_ANY : Bind the connection to all local addresses
ICTFBI 0:4edb816d21e1 69 if(err)
ICTFBI 0:4edb816d21e1 70 return NETTCPSOCKET_INUSE;
ICTFBI 0:4edb816d21e1 71
ICTFBI 0:4edb816d21e1 72 return NETTCPSOCKET_OK;
ICTFBI 0:4edb816d21e1 73 }
ICTFBI 0:4edb816d21e1 74
ICTFBI 0:4edb816d21e1 75 NetTcpSocketErr LwipNetTcpSocket::listen()
ICTFBI 0:4edb816d21e1 76 {
ICTFBI 0:4edb816d21e1 77 if(!m_pPcb)
ICTFBI 0:4edb816d21e1 78 return NETTCPSOCKET_MEM; //NetTcpSocket was not properly initialised, should destroy it & retry
ICTFBI 0:4edb816d21e1 79 /*
ICTFBI 0:4edb816d21e1 80 From doc/rawapi.txt :
ICTFBI 0:4edb816d21e1 81
ICTFBI 0:4edb816d21e1 82 The tcp_listen() function returns a new connection identifier, and
ICTFBI 0:4edb816d21e1 83 the one passed as an argument to the function will be
ICTFBI 0:4edb816d21e1 84 deallocated. The reason for this behavior is that less memory is
ICTFBI 0:4edb816d21e1 85 needed for a connection that is listening, so tcp_listen() will
ICTFBI 0:4edb816d21e1 86 reclaim the memory needed for the original connection and allocate a
ICTFBI 0:4edb816d21e1 87 new smaller memory block for the listening connection.
ICTFBI 0:4edb816d21e1 88 */
ICTFBI 0:4edb816d21e1 89
ICTFBI 0:4edb816d21e1 90 // tcp_pcb* pNewPcb = tcp_listen(m_pPcb);
ICTFBI 0:4edb816d21e1 91 tcp_pcb* pNewPcb = tcp_listen_with_backlog((tcp_pcb*)m_pPcb, 5);
ICTFBI 0:4edb816d21e1 92 if( !pNewPcb ) //Not enough memory to create the listening pcb
ICTFBI 0:4edb816d21e1 93 return NETTCPSOCKET_MEM;
ICTFBI 0:4edb816d21e1 94
ICTFBI 0:4edb816d21e1 95 m_pPcb = pNewPcb;
ICTFBI 0:4edb816d21e1 96
ICTFBI 0:4edb816d21e1 97 tcp_accept( (tcp_pcb*) m_pPcb, LwipNetTcpSocket::sAcceptCb );
ICTFBI 0:4edb816d21e1 98
ICTFBI 0:4edb816d21e1 99 return NETTCPSOCKET_OK;
ICTFBI 0:4edb816d21e1 100 }
ICTFBI 0:4edb816d21e1 101
ICTFBI 0:4edb816d21e1 102 NetTcpSocketErr LwipNetTcpSocket::connect(const Host& host)
ICTFBI 0:4edb816d21e1 103 {
ICTFBI 0:4edb816d21e1 104 if(!m_pPcb)
ICTFBI 0:4edb816d21e1 105 return NETTCPSOCKET_MEM; //NetTcpSocket was not properly initialised, should destroy it & retry
ICTFBI 0:4edb816d21e1 106
ICTFBI 0:4edb816d21e1 107 ip_addr_t ip = host.getIp().getStruct();
ICTFBI 0:4edb816d21e1 108 err_t err = tcp_connect( (tcp_pcb*) m_pPcb, &ip, host.getPort(), LwipNetTcpSocket::sConnectedCb );
ICTFBI 0:4edb816d21e1 109
ICTFBI 0:4edb816d21e1 110 if(err)
ICTFBI 0:4edb816d21e1 111 return NETTCPSOCKET_MEM;
ICTFBI 0:4edb816d21e1 112
ICTFBI 0:4edb816d21e1 113 return NETTCPSOCKET_OK;
ICTFBI 0:4edb816d21e1 114 }
ICTFBI 0:4edb816d21e1 115
ICTFBI 0:4edb816d21e1 116 NetTcpSocketErr LwipNetTcpSocket::accept(Host* pClient, NetTcpSocket** ppNewNetTcpSocket)
ICTFBI 0:4edb816d21e1 117 {
ICTFBI 0:4edb816d21e1 118 if( !m_pPcb ) //Pcb doesn't exist (anymore)
ICTFBI 0:4edb816d21e1 119 return NETTCPSOCKET_MEM;
ICTFBI 0:4edb816d21e1 120 //Dequeue a connection
ICTFBI 0:4edb816d21e1 121 //if( m_lpInPcb.empty() )
ICTFBI 0:4edb816d21e1 122 if( m_lpInNetTcpSocket.empty() )
ICTFBI 0:4edb816d21e1 123 return NETTCPSOCKET_EMPTY;
ICTFBI 0:4edb816d21e1 124
ICTFBI 0:4edb816d21e1 125 tcp_accepted( ((tcp_pcb*) m_pPcb) ); //Should fire proper events //WARN: m_pPcb is the GOOD param here (and not pInPcb)
ICTFBI 0:4edb816d21e1 126
ICTFBI 0:4edb816d21e1 127 /* tcp_pcb* pInPcb = m_lpInPcb.front();
ICTFBI 0:4edb816d21e1 128 m_lpInPcb.pop();*/
ICTFBI 0:4edb816d21e1 129
ICTFBI 0:4edb816d21e1 130 if( (m_lpInNetTcpSocket.front()) == NULL )
ICTFBI 0:4edb816d21e1 131 {
ICTFBI 0:4edb816d21e1 132 m_lpInNetTcpSocket.pop();
ICTFBI 0:4edb816d21e1 133 return NETTCPSOCKET_RST;
ICTFBI 0:4edb816d21e1 134 }
ICTFBI 0:4edb816d21e1 135
ICTFBI 0:4edb816d21e1 136 if( (m_lpInNetTcpSocket.front())->m_closed )
ICTFBI 0:4edb816d21e1 137 {
ICTFBI 0:4edb816d21e1 138 Net::releaseTcpSocket(m_lpInNetTcpSocket.front());
ICTFBI 0:4edb816d21e1 139 m_lpInNetTcpSocket.pop();
ICTFBI 0:4edb816d21e1 140 return NETTCPSOCKET_RST;
ICTFBI 0:4edb816d21e1 141 }
ICTFBI 0:4edb816d21e1 142
ICTFBI 0:4edb816d21e1 143 ip_addr_t* ip = (ip_addr_t*) &( (m_lpInNetTcpSocket.front()->m_pPcb)->remote_ip);
ICTFBI 0:4edb816d21e1 144
ICTFBI 0:4edb816d21e1 145 *ppNewNetTcpSocket = m_lpInNetTcpSocket.front();
ICTFBI 0:4edb816d21e1 146 *pClient = Host(
ICTFBI 0:4edb816d21e1 147 IpAddr(
ICTFBI 0:4edb816d21e1 148 ip
ICTFBI 0:4edb816d21e1 149 ),
ICTFBI 0:4edb816d21e1 150 m_lpInNetTcpSocket.front()->m_pPcb->remote_port
ICTFBI 0:4edb816d21e1 151 );
ICTFBI 0:4edb816d21e1 152 m_lpInNetTcpSocket.pop();
ICTFBI 0:4edb816d21e1 153 // *pClient = Host( IpAddr(pInPcb->remote_ip), pInPcb->remote_port );
ICTFBI 0:4edb816d21e1 154
ICTFBI 0:4edb816d21e1 155 //Return a new socket
ICTFBI 0:4edb816d21e1 156 // *ppNewNetTcpSocket = (NetTcpSocket*) new LwipNetTcpSocket(pInPcb);
ICTFBI 0:4edb816d21e1 157
ICTFBI 0:4edb816d21e1 158 //tcp_accepted( ((tcp_pcb*) m_pPcb) ); //Should fire proper events //WARN: m_pPcb is the GOOD param here (and not pInPcb)
ICTFBI 0:4edb816d21e1 159
ICTFBI 0:4edb816d21e1 160 /* if(*ppNewNetTcpSocket == NULL)
ICTFBI 0:4edb816d21e1 161 {
ICTFBI 0:4edb816d21e1 162 DBG("Not enough mem, socket dropped in LwipNetTcpSocket::accept.\n");
ICTFBI 0:4edb816d21e1 163 tcp_abort(pInPcb);
ICTFBI 0:4edb816d21e1 164 }*/
ICTFBI 0:4edb816d21e1 165
ICTFBI 0:4edb816d21e1 166 return NETTCPSOCKET_OK;
ICTFBI 0:4edb816d21e1 167 }
ICTFBI 0:4edb816d21e1 168
ICTFBI 0:4edb816d21e1 169 #define MAX(a,b) (((a)>(b))?(a):(b))
ICTFBI 0:4edb816d21e1 170 #define MIN(a,b) (((a)<(b))?(a):(b))
ICTFBI 0:4edb816d21e1 171
ICTFBI 0:4edb816d21e1 172 int /*if < 0 : NetTcpSocketErr*/ LwipNetTcpSocket::send(const char* buf, int len)
ICTFBI 0:4edb816d21e1 173 {
ICTFBI 0:4edb816d21e1 174 if( !m_pPcb ) //Pcb doesn't exist (anymore)
ICTFBI 0:4edb816d21e1 175 return NETTCPSOCKET_MEM;
ICTFBI 0:4edb816d21e1 176 int outLen = MIN( len, tcp_sndbuf( (tcp_pcb*) m_pPcb) );
ICTFBI 0:4edb816d21e1 177 //tcp_sndbuf() returns the number of bytes available in the output queue, so never go above it
ICTFBI 0:4edb816d21e1 178 err_t err = tcp_write( (tcp_pcb*) m_pPcb, (void*) buf, outLen, TCP_WRITE_FLAG_COPY );
ICTFBI 0:4edb816d21e1 179 //Flags are TCP_WRITE_FLAG_COPY & TCP_WRITE_FLAG_MORE (see tcp_out.c) :
ICTFBI 0:4edb816d21e1 180 //If TCP_WRITE_FLAG_MORE is not set ask client to push buffered data to app
ICTFBI 0:4edb816d21e1 181 if(err)
ICTFBI 0:4edb816d21e1 182 {
ICTFBI 0:4edb816d21e1 183 switch( err )
ICTFBI 0:4edb816d21e1 184 {
ICTFBI 0:4edb816d21e1 185 case ERR_CONN:
ICTFBI 0:4edb816d21e1 186 return (int) NETTCPSOCKET_SETUP; //Not connected properly
ICTFBI 0:4edb816d21e1 187 case ERR_ARG:
ICTFBI 0:4edb816d21e1 188 return (int) NETTCPSOCKET_SETUP; //Wrong args ! (like buf pointing to NULL)
ICTFBI 0:4edb816d21e1 189 case ERR_MEM:
ICTFBI 0:4edb816d21e1 190 default:
ICTFBI 0:4edb816d21e1 191 return (int) NETTCPSOCKET_MEM; //Not enough memory
ICTFBI 0:4edb816d21e1 192 }
ICTFBI 0:4edb816d21e1 193 }
ICTFBI 0:4edb816d21e1 194 return outLen;
ICTFBI 0:4edb816d21e1 195 }
ICTFBI 0:4edb816d21e1 196
ICTFBI 0:4edb816d21e1 197 int /*if < 0 : NetTcpSocketErr*/ LwipNetTcpSocket::recv(char* buf, int len)
ICTFBI 0:4edb816d21e1 198 {
ICTFBI 0:4edb816d21e1 199 if( !m_pPcb ) //Pcb doesn't exist (anymore)
ICTFBI 0:4edb816d21e1 200 return NETTCPSOCKET_MEM;
ICTFBI 0:4edb816d21e1 201 int inLen = 0;
ICTFBI 0:4edb816d21e1 202 int cpyLen = 0;
ICTFBI 0:4edb816d21e1 203
ICTFBI 0:4edb816d21e1 204 static int rmgLen = 0;
ICTFBI 0:4edb816d21e1 205 //Contains the remaining len in this pbuf
ICTFBI 0:4edb816d21e1 206
ICTFBI 0:4edb816d21e1 207 if( !m_pReadPbuf )
ICTFBI 0:4edb816d21e1 208 {
ICTFBI 0:4edb816d21e1 209 rmgLen = 0;
ICTFBI 0:4edb816d21e1 210 return 0;
ICTFBI 0:4edb816d21e1 211 }
ICTFBI 0:4edb816d21e1 212
ICTFBI 0:4edb816d21e1 213 if ( !rmgLen ) //We did not know m_pReadPbuf->len last time we called this fn
ICTFBI 0:4edb816d21e1 214 {
ICTFBI 0:4edb816d21e1 215 rmgLen = m_pReadPbuf->len;
ICTFBI 0:4edb816d21e1 216 }
ICTFBI 0:4edb816d21e1 217
ICTFBI 0:4edb816d21e1 218 while ( inLen < len )
ICTFBI 0:4edb816d21e1 219 {
ICTFBI 0:4edb816d21e1 220 cpyLen = MIN( (len - inLen), rmgLen ); //Remaining len to copy, remaining len in THIS pbuf
ICTFBI 0:4edb816d21e1 221 memcpy((void*)buf, (void*)((char*)(m_pReadPbuf->payload) + (m_pReadPbuf->len - rmgLen)), cpyLen);
ICTFBI 0:4edb816d21e1 222 inLen += cpyLen;
ICTFBI 0:4edb816d21e1 223 buf += cpyLen;
ICTFBI 0:4edb816d21e1 224
ICTFBI 0:4edb816d21e1 225 rmgLen = rmgLen - cpyLen; //Update rmgLen
ICTFBI 0:4edb816d21e1 226
ICTFBI 0:4edb816d21e1 227 if( rmgLen > 0 )
ICTFBI 0:4edb816d21e1 228 {
ICTFBI 0:4edb816d21e1 229 //We did not read this pbuf completely, so let's save it's pos & return
ICTFBI 0:4edb816d21e1 230 break;
ICTFBI 0:4edb816d21e1 231 }
ICTFBI 0:4edb816d21e1 232
ICTFBI 0:4edb816d21e1 233 if(m_pReadPbuf->next)
ICTFBI 0:4edb816d21e1 234 {
ICTFBI 0:4edb816d21e1 235 pbuf* pNextPBuf = m_pReadPbuf->next;
ICTFBI 0:4edb816d21e1 236 m_pReadPbuf->next = NULL; //So that it is not freed as well
ICTFBI 0:4edb816d21e1 237 //We get the reference to pNextPBuf from m_pReadPbuf
ICTFBI 0:4edb816d21e1 238 pbuf_free((pbuf*)m_pReadPbuf);
ICTFBI 0:4edb816d21e1 239 m_pReadPbuf = pNextPBuf;
ICTFBI 0:4edb816d21e1 240 rmgLen = m_pReadPbuf->len;
ICTFBI 0:4edb816d21e1 241 }
ICTFBI 0:4edb816d21e1 242 else
ICTFBI 0:4edb816d21e1 243 {
ICTFBI 0:4edb816d21e1 244 pbuf_free((pbuf*)m_pReadPbuf);
ICTFBI 0:4edb816d21e1 245 m_pReadPbuf = NULL;
ICTFBI 0:4edb816d21e1 246 rmgLen = 0;
ICTFBI 0:4edb816d21e1 247 break; //No more data to read
ICTFBI 0:4edb816d21e1 248 }
ICTFBI 0:4edb816d21e1 249
ICTFBI 0:4edb816d21e1 250 }
ICTFBI 0:4edb816d21e1 251
ICTFBI 0:4edb816d21e1 252 //tcp_recved(m_pPcb, inLen); //Acknowledge the reception
ICTFBI 0:4edb816d21e1 253
ICTFBI 0:4edb816d21e1 254 return inLen;
ICTFBI 0:4edb816d21e1 255 }
ICTFBI 0:4edb816d21e1 256
ICTFBI 0:4edb816d21e1 257 NetTcpSocketErr LwipNetTcpSocket::close()
ICTFBI 0:4edb816d21e1 258 {
ICTFBI 0:4edb816d21e1 259 //DBG("LwipNetTcpSocket::close() : Closing...\n");
ICTFBI 0:4edb816d21e1 260
ICTFBI 0:4edb816d21e1 261 if(m_closed)
ICTFBI 0:4edb816d21e1 262 return NETTCPSOCKET_OK; //Already being closed
ICTFBI 0:4edb816d21e1 263 m_closed = true;
ICTFBI 0:4edb816d21e1 264
ICTFBI 0:4edb816d21e1 265 if( !m_pPcb ) //Pcb doesn't exist (anymore)
ICTFBI 0:4edb816d21e1 266 return NETTCPSOCKET_MEM;
ICTFBI 0:4edb816d21e1 267
ICTFBI 0:4edb816d21e1 268 //Cleanup incoming data
ICTFBI 0:4edb816d21e1 269 cleanUp();
ICTFBI 0:4edb816d21e1 270
ICTFBI 0:4edb816d21e1 271 if( !!tcp_close( (tcp_pcb*) m_pPcb) )
ICTFBI 0:4edb816d21e1 272 {
ICTFBI 0:4edb816d21e1 273 DBG("LwipNetTcpSocket::close() could not close properly, abort.\n");
ICTFBI 0:4edb816d21e1 274 tcp_abort( (tcp_pcb*) m_pPcb);
ICTFBI 0:4edb816d21e1 275 m_pPcb = NULL;
ICTFBI 0:4edb816d21e1 276 return NETTCPSOCKET_MEM;
ICTFBI 0:4edb816d21e1 277 }
ICTFBI 0:4edb816d21e1 278
ICTFBI 0:4edb816d21e1 279 DBG("LwipNetTcpSocket::close() : connection closed successfully.\n");
ICTFBI 0:4edb816d21e1 280
ICTFBI 0:4edb816d21e1 281 m_pPcb = NULL;
ICTFBI 0:4edb816d21e1 282 return NETTCPSOCKET_OK;
ICTFBI 0:4edb816d21e1 283 }
ICTFBI 0:4edb816d21e1 284
ICTFBI 0:4edb816d21e1 285 NetTcpSocketErr LwipNetTcpSocket::poll()
ICTFBI 0:4edb816d21e1 286 {
ICTFBI 0:4edb816d21e1 287 NetTcpSocket::flushEvents();
ICTFBI 0:4edb816d21e1 288 return NETTCPSOCKET_OK;
ICTFBI 0:4edb816d21e1 289 }
ICTFBI 0:4edb816d21e1 290
ICTFBI 0:4edb816d21e1 291 // Callbacks events
ICTFBI 0:4edb816d21e1 292
ICTFBI 0:4edb816d21e1 293 err_t LwipNetTcpSocket::acceptCb(struct tcp_pcb *newpcb, err_t err)
ICTFBI 0:4edb816d21e1 294 {
ICTFBI 0:4edb816d21e1 295 if(err)
ICTFBI 0:4edb816d21e1 296 {
ICTFBI 0:4edb816d21e1 297 DBG("Error %d in LwipNetTcpSocket::acceptCb.\n", err);
ICTFBI 0:4edb816d21e1 298 return err;
ICTFBI 0:4edb816d21e1 299 }
ICTFBI 0:4edb816d21e1 300 //FIXME: MEM Errs
ICTFBI 0:4edb816d21e1 301 //m_lpInPcb.push(newpcb); //Add connection to the queue
ICTFBI 0:4edb816d21e1 302 LwipNetTcpSocket* pNewNetTcpSocket = new LwipNetTcpSocket(newpcb);
ICTFBI 0:4edb816d21e1 303
ICTFBI 0:4edb816d21e1 304 if(pNewNetTcpSocket == NULL)
ICTFBI 0:4edb816d21e1 305 {
ICTFBI 0:4edb816d21e1 306 DBG("Not enough mem, socket dropped in LwipNetTcpSocket::acceptCb.\n");
ICTFBI 0:4edb816d21e1 307 tcp_abort(newpcb);
ICTFBI 0:4edb816d21e1 308 return ERR_ABRT;
ICTFBI 0:4edb816d21e1 309 }
ICTFBI 0:4edb816d21e1 310
ICTFBI 0:4edb816d21e1 311 pNewNetTcpSocket->m_refs++;
ICTFBI 0:4edb816d21e1 312 m_lpInNetTcpSocket.push( pNewNetTcpSocket );
ICTFBI 0:4edb816d21e1 313
ICTFBI 0:4edb816d21e1 314 // tcp_accepted(newpcb);
ICTFBI 0:4edb816d21e1 315 // tcp_accepted( m_pPcb ); //Should fire proper events //WARN: m_pPcb is the GOOD param here (and not pInPcb)
ICTFBI 0:4edb816d21e1 316 queueEvent(NETTCPSOCKET_ACCEPT);
ICTFBI 0:4edb816d21e1 317 return ERR_OK;
ICTFBI 0:4edb816d21e1 318 }
ICTFBI 0:4edb816d21e1 319
ICTFBI 0:4edb816d21e1 320 err_t LwipNetTcpSocket::connectedCb(struct tcp_pcb *tpcb, err_t err)
ICTFBI 0:4edb816d21e1 321 {
ICTFBI 0:4edb816d21e1 322 queueEvent(NETTCPSOCKET_CONNECTED);
ICTFBI 0:4edb816d21e1 323 return ERR_OK;
ICTFBI 0:4edb816d21e1 324 }
ICTFBI 0:4edb816d21e1 325
ICTFBI 0:4edb816d21e1 326 void LwipNetTcpSocket::errCb(err_t err)
ICTFBI 0:4edb816d21e1 327 {
ICTFBI 0:4edb816d21e1 328 DBG("NetTcpSocket %p - Error %d in LwipNetTcpSocket::errCb.\n", (void*)this, err);
ICTFBI 0:4edb816d21e1 329 //WARN: At this point, m_pPcb has been freed by lwIP
ICTFBI 0:4edb816d21e1 330 m_pPcb = NULL;
ICTFBI 0:4edb816d21e1 331 //These errors are fatal, discard all events queued before so that the errors are handled first
ICTFBI 0:4edb816d21e1 332 discardEvents();
ICTFBI 0:4edb816d21e1 333 m_closed = true;
ICTFBI 0:4edb816d21e1 334 cleanUp();
ICTFBI 0:4edb816d21e1 335 if( err == ERR_ABRT)
ICTFBI 0:4edb816d21e1 336 queueEvent(NETTCPSOCKET_CONABRT);
ICTFBI 0:4edb816d21e1 337 else //if( err == ERR_RST)
ICTFBI 0:4edb816d21e1 338 queueEvent(NETTCPSOCKET_CONRST);
ICTFBI 0:4edb816d21e1 339 }
ICTFBI 0:4edb816d21e1 340
ICTFBI 0:4edb816d21e1 341 err_t LwipNetTcpSocket::sentCb(tcp_pcb* tpcb, u16_t len)
ICTFBI 0:4edb816d21e1 342 {
ICTFBI 0:4edb816d21e1 343 // DBG("%d bytes ACKed by host.\n", len);
ICTFBI 0:4edb816d21e1 344 queueEvent(NETTCPSOCKET_WRITEABLE);
ICTFBI 0:4edb816d21e1 345 return ERR_OK;
ICTFBI 0:4edb816d21e1 346 }
ICTFBI 0:4edb816d21e1 347
ICTFBI 0:4edb816d21e1 348 err_t LwipNetTcpSocket::recvCb(tcp_pcb* tpcb, pbuf *p, err_t err)
ICTFBI 0:4edb816d21e1 349 {
ICTFBI 0:4edb816d21e1 350 //Store pbuf ptr
ICTFBI 0:4edb816d21e1 351 // DBG("Receive CB with err = %d & len = %d.\n", err, p->tot_len);
ICTFBI 0:4edb816d21e1 352 // tcp_recved( (tcp_pcb*) m_pPcb, p->tot_len); //Acknowledge the reception
ICTFBI 0:4edb816d21e1 353
ICTFBI 0:4edb816d21e1 354 if(err)
ICTFBI 0:4edb816d21e1 355 {
ICTFBI 0:4edb816d21e1 356 queueEvent(NETTCPSOCKET_ERROR);
ICTFBI 0:4edb816d21e1 357 return ERR_OK; //FIXME: More robust error handling there
ICTFBI 0:4edb816d21e1 358 }
ICTFBI 0:4edb816d21e1 359 else if(!p)
ICTFBI 0:4edb816d21e1 360 {
ICTFBI 0:4edb816d21e1 361 DBG("NetTcpSocket %p - Connection closed by remote host (LwipNetTcpSocket::recvCb).\n", (void*)this);
ICTFBI 0:4edb816d21e1 362 //Buf is NULL, that means that the connection has been closed by remote host
ICTFBI 0:4edb816d21e1 363
ICTFBI 0:4edb816d21e1 364 //FIX: 27/05/2010: We do not want to deallocate the socket while some data might still be readable
ICTFBI 0:4edb816d21e1 365 //REMOVED: close();
ICTFBI 0:4edb816d21e1 366
ICTFBI 0:4edb816d21e1 367 //However we do not want to close the socket yet
ICTFBI 0:4edb816d21e1 368
ICTFBI 0:4edb816d21e1 369 queueEvent(NETTCPSOCKET_DISCONNECTED);
ICTFBI 0:4edb816d21e1 370 return ERR_OK;
ICTFBI 0:4edb816d21e1 371 }
ICTFBI 0:4edb816d21e1 372
ICTFBI 0:4edb816d21e1 373 //We asserted that p is a valid pointer
ICTFBI 0:4edb816d21e1 374
ICTFBI 0:4edb816d21e1 375 //New data processing
ICTFBI 0:4edb816d21e1 376 tcp_recved( tpcb, p->tot_len); //Acknowledge the reception
ICTFBI 0:4edb816d21e1 377 if(!m_pReadPbuf)
ICTFBI 0:4edb816d21e1 378 {
ICTFBI 0:4edb816d21e1 379 m_pReadPbuf = p;
ICTFBI 0:4edb816d21e1 380 queueEvent(NETTCPSOCKET_READABLE);
ICTFBI 0:4edb816d21e1 381 }
ICTFBI 0:4edb816d21e1 382 else
ICTFBI 0:4edb816d21e1 383 {
ICTFBI 0:4edb816d21e1 384 pbuf_cat((pbuf*)m_pReadPbuf, p); //m_pReadPbuf is not empty, tail p to it and drop our ref
ICTFBI 0:4edb816d21e1 385 //No need to queue an event in that case since the read buf has not been processed yet
ICTFBI 0:4edb816d21e1 386 }
ICTFBI 0:4edb816d21e1 387 return ERR_OK;
ICTFBI 0:4edb816d21e1 388 }
ICTFBI 0:4edb816d21e1 389
ICTFBI 0:4edb816d21e1 390
ICTFBI 0:4edb816d21e1 391 void LwipNetTcpSocket::cleanUp() //Flush input buffer
ICTFBI 0:4edb816d21e1 392 {
ICTFBI 0:4edb816d21e1 393 //Ensure that further error won't be followed to this inst (which can be destroyed)
ICTFBI 0:4edb816d21e1 394 if( m_pPcb )
ICTFBI 0:4edb816d21e1 395 {
ICTFBI 0:4edb816d21e1 396 tcp_arg( (tcp_pcb*) m_pPcb, (void*) NULL );
ICTFBI 0:4edb816d21e1 397 tcp_recv( (tcp_pcb*) m_pPcb, NULL );
ICTFBI 0:4edb816d21e1 398 tcp_sent((tcp_pcb*) m_pPcb, NULL );
ICTFBI 0:4edb816d21e1 399 tcp_err( (tcp_pcb*) m_pPcb, NULL );
ICTFBI 0:4edb816d21e1 400 }
ICTFBI 0:4edb816d21e1 401
ICTFBI 0:4edb816d21e1 402 if( m_pReadPbuf )
ICTFBI 0:4edb816d21e1 403 {
ICTFBI 0:4edb816d21e1 404 DBG("Deallocating unread data.\n");
ICTFBI 0:4edb816d21e1 405 pbuf_free((pbuf*)m_pReadPbuf); //Free all unread data
ICTFBI 0:4edb816d21e1 406 m_pReadPbuf = NULL;
ICTFBI 0:4edb816d21e1 407 recv(NULL,0); //Update recv ptr position
ICTFBI 0:4edb816d21e1 408 }
ICTFBI 0:4edb816d21e1 409 }
ICTFBI 0:4edb816d21e1 410
ICTFBI 0:4edb816d21e1 411 // Static callbacks from LwIp
ICTFBI 0:4edb816d21e1 412
ICTFBI 0:4edb816d21e1 413 err_t LwipNetTcpSocket::sAcceptCb(void *arg, struct tcp_pcb *newpcb, err_t err)
ICTFBI 0:4edb816d21e1 414 {
ICTFBI 0:4edb816d21e1 415 LwipNetTcpSocket* pMe = (LwipNetTcpSocket*) arg;
ICTFBI 0:4edb816d21e1 416 return pMe->acceptCb( newpcb, err );
ICTFBI 0:4edb816d21e1 417 }
ICTFBI 0:4edb816d21e1 418
ICTFBI 0:4edb816d21e1 419 err_t LwipNetTcpSocket::sConnectedCb(void *arg, struct tcp_pcb *tpcb, err_t err)
ICTFBI 0:4edb816d21e1 420 {
ICTFBI 0:4edb816d21e1 421 LwipNetTcpSocket* pMe = (LwipNetTcpSocket*) arg;
ICTFBI 0:4edb816d21e1 422 return pMe->connectedCb( tpcb, err );
ICTFBI 0:4edb816d21e1 423 }
ICTFBI 0:4edb816d21e1 424
ICTFBI 0:4edb816d21e1 425 void LwipNetTcpSocket::sErrCb(void *arg, err_t err)
ICTFBI 0:4edb816d21e1 426 {
ICTFBI 0:4edb816d21e1 427 if( !arg )
ICTFBI 0:4edb816d21e1 428 {
ICTFBI 0:4edb816d21e1 429 DBG("NetTcpSocket - Error %d in LwipNetTcpSocket::sErrCb.\n", err);
ICTFBI 0:4edb816d21e1 430 return; //The socket has been destroyed, discard error
ICTFBI 0:4edb816d21e1 431 }
ICTFBI 0:4edb816d21e1 432 LwipNetTcpSocket* pMe = (LwipNetTcpSocket*) arg;
ICTFBI 0:4edb816d21e1 433 return pMe->errCb( err );
ICTFBI 0:4edb816d21e1 434 }
ICTFBI 0:4edb816d21e1 435
ICTFBI 0:4edb816d21e1 436 err_t LwipNetTcpSocket::sSentCb(void *arg, struct tcp_pcb *tpcb, u16_t len)
ICTFBI 0:4edb816d21e1 437 {
ICTFBI 0:4edb816d21e1 438 LwipNetTcpSocket* pMe = (LwipNetTcpSocket*) arg;
ICTFBI 0:4edb816d21e1 439 return pMe->sentCb( tpcb, len );
ICTFBI 0:4edb816d21e1 440 }
ICTFBI 0:4edb816d21e1 441
ICTFBI 0:4edb816d21e1 442 err_t LwipNetTcpSocket::sRecvCb(void *arg, struct tcp_pcb *tpcb, struct pbuf *p, err_t err)
ICTFBI 0:4edb816d21e1 443 {
ICTFBI 0:4edb816d21e1 444 if( tpcb->flags & TF_RXCLOSED )
ICTFBI 0:4edb816d21e1 445 {
ICTFBI 0:4edb816d21e1 446 //The Pcb is in a closing state
ICTFBI 0:4edb816d21e1 447 //Discard that data here since we might have destroyed the corresponding socket object
ICTFBI 0:4edb816d21e1 448 tcp_recved( tpcb, p->tot_len);
ICTFBI 0:4edb816d21e1 449 pbuf_free( p );
ICTFBI 0:4edb816d21e1 450 return ERR_OK;
ICTFBI 0:4edb816d21e1 451 }
ICTFBI 0:4edb816d21e1 452 LwipNetTcpSocket* pMe = (LwipNetTcpSocket*) arg;
ICTFBI 0:4edb816d21e1 453 return pMe->recvCb( tpcb, p, err );
ICTFBI 0:4edb816d21e1 454 }
ICTFBI 0:4edb816d21e1 455
ICTFBI 0:4edb816d21e1 456 #endif