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
|
#!/usr/bin/perl
# **** License ****
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License version 2 as
# published by the Free Software Foundation.
#
# This program is distributed in the hope that it will be useful, but
# WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
# General Public License for more details.
#
# This code was originally developed by Vyatta, Inc.
# Portions created by Vyatta are Copyright (C) 2010 Vyatta, Inc.
# All Rights Reserved.
#
# Author: Bob Gilligan
# Date: April 30, 2010
# Description: Script to re-name a system image.
#
# **** End License ****
use strict;
use warnings;
use Getopt::Long;
use File::Temp qw/ tempfile tempdir /;
use File::Copy;
use Sys::Syslog qw/:standard :macros/;
my $UNION_BOOT = '/live/image/boot';
my $XEN_DEFAULT_IMAGE = "$UNION_BOOT/%%default_image";
my $old_name;
my $new_name;
GetOptions(
'old_name:s' => \$old_name,
'new_name:s' => \$new_name,
);
if (!defined($old_name) || !defined($new_name)) {
printf("Must specify both old ane new name.\n");
exit 1;
}
my $image_path = "/live/image/boot";
if (! -e "$image_path") {
# must be running on old non-image installed system
$image_path = "";
}
if (! -e "$image_path/$old_name") {
printf("Old name $old_name does not exist.\n");
exit 1;
}
if (("$new_name" eq "Old-non-image-installation") ||
("$new_name" eq "grub") ||
("$new_name" =~ /^initrd/) ||
("$new_name" =~ /^vmlinuz/) ||
("$new_name" =~ /^System\.map/) ||
("$new_name" =~ /^config-/) ||
("$new_name" =~ /^%%/)) {
printf("Can't use reserved image name.\n");
exit 1;
}
my $cmdline=`cat /proc/cmdline`;
my $cur_name;
($cur_name, undef) = split(' ', $cmdline);
if ($cur_name =~ s/BOOT_IMAGE=\/boot\///) {
$cur_name =~ s/\/vmlinuz.*//;
} else {
# Boot command line is not formatted as it would be for a system
# booted via grub2 with union mounted root filesystem. Another
# possibility is that it the system is Xen booted via pygrub.
#
if (-l $XEN_DEFAULT_IMAGE) {
# On Xen/pygrub systems, we figure out the running version by
# looking at the bind mount of /boot.
$cur_name = `mount | awk '/on \\/boot / { print \$1 }'`;
$cur_name =~ s/\/live\/image\/boot\///;
chomp($cur_name);
}
}
if ($old_name eq $cur_name) {
printf("Can't re-name the running image.\n");
exit 1;
}
if (-e "$image_path/$new_name") {
printf("New name $new_name already exists.\n");
exit 1;
}
printf("Renaming image $old_name to $new_name.\n");
my $tmpfh;
my $tmpfilename;
($tmpfh, $tmpfilename) = tempfile();
open (my $grubfh, '<', "${image_path}/grub/grub.cfg")
or die "Can't open grub file.\n";
# This is sensitive to the format of menu entries and boot paths
# in the grub config file.
#
my $line;
while ($line = <$grubfh>) {
$line =~ s/\/boot\/$old_name/\/boot\/$new_name/g;
$line =~ s/Vyatta $old_name/Vyatta $new_name/;
$line =~ s/Vyatta image $old_name/Vyatta image $new_name/;
$line =~ s/Lost password change $old_name/Lost password change $new_name/;
printf($tmpfh $line);
}
close($tmpfh);
close($grubfh);
mv("$image_path/$old_name", "$image_path/$new_name")
or die "rename $old_name to $new_name failed: $!\n";
cp($tmpfilename, "$image_path/grub/grub.cfg")
or die "copy $tmpfilename to grub.cfg failed: $!\n";
syslog("warning", "System image $old_name has been renamed $new_name");
printf("Done.\n");
|
