RIOT/sys/net/destiny/tcp.c

/*
 * tcp.c
 *
 *  Created on: 29.09.2011
 *      Author: Oliver
 */

#include <stdio.h>
#include <thread.h>
#include <string.h>
#include <stdlib.h>

#include "vtimer.h"
#include "tcp_timer.h"
#include "tcp_hc.h"
#include "tcp.h"
#include "in.h"
#include "socket.h"
#include "sys/net/net_help/net_help.h"
#include "sys/net/net_help/msg_help.h"

void printTCPHeader(tcp_hdr_t *tcp_header)
	{
	printf("\nBEGIN: TCP HEADER\n");
	printf("ack_nr: %lu\n", tcp_header->ack_nr);
	printf("checksum: %i\n", tcp_header->checksum);
	printf("dataOffset_reserved: %i\n", tcp_header->dataOffset_reserved);
	printf("dst_port: %i\n", tcp_header->dst_port);
	printf("reserved_flags: %i\n", tcp_header->reserved_flags);
	printf("seq_nr: %lu\n", tcp_header->seq_nr);
	printf("src_port: %i\n", tcp_header->src_port);
	printf("urg_pointer: %i\n", tcp_header->urg_pointer);
	printf("window: %i\n", tcp_header->window);
	printf("END: TCP HEADER\n");
	}

void printArrayRange_tcp(uint8_t *udp_header, uint16_t len)
	{
	int i = 0;
	printf("-------------MEMORY-------------\n");
	for (i = 0; i < len; i++)
		{
		printf("%#x ", *(udp_header+i));
		}
	printf("-------------MEMORY-------------\n");
	}

uint16_t tcp_csum(ipv6_hdr_t *ipv6_header, tcp_hdr_t *tcp_header)
	{
    uint16_t sum;
    uint16_t len = ipv6_header->length;

	sum = len + IPPROTO_TCP;
	sum = csum(sum, (uint8_t *)&ipv6_header->srcaddr, 2 * sizeof(ipv6_addr_t));
	sum = csum(sum, (uint8_t *)tcp_header, len);
    return (sum == 0) ? 0xffff : HTONS(sum);
	}

uint8_t handle_payload(ipv6_hdr_t *ipv6_header, tcp_hdr_t *tcp_header, socket_internal_t *tcp_socket, uint8_t *payload)
	{
	msg_t m_send_tcp, m_recv_tcp;
	uint8_t tcp_payload_len = ipv6_header->length-TCP_HDR_LEN;
	uint8_t acknowledged_bytes = 0;
	if (tcp_payload_len > tcp_socket->socket_values.tcp_control.rcv_wnd)
		{
		mutex_lock(&tcp_socket->tcp_buffer_mutex);
		memcpy(tcp_socket->tcp_input_buffer, payload, tcp_socket->socket_values.tcp_control.rcv_wnd);
		acknowledged_bytes = tcp_socket->socket_values.tcp_control.rcv_wnd;
		tcp_socket->socket_values.tcp_control.rcv_wnd = 0;
		tcp_socket->tcp_input_buffer_end = tcp_socket->tcp_input_buffer_end + tcp_socket->socket_values.tcp_control.rcv_wnd;
		mutex_unlock(&tcp_socket->tcp_buffer_mutex, 0);
		}
	else
		{
		mutex_lock(&tcp_socket->tcp_buffer_mutex);
		memcpy(tcp_socket->tcp_input_buffer, payload, tcp_payload_len);
		tcp_socket->socket_values.tcp_control.rcv_wnd = tcp_socket->socket_values.tcp_control.rcv_wnd - tcp_payload_len;
		acknowledged_bytes = tcp_payload_len;
		tcp_socket->tcp_input_buffer_end = tcp_socket->tcp_input_buffer_end + tcp_payload_len;
		mutex_unlock(&tcp_socket->tcp_buffer_mutex, 0);
		}
//	 msg_receive(&m_recv_tcp);
//	 printf("Sending MSG to Receiving Thread!\n");
	if (thread_getstatus(tcp_socket->recv_pid) == STATUS_RECEIVE_BLOCKED)
		{
		net_msg_send_recv(&m_send_tcp, &m_recv_tcp, tcp_socket->recv_pid, UNDEFINED);
		}

	return acknowledged_bytes;
	}

