OSDN Git Service

Add auto installation function
[android-x86/bootable-newinstaller.git] / install / scripts / 1-install
1 #
2 # By Chih-Wei Huang <cwhuang@linux.org.tw>
3 # Last updated 2018/01/26
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 auto_answer()
24 {
25         echo "$answer" > $tempfile
26         unset answer
27         test "`cat $tempfile`" != "0"
28 }
29
30 set_answer_if_auto()
31 {
32         [ -n "$AUTO_INSTALL" ] && answer="$1"
33 }
34
35 adialog()
36 {
37         if [ -n "$answer" ]; then
38                 auto_answer
39         else
40                 dialog "$@" 2> $tempfile
41         fi
42 }
43
44 choose()
45 {
46         adialog --clear --title " $1 " --menu "$2" 21 79 13 --file $menufile
47
48         retval=$?
49         choice=`cat $tempfile`
50 }
51
52 size_gb()
53 {
54         printf %0.2fGB $(dc `cat $1/size` 2097152 / p)
55 }
56
57 find_partition()
58 {
59         grep -H ^$2$ /sys/block/$1/*/partition 2> /dev/null | cut -d/ -f5
60 }
61
62 list_disks()
63 {
64         for b in /sys/block/[shv]d[a-z] /sys/block/mmcblk? /sys/block/nvme*; do
65                 [ -d $b ] && echo $b
66         done
67 }
68
69 auto_partition()
70 {
71         [ "$AUTO_INSTALL" = "force" ] || dialog --title " Auto Installer " --defaultno --yesno \
72                 "\nYou have chosen the AUTO installation.\n\nThe installer will erase the whole /dev/$1 and install Android-x86 to it.\n\nThis is the last confirmation. Are you sure to do so?" 12 61
73         [ $? -ne 0 ] && rebooting
74
75         if [ -z "$efi" ]; then
76                 echo -e "o\nn\np\n1\n\n\nw\n" | fdisk /dev/$1
77                 p=1
78         else
79                 sgdisk --zap-all /dev/$1 > /dev/tty6
80                 sgdisk --new=1::+260M --typecode=1:EF00 --largest-new=2 --typecode=2:8300 /dev/$1
81                 p=2
82         fi > /dev/tty6
83
84         while sleep 1; do
85                 answer=`find_partition $1 $p`
86                 [ -n "$answer" ] && break
87         done
88 }
89
90 partition_drive()
91 {
92         echo -n > $menufile
93         for i in `list_disks`; do
94                 echo -n `basename $i` >> $menufile
95                 if [ -f $i/removable -a `cat $i/removable` -eq 0 ]; then
96                         echo -n ' "Harddisk ' >> $menufile
97                 else
98                         echo -n ' "Removable' >> $menufile
99                 fi
100                 if [ -f $i/size ]; then
101                         sz=$(size_gb $i)
102                         [ "$sz" = "0.00GB" ] && sz="<0.01GB"
103                         printf " %10s" $sz >> $menufile
104                 fi
105                 for f in $i/device/model $i/*/name; do
106                         [ -e $f ] && echo -n " `sed $'s/\x04//g' $f`" >> $menufile && break
107                 done
108                 [ "`basename $i`" = "$booted_from" -o -d $i/$booted_from ] && echo -n " *" >> $menufile
109                 echo '"' >> $menufile
110         done
111         count=`wc -l < $menufile`
112         if [ $count -eq 0 ]; then
113                 dialog --title " Error " --msgbox \
114                         "\nOK. There is no hard drive to edit partitions." 8 49
115                 return 255
116         fi
117
118         if [ $count -eq 1 -o "$AUTO_INSTALL" = "force" ]; then
119                 drive=1
120         else
121                 drive=`basename $AUTO_INSTALL`
122         fi
123         choice=`awk -v n=$drive '{ if (n == NR || n == $1) print $1 }' $menufile`
124         if [ -b /dev/$choice ]; then
125                 retval=0
126         else
127                 choose "Choose Drive" "Please select a drive to edit partitions:\n\n* - Installer source"
128         fi
129         if [ $retval -eq 0 ]; then
130                 if [ -n "$AUTO_INSTALL" ]; then
131                         auto_partition $choice
132                         return 1
133                 fi
134                 dialog --title " Confirm " --defaultno --yesno "\n Do you want to use GPT?" 7 29
135                 if [ $? -eq 0 ]; then
136                         cgdisk /dev/$choice
137                 else
138                         cfdisk /dev/$choice
139                 fi
140                 if [ $? -eq 0 ]; then
141                         retval=1
142                 else
143                         retval=255
144                 fi
145         fi
146         return $retval
147 }
148
149 select_dev()
150 {
151         blkid | grep -v -E "^/dev/block/|^/dev/loop" | cut -b6- | sort | awk '{
152                 l=""
153                 t="unknown"
154                 sub(/:/, "", $1)
155                 for (i = NF; i > 1; --i)
156                         if (match($i, "^TYPE")) {
157                                 t=$i
158                                 gsub(/TYPE=|"/, "", t)
159                         } else if (match($i, "^LABEL")) {
160                                 l=$i
161                                 gsub(/LABEL=|"/, "", l)
162                         }
163                 printf("%-11s%-12s%-18s\n", $1, t, l)
164         }' > $tempfile
165
166         for d in `ls /sys/block`; do
167                 for i in /sys/block/$d/$d*; do
168                         [ -d $i ] || continue
169                         echo $i | grep -qE "loop|ram|sr|boot|rpmb" && continue
170                         f=$(grep "`basename $i`" $tempfile || printf "%-11s%-29s" `basename $i` unknown)
171                         sz=$(size_gb $i)
172                         [ "$sz" = "0.00GB" ] || printf "$f%10s\n" $sz
173                 done
174         done | awk -v b=$booted_from '{
175                 if (!match($1, b)) {
176                         printf("\"%s\" \"", $0)
177                         system("cd /sys/block/*/"$1"; for f in ../device/model ../device/name; do [ -e $f ] && printf %-17s \"`cat $f`\" && break; done")
178                         printf("\"\n")
179                 }
180         } END {
181                 printf("\"\" \"\"\n\"Create/Modify partitions\" \"\"\n\"Detect devices\" \"\"")
182         }' > $menufile
183         choose "Choose Partition" "Please select a partition to install Android-x86:\n\nRecommended minimum free space - 4GB  |  Optimum free space >= 8GB\n\nPartition | Filesystem | Label            | Size     | Drive name/model"
184         return $retval
185 }
186
187 progress_bar()
188 {
189         dialog --clear --title " $1 " --gauge "\n $2" 8 70
190 }
191
192 convert_fs()
193 {
194         if blkid /dev/$1 | grep -q ext2; then
195                 /system/bin/tune2fs -j /dev/$1
196                 e2fsck -fy /dev/$1
197         fi
198         if blkid /dev/$1 | grep -q ext3; then
199                 /system/bin/tune2fs -O extents,uninit_bg /dev/$1
200                 e2fsck -fy /dev/$1
201         fi
202 }
203
204 format_fs()
205 {
206         local cmd
207         echo -e '"Do not format" ""\next4 ""\nntfs ""\nfat32 ""' > $menufile
208         set_answer_if_auto $FORCE_FORMAT
209         choose "Choose filesystem" "Please select a filesystem to format $1:"
210         case "$choice" in
211                 ext4)
212                         cmd="make_ext4fs -L"
213                         ;;
214                 ntfs)
215                         cmd="mkntfs -fL"
216                         ;;
217                 fat32)
218                         cmd="mkdosfs -n"
219                         ;;
220                 *)
221                         ;;
222         esac
223         if [ -n "$cmd" ]; then
224                 [ -n "$AUTO_INSTALL" ] || dialog --title " Confirm " --defaultno --yesno \
225                         "\n You chose to format $1 to $choice.\n All data in that partition will be LOST.\n\n Are you sure to format the partition $1?" 10 59
226                 [ $? -ne 0 ] && return 1
227                 $cmd Android-x86 /dev/$1 | awk '{
228                         # FIXME: very imprecise progress
229                         if (match($0, "done"))
230                                 printf("%d\n", i+=33)
231                 }' | progress_bar "Formatting" "Formatting partition $1..."
232         elif blkid /dev/$1 | grep -q ext[23]; then
233                 dialog --clear --title " Warning " --yesno \
234                         "\nYou chose to install android-x86 to an ext2/3 filesystem. We suggest you convert it to ext4 for better reliability and performance." 9 62
235                 [ $? -eq 0 ] && convert_fs $1
236         fi
237 }
238
239 create_entry()
240 {
241         title=$1
242         shift
243         echo -e "title $title\n\tkernel /$asrc/kernel$vga $@ SRC=/$asrc\n\tinitrd /$asrc/initrd.img\n" >> $menulst
244 }
245
246 create_menulst()
247 {
248         menulst=/hd/grub/menu.lst
249         [ -n "$VESA" ] && vga=" vga=788 modeset=0"
250         echo -e "${GRUB_OPTIONS:-default=0\ntimeout=6\nsplashimage=/grub/android-x86.xpm.gz\n}root (hd0,$1)\n" > $menulst
251
252         create_entry "Android-x86 $VER" quiet $cmdline
253         create_entry "Android-x86 $VER (Debug mode)" $cmdline DEBUG=2
254         create_entry "Android-x86 $VER (Debug nomodeset)" nomodeset $cmdline DEBUG=2
255         create_entry "Android-x86 $VER (Debug video=LVDS-1:d)" video=LVDS-1:d $cmdline DEBUG=2
256 }
257
258 create_winitem()
259 {
260         win=`fdisk -l /dev/$(echo $1 | cut -b-3) | grep ^/dev | cut -b6-12,55- | awk '{
261                 if (match($2, "NTFS"))
262                         print $1
263         }' | head -1`
264         if [ -n "$win" ]; then
265                 dialog --title " Confirm " --yesno \
266                         "\nThe installer found a Windows partition in /dev/$win.\n\nDo you want to create a boot item for Windows?" 9 59
267                 [ $? -ne 0 ] && return 1
268                 wp=$((`echo $win | cut -b4-`-1))
269                 echo -e "title Windows\n\trootnoverify (hd$d,$wp)\n\tchainloader +1\n" >> $menulst
270         fi
271 }
272
273 check_data_img()
274 {
275         losetup /dev/loop7 data.img
276         if blkid /dev/loop7 | grep -q ext[23]; then
277                 dialog --clear --title " Warning " --yesno \
278                         "\nYour data.img is an ext2/3 filesystem. We suggest you convert it to ext4 for better reliability." 8 58
279                 [ $? -eq 0 ] && convert_fs loop7
280         fi
281         losetup -d /dev/loop7
282 }
283
284 create_img()
285 {
286         bname=`basename $2`
287         if [ -e $2 ]; then
288                 dialog --title " Confirm " --defaultno --yesno \
289                         "\n $bname exists. Overwrite it?" 7 38
290                 [ $? -ne 0 ] && return 255
291                 rm -f $2
292         fi
293         dialog --title " Question " --nook --nocancel --inputbox \
294                 "\nPlease input the size of the $bname in MB:" 8 63 $1 2> $tempfile
295         size=`cat $tempfile`
296         [ 0$size -le 0 ] && size=1024
297         ( dd bs=1M count=$size if=/dev/zero | pv -ns ${size}m | dd of=$2 ) 2>&1 \
298                 | progress_bar "Creating $bname" "Expect to write $size MB..."
299 }
300
301 create_data_img()
302 {
303         dialog --title " Confirm " --yesno \
304                 "\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
305
306         if [ $? -eq 0 ]; then
307                 if create_img 512 data.img; then
308                         losetup /dev/loop6 data.img
309                         make_ext4fs -L /data /dev/loop6 > /dev/tty6
310                 fi
311                 [ $? -ne 0 ] && dialog --msgbox "\n Failed to create data.img." 7 33
312         else
313                 dialog --title " Warning " --msgbox \
314                         "\nOK. So data will be save to a RAMDISK(tmpfs), and lose after power off." 8 49
315         fi
316 }
317
318 try_upgrade()
319 {
320         [ -d $1 ] && return
321         PREV_VERS="$PREV_VERS 7.1-rc2 7.1-rc1 6.0-r3 6.0-r2 6.0-r1 6.0-rc2 6.0-rc1 5.1-rc1 4.4-r5 4.4-r4 4.4-r3 4.4-r2 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"
322         for v in $PREV_VERS; do
323                 local prev
324                 if [ -d hd/$v ]; then
325                         prev=hd/$v
326                 elif [ -d hd/android-$v ]; then
327                         prev=hd/android-$v
328                 else
329                         continue
330                 fi
331                 set_answer_if_auto 1
332                 adialog --title " Question " --yesno \
333                         "\nAn older version $v is detected.\nWould you like to upgrade it?" 8 51
334                 if [ $? -eq 0 ]; then
335                         mv $prev $1
336                         rm -rf $1/data/dalvik-cache/* $1/data/system/wpa_supplicant
337                         sed -i 's/\(ctrl_interface=\)\(.*\)/\1wlan0/' $1/data/misc/wifi/wpa_supplicant.conf
338                         break
339                 fi
340         done
341 }
342
343 get_part_info()
344 {
345         d=0
346         while [ 1 ]; do
347                 h=`echo $d | awk '{ printf("%c", $1+97) }'`
348                 for part in /sys/block/[shv]d$h/$1 /sys/block/mmcblk$d/$1 /sys/block/nvme0n$(($d+1))/$1; do
349                         [ -d $part ] && break 2
350                 done
351                 d=$(($d+1))
352         done
353         p=`cat $part/partition`
354         disk=$(basename `dirname $part`)
355 }
356
357 install_to()
358 {
359         local t=`echo /sys/block/*/$1/partition`
360         [ -f "$t" ] || return 1
361         until [ -b /dev/$1 ]; do
362                 echo add > `dirname $t`/uevent
363                 sleep 1
364         done
365         cd /
366         mountpoint -q /hd && umount /hd
367         [ -n "$AUTO_UPDATE" ] && FORCE_FORMAT=no || FORCE_FORMAT=ext4
368         while [ 1 ]; do
369                 format_fs $1
370                 try_mount rw /dev/$1 /hd && break
371                 dialog --clear --title " Error " --defaultno --yesno \
372                         "\n Cannot mount /dev/$1\n Do you want to format it?" 8 37
373                 [ $? -ne 0 ] && return 255
374                 FORCE_FORMAT=ext4
375         done
376
377         fs=`cat /proc/mounts | grep /dev/$1 | awk '{ print $3 }'`
378         cmdline=`sed "s|\(initrd.*img\s*\)||; s|quiet\s*||; s|\(vga=\w\+\?\s*\)||; s|\(DPI=\w\+\?\s*\)||; s|\(AUTO_INSTALL=\w\+\?\s*\)||; s|\(INSTALL=\w\+\?\s*\)||; s|\(SRC=\S\+\?\s*\)||; s|\(DEBUG=\w\+\?\s*\)||; s|\(BOOT_IMAGE=\S\+\?\s*\)||; s|\(iso-scan/filename=\S\+\?\s*\)||; s|[[:space:]]*$||" /proc/cmdline`
379
380         [ -n "$INSTALL_PREFIX" ] && asrc=$INSTALL_PREFIX || asrc=android-$VER
381         set_answer_if_auto 1
382         [ -z "$efi" ] && adialog --title " Confirm " --no-label Skip --defaultno --yesno \
383                 "\n Do you want to install boot loader GRUB?" 7 47
384         if [ $? -eq 0 ]; then
385                 cp -af /grub /hd
386                 get_part_info $1
387                 p=$(($p-1))
388                 create_menulst $p
389                 create_winitem $1 $d
390                 rm -f /hd/boot/grub/stage1
391                 echo "(hd$d) /dev/$disk" > /hd/grub/device.map
392                 echo "setup (hd$d) (hd$d,$p)" | grub --device-map /hd/grub/device.map > /dev/tty5
393                 [ $? -ne 0 ] && return 255
394         fi
395
396         [ -n "$efi" ] && adialog --title " Confirm " --no-label Skip --defaultno --yesno \
397                 "\n Do you want to install EFI GRUB2?" 7 39
398         if [ $? -eq 0 ]; then
399                 for i in `list_disks`; do
400                         disk=`basename $i`
401                         esp=`sgdisk --print /dev/$disk 2> /dev/null | grep EF00 | awk '{print $1}'`
402                         [ -n "$esp" ] && boot=`find_partition $disk $esp` && break
403                 done
404                 if [ -z "$esp" ]; then
405                         get_part_info $1
406                         boot=`basename $(blkid /dev/$disk* | grep vfat | cut -d: -f1 | head -1)`
407                         [ -z "$boot" ] && boot=`find_partition $disk 1`
408                         esp=`cat /sys/block/$disk/$boot/partition`
409                 fi
410                 mkdir -p efi
411                 mountpoint -q efi && umount efi
412                 while [ 1 ]; do
413                         [ -n "$AUTO_INSTALL" -a -z "$AUTO_UPDATE" ] && mkdosfs -n EFI /dev/$boot
414                         try_mount rw /dev/$boot efi && break
415                         dialog --title " Confirm " --defaultno --yesno "\n Cannot mount /dev/$boot.\n Do you want to format it?" 8 37
416                         [ $? -eq 0 ] && mkdosfs -n EFI /dev/$boot
417                 done
418                 if [ "$efi" = "32" ]; then
419                         grubcfg=efi/boot/grub/i386-efi/grub.cfg
420                         bootefi=bootia32.efi
421                 else
422                         grubcfg=efi/boot/grub/x86_64-efi/grub.cfg
423                         bootefi=BOOTx64.EFI
424                 fi
425                 mkdir -p `dirname $grubcfg` efi/efi/Android
426                 cp -af grub2/efi/boot/* efi/efi/Android
427                 sed -i "s|VER|$VER|; s|CMDLINE|$cmdline|" efi/efi/Android/android.cfg
428                 [ -s efi/boot/grub/grubenv ] || ( printf %-1024s "# GRUB Environment Block%" | sed 's/k%/k\n/; s/   /###/g' > efi/boot/grub/grubenv )
429
430                 echo -e 'set timeout=5\nset debug_mode="(DEBUG mode)"' > $grubcfg
431                 # Our grub-efi doesn't support ntfs directly.
432                 # Copy boot files to ESP so grub-efi could read them
433                 if [ "$fs" = "fuseblk" ]; then
434                         cp -f src/kernel src/initrd.img efi/efi/Android
435                         echo -e "set kdir=/efi/Android\nset src=SRC=/$asrc" >> $grubcfg
436                 else
437                         echo -e "set kdir=/$asrc" >> $grubcfg
438                 fi
439                 echo -e '\nsource $cmdpath/android.cfg' >> $grubcfg
440
441                 # Checking for old EFI entries, removing them and adding new depending on bitness
442                 efibootmgr | grep -Eo ".{0,6}Android-x86" | cut -c1-4 > /tmp/efientries
443                 if [ -s /tmp/efientries ]; then
444                         set_answer_if_auto 1
445                         adialog --title " Question " --defaultno --yesno "\nEFI boot entries for previous Android-x86 installations were found.\n\nDo you wish to delete them?" 10 61
446                         [ $? -eq 0 ] && while read entry; do efibootmgr -Bb "$entry" > /dev/tty4 2>&1; done < /tmp/efientries
447                 fi
448                 efibootmgr -v -c -d /dev/$disk -p $esp -L "Android-x86 $VER" -l /efi/Android/$bootefi > /dev/tty4 2>&1
449         fi
450
451         set_answer_if_auto 0
452         adialog --title " Question " --defaultno --yesno \
453                 "\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
454         instal_rw=$?
455
456         files="mnt/$SRC/kernel mnt/$SRC/initrd.img mnt/$SRC/$RAMDISK"
457         if [ $instal_rw -eq 0 ]; then
458                 if [ "$fs" = "vfat" -o "$fs" = "fuseblk" ]; then
459                         [ -e /sfs/system.img ] && sysimg="/sfs/system.img" || sysimg="mnt/$SRC/system.*"
460                 else
461                         sysimg="android/system"
462                 fi
463         else
464                 sysimg="mnt/$SRC/system.*"
465         fi
466         files="$files $sysimg"
467         size=0
468         for s in `du -sk $files | awk '{print $1}'`; do
469                 size=$(($size+$s))
470         done
471         try_upgrade hd/$asrc
472         mkdir -p hd/$asrc
473         cd hd/$asrc
474         rm -rf system*
475         ( ( cd /; find $files | $CPIO -H newc -o ) | pv -ns ${size}k | ( $CPIO -iud > /dev/null; echo $? > /tmp/result )) 2>&1 \
476                 | progress_bar "Installing Android-x86 to $1" "Expect to write $size KB..."
477         result=$((`cat /tmp/result`*255))
478
479         if [ $result -eq 0 ]; then
480                 for d in android mnt sfs ./$SRC; do
481                         [ -d $d ] && mv $d/* . && rmdir $d
482                 done
483                 chown 0.0 *
484                 for f in *; do
485                         [ -d $f ] || chmod 644 $f
486                 done
487
488                 case "$fs" in
489                         vfat|fuseblk)
490                                 [ -e data.img ] && check_data_img || create_data_img
491                                 ;;
492                         *)
493                                 mkdir -p data
494                                 ;;
495                 esac
496         fi
497
498         dialog --infobox "\n Syncing to disk..." 5 27
499         sync
500
501         return $result
502 }
503
504 install_hd()
505 {
506         case "$AUTO_INSTALL" in
507                 [Uu]*)
508                         answer=`basename $(blkid | grep -v loop | sort | grep Android-x86 | cut -d: -f1 | head -1)`
509                         [ -z "$answer" ] && answer=`basename $(blkid | grep -v loop | sort | grep ext4 | cut -d: -f1 | head -1)`
510                         AUTO_UPDATE=$answer
511                         ;;
512                 *)
513                         [ -z "$answer" ] && set_answer_if_auto Create
514                         ;;
515         esac
516
517         select_dev || rebooting
518         retval=1
519         case "$choice" in
520                 Create*)
521                         partition_drive
522                         retval=$?
523                         ;;
524                 Detect*)
525                         dialog --title " Detecting... " --nocancel --pause "" 8 41 1
526                         ;;
527                 *)
528                         install_to $choice
529                         retval=$?
530                         ;;
531         esac
532         return $retval
533 }
534
535 do_install()
536 {
537         booted_from=`basename $dev`
538         efi=$(cat /sys/firmware/efi/fw_platform_size 2> /dev/null)
539         [ -n "$efi" ] && mount -t efivarfs none /sys/firmware/efi/efivars
540
541         until install_hd; do
542                 if [ $retval -eq 255 ]; then
543                         dialog --title ' Error! ' --yes-label Retry --no-label Reboot \
544                                 --yesno '\nInstallation failed! Please check if you have enough free disk space to install Android-x86.' 8 51
545                         [ $? -eq 1 ] && rebooting
546                 fi
547         done
548
549         [ -n "$VESA" ] || runit="Run Android-x86"
550         dialog --clear --title ' Congratulations! ' \
551                 --menu "\n Android-x86 is installed successfully.\n " 11 51 13 \
552                 "$runit" "" "Reboot" "" 2> $tempfile
553         case "`cat $tempfile`" in
554                 Run*)
555                         cd /android
556                         umount system
557                         mountpoint -q /sfs && umount /sfs
558                         if [ -e /hd/$asrc/system.sfs ]; then
559                                 mount -o loop /hd/$asrc/system.sfs /sfs
560                                 mount -o loop /sfs/system.img system
561                         elif [ -e /hd/$asrc/system.img ]; then
562                                 mount -o loop /hd/$asrc/system.img system
563                         else
564                                 mount --bind /hd/$asrc/system system
565                         fi
566                         if [ -d /hd/$asrc/data ]; then
567                                 mount --bind /hd/$asrc/data data
568                         elif [ -e /hd/$asrc/data.img ]; then
569                                 mount -o loop /hd/$asrc/data.img data
570                         fi
571                         ;;
572                 *)
573                         rebooting
574                         ;;
575         esac
576 }