label.c revision 1.28 1 1.28 martin /* $NetBSD: label.c,v 1.28 2020/10/10 19:42:19 martin Exp $ */
2 1.1 dholland
3 1.1 dholland /*
4 1.1 dholland * Copyright 1997 Jonathan Stone
5 1.1 dholland * All rights reserved.
6 1.1 dholland *
7 1.1 dholland * Redistribution and use in source and binary forms, with or without
8 1.1 dholland * modification, are permitted provided that the following conditions
9 1.1 dholland * are met:
10 1.1 dholland * 1. Redistributions of source code must retain the above copyright
11 1.1 dholland * notice, this list of conditions and the following disclaimer.
12 1.1 dholland * 2. Redistributions in binary form must reproduce the above copyright
13 1.1 dholland * notice, this list of conditions and the following disclaimer in the
14 1.1 dholland * documentation and/or other materials provided with the distribution.
15 1.1 dholland * 3. All advertising materials mentioning features or use of this software
16 1.1 dholland * must display the following acknowledgement:
17 1.1 dholland * This product includes software developed for the NetBSD Project by
18 1.1 dholland * Jonathan Stone.
19 1.1 dholland * 4. The name of Jonathan Stone may not be used to endorse
20 1.1 dholland * or promote products derived from this software without specific prior
21 1.1 dholland * written permission.
22 1.1 dholland *
23 1.1 dholland * THIS SOFTWARE IS PROVIDED BY JONATHAN STONE ``AS IS''
24 1.1 dholland * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
25 1.1 dholland * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
26 1.1 dholland * ARE DISCLAIMED. IN NO EVENT SHALL PIERMONT INFORMATION SYSTEMS INC. BE
27 1.1 dholland * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
28 1.1 dholland * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
29 1.1 dholland * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
30 1.1 dholland * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
31 1.1 dholland * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
32 1.1 dholland * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
33 1.1 dholland * THE POSSIBILITY OF SUCH DAMAGE.
34 1.1 dholland *
35 1.1 dholland */
36 1.1 dholland
37 1.1 dholland #include <sys/cdefs.h>
38 1.1 dholland #if defined(LIBC_SCCS) && !defined(lint)
39 1.28 martin __RCSID("$NetBSD: label.c,v 1.28 2020/10/10 19:42:19 martin Exp $");
40 1.1 dholland #endif
41 1.1 dholland
42 1.1 dholland #include <sys/types.h>
43 1.1 dholland #include <stddef.h>
44 1.7 martin #include <assert.h>
45 1.1 dholland #include <errno.h>
46 1.1 dholland #include <stdio.h>
47 1.1 dholland #include <fcntl.h>
48 1.1 dholland #include <util.h>
49 1.1 dholland #include <unistd.h>
50 1.1 dholland #include <sys/dkio.h>
51 1.1 dholland #include <sys/param.h>
52 1.7 martin #include <sys/bootblock.h>
53 1.1 dholland #include <ufs/ffs/fs.h>
54 1.1 dholland
55 1.1 dholland #include "defs.h"
56 1.1 dholland #include "msg_defs.h"
57 1.1 dholland #include "menu_defs.h"
58 1.1 dholland
59 1.1 dholland /*
60 1.1 dholland * local prototypes
61 1.1 dholland */
62 1.7 martin static bool boringpart(const struct disk_part_info *info);
63 1.17 martin static bool checklabel(struct disk_partitions*, char *, char *);
64 1.7 martin static void show_partition_adder(menudesc *, struct partition_usage_set*);
65 1.1 dholland
66 1.1 dholland /*
67 1.7 martin * Return 1 if a partition should be ignored when checking
68 1.1 dholland * for overlapping partitions.
69 1.1 dholland */
70 1.7 martin static bool
71 1.7 martin boringpart(const struct disk_part_info *info)
72 1.1 dholland {
73 1.1 dholland
74 1.7 martin if (info->size == 0)
75 1.7 martin return true;
76 1.7 martin if (info->flags &
77 1.7 martin (PTI_PSCHEME_INTERNAL|PTI_WHOLE_DISK|PTI_SEC_CONTAINER|
78 1.7 martin PTI_RAW_PART))
79 1.7 martin return true;
80 1.7 martin
81 1.7 martin return false;
82 1.1 dholland }
83 1.1 dholland
84 1.7 martin /*
85 1.7 martin * We have some partitions in our "wanted" list that we may not edit,
86 1.7 martin * like the RAW_PART in disklabel, some that just represent external
87 1.7 martin * mount entries for the final fstab or similar.
88 1.7 martin * We have previously sorted pset->parts and pset->infos to be in sync,
89 1.7 martin * but the former "array" may be shorter.
90 1.7 martin * Here are a few quick predicates to check for them.
91 1.7 martin */
92 1.7 martin static bool
93 1.7 martin real_partition(const struct partition_usage_set *pset, int index)
94 1.7 martin {
95 1.7 martin if (index < 0 || (size_t)index >= pset->num)
96 1.7 martin return false;
97 1.1 dholland
98 1.7 martin return pset->infos[index].cur_part_id != NO_PART;
99 1.7 martin }
100 1.1 dholland
101 1.1 dholland /*
102 1.7 martin * Check partitioning for overlapping partitions.
103 1.1 dholland * Returns 0 if no overlapping partition found, nonzero otherwise.
104 1.1 dholland * Sets reference arguments ovly1 and ovly2 to the indices of
105 1.1 dholland * overlapping partitions if any are found.
106 1.1 dholland */
107 1.7 martin static bool
108 1.7 martin checklabel(struct disk_partitions *parts,
109 1.17 martin char *ovl1, char *ovl2)
110 1.7 martin {
111 1.7 martin part_id i, j;
112 1.7 martin struct disk_part_info info;
113 1.7 martin daddr_t istart, iend, jstart, jend;
114 1.7 martin unsigned int fs_type, fs_sub_type;
115 1.1 dholland
116 1.7 martin for (i = 0; i < parts->num_part - 1; i ++ ) {
117 1.7 martin if (!parts->pscheme->get_part_info(parts, i, &info))
118 1.7 martin continue;
119 1.1 dholland
120 1.1 dholland /* skip unused or reserved partitions */
121 1.7 martin if (boringpart(&info))
122 1.1 dholland continue;
123 1.1 dholland
124 1.1 dholland /*
125 1.1 dholland * check succeeding partitions for overlap.
126 1.17 martin * O(n^2), but n is small.
127 1.1 dholland */
128 1.7 martin istart = info.start;
129 1.7 martin iend = istart + info.size;
130 1.7 martin fs_type = info.fs_type;
131 1.7 martin fs_sub_type = info.fs_sub_type;
132 1.1 dholland
133 1.7 martin for (j = i+1; j < parts->num_part; j++) {
134 1.7 martin
135 1.7 martin if (!parts->pscheme->get_part_info(parts, j, &info))
136 1.7 martin continue;
137 1.1 dholland
138 1.1 dholland /* skip unused or reserved partitions */
139 1.7 martin if (boringpart(&info))
140 1.1 dholland continue;
141 1.1 dholland
142 1.7 martin jstart = info.start;
143 1.7 martin jend = jstart + info.size;
144 1.1 dholland
145 1.1 dholland /* overlap? */
146 1.7 martin if ((istart <= jstart && jstart < iend) ||
147 1.7 martin (jstart <= istart && istart < jend)) {
148 1.17 martin snprintf(ovl1, MENUSTRSIZE,
149 1.7 martin "%" PRIu64 " - %" PRIu64 " %s, %s",
150 1.7 martin istart / sizemult, iend / sizemult,
151 1.7 martin multname,
152 1.7 martin getfslabelname(fs_type, fs_sub_type));
153 1.17 martin snprintf(ovl2, MENUSTRSIZE,
154 1.7 martin "%" PRIu64 " - %" PRIu64 " %s, %s",
155 1.7 martin jstart / sizemult, jend / sizemult,
156 1.7 martin multname,
157 1.7 martin getfslabelname(info.fs_type,
158 1.7 martin info.fs_sub_type));
159 1.7 martin return false;
160 1.1 dholland }
161 1.1 dholland }
162 1.1 dholland }
163 1.1 dholland
164 1.7 martin return true;
165 1.1 dholland }
166 1.1 dholland
167 1.2 martin int
168 1.7 martin checkoverlap(struct disk_partitions *parts)
169 1.2 martin {
170 1.7 martin char desc1[MENUSTRSIZE], desc2[MENUSTRSIZE];
171 1.7 martin if (!checklabel(parts, desc1, desc2)) {
172 1.7 martin msg_display_subst(MSG_partitions_overlap, 2, desc1, desc2);
173 1.2 martin return 1;
174 1.2 martin }
175 1.2 martin return 0;
176 1.2 martin }
177 1.2 martin
178 1.7 martin /*
179 1.7 martin * return (see post_edit_verify):
180 1.7 martin * 0 -> abort
181 1.7 martin * 1 -> re-edit
182 1.7 martin * 2 -> continue installation
183 1.7 martin */
184 1.1 dholland static int
185 1.18 martin verify_parts(struct partition_usage_set *pset, bool install)
186 1.1 dholland {
187 1.7 martin struct part_usage_info *wanted;
188 1.7 martin struct disk_partitions *parts;
189 1.7 martin size_t i, num_root;
190 1.28 martin daddr_t first_bsdstart, inst_start;
191 1.7 martin int rv;
192 1.7 martin
193 1.26 martin first_bsdstart = inst_start = -1;
194 1.7 martin num_root = 0;
195 1.7 martin parts = pset->parts;
196 1.7 martin for (i = 0; i < pset->num; i++) {
197 1.7 martin wanted = &pset->infos[i];
198 1.1 dholland
199 1.7 martin if (wanted->flags & PUIFLG_JUST_MOUNTPOINT)
200 1.1 dholland continue;
201 1.7 martin if (wanted->cur_part_id == NO_PART)
202 1.7 martin continue;
203 1.7 martin if (!(wanted->instflags & PUIINST_MOUNT))
204 1.1 dholland continue;
205 1.7 martin if (strcmp(wanted->mount, "/") != 0)
206 1.1 dholland continue;
207 1.7 martin num_root++;
208 1.7 martin
209 1.26 martin if (first_bsdstart <= 0) {
210 1.7 martin first_bsdstart = wanted->cur_start;
211 1.7 martin }
212 1.26 martin if (inst_start < 0 &&
213 1.26 martin (wanted->cur_flags & PTI_INSTALL_TARGET)) {
214 1.7 martin inst_start = wanted->cur_start;
215 1.7 martin }
216 1.7 martin }
217 1.7 martin
218 1.18 martin if ((num_root == 0 && install) ||
219 1.26 martin (num_root > 1 && inst_start < 0)) {
220 1.18 martin if (num_root == 0 && install)
221 1.7 martin msg_display_subst(MSG_must_be_one_root, 2,
222 1.7 martin msg_string(parts->pscheme->name),
223 1.7 martin msg_string(parts->pscheme->short_name));
224 1.7 martin else
225 1.7 martin msg_display_subst(MSG_multbsdpart, 2,
226 1.7 martin msg_string(parts->pscheme->name),
227 1.7 martin msg_string(parts->pscheme->short_name));
228 1.7 martin rv = ask_reedit(parts);
229 1.7 martin if (rv != 2)
230 1.7 martin return rv;
231 1.7 martin }
232 1.7 martin
233 1.7 martin /* Check for overlaps */
234 1.7 martin if (checkoverlap(parts) != 0) {
235 1.7 martin rv = ask_reedit(parts);
236 1.7 martin if (rv != 2)
237 1.7 martin return rv;
238 1.1 dholland }
239 1.7 martin
240 1.7 martin /*
241 1.7 martin * post_edit_verify returns:
242 1.7 martin * 0 -> abort
243 1.7 martin * 1 -> re-edit
244 1.7 martin * 2 -> continue installation
245 1.7 martin */
246 1.7 martin if (parts->pscheme->post_edit_verify)
247 1.7 martin return parts->pscheme->post_edit_verify(parts, false);
248 1.7 martin
249 1.7 martin return 2;
250 1.1 dholland }
251 1.1 dholland
252 1.1 dholland static int
253 1.1 dholland edit_fs_start(menudesc *m, void *arg)
254 1.1 dholland {
255 1.7 martin struct single_part_fs_edit *edit = arg;
256 1.7 martin daddr_t start, end;
257 1.1 dholland
258 1.7 martin start = getpartoff(edit->pset->parts, edit->info.start);
259 1.7 martin if (edit->info.size != 0) {
260 1.1 dholland /* Try to keep end in the same place */
261 1.7 martin end = edit->info.start + edit->info.size;
262 1.1 dholland if (end < start)
263 1.7 martin edit->info.size = edit->pset->parts->pscheme->
264 1.7 martin max_free_space_at(edit->pset->parts,
265 1.7 martin edit->info.start);
266 1.1 dholland else
267 1.7 martin edit->info.size = end - start;
268 1.1 dholland }
269 1.7 martin edit->info.start = start;
270 1.1 dholland return 0;
271 1.1 dholland }
272 1.1 dholland
273 1.1 dholland static int
274 1.1 dholland edit_fs_size(menudesc *m, void *arg)
275 1.1 dholland {
276 1.7 martin struct single_part_fs_edit *edit = arg;
277 1.25 martin struct disk_part_info pinfo;
278 1.7 martin daddr_t size;
279 1.1 dholland
280 1.25 martin /* get original partition data, in case start moved already */
281 1.25 martin edit->pset->parts->pscheme->get_part_info(edit->pset->parts,
282 1.25 martin edit->id, &pinfo);
283 1.25 martin /* ask for new size with old start and current values */
284 1.25 martin size = getpartsize(edit->pset->parts, pinfo.start,
285 1.25 martin edit->info.start, edit->info.size);
286 1.7 martin if (size < 0)
287 1.7 martin return 0;
288 1.7 martin if (size > edit->pset->parts->disk_size)
289 1.7 martin size = edit->pset->parts->disk_size - edit->info.start;
290 1.7 martin edit->info.size = size;
291 1.1 dholland return 0;
292 1.1 dholland }
293 1.1 dholland
294 1.7 martin static int
295 1.7 martin edit_fs_preserve(menudesc *m, void *arg)
296 1.1 dholland {
297 1.7 martin struct single_part_fs_edit *edit = arg;
298 1.1 dholland
299 1.7 martin edit->wanted->instflags ^= PUIINST_NEWFS;
300 1.7 martin return 0;
301 1.1 dholland }
302 1.1 dholland
303 1.7 martin static int
304 1.7 martin edit_install(menudesc *m, void *arg)
305 1.1 dholland {
306 1.7 martin struct single_part_fs_edit *edit = arg;
307 1.1 dholland
308 1.26 martin edit->info.flags ^= PTI_INSTALL_TARGET;
309 1.7 martin return 0;
310 1.1 dholland }
311 1.1 dholland
312 1.7 martin static int
313 1.7 martin edit_fs_mount(menudesc *m, void *arg)
314 1.1 dholland {
315 1.7 martin struct single_part_fs_edit *edit = arg;
316 1.1 dholland
317 1.7 martin edit->wanted->instflags ^= PUIINST_MOUNT;
318 1.7 martin return 0;
319 1.1 dholland }
320 1.1 dholland
321 1.1 dholland static int
322 1.7 martin edit_fs_mountpt(menudesc *m, void *arg)
323 1.1 dholland {
324 1.7 martin struct single_part_fs_edit *edit = arg;
325 1.7 martin char *p, *first, *last, buf[MOUNTLEN];
326 1.7 martin
327 1.7 martin strlcpy(buf, edit->wanted->mount, sizeof buf);
328 1.7 martin msg_prompt_win(MSG_mountpoint, -1, 18, 0, 0,
329 1.7 martin buf, buf, MOUNTLEN);
330 1.7 martin
331 1.7 martin /*
332 1.7 martin * Trim all leading and trailing whitespace
333 1.7 martin */
334 1.7 martin for (first = NULL, last = NULL, p = buf; *p; p++) {
335 1.7 martin if (isspace((unsigned char)*p))
336 1.7 martin continue;
337 1.7 martin if (first == NULL)
338 1.7 martin first = p;
339 1.7 martin last = p;
340 1.7 martin }
341 1.7 martin if (last != NULL)
342 1.7 martin last[1] = 0;
343 1.7 martin
344 1.21 martin if (first == NULL || *first == 0 || strcmp(first, "none") == 0) {
345 1.7 martin edit->wanted->mount[0] = 0;
346 1.7 martin edit->wanted->instflags &= ~PUIINST_MOUNT;
347 1.7 martin return 0;
348 1.7 martin }
349 1.7 martin
350 1.7 martin if (*first != '/') {
351 1.7 martin edit->wanted->mount[0] = '/';
352 1.7 martin strlcpy(&edit->wanted->mount[1], first,
353 1.7 martin sizeof(edit->wanted->mount)-1);
354 1.7 martin } else {
355 1.7 martin strlcpy(edit->wanted->mount, first, sizeof edit->wanted->mount);
356 1.7 martin }
357 1.1 dholland
358 1.1 dholland return 0;
359 1.7 martin }
360 1.1 dholland
361 1.1 dholland static int
362 1.7 martin edit_restore(menudesc *m, void *arg)
363 1.1 dholland {
364 1.7 martin struct single_part_fs_edit *edit = arg;
365 1.1 dholland
366 1.7 martin edit->info = edit->old_info;
367 1.7 martin *edit->wanted = edit->old_usage;
368 1.1 dholland return 0;
369 1.1 dholland }
370 1.1 dholland
371 1.1 dholland static int
372 1.7 martin edit_cancel(menudesc *m, void *arg)
373 1.1 dholland {
374 1.7 martin struct single_part_fs_edit *edit = arg;
375 1.1 dholland
376 1.7 martin edit->rv = -1;
377 1.7 martin return 1;
378 1.1 dholland }
379 1.1 dholland
380 1.1 dholland static int
381 1.7 martin edit_delete_ptn(menudesc *m, void *arg)
382 1.1 dholland {
383 1.7 martin struct single_part_fs_edit *edit = arg;
384 1.1 dholland
385 1.7 martin edit->rv = -2;
386 1.7 martin return 1;
387 1.7 martin }
388 1.7 martin
389 1.7 martin /*
390 1.7 martin * We have added/removed partitions, all cur_part_id values are
391 1.7 martin * out of sync. Re-fetch and reorder partitions accordingly.
392 1.7 martin */
393 1.7 martin static void
394 1.7 martin renumber_partitions(struct partition_usage_set *pset)
395 1.7 martin {
396 1.7 martin struct part_usage_info *ninfos;
397 1.7 martin struct disk_part_info info;
398 1.7 martin size_t i;
399 1.7 martin part_id pno;
400 1.7 martin
401 1.7 martin ninfos = calloc(pset->parts->num_part, sizeof(*ninfos));
402 1.7 martin if (ninfos == NULL) {
403 1.7 martin err_msg_win(err_outofmem);
404 1.7 martin return;
405 1.7 martin }
406 1.1 dholland
407 1.7 martin for (pno = 0; pno < pset->parts->num_part; pno++) {
408 1.7 martin if (!pset->parts->pscheme->get_part_info(pset->parts, pno,
409 1.7 martin &info))
410 1.7 martin continue;
411 1.7 martin for (i = 0; i < pset->parts->num_part; i++) {
412 1.7 martin if (pset->infos[i].cur_start != info.start)
413 1.7 martin continue;
414 1.18 martin if (pset->infos[i].cur_flags != info.flags)
415 1.18 martin continue;
416 1.18 martin if (pset->infos[i].type != info.nat_type->generic_ptype)
417 1.18 martin continue;
418 1.7 martin memcpy(&ninfos[pno], &pset->infos[i],
419 1.7 martin sizeof(ninfos[pno]));
420 1.7 martin ninfos[pno].cur_part_id = pno;
421 1.7 martin break;
422 1.7 martin }
423 1.1 dholland }
424 1.1 dholland
425 1.7 martin memcpy(pset->infos, ninfos, sizeof(*pset->infos)*pset->parts->num_part);
426 1.7 martin free(ninfos);
427 1.7 martin }
428 1.7 martin
429 1.7 martin /*
430 1.7 martin * Most often used file system types, we offer them in a first level menu.
431 1.7 martin */
432 1.7 martin static const uint edit_fs_common_types[] =
433 1.7 martin { FS_BSDFFS, FS_SWAP, FS_MSDOS, FS_BSDLFS, FS_EX2FS };
434 1.7 martin
435 1.7 martin /*
436 1.7 martin * Functions for uncommon file system types - we offer the full list,
437 1.7 martin * but put FFSv2 and FFSv1 at the front.
438 1.7 martin */
439 1.7 martin static void
440 1.7 martin init_fs_type_ext(menudesc *menu, void *arg)
441 1.7 martin {
442 1.7 martin struct single_part_fs_edit *edit = arg;
443 1.7 martin uint t = edit->info.fs_type;
444 1.7 martin size_t i, ndx, max = menu->numopts;
445 1.7 martin
446 1.7 martin if (t == FS_BSDFFS) {
447 1.7 martin if (edit->info.fs_sub_type == 2)
448 1.7 martin menu->cursel = 0;
449 1.7 martin else
450 1.7 martin menu->cursel = 1;
451 1.7 martin return;
452 1.15 martin } else if (t == FS_EX2FS && edit->info.fs_sub_type == 1) {
453 1.15 martin menu->cursel = FSMAXTYPES;
454 1.15 martin return;
455 1.1 dholland }
456 1.7 martin /* skip the two FFS entries, and do not add FFS later again */
457 1.7 martin for (ndx = 2, i = 0; i < FSMAXTYPES && ndx < max; i++) {
458 1.7 martin if (i == FS_UNUSED)
459 1.7 martin continue;
460 1.7 martin if (i == FS_BSDFFS)
461 1.7 martin continue;
462 1.7 martin if (fstypenames[i] == NULL)
463 1.7 martin continue;
464 1.1 dholland
465 1.7 martin if (i == t) {
466 1.7 martin menu->cursel = ndx;
467 1.7 martin break;
468 1.7 martin }
469 1.7 martin ndx++;
470 1.7 martin }
471 1.1 dholland }
472 1.1 dholland
473 1.1 dholland static int
474 1.7 martin set_fstype_ext(menudesc *menu, void *arg)
475 1.1 dholland {
476 1.7 martin struct single_part_fs_edit *edit = arg;
477 1.7 martin size_t i, ndx, max = menu->numopts;
478 1.16 martin enum part_type pt;
479 1.7 martin
480 1.7 martin if (menu->cursel == 0 || menu->cursel == 1) {
481 1.7 martin edit->info.fs_type = FS_BSDFFS;
482 1.7 martin edit->info.fs_sub_type = menu->cursel == 0 ? 2 : 1;
483 1.15 martin goto found_type;
484 1.15 martin } else if (menu->cursel == FSMAXTYPES) {
485 1.15 martin edit->info.fs_type = FS_EX2FS;
486 1.15 martin edit->info.fs_sub_type = 1;
487 1.15 martin goto found_type;
488 1.7 martin }
489 1.1 dholland
490 1.7 martin for (ndx = 2, i = 0; i < FSMAXTYPES && ndx < max; i++) {
491 1.7 martin if (i == FS_UNUSED)
492 1.7 martin continue;
493 1.7 martin if (i == FS_BSDFFS)
494 1.7 martin continue;
495 1.7 martin if (fstypenames[i] == NULL)
496 1.7 martin continue;
497 1.7 martin
498 1.7 martin if (ndx == (size_t)menu->cursel) {
499 1.7 martin edit->info.fs_type = i;
500 1.7 martin edit->info.fs_sub_type = 0;
501 1.15 martin goto found_type;
502 1.7 martin }
503 1.7 martin ndx++;
504 1.7 martin }
505 1.1 dholland return 1;
506 1.15 martin
507 1.15 martin found_type:
508 1.16 martin pt = edit->info.nat_type ? edit->info.nat_type->generic_ptype : PT_root;
509 1.15 martin edit->info.nat_type = edit->pset->parts->pscheme->
510 1.16 martin get_fs_part_type(pt, edit->info.fs_type, edit->info.fs_sub_type);
511 1.15 martin if (edit->info.nat_type == NULL)
512 1.15 martin edit->info.nat_type = edit->pset->parts->pscheme->
513 1.15 martin get_generic_part_type(PT_root);
514 1.15 martin edit->wanted->type = edit->info.nat_type->generic_ptype;
515 1.15 martin edit->wanted->fs_type = edit->info.fs_type;
516 1.15 martin edit->wanted->fs_version = edit->info.fs_sub_type;
517 1.15 martin return 1;
518 1.1 dholland }
519 1.1 dholland
520 1.7 martin /*
521 1.7 martin * Offer a menu with "exotic" file system types, start with FFSv2 and FFSv1,
522 1.7 martin * skip later FFS entry in the generic list.
523 1.7 martin */
524 1.1 dholland static int
525 1.7 martin edit_fs_type_ext(menudesc *menu, void *arg)
526 1.1 dholland {
527 1.7 martin menu_ent *opts;
528 1.7 martin int m;
529 1.7 martin size_t i, ndx, cnt;
530 1.7 martin
531 1.15 martin cnt = __arraycount(fstypenames);
532 1.7 martin opts = calloc(cnt, sizeof(*opts));
533 1.7 martin if (opts == NULL)
534 1.7 martin return 1;
535 1.7 martin
536 1.7 martin ndx = 0;
537 1.7 martin opts[ndx].opt_name = msg_string(MSG_fs_type_ffsv2);
538 1.7 martin opts[ndx].opt_action = set_fstype_ext;
539 1.7 martin ndx++;
540 1.7 martin opts[ndx].opt_name = msg_string(MSG_fs_type_ffs);
541 1.7 martin opts[ndx].opt_action = set_fstype_ext;
542 1.7 martin ndx++;
543 1.7 martin for (i = 0; i < FSMAXTYPES && ndx < cnt; i++) {
544 1.7 martin if (i == FS_UNUSED)
545 1.7 martin continue;
546 1.7 martin if (i == FS_BSDFFS)
547 1.7 martin continue;
548 1.7 martin if (fstypenames[i] == NULL)
549 1.7 martin continue;
550 1.7 martin opts[ndx].opt_name = fstypenames[i];
551 1.7 martin opts[ndx].opt_action = set_fstype_ext;
552 1.7 martin ndx++;
553 1.7 martin }
554 1.15 martin opts[ndx].opt_name = msg_string(MSG_fs_type_ext2old);
555 1.15 martin opts[ndx].opt_action = set_fstype_ext;
556 1.15 martin ndx++;
557 1.7 martin assert(ndx == cnt);
558 1.7 martin m = new_menu(MSG_Select_the_type, opts, ndx,
559 1.7 martin 30, 6, 10, 0, MC_SUBMENU | MC_SCROLL,
560 1.7 martin init_fs_type_ext, NULL, NULL, NULL, MSG_unchanged);
561 1.7 martin
562 1.7 martin if (m < 0)
563 1.7 martin return 1;
564 1.7 martin process_menu(m, arg);
565 1.7 martin free_menu(m);
566 1.7 martin free(opts);
567 1.1 dholland
568 1.1 dholland return 1;
569 1.1 dholland }
570 1.1 dholland
571 1.1 dholland static void
572 1.7 martin init_fs_type(menudesc *menu, void *arg)
573 1.7 martin {
574 1.7 martin struct single_part_fs_edit *edit = arg;
575 1.7 martin size_t i;
576 1.7 martin
577 1.7 martin /* init menu->cursel from fs type in arg */
578 1.7 martin if (edit->info.fs_type == FS_BSDFFS) {
579 1.7 martin if (edit->info.fs_sub_type == 2)
580 1.7 martin menu->cursel = 0;
581 1.7 martin else
582 1.7 martin menu->cursel = 1;
583 1.7 martin }
584 1.7 martin for (i = 1; i < __arraycount(edit_fs_common_types); i++) {
585 1.7 martin if (edit->info.fs_type == edit_fs_common_types[i]) {
586 1.7 martin menu->cursel = i+1;
587 1.7 martin break;
588 1.7 martin }
589 1.7 martin }
590 1.7 martin }
591 1.7 martin
592 1.7 martin static int
593 1.7 martin set_fstype(menudesc *menu, void *arg)
594 1.7 martin {
595 1.7 martin struct single_part_fs_edit *edit = arg;
596 1.16 martin enum part_type pt;
597 1.7 martin int ndx;
598 1.7 martin
599 1.16 martin pt = edit->info.nat_type ? edit->info.nat_type->generic_ptype : PT_root;
600 1.7 martin if (menu->cursel < 2) {
601 1.7 martin edit->info.fs_type = FS_BSDFFS;
602 1.7 martin edit->info.fs_sub_type = menu->cursel == 0 ? 2 : 1;
603 1.7 martin edit->info.nat_type = edit->pset->parts->pscheme->
604 1.16 martin get_fs_part_type(pt, FS_BSDFFS, 2);
605 1.7 martin if (edit->info.nat_type == NULL)
606 1.7 martin edit->info.nat_type = edit->pset->parts->
607 1.7 martin pscheme->get_generic_part_type(PT_root);
608 1.7 martin edit->wanted->type = edit->info.nat_type->generic_ptype;
609 1.7 martin edit->wanted->fs_type = edit->info.fs_type;
610 1.7 martin edit->wanted->fs_version = edit->info.fs_sub_type;
611 1.7 martin return 1;
612 1.7 martin }
613 1.7 martin ndx = menu->cursel-1;
614 1.7 martin
615 1.7 martin if (ndx < 0 ||
616 1.7 martin (size_t)ndx >= __arraycount(edit_fs_common_types))
617 1.7 martin return 1;
618 1.7 martin
619 1.7 martin edit->info.fs_type = edit_fs_common_types[ndx];
620 1.7 martin edit->info.fs_sub_type = 0;
621 1.7 martin edit->info.nat_type = edit->pset->parts->pscheme->
622 1.16 martin get_fs_part_type(pt, edit->info.fs_type, 0);
623 1.7 martin if (edit->info.nat_type == NULL)
624 1.7 martin edit->info.nat_type = edit->pset->parts->
625 1.7 martin pscheme->get_generic_part_type(PT_root);
626 1.7 martin edit->wanted->type = edit->info.nat_type->generic_ptype;
627 1.7 martin edit->wanted->fs_type = edit->info.fs_type;
628 1.7 martin edit->wanted->fs_version = edit->info.fs_sub_type;
629 1.7 martin return 1;
630 1.7 martin }
631 1.7 martin
632 1.7 martin /*
633 1.7 martin * Offer a menu selecting the common file system types
634 1.7 martin */
635 1.7 martin static int
636 1.7 martin edit_fs_type(menudesc *menu, void *arg)
637 1.1 dholland {
638 1.7 martin struct single_part_fs_edit *edit = arg;
639 1.7 martin menu_ent *opts;
640 1.7 martin int m, cnt;
641 1.7 martin size_t i;
642 1.7 martin
643 1.7 martin /*
644 1.7 martin * Shortcut to full menu if we have an exotic value
645 1.7 martin */
646 1.15 martin if (edit->info.fs_type == FS_EX2FS && edit->info.fs_sub_type == 1) {
647 1.15 martin edit_fs_type_ext(menu, arg);
648 1.15 martin return 0;
649 1.15 martin }
650 1.7 martin for (i = 0; i < __arraycount(edit_fs_common_types); i++)
651 1.7 martin if (edit->info.fs_type == edit_fs_common_types[i])
652 1.7 martin break;
653 1.7 martin if (i >= __arraycount(edit_fs_common_types)) {
654 1.7 martin edit_fs_type_ext(menu, arg);
655 1.7 martin return 0;
656 1.7 martin }
657 1.1 dholland
658 1.7 martin /*
659 1.7 martin * Starting with a common type, show short menu first
660 1.7 martin */
661 1.7 martin cnt = __arraycount(edit_fs_common_types) + 2;
662 1.7 martin opts = calloc(cnt, sizeof(*opts));
663 1.7 martin if (opts == NULL)
664 1.7 martin return 0;
665 1.7 martin
666 1.7 martin /* special case entry 0: two FFS entries */
667 1.7 martin for (i = 0; i < __arraycount(edit_fs_common_types); i++) {
668 1.7 martin opts[i+1].opt_name = getfslabelname(edit_fs_common_types[i], 0);
669 1.7 martin opts[i+1].opt_action = set_fstype;
670 1.7 martin }
671 1.7 martin /* duplicate FFS (at offset 1) into first entry */
672 1.7 martin opts[0] = opts[1];
673 1.7 martin opts[0].opt_name = msg_string(MSG_fs_type_ffsv2);
674 1.7 martin opts[1].opt_name = msg_string(MSG_fs_type_ffs);
675 1.7 martin /* add secondary sub-menu */
676 1.7 martin assert(i+1 < (size_t)cnt);
677 1.7 martin opts[i+1].opt_name = msg_string(MSG_other_fs_type);
678 1.7 martin opts[i+1].opt_action = edit_fs_type_ext;
679 1.7 martin
680 1.7 martin m = new_menu(MSG_Select_the_type, opts, cnt,
681 1.7 martin 30, 6, 0, 0, MC_SUBMENU | MC_SCROLL,
682 1.7 martin init_fs_type, NULL, NULL, NULL, MSG_unchanged);
683 1.7 martin
684 1.7 martin if (m < 0)
685 1.7 martin return 0;
686 1.7 martin process_menu(m, arg);
687 1.7 martin free_menu(m);
688 1.7 martin free(opts);
689 1.7 martin
690 1.7 martin return 0;
691 1.1 dholland }
692 1.1 dholland
693 1.7 martin
694 1.7 martin static void update_edit_ptn_menu(menudesc *m, void *arg);
695 1.7 martin static void draw_edit_ptn_line(menudesc *m, int opt, void *arg);
696 1.7 martin static int edit_ptn_custom_type(menudesc *m, void *arg);
697 1.1 dholland
698 1.24 martin static void
699 1.24 martin remember_deleted(struct partition_usage_set *pset,
700 1.24 martin struct disk_partitions *parts)
701 1.24 martin {
702 1.24 martin size_t i, num;
703 1.24 martin struct disk_partitions **tab;
704 1.24 martin
705 1.24 martin /* do we have parts on record already? */
706 1.24 martin for (i = 0; i < pset->num_write_back; i++)
707 1.24 martin if (pset->write_back[i] == parts)
708 1.24 martin return;
709 1.24 martin /*
710 1.24 martin * Need to record this partition table for write back
711 1.24 martin */
712 1.24 martin num = pset->num_write_back + 1;
713 1.24 martin tab = realloc(pset->write_back, num*sizeof(*pset->write_back));
714 1.24 martin if (!tab)
715 1.24 martin return;
716 1.24 martin tab[pset->num_write_back] = parts;
717 1.24 martin pset->write_back = tab;
718 1.24 martin pset->num_write_back = num;
719 1.24 martin }
720 1.24 martin
721 1.2 martin int
722 1.1 dholland edit_ptn(menudesc *menu, void *arg)
723 1.1 dholland {
724 1.7 martin struct partition_usage_set *pset = arg;
725 1.7 martin struct single_part_fs_edit edit;
726 1.7 martin int fspart_menu, num_opts;
727 1.7 martin const char *err;
728 1.7 martin menu_ent *mopts, *popt;
729 1.7 martin bool is_new_part, with_inst_opt = pset->parts->parent == NULL;
730 1.7 martin
731 1.7 martin static const menu_ent edit_ptn_fields_head[] = {
732 1.8 christos { .opt_action=edit_fs_type },
733 1.8 christos { .opt_action=edit_fs_start },
734 1.8 christos { .opt_action=edit_fs_size },
735 1.8 christos { .opt_flags=OPT_IGNORE },
736 1.7 martin };
737 1.7 martin
738 1.7 martin static const menu_ent edit_ptn_fields_head_add[] = {
739 1.8 christos { .opt_action=edit_install },
740 1.7 martin };
741 1.7 martin
742 1.7 martin static const menu_ent edit_ptn_fields_head2[] = {
743 1.8 christos { .opt_action=edit_fs_preserve },
744 1.8 christos { .opt_action=edit_fs_mount },
745 1.7 martin { .opt_menu=MENU_mountoptions, .opt_flags=OPT_SUB },
746 1.8 christos { .opt_action=edit_fs_mountpt },
747 1.7 martin };
748 1.7 martin static const menu_ent edit_ptn_fields_tail[] = {
749 1.7 martin { .opt_name=MSG_askunits, .opt_menu=MENU_sizechoice,
750 1.7 martin .opt_flags=OPT_SUB },
751 1.8 christos { .opt_name=MSG_restore,
752 1.7 martin .opt_action=edit_restore},
753 1.8 christos { .opt_name=MSG_Delete_partition,
754 1.7 martin .opt_action=edit_delete_ptn},
755 1.8 christos { .opt_name=MSG_cancel,
756 1.7 martin .opt_action=edit_cancel},
757 1.1 dholland };
758 1.7 martin
759 1.7 martin memset(&edit, 0, sizeof edit);
760 1.7 martin edit.pset = pset;
761 1.7 martin edit.index = menu->cursel;
762 1.7 martin edit.wanted = &pset->infos[edit.index];
763 1.7 martin
764 1.7 martin if (menu->cursel < 0 || (size_t)menu->cursel > pset->parts->num_part)
765 1.7 martin return 0;
766 1.7 martin is_new_part = (size_t)menu->cursel == pset->parts->num_part;
767 1.7 martin
768 1.7 martin num_opts = __arraycount(edit_ptn_fields_head) +
769 1.7 martin __arraycount(edit_ptn_fields_head2) +
770 1.7 martin __arraycount(edit_ptn_fields_tail);
771 1.7 martin if (with_inst_opt)
772 1.7 martin num_opts += __arraycount(edit_ptn_fields_head_add);
773 1.7 martin if (is_new_part)
774 1.7 martin num_opts--;
775 1.7 martin else
776 1.7 martin num_opts += pset->parts->pscheme->custom_attribute_count;
777 1.7 martin
778 1.7 martin mopts = calloc(num_opts, sizeof(*mopts));
779 1.7 martin if (mopts == NULL) {
780 1.7 martin err_msg_win(err_outofmem);
781 1.7 martin return 0;
782 1.7 martin }
783 1.7 martin memcpy(mopts, edit_ptn_fields_head, sizeof(edit_ptn_fields_head));
784 1.7 martin popt = mopts + __arraycount(edit_ptn_fields_head);
785 1.7 martin if (with_inst_opt) {
786 1.7 martin memcpy(popt, edit_ptn_fields_head_add,
787 1.7 martin sizeof(edit_ptn_fields_head_add));
788 1.7 martin popt += __arraycount(edit_ptn_fields_head_add);
789 1.7 martin }
790 1.7 martin memcpy(popt, edit_ptn_fields_head2, sizeof(edit_ptn_fields_head2));
791 1.7 martin popt += __arraycount(edit_ptn_fields_head2);
792 1.7 martin edit.first_custom_attr = popt - mopts;
793 1.7 martin if (!is_new_part) {
794 1.7 martin for (size_t i = 0;
795 1.7 martin i < pset->parts->pscheme->custom_attribute_count;
796 1.7 martin i++, popt++) {
797 1.7 martin popt->opt_action = edit_ptn_custom_type;
798 1.7 martin }
799 1.7 martin }
800 1.7 martin memcpy(popt, edit_ptn_fields_tail, sizeof(edit_ptn_fields_tail));
801 1.7 martin popt += __arraycount(edit_ptn_fields_tail) - 1;
802 1.7 martin if (is_new_part)
803 1.7 martin memcpy(popt-1, popt, sizeof(*popt));
804 1.7 martin
805 1.7 martin if (is_new_part) {
806 1.7 martin struct disk_part_free_space space;
807 1.7 martin daddr_t align = pset->parts->pscheme->get_part_alignment(
808 1.7 martin pset->parts);
809 1.7 martin
810 1.7 martin edit.id = NO_PART;
811 1.7 martin if (pset->parts->pscheme->get_free_spaces(pset->parts,
812 1.7 martin &space, 1, align, align, -1, -1) == 1) {
813 1.7 martin edit.info.start = space.start;
814 1.7 martin edit.info.size = space.size;
815 1.7 martin edit.info.fs_type = FS_BSDFFS;
816 1.7 martin edit.info.fs_sub_type = 2;
817 1.7 martin edit.info.nat_type = pset->parts->pscheme->
818 1.16 martin get_fs_part_type(PT_root, edit.info.fs_type,
819 1.7 martin edit.info.fs_sub_type);
820 1.7 martin edit.wanted->instflags = PUIINST_NEWFS;
821 1.7 martin }
822 1.7 martin } else {
823 1.7 martin edit.id = pset->infos[edit.index].cur_part_id;
824 1.7 martin if (!pset->parts->pscheme->get_part_info(pset->parts, edit.id,
825 1.7 martin &edit.info)) {
826 1.7 martin free(mopts);
827 1.7 martin return 0;
828 1.7 martin }
829 1.7 martin }
830 1.7 martin
831 1.7 martin edit.old_usage = *edit.wanted;
832 1.7 martin edit.old_info = edit.info;
833 1.7 martin
834 1.7 martin fspart_menu = new_menu(MSG_edfspart, mopts, num_opts,
835 1.7 martin 15, 2, 0, 55, MC_NOCLEAR | MC_SCROLL,
836 1.7 martin update_edit_ptn_menu, draw_edit_ptn_line, NULL,
837 1.7 martin NULL, MSG_OK);
838 1.7 martin
839 1.7 martin process_menu(fspart_menu, &edit);
840 1.7 martin free(mopts);
841 1.7 martin free_menu(fspart_menu);
842 1.7 martin
843 1.7 martin if (edit.rv == 0) { /* OK, set new data */
844 1.7 martin edit.info.last_mounted = edit.wanted->mount;
845 1.7 martin if (is_new_part) {
846 1.22 martin edit.wanted->parts = pset->parts;
847 1.7 martin edit.wanted->cur_part_id = pset->parts->pscheme->
848 1.7 martin add_partition(pset->parts, &edit.info, &err);
849 1.7 martin if (edit.wanted->cur_part_id == NO_PART)
850 1.7 martin err_msg_win(err);
851 1.7 martin else {
852 1.7 martin pset->parts->pscheme->get_part_info(
853 1.7 martin pset->parts, edit.wanted->cur_part_id,
854 1.7 martin &edit.info);
855 1.7 martin edit.wanted->cur_start = edit.info.start;
856 1.7 martin edit.wanted->size = edit.info.size;
857 1.7 martin edit.wanted->type =
858 1.7 martin edit.info.nat_type->generic_ptype;
859 1.7 martin edit.wanted->fs_type = edit.info.fs_type;
860 1.7 martin edit.wanted->fs_version = edit.info.fs_sub_type;
861 1.7 martin /* things have changed, re-sort */
862 1.7 martin renumber_partitions(pset);
863 1.7 martin }
864 1.7 martin } else {
865 1.7 martin if (!pset->parts->pscheme->set_part_info(pset->parts,
866 1.7 martin edit.id, &edit.info, &err))
867 1.7 martin err_msg_win(err);
868 1.7 martin }
869 1.7 martin
870 1.7 martin /*
871 1.7 martin * if size has changed, we may need to add or remove
872 1.7 martin * the option to add partitions
873 1.7 martin */
874 1.7 martin show_partition_adder(menu, pset);
875 1.7 martin } else if (edit.rv == -1) { /* cancel edit */
876 1.7 martin if (is_new_part) {
877 1.7 martin memmove(pset->infos+edit.index,
878 1.7 martin pset->infos+edit.index+1,
879 1.7 martin sizeof(*pset->infos)*(pset->num-edit.index));
880 1.7 martin memmove(menu->opts+edit.index,
881 1.7 martin menu->opts+edit.index+1,
882 1.7 martin sizeof(*menu->opts)*(menu->numopts-edit.index));
883 1.7 martin menu->numopts--;
884 1.7 martin menu->cursel = 0;
885 1.7 martin pset->num--;
886 1.7 martin return -1;
887 1.7 martin }
888 1.7 martin pset->infos[edit.index] = edit.old_usage;
889 1.7 martin } else if (!is_new_part && edit.rv == -2) { /* delete partition */
890 1.7 martin if (!pset->parts->pscheme->delete_partition(pset->parts,
891 1.7 martin edit.id, &err)) {
892 1.7 martin err_msg_win(err);
893 1.7 martin return 0;
894 1.7 martin }
895 1.24 martin remember_deleted(pset,
896 1.24 martin pset->infos[edit.index].parts);
897 1.7 martin memmove(pset->infos+edit.index,
898 1.7 martin pset->infos+edit.index+1,
899 1.7 martin sizeof(*pset->infos)*(pset->num-edit.index));
900 1.7 martin memmove(menu->opts+edit.index,
901 1.7 martin menu->opts+edit.index+1,
902 1.7 martin sizeof(*menu->opts)*(menu->numopts-edit.index));
903 1.7 martin menu->numopts--;
904 1.7 martin menu->cursel = 0;
905 1.23 martin if (pset->parts->num_part == 0)
906 1.23 martin menu->cursel = 1; /* skip sentinel line */
907 1.7 martin
908 1.7 martin /* things have changed, re-sort */
909 1.7 martin pset->num--;
910 1.7 martin renumber_partitions(pset);
911 1.7 martin
912 1.7 martin /* we can likely add new partitions now */
913 1.7 martin show_partition_adder(menu, pset);
914 1.7 martin
915 1.7 martin return -1;
916 1.1 dholland }
917 1.1 dholland
918 1.1 dholland return 0;
919 1.1 dholland }
920 1.1 dholland
921 1.1 dholland static void
922 1.7 martin update_edit_ptn_menu(menudesc *m, void *arg)
923 1.1 dholland {
924 1.7 martin struct single_part_fs_edit *edit = arg;
925 1.1 dholland int i;
926 1.7 martin uint t = edit->info.fs_type;
927 1.7 martin size_t attr_no;
928 1.1 dholland
929 1.7 martin /* Determine which of the properties can be changed */
930 1.7 martin for (i = 0; i < m->numopts; i++) {
931 1.7 martin if (m->opts[i].opt_action == NULL &&
932 1.7 martin m->opts[i].opt_menu != MENU_mountoptions)
933 1.7 martin continue;
934 1.1 dholland
935 1.1 dholland /* Default to disabled... */
936 1.1 dholland m->opts[i].opt_flags |= OPT_IGNORE;
937 1.7 martin if ((t == FS_UNUSED || t == FS_SWAP) &&
938 1.7 martin (m->opts[i].opt_action == edit_fs_preserve ||
939 1.7 martin m->opts[i].opt_action == edit_fs_mount ||
940 1.7 martin m->opts[i].opt_action == edit_fs_mountpt ||
941 1.7 martin m->opts[i].opt_menu == MENU_mountoptions))
942 1.1 dholland continue;
943 1.7 martin if (m->opts[i].opt_action == edit_install &&
944 1.7 martin edit->info.nat_type &&
945 1.7 martin edit->info.nat_type->generic_ptype != PT_root)
946 1.7 martin /* can only install onto PT_root partitions */
947 1.7 martin continue;
948 1.7 martin if (m->opts[i].opt_action == edit_fs_preserve &&
949 1.9 martin t != FS_BSDFFS && t != FS_BSDLFS && t != FS_APPLEUFS &&
950 1.9 martin t != FS_MSDOS && t != FS_EX2FS) {
951 1.9 martin /* Can not newfs this filesystem */
952 1.7 martin edit->wanted->instflags &= ~PUIINST_NEWFS;
953 1.1 dholland continue;
954 1.1 dholland }
955 1.7 martin if (m->opts[i].opt_action == edit_ptn_custom_type) {
956 1.7 martin attr_no = (size_t)i - edit->first_custom_attr;
957 1.7 martin if (!edit->pset->parts->pscheme->
958 1.7 martin custom_attribute_writable(
959 1.7 martin edit->pset->parts, edit->id, attr_no))
960 1.1 dholland continue;
961 1.1 dholland }
962 1.1 dholland /* Ok: we want this one */
963 1.1 dholland m->opts[i].opt_flags &= ~OPT_IGNORE;
964 1.1 dholland }
965 1.1 dholland }
966 1.1 dholland
967 1.1 dholland static void
968 1.7 martin draw_edit_ptn_line(menudesc *m, int opt, void *arg)
969 1.1 dholland {
970 1.7 martin struct single_part_fs_edit *edit = arg;
971 1.7 martin static int col_width;
972 1.7 martin static const char *ptn_type, *ptn_start, *ptn_size, *ptn_end,
973 1.7 martin *ptn_newfs, *ptn_mount, *ptn_mount_options, *ptn_mountpt,
974 1.7 martin *ptn_install;
975 1.7 martin const char *c;
976 1.7 martin char val[MENUSTRSIZE];
977 1.7 martin const char *attrname;
978 1.7 martin size_t attr_no;
979 1.7 martin
980 1.7 martin if (col_width == 0) {
981 1.7 martin int l;
982 1.7 martin
983 1.7 martin #define LOAD(STR) STR = msg_string(MSG_##STR); l = strlen(STR); \
984 1.7 martin if (l > col_width) col_width = l
985 1.7 martin
986 1.7 martin LOAD(ptn_type);
987 1.7 martin LOAD(ptn_start);
988 1.7 martin LOAD(ptn_size);
989 1.7 martin LOAD(ptn_end);
990 1.7 martin LOAD(ptn_install);
991 1.7 martin LOAD(ptn_newfs);
992 1.7 martin LOAD(ptn_mount);
993 1.7 martin LOAD(ptn_mount_options);
994 1.7 martin LOAD(ptn_mountpt);
995 1.7 martin #undef LOAD
996 1.7 martin
997 1.7 martin for (size_t i = 0;
998 1.7 martin i < edit->pset->parts->pscheme->custom_attribute_count;
999 1.7 martin i++) {
1000 1.7 martin attrname = msg_string(
1001 1.7 martin edit->pset->parts->pscheme->custom_attributes[i]
1002 1.7 martin .label);
1003 1.7 martin l = strlen(attrname);
1004 1.7 martin if (l > col_width) col_width = l;
1005 1.7 martin }
1006 1.1 dholland
1007 1.7 martin col_width += 3;
1008 1.7 martin }
1009 1.1 dholland
1010 1.1 dholland if (m->opts[opt].opt_flags & OPT_IGNORE
1011 1.7 martin && (opt != 3 || edit->info.fs_type == FS_UNUSED)
1012 1.7 martin && m->opts[opt].opt_action != edit_ptn_custom_type) {
1013 1.7 martin wprintw(m->mw, "%*s -", col_width, "");
1014 1.1 dholland return;
1015 1.1 dholland }
1016 1.1 dholland
1017 1.7 martin if (opt < 4) {
1018 1.7 martin switch (opt) {
1019 1.7 martin case 0:
1020 1.7 martin if (edit->info.fs_type == FS_BSDFFS)
1021 1.7 martin if (edit->info.fs_sub_type == 2)
1022 1.7 martin c = msg_string(MSG_fs_type_ffsv2);
1023 1.7 martin else
1024 1.7 martin c = msg_string(MSG_fs_type_ffs);
1025 1.1 dholland else
1026 1.7 martin c = getfslabelname(edit->info.fs_type,
1027 1.7 martin edit->info.fs_sub_type);
1028 1.7 martin wprintw(m->mw, "%*s : %s", col_width, ptn_type, c);
1029 1.7 martin return;
1030 1.7 martin case 1:
1031 1.7 martin wprintw(m->mw, "%*s : %" PRIu64 " %s", col_width,
1032 1.7 martin ptn_start, edit->info.start / sizemult, multname);
1033 1.7 martin return;
1034 1.7 martin case 2:
1035 1.7 martin wprintw(m->mw, "%*s : %" PRIu64 " %s", col_width,
1036 1.7 martin ptn_size, edit->info.size / sizemult, multname);
1037 1.7 martin return;
1038 1.7 martin case 3:
1039 1.7 martin wprintw(m->mw, "%*s : %" PRIu64 " %s", col_width,
1040 1.7 martin ptn_end, (edit->info.start + edit->info.size)
1041 1.7 martin / sizemult, multname);
1042 1.7 martin return;
1043 1.7 martin }
1044 1.7 martin }
1045 1.7 martin if (m->opts[opt].opt_action == edit_install) {
1046 1.7 martin wprintw(m->mw, "%*s : %s", col_width, ptn_install,
1047 1.26 martin msg_string((edit->info.flags & PTI_INSTALL_TARGET)
1048 1.7 martin ? MSG_Yes : MSG_No));
1049 1.7 martin return;
1050 1.7 martin }
1051 1.7 martin if (m->opts[opt].opt_action == edit_fs_preserve) {
1052 1.7 martin wprintw(m->mw, "%*s : %s", col_width, ptn_newfs,
1053 1.7 martin msg_string(edit->wanted->instflags & PUIINST_NEWFS
1054 1.7 martin ? MSG_Yes : MSG_No));
1055 1.7 martin return;
1056 1.7 martin }
1057 1.7 martin if (m->opts[opt].opt_action == edit_fs_mount) {
1058 1.7 martin wprintw(m->mw, "%*s : %s", col_width, ptn_mount,
1059 1.7 martin msg_string(edit->wanted->instflags & PUIINST_MOUNT
1060 1.7 martin ? MSG_Yes : MSG_No));
1061 1.7 martin return;
1062 1.7 martin }
1063 1.7 martin if (m->opts[opt].opt_menu == MENU_mountoptions) {
1064 1.7 martin wprintw(m->mw, "%*s : ", col_width, ptn_mount_options);
1065 1.7 martin if (edit->wanted->mountflags & PUIMNT_ASYNC)
1066 1.1 dholland wprintw(m->mw, "async ");
1067 1.7 martin if (edit->wanted->mountflags & PUIMNT_NOATIME)
1068 1.1 dholland wprintw(m->mw, "noatime ");
1069 1.7 martin if (edit->wanted->mountflags & PUIMNT_NODEV)
1070 1.1 dholland wprintw(m->mw, "nodev ");
1071 1.7 martin if (edit->wanted->mountflags & PUIMNT_NODEVMTIME)
1072 1.1 dholland wprintw(m->mw, "nodevmtime ");
1073 1.7 martin if (edit->wanted->mountflags & PUIMNT_NOEXEC)
1074 1.1 dholland wprintw(m->mw, "noexec ");
1075 1.7 martin if (edit->wanted->mountflags & PUIMNT_NOSUID)
1076 1.1 dholland wprintw(m->mw, "nosuid ");
1077 1.7 martin if (edit->wanted->mountflags & PUIMNT_LOG)
1078 1.1 dholland wprintw(m->mw, "log ");
1079 1.7 martin if (edit->wanted->mountflags & PUIMNT_NOAUTO)
1080 1.7 martin wprintw(m->mw, "noauto ");
1081 1.7 martin return;
1082 1.7 martin }
1083 1.7 martin if (m->opts[opt].opt_action == edit_fs_mountpt) {
1084 1.7 martin wprintw(m->mw, "%*s : %s", col_width, ptn_mountpt,
1085 1.7 martin edit->wanted->mount);
1086 1.7 martin return;
1087 1.7 martin }
1088 1.7 martin
1089 1.7 martin attr_no = opt - edit->first_custom_attr;
1090 1.7 martin edit->pset->parts->pscheme->format_custom_attribute(
1091 1.7 martin edit->pset->parts, edit->id, attr_no, &edit->info,
1092 1.7 martin val, sizeof val);
1093 1.7 martin attrname = msg_string(edit->pset->parts->pscheme->
1094 1.7 martin custom_attributes[attr_no].label);
1095 1.7 martin wprintw(m->mw, "%*s : %s", col_width, attrname, val);
1096 1.7 martin }
1097 1.7 martin
1098 1.7 martin static int
1099 1.7 martin edit_ptn_custom_type(menudesc *m, void *arg)
1100 1.7 martin {
1101 1.7 martin struct single_part_fs_edit *edit = arg;
1102 1.7 martin size_t attr_no = m->cursel - edit->first_custom_attr;
1103 1.7 martin char line[STRSIZE];
1104 1.7 martin
1105 1.7 martin switch (edit->pset->parts->pscheme->custom_attributes[attr_no].type) {
1106 1.7 martin case pet_bool:
1107 1.7 martin edit->pset->parts->pscheme->custom_attribute_toggle(
1108 1.7 martin edit->pset->parts, edit->id, attr_no);
1109 1.1 dholland break;
1110 1.7 martin case pet_cardinal:
1111 1.7 martin case pet_str:
1112 1.7 martin edit->pset->parts->pscheme->format_custom_attribute(
1113 1.7 martin edit->pset->parts, edit->id, attr_no, &edit->info,
1114 1.7 martin line, sizeof(line));
1115 1.7 martin msg_prompt_win(
1116 1.7 martin edit->pset->parts->pscheme->custom_attributes[attr_no].
1117 1.7 martin label, -1, 18, 0, 0, line, line, sizeof(line));
1118 1.7 martin edit->pset->parts->pscheme->custom_attribute_set_str(
1119 1.7 martin edit->pset->parts, edit->id, attr_no, line);
1120 1.1 dholland break;
1121 1.1 dholland }
1122 1.7 martin
1123 1.7 martin return 0;
1124 1.7 martin }
1125 1.7 martin
1126 1.7 martin
1127 1.7 martin /*
1128 1.7 martin * Some column width depend on translation, we will set these in
1129 1.7 martin * fmt_fspart_header and later use it when formatting single entries
1130 1.7 martin * in fmt_fspart_row.
1131 1.7 martin * The table consist of 3 "size like" columns, all fixed width, then
1132 1.7 martin * ptnheaders_fstype, part_header_col_flag, and finally the mount point
1133 1.7 martin * (which is variable width).
1134 1.7 martin */
1135 1.7 martin static int fstype_width, flags_width;
1136 1.7 martin static char fspart_separator[MENUSTRSIZE];
1137 1.7 martin static char fspart_title[2*MENUSTRSIZE];
1138 1.7 martin
1139 1.7 martin /*
1140 1.7 martin * Format the header of the main partition editor menu.
1141 1.7 martin */
1142 1.7 martin static void
1143 1.7 martin fmt_fspart_header(menudesc *menu, void *arg)
1144 1.7 martin {
1145 1.7 martin struct partition_usage_set *pset = arg;
1146 1.7 martin char total[6], free_space[6], scol[13], ecol[13], szcol[13],
1147 1.13 martin sepline[MENUSTRSIZE], *p, desc[MENUSTRSIZE];
1148 1.7 martin const char *fstype, *flags;
1149 1.7 martin int i;
1150 1.13 martin size_t ptn;
1151 1.13 martin bool with_clone, with_inst_flag = pset->parts->parent == NULL;
1152 1.7 martin
1153 1.13 martin with_clone = false;
1154 1.13 martin for (ptn = 0; ptn < pset->num && !with_clone; ptn++)
1155 1.13 martin if (pset->infos[ptn].flags & PUIFLG_CLONE_PARTS)
1156 1.13 martin with_clone = true;
1157 1.7 martin humanize_number(total, sizeof total,
1158 1.20 martin pset->parts->disk_size * pset->parts->bytes_per_sector,
1159 1.7 martin "", HN_AUTOSCALE, HN_B | HN_NOSPACE | HN_DECIMAL);
1160 1.7 martin humanize_number(free_space, sizeof free_space,
1161 1.20 martin pset->cur_free_space * pset->parts->bytes_per_sector,
1162 1.7 martin "", HN_AUTOSCALE, HN_B | HN_NOSPACE | HN_DECIMAL);
1163 1.7 martin
1164 1.13 martin if (with_clone)
1165 1.13 martin strlcpy(desc, msg_string(MSG_clone_flag_desc), sizeof desc);
1166 1.13 martin else
1167 1.13 martin desc[0] = 0;
1168 1.13 martin if (pset->parts->pscheme->part_flag_desc)
1169 1.13 martin strlcat(desc, msg_string(pset->parts->pscheme->part_flag_desc),
1170 1.13 martin sizeof desc);
1171 1.13 martin
1172 1.7 martin msg_display_subst(MSG_fspart, 7, pset->parts->disk,
1173 1.7 martin msg_string(pset->parts->pscheme->name),
1174 1.7 martin msg_string(pset->parts->pscheme->short_name),
1175 1.7 martin with_inst_flag ? msg_string(MSG_ptn_instflag_desc) : "",
1176 1.13 martin desc, total, free_space);
1177 1.7 martin
1178 1.7 martin snprintf(scol, sizeof scol, "%s (%s)",
1179 1.7 martin msg_string(MSG_ptnheaders_start), multname);
1180 1.7 martin snprintf(ecol, sizeof ecol, "%s (%s)",
1181 1.7 martin msg_string(MSG_ptnheaders_end), multname);
1182 1.7 martin snprintf(szcol, sizeof szcol, "%s (%s)",
1183 1.7 martin msg_string(MSG_ptnheaders_size), multname);
1184 1.7 martin
1185 1.7 martin fstype = msg_string(MSG_ptnheaders_fstype);
1186 1.7 martin flags = msg_string(MSG_part_header_col_flag);
1187 1.7 martin fstype_width = max(strlen(fstype), 8);
1188 1.7 martin flags_width = strlen(flags);
1189 1.7 martin for (i = 0, p = sepline; i < fstype_width; i++)
1190 1.7 martin *p++ = '-';
1191 1.7 martin for (i = 0, *p++ = ' '; i < flags_width; i++)
1192 1.7 martin *p++ = '-';
1193 1.7 martin *p = 0;
1194 1.7 martin
1195 1.7 martin snprintf(fspart_separator, sizeof(fspart_separator),
1196 1.7 martin "------------ ------------ ------------ %s ----------------",
1197 1.7 martin sepline);
1198 1.7 martin
1199 1.7 martin snprintf(fspart_title, sizeof(fspart_title),
1200 1.7 martin " %12.12s %12.12s %12.12s %*s %*s %s\n"
1201 1.7 martin " %s", scol, ecol, szcol, fstype_width, fstype,
1202 1.7 martin flags_width, flags, msg_string(MSG_ptnheaders_filesystem),
1203 1.7 martin fspart_separator);
1204 1.7 martin
1205 1.7 martin msg_table_add("\n\n");
1206 1.7 martin }
1207 1.7 martin
1208 1.7 martin /*
1209 1.7 martin * Format one partition entry in the main partition editor.
1210 1.7 martin */
1211 1.7 martin static void
1212 1.7 martin fmt_fspart_row(menudesc *m, int ptn, void *arg)
1213 1.7 martin {
1214 1.7 martin struct partition_usage_set *pset = arg;
1215 1.7 martin struct disk_part_info info;
1216 1.7 martin daddr_t poffset, psize, pend;
1217 1.7 martin const char *desc;
1218 1.7 martin static const char *Yes;
1219 1.7 martin char flag_str[MENUSTRSIZE], *fp;
1220 1.7 martin unsigned inst_flags;
1221 1.14 martin #ifndef NO_CLONES
1222 1.13 martin size_t clone_cnt;
1223 1.14 martin #endif
1224 1.7 martin bool with_inst_flag = pset->parts->parent == NULL;
1225 1.7 martin
1226 1.7 martin if (Yes == NULL)
1227 1.7 martin Yes = msg_string(MSG_Yes);
1228 1.7 martin
1229 1.14 martin #ifndef NO_CLONES
1230 1.13 martin if ((pset->infos[ptn].flags & PUIFLG_CLONE_PARTS) &&
1231 1.13 martin pset->infos[ptn].cur_part_id == NO_PART) {
1232 1.13 martin psize = pset->infos[ptn].size / sizemult;
1233 1.13 martin if (pset->infos[ptn].clone_ndx <
1234 1.13 martin pset->infos[ptn].clone_src->num_sel)
1235 1.13 martin clone_cnt = 1;
1236 1.13 martin else
1237 1.13 martin clone_cnt = pset->infos[ptn].clone_src->num_sel;
1238 1.13 martin if (pset->infos[ptn].cur_part_id == NO_PART)
1239 1.13 martin wprintw(m->mw, " %12" PRIu64
1240 1.13 martin " [%zu %s]", psize, clone_cnt,
1241 1.13 martin msg_string(MSG_clone_target_disp));
1242 1.13 martin else {
1243 1.13 martin poffset = pset->infos[ptn].cur_start / sizemult;
1244 1.13 martin pend = (pset->infos[ptn].cur_start +
1245 1.13 martin pset->infos[ptn].size) / sizemult - 1;
1246 1.13 martin wprintw(m->mw, "%12" PRIu64 " %12" PRIu64 " %12" PRIu64
1247 1.13 martin " [%zu %s]",
1248 1.13 martin poffset, pend, psize, clone_cnt,
1249 1.13 martin msg_string(MSG_clone_target_disp));
1250 1.13 martin }
1251 1.13 martin if (m->title == fspart_title)
1252 1.13 martin m->opts[ptn].opt_flags |= OPT_IGNORE;
1253 1.13 martin else
1254 1.13 martin m->opts[ptn].opt_flags &= ~OPT_IGNORE;
1255 1.13 martin return;
1256 1.13 martin }
1257 1.14 martin #endif
1258 1.13 martin
1259 1.7 martin if (!real_partition(pset, ptn))
1260 1.7 martin return;
1261 1.7 martin
1262 1.7 martin if (!pset->parts->pscheme->get_part_info(pset->parts,
1263 1.7 martin pset->infos[ptn].cur_part_id, &info))
1264 1.7 martin return;
1265 1.7 martin
1266 1.13 martin /*
1267 1.13 martin * We use this function in multiple menus, but only want it
1268 1.13 martin * to play with enable/disable in a single one:
1269 1.13 martin */
1270 1.13 martin if (m->title == fspart_title) {
1271 1.13 martin /*
1272 1.13 martin * Enable / disable this line if it is something
1273 1.13 martin * like RAW_PART
1274 1.13 martin */
1275 1.13 martin if ((info.flags &
1276 1.13 martin (PTI_WHOLE_DISK|PTI_PSCHEME_INTERNAL|PTI_RAW_PART))
1277 1.13 martin || (pset->infos[ptn].flags & PUIFLG_CLONE_PARTS))
1278 1.13 martin m->opts[ptn].opt_flags |= OPT_IGNORE;
1279 1.13 martin else
1280 1.13 martin m->opts[ptn].opt_flags &= ~OPT_IGNORE;
1281 1.13 martin }
1282 1.7 martin
1283 1.7 martin poffset = info.start / sizemult;
1284 1.7 martin psize = info.size / sizemult;
1285 1.7 martin if (psize == 0)
1286 1.7 martin pend = 0;
1287 1.7 martin else
1288 1.7 martin pend = (info.start + info.size) / sizemult - 1;
1289 1.7 martin
1290 1.7 martin if (info.flags & PTI_WHOLE_DISK)
1291 1.7 martin desc = msg_string(MSG_NetBSD_partition_cant_change);
1292 1.7 martin else if (info.flags & PTI_RAW_PART)
1293 1.7 martin desc = msg_string(MSG_Whole_disk_cant_change);
1294 1.7 martin else if (info.flags & PTI_BOOT)
1295 1.7 martin desc = msg_string(MSG_Boot_partition_cant_change);
1296 1.7 martin else
1297 1.7 martin desc = getfslabelname(info.fs_type, info.fs_sub_type);
1298 1.7 martin
1299 1.7 martin fp = flag_str;
1300 1.7 martin inst_flags = pset->infos[ptn].instflags;
1301 1.26 martin if (with_inst_flag && (info.flags & PTI_INSTALL_TARGET) &&
1302 1.7 martin info.nat_type->generic_ptype == PT_root) {
1303 1.7 martin static char inst_flag;
1304 1.7 martin
1305 1.7 martin if (inst_flag == 0)
1306 1.7 martin inst_flag = msg_string(MSG_install_flag)[0];
1307 1.7 martin *fp++ = inst_flag;
1308 1.7 martin }
1309 1.7 martin if (inst_flags & PUIINST_NEWFS)
1310 1.7 martin *fp++ = msg_string(MSG_newfs_flag)[0];
1311 1.13 martin if (pset->infos[ptn].flags & PUIFLG_CLONE_PARTS)
1312 1.13 martin *fp++ = msg_string(MSG_clone_flag)[0];
1313 1.7 martin *fp = 0;
1314 1.7 martin if (pset->parts->pscheme->get_part_attr_str != NULL)
1315 1.7 martin pset->parts->pscheme->get_part_attr_str(pset->parts,
1316 1.7 martin pset->infos[ptn].cur_part_id, fp, sizeof(flag_str)-1);
1317 1.7 martin
1318 1.7 martin /* if the fstype description does not fit, check if we can overrun */
1319 1.7 martin if (strlen(desc) > (size_t)fstype_width &&
1320 1.7 martin flag_str[0] == 0 && (info.last_mounted == NULL ||
1321 1.7 martin info.last_mounted[0] == 0))
1322 1.7 martin wprintw(m->mw, "%12" PRIu64 " %12" PRIu64 " %12" PRIu64
1323 1.7 martin " %s",
1324 1.7 martin poffset, pend, psize, desc);
1325 1.7 martin else
1326 1.7 martin wprintw(m->mw, "%12" PRIu64 " %12" PRIu64 " %12" PRIu64
1327 1.7 martin " %*.*s %*s %s",
1328 1.7 martin poffset, pend, psize, fstype_width, fstype_width, desc,
1329 1.7 martin -flags_width, flag_str,
1330 1.7 martin (inst_flags & PUIINST_MOUNT) && info.last_mounted &&
1331 1.7 martin info.last_mounted[0] ? info.last_mounted : "");
1332 1.1 dholland }
1333 1.1 dholland
1334 1.14 martin #ifndef NO_CLONES
1335 1.1 dholland static int
1336 1.13 martin part_ext_clone(menudesc *m, void *arg)
1337 1.13 martin {
1338 1.13 martin struct selected_partitions selected, *clone_src;
1339 1.13 martin struct clone_target_menu_data data;
1340 1.13 martin struct partition_usage_set *pset = arg;
1341 1.13 martin struct part_usage_info *p;
1342 1.13 martin struct disk_part_info sinfo, cinfo;
1343 1.13 martin struct disk_partitions *csrc;
1344 1.13 martin struct disk_part_free_space space;
1345 1.13 martin menu_ent *men;
1346 1.13 martin daddr_t clone_size, free_size, offset, align;
1347 1.13 martin int num_men, i;
1348 1.13 martin size_t s, clone_cnt;
1349 1.13 martin part_id cid;
1350 1.13 martin struct clone_data {
1351 1.13 martin struct disk_part_info info;
1352 1.13 martin part_id new_id;
1353 1.13 martin size_t ndx;
1354 1.13 martin };
1355 1.13 martin struct clone_data *clones = NULL;
1356 1.13 martin
1357 1.13 martin if (!select_partitions(&selected, pm->parts))
1358 1.13 martin return 0;
1359 1.13 martin
1360 1.13 martin clone_size = selected_parts_size(&selected);
1361 1.13 martin num_men = pset->num+1;
1362 1.13 martin men = calloc(num_men, sizeof *men);
1363 1.13 martin if (men == NULL)
1364 1.13 martin return 0;
1365 1.13 martin for (i = 0; i < num_men; i++) {
1366 1.13 martin men[i].opt_action = clone_target_select;
1367 1.13 martin if (i == 0)
1368 1.13 martin free_size = pset->infos[i].cur_start;
1369 1.13 martin else if (i > 0 && (size_t)i < pset->num)
1370 1.13 martin free_size = pset->infos[i].cur_start -
1371 1.13 martin pset->infos[i-1].cur_start - pset->infos[i-1].size;
1372 1.13 martin else
1373 1.13 martin free_size = pset->parts->free_space;
1374 1.13 martin if (free_size < clone_size)
1375 1.13 martin men[i].opt_flags = OPT_IGNORE;
1376 1.13 martin }
1377 1.13 martin men[num_men-1].opt_name = MSG_clone_target_end;
1378 1.13 martin
1379 1.13 martin memset(&data, 0, sizeof data);
1380 1.13 martin data.usage = *pset;
1381 1.13 martin data.res = -1;
1382 1.13 martin
1383 1.13 martin data.usage.menu = new_menu(MSG_clone_target_hdr,
1384 1.13 martin men, num_men, 3, 2, 0, 65, MC_SCROLL,
1385 1.13 martin NULL, fmt_fspart_row, NULL, NULL, MSG_cancel);
1386 1.13 martin process_menu(data.usage.menu, &data);
1387 1.13 martin free_menu(data.usage.menu);
1388 1.13 martin free(men);
1389 1.13 martin
1390 1.13 martin if (data.res < 0)
1391 1.13 martin goto err;
1392 1.13 martin
1393 1.13 martin /* create temporary infos for all clones that work out */
1394 1.13 martin clone_cnt = 0;
1395 1.13 martin clones = calloc(selected.num_sel, sizeof(*clones));
1396 1.13 martin if (clones == NULL)
1397 1.13 martin goto err;
1398 1.13 martin
1399 1.13 martin clone_src = malloc(sizeof(selected));
1400 1.13 martin if (clone_src == NULL)
1401 1.13 martin goto err;
1402 1.13 martin *clone_src = selected;
1403 1.13 martin
1404 1.13 martin /* find selected offset from data.res and insert clones there */
1405 1.13 martin align = pset->parts->pscheme->get_part_alignment(pset->parts);
1406 1.13 martin offset = -1;
1407 1.13 martin if (data.res > 0)
1408 1.13 martin offset = pset->infos[data.res-1].cur_start
1409 1.13 martin + pset->infos[data.res-1].size;
1410 1.13 martin else
1411 1.13 martin offset = 0;
1412 1.13 martin for (s = 0; s < selected.num_sel; s++) {
1413 1.13 martin csrc = selected.selection[s].parts;
1414 1.13 martin cid = selected.selection[s].id;
1415 1.13 martin csrc->pscheme->get_part_info(csrc, cid, &sinfo);
1416 1.13 martin if (!pset->parts->pscheme->adapt_foreign_part_info(
1417 1.13 martin pset->parts, &cinfo, csrc->pscheme, &sinfo))
1418 1.13 martin continue;
1419 1.13 martin size_t cnt = pset->parts->pscheme->get_free_spaces(
1420 1.13 martin pset->parts, &space, 1, cinfo.size-align, align,
1421 1.13 martin offset, -1);
1422 1.13 martin if (cnt == 0)
1423 1.13 martin continue;
1424 1.13 martin cinfo.start = space.start;
1425 1.13 martin cid = pset->parts->pscheme->add_partition(
1426 1.13 martin pset->parts, &cinfo, NULL);
1427 1.13 martin if (cid == NO_PART)
1428 1.13 martin continue;
1429 1.13 martin pset->parts->pscheme->get_part_info(pset->parts, cid, &cinfo);
1430 1.13 martin clones[clone_cnt].info = cinfo;
1431 1.13 martin clones[clone_cnt].new_id = cid;
1432 1.13 martin clones[clone_cnt].ndx = s;
1433 1.13 martin clone_cnt++;
1434 1.27 martin offset = roundup(cinfo.start+cinfo.size, align);
1435 1.13 martin }
1436 1.13 martin
1437 1.13 martin /* insert new clone records at offset data.res */
1438 1.13 martin men = realloc(m->opts, (m->numopts+clone_cnt)*sizeof(*m->opts));
1439 1.13 martin if (men == NULL)
1440 1.13 martin goto err;
1441 1.13 martin pset->menu_opts = men;
1442 1.13 martin m->opts = men;
1443 1.13 martin m->numopts += clone_cnt;
1444 1.13 martin
1445 1.13 martin p = realloc(pset->infos, (pset->num+clone_cnt)*sizeof(*pset->infos));
1446 1.13 martin if (p == NULL)
1447 1.13 martin goto err;
1448 1.13 martin pset->infos = p;
1449 1.13 martin
1450 1.13 martin men += data.res;
1451 1.13 martin p += data.res;
1452 1.13 martin memmove(men+clone_cnt, men,
1453 1.13 martin sizeof(*men)*(m->numopts-data.res-clone_cnt));
1454 1.13 martin if (pset->num > (size_t)data.res)
1455 1.13 martin memmove(p+clone_cnt, p, sizeof(*p)*(pset->num-data.res));
1456 1.13 martin memset(men, 0, sizeof(*men)*clone_cnt);
1457 1.13 martin memset(p, 0, sizeof(*p)*clone_cnt);
1458 1.13 martin for (s = 0; s < clone_cnt; s++) {
1459 1.13 martin p[s].cur_part_id = clones[s].new_id;
1460 1.13 martin p[s].cur_start = clones[s].info.start;
1461 1.13 martin p[s].size = clones[s].info.size;
1462 1.13 martin p[s].cur_flags = clones[s].info.flags;
1463 1.13 martin p[s].flags = PUIFLG_CLONE_PARTS;
1464 1.13 martin p[s].parts = pset->parts;
1465 1.13 martin p[s].clone_src = clone_src;
1466 1.13 martin p[s].clone_ndx = s;
1467 1.13 martin }
1468 1.13 martin free(clones);
1469 1.13 martin m->cursel = ((size_t)data.res >= pset->num) ? 0 : data.res+clone_cnt;
1470 1.13 martin pset->num += clone_cnt;
1471 1.13 martin m->h = 0;
1472 1.13 martin resize_menu_height(m);
1473 1.13 martin
1474 1.13 martin return -1;
1475 1.13 martin
1476 1.13 martin err:
1477 1.13 martin free(clones);
1478 1.13 martin free_selected_partitions(&selected);
1479 1.13 martin return 0;
1480 1.13 martin }
1481 1.14 martin #endif
1482 1.13 martin
1483 1.13 martin static int
1484 1.7 martin edit_fspart_pack(menudesc *m, void *arg)
1485 1.1 dholland {
1486 1.7 martin struct partition_usage_set *pset = arg;
1487 1.7 martin char buf[STRSIZE];
1488 1.1 dholland
1489 1.7 martin if (!pset->parts->pscheme->get_disk_pack_name(pset->parts,
1490 1.7 martin buf, sizeof buf))
1491 1.7 martin return 0;
1492 1.7 martin
1493 1.7 martin msg_prompt_win(MSG_edit_disk_pack_hdr,
1494 1.7 martin -1, 18, 0, -1, buf, buf, sizeof(buf));
1495 1.7 martin
1496 1.7 martin pset->parts->pscheme->set_disk_pack_name(pset->parts, buf);
1497 1.1 dholland return 0;
1498 1.1 dholland }
1499 1.1 dholland
1500 1.7 martin static int
1501 1.7 martin edit_fspart_add(menudesc *m, void *arg)
1502 1.7 martin {
1503 1.7 martin struct partition_usage_set *pset = arg;
1504 1.7 martin struct part_usage_info *ninfo;
1505 1.7 martin menu_ent *nmenopts;
1506 1.7 martin size_t cnt, off;
1507 1.7 martin
1508 1.7 martin ninfo = realloc(pset->infos, (pset->num+1)*sizeof(*pset->infos));
1509 1.7 martin if (ninfo == NULL)
1510 1.7 martin return 0;
1511 1.7 martin pset->infos = ninfo;
1512 1.7 martin off = pset->parts->num_part;
1513 1.7 martin cnt = pset->num-pset->parts->num_part;
1514 1.7 martin if (cnt > 0)
1515 1.7 martin memmove(pset->infos+off+1,pset->infos+off,
1516 1.7 martin cnt*sizeof(*pset->infos));
1517 1.7 martin memset(pset->infos+off, 0, sizeof(*pset->infos));
1518 1.7 martin
1519 1.7 martin nmenopts = realloc(m->opts, (m->numopts+1)*sizeof(*m->opts));
1520 1.7 martin if (nmenopts == NULL)
1521 1.7 martin return 0;
1522 1.7 martin memmove(nmenopts+off+1, nmenopts+off,
1523 1.7 martin (m->numopts-off)*sizeof(*nmenopts));
1524 1.7 martin memset(&nmenopts[off], 0, sizeof(nmenopts[off]));
1525 1.7 martin nmenopts[off].opt_action = edit_ptn;
1526 1.7 martin pset->menu_opts = m->opts = nmenopts;
1527 1.7 martin m->numopts++;
1528 1.7 martin m->cursel = off;
1529 1.7 martin pset->num++;
1530 1.7 martin
1531 1.7 martin /* now update edit menu to fit */
1532 1.7 martin m->h = 0;
1533 1.7 martin resize_menu_height(m);
1534 1.7 martin
1535 1.7 martin /* and directly invoke the partition editor for the new part */
1536 1.7 martin edit_ptn(m, arg);
1537 1.7 martin
1538 1.7 martin show_partition_adder(m, pset);
1539 1.7 martin
1540 1.7 martin return -1;
1541 1.7 martin }
1542 1.7 martin
1543 1.1 dholland static void
1544 1.7 martin add_partition_adder(menudesc *m, struct partition_usage_set *pset)
1545 1.1 dholland {
1546 1.7 martin struct part_usage_info *ninfo;
1547 1.7 martin menu_ent *nmenopts;
1548 1.7 martin size_t off;
1549 1.7 martin
1550 1.7 martin ninfo = realloc(pset->infos, (pset->num+1)*sizeof(*pset->infos));
1551 1.7 martin if (ninfo == NULL)
1552 1.7 martin return;
1553 1.7 martin pset->infos = ninfo;
1554 1.7 martin off = pset->parts->num_part+1;
1555 1.1 dholland
1556 1.7 martin nmenopts = realloc(m->opts, (m->numopts+1)*sizeof(*m->opts));
1557 1.7 martin if (nmenopts == NULL)
1558 1.7 martin return;
1559 1.7 martin memmove(nmenopts+off+1, nmenopts+off,
1560 1.7 martin (m->numopts-off)*sizeof(*nmenopts));
1561 1.7 martin memset(&nmenopts[off], 0, sizeof(nmenopts[off]));
1562 1.7 martin
1563 1.7 martin nmenopts[off].opt_name = MSG_addpart;
1564 1.7 martin nmenopts[off].opt_flags = OPT_SUB;
1565 1.7 martin nmenopts[off].opt_action = edit_fspart_add;
1566 1.7 martin
1567 1.7 martin m->opts = nmenopts;
1568 1.7 martin m->numopts++;
1569 1.7 martin }
1570 1.1 dholland
1571 1.7 martin static void
1572 1.7 martin remove_partition_adder(menudesc *m, struct partition_usage_set *pset)
1573 1.7 martin {
1574 1.7 martin size_t off;
1575 1.1 dholland
1576 1.7 martin off = pset->parts->num_part+1;
1577 1.7 martin memmove(m->opts+off, m->opts+off+1,
1578 1.7 martin (m->numopts-off-1)*sizeof(*m->opts));
1579 1.7 martin m->numopts--;
1580 1.1 dholland }
1581 1.1 dholland
1582 1.1 dholland /*
1583 1.7 martin * Called whenever the "add a partition" option may need to be removed
1584 1.7 martin * or added
1585 1.1 dholland */
1586 1.7 martin static void
1587 1.7 martin show_partition_adder(menudesc *m, struct partition_usage_set *pset)
1588 1.1 dholland {
1589 1.18 martin if (m->opts == NULL)
1590 1.18 martin return;
1591 1.18 martin
1592 1.7 martin bool can_add_partition = pset->parts->pscheme->can_add_partition(
1593 1.7 martin pset->parts);
1594 1.7 martin bool part_adder_present =
1595 1.7 martin (m->opts[pset->parts->num_part].opt_flags & OPT_IGNORE) &&
1596 1.7 martin (m->opts[pset->parts->num_part+1].opt_action == edit_fspart_add);
1597 1.1 dholland
1598 1.7 martin if (can_add_partition == part_adder_present)
1599 1.7 martin return;
1600 1.1 dholland
1601 1.7 martin if (can_add_partition)
1602 1.7 martin add_partition_adder(m, pset);
1603 1.7 martin else
1604 1.7 martin remove_partition_adder(m, pset);
1605 1.1 dholland
1606 1.7 martin /* now update edit menu to fit */
1607 1.7 martin m->h = 0;
1608 1.7 martin resize_menu_height(m);
1609 1.7 martin }
1610 1.1 dholland
1611 1.7 martin static int
1612 1.7 martin edit_fspart_abort(menudesc *m, void *arg)
1613 1.7 martin {
1614 1.7 martin struct partition_usage_set *pset = arg;
1615 1.1 dholland
1616 1.7 martin pset->ok = false;
1617 1.7 martin return 1;
1618 1.1 dholland }
1619 1.1 dholland
1620 1.1 dholland /*
1621 1.7 martin * Check a disklabel.
1622 1.7 martin * If there are overlapping active partitions,
1623 1.7 martin * Ask the user if they want to edit the partition or give up.
1624 1.1 dholland */
1625 1.1 dholland int
1626 1.18 martin edit_and_check_label(struct pm_devs *p, struct partition_usage_set *pset,
1627 1.18 martin bool install)
1628 1.1 dholland {
1629 1.7 martin menu_ent *op;
1630 1.7 martin size_t cnt, i;
1631 1.7 martin bool may_add = pset->parts->pscheme->can_add_partition(pset->parts);
1632 1.7 martin bool may_edit_pack =
1633 1.7 martin pset->parts->pscheme->get_disk_pack_name != NULL &&
1634 1.7 martin pset->parts->pscheme->set_disk_pack_name != NULL;
1635 1.7 martin
1636 1.14 martin #ifdef NO_CLONES
1637 1.14 martin #define C_M_ITEMS 0
1638 1.14 martin #else
1639 1.14 martin #define C_M_ITEMS 1
1640 1.14 martin #endif
1641 1.7 martin pset->menu_opts = calloc(pset->parts->num_part
1642 1.14 martin +3+C_M_ITEMS+may_add+may_edit_pack,
1643 1.7 martin sizeof *pset->menu_opts);
1644 1.7 martin if (pset->menu_opts == NULL)
1645 1.7 martin return 0;
1646 1.1 dholland
1647 1.7 martin op = pset->menu_opts;
1648 1.7 martin for (i = 0; i < pset->parts->num_part; i++) {
1649 1.7 martin op->opt_action = edit_ptn;
1650 1.7 martin op++;
1651 1.7 martin }
1652 1.7 martin /* separator line between partitions and actions */
1653 1.7 martin op->opt_name = fspart_separator;
1654 1.7 martin op->opt_flags = OPT_IGNORE|OPT_NOSHORT;
1655 1.7 martin op++;
1656 1.7 martin
1657 1.7 martin /* followed by new partition adder */
1658 1.7 martin if (may_add) {
1659 1.7 martin op->opt_name = MSG_addpart;
1660 1.7 martin op->opt_flags = OPT_SUB;
1661 1.7 martin op->opt_action = edit_fspart_add;
1662 1.7 martin op++;
1663 1.7 martin }
1664 1.7 martin
1665 1.7 martin /* and unit changer */
1666 1.7 martin op->opt_name = MSG_askunits;
1667 1.7 martin op->opt_menu = MENU_sizechoice;
1668 1.7 martin op->opt_flags = OPT_SUB;
1669 1.7 martin op->opt_action = NULL;
1670 1.7 martin op++;
1671 1.7 martin
1672 1.7 martin if (may_edit_pack) {
1673 1.7 martin op->opt_name = MSG_editpack;
1674 1.7 martin op->opt_flags = OPT_SUB;
1675 1.7 martin op->opt_action = edit_fspart_pack;
1676 1.7 martin op++;
1677 1.7 martin }
1678 1.13 martin
1679 1.14 martin #ifndef NO_CLONES
1680 1.13 martin /* add a clone-from-elsewhere option */
1681 1.13 martin op->opt_name = MSG_clone_from_elsewhere;
1682 1.13 martin op->opt_action = part_ext_clone;
1683 1.13 martin op++;
1684 1.14 martin #endif
1685 1.7 martin
1686 1.7 martin /* and abort option */
1687 1.7 martin op->opt_name = MSG_cancel;
1688 1.7 martin op->opt_flags = OPT_EXIT;
1689 1.7 martin op->opt_action = edit_fspart_abort;
1690 1.7 martin op++;
1691 1.7 martin cnt = op - pset->menu_opts;
1692 1.14 martin assert(cnt == pset->parts->num_part+3+C_M_ITEMS+may_add+may_edit_pack);
1693 1.7 martin
1694 1.7 martin pset->menu = new_menu(fspart_title, pset->menu_opts, cnt,
1695 1.7 martin 0, -1, 0, 74,
1696 1.7 martin MC_ALWAYS_SCROLL|MC_NOBOX|MC_DFLTEXIT|
1697 1.7 martin MC_NOCLEAR|MC_CONTINUOUS,
1698 1.7 martin fmt_fspart_header, fmt_fspart_row, NULL, NULL,
1699 1.7 martin MSG_partition_sizes_ok);
1700 1.7 martin
1701 1.7 martin if (pset->menu < 0) {
1702 1.7 martin free(pset->menu_opts);
1703 1.7 martin pset->menu_opts = NULL;
1704 1.7 martin return 0;
1705 1.7 martin }
1706 1.1 dholland
1707 1.7 martin p->current_cylsize = p->dlcylsize;
1708 1.1 dholland
1709 1.7 martin for (;;) {
1710 1.7 martin /* first give the user the option to edit the label... */
1711 1.7 martin pset->ok = true;
1712 1.7 martin process_menu(pset->menu, pset);
1713 1.7 martin if (!pset->ok) {
1714 1.7 martin i = 0;
1715 1.7 martin break;
1716 1.7 martin }
1717 1.1 dholland
1718 1.7 martin /* User thinks the label is OK. */
1719 1.18 martin i = verify_parts(pset, install);
1720 1.7 martin if (i == 1)
1721 1.7 martin continue;
1722 1.7 martin break;
1723 1.1 dholland }
1724 1.7 martin free(pset->menu_opts);
1725 1.7 martin pset->menu_opts = NULL;
1726 1.7 martin free_menu(pset->menu);
1727 1.7 martin pset->menu = -1;
1728 1.1 dholland
1729 1.7 martin return i != 0;
1730 1.1 dholland }
1731 1.1 dholland
1732 1.1 dholland /*
1733 1.10 martin * strip trailing / to avoid confusion in path comparisions later
1734 1.10 martin */
1735 1.10 martin void
1736 1.10 martin canonicalize_last_mounted(char *path)
1737 1.10 martin {
1738 1.10 martin char *p;
1739 1.10 martin
1740 1.11 martin if (path == NULL)
1741 1.11 martin return;
1742 1.11 martin
1743 1.11 martin if (strcmp(path, "/") == 0)
1744 1.11 martin return; /* in this case a "trailing" slash is allowed */
1745 1.11 martin
1746 1.10 martin for (;;) {
1747 1.10 martin p = strrchr(path, '/');
1748 1.10 martin if (p == NULL)
1749 1.10 martin return;
1750 1.10 martin if (p[1] != 0)
1751 1.10 martin return;
1752 1.10 martin p[0] = 0;
1753 1.10 martin }
1754 1.10 martin }
1755 1.10 martin
1756 1.10 martin /*
1757 1.1 dholland * Try to get 'last mounted on' (or equiv) from fs superblock.
1758 1.1 dholland */
1759 1.1 dholland const char *
1760 1.7 martin get_last_mounted(int fd, daddr_t partstart, uint *fs_type, uint *fs_sub_type,
1761 1.7 martin uint flags)
1762 1.1 dholland {
1763 1.1 dholland static char sblk[SBLOCKSIZE]; /* is this enough? */
1764 1.1 dholland struct fs *SB = (struct fs *)sblk;
1765 1.7 martin static const off_t sblocks[] = SBLOCKSEARCH;
1766 1.7 martin const off_t *sbp;
1767 1.1 dholland const char *mnt = NULL;
1768 1.1 dholland int len;
1769 1.1 dholland
1770 1.1 dholland if (fd == -1)
1771 1.1 dholland return "";
1772 1.1 dholland
1773 1.7 martin if (fs_type)
1774 1.7 martin *fs_type = 0;
1775 1.7 martin if (fs_sub_type)
1776 1.7 martin *fs_sub_type = 0;
1777 1.7 martin
1778 1.1 dholland /* Check UFS1/2 (and hence LFS) superblock */
1779 1.1 dholland for (sbp = sblocks; mnt == NULL && *sbp != -1; sbp++) {
1780 1.1 dholland if (pread(fd, sblk, sizeof sblk,
1781 1.7 martin (off_t)partstart * (off_t)512 + *sbp) != sizeof sblk)
1782 1.7 martin continue;
1783 1.7 martin
1784 1.7 martin /*
1785 1.7 martin * If start of partition and allowed by flags check
1786 1.7 martin * for other fs types
1787 1.7 martin */
1788 1.7 martin if (*sbp == 0 && (flags & GLM_MAYBE_FAT32) &&
1789 1.7 martin sblk[0x42] == 0x29 && memcmp(sblk + 0x52, "FAT", 3) == 0) {
1790 1.7 martin /* Probably a FAT filesystem, report volume name */
1791 1.7 martin size_t i;
1792 1.7 martin for (i = 0x51; i >= 0x47; i--) {
1793 1.7 martin if (sblk[i] != ' ')
1794 1.7 martin break;
1795 1.7 martin sblk[i] = 0;
1796 1.7 martin }
1797 1.7 martin sblk[0x52] = 0;
1798 1.7 martin if (fs_type)
1799 1.7 martin *fs_type = FS_MSDOS;
1800 1.7 martin if (fs_sub_type)
1801 1.7 martin *fs_sub_type = sblk[0x53];
1802 1.7 martin return sblk + 0x47;
1803 1.7 martin } else if (*sbp == 0 && (flags & GLM_MAYBE_NTFS) &&
1804 1.7 martin memcmp(sblk+3, "NTFS ", 8) == 0) {
1805 1.7 martin if (fs_type)
1806 1.7 martin *fs_type = FS_NTFS;
1807 1.7 martin if (fs_sub_type)
1808 1.7 martin *fs_sub_type = MBR_PTYPE_NTFS;
1809 1.7 martin /* XXX dig for volume name attribute ? */
1810 1.7 martin return "";
1811 1.7 martin }
1812 1.7 martin
1813 1.7 martin if (!(flags & GLM_LIKELY_FFS))
1814 1.1 dholland continue;
1815 1.7 martin
1816 1.1 dholland /* Maybe we should validate the checksum??? */
1817 1.1 dholland switch (SB->fs_magic) {
1818 1.1 dholland case FS_UFS1_MAGIC:
1819 1.1 dholland case FS_UFS1_MAGIC_SWAPPED:
1820 1.1 dholland if (!(SB->fs_old_flags & FS_FLAGS_UPDATED)) {
1821 1.1 dholland if (*sbp == SBLOCK_UFS1)
1822 1.1 dholland mnt = (const char *)SB->fs_fsmnt;
1823 1.1 dholland } else {
1824 1.1 dholland /* Check we have the main superblock */
1825 1.1 dholland if (SB->fs_sblockloc == *sbp)
1826 1.1 dholland mnt = (const char *)SB->fs_fsmnt;
1827 1.1 dholland }
1828 1.7 martin if (fs_type)
1829 1.7 martin *fs_type = FS_BSDFFS;
1830 1.7 martin if (fs_sub_type)
1831 1.12 martin *fs_sub_type = 1;
1832 1.1 dholland continue;
1833 1.1 dholland case FS_UFS2_MAGIC:
1834 1.1 dholland case FS_UFS2_MAGIC_SWAPPED:
1835 1.1 dholland /* Check we have the main superblock */
1836 1.1 dholland if (SB->fs_sblockloc == *sbp) {
1837 1.1 dholland mnt = (const char *)SB->fs_fsmnt;
1838 1.7 martin if (fs_type)
1839 1.7 martin *fs_type = FS_BSDFFS;
1840 1.7 martin if (fs_sub_type)
1841 1.7 martin *fs_sub_type = 2;
1842 1.1 dholland }
1843 1.1 dholland continue;
1844 1.1 dholland }
1845 1.1 dholland }
1846 1.1 dholland
1847 1.1 dholland if (mnt == NULL)
1848 1.1 dholland return "";
1849 1.1 dholland
1850 1.1 dholland /* If sysinst mounted this last then strip prefix */
1851 1.1 dholland len = strlen(targetroot_mnt);
1852 1.1 dholland if (memcmp(mnt, targetroot_mnt, len) == 0) {
1853 1.1 dholland if (mnt[len] == 0)
1854 1.1 dholland return "/";
1855 1.1 dholland if (mnt[len] == '/')
1856 1.1 dholland return mnt + len;
1857 1.1 dholland }
1858 1.1 dholland return mnt;
1859 1.1 dholland #undef SB
1860 1.1 dholland }
1861 1.1 dholland
1862 1.1 dholland /* Ask for a partition offset, check bounds and do the needed roundups */
1863 1.7 martin daddr_t
1864 1.7 martin getpartoff(struct disk_partitions *parts, daddr_t defpartstart)
1865 1.1 dholland {
1866 1.7 martin char defstart[24], isize[24], maxpart, minspace, maxspace,
1867 1.7 martin *prompt, *label_msg, valid_parts[4], valid_spaces[4],
1868 1.7 martin space_prompt[1024], *head, *hint_part, *hint_space, *tail;
1869 1.7 martin size_t num_freespace, spaces, ndx;
1870 1.7 martin struct disk_part_free_space *freespace;
1871 1.7 martin daddr_t i, localsizemult, ptn_alignment, min, max;
1872 1.7 martin part_id partn;
1873 1.7 martin struct disk_part_info info;
1874 1.7 martin const char *errmsg = NULL;
1875 1.7 martin
1876 1.7 martin min = parts->disk_start;
1877 1.7 martin max = min + parts->disk_size;
1878 1.7 martin
1879 1.7 martin /* upper bound on the number of free spaces, plus some slope */
1880 1.7 martin num_freespace = parts->num_part * 2 + 5;
1881 1.7 martin freespace = calloc(num_freespace, sizeof(*freespace));
1882 1.7 martin if (freespace == NULL)
1883 1.7 martin return -1;
1884 1.7 martin
1885 1.7 martin ptn_alignment = parts->pscheme->get_part_alignment(parts);
1886 1.7 martin spaces = parts->pscheme->get_free_spaces(parts, freespace,
1887 1.7 martin num_freespace, max(sizemult, ptn_alignment), ptn_alignment, -1,
1888 1.7 martin defpartstart);
1889 1.7 martin
1890 1.7 martin maxpart = 'a' + parts->num_part -1;
1891 1.7 martin if (parts->num_part > 1) {
1892 1.7 martin snprintf(valid_parts, sizeof valid_parts, "a-%c", maxpart);
1893 1.7 martin } else if (parts->num_part == 1) {
1894 1.7 martin snprintf(valid_parts, sizeof valid_parts, " %c", maxpart);
1895 1.7 martin } else {
1896 1.7 martin strcpy(valid_parts, "---");
1897 1.7 martin }
1898 1.7 martin if (spaces > 1) {
1899 1.7 martin minspace = maxpart + 1;
1900 1.7 martin maxspace = minspace + spaces -1;
1901 1.7 martin snprintf(valid_spaces, sizeof valid_spaces, "%c-%c", minspace,
1902 1.7 martin maxspace);
1903 1.7 martin } else if (spaces == 1) {
1904 1.7 martin maxspace = minspace = maxpart + 1;
1905 1.7 martin snprintf(valid_spaces, sizeof valid_spaces, " %c", minspace);
1906 1.7 martin } else {
1907 1.7 martin minspace = 0;
1908 1.7 martin maxspace = 0;
1909 1.7 martin strcpy(valid_spaces, "---");
1910 1.7 martin }
1911 1.7 martin
1912 1.7 martin /* Add description of start/size to user prompt */
1913 1.7 martin const char *mstr = msg_string(MSG_free_space_line);
1914 1.7 martin space_prompt[0] = 0;
1915 1.7 martin for (ndx = 0; ndx < spaces; ndx++) {
1916 1.7 martin char str_start[40], str_end[40], str_size[40], str_tag[4];
1917 1.7 martin
1918 1.7 martin sprintf(str_tag, "%c: ", minspace+(int)ndx);
1919 1.7 martin sprintf(str_start, "%" PRIu64, freespace[ndx].start / sizemult);
1920 1.7 martin sprintf(str_end, "%" PRIu64,
1921 1.7 martin (freespace[ndx].start + freespace[ndx].size) / sizemult);
1922 1.7 martin sprintf(str_size, "%" PRIu64, freespace[ndx].size / sizemult);
1923 1.7 martin const char *args[4] = { str_start, str_end, str_size,
1924 1.7 martin multname };
1925 1.7 martin char *line = str_arg_subst(mstr, 4, args);
1926 1.7 martin strlcat(space_prompt, str_tag, sizeof(space_prompt));
1927 1.7 martin size_t len = strlcat(space_prompt, line, sizeof(space_prompt));
1928 1.7 martin free (line);
1929 1.7 martin if (len >= sizeof space_prompt)
1930 1.7 martin break;
1931 1.7 martin }
1932 1.7 martin
1933 1.7 martin const char *args[] = { valid_parts, valid_spaces, multname };
1934 1.7 martin hint_part = NULL;
1935 1.7 martin hint_space = NULL;
1936 1.7 martin head = str_arg_subst(msg_string(MSG_label_offset_head),
1937 1.7 martin __arraycount(args), args);
1938 1.7 martin if (parts->num_part)
1939 1.7 martin hint_part = str_arg_subst(msg_string(
1940 1.7 martin MSG_label_offset_part_hint), __arraycount(args), args);
1941 1.7 martin if (spaces)
1942 1.7 martin hint_space = str_arg_subst(msg_string(
1943 1.7 martin MSG_label_offset_space_hint), __arraycount(args), args);
1944 1.7 martin tail = str_arg_subst(msg_string(MSG_label_offset_tail),
1945 1.7 martin __arraycount(args), args);
1946 1.7 martin
1947 1.7 martin if (hint_part && hint_space)
1948 1.7 martin asprintf(&label_msg, "%s\n%s\n%s\n\n%s\n%s",
1949 1.7 martin head, hint_part, hint_space, space_prompt, tail);
1950 1.7 martin else if (hint_part)
1951 1.7 martin asprintf(&label_msg, "%s\n%s\n\n%s",
1952 1.7 martin head, hint_part, tail);
1953 1.7 martin else if (hint_space)
1954 1.7 martin asprintf(&label_msg, "%s\n%s\n\n%s\n%s",
1955 1.7 martin head, hint_space, space_prompt, tail);
1956 1.7 martin else
1957 1.7 martin asprintf(&label_msg, "%s\n\n%s",
1958 1.7 martin head, tail);
1959 1.7 martin free(head); free(hint_part); free(hint_space); free(tail);
1960 1.1 dholland
1961 1.7 martin localsizemult = sizemult;
1962 1.7 martin errmsg = NULL;
1963 1.1 dholland for (;;) {
1964 1.7 martin snprintf(defstart, sizeof defstart, "%" PRIu64,
1965 1.7 martin defpartstart/sizemult);
1966 1.7 martin if (errmsg != NULL && errmsg[0] != 0)
1967 1.7 martin asprintf(&prompt, "%s\n\n%s", errmsg, label_msg);
1968 1.7 martin else
1969 1.7 martin prompt = label_msg;
1970 1.7 martin msg_prompt_win(prompt, -1, 13, 70, -1,
1971 1.7 martin (defpartstart > 0) ? defstart : NULL, isize, sizeof isize);
1972 1.7 martin if (label_msg != prompt)
1973 1.7 martin free(prompt);
1974 1.7 martin if (strcmp(defstart, isize) == 0) {
1975 1.1 dholland /* Don't do rounding if default accepted */
1976 1.7 martin i = defpartstart;
1977 1.7 martin break;
1978 1.7 martin }
1979 1.1 dholland if (isize[1] == '\0' && isize[0] >= 'a' &&
1980 1.7 martin isize[0] <= maxpart) {
1981 1.1 dholland partn = isize[0] - 'a';
1982 1.7 martin if (parts->pscheme->get_part_info(parts, partn,
1983 1.7 martin &info)) {
1984 1.7 martin i = info.start + info.size;
1985 1.7 martin localsizemult = 1;
1986 1.7 martin } else {
1987 1.7 martin errmsg = msg_string(MSG_invalid_sector_number);
1988 1.7 martin continue;
1989 1.7 martin }
1990 1.7 martin } else if (isize[1] == '\0' && isize[0] >= minspace &&
1991 1.7 martin isize[0] <= maxspace) {
1992 1.7 martin ndx = isize[0] - minspace;
1993 1.7 martin i = freespace[ndx].start;
1994 1.1 dholland localsizemult = 1;
1995 1.1 dholland } else if (atoi(isize) == -1) {
1996 1.7 martin i = min;
1997 1.1 dholland localsizemult = 1;
1998 1.1 dholland } else {
1999 1.20 martin i = parse_disk_pos(isize, &localsizemult,
2000 1.20 martin parts->bytes_per_sector,
2001 1.19 martin parts->pscheme->get_cylinder_size(parts), NULL);
2002 1.7 martin if (i < 0) {
2003 1.1 dholland errmsg = msg_string(MSG_invalid_sector_number);
2004 1.1 dholland continue;
2005 1.1 dholland }
2006 1.1 dholland }
2007 1.1 dholland /* round to cylinder size if localsizemult != 1 */
2008 1.19 martin int cylsize = parts->pscheme->get_cylinder_size(parts);
2009 1.19 martin i = NUMSEC(i, localsizemult, cylsize);
2010 1.1 dholland /* Adjust to start of slice if needed */
2011 1.7 martin if ((i < min && (min - i) < localsizemult) ||
2012 1.7 martin (i > min && (i - min) < localsizemult)) {
2013 1.7 martin i = min;
2014 1.1 dholland }
2015 1.7 martin if (max == 0 || i <= max)
2016 1.1 dholland break;
2017 1.1 dholland errmsg = msg_string(MSG_startoutsidedisk);
2018 1.1 dholland }
2019 1.7 martin free(label_msg);
2020 1.7 martin free(freespace);
2021 1.7 martin
2022 1.1 dholland return i;
2023 1.1 dholland }
2024 1.1 dholland
2025 1.1 dholland
2026 1.1 dholland /* Ask for a partition size, check bounds and do the needed roundups */
2027 1.7 martin daddr_t
2028 1.25 martin getpartsize(struct disk_partitions *parts, daddr_t orig_start,
2029 1.25 martin daddr_t partstart, daddr_t dflt)
2030 1.1 dholland {
2031 1.7 martin char dsize[24], isize[24], max_size[24], maxpartc, valid_parts[4],
2032 1.7 martin *label_msg, *prompt, *head, *hint, *tail;
2033 1.7 martin const char *errmsg = NULL;
2034 1.19 martin daddr_t i, partend, diskend, localsizemult, max, max_r, dflt_r;
2035 1.7 martin struct disk_part_info info;
2036 1.7 martin part_id partn;
2037 1.7 martin
2038 1.19 martin diskend = parts->disk_start + parts->disk_size;
2039 1.25 martin max = parts->pscheme->max_free_space_at(parts, orig_start);
2040 1.25 martin max += orig_start - partstart;
2041 1.25 martin if (sizemult == 1)
2042 1.25 martin max--; /* with hugher scale proper rounding later will be ok */
2043 1.7 martin
2044 1.7 martin /* We need to keep both the unrounded and rounded (_r) max and dflt */
2045 1.7 martin dflt_r = (partstart + dflt) / sizemult - partstart / sizemult;
2046 1.7 martin if (max == dflt)
2047 1.7 martin max_r = dflt_r;
2048 1.7 martin else
2049 1.7 martin max_r = max / sizemult;
2050 1.7 martin /* the partition may have been moved and now not fit any longer */
2051 1.7 martin if (dflt > max)
2052 1.7 martin dflt = max;
2053 1.7 martin if (dflt_r > max_r)
2054 1.7 martin dflt_r = max_r;
2055 1.7 martin
2056 1.7 martin snprintf(max_size, sizeof max_size, "%" PRIu64, max_r);
2057 1.7 martin
2058 1.7 martin maxpartc = 'a' + parts->num_part -1;
2059 1.7 martin if (parts->num_part > 1) {
2060 1.7 martin snprintf(valid_parts, sizeof valid_parts, "a-%c", maxpartc);
2061 1.7 martin } else if (parts->num_part == 1) {
2062 1.7 martin snprintf(valid_parts, sizeof valid_parts, " %c", maxpartc);
2063 1.7 martin } else {
2064 1.7 martin strcpy(valid_parts, "---");
2065 1.7 martin }
2066 1.7 martin
2067 1.7 martin const char *args[] = { valid_parts, max_size, multname };
2068 1.7 martin hint = NULL;
2069 1.7 martin head = str_arg_subst(msg_string(MSG_label_size_head),
2070 1.7 martin __arraycount(args), args);
2071 1.7 martin if (parts->num_part)
2072 1.7 martin hint = str_arg_subst(msg_string(MSG_label_size_part_hint),
2073 1.7 martin __arraycount(args), args);
2074 1.7 martin tail = str_arg_subst(msg_string(MSG_label_size_tail),
2075 1.7 martin __arraycount(args), args);
2076 1.7 martin
2077 1.7 martin if (hint != NULL)
2078 1.7 martin asprintf(&label_msg, "%s\n%s\n\n%s", head, hint, tail);
2079 1.7 martin else
2080 1.7 martin asprintf(&label_msg, "%s\n\n%s", head, tail);
2081 1.7 martin free(head); free(hint); free(tail);
2082 1.1 dholland
2083 1.7 martin localsizemult = sizemult;
2084 1.7 martin i = -1;
2085 1.1 dholland for (;;) {
2086 1.7 martin snprintf(dsize, sizeof dsize, "%" PRIu64, dflt_r);
2087 1.7 martin
2088 1.7 martin if (errmsg != NULL && errmsg[0] != 0)
2089 1.7 martin asprintf(&prompt, "%s\n\n%s", errmsg, label_msg);
2090 1.7 martin else
2091 1.7 martin prompt = label_msg;
2092 1.7 martin msg_prompt_win(prompt, -1, 12, 70, -1,
2093 1.7 martin (dflt != 0) ? dsize : 0, isize, sizeof isize);
2094 1.7 martin if (prompt != label_msg)
2095 1.7 martin free(prompt);
2096 1.7 martin
2097 1.7 martin if (strcmp(isize, dsize) == 0) {
2098 1.7 martin free(label_msg);
2099 1.7 martin return dflt;
2100 1.7 martin }
2101 1.7 martin if (parts->num_part && isize[1] == '\0' && isize[0] >= 'a' &&
2102 1.1 dholland isize[0] <= maxpartc) {
2103 1.1 dholland partn = isize[0] - 'a';
2104 1.7 martin if (parts->pscheme->get_part_info(parts, partn,
2105 1.7 martin &info)) {
2106 1.7 martin i = info.start - partstart -1;
2107 1.7 martin localsizemult = 1;
2108 1.7 martin max_r = max;
2109 1.7 martin }
2110 1.1 dholland } else if (atoi(isize) == -1) {
2111 1.7 martin i = max;
2112 1.1 dholland localsizemult = 1;
2113 1.7 martin max_r = max;
2114 1.1 dholland } else {
2115 1.7 martin i = parse_disk_pos(isize, &localsizemult,
2116 1.20 martin parts->bytes_per_sector,
2117 1.19 martin parts->pscheme->get_cylinder_size(parts), NULL);
2118 1.7 martin if (localsizemult != sizemult)
2119 1.7 martin max_r = max;
2120 1.7 martin }
2121 1.7 martin if (i < 0) {
2122 1.7 martin errmsg = msg_string(MSG_Invalid_numeric);
2123 1.7 martin continue;
2124 1.7 martin } else if (i > max_r) {
2125 1.7 martin errmsg = msg_string(MSG_Too_large);
2126 1.7 martin continue;
2127 1.1 dholland }
2128 1.1 dholland /*
2129 1.1 dholland * partend is aligned to a cylinder if localsizemult
2130 1.1 dholland * is not 1 sector
2131 1.1 dholland */
2132 1.19 martin int cylsize = parts->pscheme->get_cylinder_size(parts);
2133 1.7 martin partend = NUMSEC((partstart + i*localsizemult) / localsizemult,
2134 1.19 martin localsizemult, cylsize);
2135 1.1 dholland /* Align to end-of-disk or end-of-slice if close enough */
2136 1.19 martin if (partend > (diskend - sizemult)
2137 1.19 martin && partend < (diskend + sizemult))
2138 1.19 martin partend = diskend;
2139 1.7 martin if (partend > (partstart + max - sizemult)
2140 1.7 martin && partend < (partstart + max + sizemult))
2141 1.7 martin partend = partstart + max;
2142 1.1 dholland /* sanity checks */
2143 1.19 martin if (partend > diskend) {
2144 1.19 martin partend = diskend;
2145 1.7 martin errmsg = msg_string(MSG_endoutsidedisk);
2146 1.7 martin continue;
2147 1.1 dholland }
2148 1.7 martin free(label_msg);
2149 1.1 dholland if (partend < partstart)
2150 1.1 dholland return 0;
2151 1.1 dholland return (partend - partstart);
2152 1.1 dholland }
2153 1.1 dholland /* NOTREACHED */
2154 1.1 dholland }
2155 1.1 dholland
2156 1.1 dholland /*
2157 1.1 dholland * convert a string to a number of sectors, with a possible unit
2158 1.1 dholland * 150M = 150 Megabytes
2159 1.1 dholland * 2000c = 2000 cylinders
2160 1.1 dholland * 150256s = 150256 sectors
2161 1.7 martin * Without units, use the default (sizemult).
2162 1.7 martin * returns the raw input value, and the unit used. Caller needs to multiply!
2163 1.7 martin * On invalid inputs, returns -1.
2164 1.1 dholland */
2165 1.7 martin daddr_t
2166 1.7 martin parse_disk_pos(
2167 1.7 martin const char *str,
2168 1.7 martin daddr_t *localsizemult,
2169 1.20 martin daddr_t bps,
2170 1.7 martin daddr_t cyl_size,
2171 1.7 martin bool *extend_this)
2172 1.1 dholland {
2173 1.7 martin daddr_t val;
2174 1.7 martin char *cp;
2175 1.7 martin bool mult_found;
2176 1.1 dholland
2177 1.1 dholland if (str[0] == '\0') {
2178 1.7 martin return -1;
2179 1.1 dholland }
2180 1.7 martin val = strtoull(str, &cp, 10);
2181 1.7 martin mult_found = false;
2182 1.7 martin if (extend_this)
2183 1.7 martin *extend_this = false;
2184 1.7 martin while (*cp != 0) {
2185 1.7 martin if (*cp == 'G' || *cp == 'g') {
2186 1.7 martin if (mult_found)
2187 1.7 martin return -1;
2188 1.20 martin *localsizemult = GIG / bps;
2189 1.7 martin goto next;
2190 1.1 dholland }
2191 1.7 martin if (*cp == 'M' || *cp == 'm') {
2192 1.7 martin if (mult_found)
2193 1.7 martin return -1;
2194 1.20 martin *localsizemult = MEG / bps;
2195 1.7 martin goto next;
2196 1.1 dholland }
2197 1.7 martin if (*cp == 'c' || *cp == 'C') {
2198 1.7 martin if (mult_found)
2199 1.7 martin return -1;
2200 1.19 martin *localsizemult = cyl_size;
2201 1.7 martin goto next;
2202 1.1 dholland }
2203 1.7 martin if (*cp == 's' || *cp == 'S') {
2204 1.7 martin if (mult_found)
2205 1.7 martin return -1;
2206 1.1 dholland *localsizemult = 1;
2207 1.7 martin goto next;
2208 1.7 martin }
2209 1.7 martin if (*cp == '+' && extend_this) {
2210 1.7 martin *extend_this = true;
2211 1.7 martin cp++;
2212 1.1 dholland break;
2213 1.1 dholland }
2214 1.7 martin
2215 1.1 dholland /* not a known unit */
2216 1.7 martin return -1;
2217 1.7 martin
2218 1.7 martin next:
2219 1.7 martin mult_found = true;
2220 1.7 martin cp++;
2221 1.7 martin continue;
2222 1.1 dholland }
2223 1.7 martin if (*cp != 0)
2224 1.7 martin return -1;
2225 1.7 martin
2226 1.7 martin return val;
2227 1.1 dholland }
2228