blob: d79334bb48578740778b3c024f0b43527c21af4a [file] [log] [blame]
Tristan Muntsinger82c10502019-10-15 14:49:19 -07001#!/bin/bash
2
3# Copyright 2019 Google Inc. All rights reserved.
4
5# Licensed under the Apache License, Version 2.0 (the "License");
6# you may not use this file except in compliance with the License.
7# You may obtain a copy of the License at
8
9# http://www.apache.org/licenses/LICENSE-2.0
10
11# Unless required by applicable law or agreed to in writing, software
12# distributed under the License is distributed on an "AS IS" BASIS,
13# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14# See the License for the specific language governing permissions and
15# limitations under the License.
16
17source "${ANDROID_BUILD_TOP}/external/shflags/src/shflags"
18
Tristan Muntsinger34569112019-10-16 22:10:54 -070019DEFINE_boolean p1 \
20 false "Only generate/write the 1st partition (loader1)" "1"
21DEFINE_boolean p2 \
22 false "Only generate/write the 2nd partition (env)" "2"
23DEFINE_boolean p3 \
24 false "Only generate/write the 3rd partition (loader2)" "3"
25DEFINE_boolean p4 \
26 false "Only generate/write the 4th partition (trust)" "4"
27DEFINE_boolean p5 \
28 false "Only generate/write the 5th partition (rootfs)" "5"
29
30FLAGS_HELP="USAGE: $0 <KERNEL_DIR> [IMAGE] [flags]"
Tristan Muntsinger82c10502019-10-15 14:49:19 -070031
32FLAGS "$@" || exit $?
33eval set -- "${FLAGS_ARGV}"
34
Tristan Muntsinger34569112019-10-16 22:10:54 -070035if [ ${FLAGS_p1} -eq ${FLAGS_FALSE} ] &&
36 [ ${FLAGS_p2} -eq ${FLAGS_FALSE} ] &&
37 [ ${FLAGS_p3} -eq ${FLAGS_FALSE} ] &&
38 [ ${FLAGS_p4} -eq ${FLAGS_FALSE} ] &&
39 [ ${FLAGS_p5} -eq ${FLAGS_FALSE} ]; then
40 FLAGS_p1=${FLAGS_TRUE}
41 FLAGS_p2=${FLAGS_TRUE}
42 FLAGS_p3=${FLAGS_TRUE}
43 FLAGS_p4=${FLAGS_TRUE}
44 FLAGS_p5=${FLAGS_TRUE}
45fi
46
Tristan Muntsinger82c10502019-10-15 14:49:19 -070047for arg in "$@" ; do
48 if [ -z $KERNEL_DIR ]; then
49 KERNEL_DIR=$arg
50 elif [ -z $IMAGE ]; then
51 IMAGE=$arg
52 else
53 flags_help
54 exit 1
55 fi
56done
57
Tristan Muntsingere4eeded2019-10-15 20:26:15 -070058USE_IMAGE=`[ -z "${IMAGE}" ] && echo "0" || echo "1"`
Tristan Muntsinger1f28d1f2019-10-16 22:53:55 -070059OVERWRITE=`[ -e "${IMAGE}" ] && echo "1" || echo "0"`
60if [ -z $KERNEL_DIR ]; then
Tristan Muntsinger82c10502019-10-15 14:49:19 -070061 flags_help
62 exit 1
63fi
Tristan Muntsinger82c10502019-10-15 14:49:19 -070064if [ ! -e "${KERNEL_DIR}" ]; then
65 echo "error: can't find '${KERNEL_DIR}'. aborting..."
66 exit 1
67fi
68
69# escalate to superuser
70if [ $UID -ne 0 ]; then
71 cd ${ANDROID_BUILD_TOP}
72 . ./build/envsetup.sh
73 lunch ${TARGET_PRODUCT}-${TARGET_BUILD_VARIANT}
74 mmma external/u-boot
75 cd -
76 exec sudo -E "${0}" ${@}
77fi
78
Tristan Muntsinger1f28d1f2019-10-16 22:53:55 -070079if [ $OVERWRITE -eq 1 ]; then
80 OVERWRITE_IMAGE=${IMAGE}
81 IMAGE=`mktemp`
82fi
83
Tristan Muntsingere4eeded2019-10-15 20:26:15 -070084if [ $USE_IMAGE -eq 0 ]; then
85 init_devs=`lsblk --nodeps -oNAME -n`
86 echo "Reinsert device (to write to) into PC"
87 while true; do
88 devs=`lsblk --nodeps -oNAME -n`
89 new_devs="$(echo -e "${init_devs}\n${devs}" | sort | uniq -u | awk 'NF')"
90 num_devs=`echo "${new_devs}" | wc -l`
91 if [[ "${new_devs}" == "" ]]; then
92 num_devs=0
93 fi
94 if [[ ${num_devs} -gt 1 ]]; then
95 echo "error: too many new devices detected! aborting..."
96 exit 1
97 fi
98 if [[ ${num_devs} -eq 1 ]]; then
99 if [[ "${new_devs}" != "${mmc_dev}" ]]; then
100 if [[ "${mmc_dev}" != "" ]]; then
101 echo "error: block device name mismatch ${new_devs} != ${mmc_dev}"
102 echo "Reinsert device (to write to) into PC"
103 fi
104 mmc_dev=${new_devs}
105 continue
106 fi
107 echo "${init_devs}" | grep "${mmc_dev}" >/dev/null
108 if [[ $? -eq 0 ]]; then
109 init_devs="${devs}"
110 continue
111 fi
112 break
113 fi
114 done
115 # now inform the user
116 echo "Detected device at /dev/${mmc_dev}"
117fi
118
Tristan Muntsinger34569112019-10-16 22:10:54 -0700119if [ ${FLAGS_p1} -eq ${FLAGS_TRUE} ]; then
120 cd ${ANDROID_BUILD_TOP}/external/arm-trusted-firmware
121 CROSS_COMPILE=aarch64-linux-gnu- make PLAT=rk3399 DEBUG=0 ERROR_DEPRECATED=1 bl31
122 export BL31="${ANDROID_BUILD_TOP}/external/arm-trusted-firmware/build/rk3399/release/bl31/bl31.elf"
123 cd -
124fi
Tristan Muntsinger82c10502019-10-15 14:49:19 -0700125
Tristan Muntsinger82c10502019-10-15 14:49:19 -0700126cd ${ANDROID_BUILD_TOP}/external/u-boot
Tristan Muntsinger5f263952019-10-16 21:12:06 -0700127
Tristan Muntsinger34569112019-10-16 22:10:54 -0700128if [ ${FLAGS_p2} -eq ${FLAGS_TRUE} ]; then
129 tmpfile=`mktemp`
130 bootenv=`mktemp`
131 cat > ${tmpfile} << "EOF"
Tristan Muntsinger5f263952019-10-16 21:12:06 -0700132bootdelay=2
133baudrate=1500000
134scriptaddr=0x00500000
135boot_targets=mmc1 mmc0
136bootcmd=run distro_bootcmd
137distro_bootcmd=for target in ${boot_targets}; do run bootcmd_${target}; done
138bootcmd_mmc0=devnum=0; run mmc_boot
139bootcmd_mmc1=devnum=1; run mmc_boot
140mmc_boot=if mmc dev ${devnum}; then ; run scan_for_boot_part; fi
Tristan Muntsingerf06455a2019-10-31 16:08:19 -0700141scan_for_boot_part=part list mmc ${devnum} -bootable devplist; env exists devplist || setenv devplist 1; if test $devnum = 1; then script_type=init; else script_type=boot; fi; for distro_bootpart in ${devplist}; do if fstype mmc ${devnum}:${distro_bootpart} bootfstype; then run find_script; fi; done; setenv devplist; setenv script_type;
142find_script=if test -e mmc ${devnum}:${distro_bootpart} /boot/$script_type.scr; then echo Found U-Boot script /boot/$script_type.scr; run run_scr; fi
143run_scr=load mmc ${devnum}:${distro_bootpart} ${scriptaddr} /boot/$script_type.scr; source ${scriptaddr}
Tristan Muntsinger5f263952019-10-16 21:12:06 -0700144EOF
Tristan Muntsinger34569112019-10-16 22:10:54 -0700145 ${ANDROID_HOST_OUT}/bin/mkenvimage -s 32768 -o ${bootenv} - < ${tmpfile}
146fi
Tristan Muntsinger5f263952019-10-16 21:12:06 -0700147
Tristan Muntsinger34569112019-10-16 22:10:54 -0700148if [ ${FLAGS_p1} -eq ${FLAGS_TRUE} ] || [ ${FLAGS_p3} -eq ${FLAGS_TRUE} ]; then
149 make ARCH=arm CROSS_COMPILE=aarch64-linux-gnu- rock-pi-4-rk3399_defconfig
150 if [ ${FLAGS_p1} -eq ${FLAGS_TRUE} ]; then
151 make ARCH=arm CROSS_COMPILE=aarch64-linux-gnu- -j`nproc`
152 fi
153 if [ ${FLAGS_p3} -eq ${FLAGS_TRUE} ]; then
154 make ARCH=arm CROSS_COMPILE=aarch64-linux-gnu- u-boot.itb
155 fi
156 if [ ${FLAGS_p1} -eq ${FLAGS_TRUE} ]; then
157 idbloader=`mktemp`
158 ${ANDROID_HOST_OUT}/bin/mkimage -n rk3399 -T rksd -d tpl/u-boot-tpl.bin ${idbloader}
159 cat spl/u-boot-spl.bin >> ${idbloader}
160 fi
161fi
Tristan Muntsinger82c10502019-10-15 14:49:19 -0700162cd -
163
Tristan Muntsinger34569112019-10-16 22:10:54 -0700164if [ ${FLAGS_p5} -eq ${FLAGS_TRUE} ]; then
165 ${ANDROID_BUILD_TOP}/kernel/tests/net/test/build_rootfs.sh -a arm64 -s buster -n ${IMAGE}
166 if [ $? -ne 0 ]; then
167 echo "error: failed to build rootfs. exiting..."
168 exit 1
169 fi
170 truncate -s +3G ${IMAGE}
171 e2fsck -f ${IMAGE}
172 resize2fs ${IMAGE}
Tristan Muntsinger82c10502019-10-15 14:49:19 -0700173
Tristan Muntsinger34569112019-10-16 22:10:54 -0700174 mntdir=`mktemp -d`
175 mount ${IMAGE} ${mntdir}
176 if [ $? != 0 ]; then
177 echo "error: unable to mount ${IMAGE} ${mntdir}"
178 exit 1
179 fi
Tristan Muntsinger82c10502019-10-15 14:49:19 -0700180
Tristan Muntsinger34569112019-10-16 22:10:54 -0700181 cat > ${mntdir}/boot/init.cmd << "EOF"
Tristan Muntsinger5f263952019-10-16 21:12:06 -0700182mmc dev 1 0; mmc read 0x02080000 0x1fc0 0x40;
183ethaddr=$ethaddr
184env default -a
185setenv ethaddr $ethaddr
186setenv boot_targets 'mmc1 mmc0 usb0 pxe'
187saveenv
188mmc dev 1 0; mmc read 0x04000000 0x1fc0 0x40;
189mmc dev 0 0; mmc write 0x04000000 0x1fc0 0x40;
190mmc dev 1 0; mmc write 0x02080000 0x1fc0 0x40;
191EOF
Tristan Muntsinger34569112019-10-16 22:10:54 -0700192 ${ANDROID_BUILD_TOP}/external/u-boot/tools/mkimage \
Tristan Muntsinger5f263952019-10-16 21:12:06 -0700193 -C none -A arm -T script -d ${mntdir}/boot/init.cmd ${mntdir}/boot/init.scr
194
Tristan Muntsinger34569112019-10-16 22:10:54 -0700195 cat > ${mntdir}/boot/boot.cmd << "EOF"
Tristan Muntsingerbcdc07b2019-10-16 22:56:19 -0700196setenv bootcmd_dhcp '
197mw.b ${scriptaddr} 0 0x8000
198mmc dev 0 0
199mmc read ${scriptaddr} 0x1fc0 0x40
200env import -b ${scriptaddr} 0x8000
201mw.b ${scriptaddr} 0 0x8000
202if dhcp ${scriptaddr} manifest.txt; then
203 setenv OldSha ${Sha}
204 setenv Sha
205 env import -t ${scriptaddr} 0x8000 ManifestVersion
Tristan Muntsinger19c66ee2019-11-08 19:10:48 -0800206 echo "Manifest version $ManifestVersion";
Tristan Muntsingerbcdc07b2019-10-16 22:56:19 -0700207 if test "$ManifestVersion" = "1"; then
208 run manifest1
Tristan Muntsinger19c66ee2019-11-08 19:10:48 -0800209 elif test "$ManifestVersion" = "2"; then
210 run manifest2
Tristan Muntsingerbcdc07b2019-10-16 22:56:19 -0700211 else
212 run manifestX
213 fi
214fi'
215setenv manifestX 'echo "***** ERROR: Unknown manifest version! *****";'
216setenv manifest1 '
Tristan Muntsingerbcdc07b2019-10-16 22:56:19 -0700217env import -t ${scriptaddr} 0x8000
218if test "$Sha" != "$OldSha"; then
219 setenv serverip ${TftpServer}
220 setenv loadaddr 0x00200000
221 mmc dev 0 0;
222 file=$TplSplImg; offset=0x40; size=0x1f80; run tftpget1; setenv TplSplImg
223 file=$UbootItb; offset=0x4000; size=0x2000; run tftpget1; setenv UbootItb
224 file=$TrustImg; offset=0x6000; size=0x2000; run tftpget1; setenv TrustImg
225 file=$RootfsImg; offset=0x8000; size=0; run tftpget1; setenv RootfsImg
226 file=$UbootEnv; offset=0x1fc0; size=0x40; run tftpget1; setenv UbootEnv
227 mw.b ${scriptaddr} 0 0x8000
228 env export -b ${scriptaddr} 0x8000
229 mmc write ${scriptaddr} 0x1fc0 0x40
230else
231 echo "Already have ${Sha}. Booting..."
232fi'
Tristan Muntsinger19c66ee2019-11-08 19:10:48 -0800233setenv manifest2 '
234env import -t ${scriptaddr} 0x8000
235if test "$DFUethaddr" = "$ethaddr" || test "$DFUethaddr" = ""; then
236 if test "$Sha" != "$OldSha"; then
237 setenv serverip ${TftpServer}
238 setenv loadaddr 0x00200000
239 mmc dev 0 0;
240 file=$TplSplImg; offset=0x40; size=0x1f80; run tftpget1; setenv TplSplImg
241 file=$UbootItb; offset=0x4000; size=0x2000; run tftpget1; setenv UbootItb
242 file=$TrustImg; offset=0x6000; size=0x2000; run tftpget1; setenv TrustImg
243 file=$RootfsImg; offset=0x8000; size=0; run tftpget1; setenv RootfsImg
244 file=$UbootEnv; offset=0x1fc0; size=0x40; run tftpget1; setenv UbootEnv
245 mw.b ${scriptaddr} 0 0x8000
246 env export -b ${scriptaddr} 0x8000
247 mmc write ${scriptaddr} 0x1fc0 0x40
248 else
249 echo "Already have ${Sha}. Booting..."
250 fi
251else
252 echo "Update ${Sha} isn't for me. Booting..."
253fi'
Tristan Muntsingerbcdc07b2019-10-16 22:56:19 -0700254setenv tftpget1 "
255mw.b ${loadaddr} 0 0x400000
256&& tftp ${file}
257&& isGz=0 && setexpr isGz sub .*\\.gz\$ 1 ${file}
258&& if test $isGz = 1; then
259 setexpr boffset ${offset} * 0x200
260 && gzwrite mmc 0 ${loadaddr} 0x${filesize} 100000 ${boffset}
261 && echo Updated: ${bootfile}
262elif test ${file} = boot.env; then
263 env import -b ${loadaddr}
264 && echo Updated: boot.env
265else
266 && if test $size = 0; then
267 setexpr x $filesize - 1
268 && setexpr x $x / 0x1000
269 && setexpr x $x + 1
270 && setexpr x $x * 0x1000
271 && setexpr x $x / 0x200
272 && size=0x${x}
273 fi
274 && mmc write ${loadaddr} ${offset} ${size}
275 && echo Updated: ${bootfile}
276fi
277|| echo ** UPDATE FAILED: ${bootfile} **"
278if mmc dev 1 0; then; else
279 run bootcmd_dhcp;
280fi
Tristan Muntsinger82c10502019-10-15 14:49:19 -0700281load mmc ${devnum}:${distro_bootpart} 0x02080000 /boot/Image
282load mmc ${devnum}:${distro_bootpart} 0x04000000 /boot/uInitrd
283load mmc ${devnum}:${distro_bootpart} 0x01f00000 /boot/dtb/rockchip/rk3399-rock-pi-4.dtb
284setenv finduuid "part uuid mmc ${devnum}:${distro_bootpart} uuid"
285run finduuid
286setenv bootargs "earlycon=uart8250,mmio32,0xff1a0000 console=ttyS2,1500000n8 loglevel=7 root=PARTUUID=${uuid} rootwait rootfstype=ext4 sdhci.debug_quirks=0x20000000"
287booti 0x02080000 0x04000000 0x01f00000
288EOF
Tristan Muntsinger34569112019-10-16 22:10:54 -0700289 ${ANDROID_HOST_OUT}/bin/mkimage \
290 -C none -A arm -T script -d ${mntdir}/boot/boot.cmd ${mntdir}/boot/boot.scr
Tristan Muntsinger82c10502019-10-15 14:49:19 -0700291
Tristan Muntsinger34569112019-10-16 22:10:54 -0700292 cd ${KERNEL_DIR}
293 export PATH=${ANDROID_BUILD_TOP}/prebuilts/clang/host/linux-x86/clang-r353983c/bin:$PATH
294 export PATH=${ANDROID_BUILD_TOP}/prebuilts/gcc/linux-x86/aarch64/aarch64-linux-android-4.9/bin:$PATH
295 make ARCH=arm64 CC=clang CROSS_COMPILE=aarch64-linux-androidkernel- \
296 CLANG_TRIPLE=aarch64-linux-gnu- rockpi4_defconfig
297 make ARCH=arm64 CC=clang CROSS_COMPILE=aarch64-linux-androidkernel- \
298 CLANG_TRIPLE=aarch64-linux-gnu- -j`nproc`
Tristan Muntsinger82c10502019-10-15 14:49:19 -0700299
Tristan Muntsinger34569112019-10-16 22:10:54 -0700300 cp ${KERNEL_DIR}/arch/arm64/boot/Image ${mntdir}/boot/
301 mkdir -p ${mntdir}/boot/dtb/rockchip/
302 cp ${KERNEL_DIR}/arch/arm64/boot/dts/rockchip/rk3399-rock-pi-4.dtb ${mntdir}/boot/dtb/rockchip/
303 cd -
Tristan Muntsinger82c10502019-10-15 14:49:19 -0700304
Tristan Muntsinger34569112019-10-16 22:10:54 -0700305 mount -o bind /proc ${mntdir}/proc
306 mount -o bind /sys ${mntdir}/sys
307 mount -o bind /dev ${mntdir}/dev
Tristan Muntsinger82c10502019-10-15 14:49:19 -0700308
Tristan Muntsinger34569112019-10-16 22:10:54 -0700309 echo "Installing required packages..."
310 chroot ${mntdir} /bin/bash <<EOF
Tristan Muntsinger82c10502019-10-15 14:49:19 -0700311apt-get update
Tristan Muntsinger5ea7c6a2019-10-15 16:53:08 -0700312apt-get install -y -f initramfs-tools u-boot-tools network-manager openssh-server sudo man-db vim git dpkg-dev cdbs debhelper config-package-dev gdisk eject lzop binfmt-support ntpdate
Tristan Muntsinger82c10502019-10-15 14:49:19 -0700313EOF
314
Tristan Muntsinger34569112019-10-16 22:10:54 -0700315 echo "Turning on DHCP client..."
316 cat >${mntdir}/etc/systemd/network/dhcp.network <<EOF
Tristan Muntsinger82c10502019-10-15 14:49:19 -0700317[Match]
318Name=en*
319
320[Network]
321DHCP=yes
322EOF
323
Tristan Muntsinger34569112019-10-16 22:10:54 -0700324 chroot ${mntdir} /bin/bash << "EOT"
Tristan Muntsinger5ea7c6a2019-10-15 16:53:08 -0700325echo "Adding user vsoc-01 and groups..."
326useradd -m -G kvm,sudo -d /home/vsoc-01 --shell /bin/bash vsoc-01
327echo -e "cuttlefish\ncuttlefish" | passwd
328echo -e "cuttlefish\ncuttlefish" | passwd vsoc-01
329EOT
330
Tristan Muntsinger34569112019-10-16 22:10:54 -0700331 echo "Cloning android-cuttlefish..."
332 cd ${mntdir}/home/vsoc-01
333 git clone https://github.com/google/android-cuttlefish.git
334 cd -
Tristan Muntsinger5ea7c6a2019-10-15 16:53:08 -0700335
Tristan Muntsinger34569112019-10-16 22:10:54 -0700336 echo "Creating led script..."
337 cat > ${mntdir}/usr/local/bin/led << "EOF"
Tristan Muntsingerab89e8f2019-10-15 20:30:20 -0700338#!/bin/bash
339
340if [ "$1" == "--start" ]; then
Tristan Muntsinger34569112019-10-16 22:10:54 -0700341 echo 125 > /sys/class/gpio/export
342 echo out > /sys/class/gpio/gpio125/direction
343 chmod 666 /sys/class/gpio/gpio125/value
344 echo 0 > /sys/class/gpio/gpio125/value
345 exit 0
Tristan Muntsingerab89e8f2019-10-15 20:30:20 -0700346fi
347
348if [ "$1" == "--stop" ]; then
Tristan Muntsinger34569112019-10-16 22:10:54 -0700349 echo 0 > /sys/class/gpio/gpio125/value
350 echo 125 > /sys/class/gpio/unexport
351 exit 0
Tristan Muntsingerab89e8f2019-10-15 20:30:20 -0700352fi
353
354if [ ! -e /sys/class/gpio/gpio125/value ]; then
Tristan Muntsinger34569112019-10-16 22:10:54 -0700355 echo "error: led service not initialized"
356 exit 1
Tristan Muntsingerab89e8f2019-10-15 20:30:20 -0700357fi
358
359if [ "$1" == "0" ] || [ "$1" == "off" ] || [ "$1" == "OFF" ]; then
Tristan Muntsinger34569112019-10-16 22:10:54 -0700360 echo 0 > /sys/class/gpio/gpio125/value
361 exit 0
Tristan Muntsingerab89e8f2019-10-15 20:30:20 -0700362fi
363
364if [ "$1" == "1" ] || [ "$1" == "on" ] || [ "$1" == "ON" ]; then
Tristan Muntsinger34569112019-10-16 22:10:54 -0700365 echo 1 > /sys/class/gpio/gpio125/value
366 exit 0
Tristan Muntsingerab89e8f2019-10-15 20:30:20 -0700367fi
368
369echo "usage: led <0|1>"
370exit 1
371EOF
Tristan Muntsinger34569112019-10-16 22:10:54 -0700372 chown root:root ${mntdir}/usr/local/bin/led
373 chmod 755 ${mntdir}/usr/local/bin/led
Tristan Muntsingerab89e8f2019-10-15 20:30:20 -0700374
Tristan Muntsinger34569112019-10-16 22:10:54 -0700375 echo "Creating led service..."
376 cat > ${mntdir}/etc/systemd/system/led.service << EOF
Tristan Muntsingerab89e8f2019-10-15 20:30:20 -0700377[Unit]
378 Description=led service
379 ConditionPathExists=/usr/local/bin/led
380
381[Service]
382 Type=oneshot
383 ExecStart=/usr/local/bin/led --start
384 ExecStop=/usr/local/bin/led --stop
385 RemainAfterExit=true
386 StandardOutput=journal
387
388[Install]
389 WantedBy=multi-user.target
390EOF
391
Tristan Muntsinger34569112019-10-16 22:10:54 -0700392 echo "Creating SD duplicator script..."
393 cat > ${mntdir}/usr/local/bin/sd-dupe << "EOF"
Tristan Muntsingerab89e8f2019-10-15 20:30:20 -0700394#!/bin/bash
395led 0
396
397src_dev=mmcblk0
398dest_dev=mmcblk1
399part_num=p5
400
401if [ -e /dev/mmcblk0p5 ]; then
Tristan Muntsinger34569112019-10-16 22:10:54 -0700402 led 1
Tristan Muntsingerab89e8f2019-10-15 20:30:20 -0700403
Tristan Muntsinger34569112019-10-16 22:10:54 -0700404 sgdisk -Z -a1 /dev/${dest_dev}
405 sgdisk -a1 -n:1:64:8127 -t:1:8301 -c:1:loader1 /dev/${dest_dev}
406 sgdisk -a1 -n:2:8128:8191 -t:2:8301 -c:2:env /dev/${dest_dev}
407 sgdisk -a1 -n:3:16384:24575 -t:3:8301 -c:3:loader2 /dev/${dest_dev}
408 sgdisk -a1 -n:4:24576:32767 -t:4:8301 -c:4:trust /dev/${dest_dev}
409 sgdisk -a1 -n:5:32768:- -A:5:set:2 -t:5:8305 -c:5:rootfs /dev/${dest_dev}
Tristan Muntsingerab89e8f2019-10-15 20:30:20 -0700410
Tristan Muntsinger34569112019-10-16 22:10:54 -0700411 src_block_count=`tune2fs -l /dev/${src_dev}${part_num} | grep "Block count:" | sed 's/.*: *//'`
412 src_block_size=`tune2fs -l /dev/${src_dev}${part_num} | grep "Block size:" | sed 's/.*: *//'`
413 src_fs_size=$(( src_block_count*src_block_size ))
414 src_fs_size_m=$(( src_fs_size / 1024 / 1024 + 1 ))
Tristan Muntsingerab89e8f2019-10-15 20:30:20 -0700415
Tristan Muntsinger34569112019-10-16 22:10:54 -0700416 dd if=/dev/${src_dev}p1 of=/dev/${dest_dev}p1 conv=sync,noerror status=progress
417 dd if=/dev/${src_dev}p2 of=/dev/${dest_dev}p2 conv=sync,noerror status=progress
418 dd if=/dev/${src_dev}p3 of=/dev/${dest_dev}p3 conv=sync,noerror status=progress
419 dd if=/dev/${src_dev}p4 of=/dev/${dest_dev}p4 conv=sync,noerror status=progress
Tristan Muntsingerab89e8f2019-10-15 20:30:20 -0700420
Tristan Muntsinger34569112019-10-16 22:10:54 -0700421 echo "Writing ${src_fs_size_m} MB: /dev/${src_dev} -> /dev/${dest_dev}..."
422 dd if=/dev/${src_dev}${part_num} of=/dev/${dest_dev}${part_num} bs=1M conv=sync,noerror status=progress
Tristan Muntsingerab89e8f2019-10-15 20:30:20 -0700423
Tristan Muntsinger34569112019-10-16 22:10:54 -0700424 echo "Expanding /dev/${dest_dev}${part_num} filesystem..."
425 e2fsck -fy /dev/${dest_dev}${part_num}
426 resize2fs /dev/${dest_dev}${part_num}
427 tune2fs -O has_journal /dev/${dest_dev}${part_num}
428 e2fsck -fy /dev/${dest_dev}${part_num}
429 sync /dev/${dest_dev}
Tristan Muntsingerab89e8f2019-10-15 20:30:20 -0700430
Tristan Muntsinger34569112019-10-16 22:10:54 -0700431 echo "Cleaning up..."
432 mount /dev/${dest_dev}${part_num} /media
433 chroot /media /usr/local/bin/install-cleanup
Tristan Muntsingerab89e8f2019-10-15 20:30:20 -0700434
Tristan Muntsinger34569112019-10-16 22:10:54 -0700435 if [ $? == 0 ]; then
436 echo "Successfully copied Rock Pi image!"
437 while true; do
438 led 1; sleep 0.5
439 led 0; sleep 0.5
440 done
441 else
442 echo "Error while copying Rock Pi image"
443 while true; do
444 led 1; sleep 0.1
445 led 0; sleep 0.1
446 done
447 fi
Tristan Muntsingerab89e8f2019-10-15 20:30:20 -0700448else
Tristan Muntsinger34569112019-10-16 22:10:54 -0700449 echo "Expanding /dev/${dest_dev}${part_num} filesystem..."
450 e2fsck -fy /dev/${dest_dev}${part_num}
451 resize2fs /dev/${dest_dev}${part_num}
452 tune2fs -O has_journal /dev/${dest_dev}${part_num}
453 e2fsck -fy /dev/${dest_dev}${part_num}
454 sync /dev/${dest_dev}
Tristan Muntsingerab89e8f2019-10-15 20:30:20 -0700455
Tristan Muntsinger34569112019-10-16 22:10:54 -0700456 echo "Cleaning up..."
457 /usr/local/bin/install-cleanup
Tristan Muntsingerab89e8f2019-10-15 20:30:20 -0700458fi
459EOF
Tristan Muntsinger34569112019-10-16 22:10:54 -0700460 chmod +x ${mntdir}/usr/local/bin/sd-dupe
Tristan Muntsingerab89e8f2019-10-15 20:30:20 -0700461
Tristan Muntsinger34569112019-10-16 22:10:54 -0700462 echo "Creating SD duplicator service..."
463 cat > ${mntdir}/etc/systemd/system/sd-dupe.service << EOF
Tristan Muntsingerab89e8f2019-10-15 20:30:20 -0700464[Unit]
465 Description=Duplicate SD card rootfs to eMMC on Rock Pi
466 ConditionPathExists=/usr/local/bin/sd-dupe
467 After=led.service
468
469[Service]
470 Type=simple
471 ExecStart=/usr/local/bin/sd-dupe
472 TimeoutSec=0
473 StandardOutput=tty
474
475[Install]
476 WantedBy=multi-user.target
477EOF
478
Tristan Muntsinger34569112019-10-16 22:10:54 -0700479 echo "Creating cleanup script..."
480 cat > ${mntdir}/usr/local/bin/install-cleanup << "EOF"
Tristan Muntsingerd32ee2b2019-10-15 15:07:00 -0700481#!/bin/bash
Tristan Muntsinger5ea7c6a2019-10-15 16:53:08 -0700482echo "Installing cuttlefish-common package..."
483echo "nameserver 8.8.8.8" > /etc/resolv.conf
484MAC=`ip link | grep eth0 -A1 | grep ether | sed 's/.*\(..:..:..:..:..:..\) .*/\1/'`
485sed -i " 1 s/.*/& rockpi-${MAC}/" /etc/hosts
486sudo hostnamectl set-hostname "rockpi-${MAC}"
487
488dpkg --add-architecture amd64
489until ping -c1 ftp.debian.org; do sleep 1; done
490ntpdate time.google.com
491while true; do
492 apt-get -o Acquire::Check-Valid-Until=false update
493 if [ $? != 0 ]; then sleep 1; continue; fi
494 apt-get install -y -f libc6:amd64 qemu-user-static
495 if [ $? != 0 ]; then sleep 1; continue; fi
496 break
497done
498cd /home/vsoc-01/android-cuttlefish
499dpkg-buildpackage -d -uc -us
500apt-get install -y -f ../cuttlefish-common_*_arm64.deb
501apt-get clean
502usermod -aG cvdnetwork vsoc-01
Tristan Muntsinger5ea7c6a2019-10-15 16:53:08 -0700503chmod 660 /dev/vhost-vsock
504chown root:cvdnetwork /dev/vhost-vsock
505
Tristan Muntsingerd32ee2b2019-10-15 15:07:00 -0700506rm /etc/machine-id
507rm /var/lib/dbus/machine-id
508dbus-uuidgen --ensure
509systemd-machine-id-setup
510
Tristan Muntsingerab89e8f2019-10-15 20:30:20 -0700511systemctl disable sd-dupe
512rm /etc/systemd/system/sd-dupe.service
513rm /usr/local/bin/sd-dupe
Tristan Muntsingerd32ee2b2019-10-15 15:07:00 -0700514rm /usr/local/bin/install-cleanup
515EOF
Tristan Muntsinger34569112019-10-16 22:10:54 -0700516 chmod +x ${mntdir}/usr/local/bin/install-cleanup
Tristan Muntsingerd32ee2b2019-10-15 15:07:00 -0700517
Tristan Muntsinger34569112019-10-16 22:10:54 -0700518 chroot ${mntdir} /bin/bash << "EOT"
Tristan Muntsingerd32ee2b2019-10-15 15:07:00 -0700519echo "Enabling services..."
Tristan Muntsingerab89e8f2019-10-15 20:30:20 -0700520systemctl enable led
521systemctl enable sd-dupe
Tristan Muntsingerd32ee2b2019-10-15 15:07:00 -0700522
Tristan Muntsinger82c10502019-10-15 14:49:19 -0700523echo "Creating Initial Ramdisk..."
524update-initramfs -c -t -k "5.2.0"
525mkimage -A arm -O linux -T ramdisk -C none -a 0 -e 0 -n uInitrd -d /boot/initrd.img-5.2.0 /boot/uInitrd-5.2.0
526ln -s /boot/uInitrd-5.2.0 /boot/uInitrd
527EOT
528
Tristan Muntsinger34569112019-10-16 22:10:54 -0700529 umount ${mntdir}/sys
530 umount ${mntdir}/dev
531 umount ${mntdir}/proc
532 umount ${mntdir}
533
534 # Turn on journaling
535 tune2fs -O ^has_journal ${IMAGE}
536 e2fsck -fy ${IMAGE} >/dev/null 2>&1
537fi
Tristan Muntsinger82c10502019-10-15 14:49:19 -0700538
Tristan Muntsingere4eeded2019-10-15 20:26:15 -0700539if [ ${USE_IMAGE} -eq 0 ]; then
540 # 32GB eMMC size
Tristan Muntsinger34569112019-10-16 22:10:54 -0700541 end_sector=61071326
Tristan Muntsingere4eeded2019-10-15 20:26:15 -0700542 device=/dev/${mmc_dev}
543 devicep=${device}
544
545 sgdisk -Z -a1 ${device}
546 sgdisk -a1 -n:1:64:8127 -t:1:8301 -c:1:loader1 ${device}
547 sgdisk -a1 -n:2:8128:8191 -t:2:8301 -c:2:env ${device}
548 sgdisk -a1 -n:3:16384:24575 -t:3:8301 -c:3:loader2 ${device}
549 sgdisk -a1 -n:4:24576:32767 -t:4:8301 -c:4:trust ${device}
Tristan Muntsinger34569112019-10-16 22:10:54 -0700550 sgdisk -a1 -n:5:32768:${end_sector} -A:5:set:2 -t:5:8305 -c:5:rootfs ${device}
551 if [ ${FLAGS_p5} -eq ${FLAGS_TRUE} ]; then
552 dd if=${IMAGE} of=${devicep}5 bs=1M
553 resize2fs ${devicep}5 >/dev/null 2>&1
554 fi
Tristan Muntsingere4eeded2019-10-15 20:26:15 -0700555else
Tristan Muntsinger34569112019-10-16 22:10:54 -0700556 device=$(losetup -f)
557 devicep=${device}p
558 if [ ${FLAGS_p5} -eq ${FLAGS_FALSE} ]; then
559 fs_end=3G
560 end_sector=-
561 fi
562 if [ ${FLAGS_p5} -eq ${FLAGS_TRUE} ]; then
563 # Minimize rootfs filesystem
564 while true; do
565 out=`sudo resize2fs -M ${IMAGE} 2>&1`
566 if [[ $out =~ "Nothing to do" ]]; then
567 break
568 fi
569 done
570 # Minimize rootfs file size
571 block_count=`sudo tune2fs -l ${IMAGE} | grep "Block count:" | sed 's/.*: *//'`
572 block_size=`sudo tune2fs -l ${IMAGE} | grep "Block size:" | sed 's/.*: *//'`
573 sector_size=512
574 start_sector=32768
575 fs_size=$(( block_count*block_size ))
576 fs_sectors=$(( fs_size/sector_size ))
577 part_sectors=$(( ((fs_sectors-1)/2048+1)*2048 )) # 1MB-aligned
578 end_sector=$(( start_sector+part_sectors-1 ))
579 secondary_gpt_sectors=33
580 fs_end=$(( (end_sector+secondary_gpt_sectors+1)*sector_size ))
581 image_size=$(( part_sectors*sector_size ))
582 truncate -s ${image_size} ${IMAGE}
583 e2fsck -fy ${IMAGE} >/dev/null 2>&1
584 fi
Tristan Muntsinger82c10502019-10-15 14:49:19 -0700585
Tristan Muntsingere4eeded2019-10-15 20:26:15 -0700586 # Create final image
Tristan Muntsinger1f28d1f2019-10-16 22:53:55 -0700587 if [ $OVERWRITE -eq 1 ]; then
588 tmpimg=${OVERWRITE_IMAGE}
589 else
590 tmpimg=`mktemp`
591 fi
Tristan Muntsingere4eeded2019-10-15 20:26:15 -0700592 truncate -s ${fs_end} ${tmpimg}
Tristan Muntsinger82c10502019-10-15 14:49:19 -0700593
Tristan Muntsingere4eeded2019-10-15 20:26:15 -0700594 # Create GPT
595 sgdisk -Z -a1 ${tmpimg}
596 sgdisk -a1 -n:1:64:8127 -t:1:8301 -c:1:loader1 ${tmpimg}
597 sgdisk -a1 -n:2:8128:8191 -t:2:8301 -c:2:env ${tmpimg}
598 sgdisk -a1 -n:3:16384:24575 -t:3:8301 -c:3:loader2 ${tmpimg}
599 sgdisk -a1 -n:4:24576:32767 -t:4:8301 -c:4:trust ${tmpimg}
600 sgdisk -a1 -n:5:32768:${end_sector} -A:5:set:2 -t:5:8305 -c:5:rootfs ${tmpimg}
Tristan Muntsinger82c10502019-10-15 14:49:19 -0700601
Tristan Muntsingere4eeded2019-10-15 20:26:15 -0700602 losetup ${device} ${tmpimg}
603 partx -v --add ${device}
Tristan Muntsinger82c10502019-10-15 14:49:19 -0700604
Tristan Muntsinger34569112019-10-16 22:10:54 -0700605 if [ ${FLAGS_p5} -eq ${FLAGS_TRUE} ]; then
606 dd if=${IMAGE} of=${devicep}5 bs=1M
607 fi
Tristan Muntsingere4eeded2019-10-15 20:26:15 -0700608fi
Tristan Muntsinger34569112019-10-16 22:10:54 -0700609if [ ${FLAGS_p1} -eq ${FLAGS_TRUE} ]; then
610 dd if=${idbloader} of=${devicep}1
611fi
612if [ ${FLAGS_p2} -eq ${FLAGS_TRUE} ]; then
613 dd if=${bootenv} of=${devicep}2
614fi
615if [ ${FLAGS_p3} -eq ${FLAGS_TRUE} ]; then
616 dd if=${ANDROID_BUILD_TOP}/external/u-boot/u-boot.itb of=${devicep}3
617fi
Tristan Muntsingere4eeded2019-10-15 20:26:15 -0700618if [ ${USE_IMAGE} -eq 1 ]; then
619 chown $SUDO_USER:`id -ng $SUDO_USER` ${tmpimg}
Tristan Muntsinger1f28d1f2019-10-16 22:53:55 -0700620 if [ $OVERWRITE -eq 0 ]; then
621 mv ${tmpimg} ${IMAGE}
622 fi
Tristan Muntsingere4eeded2019-10-15 20:26:15 -0700623 partx -v --delete ${device}
624 losetup -d ${device}
625fi