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