#36 fetch remote ip number after login

This commit is contained in:
Erik Ekman 2009-06-11 20:23:05 +00:00 committed by Erik Ekman
parent 27fc039700
commit 3eef144fce

View File

@ -473,6 +473,16 @@ send_version(int fd, uint32_t version)
send_packet(fd, 'V', data, sizeof(data));
}
static void
send_ip_request(int fd, int userid)
{
char buf[512] = "I_.";
buf[1] = b32_5to8(userid);
strncat(buf, topdomain, 512 - strlen(buf));
send_query(fd, buf);
}
static void
send_case_check(int fd)
{
@ -622,6 +632,66 @@ handshake_login(int dns_fd, int seed)
return 1;
}
static int
handshake_raw_udp(int dns_fd)
{
struct timeval tv;
char in[4096];
fd_set fds;
int i;
int r;
int read;
unsigned remoteaddr = 0;
struct in_addr server;
fprintf(stderr, "Testing raw UDP data to the server");
fflush(stderr);
for (i=0; running && i<3 ;i++) {
tv.tv_sec = i + 1;
tv.tv_usec = 0;
send_ip_request(dns_fd, userid);
FD_ZERO(&fds);
FD_SET(dns_fd, &fds);
r = select(dns_fd + 1, &fds, NULL, NULL, &tv);
if(r > 0) {
read = read_dns(dns_fd, in, sizeof(in));
if (read == 5 && in[0] == 'I') {
/* Received IP address */
remoteaddr = (in[1] & 0xff);
remoteaddr <<= 8;
remoteaddr |= (in[2] & 0xff);
remoteaddr <<= 8;
remoteaddr |= (in[3] & 0xff);
remoteaddr <<= 8;
remoteaddr |= (in[4] & 0xff);
server.s_addr = ntohl(remoteaddr);
break;
}
} else {
fprintf(stderr, ".");
fflush(stderr);
}
}
if (!remoteaddr) {
fprintf(stderr, " failed to get IP.\n");
return 1;
}
fprintf(stderr, " at %s", inet_ntoa(server));
fflush(stderr);
/* TODO do login against port 53 on remote server
* based on the old seed. If reply received,
* switch to raw udp mode */
fprintf(stderr, ": not implemented\n");
return 1;
/* TODO and then return 0 on success */
}
static void
handshake_case_check(int dns_fd)
{
@ -871,6 +941,8 @@ handshake(int dns_fd, int autodetect_frag_size, int fragsize)
return r;
}
handshake_raw_udp(dns_fd);
handshake_case_check(dns_fd);
if (case_preserved) {