33
44#include "../testing/misc_tools.h"
55#include "../toxcore/mono_time.h"
6+ #include "../toxcore/network.h"
67#include "../toxcore/onion.h"
78#include "../toxcore/onion_announce.h"
89#include "../toxcore/onion_client.h"
9- #include "../toxcore/util.h"
1010#include "auto_test_support.h"
1111#include "check_compat.h"
1212
@@ -237,8 +237,10 @@ static void test_basic(void)
237237 Mono_Time * mono_time2 = mono_time_new (mem , nullptr , nullptr );
238238
239239 IP ip = get_loopback ();
240- Onion * onion1 = new_onion (log1 , mem , mono_time1 , rng , new_dht (log1 , mem , rng , ns , mono_time1 , new_networking (log1 , mem , ns , & ip , 36567 ), true, false));
241- Onion * onion2 = new_onion (log2 , mem , mono_time2 , rng , new_dht (log2 , mem , rng , ns , mono_time2 , new_networking (log2 , mem , ns , & ip , 36568 ), true, false));
240+ Networking_Core * net1 = new_networking (log1 , mem , ns , & ip , 36567 );
241+ Onion * onion1 = new_onion (log1 , mem , mono_time1 , rng , new_dht (log1 , mem , rng , ns , mono_time1 , net1 , true, false), net1 );
242+ Networking_Core * net2 = new_networking (log2 , mem , ns , & ip , 36568 );
243+ Onion * onion2 = new_onion (log2 , mem , mono_time2 , rng , new_dht (log2 , mem , rng , ns , mono_time2 , net2 , true, false), net2 );
242244 ck_assert_msg ((onion1 != nullptr ) && (onion2 != nullptr ), "Onion failed initializing." );
243245 networking_registerhandler (onion2 -> net , NET_PACKET_ANNOUNCE_REQUEST , & handle_test_1 , onion2 );
244246
@@ -281,8 +283,8 @@ static void test_basic(void)
281283 do_onion (mono_time2 , onion2 );
282284 } while (handled_test_2 == 0 );
283285
284- Onion_Announce * onion1_a = new_onion_announce (log1 , mem , rng , mono_time1 , onion1 -> dht );
285- Onion_Announce * onion2_a = new_onion_announce (log2 , mem , rng , mono_time2 , onion2 -> dht );
286+ Onion_Announce * onion1_a = new_onion_announce (log1 , mem , rng , mono_time1 , onion1 -> dht , onion1 -> net );
287+ Onion_Announce * onion2_a = new_onion_announce (log2 , mem , rng , mono_time2 , onion2 -> dht , onion2 -> net );
286288 networking_registerhandler (onion1 -> net , NET_PACKET_ANNOUNCE_RESPONSE , & handle_test_3 , onion1 );
287289 networking_registerhandler (onion1 -> net , NET_PACKET_ANNOUNCE_RESPONSE_OLD , & handle_test_3_old , onion1 );
288290 ck_assert_msg ((onion1_a != nullptr ) && (onion2_a != nullptr ), "Onion_Announce failed initializing." );
@@ -334,7 +336,8 @@ static void test_basic(void)
334336
335337 Mono_Time * mono_time3 = mono_time_new (mem , nullptr , nullptr );
336338
337- Onion * onion3 = new_onion (log3 , mem , mono_time3 , rng , new_dht (log3 , mem , rng , ns , mono_time3 , new_networking (log3 , mem , ns , & ip , 36569 ), true, false));
339+ Networking_Core * net3 = new_networking (log3 , mem , ns , & ip , 36569 );
340+ Onion * onion3 = new_onion (log3 , mem , mono_time3 , rng , new_dht (log3 , mem , rng , ns , mono_time3 , net3 , true, false), net3 );
338341 ck_assert_msg ((onion3 != nullptr ), "Onion failed initializing." );
339342
340343 random_nonce (rng , nonce );
@@ -359,7 +362,7 @@ static void test_basic(void)
359362 {
360363 Onion * onion = onion3 ;
361364
362- Networking_Core * net = dht_get_net ( onion -> dht ) ;
365+ Networking_Core * net = onion -> net ;
363366 DHT * dht = onion -> dht ;
364367 kill_onion (onion );
365368 kill_dht (dht );
@@ -371,7 +374,7 @@ static void test_basic(void)
371374 {
372375 Onion * onion = onion2 ;
373376
374- Networking_Core * net = dht_get_net ( onion -> dht ) ;
377+ Networking_Core * net = onion -> net ;
375378 DHT * dht = onion -> dht ;
376379 kill_onion (onion );
377380 kill_dht (dht );
@@ -383,7 +386,7 @@ static void test_basic(void)
383386 {
384387 Onion * onion = onion1 ;
385388
386- Networking_Core * net = dht_get_net ( onion -> dht ) ;
389+ Networking_Core * net = onion -> net ;
387390 DHT * dht = onion -> dht ;
388391 kill_onion (onion );
389392 kill_dht (dht );
@@ -396,6 +399,7 @@ static void test_basic(void)
396399typedef struct {
397400 Logger * log ;
398401 Mono_Time * mono_time ;
402+ Net_Crypto * nc ;
399403 Onion * onion ;
400404 Onion_Announce * onion_a ;
401405 Onion_Client * onion_c ;
@@ -448,7 +452,7 @@ static Onions *new_onions(const Memory *mem, const Random *rng, uint16_t port, u
448452 return nullptr ;
449453 }
450454
451- on -> onion = new_onion (on -> log , mem , on -> mono_time , rng , dht );
455+ on -> onion = new_onion (on -> log , mem , on -> mono_time , rng , dht , net );
452456
453457 if (!on -> onion ) {
454458 kill_dht (dht );
@@ -459,7 +463,7 @@ static Onions *new_onions(const Memory *mem, const Random *rng, uint16_t port, u
459463 return nullptr ;
460464 }
461465
462- on -> onion_a = new_onion_announce (on -> log , mem , rng , on -> mono_time , dht );
466+ on -> onion_a = new_onion_announce (on -> log , mem , rng , on -> mono_time , dht , net );
463467
464468 if (!on -> onion_a ) {
465469 kill_onion (on -> onion );
@@ -472,7 +476,8 @@ static Onions *new_onions(const Memory *mem, const Random *rng, uint16_t port, u
472476 }
473477
474478 TCP_Proxy_Info inf = {{{{0 }}}};
475- on -> onion_c = new_onion_client (on -> log , mem , rng , on -> mono_time , new_net_crypto (on -> log , mem , rng , ns , on -> mono_time , dht , & inf ));
479+ on -> nc = new_net_crypto (on -> log , mem , rng , ns , on -> mono_time , net , dht , & inf );
480+ on -> onion_c = new_onion_client (on -> log , mem , rng , on -> mono_time , on -> nc , dht , net );
476481
477482 if (!on -> onion_c ) {
478483 kill_onion_announce (on -> onion_a );
@@ -499,9 +504,9 @@ static void do_onions(Onions *on)
499504
500505static void kill_onions (const Memory * mem , Onions * on )
501506{
502- Networking_Core * net = dht_get_net ( on -> onion -> dht ) ;
507+ Networking_Core * net = on -> onion -> net ;
503508 DHT * dht = on -> onion -> dht ;
504- Net_Crypto * c = onion_get_net_crypto ( on -> onion_c ) ;
509+ Net_Crypto * c = on -> nc ;
505510 kill_onion_client (on -> onion_c );
506511 kill_onion_announce (on -> onion_a );
507512 kill_onion (on -> onion );
@@ -624,9 +629,9 @@ static void test_announce(void)
624629
625630 printf ("adding friend\n" );
626631 int frnum_f = onion_addfriend (onions [NUM_FIRST ]-> onion_c ,
627- nc_get_self_public_key (onion_get_net_crypto ( onions [NUM_LAST ]-> onion_c ) ));
632+ nc_get_self_public_key (onions [NUM_LAST ]-> nc ));
628633 int frnum = onion_addfriend (onions [NUM_LAST ]-> onion_c ,
629- nc_get_self_public_key (onion_get_net_crypto ( onions [NUM_FIRST ]-> onion_c ) ));
634+ nc_get_self_public_key (onions [NUM_FIRST ]-> nc ));
630635
631636 onion_dht_pk_callback (onions [NUM_FIRST ]-> onion_c , frnum_f , & dht_pk_callback , onions [NUM_FIRST ], NUM_FIRST );
632637 onion_dht_pk_callback (onions [NUM_LAST ]-> onion_c , frnum , & dht_pk_callback , onions [NUM_LAST ], NUM_LAST );
0 commit comments