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 <dialog.h> 39 #include <err.h> 40 #include <errno.h> 41 #include <stdio.h> 42 #include <stdlib.h> 43 #include <string.h> 44 #include <time.h> 45 #include <unistd.h> 46 47 #include <sys/fcntl.h> 48 #include <sys/param.h> 49 #include <sys/queue.h> 50 #include <sys/stat.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 static char path_zonetab[MAXPATHLEN], path_iso3166[MAXPATHLEN], 60 path_zoneinfo[MAXPATHLEN], path_localtime[MAXPATHLEN], 61 path_db[MAXPATHLEN], path_wall_cmos_clock[MAXPATHLEN]; 62 63 static int reallydoit = 1; 64 static int reinstall = 0; 65 static int usedialog = 1; 66 static char *chrootenv = NULL; 67 68 static void usage(void); 69 static int continent_country_menu(dialogMenuItem *); 70 static int set_zone_multi(dialogMenuItem *); 71 static int set_zone_whole_country(dialogMenuItem *); 72 static int set_zone_menu(dialogMenuItem *); 73 74 struct continent { 75 dialogMenuItem *menu; 76 int nitems; 77 int ch; 78 int sc; 79 }; 80 81 static struct continent africa, america, antarctica, arctic, asia, atlantic; 82 static struct continent australia, europe, indian, pacific; 83 84 static struct continent_names { 85 const char *name; 86 struct continent *continent; 87 } continent_names[] = { 88 { "Africa", &africa }, 89 { "America", &america }, 90 { "Antarctica", &antarctica }, 91 { "Arctic", &arctic }, 92 { "Asia", &asia }, 93 { "Atlantic", &atlantic }, 94 { "Australia", &australia }, 95 { "Europe", &europe }, 96 { "Indian", &indian }, 97 { "Pacific", &pacific } 98 }; 99 100 static struct continent_items { 101 char prompt[2]; 102 char title[30]; 103 } continent_items[] = { 104 { "1", "Africa" }, 105 { "2", "America -- North and South" }, 106 { "3", "Antarctica" }, 107 { "4", "Arctic Ocean" }, 108 { "5", "Asia" }, 109 { "6", "Atlantic Ocean" }, 110 { "7", "Australia" }, 111 { "8", "Europe" }, 112 { "9", "Indian Ocean" }, 113 { "0", "Pacific Ocean" } 114 }; 115 116 #define NCONTINENTS \ 117 (int)((sizeof(continent_items)) / (sizeof(continent_items[0]))) 118 static dialogMenuItem continents[NCONTINENTS]; 119 120 #define OCEANP(x) ((x) == 3 || (x) == 5 || (x) == 8 || (x) == 9) 121 122 static int 123 continent_country_menu(dialogMenuItem *continent) 124 { 125 char title[64], prompt[64]; 126 struct continent *contp = continent->data; 127 int isocean = OCEANP(continent - continents); 128 int menulen; 129 int rv; 130 131 /* Short cut -- if there's only one country, don't post a menu. */ 132 if (contp->nitems == 1) 133 return (contp->menu[0].fire(&contp->menu[0])); 134 135 /* It's amazing how much good grammar really matters... */ 136 if (!isocean) { 137 snprintf(title, sizeof(title), "Countries in %s", 138 continent->title); 139 snprintf(prompt, sizeof(prompt), "Select a country or region"); 140 } else { 141 snprintf(title, sizeof(title), "Islands and groups in the %s", 142 continent->title); 143 snprintf(prompt, sizeof(prompt), "Select an island or group"); 144 } 145 146 menulen = contp->nitems < 16 ? contp->nitems : 16; 147 rv = dialog_menu(title, prompt, -1, -1, menulen, -contp->nitems, 148 contp->menu, 0, &contp->ch, &contp->sc); 149 if (rv == 0) 150 return (DITEM_LEAVE_MENU); 151 return (DITEM_RECREATE); 152 } 153 154 static struct continent * 155 find_continent(const char *name) 156 { 157 int i; 158 159 for (i = 0; i < NCONTINENTS; i++) 160 if (strcmp(name, continent_names[i].name) == 0) 161 return (continent_names[i].continent); 162 return (0); 163 } 164 165 struct country { 166 char *name; 167 char *tlc; 168 int nzones; 169 char *filename; /* use iff nzones < 0 */ 170 struct continent *continent; /* use iff nzones < 0 */ 171 TAILQ_HEAD(, zone) zones; /* use iff nzones > 0 */ 172 dialogMenuItem *submenu; /* use iff nzones > 0 */ 173 }; 174 175 struct zone { 176 TAILQ_ENTRY(zone) link; 177 char *descr; 178 char *filename; 179 struct continent *continent; 180 }; 181 182 /* 183 * This is the easiest organization... we use ISO 3166 country codes, 184 * of the two-letter variety, so we just size this array to suit. 185 * Beats worrying about dynamic allocation. 186 */ 187 #define NCOUNTRIES (26 * 26) 188 static struct country countries[NCOUNTRIES]; 189 190 #define CODE2INT(s) ((s[0] - 'A') * 26 + (s[1] - 'A')) 191 192 /* 193 * Read the ISO 3166 country code database in _PATH_ISO3166 194 * (/usr/share/misc/iso3166). On error, exit via err(3). 195 */ 196 static void 197 read_iso3166_table(void) 198 { 199 FILE *fp; 200 struct country *cp; 201 size_t len; 202 char *s, *t, *name; 203 int lineno; 204 205 fp = fopen(path_iso3166, "r"); 206 if (!fp) 207 err(1, "%s", path_iso3166); 208 lineno = 0; 209 210 while ((s = fgetln(fp, &len)) != 0) { 211 lineno++; 212 if (s[len - 1] != '\n') 213 errx(1, "%s:%d: invalid format", path_iso3166, lineno); 214 s[len - 1] = '\0'; 215 if (s[0] == '#' || strspn(s, " \t") == len - 1) 216 continue; 217 218 /* Isolate the two-letter code. */ 219 t = strsep(&s, "\t"); 220 if (t == 0 || strlen(t) != 2) 221 errx(1, "%s:%d: invalid format", path_iso3166, lineno); 222 if (t[0] < 'A' || t[0] > 'Z' || t[1] < 'A' || t[1] > 'Z') 223 errx(1, "%s:%d: invalid code `%s'", path_iso3166, 224 lineno, t); 225 226 /* Now skip past the three-letter and numeric codes. */ 227 name = strsep(&s, "\t"); /* 3-let */ 228 if (name == 0 || strlen(name) != 3) 229 errx(1, "%s:%d: invalid format", path_iso3166, lineno); 230 name = strsep(&s, "\t"); /* numeric */ 231 if (name == 0 || strlen(name) != 3) 232 errx(1, "%s:%d: invalid format", path_iso3166, lineno); 233 234 name = s; 235 236 cp = &countries[CODE2INT(t)]; 237 if (cp->name) 238 errx(1, "%s:%d: country code `%s' multiply defined: %s", 239 path_iso3166, lineno, t, cp->name); 240 cp->name = strdup(name); 241 if (cp->name == NULL) 242 errx(1, "malloc failed"); 243 cp->tlc = strdup(t); 244 if (cp->tlc == NULL) 245 errx(1, "malloc failed"); 246 } 247 248 fclose(fp); 249 } 250 251 static void 252 add_zone_to_country(int lineno, const char *tlc, const char *descr, 253 const char *file, struct continent *cont) 254 { 255 struct zone *zp; 256 struct country *cp; 257 258 if (tlc[0] < 'A' || tlc[0] > 'Z' || tlc[1] < 'A' || tlc[1] > 'Z') 259 errx(1, "%s:%d: country code `%s' invalid", path_zonetab, 260 lineno, tlc); 261 262 cp = &countries[CODE2INT(tlc)]; 263 if (cp->name == 0) 264 errx(1, "%s:%d: country code `%s' unknown", path_zonetab, 265 lineno, tlc); 266 267 if (descr) { 268 if (cp->nzones < 0) 269 errx(1, "%s:%d: conflicting zone definition", 270 path_zonetab, lineno); 271 272 zp = malloc(sizeof(*zp)); 273 if (zp == 0) 274 errx(1, "malloc(%zu)", sizeof(*zp)); 275 276 if (cp->nzones == 0) 277 TAILQ_INIT(&cp->zones); 278 279 zp->descr = strdup(descr); 280 if (zp->descr == NULL) 281 errx(1, "malloc failed"); 282 zp->filename = strdup(file); 283 if (zp->filename == NULL) 284 errx(1, "malloc failed"); 285 zp->continent = cont; 286 TAILQ_INSERT_TAIL(&cp->zones, zp, link); 287 cp->nzones++; 288 } else { 289 if (cp->nzones > 0) 290 errx(1, "%s:%d: zone must have description", 291 path_zonetab, lineno); 292 if (cp->nzones < 0) 293 errx(1, "%s:%d: zone multiply defined", 294 path_zonetab, lineno); 295 cp->nzones = -1; 296 cp->filename = strdup(file); 297 if (cp->filename == NULL) 298 errx(1, "malloc failed"); 299 cp->continent = cont; 300 } 301 } 302 303 /* 304 * This comparison function intentionally sorts all of the null-named 305 * ``countries''---i.e., the codes that don't correspond to a real 306 * country---to the end. Everything else is lexical by country name. 307 */ 308 static int 309 compare_countries(const void *xa, const void *xb) 310 { 311 const struct country *a = xa, *b = xb; 312 313 if (a->name == 0 && b->name == 0) 314 return (0); 315 if (a->name == 0 && b->name != 0) 316 return (1); 317 if (b->name == 0) 318 return (-1); 319 320 return (strcmp(a->name, b->name)); 321 } 322 323 /* 324 * This must be done AFTER all zone descriptions are read, since it breaks 325 * CODE2INT(). 326 */ 327 static void 328 sort_countries(void) 329 { 330 331 qsort(countries, NCOUNTRIES, sizeof(countries[0]), compare_countries); 332 } 333 334 static void 335 read_zones(void) 336 { 337 char contbuf[16]; 338 FILE *fp; 339 struct continent *cont; 340 size_t len; 341 char *line, *tlc, *coord, *file, *descr, *p; 342 int lineno; 343 344 fp = fopen(path_zonetab, "r"); 345 if (!fp) 346 err(1, "%s", path_zonetab); 347 lineno = 0; 348 349 while ((line = fgetln(fp, &len)) != 0) { 350 lineno++; 351 if (line[len - 1] != '\n') 352 errx(1, "%s:%d: invalid format", path_zonetab, lineno); 353 line[len - 1] = '\0'; 354 if (line[0] == '#') 355 continue; 356 357 tlc = strsep(&line, "\t"); 358 if (strlen(tlc) != 2) 359 errx(1, "%s:%d: invalid country code `%s'", 360 path_zonetab, lineno, tlc); 361 coord = strsep(&line, "\t"); /* Unused */ 362 file = strsep(&line, "\t"); 363 p = strchr(file, '/'); 364 if (p == 0) 365 errx(1, "%s:%d: invalid zone name `%s'", path_zonetab, 366 lineno, file); 367 contbuf[0] = '\0'; 368 strncat(contbuf, file, p - file); 369 cont = find_continent(contbuf); 370 if (!cont) 371 errx(1, "%s:%d: invalid region `%s'", path_zonetab, 372 lineno, contbuf); 373 374 descr = (line != NULL && *line != '\0') ? line : NULL; 375 376 add_zone_to_country(lineno, tlc, descr, file, cont); 377 } 378 fclose(fp); 379 } 380 381 static void 382 make_menus(void) 383 { 384 struct country *cp; 385 struct zone *zp, *zp2; 386 struct continent *cont; 387 dialogMenuItem *dmi; 388 int i; 389 390 /* 391 * First, count up all the countries in each continent/ocean. 392 * Be careful to count those countries which have multiple zones 393 * only once for each. NB: some countries are in multiple 394 * continents/oceans. 395 */ 396 for (cp = countries; cp->name; cp++) { 397 if (cp->nzones == 0) 398 continue; 399 if (cp->nzones < 0) { 400 cp->continent->nitems++; 401 } else { 402 TAILQ_FOREACH(zp, &cp->zones, link) { 403 cont = zp->continent; 404 for (zp2 = TAILQ_FIRST(&cp->zones); 405 zp2->continent != cont; 406 zp2 = TAILQ_NEXT(zp2, link)) 407 ; 408 if (zp2 == zp) 409 zp->continent->nitems++; 410 } 411 } 412 } 413 414 /* 415 * Now allocate memory for the country menus and initialize 416 * continent menus. We set nitems back to zero so that we can 417 * use it for counting again when we actually build the menus. 418 */ 419 memset(continents, 0, sizeof(continents)); 420 for (i = 0; i < NCONTINENTS; i++) { 421 continent_names[i].continent->menu = 422 malloc(sizeof(dialogMenuItem) * 423 continent_names[i].continent->nitems); 424 if (continent_names[i].continent->menu == 0) 425 errx(1, "malloc for continent menu"); 426 continent_names[i].continent->nitems = 0; 427 continents[i].prompt = continent_items[i].prompt; 428 continents[i].title = continent_items[i].title; 429 continents[i].fire = continent_country_menu; 430 continents[i].data = continent_names[i].continent; 431 } 432 433 /* 434 * Now that memory is allocated, create the menu items for 435 * each continent. For multiple-zone countries, also create 436 * the country's zone submenu. 437 */ 438 for (cp = countries; cp->name; cp++) { 439 if (cp->nzones == 0) 440 continue; 441 if (cp->nzones < 0) { 442 dmi = &cp->continent->menu[cp->continent->nitems]; 443 memset(dmi, 0, sizeof(*dmi)); 444 asprintf(&dmi->prompt, "%d", ++cp->continent->nitems); 445 dmi->title = cp->name; 446 dmi->checked = 0; 447 dmi->fire = set_zone_whole_country; 448 dmi->selected = 0; 449 dmi->data = cp; 450 } else { 451 cp->submenu = malloc(cp->nzones * sizeof(*dmi)); 452 if (cp->submenu == 0) 453 errx(1, "malloc for submenu"); 454 cp->nzones = 0; 455 TAILQ_FOREACH(zp, &cp->zones, link) { 456 cont = zp->continent; 457 dmi = &cp->submenu[cp->nzones]; 458 memset(dmi, 0, sizeof(*dmi)); 459 asprintf(&dmi->prompt, "%d", ++cp->nzones); 460 dmi->title = zp->descr; 461 dmi->checked = 0; 462 dmi->fire = set_zone_multi; 463 dmi->selected = 0; 464 dmi->data = zp; 465 466 for (zp2 = TAILQ_FIRST(&cp->zones); 467 zp2->continent != cont; 468 zp2 = TAILQ_NEXT(zp2, link)) 469 ; 470 if (zp2 != zp) 471 continue; 472 473 dmi = &cont->menu[cont->nitems]; 474 memset(dmi, 0, sizeof(*dmi)); 475 asprintf(&dmi->prompt, "%d", ++cont->nitems); 476 dmi->title = cp->name; 477 dmi->checked = 0; 478 dmi->fire = set_zone_menu; 479 dmi->selected = 0; 480 dmi->data = cp; 481 } 482 } 483 } 484 } 485 486 static int 487 set_zone_menu(dialogMenuItem *dmi) 488 { 489 char title[64], prompt[64]; 490 struct country *cp = dmi->data; 491 int menulen; 492 int rv; 493 494 snprintf(title, sizeof(title), "%s Time Zones", cp->name); 495 snprintf(prompt, sizeof(prompt), 496 "Select a zone which observes the same time as your locality."); 497 menulen = cp->nzones < 16 ? cp->nzones : 16; 498 rv = dialog_menu(title, prompt, -1, -1, menulen, -cp->nzones, 499 cp->submenu, 0, 0, 0); 500 if (rv != 0) 501 return (DITEM_RECREATE); 502 return (DITEM_LEAVE_MENU); 503 } 504 505 static int 506 install_zoneinfo_file(const char *zoneinfo_file) 507 { 508 char buf[1024]; 509 char title[64], prompt[64]; 510 struct stat sb; 511 ssize_t len; 512 int fd1, fd2, copymode; 513 514 if (lstat(path_localtime, &sb) < 0) { 515 /* Nothing there yet... */ 516 copymode = 1; 517 } else if (S_ISLNK(sb.st_mode)) 518 copymode = 0; 519 else 520 copymode = 1; 521 522 #ifdef VERBOSE 523 if (copymode) 524 snprintf(prompt, sizeof(prompt), 525 "Copying %s to %s", zoneinfo_file, path_localtime); 526 else 527 snprintf(prompt, sizeof(prompt), 528 "Creating symbolic link %s to %s", 529 path_localtime, zoneinfo_file); 530 if (usedialog) 531 dialog_notify(prompt); 532 else 533 fprintf(stderr, "%s\n", prompt); 534 #endif 535 536 if (reallydoit) { 537 if (copymode) { 538 fd1 = open(zoneinfo_file, O_RDONLY, 0); 539 if (fd1 < 0) { 540 snprintf(title, sizeof(title), "Error"); 541 snprintf(prompt, sizeof(prompt), 542 "Could not open %s: %s", zoneinfo_file, 543 strerror(errno)); 544 if (usedialog) 545 dialog_mesgbox(title, prompt, 8, 72); 546 else 547 fprintf(stderr, "%s\n", prompt); 548 return (DITEM_FAILURE | DITEM_RECREATE); 549 } 550 551 unlink(path_localtime); 552 fd2 = open(path_localtime, O_CREAT | O_EXCL | O_WRONLY, 553 S_IRUSR | S_IRGRP | S_IROTH); 554 if (fd2 < 0) { 555 snprintf(title, sizeof(title), "Error"); 556 snprintf(prompt, sizeof(prompt), 557 "Could not open %s: %s", 558 path_localtime, strerror(errno)); 559 if (usedialog) 560 dialog_mesgbox(title, prompt, 8, 72); 561 else 562 fprintf(stderr, "%s\n", prompt); 563 return (DITEM_FAILURE | DITEM_RECREATE); 564 } 565 566 while ((len = read(fd1, buf, sizeof(buf))) > 0) 567 if ((len = write(fd2, buf, len)) < 0) 568 break; 569 570 if (len == -1) { 571 snprintf(title, sizeof(title), "Error"); 572 snprintf(prompt, sizeof(prompt), 573 "Error copying %s to %s %s", zoneinfo_file, 574 path_localtime, strerror(errno)); 575 if (usedialog) 576 dialog_mesgbox(title, prompt, 8, 72); 577 else 578 fprintf(stderr, "%s\n", prompt); 579 /* Better to leave none than a corrupt one. */ 580 unlink(path_localtime); 581 return (DITEM_FAILURE | DITEM_RECREATE); 582 } 583 close(fd1); 584 close(fd2); 585 } else { 586 if (access(zoneinfo_file, R_OK) != 0) { 587 snprintf(title, sizeof(title), "Error"); 588 snprintf(prompt, sizeof(prompt), 589 "Cannot access %s: %s", zoneinfo_file, 590 strerror(errno)); 591 if (usedialog) 592 dialog_mesgbox(title, prompt, 8, 72); 593 else 594 fprintf(stderr, "%s\n", prompt); 595 return (DITEM_FAILURE | DITEM_RECREATE); 596 } 597 unlink(path_localtime); 598 if (symlink(zoneinfo_file, path_localtime) < 0) { 599 snprintf(title, sizeof(title), "Error"); 600 snprintf(prompt, sizeof(prompt), 601 "Cannot create symbolic link %s to %s: %s", 602 path_localtime, zoneinfo_file, 603 strerror(errno)); 604 if (usedialog) 605 dialog_mesgbox(title, prompt, 8, 72); 606 else 607 fprintf(stderr, "%s\n", prompt); 608 return (DITEM_FAILURE | DITEM_RECREATE); 609 } 610 } 611 } 612 613 #ifdef VERBOSE 614 snprintf(title, sizeof(title), "Done"); 615 if (copymode) 616 snprintf(prompt, sizeof(prompt), 617 "Copied timezone file from %s to %s", zoneinfo_file, 618 path_localtime); 619 else 620 snprintf(prompt, sizeof(prompt), 621 "Created symbolic link from %s to %s", zoneinfo_file, 622 path_localtime); 623 if (usedialog) 624 dialog_mesgbox(title, prompt, 8, 72); 625 else 626 fprintf(stderr, "%s\n", prompt); 627 #endif 628 629 return (DITEM_LEAVE_MENU); 630 } 631 632 static int 633 install_zoneinfo(const char *zoneinfo) 634 { 635 int rv; 636 FILE *f; 637 char path_zoneinfo_file[MAXPATHLEN]; 638 639 sprintf(path_zoneinfo_file, "%s/%s", path_zoneinfo, zoneinfo); 640 rv = install_zoneinfo_file(path_zoneinfo_file); 641 642 /* Save knowledge for later */ 643 if ((f = fopen(path_db, "w")) != NULL) { 644 fprintf(f, "%s\n", zoneinfo); 645 fclose(f); 646 } 647 648 return (rv); 649 } 650 651 static int 652 confirm_zone(const char *filename) 653 { 654 char title[64], prompt[64]; 655 time_t t = time(0); 656 struct tm *tm; 657 int rv; 658 659 setenv("TZ", filename, 1); 660 tzset(); 661 tm = localtime(&t); 662 663 snprintf(title, sizeof(title), "Confirmation"); 664 snprintf(prompt, sizeof(prompt), 665 "Does the abbreviation `%s' look reasonable?", tm->tm_zone); 666 rv = !dialog_yesno(title, prompt, 5, 72); 667 return (rv); 668 } 669 670 static int 671 set_zone_multi(dialogMenuItem *dmi) 672 { 673 struct zone *zp = dmi->data; 674 int rv; 675 676 if (!confirm_zone(zp->filename)) 677 return (DITEM_FAILURE | DITEM_RECREATE); 678 679 rv = install_zoneinfo(zp->filename); 680 return (rv); 681 } 682 683 static int 684 set_zone_whole_country(dialogMenuItem *dmi) 685 { 686 struct country *cp = dmi->data; 687 int rv; 688 689 if (!confirm_zone(cp->filename)) 690 return (DITEM_FAILURE | DITEM_RECREATE); 691 692 rv = install_zoneinfo(cp->filename); 693 return (rv); 694 } 695 696 static void 697 usage(void) 698 { 699 700 fprintf(stderr, "usage: tzsetup [-nrs] [zoneinfo file]\n"); 701 exit(1); 702 } 703 704 #if defined(__sparc64__) 705 #define DIALOG_UTC dialog_yesno 706 #else 707 #define DIALOG_UTC dialog_noyes 708 #endif 709 710 int 711 main(int argc, char **argv) 712 { 713 char title[64], prompt[128]; 714 int c, fd, rv, skiputc; 715 716 skiputc = 0; 717 while ((c = getopt(argc, argv, "C:nrs")) != -1) { 718 switch(c) { 719 case 'C': 720 chrootenv = optarg; 721 break; 722 case 'n': 723 reallydoit = 0; 724 break; 725 case 'r': 726 reinstall = 1; 727 usedialog = 0; 728 break; 729 case 's': 730 skiputc = 1; 731 break; 732 default: 733 usage(); 734 } 735 } 736 737 if (argc - optind > 1) 738 usage(); 739 740 if (chrootenv == NULL) { 741 strcpy(path_zonetab, _PATH_ZONETAB); 742 strcpy(path_iso3166, _PATH_ISO3166); 743 strcpy(path_zoneinfo, _PATH_ZONEINFO); 744 strcpy(path_localtime, _PATH_LOCALTIME); 745 strcpy(path_db, _PATH_DB); 746 strcpy(path_wall_cmos_clock, _PATH_WALL_CMOS_CLOCK); 747 } else { 748 sprintf(path_zonetab, "%s/%s", chrootenv, _PATH_ZONETAB); 749 sprintf(path_iso3166, "%s/%s", chrootenv, _PATH_ISO3166); 750 sprintf(path_zoneinfo, "%s/%s", chrootenv, _PATH_ZONEINFO); 751 sprintf(path_localtime, "%s/%s", chrootenv, _PATH_LOCALTIME); 752 sprintf(path_db, "%s/%s", chrootenv, _PATH_DB); 753 sprintf(path_wall_cmos_clock, "%s/%s", chrootenv, 754 _PATH_WALL_CMOS_CLOCK); 755 } 756 757 758 /* Override the user-supplied umask. */ 759 (void)umask(S_IWGRP | S_IWOTH); 760 761 read_iso3166_table(); 762 read_zones(); 763 sort_countries(); 764 make_menus(); 765 766 if (reinstall == 1) { 767 FILE *f; 768 char zonefile[MAXPATHLEN]; 769 char path_db[MAXPATHLEN]; 770 771 zonefile[0] = '\0'; 772 path_db[0] = '\0'; 773 if (chrootenv != NULL) { 774 sprintf(zonefile, "%s/", chrootenv); 775 sprintf(path_db, "%s/", chrootenv); 776 } 777 strcat(zonefile, _PATH_ZONEINFO); 778 strcat(zonefile, "/"); 779 strcat(path_db, _PATH_DB); 780 781 if ((f = fopen(path_db, "r")) != NULL) { 782 if (fgets(zonefile, sizeof(zonefile), f) != NULL) { 783 zonefile[sizeof(zonefile) - 1] = 0; 784 if (strlen(zonefile) > 0) { 785 zonefile[strlen(zonefile) - 1] = 0; 786 rv = install_zoneinfo(zonefile); 787 exit(rv & ~DITEM_LEAVE_MENU); 788 } 789 errx(1, "Error reading %s.\n", path_db); 790 } 791 fclose(f); 792 errx(1, 793 "Unable to determine earlier installed zoneinfo " 794 "file. Check %s", path_db); 795 } 796 errx(1, "Cannot open %s for reading. Does it exist?", path_db); 797 } 798 799 /* 800 * If the arguments on the command-line do not specify a file, 801 * then interpret it as a zoneinfo name 802 */ 803 if (optind == argc - 1) { 804 struct stat sb; 805 806 if (stat(argv[optind], &sb) != 0) { 807 usedialog = 0; 808 rv = install_zoneinfo(argv[optind]); 809 exit(rv & ~DITEM_LEAVE_MENU); 810 } 811 /* FALLTHROUGH */ 812 } 813 814 init_dialog(); 815 if (skiputc == 0) { 816 snprintf(title, sizeof(title), 817 "Select local or UTC (Greenwich Mean Time) clock"); 818 snprintf(prompt, sizeof(prompt), 819 "Is this machine's CMOS clock set to UTC? " 820 "If it is set to local time,\n" 821 "or you don't know, please choose NO here!"); 822 if (!DIALOG_UTC(title, prompt, 7, 72)) { 823 if (reallydoit) 824 unlink(path_wall_cmos_clock); 825 } else { 826 if (reallydoit) { 827 fd = open(path_wall_cmos_clock, 828 O_WRONLY | O_CREAT | O_TRUNC, 829 S_IRUSR | S_IRGRP | S_IROTH); 830 if (fd < 0) { 831 end_dialog(); 832 err(1, "create %s", 833 path_wall_cmos_clock); 834 } 835 close(fd); 836 } 837 } 838 dialog_clear_norefresh(); 839 } 840 if (optind == argc - 1) { 841 snprintf(title, sizeof(title), "Default timezone provided"); 842 snprintf(prompt, sizeof(prompt), 843 "\nUse the default `%s' zone?", argv[optind]); 844 if (!dialog_yesno(title, prompt, 7, 72)) { 845 rv = install_zoneinfo_file(argv[optind]); 846 dialog_clear(); 847 end_dialog(); 848 exit(rv & ~DITEM_LEAVE_MENU); 849 } 850 dialog_clear_norefresh(); 851 } 852 snprintf(title, sizeof(title), "Time Zone Selector"); 853 snprintf(prompt, sizeof(prompt), "Select a region"); 854 dialog_menu(title, prompt, -1, -1, NCONTINENTS, -NCONTINENTS, 855 continents, 0, NULL, NULL); 856 857 dialog_clear(); 858 end_dialog(); 859 return (0); 860 } 861