1 /*
2 * Copyright (C) 1992 A. V. Le Blanc (LeBlanc@mcc.ac.uk)
3 * Copyright (C) 2012 Davidlohr Bueso <dave@gnu.org>
4 *
5 * Copyright (C) 2007-2013 Karel Zak <kzak@redhat.com>
6 *
7 * This program is free software. You can redistribute it and/or
8 * modify it under the terms of the GNU General Public License as
9 * published by the Free Software Foundation: either version 1 or
10 * (at your option) any later version.
11 */
12 #include <unistd.h>
13 #include <stdio.h>
14 #include <stdlib.h>
15 #include <string.h>
16 #include <fcntl.h>
17 #include <ctype.h>
18 #include <errno.h>
19 #include <getopt.h>
20 #include <sys/stat.h>
21 #include <sys/time.h>
22 #include <time.h>
23 #include <limits.h>
24 #include <signal.h>
25 #include <poll.h>
26 #include <libsmartcols.h>
27 #ifdef HAVE_LIBREADLINE
28 # define _FUNCTION_DEF
29 # include <readline/readline.h>
30 #endif
31
32 #include "c.h"
33 #include "xalloc.h"
34 #include "all-io.h"
35 #include "nls.h"
36 #include "rpmatch.h"
37 #include "blkdev.h"
38 #include "mbsalign.h"
39 #include "pathnames.h"
40 #include "canonicalize.h"
41 #include "strutils.h"
42 #include "closestream.h"
43 #include "pager.h"
44
45 #include "fdisk.h"
46
47 #include "pt-sun.h" /* to toggle flags */
48
49 #ifdef HAVE_LINUX_COMPILER_H
50 # include <linux/compiler.h>
51 #endif
52 #ifdef HAVE_LINUX_BLKPG_H
53 # include <linux/blkpg.h>
54 #endif
55
56 int pwipemode = WIPEMODE_AUTO;
57 int device_is_used;
58 int is_interactive;
59 struct fdisk_table *original_layout;
60
61 static int wipemode = WIPEMODE_AUTO;
62
63 /*
64 * fdisk debug stuff (see fdisk.h and include/debug.h)
65 */
66 UL_DEBUG_DEFINE_MASK(fdisk);
67 UL_DEBUG_DEFINE_MASKNAMES(fdisk) = UL_DEBUG_EMPTY_MASKNAMES;
68
69 static void fdiskprog_init_debug(void)
70 {
71 __UL_INIT_DEBUG_FROM_ENV(fdisk, FDISKPROG_DEBUG_, 0, FDISK_DEBUG);
72 }
73
74 static void reply_sighandler(int sig __attribute__((unused)))
75 {
76 DBG(ASK, ul_debug("got signal"));
77 }
78
79 static int reply_running;
80
81 #ifdef HAVE_LIBREADLINE
82 static char *reply_line;
83
84 static void reply_linehandler(char *line)
85 {
86 reply_line = line;
87 reply_running = 0;
88 rl_callback_handler_remove(); /* avoid duplicate prompt */
89 }
90 #endif
91
92 int get_user_reply(const char *prompt, char *buf, size_t bufsz)
93 {
94 struct sigaction oldact, act = {
95 .sa_handler = reply_sighandler
96 };
97 struct pollfd fds[] = {
98 { .fd = fileno(stdin), .events = POLLIN }
99 };
100 size_t sz;
101 int ret = 0;
102
103 DBG(ASK, ul_debug("asking for user reply %s", is_interactive ? "[interactive]" : ""));
104
105 sigemptyset(&act.sa_mask);
106 sigaction(SIGINT, &act, &oldact);
107
108 #ifdef HAVE_LIBREADLINE
109 if (is_interactive)
110 rl_callback_handler_install(prompt, reply_linehandler);
111 #endif
112 errno = 0;
113 reply_running = 1;
114 do {
115 int rc;
116
117 *buf = '\0';
118 #ifdef HAVE_LIBREADLINE
119 if (!is_interactive)
120 #endif
121 {
122 fputs(prompt, stdout);
123 fflush(stdout);
124 }
125
126 rc = poll(fds, 1, -1);
127 if (rc == -1 && errno == EINTR) { /* interrupted by signal */
128 DBG(ASK, ul_debug("cancel by CTRL+C"));
129 ret = -ECANCELED;
130 goto done;
131 }
132 if (rc == -1 && errno != EAGAIN) { /* error */
133 ret = -errno;
134 goto done;
135 }
136 #ifdef HAVE_LIBREADLINE
137 if (is_interactive) {
138 /* read input and copy to buf[] */
139 rl_callback_read_char();
140 if (!reply_running && reply_line) {
141 sz = strlen(reply_line);
142 if (sz == 0)
143 buf[sz++] = '\n';
144 else
145 memcpy(buf, reply_line, min(sz, bufsz));
146 buf[min(sz, bufsz - 1)] = '\0';
147 free(reply_line);
148 reply_line = NULL;
149 }
150 } else
151 #endif
152 {
153 if (!fgets(buf, bufsz, stdin))
154 *buf = '\0';
155 break;
156 }
157 } while (reply_running);
158
159 if (!*buf) {
160 DBG(ASK, ul_debug("cancel by CTRL+D"));
161 ret = -ECANCELED;
162 clearerr(stdin);
163 goto done;
164 }
165
166 /*
167 * cleanup the reply
168 */
169 sz = ltrim_whitespace((unsigned char *) buf);
170 if (sz && *(buf + sz - 1) == '\n')
171 *(buf + sz - 1) = '\0';
172
173 done:
174 #ifdef HAVE_LIBREADLINE
175 if (is_interactive)
176 rl_callback_handler_remove();
177 #endif
178 sigaction(SIGINT, &oldact, NULL);
179 DBG(ASK, ul_debug("user's reply: >>>%s<<< [rc=%d]", buf, ret));
180 return ret;
181 }
182
183 static int ask_menu(struct fdisk_context *cxt, struct fdisk_ask *ask,
184 char *buf, size_t bufsz)
185
186 {
187 const char *q = fdisk_ask_get_query(ask);
188 int dft = fdisk_ask_menu_get_default(ask);
189
190 if (q) {
191 fputs(q, stdout); /* print header */
192 fputc('\n', stdout);
193 }
194
195 do {
196 char prompt[128];
197 int key, c, rc;
198 const char *name, *desc;
199 size_t i = 0;
200
201 /* print menu items */
202 while (fdisk_ask_menu_get_item(ask, i++, &key, &name, &desc) == 0)
203 fprintf(stdout, " %c %s (%s)\n", key, name, desc);
204
205 /* ask for key */
206 snprintf(prompt, sizeof(prompt), _("Select (default %c): "), dft);
207 rc = get_user_reply(prompt, buf, bufsz);
208 if (rc)
209 return rc;
210 if (!*buf) {
211 fdisk_info(cxt, _("Using default response %c."), dft);
212 c = dft;
213 } else
214 c = tolower(buf[0]);
215
216 /* check result */
217 i = 0;
218 while (fdisk_ask_menu_get_item(ask, i++, &key, NULL, NULL) == 0) {
219 if (c == key) {
220 fdisk_ask_menu_set_result(ask, c);
221 return 0; /* success */
222 }
223 }
224 fdisk_warnx(cxt, _("Value out of range."));
225 } while (1);
226
227 return -EINVAL;
228 }
229
230
231 #define tochar(num) ((int) ('a' + num - 1))
232 static int ask_number(struct fdisk_context *cxt,
233 struct fdisk_ask *ask,
234 char *buf, size_t bufsz)
235 {
236 char prompt[128] = { '\0' };
237 const char *q = fdisk_ask_get_query(ask);
238 const char *range = fdisk_ask_number_get_range(ask);
239
240 uint64_t dflt = fdisk_ask_number_get_default(ask),
241 low = fdisk_ask_number_get_low(ask),
242 high = fdisk_ask_number_get_high(ask);
243 int inchar = fdisk_ask_number_inchars(ask);
244
245 assert(q);
246
247 DBG(ASK, ul_debug("asking for number "
248 "['%s', <%"PRIu64",%"PRIu64">, default=%"PRIu64", range: %s]",
249 q, low, high, dflt, range));
250
251 if (range && dflt >= low && dflt <= high) {
252 if (inchar)
253 snprintf(prompt, sizeof(prompt), _("%s (%s, default %c): "),
254 q, range, tochar(dflt));
255 else
256 snprintf(prompt, sizeof(prompt), _("%s (%s, default %"PRIu64"): "),
257 q, range, dflt);
258
259 } else if (dflt >= low && dflt <= high) {
260 if (inchar)
261 snprintf(prompt, sizeof(prompt), _("%s (%c-%c, default %c): "),
262 q, tochar(low), tochar(high), tochar(dflt));
263 else
264 snprintf(prompt, sizeof(prompt),
265 _("%s (%"PRIu64"-%"PRIu64", default %"PRIu64"): "),
266 q, low, high, dflt);
267 } else if (inchar)
268 snprintf(prompt, sizeof(prompt), _("%s (%c-%c): "),
269 q, tochar(low), tochar(high));
270 else
271 snprintf(prompt, sizeof(prompt), _("%s (%"PRIu64"-%"PRIu64"): "),
272 q, low, high);
273
274 do {
275 int rc = get_user_reply(prompt, buf, bufsz);
276 uint64_t num = 0;
277
278 if (rc)
279 return rc;
280 if (!*buf && dflt >= low && dflt <= high)
281 return fdisk_ask_number_set_result(ask, dflt);
282
283 if (isdigit_string(buf)) {
284 char *end;
285
286 errno = 0;
287 num = strtoumax(buf, &end, 10);
288 if (errno || buf == end || (end && *end))
289 continue;
290 } else if (inchar && isalpha(*buf)) {
291 num = tolower(*buf) - 'a' + 1;
292 } else
293 rc = -EINVAL;
294
295 if (rc == 0 && num >= low && num <= high)
296 return fdisk_ask_number_set_result(ask, num);
297
298 fdisk_warnx(cxt, _("Value out of range."));
299 } while (1);
300
301 return -1;
302 }
303
304 static int ask_offset(struct fdisk_context *cxt,
305 struct fdisk_ask *ask,
306 char *buf, size_t bufsz)
307 {
308 char prompt[128] = { '\0' };
309 const char *q = fdisk_ask_get_query(ask);
310 const char *range = fdisk_ask_number_get_range(ask);
311
312 uint64_t dflt = fdisk_ask_number_get_default(ask),
313 low = fdisk_ask_number_get_low(ask),
314 high = fdisk_ask_number_get_high(ask),
315 base = fdisk_ask_number_get_base(ask);
316
317 assert(q);
318
319 DBG(ASK, ul_debug("asking for offset ['%s', <%"PRIu64",%"PRIu64">, base=%"PRIu64", default=%"PRIu64", range: %s]",
320 q, low, high, base, dflt, range));
321
322 if (range && dflt >= low && dflt <= high)
323 snprintf(prompt, sizeof(prompt), _("%s (%s, default %"PRIu64"): "),
324 q, range, dflt);
325 else if (dflt >= low && dflt <= high)
326 snprintf(prompt, sizeof(prompt),
327 _("%s (%"PRIu64"-%"PRIu64", default %"PRIu64"): "),
328 q, low, high, dflt);
329 else
330 snprintf(prompt, sizeof(prompt), _("%s (%"PRIu64"-%"PRIu64"): "),
331 q, low, high);
332
333 do {
334 uintmax_t num = 0;
335 char sig = 0, *p;
336 int pwr = 0;
337
338 int rc = get_user_reply(prompt, buf, bufsz);
339 if (rc)
340 return rc;
341 if (!*buf && dflt >= low && dflt <= high)
342 return fdisk_ask_number_set_result(ask, dflt);
343
344 p = buf;
345 if (*p == '+' || *p == '-') {
346 sig = *buf;
347 p++;
348 }
349
350 rc = parse_size(p, &num, &pwr);
351 if (rc)
352 continue;
353 DBG(ASK, ul_debug("parsed size: %ju", num));
354 if (sig && pwr) {
355 /* +{size}{K,M,...} specified, the "num" is in bytes */
356 uint64_t unit = fdisk_ask_number_get_unit(ask);
357 num += unit/2; /* round */
358 num /= unit;
359 }
360 if (sig == '+')
361 num += base;
362 else if (sig == '-' && fdisk_ask_number_is_wrap_negative(ask))
363 num = high - num;
364 else if (sig == '-')
365 num = base - num;
366
367 DBG(ASK, ul_debug("final offset: %ju [sig: %c, power: %d, %s]",
368 num, sig, pwr,
369 sig ? "relative" : "absolute"));
370 if (num >= low && num <= high) {
371 if (sig && pwr)
372 fdisk_ask_number_set_relative(ask, 1);
373 return fdisk_ask_number_set_result(ask, (uint64_t)num);
374 }
375 fdisk_warnx(cxt, _("Value out of range."));
376 } while (1);
377
378 return -1;
379 }
380
381 static unsigned int info_count;
382
383 static void fputs_info(struct fdisk_ask *ask, FILE *out)
384 {
385 const char *msg;
386 assert(ask);
387
388 msg = fdisk_ask_print_get_mesg(ask);
389 if (!msg)
390 return;
391 if (info_count == 1)
392 fputc('\n', out);
393
394 fputs(msg, out);
395 fputc('\n', out);
396 }
397
398 int ask_callback(struct fdisk_context *cxt, struct fdisk_ask *ask,
399 void *data __attribute__((__unused__)))
400 {
401 int rc = 0;
402 char buf[BUFSIZ] = { '\0' };
403
404 assert(cxt);
405 assert(ask);
406
407 if (fdisk_ask_get_type(ask) != FDISK_ASKTYPE_INFO)
408 info_count = 0;
409
410 switch(fdisk_ask_get_type(ask)) {
411 case FDISK_ASKTYPE_MENU:
412 return ask_menu(cxt, ask, buf, sizeof(buf));
413 case FDISK_ASKTYPE_NUMBER:
414 return ask_number(cxt, ask, buf, sizeof(buf));
415 case FDISK_ASKTYPE_OFFSET:
416 return ask_offset(cxt, ask, buf, sizeof(buf));
417 case FDISK_ASKTYPE_INFO:
418 if (!fdisk_is_listonly(cxt))
419 info_count++;
420 fputs_info(ask, stdout);
421 break;
422 case FDISK_ASKTYPE_WARNX:
423 fflush(stdout);
424 color_scheme_fenable("warn", UL_COLOR_RED, stderr);
425 fputs(fdisk_ask_print_get_mesg(ask), stderr);
426 color_fdisable(stderr);
427 fputc('\n', stderr);
428 break;
429 case FDISK_ASKTYPE_WARN:
430 fflush(stdout);
431 color_scheme_fenable("warn", UL_COLOR_RED, stderr);
432 fputs(fdisk_ask_print_get_mesg(ask), stderr);
433 errno = fdisk_ask_print_get_errno(ask);
434 fprintf(stderr, ": %m\n");
435 color_fdisable(stderr);
436 break;
437 case FDISK_ASKTYPE_YESNO:
438 fputc('\n', stdout);
439 do {
440 int x;
441 fputs(fdisk_ask_get_query(ask), stdout);
442 rc = get_user_reply(_(" [Y]es/[N]o: "), buf, sizeof(buf));
443 if (rc)
444 break;
445 x = rpmatch(buf);
446 if (x == RPMATCH_YES || x == RPMATCH_NO) {
447 fdisk_ask_yesno_set_result(ask, x);
448 break;
449 }
450 } while(1);
451 DBG(ASK, ul_debug("yes-no ask: reply '%s' [rc=%d]", buf, rc));
452 break;
453 case FDISK_ASKTYPE_STRING:
454 {
455 char prmt[BUFSIZ];
456 snprintf(prmt, sizeof(prmt), "%s: ", fdisk_ask_get_query(ask));
457 fputc('\n', stdout);
458 rc = get_user_reply(prmt, buf, sizeof(buf));
459 if (rc == 0)
460 fdisk_ask_string_set_result(ask, xstrdup(buf));
461 DBG(ASK, ul_debug("string ask: reply '%s' [rc=%d]", buf, rc));
462 break;
463 }
464 default:
465 warnx(_("internal error: unsupported dialog type %d"), fdisk_ask_get_type(ask));
466 return -EINVAL;
467 }
468 return rc;
469 }
470
471 static struct fdisk_parttype *ask_partition_type(struct fdisk_context *cxt, int *canceled)
472 {
473 const char *q;
474 struct fdisk_label *lb;
475
476 assert(cxt);
477 lb = fdisk_get_label(cxt, NULL);
478
479 if (!lb)
480 return NULL;
481
482 *canceled = 0;
483
484 if (fdisk_label_has_parttypes_shortcuts(lb))
485 q = fdisk_label_has_code_parttypes(lb) ?
486 _("Hex code or alias (type L to list all): ") :
487 _("Partition type or alias (type L to list all): ");
488 else
489 q = fdisk_label_has_code_parttypes(lb) ?
490 _("Hex code (type L to list all codes): ") :
491 _("Partition type (type L to list all types): ");
492 do {
493 char buf[256] = { '\0' };
494 int rc = get_user_reply(q, buf, sizeof(buf));
495
496 if (rc) {
497 if (rc == -ECANCELED)
498 *canceled = 1;
499 break;
500 }
501
502 if (buf[1] == '\0' && toupper(*buf) == 'L')
503 list_partition_types(cxt);
504 else if (*buf) {
505 struct fdisk_parttype *t = fdisk_label_advparse_parttype(lb, buf,
506 FDISK_PARTTYPE_PARSE_DATA
507 | FDISK_PARTTYPE_PARSE_ALIAS
508 | FDISK_PARTTYPE_PARSE_NAME
509 | FDISK_PARTTYPE_PARSE_SEQNUM);
510 if (!t)
511 fdisk_info(cxt, _("Failed to parse '%s' partition type."), buf);
512 return t;
513 }
514 } while (1);
515
516 return NULL;
517 }
518
519
520 void list_partition_types(struct fdisk_context *cxt)
521 {
522 size_t ntypes = 0, next = 0;
523 struct fdisk_label *lb;
524 int pager = 0;
525
526 assert(cxt);
527 lb = fdisk_get_label(cxt, NULL);
528 if (!lb)
529 return;
530 ntypes = fdisk_label_get_nparttypes(lb);
531 if (!ntypes)
532 return;
533
534 if (fdisk_label_has_code_parttypes(lb)) {
535 /*
536 * Prints in 4 columns in format <hex> <name>
537 */
538 size_t last[4], done = 0, size;
539 int i;
540
541 size = ntypes;
542
543 for (i = 3; i >= 0; i--)
544 last[3 - i] = done += (size + i - done) / (i + 1);
545 i = done = 0;
546
547 do {
548 #define NAME_WIDTH 15
549 char name[NAME_WIDTH * MB_LEN_MAX];
550 size_t width = NAME_WIDTH;
551 const struct fdisk_parttype *t = fdisk_label_get_parttype(lb, next);
552 size_t ret;
553
554 if (fdisk_parttype_get_name(t)) {
555 printf("%s%02x ", i ? " " : "\n",
556 fdisk_parttype_get_code(t));
557 ret = mbsalign(_(fdisk_parttype_get_name(t)),
558 name, sizeof(name),
559 &width, MBS_ALIGN_LEFT, 0);
560
561 if (ret == (size_t)-1 || ret >= sizeof(name))
562 printf("%-15.15s",
563 _(fdisk_parttype_get_name(t)));
564 else
565 fputs(name, stdout);
566 }
567
568 next = last[i++] + done;
569 if (i > 3 || next >= last[i]) {
570 i = 0;
571 next = ++done;
572 }
573 } while (done < last[0]);
574
575 putchar('\n');
576 } else {
577 /*
578 * Prints 1 column in format <idx> <name> <typestr>
579 */
580 size_t i;
581
582 pager_open();
583 pager = 1;
584
585 for (i = 0; i < ntypes; i++) {
586 const struct fdisk_parttype *t = fdisk_label_get_parttype(lb, i);
587 printf("%3zu %-30s %s\n", i + 1,
588 fdisk_parttype_get_name(t),
589 fdisk_parttype_get_string(t));
590 }
591
592 }
593
594
595 /*
596 * Aliases
597 */
598 if (fdisk_label_has_parttypes_shortcuts(lb)) {
599 const char *alias = NULL, *typestr = NULL;
600 int rc = 0;
601
602 fputs(_("\nAliases:\n"), stdout);
603
604 for (next = 0; rc == 0 || rc == 2; next++) {
605 /* rc: <0 error, 0 success, 1 end, 2 deprecated */
606 rc = fdisk_label_get_parttype_shortcut(lb,
607 next, &typestr, NULL, &alias);
608 if (rc == 0)
609 printf(" %-14s - %s\n", alias, typestr);
610 }
611 }
612
613 if (pager)
614 pager_close();
615
616 }
617
618 void toggle_dos_compatibility_flag(struct fdisk_context *cxt)
619 {
620 struct fdisk_label *lb = fdisk_get_label(cxt, "dos");
621 int flag;
622
623 if (!lb)
624 return;
625
626 flag = !fdisk_dos_is_compatible(lb);
627 fdisk_info(cxt, flag ?
628 _("DOS Compatibility flag is set (DEPRECATED!)") :
629 _("DOS Compatibility flag is not set"));
630
631 fdisk_dos_enable_compatible(lb, flag);
632
633 if (fdisk_is_label(cxt, DOS))
634 fdisk_reset_alignment(cxt); /* reset the current label */
635 }
636
637 void change_partition_type(struct fdisk_context *cxt)
638 {
639 size_t i;
640 struct fdisk_parttype *t = NULL;
641 struct fdisk_partition *pa = NULL;
642 const char *old = NULL;
643 int canceled = 0;
644
645 assert(cxt);
646
647 if (fdisk_ask_partnum(cxt, &i, FALSE))
648 return;
649
650 if (fdisk_get_partition(cxt, i, &pa)) {
651 fdisk_warnx(cxt, _("Partition %zu does not exist yet!"), i + 1);
652 return;
653 }
654
655 t = (struct fdisk_parttype *) fdisk_partition_get_type(pa);
656 old = t ? fdisk_parttype_get_name(t) : _("Unknown");
657
658 do {
659 t = ask_partition_type(cxt, &canceled);
660 if (canceled)
661 break;
662 } while (!t);
663
664 if (canceled == 0 && t && fdisk_set_partition_type(cxt, i, t) == 0)
665 fdisk_info(cxt,
666 _("Changed type of partition '%s' to '%s'."),
667 old, t ? fdisk_parttype_get_name(t) : _("Unknown"));
668 else
669 fdisk_info(cxt,
670 _("Type of partition %zu is unchanged: %s."),
671 i + 1, old);
672
673 fdisk_unref_partition(pa);
674 fdisk_unref_parttype(t);
675 }
676
677 int print_partition_info(struct fdisk_context *cxt)
678 {
679 struct fdisk_partition *pa = NULL;
680 int rc = 0;
681 size_t i, nfields;
682 int *fields = NULL;
683 struct fdisk_label *lb = fdisk_get_label(cxt, NULL);
684
685 if ((rc = fdisk_ask_partnum(cxt, &i, FALSE)))
686 return rc;
687
688 if ((rc = fdisk_get_partition(cxt, i, &pa))) {
689 fdisk_warnx(cxt, _("Partition %zu does not exist yet!"), i + 1);
690 return rc;
691 }
692
693 if ((rc = fdisk_label_get_fields_ids_all(lb, cxt, &fields, &nfields)))
694 goto clean_data;
695
696 for (i = 0; i < nfields; ++i) {
697 int id = fields[i];
698 char *data = NULL;
699 const struct fdisk_field *fd = fdisk_label_get_field(lb, id);
700
701 if (!fd)
702 continue;
703
704 rc = fdisk_partition_to_string(pa, cxt, id, &data);
705 if (rc < 0)
706 goto clean_data;
707 if (!data || !*data)
708 continue;
709 fdisk_info(cxt, "%15s: %s", fdisk_field_get_name(fd), data);
710 free(data);
711 }
712
713 clean_data:
714 fdisk_unref_partition(pa);
715 free(fields);
716 return rc;
717 }
718
719 static size_t skip_empty(const unsigned char *buf, size_t i, size_t sz)
720 {
721 size_t next;
722 const unsigned char *p0 = buf + i;
723
724 for (next = i + 16; next < sz; next += 16) {
725 if (memcmp(p0, buf + next, 16) != 0)
726 break;
727 }
728
729 return next == i + 16 ? i : next;
730 }
731
732 static void dump_buffer(off_t base, unsigned char *buf, size_t sz)
733 {
734 size_t i, l, next = 0;
735
736 if (!buf)
737 return;
738 for (i = 0, l = 0; i < sz; i++, l++) {
739 if (l == 0) {
740 if (!next)
741 next = skip_empty(buf, i, sz);
742 printf("%08jx ", (intmax_t)base + i);
743 }
744 printf(" %02x", buf[i]);
745 if (l == 7) /* words separator */
746 fputs(" ", stdout);
747 else if (l == 15) {
748 fputc('\n', stdout); /* next line */
749 l = -1;
750 if (next > i) {
751 printf("*\n");
752 i = next - 1;
753 }
754 next = 0;
755 }
756 }
757 if (l > 0)
758 printf("\n");
759 }
760
761 static void dump_blkdev(struct fdisk_context *cxt, const char *name,
762 uint64_t offset, size_t size)
763 {
764 int fd = fdisk_get_devfd(cxt);
765
766 fdisk_info(cxt, _("\n%s: offset = %"PRIu64", size = %zu bytes."),
767 name, offset, size);
768
769 assert(fd >= 0);
770
771 if (lseek(fd, (off_t) offset, SEEK_SET) == (off_t) -1)
772 fdisk_warn(cxt, _("cannot seek"));
773 else {
774 unsigned char *buf = xmalloc(size);
775
776 if (read_all(fd, (char *) buf, size) != (ssize_t) size)
777 fdisk_warn(cxt, _("cannot read"));
778 else
779 dump_buffer(offset, buf, size);
780 free(buf);
781 }
782 }
783
784 void dump_firstsector(struct fdisk_context *cxt)
785 {
786 assert(cxt);
787
788 dump_blkdev(cxt, _("First sector"), 0, fdisk_get_sector_size(cxt));
789 }
790
791 void dump_disklabel(struct fdisk_context *cxt)
792 {
793 int i = 0;
794 const char *name = NULL;
795 uint64_t offset = 0;
796 size_t size = 0;
797
798 assert(cxt);
799
800 while (fdisk_locate_disklabel(cxt, i++, &name, &offset, &size) == 0 && size)
801 dump_blkdev(cxt, name, offset, size);
802 }
803
804 static fdisk_sector_t get_dev_blocks(char *dev)
805 {
806 int fd, ret;
807 fdisk_sector_t size;
808
809 if ((fd = open(dev, O_RDONLY|O_NONBLOCK)) < 0)
810 err(EXIT_FAILURE, _("cannot open %s"), dev);
811 ret = blkdev_get_sectors(fd, (unsigned long long *) &size);
812 close(fd);
813 if (ret < 0)
814 err(EXIT_FAILURE, _("BLKGETSIZE ioctl failed on %s"), dev);
815 return size/2;
816 }
817
818
819 void follow_wipe_mode(struct fdisk_context *cxt)
820 {
821 int dowipe = wipemode == WIPEMODE_ALWAYS ? 1 : 0;
822
823 if (isatty(STDIN_FILENO) && wipemode == WIPEMODE_AUTO)
824 dowipe = 1; /* do it in interactive mode */
825
826 if (fdisk_is_ptcollision(cxt) && wipemode != WIPEMODE_NEVER)
827 dowipe = 1; /* always remove old PT */
828
829 fdisk_enable_wipe(cxt, dowipe);
830 if (dowipe)
831 fdisk_warnx(cxt, _(
832 "The device contains '%s' signature and it will be removed by a write command. "
833 "See fdisk(8) man page and --wipe option for more details."),
834 fdisk_get_collision(cxt));
835 else
836 fdisk_warnx(cxt, _(
837 "The device contains '%s' signature and it may remain on the device. "
838 "It is recommended to wipe the device with wipefs(8) or "
839 "fdisk --wipe, in order to avoid possible collisions."),
840 fdisk_get_collision(cxt));
841 }
842
843 static void __attribute__((__noreturn__)) usage(void)
844 {
845 FILE *out = stdout;
846
847 fputs(USAGE_HEADER, out);
848
849 fprintf(out,
850 _(" %1$s [options] <disk> change partition table\n"
851 " %1$s [options] -l [<disk>...] list partition table(s)\n"),
852 program_invocation_short_name);
853
854 fputs(USAGE_SEPARATOR, out);
855 fputs(_("Display or manipulate a disk partition table.\n"), out);
856
857 fputs(USAGE_OPTIONS, out);
858 fputs(_(" -b, --sector-size <size> physical and logical sector size\n"), out);
859 fputs(_(" -B, --protect-boot don't erase bootbits when creating a new label\n"), out);
860 fputs(_(" -c, --compatibility[=<mode>] mode is 'dos' or 'nondos' (default)\n"), out);
861 fprintf(out,
862 _(" -L, --color[=<when>] colorize output (%s, %s or %s)\n"), "auto", "always", "never");
863 fprintf(out,
864 " %s\n", USAGE_COLORS_DEFAULT);
865 fputs(_(" -l, --list display partitions and exit\n"), out);
866 fputs(_(" -x, --list-details like --list but with more details\n"), out);
867
868 fputs(_(" -n, --noauto-pt don't create default partition table on empty devices\n"), out);
869 fputs(_(" -o, --output <list> output columns\n"), out);
870 fputs(_(" -t, --type <type> recognize specified partition table type only\n"), out);
871 fputs(_(" -u, --units[=<unit>] display units: 'cylinders' or 'sectors' (default)\n"), out);
872 fputs(_(" -s, --getsz display device size in 512-byte sectors [DEPRECATED]\n"), out);
873 fputs(_(" --bytes print SIZE in bytes rather than in human readable format\n"), out);
874 fprintf(out,
875 _(" --lock[=<mode>] use exclusive device lock (%s, %s or %s)\n"), "yes", "no", "nonblock");
876 fprintf(out,
877 _(" -w, --wipe <mode> wipe signatures (%s, %s or %s)\n"), "auto", "always", "never");
878 fprintf(out,
879 _(" -W, --wipe-partitions <mode> wipe signatures from new partitions (%s, %s or %s)\n"), "auto", "always", "never");
880
881 fputs(USAGE_SEPARATOR, out);
882 fputs(_(" -C, --cylinders <number> specify the number of cylinders\n"), out);
883 fputs(_(" -H, --heads <number> specify the number of heads\n"), out);
884 fputs(_(" -S, --sectors <number> specify the number of sectors per track\n"), out);
885
886 fputs(USAGE_SEPARATOR, out);
887 printf(USAGE_HELP_OPTIONS(31));
888
889 list_available_columns(out);
890
891 printf(USAGE_MAN_TAIL("fdisk(8)"));
892 exit(EXIT_SUCCESS);
893 }
894
895
896 enum {
897 ACT_FDISK = 0, /* default */
898 ACT_LIST,
899 ACT_LIST_DETAILS,
900 ACT_SHOWSIZE
901 };
902
903 int main(int argc, char **argv)
904 {
905 int rc, i, c, act = ACT_FDISK, noauto_pt = 0;
906 int colormode = UL_COLORMODE_UNDEF;
907 struct fdisk_context *cxt;
908 char *outarg = NULL;
909 const char *devname, *lockmode = NULL;
910 enum {
911 OPT_BYTES = CHAR_MAX + 1,
912 OPT_LOCK
913 };
914 static const struct option longopts[] = {
915 { "bytes", no_argument, NULL, OPT_BYTES },
916 { "color", optional_argument, NULL, 'L' },
917 { "compatibility", optional_argument, NULL, 'c' },
918 { "cylinders", required_argument, NULL, 'C' },
919 { "heads", required_argument, NULL, 'H' },
920 { "sectors", required_argument, NULL, 'S' },
921 { "getsz", no_argument, NULL, 's' },
922 { "help", no_argument, NULL, 'h' },
923 { "list", no_argument, NULL, 'l' },
924 { "list-details", no_argument, NULL, 'x' },
925 { "lock", optional_argument, NULL, OPT_LOCK },
926 { "noauto-pt", no_argument, NULL, 'n' },
927 { "sector-size", required_argument, NULL, 'b' },
928 { "type", required_argument, NULL, 't' },
929 { "units", optional_argument, NULL, 'u' },
930 { "version", no_argument, NULL, 'V' },
931 { "output", required_argument, NULL, 'o' },
932 { "protect-boot", no_argument, NULL, 'B' },
933 { "wipe", required_argument, NULL, 'w' },
934 { "wipe-partitions",required_argument, NULL, 'W' },
935 { NULL, 0, NULL, 0 }
936 };
937
938 setlocale(LC_ALL, "");
939 bindtextdomain(PACKAGE, LOCALEDIR);
940 textdomain(PACKAGE);
941 close_stdout_atexit();
942
943 fdisk_init_debug(0);
944 scols_init_debug(0);
945 fdiskprog_init_debug();
946
947 cxt = fdisk_new_context();
948 if (!cxt)
949 err(EXIT_FAILURE, _("failed to allocate libfdisk context"));
950
951 fdisk_set_ask(cxt, ask_callback, NULL);
952
953 while ((c = getopt_long(argc, argv, "b:Bc::C:hH:lL::no:sS:t:u::vVw:W:x",
954 longopts, NULL)) != -1) {
955 switch (c) {
956 case 'b':
957 {
958 size_t sz = strtou32_or_err(optarg,
959 _("invalid sector size argument"));
960 if (sz != 512 && sz != 1024 && sz != 2048 && sz != 4096)
961 errx(EXIT_FAILURE, _("invalid sector size argument"));
962 fdisk_save_user_sector_size(cxt, sz, sz);
963 break;
964 }
965 case 'B':
966 fdisk_enable_bootbits_protection(cxt, 1);
967 break;
968 case 'C':
969 fdisk_save_user_geometry(cxt,
970 strtou32_or_err(optarg,
971 _("invalid cylinders argument")),
972 0, 0);
973 break;
974 case 'c':
975 if (optarg) {
976 /* this setting is independent on the current
977 * actively used label
978 */
979 char *p = *optarg == '=' ? optarg + 1 : optarg;
980 struct fdisk_label *lb = fdisk_get_label(cxt, "dos");
981
982 if (!lb)
983 err(EXIT_FAILURE, _("not found DOS label driver"));
984 if (strcmp(p, "dos") == 0)
985 fdisk_dos_enable_compatible(lb, TRUE);
986 else if (strcmp(p, "nondos") == 0)
987 fdisk_dos_enable_compatible(lb, FALSE);
988 else
989 errx(EXIT_FAILURE, _("unknown compatibility mode '%s'"), p);
990 }
991 /* use default if no optarg specified */
992 break;
993 case 'H':
994 fdisk_save_user_geometry(cxt, 0,
995 strtou32_or_err(optarg,
996 _("invalid heads argument")),
997 0);
998 break;
999 case 'S':
1000 fdisk_save_user_geometry(cxt, 0, 0,
1001 strtou32_or_err(optarg,
1002 _("invalid sectors argument")));
1003 break;
1004 case 'l':
1005 act = ACT_LIST;
1006 break;
1007 case 'x':
1008 act = ACT_LIST_DETAILS;
1009 break;
1010 case 'L':
1011 colormode = UL_COLORMODE_AUTO;
1012 if (optarg)
1013 colormode = colormode_or_err(optarg,
1014 _("unsupported color mode"));
1015 break;
1016 case 'n':
1017 noauto_pt = 1;
1018 break;
1019 case 'o':
1020 outarg = optarg;
1021 break;
1022 case 's':
1023 act = ACT_SHOWSIZE;
1024 break;
1025 case 't':
1026 {
1027 struct fdisk_label *lb = NULL;
1028
1029 while (fdisk_next_label(cxt, &lb) == 0)
1030 fdisk_label_set_disabled(lb, 1);
1031
1032 lb = fdisk_get_label(cxt, optarg);
1033 if (!lb)
1034 errx(EXIT_FAILURE, _("unsupported disklabel: %s"), optarg);
1035 fdisk_label_set_disabled(lb, 0);
1036 break;
1037 }
1038 case 'u':
1039 if (optarg && *optarg == '=')
1040 optarg++;
1041 if (fdisk_set_unit(cxt, optarg) != 0)
1042 errx(EXIT_FAILURE, _("unsupported unit"));
1043 break;
1044 case 'V': /* preferred for util-linux */
1045 case 'v': /* for backward compatibility only */
1046 print_version(EXIT_SUCCESS);
1047 case 'w':
1048 wipemode = wipemode_from_string(optarg);
1049 if (wipemode < 0)
1050 errx(EXIT_FAILURE, _("unsupported wipe mode"));
1051 break;
1052 case 'W':
1053 pwipemode = wipemode_from_string(optarg);
1054 if (pwipemode < 0)
1055 errx(EXIT_FAILURE, _("unsupported wipe mode"));
1056 break;
1057 case 'h':
1058 usage();
1059 case OPT_BYTES:
1060 fdisk_set_size_unit(cxt, FDISK_SIZEUNIT_BYTES);
1061 break;
1062 case OPT_LOCK:
1063 lockmode = "1";
1064 if (optarg) {
1065 if (*optarg == '=')
1066 optarg++;
1067 lockmode = optarg;
1068 }
1069 break;
1070 default:
1071 errtryhelp(EXIT_FAILURE);
1072 }
1073 }
1074
1075 if (argc-optind != 1 && fdisk_has_user_device_properties(cxt))
1076 warnx(_("The device properties (sector size and geometry) should"
1077 " be used with one specified device only."));
1078
1079 colors_init(colormode, "fdisk");
1080 is_interactive = isatty(STDIN_FILENO);
1081
1082 switch (act) {
1083 case ACT_LIST:
1084 case ACT_LIST_DETAILS:
1085 fdisk_enable_listonly(cxt, 1);
1086
1087 if (act == ACT_LIST_DETAILS)
1088 fdisk_enable_details(cxt, 1);
1089
1090 init_fields(cxt, outarg, NULL);
1091
1092 if (argc > optind) {
1093 int k;
1094
1095 for (rc = 0, k = optind; k < argc; k++)
1096 rc += print_device_pt(cxt, argv[k], 1, 0, k != optind);
1097
1098 if (rc)
1099 return EXIT_FAILURE;
1100 } else
1101 print_all_devices_pt(cxt, 0);
1102 break;
1103
1104 case ACT_SHOWSIZE:
1105 /* deprecated */
1106 if (argc - optind <= 0) {
1107 warnx(_("bad usage"));
1108 errtryhelp(EXIT_FAILURE);
1109 }
1110 for (i = optind; i < argc; i++) {
1111 uintmax_t blks = get_dev_blocks(argv[i]);
1112
1113 if (argc - optind == 1)
1114 printf("%ju\n", blks);
1115 else
1116 printf("%s: %ju\n", argv[i], blks);
1117 }
1118 break;
1119
1120 case ACT_FDISK:
1121 if (argc-optind != 1) {
1122 warnx(_("bad usage"));
1123 errtryhelp(EXIT_FAILURE);
1124 }
1125
1126 /* Here starts interactive mode, use fdisk_{warn,info,..} functions */
1127 color_scheme_enable("welcome", UL_COLOR_GREEN);
1128 fdisk_info(cxt, _("Welcome to fdisk (%s)."), PACKAGE_STRING);
1129 color_disable();
1130 fdisk_info(cxt, _("Changes will remain in memory only, until you decide to write them.\n"
1131 "Be careful before using the write command.\n"));
1132
1133 devname = argv[optind];
1134 rc = fdisk_assign_device(cxt, devname, 0);
1135 if (rc == -EACCES) {
1136 rc = fdisk_assign_device(cxt, devname, 1);
1137 if (rc == 0)
1138 fdisk_warnx(cxt, _("Device is open in read-only mode."));
1139 }
1140 if (rc)
1141 err(EXIT_FAILURE, _("cannot open %s"), devname);
1142
1143 if (fdisk_device_is_used(cxt))
1144 fdisk_warnx(cxt, _(
1145 "This disk is currently in use - repartitioning is probably a bad idea.\n"
1146 "It's recommended to umount all file systems, and swapoff all swap\n"
1147 "partitions on this disk.\n"));
1148
1149 fflush(stdout);
1150
1151 if (!fdisk_is_readonly(cxt)
1152 && blkdev_lock(fdisk_get_devfd(cxt), devname, lockmode) != 0) {
1153 fdisk_deassign_device(cxt, 1);
1154 fdisk_unref_context(cxt);
1155 return EXIT_FAILURE;
1156 }
1157
1158 if (fdisk_get_collision(cxt))
1159 follow_wipe_mode(cxt);
1160
1161 if (!fdisk_has_label(cxt)) {
1162 fdisk_info(cxt, _("Device does not contain a recognized partition table."));
1163 if (!noauto_pt)
1164 fdisk_create_disklabel(cxt, NULL);
1165
1166 } else if (fdisk_is_label(cxt, GPT) && fdisk_gpt_is_hybrid(cxt))
1167 fdisk_warnx(cxt, _(
1168 "A hybrid GPT was detected. You have to sync "
1169 "the hybrid MBR manually (expert command 'M')."));
1170
1171 init_fields(cxt, outarg, NULL); /* -o <columns> */
1172
1173 if (!fdisk_is_readonly(cxt)) {
1174 fdisk_get_partitions(cxt, &original_layout);
1175 device_is_used = fdisk_device_is_used(cxt);
1176 }
1177
1178 while (1)
1179 process_fdisk_menu(&cxt);
1180 }
1181
1182 fdisk_unref_context(cxt);
1183 return EXIT_SUCCESS;
1184 }