source: MondoRescue/branches/2.2.8/mindi/parted2fdisk.pl@ 2789

Last change on this file since 2789 was 2789, checked in by Bruno Cornec, 13 years ago

r2159@localhost (orig r2158): bruno | 2009-03-05 20:01:54 +0100

  • Fixes for ia64 support from Giuseppe Grassi <giuseppe.grassi2_at_italtel.it>:
    • the m) fake output is expected in fdisk.log so printing that on stdout instead of parted2fdisk log file
    • parted2fdisk supports now lack of partition given in 't' command
    • parted2fdisk supports now LVM type of partitions correctly (using a parted flag)
    • init links now /dev/tty to /dev/ttyS0 to have a working serial support on ia64


  • Property svn:keywords set to Id
File size: 15.0 KB
Line 
1#!/usr/bin/perl -w
2#
3# $Id: parted2fdisk.pl 2789 2011-04-29 13:37:13Z bruno $
4#
5# parted2fdisk: fdisk like interface for parted
6# [developped for mindi/mondo http://www.mondorescue.org]
7#
8# Aims to be architecture independant (i386/ia64)
9# Tested on ia64 with RHAS 2.1 - Mandrake 9.0 - RHEL 3.0 - SLES 10
10#
11# (c) Bruno Cornec <Bruno.Cornec@hp.com>
12# Licensed under the GPL
13
14use strict;
15
16$ENV{LANG} = "C";
17$ENV{LANGUAGE} = "C";
18$ENV{LC_ALL} = "C";
19
20# Log
21my $flog = "/tmp/parted2fdisk.log";
22open(FLOG, "> $flog") || die "Unable to open $flog";
23
24my $fdisk = "/sbin/fdisk";
25my $parted = "/sbin/parted";
26
27my $i;
28my $l;
29my $part;
30my $wpart;
31my $start = "";
32my $end = "";
33my $cylstart;
34my $cylend;
35my %start;
36my %end;
37my %type;
38my $arch;
39my $fake = 0;
40my $mega = 1048576;
41
42# Determine on which arch we're running
43if (defined ($ENV{ARCH})) {
44 $arch = $ENV{ARCH};
45} else {
46 $arch = `uname -m`;
47 chomp($arch);
48}
49
50#
51# Looking for fdisk
52#
53$fdisk = is_lsb($fdisk);
54#
55# We always use fdisk except on ia64 with GPT types of
56# partition tables where we need parted
57# All should return fdisk like format so that callers
58# think they have called fdisk directly
59#
60my $un;
61my $type;
62my $args = "";
63my $device = "";
64my $endmax = "";
65
66if ($#ARGV < 0) {
67 printf FLOG "No arguments given exiting ...\n";
68 mysyn();
69}
70
71my %pid = ( "FAT" => "6",
72 "fat32" => "b",
73 "fat16" => "e",
74 "ext2" => "83",
75 "ext3" => "83",
76 "ext4" => "83",
77 "xfs" => "83",
78 "reiserfs" => "83",
79 "linux-swap" => "82",
80 "lvm" => "8e",
81 "" => "",
82 );
83my %pnum;
84
85# Reverse table of pid
86while (($i,$l) = each %pid) {
87 next if ($i eq "ext2");
88 $pnum{$l} = $i;
89}
90
91foreach $i (@ARGV) {
92 # We support at most one option and one device
93 print FLOG "Parameter found : $i\n";
94 if ($i =~ /^\/dev\//) {
95 $device = $i;
96 next;
97 } elsif ($i =~ /^-/) {
98 $args = $i;
99 next;
100 } else {
101 mysyn();
102 }
103}
104
105if (($args ne "") and ($device eq "")) {
106 mysyn();
107}
108
109# -s takes a partition as arg
110if ($args =~ /-s/) {
111 $wpart = $device;
112 $device =~ s/[0-9]+$//;
113}
114
115if ($args =~ /-n/) {
116 print FLOG "Fake mode. Nothing will be really done\n";
117 $fake = 1;
118}
119
120print FLOG "Called with device $device and arg $args\n";
121
122if ($arch =~ /^ia64/) {
123 # Check partition table type
124 print FLOG "We're on ia64 ...\n";
125 $parted = is_lsb($parted);
126 $type = which_type($device);
127 if ($type ne "msdos") {
128 print FLOG "Not an msdos type of disk label\n";
129 if ($args =~ /-l/) {
130 fdisk_list($device,undef,\%start,\%end, 1);
131 } elsif ($args =~ /-s/) {
132 fdisk_list($device,$wpart,\%start,\%end, 1);
133 } elsif (($args =~ /-/) and ($fake == 0)) {
134 printf FLOG "Option not supported ($args) ...\n";
135 printf FLOG "Please report to the author\n";
136 mysyn();
137 } else {
138 # Read fdisk orders on stdin and pass them to parted
139 # on the command line as parted doesn't read on stdin
140 print FLOG "Translating fdisk command to parted\n";
141 while ($i = <STDIN>) {
142 if ($i =~ /^p$/) {
143 fdisk_list($device,undef,\%start,\%end, 1);
144 } elsif ($i =~ /^n$/) {
145 fdisk_list($device,undef,\%start,\%end, 0);
146 if ($type ne "gpt") {
147 print FLOG "Forcing GPT type of disk label\n";
148 print FLOG "mklabel gpt\n";
149 system "$parted -s $device mklabel gpt\n" if ($fake == 0);
150 $type = "gpt";
151 }
152 $l = <STDIN>;
153 if (not (defined $l)) {
154 print FLOG "no primary/extended arg given for creation... assuming primary\n";
155 $l = "p";
156 }
157 chomp($l);
158 $part = <STDIN>;
159 if ((not (defined $part)) || ($part eq "")) {
160 print FLOG "no partition given for creation... skipping\n";
161 next;
162 }
163 chomp($part);
164 $cylstart = <STDIN>;
165 chomp($cylstart);
166 if ((not (defined $cylstart)) || ($cylstart eq "")) {
167 if (defined $start{$part-1}) {
168 # in MB => cyl
169 $cylstart = sprintf("%d",$end{$part-1}*$mega/$un + 1);
170 print FLOG "no start cyl given for creation... assuming the following $cylstart\n";
171 } else {
172 print FLOG "no start cyl given for creation... assuming the following 1\n";
173 $cylstart = 1;
174 }
175 }
176 $cylstart = 1 if ($cylstart < 1);
177 print FLOG "start cyl : $cylstart\n";
178 $un = get_un($device, "", 0);
179 # parted needs MB
180 if ($cylstart == 1) {
181 $start = 0.01;
182 } else {
183 $start = $cylstart* $un / $mega + 0.001;
184 }
185 # this is a size in B/KB/MB/GB
186
187 $endmax = get_max($device);
188 $cylend = <STDIN>;
189 chomp($cylend);
190 if ((not (defined $cylend)) || ($cylend eq "")) {
191 print FLOG "no end cyl given for creation... assuming full disk)\n";
192 $cylend = $endmax;
193 }
194 # Handles end syntaxes (+, K, M, ...)
195 # to give cylinders
196 if ($cylend =~ /^\+/) {
197 $cylend =~ s/^\+//;
198 # Handles suffixes; return bytes
199 $cylend = decode_Bsuf($cylend,1);
200 # This gives the number of cyl
201 $cylend /= $un;
202 $cylend = sprintf("%d",$cylend);
203 $cylend += $cylstart - 0.001;
204 # We now have the end cyl
205 }
206 $cylend = $endmax if ($cylend > $endmax);
207 print FLOG "end cyl : $cylend\n";
208 # parted needs MB
209 $end = $cylend * $un / $mega;
210 print FLOG "n $l $part $cylstart $cylend => mkpart primary $start $end\n";
211 system "$parted -s $device mkpart primary ext2 $start $end\n" if ($fake == 0);
212 print "command (m for help) send back to fake fdisk for mondorestore\n";
213 } elsif ($i =~ /^d$/) {
214 $part = <STDIN>;
215 if (not (defined $part)) {
216 print FLOG "no partition given for deletion... skipping\n";
217 next;
218 }
219 chomp($part);
220 print FLOG "d $part => rm $part\n";
221 system "$parted -s $device rm $part\n" if ($fake == 0);
222 get_parted($device,undef,\%start,\%end,undef);
223 print "command (m for help) send back to fake fdisk for mondorestore\n";
224 } elsif ($i =~ /^w$/) {
225 print FLOG "w => quit\n";
226 } elsif ($i =~ /^t$/) {
227 $part = <STDIN>;
228 if (not (defined $part)) {
229 print FLOG "no partition given for tagging... skipping\n";
230 next;
231 }
232 chomp($part);
233 # If no partition number given it's 1, and we received the type
234 if ($part !~ /\d+/) {
235 $l = $part;
236 $part = 1
237 } else {
238 $l = <STDIN>;
239 }
240 if (not (defined $l)) {
241 print FLOG "no type given for tagging partition $part... skipping\n";
242 next;
243 }
244 chomp($l);
245 if (not (defined $pnum{$l})) {
246 print FLOG "no partition number given for $l... please report to the author\n";
247 next;
248 }
249
250 if ($pnum{$l} eq "lvm") {
251 # In that case this is a flag set, not a mkfs
252 print FLOG "t $part $l => set $part $pnum{$l} on\n";
253 system "$parted -s $device set $part $pnum{$l} on\n" if ($fake == 0);
254 } else {
255 print FLOG "t $part $l => mkfs $part $pnum{$l}\n";
256 system "$parted -s $device mkfs $part $pnum{$l}\n" if ($fake == 0);
257 }
258 print "command (m for help) send back to fake fdisk for mondorestore\n";
259 } elsif ($i =~ /^a$/) {
260 $part = <STDIN>;
261 if (not (defined $part)) {
262 print FLOG "no partition given for tagging... skipping\n";
263 next;
264 }
265 chomp($part);
266
267 # Partition shouldn't be negative or null. Then take the first one.
268 $part = 1 if ($part le 0);
269
270 print FLOG "a $part => set $part boot on\n";
271 system "$parted -s $device set $part boot on\n" if ($fake == 0);
272 print "command (m for help) send back to fake fdisk for mondorestore\n";
273 } elsif ($i =~ /^q$/) {
274 print FLOG "q => quit\n";
275 } else {
276 print FLOG "Unknown command: $i\n";
277 next;
278 }
279
280 }
281 }
282 myexit(0);
283 }
284}
285
286#
287# Else everything is for fdisk
288#
289# Print only mode
290print FLOG "Passing everything to the real fdisk\n";
291my $fargs = join(@ARGV);
292
293if ($args =~ /^-/) {
294 # -l or -s
295 open (FDISK, "$fdisk $fargs 2>/dev/null |") || die "Unable to read from $fdisk";
296 while (<FDISK>) {
297 print;
298 }
299 close(FDISK);
300} else {
301 # Modification mode
302 open (FDISK, "| $fdisk $fargs 2>/dev/null") || die "Unable to modify through $fdisk";
303 while (<STDIN>) {
304 print FDISK;
305 }
306 close(FDISK);
307 close(STDIN);
308}
309myexit(0);
310
311
312# Is your system LSB ?
313sub is_lsb {
314
315my $cmd = shift;
316my $basename = basename($cmd);
317
318if (not (-x $cmd)) {
319 print FLOG "Your system is not LSB/mondo compliant: $basename was not found as $cmd\n";
320 print FLOG "Searching elswhere...";
321 foreach $i (split(':',$ENV{PATH})) {
322 if (-x "$i/$basename") {
323 $cmd = "$i/$basename";
324 print FLOG "Found $cmd, using it !\n";
325 last;
326 }
327 }
328 if (not (-x $cmd)) {
329 print FLOG "Your system doesn't provide $basename in the PATH\n";
330 print FLOG "Please correct it before relaunching\n";
331 myexit(-1);
332 }
333}
334return($cmd);
335}
336
337sub fdisk_list {
338
339my $device = shift;
340my $wpart = shift;
341my $start = shift;
342my $end = shift;
343my $verbose = shift;
344
345my $un;
346my $endmax;
347my $d;
348my $n;
349
350my %cmt = ( "FAT" => "FAT",
351 "ext2" => "Linux",
352 "ext3" => "Linux",
353 "ext4" => "Linux",
354 "xfs" => "Linux",
355 "reiserfs" => "Linux",
356 "linux-swap" => "Linux swap",
357 "lvm" => "Linux LVM",
358 "fat16" => "fat16",
359 "fat32" => "fat32",
360 "" => "Linux",
361);
362
363my $part;
364my $mstart;
365my $mend;
366my $length;
367my $pid;
368my $cmt;
369format FLOG1 =
370@<<<<<<<<<<<< @>>>>>>>>>> @>>>>>>>>>> @>>>>>>>>>> @>>> @<<<<<<<<<<<<<<<<<<<<<<<<<<<<
371$part, $mstart, $mend, $length, $pid, $cmt
372.
373format FLOG2 =
374@<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
375$part,
376 @>>>>>>>>>> @>>>>>>>>>> @>>>>>>>>>> @>>> @<<<<<<<<<<<<<<<<<<<<<<<<<<<<
377 $mstart, $mend, $length, $pid, $cmt
378.
379format STDOUT1 =
380@<<<<<<<<<<<< @>>>>>>>>>> @>>>>>>>>>> @>>>>>>>>>> @>>> @<<<<<<<<<<<<<<<<<<<<<<<<<<<<
381$part, $mstart, $mend, $length, $pid, $cmt
382.
383format STDOUT2 =
384@<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
385$part,
386 @>>>>>>>>>> @>>>>>>>>>> @>>>>>>>>>> @>>> @<<<<<<<<<<<<<<<<<<<<<<<<<<<<
387 $mstart, $mend, $length, $pid, $cmt
388.
389# Device Boot Start End Blocks Id System
390#/dev/hda1 1 77579 39099374+ ee EFI GPT
391
392
393#
394# Keep Fdisk headers
395#
396# this will return bytes
397$un = get_un ($device,$wpart,$verbose);
398
399$endmax = get_max($device);
400
401# This will return MB
402get_parted ($device,$start,$end,\%type);
403
404while (($n,$d) = each %type) {
405 # Print infos fdisk like
406 $part = ${device}.$n;
407 # start and end are in cylinder in fdisk format
408 # so return in MB * 1MB / what represents 1 cyl in B
409 $mstart = sprintf("%d",$$start{$n}*$mega/$un);
410 $mstart = 1 if ($mstart < 1);
411 $mstart = $endmax if ($mstart > $endmax);
412 $mend = sprintf("%d",$$end{$n}*$mega/$un - 1);
413 $mend = $endmax if ($mend > $endmax);
414 $mend = 1 if ($mend < 1);
415 # length is in 1K blocks
416 $length = sprintf("%d",($mend-$mstart+1)*$un/1024);
417 $pid = $pid{$type{$n}};
418 $cmt = $cmt{$type{$n}};
419 #print FLOG "$part - $mstart - $mend - $length\n";
420
421 if ($verbose == 1) {
422 if (not (defined $wpart)) {
423 if (length($part) > 13) {
424 open(STDOUT2,">&STDOUT") || die "Unable to open STDOUT2";
425 select(STDOUT2);
426 write;
427 open(FLOG2,">&FLOG") || die "Unable to open FLOG2";
428 select(FLOG2);
429 write;
430 select(STDOUT);
431 close(FLOG2);
432 close(STDOUT2);
433 } else {
434 open(STDOUT1,">&STDOUT") || die "Unable to open STDOUT1";
435 select(STDOUT1);
436 write;
437 open(FLOG1,">&FLOG") || die "Unable to open FLOG1";
438 select(FLOG1);
439 write;
440 select(STDOUT);
441 close(FLOG1);
442 close(STDOUT1);
443 }
444 } else {
445 # manage the -s option of fdisk here
446 print "$length\n" if ($part eq $wpart);
447 print FLOG "$part has $length KBytes\n" if ($part eq $wpart);
448 }
449 }
450}
451close(FDISK);
452close(PARTED);
453}
454
455#
456# Get max size from fdisk
457#
458sub get_max {
459
460my $device = shift;
461my $max = 0;
462my $foo;
463
464open (FDISK, "$fdisk -l $device 2>/dev/null |") || die "Unable to read from $fdisk";
465while (<FDISK>) {
466 if ($_ =~ /heads/) {
467 chomp;
468 $max = $_;
469 $max =~ s/.* ([0-9]+) cylinders/$1/;
470 }
471}
472close(FDISK);
473print FLOG "get_max returns $max\n";
474return($max);
475}
476
477#
478# Get units from fdisk (cylinder size)
479#
480sub get_un {
481
482my $device = shift;
483my $wpart = shift;
484my $verbose = shift;
485my $un = 0;
486my $foo;
487
488open (FDISK, "$fdisk -l $device 2>/dev/null |") || die "Unable to read from $fdisk";
489while (<FDISK>) {
490 print if (($_ !~ /^\/dev\//) and (not (defined $wpart)) and ($verbose == 1));
491 if ($_ =~ /^Units/) {
492 ($foo, $un , $foo) = split /=/;
493 $un =~ s/[A-z\s=]//g;
494 $un = eval($un);
495 }
496}
497close(FDISK);
498print FLOG "get_un returns $un\n";
499return($un);
500}
501
502#
503# Parted gives info in MB
504# (depending on versions - 1.6.25.1 provides suffixes)
505#
506sub get_parted {
507
508my $device = shift;
509my $start = shift;
510my $end = shift;
511my $type = shift;
512my $void;
513my $d;
514my $n;
515my $ret;
516my $mode;
517my $size;
518my $unit;
519
520open (PARTED, "$parted -v |") || die "Unable to read from $parted";
521$d = <PARTED>;
522print FLOG "$d";
523close(PARTED);
524
525open (PARTED, "$parted -s $device print |") || die "Unable to read from $parted";
526# Skip 3 first lines
527$d = <PARTED>;
528$d = <PARTED>;
529$d = <PARTED>;
530
531# depending on parted version, information given change:
532if ($d =~ /\bSize\b/) {
533 # SLES 10 parted >= 1.6.25
534 $mode=1;
535} else {
536 # RHEL 3 parted 1.6.3
537 # RHEL 4 parted 1.6.19
538 $mode=0;
539}
540print FLOG "mode: $mode\n";
541print FLOG "Got from parted: \n";
542print FLOG "Minor Start End Filesystem\n";
543# Get info from each partition line
544while (($n,$d) = split(/\s/, <PARTED>,2)) {
545 chomp($d);
546 next if ($n !~ /^[1-9]/);
547 $d =~ s/^\s*//;
548 $d =~ s/\s+/ /g;
549 if ($mode == 0) {
550 ($$start{$n},$$end{$n},$$type{$n},$void) = split(/ /,$d);
551 $unit = 1;
552 } elsif ($mode == 1) {
553 ($$start{$n},$$end{$n},$size,$$type{$n},$void) = split(/ /,$d);
554 $unit = $mega;
555 } else {
556 die "Undefined mode $mode";
557 }
558 $$start{$n} = "" if (not defined $$start{$n});
559 $$end{$n} = "" if (not defined $$end{$n});
560 $$type{$n} = "" if (not defined $$type{$n});
561 # Handles potential suffixes in latest parted version. Return MB
562 $ret = decode_Bsuf($$start{$n},$unit);
563 $$start{$n} = $ret;
564 $ret = decode_Bsuf($$end{$n},$unit);
565 $$end{$n} = $ret;
566 print FLOG "$n $$start{$n} $$end{$n} $$type{$n}\n";
567}
568close(PARTED);
569}
570
571sub decode_Bsuf {
572
573my $size = shift;
574my $unit = shift;
575my $ret = 0;
576
577#print FLOG "decode_Bsuf input: $size / $unit ";
578if ($size =~ /K[B]*$/i) {
579 $size =~ s/K[B]*$//i;
580 $size *= 1024;
581} elsif ($size =~ /M[B]*$/i) {
582 $size =~ s/M[B]*$//i;
583 $size *= 1048576;
584} elsif ($size =~ /G[B]*$/i) {
585 $size =~ s/G[B]*$//i;
586 $size *= 1073741824;
587} elsif ($size =~ /T[B]*$/i) {
588 $size =~ s/T[B]*$//i;
589 $size *= 1099511627776;
590} else {
591 # Nothing to do
592}
593$ret = $size / $unit;
594#print FLOG " - output : $size => $ret\n";
595return($ret);
596}
597
598
599# Based on Version 2.4 27-Sep-1996 Charles Bailey bailey@genetics.upenn.edu
600# in Basename.pm
601
602sub basename {
603
604my($fullname) = shift;
605
606my($dirpath,$basename);
607
608($dirpath,$basename) = ($fullname =~ m#^(.*/)?(.*)#s);
609
610return($basename);
611}
612
613sub myexit {
614
615my $val=shift;
616
617close(FLOG);
618exit($val);
619}
620
621sub which_type {
622
623my $device = shift;
624my $type = "";
625
626open (FDISK, "$fdisk -l $device 2>/dev/null |") || die "Unable to read from $fdisk";
627while (<FDISK>) {
628 if ($_ =~ /EFI GPT/) {
629 $type= "gpt";
630 print FLOG "Found a GPT partition format\n";
631 last;
632 }
633}
634close(FDISK);
635open (PARTED, "$parted -s $device print|") || die "Unable to read from $fdisk";
636while (<PARTED>) {
637 if ($_ =~ /Disk label type: msdos/) {
638 $type= "msdos";
639 print FLOG "Found a msdos partition format\n";
640 last;
641 }
642}
643close(FDISK);
644return ($type);
645}
646
647sub mysyn {
648 print "Syntax: $0 [-l] device | [-s] partition\n";
649 myexit(-1);
650}
Note: See TracBrowser for help on using the repository browser.