mirror of
https://github.com/RIOT-OS/RIOT.git
synced 2024-12-29 04:50:03 +01:00
988 lines
30 KiB
C
988 lines
30 KiB
C
/*
|
|
* socket.c
|
|
*
|
|
* Created on: 16.09.2011
|
|
* Author: Oliver
|
|
*/
|
|
#include <thread.h>
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <stdlib.h>
|
|
#include "udp.h"
|
|
#include "tcp.h"
|
|
#include "socket.h"
|
|
#include "vtimer.h"
|
|
#include "tcp_timer.h"
|
|
#include "sys/net/net_help/net_help.h"
|
|
#include "sys/net/net_help/msg_help.h"
|
|
|
|
msg_t socket_msg_queue[IP_PKT_RECV_BUF_SIZE];
|
|
msg_t send_msg_queue[SEND_MSG_BUF_SIZE];
|
|
|
|
void print_tcp_flags (tcp_hdr_t *tcp_header)
|
|
{
|
|
printf("FLAGS: ");
|
|
switch(tcp_header->reserved_flags)
|
|
{
|
|
case TCP_ACK:
|
|
{
|
|
printf("ACK ");
|
|
break;
|
|
}
|
|
case TCP_RST:
|
|
{
|
|
printf("RST ");
|
|
break;
|
|
}
|
|
case TCP_SYN:
|
|
{
|
|
printf("SYN ");
|
|
break;
|
|
}
|
|
case TCP_FIN:
|
|
{
|
|
printf("FIN ");
|
|
break;
|
|
}
|
|
case TCP_URG_PSH:
|
|
{
|
|
printf("URG PSH ");
|
|
break;
|
|
}
|
|
case TCP_SYN_ACK:
|
|
{
|
|
printf("SYN ACK ");
|
|
break;
|
|
}
|
|
case TCP_FIN_ACK:
|
|
{
|
|
printf("FIN ACK ");
|
|
break;
|
|
}
|
|
}
|
|
printf("\n");
|
|
}
|
|
|
|
void print_tcp_cb(tcp_cb *cb)
|
|
{
|
|
printf("Send_ISS: %lu\nSend_UNA: %lu\nSend_NXT: %lu\nSend_WND: %u\n", cb->send_iss, cb->send_una, cb->send_nxt, cb->send_wnd);
|
|
printf("Rcv_IRS: %lu\nRcv_NXT: %lu\nRcv_WND: %u\n", cb->rcv_irs, cb->rcv_nxt, cb->rcv_wnd);
|
|
printf("Time difference: %lu, No_of_retries: %u, State: %u\n\n", timex_sub(vtimer_now(), cb->last_packet_time).microseconds, cb->no_of_retry, cb->state);
|
|
}
|
|
|
|
void print_tcp_status(int in_or_out, ipv6_hdr_t *ipv6_header, tcp_hdr_t *tcp_header, socket_t *tcp_socket)
|
|
{
|
|
printf("--- %s TCP packet: ---\n", (in_or_out == INC_PACKET ? "Incoming" : "Outgoing"));
|
|
printf("IPv6 Source:");
|
|
ipv6_print_addr(&ipv6_header->srcaddr);
|
|
printf("IPv6 Dest:");
|
|
ipv6_print_addr(&ipv6_header->destaddr);
|
|
printf("TCP Length: %i\n", ipv6_header->length-TCP_HDR_LEN);
|
|
printf("Source Port: %i, Dest. Port: %i\n", tcp_header->src_port, tcp_header->dst_port);
|
|
printf("ACK: %li, SEQ: %li, Window: %i\n", tcp_header->ack_nr, tcp_header->seq_nr, tcp_header->window);
|
|
print_tcp_flags(tcp_header);
|
|
print_tcp_cb(&tcp_socket->tcp_control);
|
|
}
|
|
|
|
void print_socket(socket_t *current_socket)
|
|
{
|
|
printf("Domain: %i, Type: %i, Protocol: %i \n",
|
|
current_socket->domain,
|
|
current_socket->type,
|
|
current_socket->protocol);
|
|
// printf("Local address: \tPort: %i, \tFamily: %i\n",
|
|
// NTOHS(current_socket->local_address.sin6_port),
|
|
// current_socket->local_address.sin6_family);
|
|
ipv6_print_addr(¤t_socket->local_address.sin6_addr);
|
|
// printf("Foreign address: \tPort: %i, \tFamily: %i\n",
|
|
// NTOHS(current_socket->foreign_address.sin6_port),
|
|
// current_socket->foreign_address.sin6_family);
|
|
ipv6_print_addr(¤t_socket->foreign_address.sin6_addr);
|
|
printf("Local Port: %u, Foreign Port: %u\n", NTOHS(current_socket->local_address.sin6_port),
|
|
NTOHS(current_socket->foreign_address.sin6_port));
|
|
}
|
|
|
|
void print_internal_socket(socket_internal_t *current_socket_internal)
|
|
{
|
|
socket_t *current_socket = ¤t_socket_internal->socket_values;
|
|
printf("\n--------------------------\n");
|
|
printf("ID: %i, RECV PID: %i SEND PID: %i\n", current_socket_internal->socket_id, current_socket_internal->recv_pid, current_socket_internal->send_pid);
|
|
print_socket(current_socket);
|
|
printf("\n--------------------------\n");
|
|
}
|
|
|
|
socket_internal_t *getSocket(uint8_t s)
|
|
{
|
|
if (exists_socket(s))
|
|
{
|
|
return &(sockets[s-1]);
|
|
}
|
|
else
|
|
{
|
|
return NULL;
|
|
}
|
|
}
|
|
|
|
void print_sockets(void)
|
|
{
|
|
int i;
|
|
printf("\n--- Socket list: ---\n");
|
|
for (i = 1; i < MAX_SOCKETS+1; i++)
|
|
{
|
|
if(getSocket(i) != NULL)
|
|
{
|
|
print_internal_socket(getSocket(i));
|
|
}
|
|
}
|
|
}
|
|
|
|
bool exists_socket(uint8_t socket)
|
|
{
|
|
if (sockets[socket-1].socket_id == 0)
|
|
{
|
|
return false;
|
|
}
|
|
else
|
|
{
|
|
return true;
|
|
}
|
|
}
|
|
|
|
void close_socket(socket_internal_t *current_socket)
|
|
{
|
|
printf("Closing Socket %i with size %i!\n", current_socket->socket_id, sizeof(socket_internal_t));
|
|
memset(current_socket, 0, sizeof(socket_internal_t));
|
|
}
|
|
|
|
bool isUDPSocket(uint8_t s)
|
|
{
|
|
if ( (exists_socket(s)) &&
|
|
(getSocket(s)->socket_values.domain == PF_INET6) &&
|
|
(getSocket(s)->socket_values.type == SOCK_DGRAM) &&
|
|
((getSocket(s)->socket_values.protocol == IPPROTO_UDP) ||
|
|
(getSocket(s)->socket_values.protocol == 0)))
|
|
return true;
|
|
else
|
|
return false;
|
|
}
|
|
|
|
bool isTCPSocket(uint8_t s)
|
|
{
|
|
if ( (exists_socket(s)) &&
|
|
(getSocket(s)->socket_values.domain == PF_INET6) &&
|
|
(getSocket(s)->socket_values.type == SOCK_STREAM) &&
|
|
((getSocket(s)->socket_values.protocol == IPPROTO_TCP) ||
|
|
(getSocket(s)->socket_values.protocol == 0)))
|
|
return true;
|
|
else
|
|
return false;
|
|
}
|
|
|
|
int bind_udp_socket(int s, sockaddr6_t *name, int namelen, uint8_t pid)
|
|
{
|
|
int i;
|
|
if (!exists_socket(s))
|
|
{
|
|
return -1;
|
|
}
|
|
for (i = 1; i < MAX_SOCKETS+1; i++)
|
|
{
|
|
if (isUDPSocket(i) && (getSocket(i)->socket_values.local_address.sin6_port == name->sin6_port))
|
|
{
|
|
return -1;
|
|
}
|
|
}
|
|
memcpy(&getSocket(s)->socket_values.local_address, name, namelen);
|
|
getSocket(s)->recv_pid = pid;
|
|
return 1;
|
|
}
|
|
|
|
int bind_tcp_socket(int s, sockaddr6_t *name, int namelen, uint8_t pid)
|
|
{
|
|
int i;
|
|
if (!exists_socket(s))
|
|
{
|
|
return -1;
|
|
}
|
|
for (i = 1; i < MAX_SOCKETS+1; i++)
|
|
{
|
|
if (isTCPSocket(i) && (getSocket(i)->socket_values.local_address.sin6_port == name->sin6_port))
|
|
{
|
|
return -1;
|
|
}
|
|
}
|
|
memcpy(&getSocket(s)->socket_values.local_address, name, namelen);
|
|
getSocket(s)->recv_pid = pid;
|
|
return 1;
|
|
}
|
|
|
|
int socket(int domain, int type, int protocol)
|
|
{
|
|
int i = 1;
|
|
while (getSocket(i) != NULL)
|
|
{
|
|
i++;
|
|
}
|
|
if (i > MAX_SOCKETS+1)
|
|
{
|
|
return -1;
|
|
}
|
|
else
|
|
{
|
|
socket_t *current_socket = &sockets[i-1].socket_values;
|
|
sockets[i-1].socket_id = i;
|
|
current_socket->domain = domain;
|
|
current_socket->type = type;
|
|
current_socket->protocol = protocol;
|
|
current_socket->tcp_control.state = CLOSED;
|
|
return sockets[i-1].socket_id;
|
|
}
|
|
}
|
|
|
|
socket_internal_t *get_udp_socket(ipv6_hdr_t *ipv6_header, udp_hdr_t *udp_header)
|
|
{
|
|
uint8_t i = 1;
|
|
while (i < MAX_SOCKETS+1)
|
|
{
|
|
if ( isUDPSocket(i) &&
|
|
(getSocket(i)->socket_values.local_address.sin6_port == udp_header->dst_port))
|
|
{
|
|
return getSocket(i);
|
|
}
|
|
i++;
|
|
}
|
|
return NULL;
|
|
}
|
|
|
|
bool is_four_touple (socket_internal_t *current_socket, ipv6_hdr_t *ipv6_header, tcp_hdr_t *tcp_header)
|
|
{
|
|
// printf("Local Port: %i, TCP Dest Port: %i\n", current_socket->socket_values.local_address.sin6_port, tcp_header->dst_port);
|
|
// printf("Foreign Port: %i, TCP Src Port: %i\n", current_socket->socket_values.foreign_address.sin6_port, tcp_header->src_port);
|
|
// ipv6_print_addr(¤t_socket->socket_values.local_address.sin6_addr);
|
|
// ipv6_print_addr(&ipv6_header->destaddr);
|
|
// ipv6_print_addr(¤t_socket->socket_values.foreign_address.sin6_addr);
|
|
// ipv6_print_addr(&ipv6_header->srcaddr);
|
|
// printf("\n");
|
|
return ((ipv6_get_addr_match(¤t_socket->socket_values.local_address.sin6_addr, &ipv6_header->destaddr) == 128) &&
|
|
(current_socket->socket_values.local_address.sin6_port == tcp_header->dst_port) &&
|
|
(ipv6_get_addr_match(¤t_socket->socket_values.foreign_address.sin6_addr, &ipv6_header->srcaddr) == 128) &&
|
|
(current_socket->socket_values.foreign_address.sin6_port == tcp_header->src_port));
|
|
}
|
|
|
|
socket_internal_t *get_tcp_socket(ipv6_hdr_t *ipv6_header, tcp_hdr_t *tcp_header)
|
|
{
|
|
uint8_t i = 1;
|
|
socket_internal_t *current_socket = NULL;
|
|
socket_internal_t *listening_socket = NULL;
|
|
uint8_t compare[16];
|
|
memset(compare, 0, 16);
|
|
|
|
while (i < MAX_SOCKETS+1)
|
|
{
|
|
current_socket = getSocket(i);
|
|
// Check for matching 4 touple, ESTABLISHED connection
|
|
if( isTCPSocket(i) && is_four_touple(current_socket, ipv6_header, tcp_header))
|
|
{
|
|
return current_socket;
|
|
}
|
|
// Sockets in LISTEN and SYN_RCVD state should only be tested on local TCP values
|
|
else if ( isTCPSocket(i) &&
|
|
((current_socket->socket_values.tcp_control.state == LISTEN) || (current_socket->socket_values.tcp_control.state == SYN_RCVD)) &&
|
|
(current_socket->socket_values.local_address.sin6_addr.uint8[15] == ipv6_header->destaddr.uint8[15]) &&
|
|
(current_socket->socket_values.local_address.sin6_port == tcp_header->dst_port) &&
|
|
(current_socket->socket_values.foreign_address.sin6_addr.uint8[15] == 0x00) &&
|
|
(current_socket->socket_values.foreign_address.sin6_port == 0))
|
|
{
|
|
listening_socket = current_socket;
|
|
}
|
|
i++;
|
|
}
|
|
// Return either NULL if nothing was matched or the listening 2 touple socket
|
|
return listening_socket;
|
|
}
|
|
|
|
uint16_t get_free_source_port(uint8_t protocol)
|
|
{
|
|
int i;
|
|
uint16_t biggest_port = EPHEMERAL_PORTS-1;
|
|
// Remember biggest ephemeral port number used so far and add 1
|
|
for (i = 0; i < MAX_SOCKETS; i++)
|
|
{
|
|
if ((sockets[i].socket_values.protocol == protocol) && (sockets[i].socket_values.local_address.sin6_port > biggest_port))
|
|
{
|
|
biggest_port = sockets[i].socket_values.local_address.sin6_port;
|
|
}
|
|
}
|
|
return biggest_port + 1;
|
|
}
|
|
|
|
void set_socket_address(sockaddr6_t *sockaddr, uint8_t sin6_family, uint16_t sin6_port, uint32_t sin6_flowinfo, ipv6_addr_t *sin6_addr)
|
|
{
|
|
sockaddr->sin6_family = sin6_family;
|
|
sockaddr->sin6_port = sin6_port;
|
|
sockaddr->sin6_flowinfo = sin6_flowinfo;
|
|
memcpy(&sockaddr->sin6_addr, sin6_addr, 16);
|
|
}
|
|
|
|
void set_tcp_packet(tcp_hdr_t *tcp_hdr, uint16_t src_port, uint16_t dst_port, uint32_t seq_nr, uint32_t ack_nr,
|
|
uint8_t dataOffset_reserved, uint8_t reserved_flags, uint16_t window, uint16_t checksum, uint16_t urg_pointer)
|
|
{
|
|
tcp_hdr->ack_nr = ack_nr;
|
|
tcp_hdr->checksum = checksum;
|
|
tcp_hdr->dataOffset_reserved = dataOffset_reserved;
|
|
tcp_hdr->dst_port = dst_port;
|
|
tcp_hdr->reserved_flags = reserved_flags;
|
|
tcp_hdr->seq_nr = seq_nr;
|
|
tcp_hdr->src_port = src_port;
|
|
tcp_hdr->urg_pointer = urg_pointer;
|
|
tcp_hdr->window = window;
|
|
}
|
|
|
|
// Check for consistent ACK and SEQ number
|
|
int check_tcp_consistency(socket_t *current_tcp_socket, tcp_hdr_t *tcp_header)
|
|
{
|
|
if (IS_TCP_ACK(tcp_header->reserved_flags))
|
|
{
|
|
if(tcp_header->ack_nr > (current_tcp_socket->tcp_control.send_nxt))
|
|
{
|
|
// ACK of not yet sent byte, discard
|
|
return ACK_NO_TOO_BIG;
|
|
}
|
|
else if (tcp_header->ack_nr < (current_tcp_socket->tcp_control.send_una))
|
|
{
|
|
// ACK of previous segments, maybe dropped?
|
|
return ACK_NO_TOO_SMALL;
|
|
}
|
|
}
|
|
if ((current_tcp_socket->tcp_control.rcv_nxt > 0) && (tcp_header->seq_nr <= current_tcp_socket->tcp_control.rcv_nxt))
|
|
{
|
|
// segment repetition, maybe ACK got lost?
|
|
return SEQ_NO_TOO_SMALL;
|
|
}
|
|
return PACKET_OK;
|
|
}
|
|
|
|
int send_tcp(socket_t *current_tcp_socket, tcp_hdr_t *current_tcp_packet, ipv6_hdr_t *temp_ipv6_header, uint8_t flags, uint8_t payload_length)
|
|
{
|
|
set_tcp_packet(current_tcp_packet, current_tcp_socket->local_address.sin6_port, current_tcp_socket->foreign_address.sin6_port,
|
|
(flags == TCP_ACK ? current_tcp_socket->tcp_control.send_una-1 : current_tcp_socket->tcp_control.send_una),
|
|
current_tcp_socket->tcp_control.rcv_nxt, 0, flags, current_tcp_socket->tcp_control.rcv_wnd, 0, 0);
|
|
|
|
// Fill IPv6 Header
|
|
memcpy(&(temp_ipv6_header->destaddr), ¤t_tcp_socket->foreign_address.sin6_addr, 16);
|
|
memcpy(&(temp_ipv6_header->srcaddr), ¤t_tcp_socket->local_address.sin6_addr, 16);
|
|
temp_ipv6_header->length = TCP_HDR_LEN + payload_length;
|
|
|
|
current_tcp_packet->checksum = ~tcp_csum(temp_ipv6_header, current_tcp_packet);
|
|
|
|
// send TCP SYN packet
|
|
sixlowpan_send(¤t_tcp_socket->foreign_address.sin6_addr, (uint8_t*)(current_tcp_packet), TCP_HDR_LEN+payload_length, IPPROTO_TCP, current_tcp_socket);
|
|
|
|
return 1;
|
|
}
|
|
|
|
void set_tcp_cb(tcp_cb *tcp_control, uint32_t rcv_nxt, uint16_t rcv_wnd, uint32_t send_nxt, uint32_t send_una, uint16_t send_wnd)
|
|
{
|
|
tcp_control->rcv_nxt = rcv_nxt;
|
|
tcp_control->rcv_wnd = rcv_wnd;
|
|
tcp_control->send_nxt = send_nxt;
|
|
tcp_control->send_una = send_una;
|
|
tcp_control->send_wnd = send_wnd;
|
|
}
|
|
|
|
int connect(int socket, sockaddr6_t *addr, uint32_t addrlen)
|
|
{
|
|
// Variables
|
|
ipv6_addr_t src_addr;
|
|
socket_internal_t *current_int_tcp_socket;
|
|
socket_t *current_tcp_socket;
|
|
msg_t msg_from_server;
|
|
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]));
|
|
|
|
// Check if socket exists
|
|
current_int_tcp_socket = getSocket(socket);
|
|
if (current_int_tcp_socket == NULL)
|
|
{
|
|
return -1;
|
|
}
|
|
|
|
current_tcp_socket = ¤t_int_tcp_socket->socket_values;
|
|
|
|
current_int_tcp_socket->recv_pid = thread_getpid();
|
|
|
|
// Local address information
|
|
ipv6_get_saddr(&src_addr, &addr->sin6_addr);
|
|
set_socket_address(¤t_tcp_socket->local_address, PF_INET6, HTONS(get_free_source_port(IPPROTO_TCP)), 0, &src_addr);
|
|
|
|
// Foreign address information
|
|
set_socket_address(¤t_tcp_socket->foreign_address, addr->sin6_family, addr->sin6_port, addr->sin6_flowinfo, &addr->sin6_addr);
|
|
|
|
// TODO: random number should be generated like common BSD socket implementation with a periodic timer increasing it
|
|
// TODO: Add TCP MSS option field
|
|
// Fill lcoal TCP socket information
|
|
srand(addr->sin6_port);
|
|
|
|
current_tcp_socket->tcp_control.rcv_irs = 0;
|
|
current_tcp_socket->tcp_control.send_iss = rand();
|
|
current_tcp_socket->tcp_control.state = SYN_SENT;
|
|
|
|
set_tcp_cb(¤t_tcp_socket->tcp_control, 0, STATIC_WINDOW, current_tcp_socket->tcp_control.send_iss, current_tcp_socket->tcp_control.send_iss, 0);
|
|
|
|
// Remember current time
|
|
current_tcp_socket->tcp_control.last_packet_time = vtimer_now();
|
|
current_tcp_socket->tcp_control.no_of_retry = 0;
|
|
|
|
msg_from_server.type = TCP_RETRY;
|
|
|
|
while (msg_from_server.type == TCP_RETRY)
|
|
{
|
|
// Send packet
|
|
send_tcp(current_tcp_socket, current_tcp_packet, temp_ipv6_header, TCP_SYN, 0);
|
|
|
|
// wait for SYN ACK or RETRY
|
|
msg_receive(&msg_from_server);
|
|
if (msg_from_server.type == TCP_TIMEOUT)
|
|
{
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
// Read packet content
|
|
tcp_hdr_t *tcp_header = ((tcp_hdr_t*)(msg_from_server.content.ptr));
|
|
|
|
// Check for consistency
|
|
if (tcp_header->ack_nr != current_tcp_socket->tcp_control.send_nxt+1)
|
|
{
|
|
printf("TCP packets not consistent!\n");
|
|
}
|
|
|
|
// Got SYN ACK from Server
|
|
// Refresh foreign TCP socket information
|
|
current_tcp_socket->tcp_control.state = ESTABLISHED;
|
|
current_tcp_socket->tcp_control.rcv_irs = tcp_header->seq_nr;
|
|
set_tcp_cb(¤t_tcp_socket->tcp_control, tcp_header->seq_nr+1, current_tcp_socket->tcp_control.rcv_wnd,
|
|
current_tcp_socket->tcp_control.send_una, current_tcp_socket->tcp_control.send_una, tcp_header->window);
|
|
current_tcp_socket->tcp_control.send_una++;
|
|
current_tcp_socket->tcp_control.send_nxt++;
|
|
|
|
// Send packet
|
|
send_tcp(current_tcp_socket, current_tcp_packet, temp_ipv6_header, TCP_ACK, 0);
|
|
return 0;
|
|
}
|
|
|
|
int32_t send(int s, void *msg, uint64_t len, int flags)
|
|
{
|
|
// Variables
|
|
msg_t recv_msg;
|
|
int32_t sent_bytes = 0;
|
|
socket_internal_t *current_int_tcp_socket;
|
|
socket_t *current_tcp_socket;
|
|
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]));
|
|
msg_init_queue(send_msg_queue, SEND_MSG_BUF_SIZE);
|
|
|
|
// Check if socket exists and is TCP socket
|
|
if (!isTCPSocket(s))
|
|
{
|
|
return -1;
|
|
}
|
|
|
|
current_int_tcp_socket = getSocket(s);
|
|
current_tcp_socket = ¤t_int_tcp_socket->socket_values;
|
|
|
|
// Check for ESTABLISHED STATE
|
|
if (current_tcp_socket->tcp_control.state != ESTABLISHED)
|
|
{
|
|
return -1;
|
|
}
|
|
|
|
// Add thread PID
|
|
current_int_tcp_socket->send_pid = thread_getpid();
|
|
|
|
current_tcp_socket->tcp_control.no_of_retry = 0;
|
|
|
|
while (1)
|
|
{
|
|
// Add packet data
|
|
if (len > current_tcp_socket->tcp_control.send_wnd)
|
|
{
|
|
memcpy(&send_buffer[IPV6_HDR_LEN+TCP_HDR_LEN], msg, current_tcp_socket->tcp_control.send_wnd);
|
|
sent_bytes = current_tcp_socket->tcp_control.send_wnd;
|
|
}
|
|
else
|
|
{
|
|
memcpy(&send_buffer[IPV6_HDR_LEN+TCP_HDR_LEN], msg, len);
|
|
sent_bytes = len;
|
|
}
|
|
|
|
current_tcp_socket->tcp_control.send_nxt += sent_bytes;
|
|
current_tcp_socket->tcp_control.send_wnd -= sent_bytes;
|
|
send_tcp(current_tcp_socket, current_tcp_packet, temp_ipv6_header, 0, sent_bytes);
|
|
|
|
// Remember current time
|
|
current_tcp_socket->tcp_control.last_packet_time = vtimer_now();
|
|
|
|
net_msg_receive(&recv_msg);
|
|
// printf("Send(): Got Message from TCP Layer: %u\n", recv_msg.type);
|
|
switch (recv_msg.type)
|
|
{
|
|
case TCP_ACK:
|
|
{
|
|
tcp_hdr_t *tcp_header = ((tcp_hdr_t*)(recv_msg.content.ptr));
|
|
// printf("send nxt: %li, tcp_ack_nr: %li\n", current_tcp_socket->tcp_control.send_nxt, tcp_header->ack_nr);
|
|
if (current_tcp_socket->tcp_control.send_nxt == tcp_header->ack_nr)
|
|
{
|
|
current_tcp_socket->tcp_control.send_una = tcp_header->ack_nr;
|
|
current_tcp_socket->tcp_control.send_nxt = tcp_header->ack_nr;
|
|
current_tcp_socket->tcp_control.send_wnd = tcp_header->window;
|
|
// Got ACK for every sent byte
|
|
// printf("Returning from sen().\n");
|
|
return sent_bytes;
|
|
}
|
|
else
|
|
{
|
|
// TODO: Handle retransmit of missing bytes
|
|
break;
|
|
}
|
|
}
|
|
case TCP_RETRY:
|
|
{
|
|
current_tcp_socket->tcp_control.send_nxt -= sent_bytes;
|
|
current_tcp_socket->tcp_control.send_wnd += sent_bytes;
|
|
break;
|
|
}
|
|
case TCP_TIMEOUT:
|
|
{
|
|
current_tcp_socket->tcp_control.send_nxt -= sent_bytes;
|
|
current_tcp_socket->tcp_control.send_wnd += sent_bytes;
|
|
return -1;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
return sent_bytes;
|
|
}
|
|
|
|
uint8_t read_from_socket(socket_internal_t *current_int_tcp_socket, void *buf, int len)
|
|
{
|
|
if (len > current_int_tcp_socket->tcp_input_buffer_end)
|
|
{
|
|
mutex_lock(¤t_int_tcp_socket->tcp_buffer_mutex);
|
|
uint8_t read_bytes = current_int_tcp_socket->tcp_input_buffer_end;
|
|
memcpy(buf, current_int_tcp_socket->tcp_input_buffer, current_int_tcp_socket->tcp_input_buffer_end);
|
|
current_int_tcp_socket->tcp_input_buffer_end = 0;
|
|
current_int_tcp_socket->socket_values.tcp_control.rcv_wnd += read_bytes;
|
|
mutex_unlock(¤t_int_tcp_socket->tcp_buffer_mutex, 0);
|
|
return read_bytes;
|
|
}
|
|
else
|
|
{
|
|
mutex_lock(¤t_int_tcp_socket->tcp_buffer_mutex);
|
|
memcpy(buf, current_int_tcp_socket->tcp_input_buffer, len);
|
|
memmove(current_int_tcp_socket->tcp_input_buffer, (current_int_tcp_socket->tcp_input_buffer+len), current_int_tcp_socket->tcp_input_buffer_end-len);
|
|
current_int_tcp_socket->tcp_input_buffer_end = current_int_tcp_socket->tcp_input_buffer_end-len;
|
|
current_int_tcp_socket->socket_values.tcp_control.rcv_wnd += len;
|
|
mutex_unlock(¤t_int_tcp_socket->tcp_buffer_mutex, 0);
|
|
return len;
|
|
}
|
|
}
|
|
|
|
int recv(int s, void *buf, uint64_t len, int flags)
|
|
{
|
|
// Variables
|
|
msg_t m_recv;
|
|
socket_internal_t *current_int_tcp_socket;
|
|
socket_t *current_tcp_socket;
|
|
msg_init_queue(socket_msg_queue, IP_PKT_RECV_BUF_SIZE);
|
|
// Check if socket exists
|
|
if (!isTCPSocket(s))
|
|
{
|
|
printf("INFO: NO TCP SOCKET!\n");
|
|
return -1;
|
|
}
|
|
|
|
current_int_tcp_socket = getSocket(s);
|
|
current_tcp_socket = ¤t_int_tcp_socket->socket_values;
|
|
|
|
// Setting Thread PID
|
|
current_int_tcp_socket->recv_pid = thread_getpid();
|
|
|
|
if (current_int_tcp_socket->tcp_input_buffer_end > 0)
|
|
{
|
|
return read_from_socket(current_int_tcp_socket, buf, len);
|
|
}
|
|
|
|
msg_receive(&m_recv);
|
|
|
|
if ((exists_socket(s)) && (current_int_tcp_socket->tcp_input_buffer_end > 0))
|
|
{
|
|
return read_from_socket(current_int_tcp_socket, buf, len);
|
|
}
|
|
|
|
// Received FIN
|
|
if (m_recv.type == CLOSE_CONN)
|
|
{
|
|
// Sent FIN_ACK, wait for ACK
|
|
msg_receive(&m_recv);
|
|
// Received ACK, return with closed socket!
|
|
return -1;
|
|
}
|
|
// Received Last ACK (connection closed) or no data to read yet
|
|
printf("Returning from Recv()!\n");
|
|
return -1;
|
|
}
|
|
|
|
int32_t recvfrom(int s, void *buf, uint64_t len, int flags, sockaddr6_t *from, uint32_t *fromlen)
|
|
{
|
|
if (isUDPSocket(s))
|
|
{
|
|
msg_t m_recv, m_send;
|
|
ipv6_hdr_t *ipv6_header;
|
|
udp_hdr_t *udp_header;
|
|
uint8_t *payload;
|
|
msg_receive(&m_recv);
|
|
ipv6_header = ((ipv6_hdr_t*)m_recv.content.ptr);
|
|
udp_header = ((udp_hdr_t*)(m_recv.content.ptr + IPV6_HDR_LEN));
|
|
payload = (uint8_t*)(m_recv.content.ptr + IPV6_HDR_LEN+UDP_HDR_LEN);
|
|
|
|
memset(buf, 0, len);
|
|
memcpy(buf, payload, udp_header->length);
|
|
memcpy(&from->sin6_addr, &ipv6_header->srcaddr, 16);
|
|
from->sin6_family = AF_INET6;
|
|
from->sin6_flowinfo = 0;
|
|
from->sin6_port = udp_header->src_port;
|
|
memcpy(fromlen, (void*)(sizeof(sockaddr6_t)), sizeof(fromlen));
|
|
msg_reply(&m_recv, &m_send);
|
|
return udp_header->length;
|
|
}
|
|
else if (isTCPSocket(s))
|
|
{
|
|
return recv(s, buf, len, flags);
|
|
}
|
|
else
|
|
{
|
|
printf("Socket Type not supported!\n");
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
int32_t sendto(int s, void *msg, uint64_t len, int flags, sockaddr6_t *to, uint32_t tolen)
|
|
{
|
|
if (isUDPSocket(s) && (getSocket(s)->socket_values.foreign_address.sin6_port == 0))
|
|
{
|
|
uint8_t send_buffer[BUFFER_SIZE];
|
|
|
|
ipv6_hdr_t *temp_ipv6_header = ((ipv6_hdr_t*)(&send_buffer));
|
|
udp_hdr_t *current_udp_packet = ((udp_hdr_t*)(&send_buffer[IPV6_HDR_LEN]));
|
|
uint8_t *payload = &send_buffer[IPV6_HDR_LEN+UDP_HDR_LEN];
|
|
|
|
// ipv6_print_addr(&to->sin6_addr);
|
|
|
|
memcpy(&(temp_ipv6_header->destaddr), &to->sin6_addr, 16);
|
|
ipv6_get_saddr(&(temp_ipv6_header->srcaddr), &(temp_ipv6_header->destaddr));
|
|
|
|
current_udp_packet->src_port = get_free_source_port(IPPROTO_UDP);
|
|
current_udp_packet->dst_port = to->sin6_port;
|
|
current_udp_packet->checksum = 0;
|
|
|
|
memcpy(payload, msg, len);
|
|
current_udp_packet->length = UDP_HDR_LEN + len;
|
|
temp_ipv6_header->length = UDP_HDR_LEN + len;
|
|
|
|
current_udp_packet->checksum = ~udp_csum(temp_ipv6_header, current_udp_packet);
|
|
|
|
sixlowpan_send(&to->sin6_addr, (uint8_t*)(current_udp_packet), current_udp_packet->length, IPPROTO_UDP, NULL);
|
|
return current_udp_packet->length;
|
|
}
|
|
else
|
|
{
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
int close(int s)
|
|
{
|
|
socket_internal_t *current_socket = getSocket(s);
|
|
if (current_socket != NULL)
|
|
{
|
|
// Variables
|
|
msg_t m_recv;
|
|
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]));
|
|
|
|
// Check if socket exists and is TCP socket
|
|
if (!isTCPSocket(s))
|
|
{
|
|
return -1;
|
|
}
|
|
|
|
// Check for ESTABLISHED STATE
|
|
if (current_socket->socket_values.tcp_control.state != ESTABLISHED)
|
|
{
|
|
close_socket(current_socket);
|
|
return 1;
|
|
}
|
|
|
|
current_socket->send_pid = thread_getpid();
|
|
|
|
// Refresh local TCP socket information
|
|
current_socket->socket_values.tcp_control.send_una++;
|
|
current_socket->socket_values.tcp_control.state = FIN_WAIT_1;
|
|
|
|
send_tcp(¤t_socket->socket_values, current_tcp_packet, temp_ipv6_header, TCP_FIN, 0);
|
|
msg_receive(&m_recv);
|
|
close_socket(current_socket);
|
|
printf("Returning from Close()!\n");
|
|
return 1;
|
|
}
|
|
else
|
|
{
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
int bind(int s, sockaddr6_t *name, int namelen)
|
|
{
|
|
if (exists_socket(s))
|
|
{
|
|
socket_t *current_socket = &getSocket(s)->socket_values;
|
|
switch (current_socket->domain)
|
|
{
|
|
case (PF_INET):
|
|
{
|
|
// Not provided
|
|
return -1;
|
|
break;
|
|
}
|
|
case (PF_INET6):
|
|
{
|
|
switch (current_socket->type)
|
|
{
|
|
// TCP
|
|
case (SOCK_STREAM):
|
|
{
|
|
if ((current_socket->protocol == 0) || (current_socket->protocol == IPPROTO_TCP))
|
|
{
|
|
return bind_tcp_socket(s, name, namelen, thread_getpid());
|
|
break;
|
|
}
|
|
else
|
|
{
|
|
return -1;
|
|
break;
|
|
}
|
|
break;
|
|
}
|
|
// UDP
|
|
case (SOCK_DGRAM):
|
|
{
|
|
if ((current_socket->protocol == 0) || (current_socket->protocol == IPPROTO_UDP))
|
|
{
|
|
return bind_udp_socket(s, name, namelen, thread_getpid());
|
|
break;
|
|
}
|
|
else
|
|
{
|
|
return -1;
|
|
break;
|
|
}
|
|
break;
|
|
}
|
|
case (SOCK_SEQPACKET):
|
|
{
|
|
// not provided
|
|
return -1;
|
|
break;
|
|
}
|
|
case (SOCK_RAW):
|
|
{
|
|
// not provided
|
|
return -1;
|
|
break;
|
|
}
|
|
default:
|
|
{
|
|
return -1;
|
|
break;
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
case (PF_UNIX):
|
|
{
|
|
// Not provided
|
|
return -1;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
printf("SOCKET DOES NOT EXIST!\n");
|
|
return -1;
|
|
}
|
|
return -1;
|
|
}
|
|
|
|
int listen(int s, int backlog)
|
|
{
|
|
if (isTCPSocket(s) && getSocket(s)->socket_values.tcp_control.state == CLOSED)
|
|
{
|
|
socket_internal_t *current_socket = getSocket(s);
|
|
current_socket->socket_values.tcp_control.state = LISTEN;
|
|
return 0;
|
|
}
|
|
else
|
|
{
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
socket_internal_t *getWaitingConnectionSocket(int socket, ipv6_hdr_t *ipv6_header, tcp_hdr_t *tcp_header)
|
|
{
|
|
int i;
|
|
socket_internal_t *current_socket, *listening_socket = getSocket(socket);
|
|
for (i = 0; i < MAX_SOCKETS; i++)
|
|
{
|
|
current_socket = getSocket(i);
|
|
// Connection establishment ACK, Check for 4 touple and state
|
|
if ((ipv6_header != NULL) && (tcp_header != NULL))
|
|
{
|
|
if (is_four_touple(current_socket, ipv6_header, tcp_header) && (current_socket->socket_values.tcp_control.state == SYN_RCVD))
|
|
{
|
|
return current_socket;
|
|
}
|
|
}
|
|
// Connection establishment SYN ACK, check only for port and state
|
|
else
|
|
{
|
|
if ((current_socket->socket_values.tcp_control.state == SYN_RCVD) &&
|
|
(current_socket->socket_values.local_address.sin6_port == listening_socket->socket_values.local_address.sin6_port))
|
|
{
|
|
return current_socket;
|
|
}
|
|
}
|
|
}
|
|
return NULL;
|
|
}
|
|
|
|
int handle_new_tcp_connection(socket_internal_t *current_queued_int_socket, socket_internal_t *server_socket, uint8_t pid)
|
|
{
|
|
msg_t msg_recv_client_ack, msg_send_client_ack;
|
|
socket_t *current_queued_socket = ¤t_queued_int_socket->socket_values;
|
|
uint8_t send_buffer[BUFFER_SIZE];
|
|
ipv6_hdr_t *temp_ipv6_header = ((ipv6_hdr_t*)(&send_buffer));
|
|
tcp_hdr_t *syn_ack_packet = ((tcp_hdr_t*)(&send_buffer[IPV6_HDR_LEN]));
|
|
|
|
current_queued_int_socket->recv_pid = thread_getpid();
|
|
|
|
// Remember current time
|
|
current_queued_int_socket->socket_values.tcp_control.last_packet_time = vtimer_now();
|
|
|
|
current_queued_int_socket->socket_values.tcp_control.no_of_retry = 0;
|
|
|
|
// Set message type to Retry for while loop
|
|
msg_recv_client_ack.type = TCP_RETRY;
|
|
|
|
while (msg_recv_client_ack.type == TCP_RETRY)
|
|
{
|
|
// Send packet
|
|
send_tcp(current_queued_socket, syn_ack_packet, temp_ipv6_header, TCP_SYN_ACK, 0);
|
|
|
|
// wait for ACK from Client
|
|
msg_receive(&msg_recv_client_ack);
|
|
if (msg_recv_client_ack.type == TCP_TIMEOUT)
|
|
{
|
|
// Set status of internal socket back to LISTEN
|
|
server_socket->socket_values.tcp_control.state = LISTEN;
|
|
|
|
close_socket(current_queued_int_socket);
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
tcp_hdr_t *tcp_header;
|
|
|
|
tcp_header = ((tcp_hdr_t*)(msg_recv_client_ack.content.ptr));
|
|
|
|
// Check for consistency
|
|
if (tcp_header->ack_nr != current_queued_socket->tcp_control.send_nxt+1)
|
|
{
|
|
printf("TCP packets not consistent!\n");
|
|
}
|
|
|
|
// Got ack, connection established, refresh local and foreign tcp socket status
|
|
set_tcp_cb(¤t_queued_socket->tcp_control, tcp_header->seq_nr+1, current_queued_socket->tcp_control.rcv_wnd, tcp_header->ack_nr,
|
|
tcp_header->ack_nr, tcp_header->window);
|
|
current_queued_socket->tcp_control.state = ESTABLISHED;
|
|
|
|
// Set status of internal socket back to LISTEN
|
|
server_socket->socket_values.tcp_control.state = LISTEN;
|
|
|
|
// send a reply to the TCP handler after processing every information from the TCP ACK packet
|
|
msg_reply(&msg_recv_client_ack, &msg_send_client_ack);
|
|
|
|
print_sockets();
|
|
|
|
return current_queued_int_socket->socket_id;
|
|
}
|
|
|
|
int accept(int s, sockaddr6_t *addr, uint32_t addrlen)
|
|
{
|
|
socket_internal_t *server_socket = getSocket(s);
|
|
if (isTCPSocket(s) && (server_socket->socket_values.tcp_control.state == LISTEN))
|
|
{
|
|
socket_internal_t *current_queued_socket = getWaitingConnectionSocket(s, NULL, NULL);
|
|
if (current_queued_socket != NULL)
|
|
{
|
|
return handle_new_tcp_connection(current_queued_socket, server_socket, thread_getpid());
|
|
}
|
|
else
|
|
{
|
|
// No waiting connections, waiting for message from TCP Layer
|
|
msg_t msg_recv_client_syn;
|
|
msg_recv_client_syn.type = UNDEFINED;
|
|
while (msg_recv_client_syn.type != TCP_SYN)
|
|
{
|
|
msg_receive(&msg_recv_client_syn);
|
|
}
|
|
|
|
current_queued_socket = getWaitingConnectionSocket(s, NULL, NULL);
|
|
|
|
return handle_new_tcp_connection(current_queued_socket, server_socket, thread_getpid());
|
|
}
|
|
}
|
|
else
|
|
{
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
socket_internal_t *new_tcp_queued_socket(ipv6_hdr_t *ipv6_header, tcp_hdr_t *tcp_header)
|
|
{
|
|
int queued_socket_id;
|
|
|
|
queued_socket_id = socket(PF_INET6, SOCK_STREAM, IPPROTO_TCP);
|
|
socket_internal_t *current_queued_socket = getSocket(queued_socket_id);
|
|
|
|
// Foreign address
|
|
set_socket_address(¤t_queued_socket->socket_values.foreign_address, AF_INET6, tcp_header->src_port, ipv6_header->flowlabel, &ipv6_header->srcaddr);
|
|
|
|
// Local address
|
|
set_socket_address(¤t_queued_socket->socket_values.local_address, AF_INET6, tcp_header->dst_port, 0, &ipv6_header->destaddr);
|
|
|
|
// Foreign TCP information
|
|
current_queued_socket->socket_values.tcp_control.rcv_irs = tcp_header->seq_nr;
|
|
current_queued_socket->socket_values.tcp_control.send_iss = rand();
|
|
current_queued_socket->socket_values.tcp_control.state = SYN_RCVD;
|
|
set_tcp_cb(¤t_queued_socket->socket_values.tcp_control, tcp_header->seq_nr+1, STATIC_WINDOW,
|
|
current_queued_socket->socket_values.tcp_control.send_iss,
|
|
current_queued_socket->socket_values.tcp_control.send_iss, tcp_header->window);
|
|
|
|
return current_queued_socket;
|
|
}
|