miegl/glorytun
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Rewrite the ip size directly in the packet

Browse Source
This commit is contained in:
angt
2015-10-29 11:57:32 +01:00
parent 8a7fedcb7a
commit 0957a0c341
1 changed files with 18 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
glorytun.c
View File

@@ -305,6 +305,12 @@ static int option (int argc, char **argv, int n, struct option *opt)
return 0; return 0;
} }
static void set_ip_size (uint8_t *data, size_t size)
{
data[2] = 0xFF&(size>>8);
data[3] = 0xFF&(size);
}
static ssize_t get_ip_size (const uint8_t *data, size_t size) static ssize_t get_ip_size (const uint8_t *data, size_t size)
{ {
if (size<20) if (size<20)
@@ -419,8 +425,18 @@ int main (int argc, char **argv)
if (r<0) if (r<0)
break; break;
if (r==get_ip_size(tun.recv.write, size)) ssize_t ip_size = get_ip_size(tun.recv.write, size);
tun.recv.write += r;
if (ip_size<=0)
break;
if (r<ip_size)
set_ip_size(tun.recv.write, r);
if (r>ip_size)
break;
tun.recv.write += r;
} }
} }