networking/hostname.c

   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 "busybox.h"
  16
  17 static 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                 while (fgets(bb_common_bufsiz1, sizeof(bb_common_bufsiz1), f) != NULL) {
  33                         if (bb_common_bufsiz1[0] == '#') {
  34                                 continue;
  35                         }
  36                         chomp(bb_common_bufsiz1);
  37                         do_sethostname(bb_common_bufsiz1, 0);
  38                 }
  39                 if (ENABLE_FEATURE_CLEAN_UP)
  40                         fclose(f);
  41         }
  42 }
  43
  44 int hostname_main(int argc, char **argv)
  45 {
  46         enum {
  47                 OPT_d = 0x1,
  48                 OPT_f = 0x2,
  49                 OPT_i = 0x4,
  50                 OPT_s = 0x8,
  51                 OPT_F = 0x8,
  52                 OPT_dfis = 0xf,
  53         };
  54
  55         char buf[256];
  56         char *hostname_str;
  57
  58         if (argc < 1)
  59                 bb_show_usage();
  60
  61         getopt32(argc, argv, "dfisF:", &hostname_str);
  62
  63         /* Output in desired format */
  64         if (option_mask32 & OPT_dfis) {
  65                 struct hostent *hp;
  66                 char *p;
  67                 gethostname(buf, sizeof(buf));
  68                 hp = xgethostbyname(buf);
  69                 p = strchr(hp->h_name, '.');
  70                 if (option_mask32 & OPT_f) {
  71                         puts(hp->h_name);
  72                 } else if (option_mask32 & OPT_s) {
  73                         if (p != NULL) {
  74                                 *p = '\0';
  75                         }
  76                         puts(hp->h_name);
  77                 } else if (option_mask32 & OPT_d) {
  78                         if (p)
  79                                 puts(p + 1);
  80                 } else if (option_mask32 & OPT_i) {
  81                         while (hp->h_addr_list[0]) {
  82                                 printf("%s ", inet_ntoa(*(struct in_addr *) (*hp->h_addr_list++)));
  83                         }
  84                         puts("");
  85                 }
  86         }
  87         /* Set the hostname */
  88         else if (option_mask32 & OPT_F) {
  89                 do_sethostname(hostname_str, 1);
  90         } else if (optind < argc) {
  91                 do_sethostname(argv[optind], 0);
  92         }
  93         /* Or if all else fails,
  94          * just print the current hostname */
  95         else {
  96                 gethostname(buf, sizeof(buf));
  97                 puts(buf);
  98         }
  99         return 0;
 100 }