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