usr.sbin/utx/utx.c

   1 /*-
   2  * SPDX-License-Identifier: BSD-2-Clause-FreeBSD
   3  *
   4  * Copyright (c) 2011-2012 Ed Schouten <ed@FreeBSD.org>
   5  * All rights reserved.
   6  *
   7  * Redistribution and use in source and binary forms, with or without
   8  * modification, are permitted provided that the following conditions
   9  * are met:
  10  * 1. Redistributions of source code must retain the above copyright
  11  *    notice, this list of conditions and the following disclaimer.
  12  * 2. Redistributions in binary form must reproduce the above copyright
  13  *    notice, this list of conditions and the following disclaimer in the
  14  *    documentation and/or other materials provided with the distribution.
  15  *
  16  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  17  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  18  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  19  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  20  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  21  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  22  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  23  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  24  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  25  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  26  * SUCH DAMAGE.
  27  */
  28
  29 #include <sys/cdefs.h>
  30 __FBSDID("$FreeBSD$");
  31
  32 #include <sys/time.h>
  33 #include <errno.h>
  34 #include <ctype.h>
  35 #include <stdio.h>
  36 #include <stdlib.h>
  37 #include <string.h>
  38 #include <utmpx.h>
  39
  40 static int
  41 b16_pton(const char *in, char *out, size_t len)
  42 {
  43         size_t i;
  44
  45         for (i = 0; i < len * 2; i++)
  46                 if (!isxdigit((unsigned char)in[i]))
  47                         return (1);
  48         for (i = 0; i < len; i++)
  49                 sscanf(&in[i * 2], "%02hhx", &out[i]);
  50         return (0);
  51 }
  52
  53 static int
  54 rm(char *id[])
  55 {
  56         struct utmpx utx = { .ut_type = DEAD_PROCESS };
  57         size_t len;
  58         int ret = 0;
  59
  60         (void)gettimeofday(&utx.ut_tv, NULL);
  61         for (; *id != NULL; id++) {
  62                 len = strlen(*id);
  63                 if (len <= sizeof(utx.ut_id)) {
  64                         /* Identifier as string. */
  65                         strncpy(utx.ut_id, *id, sizeof(utx.ut_id));
  66                 } else if (len != sizeof(utx.ut_id) * 2 ||
  67                     b16_pton(*id, utx.ut_id, sizeof(utx.ut_id)) != 0) {
  68                         /* Also not hexadecimal. */
  69                         fprintf(stderr, "%s: Invalid identifier format\n", *id);
  70                         ret = 1;
  71                         continue;
  72                 }
  73
  74                 /* Zap the entry. */
  75                 if (pututxline(&utx) == NULL) {
  76                         perror(*id);
  77                         ret = 1;
  78                 }
  79         }
  80         return (ret);
  81 }
  82
  83 static int
  84 boot(short type)
  85 {
  86         struct utmpx utx = { .ut_type = type };
  87
  88         (void)gettimeofday(&utx.ut_tv, NULL);
  89         if (pututxline(&utx) == NULL) {
  90                 perror("pututxline");
  91                 return (1);
  92         }
  93         return (0);
  94 }
  95
  96 int
  97 main(int argc, char *argv[])
  98 {
  99
 100         if (argc == 2 && strcmp(argv[1], "boot") == 0)
 101                 return (boot(BOOT_TIME));
 102         else if (argc == 2 && strcmp(argv[1], "shutdown") == 0)
 103                 return (boot(SHUTDOWN_TIME));
 104         else if (argc >= 3 && strcmp(argv[1], "rm") == 0)
 105                 return (rm(&argv[2]));
 106
 107         fprintf(stderr,
 108             "usage: utx boot\n"
 109             "       utx shutdown\n"
 110             "       utx rm identifier ...\n");
 111         exit(1);
 112 }