contrib/amd/amd/amd.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: amd.c,v 1.6 1999/09/30 21:01:29 ezk Exp $
  42  * $FreeBSD: src/contrib/amd/amd/amd.c,v 1.6.2.1 2000/09/20 02:17:04 jkh Exp $
  43  * $DragonFly: src/contrib/amd/amd/amd.c,v 1.2 2003/06/17 04:23:56 dillon Exp $
  44  *
  45  */
  46
  47 /*
  48  * Automounter
  49  */
  50
  51 #ifdef HAVE_CONFIG_H
  52 # include <config.h>
  53 #endif /* HAVE_CONFIG_H */
  54 #include <am_defs.h>
  55 #include <amd.h>
  56
  57 struct amu_global_options gopt; /* where global options are stored */
  58
  59 char pid_fsname[16 + MAXHOSTNAMELEN];   /* "kiska.southseas.nz:(pid%d)" */
  60 char *hostdomain = "unknown.domain";
  61 char hostd[2 * MAXHOSTNAMELEN + 1]; /* Host+domain */
  62 char *endian = ARCH_ENDIAN;     /* Big or Little endian */
  63 char *cpu = HOST_CPU;           /* CPU type */
  64 char *PrimNetName;              /* name of primary network */
  65 char *PrimNetNum;               /* number of primary network */
  66
  67 int immediate_abort;            /* Should close-down unmounts be retried */
  68 int orig_umask = 022;
  69 int select_intr_valid;
  70
  71 jmp_buf select_intr;
  72 struct amd_stats amd_stats;     /* Server statistics */
  73 struct in_addr myipaddr;        /* (An) IP address of this host */
  74 time_t do_mapc_reload = 0;      /* mapc_reload() call required? */
  75
  76 #ifdef HAVE_SIGACTION
  77 sigset_t masked_sigs;
  78 #endif /* HAVE_SIGACTION */
  79
  80
  81 /*
  82  * Signal handler:
  83  * SIGINT - tells amd to do a full shutdown, including unmounting all
  84  *       filesystem.
  85  * SIGTERM - tells amd to shutdown now.  Just unmounts the automount nodes.
  86  */
  87 static RETSIGTYPE
  88 sigterm(int sig)
  89 {
  90 #ifdef REINSTALL_SIGNAL_HANDLER
  91   signal(sig, sigterm);
  92 #endif /* REINSTALL_SIGNAL_HANDLER */
  93
  94   switch (sig) {
  95   case SIGINT:
  96     immediate_abort = 15;
  97     break;
  98
  99   case SIGTERM:
 100     immediate_abort = -1;
 101     /* fall through... */
 102
 103   default:
 104     plog(XLOG_WARNING, "WARNING: automounter going down on signal %d", sig);
 105     break;
 106   }
 107   if (select_intr_valid)
 108     longjmp(select_intr, sig);
 109 }
 110
 111
 112 /*
 113  * Hook for cache reload.
 114  * When a SIGHUP arrives it schedules a call to mapc_reload
 115  */
 116 static RETSIGTYPE
 117 sighup(int sig)
 118 {
 119 #ifdef REINSTALL_SIGNAL_HANDLER
 120   signal(sig, sighup);
 121 #endif /* REINSTALL_SIGNAL_HANDLER */
 122
 123 #ifdef DEBUG
 124   if (sig != SIGHUP)
 125     dlog("spurious call to sighup");
 126 #endif /* DEBUG */
 127   /*
 128    * Force a reload by zero'ing the timer
 129    */
 130   if (amd_state == Run)
 131     do_mapc_reload = 0;
 132 }
 133
 134
 135 static RETSIGTYPE
 136 parent_exit(int sig)
 137 {
 138   exit(0);
 139 }
 140
 141
 142 static int
 143 daemon_mode(void)
 144 {
 145   int bgpid;
 146
 147 #ifdef HAVE_SIGACTION
 148   struct sigaction sa, osa;
 149
 150   sa.sa_handler = parent_exit;
 151   sa.sa_flags = 0;
 152   sigemptyset(&(sa.sa_mask));
 153   sigaddset(&(sa.sa_mask), SIGQUIT);
 154   sigaction(SIGQUIT, &sa, &osa);
 155 #else /* not HAVE_SIGACTION */
 156   signal(SIGQUIT, parent_exit);
 157 #endif /* not HAVE_SIGACTION */
 158
 159   bgpid = background();
 160
 161   if (bgpid != 0) {
 162     /*
 163      * Now wait for the automount points to
 164      * complete.
 165      */
 166     for (;;)
 167       pause();
 168     /* should never reach here */
 169   }
 170 #ifdef HAVE_SIGACTION
 171   sigaction(SIGQUIT, &osa, NULL);
 172 #else /* not HAVE_SIGACTION */
 173   signal(SIGQUIT, SIG_DFL);
 174 #endif /* not HAVE_SIGACTION */
 175
 176   /*
 177    * Record our pid to make it easier to kill the correct amd.
 178    */
 179   if (gopt.flags & CFM_PRINT_PID) {
 180     if (STREQ(gopt.pid_file, "/dev/stdout")) {
 181       printf("%ld\n", (long) am_mypid);
 182       fflush(stdout);
 183       /* do not fclose stdout */
 184     } else {
 185       FILE *f;
 186       mode_t prev_umask = umask(0022); /* set secure temporary umask */
 187
 188       f = fopen(gopt.pid_file, "w");
 189       if (f) {
 190         fprintf(f, "%ld\n", (long) am_mypid);
 191         (void) fclose(f);
 192       } else {
 193         fprintf(stderr, "cannot open %s (errno=%d)\n", gopt.pid_file, errno);
 194       }
 195       umask(prev_umask);        /* restore umask */
 196     }
 197   }
 198
 199   /*
 200    * Pretend we are in the foreground again
 201    */
 202   foreground = 1;
 203
 204   /*
 205    * Dissociate from the controlling terminal
 206    */
 207   amu_release_controlling_tty();
 208
 209   return getppid();
 210 }
 211
 212
 213 /*
 214  * Initialize global options structure.
 215  */
 216 static void
 217 init_global_options(void)
 218 {
 219 #if defined(HAVE_SYS_UTSNAME_H) && defined(HAVE_UNAME)
 220   static struct utsname un;
 221 #endif /* defined(HAVE_SYS_UTSNAME_H) && defined(HAVE_UNAME) */
 222
 223   memset(&gopt, 0, sizeof(struct amu_global_options));
 224
 225   /* name of current architecture */
 226   gopt.arch = HOST_ARCH;
 227
 228   /* automounter temp dir */
 229   gopt.auto_dir = "/.amd_mnt";
 230
 231   /* cluster name */
 232   gopt.cluster = NULL;
 233
 234   /*
 235    * kernel architecture: this you must get from uname() if possible.
 236    */
 237 #if defined(HAVE_SYS_UTSNAME_H) && defined(HAVE_UNAME)
 238   if (uname(&un) >= 0)
 239     gopt.karch = un.machine;
 240   else
 241 #endif /* defined(HAVE_SYS_UTSNAME_H) && defined(HAVE_UNAME) */
 242     gopt.karch = HOST_ARCH;
 243
 244   /* amd log file */
 245   gopt.logfile = NULL;
 246
 247   /* operating system name */
 248   gopt.op_sys = HOST_OS_NAME;
 249
 250   /* OS version */
 251   gopt.op_sys_ver = HOST_OS_VERSION;
 252
 253   /* full OS name and version */
 254   gopt.op_sys_full = HOST_OS;
 255
 256   /* OS version */
 257   gopt.op_sys_vendor = HOST_VENDOR;
 258
 259   /* pid file */
 260   gopt.pid_file = "/dev/stdout";
 261
 262   /* local domain */
 263   gopt.sub_domain = NULL;
 264
 265   /* NFS retransmit counter */
 266   gopt.amfs_auto_retrans = -1;
 267
 268   /* NFS retry interval */
 269   gopt.amfs_auto_timeo = -1;
 270
 271   /* cache duration */
 272   gopt.am_timeo = AM_TTL;
 273
 274   /* dismount interval */
 275   gopt.am_timeo_w = AM_TTL_W;
 276
 277   /*
 278    * various CFM_* flags.
 279    * by default, only the "plock" option is on (if available).
 280    */
 281   gopt.flags = CFM_PROCESS_LOCK;
 282
 283 #ifdef HAVE_MAP_HESIOD
 284   /* Hesiod rhs zone */
 285   gopt.hesiod_base = "automount";
 286 #endif /* HAVE_MAP_HESIOD */
 287
 288 #ifdef HAVE_MAP_LDAP
 289   /* LDAP base */
 290   gopt.ldap_base = NULL;
 291
 292   /* LDAP host ports */
 293   gopt.ldap_hostports = NULL;
 294
 295   /* LDAP cache */
 296   gopt.ldap_cache_seconds = 0;
 297   gopt.ldap_cache_maxmem = 131072;
 298 #endif /* HAVE_MAP_LDAP */
 299
 300 #ifdef HAVE_MAP_NIS
 301   /* YP domain name */
 302   gopt.nis_domain = NULL;
 303 #endif /* HAVE_MAP_NIS */
 304 }
 305
 306
 307 int
 308 main(int argc, char *argv[])
 309 {
 310   char *domdot, *verstr;
 311   int ppid = 0;
 312   int error;
 313   char *progname = NULL;                /* "amd" */
 314   char hostname[MAXHOSTNAMELEN + 1] = "localhost"; /* Hostname */
 315 #ifdef HAVE_SIGACTION
 316   struct sigaction sa;
 317 #endif /* HAVE_SIGACTION */
 318
 319   /*
 320    * Make sure some built-in assumptions are true before we start
 321    */
 322   assert(sizeof(nfscookie) >= sizeof(u_int));
 323   assert(sizeof(int) >= 4);
 324
 325   /*
 326    * Set processing status.
 327    */
 328   amd_state = Start;
 329
 330   /*
 331    * Determine program name
 332    */
 333   if (argv[0]) {
 334     progname = strrchr(argv[0], '/');
 335     if (progname && progname[1])
 336       progname++;
 337     else
 338       progname = argv[0];
 339   }
 340   if (!progname)
 341     progname = "amd";
 342   am_set_progname(progname);
 343
 344   /*
 345    * Initialize process id.  This is kept
 346    * cached since it is used for generating
 347    * and using file handles.
 348    */
 349   am_set_mypid();
 350
 351   /*
 352    * Get local machine name
 353    */
 354   if (gethostname(hostname, sizeof(hostname)) < 0) {
 355     plog(XLOG_FATAL, "gethostname: %m");
 356     going_down(1);
 357   }
 358   hostname[sizeof(hostname) - 1] = '\0';
 359
 360   /*
 361    * Check it makes sense
 362    */
 363   if (!*hostname) {
 364     plog(XLOG_FATAL, "host name is not set");
 365     going_down(1);
 366   }
 367
 368 #ifdef DEBUG
 369   /* initialize debugging flags (Register AMQ, Enter daemon mode) */
 370   debug_flags = D_AMQ | D_DAEMON;
 371 #endif /* DEBUG */
 372
 373   /*
 374    * Initialize global options structure.
 375    */
 376   init_global_options();
 377
 378   /*
 379    * Partially initialize hostd[].  This
 380    * is completed in get_args().
 381    */
 382   if ((domdot = strchr(hostname, '.'))) {
 383     /*
 384      * Hostname already contains domainname.
 385      * Split out hostname and domainname
 386      * components
 387      */
 388     *domdot++ = '\0';
 389     hostdomain = domdot;
 390   }
 391   strcpy(hostd, hostname);
 392   am_set_hostname(hostname);
 393
 394   /*
 395    * Trap interrupts for shutdowns.
 396    */
 397 #ifdef HAVE_SIGACTION
 398   sa.sa_handler = sigterm;
 399   sa.sa_flags = 0;
 400   sigemptyset(&(sa.sa_mask));
 401   sigaddset(&(sa.sa_mask), SIGINT);
 402   sigaddset(&(sa.sa_mask), SIGTERM);
 403   sigaction(SIGINT, &sa, NULL);
 404   sigaction(SIGTERM, &sa, NULL);
 405 #else /* not HAVE_SIGACTION */
 406   (void) signal(SIGINT, sigterm);
 407 #endif /* not HAVE_SIGACTION */
 408
 409   /*
 410    * Trap Terminate so that we can shutdown gracefully (some chance)
 411    */
 412 #ifdef HAVE_SIGACTION
 413   sa.sa_handler = sigterm;
 414   sa.sa_flags = 0;
 415   sigemptyset(&(sa.sa_mask));
 416   sigaddset(&(sa.sa_mask), SIGTERM);
 417   sigaction(SIGTERM, &sa, NULL);
 418 #else /* not HAVE_SIGACTION */
 419   (void) signal(SIGTERM, sigterm);
 420 #endif /* not HAVE_SIGACTION */
 421
 422   /*
 423    * Hangups tell us to reload the cache
 424    */
 425 #ifdef HAVE_SIGACTION
 426   sa.sa_handler = sighup;
 427   sa.sa_flags = 0;
 428   sigemptyset(&(sa.sa_mask));
 429   sigaddset(&(sa.sa_mask), SIGHUP);
 430   sigaction(SIGHUP, &sa, NULL);
 431 #else /* not HAVE_SIGACTION */
 432   (void) signal(SIGHUP, sighup);
 433 #endif /* not HAVE_SIGACTION */
 434
 435   /*
 436    * Trap Death-of-a-child.  These allow us to
 437    * pick up the exit status of backgrounded mounts.
 438    * See "sched.c".
 439    */
 440 #ifdef HAVE_SIGACTION
 441   sa.sa_handler = sigchld;
 442   sa.sa_flags = 0;
 443   sigemptyset(&(sa.sa_mask));
 444   sigaddset(&(sa.sa_mask), SIGCHLD);
 445   sigaction(SIGCHLD, &sa, NULL);
 446
 447   /*
 448    * construct global "masked_sigs" used in nfs_start.c
 449    */
 450   sigemptyset(&masked_sigs);
 451   sigaddset(&masked_sigs, SIGHUP);
 452   sigaddset(&masked_sigs, SIGCHLD);
 453   sigaddset(&masked_sigs, SIGTERM);
 454   sigaddset(&masked_sigs, SIGINT);
 455 #else /* not HAVE_SIGACTION */
 456   (void) signal(SIGCHLD, sigchld);
 457 #endif /* not HAVE_SIGACTION */
 458
 459   /*
 460    * Fix-up any umask problems.  Most systems default
 461    * to 002 which is not too convenient for our purposes
 462    */
 463   orig_umask = umask(0);
 464
 465   /*
 466    * Figure out primary network name
 467    */
 468   getwire(&PrimNetName, &PrimNetNum);
 469
 470   /*
 471    * Determine command-line arguments
 472    */
 473   get_args(argc, argv);
 474
 475   /*
 476    * Log version information.
 477    */
 478   verstr = strtok(get_version_string(), "\n");
 479   plog(XLOG_INFO, "AM-UTILS VERSION INFORMATION:");
 480   while (verstr) {
 481     plog(XLOG_INFO, verstr);
 482     verstr = strtok(NULL, "\n");
 483   }
 484
 485   /*
 486    * Get our own IP address so that we
 487    * can mount the automounter.
 488    */
 489   amu_get_myaddress(&myipaddr);
 490   plog(XLOG_INFO, "My ip addr is %s", inet_ntoa(myipaddr));
 491
 492   /* avoid hanging on other NFS servers if started elsewhere */
 493   if (chdir("/") < 0)
 494     plog(XLOG_INFO, "cannot chdir to /: %m");
 495
 496   /*
 497    * Now check we are root.
 498    */
 499   if (geteuid() != 0) {
 500     plog(XLOG_FATAL, "Must be root to mount filesystems (euid = %ld)", (long) geteuid());
 501     going_down(1);
 502   }
 503
 504   /*
 505    * Lock process text and data segment in memory.
 506    */
 507 #ifdef HAVE_PLOCK
 508   if (gopt.flags & CFM_PROCESS_LOCK) {
 509 # ifdef _AIX
 510     /*
 511      * On AIX you must lower the stack size using ulimit() before calling
 512      * plock.  Otherwise plock will reserve a lot of memory space based on
 513      * your maximum stack size limit.  Since it is not easily possible to
 514      * tell what should the limit be, I print a warning before calling
 515      * plock().  See the manual pages for ulimit(1,3,4) on your AIX system.
 516      */
 517     plog(XLOG_WARNING, "AIX: may need to lower stack size using ulimit(3) before calling plock");
 518 # endif /* _AIX */
 519     if (plock(PROCLOCK) != 0) {
 520       plog(XLOG_WARNING, "Couldn't lock process text and data segment in memory: %m");
 521     } else {
 522       plog(XLOG_INFO, "Locked process text and data segment in memory");
 523     }
 524   }
 525 #endif /* HAVE_PLOCK */
 526
 527 #ifdef HAVE_MAP_NIS
 528   /*
 529    * If the domain was specified then bind it here
 530    * to circumvent any default bindings that may
 531    * be done in the C library.
 532    */
 533   if (gopt.nis_domain && yp_bind(gopt.nis_domain)) {
 534     plog(XLOG_FATAL, "Can't bind to NIS domain \"%s\"", gopt.nis_domain);
 535     going_down(1);
 536   }
 537 #endif /* HAVE_MAP_NIS */
 538
 539 #ifdef DEBUG
 540   amuDebug(D_DAEMON)
 541 #endif /* DEBUG */
 542     ppid = daemon_mode();
 543
 544   sprintf(pid_fsname, "%s:(pid%ld)", am_get_hostname(), (long) am_mypid);
 545
 546   do_mapc_reload = clocktime() + ONE_HOUR;
 547
 548   /*
 549    * Register automounter with system.
 550    */
 551   error = mount_automounter(ppid);
 552   if (error && ppid)
 553     kill(ppid, SIGALRM);
 554   going_down(error);
 555
 556   abort();
 557   return 1; /* should never get here */
 558 }