blob: 8a93057b048d0a288ba99aefab958fd2fb2a047b [file] [log] [blame]
"Robert P. J. Day"63fc1a92006-07-02 19:47:05 +00001/* vi: set sw=4 ts=4: */
Eric Andersenc98c0622001-12-06 15:16:43 +00002/*
3 * Calendar implementation for busybox
4 *
Eric Andersenc98c0622001-12-06 15:16:43 +00005 * See original copyright at the end of this file
6 *
"Robert P. J. Day"801ab142006-07-12 07:56:04 +00007 * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
8 */
Manuel Novoa III cad53642003-03-19 09:13:01 +00009
10/* BB_AUDIT SUSv3 compliant with -j and -y extensions (from util-linux). */
11/* BB_AUDIT BUG: The output of 'cal -j 1752' is incorrect. The upstream
12 * BB_AUDIT BUG: version in util-linux seems to be broken as well. */
13/* http://www.opengroup.org/onlinepubs/007904975/utilities/cal.html */
14
15/* Mar 16, 2003 Manuel Novoa III (mjn3@codepoet.org)
16 *
17 * Major size reduction... over 50% (>1.5k) on i386.
18 */
Eric Andersenc98c0622001-12-06 15:16:43 +000019
Denis Vlasenkob6adbf12007-05-26 19:00:18 +000020#include "libbb.h"
Manuel Novoa III b99cb642002-05-29 19:08:41 +000021
Denis Vlasenko661f6fa2007-07-26 11:12:51 +000022/* We often use "unsigned" intead of "int", it's easier to div on most CPUs */
23
Eric Andersenc98c0622001-12-06 15:16:43 +000024#define THURSDAY 4 /* for reformation */
Tim Rikerc1ef7bd2006-01-25 00:08:53 +000025#define SATURDAY 6 /* 1 Jan 1 was a Saturday */
Eric Andersenc98c0622001-12-06 15:16:43 +000026
Tim Rikerc1ef7bd2006-01-25 00:08:53 +000027#define FIRST_MISSING_DAY 639787 /* 3 Sep 1752 */
28#define NUMBER_MISSING_DAYS 11 /* 11 day correction */
Eric Andersenc98c0622001-12-06 15:16:43 +000029
30#define MAXDAYS 42 /* max slots in a month array */
31#define SPACE -1 /* used in day array */
32
Denis Vlasenko6ca409e2007-08-12 20:58:27 +000033static const unsigned char days_in_month[] ALIGN1 = {
Manuel Novoa III cad53642003-03-19 09:13:01 +000034 0, 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31
Eric Andersenc98c0622001-12-06 15:16:43 +000035};
36
Denis Vlasenko6ca409e2007-08-12 20:58:27 +000037static const unsigned char sep1752[] ALIGN1 = {
Denis Vlasenko9213a9e2006-09-17 16:28:10 +000038 1, 2, 14, 15, 16,
Eric Andersenc98c0622001-12-06 15:16:43 +000039 17, 18, 19, 20, 21, 22, 23,
Manuel Novoa III cad53642003-03-19 09:13:01 +000040 24, 25, 26, 27, 28, 29, 30
Eric Andersenc98c0622001-12-06 15:16:43 +000041};
42
Denis Vlasenko661f6fa2007-07-26 11:12:51 +000043static unsigned julian;
Eric Andersenc98c0622001-12-06 15:16:43 +000044
Eric Andersenaff114c2004-04-14 17:51:38 +000045/* leap year -- account for Gregorian reformation in 1752 */
Denis Vlasenko661f6fa2007-07-26 11:12:51 +000046static int leap_year(unsigned yr)
Manuel Novoa III cad53642003-03-19 09:13:01 +000047{
Denis Vlasenko319f8eb2007-08-13 11:09:30 +000048 if (yr <= 1752)
Denis Vlasenko661f6fa2007-07-26 11:12:51 +000049 return !(yr % 4);
50 return (!(yr % 4) && (yr % 100)) || !(yr % 400);
Manuel Novoa III cad53642003-03-19 09:13:01 +000051}
Manuel Novoa III cad53642003-03-19 09:13:01 +000052
Eric Andersenc98c0622001-12-06 15:16:43 +000053/* number of centuries since 1700, not inclusive */
54#define centuries_since_1700(yr) \
55 ((yr) > 1700 ? (yr) / 100 - 17 : 0)
56
57/* number of centuries since 1700 whose modulo of 400 is 0 */
58#define quad_centuries_since_1700(yr) \
59 ((yr) > 1600 ? ((yr) - 1600) / 400 : 0)
60
61/* number of leap years between year 1 and this year, not inclusive */
62#define leap_years_since_year_1(yr) \
63 ((yr) / 4 - centuries_since_1700(yr) + quad_centuries_since_1700(yr))
64
Denis Vlasenko661f6fa2007-07-26 11:12:51 +000065static void center(char *, unsigned, unsigned);
66static void day_array(unsigned, unsigned, unsigned *);
67static void trim_trailing_spaces_and_print(char *);
Manuel Novoa III cad53642003-03-19 09:13:01 +000068
69static void blank_string(char *buf, size_t buflen);
Denis Vlasenko661f6fa2007-07-26 11:12:51 +000070static char *build_row(char *p, unsigned *dp);
Manuel Novoa III cad53642003-03-19 09:13:01 +000071
72#define DAY_LEN 3 /* 3 spaces per day */
73#define J_DAY_LEN (DAY_LEN + 1)
74#define WEEK_LEN 20 /* 7 * 3 - one space at the end */
75#define J_WEEK_LEN (WEEK_LEN + 7)
76#define HEAD_SEP 2 /* spaces between day headings */
Eric Andersenc98c0622001-12-06 15:16:43 +000077
Denis Vlasenko9b49a5e2007-10-11 10:05:36 +000078int cal_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
Eric Andersenc98c0622001-12-06 15:16:43 +000079int cal_main(int argc, char **argv)
80{
81 struct tm *local_time;
Manuel Novoa III cad53642003-03-19 09:13:01 +000082 struct tm zero_tm;
Eric Andersenc98c0622001-12-06 15:16:43 +000083 time_t now;
Denis Vlasenko661f6fa2007-07-26 11:12:51 +000084 unsigned month, year, flags, i;
Manuel Novoa III cad53642003-03-19 09:13:01 +000085 char *month_names[12];
86 char day_headings[28]; /* 28 for julian, 21 for nonjulian */
Eric Andersenc98c0622001-12-06 15:16:43 +000087 char buf[40];
88
Denis Vlasenkofe7cd642007-08-18 15:32:12 +000089 flags = getopt32(argv, "jy");
Manuel Novoa III cad53642003-03-19 09:13:01 +000090 julian = flags & 1;
Eric Andersenc98c0622001-12-06 15:16:43 +000091 month = 0;
Denis Vlasenko661f6fa2007-07-26 11:12:51 +000092 argv += optind;
93 argc -= optind;
Manuel Novoa III cad53642003-03-19 09:13:01 +000094
Denis Vlasenko661f6fa2007-07-26 11:12:51 +000095 if (argc > 2) {
Manuel Novoa III cad53642003-03-19 09:13:01 +000096 bb_show_usage();
97 }
98
99 if (!argc) {
Eric Andersenc98c0622001-12-06 15:16:43 +0000100 time(&now);
101 local_time = localtime(&now);
102 year = local_time->tm_year + 1900;
Manuel Novoa III cad53642003-03-19 09:13:01 +0000103 if (!(flags & 2)) {
Eric Andersenc98c0622001-12-06 15:16:43 +0000104 month = local_time->tm_mon + 1;
Manuel Novoa III cad53642003-03-19 09:13:01 +0000105 }
106 } else {
107 if (argc == 2) {
Denis Vlasenko661f6fa2007-07-26 11:12:51 +0000108 month = xatou_range(*argv++, 1, 12);
Manuel Novoa III cad53642003-03-19 09:13:01 +0000109 }
Denis Vlasenko661f6fa2007-07-26 11:12:51 +0000110 year = xatou_range(*argv, 1, 9999);
Eric Andersenc98c0622001-12-06 15:16:43 +0000111 }
112
Denis Vlasenko661f6fa2007-07-26 11:12:51 +0000113 blank_string(day_headings, sizeof(day_headings) - 7 + 7*julian);
Manuel Novoa III cad53642003-03-19 09:13:01 +0000114
115 i = 0;
116 do {
Eric Andersenc98c0622001-12-06 15:16:43 +0000117 zero_tm.tm_mon = i;
118 strftime(buf, sizeof(buf), "%B", &zero_tm);
Rob Landleyd921b2e2006-08-03 15:41:12 +0000119 month_names[i] = xstrdup(buf);
Eric Andersenc98c0622001-12-06 15:16:43 +0000120
Manuel Novoa III cad53642003-03-19 09:13:01 +0000121 if (i < 7) {
122 zero_tm.tm_wday = i;
123 strftime(buf, sizeof(buf), "%a", &zero_tm);
124 strncpy(day_headings + i * (3+julian) + julian, buf, 2);
125 }
126 } while (++i < 12);
Eric Andersenc98c0622001-12-06 15:16:43 +0000127
Manuel Novoa III cad53642003-03-19 09:13:01 +0000128 if (month) {
Denis Vlasenko661f6fa2007-07-26 11:12:51 +0000129 unsigned row, len, days[MAXDAYS];
130 unsigned *dp = days;
Manuel Novoa III cad53642003-03-19 09:13:01 +0000131 char lineout[30];
Eric Andersenc7bda1c2004-03-15 08:29:22 +0000132
Manuel Novoa III cad53642003-03-19 09:13:01 +0000133 day_array(month, year, dp);
134 len = sprintf(lineout, "%s %d", month_names[month - 1], year);
Denis Vlasenkof0ed3762006-10-26 23:21:47 +0000135 printf("%*s%s\n%s\n",
Manuel Novoa III cad53642003-03-19 09:13:01 +0000136 ((7*julian + WEEK_LEN) - len) / 2, "",
137 lineout, day_headings);
Eric Andersenc98c0622001-12-06 15:16:43 +0000138 for (row = 0; row < 6; row++) {
Manuel Novoa III cad53642003-03-19 09:13:01 +0000139 build_row(lineout, dp)[0] = '\0';
140 dp += 7;
141 trim_trailing_spaces_and_print(lineout);
142 }
143 } else {
Denis Vlasenko661f6fa2007-07-26 11:12:51 +0000144 unsigned row, which_cal, week_len, days[12][MAXDAYS];
145 unsigned *dp;
Manuel Novoa III cad53642003-03-19 09:13:01 +0000146 char lineout[80];
Eric Andersenc7bda1c2004-03-15 08:29:22 +0000147
Manuel Novoa III cad53642003-03-19 09:13:01 +0000148 sprintf(lineout, "%d", year);
149 center(lineout,
150 (WEEK_LEN * 3 + HEAD_SEP * 2)
151 + julian * (J_WEEK_LEN * 2 + HEAD_SEP
152 - (WEEK_LEN * 3 + HEAD_SEP * 2)),
153 0);
154 puts("\n"); /* two \n's */
155 for (i = 0; i < 12; i++) {
156 day_array(i + 1, year, days[i]);
157 }
158 blank_string(lineout, sizeof(lineout));
159 week_len = WEEK_LEN + julian * (J_WEEK_LEN - WEEK_LEN);
160 for (month = 0; month < 12; month += 3-julian) {
161 center(month_names[month], week_len, HEAD_SEP);
162 if (!julian) {
163 center(month_names[month + 1], week_len, HEAD_SEP);
Eric Andersenc98c0622001-12-06 15:16:43 +0000164 }
Manuel Novoa III cad53642003-03-19 09:13:01 +0000165 center(month_names[month + 2 - julian], week_len, 0);
Denis Vlasenkof0ed3762006-10-26 23:21:47 +0000166 printf("\n%s%*s%s", day_headings, HEAD_SEP, "", day_headings);
Manuel Novoa III cad53642003-03-19 09:13:01 +0000167 if (!julian) {
Denis Vlasenkof0ed3762006-10-26 23:21:47 +0000168 printf("%*s%s", HEAD_SEP, "", day_headings);
Manuel Novoa III cad53642003-03-19 09:13:01 +0000169 }
Denis Vlasenko4daad902007-09-27 10:20:47 +0000170 bb_putchar('\n');
Manuel Novoa III cad53642003-03-19 09:13:01 +0000171 for (row = 0; row < (6*7); row += 7) {
172 for (which_cal = 0; which_cal < 3-julian; which_cal++) {
173 dp = days[month + which_cal] + row;
174 build_row(lineout + which_cal * (week_len + 2), dp);
175 }
176 /* blank_string took care of nul termination. */
177 trim_trailing_spaces_and_print(lineout);
178 }
Eric Andersenc98c0622001-12-06 15:16:43 +0000179 }
180 }
Eric Andersenc98c0622001-12-06 15:16:43 +0000181
Bernhard Reutner-Fischer636a1f82008-05-19 09:29:47 +0000182 fflush_stdout_and_exit(EXIT_SUCCESS);
Eric Andersenc98c0622001-12-06 15:16:43 +0000183}
184
185/*
186 * day_array --
187 * Fill in an array of 42 integers with a calendar. Assume for a moment
188 * that you took the (maximum) 6 rows in a calendar and stretched them
189 * out end to end. You would have 42 numbers or spaces. This routine
190 * builds that array for any month from Jan. 1 through Dec. 9999.
191 */
Denis Vlasenko661f6fa2007-07-26 11:12:51 +0000192static void day_array(unsigned month, unsigned year, unsigned *days)
Eric Andersenc98c0622001-12-06 15:16:43 +0000193{
Denis Vlasenko661f6fa2007-07-26 11:12:51 +0000194 unsigned long temp;
195 unsigned i;
196 unsigned day, dw, dm;
Eric Andersenc98c0622001-12-06 15:16:43 +0000197
Manuel Novoa III cad53642003-03-19 09:13:01 +0000198 memset(days, SPACE, MAXDAYS * sizeof(int));
Eric Andersenc98c0622001-12-06 15:16:43 +0000199
Manuel Novoa III cad53642003-03-19 09:13:01 +0000200 if ((month == 9) && (year == 1752)) {
Denis Vlasenko661f6fa2007-07-26 11:12:51 +0000201 /* Assumes the Gregorian reformation eliminates
202 * 3 Sep. 1752 through 13 Sep. 1752.
203 */
204 unsigned j_offset = julian * 244;
"Vladimir N. Oleynik"676b15e2006-01-30 13:47:19 +0000205 size_t oday = 0;
Denis Vlasenko9213a9e2006-09-17 16:28:10 +0000206
Manuel Novoa III cad53642003-03-19 09:13:01 +0000207 do {
"Vladimir N. Oleynik"676b15e2006-01-30 13:47:19 +0000208 days[oday+2] = sep1752[oday] + j_offset;
209 } while (++oday < sizeof(sep1752));
Eric Andersenc98c0622001-12-06 15:16:43 +0000210
Eric Andersenc98c0622001-12-06 15:16:43 +0000211 return;
212 }
Manuel Novoa III cad53642003-03-19 09:13:01 +0000213
214 /* day_in_year
Denis Vlasenko661f6fa2007-07-26 11:12:51 +0000215 * return the 1 based day number within the year
Manuel Novoa III cad53642003-03-19 09:13:01 +0000216 */
217 day = 1;
218 if ((month > 2) && leap_year(year)) {
219 ++day;
Eric Andersenc98c0622001-12-06 15:16:43 +0000220 }
Manuel Novoa III cad53642003-03-19 09:13:01 +0000221
222 i = month;
223 while (i) {
224 day += days_in_month[--i];
225 }
226
227 /* day_in_week
Denis Vlasenko661f6fa2007-07-26 11:12:51 +0000228 * return the 0 based day number for any date from 1 Jan. 1 to
229 * 31 Dec. 9999. Assumes the Gregorian reformation eliminates
230 * 3 Sep. 1752 through 13 Sep. 1752. Returns Thursday for all
231 * missing days.
Manuel Novoa III cad53642003-03-19 09:13:01 +0000232 */
Denis Vlasenko661f6fa2007-07-26 11:12:51 +0000233 temp = (long)(year - 1) * 365 + leap_years_since_year_1(year - 1) + day;
Manuel Novoa III cad53642003-03-19 09:13:01 +0000234 if (temp < FIRST_MISSING_DAY) {
235 dw = ((temp - 1 + SATURDAY) % 7);
Denis Vlasenko661f6fa2007-07-26 11:12:51 +0000236 } else {
Manuel Novoa III cad53642003-03-19 09:13:01 +0000237 dw = (((temp - 1 + SATURDAY) - NUMBER_MISSING_DAYS) % 7);
238 }
239
240 if (!julian) {
241 day = 1;
242 }
243
244 dm = days_in_month[month];
245 if ((month == 2) && leap_year(year)) {
246 ++dm;
247 }
248
Denis Vlasenko661f6fa2007-07-26 11:12:51 +0000249 do {
Manuel Novoa III cad53642003-03-19 09:13:01 +0000250 days[dw++] = day++;
Denis Vlasenko661f6fa2007-07-26 11:12:51 +0000251 } while (--dm);
Eric Andersenc98c0622001-12-06 15:16:43 +0000252}
253
Manuel Novoa III cad53642003-03-19 09:13:01 +0000254static void trim_trailing_spaces_and_print(char *s)
Eric Andersenc98c0622001-12-06 15:16:43 +0000255{
Manuel Novoa III cad53642003-03-19 09:13:01 +0000256 char *p = s;
Eric Andersenc98c0622001-12-06 15:16:43 +0000257
Manuel Novoa III cad53642003-03-19 09:13:01 +0000258 while (*p) {
Eric Andersenc98c0622001-12-06 15:16:43 +0000259 ++p;
Manuel Novoa III cad53642003-03-19 09:13:01 +0000260 }
Denis Vlasenko661f6fa2007-07-26 11:12:51 +0000261 while (p != s) {
Manuel Novoa III cad53642003-03-19 09:13:01 +0000262 --p;
263 if (!(isspace)(*p)) { /* We want the function... not the inline. */
264 p[1] = '\0';
265 break;
266 }
267 }
268
269 puts(s);
Eric Andersenc98c0622001-12-06 15:16:43 +0000270}
271
Denis Vlasenko661f6fa2007-07-26 11:12:51 +0000272static void center(char *str, unsigned len, unsigned separate)
Eric Andersenc98c0622001-12-06 15:16:43 +0000273{
Denis Vlasenko661f6fa2007-07-26 11:12:51 +0000274 unsigned n = strlen(str);
Manuel Novoa III cad53642003-03-19 09:13:01 +0000275 len -= n;
Denis Vlasenkof0ed3762006-10-26 23:21:47 +0000276 printf("%*s%*s", (len/2) + n, str, (len/2) + (len % 2) + separate, "");
Eric Andersenc98c0622001-12-06 15:16:43 +0000277}
278
Manuel Novoa III cad53642003-03-19 09:13:01 +0000279static void blank_string(char *buf, size_t buflen)
280{
281 memset(buf, ' ', buflen);
282 buf[buflen-1] = '\0';
283}
284
Denis Vlasenko661f6fa2007-07-26 11:12:51 +0000285static char *build_row(char *p, unsigned *dp)
Manuel Novoa III cad53642003-03-19 09:13:01 +0000286{
Denis Vlasenko661f6fa2007-07-26 11:12:51 +0000287 unsigned col, val, day;
Eric Andersenc7bda1c2004-03-15 08:29:22 +0000288
Manuel Novoa III cad53642003-03-19 09:13:01 +0000289 memset(p, ' ', (julian + DAY_LEN) * 7);
290
291 col = 0;
292 do {
Denis Vlasenko661f6fa2007-07-26 11:12:51 +0000293 day = *dp++;
294 if (day != SPACE) {
Manuel Novoa III cad53642003-03-19 09:13:01 +0000295 if (julian) {
Manuel Novoa III 5b3c0562003-08-13 12:11:33 +0000296 ++p;
Manuel Novoa III cad53642003-03-19 09:13:01 +0000297 if (day >= 100) {
298 *p = '0';
299 p[-1] = (day / 100) + '0';
300 day %= 100;
301 }
302 }
Denis Vlasenko661f6fa2007-07-26 11:12:51 +0000303 val = day / 10;
304 if (val > 0) {
Manuel Novoa III cad53642003-03-19 09:13:01 +0000305 *p = val + '0';
306 }
307 *++p = day % 10 + '0';
308 p += 2;
309 } else {
310 p += DAY_LEN + julian;
311 }
312 } while (++col < 7);
313
314 return p;
315}
Eric Andersenc98c0622001-12-06 15:16:43 +0000316
317/*
318 * Copyright (c) 1989, 1993, 1994
319 * The Regents of the University of California. All rights reserved.
320 *
321 * This code is derived from software contributed to Berkeley by
322 * Kim Letkeman.
323 *
324 * Redistribution and use in source and binary forms, with or without
325 * modification, are permitted provided that the following conditions
326 * are met:
327 * 1. Redistributions of source code must retain the above copyright
328 * notice, this list of conditions and the following disclaimer.
329 * 2. Redistributions in binary form must reproduce the above copyright
330 * notice, this list of conditions and the following disclaimer in the
331 * documentation and/or other materials provided with the distribution.
Aaron Lehmann69d41782002-06-23 22:25:24 +0000332 * 3. Neither the name of the University nor the names of its contributors
Eric Andersenc98c0622001-12-06 15:16:43 +0000333 * may be used to endorse or promote products derived from this software
334 * without specific prior written permission.
335 *
336 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
337 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
338 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
339 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
340 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
341 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
342 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
343 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
344 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
345 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
346 * SUCH DAMAGE.
347 */