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 "tcp.h"
#include "in.h"
#include "socket.h"
#include "sys/net/net_help/net_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);
	}

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;
	if (getWaitingConnectionSocket(tcp_socket->socket_id)->local_tcp_status.state == SYN_RCVD)
		{
		msg_send_receive(&m_send_tcp, &m_recv_tcp, tcp_socket->pid);
		}
	}

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->in_socket.local_tcp_status.state == LISTEN)
		{
		socket_t *new_socket = new_tcp_queued_socket(ipv6_header, tcp_header, tcp_socket);
		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
			msg_send(&m_send_tcp, tcp_socket->pid, 0);
			}
		else
			{
			printf("Dropped TCP SYN Message because an error occured while requesting a new queued socket!\n");
			}
		}
	else
		{
		printf("Droppec 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_recv_tcp, m_send_tcp;
	if (tcp_socket->in_socket.local_tcp_status.state == SYN_SENT)
		{
		msg_send_receive(&m_send_tcp, &m_recv_tcp, tcp_socket->pid);
		}
	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)
	{

	}

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

	}

void handle_tcp_no_flags_packet(ipv6_hdr_t *ipv6_header, tcp_hdr_t *tcp_header, socket_internal_t *tcp_socket, uint8_t *payload)
	{
	char message[128];
	memcpy(message, payload, ipv6_header->length-TCP_HDR_LEN);
	printf("Packet-Content: %s\n", message);
	}

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*)&buffer_tcp);
		tcp_header = ((tcp_hdr_t*)(&buffer_tcp[IPV6_HDR_LEN]));
		payload = &buffer_tcp[IPV6_HDR_LEN+TCP_HDR_LEN];
		//printTCPHeader(tcp_header);
		chksum = tcp_csum(ipv6_header, tcp_header);
		printf("Checksum is %x!\n", chksum);

		tcp_socket = get_tcp_socket(ipv6_header, tcp_header);

		print_tcp_status(INC_PACKET, ipv6_header, tcp_header);

		if ((chksum == 0xffff) && (tcp_socket != NULL))
			{
			// 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);
			}

		tcp_socket = NULL;
		msg_reply(&m_recv_ip, &m_send_ip);
		}
	}