Nino Der Kinderen / Mbed 2 deprecated InetTest

Embedded C project:18/12/2014

Dependencies:   DS1307 TextLCD mbed

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