jeka
/
utun2
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
Browse Source

tests: allocate STCP traffic and link listener ports automatically

master
evgeny 3 days ago
parent
0dcecdd7b2
commit
98529b3268
4 changed files with 10 additions and 18 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 4
      src/transport_layer/etcp_connections.c
  2. 8
      tests/test_etcp_stcp.c
  3. 9
      tests/test_stcp_link.c
  4. 7
      tests/test_stcp_traffic.c

4
src/transport_layer/etcp_connections.c
Unescape View File

@ -2684,7 +2684,7 @@ int etcp_create_server_socket(struct UTUN_INSTANCE* instance, struct CFG_SERVER*
return -1; return -1;
} }
stcp_server_list_add(instance, tsrv); stcp_server_list_add(instance, tsrv);
DEBUG_INFO(DEBUG_CATEGORY_ETCP, "TCP server %s on port %u", server->name, port); DEBUG_INFO(DEBUG_CATEGORY_ETCP, "TCP server %s on port %u", server->name, ss_get_port(&ts->local_addr));
*out_sock = ts; *out_sock = ts;
return 0; return 0;
} }
@ -2786,7 +2786,7 @@ int init_sockets(struct UTUN_INSTANCE* instance) {
success_count++; success_count++;
DEBUG_INFO(DEBUG_CATEGORY_ETCP, "Initialized server %s on %s (links: %d)", DEBUG_INFO(DEBUG_CATEGORY_ETCP, "Initialized server %s on %s (links: %d)",
server->name, sockaddr_storage_to_str(&server->ip).str, queue_entry_count(sock->links_queue)); server->name, sockaddr_storage_to_str(&sock->local_addr).str, queue_entry_count(sock->links_queue));
server = server->next; server = server->next;
} }

8
tests/test_etcp_stcp.c
Unescape View File

