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

cleanup in main

Browse source
This commit is contained in:
Bjorn Andersson 2006-06-11 12:18:49 +00:00
parent 12d79f6e25
commit 7bf9c25ebc
2 changed files with 26 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
dnstun.c
View file

@ -116,9 +116,11 @@ tunnel(int tun_fd, int dns_fd)
return 0; return 0;
} }
extern char *__progname;
static void static void
usage() { usage() {
printf("Usage: dnstun [-u user] nameserver topdomain\n"); printf("Usage: %s [-u user] nameserver topdomain\n", __progname);
exit(2); exit(2);
} }
@ -132,15 +134,11 @@ main(int argc, char **argv)
struct passwd *pw; struct passwd *pw;
username = NULL; username = NULL;
while ((choice = getopt(argc, argv, "u:")) != -1) { while ((choice = getopt(argc, argv, "u:")) != -1) {
switch(choice) { switch(choice) {
case 'u': case 'u':
username = optarg; username = optarg;
pw = getpwnam(username);
if (!pw) {
printf("User %s does not exist!\n", username);
usage();
}
break; break;
default: default:
usage(); usage();
@ -150,10 +148,19 @@ main(int argc, char **argv)
argc -= optind; argc -= optind;
argv += optind; argv += optind;
if (argc != 2) { if (argc != 2) {
usage(); usage();
} }
if(username) {
pw = getpwnam(username);
if (!pw) {
printf("User %s does not exist!\n", username);
usage();
}
}
tun_fd = open_tun(); tun_fd = open_tun();
dns_fd = open_dns(argv[0], argv[1]); dns_fd = open_dns(argv[0], argv[1]);

18
dnstund.c
View file

@ -113,9 +113,11 @@ tunnel(int tun_fd, int dns_fd)
return 0; return 0;
} }
extern char *__progname;
static void static void
usage() { usage() {
printf("Usage: dnstund [-u user] topdomain\n"); printf("Usage: %s [-u user] topdomain\n", __progname);
exit(2); exit(2);
} }
@ -129,15 +131,11 @@ main(int argc, char **argv)
struct passwd *pw; struct passwd *pw;
username = NULL; username = NULL;
while ((choice = getopt(argc, argv, "u:")) != -1) { while ((choice = getopt(argc, argv, "u:")) != -1) {
switch(choice) { switch(choice) {
case 'u': case 'u':
username = optarg; username = optarg;
pw = getpwnam(username);
if (!pw) {
printf("User %s does not exist!\n", username);
usage();
}
break; break;
default: default:
usage(); usage();
@ -152,6 +150,14 @@ main(int argc, char **argv)
usage(); usage();
} }
if (username) {
pw = getpwnam(username);
if (!pw) {
printf("User %s does not exist!\n", username);
usage();
}
}
tun_fd = open_tun(); tun_fd = open_tun();
dnsd_fd = open_dnsd(argv[0]); dnsd_fd = open_dnsd(argv[0]);
printf("Listening to dns for domain %s\n", argv[0]); printf("Listening to dns for domain %s\n", argv[0]);