root / snf-image-host / common.sh.in @ edf5dd1a
History | View | Annotate | Download (8.3 kB)
1 |
# Copyright 2011 GRNET S.A. All rights reserved. |
---|---|
2 |
# |
3 |
# Redistribution and use in source and binary forms, with or without |
4 |
# modification, are permitted provided that the following conditions |
5 |
# are met: |
6 |
# |
7 |
# 1. Redistributions of source code must retain the above copyright |
8 |
# notice, this list of conditions and the following disclaimer. |
9 |
# |
10 |
# 2. Redistributions in binary form must reproduce the above copyright |
11 |
# notice, this list of conditions and the following disclaimer in the |
12 |
# documentation and/or other materials provided with the distribution. |
13 |
# |
14 |
# THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND |
15 |
# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
16 |
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE |
17 |
# ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE |
18 |
# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL |
19 |
# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS |
20 |
# OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) |
21 |
# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT |
22 |
# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY |
23 |
# OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF |
24 |
# SUCH DAMAGE. |
25 |
# |
26 |
# The views and conclusions contained in the software and documentation are |
27 |
# those of the authors and should not be interpreted as representing official |
28 |
# policies, either expressed or implied, of GRNET S.A. |
29 |
|
30 |
AWK="awk" |
31 |
KPARTX="kpartx" |
32 |
LOSETUP="losetup" |
33 |
SFDISK="sfdisk" |
34 |
PROGRESS_MONITOR="snf-progress-monitor" |
35 |
INSTALL_MBR="install-mbr" |
36 |
TIMELIMIT="timelimit" |
37 |
|
38 |
progress_monitor_support="@progress_monitor_support@" |
39 |
|
40 |
CLEANUP=( ) |
41 |
|
42 |
log_error() { |
43 |
echo "$@" >&2 |
44 |
} |
45 |
|
46 |
get_api5_arguments() { |
47 |
GETOPT_RESULT=$* |
48 |
# Note the quotes around `$TEMP': they are essential! |
49 |
eval set -- "$GETOPT_RESULT" |
50 |
while true; do |
51 |
case "$1" in |
52 |
-i|-n) instance=$2; shift 2;; |
53 |
|
54 |
-o) old_name=$2; shift 2;; |
55 |
|
56 |
-b) blockdev=$2; shift 2;; |
57 |
|
58 |
-s) swapdev=$2; shift 2;; |
59 |
|
60 |
--) shift; break;; |
61 |
|
62 |
*) log_error "Internal error!" >&2; exit 1;; |
63 |
esac |
64 |
done |
65 |
if [ -z "$instance" -o -z "$blockdev" ]; then |
66 |
log_error "Missing OS API Argument (-i, -n, or -b)" |
67 |
exit 1 |
68 |
fi |
69 |
if [ "$SCRIPT_NAME" != "export" -a -z "$swapdev" ]; then |
70 |
log_error "Missing OS API Argument -s (swapdev)" |
71 |
exit 1 |
72 |
fi |
73 |
if [ "$SCRIPT_NAME" = "rename" -a -z "$old_name" ]; then |
74 |
log_error "Missing OS API Argument -o (old_name)" |
75 |
exit 1 |
76 |
fi |
77 |
} |
78 |
|
79 |
get_api10_arguments() { |
80 |
if [ -z "$INSTANCE_NAME" -o -z "$HYPERVISOR" -o -z "$DISK_COUNT" ]; then |
81 |
log_error "Missing OS API Variable:" |
82 |
log_error "(INSTANCE_NAME HYPERVISOR or DISK_COUNT)" |
83 |
exit 1 |
84 |
fi |
85 |
instance=$INSTANCE_NAME |
86 |
if [ $DISK_COUNT -lt 1 -o -z "$DISK_0_PATH" ]; then |
87 |
log_error "At least one disk is needed" |
88 |
exit 1 |
89 |
fi |
90 |
if [ "$SCRIPT_NAME" = "export" ]; then |
91 |
if [ -z "$EXPORT_DEVICE" ]; then |
92 |
log_error "Missing OS API Variable EXPORT_DEVICE" |
93 |
fi |
94 |
blockdev=$EXPORT_DEVICE |
95 |
elif [ "$SCRIPT_NAME" = "import" ]; then |
96 |
if [ -z "$IMPORT_DEVICE" ]; then |
97 |
log_error "Missing OS API Variable IMPORT_DEVICE" |
98 |
fi |
99 |
blockdev=$IMPORT_DEVICE |
100 |
else |
101 |
blockdev=$DISK_0_PATH |
102 |
fi |
103 |
if [ "$SCRIPT_NAME" = "rename" -a -z "$OLD_INSTANCE_NAME" ]; then |
104 |
log_error "Missing OS API Variable OLD_INSTANCE_NAME" |
105 |
fi |
106 |
old_name=$OLD_INSTANCE_NAME |
107 |
} |
108 |
|
109 |
get_api20_arguments() { |
110 |
get_api10_arguments |
111 |
if [ -z "$OSP_IMG_ID" ]; then |
112 |
log_error "Missing OS API Parameter: OSP_IMG_ID" |
113 |
exit 1 |
114 |
fi |
115 |
if [ -z "$OSP_IMG_FORMAT" ]; then |
116 |
log_error "Missing OS API Parameter: OSP_IMG_FORMAT" |
117 |
exit 1 |
118 |
fi |
119 |
if [ -z "$OSP_IMG_PASSWD" ]; then |
120 |
log_error "Missing OS API Parameter: OSP_IMG_PASSWD" |
121 |
exit 1 |
122 |
fi |
123 |
|
124 |
IMG_ID=$OSP_IMG_ID |
125 |
IMG_FORMAT=$OSP_IMG_FORMAT |
126 |
IMG_PASSWD=$OSP_IMG_PASSWD |
127 |
if [ -n "$OSP_IMG_PERSONALITY" ]; then |
128 |
IMG_PERSONALITY=$OSP_IMG_PERSONALITY |
129 |
fi |
130 |
} |
131 |
|
132 |
map_disk0() { |
133 |
blockdev="$1" |
134 |
filesystem_dev_base=$($KPARTX -l -p- $blockdev | \ |
135 |
grep -m 1 -- "-1.*$blockdev" | \ |
136 |
$AWK '{print $1}') |
137 |
if [ -z "$filesystem_dev_base" ]; then |
138 |
log_error "Cannot interpret kpartx output and get partition mapping" |
139 |
exit 1 |
140 |
fi |
141 |
$KPARTX -a -p- $blockdev > /dev/null |
142 |
filesystem_dev="/dev/mapper/${filesystem_dev_base/%-1/}" |
143 |
if [ ! -b "/dev/mapper/$filesystem_dev_base" ]; then |
144 |
log_error "Can't find kpartx mapped partition:" \ |
145 |
"/dev/mapper/$filesystem_dev_base" |
146 |
exit 1 |
147 |
fi |
148 |
echo "$filesystem_dev" |
149 |
} |
150 |
|
151 |
unmap_disk0() { |
152 |
$KPARTX -d -p- $1 |
153 |
} |
154 |
|
155 |
format_disk0() { |
156 |
local device="$1" |
157 |
local image_type="$2" |
158 |
|
159 |
declare -A part_id=( ['extdump']="83" ["ntfsdump"]="7" ) |
160 |
|
161 |
# The -f is needed, because we use an optimal alignment and sfdisk complains |
162 |
# about partitions not ending on clylinder boundary. |
163 |
local sfdisk_cmd="$SFDISK -uS -H 255 -S 63 -f --quiet --Linux --DOS $device" |
164 |
|
165 |
$sfdisk_cmd > /dev/null <<EOF |
166 |
2048,,${part_id["$image_type"]},* |
167 |
EOF |
168 |
} |
169 |
|
170 |
create_floppy() { |
171 |
local img=$1 |
172 |
|
173 |
local target=$(mktemp -d) || exit 1 |
174 |
CLEANUP+=("rmdir $target") |
175 |
|
176 |
dd bs=512 count=2880 if=/dev/zero of=$img |
177 |
mkfs.ext2 -F $img > /dev/null |
178 |
mount $img $target -o loop |
179 |
set | egrep ^snf_export_\\w+=|sed -e 's/^snf_export_/SNF_IMAGE_/' | \ |
180 |
while read line; do |
181 |
echo "export $line" >> $target/rules |
182 |
done |
183 |
umount $target |
184 |
} |
185 |
|
186 |
cleanup() { |
187 |
if [ ${#CLEANUP[*]} -gt 0 ]; then |
188 |
LAST_ELEMENT=$((${#CLEANUP[*]}-1)) |
189 |
REVERSE_INDEXES=$(seq ${LAST_ELEMENT} -1 0) |
190 |
for i in $REVERSE_INDEXES; do |
191 |
${CLEANUP[$i]} |
192 |
done |
193 |
fi |
194 |
} |
195 |
|
196 |
trap cleanup EXIT |
197 |
|
198 |
DEFAULT_FILE="@sysconfdir@/default/snf-image" |
199 |
if [ -f "$DEFAULT_FILE" ]; then |
200 |
. "$DEFAULT_FILE" |
201 |
fi |
202 |
|
203 |
: ${ARCH:="x86_64"} |
204 |
: ${VARIANTS_DIR:="@sysconfdir@/ganeti/snf-image/variants"} |
205 |
: ${IMAGE_DIR:="@localstatedir@/lib/snf-image"} |
206 |
: ${HELPER:="@HELPER@"} |
207 |
: ${HELPER_KERNEL:="@HELPER_KERNEL@"} |
208 |
: ${HELPER_INITRD:="@HELPER_INITRD@"} |
209 |
: ${HELPER_SOFT_TIMEOUT=120} |
210 |
: ${HELPER_HARD_TIMEOUT=15} |
211 |
: ${HELPER_USR="nobody"} |
212 |
|
213 |
SCRIPT_NAME=$(basename $0) |
214 |
|
215 |
if [ -f /sbin/blkid -a -x /sbin/blkid ]; then |
216 |
VOL_ID="/sbin/blkid -c /dev/null -o value -s UUID" |
217 |
VOL_TYPE="/sbin/blkid -c /dev/null -o value -s TYPE" |
218 |
else |
219 |
for dir in /lib/udev /sbin; do |
220 |
if [ -f $dir/vol_id -a -x $dir/vol_id ]; then |
221 |
VOL_ID="$dir/vol_id -u" |
222 |
VOL_TYPE="$dir/vol_id -t" |
223 |
fi |
224 |
done |
225 |
fi |
226 |
|
227 |
if [ -z "$VOL_ID" ]; then |
228 |
log_error "vol_id or blkid not found, please install udev or util-linux" |
229 |
exit 1 |
230 |
fi |
231 |
|
232 |
|
233 |
if [ -z "$OS_API_VERSION" -o "$OS_API_VERSION" = "5" ]; then |
234 |
OS_API_VERSION=5 |
235 |
GETOPT_RESULT=`getopt -o o:n:i:b:s: -n '$0' -- "$@"` |
236 |
if [ $? != 0 ] ; then log_error "Terminating..."; exit 1 ; fi |
237 |
get_api5_arguments $GETOPT_RESULT |
238 |
elif [ "$OS_API_VERSION" = "10" -o "$OS_API_VERSION" = "15" ]; then |
239 |
get_api10_arguments |
240 |
elif [ "$OS_API_VERSION" = "20" ]; then |
241 |
get_api20_arguments |
242 |
IMAGE_NAME=$IMG_ID |
243 |
IMAGE_TYPE=$IMG_FORMAT |
244 |
else |
245 |
log_error "Unknown OS API VERSION $OS_API_VERSION" |
246 |
exit 1 |
247 |
fi |
248 |
|
249 |
if [ -n "$OS_VARIANT" ]; then |
250 |
if [ ! -d "$VARIANTS_DIR" ]; then |
251 |
log_error "OS Variants directory $VARIANTS_DIR doesn't exist" |
252 |
exit 1 |
253 |
fi |
254 |
VARIANT_CONFIG="$VARIANTS_DIR/$OS_VARIANT.conf" |
255 |
if [ -f "$VARIANT_CONFIG" ]; then |
256 |
. "$VARIANT_CONFIG" |
257 |
else |
258 |
if grep -qxF "$OS_VARIANT" variants.list; then |
259 |
log_error "ERROR: instance-image configuration error" |
260 |
log_error " Published variant $OS_VARIANT is missing its config" \ |
261 |
"file" |
262 |
log_error " Please create $VARIANT_CONFIG or unpublish the variant" |
263 |
log_error " (by removing $OS_VARIANT from variants.list)" |
264 |
else |
265 |
log_error "Unofficial variant $OS_VARIANT is unsupported" |
266 |
log_error "Most probably this is a user error, forcing a wrong name" |
267 |
log_error "To support this variant please create file" \ |
268 |
"$VARIANT_CONFIG" |
269 |
fi |
270 |
exit 1 |
271 |
fi |
272 |
fi |
273 |
|
274 |
# vim: set sta sts=4 shiftwidth=4 sw=4 et ai : |