onion.c now handles another packet type.

1 files changed, 41 insertions, 4 deletions

diff --git a/toxcore/onion.c b/toxcore/onion.c
index 0b0586fd..c4497078 100644
--- a/toxcore/onion.c
+++ b/toxcore/onion.c
@@ -84,13 +84,13 @@ static int handle_send_1(void *object, IP_Port source, uint8_t *packet, uint32_t
 
     if (length <= 1 + SEND_2)
         return 1;
-    
+
     uint8_t plain[MAX_ONION_SIZE];
 
     int len = decrypt_data(packet + 1 + crypto_box_NONCEBYTES, onion->dht->self_secret_key, packet + 1,
                            packet + 1 + crypto_box_NONCEBYTES + crypto_box_PUBLICKEYBYTES,
                            length - (1 + crypto_box_NONCEBYTES + crypto_box_PUBLICKEYBYTES + RETURN_1), plain);
-    
+
     if ((uint32_t)len != length - (1 + crypto_box_NONCEBYTES + crypto_box_PUBLICKEYBYTES + RETURN_1 + crypto_box_MACBYTES))
         return 1;
 
@@ -108,14 +108,15 @@ static int handle_send_1(void *object, IP_Port source, uint8_t *packet, uint32_t
     memcpy(ret_data + sizeof(IP_Port), packet + (length - RETURN_1), RETURN_1);
     len = encrypt_data_symmetric(onion->secret_symmetric_key, ret_part, ret_data, sizeof(ret_data),
                                  ret_part + crypto_secretbox_NONCEBYTES);
-    
+
     if (len != RETURN_2 - crypto_secretbox_NONCEBYTES)
         return 1;
-    
+
     data_len += len;
 
     if ((uint32_t)sendpacket(onion->net, send_to, data, data_len) != data_len)
         return 1;
+
     return 0;
 }
 
@@ -123,6 +124,42 @@ static int handle_send_2(void *object, IP_Port source, uint8_t *packet, uint32_t
 {
     Onion *onion = object;
 
+    if (length > MAX_ONION_SIZE)
+        return 1;
+
+    if (length <= 1 + SEND_3)
+        return 1;
+
+    uint8_t plain[MAX_ONION_SIZE];
+
+    int len = decrypt_data(packet + 1 + crypto_box_NONCEBYTES, onion->dht->self_secret_key, packet + 1,
+                           packet + 1 + crypto_box_NONCEBYTES + crypto_box_PUBLICKEYBYTES,
+                           length - (1 + crypto_box_NONCEBYTES + crypto_box_PUBLICKEYBYTES + RETURN_2), plain);
+
+    if ((uint32_t)len != length - (1 + crypto_box_NONCEBYTES + crypto_box_PUBLICKEYBYTES + RETURN_2 + crypto_box_MACBYTES))
+        return 1;
+
+    IP_Port send_to;
+    memcpy(&send_to, plain, sizeof(IP_Port));
+    uint8_t data[MAX_ONION_SIZE];
+    memcpy(data, plain + sizeof(IP_Port), len - sizeof(IP_Port));
+    uint32_t data_len = (len - sizeof(IP_Port));
+    uint8_t *ret_part = data + (len - sizeof(IP_Port));
+    new_nonce(ret_part);
+    uint8_t ret_data[RETURN_2 + sizeof(IP_Port)];
+    memcpy(ret_data, &source, sizeof(IP_Port));
+    memcpy(ret_data + sizeof(IP_Port), packet + (length - RETURN_2), RETURN_2);
+    len = encrypt_data_symmetric(onion->secret_symmetric_key, ret_part, ret_data, sizeof(ret_data),
+                                 ret_part + crypto_secretbox_NONCEBYTES);
+
+    if (len != RETURN_3 - crypto_secretbox_NONCEBYTES)
+        return 1;
+
+    data_len += RETURN_3;
+
+    if ((uint32_t)sendpacket(onion->net, send_to, data, data_len) != data_len)
+        return 1;
+
     return 0;
 }