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