You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
861 lines
46 KiB
861 lines
46 KiB
/** |
|
* @file topo_group.c |
|
* @brief BGP-like обмен топологией узлов (группы) |
|
*/ |
|
#include <stdlib.h> |
|
#include <string.h> |
|
#include <stdio.h> |
|
#include "../lib/platform_compat.h" |
|
#include "../lib/debug_config.h" |
|
#include "../lib/mem.h" |
|
#include "../lib/memory_pool.h" |
|
#include "utun_instance.h" |
|
#include "etcp_api.h" |
|
#include "etcp.h" |
|
#include "etcp_connections.h" |
|
#include "etcp_debug.h" |
|
#include "config_parser.h" |
|
#include "topo_node.h" |
|
#include "topo_node_sqlite.h" |
|
#include "route_lib.h" |
|
#include "topo_group.h" |
|
#include "nat_detection.h" |
|
#include "route_connectivity.h" |
|
#include "control_server.h" |
|
#include "conn_mgr.h" |
|
#include "topo_group_connect.h" |
|
#include "topo_recovery.h" |
|
|
|
|
|
#define TOPO_GROUP_UTUN 0x8000000000000000ULL |
|
|
|
|
|
// ============================================================================ |
|
// Вспомогательные функции |
|
// ============================================================================ |
|
|
|
static void topo_group_send_table_request(struct TOPO_GROUP* group, struct ETCP_CONN* conn) { |
|
if (!group || !conn) return; |
|
DEBUG_INFO(DEBUG_CATEGORY_BGP, "Sending table request to %s", conn->log_name); |
|
struct TOPOMSG_TABLE_REQ* req = u_calloc(1, sizeof(struct TOPOMSG_TABLE_REQ)); |
|
if (!req) return; |
|
req->cmd = ETCP_ID_TOPO_ENTRY; |
|
req->subcmd = TOPO_SUBCMD_REQUEST_TABLE; |
|
req->group_id = group->group_id; |
|
struct ll_entry* e = queue_entry_new(0); |
|
if (!e) { u_free(req); return; } |
|
e->dgram = (uint8_t*)req; e->len = sizeof(struct TOPOMSG_TABLE_REQ); |
|
if (etcp_send(conn, e) != 0) { u_free(req); queue_entry_free(e); } |
|
} |
|
|
|
static void topo_group_send_table_complete(struct TOPO_GROUP* group, struct ETCP_CONN* conn) { |
|
if (!group || !conn) return; |
|
struct TOPOMSG_TABLE_REQ* req = u_calloc(1, sizeof(struct TOPOMSG_TABLE_REQ)); |
|
if (!req) return; |
|
req->cmd = ETCP_ID_TOPO_ENTRY; |
|
req->subcmd = TOPO_SUBCMD_TABLE_COMPLETE; |
|
req->group_id = group->group_id; |
|
struct ll_entry* e = queue_entry_new(0); |
|
if (!e) { u_free(req); return; } |
|
e->dgram = (uint8_t*)req; e->len = sizeof(struct TOPOMSG_TABLE_REQ); |
|
if (etcp_send(conn, e) != 0) { u_free(req); queue_entry_free(e); } |
|
} |
|
|
|
static void topo_group_add_to_senders(struct TOPO_GROUP* group, struct ETCP_CONN* conn); |
|
static bool topo_group_should_send_to(const struct TOPO_GROUP_NODE* nq, uint64_t target_id); |
|
static void topo_group_send_full_table(struct TOPO_GROUP* group, struct ETCP_CONN* conn); |
|
static void topo_group_handle_request_table(struct TOPO_GROUP* group, struct ETCP_CONN* conn); |
|
|
|
static void nodeinfo_dump_log(const uint8_t* data, size_t len) { |
|
if (!data || len < sizeof(struct TOPOMSG_NODEINFO_PKT)) return; |
|
struct TOPOMSG_NODEINFO_PKT* pkt = (struct TOPOMSG_NODEINFO_PKT*)data; |
|
struct TOPOMSG_NODE* ni = &pkt->node; |
|
DEBUG_DEBUG(DEBUG_CATEGORY_BGP, "NODEINFO nid=%016llx ver=%u v4s=%u v4a=%u hops=%u", |
|
(unsigned long long)ni->node_id, ni->ver, |
|
ni->local_v4_sockets, ni->local_v4_addrs, ni->hop_count); |
|
} |
|
|
|
static char* nodeinfo_format(const uint8_t* data, size_t len) { |
|
if (!data || len < sizeof(struct TOPOMSG_NODEINFO_PKT)) return NULL; |
|
struct TOPOMSG_NODEINFO_PKT* pkt = (struct TOPOMSG_NODEINFO_PKT*)data; |
|
struct TOPOMSG_NODE* ni = &pkt->node; |
|
uint64_t node_id = ni->node_id; |
|
const uint8_t* dyn = data + sizeof(struct TOPOMSG_NODEINFO_PKT); |
|
size_t off = 0; |
|
char name_buf[64] = {0}; |
|
size_t nl = ni->node_name_len; |
|
if (nl > 0 && nl < 63 && off + nl < len - sizeof(struct TOPOMSG_NODEINFO_PKT)) { |
|
memcpy(name_buf, dyn + off, nl); off += nl; |
|
} |
|
off += ni->local_v4_sockets * sizeof(struct TOPOMSG_SOCKMETA4); |
|
off += ni->local_v4_addrs * sizeof(struct TOPOMSG_ADDR4); |
|
off += ni->local_v6_sockets * sizeof(struct TOPOMSG_SOCKMETA6); |
|
off += ni->local_v6_addrs * sizeof(struct TOPOMSG_ADDR6); |
|
char subs_buf[512] = {0}; |
|
if (ni->local_v4_subnets > 0) { |
|
const struct TOPOMSG_SUBNET4* subs = (const struct TOPOMSG_SUBNET4*)(dyn + off); |
|
size_t sl = 0; |
|
for (uint8_t i = 0; i < ni->local_v4_subnets && sl < 400; i++) { |
|
uint32_t addr = (subs[i].addr[0]<<24)|(subs[i].addr[1]<<16)|(subs[i].addr[2]<<8)|subs[i].addr[3]; |
|
char tmp[64]; snprintf(tmp,sizeof(tmp),"%d.%d.%d.%d/%u ",(addr>>24)&255,(addr>>16)&255,(addr>>8)&255,addr&255,subs[i].prefix_length); |
|
strcat(subs_buf,tmp); sl += strlen(tmp); |
|
} |
|
} |
|
int need = snprintf(NULL,0,"NODEINFO nid=%016llx ver=%u name=\"%s\" v4subs=\"%s\" v4s=%u v4a=%u hop=%u", (unsigned long long)node_id,(unsigned)ni->ver,name_buf,subs_buf,(unsigned)ni->local_v4_sockets,(unsigned)ni->local_v4_addrs,(unsigned)ni->hop_count); |
|
char* buf = u_malloc(need+1); if (!buf) return NULL; |
|
snprintf(buf,need+1,"NODEINFO nid=%016llx ver=%u name=\"%s\" v4subs=\"%s\" v4s=%u v4a=%u hop=%u", (unsigned long long)node_id,(unsigned)ni->ver,name_buf,subs_buf,(unsigned)ni->local_v4_sockets,(unsigned)ni->local_v4_addrs,(unsigned)ni->hop_count); |
|
return buf; |
|
} |
|
|
|
static const char* group_subcmd_name(uint8_t subcmd) { |
|
switch (subcmd) { |
|
case TOPO_SUBCMD_NODEINFO: return "NODEINFO"; |
|
case TOPO_SUBCMD_REQUEST_TABLE: return "REQ_TABLE"; |
|
case TOPO_SUBCMD_WITHDRAW: return "WITHDRAW"; |
|
case TOPO_SUBCMD_TABLE_COMPLETE: return "TABLE_COMPLETE"; |
|
case TOPO_SUBCMD_ERR_GROUP_MISMATCH: return "ERR_GROUP_MISMATCH"; |
|
default: return "?"; |
|
} |
|
} |
|
|
|
|
|
// ============================================================================ |
|
// Broadcast / Withdraw |
|
// ============================================================================ |
|
|
|
static void topo_group_broadcast_withdraw(struct TOPO_GROUP* group, uint64_t node_id, uint64_t wd_source, struct ETCP_CONN* exclude) { |
|
if (!group) { DEBUG_ERROR(DEBUG_CATEGORY_BGP, "group is NULL"); return; } |
|
struct TOPOMSG_WITHDRAW_PKT* pkt = u_calloc(1, sizeof(struct TOPOMSG_WITHDRAW_PKT)); |
|
if (!pkt) return; |
|
pkt->cmd = ETCP_ID_TOPO_ENTRY; pkt->subcmd = TOPO_SUBCMD_WITHDRAW; |
|
pkt->group_id = group->group_id; pkt->node_id = node_id; pkt->wd_source = wd_source; |
|
struct ll_entry* e = group->senders_list ? group->senders_list->head : NULL; |
|
while (e) { |
|
struct TOPO_GROUP_CONN_ITEM* item = (struct TOPO_GROUP_CONN_ITEM*)e->data; |
|
if (item && item->conn && item->conn != exclude) { |
|
struct ll_entry* copy = queue_entry_new(0); |
|
if (copy) { copy->dgram = u_malloc(sizeof(*pkt)); if (copy->dgram) memcpy(copy->dgram, pkt, sizeof(*pkt)); copy->len = sizeof(*pkt); etcp_send(item->conn, copy); } |
|
} |
|
e = e->next; |
|
} |
|
u_free(pkt); |
|
} |
|
|
|
|
|
// ============================================================================ |
|
// Приём пакетов |
|
// ============================================================================ |
|
|
|
static void topo_group_receive_cbk(struct ETCP_CONN* from_conn, struct ll_entry* entry) { |
|
if (!from_conn || !entry || entry->len < 2) { if (entry) { queue_dgram_free(entry); queue_entry_free(entry); } return; } |
|
struct UTUN_INSTANCE* instance = from_conn->instance; |
|
if (!instance || !instance->topo_groups) { DEBUG_ERROR(DEBUG_CATEGORY_BGP, "invalid instance/topo_groups"); queue_dgram_free(entry); queue_entry_free(entry); return; } |
|
uint8_t* data = entry->dgram; uint8_t cmd = data[0]; uint8_t subcmd = data[1]; |
|
if (cmd != ETCP_ID_TOPO_ENTRY) { queue_dgram_free(entry); queue_entry_free(entry); return; } |
|
|
|
uint64_t pkt_group_id = 0; |
|
if (subcmd == TOPO_SUBCMD_NODEINFO && entry->len >= 3) { |
|
pkt_group_id = ((struct TOPOMSG_NODEINFO_PKT*)data)->node.group_id; |
|
} else if (subcmd == TOPO_SUBCMD_WITHDRAW && entry->len >= sizeof(struct TOPOMSG_WITHDRAW_PKT)) { |
|
pkt_group_id = ((struct TOPOMSG_WITHDRAW_PKT*)data)->group_id; |
|
} else if (subcmd == TOPO_SUBCMD_REQUEST_TABLE && entry->len >= sizeof(struct TOPOMSG_TABLE_REQ)) { |
|
pkt_group_id = ((struct TOPOMSG_TABLE_REQ*)data)->group_id; |
|
} else if (subcmd == TOPO_SUBCMD_TABLE_COMPLETE && entry->len >= sizeof(struct TOPOMSG_TABLE_REQ)) { |
|
pkt_group_id = ((struct TOPOMSG_TABLE_REQ*)data)->group_id; |
|
} else if (subcmd == TOPO_SUBCMD_ERR_GROUP_MISMATCH && entry->len >= sizeof(struct TOPOMSG_ERR_GROUP_MISMATCH)) { |
|
pkt_group_id = ((struct TOPOMSG_ERR_GROUP_MISMATCH*)data)->group_id; |
|
} |
|
|
|
struct TOPO_GROUP* group = topo_groups_find(instance->topo_groups, pkt_group_id); |
|
if (!group) { |
|
DEBUG_WARN(DEBUG_CATEGORY_BGP, "BGP recv %s from %s: group %016llx not found, dropping", group_subcmd_name(subcmd), from_conn->log_name, (unsigned long long)pkt_group_id); |
|
queue_dgram_free(entry); queue_entry_free(entry); return; |
|
} |
|
|
|
DEBUG_INFO(DEBUG_CATEGORY_BGP, "BGP recv %s from %s len=%zu group=%016llx", group_subcmd_name(subcmd), from_conn->log_name, entry->len, (unsigned long long)pkt_group_id); |
|
if (subcmd == TOPO_SUBCMD_NODEINFO) { DEBUG_DEBUG(DEBUG_CATEGORY_DEBUG, "BGP recv NODEINFO from=%s nid=%016llx ver=%d grp=%016llx", from_conn->log_name, (unsigned long long)((struct TOPOMSG_NODEINFO_PKT*)data)->node.node_id, ((struct TOPOMSG_NODEINFO_PKT*)data)->node.ver, (unsigned long long)((struct TOPOMSG_NODEINFO_PKT*)data)->node.group_id); } |
|
|
|
if (subcmd == TOPO_SUBCMD_NODEINFO) { nodeinfo_dump_log(data, entry->len); topo_group_process_nodeinfo(group, from_conn, data, entry->len); } |
|
else if (subcmd == TOPO_SUBCMD_WITHDRAW) topo_group_process_withdraw(group, from_conn, data, entry->len); |
|
else if (subcmd == TOPO_SUBCMD_REQUEST_TABLE) topo_group_handle_request_table(group, from_conn); |
|
else if (subcmd == TOPO_SUBCMD_TABLE_COMPLETE) { etcp_set_routing_exchange_state(from_conn, 3); DEBUG_INFO(DEBUG_CATEGORY_BGP, "BGP initial sync complete with %s", from_conn->log_name); } |
|
else if (subcmd == TOPO_SUBCMD_ERR_GROUP_MISMATCH) { |
|
if (entry->len >= sizeof(struct TOPOMSG_ERR_GROUP_MISMATCH)) { |
|
struct TOPOMSG_ERR_GROUP_MISMATCH* err = (struct TOPOMSG_ERR_GROUP_MISMATCH*)data; |
|
DEBUG_ERROR(DEBUG_CATEGORY_BGP, "Group mismatch from %s: expected=%u received=0x%02x group=%016llx", |
|
from_conn->log_name, err->expected_type, err->received_flags, (unsigned long long)err->group_id); |
|
} |
|
} |
|
|
|
queue_dgram_free(entry); queue_entry_free(entry); |
|
} |
|
|
|
|
|
// ============================================================================ |
|
// Init / Destroy / New / Remove conn |
|
// ============================================================================ |
|
|
|
static void topo_group_conn_status(struct ETCP_CONN* conn, int status, void* arg) { |
|
struct TOPO_GROUPS* groups = (struct TOPO_GROUPS*)arg; |
|
if (!conn || !groups) return; |
|
struct ll_entry* fe = groups->group_list->head; |
|
while (fe) { struct TOPO_GROUP* g = (struct TOPO_GROUP*)fe; fe = fe->next; |
|
if (status == ETCP_CONN_STATUS_UP) topo_group_new_conn(g, conn); |
|
if (status == ETCP_CONN_STATUS_DOWN) topo_group_remove_conn(g, conn); |
|
} |
|
} |
|
|
|
static struct TOPO_GROUP* topo_group_create(struct UTUN_INSTANCE* instance, uint64_t group_id, uint8_t group_type) { |
|
if (!instance) { DEBUG_ERROR(DEBUG_CATEGORY_BGP, "instance is NULL"); return NULL; } |
|
DEBUG_INFO(DEBUG_CATEGORY_BGP, "group_id=%016llx node_id=%016llx", (unsigned long long)group_id, (unsigned long long)instance->node_id); |
|
|
|
struct ll_entry* qe = queue_entry_new(sizeof(struct TOPO_GROUP)); |
|
if (!qe) { DEBUG_ERROR(DEBUG_CATEGORY_BGP, "alloc failed"); return NULL; } |
|
struct TOPO_GROUP* group = (struct TOPO_GROUP*)qe; |
|
memset((uint8_t*)group + sizeof(struct ll_entry), 0, sizeof(*group) - sizeof(struct ll_entry)); |
|
group->group_id = group_id; |
|
group->group_type = group_type; |
|
group->instance = instance; |
|
|
|
sc_derive_ed25519_pubkey(instance->my_keys.private_key, group->ed25519_public_key); |
|
DEBUG_INFO(DEBUG_CATEGORY_BGP, "Ed25519 pubkey derived from X25519 private key"); |
|
|
|
group->nodes = queue_new(instance->ua, BGP_NODES_HASH_SIZE, offsetof(struct TOPO_GROUP_NODE, node_id) - sizeof(struct ll_entry), 8, "group_nodes"); |
|
if (!group->nodes) { queue_entry_free(qe); DEBUG_ERROR(DEBUG_CATEGORY_BGP, "nodes queue creation failed"); return NULL; } |
|
|
|
group->senders_list = queue_new(instance->ua, 0, 0, 0, "group_senders"); |
|
if (!group->senders_list) { queue_free(group->nodes); queue_entry_free(qe); DEBUG_ERROR(DEBUG_CATEGORY_BGP, "senders_list queue creation failed"); return NULL; } |
|
|
|
group->local_node = NULL; |
|
topo_group_update_my_nodeinfo(instance, group); |
|
|
|
group->conn_mgr = conn_mgr_init(group); |
|
|
|
DEBUG_INFO(DEBUG_CATEGORY_BGP, "TOPO group created: group_id=%016llx", (unsigned long long)group_id); |
|
return group; |
|
} |
|
|
|
static void topo_group_destroy(struct TOPO_GROUP* group) { |
|
if (!group) return; |
|
DEBUG_INFO(DEBUG_CATEGORY_BGP, "group_id=%016llx", (unsigned long long)group->group_id); |
|
|
|
topo_recovery_cancel_all(group); |
|
topo_group_connect_destroy(group); |
|
|
|
struct ll_entry* e; |
|
while ((e = queue_data_get(group->senders_list)) != NULL) queue_entry_free(e); |
|
queue_free(group->senders_list); |
|
|
|
if (group->local_node) { topo_nodeq_free_group_fields(group->instance->topo_groups, group->local_node); u_free(group->local_node); } |
|
|
|
if (group->nodes) queue_free(group->nodes); |
|
group->conn_mgr = NULL; /* destroyed externally in utun_instance_destroy before etcp_router_destroy */ |
|
} |
|
|
|
struct TOPO_GROUPS* topo_groups_init(struct UTUN_INSTANCE* instance) { |
|
if (!instance) { DEBUG_ERROR(DEBUG_CATEGORY_BGP, "instance is NULL"); return NULL; } |
|
DEBUG_INFO(DEBUG_CATEGORY_BGP, "node_id=%016llx", (unsigned long long)instance->node_id); |
|
|
|
struct TOPO_GROUPS* g = u_calloc(1, sizeof(struct TOPO_GROUPS)); |
|
if (!g) { DEBUG_ERROR(DEBUG_CATEGORY_BGP, "alloc failed"); return NULL; } |
|
g->instance = instance; |
|
|
|
g->group_list = queue_new(instance->ua, 16, 0, 8, "topo_groups"); |
|
if (!g->group_list) { u_free(g); DEBUG_ERROR(DEBUG_CATEGORY_BGP, "group_list queue creation failed"); return NULL; } |
|
|
|
g->node_registry = queue_new(instance->ua, TOPO_NODE_REGISTRY_HASH_SIZE, 0, 8, "node_registry"); |
|
if (!g->node_registry) { queue_free(g->group_list); u_free(g); DEBUG_ERROR(DEBUG_CATEGORY_BGP, "node_registry queue creation failed"); return NULL; } |
|
|
|
g->v4_sock_meta_pool = memory_pool_init(sizeof(struct TOPO_SOCKMETA4), "to_v4sock"); |
|
g->v4_addr_pool = memory_pool_init(sizeof(struct TOPO_ADDR4), "to_v4addr"); |
|
g->v6_sock_meta_pool = memory_pool_init(sizeof(struct TOPO_SOCKMETA6), "to_v6sock"); |
|
g->v6_addr_pool = memory_pool_init(sizeof(struct TOPO_ADDR6), "to_v6addr"); |
|
g->v4_subnet_pool = memory_pool_init(sizeof(struct TOPO_SUBNET4), "to_v4sub"); |
|
g->v6_subnet_pool = memory_pool_init(sizeof(struct TOPO_SUBNET6), "to_v6sub"); |
|
|
|
if (instance->config && instance->config->global.db_path[0]) { |
|
char db_file[512]; |
|
snprintf(db_file, sizeof(db_file), "%s/chats.db", instance->config->global.db_path); |
|
int rc = sqlite3_open(db_file, &instance->topo_sqlite_db); |
|
if (rc == SQLITE_OK && instance->topo_sqlite_db) { |
|
sqlite3_exec(instance->topo_sqlite_db, "PRAGMA journal_mode=WAL", NULL, NULL, NULL); |
|
sqlite3_exec(instance->topo_sqlite_db, "PRAGMA foreign_keys=ON", NULL, NULL, NULL); |
|
sqlite3_exec(instance->topo_sqlite_db, "PRAGMA synchronous=NORMAL", NULL, NULL, NULL); |
|
sqlite3_exec(instance->topo_sqlite_db, "PRAGMA wal_autocheckpoint=10000", NULL, NULL, NULL); |
|
sqlite3_exec(instance->topo_sqlite_db, "PRAGMA cache_size=-32768", NULL, NULL, NULL); |
|
sqlite3_exec(instance->topo_sqlite_db, "PRAGMA mmap_size=134217728", NULL, NULL, NULL); |
|
sqlite3_exec(instance->topo_sqlite_db, "PRAGMA temp_store=MEMORY", NULL, NULL, NULL); |
|
topo_node_sqlite_init(instance->topo_sqlite_db); |
|
DEBUG_INFO(DEBUG_CATEGORY_DEBUG, "SQLite opened: %s rc=%d db=%p", db_file, rc, (void*)instance->topo_sqlite_db); |
|
} else { |
|
DEBUG_ERROR(DEBUG_CATEGORY_BGP, "sqlite3_open(%s) failed: %s", |
|
db_file, instance->topo_sqlite_db ? sqlite3_errmsg(instance->topo_sqlite_db) : "null db"); |
|
if (instance->topo_sqlite_db) { sqlite3_close(instance->topo_sqlite_db); instance->topo_sqlite_db = NULL; } |
|
} |
|
} |
|
|
|
instance->topo_groups = g; |
|
|
|
struct TOPO_GROUP* default_group = topo_group_create(instance, TOPO_GROUP_UTUN, TOPO_GROUP_TYPE_UTUN); |
|
if (!default_group) { |
|
memory_pool_destroy(g->v4_sock_meta_pool); memory_pool_destroy(g->v4_addr_pool); |
|
memory_pool_destroy(g->v6_sock_meta_pool); memory_pool_destroy(g->v6_addr_pool); |
|
memory_pool_destroy(g->v4_subnet_pool); memory_pool_destroy(g->v6_subnet_pool); |
|
queue_free(g->node_registry); queue_free(g->group_list); u_free(g); |
|
DEBUG_ERROR(DEBUG_CATEGORY_BGP, "failed to create default group"); |
|
return NULL; |
|
} |
|
queue_data_put_with_index(g->group_list, &default_group->ll); |
|
|
|
instance->conn_mgr = default_group->conn_mgr; /* recv handler shortcut */ |
|
|
|
etcp_bind(instance, ETCP_ID_TOPO_ENTRY, topo_group_receive_cbk); |
|
etcp_add_conn_status_cbk(instance, topo_group_conn_status, g); |
|
|
|
DEBUG_INFO(DEBUG_CATEGORY_BGP, "TOPO groups initialized (with default group)"); |
|
return g; |
|
} |
|
|
|
void topo_groups_destroy(struct UTUN_INSTANCE* instance) { |
|
if (!instance) { DEBUG_ERROR(DEBUG_CATEGORY_BGP, "instance is NULL"); return; } |
|
if (!instance->topo_groups) { DEBUG_ERROR(DEBUG_CATEGORY_BGP, "topo_groups is NULL"); return; } |
|
DEBUG_INFO(DEBUG_CATEGORY_BGP, "node_id=%016llx", (unsigned long long)instance->node_id); |
|
|
|
etcp_unbind(instance, ETCP_ID_TOPO_ENTRY); |
|
etcp_remove_conn_status_cbk(instance, topo_group_conn_status, instance->topo_groups); |
|
route_connectivity_cancel_all(instance); |
|
|
|
struct TOPO_GROUPS* g = instance->topo_groups; |
|
struct ll_entry* ge; |
|
while ((ge = queue_data_get(g->group_list)) != NULL) { struct TOPO_GROUP* grp = (struct TOPO_GROUP*)ge; topo_group_destroy(grp); queue_entry_free(ge); } |
|
queue_free(g->group_list); |
|
|
|
if (g->node_registry) { |
|
struct ll_entry* re; |
|
while ((re = queue_data_get(g->node_registry)) != NULL) { |
|
struct TOPO_NODE* node; |
|
memcpy(&node, re->data + 8, sizeof(node)); |
|
topo_node_registry_unref(g, node->node_id); |
|
queue_entry_free(re); |
|
} |
|
queue_free(g->node_registry); |
|
} |
|
|
|
memory_pool_destroy(g->v4_sock_meta_pool); |
|
memory_pool_destroy(g->v4_addr_pool); |
|
memory_pool_destroy(g->v6_sock_meta_pool); |
|
memory_pool_destroy(g->v6_addr_pool); |
|
memory_pool_destroy(g->v4_subnet_pool); |
|
memory_pool_destroy(g->v6_subnet_pool); |
|
|
|
if (instance->topo_sqlite_db) { sqlite3_close(instance->topo_sqlite_db); instance->topo_sqlite_db = NULL; } |
|
|
|
u_free(g); instance->topo_groups = NULL; |
|
} |
|
|
|
struct TOPO_GROUP* topo_groups_get_default(struct TOPO_GROUPS* g) { |
|
return topo_groups_find(g, TOPO_GROUP_UTUN); |
|
} |
|
|
|
struct TOPO_GROUP* topo_groups_find(struct TOPO_GROUPS* g, uint64_t group_id) { |
|
if (!g || !g->group_list) return NULL; |
|
struct ll_entry* e = queue_find_data_by_index(g->group_list, &group_id); |
|
return e ? (struct TOPO_GROUP*)e : NULL; |
|
} |
|
|
|
struct TOPO_GROUP* topo_groups_create_group(struct TOPO_GROUPS* g, uint64_t group_id, uint8_t group_type, const char* channel_id) { |
|
if (!g || !g->instance) { DEBUG_ERROR(DEBUG_CATEGORY_BGP, "invalid args"); return NULL; } |
|
if (topo_groups_find(g, group_id)) { DEBUG_ERROR(DEBUG_CATEGORY_BGP, "group_id=%016llx already exists", (unsigned long long)group_id); return NULL; } |
|
struct TOPO_GROUP* group = topo_group_create(g->instance, group_id, group_type); |
|
if (!group) return NULL; |
|
if (channel_id) { strncpy(group->channel_id, channel_id, sizeof(group->channel_id) - 1); group->channel_id[sizeof(group->channel_id) - 1] = '\0'; } |
|
queue_data_put_with_index(g->group_list, &group->ll); |
|
|
|
struct ll_queue* connections = g->instance->connections; |
|
if (connections) { |
|
for (uint32_t slot = 0; slot < connections->hash_size; slot++) { |
|
struct ll_entry* entry = connections->hash_table[slot]; |
|
while (entry) { |
|
struct conn_queue_entry* cqe = (struct conn_queue_entry*)entry; |
|
if (cqe->conn && cqe->peer_node_id != 0) |
|
topo_group_new_conn(group, cqe->conn); |
|
entry = entry->hash_next; |
|
} |
|
} |
|
} |
|
|
|
DEBUG_INFO(DEBUG_CATEGORY_BGP, "Group created: group_id=%016llx type=%u ch_id=%s", (unsigned long long)group_id, group_type, group->channel_id); |
|
if (group_type == TOPO_GROUP_TYPE_CHAT) topo_group_connect_init(group); |
|
return group; |
|
} |
|
|
|
void topo_group_set_nat_check_local(struct TOPO_GROUP* group, int allow) { |
|
if (!group || !group->instance || !group->instance->nat_det) return; |
|
nat_detection_set_allow_local(group->instance->nat_det, allow); |
|
} |
|
|
|
void topo_groups_set_node_updated_cb(struct TOPO_GROUPS* groups, topo_node_updated_fn fn) { |
|
if (groups) groups->node_updated_cb = fn; |
|
} |
|
|
|
void topo_group_new_conn(struct TOPO_GROUP* group, struct ETCP_CONN* conn) { |
|
if (!group || !conn) { DEBUG_ERROR(DEBUG_CATEGORY_BGP, "invalid args"); return; } |
|
if (!conn->instance) { DEBUG_ERROR(DEBUG_CATEGORY_BGP, "conn->instance is NULL"); return; } |
|
if (!conn->instance->rt) { DEBUG_ERROR(DEBUG_CATEGORY_BGP, "conn->instance->rt is NULL"); return; } |
|
|
|
topo_recovery_cancel_for_node(group, conn->peer_node_id); |
|
|
|
struct TOPO_GROUP_NODE* peer_nq = topo_node_find_by_id(group, conn->peer_node_id); |
|
topo_group_add_to_senders(group, conn); |
|
|
|
DEBUG_DEBUG(DEBUG_CATEGORY_DEBUG, "topo_group_new_conn: peer=%016llx group=%016llx type=%d ch=%s", (unsigned long long)conn->peer_node_id, (unsigned long long)group->group_id, group->group_type, group->channel_id); |
|
|
|
topo_group_send_table_request(group, conn); |
|
topo_group_connect_on_up(group, conn); |
|
} |
|
|
|
void topo_group_remove_conn(struct TOPO_GROUP* group, struct ETCP_CONN* conn) { |
|
if (!group || !conn) { DEBUG_ERROR(DEBUG_CATEGORY_BGP, "invalid args"); return; } |
|
if (!conn->instance) { DEBUG_ERROR(DEBUG_CATEGORY_BGP, "invalid instance"); return; } |
|
|
|
bool found_in_list = false; |
|
struct ll_entry* e = group->senders_list->head; |
|
while (e) { struct TOPO_GROUP_CONN_ITEM* item = (struct TOPO_GROUP_CONN_ITEM*)e->data; if (item->conn == conn) { found_in_list = true; break; } e = e->next; } |
|
if (!found_in_list) { DEBUG_WARN(DEBUG_CATEGORY_BGP, "conn already removed or not in list, skipping"); return; } |
|
|
|
DEBUG_INFO(DEBUG_CATEGORY_BGP, "peer=%016llx", (unsigned long long)conn->peer_node_id); |
|
|
|
struct ROUTE_TABLE* rt = conn->instance->rt; |
|
int nodes_removed = 0; |
|
int cascaded = 0; |
|
struct ll_entry* node_entry = group->nodes ? group->nodes->head : NULL; |
|
while (node_entry) { |
|
struct ll_entry* next = node_entry->next; |
|
struct TOPO_GROUP_NODE* nq = (struct TOPO_GROUP_NODE*)node_entry; |
|
if (topo_group_remove_path(nq, conn) == 1) { |
|
uint64_t key = nq->node_id; |
|
if (key != conn->peer_node_id) { topo_recovery_add_node(group, nq, conn->peer_node_id); cascaded++; } |
|
if (group->group_type != TOPO_GROUP_TYPE_CHAT && rt) route_delete(rt, nq); |
|
if (conn->instance && conn->instance->control_srv) control_server_notify_node_removed(conn->instance->control_srv, key); |
|
nq->dirty = 1; |
|
route_connectivity_cancel_node(conn->instance, nq); |
|
if (nq->paths) { queue_free(nq->paths); nq->paths = NULL; } |
|
topo_nodeq_free_group_fields(group->instance->topo_groups, nq); |
|
struct ll_entry* entry = node_entry; |
|
if (entry) { queue_remove_data(group->nodes, entry); queue_entry_free(entry); } |
|
nodes_removed++; |
|
etcp_router_conn_close_all_for_node(group->instance, group->group_id, key); |
|
uint64_t wd_src = (key == conn->peer_node_id) ? conn->instance->node_id : conn->peer_node_id; |
|
topo_group_broadcast_withdraw(group, key, wd_src, NULL); |
|
DEBUG_DEBUG(DEBUG_CATEGORY_BGP, "Removed node %016llx after link down", (unsigned long long)key); |
|
} |
|
node_entry = next; |
|
} |
|
|
|
e = group->senders_list->head; |
|
while (e) { struct TOPO_GROUP_CONN_ITEM* item = (struct TOPO_GROUP_CONN_ITEM*)e->data; if (item->conn == conn) { queue_remove_data(group->senders_list, e); queue_entry_free(e); break; } e = e->next; } |
|
|
|
if (cascaded > 0) topo_recovery_start(group); |
|
|
|
topo_group_connect_on_down(group, conn); |
|
|
|
DEBUG_INFO(DEBUG_CATEGORY_BGP, "BGP peer removed: %s nodes=%d", conn->log_name, nodes_removed); |
|
} |
|
|
|
|
|
// ============================================================================ |
|
// NODEINFO BASED IMPLEMENTATION |
|
// ============================================================================ |
|
|
|
struct ETCP_CONN* topo_group_find_conn_for_node(struct TOPO_GROUP* group, uint64_t node_id) { |
|
if (!group) return NULL; |
|
struct TOPO_GROUP_NODE* nq = topo_node_find_by_id(group, node_id); |
|
DEBUG_DEBUG(DEBUG_CATEGORY_DEBUG, "find_conn_for_node %016llx: nq=%p paths=%p head=%p", |
|
(unsigned long long)node_id, (void*)nq, nq ? (void*)nq->paths : NULL, (nq && nq->paths) ? (void*)nq->paths->head : NULL); |
|
if (!nq || !nq->paths || !nq->paths->head) return NULL; |
|
struct ll_entry* e = nq->paths->head; |
|
struct TOPO_NODEPATH* best_live = NULL; uint8_t best_live_hops = 255; |
|
struct TOPO_NODEPATH* best_any = NULL; uint8_t best_any_hops = 255; |
|
while (e) { |
|
struct TOPO_NODEPATH* path = (struct TOPO_NODEPATH*)e; |
|
if (path->conn) { if (path->conn->links_up && path->hop_count < best_live_hops) { best_live = path; best_live_hops = path->hop_count; } if (path->hop_count < best_any_hops) { best_any = path; best_any_hops = path->hop_count; } } |
|
e = e->next; |
|
} |
|
if (!best_live && best_any) DEBUG_DEBUG(DEBUG_CATEGORY_BGP, "find_conn_for_node %016llx: no live paths, fallback to best (hops=%d links_up=%d)", (unsigned long long)node_id, best_any_hops, best_any->conn->links_up); |
|
return best_live ? best_live->conn : (best_any ? best_any->conn : NULL); |
|
} |
|
|
|
int topo_group_add_path(struct TOPO_GROUP_NODE* nq, struct ETCP_CONN* conn, uint64_t* hop_list, uint8_t hop_count, uint16_t cumulative_rtt) { |
|
if (!nq || !conn || hop_count > MAX_HOPS || !hop_list) { DEBUG_ERROR(DEBUG_CATEGORY_BGP, "add_path: invalid args"); return -1; } |
|
if (!nq->paths) { nq->paths = queue_new(conn->instance->ua, 0, 0, 0, "node_paths"); if (!nq->paths) return -1; } |
|
size_t path_size = sizeof(struct TOPO_NODEPATH) - sizeof(struct ll_entry) + hop_count * 8; |
|
struct ll_entry* pe = queue_entry_new(path_size); if (!pe) return -1; |
|
struct TOPO_NODEPATH* path = (struct TOPO_NODEPATH*)pe; |
|
path->conn = conn; path->hop_count = hop_count; path->cumulative_rtt = cumulative_rtt; |
|
memcpy((uint64_t*)((uint8_t*)path + sizeof(struct TOPO_NODEPATH)), hop_list, hop_count * 8); |
|
queue_data_put(nq->paths, pe); |
|
return 0; |
|
} |
|
|
|
static int topo_group_remove_path_by_hop(struct TOPO_GROUP_NODE* nq, uint64_t wd_source) { |
|
if (!nq || !nq->paths) return 0; |
|
int removed = 0; |
|
struct ll_entry* e = nq->paths->head; |
|
while (e) { |
|
struct TOPO_NODEPATH* path = (struct TOPO_NODEPATH*)e; |
|
uint64_t* hop = (uint64_t*)((uint8_t*)path + sizeof(struct TOPO_NODEPATH)); |
|
bool has_wd = false; for (uint8_t i = 0; i < path->hop_count; i++) if (hop[i] == wd_source) { has_wd = true; break; } |
|
if (has_wd) { struct ll_entry* next = e->next; queue_remove_data(nq->paths, e); queue_entry_free(e); removed++; e = next; continue; } |
|
e = e->next; |
|
} |
|
if (removed > 0 && nq->paths && queue_entry_count(nq->paths) == 0) { queue_free(nq->paths); nq->paths = NULL; return 1; } |
|
return removed; |
|
} |
|
|
|
int topo_group_remove_path(struct TOPO_GROUP_NODE* nq, struct ETCP_CONN* conn) { |
|
if (!nq || !conn || !nq->paths) return -1; |
|
struct ll_entry* e = nq->paths->head; |
|
while (e) { struct TOPO_NODEPATH* path = (struct TOPO_NODEPATH*)e; if (path->conn == conn) { queue_remove_data(nq->paths, e); queue_entry_free(e); break; } e = e->next; } |
|
int remaining = nq->paths ? queue_entry_count(nq->paths) : 0; |
|
if (remaining == 0) { if (nq->paths) { queue_free(nq->paths); nq->paths = NULL; } return 1; } |
|
return 0; |
|
} |
|
|
|
// ===== NODEINFO process ===== |
|
|
|
static void topo_group_send_err_group_mismatch(struct TOPO_GROUP* group, struct ETCP_CONN* conn, uint8_t expected_type, uint8_t received_flags) { |
|
if (!group || !conn) return; |
|
struct TOPOMSG_ERR_GROUP_MISMATCH* pkt = u_calloc(1, sizeof(struct TOPOMSG_ERR_GROUP_MISMATCH)); |
|
if (!pkt) return; |
|
pkt->cmd = ETCP_ID_TOPO_ENTRY; pkt->subcmd = TOPO_SUBCMD_ERR_GROUP_MISMATCH; |
|
pkt->group_id = group->group_id; |
|
pkt->expected_type = expected_type; pkt->received_flags = received_flags; |
|
struct ll_entry* e = queue_entry_new(0); |
|
if (!e) { u_free(pkt); return; } |
|
e->dgram = (uint8_t*)pkt; e->len = sizeof(struct TOPOMSG_ERR_GROUP_MISMATCH); |
|
if (etcp_send(conn, e) != 0) { DEBUG_ERROR(DEBUG_CATEGORY_BGP, "etcp_send ERR_GROUP_MISMATCH failed"); u_free(pkt); queue_entry_free(e); } |
|
} |
|
|
|
int topo_group_process_nodeinfo(struct TOPO_GROUP* group, struct ETCP_CONN* from, const uint8_t* data, size_t len) { |
|
if (!group || !from || len < sizeof(struct TOPOMSG_NODEINFO_PKT)) return -1; |
|
|
|
struct TOPOMSG_NODEINFO_PKT* pkt = (struct TOPOMSG_NODEINFO_PKT*)data; |
|
struct TOPOMSG_NODE* ni = &pkt->node; |
|
|
|
if (ni->group_id != group->group_id) { |
|
DEBUG_ERROR(DEBUG_CATEGORY_DEBUG, "NODEINFO group_id mismatch from %s: expected=%016llx got=%016llx nid=%016llx", |
|
from->log_name, (unsigned long long)group->group_id, (unsigned long long)ni->group_id, (unsigned long long)ni->node_id); |
|
return -1; |
|
} |
|
|
|
// Проверка соответствия типа группы |
|
uint8_t expected_flags = (group->group_type == TOPO_GROUP_TYPE_CHAT) ? 0 : TOPO_FLAG_SEND_SUBNETS; |
|
if ((ni->flags & TOPO_FLAG_SEND_SUBNETS) != expected_flags) { |
|
DEBUG_ERROR(DEBUG_CATEGORY_BGP, "Group type mismatch with %s: expected_flags=0x%02x received_flags=0x%02x (local_type=%u)", |
|
from->log_name, expected_flags, ni->flags, group->group_type); |
|
topo_group_send_err_group_mismatch(group, from, group->group_type, ni->flags); |
|
return -1; |
|
} |
|
|
|
int dyn = topo_node_dyn_size(ni); |
|
if ((size_t)dyn + sizeof(struct TOPOMSG_NODEINFO_PKT) != len) { DEBUG_WARN(DEBUG_CATEGORY_BGP, "Incorrect packet size (%s)", from->log_name); return -1; } |
|
|
|
uint64_t node_id = ni->node_id; |
|
if (ni->hop_count >= MAX_HOPS) { DEBUG_WARN(DEBUG_CATEGORY_BGP, "NODEINFO from %s dropped: too many hops (%d)", from->log_name, ni->hop_count); return -1; } |
|
if (node_id == group->instance->node_id) return 0; |
|
|
|
struct TOPO_GROUP_NODE* nodeinfo1 = topo_node_find_by_id(group, node_id); |
|
uint8_t new_ver = ni->ver; |
|
DEBUG_DEBUG(DEBUG_CATEGORY_DEBUG, "process_nodeinfo: nid=%016llx ver=%d hops=%d nodeinfo1=%p from=%s", |
|
(unsigned long long)node_id, new_ver, ni->hop_count, (void*)nodeinfo1, from->log_name); |
|
|
|
if (nodeinfo1 && (int8_t)(nodeinfo1->last_ver - new_ver) >= 0) { |
|
DEBUG_DEBUG(DEBUG_CATEGORY_DEBUG, "NODEINFO skip (stale ver): node=%016llx cur_ver=%d new_ver=%d from=%s", (unsigned long long)node_id, nodeinfo1->last_ver, new_ver, from->log_name); |
|
int new_hops = ni->hop_count + 1; |
|
if (new_hops <= MAX_HOPS) { |
|
uint64_t hop_list[MAX_HOPS]; |
|
const uint8_t* hop_src = (const uint8_t*)ni + sizeof(struct TOPOMSG_NODE) + dyn - ni->hop_count * 8; |
|
memcpy(hop_list, hop_src, ni->hop_count * 8); |
|
hop_list[ni->hop_count] = from->peer_node_id; |
|
topo_group_remove_path(nodeinfo1, from); |
|
topo_group_add_path(nodeinfo1, from, hop_list, new_hops, nodeinfo1->cumulative_rtt); |
|
} |
|
return 0; |
|
} |
|
|
|
DEBUG_INFO(DEBUG_CATEGORY_BGP, "NODEINFO: node=%016llx ver=%d v4s=%d v4a=%d v6s=%d v6a=%d hops=%d from=%s", |
|
(unsigned long long)node_id, new_ver, ni->local_v4_sockets, ni->local_v4_addrs, |
|
ni->local_v6_sockets, ni->local_v6_addrs, ni->hop_count, from->log_name); |
|
|
|
struct ll_queue* paths = NULL; |
|
int is_new_node = 0; |
|
if (nodeinfo1) { |
|
topo_nodeq_free_group_fields(group->instance->topo_groups, nodeinfo1); |
|
paths = nodeinfo1->paths; nodeinfo1->paths = NULL; |
|
} else { is_new_node = 1; } |
|
|
|
const uint8_t* ser_data = data + 2; |
|
size_t ser_len = len - 2; |
|
struct TOPO_NODE* new_ni = NULL; |
|
struct TOPO_NODESUBNETS* new_subnets = NULL; |
|
struct TOPOMSG_TRANZIT* new_tranzit = NULL; uint8_t new_tranzit_count = 0; |
|
uint64_t* new_hop_list = NULL; uint8_t new_hop_count = 0; |
|
uint16_t incoming_cumulative_rtt = 0; |
|
if (topo_node_deserialize(group, ser_data, ser_len, &new_ni, &new_subnets, &new_tranzit, &new_tranzit_count, &new_hop_list, &new_hop_count, &incoming_cumulative_rtt) != 0) { |
|
DEBUG_ERROR(DEBUG_CATEGORY_DEBUG, "NODEINFO deserialize failed from %s nid=%016llx", from->log_name, (unsigned long long)node_id); |
|
if (nodeinfo1) { queue_remove_data(group->nodes, &nodeinfo1->ll); queue_free(paths); queue_entry_free(&nodeinfo1->ll); } |
|
return -1; |
|
} |
|
DEBUG_DEBUG(DEBUG_CATEGORY_DEBUG, "NODEINFO deserialized: nid=%016llx new_ver=%d flags=0x%02x ed=%016llx", |
|
(unsigned long long)node_id, new_ni->ver, new_ni->flags, *(uint64_t*)new_ni->ed25519_public_key); |
|
|
|
/* verify Ed25519 self-signature: Ed25519_pubkey must sign X25519_pubkey */ |
|
{ |
|
uint64_t ekchk; memcpy(&ekchk, new_ni->ed25519_public_key, 8); |
|
if (ekchk == 0) { |
|
DEBUG_DEBUG(DEBUG_CATEGORY_BGP, "NODEINFO x25519_self_sig SKIP (ed25519 key zero) node=%016llx from=%s", |
|
(unsigned long long)node_id, from->log_name); |
|
} else if (sc_ed25519_verify(new_ni->ed25519_public_key, new_ni->public_key, SC_PUBKEY_SIZE, new_ni->x25519_self_sig) != SC_OK) { |
|
DEBUG_ERROR(DEBUG_CATEGORY_BGP, "NODEINFO x25519_self_sig VERIFY FAIL node=%016llx ed_pubkey=%016llx... from=%s — rejecting as forgery", |
|
(unsigned long long)node_id, ekchk, from->log_name); |
|
topo_node_destroy(group->instance->topo_groups, new_ni); |
|
u_free(new_subnets); u_free(new_tranzit); u_free(new_hop_list); |
|
if (nodeinfo1) { queue_remove_data(group->nodes, &nodeinfo1->ll); queue_free(paths); queue_entry_free(&nodeinfo1->ll); } |
|
return -1; |
|
} else { |
|
DEBUG_DEBUG(DEBUG_CATEGORY_BGP, "NODEINFO x25519_self_sig OK node=%016llx ed_pubkey=%016llx... from=%s", |
|
(unsigned long long)node_id, ekchk, from->log_name); |
|
} |
|
} |
|
|
|
if (nodeinfo1) { |
|
{ struct TOPO_NODE* stored = topo_node_registry_store(group->instance->topo_groups, new_ni); |
|
DEBUG_DEBUG(DEBUG_CATEGORY_DEBUG, "NODEINFO update: stored=%p new_ni=%p same=%d nid=%016llx", |
|
(void*)stored, (void*)new_ni, stored == new_ni, (unsigned long long)(stored ? stored->node_id : 0)); |
|
if (stored != new_ni) new_ni = stored; } |
|
new_ni->group_id = group->group_id; |
|
new_ni->flags = pkt->node.flags; |
|
nodeinfo1->node_id = new_ni ? new_ni->node_id : 0; |
|
nodeinfo1->subnets = new_subnets; |
|
nodeinfo1->cumulative_rtt = incoming_cumulative_rtt; |
|
u_free(nodeinfo1->tranzit_data); nodeinfo1->tranzit_data = new_tranzit; |
|
nodeinfo1->tranzit_count = new_tranzit_count; |
|
u_free(nodeinfo1->hop_list); nodeinfo1->hop_list = new_hop_list; |
|
nodeinfo1->hop_count = new_hop_count; |
|
} else { |
|
struct ll_entry* qe = queue_entry_new(sizeof(struct TOPO_GROUP_NODE)); |
|
if (!qe) { topo_node_destroy(group->instance->topo_groups, new_ni); u_free(new_tranzit); u_free(new_hop_list); u_free(new_subnets); return -1; } |
|
nodeinfo1 = (struct TOPO_GROUP_NODE*)qe; |
|
memset((uint8_t*)nodeinfo1 + sizeof(struct ll_entry), 0, sizeof(*nodeinfo1) - sizeof(struct ll_entry)); |
|
{ struct TOPO_NODE* stored = topo_node_registry_store(group->instance->topo_groups, new_ni); |
|
DEBUG_DEBUG(DEBUG_CATEGORY_DEBUG, "NODEINFO new: stored=%p new_ni=%p same=%d nid=%016llx", |
|
(void*)stored, (void*)new_ni, stored == new_ni, (unsigned long long)(stored ? stored->node_id : 0)); |
|
if (stored != new_ni) new_ni = stored; } |
|
new_ni->group_id = group->group_id; |
|
new_ni->flags = pkt->node.flags; |
|
nodeinfo1->node_id = new_ni ? new_ni->node_id : 0; |
|
nodeinfo1->subnets = new_subnets; |
|
nodeinfo1->cumulative_rtt = incoming_cumulative_rtt; |
|
nodeinfo1->tranzit_data = new_tranzit; nodeinfo1->tranzit_count = new_tranzit_count; |
|
nodeinfo1->hop_list = new_hop_list; nodeinfo1->hop_count = new_hop_count; |
|
nodeinfo1->connectivity.probe_status = PROBE_STATUS_NONE; |
|
nodeinfo1->connectivity.interface_status = PROBE_RESULT_UNKNOWN; |
|
nodeinfo1->connectivity.nat_status = PROBE_RESULT_UNKNOWN; |
|
nodeinfo1->connectivity.real_status = PROBE_RESULT_UNKNOWN; |
|
nodeinfo1->connectivity.ping_req_time = 0; |
|
queue_data_put_with_index(group->nodes, &nodeinfo1->ll); |
|
} |
|
|
|
nodeinfo1->paths = paths; nodeinfo1->last_ver = new_ver; |
|
|
|
uint64_t hop_list[MAX_HOPS]; |
|
if (nodeinfo1->hop_count > 0) memcpy(hop_list, nodeinfo1->hop_list, nodeinfo1->hop_count * 8); |
|
hop_list[nodeinfo1->hop_count] = from->peer_node_id; |
|
nodeinfo1->hop_count++; |
|
|
|
u_free(nodeinfo1->hop_list); |
|
nodeinfo1->hop_list = u_malloc(nodeinfo1->hop_count * 8); |
|
memcpy(nodeinfo1->hop_list, hop_list, nodeinfo1->hop_count * 8); |
|
|
|
topo_group_remove_path_by_hop(nodeinfo1, from->peer_node_id); |
|
topo_group_add_path(nodeinfo1, from, hop_list, nodeinfo1->hop_count, incoming_cumulative_rtt); |
|
|
|
if (group->group_type != TOPO_GROUP_TYPE_CHAT && group->instance->rt) route_insert(group->instance->rt, nodeinfo1); |
|
if (node_id != group->instance->node_id) { |
|
sqlite3* sdb = group->instance->topo_sqlite_db; |
|
if (sdb) { |
|
topo_node_sqlite_node_put(sdb, group->instance->topo_groups, nodeinfo1, ntp_time_get_seconds(group->instance)); |
|
if (group->group_type == TOPO_GROUP_TYPE_CHAT && group->channel_id[0]) { |
|
topo_node_sqlite_member_put(sdb, group->channel_id, node_id, NULL, 0, NULL, 0, NULL, NULL, NULL, NULL, NULL, NULL, 0); |
|
DEBUG_DEBUG(DEBUG_CATEGORY_DEBUG, "SQLite member_put: ch=%s node=%016llx", group->channel_id, (unsigned long long)node_id); |
|
} |
|
topo_node_sqlite_nodeinfo_updated(sdb, node_id); |
|
} |
|
} |
|
if (group->instance->control_srv) control_server_notify_node_change(group->instance->control_srv, nodeinfo1); |
|
|
|
DEBUG_DEBUG(DEBUG_CATEGORY_DEBUG, "node_updated_cb: node=%016llx has_cb=%d ch=%s type=%d", (unsigned long long)node_id, group->instance->topo_groups->node_updated_cb ? 1 : 0, group->channel_id, group->group_type); |
|
if (group->instance->topo_groups->node_updated_cb) |
|
group->instance->topo_groups->node_updated_cb(group->instance, node_id, ni->public_key, ni->ed25519_public_key); |
|
|
|
{ /* fire BGP node event callbacks */ |
|
int ev = is_new_node ? TOPO_NODE_EVENT_NEW : TOPO_NODE_EVENT_UPDATE; |
|
struct topo_node_cbk_entry* c = group->node_cbks; |
|
while (c) { c->fn(group, node_id, ev, c->arg); c = c->next; } |
|
} |
|
|
|
int hop_count = nodeinfo1->hop_count; |
|
struct ll_entry* se = group->senders_list ? group->senders_list->head : NULL; |
|
while (se) { |
|
struct TOPO_GROUP_CONN_ITEM* item = (struct TOPO_GROUP_CONN_ITEM*)se->data; |
|
if (item->conn) { |
|
uint64_t id = item->conn->peer_node_id; |
|
if (id != node_id) { |
|
int found = 0; for (int i = 0; i < hop_count; i++) if (hop_list[i] == id) found = 1; |
|
if (found == 0) { |
|
uint16_t out_rtt = (uint16_t)((uint32_t)item->conn->rtt_last + incoming_cumulative_rtt); |
|
topo_group_send_nodeinfo(group, nodeinfo1, item->conn, out_rtt); |
|
} |
|
} |
|
} |
|
se = se->next; |
|
} |
|
|
|
int socks_changed = 1; |
|
if (!is_new_node) socks_changed = 1; |
|
|
|
if (node_id != group->instance->node_id && (is_new_node || socks_changed)) { |
|
struct TOPO_NODE* tni = topo_node_registry_find(group->instance->topo_groups, nodeinfo1->node_id); |
|
int v4a = tni ? topo_list_count((struct _topo_head*)tni->v4_addrs) : 0; |
|
int v6a = tni ? topo_list_count((struct _topo_head*)tni->v6_addrs) : 0; |
|
if (v4a > 0 || v6a > 0) route_connectivity_probe_node(group->instance, nodeinfo1); |
|
} |
|
|
|
return 0; |
|
} |
|
|
|
int topo_group_process_withdraw(struct TOPO_GROUP* group, struct ETCP_CONN* sender, const uint8_t* data, size_t len) { |
|
if (!group || len < sizeof(struct TOPOMSG_WITHDRAW_PKT)) return -1; |
|
struct TOPOMSG_WITHDRAW_PKT* wp = (struct TOPOMSG_WITHDRAW_PKT*)data; |
|
uint64_t node_id = wp->node_id, wd_source = wp->wd_source; |
|
|
|
struct TOPO_GROUP_NODE* nq = topo_node_find_by_id(group, node_id); |
|
if (!nq) { DEBUG_INFO(DEBUG_CATEGORY_BGP, "node not found"); return 0; } |
|
int ret = topo_group_remove_path_by_hop(nq, wd_source); |
|
if (ret == 1) { |
|
if (group->group_type != TOPO_GROUP_TYPE_CHAT && group->instance && group->instance->rt) route_delete(group->instance->rt, nq); |
|
if (group->instance->control_srv) control_server_notify_node_removed(group->instance->control_srv, node_id); |
|
nq->dirty = 1; |
|
route_connectivity_cancel_node(group->instance, nq); |
|
if (nq->paths) { queue_free(nq->paths); nq->paths = NULL; } |
|
topo_nodeq_free_group_fields(group->instance->topo_groups, nq); |
|
struct ll_entry* entry = queue_find_data_by_index(group->nodes, &node_id); |
|
if (entry) { queue_remove_data(group->nodes, entry); queue_entry_free(entry); } |
|
topo_group_broadcast_withdraw(group, node_id, wd_source, sender); |
|
if (group->group_type == TOPO_GROUP_TYPE_CHAT && group->instance->topo_sqlite_db) |
|
topo_node_sqlite_member_del(group->instance->topo_sqlite_db, group->channel_id, node_id); |
|
{ /* fire BGP REMOVE callback */ |
|
struct topo_node_cbk_entry* c = group->node_cbks; |
|
while (c) { c->fn(group, node_id, TOPO_NODE_EVENT_REMOVE, c->arg); c = c->next; } |
|
} |
|
} |
|
return 0; |
|
} |
|
|
|
void topo_group_send_nodeinfo(struct TOPO_GROUP* group, struct TOPO_GROUP_NODE* node, struct ETCP_CONN* conn, uint16_t cumulative_rtt) { |
|
if (!group || !node || !conn) return; |
|
|
|
size_t max_sz = sizeof(struct TOPOMSG_NODEINFO_PKT) + 4096; |
|
uint8_t* p = u_malloc(max_sz); |
|
if (!p) return; |
|
|
|
p[0] = ETCP_ID_TOPO_ENTRY; p[1] = TOPO_SUBCMD_NODEINFO; |
|
|
|
struct TOPO_NODE* sni = topo_node_registry_find(group->instance->topo_groups, node->node_id); |
|
if (!sni) { DEBUG_ERROR(DEBUG_CATEGORY_DEBUG, "send_nodeinfo: node %016llx NOT in registry — skip forward to %s", (unsigned long long)node->node_id, conn->log_name); u_free(p); return; } |
|
DEBUG_DEBUG(DEBUG_CATEGORY_DEBUG, "send_nodeinfo: node %016llx ver=%d grp=%016llx to conn=%s", (unsigned long long)node->node_id, sni->ver, (unsigned long long)sni->group_id, conn->log_name); |
|
int ser_len = topo_node_serialize(sni, node, p + 2, max_sz - 2, cumulative_rtt); |
|
if (ser_len < 0) { DEBUG_ERROR(DEBUG_CATEGORY_DEBUG, "send_nodeinfo: serialize failed for node %016llx", (unsigned long long)node->node_id); u_free(p); return; } |
|
|
|
struct ll_entry* e = queue_entry_new(0); |
|
if (!e) { u_free(p); return; } |
|
e->dgram = p; e->len = (size_t)ser_len + 2; |
|
|
|
if (etcp_send(conn, e) != 0) { DEBUG_ERROR(DEBUG_CATEGORY_DEBUG, "send_nodeinfo: etcp_send FAILED for node %016llx to %s", (unsigned long long)node->node_id, conn->log_name); u_free(p); queue_entry_free(e); } |
|
} |
|
|
|
static void topo_group_add_to_senders(struct TOPO_GROUP* group, struct ETCP_CONN* conn) { |
|
if (!group || !conn || !group->senders_list) return; |
|
bool already = false; |
|
struct ll_entry* e = group->senders_list->head; |
|
while (e) { if (((struct TOPO_GROUP_CONN_ITEM*)e->data)->conn == conn) { already = true; break; } e = e->next; } |
|
if (!already) { |
|
struct ll_entry* item_entry = queue_entry_new(sizeof(struct TOPO_GROUP_CONN_ITEM)); |
|
if (item_entry) { ((struct TOPO_GROUP_CONN_ITEM*)item_entry->data)->conn = conn; queue_data_put(group->senders_list, item_entry); } |
|
} |
|
} |
|
|
|
static bool topo_group_should_send_to(const struct TOPO_GROUP_NODE* nq, uint64_t target_id) { |
|
if (!nq || !nq->paths) return false; |
|
if (target_id == nq->node_id) return false; |
|
struct ll_entry* e = nq->paths->head; |
|
while (e) { |
|
struct TOPO_NODEPATH* path = (struct TOPO_NODEPATH*)e; |
|
uint64_t* hop = (uint64_t*)((uint8_t*)path + sizeof(struct TOPO_NODEPATH)); |
|
bool has_id = false; for (uint8_t i = 0; i < path->hop_count; i++) if (hop[i] == target_id) { has_id = true; break; } |
|
if (!has_id) return true; |
|
e = e->next; |
|
} |
|
return false; |
|
} |
|
|
|
static void topo_group_send_full_table(struct TOPO_GROUP* group, struct ETCP_CONN* conn) { |
|
if (!group || !conn) return; |
|
uint64_t target = conn->peer_node_id; |
|
struct ll_entry* e = group->nodes ? group->nodes->head : NULL; |
|
while (e) { |
|
struct TOPO_GROUP_NODE* nq = (struct TOPO_GROUP_NODE*)e; |
|
if (nq->hop_count == 0) { e = e->next; continue; } |
|
if (!nq->paths) { DEBUG_WARN(DEBUG_CATEGORY_BGP, "node has no paths"); e = e->next; continue; } |
|
if (topo_group_should_send_to(nq, target)) topo_group_send_nodeinfo(group, nq, conn, (uint16_t)((uint32_t)conn->rtt_last + nq->cumulative_rtt)); |
|
e = e->next; |
|
} |
|
} |
|
|
|
static void topo_group_handle_request_table(struct TOPO_GROUP* group, struct ETCP_CONN* conn) { |
|
if (!group || !conn) return; |
|
topo_group_send_nodeinfo(group, group->local_node, conn, 0); |
|
topo_group_send_full_table(group, conn); |
|
topo_group_add_to_senders(group, conn); |
|
topo_group_send_table_complete(group, conn); |
|
} |
|
|
|
/* ── BGP node event callbacks ── */ |
|
|
|
void topo_group_add_node_cbk(struct TOPO_GROUP* group, topo_node_event_fn fn, void* arg) { |
|
if (!group || !fn) return; |
|
struct topo_node_cbk_entry* e = u_malloc(sizeof(*e)); |
|
if (!e) return; |
|
e->fn = fn; e->arg = arg; |
|
e->next = group->node_cbks; |
|
group->node_cbks = e; |
|
} |
|
|
|
void topo_group_remove_node_cbk(struct TOPO_GROUP* group, topo_node_event_fn fn, void* arg) { |
|
if (!group || !fn) return; |
|
struct topo_node_cbk_entry** p = &group->node_cbks; |
|
while (*p) { |
|
if ((*p)->fn == fn && (*p)->arg == arg) { |
|
struct topo_node_cbk_entry* r = *p; |
|
*p = r->next; u_free(r); |
|
return; |
|
} |
|
p = &(*p)->next; |
|
} |
|
} |
|
|
|
void topo_group_send_withdraw(struct TOPO_GROUP* group, uint64_t node_id) { |
|
if (!group) return; |
|
DEBUG_INFO(DEBUG_CATEGORY_BGP, "node=%016llx", (unsigned long long)node_id); |
|
topo_group_broadcast_withdraw(group, node_id, group->instance->node_id, NULL); |
|
}
|
|
|