Major cleanup from Charles Steinkuehler <charles@steinkuehler.net>:

- Switched to getopt argument parsing
    - Added -f option to get fully qualified domain name
    - Fixed the -s (short) and -d (domain) options, which were not
      doing a gethostbyname lookup to get the FQDN before trying to
      separate the local and domain portions of the hostname.
    - Fixed probem with 'agressive setting' of the hostname...the
      previous busybox version would try to set the hostname if called
      with a non-option argument, or the -F option, even if another
      option (like -i or -s) was given.  This behavior does not match
      the net-tools hostname, which does not attempt to set anything if
      given a 'display' option, regardless of the presence/absence of
      the -F option or additional command line arguments.
    - When using a file to set the hostname, behavior now matches
      net-tools version...previous busybox version did not handle
      comments, and simply grabbed the first line from the file.
This commit is contained in:
Eric Andersen 2001-10-31 09:59:57 +00:00
parent 950d8b496f
commit 3d61b10595

View File

@ -1,6 +1,6 @@
/* vi: set sw=4 ts=4: */ /* vi: set sw=4 ts=4: */
/* /*
* $Id: hostname.c,v 1.31 2001/10/24 04:59:56 andersen Exp $ * $Id: hostname.c,v 1.32 2001/10/31 09:59:57 andersen Exp $
* Mini hostname implementation for busybox * Mini hostname implementation for busybox
* *
* Copyright (C) 1999 by Randolph Chung <tausq@debian.org> * Copyright (C) 1999 by Randolph Chung <tausq@debian.org>
@ -30,8 +30,12 @@
#include <string.h> #include <string.h>
#include <stdio.h> #include <stdio.h>
#include <stdlib.h> #include <stdlib.h>
#include <getopt.h>
#include "busybox.h" #include "busybox.h"
extern char *optarg; /* in unistd.h */
extern int optind, opterr, optopt; /* in unistd.h */
static void do_sethostname(char *s, int isfile) static void do_sethostname(char *s, int isfile)
{ {
FILE *f; FILE *f;
@ -48,81 +52,79 @@ static void do_sethostname(char *s, int isfile)
} }
} else { } else {
f = xfopen(s, "r"); f = xfopen(s, "r");
fgets(buf, 255, f); while (fgets(buf, 255, f) != NULL) {
if (buf[0] =='#') {
continue;
}
chomp(buf);
do_sethostname(buf, 0);
}
#ifdef CONFIG_FEATURE_CLEAN_UP #ifdef CONFIG_FEATURE_CLEAN_UP
fclose(f); fclose(f);
#endif #endif
chomp(buf);
do_sethostname(buf, 0);
} }
} }
int hostname_main(int argc, char **argv) int hostname_main(int argc, char **argv)
{ {
int opt_short = 0; int opt;
int opt_domain = 0; int type = 0;
int opt_ip = 0; struct hostent *hp;
struct hostent *h;
char *filename = NULL; char *filename = NULL;
char buf[255]; char buf[255];
char *s = NULL; char *p = NULL;
if (argc < 1) if (argc < 1)
show_usage(); show_usage();
while (--argc > 0 && **(++argv) == '-') { while ((opt = getopt(argc, argv, "dfisF:")) > 0) {
while (*(++(*argv))) { switch (opt) {
switch (**argv) { case 'd':
case 's': case 'f':
opt_short = 1; case 'i':
break; case 's':
case 'i': type = opt;
opt_ip = 1; break;
break; case 'F':
case 'd': filename = optarg;
opt_domain = 1; break;
break; default:
case 'F': show_usage();
if (--argc == 0) {
show_usage();
}
filename = *(++argv);
break;
case '-':
if (strcmp(++(*argv), "file") || --argc ==0 ) {
show_usage();
}
filename = *(++argv);
break;
default:
show_usage();
}
if (filename != NULL)
break;
} }
} }
if (argc >= 1) { /* Output in desired format */
do_sethostname(*argv, 0); if (type != 0) {
} else if (filename != NULL) {
do_sethostname(filename, 1);
} else {
gethostname(buf, 255); gethostname(buf, 255);
if (opt_short) { hp = xgethostbyname(buf);
s = strchr(buf, '.'); p = strchr(hp->h_name, '.');
if (!s) if (type == 'f') {
s = buf; puts(hp->h_name);
*s = 0; } else if (type == 's') {
puts(buf); if (p != NULL) {
} else if (opt_domain) { *p = 0;
s = strchr(buf, '.'); }
puts(s ? s + 1 : "");
} else if (opt_ip) {
h = xgethostbyname(buf);
puts(inet_ntoa(*(struct in_addr *) (h->h_addr)));
} else {
puts(buf); puts(buf);
} else if (type == 'd') {
puts(p ? p + 1 : "");
} else if (type == 'i') {
while (hp->h_addr_list[0]) {
printf("%s ", inet_ntoa(*(struct in_addr *) (*hp->h_addr_list++)));
}
printf("\n");
} }
} }
/* Set the hostname */
else if (filename != NULL) {
do_sethostname(filename, 1);
} else if (optind < argc) {
do_sethostname(argv[optind], 0);
}
/* Or if all else fails,
* just print the current hostname */
else {
gethostname(buf, 255);
puts(buf);
}
return(0); return(0);
} }