void handle_tcp_ack_packet(ipv6_hdr_t *ipv6_header, tcp_hdr_t *tcp_header, socket_internal_t *tcp_socket)
	{
	msg_t m_recv_tcp, m_send_tcp;
	uint8_t target_pid;

	if (tcp_socket->socket_values.tcp_control.state == LAST_ACK)
		{
		target_pid = tcp_socket->recv_pid;
		close_socket(tcp_socket);
		msg_send(&m_send_tcp, target_pid, 0);
		return;
		}
	else if (tcp_socket->socket_values.tcp_control.state == CLOSING)
		{
		msg_send(&m_send_tcp, tcp_socket->recv_pid, 0);
		msg_send(&m_send_tcp, tcp_socket->send_pid, 0);
		return;
		}
	// TODO: Find better way of handling queued sockets ACK packets
	else if (getWaitingConnectionSocket(tcp_socket->socket_id, ipv6_header, tcp_header) != NULL)
		{
		printf("sending ACK to queued socket!\n");
		m_send_tcp.content.ptr = (char*)tcp_header;
		net_msg_send_recv(&m_send_tcp, &m_recv_tcp, tcp_socket->recv_pid, TCP_ACK);
		return;
		}
	else if (tcp_socket->socket_values.tcp_control.state == ESTABLISHED)
		{
		m_send_tcp.content.ptr = (char*)tcp_header;
		net_msg_send(&m_send_tcp, tcp_socket->send_pid, 0, TCP_ACK);
		return;
		}
	printf("NO WAY OF HANDLING THIS ACK!\n");
	}

void handle_tcp_rst_packet(ipv6_hdr_t *ipv6_header, tcp_hdr_t *tcp_header, socket_internal_t *tcp_socket)
	{

	}

void handle_tcp_syn_packet(ipv6_hdr_t *ipv6_header, tcp_hdr_t *tcp_header, socket_internal_t *tcp_socket)
	{
	msg_t m_send_tcp;
	if (tcp_socket->socket_values.tcp_control.state == LISTEN)
		{
		socket_internal_t *new_socket = new_tcp_queued_socket(ipv6_header, tcp_header);
		if (new_socket != NULL)
			{
			// notify socket function accept(..) that a new connection request has arrived
			// No need to wait for an answer because the server accept() function isnt reading from anything other than the queued sockets
			net_msg_send(&m_send_tcp, tcp_socket->recv_pid, 0, TCP_SYN);
			}
		else
			{
			printf("Dropped TCP SYN Message because an error occured while requesting a new queued socket!\n");
			}
		}
	else
		{
		printf("Dropped TCP SYN Message because socket was not in state LISTEN!");
		}
	}

void handle_tcp_syn_ack_packet(ipv6_hdr_t *ipv6_header, tcp_hdr_t *tcp_header, socket_internal_t *tcp_socket)
	{
	msg_t m_send_tcp;
	if (tcp_socket->socket_values.tcp_control.state == SYN_SENT)
		{
		m_send_tcp.content.ptr = (char*) tcp_header;
		net_msg_send(&m_send_tcp, tcp_socket->recv_pid, 0, TCP_SYN_ACK);
		}
	else
		{
		printf("Socket not in state SYN_SENT, dropping SYN-ACK-packet!");
		}
	}

void handle_tcp_fin_packet(ipv6_hdr_t *ipv6_header, tcp_hdr_t *tcp_header, socket_internal_t *tcp_socket)
	{
	msg_t m_send;
	socket_t *current_tcp_socket = &tcp_socket->socket_values;
	uint8_t send_buffer[BUFFER_SIZE];
	ipv6_hdr_t *temp_ipv6_header = ((ipv6_hdr_t*)(&send_buffer));
	tcp_hdr_t *current_tcp_packet = ((tcp_hdr_t*)(&send_buffer[IPV6_HDR_LEN]));

	set_tcp_cb(&current_tcp_socket->tcp_control, tcp_header->seq_nr+1, current_tcp_socket->tcp_control.send_wnd, tcp_header->ack_nr,
					tcp_header->ack_nr, tcp_header->window);

	if (current_tcp_socket->tcp_control.state == FIN_WAIT_1)
		{
		current_tcp_socket->tcp_control.state = CLOSING;

		send_tcp(current_tcp_socket, current_tcp_packet, temp_ipv6_header, TCP_FIN_ACK, 0);
		}
	else
		{
		current_tcp_socket->tcp_control.state = LAST_ACK;

		send_tcp(current_tcp_socket, current_tcp_packet, temp_ipv6_header, TCP_FIN_ACK, 0);
		}
	net_msg_send(&m_send, tcp_socket->recv_pid, 0, CLOSE_CONN);
	}

void handle_tcp_fin_ack_packet(ipv6_hdr_t *ipv6_header, tcp_hdr_t *tcp_header, socket_internal_t *tcp_socket)
	{
	msg_t m_send;
	socket_t *current_tcp_socket = &tcp_socket->socket_values;
	uint8_t send_buffer[BUFFER_SIZE];
	ipv6_hdr_t *temp_ipv6_header = ((ipv6_hdr_t*)(&send_buffer));
	tcp_hdr_t *current_tcp_packet = ((tcp_hdr_t*)(&send_buffer[IPV6_HDR_LEN]));

	current_tcp_socket->tcp_control.state = CLOSED;

	set_tcp_cb(&current_tcp_socket->tcp_control, tcp_header->seq_nr+1, current_tcp_socket->tcp_control.send_wnd, tcp_header->ack_nr,
			tcp_header->ack_nr, tcp_header->window);

	send_tcp(current_tcp_socket, current_tcp_packet, temp_ipv6_header, TCP_ACK, 0);

	msg_send(&m_send, tcp_socket->send_pid, 0);
	msg_send(&m_send, tcp_socket->recv_pid, 0);
	}

