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@51:ab4529a384a6, 2013-09-02 (annotated)

Committer:: tass
Date:: Mon Sep 02 08:02:21 2013 +0000
Revision:: 51:ab4529a384a6
Parent:: 49:4b404dd2c97a
Child:: 63:97f481e33cb2
Child:: 64:0225b609335e

Updated from masterbranch

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
tass	51:ab4529a384a6	464	if (pico_ipv4_nat_find(port, NULL, 0, proto)) {
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	51:ab4529a384a6	521	struct pico_socket * sock = NULL;
tass	51:ab4529a384a6	522	struct pico_tree_node *index = NULL;
tass	51:ab4529a384a6	523	struct pico_sockport *sp = NULL;
tass	49:4b404dd2c97a	524
tass	51:ab4529a384a6	525	sp = pico_get_sockport(PICO_PROTO_TCP,local);
tass	51:ab4529a384a6	526	if(sp)
tass	51:ab4529a384a6	527	{
tass	51:ab4529a384a6	528	pico_tree_foreach(index,&sp->socks)
tass	51:ab4529a384a6	529	{
tass	51:ab4529a384a6	530	if( ((struct pico_socket *)index->keyValue)->remote_port == remote)
tass	51:ab4529a384a6	531	{
tass	51:ab4529a384a6	532	sock = (struct pico_socket *)index->keyValue;
tass	51:ab4529a384a6	533	break;
tass	51:ab4529a384a6	534	}
tass	51:ab4529a384a6	535	}
tass	51:ab4529a384a6	536	}
tass	49:4b404dd2c97a	537
tass	51:ab4529a384a6	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;
tass	51:ab4529a384a6	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;
tass	51:ab4529a384a6	590	IGNORE_PARAMETER(now);
daniele	29:1a47b7151851	591	pico_free(s);
daniele	29:1a47b7151851	592	}
daniele	29:1a47b7151851	593
daniele	29:1a47b7151851	594	int pico_socket_del(struct pico_socket *s)
daniele	29:1a47b7151851	595	{
daniele	29:1a47b7151851	596	struct pico_sockport *sp = pico_get_sockport(PROTO(s), s->local_port);
daniele	29:1a47b7151851	597
daniele	29:1a47b7151851	598	if (!sp) {
daniele	29:1a47b7151851	599	pico_err = PICO_ERR_ENXIO;
daniele	29:1a47b7151851	600	return -1;
daniele	29:1a47b7151851	601	}
daniele	29:1a47b7151851	602
daniele	29:1a47b7151851	603	LOCK(Mutex);
daniele	29:1a47b7151851	604	pico_tree_delete(&sp->socks,s);
daniele	29:1a47b7151851	605	s->net = NULL;
daniele	29:1a47b7151851	606	if(pico_tree_empty(&sp->socks)){
daniele	29:1a47b7151851	607	if (PROTO(s) == PICO_PROTO_UDP)
daniele	29:1a47b7151851	608	{
daniele	29:1a47b7151851	609	pico_tree_delete(&UDPTable,sp);
daniele	29:1a47b7151851	610	}
daniele	29:1a47b7151851	611	else if (PROTO(s) == PICO_PROTO_TCP)
daniele	29:1a47b7151851	612	{
daniele	29:1a47b7151851	613	pico_tree_delete(&TCPTable,sp);
daniele	29:1a47b7151851	614	}
daniele	29:1a47b7151851	615
daniele	29:1a47b7151851	616	if(sp_tcp == sp) sp_tcp = NULL;
daniele	29:1a47b7151851	617
daniele	29:1a47b7151851	618	if(sp_udp == sp) sp_udp = NULL;
daniele	29:1a47b7151851	619
daniele	29:1a47b7151851	620	pico_free(sp);
daniele	29:1a47b7151851	621
daniele	29:1a47b7151851	622	}
daniele	29:1a47b7151851	623
daniele	29:1a47b7151851	624	#ifdef PICO_SUPPORT_MCAST
daniele	29:1a47b7151851	625	do {
tass	48:40fc4462265c	626	int filter_mode;
daniele	29:1a47b7151851	627	struct pico_tree_node index = NULL, _tmp = NULL, index2 = NULL, _tmp2 = NULL;
daniele	29:1a47b7151851	628	struct pico_mcast_listen *listen = NULL;
daniele	29:1a47b7151851	629	struct pico_ip4 *source = NULL;
daniele	29:1a47b7151851	630	if (s->MCASTListen) {
daniele	29:1a47b7151851	631	pico_tree_delete(&MCASTSockets, s);
daniele	29:1a47b7151851	632	pico_tree_foreach_safe(index, s->MCASTListen, _tmp)
daniele	29:1a47b7151851	633	{
daniele	29:1a47b7151851	634	listen = index->keyValue;
daniele	29:1a47b7151851	635	pico_tree_foreach_safe(index2, &listen->MCASTSources, _tmp2)
daniele	29:1a47b7151851	636	{
daniele	29:1a47b7151851	637	source = index->keyValue;
daniele	29:1a47b7151851	638	pico_tree_delete(&listen->MCASTSources, source);
daniele	29:1a47b7151851	639	pico_free(source);
daniele	29:1a47b7151851	640	}
daniele	29:1a47b7151851	641	filter_mode = pico_socket_aggregate_mcastfilters(&listen->mcast_link, &listen->mcast_group);
tass	48:40fc4462265c	642	if (filter_mode >= 0)
tass	48:40fc4462265c	643	pico_ipv4_mcast_leave(&listen->mcast_link, &listen->mcast_group, 1, filter_mode, &MCASTFilter);
daniele	29:1a47b7151851	644	pico_tree_delete(s->MCASTListen, listen);
daniele	29:1a47b7151851	645	pico_free(listen);
daniele	29:1a47b7151851	646	}
daniele	29:1a47b7151851	647	pico_free(s->MCASTListen);
daniele	29:1a47b7151851	648	}
daniele	29:1a47b7151851	649	} while (0);
daniele	29:1a47b7151851	650	#endif
daniele	29:1a47b7151851	651
daniele	29:1a47b7151851	652	s->state = PICO_SOCKET_STATE_CLOSED;
daniele	29:1a47b7151851	653	pico_timer_add(3000, socket_garbage_collect, s);
tass	51:ab4529a384a6	654	UNLOCK(Mutex);
daniele	29:1a47b7151851	655	return 0;
daniele	29:1a47b7151851	656	}
daniele	29:1a47b7151851	657
daniele	29:1a47b7151851	658	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	659	{
daniele	29:1a47b7151851	660	struct pico_sockport *sp;
daniele	29:1a47b7151851	661	if (more_states & PICO_SOCKET_STATE_BOUND)
daniele	29:1a47b7151851	662	return pico_socket_add(s);
daniele	29:1a47b7151851	663
daniele	29:1a47b7151851	664	if (less_states & PICO_SOCKET_STATE_BOUND)
daniele	29:1a47b7151851	665	return pico_socket_del(s);
daniele	29:1a47b7151851	666
daniele	29:1a47b7151851	667	sp = pico_get_sockport(PROTO(s), s->local_port);
daniele	29:1a47b7151851	668	if (!sp) {
daniele	29:1a47b7151851	669	pico_err = PICO_ERR_ENXIO;
daniele	29:1a47b7151851	670	return -1;
daniele	29:1a47b7151851	671	}
daniele	29:1a47b7151851	672
daniele	29:1a47b7151851	673	s->state \|= more_states;
daniele	29:1a47b7151851	674	s->state &= (~less_states);
daniele	29:1a47b7151851	675	if (tcp_state) {
daniele	29:1a47b7151851	676	s->state &= 0x00FF;
daniele	29:1a47b7151851	677	s->state \|= tcp_state;
daniele	29:1a47b7151851	678	}
daniele	29:1a47b7151851	679
daniele	29:1a47b7151851	680	return 0;
daniele	29:1a47b7151851	681	}
daniele	29:1a47b7151851	682
daniele	29:1a47b7151851	683	static int pico_socket_deliver(struct pico_protocol p, struct pico_frame f, uint16_t localport)
daniele	29:1a47b7151851	684	{
daniele	29:1a47b7151851	685	struct pico_frame *cpy = NULL;
daniele	29:1a47b7151851	686	struct pico_sockport *sp = NULL;
daniele	29:1a47b7151851	687	struct pico_socket s = NULL, found = NULL;
daniele	29:1a47b7151851	688	struct pico_tree_node *index = NULL;
daniele	29:1a47b7151851	689	struct pico_tree_node *_tmp;
daniele	29:1a47b7151851	690	struct pico_trans tr = (struct pico_trans ) f->transport_hdr;
daniele	29:1a47b7151851	691	#ifdef PICO_SUPPORT_IPV4
daniele	29:1a47b7151851	692	struct pico_ipv4_hdr *ip4hdr;
daniele	29:1a47b7151851	693	#endif
daniele	29:1a47b7151851	694	#ifdef PICO_SUPPORT_IPV6
daniele	29:1a47b7151851	695	struct pico_ipv6_hdr *ip6hdr;
daniele	29:1a47b7151851	696	#endif
daniele	29:1a47b7151851	697
daniele	29:1a47b7151851	698	if (!tr)
daniele	29:1a47b7151851	699	return -1;
daniele	29:1a47b7151851	700
daniele	29:1a47b7151851	701	sp = pico_get_sockport(p->proto_number, localport);
daniele	29:1a47b7151851	702
daniele	29:1a47b7151851	703	if (!sp) {
daniele	29:1a47b7151851	704	dbg("No such port %d\n",short_be(localport));
daniele	29:1a47b7151851	705	return -1;
daniele	29:1a47b7151851	706	}
daniele	29:1a47b7151851	707
daniele	29:1a47b7151851	708	#ifdef PICO_SUPPORT_TCP
daniele	29:1a47b7151851	709	if (p->proto_number == PICO_PROTO_TCP) {
daniele	29:1a47b7151851	710	pico_tree_foreach_safe(index,&sp->socks, _tmp){
daniele	29:1a47b7151851	711	s = index->keyValue;
daniele	29:1a47b7151851	712	/* 4-tuple identification of socket (port-IP) */
daniele	29:1a47b7151851	713	#ifdef PICO_SUPPORT_IPV4
daniele	29:1a47b7151851	714	if (IS_IPV4(f)) {
daniele	29:1a47b7151851	715	struct pico_ip4 s_local, s_remote, p_src, p_dst;
daniele	29:1a47b7151851	716	ip4hdr = (struct pico_ipv4_hdr*)(f->net_hdr);
daniele	29:1a47b7151851	717	s_local.addr = s->local_addr.ip4.addr;
daniele	29:1a47b7151851	718	s_remote.addr = s->remote_addr.ip4.addr;
daniele	29:1a47b7151851	719	p_src.addr = ip4hdr->src.addr;
daniele	29:1a47b7151851	720	p_dst.addr = ip4hdr->dst.addr;
daniele	29:1a47b7151851	721	if ( (s->remote_port == tr->sport) && /* remote port check */
daniele	29:1a47b7151851	722	(s_remote.addr == p_src.addr) && /* remote addr check */
daniele	29:1a47b7151851	723	((s_local.addr == PICO_IPV4_INADDR_ANY) \|\| (s_local.addr == p_dst.addr))) { /* Either local socket is ANY, or matches dst */
daniele	29:1a47b7151851	724	found = s;
daniele	29:1a47b7151851	725	break;
daniele	29:1a47b7151851	726	} else if ( (s->remote_port == 0) && /* not connected... listening */
daniele	29:1a47b7151851	727	((s_local.addr == PICO_IPV4_INADDR_ANY) \|\| (s_local.addr == p_dst.addr))) { /* Either local socket is ANY, or matches dst */
daniele	29:1a47b7151851	728	/* listen socket */
daniele	29:1a47b7151851	729	found = s;
daniele	29:1a47b7151851	730	}
daniele	29:1a47b7151851	731	}
daniele	29:1a47b7151851	732	#endif
daniele	29:1a47b7151851	733	#ifdef PICO_SUPPORT_IPV6 /* XXX TODO make compare for ipv6 addresses */
daniele	29:1a47b7151851	734	if (IS_IPV6(f)) {
daniele	29:1a47b7151851	735	ip6hdr = (struct pico_ipv6_hdr*)(f->net_hdr);
daniele	29:1a47b7151851	736	if ( (s->remote_port == localport) ) { // && (((struct pico_ip6) s->remote_addr.ip6).addr == ((struct pico_ip6)(ip6hdr->src)).addr) ) {
daniele	29:1a47b7151851	737	found = s;
daniele	29:1a47b7151851	738	break;
daniele	29:1a47b7151851	739	} else if (s->remote_port == 0) {
daniele	29:1a47b7151851	740	/* listen socket */
daniele	29:1a47b7151851	741	found = s;
daniele	29:1a47b7151851	742	}
daniele	29:1a47b7151851	743	}
daniele	29:1a47b7151851	744	#endif
daniele	29:1a47b7151851	745	} /* FOREACH */
daniele	29:1a47b7151851	746	if (found != NULL) {
daniele	29:1a47b7151851	747	pico_tcp_input(found,f);
daniele	29:1a47b7151851	748	if ((found->ev_pending) && found->wakeup) {
daniele	29:1a47b7151851	749	found->wakeup(found->ev_pending, found);
daniele	29:1a47b7151851	750	}
daniele	29:1a47b7151851	751	return 0;
daniele	29:1a47b7151851	752	} else {
daniele	29:1a47b7151851	753	dbg("SOCKET> mmm something wrong (prob sockport)\n");
daniele	29:1a47b7151851	754	return -1;
daniele	29:1a47b7151851	755	}
daniele	29:1a47b7151851	756	} /* TCP CASE */
daniele	29:1a47b7151851	757	#endif
daniele	29:1a47b7151851	758
daniele	29:1a47b7151851	759	#ifdef PICO_SUPPORT_UDP
daniele	29:1a47b7151851	760	if (p->proto_number == PICO_PROTO_UDP) {
daniele	29:1a47b7151851	761	pico_tree_foreach_safe(index,&sp->socks, _tmp){
daniele	29:1a47b7151851	762	s = index->keyValue;
daniele	29:1a47b7151851	763	if (IS_IPV4(f)) { /* IPV4 */
daniele	29:1a47b7151851	764	struct pico_ip4 s_local, p_dst;
daniele	29:1a47b7151851	765	ip4hdr = (struct pico_ipv4_hdr*)(f->net_hdr);
daniele	29:1a47b7151851	766	s_local.addr = s->local_addr.ip4.addr;
daniele	29:1a47b7151851	767	p_dst.addr = ip4hdr->dst.addr;
daniele	29:1a47b7151851	768	if ((pico_ipv4_is_broadcast(p_dst.addr)) \|\| pico_ipv4_is_multicast(p_dst.addr)) {
daniele	29:1a47b7151851	769	struct pico_device *dev = pico_ipv4_link_find(&s->local_addr.ip4);
daniele	29:1a47b7151851	770	if (pico_ipv4_is_multicast(p_dst.addr) && (pico_socket_mcast_filter(s, &ip4hdr->dst, &ip4hdr->src) < 0))
daniele	29:1a47b7151851	771	return -1;
daniele	29:1a47b7151851	772	if ((s_local.addr == PICO_IPV4_INADDR_ANY) \|\| /* If our local ip is ANY, or.. */
daniele	29:1a47b7151851	773	(dev == f->dev) ) { /* the source of the bcast packet is a neighbor... */
daniele	29:1a47b7151851	774	cpy = pico_frame_copy(f);
daniele	29:1a47b7151851	775	if (!cpy)
daniele	29:1a47b7151851	776	return -1;
daniele	29:1a47b7151851	777	if (pico_enqueue(&s->q_in, cpy) > 0) {
daniele	29:1a47b7151851	778	if (s->wakeup)
daniele	29:1a47b7151851	779	s->wakeup(PICO_SOCK_EV_RD, s);
daniele	29:1a47b7151851	780	}
daniele	29:1a47b7151851	781	}
daniele	29:1a47b7151851	782	} else if ((s_local.addr == PICO_IPV4_INADDR_ANY) \|\| (s_local.addr == p_dst.addr))
daniele	29:1a47b7151851	783	{ /* Either local socket is ANY, or matches dst */
daniele	29:1a47b7151851	784	cpy = pico_frame_copy(f);
daniele	29:1a47b7151851	785	if (!cpy)
daniele	29:1a47b7151851	786	return -1;
daniele	29:1a47b7151851	787	if (pico_enqueue(&s->q_in, cpy) > 0) {
daniele	29:1a47b7151851	788	if (s->wakeup)
daniele	29:1a47b7151851	789	s->wakeup(PICO_SOCK_EV_RD, s);
daniele	29:1a47b7151851	790	}
daniele	29:1a47b7151851	791	}
daniele	29:1a47b7151851	792	} else {
daniele	29:1a47b7151851	793	/... IPv6 /
daniele	29:1a47b7151851	794	}
daniele	29:1a47b7151851	795	} /* FOREACH */
daniele	29:1a47b7151851	796	pico_frame_discard(f);
daniele	29:1a47b7151851	797	if (s)
daniele	29:1a47b7151851	798	return 0;
daniele	29:1a47b7151851	799	else
daniele	29:1a47b7151851	800	return -1;
daniele	29:1a47b7151851	801	}
daniele	29:1a47b7151851	802	#endif
daniele	29:1a47b7151851	803	return -1;
daniele	29:1a47b7151851	804	}
daniele	29:1a47b7151851	805
daniele	29:1a47b7151851	806	struct pico_socket pico_socket_open(uint16_t net, uint16_t proto, void (wakeup)(uint16_t ev, struct pico_socket *))
daniele	29:1a47b7151851	807	{
daniele	29:1a47b7151851	808
daniele	29:1a47b7151851	809	struct pico_socket *s = NULL;
daniele	29:1a47b7151851	810
daniele	29:1a47b7151851	811	#ifdef PICO_SUPPORT_UDP
daniele	29:1a47b7151851	812	if (proto == PICO_PROTO_UDP) {
daniele	29:1a47b7151851	813	s = pico_udp_open();
daniele	29:1a47b7151851	814	s->proto = &pico_proto_udp;
daniele	29:1a47b7151851	815	}
daniele	29:1a47b7151851	816	#endif
daniele	29:1a47b7151851	817
daniele	29:1a47b7151851	818	#ifdef PICO_SUPPORT_TCP
daniele	29:1a47b7151851	819	if (proto == PICO_PROTO_TCP) {
daniele	29:1a47b7151851	820	s = pico_tcp_open();
daniele	29:1a47b7151851	821	s->proto = &pico_proto_tcp;
daniele	29:1a47b7151851	822	/check if Nagle enabled /
daniele	29:1a47b7151851	823	if (!IS_NAGLE_ENABLED(s))
daniele	29:1a47b7151851	824	dbg("ERROR Nagle should be enabled here\n\n");
daniele	29:1a47b7151851	825	}
daniele	29:1a47b7151851	826	#endif
daniele	29:1a47b7151851	827
daniele	29:1a47b7151851	828	if (!s) {
daniele	29:1a47b7151851	829	pico_err = PICO_ERR_EPROTONOSUPPORT;
daniele	29:1a47b7151851	830	return NULL;
daniele	29:1a47b7151851	831	}
daniele	29:1a47b7151851	832
daniele	29:1a47b7151851	833	#ifdef PICO_SUPPORT_IPV4
daniele	29:1a47b7151851	834	if (net == PICO_PROTO_IPV4)
daniele	29:1a47b7151851	835	s->net = &pico_proto_ipv4;
daniele	29:1a47b7151851	836	#endif
daniele	29:1a47b7151851	837
daniele	29:1a47b7151851	838	#ifdef PICO_SUPPORT_IPV6
daniele	29:1a47b7151851	839	if (net == PICO_PROTO_IPV6)
daniele	29:1a47b7151851	840	s->net = &pico_proto_ipv6;
daniele	29:1a47b7151851	841	#endif
daniele	29:1a47b7151851	842
daniele	29:1a47b7151851	843	s->q_in.max_size = PICO_DEFAULT_SOCKETQ;
daniele	29:1a47b7151851	844	s->q_out.max_size = PICO_DEFAULT_SOCKETQ;
daniele	29:1a47b7151851	845	s->wakeup = wakeup;
daniele	29:1a47b7151851	846
daniele	29:1a47b7151851	847	if (!s->net) {
daniele	29:1a47b7151851	848	pico_free(s);
daniele	29:1a47b7151851	849	pico_err = PICO_ERR_ENETUNREACH;
daniele	29:1a47b7151851	850	return NULL;
daniele	29:1a47b7151851	851	}
daniele	29:1a47b7151851	852	return s;
daniele	29:1a47b7151851	853	}
daniele	29:1a47b7151851	854
daniele	29:1a47b7151851	855
daniele	29:1a47b7151851	856	struct pico_socket pico_socket_clone(struct pico_socket facsimile)
daniele	29:1a47b7151851	857	{
daniele	29:1a47b7151851	858	struct pico_socket *s = NULL;
daniele	29:1a47b7151851	859
daniele	29:1a47b7151851	860	#ifdef PICO_SUPPORT_UDP
daniele	29:1a47b7151851	861	if (facsimile->proto->proto_number == PICO_PROTO_UDP) {
daniele	29:1a47b7151851	862	s = pico_udp_open();
daniele	29:1a47b7151851	863	s->proto = &pico_proto_udp;
daniele	29:1a47b7151851	864	}
daniele	29:1a47b7151851	865	#endif
daniele	29:1a47b7151851	866
daniele	29:1a47b7151851	867	#ifdef PICO_SUPPORT_TCP
daniele	29:1a47b7151851	868	if (facsimile->proto->proto_number == PICO_PROTO_TCP) {
daniele	29:1a47b7151851	869	s = pico_tcp_open();
daniele	29:1a47b7151851	870	s->proto = &pico_proto_tcp;
daniele	29:1a47b7151851	871	}
daniele	29:1a47b7151851	872	#endif
daniele	29:1a47b7151851	873
daniele	29:1a47b7151851	874	if (!s) {
daniele	29:1a47b7151851	875	pico_err = PICO_ERR_EPROTONOSUPPORT;
daniele	29:1a47b7151851	876	return NULL;
daniele	29:1a47b7151851	877	}
daniele	29:1a47b7151851	878	s->local_port = facsimile->local_port;
daniele	29:1a47b7151851	879	s->remote_port = facsimile->remote_port;
daniele	29:1a47b7151851	880	s->state = facsimile->state;
daniele	29:1a47b7151851	881
daniele	29:1a47b7151851	882	#ifdef PICO_SUPPORT_IPV4
daniele	29:1a47b7151851	883	if (facsimile->net == &pico_proto_ipv4) {
daniele	29:1a47b7151851	884	s->net = &pico_proto_ipv4;
daniele	29:1a47b7151851	885	memcpy(&s->local_addr, &facsimile->local_addr, sizeof(struct pico_ip4));
daniele	29:1a47b7151851	886	memcpy(&s->remote_addr, &facsimile->remote_addr, sizeof(struct pico_ip4));
daniele	29:1a47b7151851	887	}
daniele	29:1a47b7151851	888	#endif
daniele	29:1a47b7151851	889
daniele	29:1a47b7151851	890	#ifdef PICO_SUPPORT_IPV6
daniele	29:1a47b7151851	891	if (net == &pico_proto_ipv6) {
daniele	29:1a47b7151851	892	s->net = &pico_proto_ipv6;
daniele	29:1a47b7151851	893	memcpy(&s->local_addr, &facsimile->local_addr, sizeof(struct pico_ip6));
daniele	29:1a47b7151851	894	memcpy(&s->remote_addr, &facsimile->remote_addr, sizeof(struct pico_ip6));
daniele	29:1a47b7151851	895	}
daniele	29:1a47b7151851	896	#endif
daniele	29:1a47b7151851	897	s->q_in.max_size = PICO_DEFAULT_SOCKETQ;
daniele	29:1a47b7151851	898	s->q_out.max_size = PICO_DEFAULT_SOCKETQ;
daniele	29:1a47b7151851	899	s->wakeup = NULL;
daniele	29:1a47b7151851	900	if (!s->net) {
daniele	29:1a47b7151851	901	pico_free(s);
daniele	29:1a47b7151851	902	pico_err = PICO_ERR_ENETUNREACH;
daniele	29:1a47b7151851	903	return NULL;
daniele	29:1a47b7151851	904	}
daniele	29:1a47b7151851	905	return s;
daniele	29:1a47b7151851	906	}
daniele	29:1a47b7151851	907
daniele	29:1a47b7151851	908	int pico_socket_read(struct pico_socket s, void buf, int len)
daniele	29:1a47b7151851	909	{
daniele	29:1a47b7151851	910	if (!s \|\| buf == NULL) {
daniele	29:1a47b7151851	911	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	912	return -1;
daniele	29:1a47b7151851	913	} else {
daniele	29:1a47b7151851	914	/* check if exists in tree */
daniele	29:1a47b7151851	915	/* See task #178 */
daniele	29:1a47b7151851	916	if (pico_check_socket(s) != 0) {
daniele	29:1a47b7151851	917	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	918	return -1;
daniele	29:1a47b7151851	919	}
daniele	29:1a47b7151851	920	}
daniele	29:1a47b7151851	921
daniele	29:1a47b7151851	922	if ((s->state & PICO_SOCKET_STATE_BOUND) == 0) {
daniele	29:1a47b7151851	923	pico_err = PICO_ERR_EIO;
daniele	29:1a47b7151851	924	return -1;
daniele	29:1a47b7151851	925	}
daniele	29:1a47b7151851	926	#ifdef PICO_SUPPORT_UDP
daniele	29:1a47b7151851	927	if (PROTO(s) == PICO_PROTO_UDP)
daniele	29:1a47b7151851	928	return pico_udp_recv(s, buf, len, NULL, NULL);
daniele	29:1a47b7151851	929	#endif
daniele	29:1a47b7151851	930
daniele	29:1a47b7151851	931	#ifdef PICO_SUPPORT_TCP
daniele	29:1a47b7151851	932	if (PROTO(s) == PICO_PROTO_TCP){
daniele	29:1a47b7151851	933	/* check if in shutdown state and if no more data in tcpq_in */
daniele	29:1a47b7151851	934	if ((s->state & PICO_SOCKET_STATE_SHUT_REMOTE) && pico_tcp_queue_in_is_empty(s) ) {
daniele	29:1a47b7151851	935	pico_err = PICO_ERR_ESHUTDOWN;
daniele	29:1a47b7151851	936	return -1;
daniele	29:1a47b7151851	937	} else {
daniele	29:1a47b7151851	938	return pico_tcp_read(s, buf, len);
daniele	29:1a47b7151851	939	}
daniele	29:1a47b7151851	940	}
daniele	29:1a47b7151851	941	#endif
daniele	29:1a47b7151851	942	return 0;
daniele	29:1a47b7151851	943	}
daniele	29:1a47b7151851	944
daniele	29:1a47b7151851	945	int pico_socket_write(struct pico_socket s, void buf, int len)
daniele	29:1a47b7151851	946	{
daniele	29:1a47b7151851	947	if (!s \|\| buf == NULL) {
daniele	29:1a47b7151851	948	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	949	return -1;
daniele	29:1a47b7151851	950	} else {
daniele	29:1a47b7151851	951	/* check if exists in tree */
daniele	29:1a47b7151851	952	/* See task #178 */
daniele	29:1a47b7151851	953	if (pico_check_socket(s) != 0) {
daniele	29:1a47b7151851	954	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	955	return -1;
daniele	29:1a47b7151851	956	}
daniele	29:1a47b7151851	957	}
daniele	29:1a47b7151851	958
daniele	29:1a47b7151851	959	if ((s->state & PICO_SOCKET_STATE_BOUND) == 0) {
daniele	29:1a47b7151851	960	pico_err = PICO_ERR_EIO;
daniele	29:1a47b7151851	961	return -1;
daniele	29:1a47b7151851	962	}
daniele	29:1a47b7151851	963	if ((s->state & PICO_SOCKET_STATE_CONNECTED) == 0) {
daniele	29:1a47b7151851	964	pico_err = PICO_ERR_ENOTCONN;
daniele	29:1a47b7151851	965	return -1;
daniele	29:1a47b7151851	966	} else if (s->state & PICO_SOCKET_STATE_SHUT_LOCAL) { /* check if in shutdown state */
daniele	29:1a47b7151851	967	pico_err = PICO_ERR_ESHUTDOWN;
daniele	29:1a47b7151851	968	return -1;
daniele	29:1a47b7151851	969	} else {
daniele	29:1a47b7151851	970	return pico_socket_sendto(s, buf, len, &s->remote_addr, s->remote_port);
daniele	29:1a47b7151851	971	}
daniele	29:1a47b7151851	972	}
daniele	29:1a47b7151851	973
daniele	29:1a47b7151851	974	uint16_t pico_socket_high_port(uint16_t proto)
daniele	29:1a47b7151851	975	{
daniele	29:1a47b7151851	976	uint16_t port;
daniele	29:1a47b7151851	977	if (0 \|\|
daniele	29:1a47b7151851	978	#ifdef PICO_SUPPORT_TCP
daniele	29:1a47b7151851	979	(proto == PICO_PROTO_TCP) \|\|
daniele	29:1a47b7151851	980	#endif
daniele	29:1a47b7151851	981	#ifdef PICO_SUPPORT_TCP
daniele	29:1a47b7151851	982	(proto == PICO_PROTO_UDP) \|\|
daniele	29:1a47b7151851	983	#endif
daniele	29:1a47b7151851	984	0) {
daniele	29:1a47b7151851	985	do {
daniele	29:1a47b7151851	986	uint32_t rand = pico_rand();
daniele	29:1a47b7151851	987	port = (uint16_t) (rand & 0xFFFFU);
daniele	29:1a47b7151851	988	port = (uint16_t)(port % (65535 - 1024)) + 1024U;
daniele	29:1a47b7151851	989	if (pico_is_port_free(proto, port, NULL, NULL)) {
daniele	29:1a47b7151851	990	return short_be(port);
daniele	29:1a47b7151851	991	}
daniele	29:1a47b7151851	992	} while(1);
daniele	29:1a47b7151851	993	}
daniele	29:1a47b7151851	994	else return 0U;
daniele	29:1a47b7151851	995	}
daniele	29:1a47b7151851	996
daniele	29:1a47b7151851	997
daniele	29:1a47b7151851	998	int pico_socket_sendto(struct pico_socket s, void buf, int len, void *dst, uint16_t remote_port)
daniele	29:1a47b7151851	999	{
daniele	29:1a47b7151851	1000	struct pico_frame *f;
daniele	29:1a47b7151851	1001	struct pico_remote_duple *remote_duple = NULL;
tass	48:40fc4462265c	1002	int socket_mtu = PICO_SOCKET4_MTU;
daniele	29:1a47b7151851	1003	int header_offset = 0;
daniele	29:1a47b7151851	1004	int total_payload_written = 0;
daniele	29:1a47b7151851	1005	#ifdef PICO_SUPPORT_IPV4
daniele	29:1a47b7151851	1006	struct pico_ip4 *src4;
daniele	29:1a47b7151851	1007	#endif
daniele	29:1a47b7151851	1008
daniele	29:1a47b7151851	1009	#ifdef PICO_SUPPORT_IPV6
daniele	29:1a47b7151851	1010	struct pico_ip6 *src6;
daniele	29:1a47b7151851	1011	#endif
daniele	29:1a47b7151851	1012	if (len == 0) {
daniele	29:1a47b7151851	1013	return 0;
daniele	29:1a47b7151851	1014	} else if (len < 0) {
daniele	29:1a47b7151851	1015	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1016	return -1;
daniele	29:1a47b7151851	1017	}
daniele	29:1a47b7151851	1018
daniele	29:1a47b7151851	1019	if (buf == NULL \|\| s == NULL) {
daniele	29:1a47b7151851	1020	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1021	return -1;
daniele	29:1a47b7151851	1022	}
daniele	29:1a47b7151851	1023
daniele	29:1a47b7151851	1024	if (!dst \|\| !remote_port) {
daniele	29:1a47b7151851	1025	pico_err = PICO_ERR_EADDRNOTAVAIL;
daniele	29:1a47b7151851	1026	return -1;
daniele	29:1a47b7151851	1027	}
daniele	29:1a47b7151851	1028
daniele	29:1a47b7151851	1029	if ((s->state & PICO_SOCKET_STATE_CONNECTED) != 0) {
daniele	29:1a47b7151851	1030	if (remote_port != s->remote_port) {
daniele	29:1a47b7151851	1031	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1032	return -1;
daniele	29:1a47b7151851	1033	}
daniele	29:1a47b7151851	1034	}
daniele	29:1a47b7151851	1035
daniele	29:1a47b7151851	1036	#ifdef PICO_SUPPORT_IPV4
daniele	29:1a47b7151851	1037	if (IS_SOCK_IPV4(s)) {
tass	48:40fc4462265c	1038	socket_mtu = PICO_SOCKET4_MTU;
daniele	29:1a47b7151851	1039	if ((s->state & PICO_SOCKET_STATE_CONNECTED)) {
daniele	29:1a47b7151851	1040	if (s->remote_addr.ip4.addr != ((struct pico_ip4 *)dst)->addr ) {
daniele	29:1a47b7151851	1041	pico_err = PICO_ERR_EADDRNOTAVAIL;
daniele	29:1a47b7151851	1042	return -1;
daniele	29:1a47b7151851	1043	}
daniele	29:1a47b7151851	1044	} else {
daniele	29:1a47b7151851	1045	src4 = pico_ipv4_source_find(dst);
daniele	29:1a47b7151851	1046	if (!src4) {
daniele	29:1a47b7151851	1047	pico_err = PICO_ERR_EHOSTUNREACH;
daniele	29:1a47b7151851	1048	return -1;
daniele	29:1a47b7151851	1049	}
daniele	29:1a47b7151851	1050	if (src4->addr != PICO_IPV4_INADDR_ANY)
daniele	29:1a47b7151851	1051	s->local_addr.ip4.addr = src4->addr;
daniele	29:1a47b7151851	1052	# ifdef PICO_SUPPORT_UDP
daniele	29:1a47b7151851	1053	/* socket remote info could change in a consecutive call, make persistent */
daniele	29:1a47b7151851	1054	if (PROTO(s) == PICO_PROTO_UDP) {
daniele	29:1a47b7151851	1055	remote_duple = pico_zalloc(sizeof(struct pico_remote_duple));
daniele	29:1a47b7151851	1056	remote_duple->remote_addr.ip4.addr = ((struct pico_ip4 *)dst)->addr;
daniele	29:1a47b7151851	1057	remote_duple->remote_port = remote_port;
daniele	29:1a47b7151851	1058	}
daniele	29:1a47b7151851	1059	# endif
daniele	29:1a47b7151851	1060	}
tass	48:40fc4462265c	1061	#endif
tass	48:40fc4462265c	1062	} else if (IS_SOCK_IPV6(s)) {
tass	48:40fc4462265c	1063	socket_mtu = PICO_SOCKET6_MTU;
tass	48:40fc4462265c	1064	#ifdef PICO_SUPPORT_IPV6
daniele	29:1a47b7151851	1065	if (s->state & PICO_SOCKET_STATE_CONNECTED) {
daniele	29:1a47b7151851	1066	if (memcmp(&s->remote_addr, dst, PICO_SIZE_IP6))
daniele	29:1a47b7151851	1067	return -1;
daniele	29:1a47b7151851	1068	} else {
daniele	29:1a47b7151851	1069	src6 = pico_ipv6_source_find(dst);
daniele	29:1a47b7151851	1070	if (!src6) {
daniele	29:1a47b7151851	1071	pico_err = PICO_ERR_EHOSTUNREACH;
daniele	29:1a47b7151851	1072	return -1;
daniele	29:1a47b7151851	1073	}
daniele	29:1a47b7151851	1074	memcpy(&s->local_addr, src6, PICO_SIZE_IP6);
daniele	29:1a47b7151851	1075	memcpy(&s->remote_addr, dst, PICO_SIZE_IP6);
daniele	29:1a47b7151851	1076	# ifdef PICO_SUPPORT_UDP
daniele	29:1a47b7151851	1077	if (PROTO(s) == PICO_PROTO_UDP) {
daniele	29:1a47b7151851	1078	remote_duple = pico_zalloc(sizeof(struct pico_remote_duple));
daniele	29:1a47b7151851	1079	remote_duple->remote_addr.ip6.addr = ((struct pico_ip6 *)dst)->addr;
daniele	29:1a47b7151851	1080	remote_duple->remote_port = remote_port;
daniele	29:1a47b7151851	1081	}
daniele	29:1a47b7151851	1082	# endif
daniele	29:1a47b7151851	1083	}
tass	48:40fc4462265c	1084	#endif
daniele	29:1a47b7151851	1085	}
daniele	29:1a47b7151851	1086
daniele	29:1a47b7151851	1087	if ((s->state & PICO_SOCKET_STATE_BOUND) == 0) {
daniele	29:1a47b7151851	1088	s->local_port = pico_socket_high_port(s->proto->proto_number);
daniele	29:1a47b7151851	1089	if (s->local_port == 0) {
daniele	29:1a47b7151851	1090	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1091	return -1;
daniele	29:1a47b7151851	1092	}
daniele	29:1a47b7151851	1093	}
daniele	29:1a47b7151851	1094	if ((s->state & PICO_SOCKET_STATE_CONNECTED) == 0) {
daniele	29:1a47b7151851	1095	s->remote_port = remote_port;
daniele	29:1a47b7151851	1096	}
daniele	29:1a47b7151851	1097
daniele	29:1a47b7151851	1098	#ifdef PICO_SUPPORT_TCP
daniele	29:1a47b7151851	1099	if (PROTO(s) == PICO_PROTO_TCP)
daniele	29:1a47b7151851	1100	header_offset = pico_tcp_overhead(s);
daniele	29:1a47b7151851	1101	#endif
daniele	29:1a47b7151851	1102
daniele	29:1a47b7151851	1103	#ifdef PICO_SUPPORT_UDP
daniele	29:1a47b7151851	1104	if (PROTO(s) == PICO_PROTO_UDP)
daniele	29:1a47b7151851	1105	header_offset = sizeof(struct pico_udp_hdr);
daniele	29:1a47b7151851	1106	#endif
daniele	29:1a47b7151851	1107
daniele	29:1a47b7151851	1108	while (total_payload_written < len) {
daniele	29:1a47b7151851	1109	int transport_len = (len - total_payload_written) + header_offset;
tass	48:40fc4462265c	1110	if (transport_len > socket_mtu)
tass	48:40fc4462265c	1111	transport_len = socket_mtu;
tass	48:40fc4462265c	1112	#ifdef PICO_SUPPORT_IPFRAG
daniele	29:1a47b7151851	1113	else {
daniele	29:1a47b7151851	1114	if (total_payload_written)
daniele	29:1a47b7151851	1115	transport_len -= header_offset; /* last fragment, do not allocate memory for transport header */
daniele	29:1a47b7151851	1116	}
daniele	29:1a47b7151851	1117	#endif /* PICO_SUPPORT_IPFRAG */
daniele	29:1a47b7151851	1118
daniele	29:1a47b7151851	1119	f = pico_socket_frame_alloc(s, transport_len);
daniele	29:1a47b7151851	1120	if (!f) {
daniele	29:1a47b7151851	1121	pico_err = PICO_ERR_ENOMEM;
daniele	29:1a47b7151851	1122	return -1;
daniele	29:1a47b7151851	1123	}
daniele	29:1a47b7151851	1124	f->payload += header_offset;
daniele	29:1a47b7151851	1125	f->payload_len -= header_offset;
daniele	29:1a47b7151851	1126	f->sock = s;
daniele	29:1a47b7151851	1127	if (remote_duple) {
daniele	29:1a47b7151851	1128	f->info = pico_zalloc(sizeof(struct pico_remote_duple));
daniele	29:1a47b7151851	1129	memcpy(f->info, remote_duple, sizeof(struct pico_remote_duple));
daniele	29:1a47b7151851	1130	}
daniele	29:1a47b7151851	1131
tass	48:40fc4462265c	1132	#ifdef PICO_SUPPORT_IPFRAG
tass	48:40fc4462265c	1133	#ifdef PICO_SUPPORT_UDP
tass	48:40fc4462265c	1134	if (PROTO(s) == PICO_PROTO_UDP && ((len + header_offset) > socket_mtu)) {
daniele	29:1a47b7151851	1135	/* hacking way to identify fragmentation frames: payload != transport_hdr -> first frame */
daniele	29:1a47b7151851	1136	if (!total_payload_written) {
daniele	29:1a47b7151851	1137	frag_dbg("FRAG: first fragmented frame %p \| len = %u offset = 0\n", f, f->payload_len);
daniele	29:1a47b7151851	1138	/* transport header length field contains total length + header length */
daniele	29:1a47b7151851	1139	f->transport_len = len + header_offset;
daniele	29:1a47b7151851	1140	f->frag = short_be(PICO_IPV4_MOREFRAG);
daniele	29:1a47b7151851	1141	} else {
daniele	29:1a47b7151851	1142	/* no transport header in fragmented IP */
daniele	29:1a47b7151851	1143	f->payload = f->transport_hdr;
daniele	29:1a47b7151851	1144	f->payload_len += header_offset;
daniele	29:1a47b7151851	1145	/* set offset in octets */
daniele	29:1a47b7151851	1146	f->frag = short_be((total_payload_written + header_offset) / 8);
daniele	29:1a47b7151851	1147	if (total_payload_written + f->payload_len < len) {
daniele	29:1a47b7151851	1148	frag_dbg("FRAG: intermediate fragmented frame %p \| len = %u offset = %u\n", f, f->payload_len, short_be(f->frag));
daniele	29:1a47b7151851	1149	f->frag \|= short_be(PICO_IPV4_MOREFRAG);
daniele	29:1a47b7151851	1150	} else {
daniele	29:1a47b7151851	1151	frag_dbg("FRAG: last fragmented frame %p \| len = %u offset = %u\n", f, f->payload_len, short_be(f->frag));
daniele	29:1a47b7151851	1152	f->frag &= short_be(PICO_IPV4_FRAG_MASK);
daniele	29:1a47b7151851	1153	}
daniele	29:1a47b7151851	1154	}
daniele	29:1a47b7151851	1155	} else {
daniele	29:1a47b7151851	1156	f->frag = short_be(PICO_IPV4_DONTFRAG);
daniele	29:1a47b7151851	1157	}
daniele	29:1a47b7151851	1158	# endif /* PICO_SUPPORT_UDP */
daniele	29:1a47b7151851	1159	#endif /* PICO_SUPPORT_IPFRAG */
daniele	29:1a47b7151851	1160
daniele	29:1a47b7151851	1161	if (f->payload_len <= 0) {
daniele	29:1a47b7151851	1162	pico_frame_discard(f);
daniele	29:1a47b7151851	1163	if (remote_duple)
daniele	29:1a47b7151851	1164	pico_free(remote_duple);
daniele	29:1a47b7151851	1165	return total_payload_written;
daniele	29:1a47b7151851	1166	}
daniele	29:1a47b7151851	1167
daniele	29:1a47b7151851	1168	memcpy(f->payload, buf + total_payload_written, f->payload_len);
daniele	29:1a47b7151851	1169	//dbg("Pushing segment, hdr len: %d, payload_len: %d\n", header_offset, f->payload_len);
daniele	29:1a47b7151851	1170
daniele	29:1a47b7151851	1171	if (s->proto->push(s->proto, f) > 0) {
daniele	29:1a47b7151851	1172	total_payload_written += f->payload_len;
daniele	29:1a47b7151851	1173	} else {
daniele	29:1a47b7151851	1174	pico_frame_discard(f);
daniele	29:1a47b7151851	1175	pico_err = PICO_ERR_EAGAIN;
daniele	29:1a47b7151851	1176	break;
daniele	29:1a47b7151851	1177	}
daniele	29:1a47b7151851	1178	}
daniele	29:1a47b7151851	1179	if (remote_duple)
daniele	29:1a47b7151851	1180	pico_free(remote_duple);
daniele	29:1a47b7151851	1181	return total_payload_written;
daniele	29:1a47b7151851	1182	}
daniele	29:1a47b7151851	1183
daniele	29:1a47b7151851	1184	int pico_socket_send(struct pico_socket s, void buf, int len)
daniele	29:1a47b7151851	1185	{
daniele	29:1a47b7151851	1186	if (!s \|\| buf == NULL) {
daniele	29:1a47b7151851	1187	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1188	return -1;
daniele	29:1a47b7151851	1189	} else {
daniele	29:1a47b7151851	1190	/* check if exists in tree */
daniele	29:1a47b7151851	1191	/* See task #178 */
daniele	29:1a47b7151851	1192	if (pico_check_socket(s) != 0) {
daniele	29:1a47b7151851	1193	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1194	return -1;
daniele	29:1a47b7151851	1195	}
daniele	29:1a47b7151851	1196	}
daniele	29:1a47b7151851	1197
daniele	29:1a47b7151851	1198	if ((s->state & PICO_SOCKET_STATE_CONNECTED) == 0) {
daniele	29:1a47b7151851	1199	pico_err = PICO_ERR_ENOTCONN;
daniele	29:1a47b7151851	1200	return -1;
daniele	29:1a47b7151851	1201	}
daniele	29:1a47b7151851	1202	return pico_socket_sendto(s, buf, len, &s->remote_addr, s->remote_port);
daniele	29:1a47b7151851	1203	}
daniele	29:1a47b7151851	1204
daniele	29:1a47b7151851	1205	int pico_socket_recvfrom(struct pico_socket s, void buf, int len, void orig, uint16_t remote_port)
daniele	29:1a47b7151851	1206	{
daniele	29:1a47b7151851	1207	if (!s \|\| buf == NULL) { /// \|\| orig == NULL \|\| remote_port == NULL) {
daniele	29:1a47b7151851	1208	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1209	return -1;
daniele	29:1a47b7151851	1210	} else {
daniele	29:1a47b7151851	1211	/* check if exists in tree */
daniele	29:1a47b7151851	1212	if (pico_check_socket(s) != 0) {
daniele	29:1a47b7151851	1213	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1214	/* See task #178 */
daniele	29:1a47b7151851	1215	return -1;
daniele	29:1a47b7151851	1216	}
daniele	29:1a47b7151851	1217	}
daniele	29:1a47b7151851	1218
daniele	29:1a47b7151851	1219	if ((s->state & PICO_SOCKET_STATE_BOUND) == 0) {
daniele	29:1a47b7151851	1220	pico_err = PICO_ERR_EADDRNOTAVAIL;
daniele	29:1a47b7151851	1221	return -1;
daniele	29:1a47b7151851	1222	}
daniele	29:1a47b7151851	1223	#ifdef PICO_SUPPORT_UDP
daniele	29:1a47b7151851	1224	if (PROTO(s) == PICO_PROTO_UDP) {
daniele	29:1a47b7151851	1225	return pico_udp_recv(s, buf, len, orig, remote_port);
daniele	29:1a47b7151851	1226	}
daniele	29:1a47b7151851	1227	#endif
daniele	29:1a47b7151851	1228	#ifdef PICO_SUPPORT_TCP
daniele	29:1a47b7151851	1229	if (PROTO(s) == PICO_PROTO_TCP) {
daniele	29:1a47b7151851	1230	/* check if in shutdown state and if tcpq_in empty */
daniele	29:1a47b7151851	1231	if ((s->state & PICO_SOCKET_STATE_SHUT_REMOTE) && pico_tcp_queue_in_is_empty(s)) {
daniele	29:1a47b7151851	1232	pico_err = PICO_ERR_ESHUTDOWN;
daniele	29:1a47b7151851	1233	return -1;
daniele	29:1a47b7151851	1234	} else {
daniele	29:1a47b7151851	1235	//dbg("socket tcp recv\n");
daniele	29:1a47b7151851	1236	return pico_tcp_read(s, buf, len);
daniele	29:1a47b7151851	1237	}
daniele	29:1a47b7151851	1238	}
daniele	29:1a47b7151851	1239	#endif
daniele	29:1a47b7151851	1240	//dbg("socket return 0\n");
daniele	29:1a47b7151851	1241	return 0;
daniele	29:1a47b7151851	1242	}
daniele	29:1a47b7151851	1243
daniele	29:1a47b7151851	1244	int pico_socket_recv(struct pico_socket s, void buf, int len)
daniele	29:1a47b7151851	1245	{
daniele	29:1a47b7151851	1246	return pico_socket_recvfrom(s, buf, len, NULL, NULL);
daniele	29:1a47b7151851	1247	}
daniele	29:1a47b7151851	1248
daniele	29:1a47b7151851	1249
daniele	29:1a47b7151851	1250	int pico_socket_bind(struct pico_socket s, void local_addr, uint16_t *port)
daniele	29:1a47b7151851	1251	{
daniele	29:1a47b7151851	1252	if (!s \|\| !local_addr \|\| !port) {
daniele	29:1a47b7151851	1253	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1254	return -1;
daniele	29:1a47b7151851	1255	}
daniele	29:1a47b7151851	1256
daniele	29:1a47b7151851	1257	if (!is_sock_ipv6(s)) {
daniele	29:1a47b7151851	1258	struct pico_ip4 ip = (struct pico_ip4 )local_addr;
daniele	29:1a47b7151851	1259	if (ip->addr != PICO_IPV4_INADDR_ANY) {
daniele	29:1a47b7151851	1260	if (!pico_ipv4_link_find(local_addr)) {
daniele	29:1a47b7151851	1261	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1262	return -1;
daniele	29:1a47b7151851	1263	}
daniele	29:1a47b7151851	1264	}
daniele	29:1a47b7151851	1265	} else {
daniele	29:1a47b7151851	1266	/... IPv6 /
daniele	29:1a47b7151851	1267	}
daniele	29:1a47b7151851	1268
daniele	29:1a47b7151851	1269
daniele	29:1a47b7151851	1270	/* When given port = 0, get a random high port to bind to. */
daniele	29:1a47b7151851	1271	if (*port == 0) {
daniele	29:1a47b7151851	1272	*port = pico_socket_high_port(PROTO(s));
daniele	29:1a47b7151851	1273	if (*port == 0) {
daniele	29:1a47b7151851	1274	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1275	return -1;
daniele	29:1a47b7151851	1276	}
daniele	29:1a47b7151851	1277	}
daniele	29:1a47b7151851	1278
daniele	29:1a47b7151851	1279	if (pico_is_port_free(PROTO(s), *port, local_addr, s->net) == 0) {
daniele	29:1a47b7151851	1280	pico_err = PICO_ERR_EADDRINUSE;
daniele	29:1a47b7151851	1281	return -1;
daniele	29:1a47b7151851	1282	}
daniele	29:1a47b7151851	1283	s->local_port = *port;
daniele	29:1a47b7151851	1284
daniele	29:1a47b7151851	1285	if (is_sock_ipv6(s)) {
daniele	29:1a47b7151851	1286	struct pico_ip6 ip = (struct pico_ip6 ) local_addr;
daniele	29:1a47b7151851	1287	memcpy(s->local_addr.ip6.addr, ip, PICO_SIZE_IP6);
daniele	29:1a47b7151851	1288	/* XXX: port ipv4 functionality to ipv6 */
daniele	29:1a47b7151851	1289	/* Check for port already in use */
daniele	29:1a47b7151851	1290	if (pico_is_port_free(PROTO(s), *port, &local_addr, s->net)) {
daniele	29:1a47b7151851	1291	pico_err = PICO_ERR_EADDRINUSE;
daniele	29:1a47b7151851	1292	return -1;
daniele	29:1a47b7151851	1293	}
daniele	29:1a47b7151851	1294	} else if (is_sock_ipv4(s)) {
daniele	29:1a47b7151851	1295	struct pico_ip4 ip = (struct pico_ip4 ) local_addr;
daniele	29:1a47b7151851	1296	s->local_addr.ip4.addr = ip->addr;
daniele	29:1a47b7151851	1297	}
daniele	29:1a47b7151851	1298	return pico_socket_alter_state(s, PICO_SOCKET_STATE_BOUND, 0, 0);
daniele	29:1a47b7151851	1299	}
daniele	29:1a47b7151851	1300
daniele	29:1a47b7151851	1301	int pico_socket_connect(struct pico_socket s, void remote_addr, uint16_t remote_port)
daniele	29:1a47b7151851	1302	{
daniele	29:1a47b7151851	1303	int ret = -1;
daniele	29:1a47b7151851	1304	pico_err = PICO_ERR_EPROTONOSUPPORT;
daniele	29:1a47b7151851	1305	if (!s \|\| remote_addr == NULL \|\| remote_port == 0) {
daniele	29:1a47b7151851	1306	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1307	return -1;
daniele	29:1a47b7151851	1308	}
daniele	29:1a47b7151851	1309
daniele	29:1a47b7151851	1310	s->remote_port = remote_port;
daniele	29:1a47b7151851	1311
daniele	29:1a47b7151851	1312	if (s->local_port == 0) {
daniele	29:1a47b7151851	1313	s->local_port = pico_socket_high_port(PROTO(s));
daniele	29:1a47b7151851	1314	if (!s->local_port) {
daniele	29:1a47b7151851	1315	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1316	return -1;
daniele	29:1a47b7151851	1317	}
daniele	29:1a47b7151851	1318	}
daniele	29:1a47b7151851	1319
daniele	29:1a47b7151851	1320	if (is_sock_ipv6(s)) {
daniele	29:1a47b7151851	1321	struct pico_ip6 ip = (struct pico_ip6 ) remote_addr;
daniele	29:1a47b7151851	1322	memcpy(s->remote_addr.ip6.addr, ip, PICO_SIZE_IP6);
daniele	29:1a47b7151851	1323	} else if (is_sock_ipv4(s)) {
daniele	29:1a47b7151851	1324	struct pico_ip4 local, ip = (struct pico_ip4 *) remote_addr;
daniele	29:1a47b7151851	1325	s->remote_addr.ip4.addr = ip->addr;
daniele	29:1a47b7151851	1326	local = pico_ipv4_source_find(ip);
daniele	29:1a47b7151851	1327	if (local) {
daniele	29:1a47b7151851	1328	s->local_addr.ip4.addr = local->addr;
daniele	29:1a47b7151851	1329	} else {
daniele	29:1a47b7151851	1330	pico_err = PICO_ERR_EHOSTUNREACH;
daniele	29:1a47b7151851	1331	return -1;
daniele	29:1a47b7151851	1332	}
daniele	29:1a47b7151851	1333	}
daniele	29:1a47b7151851	1334
daniele	29:1a47b7151851	1335	pico_socket_alter_state(s, PICO_SOCKET_STATE_BOUND, 0, 0);
daniele	29:1a47b7151851	1336
daniele	29:1a47b7151851	1337	#ifdef PICO_SUPPORT_UDP
daniele	29:1a47b7151851	1338	if (PROTO(s) == PICO_PROTO_UDP) {
daniele	29:1a47b7151851	1339	pico_socket_alter_state(s, PICO_SOCKET_STATE_CONNECTED, 0, 0);
daniele	29:1a47b7151851	1340	pico_err = PICO_ERR_NOERR;
daniele	29:1a47b7151851	1341	ret = 0;
daniele	29:1a47b7151851	1342	}
daniele	29:1a47b7151851	1343	#endif
daniele	29:1a47b7151851	1344
daniele	29:1a47b7151851	1345	#ifdef PICO_SUPPORT_TCP
daniele	29:1a47b7151851	1346	if (PROTO(s) == PICO_PROTO_TCP) {
daniele	29:1a47b7151851	1347	if (pico_tcp_initconn(s) == 0) {
daniele	29:1a47b7151851	1348	pico_socket_alter_state(s, PICO_SOCKET_STATE_CONNECTED \| PICO_SOCKET_STATE_TCP_SYN_SENT, 0, 0);
daniele	29:1a47b7151851	1349	pico_err = PICO_ERR_NOERR;
daniele	29:1a47b7151851	1350	ret = 0;
daniele	29:1a47b7151851	1351	} else {
daniele	29:1a47b7151851	1352	pico_err = PICO_ERR_EHOSTUNREACH;
daniele	29:1a47b7151851	1353	}
daniele	29:1a47b7151851	1354	}
daniele	29:1a47b7151851	1355	#endif
daniele	29:1a47b7151851	1356
daniele	29:1a47b7151851	1357	return ret;
daniele	29:1a47b7151851	1358	}
daniele	29:1a47b7151851	1359
daniele	29:1a47b7151851	1360	#ifdef PICO_SUPPORT_TCP
daniele	29:1a47b7151851	1361
daniele	29:1a47b7151851	1362	int pico_socket_listen(struct pico_socket *s, int backlog)
daniele	29:1a47b7151851	1363	{
daniele	29:1a47b7151851	1364	if (!s \|\| backlog < 1) {
daniele	29:1a47b7151851	1365	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1366	return -1;
daniele	29:1a47b7151851	1367	} else {
daniele	29:1a47b7151851	1368	/* check if exists in tree */
daniele	29:1a47b7151851	1369	/* See task #178 */
daniele	29:1a47b7151851	1370	if (pico_check_socket(s) != 0) {
daniele	29:1a47b7151851	1371	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1372	return -1;
daniele	29:1a47b7151851	1373	}
daniele	29:1a47b7151851	1374	}
daniele	29:1a47b7151851	1375
daniele	29:1a47b7151851	1376	if (PROTO(s) == PICO_PROTO_UDP) {
daniele	29:1a47b7151851	1377	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1378	return -1;
daniele	29:1a47b7151851	1379	}
daniele	29:1a47b7151851	1380
daniele	29:1a47b7151851	1381	if ((s->state & PICO_SOCKET_STATE_BOUND) == 0) {
daniele	29:1a47b7151851	1382	pico_err = PICO_ERR_EISCONN;
daniele	29:1a47b7151851	1383	return -1;
daniele	29:1a47b7151851	1384	}
daniele	29:1a47b7151851	1385
daniele	29:1a47b7151851	1386	if (backlog < 1) {
daniele	29:1a47b7151851	1387	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1388	return -1;
daniele	29:1a47b7151851	1389	}
daniele	29:1a47b7151851	1390
daniele	29:1a47b7151851	1391	if (PROTO(s) == PICO_PROTO_TCP)
daniele	29:1a47b7151851	1392	pico_socket_alter_state(s, PICO_SOCKET_STATE_TCP_SYN_SENT, 0, PICO_SOCKET_STATE_TCP_LISTEN);
daniele	29:1a47b7151851	1393	s->max_backlog = backlog;
daniele	29:1a47b7151851	1394
daniele	29:1a47b7151851	1395	return 0;
daniele	29:1a47b7151851	1396	}
daniele	29:1a47b7151851	1397
daniele	29:1a47b7151851	1398	struct pico_socket pico_socket_accept(struct pico_socket s, void orig, uint16_t port)
daniele	29:1a47b7151851	1399	{
daniele	29:1a47b7151851	1400	if (!s \|\| !orig \|\| !port) {
daniele	29:1a47b7151851	1401	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1402	return NULL;
daniele	29:1a47b7151851	1403	}
daniele	29:1a47b7151851	1404
daniele	29:1a47b7151851	1405	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1406
daniele	29:1a47b7151851	1407	if ((s->state & PICO_SOCKET_STATE_BOUND) == 0) {
daniele	29:1a47b7151851	1408	return NULL;
daniele	29:1a47b7151851	1409	}
daniele	29:1a47b7151851	1410
daniele	29:1a47b7151851	1411	if (PROTO(s) == PICO_PROTO_UDP) {
daniele	29:1a47b7151851	1412	return NULL;
daniele	29:1a47b7151851	1413	}
daniele	29:1a47b7151851	1414
daniele	29:1a47b7151851	1415	if (TCPSTATE(s) == PICO_SOCKET_STATE_TCP_LISTEN) {
daniele	29:1a47b7151851	1416	struct pico_sockport *sp = pico_get_sockport(PICO_PROTO_TCP, s->local_port);
daniele	29:1a47b7151851	1417	struct pico_socket *found;
daniele	29:1a47b7151851	1418	/* If at this point no incoming connection socket is found,
daniele	29:1a47b7151851	1419	* the accept call is valid, but no connection is established yet.
daniele	29:1a47b7151851	1420	*/
daniele	29:1a47b7151851	1421	pico_err = PICO_ERR_EAGAIN;
daniele	29:1a47b7151851	1422	if (sp) {
daniele	29:1a47b7151851	1423	struct pico_tree_node * index;
daniele	29:1a47b7151851	1424	//RB_FOREACH(found, socket_tree, &sp->socks) {
daniele	29:1a47b7151851	1425	pico_tree_foreach(index,&sp->socks){
daniele	29:1a47b7151851	1426	found = index->keyValue;
daniele	29:1a47b7151851	1427	if (s == found->parent) {
daniele	29:1a47b7151851	1428	found->parent = NULL;
daniele	29:1a47b7151851	1429	pico_err = PICO_ERR_NOERR;
daniele	29:1a47b7151851	1430	memcpy(orig, &found->remote_addr, sizeof(struct pico_ip4));
daniele	29:1a47b7151851	1431	*port = found->remote_port;
daniele	29:1a47b7151851	1432	return found;
daniele	29:1a47b7151851	1433	}
daniele	29:1a47b7151851	1434	}
daniele	29:1a47b7151851	1435	}
daniele	29:1a47b7151851	1436	}
daniele	29:1a47b7151851	1437	return NULL;
daniele	29:1a47b7151851	1438	}
daniele	29:1a47b7151851	1439
daniele	29:1a47b7151851	1440	#else
daniele	29:1a47b7151851	1441
daniele	29:1a47b7151851	1442	int pico_socket_listen(struct pico_socket *s, int backlog)
daniele	29:1a47b7151851	1443	{
daniele	29:1a47b7151851	1444	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1445	return -1;
daniele	29:1a47b7151851	1446	}
daniele	29:1a47b7151851	1447
daniele	29:1a47b7151851	1448	struct pico_socket pico_socket_accept(struct pico_socket s, void orig, uint16_t local_port)
daniele	29:1a47b7151851	1449	{
daniele	29:1a47b7151851	1450	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1451	return NULL;
daniele	29:1a47b7151851	1452	}
daniele	29:1a47b7151851	1453
daniele	29:1a47b7151851	1454	#endif
daniele	29:1a47b7151851	1455
daniele	29:1a47b7151851	1456	#define PICO_SOCKET_SETOPT_EN(socket,index) (socket->opt_flags \|= (1 << index))
daniele	29:1a47b7151851	1457	#define PICO_SOCKET_SETOPT_DIS(socket,index) (socket->opt_flags &= ~(1 << index))
daniele	29:1a47b7151851	1458
daniele	29:1a47b7151851	1459	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	1460	{
daniele	29:1a47b7151851	1461	if (s == NULL) {
daniele	29:1a47b7151851	1462	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1463	return -1;
daniele	29:1a47b7151851	1464	}
daniele	29:1a47b7151851	1465
daniele	29:1a47b7151851	1466	pico_err = PICO_ERR_NOERR;
daniele	29:1a47b7151851	1467
daniele	29:1a47b7151851	1468	switch (option)
daniele	29:1a47b7151851	1469	{
daniele	29:1a47b7151851	1470	#ifdef PICO_SUPPORT_TCP
daniele	29:1a47b7151851	1471	case PICO_TCP_NODELAY:
daniele	29:1a47b7151851	1472	if (!value) {
daniele	29:1a47b7151851	1473	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1474	return -1;
daniele	29:1a47b7151851	1475	}
daniele	29:1a47b7151851	1476	if (s->proto->proto_number == PICO_PROTO_TCP) {
daniele	29:1a47b7151851	1477	int val = (int)value;
daniele	29:1a47b7151851	1478	if (*val > 0) {
daniele	29:1a47b7151851	1479	dbg("setsockopt: Nagle algorithm disabled.\n");
daniele	29:1a47b7151851	1480	PICO_SOCKET_SETOPT_EN(s,PICO_SOCKET_OPT_TCPNODELAY);
daniele	29:1a47b7151851	1481	} else {
daniele	29:1a47b7151851	1482	dbg("setsockopt: Nagle algorithm enabled.\n");
daniele	29:1a47b7151851	1483	PICO_SOCKET_SETOPT_DIS(s,PICO_SOCKET_OPT_TCPNODELAY);
daniele	29:1a47b7151851	1484	}
daniele	29:1a47b7151851	1485	} else {
daniele	29:1a47b7151851	1486	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1487	}
daniele	29:1a47b7151851	1488	break;
daniele	29:1a47b7151851	1489	#endif
daniele	29:1a47b7151851	1490
daniele	29:1a47b7151851	1491
daniele	29:1a47b7151851	1492	#ifdef PICO_SUPPORT_MCAST
daniele	29:1a47b7151851	1493	case PICO_IP_MULTICAST_IF:
daniele	29:1a47b7151851	1494	pico_err = PICO_ERR_EOPNOTSUPP;
daniele	29:1a47b7151851	1495	return -1;
daniele	29:1a47b7151851	1496	break;
daniele	29:1a47b7151851	1497
daniele	29:1a47b7151851	1498	case PICO_IP_MULTICAST_TTL:
daniele	29:1a47b7151851	1499	if (s->proto->proto_number == PICO_PROTO_UDP) {
daniele	29:1a47b7151851	1500	return pico_udp_set_mc_ttl(s, ((uint8_t ) value));
daniele	29:1a47b7151851	1501	}
daniele	29:1a47b7151851	1502	break;
daniele	29:1a47b7151851	1503
daniele	29:1a47b7151851	1504	case PICO_IP_MULTICAST_LOOP:
daniele	29:1a47b7151851	1505	if (s->proto->proto_number == PICO_PROTO_UDP) {
daniele	29:1a47b7151851	1506	switch ((uint8_t ) value)
daniele	29:1a47b7151851	1507	{
daniele	29:1a47b7151851	1508	case 0:
daniele	29:1a47b7151851	1509	/* do not loop back multicast datagram */
daniele	29:1a47b7151851	1510	PICO_SOCKET_SETOPT_DIS(s,PICO_SOCKET_OPT_MULTICAST_LOOP);
daniele	29:1a47b7151851	1511	break;
daniele	29:1a47b7151851	1512
daniele	29:1a47b7151851	1513	case 1:
daniele	29:1a47b7151851	1514	/* do loop back multicast datagram */
daniele	29:1a47b7151851	1515	PICO_SOCKET_SETOPT_EN(s,PICO_SOCKET_OPT_MULTICAST_LOOP);
daniele	29:1a47b7151851	1516	break;
daniele	29:1a47b7151851	1517
daniele	29:1a47b7151851	1518	default:
daniele	29:1a47b7151851	1519	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1520	return -1;
daniele	29:1a47b7151851	1521	}
daniele	29:1a47b7151851	1522	}
daniele	29:1a47b7151851	1523	break;
daniele	29:1a47b7151851	1524
daniele	29:1a47b7151851	1525	case PICO_IP_ADD_MEMBERSHIP:
daniele	29:1a47b7151851	1526	/* EXCLUDE mode */
daniele	29:1a47b7151851	1527	if (s->proto->proto_number == PICO_PROTO_UDP) {
tass	48:40fc4462265c	1528	int filter_mode;
daniele	29:1a47b7151851	1529	struct pico_ip_mreq *mreq = NULL;
daniele	29:1a47b7151851	1530	struct pico_mcast_listen *listen = NULL, ltest = {0};
daniele	29:1a47b7151851	1531	struct pico_ipv4_link *mcast_link = NULL;
daniele	29:1a47b7151851	1532
daniele	29:1a47b7151851	1533	if (!value) {
daniele	29:1a47b7151851	1534	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1535	return -1;
daniele	29:1a47b7151851	1536	}
daniele	29:1a47b7151851	1537	mreq = (struct pico_ip_mreq *) value;
daniele	29:1a47b7151851	1538	mcast_link = pico_socket_setoption_mcastargs_validation(mreq, NULL);
daniele	29:1a47b7151851	1539	if (!mcast_link) {
daniele	29:1a47b7151851	1540	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1541	return -1;
daniele	29:1a47b7151851	1542	}
daniele	29:1a47b7151851	1543	if (!mreq->mcast_link_addr.addr)
daniele	29:1a47b7151851	1544	mreq->mcast_link_addr.addr = mcast_link->address.addr;
daniele	29:1a47b7151851	1545
daniele	29:1a47b7151851	1546	if (!s->MCASTListen) { /* No RBTree allocated yet */
daniele	29:1a47b7151851	1547	s->MCASTListen = pico_zalloc(sizeof(struct pico_tree));
daniele	29:1a47b7151851	1548	if (!s->MCASTListen) {
daniele	29:1a47b7151851	1549	pico_err = PICO_ERR_ENOMEM;
daniele	29:1a47b7151851	1550	return -1;
daniele	29:1a47b7151851	1551	}
daniele	29:1a47b7151851	1552	s->MCASTListen->root = &LEAF;
daniele	29:1a47b7151851	1553	s->MCASTListen->compare = mcast_listen_cmp;
daniele	29:1a47b7151851	1554	}
daniele	29:1a47b7151851	1555	ltest.mcast_link = mreq->mcast_link_addr;
daniele	29:1a47b7151851	1556	ltest.mcast_group = mreq->mcast_group_addr;
daniele	29:1a47b7151851	1557	listen = pico_tree_findKey(s->MCASTListen, &ltest);
daniele	29:1a47b7151851	1558	if (listen) {
daniele	29:1a47b7151851	1559	if (listen->filter_mode != PICO_IP_MULTICAST_EXCLUDE) {
daniele	29:1a47b7151851	1560	so_mcast_dbg("pico_socket_setoption: ERROR any-source multicast (exclude) on source-specific multicast (include)\n");
daniele	29:1a47b7151851	1561	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1562	return -1;
daniele	29:1a47b7151851	1563	} else {
daniele	29:1a47b7151851	1564	so_mcast_dbg("pico_socket_setoption: ERROR duplicate PICO_IP_ADD_MEMBERSHIP\n");
daniele	29:1a47b7151851	1565	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1566	return -1;
daniele	29:1a47b7151851	1567	}
daniele	29:1a47b7151851	1568	} else {
daniele	29:1a47b7151851	1569	listen = pico_zalloc(sizeof(struct pico_mcast_listen));
daniele	29:1a47b7151851	1570	if (!listen) {
daniele	29:1a47b7151851	1571	pico_err = PICO_ERR_ENOMEM;
daniele	29:1a47b7151851	1572	return -1;
daniele	29:1a47b7151851	1573	}
daniele	29:1a47b7151851	1574	listen->filter_mode = PICO_IP_MULTICAST_EXCLUDE;
daniele	29:1a47b7151851	1575	listen->mcast_link = mreq->mcast_link_addr;
daniele	29:1a47b7151851	1576	listen->mcast_group = mreq->mcast_group_addr;
daniele	29:1a47b7151851	1577	listen->MCASTSources.root = &LEAF;
daniele	29:1a47b7151851	1578	listen->MCASTSources.compare = mcast_sources_cmp;
daniele	29:1a47b7151851	1579	pico_tree_insert(s->MCASTListen, listen);
daniele	29:1a47b7151851	1580	}
daniele	29:1a47b7151851	1581
daniele	29:1a47b7151851	1582	pico_tree_insert(&MCASTSockets, s);
daniele	29:1a47b7151851	1583	filter_mode = pico_socket_aggregate_mcastfilters(&mcast_link->address, &mreq->mcast_group_addr);
daniele	29:1a47b7151851	1584	if (filter_mode < 0)
daniele	29:1a47b7151851	1585	return -1;
daniele	29:1a47b7151851	1586
daniele	29:1a47b7151851	1587	return pico_ipv4_mcast_join(&mreq->mcast_link_addr, &mreq->mcast_group_addr, 1, filter_mode, &MCASTFilter);
daniele	29:1a47b7151851	1588	}
daniele	29:1a47b7151851	1589	break;
daniele	29:1a47b7151851	1590
daniele	29:1a47b7151851	1591	case PICO_IP_DROP_MEMBERSHIP:
daniele	29:1a47b7151851	1592	/* EXCLUDE mode */
daniele	29:1a47b7151851	1593	if (s->proto->proto_number == PICO_PROTO_UDP) {
tass	48:40fc4462265c	1594	int filter_mode = 0;
daniele	29:1a47b7151851	1595	struct pico_ip_mreq *mreq = NULL;
daniele	29:1a47b7151851	1596	struct pico_mcast_listen *listen = NULL, ltest = {0};
daniele	29:1a47b7151851	1597	struct pico_ip4 *source = NULL;
daniele	29:1a47b7151851	1598	struct pico_ipv4_link *mcast_link = NULL;
daniele	29:1a47b7151851	1599	struct pico_tree_node index, _tmp;
daniele	29:1a47b7151851	1600
daniele	29:1a47b7151851	1601	if (!value) {
daniele	29:1a47b7151851	1602	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1603	return -1;
daniele	29:1a47b7151851	1604	}
daniele	29:1a47b7151851	1605	mreq = (struct pico_ip_mreq *) value;
daniele	29:1a47b7151851	1606	mcast_link = pico_socket_setoption_mcastargs_validation(mreq, NULL);
daniele	29:1a47b7151851	1607	if (!mcast_link) {
daniele	29:1a47b7151851	1608	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1609	return -1;
daniele	29:1a47b7151851	1610	}
daniele	29:1a47b7151851	1611	if (!mreq->mcast_link_addr.addr)
daniele	29:1a47b7151851	1612	mreq->mcast_link_addr.addr = mcast_link->address.addr;
daniele	29:1a47b7151851	1613
daniele	29:1a47b7151851	1614	if (!s->MCASTListen) { /* No RBTree allocated yet */
daniele	29:1a47b7151851	1615	so_mcast_dbg("socket_setoption: ERROR PICO_IP_DROP_MEMBERSHIP before any PICO_IP_ADD_MEMBERSHIP/SOURCE_MEMBERSHIP\n");
daniele	29:1a47b7151851	1616	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1617	return -1;
daniele	29:1a47b7151851	1618	}
daniele	29:1a47b7151851	1619	ltest.mcast_link = mreq->mcast_link_addr;
daniele	29:1a47b7151851	1620	ltest.mcast_group = mreq->mcast_group_addr;
daniele	29:1a47b7151851	1621	listen = pico_tree_findKey(s->MCASTListen, &ltest);
daniele	29:1a47b7151851	1622	if (!listen) {
daniele	29:1a47b7151851	1623	so_mcast_dbg("pico_socket_setoption: ERROR PICO_IP_DROP_MEMBERSHIP before PICO_IP_ADD_MEMBERSHIP/SOURCE_MEMBERSHIP\n");
daniele	29:1a47b7151851	1624	pico_err = PICO_ERR_EADDRNOTAVAIL;
daniele	29:1a47b7151851	1625	return -1;
daniele	29:1a47b7151851	1626	} else {
daniele	29:1a47b7151851	1627	pico_tree_foreach_safe(index, &listen->MCASTSources, _tmp)
daniele	29:1a47b7151851	1628	{
daniele	29:1a47b7151851	1629	source = index->keyValue;
daniele	29:1a47b7151851	1630	pico_tree_delete(&listen->MCASTSources, source);
daniele	29:1a47b7151851	1631	pico_free(source);
daniele	29:1a47b7151851	1632	}
daniele	29:1a47b7151851	1633	pico_tree_delete(s->MCASTListen, listen);
daniele	29:1a47b7151851	1634	pico_free(listen);
daniele	29:1a47b7151851	1635	if (pico_tree_empty(s->MCASTListen)) {
daniele	29:1a47b7151851	1636	pico_free(s->MCASTListen);
daniele	29:1a47b7151851	1637	s->MCASTListen = NULL;
daniele	29:1a47b7151851	1638	pico_tree_delete(&MCASTSockets, s);
daniele	29:1a47b7151851	1639	}
daniele	29:1a47b7151851	1640	}
daniele	29:1a47b7151851	1641
daniele	29:1a47b7151851	1642	filter_mode = pico_socket_aggregate_mcastfilters(&mcast_link->address, &mreq->mcast_group_addr);
daniele	29:1a47b7151851	1643	if (filter_mode < 0)
daniele	29:1a47b7151851	1644	return -1;
daniele	29:1a47b7151851	1645
daniele	29:1a47b7151851	1646	return pico_ipv4_mcast_leave(&mreq->mcast_link_addr, &mreq->mcast_group_addr, 1, filter_mode, &MCASTFilter);
daniele	29:1a47b7151851	1647	}
daniele	29:1a47b7151851	1648	break;
daniele	29:1a47b7151851	1649
daniele	29:1a47b7151851	1650	case PICO_IP_UNBLOCK_SOURCE:
daniele	29:1a47b7151851	1651	/* EXCLUDE mode */
daniele	29:1a47b7151851	1652	if (s->proto->proto_number == PICO_PROTO_UDP) {
tass	48:40fc4462265c	1653	int filter_mode = 0;
daniele	29:1a47b7151851	1654	struct pico_ip_mreq_source *mreq = NULL;
daniele	29:1a47b7151851	1655	struct pico_mcast_listen *listen = NULL, ltest = {0};
daniele	29:1a47b7151851	1656	struct pico_ip4 *source = NULL, stest = {0};
daniele	29:1a47b7151851	1657	struct pico_ipv4_link *mcast_link = NULL;
daniele	29:1a47b7151851	1658
daniele	29:1a47b7151851	1659	if (!value) {
daniele	29:1a47b7151851	1660	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1661	return -1;
daniele	29:1a47b7151851	1662	}
daniele	29:1a47b7151851	1663	mreq = (struct pico_ip_mreq_source *) value;
daniele	29:1a47b7151851	1664	mcast_link = pico_socket_setoption_mcastargs_validation(NULL, mreq);
daniele	29:1a47b7151851	1665	if (!mcast_link) {
daniele	29:1a47b7151851	1666	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1667	return -1;
daniele	29:1a47b7151851	1668	}
daniele	29:1a47b7151851	1669	if (!mreq->mcast_link_addr.addr)
daniele	29:1a47b7151851	1670	mreq->mcast_link_addr.addr = mcast_link->address.addr;
daniele	29:1a47b7151851	1671
daniele	29:1a47b7151851	1672	if (!s->MCASTListen) { /* No RBTree allocated yet */
daniele	29:1a47b7151851	1673	so_mcast_dbg("pico_socket_setoption: ERROR PICO_IP_UNBLOCK_SOURCE before any PICO_IP_ADD_MEMBERSHIP\n");
daniele	29:1a47b7151851	1674	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1675	return -1;
daniele	29:1a47b7151851	1676	}
daniele	29:1a47b7151851	1677	ltest.mcast_link = mreq->mcast_link_addr;
daniele	29:1a47b7151851	1678	ltest.mcast_group = mreq->mcast_group_addr;
daniele	29:1a47b7151851	1679	listen = pico_tree_findKey(s->MCASTListen, &ltest);
daniele	29:1a47b7151851	1680	if (!listen) {
daniele	29:1a47b7151851	1681	so_mcast_dbg("pico_socket_setoption: ERROR PICO_IP_UNBLOCK_SOURCE before PICO_IP_ADD_MEMBERSHIP\n");
daniele	29:1a47b7151851	1682	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1683	return -1;
daniele	29:1a47b7151851	1684	} else {
daniele	29:1a47b7151851	1685	if (listen->filter_mode != PICO_IP_MULTICAST_EXCLUDE) {
daniele	29:1a47b7151851	1686	so_mcast_dbg("pico_socket_setoption: ERROR any-source multicast (exclude) on source-specific multicast (include)\n");
daniele	29:1a47b7151851	1687	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1688	return -1;
daniele	29:1a47b7151851	1689	}
daniele	29:1a47b7151851	1690	stest.addr = mreq->mcast_source_addr.addr;
daniele	29:1a47b7151851	1691	source = pico_tree_findKey(&listen->MCASTSources, &stest);
daniele	29:1a47b7151851	1692	if (!source) {
daniele	29:1a47b7151851	1693	so_mcast_dbg("pico_socket_setoption: ERROR address to unblock not in source list\n");
daniele	29:1a47b7151851	1694	pico_err = PICO_ERR_EADDRNOTAVAIL;
daniele	29:1a47b7151851	1695	return -1;
daniele	29:1a47b7151851	1696	} else {
daniele	29:1a47b7151851	1697	pico_tree_delete(&listen->MCASTSources, source);
daniele	29:1a47b7151851	1698	pico_free(source);
daniele	29:1a47b7151851	1699	}
daniele	29:1a47b7151851	1700	}
daniele	29:1a47b7151851	1701
daniele	29:1a47b7151851	1702	filter_mode = pico_socket_aggregate_mcastfilters(&mcast_link->address, &mreq->mcast_group_addr);
daniele	29:1a47b7151851	1703	if (filter_mode < 0)
daniele	29:1a47b7151851	1704	return -1;
daniele	29:1a47b7151851	1705
daniele	29:1a47b7151851	1706	return pico_ipv4_mcast_leave(&mreq->mcast_link_addr, &mreq->mcast_group_addr, 0, filter_mode, &MCASTFilter);
daniele	29:1a47b7151851	1707	}
daniele	29:1a47b7151851	1708	break;
daniele	29:1a47b7151851	1709
daniele	29:1a47b7151851	1710	case PICO_IP_BLOCK_SOURCE:
daniele	29:1a47b7151851	1711	/* EXCLUDE mode */
daniele	29:1a47b7151851	1712	if (s->proto->proto_number == PICO_PROTO_UDP) {
tass	48:40fc4462265c	1713	int filter_mode = 0;
daniele	29:1a47b7151851	1714	struct pico_ip_mreq_source *mreq = NULL;
daniele	29:1a47b7151851	1715	struct pico_mcast_listen *listen = NULL, ltest = {0};
daniele	29:1a47b7151851	1716	struct pico_ip4 *source = NULL, stest = {0};
daniele	29:1a47b7151851	1717	struct pico_ipv4_link *mcast_link = NULL;
daniele	29:1a47b7151851	1718
daniele	29:1a47b7151851	1719	if (!value) {
daniele	29:1a47b7151851	1720	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1721	return -1;
daniele	29:1a47b7151851	1722	}
daniele	29:1a47b7151851	1723	mreq = (struct pico_ip_mreq_source *) value;
daniele	29:1a47b7151851	1724	mcast_link = pico_socket_setoption_mcastargs_validation(NULL, mreq);
daniele	29:1a47b7151851	1725	if (!mcast_link) {
daniele	29:1a47b7151851	1726	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1727	return -1;
daniele	29:1a47b7151851	1728	}
daniele	29:1a47b7151851	1729	if (!mreq->mcast_link_addr.addr)
daniele	29:1a47b7151851	1730	mreq->mcast_link_addr.addr = mcast_link->address.addr;
daniele	29:1a47b7151851	1731
daniele	29:1a47b7151851	1732	if (!s->MCASTListen) { /* No RBTree allocated yet */
daniele	29:1a47b7151851	1733	so_mcast_dbg("pico_socket_setoption: ERROR PICO_IP_BLOCK_SOURCE before any PICO_IP_ADD_MEMBERSHIP\n");
daniele	29:1a47b7151851	1734	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1735	return -1;
daniele	29:1a47b7151851	1736	}
daniele	29:1a47b7151851	1737	ltest.mcast_link = mreq->mcast_link_addr;
daniele	29:1a47b7151851	1738	ltest.mcast_group = mreq->mcast_group_addr;
daniele	29:1a47b7151851	1739	listen = pico_tree_findKey(s->MCASTListen, &ltest);
daniele	29:1a47b7151851	1740	if (!listen) {
daniele	29:1a47b7151851	1741	so_mcast_dbg("pico_socket_setoption: ERROR PICO_IP_BLOCK_SOURCE before PICO_IP_ADD_MEMBERSHIP\n");
daniele	29:1a47b7151851	1742	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1743	return -1;
daniele	29:1a47b7151851	1744	} else {
daniele	29:1a47b7151851	1745	if (listen->filter_mode != PICO_IP_MULTICAST_EXCLUDE) {
daniele	29:1a47b7151851	1746	so_mcast_dbg("pico_socket_setoption: ERROR any-source multicast (exclude) on source-specific multicast (include)\n");
daniele	29:1a47b7151851	1747	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1748	return -1;
daniele	29:1a47b7151851	1749	}
daniele	29:1a47b7151851	1750	stest.addr = mreq->mcast_source_addr.addr;
daniele	29:1a47b7151851	1751	source = pico_tree_findKey(&listen->MCASTSources, &stest);
daniele	29:1a47b7151851	1752	if (source) {
daniele	29:1a47b7151851	1753	so_mcast_dbg("pico_socket_setoption: ERROR address to block already in source list\n");
daniele	29:1a47b7151851	1754	pico_err = PICO_ERR_EADDRNOTAVAIL;
daniele	29:1a47b7151851	1755	return -1;
daniele	29:1a47b7151851	1756	} else {
daniele	29:1a47b7151851	1757	source = pico_zalloc(sizeof(struct pico_ip4));
daniele	29:1a47b7151851	1758	if (!source) {
daniele	29:1a47b7151851	1759	pico_err = PICO_ERR_ENOMEM;
daniele	29:1a47b7151851	1760	return -1;
daniele	29:1a47b7151851	1761	}
daniele	29:1a47b7151851	1762	source->addr = mreq->mcast_source_addr.addr;
daniele	29:1a47b7151851	1763	pico_tree_insert(&listen->MCASTSources, source);
daniele	29:1a47b7151851	1764	}
daniele	29:1a47b7151851	1765	}
daniele	29:1a47b7151851	1766
daniele	29:1a47b7151851	1767	filter_mode = pico_socket_aggregate_mcastfilters(&mcast_link->address, &mreq->mcast_group_addr);
daniele	29:1a47b7151851	1768	if (filter_mode < 0)
daniele	29:1a47b7151851	1769	return -1;
daniele	29:1a47b7151851	1770
daniele	29:1a47b7151851	1771	return pico_ipv4_mcast_join(&mreq->mcast_link_addr, &mreq->mcast_group_addr, 0, filter_mode, &MCASTFilter);
daniele	29:1a47b7151851	1772	}
daniele	29:1a47b7151851	1773	break;
daniele	29:1a47b7151851	1774
daniele	29:1a47b7151851	1775	case PICO_IP_ADD_SOURCE_MEMBERSHIP:
daniele	29:1a47b7151851	1776	/* INCLUDE mode */
daniele	29:1a47b7151851	1777	if (s->proto->proto_number == PICO_PROTO_UDP) {
tass	48:40fc4462265c	1778	int filter_mode = 0, reference_count = 0;
daniele	29:1a47b7151851	1779	struct pico_ip_mreq_source *mreq = NULL;
daniele	29:1a47b7151851	1780	struct pico_mcast_listen *listen = NULL, ltest = {0};
daniele	29:1a47b7151851	1781	struct pico_ip4 *source = NULL, stest = {0};
daniele	29:1a47b7151851	1782	struct pico_ipv4_link *mcast_link = NULL;
daniele	29:1a47b7151851	1783
daniele	29:1a47b7151851	1784	if (!value) {
daniele	29:1a47b7151851	1785	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1786	return -1;
daniele	29:1a47b7151851	1787	}
daniele	29:1a47b7151851	1788	mreq = (struct pico_ip_mreq_source *) value;
daniele	29:1a47b7151851	1789	mcast_link = pico_socket_setoption_mcastargs_validation(NULL, mreq);
daniele	29:1a47b7151851	1790	if (!mcast_link) {
daniele	29:1a47b7151851	1791	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1792	return -1;
daniele	29:1a47b7151851	1793	}
daniele	29:1a47b7151851	1794	if (!mreq->mcast_link_addr.addr)
daniele	29:1a47b7151851	1795	mreq->mcast_link_addr.addr = mcast_link->address.addr;
daniele	29:1a47b7151851	1796
daniele	29:1a47b7151851	1797	if (!s->MCASTListen) { /* No RBTree allocated yet */
daniele	29:1a47b7151851	1798	s->MCASTListen = pico_zalloc(sizeof(struct pico_tree));
daniele	29:1a47b7151851	1799	if (!s->MCASTListen) {
daniele	29:1a47b7151851	1800	pico_err = PICO_ERR_ENOMEM;
daniele	29:1a47b7151851	1801	return -1;
daniele	29:1a47b7151851	1802	}
daniele	29:1a47b7151851	1803	s->MCASTListen->root = &LEAF;
daniele	29:1a47b7151851	1804	s->MCASTListen->compare = mcast_listen_cmp;
daniele	29:1a47b7151851	1805	}
daniele	29:1a47b7151851	1806	ltest.mcast_link = mreq->mcast_link_addr;
daniele	29:1a47b7151851	1807	ltest.mcast_group = mreq->mcast_group_addr;
daniele	29:1a47b7151851	1808	listen = pico_tree_findKey(s->MCASTListen, &ltest);
daniele	29:1a47b7151851	1809	if (listen) {
daniele	29:1a47b7151851	1810	if (listen->filter_mode != PICO_IP_MULTICAST_INCLUDE) {
daniele	29:1a47b7151851	1811	so_mcast_dbg("pico_socket_setoption: ERROR source-specific multicast (include) on any-source multicast (exclude)\n");
daniele	29:1a47b7151851	1812	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1813	return -1;
daniele	29:1a47b7151851	1814	}
daniele	29:1a47b7151851	1815	stest.addr = mreq->mcast_source_addr.addr;
daniele	29:1a47b7151851	1816	source = pico_tree_findKey(&listen->MCASTSources, &stest);
daniele	29:1a47b7151851	1817	if (source) {
daniele	29:1a47b7151851	1818	so_mcast_dbg("pico_socket_setoption: ERROR source address to allow already in source list\n");
daniele	29:1a47b7151851	1819	pico_err = PICO_ERR_EADDRNOTAVAIL;
daniele	29:1a47b7151851	1820	return -1;
daniele	29:1a47b7151851	1821	} else {
daniele	29:1a47b7151851	1822	source = pico_zalloc(sizeof(struct pico_ip4));
daniele	29:1a47b7151851	1823	if (!source) {
daniele	29:1a47b7151851	1824	pico_err = PICO_ERR_ENOMEM;
daniele	29:1a47b7151851	1825	return -1;
daniele	29:1a47b7151851	1826	}
daniele	29:1a47b7151851	1827	source->addr = mreq->mcast_source_addr.addr;
daniele	29:1a47b7151851	1828	pico_tree_insert(&listen->MCASTSources, source);
daniele	29:1a47b7151851	1829	}
daniele	29:1a47b7151851	1830	} else {
daniele	29:1a47b7151851	1831	listen = pico_zalloc(sizeof(struct pico_mcast_listen));
daniele	29:1a47b7151851	1832	if (!listen) {
daniele	29:1a47b7151851	1833	pico_err = PICO_ERR_ENOMEM;
daniele	29:1a47b7151851	1834	return -1;
daniele	29:1a47b7151851	1835	}
daniele	29:1a47b7151851	1836	listen->filter_mode = PICO_IP_MULTICAST_INCLUDE;
daniele	29:1a47b7151851	1837	listen->mcast_link = mreq->mcast_link_addr;
daniele	29:1a47b7151851	1838	listen->mcast_group = mreq->mcast_group_addr;
daniele	29:1a47b7151851	1839	listen->MCASTSources.root = &LEAF;
daniele	29:1a47b7151851	1840	listen->MCASTSources.compare = mcast_sources_cmp;
daniele	29:1a47b7151851	1841	source = pico_zalloc(sizeof(struct pico_ip4));
daniele	29:1a47b7151851	1842	if (!source) {
daniele	29:1a47b7151851	1843	pico_free(listen);
daniele	29:1a47b7151851	1844	pico_err = PICO_ERR_ENOMEM;
daniele	29:1a47b7151851	1845	return -1;
daniele	29:1a47b7151851	1846	}
daniele	29:1a47b7151851	1847	source->addr = mreq->mcast_source_addr.addr;
daniele	29:1a47b7151851	1848	pico_tree_insert(&listen->MCASTSources, source);
daniele	29:1a47b7151851	1849	pico_tree_insert(s->MCASTListen, listen);
daniele	29:1a47b7151851	1850	reference_count = 1;
daniele	29:1a47b7151851	1851	}
daniele	29:1a47b7151851	1852
daniele	29:1a47b7151851	1853	pico_tree_insert(&MCASTSockets, s);
daniele	29:1a47b7151851	1854	filter_mode = pico_socket_aggregate_mcastfilters(&mcast_link->address, &mreq->mcast_group_addr);
daniele	29:1a47b7151851	1855	if (filter_mode < 0)
daniele	29:1a47b7151851	1856	return -1;
daniele	29:1a47b7151851	1857
daniele	29:1a47b7151851	1858	return pico_ipv4_mcast_join(&mreq->mcast_link_addr, &mreq->mcast_group_addr, reference_count, filter_mode, &MCASTFilter);
daniele	29:1a47b7151851	1859	}
daniele	29:1a47b7151851	1860	break;
daniele	29:1a47b7151851	1861
daniele	29:1a47b7151851	1862	case PICO_IP_DROP_SOURCE_MEMBERSHIP:
daniele	29:1a47b7151851	1863	/* INCLUDE mode */
daniele	29:1a47b7151851	1864	if (s->proto->proto_number == PICO_PROTO_UDP) {
tass	48:40fc4462265c	1865	int filter_mode = 0, reference_count = 0;
daniele	29:1a47b7151851	1866	struct pico_ip_mreq_source *mreq = NULL;
daniele	29:1a47b7151851	1867	struct pico_mcast_listen *listen = NULL, ltest = {0};
daniele	29:1a47b7151851	1868	struct pico_ip4 *source = NULL, stest = {0};
daniele	29:1a47b7151851	1869	struct pico_ipv4_link *mcast_link = NULL;
daniele	29:1a47b7151851	1870
daniele	29:1a47b7151851	1871	if (!value) {
daniele	29:1a47b7151851	1872	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1873	return -1;
daniele	29:1a47b7151851	1874	}
daniele	29:1a47b7151851	1875	mreq = (struct pico_ip_mreq_source *) value;
daniele	29:1a47b7151851	1876	mcast_link = pico_socket_setoption_mcastargs_validation(NULL, mreq);
daniele	29:1a47b7151851	1877	if (!mcast_link) {
daniele	29:1a47b7151851	1878	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1879	return -1;
daniele	29:1a47b7151851	1880	}
daniele	29:1a47b7151851	1881	if (!mreq->mcast_link_addr.addr)
daniele	29:1a47b7151851	1882	mreq->mcast_link_addr.addr = mcast_link->address.addr;
daniele	29:1a47b7151851	1883
daniele	29:1a47b7151851	1884	if (!s->MCASTListen) { /* No RBTree allocated yet */
daniele	29:1a47b7151851	1885	so_mcast_dbg("pico_socket_setoption: ERROR PICO_IP_DROP_SOURCE_MEMBERSHIP before any PICO_IP_ADD_SOURCE_MEMBERSHIP\n");
daniele	29:1a47b7151851	1886	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1887	return -1;
daniele	29:1a47b7151851	1888	}
daniele	29:1a47b7151851	1889	ltest.mcast_link = mreq->mcast_link_addr;
daniele	29:1a47b7151851	1890	ltest.mcast_group = mreq->mcast_group_addr;
daniele	29:1a47b7151851	1891	listen = pico_tree_findKey(s->MCASTListen, &ltest);
daniele	29:1a47b7151851	1892	if (!listen) {
daniele	29:1a47b7151851	1893	so_mcast_dbg("pico_socket_setoption: ERROR PICO_IP_DROP_SOURCE_MEMBERSHIP before PICO_IP_ADD_SOURCE_MEMBERSHIP\n");
daniele	29:1a47b7151851	1894	pico_err = PICO_ERR_EADDRNOTAVAIL;
daniele	29:1a47b7151851	1895	return -1;
daniele	29:1a47b7151851	1896	} else {
daniele	29:1a47b7151851	1897	if (listen->filter_mode != PICO_IP_MULTICAST_INCLUDE) {
daniele	29:1a47b7151851	1898	so_mcast_dbg("pico_socket_setoption: ERROR source-specific multicast (include) on any-source multicast (exclude)\n");
daniele	29:1a47b7151851	1899	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1900	return -1;
daniele	29:1a47b7151851	1901	}
daniele	29:1a47b7151851	1902	stest.addr = mreq->mcast_source_addr.addr;
daniele	29:1a47b7151851	1903	source = pico_tree_findKey(&listen->MCASTSources, &stest);
daniele	29:1a47b7151851	1904	if (!source) {
daniele	29:1a47b7151851	1905	so_mcast_dbg("pico_socket_setoption: ERROR address to drop not in source list\n");
daniele	29:1a47b7151851	1906	pico_err = PICO_ERR_EADDRNOTAVAIL;
daniele	29:1a47b7151851	1907	return -1;
daniele	29:1a47b7151851	1908	} else {
daniele	29:1a47b7151851	1909	pico_tree_delete(&listen->MCASTSources, source);
daniele	29:1a47b7151851	1910	pico_free(source);
daniele	29:1a47b7151851	1911	if (pico_tree_empty(&listen->MCASTSources)) { /* 1 if empty, 0 otherwise */
daniele	29:1a47b7151851	1912	reference_count = 1;
daniele	29:1a47b7151851	1913	pico_tree_delete(s->MCASTListen, listen);
daniele	29:1a47b7151851	1914	pico_free(listen);
daniele	29:1a47b7151851	1915	if (pico_tree_empty(s->MCASTListen)) {
daniele	29:1a47b7151851	1916	pico_free(s->MCASTListen);
daniele	29:1a47b7151851	1917	s->MCASTListen = NULL;
daniele	29:1a47b7151851	1918	pico_tree_delete(&MCASTSockets, s);
daniele	29:1a47b7151851	1919	}
daniele	29:1a47b7151851	1920	}
daniele	29:1a47b7151851	1921	}
daniele	29:1a47b7151851	1922	}
daniele	29:1a47b7151851	1923
daniele	29:1a47b7151851	1924	filter_mode = pico_socket_aggregate_mcastfilters(&mcast_link->address, &mreq->mcast_group_addr);
daniele	29:1a47b7151851	1925	if (filter_mode < 0)
daniele	29:1a47b7151851	1926	return -1;
daniele	29:1a47b7151851	1927
daniele	29:1a47b7151851	1928	return pico_ipv4_mcast_leave(&mreq->mcast_link_addr, &mreq->mcast_group_addr, reference_count, filter_mode, &MCASTFilter);
daniele	29:1a47b7151851	1929	}
daniele	29:1a47b7151851	1930	break;
daniele	29:1a47b7151851	1931	#endif /* PICO_SUPPORT_MCAST */
daniele	29:1a47b7151851	1932
daniele	29:1a47b7151851	1933	default:
daniele	29:1a47b7151851	1934	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1935	return -1;
daniele	29:1a47b7151851	1936	}
daniele	29:1a47b7151851	1937
daniele	29:1a47b7151851	1938	if (pico_err != PICO_ERR_NOERR)
daniele	29:1a47b7151851	1939	return -1;
daniele	29:1a47b7151851	1940	else
daniele	29:1a47b7151851	1941	return 0;
daniele	29:1a47b7151851	1942	}
daniele	29:1a47b7151851	1943
daniele	29:1a47b7151851	1944	#define PICO_SOCKET_GETOPT(socket,index) ((socket->opt_flags & (1 << index)) != 0)
daniele	29:1a47b7151851	1945
daniele	29:1a47b7151851	1946	int pico_socket_getoption(struct pico_socket s, int option, void value)
daniele	29:1a47b7151851	1947	{
daniele	29:1a47b7151851	1948	if (!s \|\| !value) {
daniele	29:1a47b7151851	1949	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1950	return -1;
daniele	29:1a47b7151851	1951	}
daniele	29:1a47b7151851	1952
daniele	29:1a47b7151851	1953	switch (option)
daniele	29:1a47b7151851	1954	{
daniele	29:1a47b7151851	1955	#ifdef PICO_SUPPORT_TCP
daniele	29:1a47b7151851	1956	case PICO_TCP_NODELAY:
daniele	29:1a47b7151851	1957	if (s->proto->proto_number == PICO_PROTO_TCP)
daniele	29:1a47b7151851	1958	/* state of the NODELAY option */
daniele	29:1a47b7151851	1959	(int )value = PICO_SOCKET_GETOPT(s,PICO_SOCKET_OPT_TCPNODELAY);
daniele	29:1a47b7151851	1960	else
daniele	29:1a47b7151851	1961	(int )value = 0;
daniele	29:1a47b7151851	1962	break;
daniele	29:1a47b7151851	1963	#endif
daniele	29:1a47b7151851	1964
daniele	29:1a47b7151851	1965	#ifdef PICO_SUPPORT_MCAST
daniele	29:1a47b7151851	1966	case PICO_IP_MULTICAST_IF:
daniele	29:1a47b7151851	1967	pico_err = PICO_ERR_EOPNOTSUPP;
daniele	29:1a47b7151851	1968	return -1;
daniele	29:1a47b7151851	1969	break;
daniele	29:1a47b7151851	1970
daniele	29:1a47b7151851	1971	case PICO_IP_MULTICAST_TTL:
daniele	29:1a47b7151851	1972	if (s->proto->proto_number == PICO_PROTO_UDP) {
daniele	29:1a47b7151851	1973	pico_udp_get_mc_ttl(s, (uint8_t *) value);
daniele	29:1a47b7151851	1974	} else {
daniele	29:1a47b7151851	1975	(uint8_t )value = 0;
daniele	29:1a47b7151851	1976	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1977	return -1;
daniele	29:1a47b7151851	1978	}
daniele	29:1a47b7151851	1979	break;
daniele	29:1a47b7151851	1980
daniele	29:1a47b7151851	1981	case PICO_IP_MULTICAST_LOOP:
daniele	29:1a47b7151851	1982	if (s->proto->proto_number == PICO_PROTO_UDP) {
daniele	29:1a47b7151851	1983	(uint8_t )value = PICO_SOCKET_GETOPT(s,PICO_SOCKET_OPT_MULTICAST_LOOP);
daniele	29:1a47b7151851	1984	} else {
daniele	29:1a47b7151851	1985	(uint8_t )value = 0;
daniele	29:1a47b7151851	1986	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1987	return -1;
daniele	29:1a47b7151851	1988	}
daniele	29:1a47b7151851	1989	break;
daniele	29:1a47b7151851	1990	#endif /* PICO_SUPPORT_MCAST */
daniele	29:1a47b7151851	1991
daniele	29:1a47b7151851	1992	default:
daniele	29:1a47b7151851	1993	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	1994	return -1;
daniele	29:1a47b7151851	1995	}
daniele	29:1a47b7151851	1996
daniele	29:1a47b7151851	1997	return 0;
daniele	29:1a47b7151851	1998	}
daniele	29:1a47b7151851	1999
daniele	29:1a47b7151851	2000
daniele	29:1a47b7151851	2001	int pico_socket_shutdown(struct pico_socket *s, int mode)
daniele	29:1a47b7151851	2002	{
daniele	29:1a47b7151851	2003	if (!s) {
daniele	29:1a47b7151851	2004	pico_err = PICO_ERR_EINVAL;
daniele	29:1a47b7151851	2005	return -1;
daniele	29:1a47b7151851	2006	} else {
daniele	29:1a47b7151851	2007	/* check if exists in tree */
daniele	29:1a47b7151851	2008	/* See task #178 */
daniele	29:1a47b7151851	2009	if (pico_check_socket(s) != 0) {
daniele	29:1a47b7151851	2010	pico_free(s); /* close socket after bind or connect failed */
daniele	29:1a47b7151851	2011	return 0;
daniele	29:1a47b7151851	2012	}
daniele	29:1a47b7151851	2013	}
daniele	29:1a47b7151851	2014
daniele	29:1a47b7151851	2015	#ifdef PICO_SUPPORT_UDP
daniele	29:1a47b7151851	2016	if (PROTO(s) == PICO_PROTO_UDP) {
daniele	29:1a47b7151851	2017	if (mode & PICO_SHUT_RDWR)
daniele	29:1a47b7151851	2018	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	2019	else if (mode & PICO_SHUT_RD)
tass	51:ab4529a384a6	2020	pico_socket_alter_state(s, 0, PICO_SOCKET_STATE_BOUND, 0);
daniele	29:1a47b7151851	2021	}
daniele	29:1a47b7151851	2022	#endif
daniele	29:1a47b7151851	2023	#ifdef PICO_SUPPORT_TCP
daniele	29:1a47b7151851	2024	if (PROTO(s) == PICO_PROTO_TCP) {
tass	51:ab4529a384a6	2025	if(mode & PICO_SHUT_RDWR)
tass	51:ab4529a384a6	2026	pico_socket_alter_state(s, PICO_SOCKET_STATE_SHUT_LOCAL \| PICO_SOCKET_STATE_SHUT_REMOTE, 0, 0);
tass	51:ab4529a384a6	2027	else if (mode & PICO_SHUT_WR)
daniele	29:1a47b7151851	2028	pico_socket_alter_state(s, PICO_SOCKET_STATE_SHUT_LOCAL, 0, 0);
daniele	29:1a47b7151851	2029	else if (mode & PICO_SHUT_RD)
daniele	29:1a47b7151851	2030	pico_socket_alter_state(s, PICO_SOCKET_STATE_SHUT_REMOTE, 0, 0);
daniele	29:1a47b7151851	2031
daniele	29:1a47b7151851	2032	}
daniele	29:1a47b7151851	2033	#endif
daniele	29:1a47b7151851	2034	return 0;
daniele	29:1a47b7151851	2035	}
daniele	29:1a47b7151851	2036
daniele	29:1a47b7151851	2037	int pico_socket_close(struct pico_socket *s)
daniele	29:1a47b7151851	2038	{
daniele	29:1a47b7151851	2039	return pico_socket_shutdown(s, PICO_SHUT_RDWR);
daniele	29:1a47b7151851	2040	}
daniele	29:1a47b7151851	2041
daniele	29:1a47b7151851	2042	#ifdef PICO_SUPPORT_CRC
daniele	29:1a47b7151851	2043	static inline int pico_transport_crc_check(struct pico_frame *f)
daniele	29:1a47b7151851	2044	{
daniele	29:1a47b7151851	2045	struct pico_ipv4_hdr net_hdr = (struct pico_ipv4_hdr ) f->net_hdr;
daniele	29:1a47b7151851	2046	struct pico_udp_hdr *udp_hdr = NULL;
daniele	29:1a47b7151851	2047	uint16_t checksum_invalid = 1;
daniele	29:1a47b7151851	2048
daniele	29:1a47b7151851	2049	switch (net_hdr->proto)
daniele	29:1a47b7151851	2050	{
daniele	29:1a47b7151851	2051	case PICO_PROTO_TCP:
daniele	29:1a47b7151851	2052	checksum_invalid = short_be(pico_tcp_checksum_ipv4(f));
daniele	29:1a47b7151851	2053	//dbg("TCP CRC validation == %u\n", checksum_invalid);
daniele	29:1a47b7151851	2054	if (checksum_invalid) {
daniele	29:1a47b7151851	2055	//dbg("TCP CRC: validation failed!\n");
daniele	29:1a47b7151851	2056	pico_frame_discard(f);
daniele	29:1a47b7151851	2057	return 0;
daniele	29:1a47b7151851	2058	}
daniele	29:1a47b7151851	2059	break;
daniele	29:1a47b7151851	2060
daniele	29:1a47b7151851	2061	case PICO_PROTO_UDP:
daniele	29:1a47b7151851	2062	udp_hdr = (struct pico_udp_hdr *) f->transport_hdr;
daniele	29:1a47b7151851	2063	if (short_be(udp_hdr->crc)) {
daniele	29:1a47b7151851	2064	checksum_invalid = short_be(pico_udp_checksum_ipv4(f));
daniele	29:1a47b7151851	2065	//dbg("UDP CRC validation == %u\n", checksum_invalid);
daniele	29:1a47b7151851	2066	if (checksum_invalid) {
daniele	29:1a47b7151851	2067	//dbg("UDP CRC: validation failed!\n");
daniele	29:1a47b7151851	2068	pico_frame_discard(f);
daniele	29:1a47b7151851	2069	return 0;
daniele	29:1a47b7151851	2070	}
daniele	29:1a47b7151851	2071	}
daniele	29:1a47b7151851	2072	break;
daniele	29:1a47b7151851	2073
daniele	29:1a47b7151851	2074	default:
daniele	29:1a47b7151851	2075	// Do nothing
daniele	29:1a47b7151851	2076	break;
daniele	29:1a47b7151851	2077	}
daniele	29:1a47b7151851	2078	return 1;
daniele	29:1a47b7151851	2079	}
daniele	29:1a47b7151851	2080	#else
daniele	29:1a47b7151851	2081	static inline int pico_transport_crc_check(struct pico_frame *f)
daniele	29:1a47b7151851	2082	{
tass	51:ab4529a384a6	2083	IGNORE_PARAMETER(f);
tass	51:ab4529a384a6	2084	return 1;
daniele	29:1a47b7151851	2085	}
daniele	29:1a47b7151851	2086	#endif /* PICO_SUPPORT_CRC */
daniele	29:1a47b7151851	2087
daniele	29:1a47b7151851	2088	int pico_transport_process_in(struct pico_protocol self, struct pico_frame f)
daniele	29:1a47b7151851	2089	{
daniele	29:1a47b7151851	2090	struct pico_trans hdr = (struct pico_trans ) f->transport_hdr;
daniele	29:1a47b7151851	2091	int ret = 0;
daniele	29:1a47b7151851	2092
daniele	29:1a47b7151851	2093	if (!hdr) {
daniele	29:1a47b7151851	2094	pico_err = PICO_ERR_EFAULT;
daniele	29:1a47b7151851	2095	return -1;
daniele	29:1a47b7151851	2096	}
daniele	29:1a47b7151851	2097
daniele	29:1a47b7151851	2098	ret = pico_transport_crc_check(f);
daniele	29:1a47b7151851	2099	if (ret < 1)
daniele	29:1a47b7151851	2100	return ret;
daniele	29:1a47b7151851	2101	else
daniele	29:1a47b7151851	2102	ret = 0;
daniele	29:1a47b7151851	2103
daniele	29:1a47b7151851	2104	if ((hdr) && (pico_socket_deliver(self, f, hdr->dport) == 0))
daniele	29:1a47b7151851	2105	return ret;
daniele	29:1a47b7151851	2106
daniele	29:1a47b7151851	2107	if (!IS_BCAST(f)) {
daniele	29:1a47b7151851	2108	dbg("Socket not found... \n");
daniele	29:1a47b7151851	2109	pico_notify_socket_unreachable(f);
daniele	29:1a47b7151851	2110	#ifdef PICO_SUPPORT_TCP
daniele	29:1a47b7151851	2111	/* if tcp protocol send RST segment */
daniele	29:1a47b7151851	2112	//if (self->proto_number == PICO_PROTO_TCP)
daniele	29:1a47b7151851	2113	// pico_tcp_reply_rst(f);
daniele	29:1a47b7151851	2114	#endif
daniele	29:1a47b7151851	2115	ret = -1;
daniele	29:1a47b7151851	2116	pico_err = PICO_ERR_ENOENT;
daniele	29:1a47b7151851	2117	}
daniele	29:1a47b7151851	2118	pico_frame_discard(f);
daniele	29:1a47b7151851	2119	return ret;
daniele	29:1a47b7151851	2120	}
daniele	29:1a47b7151851	2121
daniele	29:1a47b7151851	2122	#define SL_LOOP_MIN 1
daniele	29:1a47b7151851	2123
daniele	29:1a47b7151851	2124
daniele	29:1a47b7151851	2125	int pico_sockets_loop(int loop_score)
daniele	29:1a47b7151851	2126	{
daniele	29:1a47b7151851	2127	static struct pico_tree_node index_udp, index_tcp;
daniele	29:1a47b7151851	2128
daniele	29:1a47b7151851	2129	struct pico_sockport *start;
daniele	29:1a47b7151851	2130	struct pico_socket *s;
daniele	29:1a47b7151851	2131
daniele	29:1a47b7151851	2132	#ifdef PICO_SUPPORT_UDP
daniele	29:1a47b7151851	2133	struct pico_frame *f;
daniele	29:1a47b7151851	2134
daniele	29:1a47b7151851	2135	if (sp_udp == NULL)
daniele	29:1a47b7151851	2136	{
daniele	29:1a47b7151851	2137	index_udp = pico_tree_firstNode(UDPTable.root);
daniele	29:1a47b7151851	2138	sp_udp = index_udp->keyValue;
daniele	29:1a47b7151851	2139	}
daniele	29:1a47b7151851	2140
daniele	29:1a47b7151851	2141	/* init start node */
daniele	29:1a47b7151851	2142	start = sp_udp;
daniele	29:1a47b7151851	2143
daniele	29:1a47b7151851	2144	/* round-robin all transport protocols, break if traversed all protocols */
daniele	29:1a47b7151851	2145	while (loop_score > SL_LOOP_MIN && sp_udp != NULL) {
daniele	29:1a47b7151851	2146	struct pico_tree_node * index;
daniele	29:1a47b7151851	2147
daniele	29:1a47b7151851	2148	pico_tree_foreach(index,&sp_udp->socks){
daniele	29:1a47b7151851	2149	s = index->keyValue;
daniele	29:1a47b7151851	2150	f = pico_dequeue(&s->q_out);
daniele	29:1a47b7151851	2151	while (f && (loop_score > 0)) {
daniele	29:1a47b7151851	2152	pico_proto_udp.push(&pico_proto_udp, f);
daniele	29:1a47b7151851	2153	loop_score -= 1;
daniele	29:1a47b7151851	2154	f = pico_dequeue(&s->q_out);
daniele	29:1a47b7151851	2155	}
daniele	29:1a47b7151851	2156	}
daniele	29:1a47b7151851	2157
daniele	29:1a47b7151851	2158	index_udp = pico_tree_next(index_udp);
daniele	29:1a47b7151851	2159	sp_udp = index_udp->keyValue;
daniele	29:1a47b7151851	2160
daniele	29:1a47b7151851	2161	if (sp_udp == NULL)
daniele	29:1a47b7151851	2162	{
daniele	29:1a47b7151851	2163	index_udp = pico_tree_firstNode(UDPTable.root);
daniele	29:1a47b7151851	2164	sp_udp = index_udp->keyValue;
daniele	29:1a47b7151851	2165	}
daniele	29:1a47b7151851	2166	if (sp_udp == start)
daniele	29:1a47b7151851	2167	break;
daniele	29:1a47b7151851	2168	}
daniele	29:1a47b7151851	2169	#endif
daniele	29:1a47b7151851	2170
daniele	29:1a47b7151851	2171	#ifdef PICO_SUPPORT_TCP
daniele	29:1a47b7151851	2172	if (sp_tcp == NULL)
daniele	29:1a47b7151851	2173	{
daniele	29:1a47b7151851	2174	index_tcp = pico_tree_firstNode(TCPTable.root);
daniele	29:1a47b7151851	2175	sp_tcp = index_tcp->keyValue;
daniele	29:1a47b7151851	2176	}
daniele	29:1a47b7151851	2177
daniele	29:1a47b7151851	2178	/* init start node */
daniele	29:1a47b7151851	2179	start = sp_tcp;
daniele	29:1a47b7151851	2180
daniele	29:1a47b7151851	2181	while (loop_score > SL_LOOP_MIN && sp_tcp != NULL) {
tass	48:40fc4462265c	2182	struct pico_tree_node * index = NULL;
daniele	29:1a47b7151851	2183	pico_tree_foreach(index, &sp_tcp->socks){
daniele	29:1a47b7151851	2184	s = index->keyValue;
daniele	29:1a47b7151851	2185	loop_score = pico_tcp_output(s, loop_score);
daniele	29:1a47b7151851	2186	if ((s->ev_pending) && s->wakeup) {
daniele	29:1a47b7151851	2187	s->wakeup(s->ev_pending, s);
daniele	29:1a47b7151851	2188	}
daniele	29:1a47b7151851	2189	if (loop_score <= 0) {
daniele	29:1a47b7151851	2190	loop_score = 0;
daniele	29:1a47b7151851	2191	break;
daniele	29:1a47b7151851	2192	}
daniele	29:1a47b7151851	2193	}
daniele	29:1a47b7151851	2194
daniele	29:1a47b7151851	2195	/* check if RB_FOREACH ended, if not, break to keep the cur sp_tcp */
tass	48:40fc4462265c	2196	if (index && index->keyValue)
daniele	29:1a47b7151851	2197	break;
daniele	29:1a47b7151851	2198
daniele	29:1a47b7151851	2199	index_tcp = pico_tree_next(index_tcp);
daniele	29:1a47b7151851	2200	sp_tcp = index_tcp->keyValue;
daniele	29:1a47b7151851	2201
daniele	29:1a47b7151851	2202	if (sp_tcp == NULL)
daniele	29:1a47b7151851	2203	{
daniele	29:1a47b7151851	2204	index_tcp = pico_tree_firstNode(TCPTable.root);
daniele	29:1a47b7151851	2205	sp_tcp = index_tcp->keyValue;
daniele	29:1a47b7151851	2206	}
daniele	29:1a47b7151851	2207	if (sp_tcp == start)
daniele	29:1a47b7151851	2208	break;
daniele	29:1a47b7151851	2209	}
daniele	29:1a47b7151851	2210	#endif
daniele	29:1a47b7151851	2211
daniele	29:1a47b7151851	2212	return loop_score;
daniele	29:1a47b7151851	2213	}
daniele	29:1a47b7151851	2214
daniele	29:1a47b7151851	2215
daniele	29:1a47b7151851	2216	struct pico_frame pico_socket_frame_alloc(struct pico_socket s, int len)
daniele	29:1a47b7151851	2217	{
daniele	29:1a47b7151851	2218	struct pico_frame *f = NULL;
daniele	29:1a47b7151851	2219
daniele	29:1a47b7151851	2220	#ifdef PICO_SUPPORT_IPV6
daniele	29:1a47b7151851	2221	if (IS_SOCK_IPV6(s))
daniele	29:1a47b7151851	2222	f = pico_proto_ipv6.alloc(&pico_proto_ipv6, len);
daniele	29:1a47b7151851	2223	#endif
daniele	29:1a47b7151851	2224
daniele	29:1a47b7151851	2225	#ifdef PICO_SUPPORT_IPV4
daniele	29:1a47b7151851	2226	if (IS_SOCK_IPV4(s))
daniele	29:1a47b7151851	2227	f = pico_proto_ipv4.alloc(&pico_proto_ipv4, len);
daniele	29:1a47b7151851	2228	#endif
daniele	29:1a47b7151851	2229	if (!f) {
daniele	29:1a47b7151851	2230	pico_err = PICO_ERR_ENOMEM;
daniele	29:1a47b7151851	2231	return f;
daniele	29:1a47b7151851	2232	}
daniele	29:1a47b7151851	2233	f->payload = f->transport_hdr;
daniele	29:1a47b7151851	2234	f->payload_len = len;
daniele	29:1a47b7151851	2235	f->sock = s;
daniele	29:1a47b7151851	2236	return f;
daniele	29:1a47b7151851	2237	}
daniele	29:1a47b7151851	2238
daniele	29:1a47b7151851	2239	int pico_transport_error(struct pico_frame *f, uint8_t proto, int code)
daniele	29:1a47b7151851	2240	{
daniele	29:1a47b7151851	2241	int ret = -1;
daniele	29:1a47b7151851	2242	struct pico_trans trans = (struct pico_trans) f->transport_hdr;
daniele	29:1a47b7151851	2243	struct pico_sockport *port = NULL;
daniele	29:1a47b7151851	2244	struct pico_socket *s = NULL;
daniele	29:1a47b7151851	2245	switch (proto) {
daniele	29:1a47b7151851	2246
daniele	29:1a47b7151851	2247
daniele	29:1a47b7151851	2248	#ifdef PICO_SUPPORT_UDP
daniele	29:1a47b7151851	2249	case PICO_PROTO_UDP:
daniele	29:1a47b7151851	2250	port = pico_get_sockport(proto, trans->sport);
daniele	29:1a47b7151851	2251	break;
daniele	29:1a47b7151851	2252	#endif
daniele	29:1a47b7151851	2253
daniele	29:1a47b7151851	2254	#ifdef PICO_SUPPORT_TCP
daniele	29:1a47b7151851	2255	case PICO_PROTO_TCP:
daniele	29:1a47b7151851	2256	port = pico_get_sockport(proto, trans->sport);
daniele	29:1a47b7151851	2257	break;
daniele	29:1a47b7151851	2258	#endif
daniele	29:1a47b7151851	2259
daniele	29:1a47b7151851	2260	default:
daniele	29:1a47b7151851	2261	/* Protocol not available */
daniele	29:1a47b7151851	2262	ret = -1;
daniele	29:1a47b7151851	2263	}
daniele	29:1a47b7151851	2264	if (port) {
daniele	29:1a47b7151851	2265	struct pico_tree_node * index;
daniele	29:1a47b7151851	2266	ret = 0;
daniele	29:1a47b7151851	2267
daniele	29:1a47b7151851	2268	pico_tree_foreach(index,&port->socks) {
daniele	29:1a47b7151851	2269	s = index->keyValue;
daniele	29:1a47b7151851	2270	if (trans->dport == s->remote_port) {
daniele	29:1a47b7151851	2271	if (s->wakeup) {
daniele	29:1a47b7151851	2272	//dbg("SOCKET ERROR FROM ICMP NOTIFICATION. (icmp code= %d)\n\n", code);
daniele	29:1a47b7151851	2273	switch(code) {
daniele	29:1a47b7151851	2274	case PICO_ICMP_UNREACH_PROTOCOL:
daniele	29:1a47b7151851	2275	pico_err = PICO_ERR_EPROTO;
daniele	29:1a47b7151851	2276	break;
daniele	29:1a47b7151851	2277
daniele	29:1a47b7151851	2278	case PICO_ICMP_UNREACH_PORT:
daniele	29:1a47b7151851	2279	pico_err = PICO_ERR_ECONNREFUSED;
daniele	29:1a47b7151851	2280	break;
daniele	29:1a47b7151851	2281
daniele	29:1a47b7151851	2282	case PICO_ICMP_UNREACH_NET:
daniele	29:1a47b7151851	2283	case PICO_ICMP_UNREACH_NET_PROHIB:
daniele	29:1a47b7151851	2284	case PICO_ICMP_UNREACH_NET_UNKNOWN:
daniele	29:1a47b7151851	2285	pico_err = PICO_ERR_ENETUNREACH;
daniele	29:1a47b7151851	2286	break;
daniele	29:1a47b7151851	2287
daniele	29:1a47b7151851	2288	default:
daniele	29:1a47b7151851	2289	pico_err = PICO_ERR_EHOSTUNREACH;
daniele	29:1a47b7151851	2290	}
daniele	29:1a47b7151851	2291	s->wakeup(PICO_SOCK_EV_ERR, s);
daniele	29:1a47b7151851	2292	}
daniele	29:1a47b7151851	2293	break;
daniele	29:1a47b7151851	2294	}
daniele	29:1a47b7151851	2295	}
daniele	29:1a47b7151851	2296	}
daniele	29:1a47b7151851	2297	pico_frame_discard(f);
daniele	29:1a47b7151851	2298	return ret;
daniele	29:1a47b7151851	2299	}
daniele	29:1a47b7151851	2300	#endif
daniele	29:1a47b7151851	2301	#endif