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 Net_Profile * tcp_np ;
400404 Onion * onion ;
401405 Onion_Announce * onion_a ;
@@ -449,7 +453,7 @@ static Onions *new_onions(const Memory *mem, const Random *rng, uint16_t port, u
449453 return nullptr ;
450454 }
451455
452- on -> onion = new_onion (on -> log , mem , on -> mono_time , rng , dht );
456+ on -> onion = new_onion (on -> log , mem , on -> mono_time , rng , dht , net );
453457
454458 if (!on -> onion ) {
455459 kill_dht (dht );
@@ -460,7 +464,7 @@ static Onions *new_onions(const Memory *mem, const Random *rng, uint16_t port, u
460464 return nullptr ;
461465 }
462466
463- on -> onion_a = new_onion_announce (on -> log , mem , rng , on -> mono_time , dht );
467+ on -> onion_a = new_onion_announce (on -> log , mem , rng , on -> mono_time , dht , net );
464468
465469 if (!on -> onion_a ) {
466470 kill_onion (on -> onion );
@@ -486,7 +490,8 @@ static Onions *new_onions(const Memory *mem, const Random *rng, uint16_t port, u
486490 }
487491
488492 TCP_Proxy_Info inf = {{{{0 }}}};
489- 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 , on -> tcp_np ));
493+ on -> nc = new_net_crypto (on -> log , mem , rng , ns , on -> mono_time , net , dht , & inf , on -> tcp_np );
494+ on -> onion_c = new_onion_client (on -> log , mem , rng , on -> mono_time , on -> nc , dht , net );
490495
491496 if (!on -> onion_c ) {
492497 netprof_kill (mem , on -> tcp_np );
@@ -514,9 +519,9 @@ static void do_onions(Onions *on)
514519
515520static void kill_onions (const Memory * mem , Onions * on )
516521{
517- Networking_Core * net = dht_get_net ( on -> onion -> dht ) ;
522+ Networking_Core * net = on -> onion -> net ;
518523 DHT * dht = on -> onion -> dht ;
519- Net_Crypto * c = onion_get_net_crypto ( on -> onion_c ) ;
524+ Net_Crypto * c = on -> nc ;
520525 kill_onion_client (on -> onion_c );
521526 kill_onion_announce (on -> onion_a );
522527 kill_onion (on -> onion );
@@ -640,9 +645,9 @@ static void test_announce(void)
640645
641646 printf ("adding friend\n" );
642647 int frnum_f = onion_addfriend (onions [NUM_FIRST ]-> onion_c ,
643- nc_get_self_public_key (onion_get_net_crypto ( onions [NUM_LAST ]-> onion_c ) ));
648+ nc_get_self_public_key (onions [NUM_LAST ]-> nc ));
644649 int frnum = onion_addfriend (onions [NUM_LAST ]-> onion_c ,
645- nc_get_self_public_key (onion_get_net_crypto ( onions [NUM_FIRST ]-> onion_c ) ));
650+ nc_get_self_public_key (onions [NUM_FIRST ]-> nc ));
646651
647652 onion_dht_pk_callback (onions [NUM_FIRST ]-> onion_c , frnum_f , & dht_pk_callback , onions [NUM_FIRST ], NUM_FIRST );
648653 onion_dht_pk_callback (onions [NUM_LAST ]-> onion_c , frnum , & dht_pk_callback , onions [NUM_LAST ], NUM_LAST );
0 commit comments