StarMirror/toxcore
1
0
Fork 0
mirror of https://github.com/irungentoo/toxcore.git synced 2024-03-22 13:30:51 +08:00
Code Issues Projects Releases Wiki Activity

Some work on TCP server done.

Browse Source
This commit is contained in:
irungentoo 2014-03-22 18:29:09 -04:00
parent 04001b2ec3
commit cb1c1d25ec
No known key found for this signature in database
GPG Key ID: 10349DC9BED89E98
2 changed files with 84 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

84
toxcore/TCP_server.c
View File

@ -165,6 +165,7 @@ static int add_accepted(TCP_Server *TCP_server, TCP_Secure_Connection *con)
}
memcpy(&TCP_server->accepted_connection_array[index], con, sizeof(TCP_Secure_Connection));
TCP_server->accepted_connection_array[index].status = TCP_STATUS_CONFIRMED;
++TCP_server->num_accepted_connections;
return index;
}
@ -385,11 +386,70 @@ static int read_connection_handshake(TCP_Secure_Connection *con, uint8_t *self_s
return 0;
}
static int disconnect_conection_index(TCP_Server *TCP_server, TCP_Secure_Connection *con, uint8_t con_number)
{
if (con_number >= NUM_CLIENT_CONNECTIONS)
return -1;
uint32_t index = con->connections[con_number].index;
uint8_t other_id = con->connections[con_number].other_id;
if (index) {
--index;
if (index >= TCP_server->size_accepted_connections)
return -1;
TCP_server->accepted_connection_array[index].connections[other_id].other_id = 0;
TCP_server->accepted_connection_array[index].connections[other_id].index = 0;
con->connections[con_number].index = 0;
con->connections[con_number].other_id = 0;
return 0;
} else {
return -1;
}
}
/* return 0 on success
* return -1 on failure
*/
static int handle_TCP_packet(TCP_Secure_Connection *con, uint8_t *data, uint16_t length)
static int handle_TCP_packet(TCP_Server *TCP_server, TCP_Secure_Connection *con, uint8_t *data, uint16_t length)
{
if (length == 0)
return -1;
switch (data[0]) {
case TCP_PACKET_ROUTING_REQUEST: {
break;
}
case TCP_PACKET_CONNECTION_NOTIFICATION: {
break;
}
case TCP_PACKET_DISCONNECT_NOTIFICATION: {
if (length != 2)
return -1;
return disconnect_conection_index(TCP_server, con, data[1] - NUM_RESERVED_PORTS);
}
case TCP_PACKET_ONION_REQUEST: {
break;
}
case TCP_PACKET_ONION_RESPONSE: {
break;
}
default: {
break;
}
}
return 0;
}
@ -574,6 +634,28 @@ static void do_TCP_confirmed(TCP_Server *TCP_server)
{
uint32_t i;
for (i = 0; i < TCP_server->size_accepted_connections; ++i) {
TCP_Secure_Connection *conn = &TCP_server->accepted_connection_array[i];
if (conn->status != TCP_STATUS_CONFIRMED)
continue;
uint8_t packet[MAX_PACKET_SIZE];
int len = read_packet_TCP_secure_connection(conn, packet, sizeof(packet));
if (len == 0) {
continue;
} else if (len == -1) {
kill_TCP_connection(conn);
del_accepted(TCP_server, i);
continue;
} else {
if (handle_TCP_packet(TCP_server, conn, packet, len) == -1) {
kill_TCP_connection(conn);
del_accepted(TCP_server, i);
}
}
}
}
void do_TCP_server(TCP_Server *TCP_server)

2
toxcore/TCP_server.h
View File

@ -60,7 +60,7 @@ typedef struct TCP_Secure_Connection {
uint8_t shared_key[crypto_box_BEFORENMBYTES];
uint16_t next_packet_length;
struct {
struct TCP_Secure_Connection *connection;
uint32_t index;
uint8_t other_id;
} connections[NUM_CLIENT_CONNECTIONS];
} TCP_Secure_Connection;