Checking commit emails.
[dragonfly.git] / bin / date / vary.c
1 /*-
2  * Copyright (c) 1997 Brian Somers <brian@Awfulhak.org>
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  *
14  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24  * SUCH DAMAGE.
25  *
26  * $FreeBSD: src/bin/date/vary.c,v 1.16 2004/08/09 13:43:39 yar Exp $
27  * $DragonFly: src/bin/date/vary.c,v 1.4 2005/07/20 06:10:51 cpressey Exp $
28  */
29
30 #include <err.h>
31 #include <time.h>
32 #include <string.h>
33 #include <stdlib.h>
34 #include "vary.h"
35
36 struct trans {
37   int val;
38   const char *str;
39 };
40
41 static struct trans trans_mon[] = {
42   { 1, "january" }, { 2, "february" }, { 3, "march" }, { 4, "april" },
43   { 5, "may"}, { 6, "june" }, { 7, "july" }, { 8, "august" },
44   { 9, "september" }, { 10, "october" }, { 11, "november" }, { 12, "december" },
45   { -1, NULL }
46 };
47
48 static struct trans trans_wday[] = {
49   { 0, "sunday" }, { 1, "monday" }, { 2, "tuesday" }, { 3, "wednesday" },
50   { 4, "thursday" }, { 5, "friday" }, { 6, "saturday" },
51   { -1, NULL }
52 };
53
54 static char digits[] = "0123456789";
55 static int adjhour(struct tm *, char, int, int);
56
57 static int
58 domktime(struct tm *t, char type)
59 {
60   time_t ret;
61
62   while ((ret = mktime(t)) == -1 && t->tm_year > 68 && t->tm_year < 138)
63     /* While mktime() fails, adjust by an hour */
64     adjhour(t, type == '-' ? type : '+', 1, 0);
65
66   return ret;
67 }
68
69 static int
70 trans(const struct trans t[], const char *arg)
71 {
72   int f;
73
74   for (f = 0; t[f].val != -1; f++)
75     if (!strncasecmp(t[f].str, arg, 3) ||
76         !strncasecmp(t[f].str, arg, strlen(t[f].str)))
77       return t[f].val;
78
79   return -1;
80 }
81
82 struct vary *
83 vary_append(struct vary *v, char *arg)
84 {
85   struct vary *result, **nextp;
86
87   if (v) {
88     result = v;
89     while (v->next)
90       v = v->next;
91     nextp = &v->next;
92   } else
93     nextp = &result;
94
95   if ((*nextp = (struct vary *)malloc(sizeof(struct vary))) == NULL)
96     err(1, "malloc");
97   (*nextp)->arg = arg;
98   (*nextp)->next = NULL;
99   return result;
100 }
101
102 static int mdays[12] = { 31, 0, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31 };
103
104 static int
105 daysinmonth(const struct tm *t)
106 {
107   int year;
108
109   year = t->tm_year + 1900;
110
111   if (t->tm_mon == 1)
112     if (!(year % 400))
113       return 29;
114     else if (!(year % 100))
115       return 28;
116     else if (!(year % 4))
117       return 29;
118     else
119       return 28;
120   else if (t->tm_mon >= 0 && t->tm_mon < 12)
121     return mdays[t->tm_mon];
122
123   return 0;
124 }
125
126
127 static int
128 adjyear(struct tm *t, char type, int val, int mk)
129 {
130   switch (type) {
131     case '+':
132       t->tm_year += val;
133       break;
134     case '-':
135       t->tm_year -= val;
136       break;
137     default:
138       t->tm_year = val;
139       if (t->tm_year < 69)
140         t->tm_year += 100;              /* as per date.c */
141       else if (t->tm_year > 1900)
142         t->tm_year -= 1900;             /* struct tm holds years since 1900 */
143       break;
144   }
145   return !mk || domktime(t, type) != -1;
146 }
147
148 static int
149 adjmon(struct tm *t, char type, int val, int istext, int mk)
150 {
151   int lmdays;
152   if (val < 0)
153     return 0;
154
155   switch (type) {
156     case '+':
157       if (istext) {
158         if (val <= t->tm_mon)
159           val += 11 - t->tm_mon;        /* early next year */
160         else
161           val -= t->tm_mon + 1;         /* later this year */
162       }
163       if (val) {
164         if (!adjyear(t, '+', (t->tm_mon + val) / 12, 0))
165           return 0;
166         val %= 12;
167         t->tm_mon += val;
168         if (t->tm_mon > 11)
169           t->tm_mon -= 12;
170       }
171       break;
172
173     case '-':
174       if (istext) {
175         if (val-1 > t->tm_mon)
176           val = 13 - val + t->tm_mon;   /* later last year */
177         else
178           val = t->tm_mon - val + 1;    /* early this year */
179       }
180       if (val) {
181         if (!adjyear(t, '-', val / 12, 0))
182           return 0;
183         val %= 12;
184         if (val > t->tm_mon) {
185           if (!adjyear(t, '-', 1, 0))
186             return 0;
187           val -= 12;
188         }
189         t->tm_mon -= val;
190       }
191       break;
192
193     default:
194       if (val > 12 || val < 1)
195         return 0;
196       t->tm_mon = --val;
197   }
198
199   /* e.g., -v-1m on March, 31 is the last day of February in common sense */
200   lmdays = daysinmonth(t);
201   if (t->tm_mday > lmdays)
202     t->tm_mday = lmdays;
203
204   return !mk || domktime(t, type) != -1;
205 }
206
207 static int
208 adjday(struct tm *t, char type, int val, int mk)
209 {
210   int daycount;
211
212   switch (type) {
213     case '+':
214       while (val) {
215         daycount = daysinmonth(t);
216         if (val > daycount - t->tm_mday) {
217           val -= daycount - t->tm_mday + 1;
218           t->tm_mday = 1;
219           if (!adjmon(t, '+', 1, 0, 0))
220             return 0;
221         } else {
222           t->tm_mday += val;
223           val = 0;
224         }
225       }
226       break;
227     case '-':
228       while (val)
229         if (val >= t->tm_mday) {
230           val -= t->tm_mday;
231           t->tm_mday = 1;
232           if (!adjmon(t, '-', 1, 0, 0))
233             return 0;
234           t->tm_mday = daysinmonth(t);
235         } else {
236           t->tm_mday -= val;
237           val = 0;
238         }
239       break;
240     default:
241       if (val > 0 && val <= daysinmonth(t))
242         t->tm_mday = val;
243       else
244         return 0;
245       break;
246   }
247
248   return !mk || domktime(t, type) != -1;
249 }
250
251 static int
252 adjwday(struct tm *t, char type, int val, int istext, int mk)
253 {
254   if (val < 0)
255     return 0;
256
257   switch (type) {
258     case '+':
259       if (istext)
260         if (val < t->tm_wday)
261           val = 7 - t->tm_wday + val;  /* early next week */
262         else
263           val -= t->tm_wday;           /* later this week */
264       else
265         val *= 7;                      /* "-v+5w" == "5 weeks in the future" */
266       return !val || adjday(t, '+', val, mk);
267     case '-':
268       if (istext) {
269         if (val > t->tm_wday)
270           val = 7 - val + t->tm_wday;  /* later last week */
271         else
272           val = t->tm_wday - val;      /* early this week */
273       } else
274         val *= 7;                      /* "-v-5w" == "5 weeks ago" */
275       return !val || adjday(t, '-', val, mk);
276     default:
277       if (val < t->tm_wday)
278         return adjday(t, '-', t->tm_wday - val, mk);
279       else if (val > 6)
280         return 0;
281       else if (val > t->tm_wday)
282         return adjday(t, '+', val - t->tm_wday, mk);
283   }
284   return 1;
285 }
286
287 static int
288 adjhour(struct tm *t, char type, int val, int mk)
289 {
290   if (val < 0)
291     return 0;
292
293   switch (type) {
294     case '+':
295       if (val) {
296         int days;
297
298         days = (t->tm_hour + val) / 24;
299         val %= 24;
300         t->tm_hour += val;
301         t->tm_hour %= 24;
302         if (!adjday(t, '+', days, 0))
303           return 0;
304       }
305       break;
306
307     case '-':
308       if (val) {
309         int days;
310
311         days = val / 24;
312         val %= 24;
313         if (val > t->tm_hour) {
314           days++;
315           val -= 24;
316         }
317         t->tm_hour -= val;
318         if (!adjday(t, '-', days, 0))
319           return 0;
320       }
321       break;
322
323     default:
324       if (val > 23)
325         return 0;
326       t->tm_hour = val;
327   }
328
329   return !mk || domktime(t, type) != -1;
330 }
331
332 static int
333 adjmin(struct tm *t, char type, int val, int mk)
334 {
335   if (val < 0)
336     return 0;
337
338   switch (type) {
339     case '+':
340       if (val) {
341         if (!adjhour(t, '+', (t->tm_min + val) / 60, 0))
342           return 0;
343         val %= 60;
344         t->tm_min += val;
345         if (t->tm_min > 59)
346           t->tm_min -= 60;
347       }
348       break;
349
350     case '-':
351       if (val) {
352         if (!adjhour(t, '-', val / 60, 0))
353           return 0;
354         val %= 60;
355         if (val > t->tm_min) {
356           if (!adjhour(t, '-', 1, 0))
357             return 0;
358           val -= 60;
359         }
360         t->tm_min -= val;
361       }
362       break;
363
364     default:
365       if (val > 59)
366         return 0;
367       t->tm_min = val;
368   }
369
370   return !mk || domktime(t, type) != -1;
371 }
372
373 static int
374 adjsec(struct tm *t, char type, int val, int mk)
375 {
376   if (val < 0)
377     return 0;
378
379   switch (type) {
380     case '+':
381       if (val) {
382         if (!adjmin(t, '+', (t->tm_sec + val) / 60, 0))
383           return 0;
384         val %= 60;
385         t->tm_sec += val;
386         if (t->tm_sec > 59)
387           t->tm_sec -= 60;
388       }
389       break;
390
391     case '-':
392       if (val) {
393         if (!adjmin(t, '-', val / 60, 0))
394           return 0;
395         val %= 60;
396         if (val > t->tm_sec) {
397           if (!adjmin(t, '-', 1, 0))
398             return 0;
399           val -= 60;
400         }
401         t->tm_sec -= val;
402       }
403       break;
404
405     default:
406       if (val > 59)
407         return 0;
408       t->tm_sec = val;
409   }
410
411   return !mk || domktime(t, type) != -1;
412 }
413
414 const struct vary *
415 vary_apply(const struct vary *v, struct tm *t)
416 {
417   char type;
418   char which;
419   char *arg;
420   size_t len;
421   int val;
422
423   for (; v; v = v->next) {
424     type = *v->arg;
425     arg = v->arg;
426     if (type == '+' || type == '-')
427       arg++;
428     else
429       type = '\0';
430     len = strlen(arg);
431     if (len < 2)
432       return v;
433
434     if (type == '\0')
435       t->tm_isdst = -1;
436
437     if (strspn(arg, digits) != len-1) {
438       val = trans(trans_wday, arg);
439       if (val != -1) {
440           if (!adjwday(t, type, val, 1, 1))
441             return v;
442       } else {
443         val = trans(trans_mon, arg);
444         if (val != -1) {
445           if (!adjmon(t, type, val, 1, 1))
446             return v;
447         } else
448           return v;
449       }
450     } else {
451       val = atoi(arg);
452       which = arg[len-1];
453       
454       switch (which) {
455         case 'S':
456           if (!adjsec(t, type, val, 1))
457             return v;
458           break;
459         case 'M':
460           if (!adjmin(t, type, val, 1))
461             return v;
462           break;
463         case 'H':
464           if (!adjhour(t, type, val, 1))
465             return v;
466           break;
467         case 'd':
468           t->tm_isdst = -1;
469           if (!adjday(t, type, val, 1))
470             return v;
471           break;
472         case 'w':
473           t->tm_isdst = -1;
474           if (!adjwday(t, type, val, 0, 1))
475             return v;
476           break;
477         case 'm':
478           t->tm_isdst = -1;
479           if (!adjmon(t, type, val, 0, 1))
480             return v;
481           break;
482         case 'y':
483           t->tm_isdst = -1;
484           if (!adjyear(t, type, val, 1))
485             return v;
486           break;
487         default:
488           return v;
489       }
490     }
491   }
492   return 0;
493 }
494
495 void
496 vary_destroy(struct vary *v)
497 {
498   struct vary *n;
499
500   while (v) {
501     n = v->next;
502     free(v);
503     v = n;
504   }
505 }