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.

modules/pico_tcp.c@93:e7a58195bfb7, 2013-10-14 (annotated)

Committer:: tass
Date:: Mon Oct 14 12:33:43 2013 +0000
Revision:: 93:e7a58195bfb7
Parent:: 91:8da42f2c4820
Child:: 95:f4ca916a26fe
Child:: 102:c6ac12223f56

Fix Issue #18

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
daniele	29:1a47b7151851	7	Authors: Daniele Lacamera, Philippe Mariman
daniele	29:1a47b7151851	8	*********************************************************************/
daniele	29:1a47b7151851	9
daniele	29:1a47b7151851	10	#include "pico_tcp.h"
daniele	29:1a47b7151851	11	#include "pico_config.h"
daniele	29:1a47b7151851	12	#include "pico_eth.h"
daniele	29:1a47b7151851	13	#include "pico_socket.h"
daniele	29:1a47b7151851	14	#include "pico_stack.h"
daniele	29:1a47b7151851	15	#include "pico_socket.h"
daniele	29:1a47b7151851	16	#include "pico_queue.h"
daniele	29:1a47b7151851	17	#include "pico_tree.h"
daniele	29:1a47b7151851	18
tass	68:0847e35d08a6	19	#define TCP_IS_STATE(s,st) (s->state & st)
daniele	29:1a47b7151851	20	#define TCP_SOCK(s) ((struct pico_socket_tcp *)s)
daniele	29:1a47b7151851	21	#define SEQN(f) (f?(long_be(((struct pico_tcp_hdr *)(f->transport_hdr))->seq)):0)
daniele	29:1a47b7151851	22	#define ACKN(f) (f?(long_be(((struct pico_tcp_hdr *)(f->transport_hdr))->ack)):0)
daniele	29:1a47b7151851	23
daniele	29:1a47b7151851	24	#define PICO_TCP_RTO_MIN 10
daniele	29:1a47b7151851	25	#define PICO_TCP_RTO_MAX 120000
tass	68:0847e35d08a6	26	#define PICO_TCP_IW 2
tass	70:cd218dd180e5	27	#define PICO_TCP_SYN_TO 1000u
tass	48:40fc4462265c	28	#define PICO_TCP_ZOMBIE_TO 30000
daniele	29:1a47b7151851	29
daniele	29:1a47b7151851	30	#define PICO_TCP_MAX_CONNECT_RETRIES 7
daniele	29:1a47b7151851	31
daniele	29:1a47b7151851	32	#define PICO_TCP_LOOKAHEAD 0x00
daniele	29:1a47b7151851	33	#define PICO_TCP_FIRST_DUPACK 0x01
daniele	29:1a47b7151851	34	#define PICO_TCP_SECOND_DUPACK 0x02
daniele	29:1a47b7151851	35	#define PICO_TCP_RECOVER 0x03
daniele	29:1a47b7151851	36	#define PICO_TCP_BLACKOUT 0x04
daniele	29:1a47b7151851	37	#define PICO_TCP_UNREACHABLE 0x05
daniele	29:1a47b7151851	38	#define PICO_TCP_WINDOW_FULL 0x06
daniele	29:1a47b7151851	39
daniele	29:1a47b7151851	40	/* check if the Nagle algorithm is enabled on the socket */
daniele	29:1a47b7151851	41	#define IS_NAGLE_ENABLED(s) (!(!(!(s->opt_flags & (1 << PICO_SOCKET_OPT_TCPNODELAY)))))
daniele	29:1a47b7151851	42	/* check if tcp connection is "idle" according to Nagle (RFC 896) */
daniele	29:1a47b7151851	43	#define IS_TCP_IDLE(t) ((t->in_flight == 0) && (t->tcpq_out.size == 0))
daniele	29:1a47b7151851	44	/* check if the hold queue contains data (again Nagle) */
daniele	29:1a47b7151851	45	#define IS_TCP_HOLDQ_EMPTY(t) (t->tcpq_hold.size == 0)
daniele	29:1a47b7151851	46
daniele	29:1a47b7151851	47	#ifdef PICO_SUPPORT_TCP
tass	45:ca30069e49bb	48	#define tcp_dbg_nagle(...) do{}while(0)
tass	45:ca30069e49bb	49	#define tcp_dbg_options(...) do{}while(0)
tass	45:ca30069e49bb	50
tass	45:ca30069e49bb	51
tass	68:0847e35d08a6	52	#define tcp_dbg(...) do{}while(0)
tass	68:0847e35d08a6	53	//#define tcp_dbg dbg
daniele	29:1a47b7151851	54
daniele	29:1a47b7151851	55	#ifdef PICO_SUPPORT_MUTEX
daniele	29:1a47b7151851	56	static void * Mutex = NULL;
daniele	29:1a47b7151851	57	#define LOCK(x) {\
daniele	29:1a47b7151851	58	if (x == NULL) \
daniele	29:1a47b7151851	59	x = pico_mutex_init(); \
daniele	29:1a47b7151851	60	pico_mutex_lock(x); \
daniele	29:1a47b7151851	61	}
daniele	29:1a47b7151851	62	#define UNLOCK(x) pico_mutex_unlock(x);
daniele	29:1a47b7151851	63
daniele	29:1a47b7151851	64	#else
daniele	29:1a47b7151851	65	#define LOCK(x) do{}while(0)
daniele	29:1a47b7151851	66	#define UNLOCK(x) do{}while(0)
daniele	29:1a47b7151851	67	#endif
daniele	29:1a47b7151851	68
daniele	29:1a47b7151851	69
daniele	29:1a47b7151851	70	static inline int seq_compare(uint32_t a, uint32_t b)
daniele	29:1a47b7151851	71	{
daniele	29:1a47b7151851	72	uint32_t thresh = ((uint32_t)(-1))>>1;
daniele	29:1a47b7151851	73	if (((a > thresh) && (b > thresh)) \|\| ((a <= thresh) && (b <= thresh))) {
daniele	29:1a47b7151851	74	if (a > b)
daniele	29:1a47b7151851	75	return 1;
daniele	29:1a47b7151851	76	if (b > a)
daniele	29:1a47b7151851	77	return -1;
daniele	29:1a47b7151851	78	} else {
daniele	29:1a47b7151851	79	if (a > b)
daniele	29:1a47b7151851	80	return -2;
daniele	29:1a47b7151851	81	if (b > a)
daniele	29:1a47b7151851	82	return 2;
daniele	29:1a47b7151851	83	}
daniele	29:1a47b7151851	84	return 0;
daniele	29:1a47b7151851	85	}
daniele	29:1a47b7151851	86
daniele	29:1a47b7151851	87	static int segment_compare(void * ka, void * kb)
daniele	29:1a47b7151851	88	{
daniele	29:1a47b7151851	89	struct pico_frame a = ka, b = kb;
daniele	29:1a47b7151851	90	return seq_compare(SEQN(a), SEQN(b));
daniele	29:1a47b7151851	91	}
daniele	29:1a47b7151851	92
daniele	29:1a47b7151851	93	struct pico_tcp_queue
daniele	29:1a47b7151851	94	{
daniele	29:1a47b7151851	95	struct pico_tree pool;
daniele	29:1a47b7151851	96	uint32_t max_size;
daniele	29:1a47b7151851	97	uint32_t size;
daniele	29:1a47b7151851	98	uint32_t frames;
daniele	29:1a47b7151851	99	};
daniele	29:1a47b7151851	100
daniele	29:1a47b7151851	101	static struct pico_frame peek_segment(struct pico_tcp_queue tq, uint32_t seq)
daniele	29:1a47b7151851	102	{
daniele	29:1a47b7151851	103	struct pico_tcp_hdr H;
tass	68:0847e35d08a6	104	struct pico_frame f = {0};
daniele	29:1a47b7151851	105	f.transport_hdr = (uint8_t *) (&H);
daniele	29:1a47b7151851	106	H.seq = long_be(seq);
daniele	29:1a47b7151851	107
daniele	29:1a47b7151851	108	return pico_tree_findKey(&tq->pool,&f);
daniele	29:1a47b7151851	109	}
daniele	29:1a47b7151851	110
daniele	29:1a47b7151851	111	static struct pico_frame first_segment(struct pico_tcp_queue tq)
daniele	29:1a47b7151851	112	{
daniele	29:1a47b7151851	113	return pico_tree_first(&tq->pool);
daniele	29:1a47b7151851	114	}
daniele	29:1a47b7151851	115
daniele	29:1a47b7151851	116	static struct pico_frame next_segment(struct pico_tcp_queue tq, struct pico_frame *cur)
daniele	29:1a47b7151851	117	{
daniele	29:1a47b7151851	118	if (!cur)
daniele	29:1a47b7151851	119	return NULL;
daniele	29:1a47b7151851	120	return peek_segment(tq, SEQN(cur) + cur->payload_len);
daniele	29:1a47b7151851	121	}
daniele	29:1a47b7151851	122
tass	70:cd218dd180e5	123	static int32_t pico_enqueue_segment(struct pico_tcp_queue tq, struct pico_frame f)
daniele	29:1a47b7151851	124	{
tass	70:cd218dd180e5	125	int32_t ret = -1;
daniele	29:1a47b7151851	126	if (f->payload_len <= 0) {
daniele	29:1a47b7151851	127	tcp_dbg("!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!! TRIED TO ENQUEUE INVALID SEGMENT!\n");
daniele	29:1a47b7151851	128	//abort();
daniele	29:1a47b7151851	129	return -1;
daniele	29:1a47b7151851	130	}
tass	68:0847e35d08a6	131	LOCK(Mutex);
daniele	29:1a47b7151851	132	if ((tq->size + f->payload_len) > tq->max_size)
daniele	29:1a47b7151851	133	{
daniele	29:1a47b7151851	134	ret = 0;
daniele	29:1a47b7151851	135	goto out;
daniele	29:1a47b7151851	136	}
daniele	29:1a47b7151851	137	if (pico_tree_insert(&tq->pool,f) != 0)
daniele	29:1a47b7151851	138	{
daniele	29:1a47b7151851	139	ret = 0;
daniele	29:1a47b7151851	140	goto out;
daniele	29:1a47b7151851	141	}
daniele	29:1a47b7151851	142	tq->size += f->payload_len;
daniele	29:1a47b7151851	143	if (f->payload_len > 0)
daniele	29:1a47b7151851	144	tq->frames++;
tass	70:cd218dd180e5	145	ret = (int32_t)f->payload_len;
daniele	29:1a47b7151851	146
daniele	29:1a47b7151851	147	out :
daniele	29:1a47b7151851	148	UNLOCK(Mutex);
daniele	29:1a47b7151851	149	return ret;
daniele	29:1a47b7151851	150	}
daniele	29:1a47b7151851	151
daniele	29:1a47b7151851	152	static void pico_discard_segment(struct pico_tcp_queue tq, struct pico_frame f)
daniele	29:1a47b7151851	153	{
daniele	29:1a47b7151851	154	struct pico_frame *f1;
daniele	29:1a47b7151851	155	LOCK(Mutex);
daniele	29:1a47b7151851	156	f1 = pico_tree_delete(&tq->pool,f);
daniele	29:1a47b7151851	157	if (f1) {
daniele	29:1a47b7151851	158	tq->size -= f->payload_len;
daniele	29:1a47b7151851	159	if (f->payload_len > 0)
daniele	29:1a47b7151851	160	tq->frames--;
daniele	29:1a47b7151851	161	}
daniele	29:1a47b7151851	162	pico_frame_discard(f);
daniele	29:1a47b7151851	163	UNLOCK(Mutex);
daniele	29:1a47b7151851	164	}
daniele	29:1a47b7151851	165
daniele	29:1a47b7151851	166	/* Structure for TCP socket */
daniele	29:1a47b7151851	167	struct tcp_sack_block {
daniele	29:1a47b7151851	168	uint32_t left;
daniele	29:1a47b7151851	169	uint32_t right;
daniele	29:1a47b7151851	170	struct tcp_sack_block *next;
daniele	29:1a47b7151851	171	};
daniele	29:1a47b7151851	172
daniele	29:1a47b7151851	173	struct pico_socket_tcp {
daniele	29:1a47b7151851	174	struct pico_socket sock;
daniele	29:1a47b7151851	175
daniele	29:1a47b7151851	176	/* Tree/queues */
daniele	29:1a47b7151851	177	struct pico_tcp_queue tcpq_in;
daniele	29:1a47b7151851	178	struct pico_tcp_queue tcpq_out;
daniele	29:1a47b7151851	179	struct pico_tcp_queue tcpq_hold; /* buffer to hold delayed frames according to Nagle */
daniele	29:1a47b7151851	180
daniele	29:1a47b7151851	181	/* tcp_output */
daniele	29:1a47b7151851	182	uint32_t snd_nxt;
daniele	29:1a47b7151851	183	uint32_t snd_last;
daniele	29:1a47b7151851	184	uint32_t snd_old_ack;
daniele	29:1a47b7151851	185	uint32_t snd_retry;
daniele	29:1a47b7151851	186	uint32_t snd_last_out;
daniele	29:1a47b7151851	187
daniele	29:1a47b7151851	188	/* congestion control */
daniele	29:1a47b7151851	189	uint32_t avg_rtt;
daniele	29:1a47b7151851	190	uint32_t rttvar;
daniele	29:1a47b7151851	191	uint32_t rto;
daniele	29:1a47b7151851	192	uint32_t in_flight;
daniele	29:1a47b7151851	193	uint8_t timer_running;
tass	70:cd218dd180e5	194	struct pico_timer * retrans_tmr;
daniele	29:1a47b7151851	195	uint8_t keepalive_timer_running;
daniele	29:1a47b7151851	196	uint16_t cwnd_counter;
daniele	29:1a47b7151851	197	uint16_t cwnd;
daniele	29:1a47b7151851	198	uint16_t ssthresh;
daniele	29:1a47b7151851	199	uint16_t recv_wnd;
daniele	29:1a47b7151851	200	uint16_t recv_wnd_scale;
daniele	29:1a47b7151851	201
daniele	29:1a47b7151851	202	/* tcp_input */
daniele	29:1a47b7151851	203	uint32_t rcv_nxt;
daniele	29:1a47b7151851	204	uint32_t rcv_ackd;
daniele	29:1a47b7151851	205	uint32_t rcv_processed;
daniele	29:1a47b7151851	206	uint16_t wnd;
daniele	29:1a47b7151851	207	uint16_t wnd_scale;
daniele	29:1a47b7151851	208
daniele	29:1a47b7151851	209	/* options */
daniele	29:1a47b7151851	210	uint32_t ts_nxt;
daniele	29:1a47b7151851	211	uint16_t mss;
daniele	29:1a47b7151851	212	uint8_t sack_ok;
daniele	29:1a47b7151851	213	uint8_t ts_ok;
daniele	29:1a47b7151851	214	uint8_t mss_ok;
daniele	29:1a47b7151851	215	uint8_t scale_ok;
daniele	29:1a47b7151851	216	struct tcp_sack_block *sacks;
daniele	29:1a47b7151851	217	uint8_t jumbo;
daniele	29:1a47b7151851	218
daniele	29:1a47b7151851	219	/* Transmission */
daniele	29:1a47b7151851	220	uint8_t x_mode;
daniele	29:1a47b7151851	221	uint8_t dupacks;
daniele	29:1a47b7151851	222	uint8_t backoff;
tass	75:a9abf10138d6	223	uint8_t localZeroWindow;
daniele	29:1a47b7151851	224	};
daniele	29:1a47b7151851	225
daniele	29:1a47b7151851	226	/* Queues */
tass	68:0847e35d08a6	227	static struct pico_queue tcp_in = {0};
tass	68:0847e35d08a6	228	static struct pico_queue tcp_out = {0};
daniele	29:1a47b7151851	229
daniele	29:1a47b7151851	230	/* If Nagle enabled, this function can make 1 new segment from smaller segments in hold queue */
daniele	29:1a47b7151851	231	static struct pico_frame * pico_hold_segment_make(struct pico_socket_tcp *t);
daniele	29:1a47b7151851	232
daniele	29:1a47b7151851	233	/* checks if tcpq_in is empty */
daniele	29:1a47b7151851	234	int pico_tcp_queue_in_is_empty(struct pico_socket *s)
daniele	29:1a47b7151851	235	{
daniele	29:1a47b7151851	236	struct pico_socket_tcp t = (struct pico_socket_tcp ) s;
daniele	29:1a47b7151851	237
daniele	29:1a47b7151851	238	if (t->tcpq_in.frames == 0)
daniele	29:1a47b7151851	239	return 1;
daniele	29:1a47b7151851	240	else
daniele	29:1a47b7151851	241	return 0;
daniele	29:1a47b7151851	242	}
daniele	29:1a47b7151851	243
daniele	29:1a47b7151851	244	/* Useful for getting rid of the beginning of the buffer (read() op) */
daniele	29:1a47b7151851	245	static int release_until(struct pico_tcp_queue *q, uint32_t seq)
daniele	29:1a47b7151851	246	{
daniele	29:1a47b7151851	247	struct pico_frame *head = first_segment(q);
daniele	29:1a47b7151851	248	int ret = 0;
daniele	29:1a47b7151851	249	while (head && (seq_compare(SEQN(head) + head->payload_len, seq) <= 0)) {
daniele	29:1a47b7151851	250	struct pico_frame *cur = head;
daniele	29:1a47b7151851	251	head = next_segment(q, cur);
tass	37:bdf736327c71	252	tcp_dbg("Releasing %p\n", q);
daniele	29:1a47b7151851	253	pico_discard_segment(q, cur);
daniele	29:1a47b7151851	254	ret++;
daniele	29:1a47b7151851	255	}
daniele	29:1a47b7151851	256	return ret;
daniele	29:1a47b7151851	257	}
daniele	29:1a47b7151851	258
daniele	29:1a47b7151851	259	static int release_all_until(struct pico_tcp_queue *q, uint32_t seq)
daniele	29:1a47b7151851	260	{
daniele	29:1a47b7151851	261	struct pico_frame f = NULL, tmp __attribute__((unused));
daniele	29:1a47b7151851	262	struct pico_tree_node * idx, * temp;
daniele	29:1a47b7151851	263	int ret = 0;
daniele	29:1a47b7151851	264
daniele	29:1a47b7151851	265	pico_tree_foreach_safe(idx,&q->pool,temp){
daniele	29:1a47b7151851	266	f = idx->keyValue;
daniele	29:1a47b7151851	267	if (seq_compare(SEQN(f) + f->payload_len, seq) <= 0) {
tass	37:bdf736327c71	268	tcp_dbg("Releasing %p\n", f);
daniele	29:1a47b7151851	269	pico_discard_segment(q, f);
daniele	29:1a47b7151851	270	ret++;
daniele	29:1a47b7151851	271	} else
daniele	29:1a47b7151851	272	return ret;
daniele	29:1a47b7151851	273	}
daniele	29:1a47b7151851	274	return ret;
daniele	29:1a47b7151851	275	}
daniele	29:1a47b7151851	276
daniele	29:1a47b7151851	277	/* API calls */
daniele	29:1a47b7151851	278
daniele	29:1a47b7151851	279	uint16_t pico_tcp_checksum_ipv4(struct pico_frame *f)
daniele	29:1a47b7151851	280	{
daniele	29:1a47b7151851	281	struct pico_ipv4_hdr hdr = (struct pico_ipv4_hdr ) f->net_hdr;
daniele	29:1a47b7151851	282	struct pico_tcp_hdr tcp_hdr = (struct pico_tcp_hdr ) f->transport_hdr;
daniele	29:1a47b7151851	283	struct pico_socket *s = f->sock;
daniele	29:1a47b7151851	284	struct pico_ipv4_pseudo_hdr pseudo;
daniele	29:1a47b7151851	285
daniele	29:1a47b7151851	286	if (s) {
daniele	29:1a47b7151851	287	/* Case of outgoing frame */
daniele	29:1a47b7151851	288	//dbg("TCP CRC: on outgoing frame\n");
daniele	29:1a47b7151851	289	pseudo.src.addr = s->local_addr.ip4.addr;
daniele	29:1a47b7151851	290	pseudo.dst.addr = s->remote_addr.ip4.addr;
daniele	29:1a47b7151851	291	} else {
daniele	29:1a47b7151851	292	/* Case of incomming frame */
daniele	29:1a47b7151851	293	//dbg("TCP CRC: on incomming frame\n");
daniele	29:1a47b7151851	294	pseudo.src.addr = hdr->src.addr;
daniele	29:1a47b7151851	295	pseudo.dst.addr = hdr->dst.addr;
daniele	29:1a47b7151851	296	}
daniele	29:1a47b7151851	297	pseudo.zeros = 0;
daniele	29:1a47b7151851	298	pseudo.proto = PICO_PROTO_TCP;
tass	70:cd218dd180e5	299	pseudo.len = (uint16_t)short_be(f->transport_len);
daniele	29:1a47b7151851	300
daniele	29:1a47b7151851	301	return pico_dualbuffer_checksum(&pseudo, sizeof(struct pico_ipv4_pseudo_hdr), tcp_hdr, f->transport_len);
daniele	29:1a47b7151851	302	}
daniele	29:1a47b7151851	303
tass	32:865c101e0874	304	static void tcp_send_fin(struct pico_socket_tcp *t);
daniele	29:1a47b7151851	305	static int pico_tcp_process_out(struct pico_protocol self, struct pico_frame f)
daniele	29:1a47b7151851	306	{
daniele	29:1a47b7151851	307	struct pico_tcp_hdr *hdr;
daniele	29:1a47b7151851	308	struct pico_socket_tcp t = (struct pico_socket_tcp )f->sock;
tass	68:0847e35d08a6	309	IGNORE_PARAMETER(self);
daniele	29:1a47b7151851	310	hdr = (struct pico_tcp_hdr *)f->transport_hdr;
tass	68:0847e35d08a6	311	f->sock->timestamp = PICO_TIME_MS();
daniele	29:1a47b7151851	312	if (f->payload_len > 0) {
daniele	29:1a47b7151851	313	tcp_dbg("Process out: sending %p (%d bytes)\n",f, f->payload_len);
daniele	29:1a47b7151851	314	} else {
daniele	29:1a47b7151851	315	tcp_dbg("Sending empty packet\n");
daniele	29:1a47b7151851	316	}
daniele	29:1a47b7151851	317
daniele	29:1a47b7151851	318	if (f->payload_len > 0) {
daniele	29:1a47b7151851	319	if (seq_compare(SEQN(f) + f->payload_len, t->snd_nxt) > 0) {
daniele	29:1a47b7151851	320	t->snd_nxt = SEQN(f) + f->payload_len;
daniele	29:1a47b7151851	321	tcp_dbg("%s: snd_nxt is now %08x\n", __FUNCTION__, t->snd_nxt);
daniele	29:1a47b7151851	322	}
daniele	29:1a47b7151851	323	} else if (hdr->flags == PICO_TCP_ACK) { /* pure ack */
tass	40:c8ab0d2bba0b	324	//hdr->seq = long_be(t->snd_nxt); /* XXX disabled this to not to mess with seq nrs of ACKs anymore */
daniele	29:1a47b7151851	325	} else {
daniele	29:1a47b7151851	326	tcp_dbg("%s: non-pure ACK with len=0, fl:%04x\n", __FUNCTION__, hdr->flags);
daniele	29:1a47b7151851	327	}
tass	32:865c101e0874	328
daniele	29:1a47b7151851	329	pico_network_send(f);
daniele	29:1a47b7151851	330	return 0;
daniele	29:1a47b7151851	331	}
daniele	29:1a47b7151851	332
daniele	29:1a47b7151851	333	int pico_tcp_push(struct pico_protocol self, struct pico_frame data);
daniele	29:1a47b7151851	334
daniele	29:1a47b7151851	335	/* Interface: protocol definition */
daniele	29:1a47b7151851	336	struct pico_protocol pico_proto_tcp = {
daniele	29:1a47b7151851	337	.name = "tcp",
daniele	29:1a47b7151851	338	.proto_number = PICO_PROTO_TCP,
daniele	29:1a47b7151851	339	.layer = PICO_LAYER_TRANSPORT,
daniele	29:1a47b7151851	340	.process_in = pico_transport_process_in,
daniele	29:1a47b7151851	341	.process_out = pico_tcp_process_out,
daniele	29:1a47b7151851	342	.push = pico_tcp_push,
daniele	29:1a47b7151851	343	.q_in = &tcp_in,
daniele	29:1a47b7151851	344	.q_out = &tcp_out,
daniele	29:1a47b7151851	345	};
daniele	29:1a47b7151851	346
daniele	29:1a47b7151851	347	static uint32_t pico_paws(void)
daniele	29:1a47b7151851	348	{
tass	74:c146c4e346c4	349	static uint32_t _paws = 0;
daniele	29:1a47b7151851	350	_paws = pico_rand();
daniele	29:1a47b7151851	351	return long_be(_paws); /XXX: implement paws /
daniele	29:1a47b7151851	352	}
daniele	29:1a47b7151851	353
tass	70:cd218dd180e5	354	static void tcp_add_options(struct pico_socket_tcp ts, struct pico_frame f, uint16_t flags, uint16_t optsiz)
daniele	29:1a47b7151851	355	{
daniele	29:1a47b7151851	356	uint32_t tsval = long_be(pico_tick);
daniele	29:1a47b7151851	357	uint32_t tsecr = long_be(ts->ts_nxt);
tass	70:cd218dd180e5	358	uint32_t i = 0;
daniele	29:1a47b7151851	359	f->start = f->transport_hdr + PICO_SIZE_TCPHDR;
daniele	29:1a47b7151851	360
daniele	29:1a47b7151851	361	memset(f->start, PICO_TCP_OPTION_NOOP, optsiz); /* fill blanks with noop */
daniele	29:1a47b7151851	362
daniele	29:1a47b7151851	363	if (flags & PICO_TCP_SYN) {
daniele	29:1a47b7151851	364	f->start[i++] = PICO_TCP_OPTION_MSS;
daniele	29:1a47b7151851	365	f->start[i++] = PICO_TCPOPTLEN_MSS;
tass	70:cd218dd180e5	366	f->start[i++] = (uint8_t)((ts->mss >> 8) & 0xFF);
tass	70:cd218dd180e5	367	f->start[i++] = (uint8_t)(ts->mss & 0xFF);
daniele	29:1a47b7151851	368	f->start[i++] = PICO_TCP_OPTION_SACK_OK;
daniele	29:1a47b7151851	369	f->start[i++] = PICO_TCPOPTLEN_SACK_OK;
daniele	29:1a47b7151851	370	}
daniele	29:1a47b7151851	371
daniele	29:1a47b7151851	372	f->start[i++] = PICO_TCP_OPTION_WS;
daniele	29:1a47b7151851	373	f->start[i++] = PICO_TCPOPTLEN_WS;
tass	70:cd218dd180e5	374	f->start[i++] = (uint8_t)(ts->wnd_scale);
daniele	29:1a47b7151851	375
daniele	29:1a47b7151851	376	if (optsiz >= 12) {
daniele	29:1a47b7151851	377	f->start[i++] = PICO_TCP_OPTION_TIMESTAMP;
daniele	29:1a47b7151851	378	f->start[i++] = PICO_TCPOPTLEN_TIMESTAMP;
daniele	29:1a47b7151851	379	memcpy(f->start + i, &tsval, 4);
daniele	29:1a47b7151851	380	i += 4;
daniele	29:1a47b7151851	381	memcpy(f->start + i, &tsecr, 4);
daniele	29:1a47b7151851	382	i += 4;
daniele	29:1a47b7151851	383	}
daniele	29:1a47b7151851	384
daniele	29:1a47b7151851	385	if (flags & PICO_TCP_ACK) {
daniele	29:1a47b7151851	386	struct tcp_sack_block *sb;
tass	70:cd218dd180e5	387	uint32_t len_off;
daniele	29:1a47b7151851	388
daniele	29:1a47b7151851	389	if (ts->sack_ok && ts->sacks) {
daniele	29:1a47b7151851	390	f->start[i++] = PICO_TCP_OPTION_SACK;
daniele	29:1a47b7151851	391	len_off = i;
daniele	29:1a47b7151851	392	f->start[i++] = PICO_TCPOPTLEN_SACK;
daniele	29:1a47b7151851	393	while(ts->sacks) {
daniele	29:1a47b7151851	394	sb = ts->sacks;
daniele	29:1a47b7151851	395	ts->sacks = sb->next;
daniele	29:1a47b7151851	396	memcpy(f->start + i, sb, 2 * sizeof(uint32_t));
tass	74:c146c4e346c4	397	i += (2 * (uint32_t)sizeof(uint32_t));
tass	73:dfb737147f6e	398	f->start[len_off] = (uint8_t)(f->start[len_off] + (2 * sizeof(uint32_t)));
daniele	29:1a47b7151851	399	pico_free(sb);
daniele	29:1a47b7151851	400	}
daniele	29:1a47b7151851	401	}
daniele	29:1a47b7151851	402	}
daniele	29:1a47b7151851	403	if (i < optsiz)
daniele	29:1a47b7151851	404	f->start[ optsiz - 1 ] = PICO_TCP_OPTION_END;
daniele	29:1a47b7151851	405	}
daniele	29:1a47b7151851	406
daniele	29:1a47b7151851	407	static void tcp_send_ack(struct pico_socket_tcp *t);
tass	75:a9abf10138d6	408	#define tcp_send_windowUpdate(t) (tcp_send_ack(t))
daniele	29:1a47b7151851	409
daniele	29:1a47b7151851	410	static void tcp_set_space(struct pico_socket_tcp *t)
daniele	29:1a47b7151851	411	{
tass	70:cd218dd180e5	412	uint32_t mtu;
tass	70:cd218dd180e5	413	int32_t space;
tass	70:cd218dd180e5	414	uint32_t shift = 0;
daniele	29:1a47b7151851	415
daniele	29:1a47b7151851	416	mtu = t->mss + PICO_SIZE_TCPHDR + PICO_SIZE_TCPOPT_SYN ;
daniele	29:1a47b7151851	417	if (t->tcpq_in.max_size == 0) {
daniele	29:1a47b7151851	418	space = 1024 * 1024 * 1024; /* One Gigabyte, for unlimited sockets. */
daniele	29:1a47b7151851	419	} else {
tass	70:cd218dd180e5	420	space = (int32_t)(((t->tcpq_in.max_size - t->tcpq_in.size) / mtu) * t->mss);
daniele	29:1a47b7151851	421	}
daniele	29:1a47b7151851	422	if (space < 0)
daniele	29:1a47b7151851	423	space = 0;
daniele	29:1a47b7151851	424	while(space > 0xFFFF) {
daniele	29:1a47b7151851	425	space >>= 1;
daniele	29:1a47b7151851	426	shift++;
daniele	29:1a47b7151851	427	}
daniele	29:1a47b7151851	428	if ((space != t->wnd) \|\| (shift != t->wnd_scale) \|\| ((space - t->wnd) > (space>>2))) {
tass	70:cd218dd180e5	429	t->wnd = (uint16_t)space;
tass	70:cd218dd180e5	430	t->wnd_scale = (uint16_t)shift;
tass	75:a9abf10138d6	431
tass	75:a9abf10138d6	432	if(t->wnd == 0) // mark the entering to zero window state
tass	75:a9abf10138d6	433	t->localZeroWindow = 1u;
tass	75:a9abf10138d6	434	else if(t->localZeroWindow)
tass	75:a9abf10138d6	435	{
tass	75:a9abf10138d6	436	t->localZeroWindow = 0u;
tass	75:a9abf10138d6	437	tcp_send_windowUpdate(t);
tass	75:a9abf10138d6	438	}
daniele	29:1a47b7151851	439	}
daniele	29:1a47b7151851	440	}
daniele	29:1a47b7151851	441
daniele	29:1a47b7151851	442	/* Return 32-bit aligned option size */
tass	70:cd218dd180e5	443	static uint16_t tcp_options_size(struct pico_socket_tcp *t, uint16_t flags)
daniele	29:1a47b7151851	444	{
tass	70:cd218dd180e5	445	uint16_t size = 0;
daniele	29:1a47b7151851	446	struct tcp_sack_block *sb = t->sacks;
daniele	29:1a47b7151851	447
daniele	29:1a47b7151851	448	if (flags & PICO_TCP_SYN) { /* Full options */
daniele	29:1a47b7151851	449	size = PICO_TCPOPTLEN_MSS + PICO_TCP_OPTION_SACK_OK + PICO_TCPOPTLEN_WS + PICO_TCPOPTLEN_TIMESTAMP;
daniele	29:1a47b7151851	450	} else {
daniele	29:1a47b7151851	451
daniele	29:1a47b7151851	452	/* Always update window scale. */
tass	73:dfb737147f6e	453	size = (uint16_t)(size + PICO_TCPOPTLEN_WS);
daniele	29:1a47b7151851	454
daniele	29:1a47b7151851	455	if (t->ts_ok)
tass	73:dfb737147f6e	456	size = (uint16_t)(size + PICO_TCPOPTLEN_TIMESTAMP);
daniele	29:1a47b7151851	457
tass	73:dfb737147f6e	458	size= (uint16_t)(size + PICO_TCPOPTLEN_END);
daniele	29:1a47b7151851	459	}
daniele	29:1a47b7151851	460	if ((flags & PICO_TCP_ACK) && (t->sack_ok && sb)) {
tass	73:dfb737147f6e	461	size = (uint16_t)(size + 2);
daniele	29:1a47b7151851	462	while(sb) {
tass	73:dfb737147f6e	463	size = (uint16_t)(size + (2 * sizeof(uint32_t)));
daniele	29:1a47b7151851	464	sb = sb->next;
daniele	29:1a47b7151851	465	}
daniele	29:1a47b7151851	466	}
tass	70:cd218dd180e5	467	size = (uint16_t)(((size + 3) >> 2) << 2);
daniele	29:1a47b7151851	468	return size;
daniele	29:1a47b7151851	469	}
daniele	29:1a47b7151851	470
tass	70:cd218dd180e5	471	uint16_t pico_tcp_overhead(struct pico_socket *s)
daniele	29:1a47b7151851	472	{
daniele	29:1a47b7151851	473	if (!s)
daniele	29:1a47b7151851	474	return 0;
daniele	29:1a47b7151851	475
tass	70:cd218dd180e5	476	return (uint16_t)(PICO_SIZE_TCPHDR + tcp_options_size((struct pico_socket_tcp )s, (uint16_t)0)); / hdr + Options size for data pkt */
daniele	29:1a47b7151851	477
daniele	29:1a47b7151851	478	}
daniele	29:1a47b7151851	479
daniele	29:1a47b7151851	480	static void tcp_process_sack(struct pico_socket_tcp *t, uint32_t start, uint32_t end)
daniele	29:1a47b7151851	481	{
daniele	29:1a47b7151851	482	struct pico_frame *f;
daniele	29:1a47b7151851	483	struct pico_tree_node * index, * temp;
daniele	29:1a47b7151851	484	int cmp;
tass	68:0847e35d08a6	485	uint16_t count = 0;
daniele	29:1a47b7151851	486
daniele	29:1a47b7151851	487	pico_tree_foreach_safe(index,&t->tcpq_out.pool,temp){
daniele	29:1a47b7151851	488	f = index->keyValue;
daniele	29:1a47b7151851	489	cmp = seq_compare(SEQN(f), start);
daniele	29:1a47b7151851	490	if (cmp > 0)
daniele	29:1a47b7151851	491	goto done;
daniele	29:1a47b7151851	492
daniele	29:1a47b7151851	493	if (cmp == 0) {
daniele	29:1a47b7151851	494	cmp = seq_compare(SEQN(f) + f->payload_len, end);
daniele	29:1a47b7151851	495	if (cmp > 0) {
daniele	29:1a47b7151851	496	tcp_dbg("Invalid SACK: ignoring.\n");
daniele	29:1a47b7151851	497	}
daniele	29:1a47b7151851	498
daniele	29:1a47b7151851	499	tcp_dbg("Marking (by SACK) segment %08x BLK:[%08x::%08x]\n", SEQN(f), start, end);
daniele	29:1a47b7151851	500	f->flags \|= PICO_FRAME_FLAG_SACKED;
daniele	29:1a47b7151851	501	count++;
daniele	29:1a47b7151851	502
daniele	29:1a47b7151851	503	if (cmp == 0) {
daniele	29:1a47b7151851	504	/* that was last segment sacked. Job done */
daniele	29:1a47b7151851	505	goto done;
daniele	29:1a47b7151851	506	}
daniele	29:1a47b7151851	507	}
daniele	29:1a47b7151851	508	}
daniele	29:1a47b7151851	509
daniele	29:1a47b7151851	510	done:
daniele	29:1a47b7151851	511	if (t->x_mode > PICO_TCP_LOOKAHEAD) {
daniele	29:1a47b7151851	512	if (t->in_flight > (count))
daniele	29:1a47b7151851	513	t->in_flight -= (count);
daniele	29:1a47b7151851	514	else
daniele	29:1a47b7151851	515	t->in_flight = 0;
daniele	29:1a47b7151851	516	}
daniele	29:1a47b7151851	517	}
daniele	29:1a47b7151851	518
tass	70:cd218dd180e5	519	inline static void tcp_add_header(struct pico_socket_tcp t, struct pico_frame f)
tass	70:cd218dd180e5	520	{
tass	70:cd218dd180e5	521	struct pico_tcp_hdr * hdr = (struct pico_tcp_hdr *)f->transport_hdr;
tass	70:cd218dd180e5	522	f->timestamp = pico_tick;
tass	70:cd218dd180e5	523	tcp_add_options(t, f, 0, (uint16_t)(f->transport_len - f->payload_len - (uint16_t)PICO_SIZE_TCPHDR));
tass	70:cd218dd180e5	524	hdr->rwnd = short_be(t->wnd);
tass	70:cd218dd180e5	525	hdr->flags \|= PICO_TCP_PSH;
tass	70:cd218dd180e5	526	hdr->ack = long_be(t->rcv_nxt);
tass	70:cd218dd180e5	527	hdr->crc = 0;
tass	70:cd218dd180e5	528	hdr->crc = short_be(pico_tcp_checksum_ipv4(f));
tass	70:cd218dd180e5	529	}
tass	70:cd218dd180e5	530
daniele	29:1a47b7151851	531	static void tcp_rcv_sack(struct pico_socket_tcp t, uint8_t opt, int len)
daniele	29:1a47b7151851	532	{
daniele	29:1a47b7151851	533	uint32_t start, end;
daniele	29:1a47b7151851	534	int i = 0;
daniele	29:1a47b7151851	535	if (len % 8) {
daniele	29:1a47b7151851	536	tcp_dbg("SACK: Invalid len.\n");
daniele	29:1a47b7151851	537	return;
daniele	29:1a47b7151851	538	}
daniele	29:1a47b7151851	539	while (i < len) {
daniele	29:1a47b7151851	540	start = long_from(opt + i);
daniele	29:1a47b7151851	541	i += 4;
daniele	29:1a47b7151851	542	end = long_from(opt + i);
daniele	29:1a47b7151851	543	i += 4;
daniele	29:1a47b7151851	544	tcp_process_sack(t, long_be(start), long_be(end));
daniele	29:1a47b7151851	545	}
daniele	29:1a47b7151851	546	}
daniele	29:1a47b7151851	547
daniele	29:1a47b7151851	548	static void tcp_parse_options(struct pico_frame *f)
daniele	29:1a47b7151851	549	{
daniele	29:1a47b7151851	550	struct pico_socket_tcp t = (struct pico_socket_tcp )f->sock;
daniele	29:1a47b7151851	551	uint8_t *opt = f->transport_hdr + PICO_SIZE_TCPHDR;
tass	68:0847e35d08a6	552	uint32_t i = 0;
tass	45:ca30069e49bb	553	f->timestamp = 0;
daniele	29:1a47b7151851	554	while (i < (f->transport_len - PICO_SIZE_TCPHDR)) {
daniele	29:1a47b7151851	555	uint8_t type = opt[i++];
daniele	29:1a47b7151851	556	uint8_t len;
daniele	29:1a47b7151851	557	if(i < (f->transport_len - PICO_SIZE_TCPHDR) && (type > 1))
daniele	29:1a47b7151851	558	len = opt[i++];
daniele	29:1a47b7151851	559	else
daniele	29:1a47b7151851	560	len = 1;
daniele	29:1a47b7151851	561	if (f->payload && ((opt + i) > f->payload))
daniele	29:1a47b7151851	562	break;
tass	45:ca30069e49bb	563	tcp_dbg_options("Received option '%d', len = %d \n", type, len);
daniele	29:1a47b7151851	564	switch (type) {
daniele	29:1a47b7151851	565	case PICO_TCP_OPTION_NOOP:
daniele	29:1a47b7151851	566	case PICO_TCP_OPTION_END:
daniele	29:1a47b7151851	567	break;
daniele	29:1a47b7151851	568	case PICO_TCP_OPTION_WS:
daniele	29:1a47b7151851	569	if (len != PICO_TCPOPTLEN_WS) {
tass	45:ca30069e49bb	570	tcp_dbg_options("TCP Window scale: bad len received (%d).\n", len);
tass	73:dfb737147f6e	571	i = i + len - 2;
daniele	29:1a47b7151851	572	break;
daniele	29:1a47b7151851	573	}
daniele	29:1a47b7151851	574	t->recv_wnd_scale = opt[i++];
tass	45:ca30069e49bb	575	tcp_dbg_options("TCP Window scale: received %d\n", t->recv_wnd_scale);
daniele	29:1a47b7151851	576	break;
daniele	29:1a47b7151851	577	case PICO_TCP_OPTION_SACK_OK:
daniele	29:1a47b7151851	578	if (len != PICO_TCPOPTLEN_SACK_OK) {
tass	45:ca30069e49bb	579	tcp_dbg_options("TCP option sack: bad len received.\n");
tass	73:dfb737147f6e	580	i = i + len - 2;
daniele	29:1a47b7151851	581	break;
daniele	29:1a47b7151851	582	}
daniele	29:1a47b7151851	583	t->sack_ok = 1;
daniele	29:1a47b7151851	584	break;
daniele	29:1a47b7151851	585	case PICO_TCP_OPTION_MSS: {
daniele	29:1a47b7151851	586	uint16_t mss;
daniele	29:1a47b7151851	587	if (len != PICO_TCPOPTLEN_MSS) {
tass	45:ca30069e49bb	588	tcp_dbg_options("TCP option mss: bad len received.\n");
tass	73:dfb737147f6e	589	i = i + len - 2;
daniele	29:1a47b7151851	590	break;
daniele	29:1a47b7151851	591	}
daniele	29:1a47b7151851	592	t->mss_ok = 1;
daniele	29:1a47b7151851	593	mss = short_from(opt + i);
tass	74:c146c4e346c4	594	i += (uint32_t)sizeof(uint16_t);
daniele	29:1a47b7151851	595	if (t->mss > short_be(mss))
daniele	29:1a47b7151851	596	t->mss = short_be(mss);
daniele	29:1a47b7151851	597	break;
daniele	29:1a47b7151851	598	}
daniele	29:1a47b7151851	599	case PICO_TCP_OPTION_TIMESTAMP: {
daniele	29:1a47b7151851	600	uint32_t tsval, tsecr;
daniele	29:1a47b7151851	601	if (len != PICO_TCPOPTLEN_TIMESTAMP) {
tass	45:ca30069e49bb	602	tcp_dbg_options("TCP option timestamp: bad len received.\n");
tass	73:dfb737147f6e	603	i = i + len - 2;
daniele	29:1a47b7151851	604	break;
daniele	29:1a47b7151851	605	}
daniele	29:1a47b7151851	606	t->ts_ok = 1;
daniele	29:1a47b7151851	607	tsval = long_from(opt + i);
tass	74:c146c4e346c4	608	i += (uint32_t)sizeof(uint32_t);
daniele	29:1a47b7151851	609	tsecr = long_from(opt + i);
daniele	29:1a47b7151851	610	f->timestamp = long_be(tsecr);
tass	74:c146c4e346c4	611	i += (uint32_t)sizeof(uint32_t);
daniele	29:1a47b7151851	612	t->ts_nxt = long_be(tsval);
daniele	29:1a47b7151851	613	break;
daniele	29:1a47b7151851	614	}
daniele	29:1a47b7151851	615	case PICO_TCP_OPTION_SACK:
daniele	29:1a47b7151851	616	{
daniele	29:1a47b7151851	617	tcp_rcv_sack(t, opt + i, len - 2);
tass	73:dfb737147f6e	618	i = i + len - 2;
daniele	29:1a47b7151851	619	break;
daniele	29:1a47b7151851	620	}
daniele	29:1a47b7151851	621	default:
tass	45:ca30069e49bb	622	tcp_dbg_options("TCP: received unsupported option %u\n", type);
tass	73:dfb737147f6e	623	i = i + len - 2;
daniele	29:1a47b7151851	624	}
daniele	29:1a47b7151851	625	}
daniele	29:1a47b7151851	626	}
daniele	29:1a47b7151851	627
daniele	29:1a47b7151851	628	static int tcp_send(struct pico_socket_tcp ts, struct pico_frame f)
daniele	29:1a47b7151851	629	{
daniele	29:1a47b7151851	630	struct pico_tcp_hdr hdr= (struct pico_tcp_hdr ) f->transport_hdr;
daniele	29:1a47b7151851	631	struct pico_frame *cpy;
daniele	29:1a47b7151851	632	hdr->trans.sport = ts->sock.local_port;
daniele	29:1a47b7151851	633	hdr->trans.dport = ts->sock.remote_port;
daniele	29:1a47b7151851	634	if (!hdr->seq)
daniele	29:1a47b7151851	635	hdr->seq = long_be(ts->snd_nxt);
daniele	29:1a47b7151851	636
daniele	29:1a47b7151851	637	if (ts->rcv_nxt != 0) {
daniele	29:1a47b7151851	638	if ( (ts->rcv_ackd == 0) \|\| (seq_compare(ts->rcv_ackd, ts->rcv_nxt) != 0) \|\| (hdr->flags & PICO_TCP_ACK)) {
daniele	29:1a47b7151851	639	hdr->flags \|= PICO_TCP_ACK;
daniele	29:1a47b7151851	640	hdr->ack = long_be(ts->rcv_nxt);
daniele	29:1a47b7151851	641	ts->rcv_ackd = ts->rcv_nxt;
daniele	29:1a47b7151851	642	}
daniele	29:1a47b7151851	643	}
daniele	29:1a47b7151851	644
daniele	29:1a47b7151851	645	if (hdr->flags & PICO_TCP_SYN) {
daniele	29:1a47b7151851	646	ts->snd_nxt++;
daniele	29:1a47b7151851	647	}
daniele	29:1a47b7151851	648	if (f->payload_len > 0) {
daniele	29:1a47b7151851	649	hdr->flags \|= PICO_TCP_PSH \| PICO_TCP_ACK;
daniele	29:1a47b7151851	650	hdr->ack = long_be(ts->rcv_nxt);
daniele	29:1a47b7151851	651	ts->rcv_ackd = ts->rcv_nxt;
tass	40:c8ab0d2bba0b	652	ts->keepalive_timer_running = 2; /* XXX TODO check fix: added 1 to counter to postpone sending keepalive, ACK is in data segments */
daniele	29:1a47b7151851	653	}
daniele	29:1a47b7151851	654
daniele	29:1a47b7151851	655	f->start = f->transport_hdr + PICO_SIZE_TCPHDR;
daniele	29:1a47b7151851	656	hdr->rwnd = short_be(ts->wnd);
daniele	29:1a47b7151851	657	hdr->crc = 0;
daniele	29:1a47b7151851	658	hdr->crc = short_be(pico_tcp_checksum_ipv4(f));
daniele	29:1a47b7151851	659
daniele	29:1a47b7151851	660	/* TCP: ENQUEUE to PROTO ( Transmit ) */
daniele	29:1a47b7151851	661	cpy = pico_frame_copy(f);
tass	40:c8ab0d2bba0b	662	if ((pico_enqueue(&tcp_out, cpy) > 0)) {
tass	40:c8ab0d2bba0b	663	if (f->payload_len > 0) {
daniele	29:1a47b7151851	664	ts->in_flight++;
tass	40:c8ab0d2bba0b	665	ts->snd_nxt += f->payload_len; /* update next pointer here to prevent sending same segment twice when called twice in same tick */
tass	40:c8ab0d2bba0b	666	}
daniele	29:1a47b7151851	667	tcp_dbg("DBG> [tcp output] state: %02x --> local port:%d remote port: %d seq: %08x ack: %08x flags: %02x = t_len: %d, hdr: %u payload: %d\n",
daniele	29:1a47b7151851	668	TCPSTATE(&ts->sock) >> 8, short_be(hdr->trans.sport), short_be(hdr->trans.dport), SEQN(f), ACKN(f), hdr->flags, f->transport_len, (hdr->len & 0xf0) >> 2 , f->payload_len );
daniele	29:1a47b7151851	669	} else {
daniele	29:1a47b7151851	670	pico_frame_discard(cpy);
daniele	29:1a47b7151851	671	}
daniele	29:1a47b7151851	672	return 0;
daniele	29:1a47b7151851	673	}
daniele	29:1a47b7151851	674
daniele	29:1a47b7151851	675	//#define PICO_TCP_SUPPORT_SOCKET_STATS
daniele	29:1a47b7151851	676
daniele	29:1a47b7151851	677	#ifdef PICO_TCP_SUPPORT_SOCKET_STATS
tass	73:dfb737147f6e	678	static void sock_stats(uint32_t when, void *arg)
daniele	29:1a47b7151851	679	{
daniele	29:1a47b7151851	680	struct pico_socket_tcp t = (struct pico_socket_tcp )arg;
daniele	29:1a47b7151851	681	tcp_dbg("STATISTIC> [%lu] socket state: %02x --> local port:%d remote port: %d queue size: %d snd_una: %08x snd_nxt: %08x timer: %d cwnd: %d\n",
daniele	29:1a47b7151851	682	when, t->sock.state, short_be(t->sock.local_port), short_be(t->sock.remote_port), t->tcpq_out.size, SEQN(first_segment(&t->tcpq_out)), t->snd_nxt, t->timer_running, t->cwnd);
daniele	29:1a47b7151851	683	pico_timer_add(2000, sock_stats, t);
daniele	29:1a47b7151851	684	}
daniele	29:1a47b7151851	685	#endif
daniele	29:1a47b7151851	686
daniele	29:1a47b7151851	687	struct pico_socket *pico_tcp_open(void)
daniele	29:1a47b7151851	688	{
daniele	29:1a47b7151851	689	struct pico_socket_tcp *t = pico_zalloc(sizeof(struct pico_socket_tcp));
daniele	29:1a47b7151851	690	if (!t)
daniele	29:1a47b7151851	691	return NULL;
tass	68:0847e35d08a6	692	t->sock.timestamp = PICO_TIME_MS();
daniele	29:1a47b7151851	693	t->mss = PICO_TCP_DEFAULT_MSS;
daniele	29:1a47b7151851	694
daniele	29:1a47b7151851	695	t->tcpq_in.pool.root = t->tcpq_hold.pool.root = t->tcpq_out.pool.root = &LEAF;
daniele	29:1a47b7151851	696	t->tcpq_hold.pool.compare = t->tcpq_in.pool.compare = t->tcpq_out.pool.compare = segment_compare;
daniele	29:1a47b7151851	697
daniele	29:1a47b7151851	698	t->tcpq_in.max_size = PICO_DEFAULT_SOCKETQ;
daniele	29:1a47b7151851	699	t->tcpq_out.max_size = PICO_DEFAULT_SOCKETQ;
daniele	29:1a47b7151851	700	t->tcpq_hold.max_size = 2*PICO_TCP_DEFAULT_MSS;
daniele	29:1a47b7151851	701
tass	40:c8ab0d2bba0b	702	/* disable Nagle by default */
tass	68:0847e35d08a6	703	//t->sock.opt_flags \|= (1 << PICO_SOCKET_OPT_TCPNODELAY);
tass	68:0847e35d08a6	704	/* Nagle is enabled by default */
tass	73:dfb737147f6e	705	t->sock.opt_flags &= (uint16_t)~(1 << PICO_SOCKET_OPT_TCPNODELAY);
daniele	29:1a47b7151851	706
daniele	29:1a47b7151851	707	#ifdef PICO_TCP_SUPPORT_SOCKET_STATS
daniele	29:1a47b7151851	708	pico_timer_add(2000, sock_stats, t);
daniele	29:1a47b7151851	709	#endif
daniele	29:1a47b7151851	710	tcp_set_space(t);
daniele	29:1a47b7151851	711
daniele	29:1a47b7151851	712	return &t->sock;
daniele	29:1a47b7151851	713	}
daniele	29:1a47b7151851	714
tass	70:cd218dd180e5	715	uint32_t pico_tcp_read(struct pico_socket s, void buf, uint32_t len)
daniele	29:1a47b7151851	716	{
daniele	29:1a47b7151851	717	struct pico_socket_tcp *t = TCP_SOCK(s);
daniele	29:1a47b7151851	718	struct pico_frame *f;
daniele	29:1a47b7151851	719	uint32_t in_frame_off, in_frame_len;
tass	70:cd218dd180e5	720	uint32_t tot_rd_len = 0;
daniele	29:1a47b7151851	721
daniele	29:1a47b7151851	722	while (tot_rd_len < len) {
daniele	29:1a47b7151851	723	/* To be sure we don't have garbage at the beginning */
daniele	29:1a47b7151851	724	release_until(&t->tcpq_in, t->rcv_processed);
daniele	29:1a47b7151851	725	f = first_segment(&t->tcpq_in);
daniele	29:1a47b7151851	726	if (!f) {
daniele	29:1a47b7151851	727	tcp_set_space(t);
daniele	29:1a47b7151851	728	goto out;
daniele	29:1a47b7151851	729	}
daniele	29:1a47b7151851	730
daniele	29:1a47b7151851	731	/* Hole at the beginning of data, awaiting retransmissions. */
daniele	29:1a47b7151851	732	if (seq_compare(t->rcv_processed, SEQN(f)) < 0) {
daniele	29:1a47b7151851	733	tcp_dbg("TCP> read hole beginning of data, %08x - %08x. rcv_nxt is %08x\n",t->rcv_processed, SEQN(f), t->rcv_nxt);
daniele	29:1a47b7151851	734	goto out;
daniele	29:1a47b7151851	735	}
daniele	29:1a47b7151851	736
daniele	29:1a47b7151851	737	if(seq_compare(t->rcv_processed, SEQN(f)) > 0) {
daniele	29:1a47b7151851	738	in_frame_off = t->rcv_processed - SEQN(f);
daniele	29:1a47b7151851	739	in_frame_len = f->payload_len - in_frame_off;
daniele	29:1a47b7151851	740	} else {
daniele	29:1a47b7151851	741	in_frame_off = 0;
daniele	29:1a47b7151851	742	in_frame_len = f->payload_len;
daniele	29:1a47b7151851	743	}
tass	68:0847e35d08a6	744	if ((in_frame_len + tot_rd_len) > (uint32_t)len) {
daniele	29:1a47b7151851	745	in_frame_len = len - tot_rd_len;
daniele	29:1a47b7151851	746	}
daniele	29:1a47b7151851	747
daniele	29:1a47b7151851	748	if (in_frame_len > f->payload_len - in_frame_off)
daniele	29:1a47b7151851	749	in_frame_len = f->payload_len - in_frame_off;
daniele	29:1a47b7151851	750
daniele	29:1a47b7151851	751	memcpy((uint8_t *)buf + tot_rd_len, f->payload + in_frame_off, in_frame_len);
daniele	29:1a47b7151851	752	tot_rd_len += in_frame_len;
daniele	29:1a47b7151851	753	t->rcv_processed += in_frame_len;
daniele	29:1a47b7151851	754
tass	68:0847e35d08a6	755	if ((in_frame_len == 0u) \|\| (in_frame_len == (uint32_t)f->payload_len)) {
daniele	29:1a47b7151851	756	pico_discard_segment(&t->tcpq_in, f);
daniele	29:1a47b7151851	757	}
daniele	29:1a47b7151851	758	}
daniele	29:1a47b7151851	759
daniele	29:1a47b7151851	760	out:
daniele	29:1a47b7151851	761	tcp_set_space(t);
daniele	29:1a47b7151851	762	if (t->tcpq_in.size == 0) {
tass	70:cd218dd180e5	763	s->ev_pending &= (uint16_t)(~PICO_SOCK_EV_RD);
daniele	29:1a47b7151851	764	}
daniele	29:1a47b7151851	765	return tot_rd_len;
daniele	29:1a47b7151851	766	}
daniele	29:1a47b7151851	767
daniele	29:1a47b7151851	768	int pico_tcp_initconn(struct pico_socket *s);
tass	73:dfb737147f6e	769	static void initconn_retry(uint32_t when, void *arg)
daniele	29:1a47b7151851	770	{
daniele	29:1a47b7151851	771	struct pico_socket_tcp t = (struct pico_socket_tcp )arg;
tass	68:0847e35d08a6	772	IGNORE_PARAMETER(when);
daniele	29:1a47b7151851	773	if (TCPSTATE(&t->sock) == PICO_SOCKET_STATE_TCP_SYN_SENT) {
daniele	29:1a47b7151851	774	if (t->backoff > PICO_TCP_MAX_CONNECT_RETRIES) {
daniele	29:1a47b7151851	775	tcp_dbg("TCP> Connection timeout. \n");
daniele	29:1a47b7151851	776	if (t->sock.wakeup)
daniele	29:1a47b7151851	777	t->sock.wakeup(PICO_SOCK_EV_ERR, &t->sock);
daniele	29:1a47b7151851	778	return;
daniele	29:1a47b7151851	779	}
daniele	29:1a47b7151851	780	tcp_dbg("TCP> SYN retry %d...\n", t->backoff);
daniele	29:1a47b7151851	781	t->backoff++;
daniele	29:1a47b7151851	782	pico_tcp_initconn(&t->sock);
daniele	29:1a47b7151851	783	} else {
daniele	29:1a47b7151851	784	tcp_dbg("TCP> Connection is already established: no retry needed. good.\n");
daniele	29:1a47b7151851	785	}
daniele	29:1a47b7151851	786	}
daniele	29:1a47b7151851	787
daniele	29:1a47b7151851	788	int pico_tcp_initconn(struct pico_socket *s)
daniele	29:1a47b7151851	789	{
daniele	29:1a47b7151851	790	struct pico_socket_tcp *ts = TCP_SOCK(s);
daniele	29:1a47b7151851	791	struct pico_frame *syn;
daniele	29:1a47b7151851	792	struct pico_tcp_hdr *hdr;
tass	70:cd218dd180e5	793	uint16_t opt_len = tcp_options_size(ts, PICO_TCP_SYN);
daniele	29:1a47b7151851	794
tass	70:cd218dd180e5	795	syn = s->net->alloc(s->net, (uint16_t)(PICO_SIZE_TCPHDR + opt_len));
daniele	29:1a47b7151851	796	if (!syn)
daniele	29:1a47b7151851	797	return -1;
daniele	29:1a47b7151851	798	hdr = (struct pico_tcp_hdr *) syn->transport_hdr;
daniele	29:1a47b7151851	799
daniele	29:1a47b7151851	800	if (!ts->snd_nxt)
daniele	29:1a47b7151851	801	ts->snd_nxt = long_be(pico_paws());
daniele	29:1a47b7151851	802	ts->snd_last = ts->snd_nxt;
daniele	29:1a47b7151851	803	ts->cwnd = PICO_TCP_IW;
daniele	29:1a47b7151851	804	ts->ssthresh = 40;
daniele	29:1a47b7151851	805	syn->sock = s;
daniele	29:1a47b7151851	806	hdr->seq = long_be(ts->snd_nxt);
tass	70:cd218dd180e5	807	hdr->len = (uint8_t)((PICO_SIZE_TCPHDR + opt_len) << 2 \| ts->jumbo);
daniele	29:1a47b7151851	808	hdr->flags = PICO_TCP_SYN;
daniele	29:1a47b7151851	809	tcp_set_space(ts);
daniele	29:1a47b7151851	810	hdr->rwnd = short_be(ts->wnd);
daniele	29:1a47b7151851	811	tcp_add_options(ts,syn, PICO_TCP_SYN, opt_len);
daniele	29:1a47b7151851	812	hdr->trans.sport = ts->sock.local_port;
daniele	29:1a47b7151851	813	hdr->trans.dport = ts->sock.remote_port;
daniele	29:1a47b7151851	814
daniele	29:1a47b7151851	815	hdr->crc = 0;
daniele	29:1a47b7151851	816	hdr->crc = short_be(pico_tcp_checksum_ipv4(syn));
daniele	29:1a47b7151851	817
daniele	29:1a47b7151851	818	/* TCP: ENQUEUE to PROTO ( SYN ) */
daniele	29:1a47b7151851	819	tcp_dbg("Sending SYN... (ports: %d - %d) size: %d\n", short_be(ts->sock.local_port), short_be(ts->sock.remote_port), syn->buffer_len);
daniele	29:1a47b7151851	820	pico_enqueue(&tcp_out, syn);
tass	37:bdf736327c71	821	pico_timer_add(PICO_TCP_SYN_TO << ts->backoff, initconn_retry, ts);
daniele	29:1a47b7151851	822	return 0;
daniele	29:1a47b7151851	823	}
daniele	29:1a47b7151851	824
daniele	29:1a47b7151851	825	static int tcp_send_synack(struct pico_socket *s)
daniele	29:1a47b7151851	826	{
daniele	29:1a47b7151851	827	struct pico_socket_tcp *ts = TCP_SOCK(s);
daniele	29:1a47b7151851	828	struct pico_frame *synack;
daniele	29:1a47b7151851	829	struct pico_tcp_hdr *hdr;
tass	70:cd218dd180e5	830	uint16_t opt_len = tcp_options_size(ts, PICO_TCP_SYN \| PICO_TCP_ACK);
daniele	29:1a47b7151851	831
tass	70:cd218dd180e5	832	synack = s->net->alloc(s->net, (uint16_t)(PICO_SIZE_TCPHDR + opt_len));
daniele	29:1a47b7151851	833	if (!synack)
daniele	29:1a47b7151851	834	return -1;
daniele	29:1a47b7151851	835	hdr = (struct pico_tcp_hdr *) synack->transport_hdr;
daniele	29:1a47b7151851	836
daniele	29:1a47b7151851	837	synack->sock = s;
tass	70:cd218dd180e5	838	hdr->len = (uint8_t)((PICO_SIZE_TCPHDR + opt_len) << 2 \| ts->jumbo);
daniele	29:1a47b7151851	839	hdr->flags = PICO_TCP_SYN \| PICO_TCP_ACK;
daniele	29:1a47b7151851	840	hdr->rwnd = short_be(ts->wnd);
daniele	29:1a47b7151851	841	hdr->seq = long_be(ts->snd_nxt);
daniele	29:1a47b7151851	842	ts->rcv_processed = long_be(hdr->seq);
daniele	29:1a47b7151851	843	ts->snd_last = ts->snd_nxt;
daniele	29:1a47b7151851	844	tcp_set_space(ts);
daniele	29:1a47b7151851	845	tcp_add_options(ts,synack, hdr->flags, opt_len);
daniele	29:1a47b7151851	846	synack->payload_len = 0;
daniele	29:1a47b7151851	847	synack->timestamp = pico_tick;
daniele	29:1a47b7151851	848	tcp_send(ts, synack);
daniele	29:1a47b7151851	849	pico_frame_discard(synack);
daniele	29:1a47b7151851	850	return 0;
daniele	29:1a47b7151851	851	}
daniele	29:1a47b7151851	852
daniele	29:1a47b7151851	853	static void tcp_send_empty(struct pico_socket_tcp *t, uint16_t flags)
daniele	29:1a47b7151851	854	{
daniele	29:1a47b7151851	855	struct pico_frame *f;
daniele	29:1a47b7151851	856	struct pico_tcp_hdr *hdr;
tass	70:cd218dd180e5	857	uint16_t opt_len = tcp_options_size(t, flags);
tass	70:cd218dd180e5	858	f = t->sock.net->alloc(t->sock.net, (uint16_t)(PICO_SIZE_TCPHDR + opt_len));
daniele	29:1a47b7151851	859	if (!f) {
daniele	29:1a47b7151851	860	return;
daniele	29:1a47b7151851	861	}
daniele	29:1a47b7151851	862	f->sock = &t->sock;
daniele	29:1a47b7151851	863	hdr = (struct pico_tcp_hdr *) f->transport_hdr;
tass	70:cd218dd180e5	864	hdr->len = (uint8_t)((PICO_SIZE_TCPHDR + opt_len) << 2 \| t->jumbo);
tass	70:cd218dd180e5	865	hdr->flags = (uint8_t)flags;
daniele	29:1a47b7151851	866	hdr->rwnd = short_be(t->wnd);
daniele	29:1a47b7151851	867	tcp_set_space(t);
daniele	29:1a47b7151851	868	tcp_add_options(t,f, flags, opt_len);
daniele	29:1a47b7151851	869	hdr->trans.sport = t->sock.local_port;
daniele	29:1a47b7151851	870	hdr->trans.dport = t->sock.remote_port;
daniele	29:1a47b7151851	871	hdr->seq = long_be(t->snd_nxt);
daniele	29:1a47b7151851	872	if ((flags & PICO_TCP_ACK) != 0)
daniele	29:1a47b7151851	873	hdr->ack = long_be(t->rcv_nxt);
daniele	29:1a47b7151851	874	t->rcv_ackd = t->rcv_nxt;
daniele	29:1a47b7151851	875
daniele	29:1a47b7151851	876	f->start = f->transport_hdr + PICO_SIZE_TCPHDR;
daniele	29:1a47b7151851	877	hdr->rwnd = short_be(t->wnd);
daniele	29:1a47b7151851	878	hdr->crc = 0;
daniele	29:1a47b7151851	879	hdr->crc = short_be(pico_tcp_checksum_ipv4(f));
daniele	29:1a47b7151851	880
daniele	29:1a47b7151851	881	/* TCP: ENQUEUE to PROTO */
daniele	29:1a47b7151851	882	pico_enqueue(&tcp_out, f);
daniele	29:1a47b7151851	883	}
daniele	29:1a47b7151851	884
daniele	29:1a47b7151851	885	static void tcp_send_ack(struct pico_socket_tcp *t)
daniele	29:1a47b7151851	886	{
daniele	29:1a47b7151851	887	return tcp_send_empty(t, PICO_TCP_ACK);
daniele	29:1a47b7151851	888	}
daniele	29:1a47b7151851	889
daniele	29:1a47b7151851	890	static int tcp_send_rst(struct pico_socket s, struct pico_frame fr)
daniele	29:1a47b7151851	891	{
daniele	29:1a47b7151851	892	struct pico_socket_tcp t = (struct pico_socket_tcp ) s;
daniele	29:1a47b7151851	893	struct pico_frame *f;
daniele	29:1a47b7151851	894	struct pico_tcp_hdr hdr, hdr_rcv;
tass	70:cd218dd180e5	895	uint16_t opt_len = tcp_options_size(t, PICO_TCP_RST);
daniele	29:1a47b7151851	896	int close;
daniele	29:1a47b7151851	897
daniele	29:1a47b7151851	898	tcp_dbg("TCP SEND_RST >>>>>>>>>>>>>>> START\n");
daniele	29:1a47b7151851	899
tass	70:cd218dd180e5	900	f = t->sock.net->alloc(t->sock.net, (uint16_t)(PICO_SIZE_TCPHDR + opt_len));
daniele	29:1a47b7151851	901
daniele	29:1a47b7151851	902	if (!f) {
daniele	29:1a47b7151851	903	return -1;
daniele	29:1a47b7151851	904	}
daniele	29:1a47b7151851	905
daniele	29:1a47b7151851	906	hdr_rcv = (struct pico_tcp_hdr *) fr->transport_hdr;
daniele	29:1a47b7151851	907
daniele	29:1a47b7151851	908	f->sock = &t->sock;
daniele	29:1a47b7151851	909	hdr = (struct pico_tcp_hdr *) f->transport_hdr;
tass	70:cd218dd180e5	910	hdr->len = (uint8_t)((PICO_SIZE_TCPHDR + opt_len) << 2 \| t->jumbo);
daniele	29:1a47b7151851	911	hdr->flags = PICO_TCP_RST;
daniele	29:1a47b7151851	912	hdr->rwnd = short_be(t->wnd);
daniele	29:1a47b7151851	913	tcp_set_space(t);
daniele	29:1a47b7151851	914	tcp_add_options(t,f, PICO_TCP_RST, opt_len);
daniele	29:1a47b7151851	915	hdr->trans.sport = t->sock.local_port;
daniele	29:1a47b7151851	916	hdr->trans.dport = t->sock.remote_port;
daniele	29:1a47b7151851	917	hdr->seq = long_be(t->snd_nxt);
daniele	29:1a47b7151851	918
daniele	29:1a47b7151851	919	/* check if state is synchronized */
daniele	29:1a47b7151851	920	if (((s->state & PICO_SOCKET_STATE_TCP) > PICO_SOCKET_STATE_TCP_SYN_RECV)) {
daniele	29:1a47b7151851	921	/* in synchronized state: send RST with seq = ack from previous segment */
daniele	29:1a47b7151851	922	hdr->seq = hdr_rcv->ack;
daniele	29:1a47b7151851	923	close = 0;
daniele	29:1a47b7151851	924	} else {
daniele	29:1a47b7151851	925	/* non-synchronized state */
daniele	29:1a47b7151851	926	/* go to CLOSED here to prevent timer callback to go on after timeout */
daniele	29:1a47b7151851	927	(t->sock).state &= 0x00FFU;
daniele	29:1a47b7151851	928	(t->sock).state \|= PICO_SOCKET_STATE_TCP_CLOSED;
daniele	29:1a47b7151851	929	close = 1;
daniele	29:1a47b7151851	930	}
daniele	29:1a47b7151851	931
daniele	29:1a47b7151851	932	hdr->ack = long_be(t->rcv_nxt);
daniele	29:1a47b7151851	933	t->rcv_ackd = t->rcv_nxt;
daniele	29:1a47b7151851	934	f->start = f->transport_hdr + PICO_SIZE_TCPHDR;
daniele	29:1a47b7151851	935	hdr->rwnd = short_be(t->wnd);
daniele	29:1a47b7151851	936	hdr->crc = 0;
daniele	29:1a47b7151851	937	hdr->crc = short_be(pico_tcp_checksum_ipv4(f));
daniele	29:1a47b7151851	938
daniele	29:1a47b7151851	939	/* TCP: ENQUEUE to PROTO */
daniele	29:1a47b7151851	940	pico_enqueue(&tcp_out, f);
daniele	29:1a47b7151851	941
daniele	29:1a47b7151851	942	/* goto CLOSED */
daniele	29:1a47b7151851	943	if (close) {
daniele	29:1a47b7151851	944	(t->sock).state &= 0xFF00U;
daniele	29:1a47b7151851	945	(t->sock).state \|= PICO_SOCKET_STATE_CLOSED;
daniele	29:1a47b7151851	946
daniele	29:1a47b7151851	947	/* call EV_FIN wakeup before deleting */
daniele	29:1a47b7151851	948	if ((t->sock).wakeup)
daniele	29:1a47b7151851	949	(t->sock).wakeup(PICO_SOCK_EV_FIN, &(t->sock));
daniele	29:1a47b7151851	950
daniele	29:1a47b7151851	951	/* delete socket */
daniele	29:1a47b7151851	952	pico_socket_del(&t->sock);
daniele	29:1a47b7151851	953
daniele	29:1a47b7151851	954	tcp_dbg("TCP SEND_RST >>>>>>>>>>>>>>> DONE, deleted socket\n");
daniele	29:1a47b7151851	955	}
daniele	29:1a47b7151851	956
daniele	29:1a47b7151851	957	return 0;
daniele	29:1a47b7151851	958	}
daniele	29:1a47b7151851	959
daniele	29:1a47b7151851	960	int pico_tcp_reply_rst(struct pico_frame *fr)
daniele	29:1a47b7151851	961	{
daniele	29:1a47b7151851	962	struct pico_tcp_hdr *hdr;
daniele	29:1a47b7151851	963	struct pico_frame *f;
tass	70:cd218dd180e5	964	uint16_t size = PICO_SIZE_TCPHDR;
daniele	29:1a47b7151851	965
daniele	29:1a47b7151851	966	tcp_dbg("TCP>>>>>>>>>>>>>>>> sending RST ... <<<<<<<<<<<<<<<<<<\n");
daniele	29:1a47b7151851	967
daniele	29:1a47b7151851	968	f = fr->sock->net->alloc(fr->sock->net, size);
daniele	29:1a47b7151851	969
daniele	29:1a47b7151851	970	/* fill in IP data from original frame */
daniele	29:1a47b7151851	971	// TODO if IPv4
daniele	29:1a47b7151851	972	((struct pico_ipv4_hdr )(f->net_hdr))->dst.addr = ((struct pico_ipv4_hdr )(fr->net_hdr))->src.addr;
daniele	29:1a47b7151851	973	((struct pico_ipv4_hdr )(f->net_hdr))->src.addr = ((struct pico_ipv4_hdr )(fr->net_hdr))->dst.addr;
daniele	29:1a47b7151851	974
daniele	29:1a47b7151851	975	/* fill in TCP data from original frame */
daniele	29:1a47b7151851	976	((struct pico_tcp_hdr )(f->transport_hdr))->trans.dport = ((struct pico_tcp_hdr )(fr->transport_hdr))->trans.sport;
daniele	29:1a47b7151851	977	((struct pico_tcp_hdr )(f->transport_hdr))->trans.sport = ((struct pico_tcp_hdr )(fr->transport_hdr))->trans.dport;
daniele	29:1a47b7151851	978	hdr = (struct pico_tcp_hdr *) f->transport_hdr;
tass	70:cd218dd180e5	979	hdr->len = (uint8_t)(size << 2);
daniele	29:1a47b7151851	980	hdr->flags = PICO_TCP_RST \| PICO_TCP_ACK;
daniele	29:1a47b7151851	981	hdr->rwnd = 0;
daniele	29:1a47b7151851	982	if (((struct pico_tcp_hdr *)(fr->transport_hdr))->flags & PICO_TCP_ACK) {
daniele	29:1a47b7151851	983	hdr->seq = ((struct pico_tcp_hdr *)(fr->transport_hdr))->ack;
daniele	29:1a47b7151851	984	} else {
daniele	29:1a47b7151851	985	hdr->seq = 0U;
daniele	29:1a47b7151851	986	}
daniele	29:1a47b7151851	987
tass	93:e7a58195bfb7	988	hdr->ack = ((struct pico_tcp_hdr *)(fr->transport_hdr))->seq + long_be(fr->payload_len);
tass	93:e7a58195bfb7	989	hdr->crc = short_be(pico_tcp_checksum_ipv4(f));
tass	93:e7a58195bfb7	990
daniele	29:1a47b7151851	991	/* enqueue for transmission */
daniele	29:1a47b7151851	992	pico_ipv4_frame_push(f,&(((struct pico_ipv4_hdr *)(f->net_hdr))->dst),PICO_PROTO_TCP);
daniele	29:1a47b7151851	993
daniele	29:1a47b7151851	994	return 0;
daniele	29:1a47b7151851	995	}
daniele	29:1a47b7151851	996
daniele	29:1a47b7151851	997	static int tcp_nosync_rst(struct pico_socket s, struct pico_frame fr)
daniele	29:1a47b7151851	998	{
daniele	29:1a47b7151851	999	struct pico_socket_tcp t = (struct pico_socket_tcp ) s;
daniele	29:1a47b7151851	1000	struct pico_frame *f;
daniele	29:1a47b7151851	1001	struct pico_tcp_hdr hdr, hdr_rcv;
tass	70:cd218dd180e5	1002	uint16_t opt_len = tcp_options_size(t, PICO_TCP_RST \| PICO_TCP_ACK);
daniele	29:1a47b7151851	1003
daniele	29:1a47b7151851	1004	tcp_dbg("TCP SEND RST (NON-SYNC) >>>>>>>>>>>>>>>>>> state %x\n",(s->state & PICO_SOCKET_STATE_TCP));
daniele	29:1a47b7151851	1005	if (((s->state & PICO_SOCKET_STATE_TCP) == PICO_SOCKET_STATE_TCP_LISTEN)) {
daniele	29:1a47b7151851	1006	/* XXX TODO NOTE: to prevent the parent socket from trying to send, because this socket has no knowledge of dst IP !!! */
daniele	29:1a47b7151851	1007	return pico_tcp_reply_rst(fr);
daniele	29:1a47b7151851	1008	}
daniele	29:1a47b7151851	1009
daniele	29:1a47b7151851	1010	/***************************************************************************/
daniele	29:1a47b7151851	1011	/* sending RST */
tass	70:cd218dd180e5	1012	f = t->sock.net->alloc(t->sock.net, (uint16_t)(PICO_SIZE_TCPHDR + opt_len));
daniele	29:1a47b7151851	1013
daniele	29:1a47b7151851	1014	if (!f) {
daniele	29:1a47b7151851	1015	return -1;
daniele	29:1a47b7151851	1016	}
daniele	29:1a47b7151851	1017
daniele	29:1a47b7151851	1018	hdr_rcv = (struct pico_tcp_hdr *) fr->transport_hdr;
daniele	29:1a47b7151851	1019
daniele	29:1a47b7151851	1020	f->sock = &t->sock;
daniele	29:1a47b7151851	1021	hdr = (struct pico_tcp_hdr *) f->transport_hdr;
tass	70:cd218dd180e5	1022	hdr->len = (uint8_t)((PICO_SIZE_TCPHDR + opt_len) << 2 \| t->jumbo);
daniele	29:1a47b7151851	1023	hdr->flags = PICO_TCP_RST \| PICO_TCP_ACK;
daniele	29:1a47b7151851	1024	hdr->rwnd = short_be(t->wnd);
daniele	29:1a47b7151851	1025	tcp_set_space(t);
daniele	29:1a47b7151851	1026	tcp_add_options(t,f, PICO_TCP_RST \| PICO_TCP_ACK, opt_len);
daniele	29:1a47b7151851	1027	hdr->trans.sport = t->sock.local_port;
daniele	29:1a47b7151851	1028	hdr->trans.dport = t->sock.remote_port;
daniele	29:1a47b7151851	1029
daniele	29:1a47b7151851	1030	/* non-synchronized state */
daniele	29:1a47b7151851	1031	if (hdr_rcv->flags & PICO_TCP_ACK) {
daniele	29:1a47b7151851	1032	hdr->seq = hdr_rcv->ack;
daniele	29:1a47b7151851	1033	} else {
daniele	29:1a47b7151851	1034	hdr->seq = 0U;
daniele	29:1a47b7151851	1035	}
daniele	29:1a47b7151851	1036
tass	93:e7a58195bfb7	1037	hdr->ack = hdr_rcv->seq + long_be(fr->payload_len);
daniele	29:1a47b7151851	1038
daniele	29:1a47b7151851	1039	t->rcv_ackd = t->rcv_nxt;
daniele	29:1a47b7151851	1040	f->start = f->transport_hdr + PICO_SIZE_TCPHDR;
daniele	29:1a47b7151851	1041	hdr->rwnd = short_be(t->wnd);
daniele	29:1a47b7151851	1042	hdr->crc = 0;
daniele	29:1a47b7151851	1043	hdr->crc = short_be(pico_tcp_checksum_ipv4(f));
daniele	29:1a47b7151851	1044
daniele	29:1a47b7151851	1045	/* TCP: ENQUEUE to PROTO */
daniele	29:1a47b7151851	1046	pico_enqueue(&tcp_out, f);
daniele	29:1a47b7151851	1047
daniele	29:1a47b7151851	1048	/***************************************************************************/
daniele	29:1a47b7151851	1049
daniele	29:1a47b7151851	1050	tcp_dbg("TCP SEND_RST (NON_SYNC) >>>>>>>>>>>>>>> DONE, ...\n");
daniele	29:1a47b7151851	1051
daniele	29:1a47b7151851	1052	return 0;
daniele	29:1a47b7151851	1053	}
daniele	29:1a47b7151851	1054
daniele	29:1a47b7151851	1055	static void tcp_send_fin(struct pico_socket_tcp *t)
daniele	29:1a47b7151851	1056	{
daniele	29:1a47b7151851	1057	struct pico_frame *f;
daniele	29:1a47b7151851	1058	struct pico_tcp_hdr *hdr;
tass	70:cd218dd180e5	1059	uint16_t opt_len = tcp_options_size(t, PICO_TCP_FIN);
tass	70:cd218dd180e5	1060	f = t->sock.net->alloc(t->sock.net, (uint16_t)(PICO_SIZE_TCPHDR + opt_len));
daniele	29:1a47b7151851	1061	if (!f) {
daniele	29:1a47b7151851	1062	return;
daniele	29:1a47b7151851	1063	}
daniele	29:1a47b7151851	1064	f->sock = &t->sock;
daniele	29:1a47b7151851	1065	hdr = (struct pico_tcp_hdr *) f->transport_hdr;
tass	70:cd218dd180e5	1066	hdr->len = (uint8_t)((PICO_SIZE_TCPHDR + opt_len) << 2 \| t->jumbo);
daniele	29:1a47b7151851	1067	hdr->flags = PICO_TCP_FIN \| PICO_TCP_ACK;
daniele	29:1a47b7151851	1068	hdr->ack = long_be(t->rcv_nxt);
daniele	29:1a47b7151851	1069	t->rcv_ackd = t->rcv_nxt;
daniele	29:1a47b7151851	1070	hdr->rwnd = short_be(t->wnd);
daniele	29:1a47b7151851	1071	tcp_set_space(t);
daniele	29:1a47b7151851	1072	tcp_add_options(t,f, PICO_TCP_FIN, opt_len);
daniele	29:1a47b7151851	1073	hdr->trans.sport = t->sock.local_port;
daniele	29:1a47b7151851	1074	hdr->trans.dport = t->sock.remote_port;
tass	40:c8ab0d2bba0b	1075	hdr->seq = long_be(t->snd_nxt); /* XXX TODO check correct ?? --> snd_last? otherwise maybe data after FIN */
daniele	29:1a47b7151851	1076
daniele	29:1a47b7151851	1077	f->start = f->transport_hdr + PICO_SIZE_TCPHDR;
daniele	29:1a47b7151851	1078	hdr->rwnd = short_be(t->wnd);
daniele	29:1a47b7151851	1079	hdr->crc = 0;
daniele	29:1a47b7151851	1080	hdr->crc = short_be(pico_tcp_checksum_ipv4(f));
daniele	29:1a47b7151851	1081	//tcp_dbg("SENDING FIN...\n");
daniele	29:1a47b7151851	1082	/* TCP: ENQUEUE to PROTO ( Pure ACK ) */
daniele	29:1a47b7151851	1083	pico_enqueue(&tcp_out, f);
daniele	29:1a47b7151851	1084	t->snd_nxt++;
daniele	29:1a47b7151851	1085	}
daniele	29:1a47b7151851	1086
daniele	29:1a47b7151851	1087	static void tcp_sack_prepare(struct pico_socket_tcp *t)
daniele	29:1a47b7151851	1088	{
daniele	29:1a47b7151851	1089	struct pico_frame *pkt;
daniele	29:1a47b7151851	1090	uint32_t left=0, right=0;
daniele	29:1a47b7151851	1091	struct tcp_sack_block *sb;
daniele	29:1a47b7151851	1092	int n = 0;
daniele	29:1a47b7151851	1093	if (t->sacks) /* previous sacks are pending */
daniele	29:1a47b7151851	1094	return;
daniele	29:1a47b7151851	1095
daniele	29:1a47b7151851	1096	pkt = first_segment(&t->tcpq_in);
daniele	29:1a47b7151851	1097	while(n < 3) {
daniele	29:1a47b7151851	1098	if (!pkt) {
daniele	29:1a47b7151851	1099	if(left) {
daniele	29:1a47b7151851	1100	sb = pico_zalloc(sizeof(struct tcp_sack_block));
daniele	29:1a47b7151851	1101	if (!sb)
daniele	29:1a47b7151851	1102	break;
daniele	29:1a47b7151851	1103	sb->left = long_be(left);
daniele	29:1a47b7151851	1104	sb->right = long_be(right);
daniele	29:1a47b7151851	1105	n++;
daniele	29:1a47b7151851	1106	sb->next = t->sacks;
daniele	29:1a47b7151851	1107	t->sacks = sb;
daniele	29:1a47b7151851	1108	left = 0;
daniele	29:1a47b7151851	1109	right = 0;
daniele	29:1a47b7151851	1110	}
daniele	29:1a47b7151851	1111	break;
daniele	29:1a47b7151851	1112	}
daniele	29:1a47b7151851	1113	if ((SEQN(pkt) < t->rcv_nxt)) {
daniele	29:1a47b7151851	1114	pkt = next_segment(&t->tcpq_in, pkt);
daniele	29:1a47b7151851	1115	continue;
daniele	29:1a47b7151851	1116	}
daniele	29:1a47b7151851	1117	if (!left) {
daniele	29:1a47b7151851	1118	left = SEQN(pkt);
daniele	29:1a47b7151851	1119	right = SEQN(pkt) + pkt->payload_len;
daniele	29:1a47b7151851	1120	pkt = next_segment(&t->tcpq_in, pkt);
daniele	29:1a47b7151851	1121	continue;
daniele	29:1a47b7151851	1122	}
daniele	29:1a47b7151851	1123	if(SEQN(pkt) == (right + 1)) {
daniele	29:1a47b7151851	1124	right += pkt->payload_len;
daniele	29:1a47b7151851	1125	pkt = next_segment(&t->tcpq_in, pkt);
daniele	29:1a47b7151851	1126	continue;
daniele	29:1a47b7151851	1127	} else {
daniele	29:1a47b7151851	1128	sb = pico_zalloc(sizeof(struct tcp_sack_block));
daniele	29:1a47b7151851	1129	if (!sb)
daniele	29:1a47b7151851	1130	break;
daniele	29:1a47b7151851	1131	sb->left = long_be(left);
daniele	29:1a47b7151851	1132	sb->right = long_be(right);
daniele	29:1a47b7151851	1133	n++;
daniele	29:1a47b7151851	1134	sb->next = t->sacks;
daniele	29:1a47b7151851	1135	t->sacks = sb;
daniele	29:1a47b7151851	1136	left = 0;
daniele	29:1a47b7151851	1137	right = 0;
daniele	29:1a47b7151851	1138	pkt = next_segment(&t->tcpq_in, pkt);
daniele	29:1a47b7151851	1139	}
daniele	29:1a47b7151851	1140	}
daniele	29:1a47b7151851	1141	}
daniele	29:1a47b7151851	1142
daniele	29:1a47b7151851	1143	static int tcp_data_in(struct pico_socket s, struct pico_frame f)
daniele	29:1a47b7151851	1144	{
daniele	29:1a47b7151851	1145	struct pico_socket_tcp t = (struct pico_socket_tcp )s;
daniele	29:1a47b7151851	1146	struct pico_tcp_hdr hdr = (struct pico_tcp_hdr ) f->transport_hdr;
daniele	29:1a47b7151851	1147
daniele	29:1a47b7151851	1148	if (((hdr->len & 0xf0) >> 2) <= f->transport_len) {
daniele	29:1a47b7151851	1149	tcp_parse_options(f);
daniele	29:1a47b7151851	1150	f->payload = f->transport_hdr + ((hdr->len & 0xf0) >>2);
tass	70:cd218dd180e5	1151	f->payload_len = (uint16_t)(f->transport_len - ((hdr->len & 0xf0) >>2));
tass	45:ca30069e49bb	1152	tcp_dbg("TCP> Received segment. (exp: %x got: %x)\n", t->rcv_nxt, SEQN(f));
daniele	29:1a47b7151851	1153
tass	45:ca30069e49bb	1154	if (seq_compare(SEQN(f), t->rcv_nxt) <= 0) {
daniele	29:1a47b7151851	1155	struct pico_frame *nxt;
daniele	29:1a47b7151851	1156	if (seq_compare(SEQN(f), t->rcv_nxt) == 0) { /* Exactly what we expected */
tass	45:ca30069e49bb	1157	struct pico_frame *cpy = pico_frame_copy(f);
tass	45:ca30069e49bb	1158	/* Enqueue: try to put into RCV buffer */
tass	45:ca30069e49bb	1159	if(pico_enqueue_segment(&t->tcpq_in, cpy) <= 0) {
tass	45:ca30069e49bb	1160	pico_frame_discard(cpy);
tass	45:ca30069e49bb	1161	}
daniele	29:1a47b7151851	1162	t->rcv_nxt = SEQN(f) + f->payload_len;
daniele	29:1a47b7151851	1163	nxt = peek_segment(&t->tcpq_in, t->rcv_nxt);
daniele	29:1a47b7151851	1164	while(nxt) {
daniele	29:1a47b7151851	1165	tcp_dbg("scrolling rcv_nxt...%08x\n", t->rcv_nxt);
daniele	29:1a47b7151851	1166	t->rcv_nxt += f->payload_len;
daniele	29:1a47b7151851	1167	nxt = peek_segment(&t->tcpq_in, t->rcv_nxt);
daniele	29:1a47b7151851	1168	}
daniele	29:1a47b7151851	1169	t->sock.ev_pending \|= PICO_SOCK_EV_RD;
daniele	29:1a47b7151851	1170	t->rcv_nxt = SEQN(f) + f->payload_len;
daniele	29:1a47b7151851	1171	} else {
tass	45:ca30069e49bb	1172	tcp_dbg("TCP> lo segment. Uninteresting retransmission. (exp: %x got: %x)\n", t->rcv_nxt, SEQN(f));
daniele	29:1a47b7151851	1173	}
daniele	29:1a47b7151851	1174	} else {
daniele	29:1a47b7151851	1175	tcp_dbg("TCP> hi segment. Possible packet loss. I'll dupack this. (exp: %x got: %x)\n", t->rcv_nxt, SEQN(f));
daniele	29:1a47b7151851	1176	if (t->sack_ok) {
daniele	29:1a47b7151851	1177	tcp_sack_prepare(t);
daniele	29:1a47b7151851	1178	}
daniele	29:1a47b7151851	1179	}
daniele	29:1a47b7151851	1180	/* In either case, ack til recv_nxt. */
daniele	29:1a47b7151851	1181	if ( ((t->sock.state & PICO_SOCKET_STATE_TCP) != PICO_SOCKET_STATE_TCP_CLOSE_WAIT) && ((t->sock.state & PICO_SOCKET_STATE_TCP) != PICO_SOCKET_STATE_TCP_SYN_SENT) && ((t->sock.state & PICO_SOCKET_STATE_TCP) != PICO_SOCKET_STATE_TCP_SYN_RECV)) {
daniele	29:1a47b7151851	1182	//tcp_dbg("SENDACK CALLED FROM OUTSIDE tcp_synack, state %x\n",t->sock.state);
daniele	29:1a47b7151851	1183	tcp_send_ack(t);
daniele	29:1a47b7151851	1184	} else {
daniele	29:1a47b7151851	1185	//tcp_dbg("SENDACK PREVENTED IN SYNSENT STATE\n");
daniele	29:1a47b7151851	1186	}
daniele	29:1a47b7151851	1187	return 0;
daniele	29:1a47b7151851	1188	} else {
daniele	29:1a47b7151851	1189	tcp_dbg("TCP: invalid data in pkt len, exp: %d, got %d\n", (hdr->len & 0xf0) >> 2, f->transport_len);
daniele	29:1a47b7151851	1190	return -1;
daniele	29:1a47b7151851	1191	}
daniele	29:1a47b7151851	1192	}
daniele	29:1a47b7151851	1193
daniele	29:1a47b7151851	1194	static int tcp_ack_advance_una(struct pico_socket_tcp t, struct pico_frame f)
daniele	29:1a47b7151851	1195	{
daniele	29:1a47b7151851	1196	int ret = release_all_until(&t->tcpq_out, ACKN(f));
daniele	29:1a47b7151851	1197	if (ret > 0)
daniele	29:1a47b7151851	1198	t->sock.ev_pending \|= PICO_SOCK_EV_WR;
daniele	29:1a47b7151851	1199	return ret;
daniele	29:1a47b7151851	1200	}
daniele	29:1a47b7151851	1201
tass	73:dfb737147f6e	1202	static uint16_t time_diff(uint32_t a, uint32_t b)
daniele	29:1a47b7151851	1203	{
daniele	29:1a47b7151851	1204	if (a >= b)
tass	73:dfb737147f6e	1205	return (uint16_t)(a - b);
daniele	29:1a47b7151851	1206	else
tass	73:dfb737147f6e	1207	return (uint16_t)(b - a);
daniele	29:1a47b7151851	1208	}
daniele	29:1a47b7151851	1209
daniele	29:1a47b7151851	1210	static void tcp_rtt(struct pico_socket_tcp *t, uint32_t rtt)
daniele	29:1a47b7151851	1211	{
daniele	29:1a47b7151851	1212
daniele	29:1a47b7151851	1213	uint32_t avg = t->avg_rtt;
daniele	29:1a47b7151851	1214	uint32_t rvar = t->rttvar;
daniele	29:1a47b7151851	1215	if (!avg) {
daniele	29:1a47b7151851	1216	/* This follows RFC2988
daniele	29:1a47b7151851	1217	* (2.2) When the first RTT measurement R is made, the host MUST set
daniele	29:1a47b7151851	1218	*
daniele	29:1a47b7151851	1219	* SRTT <- R
daniele	29:1a47b7151851	1220	* RTTVAR <- R/2
daniele	29:1a47b7151851	1221	* RTO <- SRTT + max (G, K*RTTVAR)
daniele	29:1a47b7151851	1222	*/
daniele	29:1a47b7151851	1223	t->avg_rtt = rtt;
daniele	29:1a47b7151851	1224	t->rttvar = rtt >> 1;
daniele	29:1a47b7151851	1225	t->rto = t->avg_rtt + (t->rttvar << 4);
daniele	29:1a47b7151851	1226	} else {
tass	70:cd218dd180e5	1227	int32_t var = (int32_t)(t->avg_rtt - rtt);
daniele	29:1a47b7151851	1228	if (var < 0)
daniele	29:1a47b7151851	1229	var = 0-var;
daniele	29:1a47b7151851	1230	/* RFC2988, section (2.3). Alpha and beta are the ones suggested. */
daniele	29:1a47b7151851	1231
daniele	29:1a47b7151851	1232	/* First, evaluate a new value for the rttvar */
daniele	29:1a47b7151851	1233	t->rttvar <<= 2;
daniele	29:1a47b7151851	1234	t->rttvar -= rvar;
tass	70:cd218dd180e5	1235	t->rttvar += (uint32_t)var;
daniele	29:1a47b7151851	1236	t->rttvar >>= 2;
daniele	29:1a47b7151851	1237
daniele	29:1a47b7151851	1238	/* Then, calculate the new avg_rtt */
daniele	29:1a47b7151851	1239	t->avg_rtt <<= 3;
daniele	29:1a47b7151851	1240	t->avg_rtt -= avg;
daniele	29:1a47b7151851	1241	t->avg_rtt += rtt;
daniele	29:1a47b7151851	1242	t->avg_rtt >>= 3;
daniele	29:1a47b7151851	1243
daniele	29:1a47b7151851	1244	/* Finally, assign a new value for the RTO, as specified in the RFC, with K=4 */
daniele	29:1a47b7151851	1245	t->rto = t->avg_rtt + (t->rttvar << 2);
daniele	29:1a47b7151851	1246	}
tass	45:ca30069e49bb	1247	tcp_dbg(" -----=============== RTT CUR: %u AVG: %u RTTVAR: %u RTO: %u ======================----\n", rtt, t->avg_rtt, t->rttvar, t->rto);
daniele	29:1a47b7151851	1248	}
daniele	29:1a47b7151851	1249
daniele	29:1a47b7151851	1250	static void tcp_congestion_control(struct pico_socket_tcp *t)
daniele	29:1a47b7151851	1251	{
daniele	29:1a47b7151851	1252	if (t->x_mode > PICO_TCP_LOOKAHEAD)
daniele	29:1a47b7151851	1253	return;
daniele	29:1a47b7151851	1254	if (t->cwnd > t->tcpq_out.frames) {
daniele	29:1a47b7151851	1255	tcp_dbg("Limited by app: %d\n", t->cwnd);
tass	45:ca30069e49bb	1256	if (t->sock.wakeup)
tass	45:ca30069e49bb	1257	t->sock.wakeup(PICO_SOCK_EV_WR, &t->sock);
daniele	29:1a47b7151851	1258	return;
daniele	29:1a47b7151851	1259	}
daniele	29:1a47b7151851	1260	tcp_dbg("Doing congestion control\n");
daniele	29:1a47b7151851	1261	if (t->cwnd < t->ssthresh) {
daniele	29:1a47b7151851	1262	t->cwnd++;
daniele	29:1a47b7151851	1263	} else {
daniele	29:1a47b7151851	1264	t->cwnd_counter++;
daniele	29:1a47b7151851	1265	if (t->cwnd_counter >= t->cwnd) {
daniele	29:1a47b7151851	1266	t->cwnd++;
daniele	29:1a47b7151851	1267	t->cwnd_counter = 0;
daniele	29:1a47b7151851	1268	}
daniele	29:1a47b7151851	1269	}
daniele	29:1a47b7151851	1270	tcp_dbg("TCP_CWND, %lu, %u, %u, %u\n", pico_tick, t->cwnd, t->ssthresh, t->in_flight);
daniele	29:1a47b7151851	1271	}
daniele	29:1a47b7151851	1272
tass	73:dfb737147f6e	1273	static void add_retransmission_timer(struct pico_socket_tcp *t, uint32_t next_ts);
tass	73:dfb737147f6e	1274	static void tcp_retrans_timeout(uint32_t val, void *sock)
daniele	29:1a47b7151851	1275	{
daniele	29:1a47b7151851	1276	struct pico_socket_tcp t = (struct pico_socket_tcp ) sock;
daniele	29:1a47b7151851	1277	struct pico_frame *f = NULL;
tass	73:dfb737147f6e	1278	uint32_t limit = val - t->rto;
tass	32:865c101e0874	1279	if( t->sock.net && ((t->sock.state & 0xFF00) == PICO_SOCKET_STATE_TCP_ESTABLISHED
tass	68:0847e35d08a6	1280	\|\| (t->sock.state & 0xFF00) == PICO_SOCKET_STATE_TCP_CLOSE_WAIT) )
daniele	29:1a47b7151851	1281	{
tass	68:0847e35d08a6	1282	tcp_dbg("TIMEOUT! backoff = %d\n", t->backoff);
tass	68:0847e35d08a6	1283	/* was timer cancelled? */
tass	68:0847e35d08a6	1284	if (t->timer_running == 0) {
tass	68:0847e35d08a6	1285	add_retransmission_timer(t, 0);
tass	68:0847e35d08a6	1286	return;
tass	68:0847e35d08a6	1287	}
tass	68:0847e35d08a6	1288	t->timer_running--;
daniele	29:1a47b7151851	1289
tass	68:0847e35d08a6	1290	f = first_segment(&t->tcpq_out);
tass	68:0847e35d08a6	1291	while (f) {
tass	68:0847e35d08a6	1292	if ((t->x_mode == PICO_TCP_WINDOW_FULL) \|\|
tass	68:0847e35d08a6	1293	((f->timestamp != 0) && (f->timestamp <= limit))) {
tass	68:0847e35d08a6	1294	struct pico_frame *cpy;
tass	68:0847e35d08a6	1295	tcp_dbg("TCP BLACKOUT> TIMED OUT (output) frame %08x, len= %d rto=%d Win full: %d frame flags: %04x\n", SEQN(f), f->payload_len, t->rto, t->x_mode == PICO_TCP_WINDOW_FULL, f->flags);
tass	68:0847e35d08a6	1296	if ((t->x_mode != PICO_TCP_WINDOW_FULL) ) {
tass	68:0847e35d08a6	1297	t->x_mode = PICO_TCP_BLACKOUT;
tass	68:0847e35d08a6	1298	tcp_dbg("Mode: Blackout.\n");
tass	68:0847e35d08a6	1299	t->cwnd = PICO_TCP_IW;
tass	68:0847e35d08a6	1300	t->in_flight = 0;
tass	68:0847e35d08a6	1301	}
tass	70:cd218dd180e5	1302	tcp_add_header(t,f);
tass	68:0847e35d08a6	1303	/* TCP: ENQUEUE to PROTO ( retransmit )*/
tass	68:0847e35d08a6	1304	cpy = pico_frame_copy(f);
tass	68:0847e35d08a6	1305	if (pico_enqueue(&tcp_out, cpy) > 0) {
tass	68:0847e35d08a6	1306	t->backoff++;
tass	68:0847e35d08a6	1307	add_retransmission_timer(t, (t->rto << t->backoff) + pico_tick);
tass	68:0847e35d08a6	1308	tcp_dbg("TCP_CWND, %lu, %u, %u, %u\n", pico_tick, t->cwnd, t->ssthresh, t->in_flight);
tass	68:0847e35d08a6	1309	return;
tass	68:0847e35d08a6	1310	} else {
tass	68:0847e35d08a6	1311	add_retransmission_timer(t, (t->rto << t->backoff) + pico_tick);
tass	68:0847e35d08a6	1312	pico_frame_discard(cpy);
tass	68:0847e35d08a6	1313	}
tass	68:0847e35d08a6	1314	}
tass	68:0847e35d08a6	1315	f = next_segment(&t->tcpq_out, f);
tass	68:0847e35d08a6	1316	}
tass	68:0847e35d08a6	1317	t->backoff = 0;
tass	68:0847e35d08a6	1318	add_retransmission_timer(t, 0);
tass	68:0847e35d08a6	1319	if (t->tcpq_out.size < t->tcpq_out.max_size)
tass	68:0847e35d08a6	1320	t->sock.ev_pending \|= PICO_SOCK_EV_WR;
tass	68:0847e35d08a6	1321	return;
tass	68:0847e35d08a6	1322	}
daniele	29:1a47b7151851	1323	}
daniele	29:1a47b7151851	1324
tass	73:dfb737147f6e	1325	static void add_retransmission_timer(struct pico_socket_tcp *t, uint32_t next_ts)
daniele	29:1a47b7151851	1326	{
daniele	29:1a47b7151851	1327	struct pico_tree_node * index;
daniele	29:1a47b7151851	1328
daniele	29:1a47b7151851	1329	if (t->timer_running > 0)
daniele	29:1a47b7151851	1330	return;
daniele	29:1a47b7151851	1331
daniele	29:1a47b7151851	1332	if (next_ts == 0) {
daniele	29:1a47b7151851	1333	struct pico_frame *f;
daniele	29:1a47b7151851	1334
daniele	29:1a47b7151851	1335	pico_tree_foreach(index,&t->tcpq_out.pool){
daniele	29:1a47b7151851	1336	f = index->keyValue;
daniele	29:1a47b7151851	1337	if (((next_ts == 0) \|\| (f->timestamp < next_ts)) && (f->timestamp > 0)) {
daniele	29:1a47b7151851	1338	next_ts = f->timestamp;
daniele	29:1a47b7151851	1339	}
daniele	29:1a47b7151851	1340	}
daniele	29:1a47b7151851	1341	}
daniele	29:1a47b7151851	1342	if (next_ts > 0) {
daniele	29:1a47b7151851	1343	if ((next_ts + t->rto) > pico_tick) {
tass	70:cd218dd180e5	1344	t->retrans_tmr = pico_timer_add(next_ts + t->rto - pico_tick, tcp_retrans_timeout, t);
daniele	29:1a47b7151851	1345	} else {
tass	70:cd218dd180e5	1346	t->retrans_tmr = pico_timer_add(1, tcp_retrans_timeout, t);
daniele	29:1a47b7151851	1347	}
daniele	29:1a47b7151851	1348	t->timer_running++;
daniele	29:1a47b7151851	1349	}
daniele	29:1a47b7151851	1350	}
daniele	29:1a47b7151851	1351
daniele	29:1a47b7151851	1352	static int tcp_retrans(struct pico_socket_tcp t, struct pico_frame f)
daniele	29:1a47b7151851	1353	{
daniele	29:1a47b7151851	1354	struct pico_frame *cpy;
daniele	29:1a47b7151851	1355	if (f) {
daniele	29:1a47b7151851	1356	tcp_dbg("TCP> RETRANS (by dupack) frame %08x, len= %d\n", SEQN(f), f->payload_len);
tass	70:cd218dd180e5	1357	tcp_add_header(t,f);
daniele	29:1a47b7151851	1358	/* TCP: ENQUEUE to PROTO ( retransmit )*/
daniele	29:1a47b7151851	1359	cpy = pico_frame_copy(f);
daniele	29:1a47b7151851	1360	if (pico_enqueue(&tcp_out, cpy) > 0) {
daniele	29:1a47b7151851	1361	t->in_flight++;
daniele	29:1a47b7151851	1362	t->snd_last_out = SEQN(cpy);
daniele	29:1a47b7151851	1363	add_retransmission_timer(t, pico_tick + t->rto);
daniele	29:1a47b7151851	1364	} else {
daniele	29:1a47b7151851	1365	pico_frame_discard(cpy);
daniele	29:1a47b7151851	1366	}
daniele	29:1a47b7151851	1367	return(f->payload_len);
daniele	29:1a47b7151851	1368	}
daniele	29:1a47b7151851	1369	return 0;
daniele	29:1a47b7151851	1370	}
daniele	29:1a47b7151851	1371
daniele	29:1a47b7151851	1372	#ifdef TCP_ACK_DBG
daniele	29:1a47b7151851	1373	static void tcp_ack_dbg(struct pico_socket s, struct pico_frame f)
daniele	29:1a47b7151851	1374	{
daniele	29:1a47b7151851	1375	uint32_t una, nxt, ack, cur;
daniele	29:1a47b7151851	1376	struct pico_frame una_f = NULL, cur_f;
daniele	29:1a47b7151851	1377	struct pico_tree_node *idx;
daniele	29:1a47b7151851	1378	struct pico_socket_tcp t = (struct pico_socket_tcp )s;
daniele	29:1a47b7151851	1379	char info[64];
daniele	29:1a47b7151851	1380	char tmp[64];
daniele	29:1a47b7151851	1381	ack = ACKN(f);
daniele	29:1a47b7151851	1382	nxt = t->snd_nxt;
daniele	29:1a47b7151851	1383	tcp_dbg("===================================\n");
daniele	29:1a47b7151851	1384	tcp_dbg("Queue out (%d/%d). ACKED=%08x\n", t->tcpq_out.size, t->tcpq_out.max_size, ack);
daniele	29:1a47b7151851	1385
daniele	29:1a47b7151851	1386	pico_tree_foreach(idx, &t->tcpq_out.pool) {
daniele	29:1a47b7151851	1387	info[0] = 0;
daniele	29:1a47b7151851	1388	cur_f = idx->keyValue;
daniele	29:1a47b7151851	1389	cur = SEQN(cur_f);
daniele	29:1a47b7151851	1390	if (!una_f) {
daniele	29:1a47b7151851	1391	una_f = cur_f;
daniele	29:1a47b7151851	1392	una = SEQN(una_f);
daniele	29:1a47b7151851	1393	}
daniele	29:1a47b7151851	1394
daniele	29:1a47b7151851	1395	if (cur == nxt) {
daniele	29:1a47b7151851	1396	strncpy(tmp, info, strlen(info));
daniele	29:1a47b7151851	1397	snprintf(info,64, "%s SND_NXT", tmp);
daniele	29:1a47b7151851	1398	}
daniele	29:1a47b7151851	1399	if (cur == ack) {
daniele	29:1a47b7151851	1400	strncpy(tmp, info, strlen(info));
daniele	29:1a47b7151851	1401	snprintf(info,64, "%s ACK", tmp);
daniele	29:1a47b7151851	1402	}
daniele	29:1a47b7151851	1403	if (cur == una) {
daniele	29:1a47b7151851	1404	strncpy(tmp, info, strlen(info));
daniele	29:1a47b7151851	1405	snprintf(info,64, "%s SND_UNA", tmp);
daniele	29:1a47b7151851	1406	}
daniele	29:1a47b7151851	1407	if (cur == t->snd_last) {
daniele	29:1a47b7151851	1408	strncpy(tmp, info, strlen(info));
daniele	29:1a47b7151851	1409	snprintf(info,64, "%s SND_LAST", tmp);
daniele	29:1a47b7151851	1410	}
daniele	29:1a47b7151851	1411	tcp_dbg("%08x %d%s\n", cur, cur_f->payload_len, info);
daniele	29:1a47b7151851	1412
daniele	29:1a47b7151851	1413	}
tass	82:93915e13262a	1414	tcp_dbg("SND_NXT is %08x, snd_LAST is %08x\n", nxt, t->snd_last);
daniele	29:1a47b7151851	1415	tcp_dbg("===================================\n");
daniele	29:1a47b7151851	1416	tcp_dbg("\n\n");
daniele	29:1a47b7151851	1417	}
daniele	29:1a47b7151851	1418	#endif
daniele	29:1a47b7151851	1419
daniele	29:1a47b7151851	1420	static int tcp_ack(struct pico_socket s, struct pico_frame f)
daniele	29:1a47b7151851	1421	{
daniele	29:1a47b7151851	1422	struct pico_frame f_new; / use with Nagle to push to out queue */
daniele	29:1a47b7151851	1423	struct pico_socket_tcp t = (struct pico_socket_tcp )s;
daniele	29:1a47b7151851	1424	struct pico_tcp_hdr hdr = (struct pico_tcp_hdr ) f->transport_hdr;
daniele	29:1a47b7151851	1425	uint32_t rtt = 0;
tass	68:0847e35d08a6	1426	uint16_t acked = 0;
daniele	29:1a47b7151851	1427	struct pico_frame *una = NULL;
daniele	29:1a47b7151851	1428	if ((hdr->flags & PICO_TCP_ACK) == 0)
daniele	29:1a47b7151851	1429	return -1;
daniele	29:1a47b7151851	1430
daniele	29:1a47b7151851	1431	#ifdef TCP_ACK_DBG
daniele	29:1a47b7151851	1432	tcp_ack_dbg(s,f);
daniele	29:1a47b7151851	1433	#endif
daniele	29:1a47b7151851	1434
daniele	29:1a47b7151851	1435	tcp_parse_options(f);
daniele	29:1a47b7151851	1436	t->recv_wnd = short_be(hdr->rwnd);
daniele	29:1a47b7151851	1437
tass	70:cd218dd180e5	1438	acked = (uint16_t)tcp_ack_advance_una(t, f);
daniele	29:1a47b7151851	1439	una = first_segment(&t->tcpq_out);
daniele	29:1a47b7151851	1440
daniele	29:1a47b7151851	1441	if ((t->x_mode == PICO_TCP_BLACKOUT) \|\|
daniele	29:1a47b7151851	1442	((t->x_mode == PICO_TCP_WINDOW_FULL) && ((t->recv_wnd << t->recv_wnd_scale) > t->mss))) {
daniele	29:1a47b7151851	1443	tcp_dbg("Re-entering look-ahead...\n\n\n");
daniele	29:1a47b7151851	1444	t->x_mode = PICO_TCP_LOOKAHEAD;
daniele	29:1a47b7151851	1445	t->backoff = 0;
daniele	29:1a47b7151851	1446	}
daniele	29:1a47b7151851	1447
daniele	29:1a47b7151851	1448	/* One should be acked. */
daniele	29:1a47b7151851	1449	// if ((acked == 0) && (t->in_flight > 0))
daniele	29:1a47b7151851	1450	if ((acked == 0) && (f->payload_len == 0) && (t->in_flight > 0))
daniele	29:1a47b7151851	1451	t->in_flight--;
daniele	29:1a47b7151851	1452	if (!una \|\| acked > 0) {
daniele	29:1a47b7151851	1453	t->x_mode = PICO_TCP_LOOKAHEAD;
daniele	29:1a47b7151851	1454	tcp_dbg("Mode: Look-ahead. In flight: %d/%d buf: %d\n", t->in_flight, t->cwnd, t->tcpq_out.frames);
daniele	29:1a47b7151851	1455	t->backoff = 0;
daniele	29:1a47b7151851	1456
daniele	29:1a47b7151851	1457	/* Do rtt/rttvar/rto calculations */
tass	45:ca30069e49bb	1458	/* First, try with timestamps, using the value from options */
tass	45:ca30069e49bb	1459	if(f && (f->timestamp != 0)) {
tass	45:ca30069e49bb	1460	rtt = time_diff(pico_tick, f->timestamp);
daniele	29:1a47b7151851	1461	if (rtt)
daniele	29:1a47b7151851	1462	tcp_rtt(t, rtt);
tass	45:ca30069e49bb	1463	} else if(una && (una->timestamp != 0)) {
tass	45:ca30069e49bb	1464	/* If no timestamps are there, use conservatve estimation on the una */
tass	45:ca30069e49bb	1465	rtt = time_diff(pico_tick, una->timestamp);
tass	45:ca30069e49bb	1466	if (rtt)
tass	45:ca30069e49bb	1467	tcp_rtt(t, rtt);
daniele	29:1a47b7151851	1468	}
daniele	29:1a47b7151851	1469
daniele	29:1a47b7151851	1470	tcp_dbg("TCP ACK> FRESH ACK %08x (acked %d) Queue size: %u/%u frames: %u cwnd: %u in_flight: %u snd_una: %u\n", ACKN(f), acked, t->tcpq_out.size, t->tcpq_out.max_size, t->tcpq_out.frames, t->cwnd, t->in_flight, SEQN(una));
daniele	29:1a47b7151851	1471	if (acked > t->in_flight) {
daniele	29:1a47b7151851	1472	tcp_dbg("WARNING: in flight < 0\n");
daniele	29:1a47b7151851	1473	t->in_flight = 0;
daniele	29:1a47b7151851	1474	} else
daniele	29:1a47b7151851	1475	t->in_flight -= (acked);
daniele	29:1a47b7151851	1476
daniele	29:1a47b7151851	1477	} else if ((t->snd_old_ack == ACKN(f)) && /* We've just seen this ack, and... */
daniele	29:1a47b7151851	1478	((0 == (hdr->flags & (PICO_TCP_PSH \| PICO_TCP_SYN))) &&
daniele	29:1a47b7151851	1479	(f->payload_len == 0)) && /* This is a pure ack, and... */
daniele	29:1a47b7151851	1480	(ACKN(f) != t->snd_nxt)) /* There is something in flight awaiting to be acked... */
daniele	29:1a47b7151851	1481	{
daniele	29:1a47b7151851	1482	/* Process incoming duplicate ack. */
daniele	29:1a47b7151851	1483	if (t->x_mode < PICO_TCP_RECOVER) {
daniele	29:1a47b7151851	1484	t->x_mode++;
daniele	29:1a47b7151851	1485	tcp_dbg("Mode: DUPACK %d, due to PURE ACK %0x, len = %d\n", t->x_mode, SEQN(f), f->payload_len);
daniele	29:1a47b7151851	1486	tcp_dbg("ACK: %x - QUEUE: %x\n",ACKN(f), SEQN(first_segment(&t->tcpq_out)));
daniele	29:1a47b7151851	1487	if (t->x_mode == PICO_TCP_RECOVER) { /* Switching mode */
daniele	29:1a47b7151851	1488	t->snd_retry = SEQN(first_segment(&t->tcpq_out));
daniele	29:1a47b7151851	1489	if (t->ssthresh > t->cwnd)
daniele	29:1a47b7151851	1490	t->ssthresh >>=2;
daniele	29:1a47b7151851	1491	else
daniele	29:1a47b7151851	1492	t->ssthresh = (t->cwnd >> 1);
daniele	29:1a47b7151851	1493	if (t->ssthresh < 2)
daniele	29:1a47b7151851	1494	t->ssthresh = 2;
daniele	29:1a47b7151851	1495	}
daniele	29:1a47b7151851	1496	} else if (t->x_mode == PICO_TCP_RECOVER) {
daniele	29:1a47b7151851	1497	tcp_dbg("TCP RECOVER> DUPACK! snd_una: %08x, snd_nxt: %08x, acked now: %08x\n", SEQN(first_segment(&t->tcpq_out)), t->snd_nxt, ACKN(f));
daniele	29:1a47b7151851	1498	if (t->in_flight <= t->cwnd) {
daniele	29:1a47b7151851	1499	struct pico_frame *nxt = peek_segment(&t->tcpq_out, t->snd_retry);
daniele	29:1a47b7151851	1500	if (!nxt)
daniele	29:1a47b7151851	1501	nxt = first_segment(&t->tcpq_out);
daniele	29:1a47b7151851	1502
daniele	29:1a47b7151851	1503	while (nxt && (nxt->flags & PICO_FRAME_FLAG_SACKED) && (nxt != first_segment(&t->tcpq_out))) {
daniele	29:1a47b7151851	1504	tcp_dbg("Skipping %08x because it is sacked.\n", SEQN(nxt));
daniele	29:1a47b7151851	1505	nxt = next_segment(&t->tcpq_out, nxt);
daniele	29:1a47b7151851	1506	}
daniele	29:1a47b7151851	1507
daniele	29:1a47b7151851	1508	if (nxt && (seq_compare(SEQN(nxt), t->snd_nxt)) > 0)
daniele	29:1a47b7151851	1509	nxt = NULL;
daniele	29:1a47b7151851	1510	if (nxt && (seq_compare(SEQN(nxt), SEQN(first_segment(&t->tcpq_out))) > (t->recv_wnd << t->recv_wnd_scale)))
daniele	29:1a47b7151851	1511	nxt = NULL;
daniele	29:1a47b7151851	1512
daniele	29:1a47b7151851	1513	if(!nxt)
daniele	29:1a47b7151851	1514	nxt = first_segment(&t->tcpq_out);
daniele	29:1a47b7151851	1515	if (nxt) {
daniele	29:1a47b7151851	1516	tcp_retrans(t, peek_segment(&t->tcpq_out, t->snd_retry));
daniele	29:1a47b7151851	1517	t->snd_retry = SEQN(nxt);
daniele	29:1a47b7151851	1518	}
daniele	29:1a47b7151851	1519	}
daniele	29:1a47b7151851	1520
daniele	29:1a47b7151851	1521	if (++t->cwnd_counter > 1) {
daniele	29:1a47b7151851	1522	t->cwnd--;
daniele	29:1a47b7151851	1523	if (t->cwnd < 2)
daniele	29:1a47b7151851	1524	t->cwnd = 2;
daniele	29:1a47b7151851	1525	t->cwnd_counter = 0;
daniele	29:1a47b7151851	1526	}
daniele	29:1a47b7151851	1527	} else {
daniele	29:1a47b7151851	1528	tcp_dbg("DUPACK in mode %d \n", t->x_mode);
daniele	29:1a47b7151851	1529
daniele	29:1a47b7151851	1530	}
daniele	29:1a47b7151851	1531	} /* End case duplicate ack detection */
daniele	29:1a47b7151851	1532
daniele	29:1a47b7151851	1533	/* Do congestion control */
daniele	29:1a47b7151851	1534	tcp_congestion_control(t);
daniele	29:1a47b7151851	1535	if ((acked > 0) && t->sock.wakeup) {
daniele	29:1a47b7151851	1536	if (t->tcpq_out.size < t->tcpq_out.max_size)
daniele	29:1a47b7151851	1537	t->sock.wakeup(PICO_SOCK_EV_WR, &(t->sock));
daniele	29:1a47b7151851	1538	//t->sock.ev_pending \|= PICO_SOCK_EV_WR;
daniele	29:1a47b7151851	1539	}
daniele	29:1a47b7151851	1540
daniele	29:1a47b7151851	1541	/* if Nagle enabled, check if no unack'ed data and fill out queue (till window) */
daniele	29:1a47b7151851	1542	if (IS_NAGLE_ENABLED((&(t->sock)))) {
daniele	29:1a47b7151851	1543	while (!IS_TCP_HOLDQ_EMPTY(t) && ((t->tcpq_out.max_size - t->tcpq_out.size) >= PICO_TCP_DEFAULT_MSS)) {
tass	45:ca30069e49bb	1544	tcp_dbg_nagle("TCP_ACK - NAGLE add new segment\n");
daniele	29:1a47b7151851	1545	f_new = pico_hold_segment_make(t);
daniele	29:1a47b7151851	1546	if (f_new == NULL)
daniele	29:1a47b7151851	1547	break; /* XXX corrupt !!! (or no memory) */
daniele	29:1a47b7151851	1548	if (pico_enqueue_segment(&t->tcpq_out,f_new) <= 0)
daniele	29:1a47b7151851	1549	// handle error
tass	45:ca30069e49bb	1550	tcp_dbg_nagle("TCP_ACK - NAGLE FAILED to enqueue in out\n");
daniele	29:1a47b7151851	1551	}
daniele	29:1a47b7151851	1552	}
daniele	29:1a47b7151851	1553
daniele	29:1a47b7151851	1554	/* If some space was created, put a few segments out. */
daniele	29:1a47b7151851	1555	tcp_dbg("TCP_CWND, %lu, %u, %u, %u\n", pico_tick, t->cwnd, t->ssthresh, t->in_flight);
daniele	29:1a47b7151851	1556	if (t->x_mode == PICO_TCP_LOOKAHEAD) {
daniele	29:1a47b7151851	1557	if ((t->cwnd >= t->in_flight) && (t->snd_nxt > t->snd_last_out)) {
tass	70:cd218dd180e5	1558	pico_tcp_output(&t->sock, (int)t->cwnd - (int)t->in_flight);
daniele	29:1a47b7151851	1559	}
daniele	29:1a47b7151851	1560	}
daniele	29:1a47b7151851	1561
daniele	29:1a47b7151851	1562	t->snd_old_ack = ACKN(f);
daniele	29:1a47b7151851	1563	return 0;
daniele	29:1a47b7151851	1564	}
daniele	29:1a47b7151851	1565
daniele	29:1a47b7151851	1566	static int tcp_finwaitack(struct pico_socket s, struct pico_frame f)
daniele	29:1a47b7151851	1567	{
tass	68:0847e35d08a6	1568	tcp_dbg("RECEIVED ACK IN FIN_WAIT1\n");
tass	68:0847e35d08a6	1569	tcp_dbg("TCP> IN STATE FIN_WAIT2\n");
daniele	29:1a47b7151851	1570
daniele	29:1a47b7151851	1571	/* acking part */
daniele	29:1a47b7151851	1572	tcp_ack(s,f);
daniele	29:1a47b7151851	1573	/* update TCP state */
daniele	29:1a47b7151851	1574	s->state &= 0x00FFU;
daniele	29:1a47b7151851	1575	s->state \|= PICO_SOCKET_STATE_TCP_FIN_WAIT2;
daniele	29:1a47b7151851	1576
daniele	29:1a47b7151851	1577	return 0;
daniele	29:1a47b7151851	1578	}
daniele	29:1a47b7151851	1579
tass	73:dfb737147f6e	1580	static void tcp_deltcb(uint32_t when, void *arg)
daniele	29:1a47b7151851	1581	{
daniele	29:1a47b7151851	1582	struct pico_socket_tcp t = (struct pico_socket_tcp )arg;
tass	68:0847e35d08a6	1583	IGNORE_PARAMETER(when);
tass	68:0847e35d08a6	1584
daniele	29:1a47b7151851	1585	if (TCPSTATE(&t->sock) == PICO_SOCKET_STATE_TCP_TIME_WAIT) {
daniele	29:1a47b7151851	1586	tcp_dbg("TCP> state: time_wait, final timer expired, going to closed state\n");
daniele	29:1a47b7151851	1587	/* update state */
daniele	29:1a47b7151851	1588	(t->sock).state &= 0x00FFU;
daniele	29:1a47b7151851	1589	(t->sock).state \|= PICO_SOCKET_STATE_TCP_CLOSED;
daniele	29:1a47b7151851	1590	(t->sock).state &= 0xFF00U;
daniele	29:1a47b7151851	1591	(t->sock).state \|= PICO_SOCKET_STATE_CLOSED;
daniele	29:1a47b7151851	1592	/* call EV_FIN wakeup before deleting */
daniele	29:1a47b7151851	1593	if (t->sock.wakeup) {
daniele	29:1a47b7151851	1594	(t->sock).wakeup(PICO_SOCK_EV_FIN, &(t->sock));
daniele	29:1a47b7151851	1595	}
daniele	29:1a47b7151851	1596	/* delete socket */
daniele	29:1a47b7151851	1597	pico_socket_del(&t->sock);
daniele	29:1a47b7151851	1598	} else {
daniele	29:1a47b7151851	1599	tcp_dbg("TCP> trying to go to closed, wrong state\n");
daniele	29:1a47b7151851	1600	}
daniele	29:1a47b7151851	1601	}
daniele	29:1a47b7151851	1602
daniele	29:1a47b7151851	1603	static int tcp_finwaitfin(struct pico_socket s, struct pico_frame f)
daniele	29:1a47b7151851	1604	{
daniele	29:1a47b7151851	1605	struct pico_socket_tcp t = (struct pico_socket_tcp )s;
daniele	29:1a47b7151851	1606	struct pico_tcp_hdr hdr = (struct pico_tcp_hdr ) (f->transport_hdr);
daniele	29:1a47b7151851	1607	tcp_dbg("TCP> received fin in FIN_WAIT2\n");
daniele	29:1a47b7151851	1608	/* received FIN, increase ACK nr */
daniele	29:1a47b7151851	1609	t->rcv_nxt = long_be(hdr->seq) + 1;
daniele	29:1a47b7151851	1610	s->state &= 0x00FFU;
daniele	29:1a47b7151851	1611	s->state \|= PICO_SOCKET_STATE_TCP_TIME_WAIT;
daniele	29:1a47b7151851	1612	/* set SHUT_REMOTE */
daniele	29:1a47b7151851	1613	s->state \|= PICO_SOCKET_STATE_SHUT_REMOTE;
daniele	29:1a47b7151851	1614	if (s->wakeup)
daniele	29:1a47b7151851	1615	s->wakeup(PICO_SOCK_EV_CLOSE, s);
daniele	29:1a47b7151851	1616	if (f->payload_len > 0) /* needed?? */
daniele	29:1a47b7151851	1617	tcp_data_in(s,f);
daniele	29:1a47b7151851	1618	/* send ACK */
daniele	29:1a47b7151851	1619	tcp_send_ack(t);
daniele	29:1a47b7151851	1620	/* set timer */
daniele	29:1a47b7151851	1621	pico_timer_add(200, tcp_deltcb, t);
daniele	29:1a47b7151851	1622	return 0;
daniele	29:1a47b7151851	1623	}
daniele	29:1a47b7151851	1624
daniele	29:1a47b7151851	1625	static int tcp_closewaitack(struct pico_socket s, struct pico_frame f)
daniele	29:1a47b7151851	1626	{
daniele	29:1a47b7151851	1627	struct pico_socket_tcp t = (struct pico_socket_tcp )s;
daniele	29:1a47b7151851	1628	tcp_dbg("TCP> received ack in CLOSING\n");
daniele	29:1a47b7151851	1629	/* acking part */
daniele	29:1a47b7151851	1630	tcp_ack(s,f);
daniele	29:1a47b7151851	1631	/* update TCP state */
daniele	29:1a47b7151851	1632	s->state &= 0x00FFU;
daniele	29:1a47b7151851	1633	s->state \|= PICO_SOCKET_STATE_TCP_TIME_WAIT;
daniele	29:1a47b7151851	1634	/* set timer */
daniele	29:1a47b7151851	1635	pico_timer_add(200, tcp_deltcb, t);
daniele	29:1a47b7151851	1636	return 0;
daniele	29:1a47b7151851	1637	}
daniele	29:1a47b7151851	1638
daniele	29:1a47b7151851	1639	static int tcp_lastackwait(struct pico_socket s, struct pico_frame f)
daniele	29:1a47b7151851	1640	{
tass	68:0847e35d08a6	1641	IGNORE_PARAMETER(f);
daniele	29:1a47b7151851	1642	tcp_dbg("TCP> state: last_ack, received ack, to closed\n");
daniele	29:1a47b7151851	1643	s->state &= 0x00FFU;
daniele	29:1a47b7151851	1644	s->state \|= PICO_SOCKET_STATE_TCP_CLOSED;
daniele	29:1a47b7151851	1645	s->state &= 0xFF00U;
daniele	29:1a47b7151851	1646	s->state \|= PICO_SOCKET_STATE_CLOSED;
daniele	29:1a47b7151851	1647	/* call socket wakeup with EV_FIN */
daniele	29:1a47b7151851	1648	if (s->wakeup)
daniele	29:1a47b7151851	1649	s->wakeup(PICO_SOCK_EV_FIN, s);
daniele	29:1a47b7151851	1650	/* delete socket */
daniele	29:1a47b7151851	1651	pico_socket_del(s);
daniele	29:1a47b7151851	1652	return 0;
daniele	29:1a47b7151851	1653	}
daniele	29:1a47b7151851	1654
daniele	29:1a47b7151851	1655	static int tcp_syn(struct pico_socket s, struct pico_frame f)
daniele	29:1a47b7151851	1656	{
daniele	29:1a47b7151851	1657	/* TODO: Check against backlog length */
daniele	29:1a47b7151851	1658	struct pico_socket_tcp new = (struct pico_socket_tcp )pico_socket_clone(s);
daniele	29:1a47b7151851	1659	struct pico_tcp_hdr hdr = (struct pico_tcp_hdr )f->transport_hdr;
daniele	29:1a47b7151851	1660	if (!new)
daniele	29:1a47b7151851	1661	return -1;
daniele	29:1a47b7151851	1662
daniele	29:1a47b7151851	1663	#ifdef PICO_TCP_SUPPORT_SOCKET_STATS
daniele	29:1a47b7151851	1664	pico_timer_add(2000, sock_stats, s);
daniele	29:1a47b7151851	1665	#endif
daniele	29:1a47b7151851	1666
daniele	29:1a47b7151851	1667	new->sock.remote_port = ((struct pico_trans *)f->transport_hdr)->sport;
daniele	29:1a47b7151851	1668	#ifdef PICO_SUPPORT_IPV4
daniele	29:1a47b7151851	1669	if (IS_IPV4(f)) {
daniele	29:1a47b7151851	1670	new->sock.remote_addr.ip4.addr = ((struct pico_ipv4_hdr *)(f->net_hdr))->src.addr;
daniele	29:1a47b7151851	1671	new->sock.local_addr.ip4.addr = ((struct pico_ipv4_hdr *)(f->net_hdr))->dst.addr;
daniele	29:1a47b7151851	1672	}
daniele	29:1a47b7151851	1673	#endif
daniele	29:1a47b7151851	1674	#ifdef PICO_SUPPORT_IPV6
daniele	29:1a47b7151851	1675	if (IS_IPV6(f)) {
tass	45:ca30069e49bb	1676	new->sock.remote_addr.ip6 = ((struct pico_ipv6_hdr *)(f->net_hdr))->src;
tass	45:ca30069e49bb	1677	new->sock.local_addr.ip6 = ((struct pico_ipv6_hdr *)(f->net_hdr))->dst;
daniele	29:1a47b7151851	1678	}
daniele	29:1a47b7151851	1679	#endif
daniele	29:1a47b7151851	1680
daniele	29:1a47b7151851	1681	/* Set socket limits */
daniele	29:1a47b7151851	1682	new->tcpq_in.max_size = PICO_DEFAULT_SOCKETQ;
daniele	29:1a47b7151851	1683	new->tcpq_out.max_size = PICO_DEFAULT_SOCKETQ;
daniele	29:1a47b7151851	1684	new->tcpq_hold.max_size = 2*PICO_TCP_DEFAULT_MSS;
daniele	29:1a47b7151851	1685
daniele	29:1a47b7151851	1686	f->sock = &new->sock;
daniele	29:1a47b7151851	1687	tcp_parse_options(f);
daniele	29:1a47b7151851	1688	new->mss = PICO_TCP_DEFAULT_MSS;
daniele	29:1a47b7151851	1689	new->rcv_nxt = long_be(hdr->seq) + 1;
daniele	29:1a47b7151851	1690	new->snd_nxt = long_be(pico_paws());
daniele	29:1a47b7151851	1691	new->snd_last = new->snd_nxt;
daniele	29:1a47b7151851	1692	new->cwnd = PICO_TCP_IW;
daniele	29:1a47b7151851	1693	new->ssthresh = 40;
daniele	29:1a47b7151851	1694	new->recv_wnd = short_be(hdr->rwnd);
daniele	29:1a47b7151851	1695	new->jumbo = hdr->len & 0x07;
daniele	29:1a47b7151851	1696	new->sock.parent = s;
daniele	29:1a47b7151851	1697	new->sock.wakeup = s->wakeup;
daniele	29:1a47b7151851	1698	/* Initialize timestamp values */
daniele	29:1a47b7151851	1699	new->sock.state = PICO_SOCKET_STATE_BOUND \| PICO_SOCKET_STATE_CONNECTED \| PICO_SOCKET_STATE_TCP_SYN_RECV;
daniele	29:1a47b7151851	1700	pico_socket_add(&new->sock);
daniele	29:1a47b7151851	1701	tcp_send_synack(&new->sock);
daniele	29:1a47b7151851	1702	tcp_dbg("SYNACK sent, socket added. snd_nxt is %08x\n", new->snd_nxt);
daniele	29:1a47b7151851	1703	return 0;
daniele	29:1a47b7151851	1704	}
daniele	29:1a47b7151851	1705
daniele	29:1a47b7151851	1706	static void tcp_set_init_point(struct pico_socket *s)
daniele	29:1a47b7151851	1707	{
daniele	29:1a47b7151851	1708	struct pico_socket_tcp t = (struct pico_socket_tcp )s;
daniele	29:1a47b7151851	1709	t->rcv_processed = t->rcv_nxt;
daniele	29:1a47b7151851	1710	}
daniele	29:1a47b7151851	1711
daniele	29:1a47b7151851	1712	static int tcp_synack(struct pico_socket s, struct pico_frame f)
daniele	29:1a47b7151851	1713	{
daniele	29:1a47b7151851	1714	struct pico_socket_tcp t = (struct pico_socket_tcp ) s;
daniele	29:1a47b7151851	1715	struct pico_tcp_hdr hdr = (struct pico_tcp_hdr )f->transport_hdr;
daniele	29:1a47b7151851	1716
daniele	29:1a47b7151851	1717	if (ACKN(f) == (1 + t->snd_nxt)) {
daniele	29:1a47b7151851	1718	t->rcv_nxt = long_be(hdr->seq);
daniele	29:1a47b7151851	1719	t->rcv_processed = t->rcv_nxt + 1;
daniele	29:1a47b7151851	1720	tcp_ack(s, f);
daniele	29:1a47b7151851	1721
daniele	29:1a47b7151851	1722	s->state &= 0x00FFU;
daniele	29:1a47b7151851	1723	s->state \|= PICO_SOCKET_STATE_TCP_ESTABLISHED;
daniele	29:1a47b7151851	1724	tcp_dbg("TCP> Established. State: %x\n", s->state);
daniele	29:1a47b7151851	1725
daniele	29:1a47b7151851	1726	if (s->wakeup)
daniele	29:1a47b7151851	1727	s->wakeup(PICO_SOCK_EV_CONN, s);
daniele	29:1a47b7151851	1728	s->ev_pending \|= PICO_SOCK_EV_WR;
daniele	29:1a47b7151851	1729
daniele	29:1a47b7151851	1730	t->rcv_nxt++;
daniele	29:1a47b7151851	1731	t->snd_nxt++;
daniele	29:1a47b7151851	1732	tcp_send_ack(t); /* return ACK */
daniele	29:1a47b7151851	1733
daniele	29:1a47b7151851	1734	return 0;
daniele	29:1a47b7151851	1735
daniele	29:1a47b7151851	1736	} else {
daniele	29:1a47b7151851	1737	tcp_dbg("TCP> Not established, RST sent.\n");
daniele	29:1a47b7151851	1738	tcp_nosync_rst(s,f);
daniele	29:1a47b7151851	1739	return 0;
daniele	29:1a47b7151851	1740	}
daniele	29:1a47b7151851	1741	}
daniele	29:1a47b7151851	1742
daniele	29:1a47b7151851	1743	static int tcp_first_ack(struct pico_socket s, struct pico_frame f)
daniele	29:1a47b7151851	1744	{
daniele	29:1a47b7151851	1745	struct pico_socket_tcp t = (struct pico_socket_tcp )s;
daniele	29:1a47b7151851	1746	tcp_dbg("ACK in SYN_RECV: expecting %08x got %08x\n", t->snd_nxt, ACKN(f));
daniele	29:1a47b7151851	1747	if (t->snd_nxt == ACKN(f)) {
daniele	29:1a47b7151851	1748	tcp_set_init_point(s);
daniele	29:1a47b7151851	1749	tcp_ack(s, f);
daniele	29:1a47b7151851	1750	s->state &= 0x00FFU;
daniele	29:1a47b7151851	1751	s->state \|= PICO_SOCKET_STATE_TCP_ESTABLISHED;
daniele	29:1a47b7151851	1752	tcp_dbg("TCP: Established. State now: %04x\n", s->state);
daniele	29:1a47b7151851	1753	if( !s->parent && s->wakeup) { /* If the socket has no parent, -> sending socket that has a sim_open */
daniele	29:1a47b7151851	1754	tcp_dbg("FIRST ACK - No parent found -> sending socket\n");
daniele	29:1a47b7151851	1755	s->wakeup(PICO_SOCK_EV_CONN, s);
daniele	29:1a47b7151851	1756	}
daniele	29:1a47b7151851	1757	if (s->parent && s->parent->wakeup) {
daniele	29:1a47b7151851	1758	tcp_dbg("FIRST ACK - Parent found -> listening socket\n");
daniele	29:1a47b7151851	1759	s->wakeup = s->parent->wakeup;
daniele	29:1a47b7151851	1760	s->parent->wakeup(PICO_SOCK_EV_CONN, s->parent);
daniele	29:1a47b7151851	1761	}
daniele	29:1a47b7151851	1762	s->ev_pending \|= PICO_SOCK_EV_WR;
daniele	29:1a47b7151851	1763	tcp_dbg("%s: snd_nxt is now %08x\n", __FUNCTION__, t->snd_nxt);
daniele	29:1a47b7151851	1764	return 0;
daniele	29:1a47b7151851	1765	} else {
daniele	29:1a47b7151851	1766	tcp_nosync_rst(s,f);
daniele	29:1a47b7151851	1767	return 0;
daniele	29:1a47b7151851	1768	}
daniele	29:1a47b7151851	1769	}
daniele	29:1a47b7151851	1770
daniele	29:1a47b7151851	1771	static int tcp_closewait(struct pico_socket s, struct pico_frame f)
daniele	29:1a47b7151851	1772	{
daniele	29:1a47b7151851	1773
daniele	29:1a47b7151851	1774	struct pico_socket_tcp t = (struct pico_socket_tcp )s;
daniele	29:1a47b7151851	1775	struct pico_tcp_hdr hdr = (struct pico_tcp_hdr ) (f->transport_hdr);
daniele	29:1a47b7151851	1776
tass	32:865c101e0874	1777
daniele	29:1a47b7151851	1778	if (f->payload_len > 0)
daniele	29:1a47b7151851	1779	tcp_data_in(s,f);
daniele	29:1a47b7151851	1780	if (f->flags & PICO_TCP_ACK)
daniele	29:1a47b7151851	1781	tcp_ack(s,f);
daniele	29:1a47b7151851	1782	if (seq_compare(SEQN(f), t->rcv_nxt) == 0) {
daniele	29:1a47b7151851	1783	/* received FIN, increase ACK nr */
tass	68:0847e35d08a6	1784	t->rcv_nxt = long_be(hdr->seq) + 1;
tass	68:0847e35d08a6	1785	s->state &= 0x00FFU;
tass	68:0847e35d08a6	1786	s->state \|= PICO_SOCKET_STATE_TCP_CLOSE_WAIT;
tass	68:0847e35d08a6	1787	/* set SHUT_REMOTE */
daniele	29:1a47b7151851	1788	s->state \|= PICO_SOCKET_STATE_SHUT_REMOTE;
daniele	29:1a47b7151851	1789	tcp_dbg("TCP> Close-wait\n");
tass	32:865c101e0874	1790
daniele	29:1a47b7151851	1791	if (s->wakeup){
daniele	29:1a47b7151851	1792	if(f->payload_len>0){
tass	68:0847e35d08a6	1793	struct pico_socket_tcp _t = (struct pico_socket_tcp )s;
tass	68:0847e35d08a6	1794	_t->sock.ev_pending \|=PICO_SOCK_EV_CLOSE;
daniele	29:1a47b7151851	1795	}else
daniele	29:1a47b7151851	1796	s->wakeup(PICO_SOCK_EV_CLOSE, s);
daniele	29:1a47b7151851	1797	}
daniele	29:1a47b7151851	1798	} else {
daniele	29:1a47b7151851	1799	tcp_send_ack(t); /* return ACK */
daniele	29:1a47b7151851	1800	}
daniele	29:1a47b7151851	1801	return 0;
daniele	29:1a47b7151851	1802	}
daniele	29:1a47b7151851	1803
daniele	29:1a47b7151851	1804	/static int tcp_fin(struct pico_socket s, struct pico_frame *f)
daniele	29:1a47b7151851	1805	{
daniele	29:1a47b7151851	1806	return 0;
daniele	29:1a47b7151851	1807	}*/
daniele	29:1a47b7151851	1808
daniele	29:1a47b7151851	1809	static int tcp_rcvfin(struct pico_socket s, struct pico_frame f)
daniele	29:1a47b7151851	1810	{
daniele	29:1a47b7151851	1811	struct pico_socket_tcp t = (struct pico_socket_tcp )s;
tass	68:0847e35d08a6	1812	IGNORE_PARAMETER(f);
daniele	29:1a47b7151851	1813	tcp_dbg("TCP> Received FIN in FIN_WAIT1\n");
daniele	29:1a47b7151851	1814	s->state &= 0x00FFU;
daniele	29:1a47b7151851	1815	s->state \|= PICO_SOCKET_STATE_TCP_CLOSING;
daniele	29:1a47b7151851	1816	t->rcv_processed = t->rcv_nxt + 1;
daniele	29:1a47b7151851	1817	t->rcv_nxt++;
daniele	29:1a47b7151851	1818	/* send ACK */
daniele	29:1a47b7151851	1819	tcp_send_ack(t);
daniele	29:1a47b7151851	1820	return 0;
daniele	29:1a47b7151851	1821	}
daniele	29:1a47b7151851	1822
daniele	29:1a47b7151851	1823	static int tcp_finack(struct pico_socket s, struct pico_frame f)
daniele	29:1a47b7151851	1824	{
daniele	29:1a47b7151851	1825	struct pico_socket_tcp t = (struct pico_socket_tcp )s;
tass	68:0847e35d08a6	1826	IGNORE_PARAMETER(f);
tass	68:0847e35d08a6	1827
daniele	29:1a47b7151851	1828	tcp_dbg("TCP> ENTERED finack\n");
daniele	29:1a47b7151851	1829	t->rcv_nxt++;
daniele	29:1a47b7151851	1830	/* send ACK */
daniele	29:1a47b7151851	1831	tcp_send_ack(t);
daniele	29:1a47b7151851	1832
daniele	29:1a47b7151851	1833	/* call socket wakeup with EV_FIN */
daniele	29:1a47b7151851	1834	if (s->wakeup)
daniele	29:1a47b7151851	1835	s->wakeup(PICO_SOCK_EV_FIN, s);
daniele	29:1a47b7151851	1836	s->state &= 0x00FFU;
daniele	29:1a47b7151851	1837	s->state \|= PICO_SOCKET_STATE_TCP_TIME_WAIT;
daniele	29:1a47b7151851	1838	/* set SHUT_REMOTE */
daniele	29:1a47b7151851	1839	s->state \|= PICO_SOCKET_STATE_SHUT_REMOTE;
daniele	29:1a47b7151851	1840	pico_timer_add(2000, tcp_deltcb, t);
daniele	29:1a47b7151851	1841
daniele	29:1a47b7151851	1842	return 0;
daniele	29:1a47b7151851	1843	}
daniele	29:1a47b7151851	1844
tass	70:cd218dd180e5	1845	static void tcp_force_closed(struct pico_socket *s)
tass	70:cd218dd180e5	1846	{
tass	70:cd218dd180e5	1847	struct pico_socket_tcp t = (struct pico_socket_tcp ) s;
tass	70:cd218dd180e5	1848	/* update state */
tass	70:cd218dd180e5	1849	(t->sock).state &= 0x00FFU;
tass	70:cd218dd180e5	1850	(t->sock).state \|= PICO_SOCKET_STATE_TCP_CLOSED;
tass	70:cd218dd180e5	1851	(t->sock).state &= 0xFF00U;
tass	70:cd218dd180e5	1852	(t->sock).state \|= PICO_SOCKET_STATE_CLOSED;
tass	70:cd218dd180e5	1853	}
tass	70:cd218dd180e5	1854
tass	70:cd218dd180e5	1855	static void tcp_wakeup_pending(struct pico_socket *s, uint16_t ev)
tass	70:cd218dd180e5	1856	{
tass	70:cd218dd180e5	1857	struct pico_socket_tcp t = (struct pico_socket_tcp ) s;
tass	70:cd218dd180e5	1858	if ((t->sock).wakeup)
tass	70:cd218dd180e5	1859	(t->sock).wakeup(ev, &(t->sock));
tass	70:cd218dd180e5	1860	}
tass	70:cd218dd180e5	1861
daniele	29:1a47b7151851	1862	static int tcp_rst(struct pico_socket s, struct pico_frame f)
daniele	29:1a47b7151851	1863	{
daniele	29:1a47b7151851	1864	struct pico_socket_tcp t = (struct pico_socket_tcp ) s;
daniele	29:1a47b7151851	1865	struct pico_tcp_hdr hdr = (struct pico_tcp_hdr ) (f->transport_hdr);
daniele	29:1a47b7151851	1866
daniele	29:1a47b7151851	1867	tcp_dbg("TCP >>>>>>>>>>>>>> received RST <<<<<<<<<<<<<<<<<<<<\n");
daniele	29:1a47b7151851	1868	if ((s->state & PICO_SOCKET_STATE_TCP) == PICO_SOCKET_STATE_TCP_SYN_SENT) {
daniele	29:1a47b7151851	1869	/* the RST is acceptable if the ACK field acknowledges the SYN */
daniele	29:1a47b7151851	1870	if ((t->snd_nxt + 1) == ACKN(f)) { /* valid, got to closed state */
tass	70:cd218dd180e5	1871	tcp_force_closed(s);
tass	70:cd218dd180e5	1872	tcp_wakeup_pending(s, PICO_SOCK_EV_FIN);
daniele	29:1a47b7151851	1873	pico_err = PICO_ERR_ECONNRESET;
tass	70:cd218dd180e5	1874	tcp_wakeup_pending(s, PICO_SOCK_EV_ERR);
tass	70:cd218dd180e5	1875	pico_socket_del(&t->sock); /* delete socket */
daniele	29:1a47b7151851	1876	} else { /* not valid, ignore */
daniele	29:1a47b7151851	1877	tcp_dbg("TCP RST> IGNORE\n");
daniele	29:1a47b7151851	1878	return 0;
daniele	29:1a47b7151851	1879	}
daniele	29:1a47b7151851	1880	} else { /* all other states */
daniele	29:1a47b7151851	1881	/* all reset (RST) segments are validated by checking their SEQ-fields,
daniele	29:1a47b7151851	1882	a reset is valid if its sequence number is in the window */
tass	70:cd218dd180e5	1883	if ((long_be(hdr->seq) >= t->rcv_ackd) && (long_be(hdr->seq) <= ((uint32_t)(short_be(hdr->rwnd)<<(t->wnd_scale)) + t->rcv_ackd))) {
daniele	29:1a47b7151851	1884	if ((s->state & PICO_SOCKET_STATE_TCP) == PICO_SOCKET_STATE_TCP_SYN_RECV) {
tass	70:cd218dd180e5	1885	tcp_force_closed(s);
daniele	29:1a47b7151851	1886	pico_err = PICO_ERR_ECONNRESET;
tass	70:cd218dd180e5	1887	tcp_wakeup_pending(s, PICO_SOCK_EV_ERR);
tass	70:cd218dd180e5	1888	pico_socket_del(&t->sock); /* delete socket */
daniele	29:1a47b7151851	1889	tcp_dbg("TCP RST> SOCKET BACK TO LISTEN\n");
daniele	29:1a47b7151851	1890	pico_socket_del(s);
daniele	29:1a47b7151851	1891	} else {
tass	70:cd218dd180e5	1892	tcp_force_closed(s);
tass	70:cd218dd180e5	1893	tcp_wakeup_pending(s, PICO_SOCK_EV_FIN);
daniele	29:1a47b7151851	1894	pico_err = PICO_ERR_ECONNRESET;
tass	70:cd218dd180e5	1895	tcp_wakeup_pending(s, PICO_SOCK_EV_ERR);
tass	70:cd218dd180e5	1896	pico_socket_del(&t->sock); /* delete socket */
daniele	29:1a47b7151851	1897	}
daniele	29:1a47b7151851	1898	} else { /* not valid, ignore */
daniele	29:1a47b7151851	1899	tcp_dbg("TCP RST> IGNORE\n");
daniele	29:1a47b7151851	1900	return 0;
daniele	29:1a47b7151851	1901	}
daniele	29:1a47b7151851	1902	}
daniele	29:1a47b7151851	1903
daniele	29:1a47b7151851	1904	return 0;
daniele	29:1a47b7151851	1905	}
tass	79:6d1cb906043d	1906	static int tcp_halfopencon(struct pico_socket s, struct pico_frame fr)
tass	79:6d1cb906043d	1907	{
tass	79:6d1cb906043d	1908	struct pico_socket_tcp t = (struct pico_socket_tcp ) s;
tass	79:6d1cb906043d	1909	IGNORE_PARAMETER(fr);
tass	79:6d1cb906043d	1910	tcp_send_ack(t);
tass	79:6d1cb906043d	1911	return 0;
tass	79:6d1cb906043d	1912	}
daniele	29:1a47b7151851	1913	struct tcp_action_entry {
daniele	29:1a47b7151851	1914	uint16_t tcpstate;
daniele	29:1a47b7151851	1915	int (syn)(struct pico_socket s, struct pico_frame *f);
daniele	29:1a47b7151851	1916	int (synack)(struct pico_socket s, struct pico_frame *f);
daniele	29:1a47b7151851	1917	int (ack)(struct pico_socket s, struct pico_frame *f);
daniele	29:1a47b7151851	1918	int (data)(struct pico_socket s, struct pico_frame *f);
daniele	29:1a47b7151851	1919	int (fin)(struct pico_socket s, struct pico_frame *f);
daniele	29:1a47b7151851	1920	int (finack)(struct pico_socket s, struct pico_frame *f);
daniele	29:1a47b7151851	1921	int (rst)(struct pico_socket s, struct pico_frame *f);
daniele	29:1a47b7151851	1922	};
daniele	29:1a47b7151851	1923
daniele	29:1a47b7151851	1924	static struct tcp_action_entry tcp_fsm[] = {
daniele	29:1a47b7151851	1925	/* State syn synack ack data fin finack rst*/
daniele	29:1a47b7151851	1926	{ PICO_SOCKET_STATE_TCP_UNDEF, NULL, NULL, NULL, NULL, NULL, NULL, NULL },
daniele	29:1a47b7151851	1927	{ PICO_SOCKET_STATE_TCP_CLOSED, NULL, NULL, NULL, NULL, NULL, NULL, NULL },
daniele	29:1a47b7151851	1928	{ PICO_SOCKET_STATE_TCP_LISTEN, &tcp_syn, &tcp_nosync_rst, &tcp_nosync_rst, &tcp_nosync_rst, &tcp_nosync_rst, &tcp_nosync_rst, NULL },
tass	40:c8ab0d2bba0b	1929	{ PICO_SOCKET_STATE_TCP_SYN_SENT, &tcp_nosync_rst, &tcp_synack, &tcp_nosync_rst, &tcp_nosync_rst, &tcp_nosync_rst, &tcp_nosync_rst, &tcp_rst },
tass	91:8da42f2c4820	1930	{ PICO_SOCKET_STATE_TCP_SYN_RECV, NULL, &tcp_nosync_rst, &tcp_first_ack, &tcp_data_in, &tcp_nosync_rst, &tcp_nosync_rst, &tcp_rst },
tass	79:6d1cb906043d	1931	{ PICO_SOCKET_STATE_TCP_ESTABLISHED, &tcp_halfopencon,&tcp_ack, &tcp_ack, &tcp_data_in, &tcp_closewait, &tcp_closewait, &tcp_rst },
tass	45:ca30069e49bb	1932	{ PICO_SOCKET_STATE_TCP_CLOSE_WAIT, NULL, &tcp_ack, &tcp_ack, &tcp_send_rst, &tcp_closewait, &tcp_closewait, &tcp_rst },
tass	40:c8ab0d2bba0b	1933	{ PICO_SOCKET_STATE_TCP_LAST_ACK, NULL, &tcp_ack, &tcp_lastackwait, &tcp_send_rst, &tcp_send_rst, &tcp_send_rst, &tcp_rst },
tass	40:c8ab0d2bba0b	1934	{ PICO_SOCKET_STATE_TCP_FIN_WAIT1, NULL, &tcp_ack, &tcp_finwaitack, &tcp_data_in, &tcp_rcvfin, &tcp_finack, &tcp_rst },
tass	40:c8ab0d2bba0b	1935	{ PICO_SOCKET_STATE_TCP_FIN_WAIT2, NULL, &tcp_ack, &tcp_ack, &tcp_data_in, &tcp_finwaitfin, &tcp_finack, &tcp_rst },
tass	40:c8ab0d2bba0b	1936	{ PICO_SOCKET_STATE_TCP_CLOSING, NULL, &tcp_ack, &tcp_closewaitack, &tcp_send_rst, &tcp_send_rst, &tcp_send_rst, &tcp_rst },
tass	40:c8ab0d2bba0b	1937	{ PICO_SOCKET_STATE_TCP_TIME_WAIT, NULL, &tcp_ack, &tcp_send_rst, &tcp_send_rst, &tcp_send_rst, &tcp_send_rst, &tcp_rst }
daniele	29:1a47b7151851	1938	};
daniele	29:1a47b7151851	1939
daniele	29:1a47b7151851	1940	/*
daniele	29:1a47b7151851	1941	NOTE: in SYN-RECV receiving syn when cloned by default (see yellow pos-it), should send reset.
daniele	29:1a47b7151851	1942	*/
daniele	29:1a47b7151851	1943
daniele	29:1a47b7151851	1944	int pico_tcp_input(struct pico_socket s, struct pico_frame f)
daniele	29:1a47b7151851	1945	{
daniele	29:1a47b7151851	1946	struct pico_tcp_hdr hdr = (struct pico_tcp_hdr ) (f->transport_hdr);
daniele	29:1a47b7151851	1947	int ret = 0;
daniele	29:1a47b7151851	1948	uint8_t flags = hdr->flags;
daniele	29:1a47b7151851	1949	struct tcp_action_entry *action = &tcp_fsm[s->state >> 8];
daniele	29:1a47b7151851	1950
daniele	29:1a47b7151851	1951	f->payload = (f->transport_hdr + ((hdr->len & 0xf0) >> 2));
tass	70:cd218dd180e5	1952	f->payload_len = (uint16_t)(f->transport_len - ((hdr->len & 0xf0) >> 2));
daniele	29:1a47b7151851	1953
daniele	29:1a47b7151851	1954	tcp_dbg("[%lu] TCP> [tcp input] socket: %p state: %d <-- local port:%d remote port: %d seq: %08x ack: %08x flags: %02x = t_len: %d, hdr: %u payload: %d\n", pico_tick,
daniele	29:1a47b7151851	1955	s, s->state >> 8, short_be(hdr->trans.dport), short_be(hdr->trans.sport), SEQN(f), ACKN(f), hdr->flags, f->transport_len, (hdr->len & 0xf0) >> 2, f->payload_len );
daniele	29:1a47b7151851	1956
daniele	29:1a47b7151851	1957	/* This copy of the frame has the current socket as owner */
daniele	29:1a47b7151851	1958	f->sock = s;
tass	68:0847e35d08a6	1959	s->timestamp = PICO_TIME_MS();
daniele	29:1a47b7151851	1960	/* Those are not supported at this time. */
tass	73:dfb737147f6e	1961	flags &= (uint8_t)~(PICO_TCP_CWR \| PICO_TCP_URG \| PICO_TCP_ECN);
daniele	29:1a47b7151851	1962	if (flags == PICO_TCP_SYN) {
daniele	29:1a47b7151851	1963	if (action->syn)
daniele	29:1a47b7151851	1964	action->syn(s,f);
daniele	29:1a47b7151851	1965	} else if (flags == (PICO_TCP_SYN \| PICO_TCP_ACK)) {
daniele	29:1a47b7151851	1966	if (action->synack)
daniele	29:1a47b7151851	1967	action->synack(s,f);
daniele	29:1a47b7151851	1968	} else {
daniele	29:1a47b7151851	1969	if ((flags == PICO_TCP_ACK) \|\| (flags == (PICO_TCP_ACK \| PICO_TCP_PSH))) {
daniele	29:1a47b7151851	1970	if (action->ack) {
daniele	29:1a47b7151851	1971	action->ack(s,f);
daniele	29:1a47b7151851	1972	}
daniele	29:1a47b7151851	1973	}
tass	68:0847e35d08a6	1974	if (f->payload_len > 0 && !(s->state & PICO_SOCKET_STATE_CLOSED) && !TCP_IS_STATE(s,PICO_SOCKET_STATE_TCP_LISTEN))
tass	68:0847e35d08a6	1975	{
daniele	29:1a47b7151851	1976	ret = f->payload_len;
daniele	29:1a47b7151851	1977	if (action->data)
daniele	29:1a47b7151851	1978	action->data(s,f);
daniele	29:1a47b7151851	1979	}
daniele	29:1a47b7151851	1980	if (flags == PICO_TCP_FIN) {
daniele	29:1a47b7151851	1981	if (action->fin)
daniele	29:1a47b7151851	1982	action->fin(s,f);
daniele	29:1a47b7151851	1983	}
daniele	29:1a47b7151851	1984	if ((flags == (PICO_TCP_FIN \| PICO_TCP_ACK)) \|\| (flags == (PICO_TCP_FIN \| PICO_TCP_ACK \| PICO_TCP_PSH))) {
daniele	29:1a47b7151851	1985	if (action->finack)
daniele	29:1a47b7151851	1986	action->finack(s,f);
daniele	29:1a47b7151851	1987	}
daniele	29:1a47b7151851	1988	if (flags & PICO_TCP_RST) {
daniele	29:1a47b7151851	1989	if (action->rst)
daniele	29:1a47b7151851	1990	action->rst(s,f);
daniele	29:1a47b7151851	1991	}
daniele	29:1a47b7151851	1992	}
daniele	29:1a47b7151851	1993
daniele	29:1a47b7151851	1994	//discard:
daniele	29:1a47b7151851	1995	pico_frame_discard(f);
daniele	29:1a47b7151851	1996	return ret;
daniele	29:1a47b7151851	1997	}
daniele	29:1a47b7151851	1998
tass	73:dfb737147f6e	1999	static void tcp_send_keepalive(uint32_t when, void *_t)
daniele	29:1a47b7151851	2000	{
daniele	29:1a47b7151851	2001	struct pico_socket_tcp t = (struct pico_socket_tcp )_t;
tass	68:0847e35d08a6	2002	IGNORE_PARAMETER(when);
tass	82:93915e13262a	2003	tcp_dbg("Sending keepalive (%d), [State = %d]...\n", t->backoff,t->sock.state );
daniele	29:1a47b7151851	2004	if( t->sock.net && ((t->sock.state & 0xFF00) == PICO_SOCKET_STATE_TCP_ESTABLISHED) )
daniele	29:1a47b7151851	2005	{
tass	68:0847e35d08a6	2006	tcp_send_ack(t);
daniele	29:1a47b7151851	2007
tass	68:0847e35d08a6	2008	if (t->keepalive_timer_running > 0) {
tass	68:0847e35d08a6	2009	t->keepalive_timer_running--;
tass	68:0847e35d08a6	2010	}
daniele	29:1a47b7151851	2011
tass	68:0847e35d08a6	2012	if (t->keepalive_timer_running == 0) {
tass	68:0847e35d08a6	2013	t->keepalive_timer_running++;
tass	68:0847e35d08a6	2014	tcp_dbg("[Self] Adding timer(retransmit keepalive)\n");
tass	68:0847e35d08a6	2015	pico_timer_add(t->rto << (++t->backoff), tcp_send_keepalive, t);
tass	68:0847e35d08a6	2016	}
daniele	29:1a47b7151851	2017	}
daniele	29:1a47b7151851	2018	}
daniele	29:1a47b7151851	2019
tass	76:938a140caf12	2020	inline static int checkLocalClosing(struct pico_socket *s);
tass	76:938a140caf12	2021	inline static int checkRemoteClosing(struct pico_socket *s);
tass	76:938a140caf12	2022
tass	48:40fc4462265c	2023
daniele	29:1a47b7151851	2024	int pico_tcp_output(struct pico_socket *s, int loop_score)
daniele	29:1a47b7151851	2025	{
daniele	29:1a47b7151851	2026	struct pico_socket_tcp t = (struct pico_socket_tcp )s;
daniele	29:1a47b7151851	2027	struct pico_frame f, una;
daniele	29:1a47b7151851	2028	struct pico_tcp_hdr *hdr;
daniele	29:1a47b7151851	2029	int sent = 0;
daniele	29:1a47b7151851	2030
daniele	29:1a47b7151851	2031	una = first_segment(&t->tcpq_out);
daniele	29:1a47b7151851	2032
daniele	29:1a47b7151851	2033	f = peek_segment(&t->tcpq_out, t->snd_nxt);
daniele	29:1a47b7151851	2034	while((f) && (t->cwnd >= t->in_flight)) {
daniele	29:1a47b7151851	2035	hdr = (struct pico_tcp_hdr *)f->transport_hdr;
daniele	29:1a47b7151851	2036	f->timestamp = pico_tick;
daniele	29:1a47b7151851	2037	tcp_add_options(t, f, hdr->flags, tcp_options_size(t, hdr->flags));
tass	70:cd218dd180e5	2038	if (seq_compare((SEQN(f) + f->payload_len), (SEQN(una) + (uint32_t)(t->recv_wnd << t->recv_wnd_scale))) > 0) {
tass	70:cd218dd180e5	2039	t->cwnd = (uint16_t)t->in_flight;
daniele	29:1a47b7151851	2040	if (t->cwnd < 1)
daniele	29:1a47b7151851	2041	t->cwnd = 1;
daniele	29:1a47b7151851	2042	if (t->x_mode != PICO_TCP_WINDOW_FULL) {
daniele	29:1a47b7151851	2043	tcp_dbg("TCP> RIGHT SIZING (rwnd: %d, frame len: %d\n",t->recv_wnd << t->recv_wnd_scale, f->payload_len);
daniele	29:1a47b7151851	2044	tcp_dbg("In window full...\n");
tass	45:ca30069e49bb	2045	t->snd_nxt = SEQN(una); /* XXX prevent out-of-order-packets ! / /DLA re-enabled.*/
tass	40:c8ab0d2bba0b	2046	t->snd_retry = SEQN(una); /* XXX replace by retry pointer? */
daniele	29:1a47b7151851	2047
daniele	29:1a47b7151851	2048	/* Alternative to the line above: (better performance, but seems to lock anyway with larger buffers)
daniele	29:1a47b7151851	2049	if (seq_compare(t->snd_nxt, SEQN(una)) > 0)
daniele	29:1a47b7151851	2050	t->snd_nxt -= f->payload_len;
daniele	29:1a47b7151851	2051	*/
daniele	29:1a47b7151851	2052
daniele	29:1a47b7151851	2053	t->x_mode = PICO_TCP_WINDOW_FULL;
daniele	29:1a47b7151851	2054	if (t->keepalive_timer_running == 0) {
daniele	29:1a47b7151851	2055	tcp_dbg("[Window full] Adding timer(send keepalive)\n");
daniele	29:1a47b7151851	2056	tcp_send_keepalive(0, t);
daniele	29:1a47b7151851	2057	}
daniele	29:1a47b7151851	2058	}
daniele	29:1a47b7151851	2059	break;
daniele	29:1a47b7151851	2060	}
daniele	29:1a47b7151851	2061	tcp_dbg("TCP> DEQUEUED (for output) frame %08x, acks %08x len= %d, remaining frames %d\n", SEQN(f), ACKN(f), f->payload_len,t->tcpq_out.frames);
daniele	29:1a47b7151851	2062	tcp_send(t, f);
daniele	29:1a47b7151851	2063	sent++;
daniele	29:1a47b7151851	2064	loop_score--;
daniele	29:1a47b7151851	2065	t->snd_last_out = SEQN(f);
daniele	29:1a47b7151851	2066	if (loop_score < 1)
daniele	29:1a47b7151851	2067	break;
daniele	29:1a47b7151851	2068	if (f->payload_len > 0) {
daniele	29:1a47b7151851	2069	f = next_segment(&t->tcpq_out, f);
daniele	29:1a47b7151851	2070	} else {
daniele	29:1a47b7151851	2071	f = NULL;
daniele	29:1a47b7151851	2072	}
daniele	29:1a47b7151851	2073	}
daniele	29:1a47b7151851	2074	if (sent > 0) {
daniele	29:1a47b7151851	2075	if (t->rto < PICO_TCP_RTO_MIN)
daniele	29:1a47b7151851	2076	t->rto = PICO_TCP_RTO_MIN;
tass	45:ca30069e49bb	2077	//if (s->wakeup)
tass	45:ca30069e49bb	2078	// t->sock.wakeup(PICO_SOCK_EV_WR, &t->sock);
daniele	29:1a47b7151851	2079	add_retransmission_timer(t, pico_tick + t->rto);
daniele	29:1a47b7151851	2080	} else {
tass	45:ca30069e49bb	2081	/* Nothing to transmit. */
daniele	29:1a47b7151851	2082	}
daniele	29:1a47b7151851	2083
daniele	29:1a47b7151851	2084	if ((t->tcpq_out.frames == 0) && (s->state & PICO_SOCKET_STATE_SHUT_LOCAL)) { /* if no more packets in queue, XXX replacled !f by tcpq check */
tass	76:938a140caf12	2085	if(!checkLocalClosing(&t->sock)) // check if local closing started and send fin
tass	76:938a140caf12	2086	{
tass	76:938a140caf12	2087	checkRemoteClosing(&t->sock); // check if remote closing started and send fin
tass	76:938a140caf12	2088	}
daniele	29:1a47b7151851	2089	}
daniele	29:1a47b7151851	2090	return loop_score;
daniele	29:1a47b7151851	2091	}
daniele	29:1a47b7151851	2092
daniele	29:1a47b7151851	2093	/* function to make new segment from hold queue with specific size (mss) */
daniele	29:1a47b7151851	2094	static struct pico_frame * pico_hold_segment_make(struct pico_socket_tcp *t)
daniele	29:1a47b7151851	2095	{
daniele	29:1a47b7151851	2096	struct pico_frame f_temp,f_new;
daniele	29:1a47b7151851	2097	struct pico_socket s = (struct pico_socket ) &t->sock;
daniele	29:1a47b7151851	2098	struct pico_tcp_hdr *hdr;
tass	70:cd218dd180e5	2099	uint16_t total_len = 0, total_payload_len = 0;
tass	70:cd218dd180e5	2100	uint16_t off = 0, test = 0;
daniele	29:1a47b7151851	2101
daniele	29:1a47b7151851	2102	off = pico_tcp_overhead(s);
daniele	29:1a47b7151851	2103
daniele	29:1a47b7151851	2104	/* init with first frame in hold queue */
daniele	29:1a47b7151851	2105	f_temp = first_segment(&t->tcpq_hold);
daniele	29:1a47b7151851	2106	total_len = f_temp->payload_len;
daniele	29:1a47b7151851	2107	f_temp = next_segment(&t->tcpq_hold, f_temp);
daniele	29:1a47b7151851	2108
daniele	29:1a47b7151851	2109	/* check till total_len <= MSS */
daniele	29:1a47b7151851	2110	while ((f_temp != NULL) && ((total_len+f_temp->payload_len) <= PICO_TCP_DEFAULT_MSS)) {
tass	73:dfb737147f6e	2111	total_len = (uint16_t)(total_len + f_temp->payload_len);
daniele	29:1a47b7151851	2112	f_temp = next_segment(&t->tcpq_hold, f_temp);
daniele	29:1a47b7151851	2113	if (f_temp == NULL)
daniele	29:1a47b7151851	2114	break;
daniele	29:1a47b7151851	2115	}
daniele	29:1a47b7151851	2116	/* alloc new frame with payload size = off + total_len */
tass	70:cd218dd180e5	2117	f_new = pico_socket_frame_alloc(s, (uint16_t)(off + total_len));
daniele	29:1a47b7151851	2118	if (!f_new) {
daniele	29:1a47b7151851	2119	pico_err = PICO_ERR_ENOMEM;
daniele	29:1a47b7151851	2120	return f_new;
daniele	29:1a47b7151851	2121	}
daniele	29:1a47b7151851	2122
daniele	29:1a47b7151851	2123	hdr = (struct pico_tcp_hdr *) f_new->transport_hdr;
daniele	29:1a47b7151851	2124	/* init new frame */
daniele	29:1a47b7151851	2125	f_new->payload += off;
tass	73:dfb737147f6e	2126	f_new->payload_len = (uint16_t)(f_new->payload_len - off);
daniele	29:1a47b7151851	2127	f_new->sock = s;
daniele	29:1a47b7151851	2128
daniele	29:1a47b7151851	2129	f_temp = first_segment(&t->tcpq_hold);
daniele	29:1a47b7151851	2130	hdr->seq = ((struct pico_tcp_hdr )(f_temp->transport_hdr))->seq; / get sequence number of first frame */
daniele	29:1a47b7151851	2131	hdr->trans.sport = t->sock.local_port;
daniele	29:1a47b7151851	2132	hdr->trans.dport = t->sock.remote_port;
daniele	29:1a47b7151851	2133
daniele	29:1a47b7151851	2134	/* check till total_payload_len <= MSS */
daniele	29:1a47b7151851	2135	while ((f_temp != NULL) && ((total_payload_len + f_temp->payload_len) <= PICO_TCP_DEFAULT_MSS)) {
daniele	29:1a47b7151851	2136	/* cpy data and discard frame */
daniele	29:1a47b7151851	2137	test++;
daniele	29:1a47b7151851	2138	memcpy(f_new->payload + total_payload_len, f_temp->payload, f_temp->payload_len);
tass	73:dfb737147f6e	2139	total_payload_len = (uint16_t)(total_payload_len + f_temp->payload_len);
daniele	29:1a47b7151851	2140	pico_discard_segment(&t->tcpq_hold, f_temp);
daniele	29:1a47b7151851	2141	f_temp = first_segment(&t->tcpq_hold);
daniele	29:1a47b7151851	2142	}
daniele	29:1a47b7151851	2143
tass	70:cd218dd180e5	2144	hdr->len = (uint8_t)((f_new->payload - f_new->transport_hdr) << 2 \| t->jumbo);
daniele	29:1a47b7151851	2145
tass	45:ca30069e49bb	2146	tcp_dbg_nagle("NAGLE make - joined %d segments, len %d bytes\n",test,total_payload_len);
daniele	29:1a47b7151851	2147
daniele	29:1a47b7151851	2148	return f_new;
daniele	29:1a47b7151851	2149	}
daniele	29:1a47b7151851	2150
daniele	29:1a47b7151851	2151	/* original behavior kept when Nagle disabled;
daniele	29:1a47b7151851	2152	Nagle algorithm added here, keeping hold frame queue instead of eg linked list of data */
tass	70:cd218dd180e5	2153	int32_t pico_tcp_push(struct pico_protocol self, struct pico_frame f)
daniele	29:1a47b7151851	2154	{
daniele	29:1a47b7151851	2155	struct pico_tcp_hdr hdr = (struct pico_tcp_hdr )f->transport_hdr;
daniele	29:1a47b7151851	2156	struct pico_socket_tcp t = (struct pico_socket_tcp ) f->sock;
daniele	29:1a47b7151851	2157	struct pico_frame *f_new;
tass	70:cd218dd180e5	2158	uint32_t total_len = 0;
tass	68:0847e35d08a6	2159	IGNORE_PARAMETER(self);
tass	68:0847e35d08a6	2160	pico_err = PICO_ERR_NOERR;
daniele	29:1a47b7151851	2161	hdr->trans.sport = t->sock.local_port;
daniele	29:1a47b7151851	2162	hdr->trans.dport = t->sock.remote_port;
daniele	29:1a47b7151851	2163	hdr->seq = long_be(t->snd_last + 1);
tass	70:cd218dd180e5	2164	hdr->len = (uint8_t)((f->payload - f->transport_hdr) << 2 \| t->jumbo);
daniele	29:1a47b7151851	2165
tass	68:0847e35d08a6	2166	if ((uint32_t)f->payload_len > (uint32_t)(t->tcpq_out.max_size - t->tcpq_out.size))
tass	70:cd218dd180e5	2167	t->sock.ev_pending &= (uint16_t)(~PICO_SOCK_EV_WR);
daniele	29:1a47b7151851	2168
daniele	29:1a47b7151851	2169	/***************************************************************************/
daniele	29:1a47b7151851	2170
daniele	29:1a47b7151851	2171	if (!IS_NAGLE_ENABLED((&(t->sock)))) {
daniele	29:1a47b7151851	2172	/* TCP_NODELAY enabled, original behavior */
daniele	29:1a47b7151851	2173	if (pico_enqueue_segment(&t->tcpq_out,f) > 0) {
tass	45:ca30069e49bb	2174	tcp_dbg_nagle("TCP_PUSH - NO NAGLE - Pushing segment %08x, len %08x to socket %p\n", t->snd_last + 1, f->payload_len, t);
daniele	29:1a47b7151851	2175	t->snd_last += f->payload_len;
daniele	29:1a47b7151851	2176	return f->payload_len;
daniele	29:1a47b7151851	2177	} else {
daniele	29:1a47b7151851	2178	tcp_dbg("Enqueue failed.\n");
daniele	29:1a47b7151851	2179	return 0;
daniele	29:1a47b7151851	2180	}
daniele	29:1a47b7151851	2181	}
daniele	29:1a47b7151851	2182	/***************************************************************************/
daniele	29:1a47b7151851	2183	else {
daniele	29:1a47b7151851	2184	/* Nagle's algorithm enabled, check if ready to send, or put frame in hold queue */
daniele	29:1a47b7151851	2185	if (IS_TCP_IDLE(t) && IS_TCP_HOLDQ_EMPTY(t)) { /* opt 1. send frame */
daniele	29:1a47b7151851	2186	if (pico_enqueue_segment(&t->tcpq_out,f) > 0) {
tass	45:ca30069e49bb	2187	tcp_dbg_nagle("TCP_PUSH - NAGLE - Pushing segment %08x, len %08x to socket %p\n", t->snd_last + 1, f->payload_len, t);
daniele	29:1a47b7151851	2188	t->snd_last += f->payload_len;
daniele	29:1a47b7151851	2189	return f->payload_len;
daniele	29:1a47b7151851	2190	} else {
daniele	29:1a47b7151851	2191	tcp_dbg("Enqueue failed.\n");
daniele	29:1a47b7151851	2192	return 0;
daniele	29:1a47b7151851	2193	}
daniele	29:1a47b7151851	2194	} else { /* opt 2. hold data back */
daniele	29:1a47b7151851	2195	total_len = f->payload_len + t->tcpq_hold.size;
daniele	29:1a47b7151851	2196	if ((total_len >= PICO_TCP_DEFAULT_MSS) && ((t->tcpq_out.max_size - t->tcpq_out.size) >= PICO_TCP_DEFAULT_MSS)) {/* TODO check mss socket */
daniele	29:1a47b7151851	2197	/* IF enough data in hold (>mss) AND space in out queue (>mss) */
daniele	29:1a47b7151851	2198	/* add current frame in hold and make new segment */
daniele	29:1a47b7151851	2199	if (pico_enqueue_segment(&t->tcpq_hold,f) > 0 ) {
tass	45:ca30069e49bb	2200	tcp_dbg_nagle("TCP_PUSH - NAGLE - Pushed into hold, make new (enqueued frames out %d)\n",t->tcpq_out.frames);
daniele	29:1a47b7151851	2201	t->snd_last += f->payload_len; /* XXX WATCH OUT */
daniele	29:1a47b7151851	2202	f_new = pico_hold_segment_make(t);
daniele	29:1a47b7151851	2203	} else {
tass	45:ca30069e49bb	2204	tcp_dbg_nagle("TCP_PUSH - NAGLE - enqueue hold failed 1\n");
daniele	29:1a47b7151851	2205	return 0;
daniele	29:1a47b7151851	2206	}
daniele	29:1a47b7151851	2207	/* and put new frame in out queue */
daniele	29:1a47b7151851	2208	if ((f_new != NULL) && (pico_enqueue_segment(&t->tcpq_out,f_new) > 0)) {
daniele	29:1a47b7151851	2209	return f_new->payload_len;
daniele	29:1a47b7151851	2210	} else {
tass	45:ca30069e49bb	2211	tcp_dbg_nagle("TCP_PUSH - NAGLE - enqueue out failed, f_new = %p\n",f_new);
daniele	29:1a47b7151851	2212	return -1; /* XXX something seriously wrong */
daniele	29:1a47b7151851	2213	}
daniele	29:1a47b7151851	2214	} else {
daniele	29:1a47b7151851	2215	/* ELSE put frame in hold queue */
daniele	29:1a47b7151851	2216	if (pico_enqueue_segment(&t->tcpq_hold,f) > 0) {
tass	45:ca30069e49bb	2217	tcp_dbg_nagle("TCP_PUSH - NAGLE - Pushed into hold (enqueued frames out %d)\n",t->tcpq_out.frames);
daniele	29:1a47b7151851	2218	t->snd_last += f->payload_len; /* XXX WATCH OUT */
daniele	29:1a47b7151851	2219	return f->payload_len;
daniele	29:1a47b7151851	2220	} else {
tass	68:0847e35d08a6	2221	pico_err = PICO_ERR_EAGAIN;
tass	45:ca30069e49bb	2222	tcp_dbg_nagle("TCP_PUSH - NAGLE - enqueue hold failed 2\n");
daniele	29:1a47b7151851	2223	return 0;
daniele	29:1a47b7151851	2224	}
daniele	29:1a47b7151851	2225	}
daniele	29:1a47b7151851	2226	}
daniele	29:1a47b7151851	2227	}
daniele	29:1a47b7151851	2228	/***************************************************************************/
daniele	29:1a47b7151851	2229	}
tass	68:0847e35d08a6	2230
tass	68:0847e35d08a6	2231	inline static void tcp_discard_all_segments(struct pico_tcp_queue *tq)
tass	68:0847e35d08a6	2232	{
tass	68:0847e35d08a6	2233	struct pico_tree_node index = NULL, index_safe = NULL;
tass	68:0847e35d08a6	2234	LOCK(Mutex);
tass	68:0847e35d08a6	2235	pico_tree_foreach_safe(index,&tq->pool,index_safe)
tass	68:0847e35d08a6	2236	{
tass	68:0847e35d08a6	2237	struct pico_frame *f = index->keyValue;
tass	68:0847e35d08a6	2238	if(!f)
tass	68:0847e35d08a6	2239	break;
tass	68:0847e35d08a6	2240	pico_tree_delete(&tq->pool,f);
tass	68:0847e35d08a6	2241	pico_frame_discard(f);
tass	68:0847e35d08a6	2242	}
tass	68:0847e35d08a6	2243	tq->frames = 0;
tass	68:0847e35d08a6	2244	tq->size = 0;
tass	68:0847e35d08a6	2245	UNLOCK(Mutex);
tass	68:0847e35d08a6	2246	}
tass	68:0847e35d08a6	2247
tass	68:0847e35d08a6	2248	void pico_tcp_cleanup_queues(struct pico_socket *sck)
tass	68:0847e35d08a6	2249	{
tass	68:0847e35d08a6	2250	struct pico_socket_tcp * tcp = (struct pico_socket_tcp *)sck;
tass	70:cd218dd180e5	2251	if(tcp->retrans_tmr)
tass	70:cd218dd180e5	2252	pico_timer_cancel(tcp->retrans_tmr);
tass	68:0847e35d08a6	2253	tcp_discard_all_segments(&tcp->tcpq_in);
tass	68:0847e35d08a6	2254	tcp_discard_all_segments(&tcp->tcpq_out);
tass	68:0847e35d08a6	2255	tcp_discard_all_segments(&tcp->tcpq_hold);
tass	68:0847e35d08a6	2256	}
tass	68:0847e35d08a6	2257
tass	76:938a140caf12	2258	static int checkLocalClosing(struct pico_socket *s)
tass	76:938a140caf12	2259	{
tass	76:938a140caf12	2260	struct pico_socket_tcp * t = (struct pico_socket_tcp *)s;
tass	76:938a140caf12	2261	if ((s->state & PICO_SOCKET_STATE_TCP) == PICO_SOCKET_STATE_TCP_ESTABLISHED) {
tass	76:938a140caf12	2262	tcp_dbg("TCP> buffer empty, shutdown established ...\n");
tass	76:938a140caf12	2263	/* send fin if queue empty and in state shut local (write) */
tass	76:938a140caf12	2264	tcp_send_fin(t);
tass	76:938a140caf12	2265	/* change tcp state to FIN_WAIT1 */
tass	76:938a140caf12	2266	s->state &= 0x00FFU;
tass	76:938a140caf12	2267	s->state \|= PICO_SOCKET_STATE_TCP_FIN_WAIT1;
tass	76:938a140caf12	2268	return 1;
tass	76:938a140caf12	2269	}
tass	76:938a140caf12	2270	return 0;
tass	76:938a140caf12	2271	}
tass	76:938a140caf12	2272
tass	76:938a140caf12	2273	static int checkRemoteClosing(struct pico_socket *s)
tass	76:938a140caf12	2274	{
tass	76:938a140caf12	2275	struct pico_socket_tcp * t = (struct pico_socket_tcp *)s;
tass	76:938a140caf12	2276	if ((s->state & PICO_SOCKET_STATE_TCP) == PICO_SOCKET_STATE_TCP_CLOSE_WAIT) {
tass	76:938a140caf12	2277	/* send fin if queue empty and in state shut local (write) */
tass	76:938a140caf12	2278	tcp_send_fin(t);
tass	76:938a140caf12	2279	/* change tcp state to LAST_ACK */
tass	76:938a140caf12	2280	s->state &= 0x00FFU;
tass	76:938a140caf12	2281	s->state \|= PICO_SOCKET_STATE_TCP_LAST_ACK;
tass	76:938a140caf12	2282	tcp_dbg("TCP> STATE: LAST_ACK.\n");
tass	76:938a140caf12	2283	return 1;
tass	76:938a140caf12	2284	}
tass	76:938a140caf12	2285	return 0;
tass	76:938a140caf12	2286	}
tass	76:938a140caf12	2287
tass	76:938a140caf12	2288	void pico_tcp_notify_closing(struct pico_socket *sck)
tass	76:938a140caf12	2289	{
tass	76:938a140caf12	2290	struct pico_socket_tcp t=(struct pico_socket_tcp )sck;
tass	76:938a140caf12	2291	if(t->tcpq_out.frames == 0)
tass	76:938a140caf12	2292	{
tass	76:938a140caf12	2293	checkLocalClosing(sck);
tass	76:938a140caf12	2294	}
tass	76:938a140caf12	2295	}
tass	76:938a140caf12	2296
daniele	29:1a47b7151851	2297	#endif //PICO_SUPPORT_TCP
tass	79:6d1cb906043d	2298
tass	79:6d1cb906043d	2299