/* vi: set sw=4 ts=4: */
/*
- * $Id: hostname.c,v 1.11 2000/07/14 01:51:25 kraai Exp $
* Mini hostname implementation for busybox
*
* Copyright (C) 1999 by Randolph Chung <tausq@debian.org>
*
- * adjusted by Erik Andersen <andersee@debian.org> to remove
+ * adjusted by Erik Andersen <andersen@codepoet.org> to remove
* use of long options and GNU getopt. Improved the usage info.
*
- * This program is free software; you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation; either version 2 of the License, or
- * (at your option) any later version.
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- * General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+ *
+ * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
*/
-#include "internal.h"
-#include <errno.h>
-#include <arpa/inet.h>
-#include <netdb.h>
-#include <unistd.h>
-#include <stdio.h>
-
-static const char *hostname_usage =
- "hostname [OPTION] {hostname | -F file}\n"
-#ifndef BB_FEATURE_TRIVIAL_HELP
- "\nGet or set the hostname or DNS domain name. If a hostname is given\n"
- "(or a file with the -F parameter), the host name will be set.\n\n"
- "Options:\n"
- "\t-s\t\tShort\n"
+#include "libbb.h"
- "\t-i\t\tAddresses for the hostname\n"
- "\t-d\t\tDNS domain name\n"
- "\t-F FILE\t\tUse the contents of FILE to specify the hostname\n"
-#endif
- ;
-
-
-void do_sethostname(char *s, int isfile)
+static void do_sethostname(char *s, int isfile)
{
FILE *f;
- char buf[255];
if (!s)
return;
if (!isfile) {
if (sethostname(s, strlen(s)) < 0) {
if (errno == EPERM)
- errorMsg("you must be root to change the hostname\n");
- else
- perror("sethostname");
- exit(1);
+ bb_error_msg_and_die(bb_msg_perm_denied_are_you_root);
+ bb_perror_msg_and_die("sethostname");
}
} else {
- if ((f = fopen(s, "r")) == NULL) {
- perror(s);
- exit(1);
- } else {
- fgets(buf, 255, f);
- fclose(f);
- if (buf[strlen(buf) - 1] == '\n')
- buf[strlen(buf) - 1] = 0;
- if (sethostname(buf, strlen(buf)) < 0) {
- perror("sethostname");
- exit(1);
+ f = xfopen(s, "r");
+#define strbuf bb_common_bufsiz1
+ while (fgets(strbuf, sizeof(strbuf), f) != NULL) {
+ if (strbuf[0] == '#') {
+ continue;
}
+ chomp(strbuf);
+ do_sethostname(strbuf, 0);
}
+ if (ENABLE_FEATURE_CLEAN_UP)
+ fclose(f);
}
}
+int hostname_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
int hostname_main(int argc, char **argv)
{
- int opt_short = 0;
- int opt_domain = 0;
- int opt_ip = 0;
- struct hostent *h;
- char *filename = NULL;
- char buf[255];
- char *s = NULL;
+ enum {
+ OPT_d = 0x1,
+ OPT_f = 0x2,
+ OPT_i = 0x4,
+ OPT_s = 0x8,
+ OPT_F = 0x10,
+ OPT_dfis = 0xf,
+ };
+
+ char *buf;
+ char *hostname_str;
if (argc < 1)
- usage(hostname_usage);
+ bb_show_usage();
- while (--argc > 0 && **(++argv) == '-') {
- while (*(++(*argv))) {
- switch (**argv) {
- case 's':
- opt_short = 1;
- break;
- case 'i':
- opt_ip = 1;
- break;
- case 'd':
- opt_domain = 1;
- break;
- case 'F':
- if (--argc == 0) {
- usage(hostname_usage);
- }
- filename = *(++argv);
- break;
- default:
- usage(hostname_usage);
- }
- if (filename != NULL)
- break;
- }
- }
+ getopt32(argv, "dfisF:", &hostname_str);
+ argv += optind;
+ buf = safe_gethostname();
- if (argc >= 1) {
- do_sethostname(*argv, 0);
- } else if (filename != NULL) {
- do_sethostname(filename, 1);
- } else {
- gethostname(buf, 255);
- if (opt_short) {
- s = strchr(buf, '.');
- if (!s)
- s = buf;
- *s = 0;
- printf("%s\n", buf);
- } else if (opt_domain) {
- s = strchr(buf, '.');
- printf("%s\n", (s ? s + 1 : ""));
- } else if (opt_ip) {
- h = gethostbyname(buf);
- if (!h) {
- printf("Host not found\n");
- exit(1);
+ /* Output in desired format */
+ if (option_mask32 & OPT_dfis) {
+ struct hostent *hp;
+ char *p;
+ hp = xgethostbyname(buf);
+ p = strchr(hp->h_name, '.');
+ if (option_mask32 & OPT_f) {
+ puts(hp->h_name);
+ } else if (option_mask32 & OPT_s) {
+ if (p)
+ *p = '\0';
+ puts(hp->h_name);
+ } else if (option_mask32 & OPT_d) {
+ if (p)
+ puts(p + 1);
+ } else if (option_mask32 & OPT_i) {
+ while (hp->h_addr_list[0]) {
+ printf("%s ", inet_ntoa(*(struct in_addr *) (*hp->h_addr_list++)));
}
- printf("%s\n", inet_ntoa(*(struct in_addr *) (h->h_addr)));
- } else {
- printf("%s\n", buf);
+ bb_putchar('\n');
}
}
- return(0);
+ /* Set the hostname */
+ else if (option_mask32 & OPT_F) {
+ do_sethostname(hostname_str, 1);
+ } else if (argv[0]) {
+ do_sethostname(argv[0], 0);
+ }
+ /* Or if all else fails,
+ * just print the current hostname */
+ else {
+ puts(buf);
+ }
+ if (ENABLE_FEATURE_CLEAN_UP)
+ free(buf);
+ return 0;
}
projects / oweals / busybox.git / blobdiff