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