1/* vi: set sw=4 ts=4: */
2/*
3 * Mini hostname implementation for busybox
4 *
5 * Copyright (C) 1999 by Randolph Chung <tausq@debian.org>
6 *
7 * adjusted by Erik Andersen <andersen@codepoet.org> to remove
8 * use of long options and GNU getopt.  Improved the usage info.
9 *
10 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
11 *
12 * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
13 */
14
15#include "libbb.h"
16
17static void do_sethostname(char *s, int isfile)
18{
19	FILE *f;
20
21	if (!s)
22		return;
23	if (!isfile) {
24		if (sethostname(s, strlen(s)) < 0) {
25			if (errno == EPERM)
26				bb_error_msg_and_die(bb_msg_perm_denied_are_you_root);
27			else
28				bb_perror_msg_and_die("sethostname");
29		}
30	} else {
31		f = xfopen(s, "r");
32#define strbuf bb_common_bufsiz1
33		while (fgets(strbuf, sizeof(strbuf), f) != NULL) {
34			if (strbuf[0] == '#') {
35				continue;
36			}
37			chomp(strbuf);
38			do_sethostname(strbuf, 0);
39		}
40		if (ENABLE_FEATURE_CLEAN_UP)
41			fclose(f);
42	}
43}
44
45int hostname_main(int argc, char **argv);
46int hostname_main(int argc, char **argv)
47{
48	enum {
49		OPT_d = 0x1,
50		OPT_f = 0x2,
51		OPT_i = 0x4,
52		OPT_s = 0x8,
53		OPT_F = 0x10,
54		OPT_dfis = 0xf,
55	};
56
57	char buf[256];
58	char *hostname_str;
59
60	if (argc < 1)
61		bb_show_usage();
62
63	getopt32(argv, "dfisF:", &hostname_str);
64
65	/* Output in desired format */
66	if (option_mask32 & OPT_dfis) {
67		struct hostent *hp;
68		char *p;
69		gethostname(buf, sizeof(buf));
70		hp = xgethostbyname(buf);
71		p = strchr(hp->h_name, '.');
72		if (option_mask32 & OPT_f) {
73			puts(hp->h_name);
74		} else if (option_mask32 & OPT_s) {
75			if (p != NULL) {
76				*p = '\0';
77			}
78			puts(hp->h_name);
79		} else if (option_mask32 & OPT_d) {
80			if (p)
81				puts(p + 1);
82		} else if (option_mask32 & OPT_i) {
83			while (hp->h_addr_list[0]) {
84				printf("%s ", inet_ntoa(*(struct in_addr *) (*hp->h_addr_list++)));
85			}
86			puts("");
87		}
88	}
89	/* Set the hostname */
90	else if (option_mask32 & OPT_F) {
91		do_sethostname(hostname_str, 1);
92	} else if (optind < argc) {
93		do_sethostname(argv[optind], 0);
94	}
95	/* Or if all else fails,
96	 * just print the current hostname */
97	else {
98		gethostname(buf, sizeof(buf));
99		puts(buf);
100	}
101	return 0;
102}
103