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@48:40fc4462265c, 2013-08-02 (annotated)

Committer:: tass
Date:: Fri Aug 02 07:28:05 2013 +0000
Revision:: 48:40fc4462265c
Parent:: 29:1a47b7151851
Child:: 49:4b404dd2c97a

Fixes for issues #8,#9,#10 from github.

Who changed what in which revision?

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