blob: dc834487ec24467a5ca1a3b145b684efdf5661d3 (
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
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
|
#!/bin/bash
# source in the functions
source /opt/vyatta/sbin/install-functions
# export INSTALL_LOG for the scripts invoked
export INSTALL_LOG=/tmp/install-$$.log
# export PROGRESS_PID for the scripts invoked
export PROGRESS_PID=$$
# file for get-partition output
PART_FILE=''
# Temp directory for downloaded ISO
TEMP_DIR="/var/tmp/install-image.$$"
fail_exit ()
{
echo "$*"
echo 'Exiting...'
exit 1
}
clean_up ()
{
if [ -n "$PART_FILE" ]; then
rm -f $PART_FILE >&/dev/null
fi
umount $CD_SQUASH_ROOT >&/dev/null || true
umount $CD_ROOT >&/dev/null || true
umount $INST_ROOT >&/dev/null || true
umount $READ_ROOT >&/dev/null || true
umount $WRITE_ROOT >&/dev/null || true
if [ -d "$TEMP_DIR" ]; then
rm -rf $TEMP_DIR
fi
}
sig_handler () {
echo "ERROR: Signal received. Exiting..."
clean_up
echo "Done"
trap - EXIT
exit 1
}
exit_handler () {
clean_up
}
# Try to fetch the ISO file using a URL provided by the user.
# If successful, we leave $NEW_ISO pointing to the ISO file that
# was downloaded.
fetch_iso_by_url ()
{
mkdir $TEMP_DIR
echo "Trying to fetch ISO file from $NEW_ISO"
wget -P $TEMP_DIR $NEW_ISO
if [ $? -ne 0 ]; then
echo "Unable to fetch ISO from $NEW_ISO"
exit 1
fi
filename="${TEMP_DIR}/${NEW_ISO##*/}"
if [ ! -e $filename ]; then
echo "Download of $NEW_ISO failed"
exit 1
fi
echo "ISO download suceeded."
NEW_ISO=$filename
}
# set up the specified ISO image file or URL for install
set_up_new_iso ()
{
url_scheme=${NEW_ISO%%:*}
if [ "$url_scheme" != "$NEW_ISO" ]; then
if [ "$url_scheme" = "http" -o "$url_scheme" = "https" -o \
"$url_scheme" = "ftp" ]; then
fetch_iso_by_url
fi
fi
if [ ! -f "$NEW_ISO" ] || ! (file $NEW_ISO | grep -q 9660); then
fail_exit "\"$NEW_ISO\" is not a valid ISO image file."
fi
# make sure mount points exist
mkdir -p $INST_ROOT $WRITE_ROOT $READ_ROOT $CD_ROOT $CD_SQUASH_ROOT
# mount ISO
margs="-o loop,ro $NEW_ISO $CD_ROOT"
if ! try_mount "$margs"; then
fail_exit 'Failed to mount the new image.'
fi
# check squash image
local squash_file=$CD_ROOT/live/filesystem.squashfs
if [ ! -f "$squash_file" ] || ! (file $squash_file | grep -q Squashfs) \
|| ! grep -q '^ii vyatta-version ' $CD_ROOT/live/packages.txt; then
fail_exit "\"$NEW_ISO\" is not a Vyatta ISO image file."
fi
# mount squash image
margs="-o loop,ro $squash_file $CD_SQUASH_ROOT"
if ! try_mount "$margs"; then
fail_exit 'Failed to mount the squashfs image.'
fi
}
# install new image into a newly-formatted partition.
# will exit with error if installation fails.
install_new ()
{
local root_part=$1
local inst_drv=$2
if [ ! -e "/dev/$root_part" ] || [ ! -e "/dev/$inst_drv" ]; then
fail_exit "Invalid drive/partition ($inst_drv and $root_part)."
fi
# install new image
if ! /opt/vyatta/sbin/install-image-new "$root_part"; then
exit 1
fi
# postinst operations
if ! /opt/vyatta/sbin/install-postinst-new \
"$inst_drv" "$root_part" union; then
exit 1
fi
}
# install new image into the current boot partition.
# will exit with error if installation fails.
install_existing ()
{
local ctype=$1
if ! /opt/vyatta/sbin/install-image-existing "$ctype"; then
exit 1
fi
}
# the image to be installed. only used if installing from an installed,
# running system.
NEW_ISO=$1
if [ `whoami` != 'root' ] ; then
fail_exit 'This script must be run with root privileges.'
fi
trap sig_handler INT KILL
trap exit_handler EXIT
cat <<EOF
Welcome to the Vyatta install program. This script
will walk you through the process of installing the
Vyatta image to a local hard drive.
EOF
response=''
while [ -z $response ]
do
echo -n "Would you like to continue? (Yes/No) [Yes]: "
response=$(get_response "Yes" "Yes No Y N")
if [ "$response" == "no" ] || [ "$response" == "n" ]; then
fail_exit 'Ok then.'
fi
done
if is_live_cd_boot; then
if [ -n "$NEW_ISO" ]; then
echo 'You are trying to install from a live CD boot. The live CD image'
fail_exit 'will be used. Do not specify an ISO image file or URL.'
fi
elif [ -z "$NEW_ISO" ]; then
echo 'You are trying to install from an already installed system. An ISO'
fail_exit 'image file to install or URL must be specified.'
else
# installing on an installed system. set up the new image.
set_up_new_iso
fi
# check versions
if is_live_cd_boot; then
CURVER=LIVECD
else
CURVER=$(get_cur_version)
fi
NEWVER=$(get_new_version)
if [ -z "$CURVER" ] || [ -z "$NEWVER" ]; then
fail_exit 'Cannot find release version.'
fi
if [ "$CURVER" == "$NEWVER" ]; then
fail_exit "Version \"$NEWVER\" is already installed."
fi
# get install partition
PART_FILE=$(mktemp /tmp/inst-get-part.XXXXXX) \
|| fail_exit 'Failed to create temporary file'
if ! /opt/vyatta/sbin/install-get-partition $PART_FILE; then
exit 1
fi
# get the partition params
root_part_type=''
root_part=''
inst_drv=''
eval "read root_part_type root_part inst_drv <$PART_FILE" >&/dev/null
rm -f $PART_FILE >&/dev/null
# handle different types
case "$root_part_type" in
new)
install_new "$root_part" "$inst_drv"
exit 0
;;
union|old)
install_existing "$root_part_type"
exit 0
;;
*)
fail_exit "Unknown partition type \"$root_part_type\"."
;;
esac
|
