usr.bin/nohup/nohup.c

   1 /*
   2  * Copyright (c) 1989, 1993
   3  *      The Regents of the University of California.  All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in the
  12  *    documentation and/or other materials provided with the distribution.
  13  * 3. All advertising materials mentioning features or use of this software
  14  *    must display the following acknowledgement:
  15  *      This product includes software developed by the University of
  16  *      California, Berkeley and its contributors.
  17  * 4. Neither the name of the University nor the names of its contributors
  18  *    may be used to endorse or promote products derived from this software
  19  *    without specific prior written permission.
  20  *
  21  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  22  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  24  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  25  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  26  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  27  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  28  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  29  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  30  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  31  * SUCH DAMAGE.
  32  */
  33
  34 #if 0
  35 #ifndef lint
  36 static const char copyright[] =
  37 "@(#) Copyright (c) 1989, 1993\n\
  38         The Regents of the University of California.  All rights reserved.\n";
  39 #endif /* not lint */
  40
  41 #ifndef lint
  42 static char sccsid[] = "@(#)nohup.c     8.1 (Berkeley) 6/6/93";
  43 #endif /* not lint */
  44 #endif
  45 #include <sys/cdefs.h>
  46 __FBSDID("$FreeBSD$");
  47
  48 #include <sys/param.h>
  49 #include <sys/stat.h>
  50
  51 #include <err.h>
  52 #include <errno.h>
  53 #include <fcntl.h>
  54 #include <signal.h>
  55 #include <stdio.h>
  56 #include <stdlib.h>
  57 #include <string.h>
  58 #include <unistd.h>
  59
  60 static void dofile(void);
  61 static void usage(void);
  62
  63 #define FILENAME        "nohup.out"
  64 /*
  65  * POSIX mandates that we exit with:
  66  * 126 - If the utility was found, but failed to execute.
  67  * 127 - If any other error occurred.
  68  */
  69 #define EXIT_NOEXEC     126
  70 #define EXIT_NOTFOUND   127
  71 #define EXIT_MISC       127
  72
  73 int
  74 main(int argc, char *argv[])
  75 {
  76         int exit_status;
  77
  78         while (getopt(argc, argv, "") != -1)
  79                 usage();
  80         argc -= optind;
  81         argv += optind;
  82         if (argc < 1)
  83                 usage();
  84
  85         if (isatty(STDOUT_FILENO))
  86                 dofile();
  87         if (isatty(STDERR_FILENO) && dup2(STDOUT_FILENO, STDERR_FILENO) == -1)
  88                 /* may have just closed stderr */
  89                 err(EXIT_MISC, "%s", argv[0]);
  90
  91         (void)signal(SIGHUP, SIG_IGN);
  92
  93         execvp(*argv, argv);
  94         exit_status = (errno == ENOENT) ? EXIT_NOTFOUND : EXIT_NOEXEC;
  95         err(exit_status, "%s", argv[0]);
  96 }
  97
  98 static void
  99 dofile(void)
 100 {
 101         int fd;
 102         char path[MAXPATHLEN];
 103         const char *p;
 104
 105         /*
 106          * POSIX mandates if the standard output is a terminal, the standard
 107          * output is appended to nohup.out in the working directory.  Failing
 108          * that, it will be appended to nohup.out in the directory obtained
 109          * from the HOME environment variable.  If file creation is required,
 110          * the mode_t is set to S_IRUSR | S_IWUSR.
 111          */
 112         p = FILENAME;
 113         fd = open(p, O_RDWR | O_CREAT | O_APPEND, S_IRUSR | S_IWUSR);
 114         if (fd != -1)
 115                 goto dupit;
 116         if ((p = getenv("HOME")) != NULL && *p != '\0' &&
 117             (size_t)snprintf(path, sizeof(path), "%s/%s", p, FILENAME) <
 118             sizeof(path)) {
 119                 fd = open(p = path, O_RDWR | O_CREAT | O_APPEND,
 120                     S_IRUSR | S_IWUSR);
 121                 if (fd != -1)
 122                         goto dupit;
 123         }
 124         errx(EXIT_MISC, "can't open a nohup.out file");
 125
 126 dupit:
 127         if (dup2(fd, STDOUT_FILENO) == -1)
 128                 err(EXIT_MISC, NULL);
 129         (void)fprintf(stderr, "appending output to %s\n", p);
 130 }
 131
 132 static void
 133 usage(void)
 134 {
 135         (void)fprintf(stderr, "usage: nohup [--] utility [arguments]\n");
 136         exit(EXIT_MISC);
 137 }