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: stable/11/usr.sbin/tzsetup/tzsetup.c 364903 2020-08-28 08:14:19Z philip $");
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
77static char	path_zonetab[MAXPATHLEN], path_iso3166[MAXPATHLEN],
78		path_zoneinfo[MAXPATHLEN], path_localtime[MAXPATHLEN],
79		path_db[MAXPATHLEN], path_wall_cmos_clock[MAXPATHLEN];
80
81static int reallydoit = 1;
82static int reinstall = 0;
83static char *chrootenv = NULL;
84
85static void	usage(void);
86static int	install_zoneinfo(const char *zoneinfo);
87static int	install_zoneinfo_file(const char *zoneinfo_file);
88
89#ifdef HAVE_DIALOG
90/* for use in describing more exotic behaviors */
91typedef struct dialogMenuItem {
92	char *prompt;
93	char *title;
94	int (*fire)(struct dialogMenuItem *self);
95	void *data;
96} dialogMenuItem;
97
98static int
99xdialog_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
113static int
114xdialog_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
130static int
131xdialog_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
175again:
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
206static int usedialog = 1;
207
208static int	confirm_zone(const char *filename);
209static int	continent_country_menu(dialogMenuItem *);
210static int	set_zone_multi(dialogMenuItem *);
211static int	set_zone_whole_country(dialogMenuItem *);
212static int	set_zone_menu(dialogMenuItem *);
213static int	set_zone_utc(void);
214
215struct continent {
216	dialogMenuItem *menu;
217	int		nitems;
218};
219
220static struct continent	africa, america, antarctica, asia, atlantic;
221static struct continent	australia, europe, indian, pacific, utc;
222
223static 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
239static 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])))
257static dialogMenuItem continents[NCONTINENTS];
258
259#define	OCEANP(x)	((x) == 4 || (x) == 7 || (x) == 8)
260
261static int
262continent_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
296static struct continent *
297find_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
307struct 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
317struct 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)
330static 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 */
338static void
339read_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
393static void
394add_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 */
450static int
451compare_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 */
469static void
470sort_countries(void)
471{
472
473	qsort(countries, NCOUNTRIES, sizeof(countries[0]), compare_countries);
474}
475
476static void
477read_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
530static void
531make_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
629static int
630set_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
648static int
649set_zone_utc(void)
650{
651	if (!confirm_zone("UTC"))
652		return (DITEM_FAILURE | DITEM_RECREATE);
653
654	return (install_zoneinfo("UTC"));
655}
656
657static int
658confirm_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
676static int
677set_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
689static int
690set_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
704static int
705install_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
871static int
872install_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
894static void
895usage(void)
896{
897
898	fprintf(stderr, "usage: tzsetup [-nrs] [-C chroot_directory]"
899	    " [zoneinfo_file | zoneinfo_name]\n");
900	exit(1);
901}
902
903int
904main(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}
1070