Browse Source
- STCP-хендшейк: добавлен байт флагов (STCP_HANDSHAKE_FLAG_PING), HS_ENC 46→47 - stcp_client: stcp_ping_send() — connect+handshake(PING)+rtt+close; ошибка/таймаут → success=0 - stcp_server: детект флага, ответ с эхо-флагом, мягкое закрытие (close_after_send) - etcp_connections: etcp_send_tcp_ping(), guard is_tcp/fd и проверка возврата send_ping_raw - route_connectivity: probe TCP-адресов через TCP-ping (1 попытка), UDP-кандидаты без TCP-сокетов - tests: test_stcp_ping (success+failure path), обновлена сигнатура stcp_client_connecttopo_upd
evgeny
2 months ago
13 changed files with 324 additions and 74 deletions
@ -0,0 +1,97 @@
|
||||
// test_stcp_ping.c — TCP-ping (etcp_send_tcp_ping / stcp_ping_send) end-to-end check
|
||||
#include "stcp_client.h" |
||||
#include "etcp_connections.h" |
||||
#include "etcp.h" |
||||
#include "topo_group.h" |
||||
#include "topo_node.h" |
||||
#include "../src/utun_instance.h" |
||||
#include "../lib/u_async.h" |
||||
#include "../lib/debug_config.h" |
||||
#include "../lib/mem.h" |
||||
#include <stdio.h> |
||||
#include <string.h> |
||||
#include <stdlib.h> |
||||
#include <time.h> |
||||
|
||||
static int ping_ok = -1, ping_rtt = 0; |
||||
static void ping_cb(int success, uint16_t rtt, void *arg, uint64_t nonce, |
||||
const uint8_t *resp_data, size_t resp_data_len) { |
||||
(void)arg; (void)nonce; (void)resp_data; (void)resp_data_len; |
||||
ping_ok = success; ping_rtt = rtt; |
||||
DEBUG_INFO(DEBUG_CATEGORY_GENERAL, "TCP ping cb: success=%d rtt=%u", success, rtt); |
||||
} |
||||
|
||||
static char *build_config(const char *node_id, const char *priv, const char *pub, int port) { |
||||
static char buf[512]; |
||||
snprintf(buf, sizeof(buf), |
||||
"[global]\n" |
||||
"my_node_id=0x%s\n" |
||||
"my_private_key=%s\n" |
||||
"my_public_key=%s\n" |
||||
"\n" |
||||
"[server: stcp]\n" |
||||
"addr=127.0.0.1:%d\n" |
||||
"type=public\n" |
||||
"transport=tcp\n", |
||||
node_id, priv, pub, port); |
||||
return buf; |
||||
} |
||||
|
||||
int main(void) { |
||||
debug_config_init(); |
||||
debug_set_level(DEBUG_LEVEL_INFO); |
||||
debug_set_categories(DEBUG_CATEGORY_GENERAL | DEBUG_CATEGORY_ETCP | DEBUG_CATEGORY_CONNECTION); |
||||
DEBUG_INFO(DEBUG_CATEGORY_GENERAL, "=== TCP ping (etcp_send_tcp_ping) ==="); |
||||
|
||||
srand((unsigned)time(NULL)); |
||||
struct UASYNC *ua = uasync_create(); |
||||
if (!ua) return 1; |
||||
utun_instance_set_tun_init_enabled(0); |
||||
int port = 27000 + rand() % 1000; |
||||
|
||||
struct UTUN_INSTANCE *srv = utun_instance_create_from_str(ua, build_config( |
||||
"AAAAAAAAAAAAAAAA", |
||||
"38240cb82199e504686507f11f6eaa4f740fde6f0c425c495e49a523019a5d68", |
||||
"ce8871f07fa056c636d297115f231b08c29cdf94e0d440fce83a07c34416d36a", port)); |
||||
if (!srv || utun_instance_init(srv) < 0) { DEBUG_ERROR(DEBUG_CATEGORY_GENERAL, "srv init failed"); return 1; } |
||||
|
||||
struct UTUN_INSTANCE *cli = utun_instance_create_from_str(ua, build_config( |
||||
"BBBBBBBBBBBBBBBB", |
||||
"704f2e012c8fa8768130cb0f988a997dccb628372bc5ceccacc78dcbfec5916f", |
||||
"b3193173def895bd0fcea6f86af077c7d77216f10395275f627ac18242ec0f01", port + 1)); |
||||
if (!cli || utun_instance_init(cli) < 0) { DEBUG_ERROR(DEBUG_CATEGORY_GENERAL, "cli init failed"); return 1; } |
||||
|
||||
uint8_t srv_pub[SC_PUBKEY_SIZE]; |
||||
memcpy(srv_pub, srv->my_keys.public_key, SC_PUBKEY_SIZE); |
||||
|
||||
/* success path: ping listening STCP server */ |
||||
struct sockaddr_storage addr; memset(&addr, 0, sizeof(addr)); |
||||
struct sockaddr_in *sin = (struct sockaddr_in *)&addr; |
||||
sin->sin_family = AF_INET; sin->sin_addr.s_addr = htonl(0x7f000001); sin->sin_port = htons(port); |
||||
|
||||
int rc = etcp_send_tcp_ping(cli, srv_pub, &addr, 2000, ping_cb, NULL); |
||||
if (rc != 0) { DEBUG_ERROR(DEBUG_CATEGORY_GENERAL, "etcp_send_tcp_ping failed rc=%d", rc); return 1; } |
||||
|
||||
int ticks = 0; |
||||
while (ping_ok < 0 && ticks < 5000) { uasync_poll(ua, 10); ticks++; } |
||||
if (ping_ok != 1) { DEBUG_ERROR(DEBUG_CATEGORY_GENERAL, "=== FAILED (success path): success=%d rtt=%u ===", ping_ok, ping_rtt); return 1; } |
||||
DEBUG_INFO(DEBUG_CATEGORY_GENERAL, "=== success path PASSED: rtt=%u ===", ping_rtt); |
||||
|
||||
/* failure path: port with no listener */ |
||||
ping_ok = -1; ping_rtt = 0; |
||||
struct sockaddr_storage bad_addr; memset(&bad_addr, 0, sizeof(bad_addr)); |
||||
struct sockaddr_in *bsin = (struct sockaddr_in *)&bad_addr; |
||||
bsin->sin_family = AF_INET; bsin->sin_addr.s_addr = htonl(0x7f000001); bsin->sin_port = htons(port + 5000); |
||||
rc = etcp_send_tcp_ping(cli, srv_pub, &bad_addr, 1000, ping_cb, NULL); |
||||
ticks = 0; |
||||
while (ping_ok < 0 && ticks < 3000) { uasync_poll(ua, 10); ticks++; } |
||||
if (ping_ok != 0) { DEBUG_ERROR(DEBUG_CATEGORY_GENERAL, "=== FAILED (failure path): success=%d rtt=%u rc=%d ===", ping_ok, ping_rtt, rc); return 1; } |
||||
DEBUG_INFO(DEBUG_CATEGORY_GENERAL, "=== failure path PASSED: success=%d ===", ping_ok); |
||||
|
||||
srv->running = 0; cli->running = 0; |
||||
utun_instance_destroy(srv); utun_instance_destroy(cli); |
||||
uasync_destroy(ua, 0); |
||||
|
||||
DEBUG_INFO(DEBUG_CATEGORY_GENERAL, "=== PASSED ==="); |
||||
return 0; |
||||
} |
||||
Loading…
Reference in new issue