Bonjour - First step: AutoIP compiled in and working

Users » darran » Code » Bonjour

Darran Shepherd / Mbed 2 deprecated Bonjour

First step: AutoIP compiled in and working

if/lwip/lwipNetUdpSocket.cpp@1:4218cacaf696, 2010-06-18 (annotated)

Committer:: darran
Date:: Fri Jun 18 15:54:21 2010 +0000
Revision:: 1:4218cacaf696
Parent:: 0:55a05330f8cc

Who changed what in which revision?

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

Repository toolbox

Export to desktop IDE

Build repository

Repository details

Type:	Program
Mbed OS support:	Mbed 2 deprecated
Created:	18 Jun 2010
Imports:	7
Forks:	0
Commits:	2
Dependents:	0
Dependencies:	1
Followers:	1

if/lwip/lwipNetUdpSocket.cpp@1:4218cacaf696, 2010-06-18 (annotated)

Who changed what in which revision?

Repository toolbox

Repository details

Important Information for this Arm website

Access Warning