Browse Source

WiP

pull/14/head
Max Lv 11 years ago
parent
commit
b36b7980e3
3 changed files with 209 additions and 198 deletions
  1. 38
      src/cache.c
  2. 358
      src/udprelay.c
  3. 11
      src/udprelay.h

38
src/cache.c

@ -4,7 +4,6 @@
*/ */
#include <errno.h> #include <errno.h>
#include <pthread.h>
#include <stdlib.h> #include <stdlib.h>
#include "cache.h" #include "cache.h"
#include "uthash.h" #include "uthash.h"
@ -23,7 +22,6 @@ int cache_create(struct cache **dst, const size_t capacity,
void (*free_cb) (void *element)) void (*free_cb) (void *element))
{ {
struct cache *new = NULL; struct cache *new = NULL;
int rv;
if (!dst) if (!dst)
return EINVAL; return EINVAL;
@ -31,19 +29,12 @@ int cache_create(struct cache **dst, const size_t capacity,
if ((new = malloc(sizeof(*new))) == NULL) if ((new = malloc(sizeof(*new))) == NULL)
return ENOMEM; return ENOMEM;
if ((rv = pthread_rwlock_init(&(new->cache_lock), NULL)) != 0)
goto err_out;
new->max_entries = capacity; new->max_entries = capacity;
new->entries = NULL; new->entries = NULL;
new->free_cb = free_cb; new->free_cb = free_cb;
*dst = new; *dst = new;
return 0; return 0;
err_out:
if (new)
free(new);
return rv;
} }
/** Frees an allocated cache object /** Frees an allocated cache object
@ -59,15 +50,10 @@ err_out:
int cache_delete(struct cache *cache, int keep_data) int cache_delete(struct cache *cache, int keep_data)
{ {
struct cache_entry *entry, *tmp; struct cache_entry *entry, *tmp;
int rv;
if (!cache) if (!cache)
return EINVAL; return EINVAL;
rv = pthread_rwlock_wrlock(&(cache->cache_lock));
if (rv)
return rv;
if (keep_data) { if (keep_data) {
HASH_CLEAR(hh, cache->entries); HASH_CLEAR(hh, cache->entries);
} else { } else {
@ -78,8 +64,7 @@ int cache_delete(struct cache *cache, int keep_data)
free(entry); free(entry);
} }
} }
(void)pthread_rwlock_unlock(&(cache->cache_lock));
(void)pthread_rwlock_destroy(&(cache->cache_lock));
free(cache); free(cache);
cache = NULL; cache = NULL;
return 0; return 0;
@ -111,10 +96,6 @@ int cache_lookup(struct cache *cache, char *key, void *result)
if (!cache || !key || !result) if (!cache || !key || !result)
return EINVAL; return EINVAL;
rv = pthread_rwlock_wrlock(&(cache->cache_lock));
if (rv)
return rv;
HASH_FIND_STR(cache->entries, key, tmp); HASH_FIND_STR(cache->entries, key, tmp);
if (tmp) { if (tmp) {
size_t key_len = strnlen(tmp->key, KEY_MAX_LENGTH); size_t key_len = strnlen(tmp->key, KEY_MAX_LENGTH);
@ -124,8 +105,8 @@ int cache_lookup(struct cache *cache, char *key, void *result)
} else { } else {
*dirty_hack = result = NULL; *dirty_hack = result = NULL;
} }
rv = pthread_rwlock_unlock(&(cache->cache_lock));
return rv;
return 0;
} }
/** Inserts a given <key, value> pair into the cache /** Inserts a given <key, value> pair into the cache
@ -154,9 +135,6 @@ int cache_insert(struct cache *cache, char *key, void *data)
if ((entry = malloc(sizeof(*entry))) == NULL) if ((entry = malloc(sizeof(*entry))) == NULL)
return ENOMEM; return ENOMEM;
if ((rv = pthread_rwlock_wrlock(&(cache->cache_lock))) != 0)
goto err_out;
entry->key = key; entry->key = key;
entry->data = data; entry->data = data;
key_len = strnlen(entry->key, KEY_MAX_LENGTH); key_len = strnlen(entry->key, KEY_MAX_LENGTH);
@ -175,13 +153,5 @@ int cache_insert(struct cache *cache, char *key, void *data)
} }
} }
rv = pthread_rwlock_unlock(&(cache->cache_lock));
return rv;
err_out:
if (entry)
free(entry);
(void)pthread_rwlock_unlock(&(cache->cache_lock));
return rv;
return 0;
} }

358
src/udprelay.c

@ -105,23 +105,29 @@ static int parse_udprealy_header(const char* buf, const int buf_len,
// IP V4 // IP V4
size_t in_addr_len = sizeof(struct in_addr); size_t in_addr_len = sizeof(struct in_addr);
if (buf_len > in_addbuf_len_len) { if (buf_len > in_addbuf_len_len) {
inet_ntop(AF_INET, (const void *)(buf + offset),
host, INET_ADDRSTRLEN);
if (host != NULL) {
inet_ntop(AF_INET, (const void *)(buf + offset),
host, INET_ADDRSTRLEN);
}
offset += in_addbuf_len_len; offset += in_addbuf_len_len;
} }
} else if (atyp == 3) { } else if (atyp == 3) {
// Domain name // Domain name
uint8_t name_len = *(uint8_t *)(buf + offset); uint8_t name_len = *(uint8_t *)(buf + offset);
if (name_len < buf_len && name_len < 255 && name_len > 0) { if (name_len < buf_len && name_len < 255 && name_len > 0) {
memcpy(host, buf + offset + 1, name_len);
if (host != NULL) {
memcpy(host, buf + offset + 1, name_len);
}
offset += name_len + 1; offset += name_len + 1;
} }
} else if (atyp == 4) { } else if (atyp == 4) {
// IP V6 // IP V6
size_t in6_addbuf_len_len = sizeof(stbuf_lenuct in6_addbuf_len); size_t in6_addbuf_len_len = sizeof(stbuf_lenuct in6_addbuf_len);
if (buf_len > in6_addbuf_len_len) { if (buf_len > in6_addbuf_len_len) {
inet_ntop(AF_INET6, (const void*)(buf + offset),
host, INET6_ADDRSTRLEN);
if (host != NULL) {
inet_ntop(AF_INET6, (const void*)(buf + offset),
host, INET6_ADDRSTRLEN);
}
offset += in6_addr_len; offset += in6_addr_len;
} }
} }
@ -131,8 +137,9 @@ static int parse_udprealy_header(const char* buf, const int buf_len,
return 0; return 0;
} }
sprintf(port, "%d",
ntohs(*(uint16_t *)(buf + offset)));
if (port != NULL) {
sprintf(port, "%d", ntohs(*(uint16_t *)(buf + offset)));
}
offset += 2; offset += 2;
return offset; return offset;
@ -224,23 +231,73 @@ int create_server_socket(const char *host, const char *port) {
return server_sock; return server_sock;
} }
struct remote *send_to_remote(struct addrinfo *res, const char *iface) {
connect(sockfd, res->ai_addr, res->ai_addrlen);
struct remote_ctx *new_remote_ctx(int fd) {
struct remote_ctx *ctx = malloc(sizeof(struct remote_ctx));
memset(ctx, 0, sizeof(struct remote_ctx));
ctx->buf = malloc(BUF_SIZE);
ctx->server_ctx = NULL;
ctx->fd = fd;
ev_io_init(&ctx->io, remote_recv_cb, fd, EV_READ);
return ctx;
}
return remote;
struct server_ctx * new_server_ctx(int fd) {
struct server_ctx *ctx = malloc(sizeof(struct server_ctx));
memset(ctx, 0, sizeof(struct server_ctx));
ctx->buf = malloc(BUF_SIZE);
ctx->remote_ctx = NULL;
ctx->fd = fd;
ev_io_init(&ctx.io, server_recv_cb, fd, EV_READ);
return ctx;
} }
static void server_timeout_cb(EV_P_ ev_timer *watcher, int revents) {
struct server *server = (struct server *) (((void*)watcher)
- sizeof(ev_timer));
struct remote *remote = server->remote;
#ifdef UDPRELAY_REMOTE
struct query_ctx *new_query_ctx(asyncns_query_t *query,
const uint8_t *buf, const int buf_len) {
struct query_ctx *ctx = malloc(sizeof(struct query_ctx))
memset(ctx, 0, sizeof(struct query_ctx));
ctx->buf = malloc(buf_len);
ctx->buf_len = buf_len;
memcpy(ctx->buf, buf, buf_len);
ctx->query = query;
ev_timer_init(&ctx->watcher, query_resolve_cb, 0.2, 0.5);
return ctx;
}
void close_and_free_query(EN_P_ struct query_ctx *ctx) {
if (ctx != NULL) {
ev_io_stop(EV_A_ &ctx->io);
if (ctx->buf != NULL) {
free(ctx->buf);
}
free(ctx);
}
}
void close_and_free_remote(EN_P_ struct remote_ctx *ctx) {
if (ctx != NULL) {
close(ctx->fd);
ev_timer_stop(EV_A_ &ctx->watcher);
if (ctx->addr_header != NULL) {
free(ctx->addr_header);
}
if (ctx->buf != NULL) {
free(ctx->buf);
}
free(ctx);
}
}
#endif
static void remote_timeout_cb(EV_P_ ev_timer *watcher, int revents) {
struct remote_ctx *remote_ctx = (struct remote_ctx *) (((void*)watcher)
- sizeof(ev_io));
LOGE("UDP connection timeout"); LOGE("UDP connection timeout");
ev_timer_stop(EV_A_ watcher); ev_timer_stop(EV_A_ watcher);
close_and_free_remote(EV_A_ remote);
close_and_free_server(EV_A_ server);
close_and_free_remote(EV_A_ remote_ctx);
} }
static void query_resolve_cb(EV_P_ ev_timer *watcher, int revents) { static void query_resolve_cb(EV_P_ ev_timer *watcher, int revents) {
@ -299,14 +356,18 @@ static void query_resolve_cb(EV_P_ ev_timer *watcher, int revents) {
#endif #endif
struct remote_ctx *remote_ctx = new_remote_ctx(remotefd); struct remote_ctx *remote_ctx = new_remote_ctx(remotefd);
remote_ctx->src_addr = query_ctx->src_addr;
remote_ctx->dst_addr = *rp->ai_addr; remote_ctx->dst_addr = *rp->ai_addr;
remote_ctx->server_ctx = query_ctx->server_ctx; remote_ctx->server_ctx = query_ctx->server_ctx;
remote_ctx->addr_header_len = query_ctx->addr_header_len; remote_ctx->addr_header_len = query_ctx->addr_header_len;
memcpy(remote_ctx->addr_header, query_ctx->addr_header, query_ctx->addr_header_len); memcpy(remote_ctx->addr_header, query_ctx->addr_header, query_ctx->addr_header_len);
// Add to conn cache
cache_insert(query_ctx->server_ctx->conn_cache, key, (void *)remote_ctx);
ev_io_start(EV_A_ &remote_ctx->io); ev_io_start(EV_A_ &remote_ctx->io);
int w = sendto(remote_ctx->fd, buf, buf_len, 0, &remote_ctx->dst_addr, sizeof(remote_ctx->dst_addr));
int w = sendto(remote_ctx->fd, query_ctx->buf, query_ctx->buf_len, 0, &remote_ctx->dst_addr, sizeof(remote_ctx->dst_addr));
if (w == -1) { if (w == -1) {
ERROR("udprelay_sendto_remote"); ERROR("udprelay_sendto_remote");
@ -318,137 +379,87 @@ static void query_resolve_cb(EV_P_ ev_timer *watcher, int revents) {
} }
} }
// release addrinfo
// clean up
asyncns_freeaddrinfo(result); asyncns_freeaddrinfo(result);
close_and_free_query(EV_A_ query_ctx); close_and_free_query(EV_A_ query_ctx);
} }
static void remote_recv_cb (EV_P_ ev_io *w, int revents) { static void remote_recv_cb (EV_P_ ev_io *w, int revents) {
struct remote_ctx *remote_ctx = (struct remote_ctx *)w; struct remote_ctx *remote_ctx = (struct remote_ctx *)w;
struct server_ctx *server_ctx = remote_ctx->server_ctx;
// server has been closed
if (server_ctx == NULL) { if (server_ctx == NULL) {
LOGE("invalid server."); LOGE("invalid server.");
close_and_free_remote(EV_A_ remote); close_and_free_remote(EV_A_ remote);
return; return;
} }
ev_timer_again(EV_A_ &server->recv_ctx->watcher);
// triger the timer
ev_timer_again(EV_A_ &remote_ctx->watcher);
struct sockaddr src_addr;
char *buf = remote_ctx->buf;
int addr_len = sizeof(src_addr);
int addr_header_len = remote_ctx->addr_header_len;
ssize_t r = recv(remote->fd, server->buf, BUF_SIZE, 0);
// recv
ssize_t buf_len = recvfrom(remote_ctx->fd, buf, BUF_SIZE, 0, &src_addr, &addr_len);
if (r == 0) {
// connection closed
if (buf_len == -1) {
// error on recv
// simply drop that packet
if (verbose) { if (verbose) {
LOGD("remote_recv close the connection");
ERROR("udprelay_server_recvfrom");
} }
close_and_free_remote(EV_A_ remote);
close_and_free_server(EV_A_ server);
return; return;
} else if (r < 0) {
if (errno == EAGAIN || errno == EWOULDBLOCK) {
// no data
// continue to wait for recv
return;
} else {
ERROR("remote recv");
close_and_free_remote(EV_A_ remote);
close_and_free_server(EV_A_ server);
return;
}
} }
server->buf = ss_encrypt(server->buf, &r, server->e_ctx);
if (verbose) {
LOGD("receive a packet.");
}
if (server->buf == NULL) {
LOGE("invalid password or cipher");
close_and_free_remote(EV_A_ remote);
close_and_free_server(EV_A_ server);
#ifdef UDPRELAY_LOCAL
buf = ss_decrypt_all(BUF_SIZE, buf, &buf_len, server_ctx->method);
uint8_t atyp = *(uint8_t*)buf;
int offset = 1;
int len = parse_udprealy_header(buf + offset,
buf_len - offset, udprelay_header->atyp, host, port);
if (len == 0 || len != addr_header_len) {
// error in parse header
return; return;
} }
buf_len = buf_len - offset - addr_header_len;
memmove(buf, buf + offset + addr_header_len, buf_len);
#endif
int s = send(server->fd, server->buf, r, 0);
if (s == -1) {
if (errno == EAGAIN || errno == EWOULDBLOCK) {
// no data, wait for send
server->buf_len = r;
server->buf_idx = 0;
ev_io_stop(EV_A_ &remote_recv_ctx->io);
ev_io_start(EV_A_ &server->send_ctx->io);
} else {
ERROR("remote_recv_send");
close_and_free_remote(EV_A_ remote);
close_and_free_server(EV_A_ server);
}
return;
} else if (s < r) {
server->buf_len = r - s;
server->buf_idx = s;
ev_io_stop(EV_A_ &remote_recv_ctx->io);
ev_io_start(EV_A_ &server->send_ctx->io);
#ifdef UDPRELAY_REMOTE
if (src_addr != remote_ctx->dst_addr) {
LOGE("dest addr not match.");
return; return;
} }
}
struct remote_ctx *new_remote_ctx(int fd) {
struct remote_ctx *ctx = malloc(sizeof(struct remote_ctx));
memset(ctx, 0, sizeof(struct remote_ctx));
ctx->buf = malloc(BUF_SIZE);
ctx->buf_len = 0;
ctx->server_ctx = NULL;
ctx->fd = fd;
ev_io_init(&ctx->io, remote_recv_cb, fd, EV_READ);
return ctx;
}
struct server_ctx * new_server_ctx(int fd) {
struct server_ctx *ctx = malloc(sizeof(struct server_ctx));
memset(ctx, 0, sizeof(struct server_ctx));
ctx->remote_ctx = NULL;
ctx->fd = fd;
ev_io_init(&ctx.io, server_recv_cb, fd, EV_READ);
return ctx;
}
char *tmp_buf = malloc(buf_len + addr_header_len);
memcpy(tmpbuf, remote_ctx->addr_header, addr_header_len);
memcpy(tmpbuf + addr_header_len, buf, buf_len);
free(buf);
buf = tmp_buf;
buf_len += addr_header_len;
#endif
#ifdef UDPRELAY_REMOTE
struct query_ctx *new_query_ctx(asyncns_query_t *query,
const uint8_t *buf, const int buf_len) {
struct query_ctx *ctx = malloc(sizeof(struct query_ctx))
memset(ctx, 0, sizeof(struct query_ctx));
ctx->buf = malloc(buf_len);
ctx->buf_len = buf_len;
memcpy(ctx->buf, buf, buf_len);
ctx->query = query;
ev_timer_init(&ctx->watcher, query_resolve_cb, 0.2, 0.5);
return ctx;
}
int w = sendto(remote_ctx->fd, buf, buf_len, 0, &remote_ctx->src_addr, sizeof(remote_ctx->src_addr));
void close_and_free_query(EN_P_ struct query_ctx *ctx) {
if (ctx != NULL) {
ev_io_stop(EV_A_ &ctx->io);
if (ctx->buf != NULL) {
free(ctx->buf);
}
free(ctx);
if (w == -1) {
ERROR("udprelay_sendto_remote");
} }
}
void close_and_free_remote(EN_P_ struct remote_ctx *ctx) {
if (ctx != NULL) {
ev_timer_stop(EV_A_ &ctx->watcher);
if (ctx->buf != NULL) {
free(ctx->buf);
}
free(ctx);
}
} }
#endif
static void server_recv_cb (EV_P_ ev_io *w, int revents) { static void server_recv_cb (EV_P_ ev_io *w, int revents) {
struct server_ctx *server_ctx = (struct server_ctx *)w; struct server_ctx *server_ctx = (struct server_ctx *)w;
struct udprelay_header *udprelay_header; struct udprelay_header *udprelay_header;
struct sockaddr src_addr; struct sockaddr src_addr;
uint8_t *buf = malloc(BUF_SIZE);
char *buf = server_ctx->buf;
int addr_len = sizeof(src_addr); int addr_len = sizeof(src_addr);
int offset = 0; int offset = 0;
@ -519,48 +530,67 @@ static void server_recv_cb (EV_P_ ev_io *w, int revents) {
int addr_header_len = parse_udprealy_header(buf + offset, int addr_header_len = parse_udprealy_header(buf + offset,
buf_len - offset, udprelay_header->atyp, host, port); buf_len - offset, udprelay_header->atyp, host, port);
if (addr_header_len == 0) {
// error in parse header
return;
}
char *addr_header = buf + offset - sizeof(udprelay_header->atyp); char *addr_header = buf + offset - sizeof(udprelay_header->atyp);
char *key = hash_key(addr_header, addr_header_len, &src_addr); char *key = hash_key(addr_header, addr_header_len, &src_addr);
struct *conn_cache = server_ctx->conn_cache; struct *conn_cache = server_ctx->conn_cache;
#ifdef UDPRELAY_LOCAL #ifdef UDPRELAY_LOCAL
if (udprelay_header->frag) { if (udprelay_header->frag) {
LOGE("drop a message since frag is not 0"); LOGE("drop a message since frag is not 0");
goto clean_up;
return;
} }
struct addrinfo hints;
struct addrinfo *result;
struct remote_ctx *remote_ctx = NULL;
cache_lookup(conn_cache, key, (void*)remote_ctx);
memset(&hints, 0, sizeof(struct addrinfo));
hints.ai_family = AF_UNSPEC; /* Return IPv4 and IPv6 choices */
hints.ai_socktype = SOCK_DGRAM; /* We want a UDP socket */
if (remote_ctx == NULL) {
int s = getaddrinfo(server_ctx->remote_host, server_ctx->remote_port,
&hints, &result);
if (s != 0 || result == NULL) {
LOGE("getaddrinfo: %s", gai_strerror(s));
goto clean_up;
}
struct addrinfo hints;
struct addrinfo *result;
// Bind to any port
int remotefd = create_remote_socket(result->ai_family == AF_INET6);
if (remotefd < 0) {
ERROR("udprelay bind() error..");
goto clean_up;
}
setnonblocking(remotefd);
memset(&hints, 0, sizeof(struct addrinfo));
hints.ai_family = AF_UNSPEC; /* Return IPv4 and IPv6 choices */
hints.ai_socktype = SOCK_DGRAM; /* We want a UDP socket */
int s = getaddrinfo(server_ctx->remote_host, server_ctx->remote_port,
&hints, &result);
if (s != 0 || result == NULL) {
LOGE("getaddrinfo: %s", gai_strerror(s));
return;
}
// Bind to any port
int remotefd = create_remote_socket(result->ai_family == AF_INET6);
if (remotefd < 0) {
ERROR("udprelay bind() error..");
// remember to free addrinfo
freeaddrinfo(result);
return;
}
setnonblocking(remotefd);
struct remote_ctx *remote_ctx = new_remote_ctx(remotefd);
remote_ctx->src_addr = src_addr;
remote_ctx->dst_addr = *result->ai_addr;
remote_ctx->server_ctx = server_ctx;
remote_ctx->addr_header_len = addr_header_len;
memcpy(remote_ctx->addr_header, addr_header, addr_header_len);
// Init remote_ctx
remote_ctx = new_remote_ctx(remotefd);
remote_ctx->src_addr = src_addr;
remote_ctx->dst_addr = *result->ai_addr;
remote_ctx->server_ctx = server_ctx;
remote_ctx->addr_header_len = addr_header_len;
memcpy(remote_ctx->addr_header, addr_header, addr_header_len);
ev_io_start(EV_A_ &remote_ctx->io);
// Add to conn cache
cache_insert(conn_cache, key, (void *)remote_ctx);
// Start remote io
ev_io_start(EV_A_ &remote_ctx->io);
// clean up
freeaddrinfo(result);
}
buf_len -= 3; buf_len -= 3;
memmove(buf, buf + 3, buf_len); memmove(buf, buf + 3, buf_len);
@ -571,11 +601,8 @@ static void server_recv_cb (EV_P_ ev_io *w, int revents) {
if (w == -1) { if (w == -1) {
ERROR("udprelay_sendto_remote"); ERROR("udprelay_sendto_remote");
close_and_free_remote(EV_A_ remote_ctx);
} }
freeaddrinfo(result);
#else #else
buf_len -= offset; buf_len -= offset;
@ -585,30 +612,43 @@ static void server_recv_cb (EV_P_ ev_io *w, int revents) {
LOGD("send to: %s:%s", host, port); LOGD("send to: %s:%s", host, port);
} }
struct addrinfo hints;
asyncns_query_t *query;
memset(&hints, 0, sizeof(hints));
hints.ai_family = AF_UNSPEC;
hints.ai_socktype = SOCK_STREAM;
struct remote_ctx *remote_ctx = NULL;
cache_lookup(conn_cache, key, (void*)remote_ctx);
query = asyncns_getaddrinfo(server_ctx->asyncns,
host, port, &hints);
if (remote_ctx == NULL) {
struct addrinfo hints;
asyncns_query_t *query;
memset(&hints, 0, sizeof(hints));
hints.ai_family = AF_UNSPEC;
hints.ai_socktype = SOCK_STREAM;
if (query == NULL) {
ERROR("udp_asyncns_getaddrinfo");
goto clean_up;
}
query = asyncns_getaddrinfo(server_ctx->asyncns,
host, port, &hints);
struct query_ctx *query_ctx = new_query_ctx(query, buf, buf_len);
query_ctx->server_ctx = server_ctx;
query_ctx->addr_header_len = addr_header_len;
memcpy(query_ctx->addr_header, addr_header, addr_header_len);
if (query == NULL) {
ERROR("udp_asyncns_getaddrinfo");
return;
}
struct query_ctx *query_ctx = new_query_ctx(query, buf, buf_len);
query_ctx->server_ctx = server_ctx;
query_ctx->addr_header_len = addr_header_len;
query_ctx->src_addr = src_addr;
memcpy(query_ctx->addr_header, addr_header, addr_header_len);
ev_timer_start(EV_A_ &query_ctx->watcher);
} else {
ev_timer_start(EV_A_ &query_ctx->watcher);
int w = sendto(remote_ctx->fd, buf, buf_len, 0, &remote_ctx->dst_addr, sizeof(remote_ctx->dst_addr));
if (w == -1) {
ERROR("udprelay_sendto_remote");
}
}
#endif #endif
clean_up:
free(buf);
} }
int udprelay(const char *server_host, const char *server_port, int udprelay(const char *server_host, const char *server_port,

11
src/udprelay.h

@ -21,6 +21,7 @@ struct server_ctx {
int method; int method;
char *iface; char *iface;
struct cache *conn_cache; struct cache *conn_cache;
char *buf; // server send from, remote recv into
#ifdef UDPRELAY_REMOTE #ifdef UDPRELAY_REMOTE
asyncns_t *asyncns; asyncns_t *asyncns;
#endif #endif
@ -41,16 +42,16 @@ struct query_ctx {
struct remote_ctx { struct remote_ctx {
ev_io io; ev_io io;
#ifdef UDPRELAY_REMOTE
ev_timer watcher;
#endif
int fd; int fd;
int buf_len;
char *buf; // remote send from, server recv into
char *buf; // server send from, remote recv into
int addr_header_len; int addr_header_len;
char addr_header[384]; char addr_header[384];
struct sockaddr src_addr;
struct sockaddr dst_addr; struct sockaddr dst_addr;
struct server_ctx *server_ctx; struct server_ctx *server_ctx;
#ifdef UDPRELAY_REMOTE
ev_timer watcher;
#endif
}; };
static void server_recv_cb (EV_P_ ev_io *w, int revents); static void server_recv_cb (EV_P_ ev_io *w, int revents);

Loading…
Cancel
Save