Browse Source
- New files: transport_layer/socket_monitor.c/h — socket state monitoring with per-instance ctx - etcp_api: expose get_latency/1s_pkt_count measuring, add etcp_hashrnd for tests - utun_instance: store socket_monitor per instance, add is_socket_ok check - topo_node: detect IPv4 addresses with port from listening sockets - topo_group: use node_name for display in routes - Makefile.am: add socket_monitor.c to buildtopo_upd
evgeny
2 months ago
10 changed files with 420 additions and 0 deletions
@ -0,0 +1,261 @@
|
||||
#include "socket_monitor.h" |
||||
#include "etcp_connections.h" |
||||
#include "etcp_api.h" |
||||
#include "../lib/socket_compat.h" |
||||
#include "../lib/platform_compat.h" |
||||
#include "../lib/getmyip.h" |
||||
#include "../lib/u_async.h" |
||||
#include "../lib/debug_config.h" |
||||
#include "../lib/mem.h" |
||||
#include "utun_instance.h" |
||||
#include <stdlib.h> |
||||
#include <string.h> |
||||
#include <unistd.h> |
||||
#include <errno.h> |
||||
|
||||
#ifndef _WIN32 |
||||
#include <linux/netlink.h> |
||||
#include <linux/rtnetlink.h> |
||||
#include <net/if.h> |
||||
#endif |
||||
|
||||
struct SOCKET_MONITOR { |
||||
struct UTUN_INSTANCE* instance; |
||||
socket_t nl_sock; |
||||
void* uasync_handle; |
||||
int seq; |
||||
uint32_t addr_changes; |
||||
uint32_t link_changes; |
||||
uint32_t route_changes; |
||||
uint32_t socket_updates; |
||||
}; |
||||
|
||||
static void socket_monitor_read_cb(int fd, void* arg); |
||||
|
||||
#ifndef _WIN32 |
||||
static void socket_monitor_update_if_addr(struct ETCP_SOCKET* es) { |
||||
if (es->local_addr.ss_family == AF_INET) { |
||||
if (es->netif_index > 0) { |
||||
uint32_t if_ip = get_interface_ip_by_index(es->netif_index); |
||||
if (if_ip != 0) { |
||||
struct sockaddr_in* sin = (struct sockaddr_in*)&es->interface_addr; |
||||
sin->sin_family = AF_INET; |
||||
sin->sin_addr.s_addr = if_ip; |
||||
} |
||||
} else { |
||||
struct sockaddr_in* sin = (struct sockaddr_in*)&es->local_addr; |
||||
if (sin->sin_addr.s_addr == 0) { |
||||
struct sockaddr_storage remote; |
||||
memset(&remote, 0, sizeof(remote)); |
||||
struct sockaddr_in* rem_sin = (struct sockaddr_in*)&remote; |
||||
rem_sin->sin_family = AF_INET; |
||||
rem_sin->sin_port = htons(53); |
||||
inet_pton(AF_INET, "8.8.8.8", &rem_sin->sin_addr); |
||||
get_outgoing_local_ip(&es->local_addr, &remote, &es->interface_addr); |
||||
} |
||||
} |
||||
} else if (es->local_addr.ss_family == AF_INET6) { |
||||
if (es->netif_index > 0) { |
||||
uint8_t v6addr[16]; |
||||
int got = get_interface_ipv6_addr_nl(es->netif_index, 1, v6addr); |
||||
if (got != 0) got = get_interface_ipv6_addr_nl(es->netif_index, 0, v6addr); |
||||
if (got == 0) { |
||||
struct sockaddr_in6* sin6 = (struct sockaddr_in6*)&es->interface_addr; |
||||
sin6->sin6_family = AF_INET6; |
||||
memcpy(&sin6->sin6_addr, v6addr, 16); |
||||
} |
||||
} else { |
||||
struct sockaddr_in6* sin6 = (struct sockaddr_in6*)&es->local_addr; |
||||
if (memcmp(&sin6->sin6_addr, &in6addr_any, 16) == 0) { |
||||
uint16_t v6if = get_default_route_netif_index(AF_INET6); |
||||
if (v6if > 0) { |
||||
uint8_t v6addr[16]; |
||||
int got = get_interface_ipv6_addr_nl(v6if, 1, v6addr); |
||||
if (got != 0) got = get_interface_ipv6_addr_nl(v6if, 0, v6addr); |
||||
if (got == 0) { |
||||
struct sockaddr_in6* if6 = (struct sockaddr_in6*)&es->interface_addr; |
||||
if6->sin6_family = AF_INET6; |
||||
memcpy(&if6->sin6_addr, v6addr, 16); |
||||
} |
||||
} |
||||
if (es->interface_addr.ss_family == 0) { |
||||
struct sockaddr_storage remote; |
||||
memset(&remote, 0, sizeof(remote)); |
||||
struct sockaddr_in6* rem_sin6 = (struct sockaddr_in6*)&remote; |
||||
rem_sin6->sin6_family = AF_INET6; |
||||
rem_sin6->sin6_port = htons(53); |
||||
inet_pton(AF_INET6, "2001:4860:4860::8888", &rem_sin6->sin6_addr); |
||||
get_outgoing_local_ip(&es->local_addr, &remote, &es->interface_addr); |
||||
} |
||||
} |
||||
} |
||||
} |
||||
if (es->interface_addr.ss_family == es->local_addr.ss_family && es->local_addr.ss_family != 0) { |
||||
if (es->local_addr.ss_family == AF_INET) { |
||||
struct sockaddr_in* sin_cfg = (struct sockaddr_in*)&es->local_addr; |
||||
struct sockaddr_in* sin_if = (struct sockaddr_in*)&es->interface_addr; |
||||
sin_if->sin_port = sin_cfg->sin_port; |
||||
} else if (es->local_addr.ss_family == AF_INET6) { |
||||
struct sockaddr_in6* sin6_cfg = (struct sockaddr_in6*)&es->local_addr; |
||||
struct sockaddr_in6* sin6_if = (struct sockaddr_in6*)&es->interface_addr; |
||||
sin6_if->sin6_port = sin6_cfg->sin6_port; |
||||
} |
||||
} |
||||
} |
||||
|
||||
static void socket_monitor_check_and_fire(struct SOCKET_MONITOR* sm, struct ETCP_SOCKET* es, uint32_t ifindex) { |
||||
if (es->netif_index != ifindex) return; |
||||
struct sockaddr_storage old_if_addr = es->interface_addr; |
||||
socket_monitor_update_if_addr(es); |
||||
if (memcmp(&old_if_addr, &es->interface_addr, sizeof(old_if_addr)) != 0) { |
||||
sm->socket_updates++; |
||||
DEBUG_INFO(DEBUG_CATEGORY_SOCKET, "socket %s interface_addr changed: %s -> %s", |
||||
es->name, sockaddr_storage_to_str(&old_if_addr).str, sockaddr_storage_to_str(&es->interface_addr).str); |
||||
etcp_socket_cbk_fire(es, ETCP_SOCKET_EVENT_ADDR_CHANGED); |
||||
} |
||||
} |
||||
|
||||
static void socket_monitor_handle_addr(struct SOCKET_MONITOR* sm, struct ifaddrmsg* ifa, int msg_type) { |
||||
sm->addr_changes++; |
||||
DEBUG_DEBUG(DEBUG_CATEGORY_SOCKET, "addr event: type=%s ifindex=%u family=%d", |
||||
msg_type == RTM_NEWADDR ? "NEW" : "DEL", ifa->ifa_index, ifa->ifa_family); |
||||
struct ETCP_SOCKET* es = sm->instance->etcp_sockets; |
||||
while (es) { socket_monitor_check_and_fire(sm, es, ifa->ifa_index); es = es->next; } |
||||
} |
||||
|
||||
static void socket_monitor_handle_link(struct SOCKET_MONITOR* sm, struct ifinfomsg* ifi, int msg_type) { |
||||
sm->link_changes++; |
||||
int is_up = (ifi->ifi_flags & IFF_UP) != 0; |
||||
DEBUG_DEBUG(DEBUG_CATEGORY_SOCKET, "link event: type=%s ifindex=%u flags=0x%x up=%d", |
||||
msg_type == RTM_NEWLINK ? "NEW" : "DEL", ifi->ifi_index, ifi->ifi_flags, is_up); |
||||
struct ETCP_SOCKET* es = sm->instance->etcp_sockets; |
||||
while (es) { |
||||
if (es->netif_index == ifi->ifi_index) { |
||||
etcp_socket_cbk_fire(es, ETCP_SOCKET_EVENT_STATUS_CHANGED); |
||||
if (is_up) socket_monitor_check_and_fire(sm, es, ifi->ifi_index); |
||||
} |
||||
es = es->next; |
||||
} |
||||
} |
||||
|
||||
static void socket_monitor_handle_route(struct SOCKET_MONITOR* sm, struct rtmsg* rtm, int msg_type) { |
||||
if (rtm->rtm_dst_len != 0) return; |
||||
sm->route_changes++; |
||||
DEBUG_DEBUG(DEBUG_CATEGORY_SOCKET, "default route event: type=%s family=%d", |
||||
msg_type == RTM_NEWROUTE ? "NEW" : "DEL", rtm->rtm_family); |
||||
struct ETCP_SOCKET* es = sm->instance->etcp_sockets; |
||||
while (es) { |
||||
if (es->netif_index != 0 || es->local_addr.ss_family == 0) { es = es->next; continue; } |
||||
int is_any = 0; |
||||
if (es->local_addr.ss_family == AF_INET) { |
||||
struct sockaddr_in* sin = (struct sockaddr_in*)&es->local_addr; |
||||
is_any = (sin->sin_addr.s_addr == 0); |
||||
} else if (es->local_addr.ss_family == AF_INET6) { |
||||
struct sockaddr_in6* sin6 = (struct sockaddr_in6*)&es->local_addr; |
||||
is_any = (memcmp(&sin6->sin6_addr, &in6addr_any, 16) == 0); |
||||
} |
||||
if (is_any) { |
||||
struct sockaddr_storage old_if_addr = es->interface_addr; |
||||
socket_monitor_update_if_addr(es); |
||||
if (memcmp(&old_if_addr, &es->interface_addr, sizeof(old_if_addr)) != 0) { |
||||
sm->socket_updates++; |
||||
DEBUG_INFO(DEBUG_CATEGORY_SOCKET, "socket %s interface_addr changed via default route: %s -> %s", |
||||
es->name, sockaddr_storage_to_str(&old_if_addr).str, sockaddr_storage_to_str(&es->interface_addr).str); |
||||
etcp_socket_cbk_fire(es, ETCP_SOCKET_EVENT_ADDR_CHANGED); |
||||
} |
||||
} |
||||
es = es->next; |
||||
} |
||||
} |
||||
|
||||
static void socket_monitor_read_cb(int fd, void* arg) { |
||||
struct SOCKET_MONITOR* sm = (struct SOCKET_MONITOR*)arg; |
||||
char buf[8192]; |
||||
struct sockaddr_nl nladdr; |
||||
socklen_t nladdr_len = sizeof(nladdr); |
||||
for (;;) { |
||||
ssize_t n = recvfrom(fd, buf, sizeof(buf), MSG_DONTWAIT, (struct sockaddr*)&nladdr, &nladdr_len); |
||||
if (n < 0) { |
||||
if (errno == EAGAIN || errno == EWOULDBLOCK) break; |
||||
DEBUG_ERROR(DEBUG_CATEGORY_SOCKET, "netlink recvfrom fd=%d failed: %s", fd, strerror(errno)); |
||||
break; |
||||
} |
||||
if (n == 0) break; |
||||
size_t remaining = (size_t)n; |
||||
struct nlmsghdr* nlh = (struct nlmsghdr*)buf; |
||||
while (NLMSG_OK(nlh, remaining)) { |
||||
if (nlh->nlmsg_type == RTM_NEWADDR || nlh->nlmsg_type == RTM_DELADDR) { |
||||
socket_monitor_handle_addr(sm, (struct ifaddrmsg*)NLMSG_DATA(nlh), nlh->nlmsg_type); |
||||
} else if (nlh->nlmsg_type == RTM_NEWLINK || nlh->nlmsg_type == RTM_DELLINK) { |
||||
socket_monitor_handle_link(sm, (struct ifinfomsg*)NLMSG_DATA(nlh), nlh->nlmsg_type); |
||||
} else if (nlh->nlmsg_type == RTM_NEWROUTE || nlh->nlmsg_type == RTM_DELROUTE) { |
||||
socket_monitor_handle_route(sm, (struct rtmsg*)NLMSG_DATA(nlh), nlh->nlmsg_type); |
||||
} |
||||
if (nlh->nlmsg_type == NLMSG_DONE) break; |
||||
nlh = NLMSG_NEXT(nlh, remaining); |
||||
} |
||||
} |
||||
} |
||||
#endif |
||||
|
||||
int socket_monitor_init(struct UTUN_INSTANCE* instance) { |
||||
if (!instance) { DEBUG_ERROR(DEBUG_CATEGORY_SOCKET, "instance is NULL"); return -1; } |
||||
if (instance->socket_monitor) { DEBUG_ERROR(DEBUG_CATEGORY_SOCKET, "socket monitor already initialized"); return -1; } |
||||
DEBUG_INFO(DEBUG_CATEGORY_SOCKET, "initializing socket monitor"); |
||||
|
||||
#ifdef _WIN32 |
||||
DEBUG_INFO(DEBUG_CATEGORY_SOCKET, "socket monitor not available on Windows"); |
||||
return 0; |
||||
#else |
||||
struct SOCKET_MONITOR* sm = u_calloc(1, sizeof(struct SOCKET_MONITOR)); |
||||
if (!sm) { DEBUG_ERROR(DEBUG_CATEGORY_SOCKET, "alloc failed"); return -1; } |
||||
sm->instance = instance; |
||||
sm->nl_sock = socket(AF_NETLINK, SOCK_RAW, NETLINK_ROUTE); |
||||
if (sm->nl_sock < 0) { |
||||
DEBUG_ERROR(DEBUG_CATEGORY_SOCKET, "failed to create netlink socket: %s", strerror(errno)); |
||||
u_free(sm); |
||||
return -1; |
||||
} |
||||
struct sockaddr_nl nladdr; |
||||
memset(&nladdr, 0, sizeof(nladdr)); |
||||
nladdr.nl_family = AF_NETLINK; |
||||
nladdr.nl_groups = RTMGRP_LINK | RTMGRP_IPV4_IFADDR | RTMGRP_IPV6_IFADDR |
||||
| RTMGRP_IPV4_ROUTE | RTMGRP_IPV6_ROUTE; |
||||
if (bind(sm->nl_sock, (struct sockaddr*)&nladdr, sizeof(nladdr)) < 0) { |
||||
DEBUG_ERROR(DEBUG_CATEGORY_SOCKET, "failed to bind netlink socket: %s", strerror(errno)); |
||||
close(sm->nl_sock); |
||||
u_free(sm); |
||||
return -1; |
||||
} |
||||
if (socket_set_nonblocking(sm->nl_sock) != 0) { |
||||
DEBUG_WARN(DEBUG_CATEGORY_SOCKET, "failed to set netlink socket non-blocking: %s", strerror(errno)); |
||||
} |
||||
sm->uasync_handle = uasync_add_socket(instance->ua, sm->nl_sock, socket_monitor_read_cb, NULL, NULL, sm); |
||||
if (!sm->uasync_handle) { |
||||
DEBUG_ERROR(DEBUG_CATEGORY_SOCKET, "failed to register netlink socket fd=%d with uasync", sm->nl_sock); |
||||
close(sm->nl_sock); |
||||
u_free(sm); |
||||
return -1; |
||||
} |
||||
instance->socket_monitor = sm; |
||||
DEBUG_INFO(DEBUG_CATEGORY_SOCKET, "socket monitor initialized: nl_sock=%d", sm->nl_sock); |
||||
return 0; |
||||
#endif |
||||
} |
||||
|
||||
void socket_monitor_destroy(struct UTUN_INSTANCE* instance) { |
||||
if (!instance) return; |
||||
DEBUG_INFO(DEBUG_CATEGORY_SOCKET, "destroying socket monitor"); |
||||
#ifdef _WIN32 |
||||
return; |
||||
#else |
||||
struct SOCKET_MONITOR* sm = (struct SOCKET_MONITOR*)instance->socket_monitor; |
||||
if (!sm) return; |
||||
instance->socket_monitor = NULL; |
||||
uasync_remove_socket(instance->ua, sm->uasync_handle); |
||||
close(sm->nl_sock); |
||||
DEBUG_INFO(DEBUG_CATEGORY_SOCKET, "socket monitor destroyed: nl_sock=%d addr_changes=%u link_changes=%u route_changes=%u socket_updates=%u", |
||||
sm->nl_sock, sm->addr_changes, sm->link_changes, sm->route_changes, sm->socket_updates); |
||||
u_free(sm); |
||||
#endif |
||||
} |
||||
@ -0,0 +1,16 @@
|
||||
#ifndef SOCKET_MONITOR_H |
||||
#define SOCKET_MONITOR_H |
||||
|
||||
#ifdef __cplusplus |
||||
extern "C" { |
||||
#endif |
||||
|
||||
struct UTUN_INSTANCE; |
||||
|
||||
int socket_monitor_init(struct UTUN_INSTANCE* instance); |
||||
void socket_monitor_destroy(struct UTUN_INSTANCE* instance); |
||||
|
||||
#ifdef __cplusplus |
||||
} |
||||
#endif |
||||
#endif |
||||
Loading…
Reference in new issue