Net - | Mbed

Users » mbed714 » Code » Net

if/lwip/lwipNetUdpSocket.cpp@0:d616ece2d859, 2010-09-18 (annotated)

Committer:: mbed714
Date:: Sat Sep 18 23:05:49 2010 +0000
Revision:: 0:d616ece2d859

Who changed what in which revision?

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

Repository toolbox

Export to desktop IDE

Repository details

Type:	Library
Created:	18 Sep 2010
Imports:	1
Forks:	0
Commits:	1
Dependents:	0
Dependencies:	0
Followers:	1

This repository is Public (Unlisted).

if/lwip/lwipNetUdpSocket.cpp@0:d616ece2d859, 2010-09-18 (annotated)

Who changed what in which revision?

Repository toolbox

Repository details

Important Information for this Arm website

Access Warning