1/*-
2 * SPDX-License-Identifier: BSD-2-Clause-FreeBSD
3 *
4 * Copyright (c) 2013 Nathan Whitehorn
5 * All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 * 1. Redistributions of source code must retain the above copyright
11 *    notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 *    notice, this list of conditions and the following disclaimer in the
14 *    documentation and/or other materials provided with the distribution.
15 *
16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
17 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
18 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
19 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
20 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
21 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
22 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
23 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
24 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
25 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
26 * SUCH DAMAGE.
27 *
28 * $FreeBSD$
29 */
30
31#include <sys/param.h>
32#include <errno.h>
33#include <libutil.h>
34#include <inttypes.h>
35
36#include <libgeom.h>
37#include <dialog.h>
38#include <dlg_keys.h>
39
40#include "partedit.h"
41
42static struct gprovider *
43provider_for_name(struct gmesh *mesh, const char *name)
44{
45	struct gclass *classp;
46	struct gprovider *pp = NULL;
47	struct ggeom *gp;
48
49	LIST_FOREACH(classp, &mesh->lg_class, lg_class) {
50		LIST_FOREACH(gp, &classp->lg_geom, lg_geom) {
51			if (LIST_EMPTY(&gp->lg_provider))
52				continue;
53
54			LIST_FOREACH(pp, &gp->lg_provider, lg_provider)
55				if (strcmp(pp->lg_name, name) == 0)
56					break;
57
58			if (pp != NULL) break;
59		}
60
61		if (pp != NULL) break;
62	}
63
64	return (pp);
65}
66
67static int
68part_config(char *disk, const char *scheme, char *config)
69{
70	char *partition, *ap, *size = NULL, *type = NULL, *mount = NULL;
71	struct gclass *classp;
72	struct gmesh mesh;
73	struct ggeom *gpart = NULL;
74	int error;
75
76	if (scheme == NULL)
77		scheme = default_scheme();
78
79	error = geom_gettree(&mesh);
80	if (provider_for_name(&mesh, disk) == NULL) {
81		fprintf(stderr, "GEOM provider %s not found\n", disk);
82		geom_deletetree(&mesh);
83		return (-1);
84	}
85
86	/* Remove any existing partitioning and create new scheme */
87	LIST_FOREACH(classp, &mesh.lg_class, lg_class)
88		if (strcmp(classp->lg_name, "PART") == 0)
89			break;
90	if (classp != NULL) {
91		LIST_FOREACH(gpart, &classp->lg_geom, lg_geom)
92		if (strcmp(gpart->lg_name, disk) == 0)
93			break;
94	}
95	if (gpart != NULL)
96		gpart_destroy(gpart);
97	gpart_partition(disk, scheme);
98
99	if (strcmp(scheme, "MBR") == 0) {
100		struct gmesh submesh;
101		geom_gettree(&submesh);
102		gpart_create(provider_for_name(&submesh, disk),
103		    "freebsd", NULL, NULL, &disk, 0);
104		geom_deletetree(&submesh);
105	} else {
106		disk= strdup(disk);
107	}
108
109	geom_deletetree(&mesh);
110	error = geom_gettree(&mesh);
111
112	/* Create partitions */
113	if (config == NULL) {
114		wizard_makeparts(&mesh, disk, "ufs", 0);
115		goto finished;
116	}
117
118	while ((partition = strsep(&config, ",")) != NULL) {
119		while ((ap = strsep(&partition, " \t\n")) != NULL) {
120			if (*ap == '\0')
121				continue;
122			if (size == NULL)
123				size = ap;
124			else if (type == NULL)
125				type = ap;
126			else if (mount == NULL)
127				mount = ap;
128		}
129		if (size == NULL)
130			continue;
131		if (strcmp(size, "auto") == 0)
132			size = NULL;
133		gpart_create(provider_for_name(&mesh, disk), type, size, mount,
134		    NULL, 0);
135		geom_deletetree(&mesh);
136		error = geom_gettree(&mesh);
137		size = type = mount = NULL;
138	}
139
140finished:
141	geom_deletetree(&mesh);
142	free(disk);
143
144	return (0);
145}
146
147static
148int parse_disk_config(char *input)
149{
150	char *ap;
151	char *disk = NULL, *scheme = NULL, *partconfig = NULL;
152
153	while (input != NULL && *input != 0) {
154		if (isspace(*input)) {
155			input++;
156			continue;
157		}
158
159		switch(*input) {
160		case '{':
161			input++;
162			partconfig = strchr(input, '}');
163			if (partconfig == NULL) {
164				fprintf(stderr, "Malformed partition setup "
165				    "string: %s\n", input);
166				return (1);
167			}
168			*partconfig = '\0';
169			ap = partconfig+1;
170			partconfig = input;
171			input = ap;
172			break;
173		default:
174			if (disk == NULL)
175				disk = strsep(&input, " \t\n");
176			else if (scheme == NULL)
177				scheme = strsep(&input, " \t\n");
178			else {
179				fprintf(stderr, "Unknown directive: %s\n",
180				    strsep(&input, " \t\n"));
181				return (1);
182			}
183		}
184	} while (input != NULL && *input != 0);
185
186	if (disk == NULL || strcmp(disk, "DEFAULT") == 0) {
187		struct gmesh mesh;
188		geom_gettree(&mesh);
189		disk = boot_disk_select(&mesh);
190		geom_deletetree(&mesh);
191	}
192
193	return (part_config(disk, scheme, partconfig));
194}
195
196int
197scripted_editor(int argc, const char **argv)
198{
199	char *token;
200	int i, error = 0, len = 0;
201
202	for (i = 1; i < argc; i++)
203		len += strlen(argv[i]) + 1;
204	char inputbuf[len], *input = inputbuf;
205	strcpy(input, argv[1]);
206	for (i = 2; i < argc; i++) {
207		strcat(input, " ");
208		strcat(input, argv[i]);
209	}
210
211	while ((token = strsep(&input, ";")) != NULL) {
212		error = parse_disk_config(token);
213		if (error != 0)
214			return (error);
215	}
216
217	return (0);
218}
219
220