1/*
2 * CDDL HEADER START
3 *
4 * The contents of this file are subject to the terms of the
5 * Common Development and Distribution License (the "License").
6 * You may not use this file except in compliance with the License.
7 *
8 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9 * or http://www.opensolaris.org/os/licensing.
10 * See the License for the specific language governing permissions
11 * and limitations under the License.
12 *
13 * When distributing Covered Code, include this CDDL HEADER in each
14 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15 * If applicable, add the following below this CDDL HEADER, with the
16 * fields enclosed by brackets "[]" replaced with your own identifying
17 * information: Portions Copyright [yyyy] [name of copyright owner]
18 *
19 * CDDL HEADER END
20 */
21/*
22 * Copyright (c) 1998, 2010, Oracle and/or its affiliates. All rights reserved.
23 * Copyright 2015 Nexenta Systems, Inc. All rights reserved.
24 */
25
26#include <sys/stat.h>
27#include <locale.h>
28#include <unistd.h>
29#include <stdlib.h>
30#include <stdio.h>
31#include <string.h>
32
33#include "dconf.h"
34#include "minfree.h"
35#include "utils.h"
36
37static const char USAGE[] = "\
38Usage: %s [-enuy] [-c kernel | curproc | all ]\n\
39	[-d dump-device | swap | none ] [-m min {k|m|%%} ] [-s savecore-dir]\n\
40	[-r root-dir] [-z on|off]\n";
41
42static const char OPTS[] = "einuyc:d:m:s:r:z:";
43
44static const char PATH_DEVICE[] = "/dev/dump";
45static const char PATH_CONFIG[] = "/etc/dumpadm.conf";
46
47int
48main(int argc, char *argv[])
49{
50	const char *pname = getpname(argv[0]);
51
52	u_longlong_t minf;
53	struct stat st;
54	int c;
55	int dflag = 0;			/* for checking in use during -d ops */
56	int eflag = 0;			/* print estimated dump size */
57	int dcmode = DC_CURRENT;	/* kernel settings override unless -u */
58	int modified = 0;		/* have we modified the dump config? */
59	char *minfstr = NULL;		/* string value of -m argument */
60	dumpconf_t dc;			/* current configuration */
61	int chrooted = 0;
62	int douuid = 0;
63
64	(void) setlocale(LC_ALL, "");
65	(void) textdomain(TEXT_DOMAIN);
66
67	/*
68	 * Take an initial lap through argv hunting for -r root-dir,
69	 * so that we can chroot before opening the configuration file.
70	 * We also handle -u and any bad options at this point.
71	 */
72	while (optind < argc) {
73		while ((c = getopt(argc, argv, OPTS)) != (int)EOF) {
74			if (c == 'r' && ++chrooted && chroot(optarg) == -1)
75				die(gettext("failed to chroot to %s"), optarg);
76			else if (c == 'u')
77				dcmode = DC_OVERRIDE;
78			else if (c == '?') {
79				(void) fprintf(stderr, gettext(USAGE), pname);
80				return (E_USAGE);
81			}
82		}
83
84		if (optind < argc) {
85			warn(gettext("illegal argument -- %s\n"), argv[optind]);
86			(void) fprintf(stderr, gettext(USAGE), pname);
87			return (E_USAGE);
88		}
89	}
90
91	if (geteuid() != 0)
92		die(gettext("you must be root to use %s\n"), pname);
93
94	/*
95	 * If no config file exists yet, we're going to create an empty one,
96	 * so set the modified flag to force writing out the file.
97	 */
98	if (access(PATH_CONFIG, F_OK) == -1)
99		modified++;
100
101	/*
102	 * Now open and read in the initial values from the config file.
103	 * If it doesn't exist, we create an empty file and dc is
104	 * initialized with the default values.
105	 */
106	if (dconf_open(&dc, PATH_DEVICE, PATH_CONFIG, dcmode) == -1)
107		return (E_ERROR);
108
109	/*
110	 * Take another lap through argv, processing options and
111	 * modifying the dumpconf_t as appropriate.
112	 */
113	for (optind = 1; optind < argc; optind++) {
114		while ((c = getopt(argc, argv, OPTS)) != (int)EOF) {
115			switch (c) {
116			case 'c':
117				if (dconf_str2content(&dc, optarg) == -1)
118					return (E_USAGE);
119				modified++;
120				break;
121			case 'd':
122				if (dconf_str2device(&dc, optarg) == -1)
123					return (E_USAGE);
124				dflag++;
125				modified++;
126				break;
127			case 'e':
128				eflag++;
129				break;
130			case 'i':
131				/* undocumented option */
132				if (chrooted) {
133					warn(gettext("-i and -r cannot be "
134					    "used together\n"));
135					return (E_USAGE);
136				}
137				douuid++;
138				break;
139
140			case 'm':
141				minfstr = optarg;
142				break;
143
144			case 'n':
145				dc.dc_enable = DC_OFF;
146				modified++;
147				break;
148
149			case 's':
150				if (stat(optarg, &st) == -1 ||
151				    !S_ISDIR(st.st_mode)) {
152					warn(gettext("%s is missing or not a "
153					    "directory\n"), optarg);
154					return (E_USAGE);
155				}
156
157				if (dconf_str2savdir(&dc, optarg) == -1)
158					return (E_USAGE);
159				modified++;
160				break;
161
162			case 'y':
163				dc.dc_enable = DC_ON;
164				modified++;
165				break;
166
167			case 'z':
168				if (dconf_str2csave(&dc, optarg) == -1)
169					return (E_USAGE);
170				modified++;
171				break;
172			}
173		}
174	}
175
176	if (eflag) {
177		if (argc == 2 && argv[1][0] == '-' && argv[1][1] == 'e' &&
178		    !argv[1][2])
179			return (dconf_get_dumpsize(&dc) ? E_SUCCESS : E_ERROR);
180		else
181			die(gettext("-e cannot be used with other options\n"));
182	}
183
184	if (douuid)
185		return (dconf_write_uuid(&dc) ? E_SUCCESS : E_ERROR);
186
187	if (minfstr != NULL) {
188		if (minfree_compute(dc.dc_savdir, minfstr, &minf) == -1)
189			return (E_USAGE);
190		if (minfree_write(dc.dc_savdir, minf) == -1)
191			return (E_ERROR);
192	}
193
194	if (dcmode == DC_OVERRIDE) {
195		/*
196		 * In override mode, we try to force an update.  If this
197		 * fails, we re-load the kernel configuration and write that
198		 * out to the file in order to force the file in sync.
199		 *
200		 * We allow the file to be read-only but print a warning to the
201		 * user that indicates it hasn't been updated.
202		 */
203		if (dconf_update(&dc, 0) == -1)
204			(void) dconf_getdev(&dc);
205		if (dc.dc_readonly)
206			warn(gettext("kernel settings updated, but "
207			    "%s is read-only\n"), PATH_CONFIG);
208		else if (dconf_write(&dc) == -1)
209			return (E_ERROR);
210
211	} else if (modified) {
212		/*
213		 * If we're modifying the configuration, then try
214		 * to update it, and write out the file if successful.
215		 */
216		if (dc.dc_readonly) {
217			warn(gettext("failed to update settings: %s is "
218			    "read-only\n"), PATH_CONFIG);
219			return (E_ERROR);
220		}
221
222		if (dconf_update(&dc, dflag) == -1 ||
223		    dconf_write(&dc) == -1)
224			return (E_ERROR);
225	}
226
227	if (dcmode == DC_CURRENT)
228		dconf_print(&dc, stdout);
229
230	if (dconf_close(&dc) == -1)
231		warn(gettext("failed to close configuration file"));
232
233	return (E_SUCCESS);
234}
235