Bugfix 5557: Add script to help update grub after full upgrade.

1 files changed, 217 insertions, 0 deletions

diff --git a/scripts/vyatta-update-grub.pl b/scripts/vyatta-update-grub.pl
new file mode 100644
index 00000000..3ef426b9
--- /dev/null
+++ b/scripts/vyatta-update-grub.pl
@@ -0,0 +1,217 @@
+#!/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: Jully, 2010
+# Description: Script to update the grub config file after full upgrade.
+#
+# **** End License ****
+
+use strict;
+use warnings;
+use Getopt::Long;
+use File::Temp qw/ :mktemp /;
+
+my $UNION_BOOT_DIR = '/live/image/boot';
+my $UNION_GRUB_CFG_DIR = "$UNION_BOOT_DIR/grub";
+my $DISK_BOOT_DIR = '/boot';
+my $DISK_GRUB_CFG_DIR= '/boot/grub';
+my $OLD_IMG_VER_STR = 'Old-non-image-installation';
+
+# Returns the version string of the currently running system.
+# The logic is roughly like this:
+# 
+# First, figure out whether system is image-booted or standard booted.
+# Next, locate the grub config file.
+# Next, edit all entries for running image to use the vmlinuz and
+# initrd.img symlinks instead of specific kernel and initrd files.
+# Next, change the vmlinuz and initrd.img symlinks to point to the
+# correct kernel and initrd.img files.
+#
+sub curVer {
+    my $vers = `awk '{print \$1}' /proc/cmdline`;
+
+    # In an image-booted system, the image name is the directory name
+    # directory under "/boot" in the pathname of the kernel we booted.
+    $vers =~ s/BOOT_IMAGE=\/boot\///;
+    $vers =~ s/\/?vmlinuz.*\n$//;
+
+    # In a non-image system, the kernel resides directly under "/boot".
+    # No second-level directory means that $vers will be null.
+    if ($vers eq "") {
+        $vers = $OLD_IMG_VER_STR;
+    }
+    return $vers;
+}
+
+my $boot_dir;
+my $grub_cfg_dir;
+my $grub_cfg_file;
+my $tmp_grub_cfg_file;
+my $on_disk_kernel_dir;
+my $local_kernel_dir;
+my $running_image_name;
+my $debug_flag = 0;
+
+GetOptions(
+    "debug"     => \$debug_flag,
+    );
+
+
+sub log_msg {
+    my $message = shift;
+
+    print "DEBUG: $message" if $debug_flag;
+}
+
+
+# Main
+
+if (-e $UNION_BOOT_DIR) {
+    $boot_dir = $UNION_BOOT_DIR;
+    $grub_cfg_dir = $UNION_GRUB_CFG_DIR;
+
+    $running_image_name = curVer();
+    if ($running_image_name eq $OLD_IMG_VER_STR) {
+	print "Can't find image name for union booted system.\n";
+	exit 1;
+    }
+    $on_disk_kernel_dir = "/boot/$running_image_name";
+    $local_kernel_dir = "$boot_dir/$running_image_name";
+    if (! -e $local_kernel_dir) {
+	print "Can't find kernel directory: $local_kernel_dir\n";
+	exit 1;
+    }
+    log_msg("System is image booted.\n");
+} elsif (-e $DISK_BOOT_DIR) {
+    $boot_dir = $DISK_BOOT_DIR;
+    $grub_cfg_dir = $DISK_GRUB_CFG_DIR;
+    $on_disk_kernel_dir = $boot_dir;
+    $local_kernel_dir = $boot_dir;
+    $running_image_name = "";
+    log_msg("System is disk booted.\n");
+} else {
+    print "Can't locate boot directory!\n";
+    exit 1;
+}
+
+$grub_cfg_file = "$grub_cfg_dir/grub.cfg";
+
+if (! -e $grub_cfg_file) {
+    print "can't locate grub config file: $grub_cfg_file\n";
+    exit 1;
+}
+
+if (-e "$local_kernel_dir/linux" && ! -l "$local_kernel_dir/linux") {
+    print "Linux binary is not a symlink:  $local_kernel_dir/linux\n";
+    exit 1;
+}
+
+if (-e "$local_kernel_dir/initrd.img" && ! -l "$local_kernel_dir/initrd.img") {
+    print "initrd file is not a symlink:  $local_kernel_dir/initrd.img\n";
+    exit 1;
+}
+
+# Back up the original grub config file
+system("cp $grub_cfg_file $grub_cfg_dir/orig_grub.cfg");
+if ($? >> 8) {
+    print "Couldn't back up original grub config file.\n";
+    exit 1;
+}
+
+# Make temp copy to work on
+$tmp_grub_cfg_file = $grub_cfg_file . ".tmp";
+
+system("cp $grub_cfg_file $tmp_grub_cfg_file");
+if ($? >> 8) {
+    print "Couldn't back up temp copy of grub config file.\n";
+    exit 1;
+}
+
+my $sedcmd="sed -i 's+linux $on_disk_kernel_dir\/vmlinuz-[^ ]* +linux $on_disk_kernel_dir\/vmlinuz +' $tmp_grub_cfg_file";
+
+log_msg("Executing: $sedcmd \n");
+
+system($sedcmd);
+if ($? >> 8) {
+    print "Couldn't edit linux entry in grub config file: $tmp_grub_cfg_file";
+    exit 1;
+}
+
+
+$sedcmd="sed -i 's+initrd $on_disk_kernel_dir\/initrd-[^ ]* +initrd $on_disk_kernel_dir\/initrd +' $tmp_grub_cfg_file";
+
+log_msg("Executing: $sedcmd \n");
+
+system($sedcmd);
+if ($? >> 8) {
+    print "Couldn't edit initrd entry in grub config file: $tmp_grub_cfg_file";
+    exit 1;
+}
+
+my $kern_vers=`ls $local_kernel_dir/vmlinuz-*`;
+
+my @kern_vers_list = split(' ', $kern_vers);
+
+if ($#kern_vers_list < 0) {
+    print "No kernel binary files found in grub boot dir: $local_kernel_dir\n";
+    exit 1;
+}
+
+log_msg("kern_vers_list befor subst is: @kern_vers_list\n");
+
+foreach (@kern_vers_list) {
+    s/$local_kernel_dir\/vmlinuz-//;
+}
+@kern_vers_list = sort(@kern_vers_list);
+
+log_msg("kern_vers_list after subst is: @kern_vers_list\n");
+
+# Search in reverse leacographic order (higest first) for a version to use
+foreach my $index ($#kern_vers_list..0) {
+    my $vers = $kern_vers_list[$index];
+    if (-e "$local_kernel_dir/vmlinuz-$vers" && 
+	-e "$local_kernel_dir/initrd.img-$vers") {
+	log_msg("Using version $vers\n");
+	system("rm -f $local_kernel_dir/vmlinuz");
+	system("ln -s vmlinuz-$vers $local_kernel_dir/vmlinuz");
+	if ($? >> 8) {
+	    print "Couldn't symlink kernel binary: $local_kernel_dir/vmlinuz-$vers\n";
+	    # keep going
+	}
+	
+	system("rm -f $local_kernel_dir/initrd.img");
+	system("ln -s initrd.img-$vers $local_kernel_dir/initrd.img");
+	if ($? >> 8) {
+	    print "Couldn't symlink initrd file: $local_kernel_dir/initrd.img-$vers\n";
+	    # keep going
+	}
+
+	# Move our edited grub config file back into position
+	system("mv $tmp_grub_cfg_file $grub_cfg_file");
+	if ($? >> 8) {
+	    print "Couldn't move edited grub config file into position: $grub_cfg_file\n";
+	    # keep going
+	}
+
+	# done
+	print "Done.\n";
+	exit 0;
+    }
+}
+
+print "Couldn't find matching vmlinuz and initrd.img files!\n";
+exit 1;