mirror of
https://github.com/sheumann/hush.git
synced 2025-01-16 09:30:29 +00:00
hostname: getopt_ulflags'isation
This commit is contained in:
parent
0e87d347e8
commit
8514fc5681
@ -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;
|
||||||
}
|
}
|
||||||
|
Loading…
x
Reference in New Issue
Block a user