void handle_tcp_no_flags_packet(ipv6_hdr_t *ipv6_header, tcp_hdr_t *tcp_header, socket_internal_t *tcp_socket, uint8_t *payload)
	{
	uint8_t tcp_payload_len = ipv6_header->length-TCP_HDR_LEN, read_bytes = 0;
	socket_t *current_tcp_socket = &tcp_socket->socket_values;
	uint8_t send_buffer[BUFFER_SIZE];
	ipv6_hdr_t *temp_ipv6_header = ((ipv6_hdr_t*)(&send_buffer));
	tcp_hdr_t *current_tcp_packet = ((tcp_hdr_t*)(&send_buffer[IPV6_HDR_LEN]));

	if (tcp_payload_len > 0)
		{
		read_bytes = handle_payload(ipv6_header, tcp_header, tcp_socket, payload);

		// Refresh TCP status values
		current_tcp_socket->tcp_control.state = ESTABLISHED;

//		printf("     INFOS: TCP Header Seq Nr: %lu, Read Bytes: %u, Both: %lu\n", tcp_header->seq_nr, read_bytes, tcp_header->seq_nr + read_bytes);

		set_tcp_cb(&current_tcp_socket->tcp_control,
				tcp_header->seq_nr + read_bytes,
				current_tcp_socket->tcp_control.rcv_wnd,
				current_tcp_socket->tcp_control.send_nxt,
				current_tcp_socket->tcp_control.send_una,
				current_tcp_socket->tcp_control.send_wnd);
//		printf("     INFOS: %lu\n", current_tcp_socket->tcp_control.rcv_nxt);
		// Send packet
		send_tcp(current_tcp_socket, current_tcp_packet, temp_ipv6_header, TCP_ACK, 0);
		}
	}

void tcp_packet_handler (void)
	{
	msg_t m_recv_ip, m_send_ip;
	ipv6_hdr_t *ipv6_header;
	tcp_hdr_t *tcp_header;
	uint8_t *payload;
	socket_internal_t *tcp_socket = NULL;
	uint16_t chksum;

	while (1)
		{
		msg_receive(&m_recv_ip);

		ipv6_header = ((ipv6_hdr_t*)m_recv_ip.content.ptr);
		tcp_header = ((tcp_hdr_t*)(m_recv_ip.content.ptr + IPV6_HDR_LEN));
		payload = (uint8_t*)(m_recv_ip.content.ptr + IPV6_HDR_LEN + TCP_HDR_LEN);

		chksum = tcp_csum(ipv6_header, tcp_header);
		tcp_socket = get_tcp_socket(ipv6_header, tcp_header);
		print_tcp_status(INC_PACKET, ipv6_header, tcp_header, &tcp_socket->socket_values);

		if ((chksum == 0xffff) && (tcp_socket != NULL))
			{
#ifdef TCP_HC
			update_tcp_hc_context(true, &tcp_socket->socket_values.tcp_control.tcp_context, tcp_header);
#endif
			// Remove reserved bits from tcp flags field
			uint8_t tcp_flags = tcp_header->reserved_flags & REMOVE_RESERVED;

			// TODO: URG Flag and PSH flag are currently being ignored
			switch (tcp_flags)
				{
				case TCP_ACK:
					{
					// only ACK Bit set
					handle_tcp_ack_packet(ipv6_header, tcp_header, tcp_socket);
					break;
					}
				case TCP_RST:
					{
					printf("RST Bit set!\n");
					// only RST Bit set
					handle_tcp_rst_packet(ipv6_header, tcp_header, tcp_socket);
					break;
					}
				case TCP_SYN:
					{
					// only SYN Bit set, look for matching, listening socket and request new queued socket
					printf("SYN Bit set!\n");
					handle_tcp_syn_packet(ipv6_header, tcp_header, tcp_socket);
					break;
					}
				case TCP_SYN_ACK:
					{
					// only SYN and ACK Bit set, complete three way handshake when socket in state SYN_SENT
					handle_tcp_syn_ack_packet(ipv6_header, tcp_header, tcp_socket);
					break;
					}
				case TCP_FIN:
					{
					printf("FIN Bit set!\n");
					// only FIN Bit set
					handle_tcp_fin_packet(ipv6_header, tcp_header, tcp_socket);
					break;
					}
				case TCP_FIN_ACK:
					{
					printf("FIN ACK Bit set!\n");
					// only FIN and ACK Bit set
					handle_tcp_fin_ack_packet(ipv6_header, tcp_header, tcp_socket);
					break;
					}
				default:
					{
//					printf("DEFAULT!\n");
					handle_tcp_no_flags_packet(ipv6_header, tcp_header, tcp_socket, payload);
					}
				}
			}
		else
			{
			printf("Wrong checksum (%x) or no corresponding socket found!\n", chksum);
			}

		msg_reply(&m_recv_ip, &m_send_ip);
		}
	}