contrib/amd/fsinfo/fsinfo.c

   1 /*
   2  * Copyright (c) 1997-1999 Erez Zadok
   3  * Copyright (c) 1989 Jan-Simon Pendry
   4  * Copyright (c) 1989 Imperial College of Science, Technology & Medicine
   5  * Copyright (c) 1989 The Regents of the University of California.
   6  * All rights reserved.
   7  *
   8  * This code is derived from software contributed to Berkeley by
   9  * Jan-Simon Pendry at Imperial College, London.
  10  *
  11  * Redistribution and use in source and binary forms, with or without
  12  * modification, are permitted provided that the following conditions
  13  * are met:
  14  * 1. Redistributions of source code must retain the above copyright
  15  *    notice, this list of conditions and the following disclaimer.
  16  * 2. Redistributions in binary form must reproduce the above copyright
  17  *    notice, this list of conditions and the following disclaimer in the
  18  *    documentation and/or other materials provided with the distribution.
  19  * 3. All advertising materials mentioning features or use of this software
  20  *    must display the following acknowledgment:
  21  *      This product includes software developed by the University of
  22  *      California, Berkeley and its contributors.
  23  * 4. Neither the name of the University nor the names of its contributors
  24  *    may be used to endorse or promote products derived from this software
  25  *    without specific prior written permission.
  26  *
  27  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  28  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  29  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  30  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  31  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  32  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  33  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  34  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  35  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  36  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  37  * SUCH DAMAGE.
  38  *
  39  *      %W% (Berkeley) %G%
  40  *
  41  * $Id: fsinfo.c,v 1.4 1999/02/04 07:24:44 ezk Exp $
  42  * $FreeBSD: src/contrib/amd/fsinfo/fsinfo.c,v 1.4 1999/09/15 05:45:15 obrien Exp $
  43  * $DragonFly: src/contrib/amd/fsinfo/fsinfo.c,v 1.2 2003/06/17 04:23:57 dillon Exp $
  44  *
  45  */
  46
  47 /*
  48  * fsinfo
  49  */
  50
  51 #ifdef HAVE_CONFIG_H
  52 # include <config.h>
  53 #endif /* HAVE_CONFIG_H */
  54 #include <am_defs.h>
  55 #include <fsi_data.h>
  56 #include <fsinfo.h>
  57 #include <fsi_gram.h>
  58
  59 /* globals */
  60 char **g_argv;
  61 char *autodir = "/a";
  62 char *progname;
  63 char hostname[MAXHOSTNAMELEN + 1];
  64 char *username;
  65 char idvbuf[1024];
  66 dict *dict_of_hosts;
  67 dict *dict_of_volnames;
  68 int errors;
  69 int file_io_errors;
  70 int parse_errors;
  71 int verbose;
  72 qelem *list_of_automounts;
  73 qelem *list_of_hosts;
  74
  75 /*
  76  * Output file prefixes
  77  */
  78 char *bootparams_pref;
  79 char *dumpset_pref;
  80 char *exportfs_pref;
  81 char *fstab_pref;
  82 char *mount_pref;
  83
  84
  85 /*
  86  * Argument cracking...
  87  */
  88 static void
  89 fsi_get_args(int c, char *v[])
  90 {
  91   int ch;
  92   int usage = 0;
  93   char *iptr = idvbuf;
  94
  95   /*
  96    * Determine program name
  97    */
  98   if (v[0]) {
  99     progname = strrchr(v[0], '/');
 100     if (progname && progname[1])
 101       progname++;
 102     else
 103       progname = v[0];
 104   }
 105
 106   if (!progname)
 107     progname = "fsinfo";
 108
 109   while ((ch = getopt(c, v, "a:b:d:e:f:h:m:D:U:I:qv")) != -1)
 110
 111     switch (ch) {
 112
 113     case 'a':
 114       autodir = optarg;
 115       break;
 116
 117     case 'b':
 118       if (bootparams_pref)
 119         fatal("-b option specified twice");
 120       bootparams_pref = optarg;
 121       break;
 122
 123     case 'd':
 124       if (dumpset_pref)
 125         fatal("-d option specified twice");
 126       dumpset_pref = optarg;
 127       break;
 128
 129     case 'h':
 130       strncpy(hostname, optarg, sizeof(hostname) - 1);
 131       break;
 132
 133     case 'e':
 134       if (exportfs_pref)
 135         fatal("-e option specified twice");
 136       exportfs_pref = optarg;
 137       break;
 138
 139     case 'f':
 140       if (fstab_pref)
 141         fatal("-f option specified twice");
 142       fstab_pref = optarg;
 143       break;
 144
 145     case 'm':
 146       if (mount_pref)
 147         fatal("-m option specified twice");
 148       mount_pref = optarg;
 149       break;
 150
 151     case 'q':
 152       verbose = -1;
 153       break;
 154
 155     case 'v':
 156       verbose = 1;
 157       break;
 158
 159     case 'I':
 160     case 'D':
 161     case 'U':
 162       sprintf(iptr, "-%c%s ", ch, optarg);
 163       iptr += strlen(iptr);
 164       break;
 165
 166     default:
 167       usage++;
 168       break;
 169     }
 170
 171   if (c != optind) {
 172     g_argv = v + optind - 1;
 173     if (yywrap())
 174       fatal("Cannot read any input files");
 175   } else {
 176     usage++;
 177   }
 178
 179   if (usage) {
 180     fprintf(stderr,
 181             "\
 182 Usage: %s [-v] [-a autodir] [-h hostname] [-b bootparams] [-d dumpsets]\n\
 183 \t[-e exports] [-f fstabs] [-m automounts]\n\
 184 \t[-I dir] [-D|-U string[=string]] config ...\n", progname);
 185     exit(1);
 186   }
 187
 188   if (g_argv[0])
 189     log("g_argv[0] = %s", g_argv[0]);
 190   else
 191     log("g_argv[0] = (nil)");
 192 }
 193
 194
 195 /*
 196  * Determine username of caller
 197  */
 198 static char *
 199 find_username(void)
 200 {
 201   char *u = getlogin();
 202
 203   if (!u) {
 204     struct passwd *pw = getpwuid(getuid());
 205     if (pw)
 206       u = pw->pw_name;
 207   }
 208
 209   if (!u)
 210     u = getenv("USER");
 211   if (!u)
 212     u = getenv("LOGNAME");
 213   if (!u)
 214     u = "root";
 215
 216   return strdup(u);
 217 }
 218
 219
 220 /*
 221  * MAIN
 222  */
 223 int
 224 main(int argc, char *argv[])
 225 {
 226   /*
 227    * Process arguments
 228    */
 229   fsi_get_args(argc, argv);
 230
 231   /*
 232    * If no hostname given then use the local name
 233    */
 234   if (!*hostname && gethostname(hostname, sizeof(hostname)) < 0) {
 235     perror("gethostname");
 236     exit(1);
 237   }
 238
 239   /*
 240    * Get the username
 241    */
 242   username = find_username();
 243
 244   /*
 245    * New hosts and automounts
 246    */
 247   list_of_hosts = new_que();
 248   list_of_automounts = new_que();
 249
 250   /*
 251    * New dictionaries
 252    */
 253   dict_of_volnames = new_dict();
 254   dict_of_hosts = new_dict();
 255
 256   /*
 257    * Parse input
 258    */
 259   show_area_being_processed("read config", 11);
 260   if (yyparse())
 261     errors = 1;
 262   errors += file_io_errors + parse_errors;
 263
 264   if (errors == 0) {
 265     /*
 266      * Do semantic analysis of input
 267      */
 268     analyze_hosts(list_of_hosts);
 269     analyze_automounts(list_of_automounts);
 270   }
 271
 272   /*
 273    * Give up if errors
 274    */
 275   if (errors == 0) {
 276     /*
 277      * Output data files
 278      */
 279
 280     write_atab(list_of_automounts);
 281     write_bootparams(list_of_hosts);
 282     write_dumpset(list_of_hosts);
 283     write_exportfs(list_of_hosts);
 284     write_fstab(list_of_hosts);
 285   }
 286   col_cleanup(1);
 287
 288   exit(errors);
 289   return errors; /* should never reach here */
 290 }