mirror of
https://github.com/irungentoo/toxcore.git
synced 2024-03-22 13:30:51 +08:00
Code cleanups.
Moved some functions to onion.c. Fixed possible portability issues.
This commit is contained in:
parent
3328127f18
commit
93a998d15a
|
@ -43,6 +43,7 @@ int crypto_cmp(const uint8_t *mem1, const uint8_t *mem2, size_t length)
|
|||
|
||||
unsigned int check = 0;
|
||||
size_t i;
|
||||
|
||||
for (i = 0; i < length; ++i) {
|
||||
check |= mem1[i] ^ mem2[i];
|
||||
}
|
||||
|
|
|
@ -756,41 +756,6 @@ void ipport_copy(IP_Port *target, const IP_Port *source)
|
|||
memcpy(target, source, sizeof(IP_Port));
|
||||
};
|
||||
|
||||
/* packing and unpacking functions */
|
||||
void ip_pack(uint8_t *data, const IP *source)
|
||||
{
|
||||
data[0] = source->family;
|
||||
memcpy(data + 1, &source->ip6, SIZE_IP6);
|
||||
}
|
||||
|
||||
/* return 0 on success, -1 on failure. */
|
||||
int ip_unpack(IP *target, const uint8_t *data, unsigned int data_size)
|
||||
{
|
||||
if (data_size < (1 + SIZE_IP6))
|
||||
return -1;
|
||||
|
||||
target->family = data[0];
|
||||
memcpy(&target->ip6, data + 1, SIZE_IP6);
|
||||
return 0;
|
||||
}
|
||||
|
||||
void ipport_pack(uint8_t *data, const IP_Port *source)
|
||||
{
|
||||
ip_pack(data, &source->ip);
|
||||
memcpy(data + SIZE_IP, &source->port, SIZE_PORT);
|
||||
}
|
||||
|
||||
/* return 0 on success, -1 on failure. */
|
||||
int ipport_unpack(IP_Port *target, const uint8_t *data, unsigned int data_size)
|
||||
{
|
||||
if (data_size < (SIZE_IP + SIZE_PORT))
|
||||
return -1;
|
||||
|
||||
ip_unpack(&target->ip, data, data_size);
|
||||
memcpy(&target->port, data + SIZE_IP, SIZE_PORT);
|
||||
return 0;
|
||||
}
|
||||
|
||||
/* ip_ntoa
|
||||
* converts ip into a string
|
||||
* uses a static buffer, so mustn't used multiple times in the same output
|
||||
|
|
|
@ -246,18 +246,6 @@ void ip_copy(IP *target, const IP *source);
|
|||
/* copies an ip_port structure */
|
||||
void ipport_copy(IP_Port *target, const IP_Port *source);
|
||||
|
||||
|
||||
/* packs IP into data, writes SIZE_IP bytes to data */
|
||||
void ip_pack(uint8_t *data, const IP *source);
|
||||
/* unpacks IP from data, reads SIZE_IP bytes from data
|
||||
return 0 on success, -1 on failure. */
|
||||
int ip_unpack(IP *target, const uint8_t *data, unsigned int data_size);
|
||||
/* packs IP_Port into data, writes SIZE_IPPORT bytes to data */
|
||||
void ipport_pack(uint8_t *data, const IP_Port *source);
|
||||
/* unpacks IP_Port from data, reads SIZE_IPPORT bytes to data
|
||||
return 0 on success, -1 on failure. */
|
||||
int ipport_unpack(IP_Port *target, const uint8_t *data, unsigned int data_size);
|
||||
|
||||
/*
|
||||
* addr_resolve():
|
||||
* uses getaddrinfo to resolve an address into an IP address
|
||||
|
|
|
@ -45,6 +45,58 @@ static void change_symmetric_key(Onion *onion)
|
|||
}
|
||||
}
|
||||
|
||||
/* packing and unpacking functions */
|
||||
static void ip_pack(uint8_t *data, IP source)
|
||||
{
|
||||
to_net_family(&source);
|
||||
|
||||
data[0] = source.family;
|
||||
|
||||
if (source.family == TOX_AF_INET || source.family == TOX_TCP_INET) {
|
||||
memset(data + 1, 0, SIZE_IP6);
|
||||
memcpy(data + 1, source.ip4.uint8, SIZE_IP4);
|
||||
} else {
|
||||
memcpy(data + 1, source.ip6.uint8, SIZE_IP6);
|
||||
}
|
||||
}
|
||||
|
||||
/* return 0 on success, -1 on failure. */
|
||||
static int ip_unpack(IP *target, const uint8_t *data, unsigned int data_size)
|
||||
{
|
||||
if (data_size < (1 + SIZE_IP6))
|
||||
return -1;
|
||||
|
||||
target->family = data[0];
|
||||
|
||||
if (target->family == TOX_AF_INET || target->family == TOX_TCP_INET) {
|
||||
memcpy(target->ip4.uint8, data + 1, SIZE_IP4);
|
||||
} else {
|
||||
memcpy(target->ip6.uint8, data + 1, SIZE_IP6);
|
||||
}
|
||||
|
||||
return to_host_family(target);
|
||||
}
|
||||
|
||||
static void ipport_pack(uint8_t *data, const IP_Port *source)
|
||||
{
|
||||
ip_pack(data, source->ip);
|
||||
memcpy(data + SIZE_IP, &source->port, SIZE_PORT);
|
||||
}
|
||||
|
||||
/* return 0 on success, -1 on failure. */
|
||||
static int ipport_unpack(IP_Port *target, const uint8_t *data, unsigned int data_size)
|
||||
{
|
||||
if (data_size < (SIZE_IP + SIZE_PORT))
|
||||
return -1;
|
||||
|
||||
if (ip_unpack(&target->ip, data, data_size) == -1)
|
||||
return -1;
|
||||
|
||||
memcpy(&target->port, data + SIZE_IP, SIZE_PORT);
|
||||
return 0;
|
||||
}
|
||||
|
||||
|
||||
/* Create a new onion path.
|
||||
*
|
||||
* Create a new onion path out of nodes (nodes is a list of 3 nodes)
|
||||
|
@ -81,10 +133,6 @@ int create_onion_path(const DHT *dht, Onion_Path *new_path, const Node_format *n
|
|||
memcpy(new_path->node_public_key2, nodes[1].client_id, crypto_box_PUBLICKEYBYTES);
|
||||
memcpy(new_path->node_public_key3, nodes[2].client_id, crypto_box_PUBLICKEYBYTES);
|
||||
|
||||
/* to_net_family(&new_path->ip_port1.ip); */
|
||||
to_net_family(&new_path->ip_port2.ip);
|
||||
to_net_family(&new_path->ip_port3.ip);
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
|
@ -102,9 +150,6 @@ int onion_path_to_nodes(Node_format *nodes, unsigned int num_nodes, const Onion_
|
|||
nodes[1].ip_port = path->ip_port2;
|
||||
nodes[2].ip_port = path->ip_port3;
|
||||
|
||||
to_host_family(&nodes[1].ip_port.ip);
|
||||
to_host_family(&nodes[2].ip_port.ip);
|
||||
|
||||
memcpy(nodes[0].client_id, path->node_public_key1, crypto_box_PUBLICKEYBYTES);
|
||||
memcpy(nodes[1].client_id, path->node_public_key2, crypto_box_PUBLICKEYBYTES);
|
||||
memcpy(nodes[2].client_id, path->node_public_key3, crypto_box_PUBLICKEYBYTES);
|
||||
|
@ -126,10 +171,8 @@ int create_onion_packet(uint8_t *packet, uint16_t max_packet_length, const Onion
|
|||
if (1 + length + SEND_1 > max_packet_length || length == 0)
|
||||
return -1;
|
||||
|
||||
to_net_family(&dest.ip);
|
||||
uint8_t step1[SIZE_IPPORT + length];
|
||||
|
||||
|
||||
ipport_pack(step1, &dest);
|
||||
memcpy(step1 + SIZE_IPPORT, data, length);
|
||||
|
||||
|
@ -183,10 +226,8 @@ int create_onion_packet_tcp(uint8_t *packet, uint16_t max_packet_length, const O
|
|||
if (crypto_box_NONCEBYTES + SIZE_IPPORT + SEND_BASE * 2 + length > max_packet_length || length == 0)
|
||||
return -1;
|
||||
|
||||
to_net_family(&dest.ip);
|
||||
uint8_t step1[SIZE_IPPORT + length];
|
||||
|
||||
|
||||
ipport_pack(step1, &dest);
|
||||
memcpy(step1 + SIZE_IPPORT, data, length);
|
||||
|
||||
|
@ -297,9 +338,6 @@ int onion_send_1(const Onion *onion, const uint8_t *plain, uint16_t len, IP_Port
|
|||
if (ipport_unpack(&send_to, plain, len) == -1)
|
||||
return 1;
|
||||
|
||||
if (to_host_family(&send_to.ip) == -1)
|
||||
return 1;
|
||||
|
||||
uint8_t ip_port[SIZE_IPPORT];
|
||||
ipport_pack(ip_port, &source);
|
||||
|
||||
|
@ -350,9 +388,6 @@ static int handle_send_1(void *object, IP_Port source, const uint8_t *packet, ui
|
|||
if (ipport_unpack(&send_to, plain, len) == -1)
|
||||
return 1;
|
||||
|
||||
if (to_host_family(&send_to.ip) == -1)
|
||||
return 1;
|
||||
|
||||
uint8_t data[ONION_MAX_PACKET_SIZE];
|
||||
data[0] = NET_PACKET_ONION_SEND_2;
|
||||
memcpy(data + 1, packet + 1, crypto_box_NONCEBYTES);
|
||||
|
@ -403,9 +438,6 @@ static int handle_send_2(void *object, IP_Port source, const uint8_t *packet, ui
|
|||
if (ipport_unpack(&send_to, plain, len) == -1)
|
||||
return 1;
|
||||
|
||||
if (to_host_family(&send_to.ip) == -1)
|
||||
return 1;
|
||||
|
||||
uint8_t data[ONION_MAX_PACKET_SIZE];
|
||||
memcpy(data, plain + SIZE_IPPORT, len - SIZE_IPPORT);
|
||||
uint16_t data_len = (len - SIZE_IPPORT);
|
||||
|
|
Loading…
Reference in New Issue
Block a user