1 /*- 2 * SPDX-License-Identifier: BSD-2-Clause 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 } 113