X-Git-Url: https://git.librecmc.org/?a=blobdiff_plain;f=networking%2Fhostname.c;h=862bbdfa26afa021fef9bdd15d63e89f7cf561ad;hb=cd0e80ce902f6e79b31888af4ff4545846ce1c0e;hp=4320a968c364e1dfd9f838b2449f6289c71188b6;hpb=3570a34de46b1f7dedd16999bb1687e2d6b55d40;p=oweals%2Fbusybox.git

diff --git a/networking/hostname.c b/networking/hostname.c
index 4320a968c..862bbdfa2 100644
--- a/networking/hostname.c
+++ b/networking/hostname.c
@@ -1,131 +1,102 @@
 /* vi: set sw=4 ts=4: */
 /*
- * $Id: hostname.c,v 1.13 2000/09/25 21:45:57 andersen 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 "busybox.h"
-#include <errno.h>
-#include <arpa/inet.h>
-#include <netdb.h>
-#include <unistd.h>
-#include <stdio.h>
+#include "libbb.h"
 
-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");
+				bb_error_msg_and_die(bb_msg_perm_denied_are_you_root);
 			else
-				perror("sethostname");
-			exit(1);
+				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);
 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[256];
+	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(argc, argv, "dfisF:", &hostname_str);
 
-	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;
+		gethostname(buf, sizeof(buf));
+		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 != NULL) {
+				*p = '\0';
 			}
-			printf("%s\n", inet_ntoa(*(struct in_addr *) (h->h_addr)));
-		} else {
-			printf("%s\n", buf);
+			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++)));
+			}
+			puts("");
 		}
 	}
-	return(0);
+	/* Set the hostname */
+	else if (option_mask32 & OPT_F) {
+		do_sethostname(hostname_str, 1);
+	} else if (optind < argc) {
+		do_sethostname(argv[optind], 0);
+	}
+	/* Or if all else fails,
+	 * just print the current hostname */
+	else {
+		gethostname(buf, sizeof(buf));
+		puts(buf);
+	}
+	return 0;
 }