root / snf-image-host / snf-image-update-helper.in @ d3265cf9
History | View | Annotate | Download (6 kB)
1 |
#!/bin/bash |
---|---|
2 |
|
3 |
# Copyright (C) 2011 GRNET S.A. |
4 |
# |
5 |
# This program is free software; you can redistribute it and/or modify |
6 |
# it under the terms of the GNU General Public License as published by |
7 |
# the Free Software Foundation; either version 2 of the License, or |
8 |
# (at your option) any later version. |
9 |
# |
10 |
# This program is distributed in the hope that it will be useful, but |
11 |
# WITHOUT ANY WARRANTY; without even the implied warranty of |
12 |
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
13 |
# General Public License for more details. |
14 |
# |
15 |
# You should have received a copy of the GNU General Public License |
16 |
# along with this program; if not, write to the Free Software |
17 |
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA |
18 |
# 02110-1301, USA. |
19 |
|
20 |
set -e |
21 |
set -o pipefail |
22 |
set -o errtrace |
23 |
|
24 |
. @osdir@/common.sh |
25 |
|
26 |
usage() { |
27 |
local rc="$1" |
28 |
|
29 |
cat <<EOF |
30 |
|
31 |
Usage: $0 [options] |
32 |
|
33 |
This script runs a debootstrap and creates a small Debian image populated with |
34 |
the snf-image-helper package. This needs to be done before the first use of |
35 |
ganeti's \`snf-image' guest OS type. |
36 |
|
37 |
OPTIONS: |
38 |
-d DIRECTORY |
39 |
Use this directory to host the created files, instead of the default |
40 |
[default: $HELPER_DIR] |
41 |
|
42 |
-h Print this message |
43 |
|
44 |
-p PACKAGE |
45 |
Install this deb package in the helper image, instead of the default |
46 |
|
47 |
-r Don't use any existing cache |
48 |
|
49 |
-y Assume Yes to all queries and do not prompt |
50 |
|
51 |
EOF |
52 |
|
53 |
exit "$rc" |
54 |
} |
55 |
|
56 |
while getopts "d:hp:ry" opt; do |
57 |
case $opt in |
58 |
d) HELPER_DIR="$OPTARG" |
59 |
;; |
60 |
h) usage 0 |
61 |
;; |
62 |
p) HELPER_PKG="$OPTARG" |
63 |
;; |
64 |
r) NO_CACHE="yes" |
65 |
;; |
66 |
y) NO_PROMPT="yes" |
67 |
;; |
68 |
?) log_error "Use \`-h' for help"; exit 1 |
69 |
;; |
70 |
esac |
71 |
done |
72 |
|
73 |
echo |
74 |
echo "This is the update helper image script for snf-image." |
75 |
echo "If you don't know what to do, use \`-h'." |
76 |
echo |
77 |
|
78 |
if [ ! -d "$HELPER_DIR" -o ! -w "$HELPER_DIR" ]; then |
79 |
log_error "Helper directory \`$HELPER_DIR' does not exist or the script" \ |
80 |
"has no write permission on it." |
81 |
exit 1 |
82 |
fi |
83 |
|
84 |
if [ ! -e "$MULTISTRAP_CONFIG" ]; then |
85 |
log_error "Multistrap configuration file: \`$MULTISTRAP_CONFIG' does not" \ |
86 |
"exist or is not readable by the script." |
87 |
exit 1 |
88 |
fi |
89 |
|
90 |
cat >&1 <<EOF |
91 |
This program will overwrite the following files if present: |
92 |
\`$HELPER_DIR/initrd' |
93 |
\`$HELPER_DIR/kernel' |
94 |
\`$HELPER_DIR/image' |
95 |
EOF |
96 |
|
97 |
while [[ 1 ]]; do |
98 |
echo -n "Do you want to continue [y/N]? " |
99 |
if [ "x$NO_PROMPT" = "xyes" ]; then |
100 |
echo "y"; |
101 |
break; |
102 |
fi |
103 |
|
104 |
read answer |
105 |
[ "$(echo -n "$answer" | tr [A-Z] [a-z])" = "y" ] && break |
106 |
if [ -z "$answer" -o "$(echo -n "$answer" | tr [A-Z] [a-z])" = "n" ]; then |
107 |
log_error "Abort." |
108 |
exit 1 |
109 |
fi |
110 |
done |
111 |
|
112 |
rm -f "$HELPER_DIR/initrd" "$HELPER_DIR/kernel" "$HELPER_DIR/image" |
113 |
|
114 |
echo -n "Allocating space for helper disk image..." |
115 |
helper_img=$(mktemp "$HELPER_DIR/image.XXXXXX") |
116 |
add_cleanup rm -f "$helper_img" |
117 |
dd if=/dev/zero of="$helper_img" bs=1k count=400000 &> /dev/null |
118 |
echo "done" |
119 |
|
120 |
echo "Creating partitions..." |
121 |
blockdev=$("$LOSETUP" -sf $helper_img) |
122 |
add_cleanup "$LOSETUP" -d "$blockdev" |
123 |
|
124 |
sleep 1 # sometimes losetup returns and the device is still busy.. |
125 |
|
126 |
format_disk0 "$blockdev" "extdump" 2>&1 | sed -e 's/^/CFDISK: /g' |
127 |
|
128 |
root_dev=$(map_disk0 "$blockdev")-1 |
129 |
add_cleanup unmap_disk0 "$blockdev" |
130 |
|
131 |
echo Creating and configuring filesystem... |
132 |
mkfs.ext3 "$root_dev" 2>&1 | sed -e 's/^/MKFS.EXT3: /g' |
133 |
# The helper vm should never do filesystem checks... |
134 |
tune2fs -i 0 -c 0 "$root_dev" 2>&1 | sed -e 's/^/TUNE2FS: /g' |
135 |
|
136 |
target=$(mktemp -d) |
137 |
add_cleanup rmdir "$target" |
138 |
|
139 |
mount "$root_dev" "$target" |
140 |
add_cleanup umount "$root_dev" |
141 |
|
142 |
do_multistrap "$target" |
143 |
|
144 |
# Save the package list |
145 |
chroot "$target" dpkg-query -W > "$HELPER_DIR/packages" |
146 |
|
147 |
echo -n "Configuring the helper image..." |
148 |
echo snf-image-helper > "$target/etc/hostname" |
149 |
|
150 |
cat > "$target/etc/fstab" <<EOF |
151 |
# /etc/fstab: static file system information. |
152 |
# |
153 |
# <file system> <mount point> <type> <options> <dump> <pass> |
154 |
/dev/sda1 / ext3 defaults 0 1 |
155 |
proc /proc proc defaults 0 0 |
156 |
EOF |
157 |
echo "done" |
158 |
|
159 |
echo -n "Extracting kernel..." |
160 |
if [ ! -L "$target/vmlinuz" -o ! -L "$target/vmlinuz" ]; then |
161 |
echo -e "\033[1;31mfailed\033[0m" |
162 |
log_error "vmlinuz or initrd.img link in root is missing." |
163 |
log_error "I don't know how to find a usable kernel/initrd pair." |
164 |
exit 1 |
165 |
fi |
166 |
echo "done" |
167 |
|
168 |
kernel="$target/$(chroot "$target" readlink -en /vmlinuz)" |
169 |
initrd="$target/$(chroot "$target" readlink -en /initrd.img)" |
170 |
|
171 |
echo "Moving $(basename "$kernel") and $(basename "$initrd") to \`$HELPER_DIR'" |
172 |
cp "$kernel" "$initrd" "$HELPER_DIR" |
173 |
|
174 |
kernel=$(basename "$kernel") |
175 |
initrd=$(basename "$initrd") |
176 |
|
177 |
(cd "$HELPER_DIR"; ln -fs "$kernel" kernel; ln -fs "$initrd" initrd) |
178 |
|
179 |
rm "$target/vmlinuz" "$target/initrd.img" |
180 |
|
181 |
|
182 |
pkg_installed=$(grep snf-image-helper "$HELPER_DIR/packages" > /dev/null && echo yes) |
183 |
|
184 |
if [ -z "$HELPER_PKG" -a -z "$pkg_installed" ]; then |
185 |
log_error "No helper package was specified and non was found by the apt." |
186 |
exit 1 |
187 |
fi |
188 |
|
189 |
if [ -r "$HELPER_PKG" ]; then |
190 |
echo "Installing snf-image-helper pkg in the new image..." |
191 |
cp "$HELPER_PKG" "$target/tmp/" |
192 |
pkg_name=$(basename "$HELPER_PKG") |
193 |
add_cleanup rm "$target/tmp/$pkg_name" |
194 |
chroot "$target" dpkg -i "/tmp/$pkg_name" 2>&1 | sed -e 's/^/DPKG: /g' |
195 |
|
196 |
# Recreate package list |
197 |
chroot "$target" dpkg-query -W > "$HELPER_DIR/packages" |
198 |
else |
199 |
echo "snf-image-helper pkg was installed from the apt repository." |
200 |
fi |
201 |
|
202 |
if [ "$VERSION_CHECK" == yes -a -z "$HELPER_PKG" ]; then |
203 |
helper_version="$(grep ^snf-image-helper[[:space:]] "$HELPER_DIR/packages" | cut -f2)" |
204 |
host_version="$(dpkg-query -W -f "\${Version}\n" snf-image-host)" |
205 |
if [ "$host_version" != "$helper_version" ]; then |
206 |
log_error "snf-image-host version (=$host_version) and " \ |
207 |
"snf-image-helper version (=$helper_version) don't match." |
208 |
exit 1 |
209 |
fi |
210 |
fi |
211 |
|
212 |
mv "$helper_img" "$HELPER_DIR/image" |
213 |
|
214 |
echo "Files in \`$HELPER_DIR' were updated successfully" |
215 |
exit 0 |
216 |
|
217 |
# vim: set sta sts=4 shiftwidth=4 sw=4 et ai : |