1 /* $OpenBSD: leave.c,v 1.12 2009/10/27 23:59:39 deraadt Exp $ */ 2 /* $NetBSD: leave.c,v 1.4 1995/07/03 16:50:13 phil Exp $ */ 3 4 /* 5 * Copyright (c) 1980, 1988, 1993 6 * The Regents of the University of California. All rights reserved. 7 * 8 * Redistribution and use in source and binary forms, with or without 9 * modification, are permitted provided that the following conditions 10 * are met: 11 * 1. Redistributions of source code must retain the above copyright 12 * notice, this list of conditions and the following disclaimer. 13 * 2. Redistributions in binary form must reproduce the above copyright 14 * notice, this list of conditions and the following disclaimer in the 15 * documentation and/or other materials provided with the distribution. 16 * 3. Neither the name of the University nor the names of its contributors 17 * may be used to endorse or promote products derived from this software 18 * without specific prior written permission. 19 * 20 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND 21 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 22 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 23 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE 24 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 25 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 26 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 27 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 28 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 29 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 30 * SUCH DAMAGE. 31 */ 32 33 #include <sys/param.h> 34 #include <sys/time.h> 35 #include <ctype.h> 36 #include <err.h> 37 #include <stdio.h> 38 #include <stdlib.h> 39 #include <unistd.h> 40 41 static __dead void usage(void); 42 static void doalarm(u_int secs); 43 44 #define SECOND 1 45 #define MINUTE (SECOND * 60) 46 #define FIVEMIN (5 * MINUTE) 47 #define HOUR (MINUTE * 60) 48 49 /* 50 * leave [[+]hhmm] 51 * 52 * Reminds you when you have to leave. 53 * Leave prompts for input and goes away if you hit return. 54 * It nags you like a mother hen. 55 */ 56 int 57 main(int argc, char *argv[]) 58 { 59 u_int secs; 60 int hours, minutes; 61 char c, *cp; 62 struct tm *t; 63 time_t now; 64 int plusnow = 0, twentyfour; 65 char buf[50]; 66 67 if (setlinebuf(stdout) != 0) 68 errx(1, "Cannot set stdout to line buffered."); 69 70 if (argc < 2) { 71 (void)fputs("When do you have to leave? ", stdout); 72 cp = fgets(buf, sizeof(buf), stdin); 73 if (cp == NULL || *cp == '\n') 74 exit(0); 75 } else if (argc > 2) 76 usage(); 77 else 78 cp = argv[1]; 79 80 if (*cp == '+') { 81 plusnow = 1; 82 ++cp; 83 } 84 85 for (hours = 0; (c = *cp) && c != '\n'; ++cp) { 86 if (!isdigit(c)) 87 usage(); 88 hours = hours * 10 + (c - '0'); 89 } 90 minutes = hours % 100; 91 hours /= 100; 92 /* determine 24 hours mode */ 93 twentyfour = hours > 12; 94 95 if (minutes < 0 || minutes > 59) 96 usage(); 97 if (plusnow) 98 secs = (hours * HOUR) + (minutes * MINUTE); 99 else { 100 if (hours > 23) 101 usage(); 102 (void)time(&now); 103 t = localtime(&now); 104 while (t->tm_hour > hours || 105 (t->tm_hour == hours && t->tm_min >= minutes)) { 106 if (twentyfour) 107 hours += 24; 108 else 109 hours += 12; 110 } 111 112 secs = (hours - t->tm_hour) * HOUR; 113 secs += (minutes - t->tm_min) * MINUTE; 114 } 115 doalarm(secs); 116 exit(0); 117 } 118 119 static void 120 doalarm(u_int secs) 121 { 122 int bother; 123 time_t daytime; 124 pid_t pid; 125 126 switch (pid = fork()) { 127 case 0: 128 break; 129 case -1: 130 err(1, "Fork failed"); 131 /* NOTREACHED */ 132 default: 133 (void)time(&daytime); 134 daytime += secs; 135 printf("Alarm set for %.16s. (pid %ld)\n", 136 ctime(&daytime), (long)pid); 137 exit(0); 138 } 139 sleep(2); /* let parent print set message */ 140 141 /* 142 * if write fails, we've lost the terminal through someone else 143 * causing a vhangup by logging in. 144 */ 145 if (secs >= FIVEMIN) { 146 sleep(secs - FIVEMIN); 147 if (puts("\a\aYou have to leave in 5 minutes.") == EOF) 148 exit(0); 149 secs = FIVEMIN; 150 } 151 152 if (secs >= MINUTE) { 153 sleep(secs - MINUTE); 154 if (puts("\a\aJust one more minute!") == EOF) 155 exit(0); 156 } 157 158 for (bother = 10; bother--;) { 159 sleep(MINUTE); 160 if (puts("\a\aTime to leave!") == EOF) 161 exit(0); 162 } 163 164 puts("\a\aThat was the last time I'll tell you. Bye."); 165 exit(0); 166 } 167 168 static __dead void 169 usage(void) 170 { 171 fprintf(stderr, "usage: leave [[+]hhmm]\n"); 172 exit(1); 173 } 174