mirrors/iodine
1
0
Fork 0
mirror of https://github.com/yarrick/iodine.git synced 2024-11-29 09:21:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

handshake

Browse source
This commit is contained in:
Bjorn Andersson 2006-06-11 16:13:40 +00:00
parent 25bb52c585
commit 03509b9e78
3 changed files with 65 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
dns.c
View file

@ -32,6 +32,7 @@
#include "read.h" #include "read.h"
static int host2dns(const char *, char *, int); static int host2dns(const char *, char *, int);
static int dns_write(int, int, char *, int, char);
struct sockaddr_in peer; struct sockaddr_in peer;
char topdomain[256]; char topdomain[256];
@ -202,7 +203,16 @@ dns_ping(int dns_fd)
} }
data[0] = (pingid & 0xFF00) >> 8; data[0] = (pingid & 0xFF00) >> 8;
data[1] = (pingid & 0xFF); data[1] = (pingid & 0xFF);
dns_write(dns_fd, ++pingid, data, 2, 1); dns_write(dns_fd, ++pingid, data, 2, 'P');
}
void
dns_handshake(int dns_fd)
{
char data[2];
data[0] = (pingid & 0xFF00) >> 8;
data[1] = (pingid & 0xFF);
dns_write(dns_fd, ++pingid, data, 2, 'H');
} }
void void
@ -262,8 +272,8 @@ put_hex(char *p, char h)
p[1] = to_hex[t]; p[1] = to_hex[t];
} }
int static int
dns_write(int fd, int id, char *buf, int len, int ping) dns_write(int fd, int id, char *buf, int len, char flag)
{ {
int avail; int avail;
int i; int i;
@ -285,8 +295,8 @@ dns_write(int fd, int id, char *buf, int len, int ping)
bzero(data, sizeof(data)); bzero(data, sizeof(data));
d = data; d = data;
if (ping) { if (flag != 0) {
*d = 'P'; *d = flag;
} else { } else {
// First byte is 0 for middle packet and 1 for last packet // First byte is 0 for middle packet and 1 for last packet
*d = '0' + final; *d = '0' + final;

2
dns.h
View file

@ -23,8 +23,8 @@ void close_dns(int);
int dns_sending(); int dns_sending();
void dns_handle_tun(int, char *, int); void dns_handle_tun(int, char *, int);
void dns_ping(int); void dns_ping(int);
void dns_handshake(int);
void dns_query(int, int, char *, int); void dns_query(int, int, char *, int);
int dns_write(int, int, char *, int, int);
int dns_read(int, char *, int); int dns_read(int, char *, int);
extern struct sockaddr_in peer; extern struct sockaddr_in peer;

49
iodine.c
View file

@ -101,6 +101,52 @@ tunnel(int tun_fd, int dns_fd)
return 0; return 0;
} }
static int
handshake(int dns_fd)
{
int i;
int r;
char *p;
int read;
fd_set fds;
int timeout;
char in[4096];
struct timeval tv;
timeout = 1;
for (i=0;i<5;i++) {
tv.tv_sec = timeout++;
tv.tv_usec = 0;
dns_handshake(dns_fd);
FD_ZERO(&fds);
FD_SET(dns_fd, &fds);
r = select(dns_fd + 1, &fds, NULL, NULL, &tv);
if(r > 0) {
read = dns_read(dns_fd, in, sizeof(in));
if(read <= 0) {
perror("read");
continue;
}
p = strchr(in, '-');
*p++ = '\0';
if (tun_setip(in) == 0 && tun_setmtu(atoi(p)) == 0)
return 0;
}
printf("Retrying...\n");
}
return 1;
}
extern char *__progname; extern char *__progname;
static void static void
@ -192,6 +238,9 @@ main(int argc, char **argv)
signal(SIGINT, sigint); signal(SIGINT, sigint);
if(handshake(dns_fd))
goto cleanup2;
if (newroot) { if (newroot) {
if (chroot(newroot) != 0 || chdir("/") != 0) if (chroot(newroot) != 0 || chdir("/") != 0)
err(1, "%s", newroot); err(1, "%s", newroot);