1 /*
2 * Copyright 1996 Massachusetts Institute of Technology
3 *
4 * Permission to use, copy, modify, and distribute this software and
5 * its documentation for any purpose and without fee is hereby
6 * granted, provided that both the above copyright notice and this
7 * permission notice appear in all copies, that both the above
8 * copyright notice and this permission notice appear in all
9 * supporting documentation, and that the name of M.I.T. not be used
10 * in advertising or publicity pertaining to distribution of the
11 * software without specific, written prior permission. M.I.T. makes
12 * no representations about the suitability of this software for any
13 * purpose. It is provided "as is" without express or implied
14 * warranty.
15 *
16 * THIS SOFTWARE IS PROVIDED BY M.I.T. ``AS IS''. M.I.T. DISCLAIMS
17 * ALL EXPRESS OR IMPLIED WARRANTIES WITH REGARD TO THIS SOFTWARE,
18 * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
19 * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE. IN NO EVENT
20 * SHALL M.I.T. BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
21 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
22 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
23 * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
24 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
25 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
26 * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27 * SUCH DAMAGE.
28 */
29
30 /*
31 * Second attempt at a `tzmenu' program, using the separate description
32 * files provided in newer tzdata releases.
33 */
34
35 #include <sys/cdefs.h>
36 __FBSDID("$FreeBSD$");
37
38 #include <err.h>
39 #include <errno.h>
40 #include <stdio.h>
41 #include <stdlib.h>
42 #include <string.h>
43 #include <time.h>
44 #include <unistd.h>
45
46 #include <sys/fcntl.h>
47 #include <sys/param.h>
48 #include <sys/queue.h>
49 #include <sys/stat.h>
50 #include <sys/sysctl.h>
51
52 #ifdef HAVE_DIALOG
53 #include <dialog.h>
54 #endif
55
56 #define _PATH_ZONETAB "/usr/share/zoneinfo/zone1970.tab"
57 #define _PATH_ISO3166 "/usr/share/misc/iso3166"
58 #define _PATH_ZONEINFO "/usr/share/zoneinfo"
59 #define _PATH_LOCALTIME "/etc/localtime"
60 #define _PATH_DB "/var/db/zoneinfo"
61 #define _PATH_WALL_CMOS_CLOCK "/etc/wall_cmos_clock"
62
63 #ifdef PATH_MAX
64 #define SILLY_BUFFER_SIZE 2*PATH_MAX
65 #else
66 #warning "Somebody needs to fix this to dynamically size this buffer."
67 #define SILLY_BUFFER_SIZE 2048
68 #endif
69
70 /* special return codes for `fire' actions */
71 #define DITEM_FAILURE 1
72
73 /* flags - returned in upper 16 bits of return status */
74 #define DITEM_LEAVE_MENU (1 << 16)
75 #define DITEM_RECREATE (1 << 18)
76
77 static char path_zonetab[MAXPATHLEN], path_iso3166[MAXPATHLEN],
78 path_zoneinfo[MAXPATHLEN], path_localtime[MAXPATHLEN],
79 path_db[MAXPATHLEN], path_wall_cmos_clock[MAXPATHLEN];
80
81 static int reallydoit = 1;
82 static int reinstall = 0;
83 static char *chrootenv = NULL;
84
85 static void usage(void);
86 static int install_zoneinfo(const char *zoneinfo);
87 static int install_zoneinfo_file(const char *zoneinfo_file);
88
89 #ifdef HAVE_DIALOG
90 /* for use in describing more exotic behaviors */
91 typedef struct dialogMenuItem {
92 char *prompt;
93 char *title;
94 int (*fire)(struct dialogMenuItem *self);
95 void *data;
96 } dialogMenuItem;
97
98 static int
xdialog_count_rows(const char * p)99 xdialog_count_rows(const char *p)
100 {
101 int rows = 0;
102
103 while ((p = strchr(p, '\n')) != NULL) {
104 p++;
105 if (*p == '\0')
106 break;
107 rows++;
108 }
109
110 return (rows ? rows : 1);
111 }
112
113 static int
xdialog_count_columns(const char * p)114 xdialog_count_columns(const char *p)
115 {
116 int len;
117 int max_len = 0;
118 const char *q;
119
120 for (; (q = strchr(p, '\n')) != NULL; p = q + 1) {
121 len = q - p;
122 max_len = MAX(max_len, len);
123 }
124
125 len = strlen(p);
126 max_len = MAX(max_len, len);
127 return (max_len);
128 }
129
130 static int
xdialog_menu(const char * title,const char * cprompt,int height,int width,int menu_height,int item_no,dialogMenuItem * ditems)131 xdialog_menu(const char *title, const char *cprompt, int height, int width,
132 int menu_height, int item_no, dialogMenuItem *ditems)
133 {
134 int i, result, choice = 0;
135 DIALOG_LISTITEM *listitems;
136 DIALOG_VARS save_vars;
137
138 dlg_save_vars(&save_vars);
139
140 /* initialize list items */
141 listitems = dlg_calloc(DIALOG_LISTITEM, item_no + 1);
142 assert_ptr(listitems, "xdialog_menu");
143 for (i = 0; i < item_no; i++) {
144 listitems[i].name = ditems[i].prompt;
145 listitems[i].text = ditems[i].title;
146 }
147
148 /* calculate height */
149 if (height < 0)
150 height = xdialog_count_rows(cprompt) + menu_height + 4 + 2;
151 if (height > LINES)
152 height = LINES;
153
154 /* calculate width */
155 if (width < 0) {
156 int tag_x = 0;
157
158 for (i = 0; i < item_no; i++) {
159 int j, l;
160
161 l = strlen(listitems[i].name);
162 for (j = 0; j < item_no; j++) {
163 int k = strlen(listitems[j].text);
164 tag_x = MAX(tag_x, l + k + 2);
165 }
166 }
167 width = MAX(xdialog_count_columns(cprompt), title != NULL ?
168 xdialog_count_columns(title) : 0);
169 width = MAX(width, tag_x + 4) + 4;
170 }
171 width = MAX(width, 24);
172 if (width > COLS)
173 width = COLS;
174
175 again:
176 dialog_vars.default_item = listitems[choice].name;
177 result = dlg_menu(title, cprompt, height, width,
178 menu_height, item_no, listitems, &choice, NULL);
179 switch (result) {
180 case DLG_EXIT_ESC:
181 result = -1;
182 break;
183 case DLG_EXIT_OK:
184 if (ditems[choice].fire != NULL) {
185 int status;
186
187 status = ditems[choice].fire(ditems + choice);
188 if (status & DITEM_RECREATE) {
189 dlg_clear();
190 goto again;
191 }
192 }
193 result = 0;
194 break;
195 case DLG_EXIT_CANCEL:
196 default:
197 result = 1;
198 break;
199 }
200
201 free(listitems);
202 dlg_restore_vars(&save_vars);
203 return (result);
204 }
205
206 static int usedialog = 1;
207
208 static int confirm_zone(const char *filename);
209 static int continent_country_menu(dialogMenuItem *);
210 static int set_zone_multi(dialogMenuItem *);
211 static int set_zone_whole_country(dialogMenuItem *);
212 static int set_zone_menu(dialogMenuItem *);
213 static int set_zone_utc(void);
214
215 struct continent {
216 dialogMenuItem *menu;
217 int nitems;
218 };
219
220 static struct continent africa, america, antarctica, asia, atlantic;
221 static struct continent australia, europe, indian, pacific, utc;
222
223 static struct continent_names {
224 const char *name;
225 struct continent *continent;
226 } continent_names[] = {
227 { "Africa", &africa },
228 { "America", &america },
229 { "Antarctica", &antarctica },
230 { "Asia", &asia },
231 { "Atlantic", &atlantic },
232 { "Australia", &australia },
233 { "Europe", &europe },
234 { "Indian", &indian },
235 { "Pacific", &pacific },
236 { "UTC", &utc }
237 };
238
239 static struct continent_items {
240 char prompt[2];
241 char title[30];
242 } continent_items[] = {
243 { "1", "Africa" },
244 { "2", "America -- North and South" },
245 { "3", "Antarctica" },
246 { "4", "Asia" },
247 { "5", "Atlantic Ocean" },
248 { "6", "Australia" },
249 { "7", "Europe" },
250 { "8", "Indian Ocean" },
251 { "9", "Pacific Ocean" },
252 { "0", "UTC" }
253 };
254
255 #define NCONTINENTS \
256 (int)((sizeof(continent_items)) / (sizeof(continent_items[0])))
257 static dialogMenuItem continents[NCONTINENTS];
258
259 #define OCEANP(x) ((x) == 4 || (x) == 7 || (x) == 8)
260
261 static int
continent_country_menu(dialogMenuItem * continent)262 continent_country_menu(dialogMenuItem *continent)
263 {
264 char title[64], prompt[64];
265 struct continent *contp = continent->data;
266 int isocean = OCEANP(continent - continents);
267 int menulen;
268 int rv;
269
270 if (strcmp(continent->title, "UTC") == 0)
271 return (set_zone_utc());
272
273 /* Short cut -- if there's only one country, don't post a menu. */
274 if (contp->nitems == 1)
275 return (contp->menu[0].fire(&contp->menu[0]));
276
277 /* It's amazing how much good grammar really matters... */
278 if (!isocean) {
279 snprintf(title, sizeof(title), "Countries in %s",
280 continent->title);
281 snprintf(prompt, sizeof(prompt), "Select a country or region");
282 } else {
283 snprintf(title, sizeof(title), "Islands and groups in the %s",
284 continent->title);
285 snprintf(prompt, sizeof(prompt), "Select an island or group");
286 }
287
288 menulen = contp->nitems < 16 ? contp->nitems : 16;
289 rv = xdialog_menu(title, prompt, -1, -1, menulen, contp->nitems,
290 contp->menu);
291 if (rv == 0)
292 return (DITEM_LEAVE_MENU);
293 return (DITEM_RECREATE);
294 }
295
296 static struct continent *
find_continent(const char * name)297 find_continent(const char *name)
298 {
299 int i;
300
301 for (i = 0; i < NCONTINENTS; i++)
302 if (strcmp(name, continent_names[i].name) == 0)
303 return (continent_names[i].continent);
304 return (0);
305 }
306
307 struct country {
308 char *name;
309 char *tlc;
310 int nzones;
311 char *filename; /* use iff nzones < 0 */
312 struct continent *continent; /* use iff nzones < 0 */
313 TAILQ_HEAD(, zone) zones; /* use iff nzones > 0 */
314 dialogMenuItem *submenu; /* use iff nzones > 0 */
315 };
316
317 struct zone {
318 TAILQ_ENTRY(zone) link;
319 char *descr;
320 char *filename;
321 struct continent *continent;
322 };
323
324 /*
325 * This is the easiest organization... we use ISO 3166 country codes,
326 * of the two-letter variety, so we just size this array to suit.
327 * Beats worrying about dynamic allocation.
328 */
329 #define NCOUNTRIES (26 * 26)
330 static struct country countries[NCOUNTRIES];
331
332 #define CODE2INT(s) ((s[0] - 'A') * 26 + (s[1] - 'A'))
333
334 /*
335 * Read the ISO 3166 country code database in _PATH_ISO3166
336 * (/usr/share/misc/iso3166). On error, exit via err(3).
337 */
338 static void
read_iso3166_table(void)339 read_iso3166_table(void)
340 {
341 FILE *fp;
342 struct country *cp;
343 size_t len;
344 char *s, *t, *name;
345 int lineno;
346
347 fp = fopen(path_iso3166, "r");
348 if (!fp)
349 err(1, "%s", path_iso3166);
350 lineno = 0;
351
352 while ((s = fgetln(fp, &len)) != NULL) {
353 lineno++;
354 if (s[len - 1] != '\n')
355 errx(1, "%s:%d: invalid format", path_iso3166, lineno);
356 s[len - 1] = '\0';
357 if (s[0] == '#' || strspn(s, " \t") == len - 1)
358 continue;
359
360 /* Isolate the two-letter code. */
361 t = strsep(&s, "\t");
362 if (t == NULL || strlen(t) != 2)
363 errx(1, "%s:%d: invalid format", path_iso3166, lineno);
364 if (t[0] < 'A' || t[0] > 'Z' || t[1] < 'A' || t[1] > 'Z')
365 errx(1, "%s:%d: invalid code `%s'", path_iso3166,
366 lineno, t);
367
368 /* Now skip past the three-letter and numeric codes. */
369 name = strsep(&s, "\t"); /* 3-let */
370 if (name == NULL || strlen(name) != 3)
371 errx(1, "%s:%d: invalid format", path_iso3166, lineno);
372 name = strsep(&s, "\t"); /* numeric */
373 if (name == NULL || strlen(name) != 3)
374 errx(1, "%s:%d: invalid format", path_iso3166, lineno);
375
376 name = s;
377
378 cp = &countries[CODE2INT(t)];
379 if (cp->name)
380 errx(1, "%s:%d: country code `%s' multiply defined: %s",
381 path_iso3166, lineno, t, cp->name);
382 cp->name = strdup(name);
383 if (cp->name == NULL)
384 errx(1, "malloc failed");
385 cp->tlc = strdup(t);
386 if (cp->tlc == NULL)
387 errx(1, "malloc failed");
388 }
389
390 fclose(fp);
391 }
392
393 static void
add_zone_to_country(int lineno,const char * tlc,const char * descr,const char * file,struct continent * cont)394 add_zone_to_country(int lineno, const char *tlc, const char *descr,
395 const char *file, struct continent *cont)
396 {
397 struct zone *zp;
398 struct country *cp;
399
400 if (tlc[0] < 'A' || tlc[0] > 'Z' || tlc[1] < 'A' || tlc[1] > 'Z')
401 errx(1, "%s:%d: country code `%s' invalid", path_zonetab,
402 lineno, tlc);
403
404 cp = &countries[CODE2INT(tlc)];
405 if (cp->name == 0)
406 errx(1, "%s:%d: country code `%s' unknown", path_zonetab,
407 lineno, tlc);
408
409 if (descr) {
410 if (cp->nzones < 0)
411 errx(1, "%s:%d: conflicting zone definition",
412 path_zonetab, lineno);
413
414 zp = malloc(sizeof(*zp));
415 if (zp == NULL)
416 errx(1, "malloc(%zu)", sizeof(*zp));
417
418 if (cp->nzones == 0)
419 TAILQ_INIT(&cp->zones);
420
421 zp->descr = strdup(descr);
422 if (zp->descr == NULL)
423 errx(1, "malloc failed");
424 zp->filename = strdup(file);
425 if (zp->filename == NULL)
426 errx(1, "malloc failed");
427 zp->continent = cont;
428 TAILQ_INSERT_TAIL(&cp->zones, zp, link);
429 cp->nzones++;
430 } else {
431 if (cp->nzones > 0)
432 errx(1, "%s:%d: zone must have description",
433 path_zonetab, lineno);
434 if (cp->nzones < 0)
435 errx(1, "%s:%d: zone multiply defined",
436 path_zonetab, lineno);
437 cp->nzones = -1;
438 cp->filename = strdup(file);
439 if (cp->filename == NULL)
440 errx(1, "malloc failed");
441 cp->continent = cont;
442 }
443 }
444
445 /*
446 * This comparison function intentionally sorts all of the null-named
447 * ``countries''---i.e., the codes that don't correspond to a real
448 * country---to the end. Everything else is lexical by country name.
449 */
450 static int
compare_countries(const void * xa,const void * xb)451 compare_countries(const void *xa, const void *xb)
452 {
453 const struct country *a = xa, *b = xb;
454
455 if (a->name == 0 && b->name == 0)
456 return (0);
457 if (a->name == 0 && b->name != 0)
458 return (1);
459 if (b->name == 0)
460 return (-1);
461
462 return (strcmp(a->name, b->name));
463 }
464
465 /*
466 * This must be done AFTER all zone descriptions are read, since it breaks
467 * CODE2INT().
468 */
469 static void
sort_countries(void)470 sort_countries(void)
471 {
472
473 qsort(countries, NCOUNTRIES, sizeof(countries[0]), compare_countries);
474 }
475
476 static void
read_zones(void)477 read_zones(void)
478 {
479 char contbuf[16];
480 FILE *fp;
481 struct continent *cont;
482 size_t len, contlen;
483 char *line, *country_list, *tlc, *file, *descr, *p;
484 int lineno;
485
486 fp = fopen(path_zonetab, "r");
487 if (!fp)
488 err(1, "%s", path_zonetab);
489 lineno = 0;
490
491 while ((line = fgetln(fp, &len)) != NULL) {
492 lineno++;
493 if (line[len - 1] != '\n')
494 errx(1, "%s:%d: invalid format", path_zonetab, lineno);
495 line[len - 1] = '\0';
496 if (line[0] == '#')
497 continue;
498
499 country_list = strsep(&line, "\t");
500 /* coord = */ strsep(&line, "\t"); /* Unused */
501 file = strsep(&line, "\t");
502 /* get continent portion from continent/country */
503 p = strchr(file, '/');
504 if (p == NULL)
505 errx(1, "%s:%d: invalid zone name `%s'", path_zonetab,
506 lineno, file);
507 contlen = p - file + 1; /* trailing nul */
508 if (contlen > sizeof(contbuf))
509 errx(1, "%s:%d: continent name in zone name `%s' too long",
510 path_zonetab, lineno, file);
511 strlcpy(contbuf, file, contlen);
512 cont = find_continent(contbuf);
513 if (!cont)
514 errx(1, "%s:%d: invalid region `%s'", path_zonetab,
515 lineno, contbuf);
516
517 descr = (line != NULL && *line != '\0') ? line : NULL;
518
519 while (country_list != NULL) {
520 tlc = strsep(&country_list, ",");
521 if (strlen(tlc) != 2)
522 errx(1, "%s:%d: invalid country code `%s'",
523 path_zonetab, lineno, tlc);
524 add_zone_to_country(lineno, tlc, descr, file, cont);
525 }
526 }
527 fclose(fp);
528 }
529
530 static void
make_menus(void)531 make_menus(void)
532 {
533 struct country *cp;
534 struct zone *zp, *zp2;
535 struct continent *cont;
536 dialogMenuItem *dmi;
537 int i;
538
539 /*
540 * First, count up all the countries in each continent/ocean.
541 * Be careful to count those countries which have multiple zones
542 * only once for each. NB: some countries are in multiple
543 * continents/oceans.
544 */
545 for (cp = countries; cp->name; cp++) {
546 if (cp->nzones == 0)
547 continue;
548 if (cp->nzones < 0) {
549 cp->continent->nitems++;
550 } else {
551 TAILQ_FOREACH(zp, &cp->zones, link) {
552 cont = zp->continent;
553 for (zp2 = TAILQ_FIRST(&cp->zones);
554 zp2->continent != cont;
555 zp2 = TAILQ_NEXT(zp2, link))
556 ;
557 if (zp2 == zp)
558 zp->continent->nitems++;
559 }
560 }
561 }
562
563 /*
564 * Now allocate memory for the country menus and initialize
565 * continent menus. We set nitems back to zero so that we can
566 * use it for counting again when we actually build the menus.
567 */
568 memset(continents, 0, sizeof(continents));
569 for (i = 0; i < NCONTINENTS; i++) {
570 continent_names[i].continent->menu =
571 malloc(sizeof(dialogMenuItem) *
572 continent_names[i].continent->nitems);
573 if (continent_names[i].continent->menu == NULL)
574 errx(1, "malloc for continent menu");
575 continent_names[i].continent->nitems = 0;
576 continents[i].prompt = continent_items[i].prompt;
577 continents[i].title = continent_items[i].title;
578 continents[i].fire = continent_country_menu;
579 continents[i].data = continent_names[i].continent;
580 }
581
582 /*
583 * Now that memory is allocated, create the menu items for
584 * each continent. For multiple-zone countries, also create
585 * the country's zone submenu.
586 */
587 for (cp = countries; cp->name; cp++) {
588 if (cp->nzones == 0)
589 continue;
590 if (cp->nzones < 0) {
591 dmi = &cp->continent->menu[cp->continent->nitems];
592 memset(dmi, 0, sizeof(*dmi));
593 asprintf(&dmi->prompt, "%d", ++cp->continent->nitems);
594 dmi->title = cp->name;
595 dmi->fire = set_zone_whole_country;
596 dmi->data = cp;
597 } else {
598 cp->submenu = malloc(cp->nzones * sizeof(*dmi));
599 if (cp->submenu == 0)
600 errx(1, "malloc for submenu");
601 cp->nzones = 0;
602 TAILQ_FOREACH(zp, &cp->zones, link) {
603 cont = zp->continent;
604 dmi = &cp->submenu[cp->nzones];
605 memset(dmi, 0, sizeof(*dmi));
606 asprintf(&dmi->prompt, "%d", ++cp->nzones);
607 dmi->title = zp->descr;
608 dmi->fire = set_zone_multi;
609 dmi->data = zp;
610
611 for (zp2 = TAILQ_FIRST(&cp->zones);
612 zp2->continent != cont;
613 zp2 = TAILQ_NEXT(zp2, link))
614 ;
615 if (zp2 != zp)
616 continue;
617
618 dmi = &cont->menu[cont->nitems];
619 memset(dmi, 0, sizeof(*dmi));
620 asprintf(&dmi->prompt, "%d", ++cont->nitems);
621 dmi->title = cp->name;
622 dmi->fire = set_zone_menu;
623 dmi->data = cp;
624 }
625 }
626 }
627 }
628
629 static int
set_zone_menu(dialogMenuItem * dmi)630 set_zone_menu(dialogMenuItem *dmi)
631 {
632 char title[64], prompt[64];
633 struct country *cp = dmi->data;
634 int menulen;
635 int rv;
636
637 snprintf(title, sizeof(title), "%s Time Zones", cp->name);
638 snprintf(prompt, sizeof(prompt),
639 "Select a zone which observes the same time as your locality.");
640 menulen = cp->nzones < 16 ? cp->nzones : 16;
641 rv = xdialog_menu(title, prompt, -1, -1, menulen, cp->nzones,
642 cp->submenu);
643 if (rv != 0)
644 return (DITEM_RECREATE);
645 return (DITEM_LEAVE_MENU);
646 }
647
648 static int
set_zone_utc(void)649 set_zone_utc(void)
650 {
651 if (!confirm_zone("UTC"))
652 return (DITEM_FAILURE | DITEM_RECREATE);
653
654 return (install_zoneinfo("UTC"));
655 }
656
657 static int
confirm_zone(const char * filename)658 confirm_zone(const char *filename)
659 {
660 char title[64], prompt[64];
661 time_t t = time(0);
662 struct tm *tm;
663 int rv;
664
665 setenv("TZ", filename, 1);
666 tzset();
667 tm = localtime(&t);
668
669 snprintf(title, sizeof(title), "Confirmation");
670 snprintf(prompt, sizeof(prompt),
671 "Does the abbreviation `%s' look reasonable?", tm->tm_zone);
672 rv = !dialog_yesno(title, prompt, 5, 72);
673 return (rv);
674 }
675
676 static int
set_zone_multi(dialogMenuItem * dmi)677 set_zone_multi(dialogMenuItem *dmi)
678 {
679 struct zone *zp = dmi->data;
680 int rv;
681
682 if (!confirm_zone(zp->filename))
683 return (DITEM_FAILURE | DITEM_RECREATE);
684
685 rv = install_zoneinfo(zp->filename);
686 return (rv);
687 }
688
689 static int
set_zone_whole_country(dialogMenuItem * dmi)690 set_zone_whole_country(dialogMenuItem *dmi)
691 {
692 struct country *cp = dmi->data;
693 int rv;
694
695 if (!confirm_zone(cp->filename))
696 return (DITEM_FAILURE | DITEM_RECREATE);
697
698 rv = install_zoneinfo(cp->filename);
699 return (rv);
700 }
701
702 #endif
703
704 static int
install_zoneinfo_file(const char * zoneinfo_file)705 install_zoneinfo_file(const char *zoneinfo_file)
706 {
707 char buf[1024];
708 char title[64], prompt[SILLY_BUFFER_SIZE];
709 struct stat sb;
710 ssize_t len;
711 int fd1, fd2, copymode;
712
713 if (lstat(path_localtime, &sb) < 0) {
714 /* Nothing there yet... */
715 copymode = 1;
716 } else if (S_ISLNK(sb.st_mode))
717 copymode = 0;
718 else
719 copymode = 1;
720
721 #ifdef VERBOSE
722 snprintf(title, sizeof(title), "Info");
723 if (copymode)
724 snprintf(prompt, sizeof(prompt),
725 "Copying %s to %s", zoneinfo_file, path_localtime);
726 else
727 snprintf(prompt, sizeof(prompt),
728 "Creating symbolic link %s to %s",
729 path_localtime, zoneinfo_file);
730 #ifdef HAVE_DIALOG
731 if (usedialog)
732 dialog_msgbox(title, prompt, 8, 72, 1);
733 else
734 #endif
735 fprintf(stderr, "%s\n", prompt);
736 #endif
737
738 if (reallydoit) {
739 if (copymode) {
740 fd1 = open(zoneinfo_file, O_RDONLY, 0);
741 if (fd1 < 0) {
742 snprintf(title, sizeof(title), "Error");
743 snprintf(prompt, sizeof(prompt),
744 "Could not open %s: %s", zoneinfo_file,
745 strerror(errno));
746 #ifdef HAVE_DIALOG
747 if (usedialog)
748 dialog_msgbox(title, prompt, 8, 72, 1);
749 else
750 #endif
751 fprintf(stderr, "%s\n", prompt);
752 return (DITEM_FAILURE | DITEM_RECREATE);
753 }
754
755 if (unlink(path_localtime) < 0 && errno != ENOENT) {
756 snprintf(prompt, sizeof(prompt),
757 "Could not delete %s: %s",
758 path_localtime, strerror(errno));
759 #ifdef HAVE_DIALOG
760 if (usedialog) {
761 snprintf(title, sizeof(title), "Error");
762 dialog_msgbox(title, prompt, 8, 72, 1);
763 } else
764 #endif
765 fprintf(stderr, "%s\n", prompt);
766 return (DITEM_FAILURE | DITEM_RECREATE);
767 }
768
769 fd2 = open(path_localtime, O_CREAT | O_EXCL | O_WRONLY,
770 S_IRUSR | S_IRGRP | S_IROTH);
771 if (fd2 < 0) {
772 snprintf(title, sizeof(title), "Error");
773 snprintf(prompt, sizeof(prompt),
774 "Could not open %s: %s",
775 path_localtime, strerror(errno));
776 #ifdef HAVE_DIALOG
777 if (usedialog)
778 dialog_msgbox(title, prompt, 8, 72, 1);
779 else
780 #endif
781 fprintf(stderr, "%s\n", prompt);
782 return (DITEM_FAILURE | DITEM_RECREATE);
783 }
784
785 while ((len = read(fd1, buf, sizeof(buf))) > 0)
786 if ((len = write(fd2, buf, len)) < 0)
787 break;
788
789 if (len == -1) {
790 snprintf(title, sizeof(title), "Error");
791 snprintf(prompt, sizeof(prompt),
792 "Error copying %s to %s %s", zoneinfo_file,
793 path_localtime, strerror(errno));
794 #ifdef HAVE_DIALOG
795 if (usedialog)
796 dialog_msgbox(title, prompt, 8, 72, 1);
797 else
798 #endif
799 fprintf(stderr, "%s\n", prompt);
800 /* Better to leave none than a corrupt one. */
801 unlink(path_localtime);
802 return (DITEM_FAILURE | DITEM_RECREATE);
803 }
804 close(fd1);
805 close(fd2);
806 } else {
807 if (access(zoneinfo_file, R_OK) != 0) {
808 snprintf(title, sizeof(title), "Error");
809 snprintf(prompt, sizeof(prompt),
810 "Cannot access %s: %s", zoneinfo_file,
811 strerror(errno));
812 #ifdef HAVE_DIALOG
813 if (usedialog)
814 dialog_msgbox(title, prompt, 8, 72, 1);
815 else
816 #endif
817 fprintf(stderr, "%s\n", prompt);
818 return (DITEM_FAILURE | DITEM_RECREATE);
819 }
820 if (unlink(path_localtime) < 0 && errno != ENOENT) {
821 snprintf(prompt, sizeof(prompt),
822 "Could not delete %s: %s",
823 path_localtime, strerror(errno));
824 #ifdef HAVE_DIALOG
825 if (usedialog) {
826 snprintf(title, sizeof(title), "Error");
827 dialog_msgbox(title, prompt, 8, 72, 1);
828 } else
829 #endif
830 fprintf(stderr, "%s\n", prompt);
831 return (DITEM_FAILURE | DITEM_RECREATE);
832 }
833 if (symlink(zoneinfo_file, path_localtime) < 0) {
834 snprintf(title, sizeof(title), "Error");
835 snprintf(prompt, sizeof(prompt),
836 "Cannot create symbolic link %s to %s: %s",
837 path_localtime, zoneinfo_file,
838 strerror(errno));
839 #ifdef HAVE_DIALOG
840 if (usedialog)
841 dialog_msgbox(title, prompt, 8, 72, 1);
842 else
843 #endif
844 fprintf(stderr, "%s\n", prompt);
845 return (DITEM_FAILURE | DITEM_RECREATE);
846 }
847 }
848
849 #ifdef VERBOSE
850 snprintf(title, sizeof(title), "Done");
851 if (copymode)
852 snprintf(prompt, sizeof(prompt),
853 "Copied timezone file from %s to %s",
854 zoneinfo_file, path_localtime);
855 else
856 snprintf(prompt, sizeof(prompt),
857 "Created symbolic link from %s to %s",
858 zoneinfo_file, path_localtime);
859 #ifdef HAVE_DIALOG
860 if (usedialog)
861 dialog_msgbox(title, prompt, 8, 72, 1);
862 else
863 #endif
864 fprintf(stderr, "%s\n", prompt);
865 #endif
866 } /* reallydoit */
867
868 return (DITEM_LEAVE_MENU);
869 }
870
871 static int
install_zoneinfo(const char * zoneinfo)872 install_zoneinfo(const char *zoneinfo)
873 {
874 int rv;
875 FILE *f;
876 char path_zoneinfo_file[MAXPATHLEN];
877
878 if ((size_t)snprintf(path_zoneinfo_file, sizeof(path_zoneinfo_file),
879 "%s/%s", path_zoneinfo, zoneinfo) >= sizeof(path_zoneinfo_file))
880 errx(1, "%s/%s name too long", path_zoneinfo, zoneinfo);
881 rv = install_zoneinfo_file(path_zoneinfo_file);
882
883 /* Save knowledge for later */
884 if (reallydoit && (rv & DITEM_FAILURE) == 0) {
885 if ((f = fopen(path_db, "w")) != NULL) {
886 fprintf(f, "%s\n", zoneinfo);
887 fclose(f);
888 }
889 }
890
891 return (rv);
892 }
893
894 static void
usage(void)895 usage(void)
896 {
897
898 fprintf(stderr, "usage: tzsetup [-nrs] [-C chroot_directory]"
899 " [zoneinfo_file | zoneinfo_name]\n");
900 exit(1);
901 }
902
903 int
main(int argc,char ** argv)904 main(int argc, char **argv)
905 {
906 #ifdef HAVE_DIALOG
907 char title[64], prompt[128];
908 int fd;
909 #endif
910 int c, rv, skiputc;
911 char vm_guest[16] = "";
912 size_t len = sizeof(vm_guest);
913
914 skiputc = 0;
915
916 /* Default skiputc to 1 for VM guests */
917 if (sysctlbyname("kern.vm_guest", vm_guest, &len, NULL, 0) == 0 &&
918 strcmp(vm_guest, "none") != 0)
919 skiputc = 1;
920
921 while ((c = getopt(argc, argv, "C:nrs")) != -1) {
922 switch(c) {
923 case 'C':
924 chrootenv = optarg;
925 break;
926 case 'n':
927 reallydoit = 0;
928 break;
929 case 'r':
930 reinstall = 1;
931 #ifdef HAVE_DIALOG
932 usedialog = 0;
933 #endif
934 break;
935 case 's':
936 skiputc = 1;
937 break;
938 default:
939 usage();
940 }
941 }
942
943 if (argc - optind > 1)
944 usage();
945
946 if (chrootenv == NULL) {
947 strcpy(path_zonetab, _PATH_ZONETAB);
948 strcpy(path_iso3166, _PATH_ISO3166);
949 strcpy(path_zoneinfo, _PATH_ZONEINFO);
950 strcpy(path_localtime, _PATH_LOCALTIME);
951 strcpy(path_db, _PATH_DB);
952 strcpy(path_wall_cmos_clock, _PATH_WALL_CMOS_CLOCK);
953 } else {
954 sprintf(path_zonetab, "%s/%s", chrootenv, _PATH_ZONETAB);
955 sprintf(path_iso3166, "%s/%s", chrootenv, _PATH_ISO3166);
956 sprintf(path_zoneinfo, "%s/%s", chrootenv, _PATH_ZONEINFO);
957 sprintf(path_localtime, "%s/%s", chrootenv, _PATH_LOCALTIME);
958 sprintf(path_db, "%s/%s", chrootenv, _PATH_DB);
959 sprintf(path_wall_cmos_clock, "%s/%s", chrootenv,
960 _PATH_WALL_CMOS_CLOCK);
961 }
962
963 /* Override the user-supplied umask. */
964 (void)umask(S_IWGRP | S_IWOTH);
965
966 if (reinstall == 1) {
967 FILE *f;
968 char zoneinfo[MAXPATHLEN];
969
970 if ((f = fopen(path_db, "r")) != NULL) {
971 if (fgets(zoneinfo, sizeof(zoneinfo), f) != NULL) {
972 zoneinfo[sizeof(zoneinfo) - 1] = 0;
973 if (strlen(zoneinfo) > 0) {
974 zoneinfo[strlen(zoneinfo) - 1] = 0;
975 rv = install_zoneinfo(zoneinfo);
976 exit(rv & ~DITEM_LEAVE_MENU);
977 }
978 errx(1, "Error reading %s.\n", path_db);
979 }
980 fclose(f);
981 errx(1,
982 "Unable to determine earlier installed zoneinfo "
983 "name. Check %s", path_db);
984 }
985 errx(1, "Cannot open %s for reading. Does it exist?", path_db);
986 }
987
988 /*
989 * If the arguments on the command-line do not specify a file,
990 * then interpret it as a zoneinfo name
991 */
992 if (optind == argc - 1) {
993 struct stat sb;
994
995 if (stat(argv[optind], &sb) != 0) {
996 #ifdef HAVE_DIALOG
997 usedialog = 0;
998 #endif
999 rv = install_zoneinfo(argv[optind]);
1000 exit(rv & ~DITEM_LEAVE_MENU);
1001 }
1002 /* FALLTHROUGH */
1003 }
1004 #ifdef HAVE_DIALOG
1005
1006 read_iso3166_table();
1007 read_zones();
1008 sort_countries();
1009 make_menus();
1010
1011 init_dialog(stdin, stdout);
1012 if (skiputc == 0) {
1013 DIALOG_VARS save_vars;
1014 int yesno;
1015
1016 snprintf(title, sizeof(title),
1017 "Select local or UTC (Greenwich Mean Time) clock");
1018 snprintf(prompt, sizeof(prompt),
1019 "Is this machine's CMOS clock set to UTC? "
1020 "If it is set to local time,\n"
1021 "or you don't know, please choose NO here!");
1022 dlg_save_vars(&save_vars);
1023 dialog_vars.defaultno = TRUE;
1024 yesno = dialog_yesno(title, prompt, 7, 73);
1025 dlg_restore_vars(&save_vars);
1026 if (!yesno) {
1027 if (reallydoit)
1028 unlink(path_wall_cmos_clock);
1029 } else {
1030 if (reallydoit) {
1031 fd = open(path_wall_cmos_clock,
1032 O_WRONLY | O_CREAT | O_TRUNC,
1033 S_IRUSR | S_IRGRP | S_IROTH);
1034 if (fd < 0) {
1035 end_dialog();
1036 err(1, "create %s",
1037 path_wall_cmos_clock);
1038 }
1039 close(fd);
1040 }
1041 }
1042 dlg_clear();
1043 }
1044 if (optind == argc - 1) {
1045 snprintf(title, sizeof(title), "Default timezone provided");
1046 snprintf(prompt, sizeof(prompt),
1047 "\nUse the default `%s' zone?", argv[optind]);
1048 if (!dialog_yesno(title, prompt, 7, 72)) {
1049 rv = install_zoneinfo_file(argv[optind]);
1050 dlg_clear();
1051 end_dialog();
1052 exit(rv & ~DITEM_LEAVE_MENU);
1053 }
1054 dlg_clear();
1055 }
1056 snprintf(title, sizeof(title), "Time Zone Selector");
1057 snprintf(prompt, sizeof(prompt), "Select a region");
1058 xdialog_menu(title, prompt, -1, -1, NCONTINENTS, NCONTINENTS,
1059 continents);
1060
1061 dlg_clear();
1062 end_dialog();
1063 #else
1064 usage();
1065 #endif
1066 return (0);
1067 }
1068