usr.sbin/tzsetup/tzsetup.c

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