Code cleanups.

Moved some functions to onion.c.

Fixed possible portability issues.

1 files changed, 52 insertions, 20 deletions

diff --git a/toxcore/onion.c b/toxcore/onion.c
index b444e02a..05dbaae6 100644
--- a/toxcore/onion.c
+++ b/toxcore/onion.c
@@ -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);