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