bin/pkill/pkill.c

   1 /*      $NetBSD: pkill.c,v 1.16 2005/10/10 22:13:20 kleink Exp $        */
   2
   3 /*-
   4  * Copyright (c) 2002 The NetBSD Foundation, Inc.
   5  * Copyright (c) 2005 Pawel Jakub Dawidek <pjd@FreeBSD.org>
   6  * All rights reserved.
   7  *
   8  * This code is derived from software contributed to The NetBSD Foundation
   9  * by Andrew Doran.
  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  *
  20  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
  21  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
  22  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  23  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
  24  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  25  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  26  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  27  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  28  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  29  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  30  * POSSIBILITY OF SUCH DAMAGE.
  31  */
  32
  33 #include <sys/cdefs.h>
  34 __FBSDID("$FreeBSD$");
  35
  36 #include <sys/types.h>
  37 #include <sys/param.h>
  38 #include <sys/sysctl.h>
  39 #include <sys/proc.h>
  40 #include <sys/queue.h>
  41 #include <sys/stat.h>
  42 #include <sys/time.h>
  43 #include <sys/user.h>
  44
  45 #include <assert.h>
  46 #include <stdio.h>
  47 #include <stdlib.h>
  48 #include <limits.h>
  49 #include <paths.h>
  50 #include <string.h>
  51 #include <unistd.h>
  52 #include <signal.h>
  53 #include <regex.h>
  54 #include <ctype.h>
  55 #include <fcntl.h>
  56 #include <kvm.h>
  57 #include <err.h>
  58 #include <pwd.h>
  59 #include <grp.h>
  60 #include <errno.h>
  61 #include <locale.h>
  62
  63 #define STATUS_MATCH    0
  64 #define STATUS_NOMATCH  1
  65 #define STATUS_BADUSAGE 2
  66 #define STATUS_ERROR    3
  67
  68 #define MIN_PID 5
  69 #define MAX_PID 99999
  70
  71 /* Ignore system-processes (if '-S' flag is not specified) and myself. */
  72 #define PSKIP(kp)       ((kp)->ki_pid == mypid ||                       \
  73                          (!kthreads && ((kp)->ki_flag & P_KTHREAD) != 0))
  74
  75 enum listtype {
  76         LT_GENERIC,
  77         LT_USER,
  78         LT_GROUP,
  79         LT_TTY,
  80         LT_PGRP,
  81         LT_JID,
  82         LT_SID,
  83         LT_CLASS
  84 };
  85
  86 struct list {
  87         SLIST_ENTRY(list) li_chain;
  88         long    li_number;
  89         char    *li_name;
  90 };
  91
  92 SLIST_HEAD(listhead, list);
  93
  94 static struct kinfo_proc *plist;
  95 static char     *selected;
  96 static const char *delim = "\n";
  97 static int      nproc;
  98 static int      pgrep;
  99 static int      signum = SIGTERM;
 100 static int      newest;
 101 static int      oldest;
 102 static int      interactive;
 103 static int      inverse;
 104 static int      longfmt;
 105 static int      matchargs;
 106 static int      fullmatch;
 107 static int      kthreads;
 108 static int      cflags = REG_EXTENDED;
 109 static int      quiet;
 110 static kvm_t    *kd;
 111 static pid_t    mypid;
 112
 113 static struct listhead euidlist = SLIST_HEAD_INITIALIZER(euidlist);
 114 static struct listhead ruidlist = SLIST_HEAD_INITIALIZER(ruidlist);
 115 static struct listhead rgidlist = SLIST_HEAD_INITIALIZER(rgidlist);
 116 static struct listhead pgrplist = SLIST_HEAD_INITIALIZER(pgrplist);
 117 static struct listhead ppidlist = SLIST_HEAD_INITIALIZER(ppidlist);
 118 static struct listhead tdevlist = SLIST_HEAD_INITIALIZER(tdevlist);
 119 static struct listhead sidlist = SLIST_HEAD_INITIALIZER(sidlist);
 120 static struct listhead jidlist = SLIST_HEAD_INITIALIZER(jidlist);
 121 static struct listhead classlist = SLIST_HEAD_INITIALIZER(classlist);
 122
 123 static void     usage(void) __attribute__((__noreturn__));
 124 static int      killact(const struct kinfo_proc *);
 125 static int      grepact(const struct kinfo_proc *);
 126 static void     makelist(struct listhead *, enum listtype, char *);
 127 static int      takepid(const char *, int);
 128
 129 int
 130 main(int argc, char **argv)
 131 {
 132         char buf[_POSIX2_LINE_MAX], *mstr, **pargv, *p, *q, *pidfile;
 133         const char *execf, *coref;
 134         int ancestors, debug_opt, did_action;
 135         int i, ch, bestidx, rv, criteria, pidfromfile, pidfilelock;
 136         size_t jsz;
 137         int (*action)(const struct kinfo_proc *);
 138         struct kinfo_proc *kp;
 139         struct list *li;
 140         struct timeval best_tval;
 141         regex_t reg;
 142         regmatch_t regmatch;
 143         pid_t pid;
 144
 145         setlocale(LC_ALL, "");
 146
 147         if (strcmp(getprogname(), "pgrep") == 0) {
 148                 action = grepact;
 149                 pgrep = 1;
 150         } else {
 151                 action = killact;
 152                 p = argv[1];
 153
 154                 if (argc > 1 && p[0] == '-') {
 155                         p++;
 156                         i = (int)strtol(p, &q, 10);
 157                         if (*q == '\0') {
 158                                 signum = i;
 159                                 argv++;
 160                                 argc--;
 161                         } else {
 162                                 if (strncasecmp(p, "SIG", 3) == 0)
 163                                         p += 3;
 164                                 for (i = 1; i < NSIG; i++)
 165                                         if (strcasecmp(sys_signame[i], p) == 0)
 166                                                 break;
 167                                 if (i != NSIG) {
 168                                         signum = i;
 169                                         argv++;
 170                                         argc--;
 171                                 }
 172                         }
 173                 }
 174         }
 175
 176         ancestors = 0;
 177         criteria = 0;
 178         debug_opt = 0;
 179         pidfile = NULL;
 180         pidfilelock = 0;
 181         quiet = 0;
 182         execf = NULL;
 183         coref = _PATH_DEVNULL;
 184
 185         while ((ch = getopt(argc, argv, "DF:G:ILM:N:P:SU:ac:d:fg:ij:lnoqs:t:u:vx")) != -1)
 186                 switch (ch) {
 187                 case 'D':
 188                         debug_opt++;
 189                         break;
 190                 case 'F':
 191                         pidfile = optarg;
 192                         criteria = 1;
 193                         break;
 194                 case 'G':
 195                         makelist(&rgidlist, LT_GROUP, optarg);
 196                         criteria = 1;
 197                         break;
 198                 case 'I':
 199                         if (pgrep)
 200                                 usage();
 201                         interactive = 1;
 202                         break;
 203                 case 'L':
 204                         pidfilelock = 1;
 205                         break;
 206                 case 'M':
 207                         coref = optarg;
 208                         break;
 209                 case 'N':
 210                         execf = optarg;
 211                         break;
 212                 case 'P':
 213                         makelist(&ppidlist, LT_GENERIC, optarg);
 214                         criteria = 1;
 215                         break;
 216                 case 'S':
 217                         if (!pgrep)
 218                                 usage();
 219                         kthreads = 1;
 220                         break;
 221                 case 'U':
 222                         makelist(&ruidlist, LT_USER, optarg);
 223                         criteria = 1;
 224                         break;
 225                 case 'a':
 226                         ancestors++;
 227                         break;
 228                 case 'c':
 229                         makelist(&classlist, LT_CLASS, optarg);
 230                         criteria = 1;
 231                         break;
 232                 case 'd':
 233                         if (!pgrep)
 234                                 usage();
 235                         delim = optarg;
 236                         break;
 237                 case 'f':
 238                         matchargs = 1;
 239                         break;
 240                 case 'g':
 241                         makelist(&pgrplist, LT_PGRP, optarg);
 242                         criteria = 1;
 243                         break;
 244                 case 'i':
 245                         cflags |= REG_ICASE;
 246                         break;
 247                 case 'j':
 248                         makelist(&jidlist, LT_JID, optarg);
 249                         criteria = 1;
 250                         break;
 251                 case 'l':
 252                         longfmt = 1;
 253                         break;
 254                 case 'n':
 255                         newest = 1;
 256                         criteria = 1;
 257                         break;
 258                 case 'o':
 259                         oldest = 1;
 260                         criteria = 1;
 261                         break;
 262                 case 'q':
 263                         if (!pgrep)
 264                                 usage();
 265                         quiet = 1;
 266                         break;
 267                 case 's':
 268                         makelist(&sidlist, LT_SID, optarg);
 269                         criteria = 1;
 270                         break;
 271                 case 't':
 272                         makelist(&tdevlist, LT_TTY, optarg);
 273                         criteria = 1;
 274                         break;
 275                 case 'u':
 276                         makelist(&euidlist, LT_USER, optarg);
 277                         criteria = 1;
 278                         break;
 279                 case 'v':
 280                         inverse = 1;
 281                         break;
 282                 case 'x':
 283                         fullmatch = 1;
 284                         break;
 285                 default:
 286                         usage();
 287                         /* NOTREACHED */
 288                 }
 289
 290         argc -= optind;
 291         argv += optind;
 292         if (argc != 0)
 293                 criteria = 1;
 294         if (!criteria)
 295                 usage();
 296         if (newest && oldest)
 297                 errx(STATUS_ERROR, "Options -n and -o are mutually exclusive");
 298         if (pidfile != NULL)
 299                 pidfromfile = takepid(pidfile, pidfilelock);
 300         else {
 301                 if (pidfilelock) {
 302                         errx(STATUS_ERROR,
 303                             "Option -L doesn't make sense without -F");
 304                 }
 305                 pidfromfile = -1;
 306         }
 307
 308         mypid = getpid();
 309
 310         /*
 311          * Retrieve the list of running processes from the kernel.
 312          */
 313         kd = kvm_openfiles(execf, coref, NULL, O_RDONLY, buf);
 314         if (kd == NULL)
 315                 errx(STATUS_ERROR, "Cannot open kernel files (%s)", buf);
 316
 317         /*
 318          * Use KERN_PROC_PROC instead of KERN_PROC_ALL, since we
 319          * just want processes and not individual kernel threads.
 320          */
 321         if (pidfromfile >= 0)
 322                 plist = kvm_getprocs(kd, KERN_PROC_PID, pidfromfile, &nproc);
 323         else
 324                 plist = kvm_getprocs(kd, KERN_PROC_PROC, 0, &nproc);
 325         if (plist == NULL) {
 326                 errx(STATUS_ERROR, "Cannot get process list (%s)",
 327                     kvm_geterr(kd));
 328         }
 329
 330         /*
 331          * Allocate memory which will be used to keep track of the
 332          * selection.
 333          */
 334         if ((selected = malloc(nproc)) == NULL) {
 335                 err(STATUS_ERROR, "Cannot allocate memory for %d processes",
 336                     nproc);
 337         }
 338         memset(selected, 0, nproc);
 339
 340         /*
 341          * Refine the selection.
 342          */
 343         for (; *argv != NULL; argv++) {
 344                 if ((rv = regcomp(&reg, *argv, cflags)) != 0) {
 345                         regerror(rv, &reg, buf, sizeof(buf));
 346                         errx(STATUS_BADUSAGE,
 347                             "Cannot compile regular expression `%s' (%s)",
 348                             *argv, buf);
 349                 }
 350
 351                 for (i = 0, kp = plist; i < nproc; i++, kp++) {
 352                         if (PSKIP(kp)) {
 353                                 if (debug_opt > 0)
 354                                     fprintf(stderr, "* Skipped %5d %3d %s\n",
 355                                         kp->ki_pid, kp->ki_uid, kp->ki_comm);
 356                                 continue;
 357                         }
 358
 359                         if (matchargs &&
 360                             (pargv = kvm_getargv(kd, kp, 0)) != NULL) {
 361                                 jsz = 0;
 362                                 while (jsz < sizeof(buf) && *pargv != NULL) {
 363                                         jsz += snprintf(buf + jsz,
 364                                             sizeof(buf) - jsz,
 365                                             pargv[1] != NULL ? "%s " : "%s",
 366                                             pargv[0]);
 367                                         pargv++;
 368                                 }
 369                                 mstr = buf;
 370                         } else
 371                                 mstr = kp->ki_comm;
 372
 373                         rv = regexec(&reg, mstr, 1, &regmatch, 0);
 374                         if (rv == 0) {
 375                                 if (fullmatch) {
 376                                         if (regmatch.rm_so == 0 &&
 377                                             regmatch.rm_eo ==
 378                                             (off_t)strlen(mstr))
 379                                                 selected[i] = 1;
 380                                 } else
 381                                         selected[i] = 1;
 382                         } else if (rv != REG_NOMATCH) {
 383                                 regerror(rv, &reg, buf, sizeof(buf));
 384                                 errx(STATUS_ERROR,
 385                                     "Regular expression evaluation error (%s)",
 386                                     buf);
 387                         }
 388                         if (debug_opt > 1) {
 389                                 const char *rv_res = "NoMatch";
 390                                 if (selected[i])
 391                                         rv_res = "Matched";
 392                                 fprintf(stderr, "* %s %5d %3d %s\n", rv_res,
 393                                     kp->ki_pid, kp->ki_uid, mstr);
 394                         }
 395                 }
 396
 397                 regfree(&reg);
 398         }
 399
 400         for (i = 0, kp = plist; i < nproc; i++, kp++) {
 401                 if (PSKIP(kp))
 402                         continue;
 403
 404                 if (pidfromfile >= 0 && kp->ki_pid != pidfromfile) {
 405                         selected[i] = 0;
 406                         continue;
 407                 }
 408
 409                 SLIST_FOREACH(li, &ruidlist, li_chain)
 410                         if (kp->ki_ruid == (uid_t)li->li_number)
 411                                 break;
 412                 if (SLIST_FIRST(&ruidlist) != NULL && li == NULL) {
 413                         selected[i] = 0;
 414                         continue;
 415                 }
 416
 417                 SLIST_FOREACH(li, &rgidlist, li_chain)
 418                         if (kp->ki_rgid == (gid_t)li->li_number)
 419                                 break;
 420                 if (SLIST_FIRST(&rgidlist) != NULL && li == NULL) {
 421                         selected[i] = 0;
 422                         continue;
 423                 }
 424
 425                 SLIST_FOREACH(li, &euidlist, li_chain)
 426                         if (kp->ki_uid == (uid_t)li->li_number)
 427                                 break;
 428                 if (SLIST_FIRST(&euidlist) != NULL && li == NULL) {
 429                         selected[i] = 0;
 430                         continue;
 431                 }
 432
 433                 SLIST_FOREACH(li, &ppidlist, li_chain)
 434                         if (kp->ki_ppid == (pid_t)li->li_number)
 435                                 break;
 436                 if (SLIST_FIRST(&ppidlist) != NULL && li == NULL) {
 437                         selected[i] = 0;
 438                         continue;
 439                 }
 440
 441                 SLIST_FOREACH(li, &pgrplist, li_chain)
 442                         if (kp->ki_pgid == (pid_t)li->li_number)
 443                                 break;
 444                 if (SLIST_FIRST(&pgrplist) != NULL && li == NULL) {
 445                         selected[i] = 0;
 446                         continue;
 447                 }
 448
 449                 SLIST_FOREACH(li, &tdevlist, li_chain) {
 450                         if (li->li_number == -1 &&
 451                             (kp->ki_flag & P_CONTROLT) == 0)
 452                                 break;
 453                         if (kp->ki_tdev == (dev_t)li->li_number)
 454                                 break;
 455                 }
 456                 if (SLIST_FIRST(&tdevlist) != NULL && li == NULL) {
 457                         selected[i] = 0;
 458                         continue;
 459                 }
 460
 461                 SLIST_FOREACH(li, &sidlist, li_chain)
 462                         if (kp->ki_sid == (pid_t)li->li_number)
 463                                 break;
 464                 if (SLIST_FIRST(&sidlist) != NULL && li == NULL) {
 465                         selected[i] = 0;
 466                         continue;
 467                 }
 468
 469                 SLIST_FOREACH(li, &jidlist, li_chain) {
 470                         /* A particular jail ID, including 0 (not in jail) */
 471                         if (kp->ki_jid == (int)li->li_number)
 472                                 break;
 473                         /* Any jail */
 474                         if (kp->ki_jid > 0 && li->li_number == -1)
 475                                 break;
 476                 }
 477                 if (SLIST_FIRST(&jidlist) != NULL && li == NULL) {
 478                         selected[i] = 0;
 479                         continue;
 480                 }
 481
 482                 SLIST_FOREACH(li, &classlist, li_chain) {
 483                         /*
 484                          * We skip P_SYSTEM processes to match ps(1) output.
 485                          */
 486                         if ((kp->ki_flag & P_SYSTEM) == 0 &&
 487                             strcmp(kp->ki_loginclass, li->li_name) == 0)
 488                                 break;
 489                 }
 490                 if (SLIST_FIRST(&classlist) != NULL && li == NULL) {
 491                         selected[i] = 0;
 492                         continue;
 493                 }
 494
 495                 if (argc == 0)
 496                         selected[i] = 1;
 497         }
 498
 499         if (!ancestors) {
 500                 pid = mypid;
 501                 while (pid) {
 502                         for (i = 0, kp = plist; i < nproc; i++, kp++) {
 503                                 if (PSKIP(kp))
 504                                         continue;
 505                                 if (kp->ki_pid == pid) {
 506                                         selected[i] = 0;
 507                                         pid = kp->ki_ppid;
 508                                         break;
 509                                 }
 510                         }
 511                         if (i == nproc) {
 512                                 if (pid == mypid)
 513                                         pid = getppid();
 514                                 else
 515                                         break;  /* Maybe we're in a jail ? */
 516                         }
 517                 }
 518         }
 519
 520         if (newest || oldest) {
 521                 best_tval.tv_sec = 0;
 522                 best_tval.tv_usec = 0;
 523                 bestidx = -1;
 524
 525                 for (i = 0, kp = plist; i < nproc; i++, kp++) {
 526                         if (!selected[i])
 527                                 continue;
 528                         if (bestidx == -1) {
 529                                 /* The first entry of the list which matched. */
 530                                 ;
 531                         } else if (timercmp(&kp->ki_start, &best_tval, >)) {
 532                                 /* This entry is newer than previous "best". */
 533                                 if (oldest)     /* but we want the oldest */
 534                                         continue;
 535                         } else {
 536                                 /* This entry is older than previous "best". */
 537                                 if (newest)     /* but we want the newest */
 538                                         continue;
 539                         }
 540                         /* This entry is better than previous "best" entry. */
 541                         best_tval.tv_sec = kp->ki_start.tv_sec;
 542                         best_tval.tv_usec = kp->ki_start.tv_usec;
 543                         bestidx = i;
 544                 }
 545
 546                 memset(selected, 0, nproc);
 547                 if (bestidx != -1)
 548                         selected[bestidx] = 1;
 549         }
 550
 551         /*
 552          * Take the appropriate action for each matched process, if any.
 553          */
 554         did_action = 0;
 555         for (i = 0, rv = 0, kp = plist; i < nproc; i++, kp++) {
 556                 if (PSKIP(kp))
 557                         continue;
 558                 if (selected[i]) {
 559                         if (longfmt && !pgrep) {
 560                                 did_action = 1;
 561                                 printf("kill -%d %d\n", signum, kp->ki_pid);
 562                         }
 563                         if (inverse)
 564                                 continue;
 565                 } else if (!inverse)
 566                         continue;
 567                 rv |= (*action)(kp);
 568         }
 569         if (!did_action && !pgrep && longfmt)
 570                 fprintf(stderr,
 571                     "No matching processes belonging to you were found\n");
 572
 573         exit(rv ? STATUS_MATCH : STATUS_NOMATCH);
 574 }
 575
 576 static void
 577 usage(void)
 578 {
 579         const char *ustr;
 580
 581         if (pgrep)
 582                 ustr = "[-LSfilnoqvx] [-d delim]";
 583         else
 584                 ustr = "[-signal] [-ILfilnovx]";
 585
 586         fprintf(stderr,
 587                 "usage: %s %s [-F pidfile] [-G gid] [-M core] [-N system]\n"
 588                 "             [-P ppid] [-U uid] [-c class] [-g pgrp] [-j jid]\n"
 589                 "             [-s sid] [-t tty] [-u euid] pattern ...\n",
 590                 getprogname(), ustr);
 591
 592         exit(STATUS_BADUSAGE);
 593 }
 594
 595 static void
 596 show_process(const struct kinfo_proc *kp)
 597 {
 598         char **argv;
 599
 600         if (quiet) {
 601                 assert(pgrep);
 602                 return;
 603         }
 604         if ((longfmt || !pgrep) && matchargs &&
 605             (argv = kvm_getargv(kd, kp, 0)) != NULL) {
 606                 printf("%d ", (int)kp->ki_pid);
 607                 for (; *argv != NULL; argv++) {
 608                         printf("%s", *argv);
 609                         if (argv[1] != NULL)
 610                                 putchar(' ');
 611                 }
 612         } else if (longfmt || !pgrep)
 613                 printf("%d %s", (int)kp->ki_pid, kp->ki_comm);
 614         else
 615                 printf("%d", (int)kp->ki_pid);
 616 }
 617
 618 static int
 619 killact(const struct kinfo_proc *kp)
 620 {
 621         int ch, first;
 622
 623         if (interactive) {
 624                 /*
 625                  * Be careful, ask before killing.
 626                  */
 627                 printf("kill ");
 628                 show_process(kp);
 629                 printf("? ");
 630                 fflush(stdout);
 631                 first = ch = getchar();
 632                 while (ch != '\n' && ch != EOF)
 633                         ch = getchar();
 634                 if (first != 'y' && first != 'Y')
 635                         return (1);
 636         }
 637         if (kill(kp->ki_pid, signum) == -1) {
 638                 /*
 639                  * Check for ESRCH, which indicates that the process
 640                  * disappeared between us matching it and us
 641                  * signalling it; don't issue a warning about it.
 642                  */
 643                 if (errno != ESRCH)
 644                         warn("signalling pid %d", (int)kp->ki_pid);
 645                 /*
 646                  * Return 0 to indicate that the process should not be
 647                  * considered a match, since we didn't actually get to
 648                  * signal it.
 649                  */
 650                 return (0);
 651         }
 652         return (1);
 653 }
 654
 655 static int
 656 grepact(const struct kinfo_proc *kp)
 657 {
 658
 659         show_process(kp);
 660         if (!quiet)
 661                 printf("%s", delim);
 662         return (1);
 663 }
 664
 665 static void
 666 makelist(struct listhead *head, enum listtype type, char *src)
 667 {
 668         struct list *li;
 669         struct passwd *pw;
 670         struct group *gr;
 671         struct stat st;
 672         const char *cp;
 673         char *sp, *ep, buf[MAXPATHLEN];
 674         int empty;
 675
 676         empty = 1;
 677
 678         while ((sp = strsep(&src, ",")) != NULL) {
 679                 if (*sp == '\0')
 680                         usage();
 681
 682                 if ((li = malloc(sizeof(*li))) == NULL) {
 683                         err(STATUS_ERROR, "Cannot allocate %zu bytes",
 684                             sizeof(*li));
 685                 }
 686
 687                 SLIST_INSERT_HEAD(head, li, li_chain);
 688                 empty = 0;
 689
 690                 if (type != LT_CLASS)
 691                         li->li_number = (uid_t)strtol(sp, &ep, 0);
 692
 693                 if (type != LT_CLASS && *ep == '\0') {
 694                         switch (type) {
 695                         case LT_PGRP:
 696                                 if (li->li_number == 0)
 697                                         li->li_number = getpgrp();
 698                                 break;
 699                         case LT_SID:
 700                                 if (li->li_number == 0)
 701                                         li->li_number = getsid(mypid);
 702                                 break;
 703                         case LT_JID:
 704                                 if (li->li_number < 0)
 705                                         errx(STATUS_BADUSAGE,
 706                                              "Negative jail ID `%s'", sp);
 707                                 /* For compatibility with old -j */
 708                                 if (li->li_number == 0)
 709                                         li->li_number = -1;     /* any jail */
 710                                 break;
 711                         case LT_TTY:
 712                                 if (li->li_number < 0)
 713                                         errx(STATUS_BADUSAGE,
 714                                              "Negative /dev/pts tty `%s'", sp);
 715                                 snprintf(buf, sizeof(buf), _PATH_DEV "pts/%s",
 716                                     sp);
 717                                 if (stat(buf, &st) != -1)
 718                                         goto foundtty;
 719                                 if (errno == ENOENT)
 720                                         errx(STATUS_BADUSAGE, "No such tty: `"
 721                                             _PATH_DEV "pts/%s'", sp);
 722                                 err(STATUS_ERROR, "Cannot access `"
 723                                     _PATH_DEV "pts/%s'", sp);
 724                                 break;
 725                         default:
 726                                 break;
 727                         }
 728                         continue;
 729                 }
 730
 731                 switch (type) {
 732                 case LT_USER:
 733                         if ((pw = getpwnam(sp)) == NULL)
 734                                 errx(STATUS_BADUSAGE, "Unknown user `%s'", sp);
 735                         li->li_number = pw->pw_uid;
 736                         break;
 737                 case LT_GROUP:
 738                         if ((gr = getgrnam(sp)) == NULL)
 739                                 errx(STATUS_BADUSAGE, "Unknown group `%s'", sp);
 740                         li->li_number = gr->gr_gid;
 741                         break;
 742                 case LT_TTY:
 743                         if (strcmp(sp, "-") == 0) {
 744                                 li->li_number = -1;
 745                                 break;
 746                         } else if (strcmp(sp, "co") == 0) {
 747                                 cp = "console";
 748                         } else {
 749                                 cp = sp;
 750                         }
 751
 752                         snprintf(buf, sizeof(buf), _PATH_DEV "%s", cp);
 753                         if (stat(buf, &st) != -1)
 754                                 goto foundtty;
 755
 756                         snprintf(buf, sizeof(buf), _PATH_DEV "tty%s", cp);
 757                         if (stat(buf, &st) != -1)
 758                                 goto foundtty;
 759
 760                         if (errno == ENOENT)
 761                                 errx(STATUS_BADUSAGE, "No such tty: `%s'", sp);
 762                         err(STATUS_ERROR, "Cannot access `%s'", sp);
 763
 764 foundtty:               if ((st.st_mode & S_IFCHR) == 0)
 765                                 errx(STATUS_BADUSAGE, "Not a tty: `%s'", sp);
 766
 767                         li->li_number = st.st_rdev;
 768                         break;
 769                 case LT_JID:
 770                         if (strcmp(sp, "none") == 0)
 771                                 li->li_number = 0;
 772                         else if (strcmp(sp, "any") == 0)
 773                                 li->li_number = -1;
 774                         else if (*ep != '\0')
 775                                 errx(STATUS_BADUSAGE,
 776                                      "Invalid jail ID `%s'", sp);
 777                         break;
 778                 case LT_CLASS:
 779                         li->li_number = -1;
 780                         li->li_name = strdup(sp);
 781                         if (li->li_name == NULL)
 782                                 err(STATUS_ERROR, "Cannot allocate memory");
 783                         break;
 784                 default:
 785                         usage();
 786                 }
 787         }
 788
 789         if (empty)
 790                 usage();
 791 }
 792
 793 static int
 794 takepid(const char *pidfile, int pidfilelock)
 795 {
 796         char *endp, line[BUFSIZ];
 797         FILE *fh;
 798         long rval;
 799
 800         fh = fopen(pidfile, "r");
 801         if (fh == NULL)
 802                 err(STATUS_ERROR, "Cannot open pidfile `%s'", pidfile);
 803
 804         if (pidfilelock) {
 805                 /*
 806                  * If we can lock pidfile, this means that daemon is not
 807                  * running, so would be better not to kill some random process.
 808                  */
 809                 if (flock(fileno(fh), LOCK_EX | LOCK_NB) == 0) {
 810                         (void)fclose(fh);
 811                         errx(STATUS_ERROR, "File '%s' can be locked", pidfile);
 812                 } else {
 813                         if (errno != EWOULDBLOCK) {
 814                                 errx(STATUS_ERROR,
 815                                     "Error while locking file '%s'", pidfile);
 816                         }
 817                 }
 818         }
 819
 820         if (fgets(line, sizeof(line), fh) == NULL) {
 821                 if (feof(fh)) {
 822                         (void)fclose(fh);
 823                         errx(STATUS_ERROR, "Pidfile `%s' is empty", pidfile);
 824                 }
 825                 (void)fclose(fh);
 826                 err(STATUS_ERROR, "Cannot read from pid file `%s'", pidfile);
 827         }
 828         (void)fclose(fh);
 829
 830         rval = strtol(line, &endp, 10);
 831         if (*endp != '\0' && !isspace((unsigned char)*endp))
 832                 errx(STATUS_ERROR, "Invalid pid in file `%s'", pidfile);
 833         else if (rval < MIN_PID || rval > MAX_PID)
 834                 errx(STATUS_ERROR, "Invalid pid in file `%s'", pidfile);
 835         return (rval);
 836 }