blob: 86323341a426cd737122680575fad5858b75b9f2 (
plain)
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
|
#!/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 /;
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();
if (!open (GRUBFH, "<${image_path}/grub/grub.cfg")) {
printf("Can't open grub file.\n");
exit 1;
}
# 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);
system("mv $image_path/$old_name $image_path/$new_name");
system("cp $tmpfilename $image_path/grub/grub.cfg");
system("logger -p local3.warning -t 'SystemImage' 'System image $old_name has been renamed $new_name'");
printf("Done.\n");
|
