TheArchive/iodine
1
0
Fork 0
mirror of https://github.com/yarrick/iodine.git synced 2024-11-25 11:05:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixed raw UDP mode + improved debugging

Browse Source
This commit is contained in:
frekky 2015-10-27 18:33:44 +08:00
parent c633173fea
commit 3bcfe91769
2 changed files with 78 additions and 62 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
src/client.c
View File

@ -1123,14 +1123,7 @@ tunnel_tun(int tun_fd, int dns_fd)
if ((read = read_tun(tun_fd, in, sizeof(in))) <= 0) if ((read = read_tun(tun_fd, in, sizeof(in))) <= 0)
return -1; return -1;
/* Check if outgoing buffer can hold data */ DEBUG(2, " IN: %lu bytes on tunnel, to be compressed: %d", read, compression_up);
if (window_buffer_available(outbuf) < (read / MAX_FRAGSIZE) + 1) {
DEBUG(1, " Outgoing buffer full (%lu/%lu), not adding data!",
outbuf->numitems, outbuf->length);
return -1;
}
DEBUG(2, " IN: %lu bytes on tunnel, compression %d", read, compression_up);
if (conn != CONN_DNS_NULL || compression_up) { if (conn != CONN_DNS_NULL || compression_up) {
datalen = sizeof(out); datalen = sizeof(out);
@ -1142,6 +1135,13 @@ tunnel_tun(int tun_fd, int dns_fd)
} }
if (conn == CONN_DNS_NULL) { if (conn == CONN_DNS_NULL) {
/* Check if outgoing buffer can hold data */
if (window_buffer_available(outbuf) < (read / MAX_FRAGSIZE) + 1) {
DEBUG(1, " Outgoing buffer full (%lu/%lu), not adding data!",
outbuf->numitems, outbuf->length);
return -1;
}
window_add_outgoing_data(outbuf, data, datalen, compression_up); window_add_outgoing_data(outbuf, data, datalen, compression_up);
/* Don't send anything here to respect min. send interval */ /* Don't send anything here to respect min. send interval */
} else { } else {
@ -1353,6 +1353,7 @@ client_tunnel(int tun_fd, int dns_fd)
/* TODO: detect DNS servers which drop frequent requests /* TODO: detect DNS servers which drop frequent requests
* TODO: adjust number of pending queries based on current data rate */ * TODO: adjust number of pending queries based on current data rate */
if (conn == CONN_DNS_NULL) {
sending = window_sending(outbuf); sending = window_sending(outbuf);
total = sending; total = sending;
check_pending_queries(); check_pending_queries();
@ -1379,6 +1380,7 @@ client_tunnel(int tun_fd, int dns_fd)
lazymode ? windowsize_down : 1, total); lazymode ? windowsize_down : 1, total);
if (sending > 0 || (total > 0 && lazymode)) { if (sending > 0 || (total > 0 && lazymode)) {
/* TODO: enforce min send interval even if we get new data */
tv = ms_to_timeval(min_send_interval_ms); tv = ms_to_timeval(min_send_interval_ms);
tv.tv_usec += 1; tv.tv_usec += 1;
} else if (total > 0 && !lazymode) { } else if (total > 0 && !lazymode) {
@ -1388,6 +1390,7 @@ client_tunnel(int tun_fd, int dns_fd)
send_ping_soon = 0; send_ping_soon = 0;
} }
}
if (stats) { if (stats) {
if (difftime(time(NULL), last_stats) >= stats) { if (difftime(time(NULL), last_stats) >= stats) {
@ -1404,11 +1407,12 @@ client_tunnel(int tun_fd, int dns_fd)
min_send_interval_ms, rtt_total_ms / num_immediate, server_timeout_ms); min_send_interval_ms, rtt_total_ms / num_immediate, server_timeout_ms);
fprintf(stderr, " Queries immediate: %5lu, timed out: %4lu target: %4ld ms\n", fprintf(stderr, " Queries immediate: %5lu, timed out: %4lu target: %4ld ms\n",
num_immediate, num_timeouts, max_timeout_ms); num_immediate, num_timeouts, max_timeout_ms);
if (conn == CONN_DNS_NULL) {
fprintf(stderr, " Frags resent: %4u, OOS: %4u down frag: %4ld ms\n", fprintf(stderr, " Frags resent: %4u, OOS: %4u down frag: %4ld ms\n",
outbuf->resends, inbuf->oos, downstream_timeout_ms); outbuf->resends, inbuf->oos, downstream_timeout_ms);
fprintf(stderr, " TX fragments: %8lu" ", RX: %8lu" ", pings: %8lu" "\n\n", fprintf(stderr, " TX fragments: %8lu" ", RX: %8lu" ", pings: %8lu" "\n\n",
num_frags_sent, num_frags_recv, num_pings); num_frags_sent, num_frags_recv, num_pings);
}
/* update since-last-report stats */ /* update since-last-report stats */
sent_since_report = num_sent; sent_since_report = num_sent;
recv_since_report = num_recv; recv_since_report = num_recv;
@ -1420,11 +1424,12 @@ client_tunnel(int tun_fd, int dns_fd)
if (send_ping_soon) { if (send_ping_soon) {
tv.tv_sec = 0; tv.tv_sec = 0;
tv.tv_usec = send_ping_soon * 1000; tv.tv_usec = send_ping_soon * 1000;
send_ping_soon = 0;
} }
FD_ZERO(&fds); FD_ZERO(&fds);
if (window_buffer_available(outbuf) > 16) { if (conn != CONN_DNS_NULL || window_buffer_available(outbuf) > 16) {
/* Fill up outgoing buffer with available data /* Fill up outgoing buffer with available data if it has enough space
* The windowing protocol manages data retransmits, timeouts etc. */ * The windowing protocol manages data retransmits, timeouts etc. */
FD_SET(tun_fd, &fds); FD_SET(tun_fd, &fds);
} }

43
src/server.c
View File

@ -853,14 +853,18 @@ handle_raw_login(uint8_t *packet, size_t len, struct query *q, int fd, int useri
{ {
char myhash[16]; char myhash[16];
if (len < 16) return; if (len < 16) {
DEBUG(2, "Invalid raw login packet: length %lu < 16 bytes!", len);
return;
}
/* can't use check_authenticated_user_and_ip() since IP address will be different, if (userid < 0 || userid >= created_users ||
so duplicate here except IP address */ check_authenticated_user_and_ip(userid, q) != 0) {
if (userid < 0 || userid >= created_users) return; DEBUG(2, "User %d not authenticated, ignoring raw login!", userid);
if (!check_authenticated_user_and_ip(userid, q)) return; return;
}
DEBUG(1, "IN login raw, len %lu, from user %d", len, userid); DEBUG(1, "RX-raw: login, len %lu, from user %d", len, userid);
/* User sends hash of seed + 1 */ /* User sends hash of seed + 1 */
login_calculate(myhash, 16, password, users[userid].seed + 1); login_calculate(myhash, 16, password, users[userid].seed + 1);
@ -892,9 +896,9 @@ handle_raw_data(uint8_t *packet, size_t len, struct query *q, struct dnsfd *dns_
/* Update time info for user */ /* Update time info for user */
users[userid].last_pkt = time(NULL); users[userid].last_pkt = time(NULL);
/* copy to packet buffer, update length TODO fix the raw UDP protocol */ /* copy to packet buffer, update length */
DEBUG(3, "IN pkt raw, total %lu, from user %d", len, userid); DEBUG(3, "RX-raw: full pkt raw, length %lu, from user %d", len, userid);
handle_full_packet(tun_fd, dns_fds, userid, packet, len, 1); handle_full_packet(tun_fd, dns_fds, userid, packet, len, 1);
} }
@ -910,7 +914,7 @@ handle_raw_ping(struct query *q, int dns_fd, int userid)
/* Update time info for user */ /* Update time info for user */
users[userid].last_pkt = time(NULL); users[userid].last_pkt = time(NULL);
DEBUG(3, "IN ping raw, from user %d", userid); DEBUG(3, "RX-raw: ping from user %d", userid);
/* Send ping reply */ /* Send ping reply */
send_raw(dns_fd, NULL, 0, userid, RAW_HDR_CMD_PING, &q->from, q->fromlen); send_raw(dns_fd, NULL, 0, userid, RAW_HDR_CMD_PING, &q->from, q->fromlen);
@ -920,30 +924,37 @@ static int
raw_decode(uint8_t *packet, size_t len, struct query *q, int dns_fd, struct dnsfd *dns_fds, int tun_fd) raw_decode(uint8_t *packet, size_t len, struct query *q, int dns_fd, struct dnsfd *dns_fds, int tun_fd)
{ {
int raw_user; int raw_user;
uint8_t raw_cmd;
/* minimum length */ /* minimum length */
if (len < RAW_HDR_LEN) return 0; if (len < RAW_HDR_LEN) return 0;
/* should start with header */ /* should start with header */
if (memcmp(packet, raw_header, RAW_HDR_IDENT_LEN)) return 0; if (memcmp(packet, raw_header, RAW_HDR_IDENT_LEN))
return 0;
raw_cmd = RAW_HDR_GET_CMD(packet);
raw_user = RAW_HDR_GET_USR(packet); raw_user = RAW_HDR_GET_USR(packet);
DEBUG(3, "TX-raw: client %s, user %d, raw command '%c' length %lu",
format_addr(&q->from, q->fromlen), raw_user, RAW_HDR_GET_CMD(packet), len); DEBUG(3, "RX-raw: client %s, user %d, raw command 0x%02X, length %lu",
switch (RAW_HDR_GET_CMD(packet)) { format_addr(&q->from, q->fromlen), raw_user, raw_cmd, len);
packet += RAW_HDR_LEN;
len -= RAW_HDR_LEN;
switch (raw_cmd) {
case RAW_HDR_CMD_LOGIN: case RAW_HDR_CMD_LOGIN:
/* Login challenge */ /* Login challenge */
handle_raw_login(&packet[RAW_HDR_LEN], len - RAW_HDR_LEN, q, dns_fd, raw_user); handle_raw_login(packet, len, q, dns_fd, raw_user);
break; break;
case RAW_HDR_CMD_DATA: case RAW_HDR_CMD_DATA:
/* Data packet */ /* Data packet */
handle_raw_data(&packet[RAW_HDR_LEN], len - RAW_HDR_LEN, q, dns_fds, tun_fd, raw_user); handle_raw_data(packet, len, q, dns_fds, tun_fd, raw_user);
break; break;
case RAW_HDR_CMD_PING: case RAW_HDR_CMD_PING:
/* Keepalive packet */ /* Keepalive packet */
handle_raw_ping(q, dns_fd, raw_user); handle_raw_ping(q, dns_fd, raw_user);
break; break;
default: default:
DEBUG(1, "Unhandled raw command %02X from user %d", RAW_HDR_GET_CMD(packet), raw_user); DEBUG(1, "Unhandled raw command %02X from user %d", raw_cmd, raw_user);
break; break;
} }
return 1; return 1;