Home | History | Annotate | Line # | Download | only in sysinst
partman.c revision 1.47
      1 /*	$NetBSD: partman.c,v 1.47 2020/01/10 12:55:14 martin Exp $ */
      2 
      3 /*
      4  * Copyright 2012 Eugene Lozovoy
      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  * 3. The name of Eugene Lozovoy may not be used to endorse
     16  *    or promote products derived from this software without specific prior
     17  *    written permission.
     18  *
     19  * THIS SOFTWARE IS PROVIDED BY PIERMONT INFORMATION SYSTEMS INC. ``AS IS''
     20  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     21  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
     22  * ARE DISCLAIMED. IN NO EVENT SHALL PIERMONT INFORMATION SYSTEMS INC. BE
     23  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
     24  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
     25  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
     26  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
     27  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
     28  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
     29  * THE POSSIBILITY OF SUCH DAMAGE.
     30  *
     31  */
     32 
     33 /*
     34  * Copyright 2010 The NetBSD Foundation, Inc.
     35  * All rights reserved.
     36  *
     37  * Redistribution and use in source and binary forms, with or without
     38  * modification, are permitted provided that the following conditions
     39  * are met:
     40  * 1. Redistributions of source code must retain the above copyright
     41  *    notice, this list of conditions and the following disclaimer.
     42  * 2. Redistributions in binary form must reproduce the above copyright
     43  *    notice, this list of conditions and the following disclaimer in the
     44  *    documentation and/or other materials provided with the distribution.
     45  *
     46  * THIS SOFTWARE IS PROVIDED BY PIERMONT INFORMATION SYSTEMS INC. ``AS IS''
     47  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     48  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
     49  * ARE DISCLAIMED. IN NO EVENT SHALL PIERMONT INFORMATION SYSTEMS INC. BE
     50  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
     51  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
     52  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
     53  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
     54  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
     55  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
     56  * THE POSSIBILITY OF SUCH DAMAGE.
     57  *
     58  */
     59 
     60 /* partman.c - extended partitioning */
     61 
     62 #include <assert.h>
     63 #include <fcntl.h>
     64 #include <errno.h>
     65 #include <libgen.h>
     66 #include <paths.h>
     67 #include <stdio.h>
     68 #include <stdlib.h>
     69 #include <unistd.h>
     70 #include <util.h>
     71 
     72 #include "defs.h"
     73 #include "msg_defs.h"
     74 #include "menu_defs.h"
     75 
     76 /* XXX: replace all MAX_* defines with vars that depend on kernel settings */
     77 #define MAX_ENTRIES 96
     78 
     79 #define MAX_RAID 8
     80 #define MAX_IN_RAID 48
     81 struct raid_comp {
     82 	char name[SSTRSIZE];	/* display name for this component */
     83 	struct disk_partitions *parts;	/* where is this on? */
     84 	part_id id;		/* which partition in parts */
     85 	bool is_spare;		/* is this a spare component? */
     86 };
     87 struct raid_desc {
     88 	int enabled;
     89 	int blocked;
     90 	int node;		/* the N in raid${N} */
     91 	int numRow, numCol, numSpare;
     92 	int sectPerSU, SUsPerParityUnit, SUsPerReconUnit, raid_level;
     93 	daddr_t total_size;
     94 	struct raid_comp comp[MAX_IN_RAID];
     95 };
     96 struct raid_desc *raids;
     97 
     98 #define MAX_VND 4
     99 struct vnd_desc {
    100 	int enabled;
    101 	int blocked;
    102 	int node;	/* the N in vnd${N} */
    103 	char filepath[STRSIZE];
    104 	daddr_t size;
    105 	int readonly;
    106 	int is_exist;
    107 	int manual_geom;
    108 	int secsize, nsectors, ntracks, ncylinders;
    109 	struct pm_devs *pm;	/* device this is on */
    110 	part_id pm_part;	/* which partition (in pm->parts) */
    111 };
    112 struct vnd_desc *vnds;
    113 
    114 #define MAX_CGD 4
    115 struct cgd_desc {
    116 	int enabled;
    117 	int blocked;
    118 	int node;	/* the N in cgd${N} */
    119 	char pm_name[SSTRSIZE];
    120 	const char *keygen_type;
    121 	const char *verify_type;
    122 	const char *enc_type;
    123 	const char *iv_type;
    124 	int key_size;
    125 	struct pm_devs *pm;	/* device this is on */
    126 	part_id pm_part;	/* which partition (in pm->parts) */
    127 };
    128 struct cgd_desc *cgds;
    129 
    130 #define MAX_LVM_VG 16
    131 #define MAX_LVM_PV 255
    132 #define MAX_LVM_LV 255
    133 
    134 struct lvm_pv_reg {
    135 	struct pm_devs *pm;
    136 	daddr_t start;
    137 };
    138 struct lvm_pv_reg lvm_pvs[MAX_LVM_PV];	/* XXX - make dynamic */
    139 
    140 typedef struct pv_t {
    141 	struct pm_devs *pm;
    142 	char pm_name[SSTRSIZE];
    143 	part_id pm_part;
    144 	int metadatasize;
    145 	int metadatacopies;
    146 	int labelsector;
    147 	int setphysicalvolumesize;
    148 } pv_t;
    149 typedef struct lv_t {
    150 	int blocked;
    151 	daddr_t size;
    152 	char name[SSTRSIZE];
    153 	int readonly;
    154 	int contiguous;
    155 	char extents[SSTRSIZE];
    156 	int minor;
    157 	int mirrors;
    158 	int regionsize;
    159 	int persistent;
    160 	int readahead;
    161 	int stripes;
    162 	int stripesize;
    163 	int zero;
    164 } lv_t;
    165 typedef struct lvms_t {
    166 	int enabled;
    167 	int blocked;
    168 	char name[SSTRSIZE];
    169 	int maxlogicalvolumes;
    170 	int maxphysicalvolumes;
    171 	int physicalextentsize;
    172 	daddr_t total_size;
    173 	pv_t pv[MAX_LVM_PV];
    174 	lv_t lv[MAX_LVM_LV];
    175 } lvms_t;
    176 lvms_t *lvms;
    177 
    178 typedef struct structinfo_t {
    179 	int max;
    180 	uint entry_size;
    181 	uint parent_size;
    182 	void *entry_first;
    183 	void *entry_enabled;
    184 	void *entry_blocked;
    185 	void *entry_node;
    186 } structinfo_t;
    187 structinfo_t raids_t_info, vnds_t_info, cgds_t_info, lvms_t_info, lv_t_info;
    188 
    189 typedef struct pm_upddevlist_adv_t {
    190 	const char *create_msg;
    191 	int pe_type;
    192 	structinfo_t *s;
    193 	int sub_num;
    194 	struct pm_upddevlist_adv_t *sub;
    195 } pm_upddevlist_adv_t;
    196 
    197 #define MAX_MNTS 48
    198 struct {
    199     char dev[STRSIZE];
    200     const char *mnt_opts, *on;
    201 } *mnts;
    202 
    203 static int pm_cursel; /* Number of selected entry in main menu */
    204 static int pm_changed; /* flag indicating that we have unsaved changes */
    205 static int pm_raid_curspare; /* XXX: replace by true way */
    206 static int pm_retvalue;
    207 
    208 enum { /* RAIDframe menu enum */
    209 	PMR_MENU_DEVS, PMR_MENU_DEVSSPARE, PMR_MENU_RAIDLEVEL, PMR_MENU_NUMROW,
    210 	PMR_MENU_NUMCOL, PMR_MENU_NUMSPARE,	PMR_MENU_SECTPERSU,	PMR_MENU_SUSPERPARITYUNIT,
    211 	PMR_MENU_SUSPERRECONUNIT, PMR_MENU_REMOVE, PMR_MENU_END
    212 };
    213 
    214 enum { /* VND menu enum */
    215 	PMV_MENU_FILEPATH, PMV_MENU_EXIST, PMV_MENU_SIZE, PMV_MENU_RO, PMV_MENU_MGEOM,
    216 	PMV_MENU_SECSIZE, PMV_MENU_NSECTORS, PMV_MENU_NTRACKS, PMV_MENU_NCYLINDERS,
    217 	PMV_MENU_REMOVE, PMV_MENU_END
    218 };
    219 
    220 enum { /* CGD menu enum */
    221 	PMC_MENU_DEV, PMC_MENU_ENCTYPE, PMC_MENU_KEYSIZE, PMC_MENU_IVTYPE,
    222 	PMC_MENU_KEYGENTYPE, PMC_MENU_VERIFYTYPE, PMC_MENU_REMOVE, PMC_MENU_END
    223 };
    224 
    225 enum { /* LVM menu enum */
    226 	PML_MENU_PV, PML_MENU_NAME, PML_MENU_MAXLOGICALVOLUMES,
    227 	PML_MENU_MAXPHYSICALVOLUMES, PML_MENU_PHYSICALEXTENTSIZE,
    228 	PML_MENU_REMOVE, PML_MENU_END
    229 };
    230 
    231 enum { /* LVM submenu (logical volumes) enum */
    232 	PMLV_MENU_NAME, PMLV_MENU_SIZE, PMLV_MENU_READONLY, PMLV_MENU_CONTIGUOUS,
    233 	PMLV_MENU_EXTENTS, PMLV_MENU_MINOR, PMLV_MENU_PERSISTENT,
    234 	PMLV_MENU_MIRRORS, PMLV_MENU_REGIONSIZE, PMLV_MENU_READAHEAD,
    235 	PMLV_MENU_STRIPES, PMLV_MENU_STRIPESIZE, PMLV_MENU_ZERO,
    236 	PMLV_MENU_REMOVE, PMLV_MENU_END
    237 };
    238 
    239 struct part_entry pm_dev_list(int);
    240 static int pm_raid_disk_add(menudesc *, void *);
    241 static int pm_raid_disk_del(menudesc *, void *);
    242 static int pm_cgd_disk_set(struct cgd_desc *, struct part_entry *);
    243 static int pm_mount(struct pm_devs *, int);
    244 static int pm_upddevlist(menudesc *, void *);
    245 static void pm_select(struct pm_devs *);
    246 
    247 static void
    248 pm_edit_size_value(msg prompt_msg, daddr_t cylsec, daddr_t *size)
    249 {
    250 
    251 	char answer[16], dflt[16];
    252 	daddr_t new_size_val, mult;
    253 
    254 	snprintf(dflt, sizeof dflt, "%" PRIu64 "%s", *size / sizemult,
    255 	    multname);
    256 
    257 	msg_prompt_win(prompt_msg, -1, 18, 0, 0, dflt, answer, sizeof answer);
    258 
    259 	mult = sizemult;
    260 	new_size_val = parse_disk_pos(answer, &mult, cylsec, NULL);
    261 
    262 	if (new_size_val > 0)
    263 		*size = new_size_val * mult;
    264 }
    265 
    266 static const char *
    267 pm_get_mount(struct pm_devs *p, part_id id)
    268 {
    269 
    270 	if (p->mounted == NULL)
    271 		return NULL;
    272 	if (id >= p->parts->num_part)
    273 		return NULL;
    274 	return p->mounted[id];
    275 }
    276 
    277 bool pm_set_mount(struct pm_devs *p, part_id id, const char *path);
    278 
    279 bool
    280 pm_set_mount(struct pm_devs *p, part_id id, const char *path)
    281 {
    282 
    283 	if (p->parts == NULL || id >= p->parts->num_part)
    284 		return false;
    285 
    286 	if (p->mounted == NULL) {
    287 		p->mounted = calloc(p->parts->num_part, sizeof(char*));
    288 		if (p->mounted == NULL)
    289 			return false;
    290 	}
    291 	free(p->mounted[id]);
    292 	p->mounted[id] = strdup(path);
    293 	return p->mounted[id] != NULL;
    294 }
    295 
    296 /* Universal menu for RAID/VND/CGD/LVM entry edit */
    297 static int
    298 pm_edit(int menu_entries_count, void (*menu_fmt)(menudesc *, int, void *),
    299 	int (*action)(menudesc *, void *), int (*check_fun)(void *),
    300 	void (*entry_init)(void *, void *),	void *entry_init_arg,
    301 	void *dev_ptr, int dev_ptr_delta, structinfo_t *s)
    302 {
    303 	int i, ok = 0;
    304 	menu_ent *menu_entries;
    305 
    306 	if (dev_ptr == NULL) {
    307 		/* We should create new device */
    308 		for (i = 0; i < s->max && !ok; i++)
    309 			if (*(int*)((char*)s->entry_enabled + dev_ptr_delta + s->entry_size * i) == 0) {
    310 				dev_ptr = (char*)s->entry_first + dev_ptr_delta + s->entry_size * i;
    311 				entry_init(dev_ptr, entry_init_arg);
    312 				ok = 1;
    313 			}
    314 		if (!ok) {
    315 			/* We do not have free device slots */
    316 			hit_enter_to_continue(NULL, MSG_limitcount);
    317 			return -1;
    318 		}
    319 	}
    320 
    321 	menu_entries = calloc(menu_entries_count, sizeof *menu_entries);
    322 	for (i = 0; i < menu_entries_count - 1; i++)
    323 		menu_entries[i] = (menu_ent) { .opt_action=action };
    324 	menu_entries[i] = (menu_ent) {	.opt_name=MSG_fremove,
    325 					.opt_flags=OPT_EXIT,
    326 					.opt_action=action };
    327 
    328 	int menu_no = -1;
    329 	menu_no = new_menu(NULL, menu_entries, menu_entries_count,
    330 		-1, -1, 0, 40, MC_NOCLEAR | MC_SCROLL,
    331 		NULL, menu_fmt, NULL, NULL, MSG_DONE);
    332 
    333 	process_menu(menu_no, dev_ptr);
    334 	free_menu(menu_no);
    335 	free(menu_entries);
    336 
    337 	return check_fun(dev_ptr);
    338 }
    339 
    340 /* Show filtered partitions menu */
    341 struct part_entry
    342 pm_dev_list(int type)
    343 {
    344 	int dev_num = -1, num_devs = 0;
    345 	bool ok;
    346 	part_id i;
    347 	int menu_no;
    348 	struct disk_part_info info;
    349 	menu_ent menu_entries[MAX_DISKS*MAXPARTITIONS];
    350 	struct part_entry disk_entries[MAX_DISKS*MAXPARTITIONS];
    351 	struct pm_devs *pm_i;
    352 
    353 	SLIST_FOREACH(pm_i, &pm_head, l) {
    354 		if (pm_i->parts == NULL)
    355 			continue;
    356 		for (i = 0; i < pm_i->parts->num_part; i++) {
    357 			ok = false;
    358 			if (!pm_i->parts->pscheme->get_part_info(pm_i->parts,
    359 			    i, &info))
    360 				continue;
    361 			if (info.flags &
    362 			    (PTI_WHOLE_DISK|PTI_PSCHEME_INTERNAL|PTI_RAW_PART))
    363 				continue;
    364 			switch (type) {
    365 				case PM_RAID:
    366 					if (info.fs_type == FS_RAID)
    367 						ok = 1;
    368 					break;
    369 				case PM_CGD:
    370 					if (info.fs_type == FS_CGD)
    371 						ok = 1;
    372 					break;
    373 				case PM_LVM:
    374 					if (pm_is_lvmpv(pm_i, i, &info))
    375 						ok = 1;
    376 					break;
    377 			}
    378 			if (!ok)
    379 				continue;
    380 			if (pm_partusage(pm_i, i, 0) != 0)
    381 				continue;
    382 
    383 			disk_entries[num_devs].dev_ptr = pm_i;
    384 			disk_entries[num_devs].id = i;
    385 			disk_entries[num_devs].parts = pm_i->parts;
    386 
    387 			pm_i->parts->pscheme->get_part_device(
    388 			    pm_i->parts, i, disk_entries[num_devs].fullname,
    389 			    sizeof disk_entries[num_devs].fullname,
    390 			    NULL, plain_name, false, true);
    391 
    392 			menu_entries[num_devs] = (struct menu_ent) {
    393 				.opt_name = disk_entries[num_devs].fullname,
    394 				.opt_action = set_menu_select,
    395 				.opt_flags = OPT_EXIT,
    396 			};
    397 			num_devs++;
    398 		}
    399 	}
    400 
    401 	menu_no = new_menu(MSG_avdisks,
    402 		menu_entries, num_devs, -1, -1,
    403 		(num_devs+1<3)?3:num_devs+1, 13,
    404 		MC_SCROLL | MC_NOCLEAR, NULL, NULL, NULL, NULL, MSG_cancel);
    405 	if (menu_no == -1)
    406 		return (struct part_entry) { };
    407 	process_menu(menu_no, &dev_num);
    408 	free_menu(menu_no);
    409 
    410 	if (dev_num < 0 || dev_num >= num_devs)
    411 		return (struct part_entry) { };
    412 
    413 	pm_retvalue = dev_num;
    414 	return disk_entries[dev_num];
    415 }
    416 
    417 /* Get unused raid*, cgd* or vnd* device */
    418 static int
    419 pm_manage_getfreenode(void *node, const char *d, structinfo_t *s)
    420 {
    421 	int i, ii, ok;
    422 	char buf[SSTRSIZE];
    423 	struct pm_devs *pm_i;
    424 
    425 	*(int*)node = -1;
    426 	for (i = 0; i < s->max; i++) {
    427 		ok = 1;
    428 		/* Check that node is not already reserved */
    429 		for (ii = 0; ii < s->max; ii++) {
    430 			if (*(int*)((char*)s->entry_enabled + s->entry_size
    431 			    * ii) == 0)
    432 				continue;
    433 			if (*(int*)((char*)s->entry_node + s->entry_size * ii)
    434 			    == i) {
    435 				ok = 0;
    436 				break;
    437 			}
    438 		}
    439 		if (! ok)
    440 			continue;
    441 		/* Check that node is not in the device list */
    442 		snprintf(buf, SSTRSIZE, "%s%d", d, i);
    443 		SLIST_FOREACH(pm_i, &pm_head, l)
    444 			if (! strcmp(pm_i->diskdev, buf)) {
    445 				ok = 0;
    446 				break;
    447 			}
    448 		if (ok) {
    449 			*(int*)node = i;
    450 			return i;
    451 		}
    452 	}
    453 	hit_enter_to_continue(NULL, MSG_nofreedev);
    454 	return -1;
    455 }
    456 
    457 /*
    458  * Show a line for a device, usually with full size in the right
    459  * column, alternatively (if != NULL) with no_size_display
    460  * instead in paranthesis (used for error displays or to note
    461  * a action that can be done to this device.
    462  */
    463 static void
    464 pm_fmt_disk_line(WINDOW *w, const char *line, const char *on,
    465     daddr_t total, const char *no_size_display)
    466 {
    467 	char out[STRSIZE], human[6];
    468 
    469 	if (on != NULL) {
    470 		snprintf(out, sizeof out, "%s %s %s", line,
    471 		    msg_string(MSG_pm_menu_on), on);
    472 		line = out;
    473 	}
    474 	if (no_size_display != NULL) {
    475 		wprintw(w, "   %-56s (%s)", line, no_size_display);
    476 	} else {
    477 		humanize_number(human, sizeof(human),
    478 	            total * 512, "",
    479 		    HN_AUTOSCALE, HN_B | HN_NOSPACE | HN_DECIMAL);
    480 		wprintw(w, "   %-56s %s", line, human);
    481 	}
    482 }
    483 
    484 /***
    485  RAIDs
    486  ***/
    487 
    488 static void
    489 pm_raid_menufmt(menudesc *m, int opt, void *arg)
    490 {
    491 	int i, ok = 0;
    492 	char buf[STRSIZE], rdev[STRSIZE], level[STRSIZE], *line;
    493 	struct raid_desc *dev_ptr = ((struct part_entry *)arg)[opt].dev_ptr;
    494 
    495 	if (dev_ptr->enabled == 0)
    496 		return;
    497 	buf[0] = '\0';
    498 	sprintf(rdev, "raid%d", dev_ptr->node);
    499 	for (i = 0; i < MAX_IN_RAID; i++) {
    500 		if (dev_ptr->comp[i].parts != NULL) {
    501 			strlcat(buf, dev_ptr->comp[i].name, sizeof buf);
    502 			strlcat(buf, " ", sizeof buf);
    503 			ok = 1;
    504 		}
    505 	}
    506 	if (ok) {
    507 		sprintf(level, "%u", dev_ptr->raid_level);
    508 		const char *args[] = { rdev, level };
    509 		line = str_arg_subst(msg_string(MSG_raid_menufmt),
    510 		    __arraycount(args), args);
    511 		pm_fmt_disk_line(m->mw, line, buf, dev_ptr->total_size, NULL);
    512 		free(line);
    513 	} else {
    514 		pm_fmt_disk_line(m->mw, buf, NULL, 0,
    515 		    msg_string(MSG_raid_err_menufmt));
    516 	}
    517 }
    518 
    519 static void
    520 pm_raid_edit_menufmt(menudesc *m, int opt, void *arg)
    521 {
    522 	int i;
    523 	char buf[STRSIZE];
    524 	struct raid_desc *dev_ptr = arg;
    525 
    526 	buf[0] = '\0';
    527 	switch (opt) {
    528 		case PMR_MENU_DEVS:
    529 			strlcpy(buf, msg_string(MSG_raid_disks_fmt),
    530 			    sizeof buf);
    531 			strlcat(buf, ": ", sizeof buf);
    532 			for (i = 0; i < MAX_IN_RAID; i++) {
    533 				if (dev_ptr->comp[i].parts == NULL ||
    534 				     dev_ptr->comp[i].is_spare)
    535 					continue;
    536 				strlcat(buf, " ", sizeof buf);
    537 				strlcat(buf, dev_ptr->comp[i].name, sizeof buf);
    538 			}
    539 			wprintw(m->mw, "%s", buf);
    540 			break;
    541 		case PMR_MENU_DEVSSPARE:
    542 			strlcpy(buf, msg_string(MSG_raid_spares_fmt),
    543 			    sizeof buf);
    544 			strlcat(buf, ": ", sizeof buf);
    545 			for (i = 0; i < MAX_IN_RAID; i++) {
    546 				if (dev_ptr->comp[i].parts == NULL ||
    547 				     !dev_ptr->comp[i].is_spare)
    548 					continue;
    549 				strlcat(buf, " ", sizeof buf);
    550 				strlcat(buf, dev_ptr->comp[i].name, sizeof buf);
    551 			}
    552 			wprintw(m->mw, "%s", buf);
    553 			break;
    554 		case PMR_MENU_RAIDLEVEL:
    555 			wprintw(m->mw, "%s: %u",
    556 			    msg_string(MSG_raid_level_fmt),
    557 			    dev_ptr->raid_level);
    558 			break;
    559 		case PMR_MENU_NUMROW:
    560 			wprintw(m->mw, "%s: %u",
    561 			    msg_string(MSG_raid_numrow_fmt), dev_ptr->numRow);
    562 			break;
    563 		case PMR_MENU_NUMCOL:
    564 			wprintw(m->mw, "%s: %u",
    565 			    msg_string(MSG_raid_numcol_fmt), dev_ptr->numCol);
    566 			break;
    567 		case PMR_MENU_NUMSPARE:
    568 			wprintw(m->mw, "%s: %u",
    569 			    msg_string(MSG_raid_numspare_fmt),
    570 			    dev_ptr->numSpare);
    571 			break;
    572 		case PMR_MENU_SECTPERSU:
    573 			wprintw(m->mw, "%s: %u",
    574 			    msg_string(MSG_raid_sectpersu_fmt),
    575 			    dev_ptr->sectPerSU);
    576 			break;
    577 		case PMR_MENU_SUSPERPARITYUNIT:
    578 			wprintw(m->mw, "%s: %u",
    579 			    msg_string(MSG_raid_superpar_fmt),
    580 			    dev_ptr->SUsPerParityUnit);
    581 			break;
    582 		case PMR_MENU_SUSPERRECONUNIT:
    583 			wprintw(m->mw, "%s: %u",
    584 			    msg_string(MSG_raid_superrec_fmt),
    585 			    dev_ptr->SUsPerReconUnit);
    586 			break;
    587 	}
    588 }
    589 
    590 static int
    591 pm_raid_set_value(menudesc *m, void *arg)
    592 {
    593 	int retvalue = -1;
    594 	int *out_var = NULL;
    595 	char buf[SSTRSIZE];
    596 	const char *msg_to_show = NULL;
    597 	struct raid_desc *dev_ptr = arg;
    598 
    599 	static menu_ent menuent_disk_adddel[] = {
    600 	    { .opt_name=MSG_add, .opt_flags=OPT_EXIT,
    601 	      .opt_action=pm_raid_disk_add },
    602 	    { .opt_name=MSG_remove, .opt_flags=OPT_EXIT,
    603 	      .opt_action=pm_raid_disk_del }
    604 	};
    605 	static int menu_disk_adddel = -1;
    606 	if (menu_disk_adddel == -1) {
    607 		menu_disk_adddel = new_menu(NULL, menuent_disk_adddel,
    608 			__arraycount(menuent_disk_adddel),
    609 			-1, -1, 0, 10, MC_NOCLEAR, NULL, NULL, NULL, NULL,
    610 			MSG_cancel);
    611 	}
    612 
    613 	switch (m->cursel) {
    614 		case PMR_MENU_DEVS:
    615 			pm_raid_curspare = 0;
    616 			process_menu(menu_disk_adddel, dev_ptr);
    617 			return 0;
    618 		case PMR_MENU_DEVSSPARE:
    619 			pm_raid_curspare = 1;
    620 			process_menu(menu_disk_adddel, dev_ptr);
    621 			return 0;
    622 		case PMR_MENU_RAIDLEVEL:
    623 			process_menu(MENU_raidlevel, &retvalue);
    624 			if (retvalue >= 0)
    625 				dev_ptr->raid_level = retvalue;
    626 			return 0;
    627 		case PMR_MENU_NUMROW:
    628 			hit_enter_to_continue(NULL, MSG_raid_nomultidim);
    629 			return 0;
    630 #if 0 /* notyet */
    631 			msg_to_show = MSG_raid_numrow_ask;
    632 			out_var = &(dev_ptr->numRow);
    633 			break;
    634 #endif
    635 		case PMR_MENU_NUMCOL:
    636 			msg_to_show = MSG_raid_numcol_ask;
    637 			out_var = &(dev_ptr->numCol);
    638 			break;
    639 		case PMR_MENU_NUMSPARE:
    640 			msg_to_show = MSG_raid_numspare_ask;
    641 			out_var = &(dev_ptr->numSpare);
    642 			break;
    643 		case PMR_MENU_SECTPERSU:
    644 			msg_to_show = MSG_raid_sectpersu_ask;
    645 			out_var = &(dev_ptr->sectPerSU);
    646 			break;
    647 		case PMR_MENU_SUSPERPARITYUNIT:
    648 			msg_to_show = MSG_raid_superpar_ask;
    649 			out_var = &(dev_ptr->SUsPerParityUnit);
    650 			break;
    651 		case PMR_MENU_SUSPERRECONUNIT:
    652 			msg_to_show = MSG_raid_superrec_ask;
    653 			out_var = &(dev_ptr->SUsPerReconUnit);
    654 			break;
    655 		case PMR_MENU_REMOVE:
    656 			dev_ptr->enabled = 0;
    657 			return 0;
    658 	}
    659 	if (out_var == NULL || msg_to_show == NULL)
    660 		return -1;
    661 	snprintf(buf, SSTRSIZE, "%d", *out_var);
    662 	msg_prompt_win(msg_to_show, -1, 18, 0, 0, buf, buf, SSTRSIZE);
    663 	if (atoi(buf) >= 0)
    664 		*out_var = atoi(buf);
    665 	return 0;
    666 }
    667 
    668 static void
    669 pm_raid_init(void *arg, void *none)
    670 {
    671 	struct raid_desc *dev_ptr = arg;
    672 	memset(dev_ptr, 0, sizeof(*dev_ptr));
    673 	*dev_ptr = (struct raid_desc) {
    674 		.enabled = 1,
    675 		.blocked = 0,
    676 		.sectPerSU = 32,
    677 		.SUsPerParityUnit = 1,
    678 		.SUsPerReconUnit = 1,
    679 	};
    680 }
    681 
    682 static int
    683 pm_raid_check(void *arg)
    684 {
    685 	size_t i, dev_num = 0;
    686 	daddr_t min_size = 0, cur_size = 0;
    687 	struct raid_desc *dev_ptr = arg;
    688 	struct disk_part_info info;
    689 	struct disk_partitions *parts;
    690 
    691 	if (dev_ptr->blocked)
    692 		return 0;
    693 
    694 	for (i = 0; i < MAX_IN_RAID; i++) {
    695 		if (dev_ptr->comp[i].parts != NULL) {
    696 			parts = dev_ptr->comp[i].parts;
    697 			if (!parts->pscheme->get_part_info(parts,
    698 			    dev_ptr->comp[i].id, &info))
    699 				continue;
    700 			cur_size = info.size;
    701 			if (cur_size < min_size || dev_num == 0)
    702 				min_size = cur_size;
    703 			if (dev_ptr->comp[i].is_spare)
    704 				continue;
    705 			dev_num++;
    706 		}
    707 	}
    708 
    709 	/* Calculate sum of available space */
    710 	if (dev_num > 0) {
    711 		switch (dev_ptr->raid_level) {
    712 			case 0:
    713 				dev_ptr->total_size = min_size * dev_num;
    714 				break;
    715 			case 1:
    716 				dev_ptr->total_size = min_size;
    717 				break;
    718 			case 4:
    719 			case 5:
    720 				dev_ptr->total_size = min_size * (dev_num - 1);
    721 				break;
    722 		}
    723 		pm_manage_getfreenode(&(dev_ptr->node), "raid", &raids_t_info);
    724 		if (dev_ptr->node < 0)
    725 			dev_ptr->enabled = 0;
    726 	}
    727 	else
    728 		dev_ptr->enabled = 0;
    729 	return dev_ptr->enabled;
    730 }
    731 
    732 static int
    733 pm_raid_disk_add(menudesc *m, void *arg)
    734 {
    735 	int i;
    736 	struct raid_desc *dev_ptr = arg;
    737 	struct part_entry disk_entrie = pm_dev_list(PM_RAID);
    738 	if (pm_retvalue < 0)
    739 		return pm_retvalue;
    740 
    741 	for (i = 0; i < MAX_IN_RAID; i++)
    742 		if (dev_ptr->comp[i].parts == NULL) {
    743 			dev_ptr->comp[i].parts = disk_entrie.parts;
    744 			dev_ptr->comp[i].id = disk_entrie.id;
    745 			dev_ptr->comp[i].is_spare = pm_raid_curspare;
    746 			strlcpy(dev_ptr->comp[i].name, disk_entrie.fullname,
    747 			    sizeof dev_ptr->comp[i].name);
    748 			if (pm_raid_curspare)
    749 				dev_ptr->numSpare++;
    750 			else
    751 				dev_ptr->numCol++;
    752 			dev_ptr->numRow = 1;
    753 			break;
    754 		}
    755 	return 0;
    756 }
    757 
    758 static int
    759 pm_raid_disk_del(menudesc *m, void *arg)
    760 {
    761 	int retvalue = -1, num_devs = 0;
    762 	int i, pm_cur;
    763 	int menu_no;
    764 	struct raid_desc *dev_ptr = arg;
    765 	menu_ent menu_entries[MAX_IN_RAID];
    766 	struct part_entry submenu_args[MAX_IN_RAID];
    767 
    768 	for (i = 0; i < MAX_IN_RAID; i++) {
    769 		if (dev_ptr->comp[i].parts == NULL ||
    770 			dev_ptr->comp[i].is_spare != pm_raid_curspare)
    771 			continue;
    772 		menu_entries[num_devs] = (struct menu_ent) {
    773 			.opt_name = dev_ptr->comp[i].name,
    774 			.opt_action = set_menu_select,
    775 			.opt_flags = OPT_EXIT,
    776 		};
    777 		submenu_args[num_devs].dev_ptr = dev_ptr;
    778 		submenu_args[num_devs].index = i;
    779 		num_devs++;
    780 	}
    781 
    782 	menu_no = new_menu(MSG_raid_disks,
    783 		menu_entries, num_devs, -1, -1,
    784 		(num_devs+1<3)?3:num_devs+1, 13,
    785 		MC_SCROLL | MC_NOCLEAR, NULL, NULL, NULL, NULL, MSG_cancel);
    786 	if (menu_no == -1)
    787 		return -1;
    788 	process_menu(menu_no, &retvalue);
    789 	free_menu(menu_no);
    790 
    791 	if (retvalue < 0 || retvalue >= num_devs)
    792 		return -1;
    793 
    794 	pm_cur = submenu_args[retvalue].index;
    795 
    796 	if (dev_ptr->comp[pm_cur].is_spare)
    797 		dev_ptr->numSpare--;
    798 	else
    799 		dev_ptr->numCol--;
    800 	dev_ptr->numRow = (dev_ptr->numCol)?1:0;
    801 	dev_ptr->comp[pm_cur].parts = NULL;
    802 
    803 	return 0;
    804 }
    805 
    806 static int
    807 pm_raid_commit(void)
    808 {
    809 	int i, ii;
    810 	FILE *f;
    811 	char f_name[STRSIZE], devname[STRSIZE];
    812 
    813 	for (i = 0; i < MAX_RAID; i++) {
    814 		if (! pm_raid_check(&raids[i]))
    815 			continue;
    816 
    817 		/* Generating configure file for our raid */
    818 		snprintf(f_name, SSTRSIZE, "/tmp/raid.%d.conf", raids[i].node);
    819 		f = fopen(f_name, "w");
    820 		if (f == NULL) {
    821 			endwin();
    822 			(void)fprintf(stderr,
    823 			    "Could not open %s for writing\n", f_name);
    824 			if (logfp)
    825 				(void)fprintf(logfp,
    826 				    "Could not open %s for writing\n", f_name);
    827 			return 1;
    828 		}
    829 		scripting_fprintf(NULL, "cat <<EOF >%s\n", f_name);
    830 		scripting_fprintf(f, "START array\n%d %d %d\n",
    831 		    raids[i].numRow, raids[i].numCol, raids[i].numSpare);
    832 
    833 		scripting_fprintf(f, "\nSTART disks\n");
    834 		for (ii = 0; ii < MAX_IN_RAID; ii++) {
    835 			if (raids[i].comp[ii].parts != NULL &&
    836 			    !raids[i].comp[ii].is_spare) {
    837 				strcpy(devname, raids[i].comp[ii].name);
    838 				if (raids[i].comp[ii].parts != NULL &&
    839 				    raids[i].comp[ii].id != NO_PART) {
    840 					/* wedge may have moved */
    841 					raids[i].comp[ii].parts->pscheme->
    842 					    get_part_device(
    843 					    raids[i].comp[ii].parts,
    844 					    raids[i].comp[ii].id,
    845 					    devname, sizeof devname, NULL,
    846 					    logical_name, true, true);
    847 					raids[i].comp[ii].parts->pscheme->
    848 					    get_part_device(
    849 					    raids[i].comp[ii].parts,
    850 					    raids[i].comp[ii].id,
    851 					    raids[i].comp[ii].name,
    852 					    sizeof raids[i].comp[ii].name,
    853 					    NULL, plain_name, true, true);
    854 				}
    855 				scripting_fprintf(f,  "%s\n", devname);
    856 			}
    857 		}
    858 
    859 		scripting_fprintf(f, "\nSTART spare\n");
    860 		for (ii = 0; ii < MAX_IN_RAID; ii++) {
    861 			if (raids[i].comp[ii].parts != NULL &&
    862 			    raids[i].comp[ii].is_spare) {
    863 				strcpy(devname, raids[i].comp[ii].name);
    864 				if (raids[i].comp[ii].parts != NULL &&
    865 				    raids[i].comp[ii].id != NO_PART) {
    866 					/* wedge may have moved */
    867 					raids[i].comp[ii].parts->pscheme->
    868 					    get_part_device(
    869 					    raids[i].comp[ii].parts,
    870 					    raids[i].comp[ii].id,
    871 					    devname, sizeof devname, NULL,
    872 					    logical_name, true, true);
    873 					raids[i].comp[ii].parts->pscheme->
    874 					    get_part_device(
    875 					    raids[i].comp[ii].parts,
    876 					    raids[i].comp[ii].id,
    877 					    raids[i].comp[ii].name,
    878 					    sizeof raids[i].comp[ii].name,
    879 					    NULL, plain_name, true, true);
    880 				}
    881 
    882 				scripting_fprintf(f,  "%s\n",
    883 				    devname);
    884 			}
    885 		}
    886 
    887 		scripting_fprintf(f, "\nSTART layout\n%d %d %d %d\n",
    888 		    raids[i].sectPerSU,	raids[i].SUsPerParityUnit,
    889 		    raids[i].SUsPerReconUnit, raids[i].raid_level);
    890 
    891 		scripting_fprintf(f, "\nSTART queue\nfifo 100\n\n");
    892 		scripting_fprintf(NULL, "EOF\n");
    893 		fclose (f);
    894 		fflush(NULL);
    895 
    896 		/* Raid initialization */
    897 		if (run_program(RUN_DISPLAY | RUN_PROGRESS,
    898 			"raidctl -C %s raid%d", f_name, raids[i].node) == 0 &&
    899 		    run_program(RUN_DISPLAY | RUN_PROGRESS,
    900 			"raidctl -I %d raid%d", rand(), raids[i].node) == 0 &&
    901 		    run_program(RUN_DISPLAY | RUN_PROGRESS,
    902 		        "raidctl -vi raid%d", raids[i].node) == 0 &&
    903 		    run_program(RUN_DISPLAY | RUN_PROGRESS,
    904 			"raidctl -v -A yes raid%d", raids[i].node) == 0) {
    905 			/*
    906 			 * RAID creation done, remove it from list to
    907 			 * prevent its repeated reinitialization
    908 			 */
    909 			raids[i].blocked = 1;
    910 		}
    911 	}
    912 	return 0;
    913 }
    914 
    915 /***
    916  VND
    917  ***/
    918 
    919 static void
    920 pm_vnd_menufmt(menudesc *m, int opt, void *arg)
    921 {
    922 	struct vnd_desc *dev_ptr = ((struct part_entry *)arg)[opt].dev_ptr;
    923 	char dev[STRSIZE];
    924 
    925 	if (dev_ptr->enabled == 0)
    926 		return;
    927 	sprintf(dev, "vnd%d", dev_ptr->node);
    928 	if (strlen(dev_ptr->filepath) < 1)
    929 		pm_fmt_disk_line(m->mw, dev, NULL,
    930 		    0, msg_string(MSG_vnd_err_menufmt));
    931 	else if (dev_ptr->is_exist)
    932 		pm_fmt_disk_line(m->mw, dev, dev_ptr->filepath,
    933 		    0, msg_string(MSG_vnd_assign));
    934 	else
    935 		pm_fmt_disk_line(m->mw, dev, dev_ptr->filepath,
    936 		    dev_ptr->size, NULL);
    937 }
    938 
    939 static int
    940 max_msg_length(const msg *p, size_t cnt)
    941 {
    942 	int len, m = 0;
    943 
    944 	while (cnt > 0) {
    945 		len = strlen(msg_string(*p));
    946 		if (len > m)
    947 			m = len;
    948 		cnt--; p++;
    949 	}
    950 
    951 	return m;
    952 }
    953 
    954 static void
    955 pm_vnd_edit_menufmt(menudesc *m, int opt, void *arg)
    956 {
    957 	struct vnd_desc *dev_ptr = arg;
    958 	char buf[SSTRSIZE];
    959 	strcpy(buf, "-");
    960 	static int lcol_width;
    961 	if (lcol_width == 0) {
    962 		static const msg labels[] = {
    963 		    MSG_vnd_path_fmt, MSG_vnd_assign_fmt, MSG_vnd_size_fmt,
    964 		    MSG_vnd_ro_fmt, MSG_vnd_geom_fmt, MSG_vnd_bps_fmt,
    965 		    MSG_vnd_spt_fmt, MSG_vnd_tpc_fmt, MSG_vnd_cyl_fmt
    966 		};
    967 		lcol_width = max_msg_length(labels, __arraycount(labels)) + 3;
    968 	}
    969 
    970 	switch (opt) {
    971 		case PMV_MENU_FILEPATH:
    972 			wprintw(m->mw, "%*s %s", -lcol_width,
    973 			    msg_string(MSG_vnd_path_fmt), dev_ptr->filepath);
    974 			break;
    975 		case PMV_MENU_EXIST:
    976 			wprintw(m->mw, "%*s %s", -lcol_width,
    977 			    msg_string(MSG_vnd_assign_fmt),
    978 			    dev_ptr->is_exist?
    979 				msg_string(MSG_No) : msg_string(MSG_Yes));
    980 			break;
    981 		case PMV_MENU_SIZE:
    982 			if (!dev_ptr->is_exist)
    983 				snprintf(buf, SSTRSIZE, "%" PRIu64,
    984 				    dev_ptr->size / sizemult);
    985 			wprintw(m->mw, "%*s %s", -lcol_width,
    986 			    msg_string(MSG_vnd_size_fmt), buf);
    987 			break;
    988 		case PMV_MENU_RO:
    989 			wprintw(m->mw, "%*s %s", -lcol_width,
    990 			    msg_string(MSG_vnd_ro_fmt),
    991 			    dev_ptr->readonly?
    992 				msg_string(MSG_Yes) : msg_string(MSG_No));
    993 			break;
    994 		case PMV_MENU_MGEOM:
    995 			if (!dev_ptr->is_exist)
    996 				snprintf(buf, SSTRSIZE, "%s",
    997 				    dev_ptr->manual_geom?
    998 				    msg_string(MSG_Yes) : msg_string(MSG_No));
    999 			wprintw(m->mw, "%*s %s", -lcol_width,
   1000 			    msg_string(MSG_vnd_geom_fmt), buf);
   1001 			break;
   1002 		case PMV_MENU_SECSIZE:
   1003 			if (dev_ptr->manual_geom && !dev_ptr->is_exist)
   1004 				snprintf(buf, SSTRSIZE, "%d", dev_ptr->secsize);
   1005 			wprintw(m->mw, "%*s %s", -lcol_width,
   1006 			    msg_string(MSG_vnd_bps_fmt), buf);
   1007 			break;
   1008 		case PMV_MENU_NSECTORS:
   1009 			if (dev_ptr->manual_geom && !dev_ptr->is_exist)
   1010 				snprintf(buf, SSTRSIZE, "%d",
   1011 				    dev_ptr->nsectors);
   1012 			wprintw(m->mw, "%*s %s", -lcol_width,
   1013 			    msg_string(MSG_vnd_spt_fmt), buf);
   1014 			break;
   1015 		case PMV_MENU_NTRACKS:
   1016 			if (dev_ptr->manual_geom && !dev_ptr->is_exist)
   1017 				snprintf(buf, SSTRSIZE, "%d", dev_ptr->ntracks);
   1018 			wprintw(m->mw, "%*s %s", -lcol_width,
   1019 			    msg_string(MSG_vnd_tpc_fmt), buf);
   1020 			break;
   1021 		case PMV_MENU_NCYLINDERS:
   1022 			if (dev_ptr->manual_geom && !dev_ptr->is_exist)
   1023 				snprintf(buf, SSTRSIZE, "%d",
   1024 				    dev_ptr->ncylinders);
   1025 			wprintw(m->mw, "%*s %s", -lcol_width,
   1026 			    msg_string(MSG_vnd_cyl_fmt), buf);
   1027 			break;
   1028 	}
   1029 }
   1030 
   1031 static int
   1032 pm_vnd_set_value(menudesc *m, void *arg)
   1033 {
   1034 	struct vnd_desc *dev_ptr = arg;
   1035 	char buf[STRSIZE];
   1036 	const char *msg_to_show = NULL;
   1037 	int *out_var = NULL;
   1038 
   1039 	switch (m->cursel) {
   1040 		case PMV_MENU_FILEPATH:
   1041 			msg_prompt_win(MSG_vnd_path_ask, -1, 18, 0, 0,
   1042 			    dev_ptr->filepath, dev_ptr->filepath, STRSIZE);
   1043 			if (dev_ptr->filepath[0] != '/') {
   1044 				strlcpy(buf, dev_ptr->filepath, MOUNTLEN);
   1045 				snprintf(dev_ptr->filepath, MOUNTLEN, "/%s",
   1046 				    buf);
   1047 			}
   1048 			if (dev_ptr->filepath[strlen(dev_ptr->filepath) - 1]
   1049 			    == '/')
   1050 				dev_ptr->filepath[strlen(dev_ptr->filepath)
   1051 				     - 1] = '\0';
   1052 			return 0;
   1053 		case PMV_MENU_EXIST:
   1054 			dev_ptr->is_exist = !dev_ptr->is_exist;
   1055 			return 0;
   1056 		case PMV_MENU_SIZE:
   1057 			if (dev_ptr->is_exist)
   1058 				return 0;
   1059 
   1060 			pm_edit_size_value(MSG_vnd_size_ask, pm->dlcylsize,
   1061 			    &dev_ptr->size);
   1062 
   1063 			break;
   1064 		case PMV_MENU_RO:
   1065 			dev_ptr->readonly = !dev_ptr->readonly;
   1066 			return 0;
   1067 		case PMV_MENU_MGEOM:
   1068 			if (dev_ptr->is_exist)
   1069 				return 0;
   1070 			dev_ptr->manual_geom = !dev_ptr->manual_geom;
   1071 			return 0;
   1072 		case PMV_MENU_SECSIZE:
   1073 			if (!dev_ptr->manual_geom || dev_ptr->is_exist)
   1074 				return 0;
   1075 			msg_to_show = MSG_vnd_bps_ask;
   1076 			out_var = &(dev_ptr->secsize);
   1077 			break;
   1078 		case PMV_MENU_NSECTORS:
   1079 			if (!dev_ptr->manual_geom || dev_ptr->is_exist)
   1080 				return 0;
   1081 			msg_to_show = MSG_vnd_spt_ask;
   1082 			out_var = &(dev_ptr->nsectors);
   1083 			break;
   1084 		case PMV_MENU_NTRACKS:
   1085 			if (!dev_ptr->manual_geom || dev_ptr->is_exist)
   1086 				return 0;
   1087 			msg_to_show = MSG_vnd_tpc_ask;
   1088 			out_var = &(dev_ptr->ntracks);
   1089 			break;
   1090 		case PMV_MENU_NCYLINDERS:
   1091 			if (!dev_ptr->manual_geom || dev_ptr->is_exist)
   1092 				return 0;
   1093 			msg_to_show = MSG_vnd_cyl_ask;
   1094 			out_var = &(dev_ptr->ncylinders);
   1095 			break;
   1096 		case PMV_MENU_REMOVE:
   1097 			dev_ptr->enabled = 0;
   1098 			return 0;
   1099 	}
   1100 	if (out_var == NULL || msg_to_show == NULL)
   1101 		return -1;
   1102 	snprintf(buf, SSTRSIZE, "%d", *out_var);
   1103 	msg_prompt_win(msg_to_show, -1, 18, 0, 0, buf, buf, SSTRSIZE);
   1104 	if (atoi(buf) >= 0)
   1105 		*out_var = atoi(buf);
   1106 	return 0;
   1107 }
   1108 
   1109 static void
   1110 pm_vnd_init(void *arg, void *none)
   1111 {
   1112 	struct vnd_desc *dev_ptr = arg;
   1113 	memset(dev_ptr, 0, sizeof(*dev_ptr));
   1114 	*dev_ptr = (struct vnd_desc) {
   1115 		.enabled = 1,
   1116 		.blocked = 0,
   1117 		.filepath[0] = '\0',
   1118 		.is_exist = 0,
   1119 		.size = 1024,
   1120 		.readonly = 0,
   1121 		.manual_geom = 0,
   1122 		.secsize = 512,
   1123 		.nsectors = 18,
   1124 		.ntracks = 2,
   1125 		.ncylinders = 80
   1126 	};
   1127 }
   1128 
   1129 static int
   1130 pm_vnd_check(void *arg)
   1131 {
   1132 	struct vnd_desc *dev_ptr = arg;
   1133 
   1134 	if (dev_ptr->blocked)
   1135 		return 0;
   1136 	if (strlen(dev_ptr->filepath) < 1 ||
   1137 			dev_ptr->size < 1)
   1138 		dev_ptr->enabled = 0;
   1139 	else {
   1140 		pm_manage_getfreenode(&(dev_ptr->node), "vnd", &vnds_t_info);
   1141 		if (dev_ptr->node < 0)
   1142 			dev_ptr->enabled = 0;
   1143 	}
   1144 	return dev_ptr->enabled;
   1145 }
   1146 
   1147 /* XXX: vndconfig always return 0? */
   1148 static int
   1149 pm_vnd_commit(void)
   1150 {
   1151 	int i, error;
   1152 	char r_o[3], buf[MOUNTLEN+3], resultpath[STRSIZE];
   1153 	const char *mpath, *mp_suit = NULL, *rp;
   1154 	struct pm_devs *pm_i, *pm_suit = NULL;
   1155 	part_id id, part_suit = NO_PART;
   1156 	struct disk_part_info info;
   1157 
   1158 	for (i = 0; i < MAX_VND; i++) {
   1159 		error = 0;
   1160 		if (! pm_vnd_check(&vnds[i]))
   1161 			continue;
   1162 
   1163 		/* Trying to assign target device */
   1164 		SLIST_FOREACH(pm_i, &pm_head, l) {
   1165 			for (id = 0; id < pm_i->parts->num_part; id++) {
   1166 				if (!pm_i->parts->pscheme->get_part_info(
   1167 				    pm_i->parts, id, &info))
   1168 					continue;
   1169 				if (info.flags & (PTI_SEC_CONTAINER|
   1170 				    PTI_WHOLE_DISK|PTI_PSCHEME_INTERNAL|
   1171 				    PTI_RAW_PART))
   1172 					continue;
   1173 				if (info.last_mounted == NULL ||
   1174 				    info.last_mounted[0] == 0)
   1175 					continue;
   1176 				mpath = info.last_mounted;
   1177 				strcpy(buf, mpath);
   1178 				if (buf[strlen(buf)-1] != '/')
   1179 					strcat(buf, "/");
   1180 				if (strstr(vnds[i].filepath, buf) !=
   1181 				    vnds[i].filepath)
   1182 					continue;
   1183 				if (part_suit == NO_PART || pm_suit == NULL ||
   1184 				    strlen(buf) > strlen(mp_suit)) {
   1185 					part_suit = id;
   1186 					pm_suit = pm_i;
   1187 					mp_suit = mpath;
   1188 				}
   1189 			}
   1190 		}
   1191 		if (part_suit == NO_PART || pm_suit == NULL ||
   1192 		   mp_suit == NULL)
   1193 			continue;
   1194 
   1195 		/* Mounting assigned partition and try to get real file path*/
   1196 		if (pm_mount(pm_suit, part_suit) != 0)
   1197 			continue;
   1198 		rp = pm_get_mount(pm_suit, part_suit);
   1199 		snprintf(resultpath, STRSIZE, "%s/%s",
   1200 			rp,
   1201 			&(vnds[i].filepath[strlen(rp)]));
   1202 
   1203 		strcpy(r_o, vnds[i].readonly?"-r":"");
   1204 		/* If this is a new image */
   1205 		if (!vnds[i].is_exist) {
   1206 			run_program(RUN_DISPLAY | RUN_PROGRESS, "mkdir -p %s ",
   1207 			    dirname(resultpath));
   1208 			if (error == 0)
   1209 				error = run_program(RUN_DISPLAY | RUN_PROGRESS,
   1210 				    "dd if=/dev/zero of=%s bs=1m "
   1211 				    "count=% " PRIi64 " progress=100 "
   1212 				    "msgfmt=human",
   1213 				    resultpath, vnds[i].size*(MEG/512));
   1214 		}
   1215 		if (error)
   1216 			continue;
   1217 
   1218 		/* If this is a new image with manual geometry */
   1219 		if (!vnds[i].is_exist && vnds[i].manual_geom)
   1220 			error = run_program(RUN_DISPLAY | RUN_PROGRESS,
   1221 			    "vndconfig %s vnd%d %s %d %d %d %d",
   1222 			    r_o, vnds[i].node,
   1223 			    resultpath, vnds[i].secsize,
   1224 			    vnds[i].nsectors,
   1225 			    vnds[i].ntracks, vnds[i].ncylinders);
   1226 		else
   1227 			/* If this is a existing image or image without manual
   1228 			 * geometry */
   1229 			error = run_program(RUN_DISPLAY | RUN_PROGRESS,
   1230 			    "vndconfig %s vnd%d %s",
   1231 			    r_o, vnds[i].node, resultpath);
   1232 
   1233 		if (error == 0) {
   1234 			vnds[i].blocked = 1;
   1235 			vnds[i].pm_part = part_suit;
   1236 			vnds[i].pm = pm_suit;
   1237 			vnds[i].pm->blocked++;
   1238 		}
   1239 	}
   1240 	return 0;
   1241 }
   1242 
   1243 /***
   1244  CGD
   1245  ***/
   1246 
   1247 static void
   1248 pm_cgd_menufmt(menudesc *m, int opt, void *arg)
   1249 {
   1250 	struct cgd_desc *dev_ptr = ((struct part_entry *)arg)[opt].dev_ptr;
   1251 	char desc[STRSIZE];
   1252 	struct disk_part_info info;
   1253 
   1254 	if (dev_ptr->enabled == 0)
   1255 		return;
   1256 	if (dev_ptr->pm == NULL)
   1257 		wprintw(m->mw, "%s", msg_string(MSG_cgd_err_menufmt));
   1258 	else {
   1259 		snprintf(desc, sizeof desc, "cgd%d (%s-%d)",
   1260 		    dev_ptr->node, dev_ptr->enc_type, dev_ptr->key_size);
   1261 		dev_ptr->pm->parts->pscheme->get_part_info(dev_ptr->pm->parts,
   1262 		    dev_ptr->pm_part, &info);
   1263 		pm_fmt_disk_line(m->mw, desc, dev_ptr->pm_name,
   1264 		    info.size, NULL);
   1265 	}
   1266 }
   1267 
   1268 static void
   1269 pm_cgd_edit_menufmt(menudesc *m, int opt, void *arg)
   1270 {
   1271 	struct cgd_desc *dev_ptr = arg;
   1272 	switch (opt) {
   1273 		case PMC_MENU_DEV:
   1274 			wprintw(m->mw, "%-15s: %s",
   1275 			    msg_string(MSG_cgd_dev_fmt), dev_ptr->pm_name);
   1276 			break;
   1277 		case PMC_MENU_ENCTYPE:
   1278 			wprintw(m->mw, "%-15s: %s",
   1279 			    msg_string(MSG_cgd_enc_fmt), dev_ptr->enc_type);
   1280 			break;
   1281 		case PMC_MENU_KEYSIZE:
   1282 			wprintw(m->mw, "%-15s: %d",
   1283 			    msg_string(MSG_cgd_key_fmt), dev_ptr->key_size);
   1284 			break;
   1285 		case PMC_MENU_IVTYPE:
   1286 			wprintw(m->mw, "%-15s: %s",
   1287 			    msg_string(MSG_cgd_iv_fmt), dev_ptr->iv_type);
   1288 			break;
   1289 		case PMC_MENU_KEYGENTYPE:
   1290 			wprintw(m->mw, "%-15s: %s",
   1291 			    msg_string(MSG_cgd_keygen_fmt), dev_ptr->keygen_type);
   1292 			break;
   1293 		case PMC_MENU_VERIFYTYPE:
   1294 			wprintw(m->mw, "%-15s: %s",
   1295 			    msg_string(MSG_cgd_verif_fmt), dev_ptr->verify_type);
   1296 			break;
   1297 	}
   1298 }
   1299 
   1300 static int
   1301 pm_cgd_set_value(menudesc *m, void *arg)
   1302 {
   1303 	char *retstring;
   1304 	struct cgd_desc *dev_ptr = arg;
   1305 
   1306 	switch (m->cursel) {
   1307 		case PMC_MENU_DEV:
   1308 			pm_cgd_disk_set(dev_ptr, NULL);
   1309 			return 0;
   1310 		case PMC_MENU_ENCTYPE:
   1311 			process_menu(MENU_cgd_enctype, &retstring);
   1312 			dev_ptr->enc_type = retstring;
   1313 			if (! strcmp(retstring, "aes-xts"))
   1314 				dev_ptr->key_size = 256;
   1315 			if (! strcmp(retstring, "aes-cbc"))
   1316 				dev_ptr->key_size = 192;
   1317 			if (! strcmp(retstring, "blowfish-cbc"))
   1318 				dev_ptr->key_size = 128;
   1319 			if (! strcmp(retstring, "3des-cbc"))
   1320 				dev_ptr->key_size = 192;
   1321 			return 0;
   1322 		case PMC_MENU_KEYSIZE:
   1323 			if (! strcmp(dev_ptr->enc_type, "aes-xts"))
   1324 				dev_ptr->key_size +=
   1325 					(dev_ptr->key_size < 512)? 256 : -256;
   1326 			if (! strcmp(dev_ptr->enc_type, "aes-cbc"))
   1327 				dev_ptr->key_size +=
   1328 					(dev_ptr->key_size < 256)? 64 : -128;
   1329 			if (! strcmp(dev_ptr->enc_type, "blowfish-cbc"))
   1330 				dev_ptr->key_size = 128;
   1331 			if (! strcmp(dev_ptr->enc_type, "3des-cbc"))
   1332 				dev_ptr->key_size = 192;
   1333 			return 0;
   1334 		case PMC_MENU_IVTYPE:
   1335 			process_menu(MENU_cgd_ivtype, &retstring);
   1336 			dev_ptr->iv_type = retstring;
   1337 			return 0;
   1338 		case PMC_MENU_KEYGENTYPE:
   1339 			process_menu(MENU_cgd_keygentype, &retstring);
   1340 			dev_ptr->keygen_type = retstring;
   1341 			return 0;
   1342 		case PMC_MENU_VERIFYTYPE:
   1343 			process_menu(MENU_cgd_verifytype, &retstring);
   1344 			dev_ptr->verify_type = retstring;
   1345 			return 0;
   1346 		case PMC_MENU_REMOVE:
   1347 			dev_ptr->enabled = 0;
   1348 			return 0;
   1349 	}
   1350 	return -1;
   1351 }
   1352 
   1353 static void
   1354 pm_cgd_init(void *arg1, void *arg2)
   1355 {
   1356 	struct cgd_desc *dev_ptr = arg1;
   1357 	struct part_entry *disk_entrie = arg2;
   1358 
   1359 	memset(dev_ptr, 0, sizeof(*dev_ptr));
   1360 	*dev_ptr = (struct cgd_desc) {
   1361 		.enabled = 1,
   1362 		.blocked = 0,
   1363 		.pm = NULL,
   1364 		.pm_name[0] = '\0',
   1365 		.pm_part = 0,
   1366 		.keygen_type = "pkcs5_pbkdf2/sha1",
   1367 		.verify_type = "disklabel",
   1368 		.enc_type = "aes-xts",
   1369 		.iv_type = "encblkno1",
   1370 		.key_size = 256,
   1371 	};
   1372 	if (disk_entrie != NULL) {
   1373 		disk_entrie->parts->pscheme->get_part_device(
   1374 		    disk_entrie->parts,  disk_entrie->id,
   1375 		    disk_entrie->fullname, sizeof(disk_entrie->fullname),
   1376 		    NULL, logical_name, false, true);
   1377 		pm_cgd_disk_set(dev_ptr, disk_entrie);
   1378 	}
   1379 }
   1380 
   1381 static int
   1382 pm_cgd_check(void *arg)
   1383 {
   1384 	struct cgd_desc *dev_ptr = arg;
   1385 
   1386 	if (dev_ptr->blocked)
   1387 		return 0;
   1388 	if (dev_ptr->pm == NULL)
   1389 		dev_ptr->enabled = 0;
   1390 	else {
   1391 		pm_manage_getfreenode(&(dev_ptr->node), "cgd", &cgds_t_info);
   1392 		if (dev_ptr->node < 0)
   1393 			dev_ptr->enabled = 0;
   1394 	}
   1395 	return dev_ptr->enabled;
   1396 }
   1397 
   1398 static int
   1399 pm_cgd_disk_set(struct cgd_desc *dev_ptr, struct part_entry *disk_entrie)
   1400 {
   1401 	int alloc_disk_entrie = 0;
   1402 
   1403 	if (disk_entrie == NULL) {
   1404 		alloc_disk_entrie = 1;
   1405 		disk_entrie = malloc (sizeof(struct part_entry));
   1406 		if (disk_entrie == NULL)
   1407 			return -2;
   1408 		*disk_entrie = pm_dev_list(PM_CGD);
   1409 		if (pm_retvalue < 0) {
   1410 			free(disk_entrie);
   1411 			return -1;
   1412 		}
   1413 	}
   1414 	dev_ptr->pm = disk_entrie->dev_ptr;
   1415 	dev_ptr->pm_part = disk_entrie->id;
   1416 	strlcpy(dev_ptr->pm_name, disk_entrie->fullname, SSTRSIZE);
   1417 
   1418 	if (alloc_disk_entrie)
   1419 		free(disk_entrie);
   1420 	return 0;
   1421 }
   1422 
   1423 int
   1424 pm_cgd_edit_new(struct pm_devs *mypm, part_id id)
   1425 {
   1426 	struct part_entry pe = { .id = id, .parts = mypm->parts,
   1427 	    .dev_ptr = mypm, .type = PM_CGD };
   1428 
   1429 	return pm_edit(PMC_MENU_END, pm_cgd_edit_menufmt,
   1430 		pm_cgd_set_value, pm_cgd_check, pm_cgd_init,
   1431 		&pe, NULL, 0, &cgds_t_info);
   1432 }
   1433 
   1434 int
   1435 pm_cgd_edit_old(struct part_entry *pe)
   1436 {
   1437 	return pm_edit(PMC_MENU_END, pm_cgd_edit_menufmt,
   1438 		pm_cgd_set_value, pm_cgd_check, pm_cgd_init,
   1439 		pe->dev_ptr != NULL ? pe : NULL,
   1440 		pe->dev_ptr, 0, &cgds_t_info);
   1441 }
   1442 
   1443 static int
   1444 pm_cgd_commit(void)
   1445 {
   1446 	char devname[STRSIZE];
   1447 	int i, error = 0;
   1448 
   1449 	for (i = 0; i < MAX_CGD; i++) {
   1450 		if (! pm_cgd_check(&cgds[i]))
   1451 			continue;
   1452 		if (run_program(RUN_DISPLAY | RUN_PROGRESS,
   1453 			"cgdconfig -g -V %s -i %s -k %s -o /tmp/cgd.%d.conf"
   1454 			" %s %d", cgds[i].verify_type,
   1455 			cgds[i].iv_type, cgds[i].keygen_type, cgds[i].node,
   1456 			cgds[i].enc_type, cgds[i].key_size) != 0) {
   1457 			error++;
   1458 			continue;
   1459 		}
   1460 		if (cgds[i].pm != NULL && cgds[i].pm->parts != NULL) {
   1461 			/* wedge device names may have changed */
   1462 			cgds[i].pm->parts->pscheme->get_part_device(
   1463 			    cgds[i].pm->parts, cgds[i].pm_part,
   1464 			    devname, sizeof devname, NULL,
   1465 			    logical_name, true, true);
   1466 			cgds[i].pm->parts->pscheme->get_part_device(
   1467 			    cgds[i].pm->parts, cgds[i].pm_part,
   1468 			    cgds[i].pm_name, sizeof cgds[i].pm_name, NULL,
   1469 			    plain_name, false, true);
   1470 		} else {
   1471 			continue;
   1472 		}
   1473 		if (run_program(RUN_DISPLAY | RUN_PROGRESS,
   1474 			"cgdconfig -V re-enter cgd%d '%s' /tmp/cgd.%d.conf",
   1475 			cgds[i].node, devname, cgds[i].node) != 0) {
   1476 			error++;
   1477 			continue;
   1478 		}
   1479 		cgds[i].pm->blocked++;
   1480 		cgds[i].blocked = 1;
   1481 	}
   1482 	return error;
   1483 }
   1484 
   1485 /***
   1486  LVM
   1487  ***/
   1488 
   1489 /* Add lvm logical volumes to pm list */
   1490 /* XXX: rewrite */
   1491 static void
   1492 pm_lvm_find(void)
   1493 {
   1494 	int i, ii, already_found;
   1495 	char dev[STRSIZE];
   1496 	struct pm_devs *pm_i;
   1497 
   1498 	for (i = 0; i < MAX_LVM_VG; i++) {
   1499 		if (! lvms[i].blocked)
   1500 			continue;
   1501 		for (ii = 0; ii < MAX_LVM_LV; ii++) {
   1502 			if (! lvms[i].lv[ii].blocked || lvms[i].lv[ii].size < 1)
   1503 				continue;
   1504 			snprintf(dev, STRSIZE, "%s/%s", lvms[i].name,
   1505 			    lvms[i].lv[ii].name);
   1506 			already_found = 0;
   1507 			SLIST_FOREACH(pm_i, &pm_head, l)
   1508 				if (!already_found && strcmp(pm_i->diskdev,
   1509 				    dev) == 0) {
   1510 					pm_i->found = 1;
   1511 					already_found = 1;
   1512 				}
   1513 			if (already_found)
   1514 				/* We already added this device, skipping */
   1515 				continue;
   1516 			pm_new->found = 1;
   1517 			pm_new->ptstart = 0;
   1518 			pm_new->ptsize = 0;
   1519 			pm_new->no_part = true;
   1520 			pm_new->refdev = &lvms[i].lv[ii];
   1521 			pm_new->sectorsize = 1;
   1522 			pm_new->dlcylsize = MEG;
   1523 			strlcpy(pm_new->diskdev, dev, SSTRSIZE);
   1524 			strlcpy(pm_new->diskdev_descr, dev, STRSIZE);
   1525 
   1526 			if (SLIST_EMPTY(&pm_head))
   1527 				 SLIST_INSERT_HEAD(&pm_head, pm_new, l);
   1528 			else
   1529 				 SLIST_INSERT_AFTER(pm_i, pm_new, l);
   1530 			pm_new = malloc(sizeof (struct pm_devs));
   1531 			memset(pm_new, 0, sizeof *pm_new);
   1532 		}
   1533 	}
   1534 }
   1535 
   1536 static int
   1537 pm_lvm_disk_add(menudesc *m, void *arg)
   1538 {
   1539 	int i;
   1540 	lvms_t *dev_ptr = arg;
   1541 	struct part_entry disk_entrie = pm_dev_list(PM_LVM);
   1542 	if (pm_retvalue < 0)
   1543 		return pm_retvalue;
   1544 
   1545 	for (i = 0; i < MAX_LVM_PV; i++) {
   1546 		if (dev_ptr->pv[i].pm == NULL) {
   1547 			dev_ptr->pv[i].pm = disk_entrie.dev_ptr;
   1548 			dev_ptr->pv[i].pm_part = disk_entrie.id;
   1549 			strlcpy(dev_ptr->pv[i].pm_name, disk_entrie.fullname,
   1550 			    sizeof(dev_ptr->pv[i].pm_name));
   1551 			break;
   1552 		}
   1553 	}
   1554 	pm_retvalue = 1;
   1555 	return 0;
   1556 }
   1557 
   1558 static int
   1559 pm_lvm_disk_del(menudesc *m, void *arg)
   1560 {
   1561 	int retvalue = -1, num_devs = 0;
   1562 	size_t i;
   1563 	int menu_no;
   1564 	lvms_t *dev_ptr = arg;
   1565 	menu_ent menu_entries[MAX_LVM_PV];
   1566 	struct part_entry submenu_args[MAX_LVM_PV];
   1567 
   1568 	for (i = 0; i < MAX_LVM_PV; i++) {
   1569 		if (dev_ptr->pv[i].pm == NULL)
   1570 			continue;
   1571 		menu_entries[num_devs] = (struct menu_ent) {
   1572 			.opt_name = dev_ptr->pv[i].pm_name,
   1573 			.opt_action = set_menu_select,
   1574 			.opt_flags = OPT_EXIT,
   1575 		};
   1576 		submenu_args[num_devs].index = i;
   1577 		num_devs++;
   1578 	}
   1579 
   1580 	menu_no = new_menu(MSG_lvm_disks,
   1581 		menu_entries, num_devs, -1, -1,
   1582 		(num_devs+1<3)?3:num_devs+1, 13,
   1583 		MC_SCROLL | MC_NOCLEAR, NULL, NULL, NULL, NULL, MSG_cancel);
   1584 	if (menu_no == -1)
   1585 		return -1;
   1586 	process_menu(menu_no, &retvalue);
   1587 	free_menu(menu_no);
   1588 
   1589 	if (retvalue < 0 || retvalue >= num_devs)
   1590 		return -1;
   1591 
   1592 	dev_ptr->pv[submenu_args[retvalue].index].pm = NULL;
   1593 
   1594 	return 0;
   1595 }
   1596 
   1597 static void
   1598 pm_lvm_menufmt(menudesc *m, int opt, void *arg)
   1599 {
   1600 	int i, ok = 0;
   1601 	daddr_t used_size = 0;
   1602 	char buf1[STRSIZE]; buf1[0] = 0;
   1603 	char buf2[STRSIZE]; buf2[0] = 0;
   1604 	char devs[STRSIZE]; devs[0] = 0;
   1605 	lvms_t *dev_ptr = ((struct part_entry *)arg)[opt].dev_ptr;
   1606 
   1607 	if (dev_ptr->enabled == 0)
   1608 		return;
   1609 	snprintf(buf1, STRSIZE, "VG '%s'", dev_ptr->name);
   1610 	for (i = 0; i < MAX_LVM_PV; i++)
   1611 		if (dev_ptr->pv[i].pm != NULL) {
   1612 			strlcat(devs, dev_ptr->pv[i].pm_name, STRSIZE);
   1613 			strlcat(devs, " ", STRSIZE);
   1614 			ok = 1;
   1615 		}
   1616 	for (i = 0; i < MAX_LVM_LV; i++)
   1617 		used_size += dev_ptr->lv[i].size;
   1618 	if (ok) {
   1619 		snprintf(buf2, STRSIZE, "%" PRIi64 "/%" PRIi64,
   1620 			dev_ptr->total_size - used_size, dev_ptr->total_size);
   1621 		pm_fmt_disk_line(m->mw, buf1, devs, 0, buf2);
   1622 	} else {
   1623 		pm_fmt_disk_line(m->mw, buf1, NULL, 0,
   1624 		    msg_string(MSG_lvm_err_menufmt));
   1625 	}
   1626 }
   1627 
   1628 static void
   1629 pm_lvm_edit_menufmt(menudesc *m, int opt, void *arg)
   1630 {
   1631 	int i;
   1632 	char buf[STRSIZE];
   1633 	lvms_t *dev_ptr = arg;
   1634 	strlcpy(buf, msg_string(MSG_auto), sizeof buf);
   1635 
   1636 	switch (opt) {
   1637 		case PML_MENU_PV:
   1638 			buf[0] = '\0';
   1639 			for (i = 0; i < MAX_LVM_PV; i++) {
   1640 				if (dev_ptr->pv[i].pm != NULL) {
   1641 					strlcat(buf, " ", sizeof buf);
   1642 					strlcat(buf, dev_ptr->pv[i].pm_name,
   1643 					    sizeof buf);
   1644 				}
   1645 			}
   1646 			wprintw(m->mw, "%-20s: %s",
   1647 			     msg_string(MSG_lvm_disks_fmt), buf);
   1648 			break;
   1649 		case PML_MENU_NAME:
   1650 			wprintw(m->mw, "%-20s: %s",
   1651 			    msg_string(MSG_lvm_name_fmt), dev_ptr->name);
   1652 			break;
   1653 		case PML_MENU_MAXLOGICALVOLUMES:
   1654 			if (dev_ptr->maxlogicalvolumes > 0)
   1655 				snprintf(buf, STRSIZE, "%d",
   1656 				    dev_ptr->maxlogicalvolumes);
   1657 			wprintw(m->mw, "%-20s: %s",
   1658 			    msg_string(MSG_lvm_maxlv_fmt), buf);
   1659 			break;
   1660 		case PML_MENU_MAXPHYSICALVOLUMES:
   1661 			if (dev_ptr->maxphysicalvolumes > 0)
   1662 				snprintf(buf, STRSIZE, "%d",
   1663 				    dev_ptr->maxphysicalvolumes);
   1664 			wprintw(m->mw, "%-20s: %s",
   1665 			    msg_string(MSG_lvm_maxpv_fmt), buf);
   1666 			break;
   1667 		case PML_MENU_PHYSICALEXTENTSIZE:
   1668 			if (dev_ptr->physicalextentsize > 0)
   1669 				snprintf(buf, STRSIZE, "%dM",
   1670 				    dev_ptr->physicalextentsize);
   1671 			wprintw(m->mw, "%-20s: %s",
   1672 			    msg_string(MSG_lvm_extsiz_fmt), buf);
   1673 			break;
   1674 	}
   1675 }
   1676 
   1677 static int
   1678 pm_lvm_set_value(menudesc *m, void *arg)
   1679 {
   1680 	char buf[STRSIZE];
   1681 	const char *msg_to_show = NULL;
   1682 	int *out_var = NULL;
   1683 	lvms_t *dev_ptr = arg;
   1684 
   1685 	static menu_ent menuent_disk_adddel[] = {
   1686 	    { .opt_name=MSG_add, .opt_flags=OPT_EXIT,
   1687 	      .opt_action=pm_lvm_disk_add },
   1688 	    { .opt_name=MSG_remove, .opt_flags=OPT_EXIT,
   1689 	      .opt_action=pm_lvm_disk_del }
   1690 	};
   1691 	static int menu_disk_adddel = -1;
   1692 	if (menu_disk_adddel == -1) {
   1693 		menu_disk_adddel = new_menu(NULL, menuent_disk_adddel,
   1694 			__arraycount(menuent_disk_adddel),
   1695 			-1, -1, 0, 10, MC_NOCLEAR, NULL, NULL, NULL, NULL,
   1696 			MSG_cancel);
   1697 	}
   1698 
   1699 	switch (m->cursel) {
   1700 		case PML_MENU_PV:
   1701 			process_menu(menu_disk_adddel, arg);
   1702 			return 0;
   1703 		case PML_MENU_NAME:
   1704 			msg_prompt_win(MSG_lvm_name_ask, -1, 18, 0, 0,
   1705 			    dev_ptr->name, dev_ptr->name, SSTRSIZE);
   1706 			return 0;
   1707 		case PML_MENU_MAXLOGICALVOLUMES:
   1708 			msg_to_show = MSG_lvm_maxlv_ask;
   1709 			out_var = &(dev_ptr->maxlogicalvolumes);
   1710 			break;
   1711 		case PML_MENU_MAXPHYSICALVOLUMES:
   1712 			msg_to_show = MSG_lvm_maxpv_ask;
   1713 			out_var = &(dev_ptr->maxphysicalvolumes);
   1714 			break;
   1715 		case PML_MENU_PHYSICALEXTENTSIZE:
   1716 			msg_to_show = MSG_lvm_extsiz_ask;
   1717 			out_var = &(dev_ptr->physicalextentsize);
   1718 			break;
   1719 		case PML_MENU_REMOVE:
   1720 			dev_ptr->enabled = 0;
   1721 			return 0;
   1722 	}
   1723 	if (out_var == NULL || msg_to_show == NULL)
   1724 		return -1;
   1725 	snprintf(buf, SSTRSIZE, "%d", *out_var);
   1726 	msg_prompt_win(msg_to_show, -1, 18, 0, 0, buf, buf, SSTRSIZE);
   1727 	if (atoi(buf) >= 0)
   1728 		*out_var = atoi(buf);
   1729 	return 0;
   1730 }
   1731 
   1732 static void
   1733 pm_lvm_init(void *arg, void* none)
   1734 {
   1735 	lvms_t *dev_ptr = arg;
   1736 
   1737 	memset(dev_ptr, 0, sizeof *dev_ptr);
   1738 	*dev_ptr = (struct lvms_t) {
   1739 		.enabled = 1,
   1740 		.blocked = 0,
   1741 		.maxlogicalvolumes = MAX_LVM_PV,
   1742 		.maxphysicalvolumes = MAX_LVM_LV,
   1743 		.physicalextentsize = -1,
   1744 	};
   1745 	sprintf(dev_ptr->name, "vg%.2d", rand()%100);
   1746 }
   1747 
   1748 static int
   1749 pm_lvm_check(void *arg)
   1750 {
   1751 	size_t i;
   1752 	bool ok = false;
   1753 	lvms_t *dev_ptr = arg;
   1754 	dev_ptr->total_size = 0;
   1755 	struct disk_part_info info;
   1756 
   1757 	for (i = 0; i < MAX_LVM_PV; i++) {
   1758 		if (dev_ptr->pv[i].pm != NULL) {
   1759 			if (!dev_ptr->pv[i].pm->parts->pscheme->get_part_info(
   1760 			    dev_ptr->pv[i].pm->parts, dev_ptr->pv[i].pm_part,
   1761 			    &info))
   1762 				continue;
   1763 			ok = 1;
   1764 			dev_ptr->total_size += info.size;
   1765 		}
   1766 	}
   1767 	if (!ok)
   1768 		dev_ptr->enabled = 0;
   1769 	return dev_ptr->enabled;
   1770 }
   1771 
   1772 static void
   1773 pm_lvmlv_menufmt(menudesc *m, int opt, void *arg)
   1774 {
   1775 	char buf[STRSIZE];
   1776 	lv_t *dev_ptr = ((struct part_entry *)arg)[opt].dev_ptr;
   1777 
   1778 	if (dev_ptr->size > 0) {
   1779 		snprintf(buf, STRSIZE, "'%s'", dev_ptr->name);
   1780 		pm_fmt_disk_line(m->mw, buf, NULL, dev_ptr->size, NULL);
   1781 	}
   1782 }
   1783 
   1784 static void
   1785 pm_lvmlv_edit_menufmt(menudesc *m, int opt, void *arg)
   1786 {
   1787 
   1788 	lv_t *dev_ptr = arg;
   1789 	char buf[STRSIZE];
   1790 	strlcpy(buf, msg_string(MSG_auto), STRSIZE);
   1791 
   1792 	switch (opt) {
   1793 		case PMLV_MENU_NAME:
   1794 			wprintw(m->mw, "%-20s: %s",
   1795 			    msg_string(MSG_lvmlv_name_fmt), dev_ptr->name);
   1796 			break;
   1797 		case PMLV_MENU_SIZE:
   1798 			wprintw(m->mw, "%-20s: %" PRIi64,
   1799 			    msg_string(MSG_lvmlv_size_fmt), dev_ptr->size);
   1800 			break;
   1801 		case PMLV_MENU_READONLY:
   1802 			wprintw(m->mw, "%-20s: %s",
   1803 			    msg_string(MSG_lvmlv_ro_fmt),
   1804 			    dev_ptr->readonly? msg_string(MSG_Yes) : msg_string(MSG_No));
   1805 			break;
   1806 		case PMLV_MENU_CONTIGUOUS:
   1807 			wprintw(m->mw, "%-20s: %s",
   1808 			    msg_string(MSG_lvmlv_cont_fmt),
   1809 			    dev_ptr->contiguous? msg_string(MSG_Yes) : msg_string(MSG_No));
   1810 			break;
   1811 		case PMLV_MENU_EXTENTS:
   1812 			wprintw(m->mw, "%-20s: %s",
   1813 			    msg_string(MSG_lvmlv_extnum_fmt),
   1814 			    (strlen(dev_ptr->extents) > 0) ?
   1815 			    dev_ptr->extents : msg_string(MSG_auto));
   1816 			break;
   1817 		case PMLV_MENU_MINOR:
   1818 			if (dev_ptr->minor > 0)
   1819 				snprintf(buf, STRSIZE, "%dK", dev_ptr->minor);
   1820 			wprintw(m->mw, "%-20s: %s",
   1821 			    msg_string(MSG_lvmlv_minor_fmt), buf);
   1822 			break;
   1823 		case PMLV_MENU_MIRRORS:
   1824 			wprintw(m->mw, "%-20s: %d",
   1825 			    msg_string(MSG_lvmlv_mirrors_fmt),
   1826 			    dev_ptr->mirrors);
   1827 			break;
   1828 		case PMLV_MENU_REGIONSIZE:
   1829 			if (dev_ptr->regionsize > 0)
   1830 				snprintf(buf, STRSIZE, "%dM",
   1831 				    dev_ptr->regionsize);
   1832 			wprintw(m->mw, "%-20s: %s",
   1833 			    msg_string(MSG_lvmlv_regsiz_fmt), buf);
   1834 			break;
   1835 		case PMLV_MENU_PERSISTENT:
   1836 			wprintw(m->mw, "%-20s: %s",
   1837 			    msg_string(MSG_lvmlv_pers_fmt),
   1838 			    dev_ptr->persistent ?
   1839 			    msg_string(MSG_Yes) : msg_string(MSG_No));
   1840 			break;
   1841 		case PMLV_MENU_READAHEAD:
   1842 			if (dev_ptr->readahead > 0)
   1843 				snprintf(buf, STRSIZE, "%d",
   1844 				    dev_ptr->readahead);
   1845 			wprintw(m->mw, "%-20s: %s",
   1846 			    msg_string(MSG_lvmlv_readahsect_fmt), buf);
   1847 			break;
   1848 		case PMLV_MENU_STRIPES:
   1849 			if (dev_ptr->stripes > 0)
   1850 				snprintf(buf, STRSIZE, "%d", dev_ptr->stripes);
   1851 			wprintw(m->mw, "%-20s: %s",
   1852 			    msg_string(MSG_lvmlv_stripes_fmt), buf);
   1853 			break;
   1854 		case PMLV_MENU_STRIPESIZE:
   1855 			if (dev_ptr->stripesize > 0)
   1856 				snprintf(buf, STRSIZE, "%dK", dev_ptr->stripesize);
   1857 			wprintw(m->mw, "%-20s: %s",
   1858 			    msg_string(MSG_lvmlv_stripesiz_fmt), buf);
   1859 			break;
   1860 		case PMLV_MENU_ZERO:
   1861 			wprintw(m->mw, "%-20s: %s",
   1862 			    msg_string(MSG_lvmlv_zero_fmt),
   1863 			    dev_ptr->zero ?
   1864 			    msg_string(MSG_Yes) : msg_string(MSG_No));
   1865 			break;
   1866 	}
   1867 }
   1868 
   1869 static int
   1870 pm_lvmlv_set_value(menudesc *m, void *arg)
   1871 {
   1872 	char buf[STRSIZE];
   1873 	const char *msg_to_show = NULL;
   1874 	int *out_var = NULL;
   1875 	lv_t *dev_ptr = arg;
   1876 
   1877 	switch (m->cursel) {
   1878 		case PMLV_MENU_NAME:
   1879 			msg_prompt_win(MSG_lvmlv_name_ask, -1, 18, 0, 0,
   1880 			    dev_ptr->name, dev_ptr->name, SSTRSIZE);
   1881 			return 0;
   1882 		case PMLV_MENU_SIZE:
   1883 			pm_edit_size_value(MSG_lvmlv_size_ask,
   1884 			    pm->dlcylsize, &dev_ptr->size); /* XXX cylsize? */
   1885 			break;
   1886 		case PMLV_MENU_READONLY:
   1887 			dev_ptr->readonly = !dev_ptr->readonly;
   1888 			return 0;
   1889 		case PMLV_MENU_CONTIGUOUS:
   1890 			dev_ptr->contiguous = !dev_ptr->contiguous;
   1891 			return 0;
   1892 		case PMLV_MENU_EXTENTS:
   1893 			msg_prompt_win(MSG_lvmlv_extnum_ask, -1, 18, 0, 0,
   1894 			    dev_ptr->extents, dev_ptr->extents, SSTRSIZE);
   1895 			return 0;
   1896 		case PMLV_MENU_MINOR:
   1897 			msg_to_show = MSG_lvmlv_minor_ask;
   1898 			out_var = &(dev_ptr->minor);
   1899 			break;
   1900 		case PMLV_MENU_MIRRORS:
   1901 			msg_to_show = MSG_lvmlv_mirrors_ask;
   1902 			out_var = &(dev_ptr->mirrors);
   1903 			break;
   1904 		case PMLV_MENU_REGIONSIZE:
   1905 			msg_to_show = MSG_lvmlv_regsiz_ask;
   1906 			out_var = &(dev_ptr->regionsize);
   1907 			break;
   1908 		case PMLV_MENU_PERSISTENT:
   1909 			dev_ptr->persistent = !dev_ptr->persistent;
   1910 			return 0;
   1911 		case PMLV_MENU_READAHEAD:
   1912 			msg_to_show = MSG_lvmlv_readahsect_ask;
   1913 			out_var = &(dev_ptr->readahead);
   1914 			break;
   1915 		case PMLV_MENU_STRIPES:
   1916 			msg_to_show = MSG_lvmlv_stripes_ask;
   1917 			out_var = &(dev_ptr->stripes);
   1918 			break;
   1919 		case PMLV_MENU_STRIPESIZE:
   1920 			if (dev_ptr->stripesize << 1 > 512)
   1921 				dev_ptr->stripesize = 4;
   1922 			else
   1923 				dev_ptr->stripesize <<= 1;
   1924 			return 0;
   1925 		case PMLV_MENU_ZERO:
   1926 			dev_ptr->zero = !dev_ptr->zero;
   1927 			return 0;
   1928 		case PMLV_MENU_REMOVE:
   1929 			dev_ptr->size = 0;
   1930 			return 0;
   1931 	}
   1932 	if (out_var == NULL || msg_to_show == NULL)
   1933 		return -1;
   1934 	snprintf(buf, SSTRSIZE, "%d", *out_var);
   1935 	msg_prompt_win(msg_to_show, -1, 18, 0, 0, buf, buf, SSTRSIZE);
   1936 	if (atoi(buf) >= 0)
   1937 		*out_var = atoi(buf);
   1938 	return 0;
   1939 }
   1940 
   1941 static void
   1942 pm_lvmlv_init(void *arg, void *none)
   1943 {
   1944 	lv_t *dev_ptr = arg;
   1945 	memset(dev_ptr, 0, sizeof *(dev_ptr));
   1946 	*dev_ptr = (struct lv_t) {
   1947 		.blocked = 0,
   1948 		.size = 1024,
   1949 		.stripesize = 64,
   1950 	};
   1951 	sprintf (dev_ptr->name, "lvol%.2d", rand()%100);
   1952 }
   1953 
   1954 static int
   1955 pm_lvmlv_check(void *arg)
   1956 {
   1957 	lv_t *dev_ptr = arg;
   1958 	if (dev_ptr->size > 0 && strlen(dev_ptr->name) > 0)
   1959 		return 1;
   1960 	else {
   1961 		dev_ptr->size = 0;
   1962 		return 0;
   1963 	}
   1964 }
   1965 
   1966 static int
   1967 pm_lvm_commit(void)
   1968 {
   1969 	int i, ii, error;
   1970 	uint used_size = 0;
   1971 	char params[STRSIZE*3], devs[STRSIZE*3], arg[STRSIZE];
   1972 
   1973 	for (i = 0; i < MAX_LVM_VG; i++) {
   1974 		/* Stage 0: checks */
   1975 		if (! pm_lvm_check(&lvms[i]))
   1976 			continue;
   1977 		for (ii = 0; ii < MAX_LVM_LV; ii++)
   1978 			used_size += lvms[i].lv[ii].size;
   1979 		if (used_size > lvms[i].total_size)
   1980 			continue;
   1981 
   1982 		params[0] = '\0';
   1983 		devs[0] = '\0';
   1984 		error = 0;
   1985 		/* Stage 1: creating Physical Volumes (PV's) */
   1986 		for (ii = 0; ii < MAX_LVM_PV && ! error; ii++)
   1987 			if (lvms[i].pv[ii].pm != NULL) {
   1988 				run_program(RUN_SILENT | RUN_ERROR_OK,
   1989 				    "lvm pvremove -ffy /dev/r%s",
   1990 				    (char*)lvms[i].pv[ii].pm_name);
   1991 				error += run_program(RUN_DISPLAY | RUN_PROGRESS,
   1992 				    "lvm pvcreate -ffy /dev/r%s",
   1993 				    (char*)lvms[i].pv[ii].pm_name);
   1994 				if (error)
   1995 					break;
   1996 				strlcat(devs, " /dev/r", sizeof devs);
   1997 				strlcat(devs, lvms[i].pv[ii].pm_name,
   1998 				    sizeof devs);
   1999 			}
   2000 		if (error)
   2001 			continue;
   2002 		/* Stage 2: creating Volume Groups (VG's) */
   2003 		if (lvms[i].maxlogicalvolumes > 0) {
   2004 			snprintf(arg, sizeof arg, " -l %d",
   2005 			    lvms[i].maxlogicalvolumes);
   2006 			strlcat(params, arg, sizeof params);
   2007 		}
   2008 		if (lvms[i].maxphysicalvolumes > 0) {
   2009 			snprintf(arg, sizeof arg, " -p %d",
   2010 			    lvms[i].maxphysicalvolumes);
   2011 			strlcat(params, arg, sizeof params);
   2012 		}
   2013 		if (lvms[i].physicalextentsize > 0) {
   2014 			snprintf(arg, sizeof arg, " -s %d",
   2015 			    lvms[i].physicalextentsize);
   2016 			strlcat(params, arg, sizeof params);
   2017 		}
   2018 		error += run_program(RUN_DISPLAY | RUN_PROGRESS,
   2019 		    "lvm vgcreate %s %s %s", params, lvms[i].name, devs);
   2020 		if (error)
   2021 			continue;
   2022 		/* Stage 3: creating Logical Volumes (LV's) */
   2023 		for (ii = 0; ii < MAX_LVM_LV; ii++) {
   2024 			if (lvms[i].lv[ii].size <= 0)
   2025 				continue;
   2026 
   2027 			params[0] = '\0';
   2028 			snprintf(arg, sizeof arg, " -C %c",
   2029 			    lvms[i].lv[ii].contiguous?'y':'n');
   2030 			strlcat(params, arg, sizeof params);
   2031 			snprintf(arg, sizeof arg, " -M %c",
   2032 			    lvms[i].lv[ii].persistent?'y':'n');
   2033 			strlcat(params, arg, sizeof params);
   2034 			snprintf(arg, sizeof arg, " -p %s",
   2035 			    lvms[i].lv[ii].readonly?"r":"rw");
   2036 			strlcat(params, arg, sizeof params);
   2037 			snprintf(arg, sizeof arg, " -Z %c",
   2038 			    lvms[i].lv[ii].zero?'y':'n');
   2039 			strlcat(params, arg, sizeof params);
   2040 			if (strlen(lvms[i].lv[ii].name) > 0) {
   2041 				snprintf(arg, sizeof arg, " -n %s",
   2042 				    lvms[i].lv[ii].name);
   2043 				strlcat(params, arg, sizeof params);
   2044 			}
   2045 			if (strlen(lvms[i].lv[ii].extents) > 0) {
   2046 				snprintf(arg, sizeof arg, " -l %s",
   2047 				    lvms[i].lv[ii].extents);
   2048 				strlcat(params, arg, sizeof params);
   2049 			}
   2050 			if (lvms[i].lv[ii].minor > 0) {
   2051 				snprintf(arg, sizeof arg, " --minor %d",
   2052 				    lvms[i].lv[ii].minor);
   2053 				strlcat(params, arg, sizeof params);
   2054 			}
   2055 			if (lvms[i].lv[ii].mirrors > 0) {
   2056 				snprintf(arg, sizeof arg, " -m %d",
   2057 				    lvms[i].lv[ii].mirrors);
   2058 				strlcat(params, arg, sizeof params);
   2059 				if (lvms[i].lv[ii].regionsize > 0) {
   2060 					snprintf(arg, sizeof arg, " -R %d",
   2061 					     lvms[i].lv[ii].regionsize);
   2062 					strlcat(params, arg, sizeof params);
   2063 				}
   2064 			}
   2065 			if (lvms[i].lv[ii].readahead > 0) {
   2066 				snprintf(arg, sizeof arg, " -r %d",
   2067 				    lvms[i].lv[ii].readahead);
   2068 				strlcat(params, arg, sizeof params);
   2069 			}
   2070 			if (lvms[i].lv[ii].stripes > 0) {
   2071 				snprintf(arg, sizeof arg, " -i %d",
   2072 				    lvms[i].lv[ii].stripes);
   2073 				strlcat(params, arg, sizeof params);
   2074 				if (lvms[i].lv[ii].stripesize > 0) {
   2075 					snprintf(arg, sizeof arg, " -I %d",
   2076 					    lvms[i].lv[ii].stripesize);
   2077 					strlcat(params, arg, sizeof params);
   2078 				}
   2079 			}
   2080 			snprintf(arg, sizeof arg, " -L %" PRIi64 "M",
   2081 			    lvms[i].lv[ii].size);
   2082 			strlcat(params, arg, sizeof params);
   2083 
   2084 			error += run_program(RUN_DISPLAY | RUN_PROGRESS,
   2085 			    "lvm lvcreate %s %s", params, lvms[i].name);
   2086 		}
   2087 		if (! error) {
   2088 			lvms[i].blocked = 1;
   2089 			for (ii = 0; ii < MAX_LVM_PV; ii++)
   2090 				if (lvms[i].pv[ii].pm != NULL)
   2091 					lvms[i].pv[ii].pm->blocked++;
   2092 			for (ii = 0; ii < MAX_LVM_LV; ii++)
   2093 				if (lvms[i].lv[ii].size > 0)
   2094 					lvms[i].lv[ii].blocked = 1;
   2095 		}
   2096 	}
   2097 
   2098 	return 0;
   2099 }
   2100 
   2101 /***
   2102  Partman generic functions
   2103  ***/
   2104 
   2105 int
   2106 pm_getrefdev(struct pm_devs *pm_cur)
   2107 {
   2108 	int i, ii, dev_num, num_devs, num_devs_s;
   2109 	char descr[MENUSTRSIZE], dev[MENUSTRSIZE] = "";
   2110 
   2111 	pm_cur->refdev = NULL;
   2112 	if (! strncmp(pm_cur->diskdev, "cgd", 3)) {
   2113 		dev_num = pm_cur->diskdev[3] - '0';
   2114 		for (i = 0; i < MAX_CGD; i++)
   2115 			if (cgds[i].blocked && cgds[i].node == dev_num) {
   2116 				pm_cur->refdev = &cgds[i];
   2117 				snprintf(descr, sizeof descr,
   2118 				    " (%s, %s-%d)", cgds[i].pm_name,
   2119 				    cgds[i].enc_type, cgds[i].key_size);
   2120 				strlcat(pm_cur->diskdev_descr, descr,
   2121 				    sizeof(pm_cur->diskdev_descr));
   2122 				break;
   2123 			}
   2124  	} else if (! strncmp(pm_cur->diskdev, "vnd", 3)) {
   2125  		dev_num = pm_cur->diskdev[3] - '0';
   2126  		for (i = 0; i < MAX_VND; i++)
   2127 			if (vnds[i].blocked && vnds[i].node == dev_num) {
   2128 				pm_cur->refdev = &vnds[i];
   2129 				vnds[i].pm->parts->pscheme->get_part_device(
   2130 				    vnds[i].pm->parts, vnds[i].pm_part,
   2131 				    dev, sizeof dev, NULL, plain_name, false,
   2132 				    true);
   2133 				snprintf(descr, sizeof descr, " (%s, %s)",
   2134 				    dev, vnds[i].filepath);
   2135 				strlcat(pm_cur->diskdev_descr, descr,
   2136 				    sizeof(pm_cur->diskdev_descr));
   2137 				break;
   2138 			}
   2139 	} else if (! strncmp(pm_cur->diskdev, "raid", 4)) {
   2140 		dev_num = pm_cur->diskdev[4] - '0';
   2141  		for (i = 0; i < MAX_RAID; i++)
   2142 			if (raids[i].blocked && raids[i].node == dev_num) {
   2143 				pm_cur->refdev = &raids[i];
   2144 				num_devs = 0; num_devs_s = 0;
   2145 				for (ii = 0; ii < MAX_IN_RAID; ii++)
   2146 					if (raids[i].comp[ii].parts != NULL) {
   2147 						if(raids[i].comp[ii].is_spare)
   2148 							num_devs_s++;
   2149 						else
   2150 							num_devs++;
   2151 					}
   2152 				snprintf(descr, sizeof descr,
   2153 				    " (lvl %d, %d disks, %d spare)",
   2154 				    raids[i].raid_level, num_devs,
   2155 				    num_devs_s);
   2156 				strlcat(pm_cur->diskdev_descr, descr,
   2157 				    sizeof(pm_cur->diskdev_descr));
   2158 				break;
   2159 			}
   2160 	} else
   2161 		return -1;
   2162 	return 0;
   2163 }
   2164 
   2165 /*
   2166  * Enable/disable items in the extended partition disk/partition action
   2167  * menu
   2168  */
   2169 void
   2170 pmdiskentry_enable(menudesc *menu, struct part_entry *pe)
   2171 {
   2172 	int i;
   2173 	menu_ent *m;
   2174 	bool enable;
   2175 
   2176 	for (i = 0; i < menu->numopts; i++) {
   2177 		m = &menu->opts[i];
   2178 
   2179 		enable = false;
   2180 		if (m->opt_name == MSG_unconfig) {
   2181 			if (pe->type == PM_DISK)
   2182 				enable = ((struct pm_devs *)pe->dev_ptr)
   2183 				    ->refdev != NULL;
   2184 		} else if (m->opt_name == MSG_undo) {
   2185 			if (pe->type != PM_DISK)
   2186 				continue;
   2187 			enable = ((struct pm_devs *)pe->dev_ptr)->unsaved;
   2188 		} else if (m->opt_name == MSG_switch_parts) {
   2189 			enable = pm_from_pe(pe)->parts != NULL;
   2190 		} else {
   2191 			continue;
   2192 		}
   2193 
   2194 		if (enable)
   2195 			m->opt_flags &= ~OPT_IGNORE;
   2196 		else
   2197 			m->opt_flags |= OPT_IGNORE;
   2198 	}
   2199 }
   2200 
   2201 /* Detect that partition is in use */
   2202 int
   2203 pm_partusage(struct pm_devs *pm_cur, int part_num, int do_del)
   2204 {
   2205 	int i, ii, retvalue = 0;
   2206 	struct disk_part_info info;
   2207 	part_id id;
   2208 
   2209 	if (pm_cur->parts == NULL)
   2210 		return -1;	/* nothing can be in use */
   2211 
   2212 	if (part_num < 0) {
   2213 		/* Check all partitions on device */
   2214 		for (id = 0; id < pm_cur->parts->num_part; id++) {
   2215 			if (!pm_cur->parts->pscheme->get_part_info(
   2216 			    pm_cur->parts, id, &info))
   2217 				continue;
   2218 			if (info.flags & (PTI_SEC_CONTAINER|
   2219 			    PTI_WHOLE_DISK|PTI_PSCHEME_INTERNAL|
   2220 			    PTI_RAW_PART))
   2221 				continue;
   2222 			retvalue += pm_partusage(pm_cur, id, do_del);
   2223 		}
   2224 		return retvalue;
   2225 	}
   2226 
   2227 	id = (part_id)part_num;
   2228 	if (id >= pm_cur->parts->num_part)
   2229 		return 0;
   2230 
   2231 	for (i = 0; i < MAX_CGD; i++)
   2232 		if (cgds[i].enabled &&
   2233 			cgds[i].pm == pm_cur &&
   2234 			cgds[i].pm_part == id) {
   2235 			if (do_del) {
   2236 				cgds[i].pm = NULL;
   2237 				cgds[i].pm_name[0] = '\0';
   2238 			}
   2239 			return 1;
   2240 		}
   2241 	for (i = 0; i < MAX_RAID; i++)
   2242 		for (ii = 0; ii < MAX_IN_RAID; ii++)
   2243 			if (raids[i].enabled &&
   2244 				raids[i].comp[ii].parts == pm_cur->parts &&
   2245 				raids[i].comp[ii].id == id) {
   2246 					if (do_del)
   2247 						raids[i].comp[ii].parts = NULL;
   2248 					return 1;
   2249 			}
   2250 	for (i = 0; i < MAX_LVM_VG; i++)
   2251 		for (ii = 0; ii < MAX_LVM_PV; ii++)
   2252 			if (lvms[i].enabled &&
   2253 				lvms[i].pv[ii].pm == pm_cur &&
   2254 				lvms[i].pv[ii].pm_part == id) {
   2255 					if (do_del)
   2256 						lvms[i].pv[ii].pm = NULL;
   2257 					return 1;
   2258 			}
   2259 
   2260 	return 0;
   2261 }
   2262 
   2263 static void
   2264 pm_destroy_one(struct pm_devs *pm_i)
   2265 {
   2266 	part_id i;
   2267 
   2268 	if (pm_i->parts != NULL) {
   2269 		if (pm_i->mounted != NULL) {
   2270 			for (i = 0; i < pm_i->parts->num_part; i++)
   2271 				free(pm_i->mounted[i]);
   2272 		}
   2273 
   2274 		pm_i->parts->pscheme->free(pm_i->parts);
   2275 	}
   2276 	free(pm_i);
   2277 }
   2278 
   2279 /* Clean up removed devices */
   2280 static int
   2281 pm_clean(void)
   2282 {
   2283 	int count = 0;
   2284 	struct pm_devs *pm_i, *tmp;
   2285 
   2286 	SLIST_FOREACH_SAFE(pm_i, &pm_head, l, tmp)
   2287 		if (! pm_i->found) {
   2288 			count++;
   2289 			SLIST_REMOVE(&pm_head, pm_i, pm_devs, l);
   2290 			pm_destroy_one(pm_i);
   2291 		}
   2292 	return count;
   2293 }
   2294 
   2295 /* Free all pm storage */
   2296 void
   2297 pm_destroy_all(void)
   2298 {
   2299 	struct pm_devs *pm_i, *tmp;
   2300 
   2301 	if (pm_new != pm)
   2302 		pm_destroy_one(pm_new);
   2303 	SLIST_FOREACH_SAFE(pm_i, &pm_head, l, tmp) {
   2304 		SLIST_REMOVE(&pm_head, pm_i, pm_devs, l);
   2305 		pm_destroy_one(pm_i);
   2306 	}
   2307 }
   2308 
   2309 void
   2310 pm_set_lvmpv(struct pm_devs *my_pm, part_id pno, bool add)
   2311 {
   2312 	size_t i;
   2313 	struct disk_part_info info;
   2314 
   2315 	if (!my_pm->parts->pscheme->get_part_info(my_pm->parts, pno, &info))
   2316 		return;
   2317 
   2318 	if (add) {
   2319 		for (i = 0; i < __arraycount(lvm_pvs); i++)
   2320 			if (lvm_pvs[i].pm == NULL && lvm_pvs[i].start == 0)
   2321 				break;
   2322 		if (i >= __arraycount(lvm_pvs))
   2323 			return;
   2324 		lvm_pvs[i].pm = my_pm;
   2325 		lvm_pvs[i].start = info.start;
   2326 		return;
   2327 	} else {
   2328 		for (i = 0; i < __arraycount(lvm_pvs); i++)
   2329 			if (lvm_pvs[i].pm == my_pm &&
   2330 			    lvm_pvs[i].start == info.start)
   2331 				break;
   2332 		if (i >= __arraycount(lvm_pvs))
   2333 			return;
   2334 		lvm_pvs[i].pm = NULL;
   2335 		lvm_pvs[i].start = 0;
   2336 	}
   2337 }
   2338 
   2339 bool
   2340 pm_is_lvmpv(struct pm_devs *my_pm, part_id id,
   2341     const struct disk_part_info *info)
   2342 {
   2343 	size_t i;
   2344 
   2345 	for (i = 0; i < __arraycount(lvm_pvs); i++) {
   2346 		if (lvm_pvs[i].pm != my_pm)
   2347 			continue;
   2348 		if (lvm_pvs[i].start == info->start)
   2349 			return true;
   2350 	}
   2351 
   2352 	return false;
   2353 }
   2354 
   2355 void
   2356 pm_setfstype(struct pm_devs *pm_cur, part_id id, int fstype, int fs_subtype)
   2357 {
   2358 	struct disk_part_info info;
   2359 
   2360 	if (!pm_cur->parts->pscheme->get_part_info(pm_cur->parts, id, &info))
   2361 		return;
   2362 
   2363 	info.nat_type = pm_cur->parts->pscheme->get_fs_part_type(PT_root,
   2364 	    fstype, fs_subtype);
   2365 	if (info.nat_type == NULL)
   2366 		return;
   2367 	info.fs_type = fstype;
   2368 	info.fs_sub_type = fs_subtype;
   2369 	pm_cur->parts->pscheme->set_part_info(pm_cur->parts, id, &info, NULL);
   2370 }
   2371 
   2372 static void
   2373 pm_select(struct pm_devs *pm_devs_in)
   2374 {
   2375 	pm = pm_devs_in;
   2376 	if (logfp)
   2377 		(void)fprintf(logfp, "Partman device: %s\n", pm->diskdev);
   2378 }
   2379 
   2380 void
   2381 pm_rename(struct pm_devs *pm_cur)
   2382 {
   2383 #if 0	// XXX - convert to custom attribute or handle similar
   2384 	pm_select(pm_cur);
   2385 	msg_prompt_win(MSG_packname, -1, 18, 0, 0, pm_cur->bsddiskname,
   2386 		pm_cur->bsddiskname, sizeof pm_cur->bsddiskname);
   2387 #ifndef NO_DISKLABEL
   2388 	(void) savenewlabel(pm_cur->bsdlabel, MAXPARTITIONS);
   2389 #endif
   2390 #endif
   2391 }
   2392 
   2393 int
   2394 pm_editpart(int part_num)
   2395 {
   2396 	struct partition_usage_set pset = {};
   2397 
   2398 	usage_set_from_parts(&pset, pm->parts);
   2399 	edit_ptn(&(struct menudesc){.cursel = part_num}, &pset);
   2400 	free_usage_set(&pset);
   2401 	if (checkoverlap(pm->parts)) {
   2402 		hit_enter_to_continue(MSG_cantsave, NULL);
   2403 		return -1;
   2404 	}
   2405 	pm->unsaved = 1;
   2406 	return 0;
   2407 }
   2408 
   2409 /* Safe erase of disk */
   2410 void
   2411 pm_shred(struct part_entry *pe, int shredtype)
   2412 {
   2413 	const char *srcdev;
   2414 	char dev[SSTRSIZE];
   2415 	struct pm_devs *my_pm;
   2416 
   2417 	my_pm = pe->dev_ptr;
   2418 	if (pe->type == PM_DISK) {
   2419 		snprintf(dev, sizeof dev,
   2420 		    _PATH_DEV "r%s%c", my_pm->diskdev, 'a' + RAW_PART);
   2421 		if (pe->parts != NULL) {
   2422 			pe->parts->pscheme->free(pe->parts);
   2423 			pe->parts = NULL;
   2424 			my_pm->parts = NULL;
   2425 		}
   2426 	} else if (pe->type == PM_PART) {
   2427 		pe->parts->pscheme->get_part_device(pe->parts, pe->id,
   2428 		    dev, sizeof dev, NULL, raw_dev_name, true, true);
   2429 	}
   2430 
   2431 	switch (shredtype) {
   2432 		case SHRED_ZEROS:
   2433 			srcdev = _PATH_DEVZERO;
   2434 			break;
   2435 		case SHRED_RANDOM:
   2436 			srcdev = _PATH_URANDOM;
   2437 			break;
   2438 		default:
   2439 			return;
   2440 	}
   2441 	run_program(RUN_DISPLAY | RUN_PROGRESS,
   2442 	    "progress -f %s -b 1m dd bs=1m of=%s", srcdev, dev);
   2443 	pm_partusage(my_pm, -1, 1);
   2444 }
   2445 
   2446 #if 0 // XXX
   2447 static int
   2448 pm_mountall_sort(const void *a, const void *b)
   2449 {
   2450 	return strcmp(mnts[*(const int *)a].on, mnts[*(const int *)b].on);
   2451 }
   2452 #endif
   2453 
   2454 #if 0	// XXX
   2455 /* Mount all available partitions */
   2456 static int
   2457 pm_mountall(void)
   2458 {
   2459 	int num_devs = 0;
   2460 	int mnts_order[MAX_MNTS];
   2461 	int i, ii, error, ok;
   2462 	char dev[SSTRSIZE]; dev[0] = '\0';
   2463 	struct pm_devs *pm_i;
   2464 
   2465 	localfs_dev[0] = '\0';
   2466 	if (mnts == NULL)
   2467 		mnts = calloc(MAX_MNTS, sizeof(*mnts));
   2468 
   2469 	SLIST_FOREACH(pm_i, &pm_head, l) {
   2470 		ok = 0;
   2471 		for (i = 0; i < MAXPARTITIONS; i++) {
   2472 			if (!(pm_i->bsdlabel[i].pi_flags & PIF_MOUNT &&
   2473 					pm_i->bsdlabel[i].mnt_opts != NULL))
   2474 				continue;
   2475 			mnts[num_devs].mnt_opts = pm_i->bsdlabel[i].mnt_opts;
   2476 			if (strlen(pm_i->bsdlabel[i].mounted) > 0) {
   2477 				/* Device is already mounted. So, doing mount_null */
   2478 				strlcpy(mnts[num_devs].dev, pm_i->bsdlabel[i].mounted, MOUNTLEN);
   2479 				mnts[num_devs].mnt_opts = "-t null";
   2480 			} else {
   2481 				pm_getdevstring(dev, SSTRSIZE, pm_i, i);
   2482 				snprintf(mnts[num_devs].dev, STRSIZE, "/dev/%s", dev);
   2483 			}
   2484 			mnts[num_devs].on = pm_i->bsdlabel[i].pi_mount;
   2485 			if (strcmp(pm_i->bsdlabel[i].pi_mount, "/") == 0) {
   2486 				/* Use disk with / as a default if the user has
   2487 				the sets on a local disk */
   2488 				strlcpy(localfs_dev, pm_i->diskdev, SSTRSIZE);
   2489 			}
   2490 			num_devs++;
   2491 			ok = 1;
   2492 		}
   2493 		if (ok)
   2494 			md_pre_mount(NULL, 0);
   2495 	}
   2496 	if (strlen(localfs_dev) < 1) {
   2497 		hit_enter_to_continue(MSG_noroot, NULL);
   2498 		return -1;
   2499 	}
   2500 	for (i = 0; i < num_devs; i++)
   2501 		mnts_order[i] = i;
   2502 	qsort(mnts_order, num_devs, sizeof mnts_order[0], pm_mountall_sort);
   2503 
   2504 	for (i = 0; i < num_devs; i++) {
   2505 		ii = mnts_order[i];
   2506 		make_target_dir(mnts[ii].on);
   2507 		error = target_mount_do(mnts[ii].mnt_opts, mnts[ii].dev, mnts[ii].on);
   2508 		if (error)
   2509 			return error;
   2510 	}
   2511 	return 0;
   2512 }
   2513 #endif
   2514 
   2515 /* Mount partition bypassing ordinary order */
   2516 static int
   2517 pm_mount(struct pm_devs *pm_cur, int part_num)
   2518 {
   2519 	int error = 0;
   2520 #if 0 // XXX
   2521 	char buf[MOUNTLEN];
   2522 
   2523 	if (strlen(pm_cur->bsdlabel[part_num].mounted) > 0)
   2524 		return 0;
   2525 
   2526 	snprintf(buf, sizeof(buf), "/tmp/%s%c", pm_cur->diskdev,
   2527 	    part_num + 'a');
   2528 	if (! dir_exists_p(buf))
   2529 		run_program(RUN_DISPLAY | RUN_PROGRESS, "/bin/mkdir -p %s", buf);
   2530 	if (pm_cur->bsdlabel[part_num].pi_flags & PIF_MOUNT &&
   2531 		pm_cur->bsdlabel[part_num].mnt_opts != NULL &&
   2532 		strlen(pm_cur->bsdlabel[part_num].mounted) < 1)
   2533 		error += run_program(RUN_DISPLAY | RUN_PROGRESS, "/sbin/mount %s /dev/%s%c %s",
   2534 				pm_cur->bsdlabel[part_num].mnt_opts,
   2535 				pm_cur->diskdev, part_num + 'a', buf);
   2536 
   2537 	if (error)
   2538 		pm_cur->bsdlabel[part_num].mounted[0] = '\0';
   2539 	else {
   2540 		strlcpy(pm_cur->bsdlabel[part_num].mounted, buf, MOUNTLEN);
   2541 		pm_cur->blocked++;
   2542 	}
   2543 #endif
   2544 	return error;
   2545 }
   2546 
   2547 void
   2548 pm_umount(struct pm_devs *pm_cur, int part_num)
   2549 {
   2550 	char buf[SSTRSIZE]; buf[0] = '\0';
   2551 	part_id id;
   2552 
   2553 	if (part_num < 0)
   2554 		return;
   2555 	id = (part_id)part_num;
   2556 
   2557 	pm_cur->parts->pscheme->get_part_device(pm_cur->parts, id, buf,
   2558 	    sizeof buf, NULL, plain_name, false, true);
   2559 
   2560 	if (run_program(RUN_DISPLAY | RUN_PROGRESS,
   2561 			"umount -f /dev/%s", buf) == 0) {
   2562 		free(pm_cur->mounted[id]);
   2563 		pm_cur->mounted[id] = NULL;
   2564 		if (pm_cur->blocked > 0)
   2565 			pm_cur->blocked--;
   2566 	}
   2567 }
   2568 
   2569 int
   2570 pm_unconfigure(struct pm_devs *pm_cur)
   2571 {
   2572 	int error = 0;
   2573 	if (! strncmp(pm_cur->diskdev, "cgd", 3)) {
   2574  		error = run_program(RUN_DISPLAY | RUN_PROGRESS, "cgdconfig -u %s", pm_cur->diskdev);
   2575  		if (! error && pm_cur->refdev != NULL) {
   2576 			((struct cgd_desc*)pm_cur->refdev)->pm->blocked--;
   2577 			((struct cgd_desc*)pm_cur->refdev)->blocked = 0;
   2578  		}
   2579  	} else if (! strncmp(pm_cur->diskdev, "vnd", 3)) {
   2580 		error = run_program(RUN_DISPLAY | RUN_PROGRESS, "vndconfig -u %s", pm_cur->diskdev);
   2581  		if (! error && pm_cur->refdev != NULL) {
   2582 			((struct vnd_desc*)pm_cur->refdev)->pm->blocked--;
   2583 			((struct vnd_desc*)pm_cur->refdev)->blocked = 0;
   2584  		}
   2585 	} else if (! strncmp(pm_cur->diskdev, "raid", 4)) {
   2586 		error = run_program(RUN_DISPLAY | RUN_PROGRESS, "raidctl -u %s", pm_cur->diskdev);
   2587 		if (! error && pm_cur->refdev != NULL) {
   2588 			((struct raid_desc*)pm_cur->refdev)->blocked = 0;
   2589 #if 0 // XXX
   2590 			for (i = 0; i < MAX_IN_RAID; i++)
   2591 				if (((struct raid_desc*)pm_cur->refdev)->comp[i].parts != NULL)
   2592 					((raids_t*)pm_cur->refdev)->pm[i]->blocked--;
   2593 #endif
   2594 		}
   2595 	} else if (! strncmp(pm_cur->diskdev, "dk", 2)) {
   2596 		if (pm_cur->refdev == NULL)
   2597 			return -2;
   2598 		/* error = */
   2599 		run_program(RUN_DISPLAY | RUN_PROGRESS, "dkctl %s delwedge %s",
   2600 			((struct pm_devs*)pm_cur->refdev)->diskdev, pm_cur->diskdev);
   2601 #if 0 // XXX
   2602 		if (! error) {
   2603 			if (pm_cur->refdev != NULL && ((struct pm_devs*)pm_cur->refdev)->blocked > 0)
   2604 				((struct pm_devs*)pm_cur->refdev)->blocked--;
   2605 			sscanf(pm_cur->diskdev, "dk%d", &num);
   2606 			if (num >= 0 && num < MAX_WEDGES)
   2607 				wedges[num].allocated = 0;
   2608 		}
   2609 #endif
   2610 	} /* XXX: lvm */
   2611 	else
   2612 		error = run_program(RUN_DISPLAY | RUN_PROGRESS, "eject -t disk /dev/%sd",
   2613 			pm_cur->diskdev);
   2614 	if (!error)
   2615 		pm_cur->found = 0;
   2616 	return error;
   2617 }
   2618 
   2619 /* Last checks before leaving partition manager */
   2620 #if 0
   2621 static int
   2622 pm_lastcheck(void)
   2623 {
   2624 	FILE *file_tmp = fopen(concat_paths(targetroot_mnt, "/etc/fstab"), "r");
   2625 	if (file_tmp == NULL)
   2626 		return 1;
   2627 	fclose(file_tmp);
   2628 	return 0;
   2629 }
   2630 #endif
   2631 
   2632 /* Are there unsaved changes? */
   2633 static int
   2634 pm_needsave(void)
   2635 {
   2636 	struct pm_devs *pm_i;
   2637 	SLIST_FOREACH(pm_i, &pm_head, l)
   2638 		if (pm_i->unsaved) {
   2639 			/* Oops, we have unsaved changes */
   2640 			pm_changed = 1;
   2641 			msg_display(MSG_saveprompt);
   2642 			return ask_yesno(NULL);
   2643 		}
   2644 	return 0;
   2645 }
   2646 
   2647 /* Write all changes to disk */
   2648 static int
   2649 pm_commit(menudesc *m, void *arg)
   2650 {
   2651 	int retcode;
   2652 	struct pm_devs *pm_i;
   2653 
   2654 	pm_retvalue = -1;
   2655 	SLIST_FOREACH(pm_i, &pm_head, l) {
   2656 		if (! pm_i->unsaved)
   2657 			continue;
   2658 		if (pm_i->parts == NULL) {
   2659 			pm_i->unsaved = false;
   2660 			continue;
   2661 		}
   2662 		if (!pm_i->parts->pscheme->write_to_disk(pm_i->parts)) {
   2663 			if (logfp)
   2664 				fprintf(logfp, "partitining error %s\n",
   2665 				    pm_i->diskdev);
   2666 			return -1;
   2667 		}
   2668 	}
   2669 
   2670 	/* Call all functions that may create new devices */
   2671 	if ((retcode = pm_raid_commit()) != 0) {
   2672 		if (logfp)
   2673 			fprintf(logfp, "RAIDframe configuring error #%d\n", retcode);
   2674 		return -1;
   2675 	}
   2676 	if ((retcode = pm_cgd_commit()) != 0) {
   2677 		if (logfp)
   2678 			fprintf(logfp, "CGD configuring error #%d\n", retcode);
   2679 		return -1;
   2680 	}
   2681 	if ((retcode = pm_lvm_commit()) != 0) {
   2682 		if (logfp)
   2683 			fprintf(logfp, "LVM configuring error #%d\n", retcode);
   2684 		return -1;
   2685 	}
   2686 	if ((retcode = pm_vnd_commit()) != 0) {
   2687 		if (logfp)
   2688 			fprintf(logfp, "VND configuring error #%d\n", retcode);
   2689 		return -1;
   2690 	}
   2691 	if (m != NULL && arg != NULL)
   2692 		pm_upddevlist(m, arg);
   2693 	if (logfp)
   2694 		fflush (logfp);
   2695 
   2696 	pm_retvalue = 0;
   2697 	return 0;
   2698 }
   2699 
   2700 #if 0 // XXX
   2701 static int
   2702 pm_savebootsector(void)
   2703 {
   2704 	struct pm_devs *pm_i;
   2705 	SLIST_FOREACH(pm_i, &pm_head, l)
   2706 		if (pm_i->bootable) {
   2707 			if (! strncmp("raid", pm_i->diskdev, 4))
   2708 				if (run_program(RUN_DISPLAY | RUN_PROGRESS,
   2709 					"raidctl -v -A root %s", pm_i->diskdev) != 0) {
   2710 					if (logfp)
   2711 						fprintf(logfp, "Error writting RAID bootsector to %s\n",
   2712 							pm_i->diskdev);
   2713 					continue;
   2714 				}
   2715 			if (pm_i->no_part) {
   2716 				if (pm_i->rootpart < 0 ||
   2717 					run_program(RUN_DISPLAY | RUN_PROGRESS,
   2718 					"gpt biosboot -i %d %s",
   2719 					pm_i->rootpart + 1, pm_i->diskdev) != 0) {
   2720 			 		if (logfp)
   2721 						fprintf(logfp, "Error writting GPT bootsector to %s\n",
   2722 							pm_i->diskdev);
   2723 					continue;
   2724 				}
   2725 			} else {
   2726 				pm_select(pm_i);
   2727 			 	if (
   2728 #ifndef NO_DISKLABEL
   2729 				    !check_partitions(pm_i, pm_i->parts) ||
   2730 #endif
   2731 				    md_post_newfs() != 0) {
   2732 		 			if (logfp)
   2733 						fprintf(logfp, "Error writting bootsector to %s\n",
   2734 							pm_i->diskdev);
   2735 					continue;
   2736 				}
   2737 			}
   2738 		}
   2739 	return 0;
   2740 }
   2741 #endif
   2742 
   2743 /* Function for 'Enter'-menu */
   2744 static int
   2745 pm_submenu(menudesc *m, void *arg)
   2746 {
   2747 	struct pm_devs *pm_cur = NULL;
   2748 	pm_retvalue = m->cursel + 1;
   2749 	struct part_entry *cur_pe = (struct part_entry *)arg + m->cursel;
   2750 
   2751 	switch (cur_pe->type) {
   2752 		case PM_DISK:
   2753 		case PM_PART:
   2754 		case PM_SPEC:
   2755 			if (cur_pe->dev_ptr != NULL) {
   2756 				pm_cur = cur_pe->dev_ptr;
   2757 				if (pm_cur == NULL)
   2758 					return -1;
   2759 				if (pm_cur->blocked) {
   2760 					clear();
   2761 					refresh();
   2762 					msg_display(MSG_wannaunblock);
   2763 					if (!ask_noyes(NULL))
   2764 						return -2;
   2765 					pm_cur->blocked = 0;
   2766 				}
   2767 				pm_select(pm_cur);
   2768 			}
   2769 		default:
   2770 			break;
   2771 	}
   2772 
   2773 	switch (cur_pe->type) {
   2774 		case PM_DISK:
   2775 			process_menu(MENU_pmdiskentry, cur_pe);
   2776 			break;
   2777 		case PM_PART:
   2778 			process_menu(MENU_pmpartentry, cur_pe);
   2779 			break;
   2780 		case PM_SPEC:
   2781 			process_menu(MENU_pmpartentry, cur_pe);
   2782 			break;
   2783 		case PM_RAID:
   2784 			pm_edit(PMR_MENU_END, pm_raid_edit_menufmt,
   2785 			    pm_raid_set_value, pm_raid_check, pm_raid_init,
   2786 			    NULL, cur_pe->dev_ptr, 0, &raids_t_info);
   2787 			break;
   2788 		case PM_VND:
   2789 			return pm_edit(PMV_MENU_END, pm_vnd_edit_menufmt,
   2790 			    pm_vnd_set_value, pm_vnd_check, pm_vnd_init,
   2791 			    NULL, cur_pe->dev_ptr, 0, &vnds_t_info);
   2792 		case PM_CGD:
   2793 			pm_cgd_edit_old(cur_pe);
   2794 			break;
   2795 		case PM_LVM:
   2796 			return pm_edit(PML_MENU_END, pm_lvm_edit_menufmt,
   2797 			    pm_lvm_set_value, pm_lvm_check, pm_lvm_init,
   2798 			    NULL, cur_pe->dev_ptr, 0, &lvms_t_info);
   2799 		case PM_LVMLV:
   2800 			return pm_edit(PMLV_MENU_END, pm_lvmlv_edit_menufmt,
   2801 			    pm_lvmlv_set_value, pm_lvmlv_check, pm_lvmlv_init,
   2802 			    NULL, cur_pe->dev_ptr,
   2803 			    cur_pe->dev_ptr_delta, &lv_t_info);
   2804 	}
   2805 	return 0;
   2806 }
   2807 
   2808 /* Functions that generate menu entries text */
   2809 static void
   2810 pm_menufmt(menudesc *m, int opt, void *arg)
   2811 {
   2812 	const char *dev_status = "";
   2813 	char buf[STRSIZE], dev[STRSIZE];
   2814 	part_id part_num = ((struct part_entry *)arg)[opt].id;
   2815 	struct pm_devs *pm_cur = ((struct part_entry *)arg)[opt].dev_ptr;
   2816 	struct disk_partitions *parts = ((struct part_entry *)arg)[opt].parts;
   2817 	struct disk_part_info info;
   2818 	const char *mount_point, *fstype;
   2819 
   2820 	switch (((struct part_entry *)arg)[opt].type) {
   2821 		case PM_DISK:
   2822 			if (pm_cur->blocked)
   2823 				dev_status = msg_string(MSG_pmblocked);
   2824 			else if (! pm_cur->unsaved)
   2825 				dev_status = msg_string(MSG_pmunchanged);
   2826 			else
   2827 				dev_status = msg_string(MSG_pmused);
   2828 			wprintw(m->mw, "%-43.42s %25.24s",
   2829 				pm_cur->diskdev_descr,
   2830 				dev_status);
   2831 			break;
   2832 		case PM_PART:
   2833 			if (parts->pscheme->get_part_device != NULL)
   2834 				parts->pscheme->get_part_device(
   2835 				    parts,  part_num,
   2836 				    dev, sizeof dev, NULL, plain_name, false,
   2837 				    true);
   2838 			else
   2839 				strcpy(dev, "-");
   2840 			parts->pscheme->get_part_info(parts,
   2841 			    part_num, &info);
   2842 			if (pm_cur->mounted != NULL &&
   2843 			    pm_cur->mounted[part_num] != NULL &&
   2844 			    pm_cur->mounted[part_num][0] != 0)
   2845 				mount_point = msg_string(MSG_pmmounted);
   2846 			else
   2847 				mount_point = msg_string(MSG_pmunused);
   2848 			fstype = getfslabelname(info.fs_type,
   2849 			    info.fs_sub_type);
   2850 			if (info.last_mounted != NULL) {
   2851 				snprintf(buf, STRSIZE, "%s (%s) %s",
   2852 				    info.last_mounted, fstype,
   2853 				     mount_point);
   2854 				pm_fmt_disk_line(m->mw, dev, buf,
   2855 				    info.size, NULL);
   2856 			} else {
   2857 				if (fstype != NULL) {
   2858 					strlcat(dev, " (", sizeof(dev));
   2859 					strlcat(dev, fstype, sizeof(dev));
   2860 					strlcat(dev, ")", sizeof(dev));
   2861 				}
   2862 				pm_fmt_disk_line(m->mw, dev, NULL,
   2863 				    info.size, NULL);
   2864 			}
   2865 			break;
   2866 		case PM_SPEC:
   2867 			/* XXX ? */
   2868 			pm_fmt_disk_line(m->mw, pm_cur->diskdev_descr, NULL,
   2869 			    pm_cur->dlsize, NULL);
   2870 			break;
   2871 		case PM_RAID:
   2872 			pm_raid_menufmt(m, opt, arg);
   2873 			break;
   2874 		case PM_VND:
   2875 			pm_vnd_menufmt(m, opt, arg);
   2876 			break;
   2877 		case PM_CGD:
   2878 			pm_cgd_menufmt(m, opt, arg);
   2879 			break;
   2880 		case PM_LVM:
   2881 			pm_lvm_menufmt(m, opt, arg);
   2882 			break;
   2883 		case PM_LVMLV:
   2884 			pm_lvmlv_menufmt(m, opt, arg);
   2885 			break;
   2886 	}
   2887 }
   2888 
   2889 /* Submenu for RAID/LVM/CGD/VND */
   2890 static void
   2891 pm_upddevlist_adv(menudesc *m, void *arg, int *i,
   2892 	pm_upddevlist_adv_t *d)
   2893 {
   2894 	int ii;
   2895 	if (d->create_msg != NULL) {
   2896 		/* We want to have menu entry that creates a new device */
   2897 		((struct part_entry *)arg)[*i].type = d->pe_type;
   2898 		((struct part_entry *)arg)[*i].dev_ptr = NULL;
   2899 		((struct part_entry *)arg)[*i].dev_ptr_delta = d->s->parent_size * d->sub_num;
   2900 		m->opts[(*i)++] = (struct menu_ent) {
   2901 			.opt_name = d->create_msg,
   2902 			.opt_action = pm_submenu,
   2903 		};
   2904 	}
   2905 	for (ii = 0; ii < d->s->max; ii++) {
   2906 		if (d->s->entry_enabled == NULL ||
   2907 			d->s->entry_blocked == NULL ||
   2908 			*(int*)((char*)d->s->entry_enabled + d->s->entry_size * ii +
   2909 				d->s->parent_size * d->sub_num) == 0 ||
   2910 			*(int*)((char*)d->s->entry_blocked + d->s->entry_size * ii +
   2911 				d->s->parent_size * d->sub_num) != 0)
   2912 			continue;
   2913 		/* We have a entry for displaying */
   2914 		pm_changed = 1;
   2915 		m->opts[*i] = (struct menu_ent) {
   2916 			.opt_name = NULL,
   2917 			.opt_action = pm_submenu,
   2918 		};
   2919 		((struct part_entry *)arg)[*i].type = d->pe_type;
   2920 		((struct part_entry *)arg)[*i].dev_ptr = (char*)d->s->entry_first +
   2921 			d->s->entry_size * ii + d->s->parent_size * d->sub_num;
   2922 		(*i)++;
   2923 		/* We should show submenu for current entry */
   2924 		if (d->sub != NULL) {
   2925 			d->sub->sub_num = ii;
   2926 			pm_upddevlist_adv(m, arg, i, d->sub);
   2927 		}
   2928 	}
   2929 }
   2930 
   2931 /* Update partman main menu with devices list */
   2932 static int
   2933 pm_upddevlist(menudesc *m, void *arg)
   2934 {
   2935 	int i = 0;
   2936 	size_t ii;
   2937 	struct pm_devs *pm_i;
   2938 	struct disk_partitions *secondary;
   2939 	const struct disk_partitioning_scheme *ps;
   2940 	struct disk_part_info info;
   2941 
   2942 	if (arg != NULL)
   2943 		pm_retvalue = m->cursel + 1;
   2944 
   2945 	pm_changed = 0;
   2946 	/* Mark all devices as not found */
   2947 	SLIST_FOREACH(pm_i, &pm_head, l) {
   2948 		if (pm_i->parts != NULL && !pm_i->unsaved) {
   2949 			pm_i->parts->pscheme->free(pm_i->parts);
   2950 			pm_i->parts = NULL;
   2951 		}
   2952 		if (pm_i->found > 0)
   2953 			pm_i->found = 0;
   2954 	}
   2955 
   2956 	/* Detect all present devices */
   2957 	(void)find_disks("partman", false);
   2958 	if (have_lvm)
   2959 		pm_lvm_find();
   2960 	pm_clean();
   2961 
   2962 	if (m == NULL || arg == NULL)
   2963 		return -1;
   2964 
   2965 	SLIST_FOREACH(pm_i, &pm_head, l) {
   2966 		memset(&m->opts[i], 0, sizeof m->opts[i]);
   2967 		m->opts[i].opt_action = pm_submenu;
   2968 		((struct part_entry *)arg)[i].dev_ptr = pm_i;
   2969 		((struct part_entry *)arg)[i].id = NO_PART;
   2970 		if (pm_i->no_part)
   2971 			((struct part_entry *)arg)[i].type = PM_SPEC;
   2972 		else {
   2973 			ps = pm_i->parts != NULL ? pm_i->parts->pscheme : NULL;
   2974 			secondary = NULL;
   2975 
   2976 			((struct part_entry *)arg)[i].type = PM_DISK;
   2977 
   2978 			for (ii = 0; pm_i->parts != NULL &&
   2979 			    ii < pm_i->parts->num_part; ii++) {
   2980 				if (!ps->get_part_info(
   2981 				    pm_i->parts, ii, &info))
   2982 					continue;
   2983 				if (info.flags & PTI_SEC_CONTAINER) {
   2984 					if (secondary == NULL &&
   2985 					    ps->secondary_scheme != NULL)
   2986 						secondary = ps->
   2987 						    secondary_partitions(
   2988 						    pm_i->parts,
   2989 						    info.start, false);
   2990 					continue;
   2991 				}
   2992 				if (info.flags & (PTI_WHOLE_DISK|
   2993 				    PTI_PSCHEME_INTERNAL|PTI_RAW_PART))
   2994 					continue;
   2995 				if (info.fs_type == FS_UNUSED)
   2996 					continue;
   2997 				if (i >= MAX_ENTRIES)
   2998 					break;
   2999 				i++;
   3000 				memset(&m->opts[i], 0, sizeof m->opts[i]);
   3001 				m->opts[i].opt_action = pm_submenu;
   3002 				((struct part_entry *)arg)[i].parts =
   3003 				    pm_i->parts;
   3004 				((struct part_entry *)arg)[i].dev_ptr = pm_i;
   3005 				((struct part_entry *)arg)[i].id = ii;
   3006 				((struct part_entry *)arg)[i].type = PM_PART;
   3007 			}
   3008 
   3009 			for (ii = 0; secondary != NULL &&
   3010 			    ii < secondary->num_part; ii++) {
   3011 				if (!secondary->pscheme->get_part_info(
   3012 				    secondary, ii, &info))
   3013 					continue;
   3014 				if (info.flags & (PTI_WHOLE_DISK|
   3015 				    PTI_PSCHEME_INTERNAL|PTI_RAW_PART))
   3016 					continue;
   3017 				if (info.fs_type == FS_UNUSED)
   3018 					continue;
   3019 				if (i >= MAX_ENTRIES)
   3020 					break;
   3021 				i++;
   3022 				memset(&m->opts[i], 0, sizeof m->opts[i]);
   3023 				m->opts[i].opt_action = pm_submenu;
   3024 				((struct part_entry *)arg)[i].parts = secondary;
   3025 				((struct part_entry *)arg)[i].dev_ptr = pm_i;
   3026 				((struct part_entry *)arg)[i].id = ii;
   3027 				((struct part_entry *)arg)[i].type = PM_PART;
   3028 			}
   3029 		}
   3030 		i++;
   3031 	}
   3032 	for (ii = 0; ii <= (size_t)i; ii++) {
   3033 		m->opts[ii].opt_flags = OPT_EXIT;
   3034 	}
   3035 	if (have_cgd) {
   3036 		pm_upddevlist_adv(m, arg, &i,
   3037 		    &(pm_upddevlist_adv_t) {MSG_create_cgd, PM_CGD,
   3038 		    &cgds_t_info, 0, NULL});
   3039 	}
   3040 	pm_upddevlist_adv(m, arg, &i,
   3041 	    &(pm_upddevlist_adv_t) {MSG_create_vnd, PM_VND,
   3042 	    &vnds_t_info, 0, NULL});
   3043 	if (have_lvm) {
   3044 		pm_upddevlist_adv(m, arg, &i,
   3045 		    &(pm_upddevlist_adv_t) {MSG_create_vg, PM_LVM,
   3046 		    &lvms_t_info, 0,
   3047 		    &(pm_upddevlist_adv_t) {MSG_create_lv, PM_LVMLV,
   3048 		    &lv_t_info, 0,
   3049 		    NULL}});
   3050 	}
   3051 	if (have_raid) {
   3052 		pm_upddevlist_adv(m, arg, &i,
   3053 		    &(pm_upddevlist_adv_t) {MSG_create_raid, PM_RAID, &raids_t_info, 0, NULL});
   3054 	}
   3055 
   3056 	m->opts[i++] = (struct menu_ent) {
   3057 		.opt_name = MSG_updpmlist,
   3058 		.opt_action = pm_upddevlist,
   3059 	};
   3060 	m->opts[i  ] = (struct menu_ent) {
   3061 		.opt_name = MSG_savepm,
   3062 		.opt_action = pm_commit,
   3063 	};
   3064 
   3065 	if (pm_retvalue >= 0)
   3066 		m->cursel = pm_retvalue - 1;
   3067 	return i;
   3068 }
   3069 
   3070 static void
   3071 pm_menuin(menudesc *m, void *arg)
   3072 {
   3073 	if (pm_cursel > m->numopts)
   3074 		m->cursel = m->numopts;
   3075 	else if (pm_cursel < 0)
   3076 		m->cursel = 0;
   3077 	else
   3078 		m->cursel = pm_cursel;
   3079 }
   3080 
   3081 static void
   3082 pm_menuout(menudesc *m, void *arg)
   3083 {
   3084 	pm_cursel = m->cursel;
   3085 }
   3086 
   3087 /* Main partman function */
   3088 int
   3089 partman(void)
   3090 {
   3091 	int menu_no, menu_num_entries;
   3092 	static int firstrun = 1;
   3093 	menu_ent menu_entries[MAX_ENTRIES+6];
   3094 	struct part_entry args[MAX_ENTRIES];
   3095 
   3096 	if (firstrun) {
   3097 		check_available_binaries();
   3098 
   3099 		if (!have_raid)
   3100 			remove_raid_options();
   3101 		else if (!(raids = calloc(MAX_RAID, sizeof(*raids))))
   3102 			have_raid = 0;
   3103 
   3104 #define remove_vnd_options() (void)0
   3105 		if (!have_vnd)
   3106 			remove_vnd_options();
   3107 		else if (!(vnds = calloc(MAX_VND, sizeof(*vnds))))
   3108 			have_vnd = 0;
   3109 
   3110 		if (!have_cgd)
   3111 			remove_cgd_options();
   3112 		else if (!(cgds = calloc(MAX_CGD, sizeof(*cgds))))
   3113 			have_cgd = 0;
   3114 
   3115 		if (!have_lvm)
   3116 			remove_lvm_options();
   3117 		else if (!(lvms = calloc(MAX_LVM_VG, sizeof(*lvms))))
   3118 			have_lvm = 0;
   3119 
   3120 		raids_t_info = (structinfo_t) {
   3121 			.max = MAX_RAID,
   3122 			.entry_size = sizeof raids[0],
   3123 			.entry_first = &raids[0],
   3124 			.entry_enabled = &(raids[0].enabled),
   3125 			.entry_blocked = &(raids[0].blocked),
   3126 			.entry_node = &(raids[0].node),
   3127 		};
   3128 		vnds_t_info = (structinfo_t) {
   3129 			.max = MAX_VND,
   3130 			.entry_size = sizeof vnds[0],
   3131 			.entry_first = &vnds[0],
   3132 			.entry_enabled = &(vnds[0].enabled),
   3133 			.entry_blocked = &(vnds[0].blocked),
   3134 			.entry_node = &(vnds[0].node),
   3135 		};
   3136 		cgds_t_info = (structinfo_t) {
   3137 			.max = MAX_CGD,
   3138 			.entry_size = sizeof cgds[0],
   3139 			.entry_first = &cgds[0],
   3140 			.entry_enabled = &(cgds[0].enabled),
   3141 			.entry_blocked = &(cgds[0].blocked),
   3142 			.entry_node = &(cgds[0].node),
   3143 		};
   3144 		lvms_t_info = (structinfo_t) {
   3145 			.max = MAX_LVM_VG,
   3146 			.entry_size = sizeof lvms[0],
   3147 			.entry_first = &lvms[0],
   3148 			.entry_enabled = &(lvms[0].enabled),
   3149 			.entry_blocked = &(lvms[0].blocked),
   3150 			.entry_node = NULL,
   3151 		};
   3152 		lv_t_info = (structinfo_t) {
   3153 			.max = MAX_LVM_LV,
   3154 			.entry_size = sizeof lvms[0].lv[0],
   3155 			.entry_first = &lvms[0].lv[0],
   3156 			.entry_enabled = &(lvms[0].lv[0].size),
   3157 			.entry_blocked = &(lvms[0].lv[0].blocked),
   3158 			.parent_size = sizeof lvms[0],
   3159 		};
   3160 
   3161 		pm_cursel = 0;
   3162 		pm_changed = 0;
   3163 		firstrun = 0;
   3164 	}
   3165 
   3166 	do {
   3167 		menu_num_entries = pm_upddevlist(&(menudesc){.opts = menu_entries}, args);
   3168 		menu_no = new_menu(MSG_partman_header,
   3169 			menu_entries, menu_num_entries+1, 1, 1, 0, 75, /* Fixed width */
   3170 			MC_ALWAYS_SCROLL | MC_NOBOX | MC_NOCLEAR,
   3171 			pm_menuin, pm_menufmt, pm_menuout, NULL, MSG_finishpm);
   3172 		if (menu_no == -1)
   3173 			pm_retvalue = -1;
   3174 		else {
   3175 			pm_retvalue = 0;
   3176 			clear();
   3177 			refresh();
   3178 			process_menu(menu_no, &args);
   3179 			free_menu(menu_no);
   3180 		}
   3181 
   3182 		if (pm_retvalue == 0 && pm->parts != NULL)
   3183 			if (pm_needsave())
   3184 				pm_commit(NULL, NULL);
   3185 
   3186 	} while (pm_retvalue > 0);
   3187 
   3188 	/* retvalue <0 - error, retvalue ==0 - user quits, retvalue >0 - all ok */
   3189 	return (pm_retvalue >= 0)?0:-1;
   3190 }
   3191 
   3192 void
   3193 update_wedges(const char *disk)
   3194 {
   3195 	check_available_binaries();
   3196 
   3197 	if (!have_dk)
   3198 		return;
   3199 
   3200 	run_program(RUN_SILENT | RUN_ERROR_OK,
   3201 	    "dkctl %s makewedges", disk);
   3202 }
   3203 
   3204 bool
   3205 pm_force_parts(struct pm_devs *my_pm)
   3206 {
   3207 	if (my_pm == NULL)
   3208 		return false;
   3209 	if (my_pm->parts != NULL)
   3210 		return true;
   3211 
   3212 	const struct disk_partitioning_scheme *ps =
   3213 	    select_part_scheme(my_pm, NULL, false, NULL);
   3214 	if (ps == NULL)
   3215 		return false;
   3216 
   3217 	struct disk_partitions *parts =
   3218 	   (*ps->create_new_for_disk)(my_pm->diskdev, 0, my_pm->dlsize,
   3219 	    my_pm->dlsize, false, NULL);
   3220 	if (parts == NULL)
   3221 		return false;
   3222 
   3223 	my_pm->parts = parts;
   3224 	if (pm->dlsize > ps->size_limit)
   3225 		pm->dlsize = ps->size_limit;
   3226 
   3227 	return true;
   3228 }
   3229 
   3230 void
   3231 pm_edit_partitions(struct part_entry *pe)
   3232 {
   3233 	struct pm_devs *my_pm = pm_from_pe(pe);
   3234 	struct partition_usage_set pset = { 0 };
   3235 
   3236 	if (!my_pm)
   3237 		return;
   3238 
   3239 	if (!pm_force_parts(my_pm))
   3240 		return;
   3241 
   3242 	clear();
   3243 	refresh();
   3244 
   3245 	usage_set_from_parts(&pset, my_pm->parts);
   3246 	edit_and_check_label(my_pm, &pset, false);
   3247 	free_usage_set(&pset);
   3248 
   3249 	pm_partusage(my_pm, -1, -1);
   3250 	my_pm->unsaved = true;
   3251 	pm_retvalue = 1;
   3252 }
   3253 
   3254 part_id
   3255 pm_whole_disk(struct part_entry *pe, int t)
   3256 {
   3257 	struct pm_devs *my_pm = pm_from_pe(pe);
   3258 	struct disk_partitions *parts, *outer;
   3259 	struct disk_part_info info, oinfo;
   3260 	struct disk_part_free_space space;
   3261 	daddr_t align;
   3262 	int fst;
   3263 	struct partition_usage_set pset = { 0 };
   3264 	part_id new_part, id;
   3265 	size_t i, cnt;
   3266 
   3267 	if (!my_pm)
   3268 		return NO_PART;
   3269 
   3270 	if (!pm_force_parts(my_pm))
   3271 		return NO_PART;
   3272 
   3273 	parts = my_pm->parts;
   3274 	parts->pscheme->delete_all_partitions(parts);
   3275 	if (parts->pscheme->secondary_scheme != NULL) {
   3276 		outer = parts;
   3277 		parts = parts->pscheme->secondary_partitions(outer,
   3278 		    0, true);
   3279 		if (parts == NULL) {
   3280 			parts = outer;
   3281 		} else {
   3282 			if (outer->pscheme->write_to_disk(outer))
   3283 				my_pm->parts = parts;
   3284 		}
   3285 	}
   3286 
   3287 	align = parts->pscheme->get_part_alignment(parts);
   3288 
   3289 	memset(&info, 0, sizeof info);
   3290 	switch (t) {
   3291 	case SY_NEWRAID:
   3292 		fst = FS_RAID;
   3293 		break;
   3294 	case SY_NEWLVM:
   3295 		fst = FS_BSDFFS;
   3296 		break;
   3297 	case SY_NEWCGD:
   3298 		fst = FS_CGD;
   3299 		break;
   3300 	default:
   3301 		assert(false);
   3302 		return NO_PART;
   3303 	}
   3304 	info.nat_type = parts->pscheme->get_fs_part_type(PT_root, fst, 0);
   3305 	if (info.nat_type != NULL && parts->pscheme->get_default_fstype != NULL)
   3306 		parts->pscheme->get_default_fstype(info.nat_type,
   3307 		    &info.fs_type, &info.fs_sub_type);
   3308 	if (parts->pscheme->get_free_spaces(parts, &space, 1,
   3309 	    5*align, align, -1, -1) != 1)
   3310 		return NO_PART;
   3311 	info.start = space.start;
   3312 	info.size = space.size;
   3313 	new_part = parts->pscheme->add_partition(parts, &info, NULL);
   3314 	if (new_part == NO_PART)
   3315 		return NO_PART;
   3316 
   3317 	parts->pscheme->get_part_info(parts, new_part, &oinfo);
   3318 
   3319 	clear();
   3320 	refresh();
   3321 
   3322 	usage_set_from_parts(&pset, parts);
   3323 	edit_and_check_label(my_pm, &pset, false);
   3324 	free_usage_set(&pset);
   3325 
   3326 	/*
   3327 	 * Try to match our new partition after user edit
   3328 	 */
   3329 	new_part = NO_PART;
   3330 	for (cnt = i = 0; i < parts->num_part; i++) {
   3331 		if (!parts->pscheme->get_part_info(parts,i, &info))
   3332 			continue;
   3333 		if (info.flags & (PTI_SEC_CONTAINER|PTI_WHOLE_DISK|
   3334 		    PTI_PSCHEME_INTERNAL|PTI_RAW_PART))
   3335 			continue;
   3336 		if (info.nat_type != oinfo.nat_type)
   3337 			continue;
   3338 		if (new_part == NO_PART)
   3339 			new_part = i;
   3340 		cnt++;
   3341 	}
   3342 	if (cnt > 1) {
   3343 		/* multiple matches, retry matching with start */
   3344 		id = NO_PART;
   3345 		for (cnt = i = 0; i < parts->num_part; i++) {
   3346 			if (!parts->pscheme->get_part_info(parts, i, &info))
   3347 				continue;
   3348 			if (info.flags & (PTI_SEC_CONTAINER|PTI_WHOLE_DISK|
   3349 			    PTI_PSCHEME_INTERNAL|PTI_RAW_PART))
   3350 				continue;
   3351 			if (info.nat_type != oinfo.nat_type)
   3352 				continue;
   3353 			if (info.start != oinfo.start)
   3354 				continue;
   3355 			if (id == NO_PART)
   3356 				id = i;
   3357 			cnt++;
   3358 		}
   3359 		if (id != NO_PART)
   3360 			new_part = id;
   3361 	}
   3362 
   3363 	clear();
   3364 	refresh();
   3365 
   3366 	pm_partusage(my_pm, -1, -1);
   3367 	my_pm->unsaved = true;
   3368 	pm_retvalue = 1;
   3369 
   3370 	return new_part;
   3371 }
   3372 
   3373 struct pm_devs *
   3374 pm_from_pe(struct part_entry *pe)
   3375 {
   3376 	switch (pe->type) {
   3377 	case PM_DISK:
   3378 		return pe->dev_ptr;
   3379 	default:
   3380 		assert(false);
   3381 	}
   3382 	return NULL;
   3383 }
   3384