1 |
|
|
/* $OpenBSD: leave.c,v 1.17 2015/10/09 01:37:08 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/time.h> |
34 |
|
|
#include <ctype.h> |
35 |
|
|
#include <err.h> |
36 |
|
|
#include <stdio.h> |
37 |
|
|
#include <stdlib.h> |
38 |
|
|
#include <unistd.h> |
39 |
|
|
|
40 |
|
|
static __dead void usage(void); |
41 |
|
|
static void doalarm(u_int secs); |
42 |
|
|
|
43 |
|
|
#define SECOND 1 |
44 |
|
|
#define MINUTE (SECOND * 60) |
45 |
|
|
#define FIVEMIN (5 * MINUTE) |
46 |
|
|
#define HOUR (MINUTE * 60) |
47 |
|
|
|
48 |
|
|
/* |
49 |
|
|
* leave [[+]hhmm] |
50 |
|
|
* |
51 |
|
|
* Reminds you when you have to leave. |
52 |
|
|
* Leave prompts for input and goes away if you hit return. |
53 |
|
|
* It nags you like a mother hen. |
54 |
|
|
*/ |
55 |
|
|
int |
56 |
|
|
main(int argc, char *argv[]) |
57 |
|
|
{ |
58 |
|
|
u_int secs; |
59 |
|
|
int hours, minutes; |
60 |
|
|
char c, *cp; |
61 |
|
|
struct tm *t; |
62 |
|
|
time_t now; |
63 |
|
|
int plusnow = 0, twentyfour; |
64 |
|
|
char buf[50]; |
65 |
|
|
|
66 |
|
|
if (pledge("stdio proc flock rpath cpath wpath", NULL) == -1) |
67 |
|
|
err(1, "pledge"); |
68 |
|
|
|
69 |
|
|
if (setvbuf(stdout, NULL, _IOLBF, 0) != 0) |
70 |
|
|
errx(1, "Cannot set stdout to line buffered."); |
71 |
|
|
|
72 |
|
|
if (argc < 2) { |
73 |
|
|
(void)fputs("When do you have to leave? ", stdout); |
74 |
|
|
cp = fgets(buf, sizeof(buf), stdin); |
75 |
|
|
if (cp == NULL || *cp == '\n') |
76 |
|
|
exit(0); |
77 |
|
|
} else if (argc > 2) |
78 |
|
|
usage(); |
79 |
|
|
else |
80 |
|
|
cp = argv[1]; |
81 |
|
|
|
82 |
|
|
if (*cp == '+') { |
83 |
|
|
plusnow = 1; |
84 |
|
|
++cp; |
85 |
|
|
} |
86 |
|
|
|
87 |
|
|
for (hours = 0; (c = *cp) && c != '\n'; ++cp) { |
88 |
|
|
if (!isdigit((unsigned char)c)) |
89 |
|
|
usage(); |
90 |
|
|
hours = hours * 10 + (c - '0'); |
91 |
|
|
} |
92 |
|
|
minutes = hours % 100; |
93 |
|
|
hours /= 100; |
94 |
|
|
/* determine 24 hours mode */ |
95 |
|
|
twentyfour = hours > 12; |
96 |
|
|
|
97 |
|
|
if (minutes < 0 || minutes > 59) |
98 |
|
|
usage(); |
99 |
|
|
if (plusnow) |
100 |
|
|
secs = (hours * HOUR) + (minutes * MINUTE); |
101 |
|
|
else { |
102 |
|
|
if (hours > 23) |
103 |
|
|
usage(); |
104 |
|
|
(void)time(&now); |
105 |
|
|
t = localtime(&now); |
106 |
|
|
while (t->tm_hour > hours || |
107 |
|
|
(t->tm_hour == hours && t->tm_min >= minutes)) { |
108 |
|
|
if (twentyfour) |
109 |
|
|
hours += 24; |
110 |
|
|
else |
111 |
|
|
hours += 12; |
112 |
|
|
} |
113 |
|
|
|
114 |
|
|
secs = (hours - t->tm_hour) * HOUR; |
115 |
|
|
secs += (minutes - t->tm_min) * MINUTE; |
116 |
|
|
} |
117 |
|
|
doalarm(secs); |
118 |
|
|
exit(0); |
119 |
|
|
} |
120 |
|
|
|
121 |
|
|
static void |
122 |
|
|
doalarm(u_int secs) |
123 |
|
|
{ |
124 |
|
|
int bother; |
125 |
|
|
time_t daytime; |
126 |
|
|
pid_t pid; |
127 |
|
|
|
128 |
|
|
switch (pid = fork()) { |
129 |
|
|
case 0: |
130 |
|
|
break; |
131 |
|
|
case -1: |
132 |
|
|
err(1, "Fork failed"); |
133 |
|
|
/* NOTREACHED */ |
134 |
|
|
default: |
135 |
|
|
(void)time(&daytime); |
136 |
|
|
daytime += secs; |
137 |
|
|
printf("Alarm set for %.16s. (pid %ld)\n", |
138 |
|
|
ctime(&daytime), (long)pid); |
139 |
|
|
exit(0); |
140 |
|
|
} |
141 |
|
|
sleep(2); /* let parent print set message */ |
142 |
|
|
|
143 |
|
|
/* |
144 |
|
|
* if write fails, we've lost the terminal through someone else |
145 |
|
|
* causing a vhangup by logging in. |
146 |
|
|
*/ |
147 |
|
|
if (secs >= FIVEMIN) { |
148 |
|
|
sleep(secs - FIVEMIN); |
149 |
|
|
if (puts("\a\aYou have to leave in 5 minutes.") == EOF) |
150 |
|
|
exit(0); |
151 |
|
|
secs = FIVEMIN; |
152 |
|
|
} |
153 |
|
|
|
154 |
|
|
if (secs >= MINUTE) { |
155 |
|
|
sleep(secs - MINUTE); |
156 |
|
|
if (puts("\a\aJust one more minute!") == EOF) |
157 |
|
|
exit(0); |
158 |
|
|
} |
159 |
|
|
|
160 |
|
|
for (bother = 10; bother--;) { |
161 |
|
|
sleep(MINUTE); |
162 |
|
|
if (puts("\a\aTime to leave!") == EOF) |
163 |
|
|
exit(0); |
164 |
|
|
} |
165 |
|
|
|
166 |
|
|
puts("\a\aThat was the last time I'll tell you. Bye."); |
167 |
|
|
exit(0); |
168 |
|
|
} |
169 |
|
|
|
170 |
|
|
static __dead void |
171 |
|
|
usage(void) |
172 |
|
|
{ |
173 |
|
|
fprintf(stderr, "usage: leave [[+]hhmm]\n"); |
174 |
|
|
exit(1); |
175 |
|
|
} |