mirror
/
NetGuard
mirror of https://github.com/M66B/NetGuard.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Revert "Fixed send hanging by simplifying" 

This reverts commit 679a22ef4c.
This commit is contained in:
M66B 2019-05-05 19:20:49 +02:00
parent 697b7bdf2d
commit 06733982bf
2 changed files with 22 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
app/src/main/jni/netguard/netguard.h
View File

@ -96,6 +96,7 @@ struct allowed {
struct segment {
uint32_t seq;
uint16_t len;
uint16_t sent;
int psh;
uint8_t *data;
struct segment *next;

37
app/src/main/jni/netguard/tcp.c
View File

@ -145,8 +145,8 @@ int monitor_tcp_session(const struct arguments *args, struct ng_session *s, int
// Check for outgoing data
if (s->tcp.forward != NULL) {
uint32_t buffer_size = (uint32_t) get_receive_buffer(s);
if (s->tcp.forward->seq == s->tcp.remote_seq &&
s->tcp.forward->len < buffer_size)
if (s->tcp.forward->seq + s->tcp.forward->sent == s->tcp.remote_seq &&
s->tcp.forward->len - s->tcp.forward->sent < buffer_size)
events = events | EPOLLOUT;
else
recheck = 1;
@ -205,7 +205,7 @@ uint32_t get_receive_window(const struct ng_session *cur) {
uint32_t toforward = 0;
struct segment *q = cur->tcp.forward;
while (q != NULL) {
toforward += q->len;
toforward += (q->len - q->sent);
q = q->next;
}
@ -459,15 +459,17 @@ void check_tcp_socket(const struct arguments *args,
// Forward data
uint32_t buffer_size = (uint32_t) get_receive_buffer(s);
while (s->tcp.forward != NULL &&
s->tcp.forward->seq == s->tcp.remote_seq &&
s->tcp.forward->len < buffer_size) {
log_android(ANDROID_LOG_DEBUG, "%s fwd %u...%u",
s->tcp.forward->seq + s->tcp.forward->sent == s->tcp.remote_seq &&
s->tcp.forward->len - s->tcp.forward->sent < buffer_size) {
log_android(ANDROID_LOG_DEBUG, "%s fwd %u...%u sent %u",
session,
s->tcp.forward->seq - s->tcp.remote_start,
s->tcp.forward->seq + s->tcp.forward->len - s->tcp.remote_start);
s->tcp.forward->seq + s->tcp.forward->len - s->tcp.remote_start,
s->tcp.forward->sent);
ssize_t sent = send(s->socket,
s->tcp.forward->data, s->tcp.forward->len,
s->tcp.forward->data + s->tcp.forward->sent,
s->tcp.forward->len - s->tcp.forward->sent,
(unsigned int) (MSG_NOSIGNAL | (s->tcp.forward->psh
? 0
: MSG_MORE)));
@ -485,18 +487,18 @@ void check_tcp_socket(const struct arguments *args,
fwd = 1;
buffer_size -= sent;
s->tcp.sent += sent;
s->tcp.forward->seq += sent;
s->tcp.forward->len -= sent;
s->tcp.remote_seq = s->tcp.forward->seq;
s->tcp.forward->sent += sent;
s->tcp.remote_seq = s->tcp.forward->seq + s->tcp.forward->sent;
if (s->tcp.forward->len == 0) {
if (s->tcp.forward->len == s->tcp.forward->sent) {
struct segment *p = s->tcp.forward;
s->tcp.forward = s->tcp.forward->next;
free(p->data);
free(p);
} else {
log_android(ANDROID_LOG_WARN, "%s partial send %u/%u",
session, sent, s->tcp.forward->len);
log_android(ANDROID_LOG_WARN,
"%s partial send %u/%u",
session, s->tcp.forward->sent, s->tcp.forward->len);
break;
}
}
@ -505,10 +507,11 @@ void check_tcp_socket(const struct arguments *args,
// Log data buffered
struct segment *seg = s->tcp.forward;
while (seg != NULL) {
log_android(ANDROID_LOG_WARN, "%s queued %u...%u",
log_android(ANDROID_LOG_WARN, "%s queued %u...%u sent %u",
session,
seg->seq - s->tcp.remote_start,
seg->seq + seg->len - s->tcp.remote_start);
seg->seq + seg->len - s->tcp.remote_start,
seg->sent);
seg = seg->next;
}
}
@ -747,6 +750,7 @@ jboolean handle_tcp(const struct arguments *args,
s->tcp.forward = malloc(sizeof(struct segment));
s->tcp.forward->seq = s->tcp.remote_seq;
s->tcp.forward->len = datalen;
s->tcp.forward->sent = 0;
s->tcp.forward->psh = tcphdr->psh;
s->tcp.forward->data = malloc(datalen);
memcpy(s->tcp.forward->data, data, datalen);
@ -981,6 +985,7 @@ void queue_tcp(const struct arguments *args,
struct segment *n = malloc(sizeof(struct segment));
n->seq = seq;
n->len = datalen;
n->sent = 0;
n->psh = tcphdr->psh;
n->data = malloc(datalen);
memcpy(n->data, data, datalen);