source: mainline/uspace/app/fdisk/fdisk.c@ 2d78d88

lfn serial ticket/834-toolchain-update topic/msim-upgrade topic/simplify-dev-export
Last change on this file since 2d78d88 was 2d78d88, checked in by Jiri Svoboda <jiri@…>, 7 years ago

Modifying mount point for a partition.
  • Property mode set to 100644
File size: 24.5 KB
RevLine 
[e96047c]1/*
[64ffd83]2 * Copyright (c) 2018 Jiri Svoboda
[e96047c]3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 *
9 * - Redistributions of source code must retain the above copyright
10 * notice, this list of conditions and the following disclaimer.
11 * - Redistributions in binary form must reproduce the above copyright
12 * notice, this list of conditions and the following disclaimer in the
13 * documentation and/or other materials provided with the distribution.
14 * - The name of the author may not be used to endorse or promote products
15 * derived from this software without specific prior written permission.
16 *
17 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
18 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
19 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
20 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
21 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
22 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
23 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
24 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
25 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
26 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27 */
28
29/** @addtogroup fdisk
30 * @brief Disk management tool
31 * @{
32 */
33/**
34 * @file
35 */
36
[6c4eedf]37#include <cap.h>
[e96047c]38#include <errno.h>
[db9c889]39#include <fdisk.h>
40#include <io/label.h>
[e96047c]41#include <nchoice.h>
42#include <stdbool.h>
43#include <stdio.h>
44#include <stdlib.h>
[1d6dd2a]45#include <str.h>
[db9c889]46#include <vol.h>
[e96047c]47
[9c2c7d2]48#define NO_LABEL_CAPTION "(No name)"
49
[e96047c]50static bool quit = false;
[22fb7ab]51static fdisk_t *fdisk;
[e96047c]52
53/** Device menu actions */
54typedef enum {
55 /** Create label */
56 devac_create_label,
57 /** Delete label */
58 devac_delete_label,
[ea0ff6b]59 /** Erase disk */
60 devac_erase_disk,
[b7a4d06]61 /** Create (primary) partition */
62 devac_create_pri_part,
63 /** Create extended partition */
64 devac_create_ext_part,
65 /** Create logical partition */
66 devac_create_log_part,
[2d78d88]67 /** Modify partition */
68 devac_modify_part,
[e96047c]69 /** Delete partition */
70 devac_delete_part,
71 /** Exit */
72 devac_exit
73} devac_t;
74
[2d78d88]75/** Partition property to modify */
76typedef enum {
77 /** Modify mount point */
78 pm_mountp,
79 /** Cancel */
80 pm_cancel
81} pmprop_t;
82
[ef9dac04]83/** Confirm user selection. */
[b7fd2a0]84static errno_t fdsk_confirm(const char *msg, bool *rconfirm)
[ef9dac04]85{
86 tinput_t *tinput = NULL;
87 char *answer;
[b7fd2a0]88 errno_t rc;
[ef9dac04]89
90 tinput = tinput_new();
91 if (tinput == NULL) {
92 rc = ENOMEM;
93 goto error;
94 }
95
96 rc = tinput_set_prompt(tinput, "y/n> ");
97 if (rc != EOK)
98 goto error;
99
100 while (true) {
101 printf("%s\n", msg);
102
103 rc = tinput_read(tinput, &answer);
104 if (rc == ENOENT) {
105 *rconfirm = false;
106 free(answer);
107 break;
108 }
109
110 if (rc != EOK)
111 goto error;
112
113 if (str_cmp(answer, "y") == 0) {
114 *rconfirm = true;
115 free(answer);
116 break;
117 } else if (str_cmp(answer, "n") == 0) {
118 *rconfirm = false;
119 free(answer);
120 break;
121 }
122 }
123
124 tinput_destroy(tinput);
125 return EOK;
126error:
127 if (tinput != NULL)
128 tinput_destroy(tinput);
129 return rc;
130}
131
[e96047c]132/** Device selection */
[b7fd2a0]133static errno_t fdsk_dev_sel_choice(service_id_t *rsvcid)
[e96047c]134{
135 fdisk_dev_list_t *devlist = NULL;
136 fdisk_dev_info_t *info;
137 nchoice_t *choice = NULL;
138 char *svcname = NULL;
[6c4eedf]139 cap_spec_t cap;
[8227d63]140 fdisk_dev_info_t *sdev;
[e96047c]141 char *scap = NULL;
142 char *dtext = NULL;
143 service_id_t svcid;
144 void *sel;
[8227d63]145 int ndevs;
[b7fd2a0]146 errno_t rc;
[e96047c]147
148 rc = nchoice_create(&choice);
149 if (rc != EOK) {
150 assert(rc == ENOMEM);
151 printf("Out of memory.\n");
152 goto error;
153 }
154
155 rc = nchoice_set_prompt(choice, "Select device");
156 if (rc != EOK) {
157 assert(rc == ENOMEM);
158 printf("Out of memory.\n");
159 goto error;
160 }
161
[22fb7ab]162 rc = fdisk_dev_list_get(fdisk, &devlist);
[e96047c]163 if (rc != EOK) {
164 printf("Error getting device list.\n");
165 goto error;
166 }
167
168 info = fdisk_dev_first(devlist);
[8227d63]169 ndevs = 0;
[e96047c]170 while (info != NULL) {
[8227d63]171 rc = fdisk_dev_info_get_svcname(info, &svcname);
[e96047c]172 if (rc != EOK) {
[4627314]173 fdisk_dev_info_get_svcid(info, &svcid);
174 printf("Error getting device service name "
175 "(service ID %zu).\n", svcid);
176 info = fdisk_dev_next(info);
177 continue;
[e96047c]178 }
179
[8227d63]180 rc = fdisk_dev_info_capacity(info, &cap);
[e96047c]181 if (rc != EOK) {
[4627314]182 printf("Error getting device capacity "
183 "(device %s).\n", svcname);
184 info = fdisk_dev_next(info);
185 continue;
[e96047c]186 }
187
[6c4eedf]188 cap_simplify(&cap);
[9854a8f]189
[6c4eedf]190 rc = cap_format(&cap, &scap);
[e96047c]191 if (rc != EOK) {
192 assert(rc == ENOMEM);
193 printf("Out of memory.\n");
194 goto error;
195 }
196
[d5c1051]197 int ret = asprintf(&dtext, "%s (%s)", svcname, scap);
198 if (ret < 0) {
[e96047c]199 rc = ENOMEM;
200 printf("Out of memory.\n");
201 goto error;
202 }
203
204 free(svcname);
205 svcname = NULL;
206 free(scap);
207 scap = NULL;
208
[68b5dd11]209 rc = nchoice_add(choice, dtext, info, 0);
[e96047c]210 if (rc != EOK) {
211 assert(rc == ENOMEM);
212 printf("Out of memory.\n");
213 goto error;
214 }
215
[4627314]216 ++ndevs;
217
[e96047c]218 free(dtext);
219 dtext = NULL;
220
221 info = fdisk_dev_next(info);
222 }
223
[8227d63]224 if (ndevs == 0) {
225 printf("No disk devices found.\n");
226 rc = ENOENT;
227 goto error;
228 }
229
[68b5dd11]230 rc = nchoice_add(choice, "Exit", NULL, 0);
[8227d63]231 if (rc != EOK) {
232 assert(rc == ENOMEM);
233 printf("Out of memory.\n");
234 goto error;
235 }
236
[e96047c]237 rc = nchoice_get(choice, &sel);
238 if (rc != EOK) {
239 printf("Error getting user selection.\n");
240 return rc;
241 }
242
[8227d63]243 sdev = (fdisk_dev_info_t *)sel;
244 if (sdev != NULL) {
245 fdisk_dev_info_get_svcid(sdev, &svcid);
246 } else {
247 svcid = 0;
248 }
[e96047c]249
[8227d63]250 fdisk_dev_list_free(devlist);
[e96047c]251
252 nchoice_destroy(choice);
253 *rsvcid = svcid;
254 return EOK;
255error:
[9b07fba]256 assert(rc != EOK);
257 *rsvcid = 0;
[e96047c]258 if (devlist != NULL)
259 fdisk_dev_list_free(devlist);
260 if (choice != NULL)
261 nchoice_destroy(choice);
262 free(dtext);
263 free(svcname);
264 free(scap);
265 return rc;
266}
267
[b7fd2a0]268static errno_t fdsk_create_label(fdisk_dev_t *dev)
[e96047c]269{
270 nchoice_t *choice = NULL;
271 void *sel;
[8227d63]272 char *sltype = NULL;
273 int i;
[b7fd2a0]274 errno_t rc;
[e96047c]275
276 rc = nchoice_create(&choice);
277 if (rc != EOK) {
278 assert(rc == ENOMEM);
279 printf("Out of memory.\n");
280 goto error;
281 }
282
283 rc = nchoice_set_prompt(choice, "Select label type");
284 if (rc != EOK) {
285 assert(rc == ENOMEM);
286 printf("Out of memory.\n");
287 goto error;
288 }
289
[372df8f]290 for (i = LT_FIRST; i < LT_LIMIT; i++) {
[db9c889]291 rc = label_type_format(i, &sltype);
[8227d63]292 if (rc != EOK)
293 goto error;
294
[68b5dd11]295 rc = nchoice_add(choice, sltype, (void *)(uintptr_t)i,
296 i == LT_DEFAULT);
[8227d63]297 if (rc != EOK) {
298 assert(rc == ENOMEM);
299 printf("Out of memory.\n");
300 goto error;
301 }
302
303 free(sltype);
304 sltype = NULL;
305 }
306
307 rc = nchoice_get(choice, &sel);
308 if (rc != EOK) {
309 printf("Error getting user selection.\n");
310 goto error;
311 }
312
[22fb7ab]313 rc = fdisk_label_create(dev, (label_type_t)sel);
[8227d63]314 if (rc != EOK) {
315 printf("Error creating label.\n");
316 goto error;
317 }
318
319 nchoice_destroy(choice);
320 return EOK;
321error:
322 free(sltype);
323 if (choice != NULL)
324 nchoice_destroy(choice);
325 return rc;
326}
327
[b7fd2a0]328static errno_t fdsk_delete_label(fdisk_dev_t *dev)
[8227d63]329{
[ef9dac04]330 bool confirm;
[b7fd2a0]331 errno_t rc;
[8227d63]332
[ef9dac04]333 rc = fdsk_confirm("Warning. Any data on disk will be lost. "
334 "Really delete label?", &confirm);
335 if (rc != EOK) {
336 printf("Error getting user confirmation.\n");
337 return rc;
338 }
339
340 if (!confirm)
341 return EOK;
342
[8227d63]343 rc = fdisk_label_destroy(dev);
344 if (rc != EOK) {
345 printf("Error deleting label.\n");
346 return rc;
347 }
348
349 return EOK;
350}
351
[b7fd2a0]352static errno_t fdsk_erase_disk(fdisk_dev_t *dev)
[ea0ff6b]353{
[ef9dac04]354 bool confirm;
[b7fd2a0]355 errno_t rc;
[ea0ff6b]356
[ef9dac04]357 rc = fdsk_confirm("Warning. Any data on disk will be lost. "
358 "Really erase disk?", &confirm);
359 if (rc != EOK) {
360 printf("Error getting user confirmation.\n");
361 return rc;
362 }
363
364 if (!confirm)
365 return EOK;
366
[ea0ff6b]367 rc = fdisk_dev_erase(dev);
368 if (rc != EOK) {
369 printf("Error erasing disk.\n");
370 return rc;
371 }
372
373 return EOK;
374}
375
[b7fd2a0]376static errno_t fdsk_select_fstype(vol_fstype_t *fstype)
[8227d63]377{
378 nchoice_t *choice = NULL;
379 void *sel;
380 char *sfstype;
381 int i;
[b7fd2a0]382 errno_t rc;
[8227d63]383
384 rc = nchoice_create(&choice);
[e96047c]385 if (rc != EOK) {
386 assert(rc == ENOMEM);
387 printf("Out of memory.\n");
388 goto error;
389 }
390
[8227d63]391 rc = nchoice_set_prompt(choice, "Select file system type");
[e96047c]392 if (rc != EOK) {
393 assert(rc == ENOMEM);
394 printf("Out of memory.\n");
395 goto error;
396 }
397
[4b6635a7]398 for (i = 0; i < VOL_FSTYPE_LIMIT; i++) {
[db9c889]399 rc = vol_fstype_format(i, &sfstype);
[8227d63]400 if (rc != EOK)
401 goto error;
402
[68b5dd11]403 rc = nchoice_add(choice, sfstype, (void *)(uintptr_t)i,
404 i == VOL_FSTYPE_DEFAULT);
[8227d63]405 if (rc != EOK) {
406 assert(rc == ENOMEM);
407 printf("Out of memory.\n");
408 goto error;
409 }
410
411 free(sfstype);
412 sfstype = NULL;
413 }
414
[e96047c]415 rc = nchoice_get(choice, &sel);
416 if (rc != EOK) {
417 printf("Error getting user selection.\n");
418 goto error;
419 }
420
[8227d63]421 nchoice_destroy(choice);
[4b6635a7]422 *fstype = (vol_fstype_t)sel;
[8227d63]423 return EOK;
424error:
425 free(sfstype);
426 if (choice != NULL)
427 nchoice_destroy(choice);
428 return rc;
429}
430
[b7fd2a0]431static errno_t fdsk_create_part(fdisk_dev_t *dev, label_pkind_t pkind)
[8227d63]432{
[b7fd2a0]433 errno_t rc;
[8227d63]434 fdisk_part_spec_t pspec;
[6c4eedf]435 cap_spec_t cap;
436 cap_spec_t mcap;
[9c2c7d2]437 vol_label_supp_t vlsupp;
[4b6635a7]438 vol_fstype_t fstype = 0;
[8227d63]439 tinput_t *tinput = NULL;
[b598460a]440 fdisk_spc_t spc;
[8227d63]441 char *scap;
[b598460a]442 char *smcap = NULL;
[9c2c7d2]443 char *label = NULL;
[64ffd83]444 char *mountp = NULL;
[b598460a]445
446 if (pkind == lpk_logical)
447 spc = spc_log;
448 else
449 spc = spc_pri;
450
451 rc = fdisk_part_get_max_avail(dev, spc, &mcap);
452 if (rc != EOK) {
453 rc = EIO;
454 goto error;
455 }
456
[6c4eedf]457 cap_simplify(&mcap);
[03661d19]458
[6c4eedf]459 rc = cap_format(&mcap, &smcap);
[b598460a]460 if (rc != EOK) {
461 rc = ENOMEM;
462 goto error;
463 }
[8227d63]464
465 tinput = tinput_new();
466 if (tinput == NULL) {
467 rc = ENOMEM;
468 goto error;
469 }
470
471 rc = tinput_set_prompt(tinput, "?> ");
472 if (rc != EOK)
473 goto error;
474
475 while (true) {
476 printf("Enter capacity of new partition.\n");
[b598460a]477 rc = tinput_read_i(tinput, smcap, &scap);
[8227d63]478 if (rc != EOK)
479 goto error;
480
[6c4eedf]481 rc = cap_parse(scap, &cap);
[8227d63]482 if (rc == EOK)
483 break;
484 }
485
486 tinput_destroy(tinput);
[78d50bd]487 tinput = NULL;
[b598460a]488 free(smcap);
489 smcap = NULL;
[8227d63]490
[b7a4d06]491 if (pkind != lpk_extended) {
492 rc = fdsk_select_fstype(&fstype);
493 if (rc != EOK)
494 goto error;
495 }
[8227d63]496
[9c2c7d2]497 fdisk_get_vollabel_support(dev, fstype, &vlsupp);
498 if (vlsupp.supported) {
499 tinput = tinput_new();
500 if (tinput == NULL) {
501 rc = ENOMEM;
502 goto error;
503 }
504
505 rc = tinput_set_prompt(tinput, "?> ");
506 if (rc != EOK)
507 goto error;
508
509 /* Ask for volume label */
510 printf("Enter volume label for new partition.\n");
[f3504c1]511 rc = tinput_read_i(tinput, "New volume", &label);
[9c2c7d2]512 if (rc != EOK)
513 goto error;
514
[1433ecda]515 tinput_destroy(tinput);
[9c2c7d2]516 tinput = NULL;
517 }
518
[64ffd83]519
[8227d63]520 fdisk_pspec_init(&pspec);
521 pspec.capacity = cap;
[b7a4d06]522 pspec.pkind = pkind;
[8227d63]523 pspec.fstype = fstype;
[9c2c7d2]524 pspec.label = label;
[64ffd83]525 pspec.mountp = mountp;
[8227d63]526
527 rc = fdisk_part_create(dev, &pspec, NULL);
[e96047c]528 if (rc != EOK) {
[8227d63]529 printf("Error creating partition.\n");
[e96047c]530 goto error;
531 }
532
[9c2c7d2]533 free(label);
[64ffd83]534 free(mountp);
[8227d63]535 return EOK;
536error:
[b598460a]537 free(smcap);
[9c2c7d2]538 free(label);
[64ffd83]539 free(mountp);
[8227d63]540 if (tinput != NULL)
541 tinput_destroy(tinput);
542 return rc;
543}
544
[2d78d88]545/** Add an option to choice for each partition.
546 *
547 * @param dev Fdisk device
548 * @param choice Choice to add optionts to
549 *
550 * @return EOK on sucess or error code
551 */
552static errno_t fdsk_add_part_choices(fdisk_dev_t *dev,
553 nchoice_t *choice)
[8227d63]554{
555 fdisk_part_t *part;
556 fdisk_part_info_t pinfo;
557 char *scap = NULL;
[21f1543]558 char *spkind = NULL;
[8227d63]559 char *sfstype = NULL;
560 char *sdesc = NULL;
[9c2c7d2]561 const char *label;
[b7fd2a0]562 errno_t rc;
[8227d63]563
564 part = fdisk_part_first(dev);
565 while (part != NULL) {
566 rc = fdisk_part_get_info(part, &pinfo);
567 if (rc != EOK) {
568 printf("Error getting partition information.\n");
569 goto error;
570 }
571
[6c4eedf]572 cap_simplify(&pinfo.capacity);
[9854a8f]573
[6c4eedf]574 rc = cap_format(&pinfo.capacity, &scap);
[8227d63]575 if (rc != EOK) {
576 printf("Out of memory.\n");
577 goto error;
578 }
579
[db9c889]580 rc = label_pkind_format(pinfo.pkind, &spkind);
[8227d63]581 if (rc != EOK) {
[21f1543]582 printf("\nOut of memory.\n");
[8227d63]583 goto error;
584 }
585
[21f1543]586 if (pinfo.pkind != lpk_extended) {
[db9c889]587 rc = vol_pcnt_fs_format(pinfo.pcnt, pinfo.fstype, &sfstype);
[21f1543]588 if (rc != EOK) {
589 printf("Out of memory.\n");
590 goto error;
591 }
592
[9c2c7d2]593 if (str_size(pinfo.label) > 0)
594 label = pinfo.label;
595 else
596 label = "(No name)";
597
[d5c1051]598 int ret = asprintf(&sdesc, "%s %s, %s, %s", label,
[9c2c7d2]599 scap, spkind, sfstype);
[d5c1051]600 if (ret < 0) {
[21f1543]601 rc = ENOMEM;
602 goto error;
603 }
604
605 } else {
[d5c1051]606 int ret = asprintf(&sdesc, "%s, %s", scap, spkind);
607 if (ret < 0) {
[21f1543]608 rc = ENOMEM;
609 goto error;
610 }
[8227d63]611 }
612
[68b5dd11]613 rc = nchoice_add(choice, sdesc, (void *)part, 0);
[8227d63]614 if (rc != EOK) {
615 assert(rc == ENOMEM);
616 printf("Out of memory.\n");
617 goto error;
618 }
619
620 free(scap);
621 scap = NULL;
[21f1543]622 free(spkind);
623 spkind = NULL;
[8227d63]624 free(sfstype);
625 sfstype = NULL;
626 free(sdesc);
627 sdesc = NULL;
628
629 part = fdisk_part_next(part);
630 }
631
[2d78d88]632 return EOK;
633error:
634 free(scap);
635 free(spkind);
636 free(sfstype);
637 free(sdesc);
638
639 return rc;
640}
641
642/** Modify partition mount point.
643 *
644 * Run the interaction to modify a partition mount point
645 *
646 * @part Fdisk partition
647 * @return EOK on success or error code
648 */
649static errno_t fdsk_modify_mountp(fdisk_part_t *part)
650{
651 tinput_t *tinput = NULL;
652 errno_t rc;
653 char *mountp = NULL;
654
655 /* Ask for mount point */
656 tinput = tinput_new();
657 if (tinput == NULL) {
658 rc = ENOMEM;
659 goto error;
660 }
661
662 rc = tinput_set_prompt(tinput, "?> ");
663 if (rc != EOK)
664 goto error;
665
666 while (true) {
667 printf("Enter mount point for new partition (Auto, None or /path).\n");
668 rc = tinput_read_i(tinput, "Auto", &mountp);
669 if (rc != EOK)
670 goto error;
671
672 rc = vol_mountp_validate(mountp);
673 if (rc == EOK)
674 break;
675
676 free(mountp);
677 mountp = NULL;
678 }
679
680 rc = fdisk_part_set_mountp(part, mountp);
681 if (rc != EOK)
682 goto error;
683
684 free(mountp);
685
686 tinput_destroy(tinput);
687 tinput = NULL;
688 return EOK;
689error:
690 if (mountp != NULL)
691 free(mountp);
692 if (tinput != NULL)
693 tinput_destroy(tinput);
694 return rc;
695}
696
697/** Modify partition.
698 *
699 * Run the interaction to modify a partition.
700 *
701 * @param dev Fdisk device
702 * @return EOK on success or error code
703 */
704static errno_t fdsk_modify_part(fdisk_dev_t *dev)
705{
706 nchoice_t *choice = NULL;
707 fdisk_part_t *part;
708 void *sel;
709 errno_t rc;
710
711 rc = nchoice_create(&choice);
712 if (rc != EOK) {
713 assert(rc == ENOMEM);
714 printf("Out of memory.\n");
715 goto error;
716 }
717
718 rc = nchoice_set_prompt(choice, "Select partition to modify");
719 if (rc != EOK) {
720 assert(rc == ENOMEM);
721 printf("Out of memory.\n");
722 goto error;
723 }
724
725 rc = fdsk_add_part_choices(dev, choice);
726 if (rc != EOK)
727 goto error;
728
729 rc = nchoice_add(choice, "Cancel", NULL, 0);
730 if (rc != EOK) {
731 assert(rc == ENOMEM);
732 printf("Out of memory.\n");
733 goto error;
734 }
735
736 rc = nchoice_get(choice, &sel);
737 if (rc == ENOENT)
738 return EOK;
739 if (rc != EOK) {
740 printf("Error getting user selection.\n");
741 goto error;
742 }
743
744 nchoice_destroy(choice);
745 choice = NULL;
746
747 if (sel == NULL)
748 return EOK;
749
750 part = (fdisk_part_t *)sel;
751
752 rc = nchoice_create(&choice);
753 if (rc != EOK) {
754 assert(rc == ENOMEM);
755 printf("Out of memory.\n");
756 goto error;
757 }
758
759 rc = nchoice_set_prompt(choice, "Select property to modify");
760 if (rc != EOK) {
761 assert(rc == ENOMEM);
762 printf("Out of memory.\n");
763 goto error;
764 }
765
766 rc = nchoice_add(choice, "Mount point", (void *)pm_mountp, 0);
767 if (rc != EOK) {
768 assert(rc == ENOMEM);
769 printf("Out of memory.\n");
770 goto error;
771 }
772
773 rc = nchoice_add(choice, "Cancel", (void *)pm_cancel, 0);
774 if (rc != EOK) {
775 assert(rc == ENOMEM);
776 printf("Out of memory.\n");
777 goto error;
778 }
779
780 rc = nchoice_get(choice, &sel);
781 if (rc == ENOENT)
782 return EOK;
783 if (rc != EOK) {
784 printf("Error getting user selection.\n");
785 goto error;
786 }
787
788 nchoice_destroy(choice);
789 choice = NULL;
790
791 switch ((pmprop_t)sel) {
792 case pm_mountp:
793 rc = fdsk_modify_mountp(part);
794 break;
795 case pm_cancel:
796 rc = EOK;
797 break;
798 }
799
800 return rc;
801error:
802 if (choice != NULL)
803 nchoice_destroy(choice);
804 return rc;
805}
806
807static errno_t fdsk_delete_part(fdisk_dev_t *dev)
808{
809 nchoice_t *choice = NULL;
810 bool confirm;
811 void *sel;
812 errno_t rc;
813
814 rc = nchoice_create(&choice);
815 if (rc != EOK) {
816 assert(rc == ENOMEM);
817 printf("Out of memory.\n");
818 goto error;
819 }
820
821 rc = nchoice_set_prompt(choice, "Select partition to delete");
822 if (rc != EOK) {
823 assert(rc == ENOMEM);
824 printf("Out of memory.\n");
825 goto error;
826 }
827
828 rc = fdsk_add_part_choices(dev, choice);
829 if (rc != EOK)
830 goto error;
831
[21f1543]832 rc = nchoice_add(choice, "Cancel", NULL, 0);
833 if (rc != EOK) {
834 assert(rc == ENOMEM);
835 printf("Out of memory.\n");
836 goto error;
837 }
838
[8227d63]839 rc = nchoice_get(choice, &sel);
[ef9dac04]840 if (rc == ENOENT)
841 return EOK;
[8227d63]842 if (rc != EOK) {
843 printf("Error getting user selection.\n");
844 goto error;
845 }
846
847
[e96047c]848 nchoice_destroy(choice);
[ef9dac04]849 choice = NULL;
850
851 if (sel == NULL)
852 return EOK;
853
854 rc = fdsk_confirm("Warning. Any data in partition will be lost. "
855 "Really delete partition?", &confirm);
856 if (rc != EOK) {
857 printf("Error getting user confirmation.\n");
858 goto error;
859 }
860
861 if (!confirm)
862 return EOK;
863
864 rc = fdisk_part_destroy((fdisk_part_t *)sel);
865 if (rc != EOK) {
866 printf("Error deleting partition.\n");
867 return rc;
868 }
869
[e96047c]870 return EOK;
871error:
872 if (choice != NULL)
873 nchoice_destroy(choice);
874 return rc;
875}
876
877/** Device menu */
[b7fd2a0]878static errno_t fdsk_dev_menu(fdisk_dev_t *dev)
[e96047c]879{
880 nchoice_t *choice = NULL;
881 fdisk_label_info_t linfo;
[8227d63]882 fdisk_part_t *part;
883 fdisk_part_info_t pinfo;
[6c4eedf]884 cap_spec_t cap;
885 cap_spec_t mcap;
[ea0ff6b]886 fdisk_dev_flags_t dflags;
[e96047c]887 char *sltype = NULL;
[8227d63]888 char *sdcap = NULL;
889 char *scap = NULL;
[852664b9]890 char *smcap = NULL;
[8227d63]891 char *sfstype = NULL;
892 char *svcname = NULL;
[b7a4d06]893 char *spkind;
[d858a660]894 const char *label;
[b7fd2a0]895 errno_t rc;
[8227d63]896 int npart;
[e96047c]897 void *sel;
898
899 rc = nchoice_create(&choice);
900 if (rc != EOK) {
901 assert(rc == ENOMEM);
902 printf("Out of memory.\n");
903 goto error;
904 }
905
906 rc = nchoice_set_prompt(choice, "Select action");
907 if (rc != EOK) {
908 assert(rc == ENOMEM);
909 printf("Out of memory.\n");
910 goto error;
911 }
912
[8227d63]913 rc = fdisk_dev_capacity(dev, &cap);
914 if (rc != EOK) {
915 printf("Error getting device capacity.\n");
916 goto error;
917 }
918
[6c4eedf]919 cap_simplify(&cap);
[9854a8f]920
[6c4eedf]921 rc = cap_format(&cap, &sdcap);
[8227d63]922 if (rc != EOK) {
923 printf("Out of memory.\n");
924 goto error;
925 }
926
927 rc = fdisk_dev_get_svcname(dev, &svcname);
928 if (rc != EOK) {
929 printf("Error getting device service name.\n");
930 goto error;
931 }
932
[ea0ff6b]933 fdisk_dev_get_flags(dev, &dflags);
934
[d858a660]935 printf("Device: %s (%s)\n", svcname, sdcap);
[22fb7ab]936 free(sdcap);
937 sdcap = NULL;
938
[e96047c]939 rc = fdisk_label_get_info(dev, &linfo);
940 if (rc != EOK) {
941 printf("Error getting label information.\n");
942 goto error;
943 }
944
[0ecfc62]945 switch (linfo.ltype) {
946 case lt_none:
947 printf("Disk contains no label.\n");
[22fb7ab]948 break;
[0ecfc62]949 default:
[db9c889]950 rc = label_type_format(linfo.ltype, &sltype);
[22fb7ab]951 if (rc != EOK) {
952 assert(rc == ENOMEM);
953 printf("Out of memory.\n");
954 goto error;
955 }
[e96047c]956
[22fb7ab]957 printf("Label type: %s\n", sltype);
958 free(sltype);
959 sltype = NULL;
960 break;
961 }
[8227d63]962
963 part = fdisk_part_first(dev);
964 npart = 0;
965 while (part != NULL) {
966 ++npart;
967 rc = fdisk_part_get_info(part, &pinfo);
968 if (rc != EOK) {
969 printf("Error getting partition information.\n");
970 goto error;
971 }
972
[6c4eedf]973 cap_simplify(&pinfo.capacity);
[9854a8f]974
[6c4eedf]975 rc = cap_format(&pinfo.capacity, &scap);
[8227d63]976 if (rc != EOK) {
977 printf("Out of memory.\n");
978 goto error;
979 }
980
[db9c889]981 rc = vol_pcnt_fs_format(pinfo.pcnt, pinfo.fstype, &sfstype);
[8227d63]982 if (rc != EOK) {
983 printf("Out of memory.\n");
984 goto error;
985 }
986
[d858a660]987 if (str_size(pinfo.label) > 0)
988 label = pinfo.label;
989 else
[9c2c7d2]990 label = "(No name)";
[d858a660]991
[edebb4a1]992 if (linfo.ltype == lt_none)
[d858a660]993 printf("Entire disk: %s %s", label, scap);
[edebb4a1]994 else
[d858a660]995 printf("Partition %d: %s %s", npart, label, scap);
[edebb4a1]996
[b7a4d06]997 if ((linfo.flags & lf_ext_supp) != 0) {
[db9c889]998 rc = label_pkind_format(pinfo.pkind, &spkind);
[b7a4d06]999 if (rc != EOK) {
1000 printf("\nOut of memory.\n");
1001 goto error;
1002 }
1003
1004 printf(", %s", spkind);
1005 free(spkind);
1006 }
1007
[4b6635a7]1008 if (pinfo.pkind != lpk_extended) {
[2dab624]1009 printf(", %s", sfstype);
[4b6635a7]1010 }
1011
[b7a4d06]1012 printf("\n");
1013
[8227d63]1014 free(scap);
1015 scap = NULL;
1016 free(sfstype);
1017 sfstype = NULL;
1018
1019 part = fdisk_part_next(part);
1020 }
[e96047c]1021
[852664b9]1022 /* Display available space */
1023 if ((linfo.flags & lf_can_create_pri) != 0) {
1024 rc = fdisk_part_get_max_avail(dev, spc_pri, &mcap);
1025 if (rc != EOK) {
1026 rc = EIO;
1027 goto error;
1028 }
1029
[6c4eedf]1030 cap_simplify(&mcap);
[9854a8f]1031
[6c4eedf]1032 rc = cap_format(&mcap, &smcap);
[852664b9]1033 if (rc != EOK) {
1034 rc = ENOMEM;
1035 goto error;
1036 }
1037
1038 if ((linfo.flags & lf_ext_supp) != 0)
1039 printf("Maximum free primary block: %s\n", smcap);
1040 else
1041 printf("Maximum free block: %s\n", smcap);
1042
1043 free(smcap);
1044 smcap = NULL;
1045
1046 rc = fdisk_part_get_tot_avail(dev, spc_pri, &mcap);
1047 if (rc != EOK) {
1048 rc = EIO;
1049 goto error;
1050 }
1051
[6c4eedf]1052 cap_simplify(&mcap);
[9854a8f]1053
[6c4eedf]1054 rc = cap_format(&mcap, &smcap);
[852664b9]1055 if (rc != EOK) {
1056 rc = ENOMEM;
1057 goto error;
1058 }
1059
1060 if ((linfo.flags & lf_ext_supp) != 0)
1061 printf("Total free primary space: %s\n", smcap);
1062 else
1063 printf("Total free space: %s\n", smcap);
1064
1065 free(smcap);
1066 smcap = NULL;
1067 }
1068
1069 /* Display available space */
1070 if ((linfo.flags & lf_can_create_log) != 0) {
1071 rc = fdisk_part_get_max_avail(dev, spc_log, &mcap);
1072 if (rc != EOK) {
1073 rc = EIO;
1074 goto error;
1075 }
1076
[6c4eedf]1077 cap_simplify(&mcap);
[9854a8f]1078
[6c4eedf]1079 rc = cap_format(&mcap, &smcap);
[852664b9]1080 if (rc != EOK) {
1081 rc = ENOMEM;
1082 goto error;
1083 }
1084
1085 printf("Maximum free logical block: %s\n", smcap);
1086 free(smcap);
1087 smcap = NULL;
1088
1089 rc = fdisk_part_get_tot_avail(dev, spc_log, &mcap);
1090 if (rc != EOK) {
1091 rc = EIO;
1092 goto error;
1093 }
1094
[6c4eedf]1095 cap_simplify(&mcap);
[9854a8f]1096
[6c4eedf]1097 rc = cap_format(&mcap, &smcap);
[852664b9]1098 if (rc != EOK) {
1099 rc = ENOMEM;
1100 goto error;
1101 }
1102
1103 printf("Total free logical space: %s\n", smcap);
1104 free(smcap);
1105 smcap = NULL;
1106 }
1107
[e96047c]1108 rc = nchoice_set_prompt(choice, "Select action");
1109 if (rc != EOK) {
1110 assert(rc == ENOMEM);
1111 printf("Out of memory.\n");
1112 goto error;
1113 }
1114
[0ecfc62]1115 if ((linfo.flags & lf_ext_supp) != 0) {
1116 if ((linfo.flags & lf_can_create_pri) != 0) {
1117 rc = nchoice_add(choice, "Create primary "
1118 "partition",
[68b5dd11]1119 (void *)devac_create_pri_part, 0);
[0ecfc62]1120 if (rc != EOK) {
1121 assert(rc == ENOMEM);
1122 printf("Out of memory.\n");
1123 goto error;
[b7a4d06]1124 }
[0ecfc62]1125 }
[b7a4d06]1126
[0ecfc62]1127 if ((linfo.flags & lf_can_create_ext) != 0) {
1128 rc = nchoice_add(choice, "Create extended "
1129 "partition",
[68b5dd11]1130 (void *)devac_create_ext_part, 0);
[0ecfc62]1131 if (rc != EOK) {
1132 assert(rc == ENOMEM);
1133 printf("Out of memory.\n");
1134 goto error;
[b7a4d06]1135 }
[0ecfc62]1136 }
[b7a4d06]1137
[0ecfc62]1138 if ((linfo.flags & lf_can_create_log) != 0) {
1139 rc = nchoice_add(choice, "Create logical "
1140 "partition",
[68b5dd11]1141 (void *)devac_create_log_part, 0);
[0ecfc62]1142 if (rc != EOK) {
1143 assert(rc == ENOMEM);
1144 printf("Out of memory.\n");
1145 goto error;
[b7a4d06]1146 }
[0ecfc62]1147 }
1148 } else { /* (linfo.flags & lf_ext_supp) == 0 */
1149 if ((linfo.flags & lf_can_create_pri) != 0) {
1150 rc = nchoice_add(choice, "Create partition",
[68b5dd11]1151 (void *)devac_create_pri_part, 0);
[0ecfc62]1152 if (rc != EOK) {
1153 assert(rc == ENOMEM);
[1433ecda]1154 printf("Out of memory.\n");
1155 goto error;
1156 }
[8227d63]1157 }
1158 }
1159
[2d78d88]1160 if ((linfo.flags & lf_can_modify_part) != 0) {
1161 rc = nchoice_add(choice, "Modify partition",
1162 (void *)devac_modify_part, 0);
1163 if (rc != EOK) {
1164 assert(rc == ENOMEM);
1165 printf("Out of memory.\n");
1166 goto error;
1167 }
1168 }
1169
[edebb4a1]1170 if ((linfo.flags & lf_can_delete_part) != 0) {
[8227d63]1171 rc = nchoice_add(choice, "Delete partition",
[68b5dd11]1172 (void *)devac_delete_part, 0);
[8227d63]1173 if (rc != EOK) {
1174 assert(rc == ENOMEM);
1175 printf("Out of memory.\n");
1176 goto error;
1177 }
1178 }
1179
[ea0ff6b]1180 if ((dflags & fdf_can_create_label) != 0) {
[e96047c]1181 rc = nchoice_add(choice, "Create label",
[68b5dd11]1182 (void *)devac_create_label, 0);
[e96047c]1183 if (rc != EOK) {
1184 assert(rc == ENOMEM);
1185 printf("Out of memory.\n");
1186 goto error;
1187 }
[ea0ff6b]1188 }
1189
1190 if ((dflags & fdf_can_delete_label) != 0) {
[e96047c]1191 rc = nchoice_add(choice, "Delete label",
[68b5dd11]1192 (void *)devac_delete_label, 0);
[e96047c]1193 if (rc != EOK) {
1194 assert(rc == ENOMEM);
1195 printf("Out of memory.\n");
1196 goto error;
1197 }
1198 }
1199
[ea0ff6b]1200 if ((dflags & fdf_can_erase_dev) != 0) {
1201 rc = nchoice_add(choice, "Erase disk",
[68b5dd11]1202 (void *)devac_erase_disk, 0);
[ea0ff6b]1203 if (rc != EOK) {
1204 assert(rc == ENOMEM);
1205 printf("Out of memory.\n");
1206 goto error;
1207 }
1208 }
1209
[68b5dd11]1210 rc = nchoice_add(choice, "Exit", (void *)devac_exit, 0);
[e96047c]1211 if (rc != EOK) {
1212 assert(rc == ENOMEM);
1213 printf("Out of memory.\n");
1214 goto error;
1215 }
1216
1217 rc = nchoice_get(choice, &sel);
1218 if (rc != EOK) {
1219 printf("Error getting user selection.\n");
1220 return rc;
1221 }
1222
1223 switch ((devac_t)sel) {
1224 case devac_create_label:
[22fb7ab]1225 (void) fdsk_create_label(dev);
[e96047c]1226 break;
1227 case devac_delete_label:
[22fb7ab]1228 (void) fdsk_delete_label(dev);
[e96047c]1229 break;
[ea0ff6b]1230 case devac_erase_disk:
1231 (void) fdsk_erase_disk(dev);
1232 break;
[b7a4d06]1233 case devac_create_pri_part:
1234 (void) fdsk_create_part(dev, lpk_primary);
1235 break;
1236 case devac_create_ext_part:
1237 (void) fdsk_create_part(dev, lpk_extended);
1238 break;
1239 case devac_create_log_part:
1240 (void) fdsk_create_part(dev, lpk_logical);
[e96047c]1241 break;
[2d78d88]1242 case devac_modify_part:
1243 (void) fdsk_modify_part(dev);
1244 break;
[e96047c]1245 case devac_delete_part:
[22fb7ab]1246 (void) fdsk_delete_part(dev);
[e96047c]1247 break;
1248 case devac_exit:
1249 quit = true;
1250 break;
1251 }
1252
1253 nchoice_destroy(choice);
1254 return EOK;
1255error:
[8227d63]1256 free(sdcap);
1257 free(scap);
[852664b9]1258 free(smcap);
[8227d63]1259 free(sfstype);
1260 free(svcname);
[e96047c]1261 if (choice != NULL)
1262 nchoice_destroy(choice);
1263 return rc;
1264}
1265
1266int main(int argc, char *argv[])
1267{
1268 service_id_t svcid;
1269 fdisk_dev_t *dev;
[b7fd2a0]1270 errno_t rc;
[e96047c]1271
[22fb7ab]1272 rc = fdisk_create(&fdisk);
1273 if (rc != EOK) {
1274 printf("Error initializing Fdisk.\n");
1275 return 1;
1276 }
1277
[e96047c]1278 rc = fdsk_dev_sel_choice(&svcid);
1279 if (rc != EOK)
1280 return 1;
1281
[8227d63]1282 if (svcid == 0)
1283 return 0;
1284
[22fb7ab]1285 rc = fdisk_dev_open(fdisk, svcid, &dev);
[e96047c]1286 if (rc != EOK) {
1287 printf("Error opening device.\n");
1288 return 1;
1289 }
1290
1291 while (!quit) {
1292 rc = fdsk_dev_menu(dev);
1293 if (rc != EOK) {
1294 fdisk_dev_close(dev);
1295 return 1;
1296 }
1297 }
1298
1299 fdisk_dev_close(dev);
[22fb7ab]1300 fdisk_destroy(fdisk);
[e96047c]1301
1302 return 0;
1303}
1304
1305
1306/** @}
1307 */
Note: See TracBrowser for help on using the repository browser.