mirror of
https://github.com/irungentoo/toxcore.git
synced 2024-03-22 13:30:51 +08:00
1057 lines
29 KiB
C
1057 lines
29 KiB
C
/* SPDX-License-Identifier: GPL-3.0-or-later
|
|
* Copyright © 2016-2018 The TokTok team.
|
|
* Copyright © 2014 Tox project.
|
|
*/
|
|
|
|
/*
|
|
* Implementation of the TCP relay client part of Tox.
|
|
*/
|
|
#ifdef HAVE_CONFIG_H
|
|
#include "config.h"
|
|
#endif
|
|
|
|
#include "TCP_client.h"
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
#include "mono_time.h"
|
|
#include "util.h"
|
|
|
|
typedef struct TCP_Client_Conn {
|
|
// TODO(iphydf): Add an enum for this.
|
|
uint8_t status; /* 0 if not used, 1 if other is offline, 2 if other is online. */
|
|
uint8_t public_key[CRYPTO_PUBLIC_KEY_SIZE];
|
|
uint32_t number;
|
|
} TCP_Client_Conn;
|
|
|
|
struct TCP_Client_Connection {
|
|
TCP_Client_Status status;
|
|
Socket sock;
|
|
uint8_t self_public_key[CRYPTO_PUBLIC_KEY_SIZE]; /* our public key */
|
|
uint8_t public_key[CRYPTO_PUBLIC_KEY_SIZE]; /* public key of the server */
|
|
IP_Port ip_port; /* The ip and port of the server */
|
|
TCP_Proxy_Info proxy_info;
|
|
uint8_t recv_nonce[CRYPTO_NONCE_SIZE]; /* Nonce of received packets. */
|
|
uint8_t sent_nonce[CRYPTO_NONCE_SIZE]; /* Nonce of sent packets. */
|
|
uint8_t shared_key[CRYPTO_SHARED_KEY_SIZE];
|
|
uint16_t next_packet_length;
|
|
|
|
uint8_t temp_secret_key[CRYPTO_SECRET_KEY_SIZE];
|
|
|
|
uint8_t last_packet[2 + MAX_PACKET_SIZE];
|
|
uint16_t last_packet_length;
|
|
uint16_t last_packet_sent;
|
|
|
|
TCP_Priority_List *priority_queue_start;
|
|
TCP_Priority_List *priority_queue_end;
|
|
|
|
uint64_t kill_at;
|
|
|
|
uint64_t last_pinged;
|
|
uint64_t ping_id;
|
|
|
|
uint64_t ping_response_id;
|
|
uint64_t ping_request_id;
|
|
|
|
TCP_Client_Conn connections[NUM_CLIENT_CONNECTIONS];
|
|
tcp_routing_response_cb *response_callback;
|
|
void *response_callback_object;
|
|
tcp_routing_status_cb *status_callback;
|
|
void *status_callback_object;
|
|
tcp_routing_data_cb *data_callback;
|
|
void *data_callback_object;
|
|
tcp_oob_data_cb *oob_data_callback;
|
|
void *oob_data_callback_object;
|
|
|
|
tcp_onion_response_cb *onion_callback;
|
|
void *onion_callback_object;
|
|
|
|
/* Can be used by user. */
|
|
void *custom_object;
|
|
uint32_t custom_uint;
|
|
};
|
|
|
|
const uint8_t *tcp_con_public_key(const TCP_Client_Connection *con)
|
|
{
|
|
return con->public_key;
|
|
}
|
|
|
|
IP_Port tcp_con_ip_port(const TCP_Client_Connection *con)
|
|
{
|
|
return con->ip_port;
|
|
}
|
|
|
|
TCP_Client_Status tcp_con_status(const TCP_Client_Connection *con)
|
|
{
|
|
return con->status;
|
|
}
|
|
void *tcp_con_custom_object(const TCP_Client_Connection *con)
|
|
{
|
|
return con->custom_object;
|
|
}
|
|
uint32_t tcp_con_custom_uint(const TCP_Client_Connection *con)
|
|
{
|
|
return con->custom_uint;
|
|
}
|
|
void tcp_con_set_custom_object(TCP_Client_Connection *con, void *object)
|
|
{
|
|
con->custom_object = object;
|
|
}
|
|
void tcp_con_set_custom_uint(TCP_Client_Connection *con, uint32_t value)
|
|
{
|
|
con->custom_uint = value;
|
|
}
|
|
|
|
/* return 1 on success
|
|
* return 0 on failure
|
|
*/
|
|
static int connect_sock_to(Socket sock, IP_Port ip_port, TCP_Proxy_Info *proxy_info)
|
|
{
|
|
if (proxy_info->proxy_type != TCP_PROXY_NONE) {
|
|
ip_port = proxy_info->ip_port;
|
|
}
|
|
|
|
/* nonblocking socket, connect will never return success */
|
|
net_connect(sock, ip_port);
|
|
return 1;
|
|
}
|
|
|
|
/* return 1 on success.
|
|
* return 0 on failure.
|
|
*/
|
|
static int proxy_http_generate_connection_request(TCP_Client_Connection *tcp_conn)
|
|
{
|
|
char one[] = "CONNECT ";
|
|
char two[] = " HTTP/1.1\nHost: ";
|
|
char three[] = "\r\n\r\n";
|
|
|
|
char ip[TOX_INET6_ADDRSTRLEN];
|
|
|
|
if (!ip_parse_addr(&tcp_conn->ip_port.ip, ip, sizeof(ip))) {
|
|
return 0;
|
|
}
|
|
|
|
const uint16_t port = net_ntohs(tcp_conn->ip_port.port);
|
|
const int written = snprintf((char *)tcp_conn->last_packet, MAX_PACKET_SIZE, "%s%s:%hu%s%s:%hu%s", one, ip, port, two,
|
|
ip, port, three);
|
|
|
|
if (written < 0 || MAX_PACKET_SIZE < written) {
|
|
return 0;
|
|
}
|
|
|
|
tcp_conn->last_packet_length = written;
|
|
tcp_conn->last_packet_sent = 0;
|
|
|
|
return 1;
|
|
}
|
|
|
|
/* return 1 on success.
|
|
* return 0 if no data received.
|
|
* return -1 on failure (connection refused).
|
|
*/
|
|
static int proxy_http_read_connection_response(TCP_Client_Connection *tcp_conn)
|
|
{
|
|
char success[] = "200";
|
|
uint8_t data[16]; // draining works the best if the length is a power of 2
|
|
|
|
int ret = read_TCP_packet(tcp_conn->sock, data, sizeof(data) - 1);
|
|
|
|
if (ret == -1) {
|
|
return 0;
|
|
}
|
|
|
|
data[sizeof(data) - 1] = 0;
|
|
|
|
if (strstr((char *)data, success)) {
|
|
// drain all data
|
|
unsigned int data_left = net_socket_data_recv_buffer(tcp_conn->sock);
|
|
|
|
if (data_left) {
|
|
VLA(uint8_t, temp_data, data_left);
|
|
read_TCP_packet(tcp_conn->sock, temp_data, data_left);
|
|
}
|
|
|
|
return 1;
|
|
}
|
|
|
|
return -1;
|
|
}
|
|
|
|
static void proxy_socks5_generate_handshake(TCP_Client_Connection *tcp_conn)
|
|
{
|
|
tcp_conn->last_packet[0] = 5; /* SOCKSv5 */
|
|
tcp_conn->last_packet[1] = 1; /* number of authentication methods supported */
|
|
tcp_conn->last_packet[2] = 0; /* No authentication */
|
|
|
|
tcp_conn->last_packet_length = 3;
|
|
tcp_conn->last_packet_sent = 0;
|
|
}
|
|
|
|
/* return 1 on success.
|
|
* return 0 if no data received.
|
|
* return -1 on failure (connection refused).
|
|
*/
|
|
static int socks5_read_handshake_response(TCP_Client_Connection *tcp_conn)
|
|
{
|
|
uint8_t data[2];
|
|
int ret = read_TCP_packet(tcp_conn->sock, data, sizeof(data));
|
|
|
|
if (ret == -1) {
|
|
return 0;
|
|
}
|
|
|
|
if (data[0] == 5 && data[1] == 0) { // TODO(irungentoo): magic numbers
|
|
return 1;
|
|
}
|
|
|
|
return -1;
|
|
}
|
|
|
|
static void proxy_socks5_generate_connection_request(TCP_Client_Connection *tcp_conn)
|
|
{
|
|
tcp_conn->last_packet[0] = 5; /* SOCKSv5 */
|
|
tcp_conn->last_packet[1] = 1; /* command code: establish a TCP/IP stream connection */
|
|
tcp_conn->last_packet[2] = 0; /* reserved, must be 0 */
|
|
uint16_t length = 3;
|
|
|
|
if (net_family_is_ipv4(tcp_conn->ip_port.ip.family)) {
|
|
tcp_conn->last_packet[3] = 1; /* IPv4 address */
|
|
++length;
|
|
memcpy(tcp_conn->last_packet + length, tcp_conn->ip_port.ip.ip.v4.uint8, sizeof(IP4));
|
|
length += sizeof(IP4);
|
|
} else {
|
|
tcp_conn->last_packet[3] = 4; /* IPv6 address */
|
|
++length;
|
|
memcpy(tcp_conn->last_packet + length, tcp_conn->ip_port.ip.ip.v6.uint8, sizeof(IP6));
|
|
length += sizeof(IP6);
|
|
}
|
|
|
|
memcpy(tcp_conn->last_packet + length, &tcp_conn->ip_port.port, sizeof(uint16_t));
|
|
length += sizeof(uint16_t);
|
|
|
|
tcp_conn->last_packet_length = length;
|
|
tcp_conn->last_packet_sent = 0;
|
|
}
|
|
|
|
/* return 1 on success.
|
|
* return 0 if no data received.
|
|
* return -1 on failure (connection refused).
|
|
*/
|
|
static int proxy_socks5_read_connection_response(TCP_Client_Connection *tcp_conn)
|
|
{
|
|
if (net_family_is_ipv4(tcp_conn->ip_port.ip.family)) {
|
|
uint8_t data[4 + sizeof(IP4) + sizeof(uint16_t)];
|
|
int ret = read_TCP_packet(tcp_conn->sock, data, sizeof(data));
|
|
|
|
if (ret == -1) {
|
|
return 0;
|
|
}
|
|
|
|
if (data[0] == 5 && data[1] == 0) {
|
|
return 1;
|
|
}
|
|
} else {
|
|
uint8_t data[4 + sizeof(IP6) + sizeof(uint16_t)];
|
|
int ret = read_TCP_packet(tcp_conn->sock, data, sizeof(data));
|
|
|
|
if (ret == -1) {
|
|
return 0;
|
|
}
|
|
|
|
if (data[0] == 5 && data[1] == 0) {
|
|
return 1;
|
|
}
|
|
}
|
|
|
|
return -1;
|
|
}
|
|
|
|
/* return 0 on success.
|
|
* return -1 on failure.
|
|
*/
|
|
static int generate_handshake(TCP_Client_Connection *tcp_conn)
|
|
{
|
|
uint8_t plain[CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE];
|
|
crypto_new_keypair(plain, tcp_conn->temp_secret_key);
|
|
random_nonce(tcp_conn->sent_nonce);
|
|
memcpy(plain + CRYPTO_PUBLIC_KEY_SIZE, tcp_conn->sent_nonce, CRYPTO_NONCE_SIZE);
|
|
memcpy(tcp_conn->last_packet, tcp_conn->self_public_key, CRYPTO_PUBLIC_KEY_SIZE);
|
|
random_nonce(tcp_conn->last_packet + CRYPTO_PUBLIC_KEY_SIZE);
|
|
int len = encrypt_data_symmetric(tcp_conn->shared_key, tcp_conn->last_packet + CRYPTO_PUBLIC_KEY_SIZE, plain,
|
|
sizeof(plain), tcp_conn->last_packet + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE);
|
|
|
|
if (len != sizeof(plain) + CRYPTO_MAC_SIZE) {
|
|
return -1;
|
|
}
|
|
|
|
tcp_conn->last_packet_length = CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE + sizeof(plain) + CRYPTO_MAC_SIZE;
|
|
tcp_conn->last_packet_sent = 0;
|
|
return 0;
|
|
}
|
|
|
|
/* data must be of length TCP_SERVER_HANDSHAKE_SIZE
|
|
*
|
|
* return 0 on success.
|
|
* return -1 on failure.
|
|
*/
|
|
static int handle_handshake(TCP_Client_Connection *tcp_conn, const uint8_t *data)
|
|
{
|
|
uint8_t plain[CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE];
|
|
int len = decrypt_data_symmetric(tcp_conn->shared_key, data, data + CRYPTO_NONCE_SIZE,
|
|
TCP_SERVER_HANDSHAKE_SIZE - CRYPTO_NONCE_SIZE, plain);
|
|
|
|
if (len != sizeof(plain)) {
|
|
return -1;
|
|
}
|
|
|
|
memcpy(tcp_conn->recv_nonce, plain + CRYPTO_PUBLIC_KEY_SIZE, CRYPTO_NONCE_SIZE);
|
|
encrypt_precompute(plain, tcp_conn->temp_secret_key, tcp_conn->shared_key);
|
|
crypto_memzero(tcp_conn->temp_secret_key, CRYPTO_SECRET_KEY_SIZE);
|
|
return 0;
|
|
}
|
|
|
|
/* return 0 if pending data was sent completely
|
|
* return -1 if it wasn't
|
|
*/
|
|
static int client_send_pending_data_nonpriority(TCP_Client_Connection *con)
|
|
{
|
|
if (con->last_packet_length == 0) {
|
|
return 0;
|
|
}
|
|
|
|
const uint16_t left = con->last_packet_length - con->last_packet_sent;
|
|
const int len = net_send(con->sock, con->last_packet + con->last_packet_sent, left);
|
|
|
|
if (len <= 0) {
|
|
return -1;
|
|
}
|
|
|
|
if (len == left) {
|
|
con->last_packet_length = 0;
|
|
con->last_packet_sent = 0;
|
|
return 0;
|
|
}
|
|
|
|
con->last_packet_sent += len;
|
|
return -1;
|
|
}
|
|
|
|
/* return 0 if pending data was sent completely
|
|
* return -1 if it wasn't
|
|
*/
|
|
static int client_send_pending_data(TCP_Client_Connection *con)
|
|
{
|
|
/* finish sending current non-priority packet */
|
|
if (client_send_pending_data_nonpriority(con) == -1) {
|
|
return -1;
|
|
}
|
|
|
|
TCP_Priority_List *p = con->priority_queue_start;
|
|
|
|
while (p) {
|
|
const uint16_t left = p->size - p->sent;
|
|
const int len = net_send(con->sock, p->data + p->sent, left);
|
|
|
|
if (len != left) {
|
|
if (len > 0) {
|
|
p->sent += len;
|
|
}
|
|
|
|
break;
|
|
}
|
|
|
|
TCP_Priority_List *pp = p;
|
|
p = p->next;
|
|
free(pp);
|
|
}
|
|
|
|
con->priority_queue_start = p;
|
|
|
|
if (!p) {
|
|
con->priority_queue_end = nullptr;
|
|
return 0;
|
|
}
|
|
|
|
return -1;
|
|
}
|
|
|
|
/* return 0 on failure (only if malloc fails)
|
|
* return 1 on success
|
|
*/
|
|
static bool client_add_priority(TCP_Client_Connection *con, const uint8_t *packet, uint16_t size, uint16_t sent)
|
|
{
|
|
TCP_Priority_List *p = con->priority_queue_end;
|
|
TCP_Priority_List *new_list = (TCP_Priority_List *)malloc(sizeof(TCP_Priority_List) + size);
|
|
|
|
if (!new_list) {
|
|
return 0;
|
|
}
|
|
|
|
new_list->next = nullptr;
|
|
new_list->size = size;
|
|
new_list->sent = sent;
|
|
memcpy(new_list->data, packet, size);
|
|
|
|
if (p) {
|
|
p->next = new_list;
|
|
} else {
|
|
con->priority_queue_start = new_list;
|
|
}
|
|
|
|
con->priority_queue_end = new_list;
|
|
return 1;
|
|
}
|
|
|
|
/* return 1 on success.
|
|
* return 0 if could not send packet.
|
|
* return -1 on failure (connection must be killed).
|
|
*/
|
|
static int write_packet_TCP_client_secure_connection(TCP_Client_Connection *con, const uint8_t *data, uint16_t length,
|
|
bool priority)
|
|
{
|
|
if (length + CRYPTO_MAC_SIZE > MAX_PACKET_SIZE) {
|
|
return -1;
|
|
}
|
|
|
|
bool sendpriority = 1;
|
|
|
|
if (client_send_pending_data(con) == -1) {
|
|
if (priority) {
|
|
sendpriority = 0;
|
|
} else {
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
VLA(uint8_t, packet, sizeof(uint16_t) + length + CRYPTO_MAC_SIZE);
|
|
|
|
uint16_t c_length = net_htons(length + CRYPTO_MAC_SIZE);
|
|
memcpy(packet, &c_length, sizeof(uint16_t));
|
|
int len = encrypt_data_symmetric(con->shared_key, con->sent_nonce, data, length, packet + sizeof(uint16_t));
|
|
|
|
if ((unsigned int)len != (SIZEOF_VLA(packet) - sizeof(uint16_t))) {
|
|
return -1;
|
|
}
|
|
|
|
if (priority) {
|
|
len = sendpriority ? net_send(con->sock, packet, SIZEOF_VLA(packet)) : 0;
|
|
|
|
if (len <= 0) {
|
|
len = 0;
|
|
}
|
|
|
|
increment_nonce(con->sent_nonce);
|
|
|
|
if ((unsigned int)len == SIZEOF_VLA(packet)) {
|
|
return 1;
|
|
}
|
|
|
|
return client_add_priority(con, packet, SIZEOF_VLA(packet), len);
|
|
}
|
|
|
|
len = net_send(con->sock, packet, SIZEOF_VLA(packet));
|
|
|
|
if (len <= 0) {
|
|
return 0;
|
|
}
|
|
|
|
increment_nonce(con->sent_nonce);
|
|
|
|
if ((unsigned int)len == SIZEOF_VLA(packet)) {
|
|
return 1;
|
|
}
|
|
|
|
memcpy(con->last_packet, packet, SIZEOF_VLA(packet));
|
|
con->last_packet_length = SIZEOF_VLA(packet);
|
|
con->last_packet_sent = len;
|
|
return 1;
|
|
}
|
|
|
|
/* return 1 on success.
|
|
* return 0 if could not send packet.
|
|
* return -1 on failure (connection must be killed).
|
|
*/
|
|
int send_routing_request(TCP_Client_Connection *con, uint8_t *public_key)
|
|
{
|
|
uint8_t packet[1 + CRYPTO_PUBLIC_KEY_SIZE];
|
|
packet[0] = TCP_PACKET_ROUTING_REQUEST;
|
|
memcpy(packet + 1, public_key, CRYPTO_PUBLIC_KEY_SIZE);
|
|
return write_packet_TCP_client_secure_connection(con, packet, sizeof(packet), 1);
|
|
}
|
|
|
|
void routing_response_handler(TCP_Client_Connection *con, tcp_routing_response_cb *response_callback, void *object)
|
|
{
|
|
con->response_callback = response_callback;
|
|
con->response_callback_object = object;
|
|
}
|
|
|
|
void routing_status_handler(TCP_Client_Connection *con, tcp_routing_status_cb *status_callback, void *object)
|
|
{
|
|
con->status_callback = status_callback;
|
|
con->status_callback_object = object;
|
|
}
|
|
|
|
static int tcp_send_ping_response(TCP_Client_Connection *con);
|
|
static int tcp_send_ping_request(TCP_Client_Connection *con);
|
|
|
|
/* return 1 on success.
|
|
* return 0 if could not send packet.
|
|
* return -1 on failure.
|
|
*/
|
|
int send_data(TCP_Client_Connection *con, uint8_t con_id, const uint8_t *data, uint16_t length)
|
|
{
|
|
if (con_id >= NUM_CLIENT_CONNECTIONS) {
|
|
return -1;
|
|
}
|
|
|
|
if (con->connections[con_id].status != 2) {
|
|
return -1;
|
|
}
|
|
|
|
if (tcp_send_ping_response(con) == 0 || tcp_send_ping_request(con) == 0) {
|
|
return 0;
|
|
}
|
|
|
|
VLA(uint8_t, packet, 1 + length);
|
|
packet[0] = con_id + NUM_RESERVED_PORTS;
|
|
memcpy(packet + 1, data, length);
|
|
return write_packet_TCP_client_secure_connection(con, packet, SIZEOF_VLA(packet), 0);
|
|
}
|
|
|
|
/* return 1 on success.
|
|
* return 0 if could not send packet.
|
|
* return -1 on failure.
|
|
*/
|
|
int send_oob_packet(TCP_Client_Connection *con, const uint8_t *public_key, const uint8_t *data, uint16_t length)
|
|
{
|
|
if (length == 0 || length > TCP_MAX_OOB_DATA_LENGTH) {
|
|
return -1;
|
|
}
|
|
|
|
VLA(uint8_t, packet, 1 + CRYPTO_PUBLIC_KEY_SIZE + length);
|
|
packet[0] = TCP_PACKET_OOB_SEND;
|
|
memcpy(packet + 1, public_key, CRYPTO_PUBLIC_KEY_SIZE);
|
|
memcpy(packet + 1 + CRYPTO_PUBLIC_KEY_SIZE, data, length);
|
|
return write_packet_TCP_client_secure_connection(con, packet, SIZEOF_VLA(packet), 0);
|
|
}
|
|
|
|
|
|
/* Set the number that will be used as an argument in the callbacks related to con_id.
|
|
*
|
|
* When not set by this function, the number is -1.
|
|
*
|
|
* return 0 on success.
|
|
* return -1 on failure.
|
|
*/
|
|
int set_tcp_connection_number(TCP_Client_Connection *con, uint8_t con_id, uint32_t number)
|
|
{
|
|
if (con_id >= NUM_CLIENT_CONNECTIONS) {
|
|
return -1;
|
|
}
|
|
|
|
if (con->connections[con_id].status == 0) {
|
|
return -1;
|
|
}
|
|
|
|
con->connections[con_id].number = number;
|
|
return 0;
|
|
}
|
|
|
|
void routing_data_handler(TCP_Client_Connection *con, tcp_routing_data_cb *data_callback, void *object)
|
|
{
|
|
con->data_callback = data_callback;
|
|
con->data_callback_object = object;
|
|
}
|
|
|
|
void oob_data_handler(TCP_Client_Connection *con, tcp_oob_data_cb *oob_data_callback, void *object)
|
|
{
|
|
con->oob_data_callback = oob_data_callback;
|
|
con->oob_data_callback_object = object;
|
|
}
|
|
|
|
/* return 1 on success.
|
|
* return 0 if could not send packet.
|
|
* return -1 on failure (connection must be killed).
|
|
*/
|
|
static int client_send_disconnect_notification(TCP_Client_Connection *con, uint8_t id)
|
|
{
|
|
uint8_t packet[1 + 1];
|
|
packet[0] = TCP_PACKET_DISCONNECT_NOTIFICATION;
|
|
packet[1] = id;
|
|
return write_packet_TCP_client_secure_connection(con, packet, sizeof(packet), 1);
|
|
}
|
|
|
|
/* return 1 on success.
|
|
* return 0 if could not send packet.
|
|
* return -1 on failure (connection must be killed).
|
|
*/
|
|
static int tcp_send_ping_request(TCP_Client_Connection *con)
|
|
{
|
|
if (!con->ping_request_id) {
|
|
return 1;
|
|
}
|
|
|
|
uint8_t packet[1 + sizeof(uint64_t)];
|
|
packet[0] = TCP_PACKET_PING;
|
|
memcpy(packet + 1, &con->ping_request_id, sizeof(uint64_t));
|
|
const int ret = write_packet_TCP_client_secure_connection(con, packet, sizeof(packet), 1);
|
|
|
|
if (ret == 1) {
|
|
con->ping_request_id = 0;
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
/* return 1 on success.
|
|
* return 0 if could not send packet.
|
|
* return -1 on failure (connection must be killed).
|
|
*/
|
|
static int tcp_send_ping_response(TCP_Client_Connection *con)
|
|
{
|
|
if (!con->ping_response_id) {
|
|
return 1;
|
|
}
|
|
|
|
uint8_t packet[1 + sizeof(uint64_t)];
|
|
packet[0] = TCP_PACKET_PONG;
|
|
memcpy(packet + 1, &con->ping_response_id, sizeof(uint64_t));
|
|
const int ret = write_packet_TCP_client_secure_connection(con, packet, sizeof(packet), 1);
|
|
|
|
if (ret == 1) {
|
|
con->ping_response_id = 0;
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
/* return 1 on success.
|
|
* return 0 if could not send packet.
|
|
* return -1 on failure (connection must be killed).
|
|
*/
|
|
int send_disconnect_request(TCP_Client_Connection *con, uint8_t con_id)
|
|
{
|
|
if (con_id >= NUM_CLIENT_CONNECTIONS) {
|
|
return -1;
|
|
}
|
|
|
|
con->connections[con_id].status = 0;
|
|
con->connections[con_id].number = 0;
|
|
return client_send_disconnect_notification(con, con_id + NUM_RESERVED_PORTS);
|
|
}
|
|
|
|
/* return 1 on success.
|
|
* return 0 if could not send packet.
|
|
* return -1 on failure (connection must be killed).
|
|
*/
|
|
int send_onion_request(TCP_Client_Connection *con, const uint8_t *data, uint16_t length)
|
|
{
|
|
VLA(uint8_t, packet, 1 + length);
|
|
packet[0] = TCP_PACKET_ONION_REQUEST;
|
|
memcpy(packet + 1, data, length);
|
|
return write_packet_TCP_client_secure_connection(con, packet, SIZEOF_VLA(packet), 0);
|
|
}
|
|
|
|
void onion_response_handler(TCP_Client_Connection *con, tcp_onion_response_cb *onion_callback, void *object)
|
|
{
|
|
con->onion_callback = onion_callback;
|
|
con->onion_callback_object = object;
|
|
}
|
|
|
|
/* Create new TCP connection to ip_port/public_key
|
|
*/
|
|
TCP_Client_Connection *new_TCP_connection(const Mono_Time *mono_time, IP_Port ip_port, const uint8_t *public_key,
|
|
const uint8_t *self_public_key, const uint8_t *self_secret_key, TCP_Proxy_Info *proxy_info)
|
|
{
|
|
if (networking_at_startup() != 0) {
|
|
return nullptr;
|
|
}
|
|
|
|
if (!net_family_is_ipv4(ip_port.ip.family) && !net_family_is_ipv6(ip_port.ip.family)) {
|
|
return nullptr;
|
|
}
|
|
|
|
TCP_Proxy_Info default_proxyinfo;
|
|
|
|
if (proxy_info == nullptr) {
|
|
default_proxyinfo.proxy_type = TCP_PROXY_NONE;
|
|
proxy_info = &default_proxyinfo;
|
|
}
|
|
|
|
Family family = ip_port.ip.family;
|
|
|
|
if (proxy_info->proxy_type != TCP_PROXY_NONE) {
|
|
family = proxy_info->ip_port.ip.family;
|
|
}
|
|
|
|
Socket sock = net_socket(family, TOX_SOCK_STREAM, TOX_PROTO_TCP);
|
|
|
|
if (!sock_valid(sock)) {
|
|
return nullptr;
|
|
}
|
|
|
|
if (!set_socket_nosigpipe(sock)) {
|
|
kill_sock(sock);
|
|
return nullptr;
|
|
}
|
|
|
|
if (!(set_socket_nonblock(sock) && connect_sock_to(sock, ip_port, proxy_info))) {
|
|
kill_sock(sock);
|
|
return nullptr;
|
|
}
|
|
|
|
TCP_Client_Connection *temp = (TCP_Client_Connection *)calloc(sizeof(TCP_Client_Connection), 1);
|
|
|
|
if (temp == nullptr) {
|
|
kill_sock(sock);
|
|
return nullptr;
|
|
}
|
|
|
|
temp->sock = sock;
|
|
memcpy(temp->public_key, public_key, CRYPTO_PUBLIC_KEY_SIZE);
|
|
memcpy(temp->self_public_key, self_public_key, CRYPTO_PUBLIC_KEY_SIZE);
|
|
encrypt_precompute(temp->public_key, self_secret_key, temp->shared_key);
|
|
temp->ip_port = ip_port;
|
|
temp->proxy_info = *proxy_info;
|
|
|
|
switch (proxy_info->proxy_type) {
|
|
case TCP_PROXY_HTTP:
|
|
temp->status = TCP_CLIENT_PROXY_HTTP_CONNECTING;
|
|
proxy_http_generate_connection_request(temp);
|
|
break;
|
|
|
|
case TCP_PROXY_SOCKS5:
|
|
temp->status = TCP_CLIENT_PROXY_SOCKS5_CONNECTING;
|
|
proxy_socks5_generate_handshake(temp);
|
|
break;
|
|
|
|
case TCP_PROXY_NONE:
|
|
temp->status = TCP_CLIENT_CONNECTING;
|
|
|
|
if (generate_handshake(temp) == -1) {
|
|
kill_sock(sock);
|
|
free(temp);
|
|
return nullptr;
|
|
}
|
|
|
|
break;
|
|
}
|
|
|
|
temp->kill_at = mono_time_get(mono_time) + TCP_CONNECTION_TIMEOUT;
|
|
|
|
return temp;
|
|
}
|
|
|
|
/* return 0 on success
|
|
* return -1 on failure
|
|
*/
|
|
static int handle_TCP_client_packet(TCP_Client_Connection *conn, const uint8_t *data, uint16_t length, void *userdata)
|
|
{
|
|
if (length <= 1) {
|
|
return -1;
|
|
}
|
|
|
|
switch (data[0]) {
|
|
case TCP_PACKET_ROUTING_RESPONSE: {
|
|
if (length != 1 + 1 + CRYPTO_PUBLIC_KEY_SIZE) {
|
|
return -1;
|
|
}
|
|
|
|
if (data[1] < NUM_RESERVED_PORTS) {
|
|
return 0;
|
|
}
|
|
|
|
uint8_t con_id = data[1] - NUM_RESERVED_PORTS;
|
|
|
|
if (conn->connections[con_id].status != 0) {
|
|
return 0;
|
|
}
|
|
|
|
conn->connections[con_id].status = 1;
|
|
conn->connections[con_id].number = -1;
|
|
memcpy(conn->connections[con_id].public_key, data + 2, CRYPTO_PUBLIC_KEY_SIZE);
|
|
|
|
if (conn->response_callback) {
|
|
conn->response_callback(conn->response_callback_object, con_id, conn->connections[con_id].public_key);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
case TCP_PACKET_CONNECTION_NOTIFICATION: {
|
|
if (length != 1 + 1) {
|
|
return -1;
|
|
}
|
|
|
|
if (data[1] < NUM_RESERVED_PORTS) {
|
|
return -1;
|
|
}
|
|
|
|
uint8_t con_id = data[1] - NUM_RESERVED_PORTS;
|
|
|
|
if (conn->connections[con_id].status != 1) {
|
|
return 0;
|
|
}
|
|
|
|
conn->connections[con_id].status = 2;
|
|
|
|
if (conn->status_callback) {
|
|
conn->status_callback(conn->status_callback_object, conn->connections[con_id].number, con_id,
|
|
conn->connections[con_id].status);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
case TCP_PACKET_DISCONNECT_NOTIFICATION: {
|
|
if (length != 1 + 1) {
|
|
return -1;
|
|
}
|
|
|
|
if (data[1] < NUM_RESERVED_PORTS) {
|
|
return -1;
|
|
}
|
|
|
|
uint8_t con_id = data[1] - NUM_RESERVED_PORTS;
|
|
|
|
if (conn->connections[con_id].status == 0) {
|
|
return 0;
|
|
}
|
|
|
|
if (conn->connections[con_id].status != 2) {
|
|
return 0;
|
|
}
|
|
|
|
conn->connections[con_id].status = 1;
|
|
|
|
if (conn->status_callback) {
|
|
conn->status_callback(conn->status_callback_object, conn->connections[con_id].number, con_id,
|
|
conn->connections[con_id].status);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
case TCP_PACKET_PING: {
|
|
if (length != 1 + sizeof(uint64_t)) {
|
|
return -1;
|
|
}
|
|
|
|
uint64_t ping_id;
|
|
memcpy(&ping_id, data + 1, sizeof(uint64_t));
|
|
conn->ping_response_id = ping_id;
|
|
tcp_send_ping_response(conn);
|
|
return 0;
|
|
}
|
|
|
|
case TCP_PACKET_PONG: {
|
|
if (length != 1 + sizeof(uint64_t)) {
|
|
return -1;
|
|
}
|
|
|
|
uint64_t ping_id;
|
|
memcpy(&ping_id, data + 1, sizeof(uint64_t));
|
|
|
|
if (ping_id) {
|
|
if (ping_id == conn->ping_id) {
|
|
conn->ping_id = 0;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
return -1;
|
|
}
|
|
|
|
case TCP_PACKET_OOB_RECV: {
|
|
if (length <= 1 + CRYPTO_PUBLIC_KEY_SIZE) {
|
|
return -1;
|
|
}
|
|
|
|
if (conn->oob_data_callback) {
|
|
conn->oob_data_callback(conn->oob_data_callback_object, data + 1, data + 1 + CRYPTO_PUBLIC_KEY_SIZE,
|
|
length - (1 + CRYPTO_PUBLIC_KEY_SIZE), userdata);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
case TCP_PACKET_ONION_RESPONSE: {
|
|
conn->onion_callback(conn->onion_callback_object, data + 1, length - 1, userdata);
|
|
return 0;
|
|
}
|
|
|
|
default: {
|
|
if (data[0] < NUM_RESERVED_PORTS) {
|
|
return -1;
|
|
}
|
|
|
|
uint8_t con_id = data[0] - NUM_RESERVED_PORTS;
|
|
|
|
if (conn->data_callback) {
|
|
conn->data_callback(conn->data_callback_object, conn->connections[con_id].number, con_id, data + 1, length - 1,
|
|
userdata);
|
|
}
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static bool tcp_process_packet(TCP_Client_Connection *conn, void *userdata)
|
|
{
|
|
uint8_t packet[MAX_PACKET_SIZE];
|
|
const int len = read_packet_TCP_secure_connection(conn->sock, &conn->next_packet_length, conn->shared_key,
|
|
conn->recv_nonce, packet, sizeof(packet));
|
|
|
|
if (len == 0) {
|
|
return false;
|
|
}
|
|
|
|
if (len == -1) {
|
|
conn->status = TCP_CLIENT_DISCONNECTED;
|
|
return false;
|
|
}
|
|
|
|
if (handle_TCP_client_packet(conn, packet, len, userdata) == -1) {
|
|
conn->status = TCP_CLIENT_DISCONNECTED;
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
static int do_confirmed_TCP(TCP_Client_Connection *conn, const Mono_Time *mono_time, void *userdata)
|
|
{
|
|
client_send_pending_data(conn);
|
|
tcp_send_ping_response(conn);
|
|
tcp_send_ping_request(conn);
|
|
|
|
if (mono_time_is_timeout(mono_time, conn->last_pinged, TCP_PING_FREQUENCY)) {
|
|
uint64_t ping_id = random_u64();
|
|
|
|
if (!ping_id) {
|
|
++ping_id;
|
|
}
|
|
|
|
conn->ping_request_id = ping_id;
|
|
conn->ping_id = ping_id;
|
|
tcp_send_ping_request(conn);
|
|
conn->last_pinged = mono_time_get(mono_time);
|
|
}
|
|
|
|
if (conn->ping_id && mono_time_is_timeout(mono_time, conn->last_pinged, TCP_PING_TIMEOUT)) {
|
|
conn->status = TCP_CLIENT_DISCONNECTED;
|
|
return 0;
|
|
}
|
|
|
|
while (tcp_process_packet(conn, userdata)) {
|
|
// Keep reading until error or out of data.
|
|
continue;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
/* Run the TCP connection
|
|
*/
|
|
void do_TCP_connection(Mono_Time *mono_time, TCP_Client_Connection *tcp_connection, void *userdata)
|
|
{
|
|
if (tcp_connection->status == TCP_CLIENT_DISCONNECTED) {
|
|
return;
|
|
}
|
|
|
|
if (tcp_connection->status == TCP_CLIENT_PROXY_HTTP_CONNECTING) {
|
|
if (client_send_pending_data(tcp_connection) == 0) {
|
|
int ret = proxy_http_read_connection_response(tcp_connection);
|
|
|
|
if (ret == -1) {
|
|
tcp_connection->kill_at = 0;
|
|
tcp_connection->status = TCP_CLIENT_DISCONNECTED;
|
|
}
|
|
|
|
if (ret == 1) {
|
|
generate_handshake(tcp_connection);
|
|
tcp_connection->status = TCP_CLIENT_CONNECTING;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (tcp_connection->status == TCP_CLIENT_PROXY_SOCKS5_CONNECTING) {
|
|
if (client_send_pending_data(tcp_connection) == 0) {
|
|
int ret = socks5_read_handshake_response(tcp_connection);
|
|
|
|
if (ret == -1) {
|
|
tcp_connection->kill_at = 0;
|
|
tcp_connection->status = TCP_CLIENT_DISCONNECTED;
|
|
}
|
|
|
|
if (ret == 1) {
|
|
proxy_socks5_generate_connection_request(tcp_connection);
|
|
tcp_connection->status = TCP_CLIENT_PROXY_SOCKS5_UNCONFIRMED;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (tcp_connection->status == TCP_CLIENT_PROXY_SOCKS5_UNCONFIRMED) {
|
|
if (client_send_pending_data(tcp_connection) == 0) {
|
|
int ret = proxy_socks5_read_connection_response(tcp_connection);
|
|
|
|
if (ret == -1) {
|
|
tcp_connection->kill_at = 0;
|
|
tcp_connection->status = TCP_CLIENT_DISCONNECTED;
|
|
}
|
|
|
|
if (ret == 1) {
|
|
generate_handshake(tcp_connection);
|
|
tcp_connection->status = TCP_CLIENT_CONNECTING;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (tcp_connection->status == TCP_CLIENT_CONNECTING) {
|
|
if (client_send_pending_data(tcp_connection) == 0) {
|
|
tcp_connection->status = TCP_CLIENT_UNCONFIRMED;
|
|
}
|
|
}
|
|
|
|
if (tcp_connection->status == TCP_CLIENT_UNCONFIRMED) {
|
|
uint8_t data[TCP_SERVER_HANDSHAKE_SIZE];
|
|
int len = read_TCP_packet(tcp_connection->sock, data, sizeof(data));
|
|
|
|
if (sizeof(data) == len) {
|
|
if (handle_handshake(tcp_connection, data) == 0) {
|
|
tcp_connection->kill_at = -1;
|
|
tcp_connection->status = TCP_CLIENT_CONFIRMED;
|
|
} else {
|
|
tcp_connection->kill_at = 0;
|
|
tcp_connection->status = TCP_CLIENT_DISCONNECTED;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (tcp_connection->status == TCP_CLIENT_CONFIRMED) {
|
|
do_confirmed_TCP(tcp_connection, mono_time, userdata);
|
|
}
|
|
|
|
if (tcp_connection->kill_at <= mono_time_get(mono_time)) {
|
|
tcp_connection->status = TCP_CLIENT_DISCONNECTED;
|
|
}
|
|
}
|
|
|
|
/* Kill the TCP connection
|
|
*/
|
|
void kill_TCP_connection(TCP_Client_Connection *tcp_connection)
|
|
{
|
|
if (tcp_connection == nullptr) {
|
|
return;
|
|
}
|
|
|
|
wipe_priority_list(tcp_connection->priority_queue_start);
|
|
kill_sock(tcp_connection->sock);
|
|
crypto_memzero(tcp_connection, sizeof(TCP_Client_Connection));
|
|
free(tcp_connection);
|
|
}
|