-
Notifications
You must be signed in to change notification settings - Fork 1.2k
/
udp_proxy_server.c
53 lines (47 loc) · 1.4 KB
/
udp_proxy_server.c
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
/*
* udp proxy server
*
* @build: make examples
* @udp_server: bin/udp_echo_server 1234
* @proxy_server: bin/udp_proxy_server 2222 127.0.0.1:1234
* @client: bin/nc -u 127.0.0.1 2222
* > hello
* < hello
*/
#include "hloop.h"
static char proxy_host[64] = "0.0.0.0";
static int proxy_port = 1080;
static char backend_host[64] = "127.0.0.1";
static int backend_port = 80;
// hloop_create_udp_server -> hio_setup_udp_upstream
int main(int argc, char** argv) {
if (argc < 3) {
printf("Usage: %s proxy_port backend_host:backend_port\n", argv[0]);
return -10;
}
proxy_port = atoi(argv[1]);
char* pos = strchr(argv[2], ':');
if (pos) {
int len = pos - argv[2];
if (len > 0) {
memcpy(backend_host, argv[2], len);
backend_host[len] = '\0';
}
backend_port = atoi(pos + 1);
} else {
strncpy(backend_host, argv[2], sizeof(backend_host));
}
printf("%s:%d proxy %s:%d\n", proxy_host, proxy_port, backend_host, backend_port);
hloop_t* loop = hloop_new(0);
hio_t* io = hloop_create_udp_server(loop, proxy_host, proxy_port);
if (io == NULL) {
return -20;
}
hio_t* upstream_io = hio_setup_udp_upstream(io, backend_host, backend_port);
if (upstream_io == NULL) {
return -30;
}
hloop_run(loop);
hloop_free(&loop);
return 0;
}