PicoTCP - Free (GPLv2) TCP/IP stack developed by TASS Belgi…

Users » daniele » Code » PicoTCP

Free (GPLv2) TCP/IP stack developed by TASS Belgium

Dependents: lpc1768-picotcp-demo ZeroMQ_PicoTCP_Publisher_demo TCPSocket_HelloWorld_PicoTCP Pico_TCP_UDP_Test ... more

PicoTCP. Copyright (c) 2013 TASS Belgium NV.

Released under the GNU General Public License, version 2.

Different licensing models may exist, at the sole discretion of the Copyright holders.

Official homepage: http://www.picotcp.com

Bug tracker: https://github.com/tass-belgium/picotcp/issues

Development steps:

~~initial integration with mbed RTOS~~
~~generic mbed Ethernet driver~~
~~high performance NXP LPC1768 specific Ethernet driver~~
~~Multi-threading support for mbed RTOS~~
~~Berkeley sockets and integration with the New Socket API~~
~~Fork of the apps running on top of the New Socket API~~
~~Scheduling optimizations~~
Debugging/benchmarking/testing

Demo application (measuring TCP sender performance):

Import program lpc1768-picotcp-demo

A PicoTCP demo app testing the ethernet throughput on the lpc1768 mbed board.

stack/pico_socket.c@49:4b404dd2c97a, 2013-08-05 (annotated)

Committer:: tass
Date:: Mon Aug 05 12:58:56 2013 +0000
Revision:: 49:4b404dd2c97a
Parent:: 48:40fc4462265c
Child:: 51:ab4529a384a6

updated from master branch.

Who changed what in which revision?

