Browse Source

Merge branch 'master' of http://git.syqiq.tech/jeka/utun2

nodeinfo-routing-update
jeka 1 week ago
parent
commit
d3e211793c
  1. 4
      src/etcp_connections.c
  2. 22
      tools/proxy/Makefile
  3. 239
      tools/proxy/udp_proxy.c

4
src/etcp_connections.c

@ -723,8 +723,8 @@ static void link_stats_timer_cb(void* arg) {
link->stat_win[link->win_ptr].pkt_loss = (uint16_t)link->window_retransmissions; link->stat_win[link->win_ptr].pkt_loss = (uint16_t)link->window_retransmissions;
link->stat_win[link->win_ptr].pkt_transmitted = link->window_pkt_transmitted; link->stat_win[link->win_ptr].pkt_transmitted = link->window_pkt_transmitted;
DEBUG_DEBUG(DEBUG_CATEGORY_ETCP, "[%s] stats window updated (win_timebase=%u us, rtt=%u, retrans=%u, transmitted=%u)", // DEBUG_DEBUG(DEBUG_CATEGORY_ETCP, "[%s] stats window updated (win_timebase=%u us, rtt=%u, retrans=%u, transmitted=%u)",
link->etcp->log_name, link->win_timebase, link->rtt_avg10, link->window_retransmissions, link->window_pkt_transmitted); // link->etcp->log_name, link->win_timebase, link->rtt_avg10, link->window_retransmissions, link->window_pkt_transmitted);
// 2. Переходим к следующему слоту // 2. Переходим к следующему слоту
link->win_ptr = (link->win_ptr + 1) % 32; link->win_ptr = (link->win_ptr + 1) % 32;

22
tools/proxy/Makefile

@ -0,0 +1,22 @@
# Makefile for udp_proxy - UDP network emulator proxy
CC = gcc
CFLAGS = -O2 -Wall -Wextra -std=gnu11 -I../../lib -I../../tinycrypt/lib/include
LDFLAGS = -L../../lib -luasync -lpthread
TARGET = udp_proxy
SRC = udp_proxy.c
.PHONY: all clean
all: $(TARGET)
$(TARGET): $(SRC)
$(CC) $(CFLAGS) $(SRC) -o $@ $(LDFLAGS)
@echo "✅ UDP proxy built: $(TARGET)"
clean:
rm -f $(TARGET)
@echo "🧹 Cleaned"
debug:
$(CC) $(CFLAGS) -g $(SRC) -o $(TARGET) $(LDFLAGS)

239
tools/proxy/udp_proxy.c

@ -0,0 +1,239 @@
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <time.h>
#include <errno.h>
#include <u_async.h>
#include <socket_compat.h>
#include <debug_config.h>
#include <mem.h>
#define BUFFER_SIZE 65536
typedef struct udp_flow {
struct sockaddr_in client_addr;
socket_t backend_sock;
void* socket_id;
struct udp_flow* next;
} udp_flow_t;
typedef struct {
uasync_t* ua;
socket_t listen_sock;
void* listen_id;
struct sockaddr_in target_addr;
char listen_ip[64];
int listen_port;
int loss_forward;
int loss_back;
int delay_min_fwd;
int delay_max_fwd;
int delay_min_back;
int delay_max_back;
udp_flow_t* flows;
char buffer[BUFFER_SIZE];
} proxy_context_t;
static void client_read_cb(socket_t sock, void* arg);
static void backend_read_cb(socket_t sock, void* arg);
static udp_flow_t* find_flow_by_client(proxy_context_t* ctx, struct sockaddr_in* client) {
for (udp_flow_t* f = ctx->flows; f; f = f->next) {
if (memcmp(&f->client_addr, client, sizeof(*client)) == 0) return f;
}
return NULL;
}
static udp_flow_t* find_flow_by_sock(proxy_context_t* ctx, socket_t sock) {
for (udp_flow_t* f = ctx->flows; f; f = f->next) {
if (f->backend_sock == sock) return f;
}
return NULL;
}
static udp_flow_t* create_flow(proxy_context_t* ctx, struct sockaddr_in* client) {
udp_flow_t* flow = (udp_flow_t*)u_malloc(sizeof(udp_flow_t));
if (!flow) return NULL;
memcpy(&flow->client_addr, client, sizeof(*client));
flow->backend_sock = socket_create_udp(AF_INET);
if (flow->backend_sock == SOCKET_INVALID) {
u_free(flow);
return NULL;
}
socket_set_nonblocking(flow->backend_sock);
socket_set_reuseaddr(flow->backend_sock, 1);
struct sockaddr_in baddr = {0};
baddr.sin_family = AF_INET;
baddr.sin_addr.s_addr = INADDR_ANY;
baddr.sin_port = 0;
bind((int)flow->backend_sock, (struct sockaddr*)&baddr, sizeof(baddr));
flow->socket_id = uasync_add_socket_t(ctx->ua, flow->backend_sock, backend_read_cb, NULL, NULL, ctx);
if (!flow->socket_id) {
socket_close_wrapper(flow->backend_sock);
u_free(flow);
return NULL;
}
flow->next = ctx->flows;
ctx->flows = flow;
DEBUG_INFO(DEBUG_CATEGORY_SOCKET, "New flow: client %s:%d -> backend socket %d",
inet_ntoa(client->sin_addr), ntohs(client->sin_port), (int)flow->backend_sock);
return flow;
}
static void client_read_cb(socket_t sock, void* arg) {
proxy_context_t* ctx = (proxy_context_t*)arg;
struct sockaddr_in client;
socklen_t alen = sizeof(client);
ssize_t n = socket_recvfrom(sock, ctx->buffer, BUFFER_SIZE, (struct sockaddr*)&client, &alen);
if (n <= 0) return;
udp_flow_t* flow = find_flow_by_client(ctx, &client);
if (!flow) {
flow = create_flow(ctx, &client);
if (!flow) return;
}
if (ctx->loss_forward > 0 && (rand() % 100 < ctx->loss_forward)) {
DEBUG_WARN(DEBUG_CATEGORY_SOCKET, "Forward packet dropped");
return;
}
int delay = ctx->delay_min_fwd;
if (ctx->delay_max_fwd > ctx->delay_min_fwd) delay += rand() % (ctx->delay_max_fwd - ctx->delay_min_fwd + 1);
if (delay > 0) {
// simple immediate for now, full timeout later
usleep(delay * 1000); // temporary
}
socket_sendto(flow->backend_sock, ctx->buffer, n, (struct sockaddr*)&ctx->target_addr, sizeof(ctx->target_addr));
}
static void backend_read_cb(socket_t sock, void* arg) {
proxy_context_t* ctx = (proxy_context_t*)arg;
udp_flow_t* flow = find_flow_by_sock(ctx, sock);
if (!flow) return;
struct sockaddr_in from;
socklen_t alen = sizeof(from);
ssize_t n = socket_recvfrom(sock, ctx->buffer, BUFFER_SIZE, (struct sockaddr*)&from, &alen);
if (n <= 0) return;
if (ctx->loss_back > 0 && (rand() % 100 < ctx->loss_back)) {
DEBUG_WARN(DEBUG_CATEGORY_SOCKET, "Back packet dropped");
return;
}
int delay = ctx->delay_min_back;
if (ctx->delay_max_back > ctx->delay_min_back) delay += rand() % (ctx->delay_max_back - ctx->delay_min_back + 1);
if (delay > 0) {
usleep(delay * 1000); // temporary
}
socket_sendto(ctx->listen_sock, ctx->buffer, n, (struct sockaddr*)&flow->client_addr, sizeof(flow->client_addr));
}
static void print_usage(const char* prog) {
printf("UDP Proxy - network loss/delay emulator\n\n");
printf("Usage: %s --listen [IP:]PORT --target IP:PORT [options]\n\n", prog);
printf("Required:\n");
printf(" --listen [IP:]PORT Listen IP:port (default 0.0.0.0)\n");
printf(" --target IP:PORT Destination IP:port\n\n");
printf("Options:\n");
printf(" --loss-forward N Forward loss %% (0-100, default 0)\n");
printf(" --loss-back N Backward loss %% (0-100, default 0)\n");
printf(" --delay-forward A:B Forward delay ms range (default 0:0)\n");
printf(" --delay-back A:B Backward delay ms range (default 0:0)\n");
printf(" --help Show this help\n\n");
printf("Example: %s --listen 0.0.0.0:12345 --target 8.8.8.8:53 --loss-forward 5 --delay-forward 10:50\n", prog);
exit(0);
}
int main(int argc, char* argv[]) {
if (argc < 3) print_usage(argv[0]);
srand(time(NULL));
socket_platform_init();
proxy_context_t ctx = {0};
ctx.ua = uasync_create();
if (!ctx.ua) {
DEBUG_ERROR(DEBUG_CATEGORY_SOCKET, "uasync_create failed");
return 1;
}
ctx.loss_forward = 0; ctx.loss_back = 0;
ctx.delay_min_fwd = ctx.delay_max_fwd = 0;
ctx.delay_min_back = ctx.delay_max_back = 0;
for (int i = 1; i < argc; i++) {
if (strcmp(argv[i], "--help") == 0) print_usage(argv[0]);
if (strcmp(argv[i], "--listen") == 0 && i+1 < argc) {
char ip[64] = "0.0.0.0"; int p = 0;
const char* val = argv[++i];
if (strchr(val, ':')) {
sscanf(val, "%63[^:]:%d", ip, &p);
} else {
p = atoi(val);
}
struct sockaddr_in addr = {0};
addr.sin_family = AF_INET;
addr.sin_port = htons(p);
inet_aton(ip, &addr.sin_addr);
strncpy(ctx.listen_ip, ip, sizeof(ctx.listen_ip)-1);
ctx.listen_port = p;
ctx.listen_sock = socket_create_udp(AF_INET);
socket_set_nonblocking(ctx.listen_sock);
socket_set_reuseaddr(ctx.listen_sock, 1);
if (bind((int)ctx.listen_sock, (struct sockaddr*)&addr, sizeof(addr)) < 0) {
DEBUG_ERROR(DEBUG_CATEGORY_SOCKET, "bind listen failed on %s:%d", ip, p);
return 1;
}
ctx.listen_id = uasync_add_socket_t(ctx.ua, ctx.listen_sock, client_read_cb, NULL, NULL, &ctx);
} else if (strcmp(argv[i], "--target") == 0 && i+1 < argc) {
char ip[64] = {0}; int port = 0;
if (sscanf(argv[++i], "%63[^:]:%d", ip, &port) == 2) {
ctx.target_addr.sin_family = AF_INET;
ctx.target_addr.sin_port = htons(port);
inet_aton(ip, &ctx.target_addr.sin_addr);
}
} else if (strcmp(argv[i], "--loss-forward") == 0 && i+1 < argc) {
ctx.loss_forward = atoi(argv[++i]);
} else if (strcmp(argv[i], "--loss-back") == 0 && i+1 < argc) {
ctx.loss_back = atoi(argv[++i]);
} else if (strcmp(argv[i], "--delay-forward") == 0 && i+1 < argc) {
int mn=0,mx=0; sscanf(argv[++i], "%d:%d", &mn, &mx);
ctx.delay_min_fwd = mn; ctx.delay_max_fwd = mx > mn ? mx : mn;
} else if (strcmp(argv[i], "--delay-back") == 0 && i+1 < argc) {
int mn=0,mx=0; sscanf(argv[++i], "%d:%d", &mn, &mx);
ctx.delay_min_back = mn; ctx.delay_max_back = mx > mn ? mx : mn;
}
}
if (ctx.listen_sock == 0 || ctx.target_addr.sin_port == 0) {
fprintf(stderr, "Error: --listen and --target required\n");
print_usage(argv[0]);
}
printf("UDP Proxy: listen %s:%d → %s:%d | loss %d%%/%d%% | delay %d-%d/%d-%d ms\n",
ctx.listen_ip, ctx.listen_port,
inet_ntoa(ctx.target_addr.sin_addr), ntohs(ctx.target_addr.sin_port),
ctx.loss_forward, ctx.loss_back,
ctx.delay_min_fwd, ctx.delay_max_fwd, ctx.delay_min_back, ctx.delay_max_back);
uasync_mainloop(ctx.ua);
return 0;
}
Loading…
Cancel
Save