git.gag.com Git - debian/amanda/blob - client-src/killpgrp.c

   1 /*
   2  * Amanda, The Advanced Maryland Automatic Network Disk Archiver
   3  * Copyright (c) 1991-1998 University of Maryland at College Park
   4  * All Rights Reserved.
   5  *
   6  * Permission to use, copy, modify, distribute, and sell this software and its
   7  * documentation for any purpose is hereby granted without fee, provided that
   8  * the above copyright notice appear in all copies and that both that
   9  * copyright notice and this permission notice appear in supporting
  10  * documentation, and that the name of U.M. not be used in advertising or
  11  * publicity pertaining to distribution of the software without specific,
  12  * written prior permission.  U.M. makes no representations about the
  13  * suitability of this software for any purpose.  It is provided "as is"
  14  * without express or implied warranty.
  15  *
  16  * U.M. DISCLAIMS ALL WARRANTIES WITH REGARD TO THIS SOFTWARE, INCLUDING ALL
  17  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS, IN NO EVENT SHALL U.M.
  18  * BE LIABLE FOR ANY SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  19  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION
  20  * OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN
  21  * CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  22  *
  23  * Authors: the Amanda Development Team.  Its members are listed in a
  24  * file named AUTHORS, in the root directory of this distribution.
  25  */
  26 /*
  27  * $Id: killpgrp.c,v 1.17 2006/07/25 18:27:56 martinea Exp $
  28  *
  29  * if it is the process group leader, it kills all processes in its
  30  * process group when it is killed itself.
  31  *
  32  * argv[0] is the killpgrp program name
  33  * argv[1] is the config name or NOCONFIG
  34  *
  35  */
  36 #include "amanda.h"
  37 #include "version.h"
  38 #include "util.h"
  39
  40 #ifdef HAVE_GETPGRP
  41 #ifdef GETPGRP_VOID
  42 #define AM_GETPGRP() getpgrp()
  43 #else
  44 #define AM_GETPGRP() getpgrp(getpid())
  45 #endif
  46 #else
  47 /* we cannot check it, so let us assume it is ok */
  48 #define AM_GETPGRP() getpid()
  49 #endif
  50
  51 int main(int argc, char **argv);
  52 static void term_kill_soft(int sig);
  53 static void term_kill_hard(int sig);
  54
  55 int
  56 main(
  57     int         argc,
  58     char **     argv)
  59 {
  60     int ch;
  61     char *exitstr;
  62     amwait_t status;
  63
  64     /*
  65      * Configure program for internationalization:
  66      *   1) Only set the message locale for now.
  67      *   2) Set textdomain for all amanda related programs to "amanda"
  68      *      We don't want to be forced to support dozens of message catalogs.
  69      */
  70     setlocale(LC_MESSAGES, "C");
  71     textdomain("amanda");
  72
  73     safe_fd(-1, 0);
  74     safe_cd();
  75
  76     set_pname("killpgrp");
  77
  78     dbopen(DBG_SUBDIR_CLIENT);
  79     if (argc < 2) {
  80         error("Need at least 2 arguments\n");
  81         /*NOTREACHED*/
  82     }
  83     dbprintf(_("version %s\n"), version());
  84     dbprintf(_("config: %s\n"), argv[1]);
  85     if (strcmp(argv[1], "NOCONFIG") != 0)
  86         dbrename(argv[1], DBG_SUBDIR_CLIENT);
  87
  88 #ifdef WANT_SETUID_CLIENT
  89     check_running_as(RUNNING_AS_CLIENT_LOGIN | RUNNING_AS_UID_ONLY);
  90     if (!become_root()) {
  91         error(_("error [%s could not become root (is the setuid bit set?)]\n"), get_pname());
  92         /*NOTREACHED*/
  93     }
  94 #else
  95     check_running_as(RUNNING_AS_CLIENT_LOGIN);
  96 #endif
  97
  98     if (AM_GETPGRP() != getpid()) {
  99         error(_("error [must be the process group leader]"));
 100         /*NOTREACHED*/
 101     }
 102
 103     signal(SIGTERM, term_kill_soft);
 104
 105     /* Consume any extranious input */
 106     do {
 107         ch = getchar();
 108         /* wait until EOF */
 109     } while (ch != EOF);
 110
 111     term_kill_soft(0);
 112
 113     for(;;) {
 114         if (wait(&status) != -1)
 115             break;
 116         if (errno != EINTR) {
 117             error(_("error [wait() failed: %s]"), strerror(errno));
 118             /*NOTREACHED*/
 119         }
 120     }
 121     exitstr = str_exit_status("child", status);
 122     dbprintf("%s\n", exitstr);
 123     amfree(exitstr);
 124
 125     /*@ignore@*/
 126     return WIFEXITED(status)?WEXITSTATUS(status):1;
 127     /*@end@*/
 128 }
 129
 130 static void term_kill_soft(
 131     int sig)
 132 {
 133     pid_t dumppid = getpid();
 134     int killerr;
 135
 136     (void)sig;  /* Quiet unused parameter warning */
 137
 138     signal(SIGTERM, SIG_IGN);
 139     signal(SIGALRM, term_kill_hard);
 140     alarm(3);
 141     /*
 142      * First, try to kill the dump process nicely.  If it ignores us
 143      * for three seconds, hit it harder.
 144      */
 145     dbprintf(_("sending SIGTERM to process group %ld\n"), (long) dumppid);
 146     killerr = kill(-dumppid, SIGTERM);
 147     if (killerr == -1) {
 148         dbprintf(_("kill failed: %s\n"), strerror(errno));
 149     }
 150 }
 151
 152 static void term_kill_hard(
 153     int sig)
 154 {
 155     pid_t dumppid = getpid();
 156     int killerr;
 157
 158     (void)sig;  /* Quiet unused parameter warning */
 159
 160     dbprintf(_("It won\'t die with SIGTERM, but SIGKILL should do.\n"));
 161     dbprintf(_("Don't expect any further output, this will be suicide.\n"));
 162     killerr = kill(-dumppid, SIGKILL);
 163     /* should never reach this point, but so what? */
 164     if (killerr == -1) {
 165         dbprintf(_("kill failed: %s\n"), strerror(errno));
 166         dbprintf(_("waiting until child terminates\n"));
 167     }
 168 }