X-Git-Url: https://git.librecmc.org/?a=blobdiff_plain;f=networking%2Fhostname.c;h=6522bb21bbd2a074adc3f40fdbb3e46872270284;hb=fbf0b8af41eb2662c40d0fb932ae85c01ea0743e;hp=3c22cf8c0adb277399c74d378b0f1b5f5afe1ecb;hpb=d29edf34d6432a310dbb8124cd7963c34d31a69c;p=oweals%2Fbusybox.git

diff --git a/networking/hostname.c b/networking/hostname.c
index 3c22cf8c0..6522bb21b 100644
--- a/networking/hostname.c
+++ b/networking/hostname.c
@@ -1,5 +1,6 @@
+/* vi: set sw=4 ts=4: */
 /*
- * $Id: hostname.c,v 1.2 1999/12/08 04:13:44 andersen Exp $
+ * $Id: hostname.c,v 1.35 2003/03/19 09:12:37 mjn3 Exp $
  * Mini hostname implementation for busybox
  *
  * Copyright (C) 1999 by Randolph Chung <tausq@debian.org>
@@ -22,117 +23,108 @@
  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  */
 
-#include "internal.h"
 #include <errno.h>
 #include <arpa/inet.h>
 #include <netdb.h>
 #include <unistd.h>
+#include <string.h>
 #include <stdio.h>
+#include <stdlib.h>
+#include <getopt.h>
+#include "busybox.h"
 
-static const char* hostname_usage = 
-"hostname [OPTION] {hostname | -F file}\n\n"
-"Get 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"
-"\t-i\t\tAddresses for the hostname\n"
-"\t-d\t\tDNS domain name\n"
-"\t-F file\tUse FILE to specify the hostname\n";
+extern char *optarg; /* in unistd.h */
+extern int  optind, opterr, optopt; /* in unistd.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) 
-	        fprintf(stderr, "hostname: you must be root to change the hostname\n");
-	    else	  	
-	        perror("sethostname");
-	    exit(1);
-	}	
-    } else {
-        if ((f = fopen(s, "r")) == NULL) {
-	    perror(s);
-	    exit(1);
+	FILE *f;
+	char buf[255];
+
+	if (!s)
+		return;
+	if (!isfile) {
+		if (sethostname(s, strlen(s)) < 0) {
+			if (errno == EPERM)
+				bb_error_msg_and_die("you must be root to change the hostname");
+			else
+				bb_perror_msg_and_die("sethostname");
+		}
 	} 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 = bb_xfopen(s, "r");
+		while (fgets(buf, 255, f) != NULL) {
+			if (buf[0] =='#') {
+				continue;
+			}
+			chomp(buf);
+			do_sethostname(buf, 0);
+		}
+#ifdef CONFIG_FEATURE_CLEAN_UP
+		fclose(f);
+#endif
 	}
-    }	
 }
 
 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;
-    
-    if (argc < 1) usage(hostname_usage);
+	int opt;
+	int type = 0;
+	struct hostent *hp;
+	char *filename = NULL;
+	char buf[255];
+	char *p = NULL;
+
+	if (argc < 1)
+		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':
-		filename = optarg;
-		if (--argc == 0) {
-		    usage(hostname_usage);
+        while ((opt = getopt(argc, argv, "dfisF:")) > 0) {
+                switch (opt) {
+		case 'd':
+		case 'f':
+		case 'i':
+		case 's':
+			type = opt;
+			break;
+		case 'F':
+			filename = optarg;
+			break;
+		default:
+			bb_show_usage();
 		}
-		filename = *(++argv);
-		break;
-	    default:
-		usage(hostname_usage);
-	    }
-	    if (filename!=NULL)
-		break;
 	}
-    }
 
-    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);
-	    }
-	    printf("%s\n", inet_ntoa(*(struct in_addr *)(h->h_addr)));
-        } else {
-	  printf("%s\n", buf);
-        }
-    }
-    exit( 0);
+	/* Output in desired format */
+	if (type != 0) {
+		gethostname(buf, 255);
+		hp = xgethostbyname(buf);
+		p = strchr(hp->h_name, '.');
+		if (type == 'f') {
+			puts(hp->h_name);
+		} else if (type == 's') {
+			if (p != NULL) {
+				*p = 0;
+			}
+			puts(hp->h_name);
+		} else if (type == 'd') {
+			if (p) puts(p + 1);
+		} else if (type == 'i') {
+			while (hp->h_addr_list[0]) {
+				printf("%s ", inet_ntoa(*(struct in_addr *) (*hp->h_addr_list++)));
+			}
+			printf("\n");
+		}
+	}
+	/* Set the hostname */
+	else if (filename != NULL) {
+		do_sethostname(filename, 1);
+	} else if (optind < argc) {
+		do_sethostname(argv[optind], 0);
+	}
+	/* Or if all else fails,
+	 * just print the current hostname */
+	 else {
+		gethostname(buf, 255);
+		puts(buf);
+	}
+	return(0);
 }
-