]>
Commit | Line | Data |
---|---|---|
ff3bae7b MT |
1 | #!/bin/bash |
2 | ############################################################################ | |
3 | # # | |
4 | # This file is part of the IPFire Firewall. # | |
5 | # # | |
6 | # IPFire is free software; you can redistribute it and/or modify # | |
7 | # it under the terms of the GNU General Public License as published by # | |
8 | # the Free Software Foundation; either version 2 of the License, or # | |
9 | # (at your option) any later version. # | |
10 | # # | |
11 | # IPFire is distributed in the hope that it will be useful, # | |
12 | # but WITHOUT ANY WARRANTY; without even the implied warranty of # | |
13 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # | |
14 | # GNU General Public License for more details. # | |
15 | # # | |
16 | # You should have received a copy of the GNU General Public License # | |
17 | # along with IPFire; if not, write to the Free Software # | |
18 | # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA # | |
19 | # # | |
20 | # Copyright (C) 2014 IPFire Team <info@ipfire.org>. # | |
21 | # # | |
22 | ############################################################################ | |
23 | ||
24 | GRUB_INSTALL_ARGS="--no-floppy --recheck" | |
25 | ||
5946f1f9 MT |
26 | function find_bootloader_device() { |
27 | local mp | |
28 | for mp in /boot /; do | |
29 | if find_device "${mp}"; then | |
30 | return 0 | |
31 | fi | |
32 | done | |
33 | ||
34 | return 1 | |
35 | } | |
36 | ||
37 | function find_device() { | |
38 | local mountpoint="${1}" | |
ff3bae7b MT |
39 | |
40 | local root | |
41 | local dev mp fs flags rest | |
42 | while read -r dev mp fs flags rest; do | |
43 | # Skip unwanted entries | |
44 | [ "${dev}" = "rootfs" ] && continue | |
45 | ||
5946f1f9 | 46 | if [ "${mp}" = "${mountpoint}" ] && [ -b "${dev}" ]; then |
ff3bae7b MT |
47 | root="$(basename "${dev}")" |
48 | break | |
49 | fi | |
50 | done < /proc/mounts | |
51 | ||
52 | # Get the actual device from the partition that holds / | |
5946f1f9 MT |
53 | while [ -n "${root}" ]; do |
54 | if [ -e "/sys/block/${root}" ]; then | |
55 | echo "${root}" | |
56 | return 0 | |
57 | fi | |
58 | ||
59 | # Remove last character | |
60 | root="${root::-1}" | |
61 | done | |
ff3bae7b MT |
62 | |
63 | return 1 | |
64 | } | |
65 | ||
5946f1f9 | 66 | function device_is_mdraid() { |
ff3bae7b MT |
67 | local device="${1}" |
68 | ||
69 | [ -d "/sys/block/${device}/md" ] | |
70 | } | |
71 | ||
72 | function mdraid_get_slaves() { | |
73 | local device="${1}" | |
74 | ||
75 | local slave | |
76 | for slave in /sys/block/${device}/slaves/*; do | |
77 | basename "${slave}" | |
78 | done 2>/dev/null | |
79 | } | |
80 | ||
81 | function grub_update_config() { | |
82 | echo "Updating configuration..." | |
83 | ||
84 | if ! grub-mkconfig -o /boot/grub/grub.cfg &>/dev/null; then | |
85 | echo "Could not update configuration. Aborting." >&2 | |
86 | return 1 | |
87 | fi | |
88 | ||
89 | return 0 | |
90 | } | |
91 | ||
92 | function grub_install() { | |
93 | local device="${1}" | |
94 | ||
95 | echo "Installing GRUB on ${device}..." | |
96 | ||
97 | if [ ! -b "${device}" ]; then | |
98 | echo "${device} does not exist or is not a block device" >&2 | |
99 | return 1 | |
100 | fi | |
101 | ||
102 | local args | |
103 | for args in "" "--force"; do | |
104 | if grub-install ${GRUB_INSTALL_ARGS} ${args} "${device}" &>/dev/null; then | |
105 | return 0 | |
106 | fi | |
107 | done | |
108 | ||
109 | echo "Could not install GRUB on ${device}" >&2 | |
110 | return 1 | |
111 | } | |
112 | ||
113 | function main() { | |
114 | # Find the root device | |
5946f1f9 | 115 | local device="$(find_bootloader_device)" |
ff3bae7b MT |
116 | if [ -z "${device}" ]; then |
117 | echo "Could not find root device. Aborting." >&2 | |
118 | exit 1 | |
119 | fi | |
120 | ||
5946f1f9 | 121 | echo "Found bootloader device: /dev/${device}" |
ff3bae7b MT |
122 | |
123 | # Update configuration files | |
124 | grub_update_config || exit $? | |
125 | ||
126 | # Handle mdraid devices | |
5946f1f9 | 127 | if device_is_mdraid "${device}"; then |
ff3bae7b MT |
128 | local slave |
129 | for slave in $(mdraid_get_slaves "${device}"); do | |
130 | grub_install "/dev/${slave}" | |
131 | done | |
132 | ||
133 | # Handle normal block devices | |
134 | else | |
135 | grub_install "/dev/${device}" | |
136 | fi | |
137 | ||
138 | return 0 | |
139 | } | |
140 | ||
141 | # Run main function | |
142 | main |