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
42 static struct gprovider *
provider_for_name(struct gmesh * mesh,const char * name)43 provider_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
67 static int
part_config(char * disk,const char * scheme,char * config)68 part_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
140 finished:
141 geom_deletetree(&mesh);
142 free(disk);
143
144 return (0);
145 }
146
147 static
parse_disk_config(char * input)148 int 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
196 int
scripted_editor(int argc,const char ** argv)197 scripted_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