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/zone.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 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 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 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, arctic, 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 { "Arctic", &arctic }, 231 { "Asia", &asia }, 232 { "Atlantic", &atlantic }, 233 { "Australia", &australia }, 234 { "Europe", &europe }, 235 { "Indian", &indian }, 236 { "Pacific", &pacific }, 237 { "UTC", &utc } 238 }; 239 240 static struct continent_items { 241 char prompt[2]; 242 char title[30]; 243 } continent_items[] = { 244 { "1", "Africa" }, 245 { "2", "America -- North and South" }, 246 { "3", "Antarctica" }, 247 { "4", "Arctic Ocean" }, 248 { "5", "Asia" }, 249 { "6", "Atlantic Ocean" }, 250 { "7", "Australia" }, 251 { "8", "Europe" }, 252 { "9", "Indian Ocean" }, 253 { "0", "Pacific Ocean" }, 254 { "a", "UTC" } 255 }; 256 257 #define NCONTINENTS \ 258 (int)((sizeof(continent_items)) / (sizeof(continent_items[0]))) 259 static dialogMenuItem continents[NCONTINENTS]; 260 261 #define OCEANP(x) ((x) == 3 || (x) == 5 || (x) == 8 || (x) == 9) 262 263 static int 264 continent_country_menu(dialogMenuItem *continent) 265 { 266 char title[64], prompt[64]; 267 struct continent *contp = continent->data; 268 int isocean = OCEANP(continent - continents); 269 int menulen; 270 int rv; 271 272 if (strcmp(continent->title, "UTC") == 0) 273 return (set_zone_utc()); 274 275 /* Short cut -- if there's only one country, don't post a menu. */ 276 if (contp->nitems == 1) 277 return (contp->menu[0].fire(&contp->menu[0])); 278 279 /* It's amazing how much good grammar really matters... */ 280 if (!isocean) { 281 snprintf(title, sizeof(title), "Countries in %s", 282 continent->title); 283 snprintf(prompt, sizeof(prompt), "Select a country or region"); 284 } else { 285 snprintf(title, sizeof(title), "Islands and groups in the %s", 286 continent->title); 287 snprintf(prompt, sizeof(prompt), "Select an island or group"); 288 } 289 290 menulen = contp->nitems < 16 ? contp->nitems : 16; 291 rv = xdialog_menu(title, prompt, -1, -1, menulen, contp->nitems, 292 contp->menu); 293 if (rv == 0) 294 return (DITEM_LEAVE_MENU); 295 return (DITEM_RECREATE); 296 } 297 298 static struct continent * 299 find_continent(const char *name) 300 { 301 int i; 302 303 for (i = 0; i < NCONTINENTS; i++) 304 if (strcmp(name, continent_names[i].name) == 0) 305 return (continent_names[i].continent); 306 return (0); 307 } 308 309 struct country { 310 char *name; 311 char *tlc; 312 int nzones; 313 char *filename; /* use iff nzones < 0 */ 314 struct continent *continent; /* use iff nzones < 0 */ 315 TAILQ_HEAD(, zone) zones; /* use iff nzones > 0 */ 316 dialogMenuItem *submenu; /* use iff nzones > 0 */ 317 }; 318 319 struct zone { 320 TAILQ_ENTRY(zone) link; 321 char *descr; 322 char *filename; 323 struct continent *continent; 324 }; 325 326 /* 327 * This is the easiest organization... we use ISO 3166 country codes, 328 * of the two-letter variety, so we just size this array to suit. 329 * Beats worrying about dynamic allocation. 330 */ 331 #define NCOUNTRIES (26 * 26) 332 static struct country countries[NCOUNTRIES]; 333 334 #define CODE2INT(s) ((s[0] - 'A') * 26 + (s[1] - 'A')) 335 336 /* 337 * Read the ISO 3166 country code database in _PATH_ISO3166 338 * (/usr/share/misc/iso3166). On error, exit via err(3). 339 */ 340 static void 341 read_iso3166_table(void) 342 { 343 FILE *fp; 344 struct country *cp; 345 size_t len; 346 char *s, *t, *name; 347 int lineno; 348 349 fp = fopen(path_iso3166, "r"); 350 if (!fp) 351 err(1, "%s", path_iso3166); 352 lineno = 0; 353 354 while ((s = fgetln(fp, &len)) != NULL) { 355 lineno++; 356 if (s[len - 1] != '\n') 357 errx(1, "%s:%d: invalid format", path_iso3166, lineno); 358 s[len - 1] = '\0'; 359 if (s[0] == '#' || strspn(s, " \t") == len - 1) 360 continue; 361 362 /* Isolate the two-letter code. */ 363 t = strsep(&s, "\t"); 364 if (t == NULL || strlen(t) != 2) 365 errx(1, "%s:%d: invalid format", path_iso3166, lineno); 366 if (t[0] < 'A' || t[0] > 'Z' || t[1] < 'A' || t[1] > 'Z') 367 errx(1, "%s:%d: invalid code `%s'", path_iso3166, 368 lineno, t); 369 370 /* Now skip past the three-letter and numeric codes. */ 371 name = strsep(&s, "\t"); /* 3-let */ 372 if (name == NULL || strlen(name) != 3) 373 errx(1, "%s:%d: invalid format", path_iso3166, lineno); 374 name = strsep(&s, "\t"); /* numeric */ 375 if (name == NULL || strlen(name) != 3) 376 errx(1, "%s:%d: invalid format", path_iso3166, lineno); 377 378 name = s; 379 380 cp = &countries[CODE2INT(t)]; 381 if (cp->name) 382 errx(1, "%s:%d: country code `%s' multiply defined: %s", 383 path_iso3166, lineno, t, cp->name); 384 cp->name = strdup(name); 385 if (cp->name == NULL) 386 errx(1, "malloc failed"); 387 cp->tlc = strdup(t); 388 if (cp->tlc == NULL) 389 errx(1, "malloc failed"); 390 } 391 392 fclose(fp); 393 } 394 395 static void 396 add_zone_to_country(int lineno, const char *tlc, const char *descr, 397 const char *file, struct continent *cont) 398 { 399 struct zone *zp; 400 struct country *cp; 401 402 if (tlc[0] < 'A' || tlc[0] > 'Z' || tlc[1] < 'A' || tlc[1] > 'Z') 403 errx(1, "%s:%d: country code `%s' invalid", path_zonetab, 404 lineno, tlc); 405 406 cp = &countries[CODE2INT(tlc)]; 407 if (cp->name == 0) 408 errx(1, "%s:%d: country code `%s' unknown", path_zonetab, 409 lineno, tlc); 410 411 if (descr) { 412 if (cp->nzones < 0) 413 errx(1, "%s:%d: conflicting zone definition", 414 path_zonetab, lineno); 415 416 zp = malloc(sizeof(*zp)); 417 if (zp == NULL) 418 errx(1, "malloc(%zu)", sizeof(*zp)); 419 420 if (cp->nzones == 0) 421 TAILQ_INIT(&cp->zones); 422 423 zp->descr = strdup(descr); 424 if (zp->descr == NULL) 425 errx(1, "malloc failed"); 426 zp->filename = strdup(file); 427 if (zp->filename == NULL) 428 errx(1, "malloc failed"); 429 zp->continent = cont; 430 TAILQ_INSERT_TAIL(&cp->zones, zp, link); 431 cp->nzones++; 432 } else { 433 if (cp->nzones > 0) 434 errx(1, "%s:%d: zone must have description", 435 path_zonetab, lineno); 436 if (cp->nzones < 0) 437 errx(1, "%s:%d: zone multiply defined", 438 path_zonetab, lineno); 439 cp->nzones = -1; 440 cp->filename = strdup(file); 441 if (cp->filename == NULL) 442 errx(1, "malloc failed"); 443 cp->continent = cont; 444 } 445 } 446 447 /* 448 * This comparison function intentionally sorts all of the null-named 449 * ``countries''---i.e., the codes that don't correspond to a real 450 * country---to the end. Everything else is lexical by country name. 451 */ 452 static int 453 compare_countries(const void *xa, const void *xb) 454 { 455 const struct country *a = xa, *b = xb; 456 457 if (a->name == 0 && b->name == 0) 458 return (0); 459 if (a->name == 0 && b->name != 0) 460 return (1); 461 if (b->name == 0) 462 return (-1); 463 464 return (strcmp(a->name, b->name)); 465 } 466 467 /* 468 * This must be done AFTER all zone descriptions are read, since it breaks 469 * CODE2INT(). 470 */ 471 static void 472 sort_countries(void) 473 { 474 475 qsort(countries, NCOUNTRIES, sizeof(countries[0]), compare_countries); 476 } 477 478 static void 479 read_zones(void) 480 { 481 char contbuf[16]; 482 FILE *fp; 483 struct continent *cont; 484 size_t len; 485 char *line, *tlc, *file, *descr, *p; 486 int lineno; 487 488 fp = fopen(path_zonetab, "r"); 489 if (!fp) 490 err(1, "%s", path_zonetab); 491 lineno = 0; 492 493 while ((line = fgetln(fp, &len)) != NULL) { 494 lineno++; 495 if (line[len - 1] != '\n') 496 errx(1, "%s:%d: invalid format", path_zonetab, lineno); 497 line[len - 1] = '\0'; 498 if (line[0] == '#') 499 continue; 500 501 tlc = strsep(&line, "\t"); 502 if (strlen(tlc) != 2) 503 errx(1, "%s:%d: invalid country code `%s'", 504 path_zonetab, lineno, tlc); 505 /* coord = */ strsep(&line, "\t"); /* Unused */ 506 file = strsep(&line, "\t"); 507 p = strchr(file, '/'); 508 if (p == NULL) 509 errx(1, "%s:%d: invalid zone name `%s'", path_zonetab, 510 lineno, file); 511 contbuf[0] = '\0'; 512 strncat(contbuf, file, p - file); 513 cont = find_continent(contbuf); 514 if (!cont) 515 errx(1, "%s:%d: invalid region `%s'", path_zonetab, 516 lineno, contbuf); 517 518 descr = (line != NULL && *line != '\0') ? line : NULL; 519 520 add_zone_to_country(lineno, tlc, descr, file, cont); 521 } 522 fclose(fp); 523 } 524 525 static void 526 make_menus(void) 527 { 528 struct country *cp; 529 struct zone *zp, *zp2; 530 struct continent *cont; 531 dialogMenuItem *dmi; 532 int i; 533 534 /* 535 * First, count up all the countries in each continent/ocean. 536 * Be careful to count those countries which have multiple zones 537 * only once for each. NB: some countries are in multiple 538 * continents/oceans. 539 */ 540 for (cp = countries; cp->name; cp++) { 541 if (cp->nzones == 0) 542 continue; 543 if (cp->nzones < 0) { 544 cp->continent->nitems++; 545 } else { 546 TAILQ_FOREACH(zp, &cp->zones, link) { 547 cont = zp->continent; 548 for (zp2 = TAILQ_FIRST(&cp->zones); 549 zp2->continent != cont; 550 zp2 = TAILQ_NEXT(zp2, link)) 551 ; 552 if (zp2 == zp) 553 zp->continent->nitems++; 554 } 555 } 556 } 557 558 /* 559 * Now allocate memory for the country menus and initialize 560 * continent menus. We set nitems back to zero so that we can 561 * use it for counting again when we actually build the menus. 562 */ 563 memset(continents, 0, sizeof(continents)); 564 for (i = 0; i < NCONTINENTS; i++) { 565 continent_names[i].continent->menu = 566 malloc(sizeof(dialogMenuItem) * 567 continent_names[i].continent->nitems); 568 if (continent_names[i].continent->menu == NULL) 569 errx(1, "malloc for continent menu"); 570 continent_names[i].continent->nitems = 0; 571 continents[i].prompt = continent_items[i].prompt; 572 continents[i].title = continent_items[i].title; 573 continents[i].fire = continent_country_menu; 574 continents[i].data = continent_names[i].continent; 575 } 576 577 /* 578 * Now that memory is allocated, create the menu items for 579 * each continent. For multiple-zone countries, also create 580 * the country's zone submenu. 581 */ 582 for (cp = countries; cp->name; cp++) { 583 if (cp->nzones == 0) 584 continue; 585 if (cp->nzones < 0) { 586 dmi = &cp->continent->menu[cp->continent->nitems]; 587 memset(dmi, 0, sizeof(*dmi)); 588 asprintf(&dmi->prompt, "%d", ++cp->continent->nitems); 589 dmi->title = cp->name; 590 dmi->fire = set_zone_whole_country; 591 dmi->data = cp; 592 } else { 593 cp->submenu = malloc(cp->nzones * sizeof(*dmi)); 594 if (cp->submenu == 0) 595 errx(1, "malloc for submenu"); 596 cp->nzones = 0; 597 TAILQ_FOREACH(zp, &cp->zones, link) { 598 cont = zp->continent; 599 dmi = &cp->submenu[cp->nzones]; 600 memset(dmi, 0, sizeof(*dmi)); 601 asprintf(&dmi->prompt, "%d", ++cp->nzones); 602 dmi->title = zp->descr; 603 dmi->fire = set_zone_multi; 604 dmi->data = zp; 605 606 for (zp2 = TAILQ_FIRST(&cp->zones); 607 zp2->continent != cont; 608 zp2 = TAILQ_NEXT(zp2, link)) 609 ; 610 if (zp2 != zp) 611 continue; 612 613 dmi = &cont->menu[cont->nitems]; 614 memset(dmi, 0, sizeof(*dmi)); 615 asprintf(&dmi->prompt, "%d", ++cont->nitems); 616 dmi->title = cp->name; 617 dmi->fire = set_zone_menu; 618 dmi->data = cp; 619 } 620 } 621 } 622 } 623 624 static int 625 set_zone_menu(dialogMenuItem *dmi) 626 { 627 char title[64], prompt[64]; 628 struct country *cp = dmi->data; 629 int menulen; 630 int rv; 631 632 snprintf(title, sizeof(title), "%s Time Zones", cp->name); 633 snprintf(prompt, sizeof(prompt), 634 "Select a zone which observes the same time as your locality."); 635 menulen = cp->nzones < 16 ? cp->nzones : 16; 636 rv = xdialog_menu(title, prompt, -1, -1, menulen, cp->nzones, 637 cp->submenu); 638 if (rv != 0) 639 return (DITEM_RECREATE); 640 return (DITEM_LEAVE_MENU); 641 } 642 643 static int 644 set_zone_utc(void) 645 { 646 if (!confirm_zone("UTC")) 647 return (DITEM_FAILURE | DITEM_RECREATE); 648 649 return (install_zoneinfo("UTC")); 650 } 651 652 static int 653 confirm_zone(const char *filename) 654 { 655 char title[64], prompt[64]; 656 time_t t = time(0); 657 struct tm *tm; 658 int rv; 659 660 setenv("TZ", filename, 1); 661 tzset(); 662 tm = localtime(&t); 663 664 snprintf(title, sizeof(title), "Confirmation"); 665 snprintf(prompt, sizeof(prompt), 666 "Does the abbreviation `%s' look reasonable?", tm->tm_zone); 667 rv = !dialog_yesno(title, prompt, 5, 72); 668 return (rv); 669 } 670 671 static int 672 set_zone_multi(dialogMenuItem *dmi) 673 { 674 struct zone *zp = dmi->data; 675 int rv; 676 677 if (!confirm_zone(zp->filename)) 678 return (DITEM_FAILURE | DITEM_RECREATE); 679 680 rv = install_zoneinfo(zp->filename); 681 return (rv); 682 } 683 684 static int 685 set_zone_whole_country(dialogMenuItem *dmi) 686 { 687 struct country *cp = dmi->data; 688 int rv; 689 690 if (!confirm_zone(cp->filename)) 691 return (DITEM_FAILURE | DITEM_RECREATE); 692 693 rv = install_zoneinfo(cp->filename); 694 return (rv); 695 } 696 697 #endif 698 699 static int 700 install_zoneinfo_file(const char *zoneinfo_file) 701 { 702 char buf[1024]; 703 char title[64], prompt[SILLY_BUFFER_SIZE]; 704 struct stat sb; 705 ssize_t len; 706 int fd1, fd2, copymode; 707 708 if (lstat(path_localtime, &sb) < 0) { 709 /* Nothing there yet... */ 710 copymode = 1; 711 } else if (S_ISLNK(sb.st_mode)) 712 copymode = 0; 713 else 714 copymode = 1; 715 716 #ifdef VERBOSE 717 snprintf(title, sizeof(title), "Info"); 718 if (copymode) 719 snprintf(prompt, sizeof(prompt), 720 "Copying %s to %s", zoneinfo_file, path_localtime); 721 else 722 snprintf(prompt, sizeof(prompt), 723 "Creating symbolic link %s to %s", 724 path_localtime, zoneinfo_file); 725 #ifdef HAVE_DIALOG 726 if (usedialog) 727 dialog_msgbox(title, prompt, 8, 72, 1); 728 else 729 #endif 730 fprintf(stderr, "%s\n", prompt); 731 #endif 732 733 if (reallydoit) { 734 if (copymode) { 735 fd1 = open(zoneinfo_file, O_RDONLY, 0); 736 if (fd1 < 0) { 737 snprintf(title, sizeof(title), "Error"); 738 snprintf(prompt, sizeof(prompt), 739 "Could not open %s: %s", zoneinfo_file, 740 strerror(errno)); 741 #ifdef HAVE_DIALOG 742 if (usedialog) 743 dialog_msgbox(title, prompt, 8, 72, 1); 744 else 745 #endif 746 fprintf(stderr, "%s\n", prompt); 747 return (DITEM_FAILURE | DITEM_RECREATE); 748 } 749 750 if (unlink(path_localtime) < 0 && errno != ENOENT) { 751 snprintf(prompt, sizeof(prompt), 752 "Could not delete %s: %s", 753 path_localtime, strerror(errno)); 754 #ifdef HAVE_DIALOG 755 if (usedialog) { 756 snprintf(title, sizeof(title), "Error"); 757 dialog_msgbox(title, prompt, 8, 72, 1); 758 } else 759 #endif 760 fprintf(stderr, "%s\n", prompt); 761 return (DITEM_FAILURE | DITEM_RECREATE); 762 } 763 764 fd2 = open(path_localtime, O_CREAT | O_EXCL | O_WRONLY, 765 S_IRUSR | S_IRGRP | S_IROTH); 766 if (fd2 < 0) { 767 snprintf(title, sizeof(title), "Error"); 768 snprintf(prompt, sizeof(prompt), 769 "Could not open %s: %s", 770 path_localtime, strerror(errno)); 771 #ifdef HAVE_DIALOG 772 if (usedialog) 773 dialog_msgbox(title, prompt, 8, 72, 1); 774 else 775 #endif 776 fprintf(stderr, "%s\n", prompt); 777 return (DITEM_FAILURE | DITEM_RECREATE); 778 } 779 780 while ((len = read(fd1, buf, sizeof(buf))) > 0) 781 if ((len = write(fd2, buf, len)) < 0) 782 break; 783 784 if (len == -1) { 785 snprintf(title, sizeof(title), "Error"); 786 snprintf(prompt, sizeof(prompt), 787 "Error copying %s to %s %s", zoneinfo_file, 788 path_localtime, strerror(errno)); 789 #ifdef HAVE_DIALOG 790 if (usedialog) 791 dialog_msgbox(title, prompt, 8, 72, 1); 792 else 793 #endif 794 fprintf(stderr, "%s\n", prompt); 795 /* Better to leave none than a corrupt one. */ 796 unlink(path_localtime); 797 return (DITEM_FAILURE | DITEM_RECREATE); 798 } 799 close(fd1); 800 close(fd2); 801 } else { 802 if (access(zoneinfo_file, R_OK) != 0) { 803 snprintf(title, sizeof(title), "Error"); 804 snprintf(prompt, sizeof(prompt), 805 "Cannot access %s: %s", zoneinfo_file, 806 strerror(errno)); 807 #ifdef HAVE_DIALOG 808 if (usedialog) 809 dialog_msgbox(title, prompt, 8, 72, 1); 810 else 811 #endif 812 fprintf(stderr, "%s\n", prompt); 813 return (DITEM_FAILURE | DITEM_RECREATE); 814 } 815 if (unlink(path_localtime) < 0 && errno != ENOENT) { 816 snprintf(prompt, sizeof(prompt), 817 "Could not delete %s: %s", 818 path_localtime, strerror(errno)); 819 #ifdef HAVE_DIALOG 820 if (usedialog) { 821 snprintf(title, sizeof(title), "Error"); 822 dialog_msgbox(title, prompt, 8, 72, 1); 823 } else 824 #endif 825 fprintf(stderr, "%s\n", prompt); 826 return (DITEM_FAILURE | DITEM_RECREATE); 827 } 828 if (symlink(zoneinfo_file, path_localtime) < 0) { 829 snprintf(title, sizeof(title), "Error"); 830 snprintf(prompt, sizeof(prompt), 831 "Cannot create symbolic link %s to %s: %s", 832 path_localtime, zoneinfo_file, 833 strerror(errno)); 834 #ifdef HAVE_DIALOG 835 if (usedialog) 836 dialog_msgbox(title, prompt, 8, 72, 1); 837 else 838 #endif 839 fprintf(stderr, "%s\n", prompt); 840 return (DITEM_FAILURE | DITEM_RECREATE); 841 } 842 } 843 844 #ifdef VERBOSE 845 snprintf(title, sizeof(title), "Done"); 846 if (copymode) 847 snprintf(prompt, sizeof(prompt), 848 "Copied timezone file from %s to %s", 849 zoneinfo_file, path_localtime); 850 else 851 snprintf(prompt, sizeof(prompt), 852 "Created symbolic link from %s to %s", 853 zoneinfo_file, path_localtime); 854 #ifdef HAVE_DIALOG 855 if (usedialog) 856 dialog_msgbox(title, prompt, 8, 72, 1); 857 else 858 #endif 859 fprintf(stderr, "%s\n", prompt); 860 #endif 861 } /* reallydoit */ 862 863 return (DITEM_LEAVE_MENU); 864 } 865 866 static int 867 install_zoneinfo(const char *zoneinfo) 868 { 869 int rv; 870 FILE *f; 871 char path_zoneinfo_file[MAXPATHLEN]; 872 873 if ((size_t)snprintf(path_zoneinfo_file, sizeof(path_zoneinfo_file), 874 "%s/%s", path_zoneinfo, zoneinfo) >= sizeof(path_zoneinfo_file)) 875 errx(1, "%s/%s name too long", path_zoneinfo, zoneinfo); 876 rv = install_zoneinfo_file(path_zoneinfo_file); 877 878 /* Save knowledge for later */ 879 if (reallydoit && (rv & DITEM_FAILURE) == 0) { 880 if ((f = fopen(path_db, "w")) != NULL) { 881 fprintf(f, "%s\n", zoneinfo); 882 fclose(f); 883 } 884 } 885 886 return (rv); 887 } 888 889 static void 890 usage(void) 891 { 892 893 fprintf(stderr, "usage: tzsetup [-nrs] [-C chroot_directory]" 894 " [zoneinfo_file | zoneinfo_name]\n"); 895 exit(1); 896 } 897 898 int 899 main(int argc, char **argv) 900 { 901 #ifdef HAVE_DIALOG 902 char title[64], prompt[128]; 903 int fd; 904 #endif 905 int c, rv, skiputc; 906 char vm_guest[16] = ""; 907 size_t len = sizeof(vm_guest); 908 909 skiputc = 0; 910 911 /* Default skiputc to 1 for VM guests */ 912 if (sysctlbyname("kern.vm_guest", vm_guest, &len, NULL, 0) == 0 && 913 strcmp(vm_guest, "none") != 0) 914 skiputc = 1; 915 916 while ((c = getopt(argc, argv, "C:nrs")) != -1) { 917 switch(c) { 918 case 'C': 919 chrootenv = optarg; 920 break; 921 case 'n': 922 reallydoit = 0; 923 break; 924 case 'r': 925 reinstall = 1; 926 #ifdef HAVE_DIALOG 927 usedialog = 0; 928 #endif 929 break; 930 case 's': 931 skiputc = 1; 932 break; 933 default: 934 usage(); 935 } 936 } 937 938 if (argc - optind > 1) 939 usage(); 940 941 if (chrootenv == NULL) { 942 strcpy(path_zonetab, _PATH_ZONETAB); 943 strcpy(path_iso3166, _PATH_ISO3166); 944 strcpy(path_zoneinfo, _PATH_ZONEINFO); 945 strcpy(path_localtime, _PATH_LOCALTIME); 946 strcpy(path_db, _PATH_DB); 947 strcpy(path_wall_cmos_clock, _PATH_WALL_CMOS_CLOCK); 948 } else { 949 sprintf(path_zonetab, "%s/%s", chrootenv, _PATH_ZONETAB); 950 sprintf(path_iso3166, "%s/%s", chrootenv, _PATH_ISO3166); 951 sprintf(path_zoneinfo, "%s/%s", chrootenv, _PATH_ZONEINFO); 952 sprintf(path_localtime, "%s/%s", chrootenv, _PATH_LOCALTIME); 953 sprintf(path_db, "%s/%s", chrootenv, _PATH_DB); 954 sprintf(path_wall_cmos_clock, "%s/%s", chrootenv, 955 _PATH_WALL_CMOS_CLOCK); 956 } 957 958 /* Override the user-supplied umask. */ 959 (void)umask(S_IWGRP | S_IWOTH); 960 961 if (reinstall == 1) { 962 FILE *f; 963 char zoneinfo[MAXPATHLEN]; 964 965 if ((f = fopen(path_db, "r")) != NULL) { 966 if (fgets(zoneinfo, sizeof(zoneinfo), f) != NULL) { 967 zoneinfo[sizeof(zoneinfo) - 1] = 0; 968 if (strlen(zoneinfo) > 0) { 969 zoneinfo[strlen(zoneinfo) - 1] = 0; 970 rv = install_zoneinfo(zoneinfo); 971 exit(rv & ~DITEM_LEAVE_MENU); 972 } 973 errx(1, "Error reading %s.\n", path_db); 974 } 975 fclose(f); 976 errx(1, 977 "Unable to determine earlier installed zoneinfo " 978 "name. Check %s", path_db); 979 } 980 errx(1, "Cannot open %s for reading. Does it exist?", path_db); 981 } 982 983 /* 984 * If the arguments on the command-line do not specify a file, 985 * then interpret it as a zoneinfo name 986 */ 987 if (optind == argc - 1) { 988 struct stat sb; 989 990 if (stat(argv[optind], &sb) != 0) { 991 #ifdef HAVE_DIALOG 992 usedialog = 0; 993 #endif 994 rv = install_zoneinfo(argv[optind]); 995 exit(rv & ~DITEM_LEAVE_MENU); 996 } 997 /* FALLTHROUGH */ 998 } 999 #ifdef HAVE_DIALOG 1000 1001 read_iso3166_table(); 1002 read_zones(); 1003 sort_countries(); 1004 make_menus(); 1005 1006 init_dialog(stdin, stdout); 1007 if (skiputc == 0) { 1008 DIALOG_VARS save_vars; 1009 int yesno; 1010 1011 snprintf(title, sizeof(title), 1012 "Select local or UTC (Greenwich Mean Time) clock"); 1013 snprintf(prompt, sizeof(prompt), 1014 "Is this machine's CMOS clock set to UTC? " 1015 "If it is set to local time,\n" 1016 "or you don't know, please choose NO here!"); 1017 dlg_save_vars(&save_vars); 1018 #if !defined(__sparc64__) 1019 dialog_vars.defaultno = TRUE; 1020 #endif 1021 yesno = dialog_yesno(title, prompt, 7, 73); 1022 dlg_restore_vars(&save_vars); 1023 if (!yesno) { 1024 if (reallydoit) 1025 unlink(path_wall_cmos_clock); 1026 } else { 1027 if (reallydoit) { 1028 fd = open(path_wall_cmos_clock, 1029 O_WRONLY | O_CREAT | O_TRUNC, 1030 S_IRUSR | S_IRGRP | S_IROTH); 1031 if (fd < 0) { 1032 end_dialog(); 1033 err(1, "create %s", 1034 path_wall_cmos_clock); 1035 } 1036 close(fd); 1037 } 1038 } 1039 dlg_clear(); 1040 } 1041 if (optind == argc - 1) { 1042 snprintf(title, sizeof(title), "Default timezone provided"); 1043 snprintf(prompt, sizeof(prompt), 1044 "\nUse the default `%s' zone?", argv[optind]); 1045 if (!dialog_yesno(title, prompt, 7, 72)) { 1046 rv = install_zoneinfo_file(argv[optind]); 1047 dlg_clear(); 1048 end_dialog(); 1049 exit(rv & ~DITEM_LEAVE_MENU); 1050 } 1051 dlg_clear(); 1052 } 1053 snprintf(title, sizeof(title), "Time Zone Selector"); 1054 snprintf(prompt, sizeof(prompt), "Select a region"); 1055 xdialog_menu(title, prompt, -1, -1, NCONTINENTS, NCONTINENTS, 1056 continents); 1057 1058 dlg_clear(); 1059 end_dialog(); 1060 #else 1061 usage(); 1062 #endif 1063 return (0); 1064 } 1065