hostname: getopt_ulflags'isation

This commit is contained in:
Denis Vlasenko 2006-09-22 08:53:14 +00:00
parent 0e87d347e8
commit 8514fc5681

View File

@ -14,15 +14,11 @@
*/ */
#include "busybox.h" #include "busybox.h"
#include <getopt.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;
char buf[255]; char buf[256];
if (!s) if (!s)
return; return;
@ -35,7 +31,7 @@ static void do_sethostname(char *s, int isfile)
} }
} else { } else {
f = xfopen(s, "r"); f = xfopen(s, "r");
while (fgets(buf, 255, f) != NULL) { while (fgets(buf, sizeof(buf), f) != NULL) {
if (buf[0] =='#') { if (buf[0] =='#') {
continue; continue;
} }
@ -50,64 +46,57 @@ static void do_sethostname(char *s, int isfile)
int hostname_main(int argc, char **argv) int hostname_main(int argc, char **argv)
{ {
int opt; enum {
int type = 0; OPT_d = 0x1,
struct hostent *hp; OPT_f = 0x2,
char *filename = NULL; OPT_i = 0x4,
char buf[255]; OPT_s = 0x8,
char *p = NULL; OPT_dfis = 0xf,
};
char buf[256];
unsigned long opt;
char *hostname_str = NULL;
if (argc < 1) if (argc < 1)
bb_show_usage(); bb_show_usage();
while ((opt = getopt(argc, argv, "dfisF:")) > 0) { opt = bb_getopt_ulflags(argc, argv, "dfisF:", &hostname_str);
switch (opt) {
case 'd':
case 'f':
case 'i':
case 's':
type = opt;
break;
case 'F':
filename = optarg;
break;
default:
bb_show_usage();
}
}
/* Output in desired format */ /* Output in desired format */
if (type != 0) { if (opt & OPT_dfis) {
gethostname(buf, 255); struct hostent *hp;
char *p;
gethostname(buf, sizeof(buf));
hp = xgethostbyname(buf); hp = xgethostbyname(buf);
p = strchr(hp->h_name, '.'); p = strchr(hp->h_name, '.');
if (type == 'f') { if (opt & OPT_f) {
puts(hp->h_name); puts(hp->h_name);
} else if (type == 's') { } else if (opt & OPT_s) {
if (p != NULL) { if (p != NULL) {
*p = 0; *p = 0;
} }
puts(hp->h_name); puts(hp->h_name);
} else if (type == 'd') { } else if (opt & OPT_d) {
if (p) puts(p + 1); if (p) puts(p + 1);
} else if (type == 'i') { } else if (opt & OPT_i) {
while (hp->h_addr_list[0]) { while (hp->h_addr_list[0]) {
printf("%s ", inet_ntoa(*(struct in_addr *) (*hp->h_addr_list++))); printf("%s ", inet_ntoa(*(struct in_addr *) (*hp->h_addr_list++)));
} }
printf("\n"); puts("");
} }
} }
/* Set the hostname */ /* Set the hostname */
else if (filename != NULL) { else if (hostname_str != NULL) {
do_sethostname(filename, 1); do_sethostname(hostname_str, 1);
} else if (optind < argc) { } else if (optind < argc) {
do_sethostname(argv[optind], 0); do_sethostname(argv[optind], 0);
} }
/* Or if all else fails, /* Or if all else fails,
* just print the current hostname */ * just print the current hostname */
else { else {
gethostname(buf, 255); gethostname(buf, sizeof(buf));
puts(buf); puts(buf);
} }
return(0); return 0;
} }