@ -17,8 +17,6 @@
#include <stdlib.h> #include <stdlib.h>
#include <time.h> #include <time.h>
#define TEST_PORT 25678
static int test_failed = 0; static int test_failed = 0;
static struct SC_MYKEYS s_keys, c_keys; static struct SC_MYKEYS s_keys, c_keys;
static struct UTUN_INSTANCE *srv_inst, *cli_inst; static struct UTUN_INSTANCE *srv_inst, *cli_inst;
@ -92,13 +90,11 @@ int main(void) {
struct UASYNC *ua = uasync_create(); TASSERT(ua); struct UASYNC *ua = uasync_create(); TASSERT(ua);
utun_instance_set_tun_init_enabled(0); utun_instance_set_tun_init_enabled(0);
int port = TEST_PORT + rand() % 1000; srv_inst = utun_instance_create_from_str(ua, build_server_config(0));
srv_inst = utun_instance_create_from_str(ua, build_server_config(port));
TASSERT(srv_inst); TASSERT(srv_inst);
TASSERT(utun_instance_init(srv_inst) >= 0); TASSERT(utun_instance_init(srv_inst) >= 0);
cli_inst = utun_instance_create_from_str(ua, build_client_config(port + 1)); cli_inst = utun_instance_create_from_str(ua, build_client_config(0));
TASSERT(cli_inst); TASSERT(cli_inst);
TASSERT(utun_instance_init(cli_inst) >= 0); TASSERT(utun_instance_init(cli_inst) >= 0);

9
tests/test_stcp_link.c
Unescape View File

@ -14,7 +14,6 @@ static int tests_passed = 0, tests_total = 0;
static struct SC_MYKEYS s_keys, c_keys; static struct SC_MYKEYS s_keys, c_keys;
static int test_failed = 0; static int test_failed = 0;
#define BASE_PORT 24567
#define TASSERT(cond) do { \ #define TASSERT(cond) do { \
if (!(cond)) { DEBUG_ERROR(DEBUG_CATEGORY_GENERAL, " FAIL: %s", #cond); test_failed = 1; return test_failed; } \ if (!(cond)) { DEBUG_ERROR(DEBUG_CATEGORY_GENERAL, " FAIL: %s", #cond); test_failed = 1; return test_failed; } \
} while(0) } while(0)
@ -45,11 +44,11 @@ static int test1_basic(void) {
struct stcp_link_config cfg = {.inst = inst, .listen_family = AF_INET}; struct stcp_link_config cfg = {.inst = inst, .listen_family = AF_INET};
g_on_srv_link = NULL; g_on_srv_link = NULL;
struct stcp_server *ss = stcp_server_listen(&cfg, BASE_PORT + 1, on_srv_link, NULL); struct stcp_server *ss = stcp_server_listen(&cfg, 0, on_srv_link, NULL);
TASSERT(ss); TASSERT(ss);
int cli_ready = 0; int cli_ready = 0;
struct stcp_client *sc = stcp_client_connect(ua, "127.0.0.1", BASE_PORT + 1, &c_keys, struct stcp_client *sc = stcp_client_connect(ua, "127.0.0.1", stcp_link_server_get_port(ss), &c_keys,
s_keys.public_key, NULL, s_keys.public_key, NULL,
0, NULL, 0, NULL,
0, 200, 0, on_cli_ready, &cli_ready, NULL, NULL, NULL, NULL, NULL, 0, NULL, NULL, NULL); 0, 200, 0, on_cli_ready, &cli_ready, NULL, NULL, NULL, NULL, NULL, 0, NULL, NULL, NULL);
@ -73,12 +72,12 @@ static int test2_wrong_key(void) {
struct stcp_link_config cfg = {.inst = inst, .listen_family = AF_INET}; struct stcp_link_config cfg = {.inst = inst, .listen_family = AF_INET};
g_on_srv_link = NULL; g_on_srv_link = NULL;
struct stcp_server *ss = stcp_server_listen(&cfg, BASE_PORT + 2, on_srv_link, NULL); struct stcp_server *ss = stcp_server_listen(&cfg, 0, on_srv_link, NULL);
TASSERT(ss); TASSERT(ss);
struct SC_MYKEYS rogue; TASSERT(sc_generate_keypair(&rogue) == SC_OK); struct SC_MYKEYS rogue; TASSERT(sc_generate_keypair(&rogue) == SC_OK);
int cli_ready = 0; int cli_ready = 0;
struct stcp_client *sc = stcp_client_connect(ua, "127.0.0.1", BASE_PORT + 2, &c_keys, struct stcp_client *sc = stcp_client_connect(ua, "127.0.0.1", stcp_link_server_get_port(ss), &c_keys,
rogue.public_key, NULL, rogue.public_key, NULL,
0, NULL, 0, NULL,
0, 200, 0, on_cli_ready, &cli_ready, NULL, NULL, NULL, NULL, NULL, 0, NULL, NULL, NULL); 0, 200, 0, on_cli_ready, &cli_ready, NULL, NULL, NULL, NULL, NULL, 0, NULL, NULL, NULL);

7
tests/test_stcp_traffic.c
Unescape View File

@ -16,7 +16,6 @@
#include <stdlib.h> #include <stdlib.h>
#include <time.h> #include <time.h>
#define TEST_PORT 25680
#define TRAFFIC_MB (1024*1024) #define TRAFFIC_MB (1024*1024)
#define PKT_MIN 10 #define PKT_MIN 10
#define PKT_MAX 1800 #define PKT_MAX 1800
@ -143,13 +142,11 @@ int main(void) {
struct UASYNC *ua = uasync_create(); TASSERT(ua); struct UASYNC *ua = uasync_create(); TASSERT(ua);
utun_instance_set_tun_init_enabled(0); utun_instance_set_tun_init_enabled(0);
int port = TEST_PORT + rand() % 1000; srv_inst = utun_instance_create_from_str(ua, build_server_config(0));
srv_inst = utun_instance_create_from_str(ua, build_server_config(port));
TASSERT(srv_inst); TASSERT(srv_inst);
TASSERT(utun_instance_init(srv_inst) >= 0); TASSERT(utun_instance_init(srv_inst) >= 0);
cli_inst = utun_instance_create_from_str(ua, build_client_config(port + 1)); cli_inst = utun_instance_create_from_str(ua, build_client_config(0));
TASSERT(cli_inst); TASSERT(cli_inst);
TASSERT(utun_instance_init(cli_inst) >= 0); TASSERT(utun_instance_init(cli_inst) >= 0);

Write Preview
Loading…
Cancel
Save