38#endif /* not lint */
39
40#include <sys/types.h>
41#include <dialog.h>
42#include <err.h>
43#include <errno.h>
44#include <stdio.h>
45#include <stdlib.h>
46#include <string.h>
47#include <unistd.h>
48
49#include <sys/fcntl.h>
50#include <sys/queue.h>
51#include <sys/stat.h>
52
53#include "paths.h"
54
55static int reallydoit = 1;
56
57static int continent_country_menu(dialogMenuItem *);
58static int set_zone_multi(dialogMenuItem *);
59static int set_zone_whole_country(dialogMenuItem *);
60static int set_zone_menu(dialogMenuItem *);
61
62struct continent {
63 dialogMenuItem *menu;
64 int nitems;
65 int ch;
66 int sc;
67};
68
69static struct continent africa, america, antarctica, arctic, asia, atlantic;
70static struct continent australia, europe, indian, pacific;
71
72static struct continent_names {
73 char *name;
74 struct continent *continent;
75} continent_names[] = {
76 { "Africa", &africa }, { "America", &america },
77 { "Antarctica", &antarctica }, { "Arctic", &arctic },
78 { "Asia", &asia },
79 { "Atlantic", &atlantic }, { "Australia", &australia },
80 { "Europe", &europe }, { "Indian", &indian }, { "Pacific", &pacific }
81};
82
83static dialogMenuItem continents[] = {
84 { "1", "Africa", 0, continent_country_menu, 0, &africa },
85 { "2", "America -- North and South", 0, continent_country_menu, 0,
86 &america },
87 { "3", "Antarctica", 0, continent_country_menu, 0, &antarctica },
88 { "4", "Arctic Ocean", 0, continent_country_menu, 0, &arctic },
89 { "5", "Asia", 0, continent_country_menu, 0, &asia },
90 { "6", "Atlantic Ocean", 0, continent_country_menu, 0, &atlantic },
91 { "7", "Australia", 0, continent_country_menu, 0, &australia },
92 { "8", "Europe", 0, continent_country_menu, 0, &europe },
93 { "9", "Indian Ocean", 0, continent_country_menu, 0, &indian },
94 { "0", "Pacific Ocean", 0, continent_country_menu, 0, &pacific }
95};
96#define NCONTINENTS ((sizeof continents)/(sizeof continents[0]))
97#define OCEANP(x) ((x) == 3 || (x) == 5 || (x) == 8 || (x) == 9)
98
99static int
100continent_country_menu(dialogMenuItem *continent)
101{
102 int rv;
103 struct continent *contp = continent->data;
104 char title[256];
105 int isocean = OCEANP(continent - continents);
106 int menulen;
107
108 /* Short cut -- if there's only one country, don't post a menu. */
109 if (contp->nitems == 1) {
110 return set_zone_menu(&contp->menu[0]);
111 }
112
113 /* It's amazing how much good grammar really matters... */
114 if (!isocean)
115 snprintf(title, sizeof title, "Countries in %s",
116 continent->title);
117 else
118 snprintf(title, sizeof title, "Islands and groups in the %s",
119 continent->title);
120
121 menulen = contp->nitems < 16 ? contp->nitems : 16;
122 rv = dialog_menu(title, (isocean ? "Select an island or group"
123 : "Select a country"), -1, -1, menulen,
124 -contp->nitems, contp->menu, 0, &contp->ch,
125 &contp->sc);
126 if (rv == 0)
127 return DITEM_LEAVE_MENU;
128 return DITEM_RECREATE;
129}
130
131static struct continent *
132find_continent(const char *name)
133{
134 int i;
135
136 for (i = 0; i < NCONTINENTS; i++) {
137 if (strcmp(name, continent_names[i].name) == 0)
138 return continent_names[i].continent;
139 }
140 return 0;
141}
142
143struct country {
144 char *name;
145 char *tlc;
146 int nzones;
147 char *filename; /* use iff nzones < 0 */
148 struct continent *continent; /* use iff nzones < 0 */
149 TAILQ_HEAD(, zone) zones; /* use iff nzones > 0 */
150 dialogMenuItem *submenu; /* use iff nzones > 0 */
151};
152
153struct zone {
154 TAILQ_ENTRY(zone) link;
155 char *descr;
156 char *filename;
157 struct continent *continent;
158};
159
160/*
161 * This is the easiest organization... we use ISO 3166 country codes,
162 * of the two-letter variety, so we just size this array to suit.
163 * Beats worrying about dynamic allocation.
164 */
165#define NCOUNTRIES (26*26)
166static struct country countries[NCOUNTRIES];
167#define CODE2INT(s) ((s[0] - 'A') * 26 + (s[1] - 'A'))
168
169/*
170 * Read the ISO 3166 country code database in _PATH_ISO3166
171 * (/usr/share/misc/iso3166). On error, exit via err(3).
172 */
173static void
174read_iso3166_table(void)
175{
176 FILE *fp;
177 char *s, *t, *name;
178 size_t len;
179 int lineno;
180 struct country *cp;
181
182 fp = fopen(_PATH_ISO3166, "r");
183 if (!fp)
184 err(1, _PATH_ISO3166);
185 lineno = 0;
186
187 while ((s = fgetln(fp, &len)) != 0) {
188 lineno++;
189 if (s[len - 1] != '\n')
190 errx(1, _PATH_ISO3166 ":%d: invalid format", lineno);
191 s[len - 1] = '\0';
192 if (s[0] == '#' || strspn(s, " \t") == len - 1)
193 continue;
194
195 /* Isolate the two-letter code. */
196 t = strsep(&s, "\t");
197 if (t == 0 || strlen(t) != 2)
198 errx(1, _PATH_ISO3166 ":%d: invalid format", lineno);
199 if (t[0] < 'A' || t[0] > 'Z' || t[1] < 'A' || t[1] > 'Z')
200 errx(1, _PATH_ISO3166 ":%d: invalid code `%s'",
201 lineno, t);
202
203 /* Now skip past the three-letter and numeric codes. */
204 name = strsep(&s, "\t"); /* 3-let */
205 if (name == 0 || strlen(name) != 3)
206 errx(1, _PATH_ISO3166 ":%d: invalid format", lineno);
207 name = strsep(&s, "\t"); /* numeric */
208 if (name == 0 || strlen(name) != 3)
209 errx(1, _PATH_ISO3166 ":%d: invalid format", lineno);
210
211 name = s;
212
213 cp = &countries[CODE2INT(t)];
214 if (cp->name)
215 errx(1, _PATH_ISO3166
216 ":%d: country code `%s' multiply defined: %s",
217 lineno, t, cp->name);
218 cp->name = strdup(name);
219 cp->tlc = strdup(t);
220 }
221
222 fclose(fp);
223}
224
225static void
226add_zone_to_country(int lineno, const char *tlc, const char *descr,
227 const char *file, struct continent *cont)
228{
229 struct zone *zp;
230 struct country *cp;
231
232 if (tlc[0] < 'A' || tlc[0] > 'Z' || tlc[1] < 'A' || tlc[1] > 'Z')
233 errx(1, _PATH_ZONETAB ":%d: country code `%s' invalid",
234 lineno, tlc);
235
236 cp = &countries[CODE2INT(tlc)];
237 if (cp->name == 0)
238 errx(1, _PATH_ZONETAB ":%d: country code `%s' unknown",
239 lineno, tlc);
240
241 if (descr) {
242 if (cp->nzones < 0)
243 errx(1, _PATH_ZONETAB
244 ":%d: conflicting zone definition", lineno);
245
246 zp = malloc(sizeof *zp);
247 if (zp == 0)
248 err(1, "malloc(%lu)", (unsigned long)sizeof *zp);
249
250 if (cp->nzones == 0)
251 TAILQ_INIT(&cp->zones);
252
253 zp->descr = strdup(descr);
254 zp->filename = strdup(file);
255 zp->continent = cont;
256 TAILQ_INSERT_TAIL(&cp->zones, zp, link);
257 cp->nzones++;
258 } else {
259 if (cp->nzones > 0)
260 errx(1, _PATH_ZONETAB
261 ":%d: zone must have description", lineno);
262 if (cp->nzones < 0)
263 errx(1, _PATH_ZONETAB
264 ":%d: zone multiply defined", lineno);
265 cp->nzones = -1;
266 cp->filename = strdup(file);
267 cp->continent = cont;
268 }
269}
270
271/*
272 * This comparison function intentionally sorts all of the null-named
273 * ``countries''---i.e., the codes that don't correspond to a real
274 * country---to the end. Everything else is lexical by country name.
275 */
276static int
277compare_countries(const void *xa, const void *xb)
278{
279 const struct country *a = xa, *b = xb;
280
281 if (a->name == 0 && b->name == 0)
282 return 0;
283 if (a->name == 0 && b->name != 0)
284 return 1;
285 if (b->name == 0)
286 return -1;
287
288 return strcmp(a->name, b->name);
289}
290
291/*
292 * This must be done AFTER all zone descriptions are read, since it breaks
293 * CODE2INT().
294 */
295static void
296sort_countries(void)
297{
298 qsort(countries, NCOUNTRIES, sizeof countries[0], compare_countries);
299}
300
301static void
302read_zones(void)
303{
304 FILE *fp;
305 char *line;
306 size_t len;
307 int lineno;
308 char *tlc, *coord, *file, *descr, *p;
309 char contbuf[16];
310 struct continent *cont;
311
312 fp = fopen(_PATH_ZONETAB, "r");
313 if (!fp)
314 err(1, _PATH_ZONETAB);
315 lineno = 0;
316
317 while ((line = fgetln(fp, &len)) != 0) {
318 lineno++;
319 if (line[len - 1] != '\n')
320 errx(1, _PATH_ZONETAB ":%d: invalid format", lineno);
321 line[len - 1] = '\0';
322 if (line[0] == '#')
323 continue;
324
325 tlc = strsep(&line, "\t");
326 if (strlen(tlc) != 2)
327 errx(1, _PATH_ZONETAB ":%d: invalid country code `%s'",
328 lineno, tlc);
329 coord = strsep(&line, "\t");
330 file = strsep(&line, "\t");
331 p = strchr(file, '/');
332 if (p == 0)
333 errx(1, _PATH_ZONETAB ":%d: invalid zone name `%s'",
334 lineno, file);
335 contbuf[0] = '\0';
336 strncat(contbuf, file, p - file);
337 cont = find_continent(contbuf);
338 if (!cont)
339 errx(1, _PATH_ZONETAB ":%d: invalid region `%s'",
340 lineno, contbuf);
341
342 descr = (line && *line) ? line : 0;
343
344 add_zone_to_country(lineno, tlc, descr, file, cont);
345 }
346 fclose(fp);
347}
348
349static void
350make_menus(void)
351{
352 struct country *cp;
353 struct zone *zp, *zp2;
354 struct continent *cont;
355 dialogMenuItem *dmi;
356 int i;
357
358 /*
359 * First, count up all the countries in each continent/ocean.
360 * Be careful to count those countries which have multiple zones
361 * only once for each. NB: some countries are in multiple
362 * continents/oceans.
363 */
364 for (cp = countries; cp->name; cp++) {
365 if (cp->nzones == 0)
366 continue;
367 if (cp->nzones < 0) {
368 cp->continent->nitems++;
369 } else {
370 for (zp = cp->zones.tqh_first; zp;
371 zp = zp->link.tqe_next) {
372 cont = zp->continent;
373 for (zp2 = cp->zones.tqh_first;
374 zp2->continent != cont;
375 zp2 = zp2->link.tqe_next)
376 ;
377 if (zp2 == zp)
378 zp->continent->nitems++;
379 }
380 }
381 }
382
383 /*
384 * Now allocate memory for the country menus. We set
385 * nitems back to zero so that we can use it for counting
386 * again when we actually build the menus.
387 */
388 for (i = 0; i < NCONTINENTS; i++) {
389 continent_names[i].continent->menu =
390 malloc(sizeof(dialogMenuItem) *
391 continent_names[i].continent->nitems);
392 if (continent_names[i].continent->menu == 0)
393 err(1, "malloc for continent menu");
394 continent_names[i].continent->nitems = 0;
395 }
396
397 /*
398 * Now that memory is allocated, create the menu items for
399 * each continent. For multiple-zone countries, also create
400 * the country's zone submenu.
401 */
402 for (cp = countries; cp->name; cp++) {
403 if (cp->nzones == 0)
404 continue;
405 if (cp->nzones < 0) {
406 dmi = &cp->continent->menu[cp->continent->nitems];
407 memset(dmi, 0, sizeof *dmi);
408 asprintf(&dmi->prompt, "%d",
409 ++cp->continent->nitems);
410 dmi->title = cp->name;
411 dmi->checked = 0;
412 dmi->fire = set_zone_whole_country;
413 dmi->selected = 0;
414 dmi->data = cp;
415 } else {
416 cp->submenu = malloc(cp->nzones * sizeof *dmi);
417 if (cp->submenu == 0)
418 err(1, "malloc for submenu");
419 cp->nzones = 0;
420 for (zp = cp->zones.tqh_first; zp;
421 zp = zp->link.tqe_next) {
422 cont = zp->continent;
423 dmi = &cp->submenu[cp->nzones];
424 memset(dmi, 0, sizeof *dmi);
425 asprintf(&dmi->prompt, "%d",
426 ++cp->nzones);
427 dmi->title = zp->descr;
428 dmi->checked = 0;
429 dmi->fire = set_zone_multi;
430 dmi->selected = 0;
431 dmi->data = zp;
432
433 for (zp2 = cp->zones.tqh_first;
434 zp2->continent != cont;
435 zp2 = zp2->link.tqe_next)
436 ;
437 if (zp2 != zp)
438 continue;
439
440 dmi = &cont->menu[cont->nitems];
441 memset(dmi, 0, sizeof *dmi);
442 asprintf(&dmi->prompt, "%d", ++cont->nitems);
443 dmi->title = cp->name;
444 dmi->checked = 0;
445 dmi->fire = set_zone_menu;
446 dmi->selected = 0;
447 dmi->data = cp;
448 }
449 }
450 }
451}
452
453static int
454set_zone_menu(dialogMenuItem *dmi)
455{
456 int rv;
457 char buf[256];
458 struct country *cp = dmi->data;
459 int menulen;
460
461 snprintf(buf, sizeof buf, "%s Time Zones", cp->name);
462 menulen = cp->nzones < 16 ? cp->nzones : 16;
463 rv = dialog_menu(buf, "Select a zone which observes the same time as "
464 "your locality.", -1, -1, menulen, -cp->nzones,
465 cp->submenu, 0, 0, 0);
466 if (rv != 0)
467 return DITEM_RECREATE;
468 return DITEM_LEAVE_MENU;
469}
470
471static int
472install_zone_file(const char *filename)
473{
474 struct stat sb;
475 int fd1, fd2;
476 int copymode;
477 char *msg;
478 ssize_t len;
479 char buf[1024];
480
481 if (lstat(_PATH_LOCALTIME, &sb) < 0)
482 /* Nothing there yet... */
483 copymode = 1;
484 else if(S_ISLNK(sb.st_mode))
485 copymode = 0;
486 else
487 copymode = 1;
488
489#ifdef VERBOSE
490 if (copymode)
491 asprintf(&msg, "Copying %s to " _PATH_LOCALTIME, filename);
492 else
493 asprintf(&msg, "Creating symbolic link " _PATH_LOCALTIME
494 " to %s", filename);
495
496 dialog_notify(msg);
497 free(msg);
498#endif
499
500 if (reallydoit) {
501 if (copymode) {
502 fd1 = open(filename, O_RDONLY, 0);
503 if (fd1 < 0) {
504 asprintf(&msg, "Could not open %s: %s",
505 filename, strerror(errno));
506 dialog_mesgbox("Error", msg, 8, 72);
507 free(msg);
508 return DITEM_FAILURE | DITEM_RECREATE;
509 }
510
511 unlink(_PATH_LOCALTIME);
512 fd2 = open(_PATH_LOCALTIME,
513 O_CREAT | O_EXCL | O_WRONLY,
514 0444);
515 if (fd2 < 0) {
516 asprintf(&msg, "Could not open "
517 _PATH_LOCALTIME ": %s",
518 strerror(errno));
519 dialog_mesgbox("Error", msg, 8, 72);
520 free(msg);
521 return DITEM_FAILURE | DITEM_RECREATE;
522 }
523
524 while ((len = read(fd1, buf, sizeof buf)) > 0)
525 len = write(fd2, buf, len);
526
527 if (len == -1) {
528 asprintf(&msg, "Error copying %s to "
529 _PATH_LOCALTIME ": %s",
530 strerror(errno));
531 dialog_mesgbox("Error", msg, 8, 72);
532 free(msg);
533 /* Better to leave none than a corrupt one. */
534 unlink(_PATH_LOCALTIME);
535 return DITEM_FAILURE | DITEM_RECREATE;
536 }
537 close(fd1);
538 close(fd2);
539 } else {
540 if (access(filename, R_OK) != 0) {
541 asprintf(&msg, "Cannot access %s: %s",
542 filename, strerror(errno));
543 dialog_mesgbox("Error", msg, 8, 72);
544 free(msg);
545 return DITEM_FAILURE | DITEM_RECREATE;
546 }
547 unlink(_PATH_LOCALTIME);
548 if (symlink(filename, _PATH_LOCALTIME) < 0) {
549 asprintf(&msg, "Cannot create symbolic link "
550 _PATH_LOCALTIME " to %s: %s",
551 filename, strerror(errno));
552 dialog_mesgbox("Error", msg, 8, 72);
553 free(msg);
554 return DITEM_FAILURE | DITEM_RECREATE;
555 }
556 }
557 }
558
559#ifdef VERBOSE
560 if (copymode)
561 asprintf(&msg, "Copied timezone file from %s to "
562 _PATH_LOCALTIME, filename);
563 else
564 asprintf(&msg, "Created symbolic link from " _PATH_LOCALTIME
565 " to %s", filename);
566
567 dialog_mesgbox("Done", msg, 8, 72);
568 free(msg);
569#endif
570 return DITEM_LEAVE_MENU;
571}
572
573static int
574confirm_zone(const char *filename)
575{
576 char *msg;
577 struct tm *tm;
578 time_t t = time(0);
579 int rv;
580
581 setenv("TZ", filename, 1);
582 tzset();
583 tm = localtime(&t);
584
585 asprintf(&msg, "Does the abbreviation `%s' look reasonable?",
586 tm->tm_zone);
587 rv = !dialog_yesno("Confirmation", msg, 4, 72);
588 free(msg);
589 return rv;
590}
591
592static int
593set_zone_multi(dialogMenuItem *dmi)
594{
595 char *fn;
596 struct zone *zp = dmi->data;
597 int rv;
598
599 if (!confirm_zone(zp->filename))
600 return DITEM_FAILURE | DITEM_RECREATE;
601
602 asprintf(&fn, "%s/%s", _PATH_ZONEINFO, zp->filename);
603 rv = install_zone_file(fn);
604 free(fn);
605 return rv;
606}
607
608static int
609set_zone_whole_country(dialogMenuItem *dmi)
610{
611 char *fn;
612 struct country *cp = dmi->data;
613 int rv;
614
615 if (!confirm_zone(cp->filename))
616 return DITEM_FAILURE | DITEM_RECREATE;
617
618 asprintf(&fn, "%s/%s", _PATH_ZONEINFO, cp->filename);
619 rv = install_zone_file(fn);
620 free(fn);
621 return rv;
622}
623
624static void
625usage()
626{
627 fprintf(stderr, "usage: tzsetup [-n]\n");
628 exit(1);
629}
630
631int
632main(int argc, char **argv)
633{
634 int c, fd;
635
636 while ((c = getopt(argc, argv, "n")) != -1) {
637 switch(c) {
638 case 'n':
639 reallydoit = 0;
640 break;
641
642 default:
643 usage();
644 }
645 }
646
647 if (argc - optind > 1)
648 usage();
649
650 read_iso3166_table();
651 read_zones();
652 sort_countries();
653 make_menus();
654
655 init_dialog();
656 if (!dialog_yesno("Select local or UTC (Greenwich Mean Time) clock",
657 "Is this machine's CMOS clock set to UTC? If it is set to local time,\n"
| 38#endif /* not lint */
39
40#include <sys/types.h>
41#include <dialog.h>
42#include <err.h>
43#include <errno.h>
44#include <stdio.h>
45#include <stdlib.h>
46#include <string.h>
47#include <unistd.h>
48
49#include <sys/fcntl.h>
50#include <sys/queue.h>
51#include <sys/stat.h>
52
53#include "paths.h"
54
55static int reallydoit = 1;
56
57static int continent_country_menu(dialogMenuItem *);
58static int set_zone_multi(dialogMenuItem *);
59static int set_zone_whole_country(dialogMenuItem *);
60static int set_zone_menu(dialogMenuItem *);
61
62struct continent {
63 dialogMenuItem *menu;
64 int nitems;
65 int ch;
66 int sc;
67};
68
69static struct continent africa, america, antarctica, arctic, asia, atlantic;
70static struct continent australia, europe, indian, pacific;
71
72static struct continent_names {
73 char *name;
74 struct continent *continent;
75} continent_names[] = {
76 { "Africa", &africa }, { "America", &america },
77 { "Antarctica", &antarctica }, { "Arctic", &arctic },
78 { "Asia", &asia },
79 { "Atlantic", &atlantic }, { "Australia", &australia },
80 { "Europe", &europe }, { "Indian", &indian }, { "Pacific", &pacific }
81};
82
83static dialogMenuItem continents[] = {
84 { "1", "Africa", 0, continent_country_menu, 0, &africa },
85 { "2", "America -- North and South", 0, continent_country_menu, 0,
86 &america },
87 { "3", "Antarctica", 0, continent_country_menu, 0, &antarctica },
88 { "4", "Arctic Ocean", 0, continent_country_menu, 0, &arctic },
89 { "5", "Asia", 0, continent_country_menu, 0, &asia },
90 { "6", "Atlantic Ocean", 0, continent_country_menu, 0, &atlantic },
91 { "7", "Australia", 0, continent_country_menu, 0, &australia },
92 { "8", "Europe", 0, continent_country_menu, 0, &europe },
93 { "9", "Indian Ocean", 0, continent_country_menu, 0, &indian },
94 { "0", "Pacific Ocean", 0, continent_country_menu, 0, &pacific }
95};
96#define NCONTINENTS ((sizeof continents)/(sizeof continents[0]))
97#define OCEANP(x) ((x) == 3 || (x) == 5 || (x) == 8 || (x) == 9)
98
99static int
100continent_country_menu(dialogMenuItem *continent)
101{
102 int rv;
103 struct continent *contp = continent->data;
104 char title[256];
105 int isocean = OCEANP(continent - continents);
106 int menulen;
107
108 /* Short cut -- if there's only one country, don't post a menu. */
109 if (contp->nitems == 1) {
110 return set_zone_menu(&contp->menu[0]);
111 }
112
113 /* It's amazing how much good grammar really matters... */
114 if (!isocean)
115 snprintf(title, sizeof title, "Countries in %s",
116 continent->title);
117 else
118 snprintf(title, sizeof title, "Islands and groups in the %s",
119 continent->title);
120
121 menulen = contp->nitems < 16 ? contp->nitems : 16;
122 rv = dialog_menu(title, (isocean ? "Select an island or group"
123 : "Select a country"), -1, -1, menulen,
124 -contp->nitems, contp->menu, 0, &contp->ch,
125 &contp->sc);
126 if (rv == 0)
127 return DITEM_LEAVE_MENU;
128 return DITEM_RECREATE;
129}
130
131static struct continent *
132find_continent(const char *name)
133{
134 int i;
135
136 for (i = 0; i < NCONTINENTS; i++) {
137 if (strcmp(name, continent_names[i].name) == 0)
138 return continent_names[i].continent;
139 }
140 return 0;
141}
142
143struct country {
144 char *name;
145 char *tlc;
146 int nzones;
147 char *filename; /* use iff nzones < 0 */
148 struct continent *continent; /* use iff nzones < 0 */
149 TAILQ_HEAD(, zone) zones; /* use iff nzones > 0 */
150 dialogMenuItem *submenu; /* use iff nzones > 0 */
151};
152
153struct zone {
154 TAILQ_ENTRY(zone) link;
155 char *descr;
156 char *filename;
157 struct continent *continent;
158};
159
160/*
161 * This is the easiest organization... we use ISO 3166 country codes,
162 * of the two-letter variety, so we just size this array to suit.
163 * Beats worrying about dynamic allocation.
164 */
165#define NCOUNTRIES (26*26)
166static struct country countries[NCOUNTRIES];
167#define CODE2INT(s) ((s[0] - 'A') * 26 + (s[1] - 'A'))
168
169/*
170 * Read the ISO 3166 country code database in _PATH_ISO3166
171 * (/usr/share/misc/iso3166). On error, exit via err(3).
172 */
173static void
174read_iso3166_table(void)
175{
176 FILE *fp;
177 char *s, *t, *name;
178 size_t len;
179 int lineno;
180 struct country *cp;
181
182 fp = fopen(_PATH_ISO3166, "r");
183 if (!fp)
184 err(1, _PATH_ISO3166);
185 lineno = 0;
186
187 while ((s = fgetln(fp, &len)) != 0) {
188 lineno++;
189 if (s[len - 1] != '\n')
190 errx(1, _PATH_ISO3166 ":%d: invalid format", lineno);
191 s[len - 1] = '\0';
192 if (s[0] == '#' || strspn(s, " \t") == len - 1)
193 continue;
194
195 /* Isolate the two-letter code. */
196 t = strsep(&s, "\t");
197 if (t == 0 || strlen(t) != 2)
198 errx(1, _PATH_ISO3166 ":%d: invalid format", lineno);
199 if (t[0] < 'A' || t[0] > 'Z' || t[1] < 'A' || t[1] > 'Z')
200 errx(1, _PATH_ISO3166 ":%d: invalid code `%s'",
201 lineno, t);
202
203 /* Now skip past the three-letter and numeric codes. */
204 name = strsep(&s, "\t"); /* 3-let */
205 if (name == 0 || strlen(name) != 3)
206 errx(1, _PATH_ISO3166 ":%d: invalid format", lineno);
207 name = strsep(&s, "\t"); /* numeric */
208 if (name == 0 || strlen(name) != 3)
209 errx(1, _PATH_ISO3166 ":%d: invalid format", lineno);
210
211 name = s;
212
213 cp = &countries[CODE2INT(t)];
214 if (cp->name)
215 errx(1, _PATH_ISO3166
216 ":%d: country code `%s' multiply defined: %s",
217 lineno, t, cp->name);
218 cp->name = strdup(name);
219 cp->tlc = strdup(t);
220 }
221
222 fclose(fp);
223}
224
225static void
226add_zone_to_country(int lineno, const char *tlc, const char *descr,
227 const char *file, struct continent *cont)
228{
229 struct zone *zp;
230 struct country *cp;
231
232 if (tlc[0] < 'A' || tlc[0] > 'Z' || tlc[1] < 'A' || tlc[1] > 'Z')
233 errx(1, _PATH_ZONETAB ":%d: country code `%s' invalid",
234 lineno, tlc);
235
236 cp = &countries[CODE2INT(tlc)];
237 if (cp->name == 0)
238 errx(1, _PATH_ZONETAB ":%d: country code `%s' unknown",
239 lineno, tlc);
240
241 if (descr) {
242 if (cp->nzones < 0)
243 errx(1, _PATH_ZONETAB
244 ":%d: conflicting zone definition", lineno);
245
246 zp = malloc(sizeof *zp);
247 if (zp == 0)
248 err(1, "malloc(%lu)", (unsigned long)sizeof *zp);
249
250 if (cp->nzones == 0)
251 TAILQ_INIT(&cp->zones);
252
253 zp->descr = strdup(descr);
254 zp->filename = strdup(file);
255 zp->continent = cont;
256 TAILQ_INSERT_TAIL(&cp->zones, zp, link);
257 cp->nzones++;
258 } else {
259 if (cp->nzones > 0)
260 errx(1, _PATH_ZONETAB
261 ":%d: zone must have description", lineno);
262 if (cp->nzones < 0)
263 errx(1, _PATH_ZONETAB
264 ":%d: zone multiply defined", lineno);
265 cp->nzones = -1;
266 cp->filename = strdup(file);
267 cp->continent = cont;
268 }
269}
270
271/*
272 * This comparison function intentionally sorts all of the null-named
273 * ``countries''---i.e., the codes that don't correspond to a real
274 * country---to the end. Everything else is lexical by country name.
275 */
276static int
277compare_countries(const void *xa, const void *xb)
278{
279 const struct country *a = xa, *b = xb;
280
281 if (a->name == 0 && b->name == 0)
282 return 0;
283 if (a->name == 0 && b->name != 0)
284 return 1;
285 if (b->name == 0)
286 return -1;
287
288 return strcmp(a->name, b->name);
289}
290
291/*
292 * This must be done AFTER all zone descriptions are read, since it breaks
293 * CODE2INT().
294 */
295static void
296sort_countries(void)
297{
298 qsort(countries, NCOUNTRIES, sizeof countries[0], compare_countries);
299}
300
301static void
302read_zones(void)
303{
304 FILE *fp;
305 char *line;
306 size_t len;
307 int lineno;
308 char *tlc, *coord, *file, *descr, *p;
309 char contbuf[16];
310 struct continent *cont;
311
312 fp = fopen(_PATH_ZONETAB, "r");
313 if (!fp)
314 err(1, _PATH_ZONETAB);
315 lineno = 0;
316
317 while ((line = fgetln(fp, &len)) != 0) {
318 lineno++;
319 if (line[len - 1] != '\n')
320 errx(1, _PATH_ZONETAB ":%d: invalid format", lineno);
321 line[len - 1] = '\0';
322 if (line[0] == '#')
323 continue;
324
325 tlc = strsep(&line, "\t");
326 if (strlen(tlc) != 2)
327 errx(1, _PATH_ZONETAB ":%d: invalid country code `%s'",
328 lineno, tlc);
329 coord = strsep(&line, "\t");
330 file = strsep(&line, "\t");
331 p = strchr(file, '/');
332 if (p == 0)
333 errx(1, _PATH_ZONETAB ":%d: invalid zone name `%s'",
334 lineno, file);
335 contbuf[0] = '\0';
336 strncat(contbuf, file, p - file);
337 cont = find_continent(contbuf);
338 if (!cont)
339 errx(1, _PATH_ZONETAB ":%d: invalid region `%s'",
340 lineno, contbuf);
341
342 descr = (line && *line) ? line : 0;
343
344 add_zone_to_country(lineno, tlc, descr, file, cont);
345 }
346 fclose(fp);
347}
348
349static void
350make_menus(void)
351{
352 struct country *cp;
353 struct zone *zp, *zp2;
354 struct continent *cont;
355 dialogMenuItem *dmi;
356 int i;
357
358 /*
359 * First, count up all the countries in each continent/ocean.
360 * Be careful to count those countries which have multiple zones
361 * only once for each. NB: some countries are in multiple
362 * continents/oceans.
363 */
364 for (cp = countries; cp->name; cp++) {
365 if (cp->nzones == 0)
366 continue;
367 if (cp->nzones < 0) {
368 cp->continent->nitems++;
369 } else {
370 for (zp = cp->zones.tqh_first; zp;
371 zp = zp->link.tqe_next) {
372 cont = zp->continent;
373 for (zp2 = cp->zones.tqh_first;
374 zp2->continent != cont;
375 zp2 = zp2->link.tqe_next)
376 ;
377 if (zp2 == zp)
378 zp->continent->nitems++;
379 }
380 }
381 }
382
383 /*
384 * Now allocate memory for the country menus. We set
385 * nitems back to zero so that we can use it for counting
386 * again when we actually build the menus.
387 */
388 for (i = 0; i < NCONTINENTS; i++) {
389 continent_names[i].continent->menu =
390 malloc(sizeof(dialogMenuItem) *
391 continent_names[i].continent->nitems);
392 if (continent_names[i].continent->menu == 0)
393 err(1, "malloc for continent menu");
394 continent_names[i].continent->nitems = 0;
395 }
396
397 /*
398 * Now that memory is allocated, create the menu items for
399 * each continent. For multiple-zone countries, also create
400 * the country's zone submenu.
401 */
402 for (cp = countries; cp->name; cp++) {
403 if (cp->nzones == 0)
404 continue;
405 if (cp->nzones < 0) {
406 dmi = &cp->continent->menu[cp->continent->nitems];
407 memset(dmi, 0, sizeof *dmi);
408 asprintf(&dmi->prompt, "%d",
409 ++cp->continent->nitems);
410 dmi->title = cp->name;
411 dmi->checked = 0;
412 dmi->fire = set_zone_whole_country;
413 dmi->selected = 0;
414 dmi->data = cp;
415 } else {
416 cp->submenu = malloc(cp->nzones * sizeof *dmi);
417 if (cp->submenu == 0)
418 err(1, "malloc for submenu");
419 cp->nzones = 0;
420 for (zp = cp->zones.tqh_first; zp;
421 zp = zp->link.tqe_next) {
422 cont = zp->continent;
423 dmi = &cp->submenu[cp->nzones];
424 memset(dmi, 0, sizeof *dmi);
425 asprintf(&dmi->prompt, "%d",
426 ++cp->nzones);
427 dmi->title = zp->descr;
428 dmi->checked = 0;
429 dmi->fire = set_zone_multi;
430 dmi->selected = 0;
431 dmi->data = zp;
432
433 for (zp2 = cp->zones.tqh_first;
434 zp2->continent != cont;
435 zp2 = zp2->link.tqe_next)
436 ;
437 if (zp2 != zp)
438 continue;
439
440 dmi = &cont->menu[cont->nitems];
441 memset(dmi, 0, sizeof *dmi);
442 asprintf(&dmi->prompt, "%d", ++cont->nitems);
443 dmi->title = cp->name;
444 dmi->checked = 0;
445 dmi->fire = set_zone_menu;
446 dmi->selected = 0;
447 dmi->data = cp;
448 }
449 }
450 }
451}
452
453static int
454set_zone_menu(dialogMenuItem *dmi)
455{
456 int rv;
457 char buf[256];
458 struct country *cp = dmi->data;
459 int menulen;
460
461 snprintf(buf, sizeof buf, "%s Time Zones", cp->name);
462 menulen = cp->nzones < 16 ? cp->nzones : 16;
463 rv = dialog_menu(buf, "Select a zone which observes the same time as "
464 "your locality.", -1, -1, menulen, -cp->nzones,
465 cp->submenu, 0, 0, 0);
466 if (rv != 0)
467 return DITEM_RECREATE;
468 return DITEM_LEAVE_MENU;
469}
470
471static int
472install_zone_file(const char *filename)
473{
474 struct stat sb;
475 int fd1, fd2;
476 int copymode;
477 char *msg;
478 ssize_t len;
479 char buf[1024];
480
481 if (lstat(_PATH_LOCALTIME, &sb) < 0)
482 /* Nothing there yet... */
483 copymode = 1;
484 else if(S_ISLNK(sb.st_mode))
485 copymode = 0;
486 else
487 copymode = 1;
488
489#ifdef VERBOSE
490 if (copymode)
491 asprintf(&msg, "Copying %s to " _PATH_LOCALTIME, filename);
492 else
493 asprintf(&msg, "Creating symbolic link " _PATH_LOCALTIME
494 " to %s", filename);
495
496 dialog_notify(msg);
497 free(msg);
498#endif
499
500 if (reallydoit) {
501 if (copymode) {
502 fd1 = open(filename, O_RDONLY, 0);
503 if (fd1 < 0) {
504 asprintf(&msg, "Could not open %s: %s",
505 filename, strerror(errno));
506 dialog_mesgbox("Error", msg, 8, 72);
507 free(msg);
508 return DITEM_FAILURE | DITEM_RECREATE;
509 }
510
511 unlink(_PATH_LOCALTIME);
512 fd2 = open(_PATH_LOCALTIME,
513 O_CREAT | O_EXCL | O_WRONLY,
514 0444);
515 if (fd2 < 0) {
516 asprintf(&msg, "Could not open "
517 _PATH_LOCALTIME ": %s",
518 strerror(errno));
519 dialog_mesgbox("Error", msg, 8, 72);
520 free(msg);
521 return DITEM_FAILURE | DITEM_RECREATE;
522 }
523
524 while ((len = read(fd1, buf, sizeof buf)) > 0)
525 len = write(fd2, buf, len);
526
527 if (len == -1) {
528 asprintf(&msg, "Error copying %s to "
529 _PATH_LOCALTIME ": %s",
530 strerror(errno));
531 dialog_mesgbox("Error", msg, 8, 72);
532 free(msg);
533 /* Better to leave none than a corrupt one. */
534 unlink(_PATH_LOCALTIME);
535 return DITEM_FAILURE | DITEM_RECREATE;
536 }
537 close(fd1);
538 close(fd2);
539 } else {
540 if (access(filename, R_OK) != 0) {
541 asprintf(&msg, "Cannot access %s: %s",
542 filename, strerror(errno));
543 dialog_mesgbox("Error", msg, 8, 72);
544 free(msg);
545 return DITEM_FAILURE | DITEM_RECREATE;
546 }
547 unlink(_PATH_LOCALTIME);
548 if (symlink(filename, _PATH_LOCALTIME) < 0) {
549 asprintf(&msg, "Cannot create symbolic link "
550 _PATH_LOCALTIME " to %s: %s",
551 filename, strerror(errno));
552 dialog_mesgbox("Error", msg, 8, 72);
553 free(msg);
554 return DITEM_FAILURE | DITEM_RECREATE;
555 }
556 }
557 }
558
559#ifdef VERBOSE
560 if (copymode)
561 asprintf(&msg, "Copied timezone file from %s to "
562 _PATH_LOCALTIME, filename);
563 else
564 asprintf(&msg, "Created symbolic link from " _PATH_LOCALTIME
565 " to %s", filename);
566
567 dialog_mesgbox("Done", msg, 8, 72);
568 free(msg);
569#endif
570 return DITEM_LEAVE_MENU;
571}
572
573static int
574confirm_zone(const char *filename)
575{
576 char *msg;
577 struct tm *tm;
578 time_t t = time(0);
579 int rv;
580
581 setenv("TZ", filename, 1);
582 tzset();
583 tm = localtime(&t);
584
585 asprintf(&msg, "Does the abbreviation `%s' look reasonable?",
586 tm->tm_zone);
587 rv = !dialog_yesno("Confirmation", msg, 4, 72);
588 free(msg);
589 return rv;
590}
591
592static int
593set_zone_multi(dialogMenuItem *dmi)
594{
595 char *fn;
596 struct zone *zp = dmi->data;
597 int rv;
598
599 if (!confirm_zone(zp->filename))
600 return DITEM_FAILURE | DITEM_RECREATE;
601
602 asprintf(&fn, "%s/%s", _PATH_ZONEINFO, zp->filename);
603 rv = install_zone_file(fn);
604 free(fn);
605 return rv;
606}
607
608static int
609set_zone_whole_country(dialogMenuItem *dmi)
610{
611 char *fn;
612 struct country *cp = dmi->data;
613 int rv;
614
615 if (!confirm_zone(cp->filename))
616 return DITEM_FAILURE | DITEM_RECREATE;
617
618 asprintf(&fn, "%s/%s", _PATH_ZONEINFO, cp->filename);
619 rv = install_zone_file(fn);
620 free(fn);
621 return rv;
622}
623
624static void
625usage()
626{
627 fprintf(stderr, "usage: tzsetup [-n]\n");
628 exit(1);
629}
630
631int
632main(int argc, char **argv)
633{
634 int c, fd;
635
636 while ((c = getopt(argc, argv, "n")) != -1) {
637 switch(c) {
638 case 'n':
639 reallydoit = 0;
640 break;
641
642 default:
643 usage();
644 }
645 }
646
647 if (argc - optind > 1)
648 usage();
649
650 read_iso3166_table();
651 read_zones();
652 sort_countries();
653 make_menus();
654
655 init_dialog();
656 if (!dialog_yesno("Select local or UTC (Greenwich Mean Time) clock",
657 "Is this machine's CMOS clock set to UTC? If it is set to local time,\n"
|