1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
|
package partition_table::gpt;
use diagnostics;
use strict;
use vars qw(@ISA);
@ISA = qw(partition_table::raw);
use fs::type;
use partition_table::raw;
use c;
my $nb_primary = 128;
# See https://en.wikipedia.org/wiki/GUID_Partition_Table#Partition_type_GUIDs for a list of exitings GUIDs
my %parted_mapping = (
'linux-swap(v1)' => 'swap',
'ntfs' => 'ntfs-3g',
'fat16' => 'vfat',
'fat32' => 'vfat',
);
my %rev_parted_mapping = reverse %parted_mapping;
# prefer 'fat32' over 'fat16':
$rev_parted_mapping{vfat} = 'fat32';
sub read_one {
my ($hd, $_sector) = @_;
c::get_disk_type($hd->{file}) eq "gpt" or die "$hd->{device} not a GPT disk ($hd->{file})";
my @pt;
foreach (c::get_disk_partitions($hd->{file})) {
# compatibility with MBR partitions tables:
$_->{pt_type} = 0x82 if $_->{fs_type} eq 'swap';
$_->{pt_type} = 0x0b if $_->{fs_type} eq 'vfat';
$_->{pt_type} = 0x83 if $_->{fs_type} =~ /^ext/;
# fix detecting ESP (special case are they're detected through pt_type):
if ($_->{flag} eq 'ESP') {
$_->{pt_type} = 0xef;
} elsif ($_->{flag} eq 'BIOS_GRUB') {
$_->{fs_type} = $_->{flag}; # hack to prevent it to land in hd->{raw}
$_->{pt_type} = $_->{flag}; # hack...
} elsif ($_->{flag} eq 'LVM') {
$_->{pt_type} = 0x8e;
} elsif ($_->{flag} eq 'RAID') {
$_->{pt_type} = 0xfd;
} elsif ($_->{flag} eq 'RECOVERY') {
$_->{pt_type} = 0x12;
}
$_->{fs_type} = $parted_mapping{$_->{fs_type}} if $parted_mapping{$_->{fs_type}};
@pt[$_->{part_number}-1] = $_;
}
for (my $part_number = 1; $part_number < $nb_primary; $part_number++) {
next if exists($pt[$part_number-1]);
$pt[$part_number-1] = { part_number => $part_number };
}
\@pt;
}
sub write {
my ($hd, $_sector, $pt, $_info) = @_;
my $partitions_killed;
# Initialize the disk if current partition table is not gpt
if (c::get_disk_type($hd->{file}) ne "gpt") {
c::set_disk_type($hd->{file}, "gpt");
$partitions_killed = 1;
}
foreach (@{$hd->{will_tell_kernel}}) {
my ($action, $part_number, $o_start, $o_size) = @$_;
my ($part) = grep { $_->{start} == $o_start && $_->{size} == $o_size } @$pt;
print "($action, $part_number, $o_start, $o_size)\n";
if ($action eq 'add') {
local $part->{fs_type} = $rev_parted_mapping{$part->{fs_type}} if $rev_parted_mapping{$part->{fs_type}};
c::disk_add_partition($hd->{file}, $o_start, $o_size, $part->{fs_type}) or die "failed to add partition #$part_number on $hd->{file}";
my $flag;
if (isESP($part)) {
$flag = 'ESP';
} elsif (isBIOS_GRUB($part)) {
$flag = 'BIOS_GRUB';
} elsif (isRawLVM($part)) {
$flag = 'LVM';
} elsif (isRawRAID($part)) {
$flag = 'RAID';
}
if ($flag) {
c::set_partition_flag($hd->{file}, $part_number, $flag)
or die "failed to set type '$flag' for $part->{file} on $part->{mntpoint}";
}
} elsif ($action eq 'del' && !$partitions_killed) {
c::disk_del_partition($hd->{file}, $part_number) or die "failed to del partition #$part_number on $hd->{file}";
}
}
# prevent errors when telling kernel to reread partition table:
# (above add/del_partition result in udev events)
system(qw(udevadm settle));
common::sync();
1;
}
sub initialize {
my ($class, $hd) = @_;
my @raw;
for (my $part_number = 0; $part_number < $nb_primary-1; $part_number++) {
# part_number starts at 1
$raw[$part_number] = { part_number => $part_number + 1 };
}
$hd->{primary} = { raw => \@raw };
bless $hd, $class;
}
sub can_add { &can_raw_add }
sub adjustStart {}
sub adjustEnd {}
1;
|
