contrib/amd/hlfsd/hlfsd.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: hlfsd.c,v 1.5 1999/09/08 23:36:51 ezk Exp $
  42  * $FreeBSD: src/contrib/amd/hlfsd/hlfsd.c,v 1.5 1999/09/15 05:45:15 obrien Exp $
  43  * $DragonFly: src/contrib/amd/hlfsd/hlfsd.c,v 1.2 2003/06/17 04:23:57 dillon Exp $
  44  *
  45  * HLFSD was written at Columbia University Computer Science Department, by
  46  * Erez Zadok <ezk@cs.columbia.edu> and Alexander Dupuy <dupuy@cs.columbia.edu>
  47  * It is being distributed under the same terms and conditions as amd does.
  48  */
  49
  50 #ifdef HAVE_CONFIG_H
  51 # include <config.h>
  52 #endif /* HAVE_CONFIG_H */
  53 #include <am_defs.h>
  54 #include <hlfsd.h>
  55
  56 /*
  57  * STATIC VARIABLES:
  58  */
  59 static RETSIGTYPE proceed(int);
  60 static RETSIGTYPE reaper(int);
  61 static RETSIGTYPE reload(int);
  62 static char *hlfs_group = DEFAULT_HLFS_GROUP;
  63 static char default_dir_name[] = DEFAULT_DIRNAME;
  64 static char *dir_name = default_dir_name;
  65 static int printpid = 0;
  66 static int stoplight = 0;
  67 static void hlfsd_init(void);
  68 static void usage(void);
  69
  70 static struct itimerval reloadinterval = {
  71   {DEFAULT_INTERVAL, 0},
  72   {DEFAULT_INTERVAL, 0}
  73 };
  74
  75 /*
  76  * default mount options.
  77  */
  78 static char default_mntopts[] = "ro,noac";
  79
  80 /*
  81  * GLOBALS:
  82  */
  83 SVCXPRT *nfsxprt;
  84 char *alt_spooldir = ALT_SPOOLDIR;
  85 char *home_subdir = HOME_SUBDIR;
  86 char *logfile = DEFAULT_LOGFILE;
  87 char *passwdfile = NULL;        /* alternate passwd file to use */
  88 char *slinkname = 0;
  89 char hostname[MAXHOSTNAMELEN + 1] = "localhost";
  90 int cache_interval = DEFAULT_CACHE_INTERVAL;
  91 gid_t hlfs_gid = (gid_t) INVALIDID;
  92 int masterpid = 0;
  93 int noverify = 0;
  94 int orig_umask = 022;
  95 int serverpid = 0;
  96 nfstime startup;
  97 u_short nfs_port;
  98
  99 /* symbol must be available always */
 100 #ifdef MOUNT_TABLE_ON_FILE
 101 char *mnttab_file_name = MNTTAB_FILE_NAME;
 102 #else /* not MOUNT_TABLE_ON_FILE */
 103 char *mnttab_file_name = NULL;
 104 #endif /* not MOUNT_TABLE_ON_FILE */
 105
 106 /* forward declarations */
 107 void hlfsd_going_down(int rc);
 108
 109
 110 static void
 111 usage(void)
 112 {
 113   fprintf(stderr,
 114           "Usage: %s [-Cfhnpv] [-a altdir] [-c cache-interval] [-g group]\n",
 115           am_get_progname());
 116   fprintf(stderr, "\t[-i interval] [-l logfile] [-o mntopts] [-P passwdfile]\n");
 117   show_opts('x', xlog_opt);
 118 #ifdef DEBUG
 119   show_opts('D', dbg_opt);
 120 #endif /* DEBUG */
 121   fprintf(stderr, "\t[dir_name [subdir]]\n");
 122   exit(2);
 123 }
 124
 125
 126 int
 127 main(int argc, char *argv[])
 128 {
 129   char *dot;
 130   char *mntopts = (char *) NULL;
 131   char hostpid_fs[MAXHOSTNAMELEN + 1 + 16];     /* room for ":(pid###)" */
 132   char progpid_fs[PROGNAMESZ + 1 + 11];         /* room for ":pid" */
 133   char preopts[128];
 134   char *progname;
 135   int forcecache = 0;
 136   int forcefast = 0;
 137   int genflags = 0;
 138   int opt, ret;
 139   int opterrs = 0;
 140   int retry;
 141   int soNFS;                    /* NFS socket */
 142   int s = -99;
 143   mntent_t mnt;
 144   nfs_args_t nfs_args;
 145   am_nfs_handle_t anh;
 146   struct dirent *direntry;
 147   struct group *grp;
 148   struct stat stmodes;
 149   DIR *mountdir;
 150   MTYPE_TYPE type = MOUNT_TYPE_NFS;
 151
 152 #ifdef HAVE_SIGACTION
 153   struct sigaction sa;
 154 #endif /* not HAVE_SIGACTION */
 155
 156 #ifndef HAVE_TRANSPORT_TYPE_TLI
 157   struct sockaddr_in localsocket;
 158 #endif /* not HAVE_TRANSPORT_TYPE_TLI */
 159
 160
 161   /* get program name and truncate so we don't overflow progpid_fs */
 162
 163   if ((progname = strrchr(argv[0], '/')) != NULL)
 164     progname++;
 165   else
 166     progname = argv[0];
 167   if ((int) strlen(progname) > PROGNAMESZ) /* truncate to reasonable size */
 168     progname[PROGNAMESZ] = '\0';
 169   am_set_progname(progname);
 170
 171   while ((opt = getopt(argc, argv, "a:c:CD:fg:hi:l:no:pP:x:v")) != -1)
 172     switch (opt) {
 173
 174     case 'a':
 175       if (!optarg || optarg[0] != '/') {
 176         printf("%s: invalid directory for -a: %s\n",
 177                am_get_progname(), optarg);
 178         exit(3);
 179       }
 180       alt_spooldir = optarg;
 181       break;
 182
 183     case 'c':
 184       if (!atoi(optarg)) {
 185         printf("%s: invalid interval for -c: %s\n",
 186                am_get_progname(), optarg);
 187         exit(3);
 188       }
 189       cache_interval = atoi(optarg);
 190       break;
 191
 192     case 'C':
 193       forcecache++;
 194       break;
 195
 196     case 'f':
 197       forcefast++;
 198       break;
 199
 200     case 'g':
 201       hlfs_group = optarg;
 202       break;
 203
 204     case 'i':
 205       if (!atoi(optarg)) {
 206         printf("%s: invalid interval for -i: %s\n",
 207                am_get_progname(), optarg);
 208         exit(3);
 209       }
 210       reloadinterval.it_interval.tv_sec = atoi(optarg);
 211       reloadinterval.it_value.tv_sec = atoi(optarg);
 212       break;
 213
 214     case 'l':
 215       logfile = optarg;
 216       break;
 217
 218     case 'n':
 219       noverify++;
 220       break;
 221
 222     case 'o':
 223       mntopts = optarg;
 224       break;
 225
 226     case 'p':
 227       printpid++;
 228       break;
 229
 230     case 'P':
 231       passwdfile = optarg;
 232       break;
 233
 234     case 'v':
 235       fprintf(stderr, "%s\n", HLFSD_VERSION);
 236       exit(0);
 237
 238     case 'x':
 239       opterrs += switch_option(optarg);
 240       break;
 241
 242     case 'D':
 243 #ifdef DEBUG
 244       opterrs += debug_option(optarg);
 245 #else /* not DEBUG */
 246       fprintf(stderr, "%s: not compiled with DEBUG -- sorry.\n", am_get_progname());
 247 #endif /* not DEBUG */
 248       break;
 249
 250     case 'h':
 251     case '?':
 252       opterrs++;
 253     }
 254
 255   /* set some default debugging options */
 256   if (xlog_level_init == ~0)
 257     switch_option("");
 258   /* need my pid before any dlog/plog */
 259   am_set_mypid();
 260 #ifdef DEBUG
 261   switch_option("debug");
 262 #endif /* DEBUG */
 263
 264 /*
 265  * Terminate if did not ask to forcecache (-C) and hlfsd would not be able
 266  * to set the minimum cache intervals.
 267  */
 268 #if !defined(MNT2_NFS_OPT_ACREGMIN) && !defined(MNT2_NFS_OPT_NOAC) && !defined(HAVE_FIELD_NFS_ARGS_T_ACREGMIN)
 269   if (!forcecache) {
 270     fprintf(stderr, "%s: will not be able to turn off attribute caches.\n", am_get_progname());
 271     exit(1);
 272   }
 273 #endif /* !defined(MNT2_NFS_OPT_ACREGMIN) && !defined(MNT2_NFS_OPT_NOAC) && !defined(HAVE_FIELD_NFS_ARGS_T_ACREGMIN) */
 274
 275
 276   switch (argc - optind) {
 277   case 2:
 278     home_subdir = argv[optind + 1];
 279   case 1:
 280     dir_name = argv[optind];
 281   case 0:
 282     break;
 283   default:
 284     opterrs++;
 285   }
 286
 287   if (opterrs)
 288     usage();
 289
 290   /* ensure that only root can run hlfsd */
 291   if (geteuid()) {
 292     fprintf(stderr, "hlfsd can only be run as root\n");
 293     exit(1);
 294   }
 295   setbuf(stdout, (char *) NULL);
 296   umask(0);
 297
 298   /* find gid for hlfs_group */
 299   if ((grp = getgrnam(hlfs_group)) == (struct group *) NULL) {
 300     fprintf(stderr, "%s: cannot get gid for group \"%s\".\n",
 301             am_get_progname(), hlfs_group);
 302   } else {
 303     hlfs_gid = grp->gr_gid;
 304   }
 305
 306   /* get hostname for logging and open log before we reset umask */
 307   gethostname(hostname, sizeof(hostname));
 308   hostname[sizeof(hostname) - 1] = '\0';
 309   if ((dot = strchr(hostname, '.')) != NULL)
 310     *dot = '\0';
 311   orig_umask = umask(0);
 312   if (logfile)
 313     switch_to_logfile(logfile, orig_umask);
 314
 315 #if defined(DEBUG) && !defined(MOUNT_TABLE_ON_FILE)
 316   if (debug_flags & D_MTAB)
 317     dlog("-D mtab option ignored");
 318 #endif /* defined(DEBUG) && !defined(MOUNT_TABLE_ON_FILE) */
 319
 320   /* avoid hanging on other NFS servers if started elsewhere */
 321   if (chdir("/") < 0)
 322     fatal("cannot chdir to /: %m");
 323
 324   if (geteuid() != 0)
 325     fatal("must be root to mount filesystems");
 326
 327   /*
 328    * dir_name must match "^(/.*)/([^/]+)$", and is split at last '/' with
 329    * slinkname = `basename $dir_name` - requires dir_name be writable
 330    */
 331
 332   if (dir_name[0] != '/'
 333       || ((slinkname = strrchr(dir_name, '/')), *slinkname++ = '\0',
 334           (dir_name[0] == '\0' || slinkname[0] == '\0'))) {
 335     if (slinkname)
 336       *--slinkname = '/';
 337     printf("%s: invalid mount directory/link %s\n",
 338            am_get_progname(), dir_name);
 339     exit(3);
 340   }
 341
 342   clock_valid = 0;              /* invalidate logging clock */
 343
 344   if (!forcefast) {
 345     /* make sure mount point exists and is at least mode 555 */
 346     if (stat(dir_name, &stmodes) < 0)
 347       if (errno != ENOENT || mkdirs(dir_name, 0555) < 0
 348           || stat(dir_name, &stmodes) < 0)
 349         fatalerror(dir_name);
 350
 351     if ((stmodes.st_mode & 0555) != 0555) {
 352       fprintf(stderr, "%s: directory %s not read/executable\n",
 353               am_get_progname(), dir_name);
 354       plog(XLOG_WARNING, "directory %s not read/executable",
 355            dir_name);
 356     }
 357
 358     /* warn if extraneous stuff will be hidden by mount */
 359     if ((mountdir = opendir(dir_name)) == NULL)
 360       fatalerror(dir_name);
 361
 362     while ((direntry = readdir(mountdir)) != NULL) {
 363       if (!NSTREQ(".", direntry->d_name, NAMLEN(direntry)) &&
 364           !NSTREQ("..", direntry->d_name, NAMLEN(direntry)) &&
 365           !NSTREQ(slinkname, direntry->d_name, NAMLEN(direntry)))
 366         break;
 367     }
 368
 369     if (direntry != NULL) {
 370       fprintf(stderr, "%s: %s/%s will be hidden by mount\n",
 371               am_get_progname(), dir_name, direntry->d_name);
 372       plog(XLOG_WARNING, "%s/%s will be hidden by mount\n",
 373            dir_name, direntry->d_name);
 374     }
 375     closedir(mountdir);
 376
 377     /* make sure alternate spool dir exists */
 378     if ((errno = mkdirs(alt_spooldir, OPEN_SPOOLMODE))) {
 379       fprintf(stderr, "%s: cannot create alternate dir ",
 380               am_get_progname());
 381       perror(alt_spooldir);
 382       plog(XLOG_ERROR, "cannot create alternate dir %s: %m",
 383            alt_spooldir);
 384     }
 385     chmod(alt_spooldir, OPEN_SPOOLMODE);
 386
 387     /* create failsafe link to alternate spool directory */
 388     slinkname[-1] = '/';        /* unsplit dir_name to include link */
 389     if (lstat(dir_name, &stmodes) == 0 &&
 390         (stmodes.st_mode & S_IFMT) != S_IFLNK) {
 391       fprintf(stderr, "%s: failsafe %s not a symlink\n",
 392               am_get_progname(), dir_name);
 393       plog(XLOG_WARNING, "failsafe %s not a symlink\n",
 394            dir_name);
 395     } else {
 396       unlink(dir_name);
 397
 398       if (symlink(alt_spooldir, dir_name) < 0) {
 399         fprintf(stderr,
 400                 "%s: cannot create failsafe symlink %s -> ",
 401                 am_get_progname(), dir_name);
 402         perror(alt_spooldir);
 403         plog(XLOG_WARNING,
 404              "cannot create failsafe symlink %s -> %s: %m",
 405              dir_name, alt_spooldir);
 406       }
 407     }
 408
 409     slinkname[-1] = '\0';       /* resplit dir_name */
 410   } /* end of "if (!forcefast) {" */
 411
 412   /*
 413    * Register hlfsd as an nfs service with the portmapper.
 414    */
 415 #ifdef HAVE_TRANSPORT_TYPE_TLI
 416   ret = create_nfs_service(&soNFS, &nfs_port, &nfsxprt, nfs_program_2);
 417 #else /* not HAVE_TRANSPORT_TYPE_TLI */
 418   ret = create_nfs_service(&soNFS, &nfs_port, &nfsxprt, nfs_program_2);
 419 #endif /* not HAVE_TRANSPORT_TYPE_TLI */
 420   if (ret != 0)
 421     fatal("cannot create NFS service");
 422
 423 #ifdef HAVE_SIGACTION
 424   sa.sa_handler = proceed;
 425   sa.sa_flags = 0;
 426   sigemptyset(&(sa.sa_mask));
 427   sigaddset(&(sa.sa_mask), SIGUSR2);
 428   sigaction(SIGUSR2, &sa, NULL);
 429 #else /* not HAVE_SIGACTION */
 430   signal(SIGUSR2, proceed);
 431 #endif /* not HAVE_SIGACTION */
 432
 433   plog(XLOG_INFO, "Initializing hlfsd...");
 434   hlfsd_init();                 /* start up child (forking) to run svc_run */
 435
 436 #ifdef HAVE_SIGACTION
 437   sa.sa_handler = reaper;
 438   sa.sa_flags = 0;
 439   sigemptyset(&(sa.sa_mask));
 440   sigaddset(&(sa.sa_mask), SIGCHLD);
 441   sigaction(SIGCHLD, &sa, NULL);
 442 #else /* not HAVE_SIGACTION */
 443   signal(SIGCHLD, reaper);
 444 #endif /* not HAVE_SIGACTION */
 445
 446 #ifdef DEBUG
 447   /*
 448    * In the parent, if -D nodaemon (or -D daemon) , we don't need to
 449    * set this signal handler.
 450    */
 451   amuDebug(D_DAEMON) {
 452 #endif /* DEBUG */
 453     /* XXX: port to use pure svr4 signals */
 454     s = -99;
 455     while (stoplight != SIGUSR2) {
 456       plog(XLOG_INFO, "parent waits for child to setup (stoplight=%d)", stoplight);
 457       s = sigpause(0);          /* wait for child to set up */
 458       sleep(1);
 459     }
 460 #ifdef DEBUG
 461   }
 462 #endif /* DEBUG */
 463
 464   /*
 465    * setup options to mount table (/etc/{mtab,mnttab}) entry
 466    */
 467   sprintf(hostpid_fs, "%s:(pid%d)", hostname, masterpid);
 468   memset((char *) &mnt, 0, sizeof(mnt));
 469   mnt.mnt_dir = dir_name;       /* i.e., "/mail" */
 470   mnt.mnt_fsname = hostpid_fs;
 471   if (mntopts) {
 472     mnt.mnt_opts = mntopts;
 473   } else {
 474     strcpy(preopts, default_mntopts);
 475     /*
 476      * Turn off all kinds of attribute and symlink caches as
 477      * much as possible.  Also make sure that mount does not
 478      * show up to df.
 479      */
 480 #ifdef MNTTAB_OPT_INTR
 481     strcat(preopts, ",");
 482     strcat(preopts, MNTTAB_OPT_INTR);
 483 #endif /* MNTTAB_OPT_INTR */
 484 #ifdef MNTTAB_OPT_IGNORE
 485     strcat(preopts, ",");
 486     strcat(preopts, MNTTAB_OPT_IGNORE);
 487 #endif /* MNTTAB_OPT_IGNORE */
 488 #ifdef MNT2_GEN_OPT_CACHE
 489     strcat(preopts, ",nocache");
 490 #endif /* MNT2_GEN_OPT_CACHE */
 491 #ifdef MNT2_NFS_OPT_SYMTTL
 492     strcat(preopts, ",symttl=0");
 493 #endif /* MNT2_NFS_OPT_SYMTTL */
 494     mnt.mnt_opts = preopts;
 495   }
 496
 497   /*
 498    * Make sure that amd's top-level NFS mounts are hidden by default
 499    * from df.
 500    * If they don't appear to support the either the "ignore" mnttab
 501    * option entry, or the "auto" one, set the mount type to "nfs".
 502    */
 503   mnt.mnt_type = HIDE_MOUNT_TYPE;
 504   /* some systems don't have a mount type, but a mount flag */
 505
 506 #ifndef HAVE_TRANSPORT_TYPE_TLI
 507   amu_get_myaddress(&localsocket.sin_addr);
 508   localsocket.sin_family = AF_INET;
 509   localsocket.sin_port = htons(nfsxprt->xp_port);
 510 #endif /* not HAVE_TRANSPORT_TYPE_TLI */
 511
 512   /*
 513    * Update hostname field.
 514    * Make some name prog:pid (i.e., hlfsd:174) for hostname
 515    */
 516   sprintf(progpid_fs, "%s:%d", am_get_progname(), masterpid);
 517
 518   /* Most kernels have a name length restriction. */
 519   if ((int) strlen(progpid_fs) >= (int) MAXHOSTNAMELEN)
 520     strcpy(progpid_fs + MAXHOSTNAMELEN - 3, "..");
 521
 522   genflags = compute_mount_flags(&mnt);
 523
 524   retry = hasmntval(&mnt, MNTTAB_OPT_RETRY);
 525   if (retry <= 0)
 526     retry = 1;                  /* XXX */
 527
 528   memmove(&anh.v2.fhs_fh, root_fhp, sizeof(*root_fhp));
 529 #ifdef HAVE_TRANSPORT_TYPE_TLI
 530   compute_nfs_args(&nfs_args,
 531                    &mnt,
 532                    genflags,
 533                    nfsncp,
 534                    NULL,        /* remote host IP addr is set below */
 535                    NFS_VERSION, /* version 2 */
 536                    "udp",       /* XXX: shouldn't this be "udp"? */
 537                    &anh,
 538                    progpid_fs,  /* host name for kernel */
 539                    hostpid_fs); /* filesystem name for kernel */
 540   /*
 541    * IMPORTANT: set the correct IP address AFTERWARDS.  It cannot
 542    * be done using the normal mechanism of compute_nfs_args(), because
 543    * that one will allocate a new address and use NFS_SA_DREF() to copy
 544    * parts to it, while assuming that the ip_addr passed is always
 545    * a "struct sockaddr_in".  That assumption is incorrect on TLI systems,
 546    * because they define a special macro HOST_SELF which is DIFFERENT
 547    * than localhost (127.0.0.1)!
 548    */
 549   nfs_args.addr = &nfsxprt->xp_ltaddr;
 550 #else /* not HAVE_TRANSPORT_TYPE_TLI */
 551   compute_nfs_args(&nfs_args,
 552                    &mnt,
 553                    genflags,
 554                    &localsocket,
 555                    NFS_VERSION, /* version 2 */
 556                    "udp",       /* XXX: shouldn't this be "udp"? */
 557                    &anh,
 558                    progpid_fs,  /* host name for kernel */
 559                    hostpid_fs); /* filesystem name for kernel */
 560 #endif /* not HAVE_TRANSPORT_TYPE_TLI */
 561
 562   /*************************************************************************
 563    * NOTE: while compute_nfs_args() works ok for regular NFS mounts        *
 564    * the toplvl one is not, and so some options must be corrected by hand  *
 565    * more carefully, *after* compute_nfs_args() runs.                      *
 566    *************************************************************************/
 567   compute_automounter_nfs_args(&nfs_args, &mnt);
 568
 569   clock_valid = 0;              /* invalidate logging clock */
 570
 571 /*
 572  * The following code could be cleverly ifdef-ed, but I duplicated the
 573  * mount_fs call three times for simplicity and readability.
 574  */
 575 #ifdef DEBUG
 576 /*
 577  * For some reason, this mount may have to be done in the background, if I am
 578  * using -D nodebug.  I suspect that the actual act of mounting requires
 579  * calling to hlfsd itself to invoke one or more of its nfs calls, to stat
 580  * /mail.  That means that even if you say -D nodaemon, at least the mount
 581  * of hlfsd itself on top of /mail will be done in the background.
 582  * The other alternative I have is to run svc_run, but set a special
 583  * signal handler to perform the mount in N seconds via some alarm.
 584  *      -Erez Zadok.
 585  */
 586   if (debug_flags & D_DAEMON) { /* asked for -D daemon */
 587     plog(XLOG_INFO, "parent NFS mounting hlfsd service points");
 588     if (mount_fs(&mnt, genflags, (caddr_t) &nfs_args, retry, type, 0, NULL, mnttab_file_name) < 0)
 589       fatal("nfsmount: %m");
 590   } else {                      /* asked for -D nodaemon */
 591     if (fork() == 0) {          /* child runs mount */
 592       am_set_mypid();
 593       foreground = 0;
 594       plog(XLOG_INFO, "child NFS mounting hlfsd service points");
 595       if (mount_fs(&mnt, genflags, (caddr_t) &nfs_args, retry, type, 0, NULL, mnttab_file_name) < 0) {
 596         fatal("nfsmount: %m");
 597       }
 598       exit(0);                  /* all went well */
 599     } else { /* fork failed or parent running */
 600       plog(XLOG_INFO, "parent waiting 1sec for mount...");
 601     }
 602   }
 603 #else /* not DEBUG */
 604   plog(XLOG_INFO, "normal NFS mounting hlfsd service points");
 605   if (mount_fs(&mnt, genflags, (caddr_t) &nfs_args, retry, type, 2, "udp", mnttab_file_name) < 0)
 606     fatal("nfsmount: %m");
 607 #endif /* not DEBUG */
 608
 609 #ifdef HAVE_TRANSPORT_TYPE_TLI
 610   /*
 611    * XXX: this free_knetconfig() was not done for hlfsd before,
 612    * and apparently there was a reason for it, but why? -Erez
 613    */
 614   free_knetconfig(nfs_args.knconf);
 615   /*
 616    * local automounter mounts do not allocate a special address, so
 617    * no need to XFREE(nfs_args.addr) under TLI.
 618    */
 619 #endif /* HAVE_TRANSPORT_TYPE_TLI */
 620
 621   if (printpid)
 622     printf("%d\n", masterpid);
 623
 624   plog(XLOG_INFO, "hlfsd ready to serve");
 625 #ifdef DEBUG
 626   /*
 627    * If asked not to fork a daemon (-D nodaemon), then hlfsd_init()
 628    * will not run svc_run.  We must start svc_run here.
 629    */
 630   dlog("starting no-daemon debugging svc_run");
 631   amuDebugNo(D_DAEMON)
 632     svc_run();
 633 #endif /* DEBUG */
 634
 635   cleanup(0);                   /* should never happen here */
 636   return (0);                   /* everything went fine? */
 637 }
 638
 639
 640 static void
 641 hlfsd_init(void)
 642 {
 643   int child = 0;
 644 #ifdef HAVE_SIGACTION
 645   struct sigaction sa;
 646 #endif /* HAVE_SIGACTION */
 647
 648   clock_valid = 0;              /* invalidate logging clock */
 649
 650   /*
 651    * Initialize file handles.
 652    */
 653   plog(XLOG_INFO, "initializing hlfsd file handles");
 654   hlfsd_init_filehandles();
 655
 656 #ifdef DEBUG
 657   /*
 658    * If -D daemon then we must fork.
 659    */
 660   amuDebug(D_DAEMON)
 661 #endif /* DEBUG */
 662     child = fork();
 663
 664   if (child < 0)
 665     fatal("fork: %m");
 666
 667   if (child != 0) {             /* parent process - save child pid */
 668     masterpid = child;
 669     am_set_mypid();             /* for logging routines */
 670     return;
 671   }
 672
 673   /*
 674    * CHILD CODE:
 675    * initialize server
 676    */
 677
 678   plog(XLOG_INFO, "initializing home directory database");
 679   plt_init();                   /* initialize database */
 680   plog(XLOG_INFO, "home directory database initialized");
 681
 682   masterpid = serverpid = am_set_mypid(); /* for logging routines */
 683
 684   /*
 685    * SIGALRM/SIGHUP: reload password database if timer expired
 686    * or user sent HUP signal.
 687    */
 688 #ifdef HAVE_SIGACTION
 689   sa.sa_handler = reload;
 690   sa.sa_flags = 0;
 691   sigemptyset(&(sa.sa_mask));
 692   sigaddset(&(sa.sa_mask), SIGALRM);
 693   sigaddset(&(sa.sa_mask), SIGHUP);
 694   sigaction(SIGALRM, &sa, NULL);
 695   sigaction(SIGHUP, &sa, NULL);
 696 #else /* not HAVE_SIGACTION */
 697   signal(SIGALRM, reload);
 698   signal(SIGHUP, reload);
 699 #endif /* not HAVE_SIGACTION */
 700
 701   /*
 702    * SIGTERM: cleanup and exit.
 703    */
 704 #ifdef HAVE_SIGACTION
 705   sa.sa_handler = cleanup;
 706   sa.sa_flags = 0;
 707   sigemptyset(&(sa.sa_mask));
 708   sigaddset(&(sa.sa_mask), SIGTERM);
 709   sigaction(SIGTERM, &sa, NULL);
 710 #else /* not HAVE_SIGACTION */
 711   signal(SIGTERM, cleanup);
 712 #endif /* not HAVE_SIGACTION */
 713
 714   /*
 715    * SIGCHLD: interlock synchronization and testing
 716    */
 717 #ifdef HAVE_SIGACTION
 718   sa.sa_handler = interlock;
 719   sa.sa_flags = 0;
 720   sigemptyset(&(sa.sa_mask));
 721   sigaddset(&(sa.sa_mask), SIGCHLD);
 722   sigaction(SIGCHLD, &sa, NULL);
 723 #else /* not HAVE_SIGACTION */
 724   signal(SIGCHLD, interlock);
 725 #endif /* not HAVE_SIGACTION */
 726
 727   /*
 728    * SIGUSR1: dump internal hlfsd maps/cache to file
 729    */
 730 #ifdef HAVE_SIGACTION
 731 # if defined(DEBUG) || defined(DEBUG_PRINT)
 732   sa.sa_handler = plt_print;
 733 # else /* not defined(DEBUG) || defined(DEBUG_PRINT) */
 734   sa.sa_handler = SIG_IGN;
 735 # endif /* not defined(DEBUG) || defined(DEBUG_PRINT) */
 736   sa.sa_flags = 0;
 737   sigemptyset(&(sa.sa_mask));
 738   sigaddset(&(sa.sa_mask), SIGUSR1);
 739   sigaction(SIGUSR1, &sa, NULL);
 740 #else /* not HAVE_SIGACTION */
 741 # if defined(DEBUG) || defined(DEBUG_PRINT)
 742   signal(SIGUSR1, plt_print);
 743 # else /* not defined(DEBUG) || defined(DEBUG_PRINT) */
 744   signal(SIGUSR1, SIG_IGN);
 745 # endif /* not defined(DEBUG) || defined(DEBUG_PRINT) */
 746 #endif /* not HAVE_SIGACTION */
 747
 748   if (setitimer(ITIMER_REAL, &reloadinterval, (struct itimerval *) 0) < 0)
 749     fatal("setitimer: %m");
 750
 751   gettimeofday((struct timeval *) &startup, (struct timezone *) 0);
 752
 753 #ifdef DEBUG
 754   /*
 755    * If -D daemon, then start serving here in the child,
 756    * and the parent will exit.  But if -D nodaemon, then
 757    * skip this code and make sure svc_run is entered elsewhere.
 758    */
 759   amuDebug(D_DAEMON) {
 760 #endif /* DEBUG */
 761
 762     /*
 763      * Dissociate from the controlling terminal
 764      */
 765     amu_release_controlling_tty();
 766
 767     /*
 768      * signal parent we are ready. parent should
 769      * mount(2) and die.
 770      */
 771     if (kill(getppid(), SIGUSR2) < 0)
 772       fatal("kill: %m");
 773     plog(XLOG_INFO, "starting svc_run");
 774     svc_run();
 775     cleanup(0);         /* should never happen, just in case */
 776 #ifdef DEBUG
 777   } /* end of code that runs iff hlfsd daemonizes */
 778 #endif /* DEBUG */
 779
 780 }
 781
 782
 783 static RETSIGTYPE
 784 proceed(int signum)
 785 {
 786   stoplight = signum;
 787 }
 788
 789
 790 static RETSIGTYPE
 791 reload(int signum)
 792 {
 793   int child;
 794   int status;
 795
 796   clock_valid = 0;              /* invalidate logging clock */
 797
 798   if (getpid() != masterpid)
 799     return;
 800
 801   /*
 802    * If received a SIGHUP, close and reopen the log file (so that it
 803    * can be rotated)
 804    */
 805   if (signum == SIGHUP && logfile)
 806     switch_to_logfile(logfile, orig_umask);
 807
 808   /*
 809    * parent performs the reload, while the child continues to serve
 810    * clients accessing the home dir link.
 811    */
 812   if ((child = fork()) > 0) {
 813     serverpid = child;          /* parent runs here */
 814     am_set_mypid();
 815
 816     plt_init();
 817
 818     if (kill(child, SIGKILL) < 0) {
 819       plog(XLOG_ERROR, "kill child: %m");
 820     } else {                    /* wait for child to die before continue */
 821       if (wait(&status) != child) {
 822         /*
 823          * I took out this line because it generates annoying output.  It
 824          * indicates a very small bug in hlfsd which is totally harmless.
 825          * It causes hlfsd to work a bit harder than it should.
 826          * Nevertheless, I intend on fixing it in a future release.
 827          * -Erez Zadok <ezk@cs.columbia.edu>
 828          */
 829         /* plog(XLOG_ERROR, "unknown child"); */
 830       }
 831     }
 832     serverpid = masterpid;
 833   } else if (child < 0) {
 834     plog(XLOG_ERROR, "unable to fork: %m");
 835   } else {
 836     /* let child handle requests while we reload */
 837     serverpid = getpid();
 838     am_set_mypid();
 839   }
 840 }
 841
 842
 843 RETSIGTYPE
 844 cleanup(int signum)
 845 {
 846   struct stat stbuf;
 847   int umount_result;
 848
 849   clock_valid = 0;              /* invalidate logging clock */
 850
 851 #ifdef DEBUG
 852   amuDebug(D_DAEMON)
 853 #endif /* DEBUG */
 854     if (getpid() != masterpid)
 855     return;
 856
 857 #ifdef DEBUG
 858   amuDebug(D_DAEMON)
 859 #endif /* DEBUG */
 860     if (fork() != 0) {
 861     masterpid = 0;
 862     am_set_mypid();
 863     return;
 864   }
 865   am_set_mypid();
 866
 867   for (;;) {
 868     while ((umount_result = UMOUNT_FS(dir_name, mnttab_file_name)) == EBUSY) {
 869 #ifdef DEBUG
 870       dlog("cleanup(): umount delaying for 10 seconds");
 871 #endif /* DEBUG */
 872       sleep(10);
 873     }
 874     if (stat(dir_name, &stbuf) == 0 && stbuf.st_ino == ROOTID) {
 875       plog(XLOG_ERROR, "unable to unmount %s", dir_name);
 876       plog(XLOG_ERROR, "suspending, unmount before terminating");
 877       kill(am_mypid, SIGSTOP);
 878       continue;                 /* retry unmount */
 879     }
 880     break;
 881   }
 882
 883 #ifdef DEBUG
 884   dlog("cleanup(): killing processes and terminating");
 885   amuDebug(D_DAEMON)
 886 #endif /* DEBUG */
 887     kill(masterpid, SIGKILL);
 888
 889 #ifdef DEBUG
 890   amuDebug(D_DAEMON)
 891 #endif /* DEBUG */
 892     kill(serverpid, SIGKILL);
 893
 894   plog(XLOG_INFO, "hlfsd terminating with status 0\n");
 895   exit(0);
 896 }
 897
 898
 899 static RETSIGTYPE
 900 reaper(int signum)
 901 {
 902   int result;
 903
 904   if (wait(&result) == masterpid) {
 905     exit(4);
 906   }
 907 }
 908
 909
 910 void
 911 hlfsd_going_down(int rc)
 912 {
 913   int mypid = getpid();         /* XXX: should this be the global am_mypid */
 914
 915   if (mypid == masterpid)
 916     cleanup(0);
 917   else if (mypid == serverpid)
 918     kill(masterpid, SIGTERM);
 919
 920   exit(rc);
 921 }
 922
 923
 924 void
 925 fatal(char *mess)
 926 {
 927   if (logfile && !STREQ(logfile, "stderr")) {
 928     char lessmess[128];
 929     int messlen;
 930
 931     messlen = strlen(mess);
 932
 933     if (!STREQ(&mess[messlen + 1 - sizeof(ERRM)], ERRM))
 934       fprintf(stderr, "%s: %s\n", am_get_progname(), mess);
 935     else {
 936       strcpy(lessmess, mess);
 937       lessmess[messlen - 4] = '\0';
 938
 939       if (errno < sys_nerr)
 940         fprintf(stderr, "%s: %s: %s\n", am_get_progname(), lessmess,
 941 #ifdef HAVE_STRERROR
 942                 strerror(errno)
 943 #else /* not HAVE_STRERROR */
 944                 sys_errlist[errno]
 945 #endif /* not HAVE_STRERROR */
 946                 );
 947       else
 948         fprintf(stderr, "%s: %s: Error %d\n",
 949                 am_get_progname(), lessmess, errno);
 950     }
 951   }
 952   plog(XLOG_FATAL, mess);
 953
 954   hlfsd_going_down(1);
 955 }