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/lwipNetTcpSocket.cpp@2:55b6fd49b738, 2014-12-18 (annotated)

Committer:
rikvandyck
Date:
Thu Dec 18 10:43:07 2014 +0000
Revision:
2:55b6fd49b738
Parent:
0:e1edd52b1ee2 
Project_Embedded_C

Who changed what in which revision?

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