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
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
|
#!/usr/bin/perl
use lib qw(/usr/lib/libDrakX ../perl-install);
use standalone;
use fsedit;
use fs::format;
use fs::type;
use resize_fat::main;
use diskdrake::resize_ntfs;
use common;
use partition_table::empty;
use Carp::Heavy;
#- help getting the file in make_rescue_img
BEGIN { partition_table::raw::default_type() }
my ($action, $dir, $server) = @ARGV;
sub usage() { die "partimage_whole_disk <save_all | rest_all> <dir> [<server>]\n" }
@ARGV == 2 || @ARGV == 3 or usage();
$ENV{PATH} = "/sbin:/usr/sbin:$ENV{PATH}";
log::openLog("/var/log/partimage_whole_disk.log");
my @partimage_cmd = ('partimage', if_($server, '-s', $server));
my $all_hds = fsedit::get_hds({});
if ($action eq 'save_all') {
save_all();
} elsif ($action eq 'rest_all') {
rest_all();
} else {
usage();
}
sub save_all() {
my $hd = $all_hds->{hds}[0] or die "no main hard drive\n";
log::l("save_all on $hd->{device}");
my $part_list = [ partition_table::get_normal_parts($hd) ];
foreach (@$part_list) {
$_->{saved} = !member($_->{fs_type}, 'ntfs', 'vfat', 'swap');
if ($_->{saved}) {
run_or_die(@partimage_cmd,
'-V', 0, '--nombr', '--nodesc', '--nocheck', '-b', '-o',
'save', devices::make($_->{device}), "$dir/$_->{device}");
}
}
save_part_list($hd->{geom}, $part_list);
}
sub rest_all() {
my ($forced_geom, $part_list) = read_part_list();
fs::type::set_fs_type($_, $_->{fs_type}) foreach @$part_list;
my @used_hds;
foreach my $part (@$part_list) {
if (my $hd = find { $part->{device} =~ /^\Q$_->{device}\E./ } fs::get::hds($all_hds)) {
push @used_hds, $hd;
put_in_hash($part, partition_table::hd2minimal_part($hd));
}
}
my ($from_partimage, $other) = partition { $_->{saved} } @$part_list;
my ($from_resize, $created) = partition { member($_->{fs_type}, 'vfat', 'ntfs') } @$other;
foreach (@$from_resize) {
#- resize first
my $part = fs::get::device2part($_->{device}, [ fs::get::fstab($all_hds) ]) or log::l("partition to resize is missing ($_->{device})"), next;
$_->{fs_type} eq $_->{fs_type} or log::l("partition $_->{device} doesn't have the right filesystem ($part->{fs_type} != $_->{fs_type})"), next;
$_->{start} = $part->{start};
if ($_->{size} < $part->{size}) {
log::l("resizing $_->{device} to $_->{size} (it is $part->{size})");
my $resize_pkg = $_->{fs_type} eq 'vfat' ? 'resize_fat::main' : 'diskdrake::resize_ntfs';
my $resize = $resize_pkg->new($_->{device}, devices::make($_->{device}));
$resize->resize($_->{size});
} else {
log::l("no need to resize $_->{device} since $_->{size} >= $part->{size}");
}
}
foreach my $hd (uniq(@used_hds)) {
put_in_hash($hd->{geom}, $forced_geom);
#- write the partition table
partition_table::raw::zero_MBR($hd);
foreach my $part (grep { $_->{rootDevice} eq $hd->{device} } @$part_list) {
my $hole = find { isEmpty($_) && $_->{size} >= $part->{size} } partition_table::get_normal_parts_and_holes($hd) or die "not enough room";
$part->{start} = $hole->{start};
log::l("handling $part->{device}");
my $extended = $part->{device} =~ /(\d+)$/ && $1 > 4 && $hd->hasExtended;
my %wanted_part = %$part;
if ($extended || $part->{start} == 1) {
$part->{size} += $hd->{geom}{sectors};
}
partition_table::add($hd, $part, $extended && 'Extended');
foreach ('device', 'size') {
$part->{$_} eq $wanted_part{$_} or log::l("bad $_ for $part->{device}: $part->{$_} != $wanted_part{$_}");
}
}
partition_table::write($hd);
}
#- restore from partimage
foreach (@$from_partimage) {
run_or_die(@partimage_cmd, 'restore', '-b', devices::make($_->{device}), "$dir/$_->{device}");
}
foreach (@$created) {
fs::format::part_raw($_, undef);
}
run_program::run('install_bootloader');
}
sub lst_fields() { qw(device size fs_type saved) }
sub save_part_list {
my ($geom, $part_list) = @_;
my @l = map { join(' ', @$_{lst_fields()}) } @$part_list;
log::l("save_part_list: $_") foreach @l;
my $partimage = join(' ', @partimage_cmd);
open(my $F, "| $partimage -z0 -Bfoo=bar -o save_file $dir/lst");
print $F join("/", $geom->{heads}, $geom->{sectors}), "\n";
print $F "$_\n" foreach @l;
}
sub read_part_list() {
my $partimage = join(' ', @partimage_cmd);
open(my $F, "$partimage -z0 -Bfoo=bar rest_file $dir/lst |");
my $geom_string = <$F>;
my %geom; @geom{'heads', 'sectors'} = split('/', chomp_($geom_string));
my @l = chomp_(cat__($F));
log::l("read_part_list: $_") foreach @l;
\%geom, [ map { my %l; @l{lst_fields()} = split; \%l } @l ];
}
sub run_or_die {
my (@l) = @_;
run_program::raw({ timeout => 4 * 60 * 60 }, @l) or die join(' ', @l) . " failed\n";
}
|
