home *** CD-ROM | disk | FTP | other *** search
/ Inter.Net 55-2 / Inter.Net 55-2.iso / Mandrake / mdkinst / usr / bin / perl-install / partition_table_raw.pm < prev    next >
Encoding:
Perl POD Document  |  2000-01-12  |  3.9 KB  |  132 lines
  1. package partition_table_raw;
  2.  
  3.  
  4.  
  5.  
  6. use common qw(:common :system :file);
  7. use devices;
  8. use c;
  9.  
  10. my @fields = qw(active start_head start_sec start_cyl type end_head end_sec end_cyl start size);
  11. my $format = "C8 I2";
  12. my $magic = "\x55\xAA";
  13. my $nb_primary = 4;
  14.  
  15. my $offset = $common::SECTORSIZE - length($magic) - $nb_primary * common::psizeof($format);
  16.  
  17. my @MBR_signatures = (
  18.     [ 'empty', 0, "\0\0\0\0" ],
  19.     [ 'lilo', 0x2,  "LILO" ],
  20.     [ 'lilo', 0x6,  "LILO" ],
  21.     [ 'osbs', 0x2,  "OSBS" ], 
  22.     [ 'pqmagic', 0xef, "PQV" ],
  23.     [ 'BootStar', 0x130, "BootStar:" ],
  24.     [ 'DocsBoot', 0x148, 'DocsBoot' ],
  25.     [ 'system_commander', 0x1ad, "SYSCMNDRSYS" ],
  26.     [ 'Be Os', 0x24, 'Boot Manager' ],
  27.     [ 'TimO', 0, 'IBM Thinkpad hibernation partition' ],
  28.     [ 'os2', 0x1c2, "\xA" ],
  29.     [ 'dos', 0xa0, "\x25\x03\x4E\x02\xCD\x13" ],
  30.     [ 'dos', 0x60, "\xBB\x00\x7C\xB8\x01\x02\x57\xCD\x13\x5F\x73\x0C\x33\xC0\xCD\x13" ], 
  31.     [ 'freebsd', 0xC0, "\x00\x30\xE4\xCD\x16\xCD\x19\xBB\x07\x00\xB4" ],
  32.     [ 'dummy', 0xAC, "\x0E\xB3\x07\x56\xCD\x10\x5E\xEB" ], 
  33.     [ 'ranish', 0x100, "\x6A\x10\xB4\x42\x8B\xF4\xCD\x13\x8B\xE5\x73" ],
  34. );
  35.  
  36. sub typeOfMBR($) { typeFromMagic(devices::make($_[0]), @MBR_signatures) }
  37. sub typeOfMBR_($) { typeFromMagic($_[0], @MBR_signatures) }
  38.  
  39. sub compute_CHS($$) {
  40.     my ($hd, $e) = @_;
  41.     my @l = qw(cyl head sec);
  42.     @$e{map { "start_$_" } @l} = $e->{start} || $e->{type} ? CHS2rawCHS(sector2CHS($hd, $e->{start})) : (0,0,0);
  43.     @$e{map { "end_$_"   } @l} = $e->{start} || $e->{type} ? CHS2rawCHS(sector2CHS($hd, $e->{start} + $e->{size} - 1)) : (0,0,0);
  44.     1;
  45. }
  46.  
  47. sub CHS2rawCHS($$$) {
  48.     my ($c, $h, $s) = @_;
  49.     $c = min($c, 1023); 
  50.     ($c & 0xff, $h, $s | ($c >> 2 & 0xc0));
  51. }
  52.  
  53. # returns (cylinder, head, sector)
  54. sub sector2CHS($$) {
  55.     my ($hd, $start) = @_;
  56.     my ($s, $h);
  57.     ($start, $s) = divide($start, $hd->{geom}{sectors});
  58.     ($start, $h) = divide($start, $hd->{geom}{heads});
  59.     ($start, $h, $s + 1);
  60. }
  61.  
  62. sub get_geometry($) {
  63.     my ($dev) = @_;
  64.     my $g = "";
  65.  
  66.     local *F; sysopen F, $dev, 0 or return;
  67.     ioctl(F, c::HDIO_GETGEO(), $g) or return;
  68.  
  69.     my %geom; @geom{qw(heads sectors cylinders start)} = unpack "CCSL", $g;
  70.  
  71.     { geom => \%geom, totalsectors => $geom{heads} * $geom{sectors} * $geom{cylinders} };
  72. }
  73.  
  74. sub openit($$;$) { sysopen $_[1], $_[0]{file}, $_[2] || 0; }
  75.  
  76. # cause kernel to re-read partition table
  77. sub kernel_read($) {
  78.     my ($hd) = @_;
  79.     local *F; openit($hd, *F) or return 0;
  80.     $hd->{rebootNeeded} = !ioctl(F, c::BLKRRPART(), 0);
  81. }
  82.  
  83. sub read($$) {
  84.     my ($hd, $sector) = @_;
  85.     my $tmp;
  86.  
  87.     local *F; openit($hd, *F) or return;
  88.     c::lseek_sector(fileno(F), $sector, $offset) or die "reading of partition in sector $sector failed";
  89.  
  90.     my @pt = map {
  91.     sysread F, $tmp, psizeof($format) or return "error while reading partition table in sector $sector";
  92.     my %h; @h{@fields} = unpack $format, $tmp;
  93.     \%h;
  94.     } (1..$nb_primary);
  95.  
  96.     
  97.     sysread F, $tmp, length $magic or die "error reading magic number";
  98.     $tmp eq $magic or die "bad magic number";
  99.  
  100.     [ @pt ];
  101. }
  102.  
  103. # write the partition table (and extended ones)
  104. # for each entry, it uses fields: start, size, type, active
  105. sub write($$$) {
  106.     my ($hd, $sector, $pt) = @_;
  107.  
  108.     local *F; openit($hd, *F, 2) or die "error opening device $hd->{device} for writing";
  109.     c::lseek_sector(fileno(F), $sector, $offset) or return 0;
  110.  
  111.     @$pt == $nb_primary or die "partition table does not have $nb_primary entries";
  112.     foreach (@$pt) {
  113.     compute_CHS($hd, $_);
  114.     local $_->{start} = $_->{local_start} || 0;
  115.     $_->{active} ||= 0; $_->{type} ||= 0; $_->{size} ||= 0; 
  116.     syswrite F, pack($format, @$_{@fields}), psizeof($format) or return 0;
  117.     }
  118.     syswrite F, $magic, length $magic or return 0;
  119.     1;
  120. }
  121.  
  122. sub clear_raw { { raw => [ ({}) x $nb_primary ] } }
  123.  
  124. sub zero_MBR($) {
  125.     my ($hd) = @_;
  126.     $hd->{isDirty} = $hd->{needKernelReread} = 1;
  127.     $hd->{primary} = clear_raw();
  128.     delete $hd->{extended};
  129. }
  130.  
  131. 1;
  132.