User	Revision	Line number	New contents of line
daniele	29:1a47b7151851	1	/*********************************************************************
daniele	29:1a47b7151851	2	PicoTCP. Copyright (c) 2012 TASS Belgium NV. Some rights reserved.
daniele	29:1a47b7151851	3	See LICENSE and COPYING for usage.
daniele	29:1a47b7151851	4
daniele	29:1a47b7151851	5
daniele	29:1a47b7151851	6	Authors: Daniele Lacamera
daniele	29:1a47b7151851	7	*********************************************************************/
daniele	29:1a47b7151851	8
daniele	29:1a47b7151851	9
daniele	29:1a47b7151851	10	#include "pico_config.h"
daniele	29:1a47b7151851	11	#include "pico_queue.h"
daniele	29:1a47b7151851	12	#include "pico_socket.h"
daniele	29:1a47b7151851	13	#include "pico_ipv4.h"
daniele	29:1a47b7151851	14	#include "pico_ipv6.h"
daniele	29:1a47b7151851	15	#include "pico_udp.h"
daniele	29:1a47b7151851	16	#include "pico_tcp.h"
daniele	29:1a47b7151851	17	#include "pico_stack.h"
daniele	29:1a47b7151851	18	#include "pico_icmp4.h"
daniele	29:1a47b7151851	19	#include "pico_nat.h"
daniele	29:1a47b7151851	20	#include "pico_tree.h"
daniele	29:1a47b7151851	21	#include "pico_device.h"
daniele	29:1a47b7151851	22
daniele	29:1a47b7151851	23	#if defined (PICO_SUPPORT_IPV4) \|\| defined (PICO_SUPPORT_IPV6)
daniele	29:1a47b7151851	24	#if defined (PICO_SUPPORT_TCP) \|\| defined (PICO_SUPPORT_UDP)
daniele	29:1a47b7151851	25
daniele	29:1a47b7151851	26
tass	48:40fc4462265c	27	#define PROTO(s) ((s)->proto->proto_number)
tass	48:40fc4462265c	28	#define PICO_SOCKET4_MTU 1480 /* Ethernet MTU(1500) - IP header size(20) */
tass	48:40fc4462265c	29	#define PICO_SOCKET6_MTU 1460 /* Ethernet MTU(1500) - IP header size(40) */
tass	48:40fc4462265c	30
daniele	29:1a47b7151851	31	#ifdef PICO_SUPPORT_MUTEX
daniele	29:1a47b7151851	32	static void * Mutex = NULL;
daniele	29:1a47b7151851	33	#define LOCK(x) {\
daniele	29:1a47b7151851	34	if (x == NULL) \
daniele	29:1a47b7151851	35	x = pico_mutex_init(); \
daniele	29:1a47b7151851	36	pico_mutex_lock(x); \
daniele	29:1a47b7151851	37	}
daniele	29:1a47b7151851	38	#define UNLOCK(x) pico_mutex_unlock(x);
daniele	29:1a47b7151851	39
daniele	29:1a47b7151851	40	#else
daniele	29:1a47b7151851	41	#define LOCK(x) do{}while(0)
daniele	29:1a47b7151851	42	#define UNLOCK(x) do{}while(0)
daniele	29:1a47b7151851	43	#endif
daniele	29:1a47b7151851	44
daniele	29:1a47b7151851	45
daniele	29:1a47b7151851	46	#define PROTO(s) ((s)->proto->proto_number)
daniele	29:1a47b7151851	47
daniele	29:1a47b7151851	48	#ifdef PICO_SUPPORT_TCP
daniele	29:1a47b7151851	49	# define IS_NAGLE_ENABLED(s) (!(!(!(s->opt_flags & (1 << PICO_SOCKET_OPT_TCPNODELAY)))))
daniele	29:1a47b7151851	50	#endif
daniele	29:1a47b7151851	51
daniele	29:1a47b7151851	52	#define PICO_SOCKET_MTU 1480 /* Ethernet MTU(1500) - IP header size(20) */
daniele	29:1a47b7151851	53
daniele	29:1a47b7151851	54	#ifdef PICO_SUPPORT_IPV4
daniele	29:1a47b7151851	55	# define IS_SOCK_IPV4(s) ((s->net == &pico_proto_ipv4))
daniele	29:1a47b7151851	56	#else
daniele	29:1a47b7151851	57	# define IS_SOCK_IPV4(s) (0)
daniele	29:1a47b7151851	58	#endif
daniele	29:1a47b7151851	59
daniele	29:1a47b7151851	60	#ifdef PICO_SUPPORT_IPV6
daniele	29:1a47b7151851	61	# define IS_SOCK_IPV6(s) ((s->net == &pico_proto_ipv6))
daniele	29:1a47b7151851	62	#else
daniele	29:1a47b7151851	63	# define IS_SOCK_IPV6(s) (0)
daniele	29:1a47b7151851	64	#endif
daniele	29:1a47b7151851	65
daniele	29:1a47b7151851	66	#ifdef PICO_SUPPORT_IPFRAG
daniele	29:1a47b7151851	67	# define frag_dbg(...) do{}while(0)
daniele	29:1a47b7151851	68	#endif
daniele	29:1a47b7151851	69
daniele	29:1a47b7151851	70	#ifdef PICO_SUPPORT_MCAST
daniele	29:1a47b7151851	71	# define so_mcast_dbg(...) do{}while(0) /* ip_mcast_dbg in pico_ipv4.c */
daniele	29:1a47b7151851	72	#endif
daniele	29:1a47b7151851	73
daniele	29:1a47b7151851	74	static struct pico_sockport sp_udp = NULL ,sp_tcp = NULL;
daniele	29:1a47b7151851	75
daniele	29:1a47b7151851	76	struct pico_frame pico_socket_frame_alloc(struct pico_socket s, int len);
daniele	29:1a47b7151851	77
daniele	29:1a47b7151851	78	static int socket_cmp(void * ka, void * kb)
daniele	29:1a47b7151851	79	{
daniele	29:1a47b7151851	80	struct pico_socket a = ka, b = kb;
daniele	29:1a47b7151851	81	int a_is_ip6 = is_sock_ipv6(a);
daniele	29:1a47b7151851	82	int b_is_ip6 = is_sock_ipv6(b);
daniele	29:1a47b7151851	83
daniele	29:1a47b7151851	84	int diff;
daniele	29:1a47b7151851	85
daniele	29:1a47b7151851	86	/* First, order by network ver */
daniele	29:1a47b7151851	87	if (a_is_ip6 < b_is_ip6)
daniele	29:1a47b7151851	88	return -1;
daniele	29:1a47b7151851	89	if (a_is_ip6 > b_is_ip6)
daniele	29:1a47b7151851	90	return 1;
daniele	29:1a47b7151851	91
daniele	29:1a47b7151851	92	/* If either socket is PICO_IPV4_INADDR_ANY mode, skip local address comparison */
daniele	29:1a47b7151851	93
daniele	29:1a47b7151851	94	/* At this point, sort by local host */
daniele	29:1a47b7151851	95
daniele	29:1a47b7151851	96	if (0) {
daniele	29:1a47b7151851	97	#ifdef PICO_SUPPORT_IPV6
daniele	29:1a47b7151851	98	} else if (a_is_ip6) {
daniele	29:1a47b7151851	99	if ((memcmp(a->local_addr.ip6.addr, PICO_IP6_ANY, PICO_SIZE_IP6)==0) \|\| memcmp((b->local_addr.ip6.addr, PICO_IP6_ANY, PICO_SIZE_IP6) == 0))
daniele	29:1a47b7151851	100	diff = 0;
daniele	29:1a47b7151851	101	else
daniele	29:1a47b7151851	102	diff = memcmp(a->local_addr.ip6.addr, b->local_addr.ip6.addr, PICO_SIZE_IP6);
daniele	29:1a47b7151851	103	#endif
daniele	29:1a47b7151851	104	} else {
daniele	29:1a47b7151851	105	if ((a->local_addr.ip4.addr == PICO_IP4_ANY) \|\| (b->local_addr.ip4.addr == PICO_IP4_ANY))
daniele	29:1a47b7151851	106	diff = 0;
daniele	29:1a47b7151851	107	else
daniele	29:1a47b7151851	108	diff = a->local_addr.ip4.addr - b->local_addr.ip4.addr;
daniele	29:1a47b7151851	109	}
daniele	29:1a47b7151851	110
daniele	29:1a47b7151851	111	if (diff)
daniele	29:1a47b7151851	112	return diff;
daniele	29:1a47b7151851	113
daniele	29:1a47b7151851	114
daniele	29:1a47b7151851	115	/* Sort by remote host */
daniele	29:1a47b7151851	116	if (a_is_ip6)
daniele	29:1a47b7151851	117	diff = memcmp(a->remote_addr.ip6.addr, b->remote_addr.ip6.addr, PICO_SIZE_IP6);
daniele	29:1a47b7151851	118	else
daniele	29:1a47b7151851	119	diff = a->remote_addr.ip4.addr - b->remote_addr.ip4.addr;
daniele	29:1a47b7151851	120
daniele	29:1a47b7151851	121	if (diff)
daniele	29:1a47b7151851	122	return diff;
daniele	29:1a47b7151851	123
daniele	29:1a47b7151851	124	/* And finally by remote port. The two sockets are coincident if the quad is the same. */
daniele	29:1a47b7151851	125	return b->remote_port - a->remote_port;
daniele	29:1a47b7151851	126	}
daniele	29:1a47b7151851	127
daniele	29:1a47b7151851	128	struct pico_sockport
daniele	29:1a47b7151851	129	{
daniele	29:1a47b7151851	130	struct pico_tree socks; // how you make the connection ?
daniele	29:1a47b7151851	131	uint16_t number;
daniele	29:1a47b7151851	132	uint16_t proto;
daniele	29:1a47b7151851	133	};
daniele	29:1a47b7151851	134
daniele	29:1a47b7151851	135	#define INIT_SOCKPORT { {&LEAF , socket_cmp}, 0, 0 }
daniele	29:1a47b7151851	136
daniele	29:1a47b7151851	137	int sockport_cmp(void * ka, void * kb)
daniele	29:1a47b7151851	138	{
daniele	29:1a47b7151851	139	struct pico_sockport a = ka, b = kb;
daniele	29:1a47b7151851	140	if (a->number < b->number)
daniele	29:1a47b7151851	141	return -1;
daniele	29:1a47b7151851	142	if (a->number > b->number)
daniele	29:1a47b7151851	143	return 1;
daniele	29:1a47b7151851	144	return 0;
daniele	29:1a47b7151851	145	}
daniele	29:1a47b7151851	146
daniele	29:1a47b7151851	147	PICO_TREE_DECLARE(UDPTable,sockport_cmp);
daniele	29:1a47b7151851	148	PICO_TREE_DECLARE(TCPTable,sockport_cmp);
daniele	29:1a47b7151851	149
daniele	29:1a47b7151851	150	#ifdef PICO_SUPPORT_MCAST
daniele	29:1a47b7151851	151	/* socket
daniele	29:1a47b7151851	152	* \|
daniele	29:1a47b7151851	153	* MCASTListen
daniele	29:1a47b7151851	154	* \| \| \|
daniele	29:1a47b7151851	155	* ------------ \| ------------
daniele	29:1a47b7151851	156	* \| \| \|
daniele	29:1a47b7151851	157	* MCASTSources MCASTSources MCASTSources
daniele	29:1a47b7151851	158	* \| \| \| \| \| \| \| \| \| \| \| \|
daniele	29:1a47b7151851	159	* S S S S S S S S S S S S
daniele	29:1a47b7151851	160	*
daniele	29:1a47b7151851	161	* MCASTListen: RBTree(mcast_link, mcast_group)
daniele	29:1a47b7151851	162	* MCASTSources: RBTree(source)
daniele	29:1a47b7151851	163	*/
daniele	29:1a47b7151851	164	struct pico_mcast_listen
daniele	29:1a47b7151851	165	{
daniele	29:1a47b7151851	166	uint8_t filter_mode;
daniele	29:1a47b7151851	167	struct pico_ip4 mcast_link;
daniele	29:1a47b7151851	168	struct pico_ip4 mcast_group;
daniele	29:1a47b7151851	169	struct pico_tree MCASTSources;
daniele	29:1a47b7151851	170	};
daniele	29:1a47b7151851	171
daniele	29:1a47b7151851	172	static int mcast_listen_cmp(void ka, void kb)
daniele	29:1a47b7151851	173	{
daniele	29:1a47b7151851	174	struct pico_mcast_listen a = ka, b = kb;
daniele	29:1a47b7151851	175	if (a->mcast_group.addr < b->mcast_group.addr)
daniele	29:1a47b7151851	176	return -1;
daniele	29:1a47b7151851	177	if (a->mcast_group.addr > b->mcast_group.addr)
daniele	29:1a47b7151851	178	return 1;
daniele	29:1a47b7151851	179
daniele	29:1a47b7151851	180	if (a->mcast_link.addr < b->mcast_link.addr)
daniele	29:1a47b7151851	181	return -1;
daniele	29:1a47b7151851	182	if (a->mcast_link.addr > b->mcast_link.addr)
daniele	29:1a47b7151851	183	return 1;
daniele	29:1a47b7151851	184
daniele	29:1a47b7151851	185	return 0;
daniele	29:1a47b7151851	186	}
daniele	29:1a47b7151851	187
daniele	29:1a47b7151851	188	static int mcast_sources_cmp(void ka, void kb)
daniele	29:1a47b7151851	189	{
daniele	29:1a47b7151851	190	struct pico_ip4 a = ka, b = kb;
daniele	29:1a47b7151851	191	if (a->addr < b->addr)
daniele	29:1a47b7151851	192	return -1;
daniele	29:1a47b7151851	193	if (a->addr > b->addr)
daniele	29:1a47b7151851	194	return 1;
daniele	29:1a47b7151851	195	return 0;
daniele	29:1a47b7151851	196	}
daniele	29:1a47b7151851	197
daniele	29:1a47b7151851	198	static int mcast_socket_cmp(void ka, void kb)
daniele	29:1a47b7151851	199	{
daniele	29:1a47b7151851	200	struct pico_socket a = ka, b = kb;
daniele	29:1a47b7151851	201	if (a < b)
daniele	29:1a47b7151851	202	return -1;
daniele	29:1a47b7151851	203	if (a > b)
daniele	29:1a47b7151851	204	return 1;
daniele	29:1a47b7151851	205	return 0;
daniele	29:1a47b7151851	206	}
daniele	29:1a47b7151851	207	/* gather all multicast sockets to hasten filter aggregation */
daniele	29:1a47b7151851	208	PICO_TREE_DECLARE(MCASTSockets, mcast_socket_cmp);
daniele	29:1a47b7151851	209
daniele	29:1a47b7151851	210	static int mcast_filter_cmp(void ka, void kb)
daniele	29:1a47b7151851	211	{
daniele	29:1a47b7151851	212	struct pico_ip4 a = ka, b = kb;
daniele	29:1a47b7151851	213	if (a->addr < b->addr)
daniele	29:1a47b7151851	214	return -1;
daniele	29:1a47b7151851	215	if (a->addr > b->addr)
daniele	29:1a47b7151851	216	return 1;
daniele	29:1a47b7151851	217	return 0;
daniele	29:1a47b7151851	218	}
daniele	29:1a47b7151851	219	/* gather sources to be filtered */
daniele	29:1a47b7151851	220	PICO_TREE_DECLARE(MCASTFilter, mcast_filter_cmp);
daniele	29:1a47b7151851	221
daniele	29:1a47b7151851	222	/* MCASTFilter will be empty if no socket is listening on mcast_group on mcast_link anymore */
daniele	29:1a47b7151851	223	static int pico_socket_aggregate_mcastfilters(struct pico_ip4 mcast_link, struct pico_ip4 mcast_group)
daniele	29:1a47b7151851	224	{
daniele	29:1a47b7151851	225	uint8_t filter_mode = PICO_IP_MULTICAST_INCLUDE;
daniele	29:1a47b7151851	226	struct pico_mcast_listen *listen = NULL, ltest = {0};
daniele	29:1a47b7151851	227	struct pico_ip4 *source = NULL;
daniele	29:1a47b7151851	228	struct pico_socket *mcast_sock = NULL;
daniele	29:1a47b7151851	229	struct pico_tree_node index = NULL, _tmp = NULL, index2 = NULL, _tmp2 = NULL;
daniele	29:1a47b7151851	230
daniele	29:1a47b7151851	231	ltest.mcast_link = *mcast_link;
daniele	29:1a47b7151851	232	ltest.mcast_group = *mcast_group;
daniele	29:1a47b7151851	233
daniele	29:1a47b7151851	234	/* cleanup old filter */
daniele	29:1a47b7151851	235	pico_tree_foreach_safe(index, &MCASTFilter, _tmp)
daniele	29:1a47b7151851	236	{
daniele	29:1a47b7151851	237	pico_tree_delete(&MCASTFilter, index->keyValue);
daniele	29:1a47b7151851	238	}
daniele	29:1a47b7151851	239
daniele	29:1a47b7151851	240	/* construct new filter */
daniele	29:1a47b7151851	241	pico_tree_foreach(index, &MCASTSockets)
daniele	29:1a47b7151851	242	{
daniele	29:1a47b7151851	243	mcast_sock = index->keyValue;
daniele	29:1a47b7151851	244	listen = pico_tree_findKey(mcast_sock->MCASTListen, &ltest);
daniele	29:1a47b7151851	245	if (listen) {
daniele	29:1a47b7151851	246	/* aggregate filter */
daniele	29:1a47b7151851	247	switch(filter_mode)
daniele	29:1a47b7151851	248	{
daniele	29:1a47b7151851	249	case PICO_IP_MULTICAST_INCLUDE:
daniele	29:1a47b7151851	250	switch (listen->filter_mode)
daniele	29:1a47b7151851	251	{
daniele	29:1a47b7151851	252	case PICO_IP_MULTICAST_INCLUDE:
daniele	29:1a47b7151851	253	/* filter = summation of INCLUDEs */
daniele	29:1a47b7151851	254	/* mode stays INCLUDE, add all sources to filter */
daniele	29:1a47b7151851	255	pico_tree_foreach(index2, &listen->MCASTSources)
daniele	29:1a47b7151851	256	{
daniele	29:1a47b7151851	257	source = index2->keyValue;
daniele	29:1a47b7151851	258	pico_tree_insert(&MCASTFilter, source);
daniele	29:1a47b7151851	259	}
daniele	29:1a47b7151851	260	break;
daniele	29:1a47b7151851	261
daniele	29:1a47b7151851	262	case PICO_IP_MULTICAST_EXCLUDE:
daniele	29:1a47b7151851	263	/* filter = EXCLUDE - INCLUDE */
daniele	29:1a47b7151851	264	/* delete from the interface INCLUDE filter any source NOT in the socket EXCLUDE filter */
daniele	29:1a47b7151851	265	pico_tree_foreach_safe(index2, &MCASTFilter, _tmp2)
daniele	29:1a47b7151851	266	{
daniele	29:1a47b7151851	267	source = pico_tree_findKey(&listen->MCASTSources, index2->keyValue);
daniele	29:1a47b7151851	268	if (!source)
daniele	29:1a47b7151851	269	pico_tree_delete(&MCASTFilter, index2->keyValue);
daniele	29:1a47b7151851	270	}
daniele	29:1a47b7151851	271	/* any record with filter mode EXCLUDE, causes the interface mode to be EXCLUDE */
daniele	29:1a47b7151851	272	filter_mode = PICO_IP_MULTICAST_EXCLUDE;
daniele	29:1a47b7151851	273	/* add to the interface EXCLUDE filter any socket source NOT in the former interface INCLUDE filter */
daniele	29:1a47b7151851	274	pico_tree_foreach(index2, &listen->MCASTSources)
daniele	29:1a47b7151851	275	{
daniele	29:1a47b7151851	276	source = pico_tree_insert(&MCASTFilter, index2->keyValue);
daniele	29:1a47b7151851	277	if (source)
daniele	29:1a47b7151851	278	pico_tree_delete(&MCASTFilter, source);
daniele	29:1a47b7151851	279	}
daniele	29:1a47b7151851	280	break;
daniele	29:1a47b7151851	281
daniele	29:1a47b7151851	282	default:
daniele	29:1a47b7151851	283	return -1;
daniele	29:1a47b7151851	284	}
daniele	29:1a47b7151851	285	break;
daniele	29:1a47b7151851	286
daniele	29:1a47b7151851	287	case PICO_IP_MULTICAST_EXCLUDE:
daniele	29:1a47b7151851	288	switch (listen->filter_mode)
daniele	29:1a47b7151851	289	{
daniele	29:1a47b7151851	290	case PICO_IP_MULTICAST_INCLUDE:
daniele	29:1a47b7151851	291	/* filter = EXCLUDE - INCLUDE */
daniele	29:1a47b7151851	292	/* any record with filter mode EXCLUDE, causes the interface mode to be EXCLUDE */
daniele	29:1a47b7151851	293	/* remove from the interface EXCLUDE filter any source in the socket INCLUDE filter */
daniele	29:1a47b7151851	294	pico_tree_foreach(index2, &listen->MCASTSources)
daniele	29:1a47b7151851	295	{
daniele	29:1a47b7151851	296	source = pico_tree_findKey(&MCASTFilter, index2->keyValue);
daniele	29:1a47b7151851	297	if (source)
daniele	29:1a47b7151851	298	pico_tree_delete(&MCASTFilter, source);
daniele	29:1a47b7151851	299	}
daniele	29:1a47b7151851	300	break;
daniele	29:1a47b7151851	301
daniele	29:1a47b7151851	302	case PICO_IP_MULTICAST_EXCLUDE:
daniele	29:1a47b7151851	303	/* filter = intersection of EXCLUDEs */
daniele	29:1a47b7151851	304	/* any record with filter mode EXCLUDE, causes the interface mode to be EXCLUDE */
daniele	29:1a47b7151851	305	/* remove from the interface EXCLUDE filter any source not in the socket EXCLUDE filter */
daniele	29:1a47b7151851	306	pico_tree_foreach_safe(index2, &MCASTFilter, _tmp2)
daniele	29:1a47b7151851	307	{
daniele	29:1a47b7151851	308	source = pico_tree_findKey(&listen->MCASTSources, index2->keyValue);
daniele	29:1a47b7151851	309	if (!source)
daniele	29:1a47b7151851	310	pico_tree_delete(&MCASTFilter, index2->keyValue);
daniele	29:1a47b7151851	311	}
daniele	29:1a47b7151851	312	break;
daniele	29:1a47b7151851	313
daniele	29:1a47b7151851	314	default:
daniele	29:1a47b7151851	315	return -1;
daniele	29:1a47b7151851	316	}
daniele	29:1a47b7151851	317	break;
daniele	29:1a47b7151851	318
daniele	29:1a47b7151851	319	default:
daniele	29:1a47b7151851	320	return -1;
daniele	29:1a47b7151851	321	}
daniele	29:1a47b7151851	322	}
daniele	29:1a47b7151851	323	}
daniele	29:1a47b7151851	324	return filter_mode;
daniele	29:1a47b7151851	325	}
daniele	29:1a47b7151851	326
daniele	29:1a47b7151851	327	static int pico_socket_mcast_filter(struct pico_socket s, struct pico_ip4 mcast_group, struct pico_ip4 *src)
daniele	29:1a47b7151851	328	{
daniele	29:1a47b7151851	329	struct pico_ipv4_link *mcast_link = NULL;
daniele	29:1a47b7151851	330	struct pico_mcast_listen *listen = NULL, ltest = {0};
daniele	29:1a47b7151851	331	struct pico_tree_node *index = NULL;
daniele	29:1a47b7151851	332
daniele	29:1a47b7151851	333	/* no multicast enabled on socket */
daniele	29:1a47b7151851	334	if (!s->MCASTListen)
daniele	29:1a47b7151851	335	return 0;
daniele	29:1a47b7151851	336
daniele	29:1a47b7151851	337	mcast_link = pico_ipv4_link_get(&s->local_addr.ip4);
daniele	29:1a47b7151851	338	if (!mcast_link)
daniele	29:1a47b7151851	339	return -1;
daniele	29:1a47b7151851	340
daniele	29:1a47b7151851	341	ltest.mcast_link.addr = mcast_link->address.addr;
daniele	29:1a47b7151851	342	ltest.mcast_group = *mcast_group;
daniele	29:1a47b7151851	343	listen = pico_tree_findKey(s->MCASTListen, &ltest);
daniele	29:1a47b7151851	344	if (!listen)
daniele	29:1a47b7151851	345	return -1;
daniele	29:1a47b7151851	346
daniele	29:1a47b7151851	347	/* perform source filtering */
daniele	29:1a47b7151851	348	switch (listen->filter_mode)
daniele	29:1a47b7151851	349	{
daniele	29:1a47b7151851	350	case PICO_IP_MULTICAST_INCLUDE:
daniele	29:1a47b7151851	351	pico_tree_foreach(index, &listen->MCASTSources)
daniele	29:1a47b7151851	352	{
daniele	29:1a47b7151851	353	if (src->addr == ((struct pico_ip4 *)index->keyValue)->addr) {
daniele	29:1a47b7151851	354	so_mcast_dbg("MCAST: IP %08X in included socket source list\n", src->addr);
daniele	29:1a47b7151851	355	return 0;
daniele	29:1a47b7151851	356	}
daniele	29:1a47b7151851	357	}
daniele	29:1a47b7151851	358	so_mcast_dbg("MCAST: IP %08X NOT in included socket source list\n", src->addr);
daniele	29:1a47b7151851	359	return -1;
daniele	29:1a47b7151851	360	break;
daniele	29:1a47b7151851	361
daniele	29:1a47b7151851	362	case PICO_IP_MULTICAST_EXCLUDE:
daniele	29:1a47b7151851	363	pico_tree_foreach(index, &listen->MCASTSources)
daniele	29:1a47b7151851	364	{
daniele	29:1a47b7151851	365	if (src->addr == ((struct pico_ip4 *)index->keyValue)->addr) {
daniele	29:1a47b7151851	366	so_mcast_dbg("MCAST: IP %08X in excluded socket source list\n", src->addr);
daniele	29:1a47b7151851	367	return -1;
daniele	29:1a47b7151851	368	}
daniele	29:1a47b7151851	369	}
daniele	29:1a47b7151851	370	so_mcast_dbg("MCAST: IP %08X NOT in excluded socket source list\n", src->addr);
daniele	29:1a47b7151851	371	return 0;
daniele	29:1a47b7151851	372	break;
daniele	29:1a47b7151851	373
daniele	29:1a47b7151851	374	default:
daniele	29:1a47b7151851	375	return -1;
daniele	29:1a47b7151851	376	break;
daniele	29:1a47b7151851	377	}
daniele	29:1a47b7151851	378	return -1;
daniele	29:1a47b7151851	379	}
daniele	29:1a47b7151851	380
daniele	29:1a47b7151851	381	static inline struct pico_ipv4_link pico_socket_setoption_mcastargs_validation(struct pico_ip_mreq mreq, struct pico_ip_mreq_source *mreq_source)
daniele	29:1a47b7151851	382	{
daniele	29:1a47b7151851	383	struct pico_ipv4_link *mcast_link = NULL;
daniele	29:1a47b7151851	384
daniele	29:1a47b7151851	385	if (!mreq && !mreq_source)
daniele	29:1a47b7151851	386	return NULL;
daniele	29:1a47b7151851	387
daniele	29:1a47b7151851	388	if (mreq) {
daniele	29:1a47b7151851	389	if (!mreq->mcast_group_addr.addr)
daniele	29:1a47b7151851	390	return NULL;
daniele	29:1a47b7151851	391	if (pico_ipv4_is_unicast(mreq->mcast_group_addr.addr))
daniele	29:1a47b7151851	392	return NULL;
daniele	29:1a47b7151851	393
daniele	29:1a47b7151851	394	if (!mreq->mcast_link_addr.addr) {
daniele	29:1a47b7151851	395	mcast_link = pico_ipv4_get_default_mcastlink();
daniele	29:1a47b7151851	396	if (!mcast_link)
daniele	29:1a47b7151851	397	return NULL;
daniele	29:1a47b7151851	398	} else {
daniele	29:1a47b7151851	399	mcast_link = pico_ipv4_link_get(&mreq->mcast_link_addr);
daniele	29:1a47b7151851	400	if (!mcast_link)
daniele	29:1a47b7151851	401	return NULL;
daniele	29:1a47b7151851	402	}
daniele	29:1a47b7151851	403	}
daniele	29:1a47b7151851	404	if (mreq_source) {
daniele	29:1a47b7151851	405	if (!mreq_source->mcast_group_addr.addr)
daniele	29:1a47b7151851	406	return NULL;
daniele	29:1a47b7151851	407	if (pico_ipv4_is_unicast(mreq_source->mcast_group_addr.addr))
daniele	29:1a47b7151851	408	return NULL;
daniele	29:1a47b7151851	409
daniele	29:1a47b7151851	410	if (!mreq_source->mcast_source_addr.addr)
daniele	29:1a47b7151851	411	return NULL;
daniele	29:1a47b7151851	412	if (!pico_ipv4_is_unicast(mreq_source->mcast_source_addr.addr))
daniele	29:1a47b7151851	413	return NULL;
daniele	29:1a47b7151851	414
daniele	29:1a47b7151851	415	if (!mreq_source->mcast_link_addr.addr) {
daniele	29:1a47b7151851	416	mcast_link = pico_ipv4_get_default_mcastlink();
daniele	29:1a47b7151851	417	if (!mcast_link)
daniele	29:1a47b7151851	418	return NULL;
daniele	29:1a47b7151851	419	} else {
daniele	29:1a47b7151851	420	mcast_link = pico_ipv4_link_get(&mreq_source->mcast_link_addr);
daniele	29:1a47b7151851	421	if (!mcast_link)
daniele	29:1a47b7151851	422	return NULL;
daniele	29:1a47b7151851	423	}
daniele	29:1a47b7151851	424	}
daniele	29:1a47b7151851	425	return mcast_link;
daniele	29:1a47b7151851	426	}
daniele	29:1a47b7151851	427	#else
daniele	29:1a47b7151851	428	static int pico_socket_mcast_filter(struct pico_socket s, struct pico_ip4 mcast_group, struct pico_ip4 *src) {
daniele	29:1a47b7151851	429	return 0;
daniele	29:1a47b7151851	430	}
daniele	29:1a47b7151851	431	#endif /* PICO_SUPPORT_MCAST */
daniele	29:1a47b7151851	432
daniele	29:1a47b7151851	433	static struct pico_sockport *pico_get_sockport(uint16_t proto, uint16_t port)
daniele	29:1a47b7151851	434	{
daniele	29:1a47b7151851	435	struct pico_sockport test = INIT_SOCKPORT;
daniele	29:1a47b7151851	436	test.number = port;
daniele	29:1a47b7151851	437
daniele	29:1a47b7151851	438	if (proto == PICO_PROTO_UDP)
daniele	29:1a47b7151851	439	return pico_tree_findKey(&UDPTable,&test);
daniele	29:1a47b7151851	440
daniele	29:1a47b7151851	441	else if (proto == PICO_PROTO_TCP)
daniele	29:1a47b7151851	442	return pico_tree_findKey(&TCPTable,&test);
daniele	29:1a47b7151851	443
daniele	29:1a47b7151851	444	else return NULL;
daniele	29:1a47b7151851	445	}
daniele	29:1a47b7151851	446
daniele	29:1a47b7151851	447	int pico_is_port_free(uint16_t proto, uint16_t port, void addr, void net)
daniele	29:1a47b7151851	448	{
daniele	29:1a47b7151851	449	struct pico_sockport *sp;
daniele	29:1a47b7151851	450	struct pico_ip4 ip;
daniele	29:1a47b7151851	451	sp = pico_get_sockport(proto, port);
daniele	29:1a47b7151851	452
daniele	29:1a47b7151851	453	if (!net)
daniele	29:1a47b7151851	454	net = &pico_proto_ipv4;
daniele	29:1a47b7151851	455
daniele	29:1a47b7151851	456	/ IPv6 (wip) */
daniele	29:1a47b7151851	457	if (net != &pico_proto_ipv4) {
daniele	29:1a47b7151851	458	dbg("IPV6!!!!!\n");
daniele	29:1a47b7151851	459	return (!sp);
daniele	29:1a47b7151851	460	}
daniele	29:1a47b7151851	461
daniele	29:1a47b7151851	462	/* IPv4 */
daniele	29:1a47b7151851	463	#ifdef PICO_SUPPORT_NAT
daniele	29:1a47b7151851	464	if (pico_ipv4_nat_find(port,NULL, 0,proto) == 0) {
daniele	29:1a47b7151851	465	dbg("In use by nat....\n");
daniele	29:1a47b7151851	466	return 0;
daniele	29:1a47b7151851	467	}
daniele	29:1a47b7151851	468	#endif
daniele	29:1a47b7151851	469	if (addr)
daniele	29:1a47b7151851	470	ip.addr = ((struct pico_ip4 *)addr)->addr;
daniele	29:1a47b7151851	471	else
daniele	29:1a47b7151851	472	ip.addr = PICO_IPV4_INADDR_ANY;
daniele	29:1a47b7151851	473
daniele	29:1a47b7151851	474	if (ip.addr == PICO_IPV4_INADDR_ANY) {
daniele	29:1a47b7151851	475	if (!sp) return 1;
daniele	29:1a47b7151851	476	else {
daniele	29:1a47b7151851	477	dbg("In use, and asked for ANY\n");
daniele	29:1a47b7151851	478	return 0;
daniele	29:1a47b7151851	479	}
daniele	29:1a47b7151851	480	}
daniele	29:1a47b7151851	481	if (sp) {
daniele	29:1a47b7151851	482	struct pico_ip4 *s_local;
daniele	29:1a47b7151851	483	struct pico_tree_node *idx;
daniele	29:1a47b7151851	484	struct pico_socket *s;
daniele	29:1a47b7151851	485	pico_tree_foreach(idx, &sp->socks) {
daniele	29:1a47b7151851	486	s = idx->keyValue;
daniele	29:1a47b7151851	487	if (s->net == &pico_proto_ipv4) {
daniele	29:1a47b7151851	488	s_local = (struct pico_ip4*) &s->local_addr;
daniele	29:1a47b7151851	489	if ((s_local->addr == PICO_IPV4_INADDR_ANY) \|\| (s_local->addr == ip.addr))
daniele	29:1a47b7151851	490	return 0;
daniele	29:1a47b7151851	491	}
daniele	29:1a47b7151851	492	}
daniele	29:1a47b7151851	493	}
daniele	29:1a47b7151851	494	return 1;
daniele	29:1a47b7151851	495	}
daniele	29:1a47b7151851	496
daniele	29:1a47b7151851	497	static int pico_check_socket(struct pico_socket *s)
daniele	29:1a47b7151851	498	{
daniele	29:1a47b7151851	499	struct pico_sockport *test;
daniele	29:1a47b7151851	500	struct pico_socket *found;
daniele	29:1a47b7151851	501	struct pico_tree_node * index;
daniele	29:1a47b7151851	502
daniele	29:1a47b7151851	503	test = pico_get_sockport(PROTO(s), s->local_port);
daniele	29:1a47b7151851	504
daniele	29:1a47b7151851	505	if (!test) {
daniele	29:1a47b7151851	506	return -1;
daniele	29:1a47b7151851	507	}
daniele	29:1a47b7151851	508
daniele	29:1a47b7151851	509	pico_tree_foreach(index,&test->socks){
daniele	29:1a47b7151851	510	found = index->keyValue;
daniele	29:1a47b7151851	511	if (s == found) {
daniele	29:1a47b7151851	512	return 0;
daniele	29:1a47b7151851	513	}
daniele	29:1a47b7151851	514	}
daniele	29:1a47b7151851	515
daniele	29:1a47b7151851	516	return -1;
daniele	29:1a47b7151851	517	}
daniele	29:1a47b7151851	518
tass	49:4b404dd2c97a	519	struct pico_socket* pico_sockets_find(uint16_t local,uint16_t remote)
tass	49:4b404dd2c97a	520	{
tass	49:4b404dd2c97a	521	struct pico_socket * sock = NULL;
tass	49:4b404dd2c97a	522	struct pico_tree_node *index = NULL;
tass	49:4b404dd2c97a	523	struct pico_sockport *sp = NULL;
tass	49:4b404dd2c97a	524
tass	49:4b404dd2c97a	525	sp = pico_get_sockport(PICO_PROTO_TCP,local);
tass	49:4b404dd2c97a	526	if(sp)
tass	49:4b404dd2c97a	527	{
tass	49:4b404dd2c97a	528	pico_tree_foreach(index,&sp->socks)
tass	49:4b404dd2c97a	529	{
tass	49:4b404dd2c97a	530	if( ((struct pico_socket *)index->keyValue)->remote_port == remote)
tass	49:4b404dd2c97a	531	{
tass	49:4b404dd2c97a	532	sock = (struct pico_socket *)index->keyValue;
tass	49:4b404dd2c97a	533	break;
tass	49:4b404dd2c97a	534	}
tass	49:4b404dd2c97a	535	}
tass	49:4b404dd2c97a	536	}
tass	49:4b404dd2c97a	537
tass	49:4b404dd2c97a	538	return sock;
tass	49:4b404dd2c97a	539	}
tass	49:4b404dd2c97a	540
daniele	29:1a47b7151851	541
daniele	29:1a47b7151851	542	int pico_socket_add(struct pico_socket *s)
daniele	29:1a47b7151851	543	{
daniele	29:1a47b7151851	544	struct pico_sockport *sp = pico_get_sockport(PROTO(s), s->local_port);
daniele	29:1a47b7151851	545	LOCK(Mutex);
daniele	29:1a47b7151851	546	if (!sp) {
daniele	29:1a47b7151851	547	//dbg("Creating sockport..%04x\n", s->local_port); /* In comment due to spam during test */
daniele	29:1a47b7151851	548	sp = pico_zalloc(sizeof(struct pico_sockport));
daniele	29:1a47b7151851	549
daniele	29:1a47b7151851	550	if (!sp) {
daniele	29:1a47b7151851	551	pico_err = PICO_ERR_ENOMEM;
daniele	29:1a47b7151851	552	UNLOCK(Mutex);
daniele	29:1a47b7151851	553	return -1;
daniele	29:1a47b7151851	554	}
daniele	29:1a47b7151851	555	sp->proto = PROTO(s);
daniele	29:1a47b7151851	556	sp->number = s->local_port;
daniele	29:1a47b7151851	557	sp->socks.root = &LEAF;
daniele	29:1a47b7151851	558	sp->socks.compare = socket_cmp;
daniele	29:1a47b7151851	559
daniele	29:1a47b7151851	560	if (PROTO(s) == PICO_PROTO_UDP)
daniele	29:1a47b7151851	561	{
daniele	29:1a47b7151851	562	pico_tree_insert(&UDPTable,sp);
daniele	29:1a47b7151851	563	}
daniele	29:1a47b7151851	564	else if (PROTO(s) == PICO_PROTO_TCP)
daniele	29:1a47b7151851	565	{
daniele	29:1a47b7151851	566	pico_tree_insert(&TCPTable,sp);
daniele	29:1a47b7151851	567	}
daniele	29:1a47b7151851	568	}
daniele	29:1a47b7151851	569
daniele	29:1a47b7151851	570	pico_tree_insert(&sp->socks,s);
daniele	29:1a47b7151851	571	s->state \|= PICO_SOCKET_STATE_BOUND;
daniele	29:1a47b7151851	572	UNLOCK(Mutex);
daniele	29:1a47b7151851	573	#if DEBUG_SOCKET_TREE
daniele	29:1a47b7151851	574	{
daniele	29:1a47b7151851	575	struct pico_tree_node * index;
daniele	29:1a47b7151851	576	//RB_FOREACH(s, socket_tree, &sp->socks) {
daniele	29:1a47b7151851	577	pico_tree_foreach(index,&sp->socks){
daniele	29:1a47b7151851	578	s = index->keyValue;
daniele	29:1a47b7151851	579	dbg(">>>> List Socket lc=%hu rm=%hu\n", short_be(s->local_port), short_be(s->remote_port));
daniele	29:1a47b7151851	580	}
daniele	29:1a47b7151851	581
daniele	29:1a47b7151851	582	}
daniele	29:1a47b7151851	583	#endif
daniele	29:1a47b7151851	584	return 0;
daniele	29:1a47b7151851	585	}
daniele	29:1a47b7151851	586
daniele	29:1a47b7151851	587	static void socket_garbage_collect(unsigned long now, void *arg)
daniele	29:1a47b7151851	588	{
daniele	29:1a47b7151851	589	struct pico_socket s = (struct pico_socket ) arg;
daniele	29:1a47b7151851	590	pico_free(s);
daniele	29:1a47b7151851	591	}
daniele	29:1a47b7151851	592
daniele	29:1a47b7151851	593	int pico_socket_del(struct pico_socket *s)
daniele	29:1a47b7151851	594	{
daniele	29:1a47b7151851	595	struct pico_sockport *sp = pico_get_sockport(PROTO(s), s->local_port);
daniele	29:1a47b7151851	596
daniele	29:1a47b7151851	597	if (!sp) {
daniele	29:1a47b7151851	598	pico_err = PICO_ERR_ENXIO;
daniele	29:1a47b7151851	599	return -1;
daniele	29:1a47b7151851	600	}
daniele	29:1a47b7151851	601
daniele	29:1a47b7151851	602	LOCK(Mutex);
daniele	29:1a47b7151851	603	pico_tree_delete(&sp->socks,s);
daniele	29:1a47b7151851	604	s->net = NULL;
daniele	29:1a47b7151851	605	if(pico_tree_empty(&sp->socks)){
daniele	29:1a47b7151851	606	if (PROTO(s) == PICO_PROTO_UDP)
daniele	29:1a47b7151851	607	{
daniele	29:1a47b7151851	608	pico_tree_delete(&UDPTable,sp);
daniele	29:1a47b7151851	609	}
daniele	29:1a47b7151851	610	else if (PROTO(s) == PICO_PROTO_TCP)
daniele	29:1a47b7151851	611	{
daniele	29:1a47b7151851	612	pico_tree_delete(&TCPTable,sp);
daniele	29:1a47b7151851	613	}
daniele	29:1a47b7151851	614
daniele	29:1a47b7151851	615	if(sp_tcp == sp) sp_tcp = NULL;
daniele	29:1a47b7151851	616
daniele	29:1a47b7151851	617	if(sp_udp == sp) sp_udp = NULL;
daniele	29:1a47b7151851	618
daniele	29:1a47b7151851	619	pico_free(sp);
daniele	29:1a47b7151851	620
daniele	29:1a47b7151851	621	}
daniele	29:1a47b7151851	622
daniele	29:1a47b7151851	623	#ifdef PICO_SUPPORT_MCAST
daniele	29:1a47b7151851	624	do {
tass	48:40fc4462265c	625	int filter_mode;
daniele	29:1a47b7151851	626	struct pico_tree_node index = NULL, _tmp = NULL, index2 = NULL, _tmp2 = NULL;
daniele	29:1a47b7151851	627	struct pico_mcast_listen *listen = NULL;
daniele	29:1a47b7151851	628	struct pico_ip4 *source = NULL;
daniele	29:1a47b7151851	629	if (s->MCASTListen) {
daniele	29:1a47b7151851	630	pico_tree_delete(&MCASTSockets, s);
daniele	29:1a47b7151851	631	pico_tree_foreach_safe(index, s->MCASTListen, _tmp)
daniele	29:1a47b7151851	632	{
daniele	29:1a47b7151851	633	listen = index->keyValue;
daniele	29:1a47b7151851	634	pico_tree_foreach_safe(index2, &listen->MCASTSources, _tmp2)
daniele	29:1a47b7151851	635	{
daniele	29:1a47b7151851	636	source = index->keyValue;
daniele	29:1a47b7151851	637	pico_tree_delete(&listen->MCASTSources, source);
daniele	29:1a47b7151851	638	pico_free(source);
daniele	29:1a47b7151851	639	}
daniele	29:1a47b7151851	640	filter_mode = pico_socket_aggregate_mcastfilters(&listen->mcast_link, &listen->mcast_group);
tass	48:40fc4462265c	641	if (filter_mode >= 0)
tass	48:40fc4462265c	642	pico_ipv4_mcast_leave(&listen->mcast_link, &listen->mcast_group, 1, filter_mode, &MCASTFilter);
daniele	29:1a47b7151851	643	pico_tree_delete(s->MCASTListen, listen);
daniele	29:1a47b7151851	644	pico_free(listen);
daniele	29:1a47b7151851	645	}
daniele	29:1a47b7151851	646	pico_free(s->MCASTListen);
daniele	29:1a47b7151851	647	}
daniele	29:1a47b7151851	648	} while (0);
daniele	29:1a47b7151851	649	#endif
daniele	29:1a47b7151851	650
daniele	29:1a47b7151851	651	s->state = PICO_SOCKET_STATE_CLOSED;
daniele	29:1a47b7151851	652	pico_timer_add(3000, socket_garbage_collect, s);
daniele	29:1a47b7151851	653	UNLOCK(Mutex);
daniele	29:1a47b7151851	654	return 0;
daniele	29:1a47b7151851	655	}
daniele	29:1a47b7151851	656
daniele	29:1a47b7151851	657	static int pico_socket_alter_state(struct pico_socket *s, uint16_t more_states, uint16_t less_states, uint16_t tcp_state)
daniele	29:1a47b7151851	658	{
daniele	29:1a47b7151851	659	struct pico_sockport *sp;
daniele	29:1a47b7151851	660	if (more_states & PICO_SOCKET_STATE_BOUND)
daniele	29:1a47b7151851	661	return pico_socket_add(s);
daniele	29:1a47b7151851	662
daniele	29:1a47b7151851	663	if (less_states & PICO_SOCKET_STATE_BOUND)
daniele	29:1a47b7151851	664	return pico_socket_del(s);
daniele	29:1a47b7151851	665
daniele	29:1a47b7151851	666	sp = pico_get_sockport(PROTO(s), s->local_port);
daniele	29:1a47b7151851	667	if (!sp) {
daniele	29:1a47b7151851	668	pico_err = PICO_ERR_ENXIO;
daniele	29:1a47b7151851	669	return -1;
daniele	29:1a47b7151851	670	}
daniele	29:1a47b7151851	671
daniele	29:1a47b7151851	672	s->state \|= more_states;
daniele	29:1a47b7151851	673	s->state &= (~less_states);
daniele	29:1a47b7151851	674	if (tcp_state) {
daniele	29:1a47b7151851	675	s->state &= 0x00FF;
daniele	29:1a47b7151851	676	s->state \|= tcp_state;
daniele	29:1a47b7151851	677	}
daniele	29:1a47b7151851	678
daniele	29:1a47b7151851	679	return 0;
daniele	29:1a47b7151851	680	}
daniele	29:1a47b7151851	681
daniele	29:1a47b7151851	682	static int pico_socket_deliver(struct pico_protocol p, struct pico_frame f, uint16_t localport)
daniele	29:1a47b7151851	683	{
daniele	29:1a47b7151851	684	struct pico_frame *cpy = NULL;
daniele	29:1a47b7151851	685	struct pico_sockport *sp = NULL;
daniele	29:1a47b7151851	686	struct pico_socket s = NULL, found = NULL;
daniele	29:1a47b7151851	687	struct pico_tree_node *index = NULL;
daniele	29:1a47b7151851	688	struct pico_tree_node *_tmp;
daniele	29:1a47b7151851	689	struct pico_trans tr = (struct pico_trans ) f->transport_hdr;
daniele	29:1a47b7151851	690	#ifdef PICO_SUPPORT_IPV4
daniele	29:1a47b7151851	691	struct pico_ipv4_hdr *ip4hdr;
daniele	29:1a47b7151851	692	#endif
daniele	29:1a47b7151851	693	#ifdef PICO_SUPPORT_IPV6
daniele	29:1a47b7151851	694	struct pico_ipv6_hdr *ip6hdr;
daniele	29:1a47b7151851	695	#endif
daniele	29:1a47b7151851	696
daniele	29:1a47b7151851	697	if (!tr)
daniele	29:1a47b7151851	698	return -1;
daniele	29:1a47b7151851	699
daniele	29:1a47b7151851	700	sp = pico_get_sockport(p->proto_number, localport);
daniele	29:1a47b7151851	701
daniele	29:1a47b7151851	702	if (!sp) {
daniele	29:1a47b7151851	703	dbg("No such port %d\n",short_be(localport));
daniele	29:1a47b7151851	704	return -1;
daniele	29:1a47b7151851	705	}
daniele	29:1a47b7151851	706
daniele	29:1a47b7151851	707	#ifdef PICO_SUPPORT_TCP
daniele	29:1a47b7151851	708	if (p->proto_number == PICO_PROTO_TCP) {
daniele	29:1a47b7151851	709	pico_tree_foreach_safe(index,&sp->socks, _tmp){
daniele	29:1a47b7151851	710	s = index->keyValue;
daniele	29:1a47b7151851	711	/* 4-tuple identification of socket (port-IP) */
daniele	29:1a47b7151851	712	#ifdef PICO_SUPPORT_IPV4
daniele	29:1a47b7151851	713	if (IS_IPV4(f)) {
daniele	29:1a47b7151851	714	struct pico_ip4 s_local, s_remote, p_src, p_dst;
daniele	29:1a47b7151851	715	ip4hdr = (struct pico_ipv4_hdr*)(f->net_hdr);
daniele	29:1a47b7151851	716	s_local.addr = s->local_addr.ip4.addr;
daniele	29:1a47b7151851	717	s_remote.addr = s->remote_addr.ip4.addr;
daniele	29:1a47b7151851	718	p_src.addr = ip4hdr->src.addr;
daniele	29:1a47b7151851	719	p_dst.addr = ip4hdr->dst.addr;
daniele	29:1a47b7151851	720	if ( (s->remote_port == tr->sport) && /* remote port check */
daniele	29:1a47b7151851	721	(s_remote.addr == p_src.addr) && /* remote addr check */
daniele	29:1a47b7151851	722	((s_local.addr == PICO_IPV4_INADDR_ANY) \|\| (s_local.addr == p_dst.addr))) { /* Either local socket is ANY, or matches dst */
daniele	29:1a47b7151851	723	found = s;
daniele	29:1a47b7151851	724	break;
daniele	29:1a47b7151851	725	} else if ( (s->remote_port == 0) && /* not connected... listening */
daniele	29:1a47b7151851	726	((s_local.addr == PICO_IPV4_INADDR_ANY) \|\| (s_local.addr == p_dst.addr))) { /* Either local socket is ANY, or matches dst */
daniele	29:1a47b7151851	727	/* listen socket */
daniele	29:1a47b7151851	728	found = s;
daniele	29:1a47b7151851	729	}
daniele	29:1a47b7151851	730	}
daniele	29:1a47b7151851	731	#endif
daniele	29:1a47b7151851	732	#ifdef PICO_SUPPORT_IPV6 /* XXX TODO make compare for ipv6 addresses */
daniele	29:1a47b7151851	733	if (IS_IPV6(f)) {
daniele	29:1a47b7151851	734	ip6hdr = (struct pico_ipv6_hdr*)(f->net_hdr);
daniele	29:1a47b7151851	735	if ( (s->remote_port == localport) ) { // && (((struct pico_ip6) s->remote_addr.ip6).addr == ((struct pico_ip6)(ip6hdr->src)).addr) ) {
daniele	29:1a47b7151851	736	found = s;
daniele	29:1a47b7151851	737	break;
daniele	29:1a47b7151851	738	} else if (s->remote_port == 0) {
daniele	29:1a47b7151851	739	/* listen socket */
daniele	29:1a47b7151851	740	found = s;
daniele	29:1a47b7151851	741	}
daniele	29:1a47b7151851	742	}
daniele	29:1a47b7151851	743	#endif
daniele	29:1a47b7151851	744	} /* FOREACH */
daniele	29:1a47b7151851	745	if (found != NULL) {
daniele	29:1a47b7151851	746	pico_tcp_input(found,f);
daniele	29:1a47b7151851	747	if ((found->ev_pending) && found->wakeup) {
daniele	29:1a47b7151851	748	found->wakeup(found->ev_pending, found);
daniele	29:1a47b7151851	749	}
daniele	29:1a47b7151851	750	return 0;
daniele	29:1a47b7151851	751	} else {
daniele	29:1a47b7151851	752	dbg("SOCKET> mmm something wrong (prob sockport)\n");
daniele	29:1a47b7151851	753	return -1;
daniele	29:1a47b7151851	754	}
daniele	29:1a47b7151851	755	} /* TCP CASE */
daniele	29:1a47b7151851	756	#endif
daniele	29:1a47b7151851	757
daniele	29:1a47b7151851	758	#ifdef PICO_SUPPORT_UDP
daniele	29:1a47b7151851	759	if (p->proto_number == PICO_PROTO_UDP) {
daniele	29:1a47b7151851	760	pico_tree_foreach_safe(index,&sp->socks, _tmp){
daniele	29:1a47b7151851	761	s = index->keyValue;
daniele	29:1a47b7151851	762	if (IS_IPV4(f)) { /* IPV4 */
daniele	29:1a47b7151851	763	struct pico_ip4 s_local, p_dst;
daniele	29:1a47b7151851	764	ip4hdr = (struct pico_ipv4_hdr*)(f->net_hdr);
daniele	29:1a47b7151851	765	s_local.addr = s->local_addr.ip4.addr;
daniele	29:1a47b7151851	766	p_dst.addr = ip4hdr->dst.addr;
daniele	29:1a47b7151851	767	if ((pico_ipv4_is_broadcast(p_dst.addr)) \|\| pico_ipv4_is_multicast(p_dst.addr)) {
daniele	29:1a47b7151851	768	struct pico_device *dev = pico_ipv4_link_find(&s->local_addr.ip4);
daniele	29:1a47b7151851	769	if (pico_ipv4_is_multicast(p_dst.addr) && (pico_socket_mcast_filter(s, &ip4hdr->dst, &ip4hdr->src) < 0))
daniele	29:1a47b7151851	770	return -1;
daniele	29:1a47b7151851	771	if ((s_local.addr == PICO_IPV4_INADDR_ANY) \|\| /* If our local ip is ANY, or.. */
daniele	29:1a47b7151851	772	(dev == f->dev) ) { /* the source of the bcast packet is a neighbor... */
daniele	29:1a47b7151851	773	cpy = pico_frame_copy(f);
daniele	29:1a47b7151851	774	if (!cpy)
daniele	29:1a47b7151851	775	return -1;
daniele	29:1a47b7151851	776	if (pico_enqueue(&s->q_in, cpy) > 0) {
daniele	29:1a47b7151851	777	if (s->wakeup)
daniele	29:1a47b7151851	778	s->wakeup(PICO_SOCK_EV_RD, s);
daniele	29:1a47b7151851	779	}
daniele	29:1a47b7151851	780	}
daniele	29:1a47b7151851	781	} else if ((s_local.addr == PICO_IPV4_INADDR_ANY) \|\| (s_local.addr == p_dst.addr))
daniele	29:1a47b7151851	782	{ /* Either local socket is ANY, or matches dst */
daniele	29:1a47b7151851	783	cpy = pico_frame_copy(f);
daniele	29:1a47b7151851	784	if (!cpy)
daniele	29:1a47b7151851	785	return -1;
daniele	29:1a47b7151851	786	if (pico_enqueue(&s->q_in, cpy) > 0) {
daniele	29:1a47b7151851	787	if (s->wakeup)
daniele	29:1a47b7151851	788	s->wakeup(PICO_SOCK_EV_RD, s);
daniele	29:1a47b7151851	789	}
daniele	29:1a47b7151851	790	}
daniele	29:1a47b7151851	791	} else {
daniele	29:1a47b7151851	792	/... IPv6 /
daniele	29:1a47b7151851	793	}
daniele	29:1a47b7151851	794	} /* FOREACH */
daniele	29:1a47b7151851	795	pico_frame_discard(f);
daniele	29:1a47b7151851	796	if (s)
daniele	29:1a47b7151851	797	return 0;
daniele	29:1a47b7151851	798	else
daniele	29:1a47b7151851	799	return -1;
daniele	29:1a47b7151851	800	}
daniele	29:1a47b7151851	801	#endif
daniele	29:1a47b7151851	802	return -1;
daniele	29:1a47b7151851	803	}
daniele	29:1a47b7151851	804
daniele	29:1a47b7151851	805	struct pico_socket pico_socket_open(uint16_t net, uint16_t proto, void (wakeup)(uint16_t ev, struct pico_socket *))
daniele	29:1a47b7151851	806	{
daniele	29:1a47b7151851	807
daniele	29:1a47b7151851	808	struct pico_socket *s = NULL;
daniele	29:1a47b7151851	809
daniele	29:1a47b7151851	810	#ifdef PICO_SUPPORT_UDP
daniele	29:1a47b7151851	811	if (proto == PICO_PROTO_UDP) {
daniele	29:1a47b7151851	812	s = pico_udp_open();
daniele	29:1a47b7151851	813	s->proto = &pico_proto_udp;
daniele	29:1a47b7151851	814	}
daniele	29:1a47b7151851	815	#endif
daniele	29:1a47b7151851	816
daniele	29:1a47b7151851	817	#ifdef PICO_SUPPORT_TCP
daniele	29:1a47b7151851	818	if (proto == PICO_PROTO_TCP) {
daniele	29:1a47b7151851	819	s = pico_tcp_open();
daniele	29:1a47b7151851	820	s->proto = &pico_proto_tcp;
daniele	29:1a47b7151851	821	/check if Nagle enabled /
daniele	29:1a47b7151851	822	if (!IS_NAGLE_ENABLED(s))
daniele	29:1a47b7151851	823	dbg("ERROR Nagle should be enabled here\n\n");
daniele	29:1a47b7151851	824	}
daniele	29:1a47b7151851	825	#endif
daniele	29:1a47b7151851	826
daniele	29:1a47b7151851	827	if (!s) {
daniele	29:1a47b7151851	828	pico_err = PICO_ERR_EPROTONOSUPPORT;
daniele	29:1a47b7151851	829	return NULL;
daniele	29:1a47b7151851	830	}
daniele	29:1a47b7151851	831
daniele	29:1a47b7151851	832	#ifdef PICO_SUPPORT_IPV4
daniele	29:1a47b7151851	833	if (net == PICO_PROTO_IPV4)
daniele	29:1a47b7151851	834	s->net = &pico_proto_ipv4;
daniele	29:1a47b7151851	835	#endif
daniele	29:1a47b7151851	836
daniele	29:1a47b7151851	837	#ifdef PICO_SUPPORT_IPV6
daniele	29:1a47b7151851	838	if (net == PICO_PROTO_IPV6)
daniele	29:1a47b7151851	839	s->net = &pico_proto_ipv6;
daniele	29:1a47b7151851	840	#endif
daniele	29:1a47b7151851	841
daniele	29:1a47b7151851	842	s->q_in.max_size = PICO_DEFAULT_SOCKETQ;
daniele	29:1a47b7151851	843	s->q_out.max_size = PICO_DEFAULT_SOCKETQ;
daniele	29:1a47b7151851	844	s->wakeup = wakeup;
daniele	29:1a47b7151851	845
daniele	29:1a47b7151851	846	if (!s->net) {
daniele	29:1a47b7151851	847	pico_free(s);
daniele	29:1a47b7151851	848	pico_err = PICO_ERR_ENETUNREACH;
daniele	29:1a47b7151851	849	return NULL;
daniele	29:1a47b7151851	850	}
daniele	29:1a47b7151851	851	return s;
daniele	29:1a47b7151851	852	}
daniele	29:1a47b7151851	853
daniele	29:1a47b7151851	854
daniele	29:1a47b7151851	855	struct pico_socket pico_socket_clone(struct pico_socket facsimile)
daniele	29:1a47b7151851	856	{
daniele	29:1a47b7151851	857	struct pico_socket *s = NULL;
daniele	29:1a47b7151851	858
daniele	29:1a47b7151851	859	#ifdef PICO_SUPPORT_UDP
daniele	29:1a47b7151851	860	if (facsimile->proto->proto_number == PICO_PROTO_UDP) {
daniele	29:1a47b7151851	861	s = pico_udp_open();
daniele	29:1a47b7151851	862	s->proto = &pico_proto_udp;
daniele	29:1a47b7151851	863	}
daniele	29:1a47b7151851	864	#endif
daniele	29:1a47b7151851	865
daniele	29:1a47b7151851	866	#ifdef PICO_SUPPORT_TCP
daniele	29:1a47b7151851	867	if (facsimile->proto->proto_number == PICO_PROTO_TCP) {
daniele	29:1a47b7151851	868	s = pico_tcp_open();
daniele	29:1a47b7151851	869	s->proto = &pico_proto_tcp;
daniele	29:1a47b7151851	870	}
daniele	29:1a47b7151851	871	#endif
daniele	29:1a47b7151851	872
daniele	29:1a47b7151851	873	if (!s) {
daniele	29:1a47b7151851	874	pico_err = PICO_ERR_EPROTONOSUPPORT;
daniele	29:1a47b7151851	875	return NULL;
daniele	29:1a47b7151851	876	}
daniele	29:1a47b7151851	877	s->local_port = facsimile->local_port;
daniele	29:1a47b7151851	878	s->remote_port = facsimile->remote_port;
daniele	29:1a47b7151851	879	s->state = facsimile->state;
daniele	29:1a47b7151851	880
daniele	29:1a47b7151851	881	#ifdef PICO_SUPPORT_IPV4
daniele	29:1a47b7151851	882	if (facsimile->net == &pico_proto_ipv4) {
daniele	29:1a47b7151851	883	s->net = &pico_proto_ipv4;
daniele	29:1a47b7151851	884	memcpy(&s->local_addr, &facsimile->local_addr, sizeof(struct pico_ip4));
daniele	29:1a47b7151851	885	memcpy(&s->remote_addr, &facsimile->remote_addr, sizeof(struct pico_ip4));
daniele	29:1a47b7151851	886	}
daniele	29:1a47b7151851	887	#endif
daniele	29:1a47b7151851	888
daniele	29:1a47b7151851	889	#ifdef PICO_SUPPORT_IPV6
daniele	29:1a47b7151851	890	if (net == &pico_proto_ipv6) {
daniele	29:1a47b7151851	891	s->net = &pico_proto_ipv6;
daniele	29:1a47b7151851	892	memcpy(&s->local_addr, &facsimile->local_addr, sizeof(struct pico_ip6));
daniele	29:1a47b7151851	893	memcpy(&s->remote_addr, &facsimile->remote_addr, sizeof(struct pico_ip6));
daniele	29:1a47b7151851	894	}
daniele	29:1a47b7151851	895	#endif
daniele	29:1a47b7151851	896	s->q_in.max_size = PICO_DEFAULT_SOCKETQ;
daniele	29:1a47b7151851	897	s->q_out.max_size = PICO_DEFAULT_SOCKETQ;
daniele	29:1a47b7151851	898	s->wakeup = NULL;
daniele	29:1a47b7151851	899	if (!s->net) {
daniele	29:1a47b7151851	900	pico_free(s);
daniele	29:1a47b7151851	901	pico_err = PICO_ERR_ENETUNREACH;
daniele	29:1a47b7151851	902	return NULL;
daniele	29:1a47b7151851	903	}
daniele	29:1a47b7151851	904	return s;
daniele	29:1a47b7151851	905	}
daniele	29:1a47b7151851	906
daniele	29:1a47b7151851	907	int pico_socket_read(struct pico_socket s, void buf, int len)
daniele	29:1a47b7151851	908	{
daniele	29:1a47b7151851	909	if (!s \|\| buf == NULL) {
daniele	29:1a47b7151851	910	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	911	return -1;
daniele	29:1a47b7151851	912	} else {
daniele	29:1a47b7151851	913	/* check if exists in tree */
daniele	29:1a47b7151851	914	/* See task #178 */
daniele	29:1a47b7151851	915	if (pico_check_socket(s) != 0) {
daniele	29:1a47b7151851	916	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	917	return -1;
daniele	29:1a47b7151851	918	}
daniele	29:1a47b7151851	919	}
daniele	29:1a47b7151851	920
daniele	29:1a47b7151851	921	if ((s->state & PICO_SOCKET_STATE_BOUND) == 0) {
daniele	29:1a47b7151851	922	pico_err = PICO_ERR_EIO;
daniele	29:1a47b7151851	923	return -1;
daniele	29:1a47b7151851	924	}
daniele	29:1a47b7151851	925	#ifdef PICO_SUPPORT_UDP
daniele	29:1a47b7151851	926	if (PROTO(s) == PICO_PROTO_UDP)
daniele	29:1a47b7151851	927	return pico_udp_recv(s, buf, len, NULL, NULL);
daniele	29:1a47b7151851	928	#endif
daniele	29:1a47b7151851	929
daniele	29:1a47b7151851	930	#ifdef PICO_SUPPORT_TCP
daniele	29:1a47b7151851	931	if (PROTO(s) == PICO_PROTO_TCP){
daniele	29:1a47b7151851	932	/* check if in shutdown state and if no more data in tcpq_in */
daniele	29:1a47b7151851	933	if ((s->state & PICO_SOCKET_STATE_SHUT_REMOTE) && pico_tcp_queue_in_is_empty(s) ) {
daniele	29:1a47b7151851	934	pico_err = PICO_ERR_ESHUTDOWN;
daniele	29:1a47b7151851	935	return -1;
daniele	29:1a47b7151851	936	} else {
daniele	29:1a47b7151851	937	return pico_tcp_read(s, buf, len);
daniele	29:1a47b7151851	938	}
daniele	29:1a47b7151851	939	}
daniele	29:1a47b7151851	940	#endif
daniele	29:1a47b7151851	941	return 0;
daniele	29:1a47b7151851	942	}
daniele	29:1a47b7151851	943
daniele	29:1a47b7151851	944	int pico_socket_write(struct pico_socket s, void buf, int len)
daniele	29:1a47b7151851	945	{
daniele	29:1a47b7151851	946	if (!s \|\| buf == NULL) {
daniele	29:1a47b7151851	947	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	948	return -1;
daniele	29:1a47b7151851	949	} else {
daniele	29:1a47b7151851	950	/* check if exists in tree */
daniele	29:1a47b7151851	951	/* See task #178 */
daniele	29:1a47b7151851	952	if (pico_check_socket(s) != 0) {
daniele	29:1a47b7151851	953	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	954	return -1;
daniele	29:1a47b7151851	955	}
daniele	29:1a47b7151851	956	}
daniele	29:1a47b7151851	957
daniele	29:1a47b7151851	958	if ((s->state & PICO_SOCKET_STATE_BOUND) == 0) {
daniele	29:1a47b7151851	959	pico_err = PICO_ERR_EIO;
daniele	29:1a47b7151851	960	return -1;
daniele	29:1a47b7151851	961	}
daniele	29:1a47b7151851	962	if ((s->state & PICO_SOCKET_STATE_CONNECTED) == 0) {
daniele	29:1a47b7151851	963	pico_err = PICO_ERR_ENOTCONN;
daniele	29:1a47b7151851	964	return -1;
daniele	29:1a47b7151851	965	} else if (s->state & PICO_SOCKET_STATE_SHUT_LOCAL) { /* check if in shutdown state */
daniele	29:1a47b7151851	966	pico_err = PICO_ERR_ESHUTDOWN;
daniele	29:1a47b7151851	967	return -1;
daniele	29:1a47b7151851	968	} else {
daniele	29:1a47b7151851	969	return pico_socket_sendto(s, buf, len, &s->remote_addr, s->remote_port);
daniele	29:1a47b7151851	970	}
daniele	29:1a47b7151851	971	}
daniele	29:1a47b7151851	972
daniele	29:1a47b7151851	973	uint16_t pico_socket_high_port(uint16_t proto)
daniele	29:1a47b7151851	974	{
daniele	29:1a47b7151851	975	uint16_t port;
daniele	29:1a47b7151851	976	if (0 \|\|
daniele	29:1a47b7151851	977	#ifdef PICO_SUPPORT_TCP
daniele	29:1a47b7151851	978	(proto == PICO_PROTO_TCP) \|\|
daniele	29:1a47b7151851	979	#endif
daniele	29:1a47b7151851	980	#ifdef PICO_SUPPORT_TCP
daniele	29:1a47b7151851	981	(proto == PICO_PROTO_UDP) \|\|
daniele	29:1a47b7151851	982	#endif
daniele	29:1a47b7151851	983	0) {
daniele	29:1a47b7151851	984	do {
daniele	29:1a47b7151851	985	uint32_t rand = pico_rand();
daniele	29:1a47b7151851	986	port = (uint16_t) (rand & 0xFFFFU);
daniele	29:1a47b7151851	987	port = (uint16_t)(port % (65535 - 1024)) + 1024U;
daniele	29:1a47b7151851	988	if (pico_is_port_free(proto, port, NULL, NULL)) {
daniele	29:1a47b7151851	989	return short_be(port);
daniele	29:1a47b7151851	990	}
daniele	29:1a47b7151851	991	} while(1);
daniele	29:1a47b7151851	992	}
daniele	29:1a47b7151851	993	else return 0U;
daniele	29:1a47b7151851	994	}
daniele	29:1a47b7151851	995
daniele	29:1a47b7151851	996
daniele	29:1a47b7151851	997	int pico_socket_sendto(struct pico_socket s, void buf, int len, void *dst, uint16_t remote_port)
daniele	29:1a47b7151851	998	{
daniele	29:1a47b7151851	999	struct pico_frame *f;
daniele	29:1a47b7151851	1000	struct pico_remote_duple *remote_duple = NULL;
tass	48:40fc4462265c	1001	int socket_mtu = PICO_SOCKET4_MTU;
daniele	29:1a47b7151851	1002	int header_offset = 0;
daniele	29:1a47b7151851	1003	int total_payload_written = 0;
daniele	29:1a47b7151851	1004	#ifdef PICO_SUPPORT_IPV4
daniele	29:1a47b7151851	1005	struct pico_ip4 *src4;
daniele	29:1a47b7151851	1006	#endif
daniele	29:1a47b7151851	1007
daniele	29:1a47b7151851	1008	#ifdef PICO_SUPPORT_IPV6
daniele	29:1a47b7151851	1009	struct pico_ip6 *src6;
daniele	29:1a47b7151851	1010	#endif
daniele	29:1a47b7151851	1011	if (len == 0) {
daniele	29:1a47b7151851	1012	return 0;
daniele	29:1a47b7151851	1013	} else if (len < 0) {
daniele	29:1a47b7151851	1014	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1015	return -1;
daniele	29:1a47b7151851	1016	}
daniele	29:1a47b7151851	1017
daniele	29:1a47b7151851	1018	if (buf == NULL \|\| s == NULL) {
daniele	29:1a47b7151851	1019	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1020	return -1;
daniele	29:1a47b7151851	1021	}
daniele	29:1a47b7151851	1022
daniele	29:1a47b7151851	1023	if (!dst \|\| !remote_port) {
daniele	29:1a47b7151851	1024	pico_err = PICO_ERR_EADDRNOTAVAIL;
daniele	29:1a47b7151851	1025	return -1;
daniele	29:1a47b7151851	1026	}
daniele	29:1a47b7151851	1027
daniele	29:1a47b7151851	1028	if ((s->state & PICO_SOCKET_STATE_CONNECTED) != 0) {
daniele	29:1a47b7151851	1029	if (remote_port != s->remote_port) {
daniele	29:1a47b7151851	1030	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1031	return -1;
daniele	29:1a47b7151851	1032	}
daniele	29:1a47b7151851	1033	}
daniele	29:1a47b7151851	1034
daniele	29:1a47b7151851	1035	#ifdef PICO_SUPPORT_IPV4
daniele	29:1a47b7151851	1036	if (IS_SOCK_IPV4(s)) {
tass	48:40fc4462265c	1037	socket_mtu = PICO_SOCKET4_MTU;
daniele	29:1a47b7151851	1038	if ((s->state & PICO_SOCKET_STATE_CONNECTED)) {
daniele	29:1a47b7151851	1039	if (s->remote_addr.ip4.addr != ((struct pico_ip4 *)dst)->addr ) {
daniele	29:1a47b7151851	1040	pico_err = PICO_ERR_EADDRNOTAVAIL;
daniele	29:1a47b7151851	1041	return -1;
daniele	29:1a47b7151851	1042	}
daniele	29:1a47b7151851	1043	} else {
daniele	29:1a47b7151851	1044	src4 = pico_ipv4_source_find(dst);
daniele	29:1a47b7151851	1045	if (!src4) {
daniele	29:1a47b7151851	1046	pico_err = PICO_ERR_EHOSTUNREACH;
daniele	29:1a47b7151851	1047	return -1;
daniele	29:1a47b7151851	1048	}
daniele	29:1a47b7151851	1049	if (src4->addr != PICO_IPV4_INADDR_ANY)
daniele	29:1a47b7151851	1050	s->local_addr.ip4.addr = src4->addr;
daniele	29:1a47b7151851	1051	# ifdef PICO_SUPPORT_UDP
daniele	29:1a47b7151851	1052	/* socket remote info could change in a consecutive call, make persistent */
daniele	29:1a47b7151851	1053	if (PROTO(s) == PICO_PROTO_UDP) {
daniele	29:1a47b7151851	1054	remote_duple = pico_zalloc(sizeof(struct pico_remote_duple));
daniele	29:1a47b7151851	1055	remote_duple->remote_addr.ip4.addr = ((struct pico_ip4 *)dst)->addr;
daniele	29:1a47b7151851	1056	remote_duple->remote_port = remote_port;
daniele	29:1a47b7151851	1057	}
daniele	29:1a47b7151851	1058	# endif
daniele	29:1a47b7151851	1059	}
tass	48:40fc4462265c	1060	#endif
tass	48:40fc4462265c	1061	} else if (IS_SOCK_IPV6(s)) {
tass	48:40fc4462265c	1062	socket_mtu = PICO_SOCKET6_MTU;
tass	48:40fc4462265c	1063	#ifdef PICO_SUPPORT_IPV6
daniele	29:1a47b7151851	1064	if (s->state & PICO_SOCKET_STATE_CONNECTED) {
daniele	29:1a47b7151851	1065	if (memcmp(&s->remote_addr, dst, PICO_SIZE_IP6))
daniele	29:1a47b7151851	1066	return -1;
daniele	29:1a47b7151851	1067	} else {
daniele	29:1a47b7151851	1068	src6 = pico_ipv6_source_find(dst);
daniele	29:1a47b7151851	1069	if (!src6) {
daniele	29:1a47b7151851	1070	pico_err = PICO_ERR_EHOSTUNREACH;
daniele	29:1a47b7151851	1071	return -1;
daniele	29:1a47b7151851	1072	}
daniele	29:1a47b7151851	1073	memcpy(&s->local_addr, src6, PICO_SIZE_IP6);
daniele	29:1a47b7151851	1074	memcpy(&s->remote_addr, dst, PICO_SIZE_IP6);
daniele	29:1a47b7151851	1075	# ifdef PICO_SUPPORT_UDP
daniele	29:1a47b7151851	1076	if (PROTO(s) == PICO_PROTO_UDP) {
daniele	29:1a47b7151851	1077	remote_duple = pico_zalloc(sizeof(struct pico_remote_duple));
daniele	29:1a47b7151851	1078	remote_duple->remote_addr.ip6.addr = ((struct pico_ip6 *)dst)->addr;
daniele	29:1a47b7151851	1079	remote_duple->remote_port = remote_port;
daniele	29:1a47b7151851	1080	}
daniele	29:1a47b7151851	1081	# endif
daniele	29:1a47b7151851	1082	}
tass	48:40fc4462265c	1083	#endif
daniele	29:1a47b7151851	1084	}
daniele	29:1a47b7151851	1085
daniele	29:1a47b7151851	1086	if ((s->state & PICO_SOCKET_STATE_BOUND) == 0) {
daniele	29:1a47b7151851	1087	s->local_port = pico_socket_high_port(s->proto->proto_number);
daniele	29:1a47b7151851	1088	if (s->local_port == 0) {
daniele	29:1a47b7151851	1089	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1090	return -1;
daniele	29:1a47b7151851	1091	}
daniele	29:1a47b7151851	1092	}
daniele	29:1a47b7151851	1093	if ((s->state & PICO_SOCKET_STATE_CONNECTED) == 0) {
daniele	29:1a47b7151851	1094	s->remote_port = remote_port;
daniele	29:1a47b7151851	1095	}
daniele	29:1a47b7151851	1096
daniele	29:1a47b7151851	1097	#ifdef PICO_SUPPORT_TCP
daniele	29:1a47b7151851	1098	if (PROTO(s) == PICO_PROTO_TCP)
daniele	29:1a47b7151851	1099	header_offset = pico_tcp_overhead(s);
daniele	29:1a47b7151851	1100	#endif
daniele	29:1a47b7151851	1101
daniele	29:1a47b7151851	1102	#ifdef PICO_SUPPORT_UDP
daniele	29:1a47b7151851	1103	if (PROTO(s) == PICO_PROTO_UDP)
daniele	29:1a47b7151851	1104	header_offset = sizeof(struct pico_udp_hdr);
daniele	29:1a47b7151851	1105	#endif
daniele	29:1a47b7151851	1106
daniele	29:1a47b7151851	1107	while (total_payload_written < len) {
daniele	29:1a47b7151851	1108	int transport_len = (len - total_payload_written) + header_offset;
tass	48:40fc4462265c	1109	if (transport_len > socket_mtu)
tass	48:40fc4462265c	1110	transport_len = socket_mtu;
tass	48:40fc4462265c	1111	#ifdef PICO_SUPPORT_IPFRAG
daniele	29:1a47b7151851	1112	else {
daniele	29:1a47b7151851	1113	if (total_payload_written)
daniele	29:1a47b7151851	1114	transport_len -= header_offset; /* last fragment, do not allocate memory for transport header */
daniele	29:1a47b7151851	1115	}
daniele	29:1a47b7151851	1116	#endif /* PICO_SUPPORT_IPFRAG */
daniele	29:1a47b7151851	1117
daniele	29:1a47b7151851	1118	f = pico_socket_frame_alloc(s, transport_len);
daniele	29:1a47b7151851	1119	if (!f) {
daniele	29:1a47b7151851	1120	pico_err = PICO_ERR_ENOMEM;
daniele	29:1a47b7151851	1121	return -1;
daniele	29:1a47b7151851	1122	}
daniele	29:1a47b7151851	1123	f->payload += header_offset;
daniele	29:1a47b7151851	1124	f->payload_len -= header_offset;
daniele	29:1a47b7151851	1125	f->sock = s;
daniele	29:1a47b7151851	1126	if (remote_duple) {
daniele	29:1a47b7151851	1127	f->info = pico_zalloc(sizeof(struct pico_remote_duple));
daniele	29:1a47b7151851	1128	memcpy(f->info, remote_duple, sizeof(struct pico_remote_duple));
daniele	29:1a47b7151851	1129	}
daniele	29:1a47b7151851	1130
tass	48:40fc4462265c	1131	#ifdef PICO_SUPPORT_IPFRAG
tass	48:40fc4462265c	1132	#ifdef PICO_SUPPORT_UDP
tass	48:40fc4462265c	1133	if (PROTO(s) == PICO_PROTO_UDP && ((len + header_offset) > socket_mtu)) {
daniele	29:1a47b7151851	1134	/* hacking way to identify fragmentation frames: payload != transport_hdr -> first frame */
daniele	29:1a47b7151851	1135	if (!total_payload_written) {
daniele	29:1a47b7151851	1136	frag_dbg("FRAG: first fragmented frame %p \| len = %u offset = 0\n", f, f->payload_len);
daniele	29:1a47b7151851	1137	/* transport header length field contains total length + header length */
daniele	29:1a47b7151851	1138	f->transport_len = len + header_offset;
daniele	29:1a47b7151851	1139	f->frag = short_be(PICO_IPV4_MOREFRAG);
daniele	29:1a47b7151851	1140	} else {
daniele	29:1a47b7151851	1141	/* no transport header in fragmented IP */
daniele	29:1a47b7151851	1142	f->payload = f->transport_hdr;
daniele	29:1a47b7151851	1143	f->payload_len += header_offset;
daniele	29:1a47b7151851	1144	/* set offset in octets */
daniele	29:1a47b7151851	1145	f->frag = short_be((total_payload_written + header_offset) / 8);
daniele	29:1a47b7151851	1146	if (total_payload_written + f->payload_len < len) {
daniele	29:1a47b7151851	1147	frag_dbg("FRAG: intermediate fragmented frame %p \| len = %u offset = %u\n", f, f->payload_len, short_be(f->frag));
daniele	29:1a47b7151851	1148	f->frag \|= short_be(PICO_IPV4_MOREFRAG);
daniele	29:1a47b7151851	1149	} else {
daniele	29:1a47b7151851	1150	frag_dbg("FRAG: last fragmented frame %p \| len = %u offset = %u\n", f, f->payload_len, short_be(f->frag));
daniele	29:1a47b7151851	1151	f->frag &= short_be(PICO_IPV4_FRAG_MASK);
daniele	29:1a47b7151851	1152	}
daniele	29:1a47b7151851	1153	}
daniele	29:1a47b7151851	1154	} else {
daniele	29:1a47b7151851	1155	f->frag = short_be(PICO_IPV4_DONTFRAG);
daniele	29:1a47b7151851	1156	}
daniele	29:1a47b7151851	1157	# endif /* PICO_SUPPORT_UDP */
daniele	29:1a47b7151851	1158	#endif /* PICO_SUPPORT_IPFRAG */
daniele	29:1a47b7151851	1159
daniele	29:1a47b7151851	1160	if (f->payload_len <= 0) {
daniele	29:1a47b7151851	1161	pico_frame_discard(f);
daniele	29:1a47b7151851	1162	if (remote_duple)
daniele	29:1a47b7151851	1163	pico_free(remote_duple);
daniele	29:1a47b7151851	1164	return total_payload_written;
daniele	29:1a47b7151851	1165	}
daniele	29:1a47b7151851	1166
daniele	29:1a47b7151851	1167	memcpy(f->payload, buf + total_payload_written, f->payload_len);
daniele	29:1a47b7151851	1168	//dbg("Pushing segment, hdr len: %d, payload_len: %d\n", header_offset, f->payload_len);
daniele	29:1a47b7151851	1169
daniele	29:1a47b7151851	1170	if (s->proto->push(s->proto, f) > 0) {
daniele	29:1a47b7151851	1171	total_payload_written += f->payload_len;
daniele	29:1a47b7151851	1172	} else {
daniele	29:1a47b7151851	1173	pico_frame_discard(f);
daniele	29:1a47b7151851	1174	pico_err = PICO_ERR_EAGAIN;
daniele	29:1a47b7151851	1175	break;
daniele	29:1a47b7151851	1176	}
daniele	29:1a47b7151851	1177	}
daniele	29:1a47b7151851	1178	if (remote_duple)
daniele	29:1a47b7151851	1179	pico_free(remote_duple);
daniele	29:1a47b7151851	1180	return total_payload_written;
daniele	29:1a47b7151851	1181	}
daniele	29:1a47b7151851	1182
daniele	29:1a47b7151851	1183	int pico_socket_send(struct pico_socket s, void buf, int len)
daniele	29:1a47b7151851	1184	{
daniele	29:1a47b7151851	1185	if (!s \|\| buf == NULL) {
daniele	29:1a47b7151851	1186	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1187	return -1;
daniele	29:1a47b7151851	1188	} else {
daniele	29:1a47b7151851	1189	/* check if exists in tree */
daniele	29:1a47b7151851	1190	/* See task #178 */
daniele	29:1a47b7151851	1191	if (pico_check_socket(s) != 0) {
daniele	29:1a47b7151851	1192	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1193	return -1;
daniele	29:1a47b7151851	1194	}
daniele	29:1a47b7151851	1195	}
daniele	29:1a47b7151851	1196
daniele	29:1a47b7151851	1197	if ((s->state & PICO_SOCKET_STATE_CONNECTED) == 0) {
daniele	29:1a47b7151851	1198	pico_err = PICO_ERR_ENOTCONN;
daniele	29:1a47b7151851	1199	return -1;
daniele	29:1a47b7151851	1200	}
daniele	29:1a47b7151851	1201	return pico_socket_sendto(s, buf, len, &s->remote_addr, s->remote_port);
daniele	29:1a47b7151851	1202	}
daniele	29:1a47b7151851	1203
daniele	29:1a47b7151851	1204	int pico_socket_recvfrom(struct pico_socket s, void buf, int len, void orig, uint16_t remote_port)
daniele	29:1a47b7151851	1205	{
daniele	29:1a47b7151851	1206	if (!s \|\| buf == NULL) { /// \|\| orig == NULL \|\| remote_port == NULL) {
daniele	29:1a47b7151851	1207	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1208	return -1;
daniele	29:1a47b7151851	1209	} else {
daniele	29:1a47b7151851	1210	/* check if exists in tree */
daniele	29:1a47b7151851	1211	if (pico_check_socket(s) != 0) {
daniele	29:1a47b7151851	1212	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1213	/* See task #178 */
daniele	29:1a47b7151851	1214	return -1;
daniele	29:1a47b7151851	1215	}
daniele	29:1a47b7151851	1216	}
daniele	29:1a47b7151851	1217
daniele	29:1a47b7151851	1218	if ((s->state & PICO_SOCKET_STATE_BOUND) == 0) {
daniele	29:1a47b7151851	1219	pico_err = PICO_ERR_EADDRNOTAVAIL;
daniele	29:1a47b7151851	1220	return -1;
daniele	29:1a47b7151851	1221	}
daniele	29:1a47b7151851	1222	#ifdef PICO_SUPPORT_UDP
daniele	29:1a47b7151851	1223	if (PROTO(s) == PICO_PROTO_UDP) {
daniele	29:1a47b7151851	1224	return pico_udp_recv(s, buf, len, orig, remote_port);
daniele	29:1a47b7151851	1225	}
daniele	29:1a47b7151851	1226	#endif
daniele	29:1a47b7151851	1227	#ifdef PICO_SUPPORT_TCP
daniele	29:1a47b7151851	1228	if (PROTO(s) == PICO_PROTO_TCP) {
daniele	29:1a47b7151851	1229	/* check if in shutdown state and if tcpq_in empty */
daniele	29:1a47b7151851	1230	if ((s->state & PICO_SOCKET_STATE_SHUT_REMOTE) && pico_tcp_queue_in_is_empty(s)) {
daniele	29:1a47b7151851	1231	pico_err = PICO_ERR_ESHUTDOWN;
daniele	29:1a47b7151851	1232	return -1;
daniele	29:1a47b7151851	1233	} else {
daniele	29:1a47b7151851	1234	//dbg("socket tcp recv\n");
daniele	29:1a47b7151851	1235	return pico_tcp_read(s, buf, len);
daniele	29:1a47b7151851	1236	}
daniele	29:1a47b7151851	1237	}
daniele	29:1a47b7151851	1238	#endif
daniele	29:1a47b7151851	1239	//dbg("socket return 0\n");
daniele	29:1a47b7151851	1240	return 0;
daniele	29:1a47b7151851	1241	}
daniele	29:1a47b7151851	1242
daniele	29:1a47b7151851	1243	int pico_socket_recv(struct pico_socket s, void buf, int len)
daniele	29:1a47b7151851	1244	{
daniele	29:1a47b7151851	1245	return pico_socket_recvfrom(s, buf, len, NULL, NULL);
daniele	29:1a47b7151851	1246	}
daniele	29:1a47b7151851	1247
daniele	29:1a47b7151851	1248
daniele	29:1a47b7151851	1249	int pico_socket_bind(struct pico_socket s, void local_addr, uint16_t *port)
daniele	29:1a47b7151851	1250	{
daniele	29:1a47b7151851	1251	if (!s \|\| !local_addr \|\| !port) {
daniele	29:1a47b7151851	1252	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1253	return -1;
daniele	29:1a47b7151851	1254	}
daniele	29:1a47b7151851	1255
daniele	29:1a47b7151851	1256	if (!is_sock_ipv6(s)) {
daniele	29:1a47b7151851	1257	struct pico_ip4 ip = (struct pico_ip4 )local_addr;
daniele	29:1a47b7151851	1258	if (ip->addr != PICO_IPV4_INADDR_ANY) {
daniele	29:1a47b7151851	1259	if (!pico_ipv4_link_find(local_addr)) {
daniele	29:1a47b7151851	1260	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1261	return -1;
daniele	29:1a47b7151851	1262	}
daniele	29:1a47b7151851	1263	}
daniele	29:1a47b7151851	1264	} else {
daniele	29:1a47b7151851	1265	/... IPv6 /
daniele	29:1a47b7151851	1266	}
daniele	29:1a47b7151851	1267
daniele	29:1a47b7151851	1268
daniele	29:1a47b7151851	1269	/* When given port = 0, get a random high port to bind to. */
daniele	29:1a47b7151851	1270	if (*port == 0) {
daniele	29:1a47b7151851	1271	*port = pico_socket_high_port(PROTO(s));
daniele	29:1a47b7151851	1272	if (*port == 0) {
daniele	29:1a47b7151851	1273	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1274	return -1;
daniele	29:1a47b7151851	1275	}
daniele	29:1a47b7151851	1276	}
daniele	29:1a47b7151851	1277
daniele	29:1a47b7151851	1278	if (pico_is_port_free(PROTO(s), *port, local_addr, s->net) == 0) {
daniele	29:1a47b7151851	1279	pico_err = PICO_ERR_EADDRINUSE;
daniele	29:1a47b7151851	1280	return -1;
daniele	29:1a47b7151851	1281	}
daniele	29:1a47b7151851	1282	s->local_port = *port;
daniele	29:1a47b7151851	1283
daniele	29:1a47b7151851	1284	if (is_sock_ipv6(s)) {
daniele	29:1a47b7151851	1285	struct pico_ip6 ip = (struct pico_ip6 ) local_addr;
daniele	29:1a47b7151851	1286	memcpy(s->local_addr.ip6.addr, ip, PICO_SIZE_IP6);
daniele	29:1a47b7151851	1287	/* XXX: port ipv4 functionality to ipv6 */
daniele	29:1a47b7151851	1288	/* Check for port already in use */
daniele	29:1a47b7151851	1289	if (pico_is_port_free(PROTO(s), *port, &local_addr, s->net)) {
daniele	29:1a47b7151851	1290	pico_err = PICO_ERR_EADDRINUSE;
daniele	29:1a47b7151851	1291	return -1;
daniele	29:1a47b7151851	1292	}
daniele	29:1a47b7151851	1293	} else if (is_sock_ipv4(s)) {
daniele	29:1a47b7151851	1294	struct pico_ip4 ip = (struct pico_ip4 ) local_addr;
daniele	29:1a47b7151851	1295	s->local_addr.ip4.addr = ip->addr;
daniele	29:1a47b7151851	1296	}
daniele	29:1a47b7151851	1297	return pico_socket_alter_state(s, PICO_SOCKET_STATE_BOUND, 0, 0);
daniele	29:1a47b7151851	1298	}
daniele	29:1a47b7151851	1299
daniele	29:1a47b7151851	1300	int pico_socket_connect(struct pico_socket s, void remote_addr, uint16_t remote_port)
daniele	29:1a47b7151851	1301	{
daniele	29:1a47b7151851	1302	int ret = -1;
daniele	29:1a47b7151851	1303	pico_err = PICO_ERR_EPROTONOSUPPORT;
daniele	29:1a47b7151851	1304	if (!s \|\| remote_addr == NULL \|\| remote_port == 0) {
daniele	29:1a47b7151851	1305	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1306	return -1;
daniele	29:1a47b7151851	1307	}
daniele	29:1a47b7151851	1308
daniele	29:1a47b7151851	1309	s->remote_port = remote_port;
daniele	29:1a47b7151851	1310
daniele	29:1a47b7151851	1311	if (s->local_port == 0) {
daniele	29:1a47b7151851	1312	s->local_port = pico_socket_high_port(PROTO(s));
daniele	29:1a47b7151851	1313	if (!s->local_port) {
daniele	29:1a47b7151851	1314	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1315	return -1;
daniele	29:1a47b7151851	1316	}
daniele	29:1a47b7151851	1317	}
daniele	29:1a47b7151851	1318
daniele	29:1a47b7151851	1319	if (is_sock_ipv6(s)) {
daniele	29:1a47b7151851	1320	struct pico_ip6 ip = (struct pico_ip6 ) remote_addr;
daniele	29:1a47b7151851	1321	memcpy(s->remote_addr.ip6.addr, ip, PICO_SIZE_IP6);
daniele	29:1a47b7151851	1322	} else if (is_sock_ipv4(s)) {
daniele	29:1a47b7151851	1323	struct pico_ip4 local, ip = (struct pico_ip4 *) remote_addr;
daniele	29:1a47b7151851	1324	s->remote_addr.ip4.addr = ip->addr;
daniele	29:1a47b7151851	1325	local = pico_ipv4_source_find(ip);
daniele	29:1a47b7151851	1326	if (local) {
daniele	29:1a47b7151851	1327	s->local_addr.ip4.addr = local->addr;
daniele	29:1a47b7151851	1328	} else {
daniele	29:1a47b7151851	1329	pico_err = PICO_ERR_EHOSTUNREACH;
daniele	29:1a47b7151851	1330	return -1;
daniele	29:1a47b7151851	1331	}
daniele	29:1a47b7151851	1332	}
daniele	29:1a47b7151851	1333
daniele	29:1a47b7151851	1334	pico_socket_alter_state(s, PICO_SOCKET_STATE_BOUND, 0, 0);
daniele	29:1a47b7151851	1335
daniele	29:1a47b7151851	1336	#ifdef PICO_SUPPORT_UDP
daniele	29:1a47b7151851	1337	if (PROTO(s) == PICO_PROTO_UDP) {
daniele	29:1a47b7151851	1338	pico_socket_alter_state(s, PICO_SOCKET_STATE_CONNECTED, 0, 0);
daniele	29:1a47b7151851	1339	pico_err = PICO_ERR_NOERR;
daniele	29:1a47b7151851	1340	ret = 0;
daniele	29:1a47b7151851	1341	}
daniele	29:1a47b7151851	1342	#endif
daniele	29:1a47b7151851	1343
daniele	29:1a47b7151851	1344	#ifdef PICO_SUPPORT_TCP
daniele	29:1a47b7151851	1345	if (PROTO(s) == PICO_PROTO_TCP) {
daniele	29:1a47b7151851	1346	if (pico_tcp_initconn(s) == 0) {
daniele	29:1a47b7151851	1347	pico_socket_alter_state(s, PICO_SOCKET_STATE_CONNECTED \| PICO_SOCKET_STATE_TCP_SYN_SENT, 0, 0);
daniele	29:1a47b7151851	1348	pico_err = PICO_ERR_NOERR;
daniele	29:1a47b7151851	1349	ret = 0;
daniele	29:1a47b7151851	1350	} else {
daniele	29:1a47b7151851	1351	pico_err = PICO_ERR_EHOSTUNREACH;
daniele	29:1a47b7151851	1352	}
daniele	29:1a47b7151851	1353	}
daniele	29:1a47b7151851	1354	#endif
daniele	29:1a47b7151851	1355
daniele	29:1a47b7151851	1356	return ret;
daniele	29:1a47b7151851	1357	}
daniele	29:1a47b7151851	1358
daniele	29:1a47b7151851	1359	#ifdef PICO_SUPPORT_TCP
daniele	29:1a47b7151851	1360
daniele	29:1a47b7151851	1361	int pico_socket_listen(struct pico_socket *s, int backlog)
daniele	29:1a47b7151851	1362	{
daniele	29:1a47b7151851	1363	if (!s \|\| backlog < 1) {
daniele	29:1a47b7151851	1364	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1365	return -1;
daniele	29:1a47b7151851	1366	} else {
daniele	29:1a47b7151851	1367	/* check if exists in tree */
daniele	29:1a47b7151851	1368	/* See task #178 */
daniele	29:1a47b7151851	1369	if (pico_check_socket(s) != 0) {
daniele	29:1a47b7151851	1370	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1371	return -1;
daniele	29:1a47b7151851	1372	}
daniele	29:1a47b7151851	1373	}
daniele	29:1a47b7151851	1374
daniele	29:1a47b7151851	1375	if (PROTO(s) == PICO_PROTO_UDP) {
daniele	29:1a47b7151851	1376	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1377	return -1;
daniele	29:1a47b7151851	1378	}
daniele	29:1a47b7151851	1379
daniele	29:1a47b7151851	1380	if ((s->state & PICO_SOCKET_STATE_BOUND) == 0) {
daniele	29:1a47b7151851	1381	pico_err = PICO_ERR_EISCONN;
daniele	29:1a47b7151851	1382	return -1;
daniele	29:1a47b7151851	1383	}
daniele	29:1a47b7151851	1384
daniele	29:1a47b7151851	1385	if (backlog < 1) {
daniele	29:1a47b7151851	1386	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1387	return -1;
daniele	29:1a47b7151851	1388	}
daniele	29:1a47b7151851	1389
daniele	29:1a47b7151851	1390	if (PROTO(s) == PICO_PROTO_TCP)
daniele	29:1a47b7151851	1391	pico_socket_alter_state(s, PICO_SOCKET_STATE_TCP_SYN_SENT, 0, PICO_SOCKET_STATE_TCP_LISTEN);
daniele	29:1a47b7151851	1392	s->max_backlog = backlog;
daniele	29:1a47b7151851	1393
daniele	29:1a47b7151851	1394	return 0;
daniele	29:1a47b7151851	1395	}
daniele	29:1a47b7151851	1396
daniele	29:1a47b7151851	1397	struct pico_socket pico_socket_accept(struct pico_socket s, void orig, uint16_t port)
daniele	29:1a47b7151851	1398	{
daniele	29:1a47b7151851	1399	if (!s \|\| !orig \|\| !port) {
daniele	29:1a47b7151851	1400	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1401	return NULL;
daniele	29:1a47b7151851	1402	}
daniele	29:1a47b7151851	1403
daniele	29:1a47b7151851	1404	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1405
daniele	29:1a47b7151851	1406	if ((s->state & PICO_SOCKET_STATE_BOUND) == 0) {
daniele	29:1a47b7151851	1407	return NULL;
daniele	29:1a47b7151851	1408	}
daniele	29:1a47b7151851	1409
daniele	29:1a47b7151851	1410	if (PROTO(s) == PICO_PROTO_UDP) {
daniele	29:1a47b7151851	1411	return NULL;
daniele	29:1a47b7151851	1412	}
daniele	29:1a47b7151851	1413
daniele	29:1a47b7151851	1414	if (TCPSTATE(s) == PICO_SOCKET_STATE_TCP_LISTEN) {
daniele	29:1a47b7151851	1415	struct pico_sockport *sp = pico_get_sockport(PICO_PROTO_TCP, s->local_port);
daniele	29:1a47b7151851	1416	struct pico_socket *found;
daniele	29:1a47b7151851	1417	/* If at this point no incoming connection socket is found,
daniele	29:1a47b7151851	1418	* the accept call is valid, but no connection is established yet.
daniele	29:1a47b7151851	1419	*/
daniele	29:1a47b7151851	1420	pico_err = PICO_ERR_EAGAIN;
daniele	29:1a47b7151851	1421	if (sp) {
daniele	29:1a47b7151851	1422	struct pico_tree_node * index;
daniele	29:1a47b7151851	1423	//RB_FOREACH(found, socket_tree, &sp->socks) {
daniele	29:1a47b7151851	1424	pico_tree_foreach(index,&sp->socks){
daniele	29:1a47b7151851	1425	found = index->keyValue;
daniele	29:1a47b7151851	1426	if (s == found->parent) {
daniele	29:1a47b7151851	1427	found->parent = NULL;
daniele	29:1a47b7151851	1428	pico_err = PICO_ERR_NOERR;
daniele	29:1a47b7151851	1429	memcpy(orig, &found->remote_addr, sizeof(struct pico_ip4));
daniele	29:1a47b7151851	1430	*port = found->remote_port;
daniele	29:1a47b7151851	1431	return found;
daniele	29:1a47b7151851	1432	}
daniele	29:1a47b7151851	1433	}
daniele	29:1a47b7151851	1434	}
daniele	29:1a47b7151851	1435	}
daniele	29:1a47b7151851	1436	return NULL;
daniele	29:1a47b7151851	1437	}
daniele	29:1a47b7151851	1438
daniele	29:1a47b7151851	1439	#else
daniele	29:1a47b7151851	1440
daniele	29:1a47b7151851	1441	int pico_socket_listen(struct pico_socket *s, int backlog)
daniele	29:1a47b7151851	1442	{
daniele	29:1a47b7151851	1443	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1444	return -1;
daniele	29:1a47b7151851	1445	}
daniele	29:1a47b7151851	1446
daniele	29:1a47b7151851	1447	struct pico_socket pico_socket_accept(struct pico_socket s, void orig, uint16_t local_port)
daniele	29:1a47b7151851	1448	{
daniele	29:1a47b7151851	1449	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1450	return NULL;
daniele	29:1a47b7151851	1451	}
daniele	29:1a47b7151851	1452
daniele	29:1a47b7151851	1453	#endif
daniele	29:1a47b7151851	1454
daniele	29:1a47b7151851	1455	#define PICO_SOCKET_SETOPT_EN(socket,index) (socket->opt_flags \|= (1 << index))
daniele	29:1a47b7151851	1456	#define PICO_SOCKET_SETOPT_DIS(socket,index) (socket->opt_flags &= ~(1 << index))
daniele	29:1a47b7151851	1457
daniele	29:1a47b7151851	1458	int pico_socket_setoption(struct pico_socket s, int option, void value) // XXX no check against proto (vs setsockopt) or implicit by socket?
daniele	29:1a47b7151851	1459	{
daniele	29:1a47b7151851	1460	if (s == NULL) {
daniele	29:1a47b7151851	1461	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1462	return -1;
daniele	29:1a47b7151851	1463	}
daniele	29:1a47b7151851	1464
daniele	29:1a47b7151851	1465	pico_err = PICO_ERR_NOERR;
daniele	29:1a47b7151851	1466
daniele	29:1a47b7151851	1467	switch (option)
daniele	29:1a47b7151851	1468	{
daniele	29:1a47b7151851	1469	#ifdef PICO_SUPPORT_TCP
daniele	29:1a47b7151851	1470	case PICO_TCP_NODELAY:
daniele	29:1a47b7151851	1471	if (!value) {
daniele	29:1a47b7151851	1472	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1473	return -1;
daniele	29:1a47b7151851	1474	}
daniele	29:1a47b7151851	1475	if (s->proto->proto_number == PICO_PROTO_TCP) {
daniele	29:1a47b7151851	1476	int val = (int)value;
daniele	29:1a47b7151851	1477	if (*val > 0) {
daniele	29:1a47b7151851	1478	dbg("setsockopt: Nagle algorithm disabled.\n");
daniele	29:1a47b7151851	1479	PICO_SOCKET_SETOPT_EN(s,PICO_SOCKET_OPT_TCPNODELAY);
daniele	29:1a47b7151851	1480	} else {
daniele	29:1a47b7151851	1481	dbg("setsockopt: Nagle algorithm enabled.\n");
daniele	29:1a47b7151851	1482	PICO_SOCKET_SETOPT_DIS(s,PICO_SOCKET_OPT_TCPNODELAY);
daniele	29:1a47b7151851	1483	}
daniele	29:1a47b7151851	1484	} else {
daniele	29:1a47b7151851	1485	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1486	}
daniele	29:1a47b7151851	1487	break;
daniele	29:1a47b7151851	1488	#endif
daniele	29:1a47b7151851	1489
daniele	29:1a47b7151851	1490
daniele	29:1a47b7151851	1491	#ifdef PICO_SUPPORT_MCAST
daniele	29:1a47b7151851	1492	case PICO_IP_MULTICAST_IF:
daniele	29:1a47b7151851	1493	pico_err = PICO_ERR_EOPNOTSUPP;
daniele	29:1a47b7151851	1494	return -1;
daniele	29:1a47b7151851	1495	break;
daniele	29:1a47b7151851	1496
daniele	29:1a47b7151851	1497	case PICO_IP_MULTICAST_TTL:
daniele	29:1a47b7151851	1498	if (s->proto->proto_number == PICO_PROTO_UDP) {
daniele	29:1a47b7151851	1499	return pico_udp_set_mc_ttl(s, ((uint8_t ) value));
daniele	29:1a47b7151851	1500	}
daniele	29:1a47b7151851	1501	break;
daniele	29:1a47b7151851	1502
daniele	29:1a47b7151851	1503	case PICO_IP_MULTICAST_LOOP:
daniele	29:1a47b7151851	1504	if (s->proto->proto_number == PICO_PROTO_UDP) {
daniele	29:1a47b7151851	1505	switch ((uint8_t ) value)
daniele	29:1a47b7151851	1506	{
daniele	29:1a47b7151851	1507	case 0:
daniele	29:1a47b7151851	1508	/* do not loop back multicast datagram */
daniele	29:1a47b7151851	1509	PICO_SOCKET_SETOPT_DIS(s,PICO_SOCKET_OPT_MULTICAST_LOOP);
daniele	29:1a47b7151851	1510	break;
daniele	29:1a47b7151851	1511
daniele	29:1a47b7151851	1512	case 1:
daniele	29:1a47b7151851	1513	/* do loop back multicast datagram */
daniele	29:1a47b7151851	1514	PICO_SOCKET_SETOPT_EN(s,PICO_SOCKET_OPT_MULTICAST_LOOP);
daniele	29:1a47b7151851	1515	break;
daniele	29:1a47b7151851	1516
daniele	29:1a47b7151851	1517	default:
daniele	29:1a47b7151851	1518	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1519	return -1;
daniele	29:1a47b7151851	1520	}
daniele	29:1a47b7151851	1521	}
daniele	29:1a47b7151851	1522	break;
daniele	29:1a47b7151851	1523
daniele	29:1a47b7151851	1524	case PICO_IP_ADD_MEMBERSHIP:
daniele	29:1a47b7151851	1525	/* EXCLUDE mode */
daniele	29:1a47b7151851	1526	if (s->proto->proto_number == PICO_PROTO_UDP) {
tass	48:40fc4462265c	1527	int filter_mode;
daniele	29:1a47b7151851	1528	struct pico_ip_mreq *mreq = NULL;
daniele	29:1a47b7151851	1529	struct pico_mcast_listen *listen = NULL, ltest = {0};
daniele	29:1a47b7151851	1530	struct pico_ipv4_link *mcast_link = NULL;
daniele	29:1a47b7151851	1531
daniele	29:1a47b7151851	1532	if (!value) {
daniele	29:1a47b7151851	1533	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1534	return -1;
daniele	29:1a47b7151851	1535	}
daniele	29:1a47b7151851	1536	mreq = (struct pico_ip_mreq *) value;
daniele	29:1a47b7151851	1537	mcast_link = pico_socket_setoption_mcastargs_validation(mreq, NULL);
daniele	29:1a47b7151851	1538	if (!mcast_link) {
daniele	29:1a47b7151851	1539	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1540	return -1;
daniele	29:1a47b7151851	1541	}
daniele	29:1a47b7151851	1542	if (!mreq->mcast_link_addr.addr)
daniele	29:1a47b7151851	1543	mreq->mcast_link_addr.addr = mcast_link->address.addr;
daniele	29:1a47b7151851	1544
daniele	29:1a47b7151851	1545	if (!s->MCASTListen) { /* No RBTree allocated yet */
daniele	29:1a47b7151851	1546	s->MCASTListen = pico_zalloc(sizeof(struct pico_tree));
daniele	29:1a47b7151851	1547	if (!s->MCASTListen) {
daniele	29:1a47b7151851	1548	pico_err = PICO_ERR_ENOMEM;
daniele	29:1a47b7151851	1549	return -1;
daniele	29:1a47b7151851	1550	}
daniele	29:1a47b7151851	1551	s->MCASTListen->root = &LEAF;
daniele	29:1a47b7151851	1552	s->MCASTListen->compare = mcast_listen_cmp;
daniele	29:1a47b7151851	1553	}
daniele	29:1a47b7151851	1554	ltest.mcast_link = mreq->mcast_link_addr;
daniele	29:1a47b7151851	1555	ltest.mcast_group = mreq->mcast_group_addr;
daniele	29:1a47b7151851	1556	listen = pico_tree_findKey(s->MCASTListen, &ltest);
daniele	29:1a47b7151851	1557	if (listen) {
daniele	29:1a47b7151851	1558	if (listen->filter_mode != PICO_IP_MULTICAST_EXCLUDE) {
daniele	29:1a47b7151851	1559	so_mcast_dbg("pico_socket_setoption: ERROR any-source multicast (exclude) on source-specific multicast (include)\n");
daniele	29:1a47b7151851	1560	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1561	return -1;
daniele	29:1a47b7151851	1562	} else {
daniele	29:1a47b7151851	1563	so_mcast_dbg("pico_socket_setoption: ERROR duplicate PICO_IP_ADD_MEMBERSHIP\n");
daniele	29:1a47b7151851	1564	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1565	return -1;
daniele	29:1a47b7151851	1566	}
daniele	29:1a47b7151851	1567	} else {
daniele	29:1a47b7151851	1568	listen = pico_zalloc(sizeof(struct pico_mcast_listen));
daniele	29:1a47b7151851	1569	if (!listen) {
daniele	29:1a47b7151851	1570	pico_err = PICO_ERR_ENOMEM;
daniele	29:1a47b7151851	1571	return -1;
daniele	29:1a47b7151851	1572	}
daniele	29:1a47b7151851	1573	listen->filter_mode = PICO_IP_MULTICAST_EXCLUDE;
daniele	29:1a47b7151851	1574	listen->mcast_link = mreq->mcast_link_addr;
daniele	29:1a47b7151851	1575	listen->mcast_group = mreq->mcast_group_addr;
daniele	29:1a47b7151851	1576	listen->MCASTSources.root = &LEAF;
daniele	29:1a47b7151851	1577	listen->MCASTSources.compare = mcast_sources_cmp;
daniele	29:1a47b7151851	1578	pico_tree_insert(s->MCASTListen, listen);
daniele	29:1a47b7151851	1579	}
daniele	29:1a47b7151851	1580
daniele	29:1a47b7151851	1581	pico_tree_insert(&MCASTSockets, s);
daniele	29:1a47b7151851	1582	filter_mode = pico_socket_aggregate_mcastfilters(&mcast_link->address, &mreq->mcast_group_addr);
daniele	29:1a47b7151851	1583	if (filter_mode < 0)
daniele	29:1a47b7151851	1584	return -1;
daniele	29:1a47b7151851	1585
daniele	29:1a47b7151851	1586	return pico_ipv4_mcast_join(&mreq->mcast_link_addr, &mreq->mcast_group_addr, 1, filter_mode, &MCASTFilter);
daniele	29:1a47b7151851	1587	}
daniele	29:1a47b7151851	1588	break;
daniele	29:1a47b7151851	1589
daniele	29:1a47b7151851	1590	case PICO_IP_DROP_MEMBERSHIP:
daniele	29:1a47b7151851	1591	/* EXCLUDE mode */
daniele	29:1a47b7151851	1592	if (s->proto->proto_number == PICO_PROTO_UDP) {
tass	48:40fc4462265c	1593	int filter_mode = 0;
daniele	29:1a47b7151851	1594	struct pico_ip_mreq *mreq = NULL;
daniele	29:1a47b7151851	1595	struct pico_mcast_listen *listen = NULL, ltest = {0};
daniele	29:1a47b7151851	1596	struct pico_ip4 *source = NULL;
daniele	29:1a47b7151851	1597	struct pico_ipv4_link *mcast_link = NULL;
daniele	29:1a47b7151851	1598	struct pico_tree_node index, _tmp;
daniele	29:1a47b7151851	1599
daniele	29:1a47b7151851	1600	if (!value) {
daniele	29:1a47b7151851	1601	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1602	return -1;
daniele	29:1a47b7151851	1603	}
daniele	29:1a47b7151851	1604	mreq = (struct pico_ip_mreq *) value;
daniele	29:1a47b7151851	1605	mcast_link = pico_socket_setoption_mcastargs_validation(mreq, NULL);
daniele	29:1a47b7151851	1606	if (!mcast_link) {
daniele	29:1a47b7151851	1607	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1608	return -1;
daniele	29:1a47b7151851	1609	}
daniele	29:1a47b7151851	1610	if (!mreq->mcast_link_addr.addr)
daniele	29:1a47b7151851	1611	mreq->mcast_link_addr.addr = mcast_link->address.addr;
daniele	29:1a47b7151851	1612
daniele	29:1a47b7151851	1613	if (!s->MCASTListen) { /* No RBTree allocated yet */
daniele	29:1a47b7151851	1614	so_mcast_dbg("socket_setoption: ERROR PICO_IP_DROP_MEMBERSHIP before any PICO_IP_ADD_MEMBERSHIP/SOURCE_MEMBERSHIP\n");
daniele	29:1a47b7151851	1615	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1616	return -1;
daniele	29:1a47b7151851	1617	}
daniele	29:1a47b7151851	1618	ltest.mcast_link = mreq->mcast_link_addr;
daniele	29:1a47b7151851	1619	ltest.mcast_group = mreq->mcast_group_addr;
daniele	29:1a47b7151851	1620	listen = pico_tree_findKey(s->MCASTListen, &ltest);
daniele	29:1a47b7151851	1621	if (!listen) {
daniele	29:1a47b7151851	1622	so_mcast_dbg("pico_socket_setoption: ERROR PICO_IP_DROP_MEMBERSHIP before PICO_IP_ADD_MEMBERSHIP/SOURCE_MEMBERSHIP\n");
daniele	29:1a47b7151851	1623	pico_err = PICO_ERR_EADDRNOTAVAIL;
daniele	29:1a47b7151851	1624	return -1;
daniele	29:1a47b7151851	1625	} else {
daniele	29:1a47b7151851	1626	pico_tree_foreach_safe(index, &listen->MCASTSources, _tmp)
daniele	29:1a47b7151851	1627	{
daniele	29:1a47b7151851	1628	source = index->keyValue;
daniele	29:1a47b7151851	1629	pico_tree_delete(&listen->MCASTSources, source);
daniele	29:1a47b7151851	1630	pico_free(source);
daniele	29:1a47b7151851	1631	}
daniele	29:1a47b7151851	1632	pico_tree_delete(s->MCASTListen, listen);
daniele	29:1a47b7151851	1633	pico_free(listen);
daniele	29:1a47b7151851	1634	if (pico_tree_empty(s->MCASTListen)) {
daniele	29:1a47b7151851	1635	pico_free(s->MCASTListen);
daniele	29:1a47b7151851	1636	s->MCASTListen = NULL;
daniele	29:1a47b7151851	1637	pico_tree_delete(&MCASTSockets, s);
daniele	29:1a47b7151851	1638	}
daniele	29:1a47b7151851	1639	}
daniele	29:1a47b7151851	1640
daniele	29:1a47b7151851	1641	filter_mode = pico_socket_aggregate_mcastfilters(&mcast_link->address, &mreq->mcast_group_addr);
daniele	29:1a47b7151851	1642	if (filter_mode < 0)
daniele	29:1a47b7151851	1643	return -1;
daniele	29:1a47b7151851	1644
daniele	29:1a47b7151851	1645	return pico_ipv4_mcast_leave(&mreq->mcast_link_addr, &mreq->mcast_group_addr, 1, filter_mode, &MCASTFilter);
daniele	29:1a47b7151851	1646	}
daniele	29:1a47b7151851	1647	break;
daniele	29:1a47b7151851	1648
daniele	29:1a47b7151851	1649	case PICO_IP_UNBLOCK_SOURCE:
daniele	29:1a47b7151851	1650	/* EXCLUDE mode */
daniele	29:1a47b7151851	1651	if (s->proto->proto_number == PICO_PROTO_UDP) {
tass	48:40fc4462265c	1652	int filter_mode = 0;
daniele	29:1a47b7151851	1653	struct pico_ip_mreq_source *mreq = NULL;
daniele	29:1a47b7151851	1654	struct pico_mcast_listen *listen = NULL, ltest = {0};
daniele	29:1a47b7151851	1655	struct pico_ip4 *source = NULL, stest = {0};
daniele	29:1a47b7151851	1656	struct pico_ipv4_link *mcast_link = NULL;
daniele	29:1a47b7151851	1657
daniele	29:1a47b7151851	1658	if (!value) {
daniele	29:1a47b7151851	1659	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1660	return -1;
daniele	29:1a47b7151851	1661	}
daniele	29:1a47b7151851	1662	mreq = (struct pico_ip_mreq_source *) value;
daniele	29:1a47b7151851	1663	mcast_link = pico_socket_setoption_mcastargs_validation(NULL, mreq);
daniele	29:1a47b7151851	1664	if (!mcast_link) {
daniele	29:1a47b7151851	1665	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1666	return -1;
daniele	29:1a47b7151851	1667	}
daniele	29:1a47b7151851	1668	if (!mreq->mcast_link_addr.addr)
daniele	29:1a47b7151851	1669	mreq->mcast_link_addr.addr = mcast_link->address.addr;
daniele	29:1a47b7151851	1670
daniele	29:1a47b7151851	1671	if (!s->MCASTListen) { /* No RBTree allocated yet */
daniele	29:1a47b7151851	1672	so_mcast_dbg("pico_socket_setoption: ERROR PICO_IP_UNBLOCK_SOURCE before any PICO_IP_ADD_MEMBERSHIP\n");
daniele	29:1a47b7151851	1673	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1674	return -1;
daniele	29:1a47b7151851	1675	}
daniele	29:1a47b7151851	1676	ltest.mcast_link = mreq->mcast_link_addr;
daniele	29:1a47b7151851	1677	ltest.mcast_group = mreq->mcast_group_addr;
daniele	29:1a47b7151851	1678	listen = pico_tree_findKey(s->MCASTListen, &ltest);
daniele	29:1a47b7151851	1679	if (!listen) {
daniele	29:1a47b7151851	1680	so_mcast_dbg("pico_socket_setoption: ERROR PICO_IP_UNBLOCK_SOURCE before PICO_IP_ADD_MEMBERSHIP\n");
daniele	29:1a47b7151851	1681	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1682	return -1;
daniele	29:1a47b7151851	1683	} else {
daniele	29:1a47b7151851	1684	if (listen->filter_mode != PICO_IP_MULTICAST_EXCLUDE) {
daniele	29:1a47b7151851	1685	so_mcast_dbg("pico_socket_setoption: ERROR any-source multicast (exclude) on source-specific multicast (include)\n");
daniele	29:1a47b7151851	1686	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1687	return -1;
daniele	29:1a47b7151851	1688	}
daniele	29:1a47b7151851	1689	stest.addr = mreq->mcast_source_addr.addr;
daniele	29:1a47b7151851	1690	source = pico_tree_findKey(&listen->MCASTSources, &stest);
daniele	29:1a47b7151851	1691	if (!source) {
daniele	29:1a47b7151851	1692	so_mcast_dbg("pico_socket_setoption: ERROR address to unblock not in source list\n");
daniele	29:1a47b7151851	1693	pico_err = PICO_ERR_EADDRNOTAVAIL;
daniele	29:1a47b7151851	1694	return -1;
daniele	29:1a47b7151851	1695	} else {
daniele	29:1a47b7151851	1696	pico_tree_delete(&listen->MCASTSources, source);
daniele	29:1a47b7151851	1697	pico_free(source);
daniele	29:1a47b7151851	1698	}
daniele	29:1a47b7151851	1699	}
daniele	29:1a47b7151851	1700
daniele	29:1a47b7151851	1701	filter_mode = pico_socket_aggregate_mcastfilters(&mcast_link->address, &mreq->mcast_group_addr);
daniele	29:1a47b7151851	1702	if (filter_mode < 0)
daniele	29:1a47b7151851	1703	return -1;
daniele	29:1a47b7151851	1704
daniele	29:1a47b7151851	1705	return pico_ipv4_mcast_leave(&mreq->mcast_link_addr, &mreq->mcast_group_addr, 0, filter_mode, &MCASTFilter);
daniele	29:1a47b7151851	1706	}
daniele	29:1a47b7151851	1707	break;
daniele	29:1a47b7151851	1708
daniele	29:1a47b7151851	1709	case PICO_IP_BLOCK_SOURCE:
daniele	29:1a47b7151851	1710	/* EXCLUDE mode */
daniele	29:1a47b7151851	1711	if (s->proto->proto_number == PICO_PROTO_UDP) {
tass	48:40fc4462265c	1712	int filter_mode = 0;
daniele	29:1a47b7151851	1713	struct pico_ip_mreq_source *mreq = NULL;
daniele	29:1a47b7151851	1714	struct pico_mcast_listen *listen = NULL, ltest = {0};
daniele	29:1a47b7151851	1715	struct pico_ip4 *source = NULL, stest = {0};
daniele	29:1a47b7151851	1716	struct pico_ipv4_link *mcast_link = NULL;
daniele	29:1a47b7151851	1717
daniele	29:1a47b7151851	1718	if (!value) {
daniele	29:1a47b7151851	1719	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1720	return -1;
daniele	29:1a47b7151851	1721	}
daniele	29:1a47b7151851	1722	mreq = (struct pico_ip_mreq_source *) value;
daniele	29:1a47b7151851	1723	mcast_link = pico_socket_setoption_mcastargs_validation(NULL, mreq);
daniele	29:1a47b7151851	1724	if (!mcast_link) {
daniele	29:1a47b7151851	1725	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1726	return -1;
daniele	29:1a47b7151851	1727	}
daniele	29:1a47b7151851	1728	if (!mreq->mcast_link_addr.addr)
daniele	29:1a47b7151851	1729	mreq->mcast_link_addr.addr = mcast_link->address.addr;
daniele	29:1a47b7151851	1730
daniele	29:1a47b7151851	1731	if (!s->MCASTListen) { /* No RBTree allocated yet */
daniele	29:1a47b7151851	1732	so_mcast_dbg("pico_socket_setoption: ERROR PICO_IP_BLOCK_SOURCE before any PICO_IP_ADD_MEMBERSHIP\n");
daniele	29:1a47b7151851	1733	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1734	return -1;
daniele	29:1a47b7151851	1735	}
daniele	29:1a47b7151851	1736	ltest.mcast_link = mreq->mcast_link_addr;
daniele	29:1a47b7151851	1737	ltest.mcast_group = mreq->mcast_group_addr;
daniele	29:1a47b7151851	1738	listen = pico_tree_findKey(s->MCASTListen, &ltest);
daniele	29:1a47b7151851	1739	if (!listen) {
daniele	29:1a47b7151851	1740	so_mcast_dbg("pico_socket_setoption: ERROR PICO_IP_BLOCK_SOURCE before PICO_IP_ADD_MEMBERSHIP\n");
daniele	29:1a47b7151851	1741	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1742	return -1;
daniele	29:1a47b7151851	1743	} else {
daniele	29:1a47b7151851	1744	if (listen->filter_mode != PICO_IP_MULTICAST_EXCLUDE) {
daniele	29:1a47b7151851	1745	so_mcast_dbg("pico_socket_setoption: ERROR any-source multicast (exclude) on source-specific multicast (include)\n");
daniele	29:1a47b7151851	1746	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1747	return -1;
daniele	29:1a47b7151851	1748	}
daniele	29:1a47b7151851	1749	stest.addr = mreq->mcast_source_addr.addr;
daniele	29:1a47b7151851	1750	source = pico_tree_findKey(&listen->MCASTSources, &stest);
daniele	29:1a47b7151851	1751	if (source) {
daniele	29:1a47b7151851	1752	so_mcast_dbg("pico_socket_setoption: ERROR address to block already in source list\n");
daniele	29:1a47b7151851	1753	pico_err = PICO_ERR_EADDRNOTAVAIL;
daniele	29:1a47b7151851	1754	return -1;
daniele	29:1a47b7151851	1755	} else {
daniele	29:1a47b7151851	1756	source = pico_zalloc(sizeof(struct pico_ip4));
daniele	29:1a47b7151851	1757	if (!source) {
daniele	29:1a47b7151851	1758	pico_err = PICO_ERR_ENOMEM;
daniele	29:1a47b7151851	1759	return -1;
daniele	29:1a47b7151851	1760	}
daniele	29:1a47b7151851	1761	source->addr = mreq->mcast_source_addr.addr;
daniele	29:1a47b7151851	1762	pico_tree_insert(&listen->MCASTSources, source);
daniele	29:1a47b7151851	1763	}
daniele	29:1a47b7151851	1764	}
daniele	29:1a47b7151851	1765
daniele	29:1a47b7151851	1766	filter_mode = pico_socket_aggregate_mcastfilters(&mcast_link->address, &mreq->mcast_group_addr);
daniele	29:1a47b7151851	1767	if (filter_mode < 0)
daniele	29:1a47b7151851	1768	return -1;
daniele	29:1a47b7151851	1769
daniele	29:1a47b7151851	1770	return pico_ipv4_mcast_join(&mreq->mcast_link_addr, &mreq->mcast_group_addr, 0, filter_mode, &MCASTFilter);
daniele	29:1a47b7151851	1771	}
daniele	29:1a47b7151851	1772	break;
daniele	29:1a47b7151851	1773
daniele	29:1a47b7151851	1774	case PICO_IP_ADD_SOURCE_MEMBERSHIP:
daniele	29:1a47b7151851	1775	/* INCLUDE mode */
daniele	29:1a47b7151851	1776	if (s->proto->proto_number == PICO_PROTO_UDP) {
tass	48:40fc4462265c	1777	int filter_mode = 0, reference_count = 0;
daniele	29:1a47b7151851	1778	struct pico_ip_mreq_source *mreq = NULL;
daniele	29:1a47b7151851	1779	struct pico_mcast_listen *listen = NULL, ltest = {0};
daniele	29:1a47b7151851	1780	struct pico_ip4 *source = NULL, stest = {0};
daniele	29:1a47b7151851	1781	struct pico_ipv4_link *mcast_link = NULL;
daniele	29:1a47b7151851	1782
daniele	29:1a47b7151851	1783	if (!value) {
daniele	29:1a47b7151851	1784	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1785	return -1;
daniele	29:1a47b7151851	1786	}
daniele	29:1a47b7151851	1787	mreq = (struct pico_ip_mreq_source *) value;
daniele	29:1a47b7151851	1788	mcast_link = pico_socket_setoption_mcastargs_validation(NULL, mreq);
daniele	29:1a47b7151851	1789	if (!mcast_link) {
daniele	29:1a47b7151851	1790	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1791	return -1;
daniele	29:1a47b7151851	1792	}
daniele	29:1a47b7151851	1793	if (!mreq->mcast_link_addr.addr)
daniele	29:1a47b7151851	1794	mreq->mcast_link_addr.addr = mcast_link->address.addr;
daniele	29:1a47b7151851	1795
daniele	29:1a47b7151851	1796	if (!s->MCASTListen) { /* No RBTree allocated yet */
daniele	29:1a47b7151851	1797	s->MCASTListen = pico_zalloc(sizeof(struct pico_tree));
daniele	29:1a47b7151851	1798	if (!s->MCASTListen) {
daniele	29:1a47b7151851	1799	pico_err = PICO_ERR_ENOMEM;
daniele	29:1a47b7151851	1800	return -1;
daniele	29:1a47b7151851	1801	}
daniele	29:1a47b7151851	1802	s->MCASTListen->root = &LEAF;
daniele	29:1a47b7151851	1803	s->MCASTListen->compare = mcast_listen_cmp;
daniele	29:1a47b7151851	1804	}
daniele	29:1a47b7151851	1805	ltest.mcast_link = mreq->mcast_link_addr;
daniele	29:1a47b7151851	1806	ltest.mcast_group = mreq->mcast_group_addr;
daniele	29:1a47b7151851	1807	listen = pico_tree_findKey(s->MCASTListen, &ltest);
daniele	29:1a47b7151851	1808	if (listen) {
daniele	29:1a47b7151851	1809	if (listen->filter_mode != PICO_IP_MULTICAST_INCLUDE) {
daniele	29:1a47b7151851	1810	so_mcast_dbg("pico_socket_setoption: ERROR source-specific multicast (include) on any-source multicast (exclude)\n");
daniele	29:1a47b7151851	1811	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1812	return -1;
daniele	29:1a47b7151851	1813	}
daniele	29:1a47b7151851	1814	stest.addr = mreq->mcast_source_addr.addr;
daniele	29:1a47b7151851	1815	source = pico_tree_findKey(&listen->MCASTSources, &stest);
daniele	29:1a47b7151851	1816	if (source) {
daniele	29:1a47b7151851	1817	so_mcast_dbg("pico_socket_setoption: ERROR source address to allow already in source list\n");
daniele	29:1a47b7151851	1818	pico_err = PICO_ERR_EADDRNOTAVAIL;
daniele	29:1a47b7151851	1819	return -1;
daniele	29:1a47b7151851	1820	} else {
daniele	29:1a47b7151851	1821	source = pico_zalloc(sizeof(struct pico_ip4));
daniele	29:1a47b7151851	1822	if (!source) {
daniele	29:1a47b7151851	1823	pico_err = PICO_ERR_ENOMEM;
daniele	29:1a47b7151851	1824	return -1;
daniele	29:1a47b7151851	1825	}
daniele	29:1a47b7151851	1826	source->addr = mreq->mcast_source_addr.addr;
daniele	29:1a47b7151851	1827	pico_tree_insert(&listen->MCASTSources, source);
daniele	29:1a47b7151851	1828	}
daniele	29:1a47b7151851	1829	} else {
daniele	29:1a47b7151851	1830	listen = pico_zalloc(sizeof(struct pico_mcast_listen));
daniele	29:1a47b7151851	1831	if (!listen) {
daniele	29:1a47b7151851	1832	pico_err = PICO_ERR_ENOMEM;
daniele	29:1a47b7151851	1833	return -1;
daniele	29:1a47b7151851	1834	}
daniele	29:1a47b7151851	1835	listen->filter_mode = PICO_IP_MULTICAST_INCLUDE;
daniele	29:1a47b7151851	1836	listen->mcast_link = mreq->mcast_link_addr;
daniele	29:1a47b7151851	1837	listen->mcast_group = mreq->mcast_group_addr;
daniele	29:1a47b7151851	1838	listen->MCASTSources.root = &LEAF;
daniele	29:1a47b7151851	1839	listen->MCASTSources.compare = mcast_sources_cmp;
daniele	29:1a47b7151851	1840	source = pico_zalloc(sizeof(struct pico_ip4));
daniele	29:1a47b7151851	1841	if (!source) {
daniele	29:1a47b7151851	1842	pico_free(listen);
daniele	29:1a47b7151851	1843	pico_err = PICO_ERR_ENOMEM;
daniele	29:1a47b7151851	1844	return -1;
daniele	29:1a47b7151851	1845	}
daniele	29:1a47b7151851	1846	source->addr = mreq->mcast_source_addr.addr;
daniele	29:1a47b7151851	1847	pico_tree_insert(&listen->MCASTSources, source);
daniele	29:1a47b7151851	1848	pico_tree_insert(s->MCASTListen, listen);
daniele	29:1a47b7151851	1849	reference_count = 1;
daniele	29:1a47b7151851	1850	}
daniele	29:1a47b7151851	1851
daniele	29:1a47b7151851	1852	pico_tree_insert(&MCASTSockets, s);
daniele	29:1a47b7151851	1853	filter_mode = pico_socket_aggregate_mcastfilters(&mcast_link->address, &mreq->mcast_group_addr);
daniele	29:1a47b7151851	1854	if (filter_mode < 0)
daniele	29:1a47b7151851	1855	return -1;
daniele	29:1a47b7151851	1856
daniele	29:1a47b7151851	1857	return pico_ipv4_mcast_join(&mreq->mcast_link_addr, &mreq->mcast_group_addr, reference_count, filter_mode, &MCASTFilter);
daniele	29:1a47b7151851	1858	}
daniele	29:1a47b7151851	1859	break;
daniele	29:1a47b7151851	1860
daniele	29:1a47b7151851	1861	case PICO_IP_DROP_SOURCE_MEMBERSHIP:
daniele	29:1a47b7151851	1862	/* INCLUDE mode */
daniele	29:1a47b7151851	1863	if (s->proto->proto_number == PICO_PROTO_UDP) {
tass	48:40fc4462265c	1864	int filter_mode = 0, reference_count = 0;
daniele	29:1a47b7151851	1865	struct pico_ip_mreq_source *mreq = NULL;
daniele	29:1a47b7151851	1866	struct pico_mcast_listen *listen = NULL, ltest = {0};
daniele	29:1a47b7151851	1867	struct pico_ip4 *source = NULL, stest = {0};
daniele	29:1a47b7151851	1868	struct pico_ipv4_link *mcast_link = NULL;
daniele	29:1a47b7151851	1869
daniele	29:1a47b7151851	1870	if (!value) {
daniele	29:1a47b7151851	1871	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1872	return -1;
daniele	29:1a47b7151851	1873	}
daniele	29:1a47b7151851	1874	mreq = (struct pico_ip_mreq_source *) value;
daniele	29:1a47b7151851	1875	mcast_link = pico_socket_setoption_mcastargs_validation(NULL, mreq);
daniele	29:1a47b7151851	1876	if (!mcast_link) {
daniele	29:1a47b7151851	1877	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1878	return -1;
daniele	29:1a47b7151851	1879	}
daniele	29:1a47b7151851	1880	if (!mreq->mcast_link_addr.addr)
daniele	29:1a47b7151851	1881	mreq->mcast_link_addr.addr = mcast_link->address.addr;
daniele	29:1a47b7151851	1882
daniele	29:1a47b7151851	1883	if (!s->MCASTListen) { /* No RBTree allocated yet */
daniele	29:1a47b7151851	1884	so_mcast_dbg("pico_socket_setoption: ERROR PICO_IP_DROP_SOURCE_MEMBERSHIP before any PICO_IP_ADD_SOURCE_MEMBERSHIP\n");
daniele	29:1a47b7151851	1885	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1886	return -1;
daniele	29:1a47b7151851	1887	}
daniele	29:1a47b7151851	1888	ltest.mcast_link = mreq->mcast_link_addr;
daniele	29:1a47b7151851	1889	ltest.mcast_group = mreq->mcast_group_addr;
daniele	29:1a47b7151851	1890	listen = pico_tree_findKey(s->MCASTListen, &ltest);
daniele	29:1a47b7151851	1891	if (!listen) {
daniele	29:1a47b7151851	1892	so_mcast_dbg("pico_socket_setoption: ERROR PICO_IP_DROP_SOURCE_MEMBERSHIP before PICO_IP_ADD_SOURCE_MEMBERSHIP\n");
daniele	29:1a47b7151851	1893	pico_err = PICO_ERR_EADDRNOTAVAIL;
daniele	29:1a47b7151851	1894	return -1;
daniele	29:1a47b7151851	1895	} else {
daniele	29:1a47b7151851	1896	if (listen->filter_mode != PICO_IP_MULTICAST_INCLUDE) {
daniele	29:1a47b7151851	1897	so_mcast_dbg("pico_socket_setoption: ERROR source-specific multicast (include) on any-source multicast (exclude)\n");
daniele	29:1a47b7151851	1898	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1899	return -1;
daniele	29:1a47b7151851	1900	}
daniele	29:1a47b7151851	1901	stest.addr = mreq->mcast_source_addr.addr;
daniele	29:1a47b7151851	1902	source = pico_tree_findKey(&listen->MCASTSources, &stest);
daniele	29:1a47b7151851	1903	if (!source) {
daniele	29:1a47b7151851	1904	so_mcast_dbg("pico_socket_setoption: ERROR address to drop not in source list\n");
daniele	29:1a47b7151851	1905	pico_err = PICO_ERR_EADDRNOTAVAIL;
daniele	29:1a47b7151851	1906	return -1;
daniele	29:1a47b7151851	1907	} else {
daniele	29:1a47b7151851	1908	pico_tree_delete(&listen->MCASTSources, source);
daniele	29:1a47b7151851	1909	pico_free(source);
daniele	29:1a47b7151851	1910	if (pico_tree_empty(&listen->MCASTSources)) { /* 1 if empty, 0 otherwise */
daniele	29:1a47b7151851	1911	reference_count = 1;
daniele	29:1a47b7151851	1912	pico_tree_delete(s->MCASTListen, listen);
daniele	29:1a47b7151851	1913	pico_free(listen);
daniele	29:1a47b7151851	1914	if (pico_tree_empty(s->MCASTListen)) {
daniele	29:1a47b7151851	1915	pico_free(s->MCASTListen);
daniele	29:1a47b7151851	1916	s->MCASTListen = NULL;
daniele	29:1a47b7151851	1917	pico_tree_delete(&MCASTSockets, s);
daniele	29:1a47b7151851	1918	}
daniele	29:1a47b7151851	1919	}
daniele	29:1a47b7151851	1920	}
daniele	29:1a47b7151851	1921	}
daniele	29:1a47b7151851	1922
daniele	29:1a47b7151851	1923	filter_mode = pico_socket_aggregate_mcastfilters(&mcast_link->address, &mreq->mcast_group_addr);
daniele	29:1a47b7151851	1924	if (filter_mode < 0)
daniele	29:1a47b7151851	1925	return -1;
daniele	29:1a47b7151851	1926
daniele	29:1a47b7151851	1927	return pico_ipv4_mcast_leave(&mreq->mcast_link_addr, &mreq->mcast_group_addr, reference_count, filter_mode, &MCASTFilter);
daniele	29:1a47b7151851	1928	}
daniele	29:1a47b7151851	1929	break;
daniele	29:1a47b7151851	1930	#endif /* PICO_SUPPORT_MCAST */
daniele	29:1a47b7151851	1931
daniele	29:1a47b7151851	1932	default:
daniele	29:1a47b7151851	1933	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1934	return -1;
daniele	29:1a47b7151851	1935	}
daniele	29:1a47b7151851	1936
daniele	29:1a47b7151851	1937	if (pico_err != PICO_ERR_NOERR)
daniele	29:1a47b7151851	1938	return -1;
daniele	29:1a47b7151851	1939	else
daniele	29:1a47b7151851	1940	return 0;
daniele	29:1a47b7151851	1941	}
daniele	29:1a47b7151851	1942
daniele	29:1a47b7151851	1943	#define PICO_SOCKET_GETOPT(socket,index) ((socket->opt_flags & (1 << index)) != 0)
daniele	29:1a47b7151851	1944
daniele	29:1a47b7151851	1945	int pico_socket_getoption(struct pico_socket s, int option, void value)
daniele	29:1a47b7151851	1946	{
daniele	29:1a47b7151851	1947	if (!s \|\| !value) {
daniele	29:1a47b7151851	1948	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1949	return -1;
daniele	29:1a47b7151851	1950	}
daniele	29:1a47b7151851	1951
daniele	29:1a47b7151851	1952	switch (option)
daniele	29:1a47b7151851	1953	{
daniele	29:1a47b7151851	1954	#ifdef PICO_SUPPORT_TCP
daniele	29:1a47b7151851	1955	case PICO_TCP_NODELAY:
daniele	29:1a47b7151851	1956	if (s->proto->proto_number == PICO_PROTO_TCP)
daniele	29:1a47b7151851	1957	/* state of the NODELAY option */
daniele	29:1a47b7151851	1958	(int )value = PICO_SOCKET_GETOPT(s,PICO_SOCKET_OPT_TCPNODELAY);
daniele	29:1a47b7151851	1959	else
daniele	29:1a47b7151851	1960	(int )value = 0;
daniele	29:1a47b7151851	1961	break;
daniele	29:1a47b7151851	1962	#endif
daniele	29:1a47b7151851	1963
daniele	29:1a47b7151851	1964	#ifdef PICO_SUPPORT_MCAST
daniele	29:1a47b7151851	1965	case PICO_IP_MULTICAST_IF:
daniele	29:1a47b7151851	1966	pico_err = PICO_ERR_EOPNOTSUPP;
daniele	29:1a47b7151851	1967	return -1;
daniele	29:1a47b7151851	1968	break;
daniele	29:1a47b7151851	1969
daniele	29:1a47b7151851	1970	case PICO_IP_MULTICAST_TTL:
daniele	29:1a47b7151851	1971	if (s->proto->proto_number == PICO_PROTO_UDP) {
daniele	29:1a47b7151851	1972	pico_udp_get_mc_ttl(s, (uint8_t *) value);
daniele	29:1a47b7151851	1973	} else {
daniele	29:1a47b7151851	1974	(uint8_t )value = 0;
daniele	29:1a47b7151851	1975	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1976	return -1;
daniele	29:1a47b7151851	1977	}
daniele	29:1a47b7151851	1978	break;
daniele	29:1a47b7151851	1979
daniele	29:1a47b7151851	1980	case PICO_IP_MULTICAST_LOOP:
daniele	29:1a47b7151851	1981	if (s->proto->proto_number == PICO_PROTO_UDP) {
daniele	29:1a47b7151851	1982	(uint8_t )value = PICO_SOCKET_GETOPT(s,PICO_SOCKET_OPT_MULTICAST_LOOP);
daniele	29:1a47b7151851	1983	} else {
daniele	29:1a47b7151851	1984	(uint8_t )value = 0;
daniele	29:1a47b7151851	1985	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1986	return -1;
daniele	29:1a47b7151851	1987	}
daniele	29:1a47b7151851	1988	break;
daniele	29:1a47b7151851	1989	#endif /* PICO_SUPPORT_MCAST */
daniele	29:1a47b7151851	1990
daniele	29:1a47b7151851	1991	default:
daniele	29:1a47b7151851	1992	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1993	return -1;
daniele	29:1a47b7151851	1994	}
daniele	29:1a47b7151851	1995
daniele	29:1a47b7151851	1996	return 0;
daniele	29:1a47b7151851	1997	}
daniele	29:1a47b7151851	1998
daniele	29:1a47b7151851	1999
daniele	29:1a47b7151851	2000	int pico_socket_shutdown(struct pico_socket *s, int mode)
daniele	29:1a47b7151851	2001	{
daniele	29:1a47b7151851	2002	if (!s) {
daniele	29:1a47b7151851	2003	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	2004	return -1;
daniele	29:1a47b7151851	2005	} else {
daniele	29:1a47b7151851	2006	/* check if exists in tree */
daniele	29:1a47b7151851	2007	/* See task #178 */
daniele	29:1a47b7151851	2008	if (pico_check_socket(s) != 0) {
daniele	29:1a47b7151851	2009	pico_free(s); /* close socket after bind or connect failed */
daniele	29:1a47b7151851	2010	return 0;
daniele	29:1a47b7151851	2011	}
daniele	29:1a47b7151851	2012	}
daniele	29:1a47b7151851	2013
daniele	29:1a47b7151851	2014	#ifdef PICO_SUPPORT_UDP
daniele	29:1a47b7151851	2015	if (PROTO(s) == PICO_PROTO_UDP) {
daniele	29:1a47b7151851	2016	if (mode & PICO_SHUT_RDWR)
daniele	29:1a47b7151851	2017	pico_socket_alter_state(s, PICO_SOCKET_STATE_CLOSED, PICO_SOCKET_STATE_CLOSING \|PICO_SOCKET_STATE_BOUND \| PICO_SOCKET_STATE_CONNECTED, 0);
daniele	29:1a47b7151851	2018	else if (mode & PICO_SHUT_RD)
daniele	29:1a47b7151851	2019	pico_socket_alter_state(s, PICO_SOCKET_STATE_BOUND, 0, 0);
daniele	29:1a47b7151851	2020	}
daniele	29:1a47b7151851	2021	#endif
daniele	29:1a47b7151851	2022	#ifdef PICO_SUPPORT_TCP
daniele	29:1a47b7151851	2023	if (PROTO(s) == PICO_PROTO_TCP) {
tass	48:40fc4462265c	2024	if(mode & PICO_SHUT_RDWR)
tass	48:40fc4462265c	2025	pico_socket_alter_state(s, PICO_SOCKET_STATE_SHUT_LOCAL \| PICO_SOCKET_STATE_SHUT_REMOTE, 0, 0);
tass	48:40fc4462265c	2026	else if (mode & PICO_SHUT_WR)
daniele	29:1a47b7151851	2027	pico_socket_alter_state(s, PICO_SOCKET_STATE_SHUT_LOCAL, 0, 0);
daniele	29:1a47b7151851	2028	else if (mode & PICO_SHUT_RD)
daniele	29:1a47b7151851	2029	pico_socket_alter_state(s, PICO_SOCKET_STATE_SHUT_REMOTE, 0, 0);
daniele	29:1a47b7151851	2030
daniele	29:1a47b7151851	2031	}
daniele	29:1a47b7151851	2032	#endif
daniele	29:1a47b7151851	2033	return 0;
daniele	29:1a47b7151851	2034	}
daniele	29:1a47b7151851	2035
daniele	29:1a47b7151851	2036	int pico_socket_close(struct pico_socket *s)
daniele	29:1a47b7151851	2037	{
daniele	29:1a47b7151851	2038	return pico_socket_shutdown(s, PICO_SHUT_RDWR);
daniele	29:1a47b7151851	2039	}
daniele	29:1a47b7151851	2040
daniele	29:1a47b7151851	2041	#ifdef PICO_SUPPORT_CRC
daniele	29:1a47b7151851	2042	static inline int pico_transport_crc_check(struct pico_frame *f)
daniele	29:1a47b7151851	2043	{
daniele	29:1a47b7151851	2044	struct pico_ipv4_hdr net_hdr = (struct pico_ipv4_hdr ) f->net_hdr;
daniele	29:1a47b7151851	2045	struct pico_udp_hdr *udp_hdr = NULL;
daniele	29:1a47b7151851	2046	uint16_t checksum_invalid = 1;
daniele	29:1a47b7151851	2047
daniele	29:1a47b7151851	2048	switch (net_hdr->proto)
daniele	29:1a47b7151851	2049	{
daniele	29:1a47b7151851	2050	case PICO_PROTO_TCP:
daniele	29:1a47b7151851	2051	checksum_invalid = short_be(pico_tcp_checksum_ipv4(f));
daniele	29:1a47b7151851	2052	//dbg("TCP CRC validation == %u\n", checksum_invalid);
daniele	29:1a47b7151851	2053	if (checksum_invalid) {
daniele	29:1a47b7151851	2054	//dbg("TCP CRC: validation failed!\n");
daniele	29:1a47b7151851	2055	pico_frame_discard(f);
daniele	29:1a47b7151851	2056	return 0;
daniele	29:1a47b7151851	2057	}
daniele	29:1a47b7151851	2058	break;
daniele	29:1a47b7151851	2059
daniele	29:1a47b7151851	2060	case PICO_PROTO_UDP:
daniele	29:1a47b7151851	2061	udp_hdr = (struct pico_udp_hdr *) f->transport_hdr;
daniele	29:1a47b7151851	2062	if (short_be(udp_hdr->crc)) {
daniele	29:1a47b7151851	2063	checksum_invalid = short_be(pico_udp_checksum_ipv4(f));
daniele	29:1a47b7151851	2064	//dbg("UDP CRC validation == %u\n", checksum_invalid);
daniele	29:1a47b7151851	2065	if (checksum_invalid) {
daniele	29:1a47b7151851	2066	//dbg("UDP CRC: validation failed!\n");
daniele	29:1a47b7151851	2067	pico_frame_discard(f);
daniele	29:1a47b7151851	2068	return 0;
daniele	29:1a47b7151851	2069	}
daniele	29:1a47b7151851	2070	}
daniele	29:1a47b7151851	2071	break;
daniele	29:1a47b7151851	2072
daniele	29:1a47b7151851	2073	default:
daniele	29:1a47b7151851	2074	// Do nothing
daniele	29:1a47b7151851	2075	break;
daniele	29:1a47b7151851	2076	}
daniele	29:1a47b7151851	2077	return 1;
daniele	29:1a47b7151851	2078	}
daniele	29:1a47b7151851	2079	#else
daniele	29:1a47b7151851	2080	static inline int pico_transport_crc_check(struct pico_frame *f)
daniele	29:1a47b7151851	2081	{
daniele	29:1a47b7151851	2082	return 1;
daniele	29:1a47b7151851	2083	}
daniele	29:1a47b7151851	2084	#endif /* PICO_SUPPORT_CRC */
daniele	29:1a47b7151851	2085
daniele	29:1a47b7151851	2086	int pico_transport_process_in(struct pico_protocol self, struct pico_frame f)
daniele	29:1a47b7151851	2087	{
daniele	29:1a47b7151851	2088	struct pico_trans hdr = (struct pico_trans ) f->transport_hdr;
daniele	29:1a47b7151851	2089	int ret = 0;
daniele	29:1a47b7151851	2090
daniele	29:1a47b7151851	2091	if (!hdr) {
daniele	29:1a47b7151851	2092	pico_err = PICO_ERR_EFAULT;
daniele	29:1a47b7151851	2093	return -1;
daniele	29:1a47b7151851	2094	}
daniele	29:1a47b7151851	2095
daniele	29:1a47b7151851	2096	ret = pico_transport_crc_check(f);
daniele	29:1a47b7151851	2097	if (ret < 1)
daniele	29:1a47b7151851	2098	return ret;
daniele	29:1a47b7151851	2099	else
daniele	29:1a47b7151851	2100	ret = 0;
daniele	29:1a47b7151851	2101
daniele	29:1a47b7151851	2102	if ((hdr) && (pico_socket_deliver(self, f, hdr->dport) == 0))
daniele	29:1a47b7151851	2103	return ret;
daniele	29:1a47b7151851	2104
daniele	29:1a47b7151851	2105	if (!IS_BCAST(f)) {
daniele	29:1a47b7151851	2106	dbg("Socket not found... \n");
daniele	29:1a47b7151851	2107	pico_notify_socket_unreachable(f);
daniele	29:1a47b7151851	2108	#ifdef PICO_SUPPORT_TCP
daniele	29:1a47b7151851	2109	/* if tcp protocol send RST segment */
daniele	29:1a47b7151851	2110	//if (self->proto_number == PICO_PROTO_TCP)
daniele	29:1a47b7151851	2111	// pico_tcp_reply_rst(f);
daniele	29:1a47b7151851	2112	#endif
daniele	29:1a47b7151851	2113	ret = -1;
daniele	29:1a47b7151851	2114	pico_err = PICO_ERR_ENOENT;
daniele	29:1a47b7151851	2115	}
daniele	29:1a47b7151851	2116	pico_frame_discard(f);
daniele	29:1a47b7151851	2117	return ret;
daniele	29:1a47b7151851	2118	}
daniele	29:1a47b7151851	2119
daniele	29:1a47b7151851	2120	#define SL_LOOP_MIN 1
daniele	29:1a47b7151851	2121
daniele	29:1a47b7151851	2122
daniele	29:1a47b7151851	2123	int pico_sockets_loop(int loop_score)
daniele	29:1a47b7151851	2124	{
daniele	29:1a47b7151851	2125	static struct pico_tree_node index_udp, index_tcp;
daniele	29:1a47b7151851	2126
daniele	29:1a47b7151851	2127	struct pico_sockport *start;
daniele	29:1a47b7151851	2128	struct pico_socket *s;
daniele	29:1a47b7151851	2129
daniele	29:1a47b7151851	2130	#ifdef PICO_SUPPORT_UDP
daniele	29:1a47b7151851	2131	struct pico_frame *f;
daniele	29:1a47b7151851	2132
daniele	29:1a47b7151851	2133	if (sp_udp == NULL)
daniele	29:1a47b7151851	2134	{
daniele	29:1a47b7151851	2135	index_udp = pico_tree_firstNode(UDPTable.root);
daniele	29:1a47b7151851	2136	sp_udp = index_udp->keyValue;
daniele	29:1a47b7151851	2137	}
daniele	29:1a47b7151851	2138
daniele	29:1a47b7151851	2139	/* init start node */
daniele	29:1a47b7151851	2140	start = sp_udp;
daniele	29:1a47b7151851	2141
daniele	29:1a47b7151851	2142	/* round-robin all transport protocols, break if traversed all protocols */
daniele	29:1a47b7151851	2143	while (loop_score > SL_LOOP_MIN && sp_udp != NULL) {
daniele	29:1a47b7151851	2144	struct pico_tree_node * index;
daniele	29:1a47b7151851	2145
daniele	29:1a47b7151851	2146	pico_tree_foreach(index,&sp_udp->socks){
daniele	29:1a47b7151851	2147	s = index->keyValue;
daniele	29:1a47b7151851	2148	f = pico_dequeue(&s->q_out);
daniele	29:1a47b7151851	2149	while (f && (loop_score > 0)) {
daniele	29:1a47b7151851	2150	pico_proto_udp.push(&pico_proto_udp, f);
daniele	29:1a47b7151851	2151	loop_score -= 1;
daniele	29:1a47b7151851	2152	f = pico_dequeue(&s->q_out);
daniele	29:1a47b7151851	2153	}
daniele	29:1a47b7151851	2154	}
daniele	29:1a47b7151851	2155
daniele	29:1a47b7151851	2156	index_udp = pico_tree_next(index_udp);
daniele	29:1a47b7151851	2157	sp_udp = index_udp->keyValue;
daniele	29:1a47b7151851	2158
daniele	29:1a47b7151851	2159	if (sp_udp == NULL)
daniele	29:1a47b7151851	2160	{
daniele	29:1a47b7151851	2161	index_udp = pico_tree_firstNode(UDPTable.root);
daniele	29:1a47b7151851	2162	sp_udp = index_udp->keyValue;
daniele	29:1a47b7151851	2163	}
daniele	29:1a47b7151851	2164	if (sp_udp == start)
daniele	29:1a47b7151851	2165	break;
daniele	29:1a47b7151851	2166	}
daniele	29:1a47b7151851	2167	#endif
daniele	29:1a47b7151851	2168
daniele	29:1a47b7151851	2169	#ifdef PICO_SUPPORT_TCP
daniele	29:1a47b7151851	2170	if (sp_tcp == NULL)
daniele	29:1a47b7151851	2171	{
daniele	29:1a47b7151851	2172	index_tcp = pico_tree_firstNode(TCPTable.root);
daniele	29:1a47b7151851	2173	sp_tcp = index_tcp->keyValue;
daniele	29:1a47b7151851	2174	}
daniele	29:1a47b7151851	2175
daniele	29:1a47b7151851	2176	/* init start node */
daniele	29:1a47b7151851	2177	start = sp_tcp;
daniele	29:1a47b7151851	2178
daniele	29:1a47b7151851	2179	while (loop_score > SL_LOOP_MIN && sp_tcp != NULL) {
tass	48:40fc4462265c	2180	struct pico_tree_node * index = NULL;
daniele	29:1a47b7151851	2181	pico_tree_foreach(index, &sp_tcp->socks){
daniele	29:1a47b7151851	2182	s = index->keyValue;
daniele	29:1a47b7151851	2183	loop_score = pico_tcp_output(s, loop_score);
daniele	29:1a47b7151851	2184	if ((s->ev_pending) && s->wakeup) {
daniele	29:1a47b7151851	2185	s->wakeup(s->ev_pending, s);
daniele	29:1a47b7151851	2186	}
daniele	29:1a47b7151851	2187	if (loop_score <= 0) {
daniele	29:1a47b7151851	2188	loop_score = 0;
daniele	29:1a47b7151851	2189	break;
daniele	29:1a47b7151851	2190	}
daniele	29:1a47b7151851	2191	}
daniele	29:1a47b7151851	2192
daniele	29:1a47b7151851	2193	/* check if RB_FOREACH ended, if not, break to keep the cur sp_tcp */
tass	48:40fc4462265c	2194	if (index && index->keyValue)
daniele	29:1a47b7151851	2195	break;
daniele	29:1a47b7151851	2196
daniele	29:1a47b7151851	2197	index_tcp = pico_tree_next(index_tcp);
daniele	29:1a47b7151851	2198	sp_tcp = index_tcp->keyValue;
daniele	29:1a47b7151851	2199
daniele	29:1a47b7151851	2200	if (sp_tcp == NULL)
daniele	29:1a47b7151851	2201	{
daniele	29:1a47b7151851	2202	index_tcp = pico_tree_firstNode(TCPTable.root);
daniele	29:1a47b7151851	2203	sp_tcp = index_tcp->keyValue;
daniele	29:1a47b7151851	2204	}
daniele	29:1a47b7151851	2205	if (sp_tcp == start)
daniele	29:1a47b7151851	2206	break;
daniele	29:1a47b7151851	2207	}
daniele	29:1a47b7151851	2208	#endif
daniele	29:1a47b7151851	2209
daniele	29:1a47b7151851	2210	return loop_score;
daniele	29:1a47b7151851	2211	}
daniele	29:1a47b7151851	2212
daniele	29:1a47b7151851	2213
daniele	29:1a47b7151851	2214	struct pico_frame pico_socket_frame_alloc(struct pico_socket s, int len)
daniele	29:1a47b7151851	2215	{
daniele	29:1a47b7151851	2216	struct pico_frame *f = NULL;
daniele	29:1a47b7151851	2217
daniele	29:1a47b7151851	2218	#ifdef PICO_SUPPORT_IPV6
daniele	29:1a47b7151851	2219	if (IS_SOCK_IPV6(s))
daniele	29:1a47b7151851	2220	f = pico_proto_ipv6.alloc(&pico_proto_ipv6, len);
daniele	29:1a47b7151851	2221	#endif
daniele	29:1a47b7151851	2222
daniele	29:1a47b7151851	2223	#ifdef PICO_SUPPORT_IPV4
daniele	29:1a47b7151851	2224	if (IS_SOCK_IPV4(s))
daniele	29:1a47b7151851	2225	f = pico_proto_ipv4.alloc(&pico_proto_ipv4, len);
daniele	29:1a47b7151851	2226	#endif
daniele	29:1a47b7151851	2227	if (!f) {
daniele	29:1a47b7151851	2228	pico_err = PICO_ERR_ENOMEM;
daniele	29:1a47b7151851	2229	return f;
daniele	29:1a47b7151851	2230	}
daniele	29:1a47b7151851	2231	f->payload = f->transport_hdr;
daniele	29:1a47b7151851	2232	f->payload_len = len;
daniele	29:1a47b7151851	2233	f->sock = s;
daniele	29:1a47b7151851	2234	return f;
daniele	29:1a47b7151851	2235	}
daniele	29:1a47b7151851	2236
daniele	29:1a47b7151851	2237	int pico_transport_error(struct pico_frame *f, uint8_t proto, int code)
daniele	29:1a47b7151851	2238	{
daniele	29:1a47b7151851	2239	int ret = -1;
daniele	29:1a47b7151851	2240	struct pico_trans trans = (struct pico_trans) f->transport_hdr;
daniele	29:1a47b7151851	2241	struct pico_sockport *port = NULL;
daniele	29:1a47b7151851	2242	struct pico_socket *s = NULL;
daniele	29:1a47b7151851	2243	switch (proto) {
daniele	29:1a47b7151851	2244
daniele	29:1a47b7151851	2245
daniele	29:1a47b7151851	2246	#ifdef PICO_SUPPORT_UDP
daniele	29:1a47b7151851	2247	case PICO_PROTO_UDP:
daniele	29:1a47b7151851	2248	port = pico_get_sockport(proto, trans->sport);
daniele	29:1a47b7151851	2249	break;
daniele	29:1a47b7151851	2250	#endif
daniele	29:1a47b7151851	2251
daniele	29:1a47b7151851	2252	#ifdef PICO_SUPPORT_TCP
daniele	29:1a47b7151851	2253	case PICO_PROTO_TCP:
daniele	29:1a47b7151851	2254	port = pico_get_sockport(proto, trans->sport);
daniele	29:1a47b7151851	2255	break;
daniele	29:1a47b7151851	2256	#endif
daniele	29:1a47b7151851	2257
daniele	29:1a47b7151851	2258	default:
daniele	29:1a47b7151851	2259	/* Protocol not available */
daniele	29:1a47b7151851	2260	ret = -1;
daniele	29:1a47b7151851	2261	}
daniele	29:1a47b7151851	2262	if (port) {
daniele	29:1a47b7151851	2263	struct pico_tree_node * index;
daniele	29:1a47b7151851	2264	ret = 0;
daniele	29:1a47b7151851	2265
daniele	29:1a47b7151851	2266	pico_tree_foreach(index,&port->socks) {
daniele	29:1a47b7151851	2267	s = index->keyValue;
daniele	29:1a47b7151851	2268	if (trans->dport == s->remote_port) {
daniele	29:1a47b7151851	2269	if (s->wakeup) {
daniele	29:1a47b7151851	2270	//dbg("SOCKET ERROR FROM ICMP NOTIFICATION. (icmp code= %d)\n\n", code);
daniele	29:1a47b7151851	2271	switch(code) {
daniele	29:1a47b7151851	2272	case PICO_ICMP_UNREACH_PROTOCOL:
daniele	29:1a47b7151851	2273	pico_err = PICO_ERR_EPROTO;
daniele	29:1a47b7151851	2274	break;
daniele	29:1a47b7151851	2275
daniele	29:1a47b7151851	2276	case PICO_ICMP_UNREACH_PORT:
daniele	29:1a47b7151851	2277	pico_err = PICO_ERR_ECONNREFUSED;
daniele	29:1a47b7151851	2278	break;
daniele	29:1a47b7151851	2279
daniele	29:1a47b7151851	2280	case PICO_ICMP_UNREACH_NET:
daniele	29:1a47b7151851	2281	case PICO_ICMP_UNREACH_NET_PROHIB:
daniele	29:1a47b7151851	2282	case PICO_ICMP_UNREACH_NET_UNKNOWN:
daniele	29:1a47b7151851	2283	pico_err = PICO_ERR_ENETUNREACH;
daniele	29:1a47b7151851	2284	break;
daniele	29:1a47b7151851	2285
daniele	29:1a47b7151851	2286	default:
daniele	29:1a47b7151851	2287	pico_err = PICO_ERR_EHOSTUNREACH;
daniele	29:1a47b7151851	2288	}
daniele	29:1a47b7151851	2289	s->wakeup(PICO_SOCK_EV_ERR, s);
daniele	29:1a47b7151851	2290	}
daniele	29:1a47b7151851	2291	break;
daniele	29:1a47b7151851	2292	}
daniele	29:1a47b7151851	2293	}
daniele	29:1a47b7151851	2294	}
daniele	29:1a47b7151851	2295	pico_frame_discard(f);
daniele	29:1a47b7151851	2296	return ret;
daniele	29:1a47b7151851	2297	}
daniele	29:1a47b7151851	2298	#endif
daniele	29:1a47b7151851	2299	#endif