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@0:55a05330f8cc, 2010-06-18 (annotated)

Committer:: darran
Date:: Fri Jun 18 09:11:35 2010 +0000
Revision:: 0:55a05330f8cc
Child:: 1:4218cacaf696

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	0:55a05330f8cc	33	LwipNetUdpSocket::LwipNetUdpSocket(udp_pcb* pPcb /= NULL/) : 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	0:55a05330f8cc	35	DBG("\r\nNew NetUdpSocket %p\r\n", (void*)this);
darran	0:55a05330f8cc	36	if(!m_pPcb)
darran	0:55a05330f8cc	37	m_pPcb = udp_new();
darran	0:55a05330f8cc	38	/* if(m_pPcb)
darran	0:55a05330f8cc	39	{
darran	0:55a05330f8cc	40	//Setup callback
darran	0:55a05330f8cc	41	// udp_recv( (udp_pcb) m_pPcb, LwipNetUdpSocket::sRecvCb, (void) this );
darran	0:55a05330f8cc	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	0:55a05330f8cc	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	0:55a05330f8cc	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	0:55a05330f8cc	72	pbuf* p = pbuf_alloc(PBUF_RAW, len, PBUF_RAM);
darran	0:55a05330f8cc	73	if( !p )
darran	0:55a05330f8cc	74	return NETUDPSOCKET_MEM;
darran	0:55a05330f8cc	75	memcpy (p->payload, (void*)buf, len);
darran	0:55a05330f8cc	76	//udp_connect( (udp_pcb*) m_pPcb, &(pHost->getIp().getStruct()), pHost->getPort() );
darran	0:55a05330f8cc	77	//err_t err = udp_send( (udp_pcb*) m_pPcb, p);
darran	0:55a05330f8cc	78	err_t err = udp_sendto( (udp_pcb*) m_pPcb, p, &(pHost->getIp().getStruct()), pHost->getPort() );
darran	0:55a05330f8cc	79	pbuf_free( p );
darran	0:55a05330f8cc	80	if(err)
darran	0:55a05330f8cc	81	return NETUDPSOCKET_SETUP; //Connection problem
darran	0:55a05330f8cc	82	DBG("\r\n%d bytes sent in UDP Socket.\r\n", len);
darran	0:55a05330f8cc	83	return len;
darran	0:55a05330f8cc	84	}
darran	0:55a05330f8cc	85
darran	0:55a05330f8cc	86	int /if < 0 : NetUdpSocketErr/ LwipNetUdpSocket::recvfrom(char* buf, int len, Host* pHost)
darran	0:55a05330f8cc	87	{
darran	0:55a05330f8cc	88	if( !m_pPcb ) //Pcb doesn't exist (anymore)
darran	0:55a05330f8cc	89	return NETUDPSOCKET_MEM;
darran	0:55a05330f8cc	90	int inLen = 0;
darran	0:55a05330f8cc	91	int cpyLen = 0;
darran	0:55a05330f8cc	92
darran	0:55a05330f8cc	93	static int rmgLen = 0;
darran	0:55a05330f8cc	94	//Contains the remaining len in this pbuf
darran	0:55a05330f8cc	95
darran	0:55a05330f8cc	96	if( m_lInPkt.empty() )
darran	0:55a05330f8cc	97	return 0;
darran	0:55a05330f8cc	98
darran	0:55a05330f8cc	99	pbuf* pBuf = (pbuf*) m_lInPkt.front().pBuf;
darran	0:55a05330f8cc	100
darran	0:55a05330f8cc	101	if(pHost)
darran	0:55a05330f8cc	102	*pHost = Host( IpAddr(&m_lInPkt.front().addr), m_lInPkt.front().port );
darran	0:55a05330f8cc	103
darran	0:55a05330f8cc	104	if( !pBuf )
darran	0:55a05330f8cc	105	{
darran	0:55a05330f8cc	106	rmgLen = 0;
darran	0:55a05330f8cc	107	return 0;
darran	0:55a05330f8cc	108	}
darran	0:55a05330f8cc	109
darran	0:55a05330f8cc	110	if ( !rmgLen ) //We did not know m_pReadPbuf->len last time we called this fn
darran	0:55a05330f8cc	111	{
darran	0:55a05330f8cc	112	rmgLen = pBuf->len;
darran	0:55a05330f8cc	113	}
darran	0:55a05330f8cc	114
darran	0:55a05330f8cc	115	while ( inLen < len )
darran	0:55a05330f8cc	116	{
darran	0:55a05330f8cc	117	cpyLen = MIN( (len - inLen), rmgLen ); //Remaining len to copy, remaining len in THIS pbuf
darran	0:55a05330f8cc	118	memcpy((void)buf, (void)((char*)(pBuf->payload) + (pBuf->len - rmgLen)), cpyLen);
darran	0:55a05330f8cc	119	inLen += cpyLen;
darran	0:55a05330f8cc	120	buf += cpyLen;
darran	0:55a05330f8cc	121
darran	0:55a05330f8cc	122	rmgLen = rmgLen - cpyLen; //Update rmgLen
darran	0:55a05330f8cc	123
darran	0:55a05330f8cc	124	if( rmgLen > 0 )
darran	0:55a05330f8cc	125	{
darran	0:55a05330f8cc	126	//We did not read this pbuf completely, so let's save it's pos & return
darran	0:55a05330f8cc	127	break;
darran	0:55a05330f8cc	128	}
darran	0:55a05330f8cc	129
darran	0:55a05330f8cc	130	if(pBuf->next)
darran	0:55a05330f8cc	131	{
darran	0:55a05330f8cc	132	pbuf* pNextPBuf = pBuf->next;
darran	0:55a05330f8cc	133	pBuf->next = NULL; //So that it is not freed as well
darran	0:55a05330f8cc	134	//We get the reference to pNextPBuf from m_pReadPbuf
darran	0:55a05330f8cc	135	pbuf_free((pbuf*)pBuf);
darran	0:55a05330f8cc	136	pBuf = pNextPBuf;
darran	0:55a05330f8cc	137	rmgLen = pBuf->len;
darran	0:55a05330f8cc	138	}
darran	0:55a05330f8cc	139	else
darran	0:55a05330f8cc	140	{
darran	0:55a05330f8cc	141	pbuf_free((pbuf*)pBuf);
darran	0:55a05330f8cc	142	pBuf = NULL;
darran	0:55a05330f8cc	143	rmgLen = 0;
darran	0:55a05330f8cc	144	m_lInPkt.pop_front();
darran	0:55a05330f8cc	145	break; //No more data to read
darran	0:55a05330f8cc	146	}
darran	0:55a05330f8cc	147	}
darran	0:55a05330f8cc	148
darran	0:55a05330f8cc	149	return inLen;
darran	0:55a05330f8cc	150	}
darran	0:55a05330f8cc	151
darran	0:55a05330f8cc	152	NetUdpSocketErr LwipNetUdpSocket::close()
darran	0:55a05330f8cc	153	{
darran	0:55a05330f8cc	154	//DBG("\r\nLwipNetUdpSocket::close() : Closing...\r\n");
darran	0:55a05330f8cc	155
darran	0:55a05330f8cc	156	if(m_closed)
darran	0:55a05330f8cc	157	return NETUDPSOCKET_OK; //Already being closed
darran	0:55a05330f8cc	158	m_closed = true;
darran	0:55a05330f8cc	159
darran	0:55a05330f8cc	160	if( !m_pPcb ) //Pcb doesn't exist (anymore)
darran	0:55a05330f8cc	161	return NETUDPSOCKET_MEM;
darran	0:55a05330f8cc	162
darran	0:55a05330f8cc	163	//Cleanup incoming data
darran	0:55a05330f8cc	164	cleanUp();
darran	0:55a05330f8cc	165
darran	0:55a05330f8cc	166	udp_remove( (udp_pcb*) m_pPcb);
darran	0:55a05330f8cc	167
darran	0:55a05330f8cc	168	m_pPcb = NULL;
darran	0:55a05330f8cc	169	return NETUDPSOCKET_OK;
darran	0:55a05330f8cc	170	}
darran	0:55a05330f8cc	171
darran	0:55a05330f8cc	172	NetUdpSocketErr LwipNetUdpSocket::poll()
darran	0:55a05330f8cc	173	{
darran	0:55a05330f8cc	174	NetUdpSocket::flushEvents();
darran	0:55a05330f8cc	175	return NETUDPSOCKET_OK;
darran	0:55a05330f8cc	176	}
darran	0:55a05330f8cc	177
darran	0:55a05330f8cc	178	// Callbacks events
darran	0:55a05330f8cc	179
darran	0:55a05330f8cc	180	void LwipNetUdpSocket::recvCb(udp_pcb* pcb, struct pbuf* p, ip_addr_t* addr, u16_t port)
darran	0:55a05330f8cc	181	{
darran	0:55a05330f8cc	182	DBG("\r\n Packet of length %d arrived in UDP Socket.\r\n", p->tot_len);
darran	0:55a05330f8cc	183	list<InPacket>::iterator it;
darran	0:55a05330f8cc	184	for ( it = m_lInPkt.begin(); it != m_lInPkt.end(); it++ )
darran	0:55a05330f8cc	185	{
darran	0:55a05330f8cc	186	if( ip_addr_cmp((&((it).addr)), addr) && ((it).port == port) )
darran	0:55a05330f8cc	187	{
darran	0:55a05330f8cc	188	//Let's tail this packet to the previous one
darran	0:55a05330f8cc	189	pbuf_cat((pbuf)(it).pBuf, p);
darran	0:55a05330f8cc	190	//No need to queue an event in that case since the read buf has not been processed yet
darran	0:55a05330f8cc	191	return;
darran	0:55a05330f8cc	192	}
darran	0:55a05330f8cc	193	}
darran	0:55a05330f8cc	194
darran	0:55a05330f8cc	195	//New host, add a packet to the queue
darran	0:55a05330f8cc	196	InPacket pkt;
darran	0:55a05330f8cc	197	pkt.pBuf = p;
darran	0:55a05330f8cc	198	pkt.addr = *addr;
darran	0:55a05330f8cc	199	pkt.port = port;
darran	0:55a05330f8cc	200	m_lInPkt.push_back(pkt);
darran	0:55a05330f8cc	201
darran	0:55a05330f8cc	202	queueEvent(NETUDPSOCKET_READABLE);
darran	0:55a05330f8cc	203	}
darran	0:55a05330f8cc	204
darran	0:55a05330f8cc	205	void LwipNetUdpSocket::cleanUp() //Flush input buffer
darran	0:55a05330f8cc	206	{
darran	0:55a05330f8cc	207	list<InPacket>::iterator it;
darran	0:55a05330f8cc	208	for ( it = m_lInPkt.begin(); it != m_lInPkt.end(); it++ )
darran	0:55a05330f8cc	209	{
darran	0:55a05330f8cc	210	//Free buf
darran	0:55a05330f8cc	211	pbuf_free((pbuf)(it).pBuf);
darran	0:55a05330f8cc	212	}
darran	0:55a05330f8cc	213	recvfrom(NULL, 0, NULL);
darran	0:55a05330f8cc	214	m_lInPkt.clear();
darran	0:55a05330f8cc	215	}
darran	0:55a05330f8cc	216
darran	0:55a05330f8cc	217	// Static callback from LwIp
darran	0:55a05330f8cc	218
darran	0:55a05330f8cc	219	void LwipNetUdpSocket::sRecvCb(void arg, struct udp_pcb pcb, struct pbuf p, ip_addr_t addr, u16_t port)
darran	0:55a05330f8cc	220	{
darran	0:55a05330f8cc	221	LwipNetUdpSocket* pMe = (LwipNetUdpSocket*) arg;
darran	0:55a05330f8cc	222	return pMe->recvCb( pcb, p, addr, port );
darran	0:55a05330f8cc	223	}
darran	0:55a05330f8cc	224
darran	0:55a05330f8cc	225	#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@0:55a05330f8cc, 2010-06-18 (annotated)

Who changed what in which revision?

Repository toolbox

Repository details

Important Information for this Arm website

Access Warning