OSDN Git Service

Fail more gracefully if pathname expansion fails
[android-x86/bootable-newinstaller.git] / install / scripts / 1-install
1 #
2 # By Chih-Wei Huang <cwhuang@linux.org.tw>
3 # Last updated 2012/07/07
4 #
5 # License: GNU Public License
6 # We explicitely grant the right to use the scripts
7 # with Android-x86 project.
8 #
9
10 tempfile=/tmp/temp-$$
11 menufile=/tmp/menu-$$
12
13 CPIO=cpio
14
15 rebooting()
16 {
17         dialog --title " Rebooting... " --nocancel --pause "" 8 41 1
18         sync
19         umount -a
20         reboot -f
21 }
22
23 choose()
24 {
25         dialog --clear --title " $1 " \
26                 --menu "$2" 20 71 13 --file $menufile 2> $tempfile
27
28         retval=$?
29         choice=`cat $tempfile`
30 }
31
32 partition_drive()
33 {
34         echo -n > $menufile
35         for i in /sys/block/[sh]d[a-z]; do
36                 if [ ! -d $i ]; then  # pathname expansion failed
37                         continue
38                 fi
39                 echo -n `basename $i` >> $menufile
40                 if [ -f $i/removable -a `cat $i/removable` -eq 0 ]; then
41                         echo -n ' "Harddisk ' >> $menufile
42                 else
43                         echo -n ' "Removable' >> $menufile
44                 fi
45                 if [ -f $i/size ]; then
46                         echo -n " (" `cat $i/size` "blocks)" >> $menufile
47                 fi
48                 echo '"' >> $menufile
49         done
50         count=`wc -l $menufile | awk '{ print $1 }'`
51         if [ $count -eq 0 ]; then
52                 dialog --title " Error " --msgbox \
53                         "\nOK. There is no hard drive to edit partitions." 8 49
54                 return 255
55         fi
56         if [ $count -eq 1 ]; then
57                 choice=`awk '{ print $1 }' $menufile`
58                 retval=0
59         else
60                 choose "Choose Drive" "Please select a drive to edit partitions:"
61         fi
62         if [ $retval -eq 0 ]; then
63                 cfdisk /dev/$choice
64                 if [ $? -eq 0 ]; then
65                         retval=1
66                 else
67                         retval=255
68                 fi
69         fi
70         return $retval
71 }
72
73 select_dev()
74 {
75         fdisk -l | grep ^/dev | cut -b6-12,55- | awk '{
76                 if (!match($2, "Extended")) {
77                         printf("\"%-28s", $0)
78                         system("echo -n `cat /sys/block/*/"$1"/../device/model`")
79                         printf("\" \"\"\n")
80                 }
81         } END {
82                 printf("\"Create/Modify partitions\" \"\"\n\"Detect devices\" \"\"")
83         }' > $menufile
84         choose "Choose Partition" "Please select a partition to install Android-x86:"
85         return $retval
86 }
87
88 progress_bar()
89 {
90         dialog --clear --title " $1 " --gauge "\n $2" 8 70
91 }
92
93 format_fs()
94 {
95         local cmd
96         echo -e '"Do not format" ""\next3 ""\next2 ""\nntfs ""\nfat32 ""' > $menufile
97         choose "Choose filesystem" "Please select a filesystem to format $1:"
98         case "$choice" in
99                 ext3)
100                         cmd="mke2fs -jL"
101                         ;;
102                 ext2)
103                         cmd="mke2fs -L"
104                         ;;
105                 ntfs)
106                         cmd="mkntfs -fL"
107                         ;;
108                 fat32)
109                         cmd="mkdosfs -n"
110                         ;;
111                 *)
112                         ;;
113         esac
114         if [ -n "$cmd" ]; then
115                 dialog --title " Confirm " --no-label Skip --yesno \
116                         "\n You chose to format $1 to $choice.\n All data in that partition will LOSE.\n\n Are you sure to format the partition $1?" 10 51
117                 [ $? -ne 0 ] && return 1
118                 $cmd Android-x86 /dev/$1 | awk '{
119                         # FIXME: very imprecise progress
120                         if (match($0, "done"))
121                                 printf("%d\n", i+=33)
122                 }' | progress_bar "Formatting" "Formatting partition $1..."
123         fi
124 }
125
126 create_entry()
127 {
128         title=$1
129         shift
130         echo -e "title $title\n\tkernel /$asrc/kernel$vga $@ SRC=/$asrc\n\tinitrd /$asrc/initrd.img\n" >> $menulst
131 }
132
133 create_menulst()
134 {
135         menulst=/hd/grub/menu.lst
136         [ -n "$VESA" ] && vga=" vga=788 modeset=0"
137         echo -e "${GRUB_OPTIONS:-default=0\ntimeout=6\nsplashimage=/grub/android-x86.xpm.gz\n}root (hd0,$1)\n" > $menulst
138         cmdline=`cat /proc/cmdline | sed "s|\(initrd.*img\s*\)||; s|quiet\s*||; s|\(vga=\w\+\?\s*\)||; s|\(DPI=\w\+\?\s*\)||; s|\(INSTALL=\w\+\?\s*\)||; s|\(SRC=\S\+\?\s*\)||; s|\(DEBUG=\w\+\?\s*\)||; s|\(BOOT_IMAGE.*\)||"`
139
140         create_entry "Android-x86 $VER" quiet $cmdline
141         create_entry "Android-x86 $VER (Debug mode)" $cmdline DEBUG=2
142         create_entry "Android-x86 $VER (Debug nomodeset)" nomodeset $cmdline DEBUG=2
143         create_entry "Android-x86 $VER (Debug video=LVDS-1:d)" video=LVDS-1:d $cmdline DEBUG=2
144 }
145
146 create_winitem()
147 {
148         win=`fdisk -l /dev/$(echo $1 | cut -b-3) | grep ^/dev | cut -b6-12,55- | awk '{
149                 if (match($2, "NTFS"))
150                         print $1
151         }' | head -1`
152         if [ -n "$win" ]; then
153                 dialog --title " Confirm " --yesno \
154                         "\nThe installer found a Windows partition in /dev/$win.\n\nDo you want to create a boot item for Windows?" 9 59
155                 [ $? -ne 0 ] && return 1
156                 wp=$((`echo $win | cut -b4-`-1))
157                 echo -e "title Windows\n\trootnoverify (hd$d,$wp)\n\tchainloader +1\n" >> $menulst
158         fi
159 }
160
161 create_img()
162 {
163         bname=`basename $2`
164         if [ -e $2 ]; then
165                 dialog --title " Confirm " --defaultno --yesno \
166                         "\n $bname exists. Overwrite it?" 7 38
167                 [ $? -ne 0 ] && return 255
168                 rm -f $2
169         fi
170         dialog --title " Question " --nook --nocancel --inputbox \
171                 "\nPlease input the size of the $bname in MB (max 2047):" 8 63 $1 2> $tempfile
172         size=`cat $tempfile`
173         [ 0$size -le 0 -o 0$size -gt 2047 ] && size=2047
174         ( dd bs=1M count=$size if=/dev/zero | pv -ns ${size}m | dd of=$2 ) 2>&1 \
175                 | progress_bar "Creating $bname" "Expect to write $size MB..."
176 }
177
178 create_data_img()
179 {
180         dialog --title " Confirm " --yesno \
181                 "\nThe installer is going to create a disk image to save the user data. At least 512MB free disk space is recommended.\n\nAre you sure to create the image?" 11 62
182
183         if [ $? -eq 0 ]; then
184                 if create_img 512 data.img; then
185                         losetup /dev/loop6 data.img
186                         mke2fs -jL /data /dev/loop6 > /dev/tty6
187                 fi
188                 [ $? -ne 0 ] && dialog --msgbox "\n Failed to create data.img." 7 33
189         else
190                 dialog --title " Warning " --msgbox \
191                         "\nOK. So data will be save to a RAMDISK(tmpfs), and lose after power off." 8 49
192         fi
193 }
194
195 try_upgrade()
196 {
197         [ -d $1 ] && return
198         PREV_VERS="4.4-r1 4.4-RC2 4.4-RC1 4.4-test 4.3-test 4.2-test 4.0-r1 4.0-RC2 4.0-RC1"
199         for v in $PREV_VERS; do
200                 prev=hd/android-$v
201                 if [ -d $prev ]; then
202                         dialog --title " Question " --yesno \
203                                 "\nAn older Android-x86 version $v is detected.\nWould you like to upgrade it?" 8 55
204                         if [ $? -eq 0 ]; then
205                                 mv $prev $1
206                                 rm -rf $1/data/dalvik-cache/* $1/data/system/wpa_supplicant
207                                 sed -i 's/\(ctrl_interface=\)\(.*\)/\1wlan0/' $1/data/misc/wifi/wpa_supplicant.conf
208                                 break
209                         fi
210                 fi
211         done
212 }
213
214 install_to()
215 {
216         cd /
217         mountpoint -q /hd && umount /hd
218         while [ 1 ]; do
219                 format_fs $1
220                 try_mount rw /dev/$1 /hd && break
221                 dialog --clear --title " Error " --defaultno --yesno \
222                         "\n Cannot mount /dev/$1\n Do you want to format it?" 8 37
223                 [ $? -ne 0 ] && return 255
224         done
225
226         fs=`cat /proc/mounts | grep /dev/$1 | awk '{ print $3 }'`
227
228         asrc=android-$VER
229         dialog --title " Confirm " --no-label Skip --defaultno --yesno \
230                 "\n Do you want to install boot loader GRUB?" 7 47
231         if [ $? -eq 0 ]; then
232                 cp -af /grub /hd
233                 d=0
234                 while [ 1 ]; do
235                         h=`echo $d | awk '{ printf("%c", $1+97) }'`
236                         [ -d /sys/block/[sh]d$h/$1 ] && break
237                         d=$(($d+1))
238                 done
239                 p=$((`echo $1 | cut -b4-`-1))
240                 create_menulst $p
241                 create_winitem $1 $d
242                 rm -f /hd/boot/grub/stage1
243                 echo "setup (hd$d) (hd$d,$p)" | grub > /dev/tty5
244                 [ $? -ne 0 ] && return 255
245         fi
246
247         dialog --title " Question " --yesno \
248                 "\nDo you want to install /system directory as read-write?\n\nMaking /system be read-write is easier for debugging, but it needs more disk space and longer installation time." 10 61
249         instal_rw=$?
250
251         files="mnt/$SRC/kernel mnt/$SRC/initrd.img mnt/$SRC/ramdisk.img"
252         if [ $instal_rw -eq 0 ]; then
253                 if [ "$fs" = "vfat" -o "$fs" = "fuseblk" ]; then
254                         [ -e /sfs/system.img ] && sysimg="/sfs/system.img" || sysimg="mnt/$SRC/system.*"
255                 else
256                         sysimg="android/system"
257                 fi
258         else
259                 sysimg="mnt/$SRC/system.*"
260         fi
261         files="$files $sysimg"
262         size=0
263         for s in `du -sk $files | awk '{print $1}'`; do
264                 size=$(($size+$s))
265         done
266         try_upgrade hd/$asrc
267         mkdir -p hd/$asrc
268         cd hd/$asrc
269         rm -rf system*
270         ( ( cd /; find $files | $CPIO -H newc -o ) | pv -ns ${size}k | ( $CPIO -iud > /dev/null; echo $? > /tmp/result )) 2>&1 \
271                 | progress_bar "Installing Android-x86" "Expect to write $size KB..."
272         result=$((`cat /tmp/result`*255))
273
274         if [ $result -eq 0 ]; then
275                 chmod 644 *
276                 chown 0.0 *
277                 for d in android mnt sfs ./$SRC; do
278                         [ -d $d ] && mv $d/* . && rmdir $d
279                 done
280
281                 case "$fs" in
282                         vfat|fuseblk)
283                                 create_data_img
284                                 ;;
285                         *)
286                                 mkdir -p data
287                                 ;;
288                 esac
289         fi
290
291         sync
292
293         return $result
294 }
295
296 install_hd()
297 {
298         select_dev || rebooting
299         retval=1
300         case "$choice" in
301                 [sh]d*)
302                         install_to $choice
303                         retval=$?
304                         ;;
305                 Create*)
306                         partition_drive
307                         retval=$?
308                         ;;
309                 Detect*)
310                         dialog --title " Detecting... " --nocancel --pause "" 8 41 1
311                         ;;
312         esac
313         return $retval
314 }
315
316 do_install()
317 {
318         until install_hd; do
319                 if [ $retval -eq 255 ]; then
320                         dialog --title ' Error! ' --yes-label Retry --no-label Reboot \
321                                 --yesno '\nInstallation failed! Please check if you have enough free disk space to install Android-x86.' 8 51
322                         [ $? -eq 1 ] && rebooting
323                 fi
324         done
325
326         [ -n "$VESA" ] || runit="Run Android-x86"
327         dialog --clear --title ' Congratulations! ' \
328                 --menu "\n Android-x86 is installed successfully.\n " 11 51 13 \
329                 "$runit" "" "Reboot" "" 2> $tempfile
330         case "`cat $tempfile`" in
331                 Run*)
332                         cd /android
333                         umount system
334                         if mountpoint -q /sfs; then
335                                 umount /sfs
336                                 if [ -e /hd/$asrc/system.sfs ]; then
337                                         mount -o loop /hd/$asrc/system.sfs /sfs
338                                         mount -o loop /sfs/system.img system
339                                 else
340                                         mount -o loop /hd/$asrc/system.img system
341                                 fi
342                         else
343                                 mount -o loop /hd/$asrc/system.img system
344                         fi
345                         if [ -d /hd/$asrc/data ]; then
346                                 mount --bind /hd/$asrc/data data
347                         elif [ -e /hd/$asrc/data.img ]; then
348                                 mount -o loop /hd/$asrc/data.img data
349                         fi
350                         ;;
351                 *)
352                         rebooting
353                         ;;
354         esac
355 }