return 1 if connection failed, #62

This commit is contained in:
Erik Ekman 2009-06-24 16:29:30 +00:00 committed by Erik Ekman
parent e84b317e0e
commit da50020f5b

View File

@ -1157,6 +1157,7 @@ main(int argc, char **argv)
int dns_fd; int dns_fd;
int max_downstream_frag_size; int max_downstream_frag_size;
int autodetect_frag_size; int autodetect_frag_size;
int retval;
memset(password, 0, 33); memset(password, 0, 33);
username = NULL; username = NULL;
@ -1174,6 +1175,8 @@ main(int argc, char **argv)
b32 = get_base32_encoder(); b32 = get_base32_encoder();
dataenc = get_base32_encoder(); dataenc = get_base32_encoder();
retval = 0;
#ifdef WINDOWS32 #ifdef WINDOWS32
WSAStartup(req_version, &wsa_data); WSAStartup(req_version, &wsa_data);
#endif #endif
@ -1282,16 +1285,22 @@ main(int argc, char **argv)
if (strlen(password) == 0) if (strlen(password) == 0)
read_password(password, sizeof(password)); read_password(password, sizeof(password));
if ((tun_fd = open_tun(device)) == -1) if ((tun_fd = open_tun(device)) == -1) {
retval = 1;
goto cleanup1; goto cleanup1;
if ((dns_fd = open_dns(0, INADDR_ANY)) == -1) }
if ((dns_fd = open_dns(0, INADDR_ANY)) == -1) {
retval = 1;
goto cleanup2; goto cleanup2;
}
signal(SIGINT, sighandler); signal(SIGINT, sighandler);
signal(SIGTERM, sighandler); signal(SIGTERM, sighandler);
if(handshake(dns_fd, autodetect_frag_size, max_downstream_frag_size)) if (handshake(dns_fd, autodetect_frag_size, max_downstream_frag_size)) {
retval = 1;
goto cleanup2; goto cleanup2;
}
fprintf(stderr, "Sending queries for %s to %s\n", topdomain, nameserv_addr); fprintf(stderr, "Sending queries for %s to %s\n", topdomain, nameserv_addr);
@ -1325,5 +1334,6 @@ cleanup2:
close_tun(tun_fd); close_tun(tun_fd);
cleanup1: cleanup1:
return 0; return retval;
} }