Files
anykernel3-rhode/tools/ak3-core.sh
T

999 lines
34 KiB
Bash
Executable File

### AnyKernel methods (DO NOT CHANGE)
## osm0sis @ xda-developers
[ "$OUTFD" ] || OUTFD=$1;
# set up working directory variables
[ "$AKHOME" ] || AKHOME=$PWD;
BOOTIMG=$AKHOME/boot.img;
BIN=$AKHOME/tools;
PATCH=$AKHOME/patch;
RAMDISK=$AKHOME/ramdisk;
SPLITIMG=$AKHOME/split_img;
VENDORRD=$AKHOME/vendor_ramdisk;
### output/testing functions:
# ui_print "<text>" [...]
ui_print() {
until [ ! "$1" ]; do
echo "ui_print $1
ui_print" >> /proc/self/fd/$OUTFD;
shift;
done;
}
# abort ["<text>" [...]]
abort() {
ui_print " " "$@";
exit 1;
}
# contains <string> <substring>
contains() {
[ "${1#*$2}" != "$1" ];
}
# file_getprop <file> <property>
file_getprop() {
grep "^$2=" "$1" | tail -n1 | cut -d= -f2-;
}
###
### file/directory attributes functions:
# set_perm <owner> <group> <mode> <file> [<file2> ...]
set_perm() {
local uid gid mod;
uid=$1; gid=$2; mod=$3;
shift 3;
chown $uid:$gid "$@" || chown $uid.$gid "$@";
chmod $mod "$@";
}
# set_perm_recursive <owner> <group> <dir_mode> <file_mode> <dir> [<dir2> ...]
set_perm_recursive() {
local uid gid dmod fmod;
uid=$1; gid=$2; dmod=$3; fmod=$4;
shift 4;
while [ "$1" ]; do
chown -R $uid:$gid "$1" || chown -R $uid.$gid "$1";
find "$1" -type d -exec chmod $dmod {} +;
find "$1" -type f -exec chmod $fmod {} +;
shift;
done;
}
###
### dump_boot functions:
# split_boot (dump and split image only)
split_boot() {
local splitfail;
if [ ! -e "$(echo "$BLOCK" | cut -d\ -f1)" ]; then
abort "Invalid partition. Aborting...";
fi;
if echo "$BLOCK" | grep -q ' '; then
BLOCK=$(echo "$BLOCK" | cut -d\ -f1);
CUSTOMDD=$(echo "$BLOCK" | cut -d\ -f2-);
elif [ ! "$CUSTOMDD" ]; then
CUSTOMDD="bs=1048576";
fi;
if [ -f "$BIN/nanddump" ]; then
nanddump -f $BOOTIMG $BLOCK;
else
dd if=$BLOCK of=$BOOTIMG $CUSTOMDD;
fi;
if [ $? != 0 ]; then
abort "Dumping image failed. Aborting...";
fi;
mkdir -p $SPLITIMG;
cd $SPLITIMG;
if [ -f "$BIN/unpackelf" ] && unpackelf -i $BOOTIMG -h -q 2>/dev/null; then
if [ -f "$BIN/elftool" ]; then
mkdir elftool_out;
elftool unpack -i $BOOTIMG -o elftool_out;
fi;
unpackelf -i $BOOTIMG;
[ $? != 0 ] && splitfail=1;
mv -f boot.img-kernel kernel.gz;
mv -f boot.img-ramdisk ramdisk.cpio.gz;
mv -f boot.img-cmdline cmdline.txt 2>/dev/null;
if [ -f boot.img-dt -a ! -f "$BIN/elftool" ]; then
case $(od -ta -An -N4 boot.img-dt | sed -e 's/ del//' -e 's/ //g') in
QCDT|ELF) mv -f boot.img-dt dt;;
*)
gzip -c kernel.gz > kernel.gz-dtb;
cat boot.img-dt >> kernel.gz-dtb;
rm -f boot.img-dt kernel.gz;
;;
esac;
fi;
elif [ -f "$BIN/mboot" ]; then
mboot -u -f $BOOTIMG;
elif [ -f "$BIN/dumpimage" ]; then
dd bs=$(($(printf '%d\n' 0x$(hexdump -n 4 -s 12 -e '16/1 "%02x""\n"' $BOOTIMG)) + 64)) count=1 conv=notrunc if=$BOOTIMG of=boot-trimmed.img;
dumpimage -l boot-trimmed.img > header;
grep "Name:" header | cut -c15- > boot.img-name;
grep "Type:" header | cut -c15- | cut -d\ -f1 > boot.img-arch;
grep "Type:" header | cut -c15- | cut -d\ -f2 > boot.img-os;
grep "Type:" header | cut -c15- | cut -d\ -f3 | cut -d- -f1 > boot.img-type;
grep "Type:" header | cut -d\( -f2 | cut -d\) -f1 | cut -d\ -f1 | cut -d- -f1 > boot.img-comp;
grep "Address:" header | cut -c15- > boot.img-addr;
grep "Point:" header | cut -c15- > boot.img-ep;
dumpimage -p 0 -o kernel.gz boot-trimmed.img;
[ $? != 0 ] && splitfail=1;
case $(cat boot.img-type) in
Multi) dumpimage -p 1 -o ramdisk.cpio.gz boot-trimmed.img;;
RAMDisk) mv -f kernel.gz ramdisk.cpio.gz;;
esac;
elif [ -f "$BIN/rkcrc" ]; then
dd bs=4096 skip=8 iflag=skip_bytes conv=notrunc if=$BOOTIMG of=ramdisk.cpio.gz;
else
(set -o pipefail; magiskboot unpack -h $BOOTIMG 2>&1 | tee infotmp >&2);
case $? in
1) splitfail=1;;
2) touch chromeos;;
esac;
fi;
if [ $? != 0 -o "$splitfail" ]; then
abort "Splitting image failed. Aborting...";
fi;
cd $AKHOME;
}
# unpack_ramdisk (extract ramdisk only)
unpack_ramdisk() {
local comp cpio vndrname;
cd $SPLITIMG;
if [ -f ramdisk.cpio.gz ]; then
if [ -f "$BIN/mkmtkhdr" ]; then
mv -f ramdisk.cpio.gz ramdisk.cpio.gz-mtk;
dd bs=512 skip=1 conv=notrunc if=ramdisk.cpio.gz-mtk of=ramdisk.cpio.gz;
fi;
mv -f ramdisk.cpio.gz ramdisk.cpio;
fi;
if [ -f ramdisk.cpio ]; then
[ -d $RAMDISK ] && mv -f $RAMDISK $AKHOME/rdtmp;
comp=$(magiskboot decompress ramdisk.cpio 2>&1 | grep -v 'raw' | sed -n 's;.*\[\(.*\)\];\1;p');
if [ "$comp" ]; then
mv -f ramdisk.cpio ramdisk.cpio.$comp;
magiskboot decompress ramdisk.cpio.$comp ramdisk.cpio;
if [ $? != 0 ] && $comp --help 2>/dev/null; then
echo "Attempting ramdisk unpack with busybox $comp..." >&2;
$comp -dc ramdisk.cpio.$comp > ramdisk.cpio;
fi;
fi;
mkdir -p $RAMDISK;
chmod 755 $RAMDISK;
cd $RAMDISK;
EXTRACT_UNSAFE_SYMLINKS=1 cpio -d -F $SPLITIMG/ramdisk.cpio -i;
if [ $? != 0 -o ! "$(ls)" ]; then
abort "Unpacking ramdisk failed. Aborting...";
fi;
if [ -d "$AKHOME/rdtmp" ]; then
cp -af $AKHOME/rdtmp/* .;
fi;
elif [ -d vendor_ramdisk ]; then
[ -d $VENDORRD ] && mv -f $VENDORRD $AKHOME/vrdtmp;
for cpio in vendor_ramdisk/*.cpio; do
comp=$(magiskboot decompress $cpio 2>&1 | grep -v 'raw' | sed -n 's;.*\[\(.*\)\];\1;p');
if [ "$comp" ]; then
mv -f $cpio $cpio.$comp;
magiskboot decompress $cpio.$comp $cpio;
if [ $? != 0 ] && $comp --help 2>/dev/null; then
echo "Attempting $cpio unpack with busybox $comp..." >&2;
$comp -dc $cpio.$comp > $cpio;
fi;
fi;
vndrname=$(basename $cpio .cpio);
mkdir -p $VENDORRD/$vndrname;
chmod 755 $VENDORRD/$vndrname;
cd $VENDORRD/$vndrname;
EXTRACT_UNSAFE_SYMLINKS=1 cpio -d -F $SPLITIMG/$cpio -i;
if [ $? != 0 -o ! "$(ls)" ]; then
abort "Unpacking vendor ramdisk \"$vndrname\" failed. Aborting...";
fi;
if [ -d "$AKHOME/vrdtmp/$vndrname" ]; then
cp -af $AKHOME/vrdtmp/$vndrname/* .;
fi;
done;
cd $VENDORRD;
else
abort "No ramdisk found to unpack. Aborting...";
fi;
}
### dump_boot (dump and split image, then extract ramdisk)
dump_boot() {
split_boot;
unpack_ramdisk;
}
###
### write_boot functions:
# repack_ramdisk (repack ramdisk only)
repack_ramdisk() {
local comp packfail vndrname cpio mtktype;
cd $AKHOME;
if [ "$RAMDISK_COMPRESSION" != "auto" ] && [ "$(grep HEADER_VER $SPLITIMG/infotmp | sed -n 's;.*\[\(.*\)\];\1;p')" -gt 3 ]; then
ui_print " " "Warning: Only lz4-l ramdisk compression is allowed with hdr v4+ images. Resetting to auto...";
RAMDISK_COMPRESSION=auto;
fi;
case $RAMDISK_COMPRESSION in
auto|"") comp=$(ls $SPLITIMG/ramdisk.cpio.* $SPLITIMG/vendor_ramdisk/ramdisk.cpio.* 2>/dev/null | tail -n1 | grep -v 'mtk' | rev | cut -d. -f1 | rev);;
none|cpio) comp="";;
gz) comp=gzip;;
lzo) comp=lzop;;
bz2) comp=bzip2;;
lz4-l) comp=lz4_legacy;;
*) comp=$RAMDISK_COMPRESSION;;
esac;
if [ -f $SPLITIMG/ramdisk.cpio ]; then
if [ -f "$BIN/mkbootfs" ]; then
mkbootfs $RAMDISK > ramdisk-new.cpio;
else
cd $RAMDISK;
find . | cpio -H newc -o > $AKHOME/ramdisk-new.cpio;
fi;
[ $? != 0 ] && packfail=1;
elif [ -d $SPLITIMG/vendor_ramdisk ]; then
cd $VENDORRD;
for vndrname in *; do
if [ -f "$BIN/mkbootfs" ]; then
mkbootfs $vndrname > $AKHOME/$vndrname-new.cpio;
else
cd $VENDORRD/$vndrname;
find . | cpio -H newc -o > $AKHOME/$vndrname-new.cpio;
fi;
[ $? != 0 ] && packfail=1;
done;
fi;
cd $AKHOME;
if [ ! "$NO_MAGISK_CHECK" ]; then
magiskboot cpio ramdisk-new.cpio test;
magisk_patched=$?;
fi;
[ "$magisk_patched" == 1 ] && magiskboot cpio ramdisk-new.cpio "extract .backup/.magisk $SPLITIMG/.magisk";
if [ "$comp" ]; then
for cpio in *-new.cpio; do
magiskboot compress=$comp $cpio;
if [ $? != 0 ] && $comp --help 2>/dev/null; then
echo "Attempting $cpio repack with busybox $comp..." >&2;
$comp -9c $cpio > $cpio.$comp;
[ $? != 0 ] && packfail=1;
rm -f $cpio;
fi;
done;
fi;
if [ "$packfail" ]; then
abort "Repacking ramdisk failed. Aborting...";
fi;
if [ -f "$BIN/mkmtkhdr" -a -f "$SPLITIMG/boot.img-base" ]; then
mtktype=$(od -ta -An -N8 -j8 $SPLITIMG/ramdisk.cpio.gz-mtk | sed -e 's/ nul//g' -e 's/ //g' | tr '[:upper:]' '[:lower:]');
case $mtktype in
rootfs|recovery) mkmtkhdr --$mtktype ramdisk-new.cpio*;;
esac;
fi;
}
# flash_boot (build, sign and write image only)
flash_boot() {
local varlist i kernel ramdisk fdt cmdline comp part0 part1 vndrname needskernelpatch nocompflag signfail pk8 cert avbtype;
cd $SPLITIMG;
if [ -f "$BIN/mkimage" ]; then
varlist="name arch os type comp addr ep";
elif [ -f "$BIN/mkbootimg" -a -f "$BIN/unpackelf" -a -f boot.img-base ]; then
mv -f cmdline.txt boot.img-cmdline 2>/dev/null;
varlist="cmdline base pagesize kernel_offset ramdisk_offset tags_offset";
fi;
for i in $varlist; do
if [ -f boot.img-$i ]; then
eval local $i=\"$(cat boot.img-$i)\";
fi;
done;
cd $AKHOME;
for i in zImage zImage-dtb Image Image-dtb Image.gz Image.gz-dtb Image.bz2 Image.bz2-dtb Image.lzo Image.lzo-dtb Image.lzma Image.lzma-dtb Image.xz Image.xz-dtb Image.lz4 Image.lz4-dtb Image.fit; do
if [ -f $i ]; then
kernel=$AKHOME/$i;
break;
fi;
done;
if [ "$kernel" ]; then
if [ -f "$BIN/mkmtkhdr" -a -f "$SPLITIMG/boot.img-base" ]; then
mkmtkhdr --kernel $kernel;
kernel=$kernel-mtk;
fi;
elif [ "$(ls $SPLITIMG/kernel* 2>/dev/null)" ]; then
kernel=$(ls $SPLITIMG/kernel* | grep -v 'kernel_dtb' | tail -n1);
fi;
if [ "$(ls ramdisk-new.cpio* 2>/dev/null)" ]; then
ramdisk=$AKHOME/$(ls ramdisk-new.cpio* | tail -n1);
elif [ -f "$BIN/mkmtkhdr" -a -f "$SPLITIMG/boot.img-base" ]; then
ramdisk=$SPLITIMG/ramdisk.cpio.gz-mtk;
else
ramdisk=$(ls $SPLITIMG/ramdisk.cpio* 2>/dev/null | tail -n1);
fi;
for fdt in dt recovery_dtbo dtb; do
for i in $AKHOME/$fdt $AKHOME/$fdt.img $SPLITIMG/$fdt; do
if [ -f $i ]; then
eval local $fdt=$i;
break;
fi;
done;
done;
cd $SPLITIMG;
if [ -f "$BIN/mkimage" ]; then
[ "$comp" == "uncompressed" ] && comp=none;
part0=$kernel;
case $type in
Multi) part1=":$ramdisk";;
RAMDisk) part0=$ramdisk;;
esac;
mkimage -A $arch -O $os -T $type -C $comp -a $addr -e $ep -n "$name" -d $part0$part1 $AKHOME/boot-new.img;
elif [ -f "$BIN/elftool" ]; then
[ "$dt" ] && dt="$dt,rpm";
[ -f cmdline.txt ] && cmdline="cmdline.txt@cmdline";
elftool pack -o $AKHOME/boot-new.img header=elftool_out/header $kernel $ramdisk,ramdisk $dt $cmdline;
elif [ -f "$BIN/mboot" ]; then
cp -f $kernel kernel;
cp -f $ramdisk ramdisk.cpio.gz;
mboot -d $SPLITIMG -f $AKHOME/boot-new.img;
elif [ -f "$BIN/rkcrc" ]; then
rkcrc -k $ramdisk $AKHOME/boot-new.img;
elif [ -f "$BIN/mkbootimg" -a -f "$BIN/unpackelf" -a -f boot.img-base ]; then
[ "$dt" ] && dt="--dt $dt";
mkbootimg --kernel $kernel --ramdisk $ramdisk --cmdline "$cmdline" --base $base --pagesize $pagesize --kernel_offset $kernel_offset --ramdisk_offset $ramdisk_offset --tags_offset "$tags_offset" $dt --output $AKHOME/boot-new.img;
else
[ "$kernel" ] && cp -f $kernel kernel;
[ "$ramdisk" ] && cp -f $ramdisk ramdisk.cpio;
if [ -d vendor_ramdisk ]; then
for i in vendor_ramdisk/*.cpio; do
vndrname=$(basename $i .cpio);
cp -f $AKHOME/$vndrname-new.cpio vendor_ramdisk/$vndrname.cpio;
done;
fi;
[ "$dt" -a -f extra ] && cp -f $dt extra;
for i in dtb recovery_dtbo; do
[ "$(eval echo \$$i)" -a -f $i ] && cp -f $(eval echo \$$i) $i;
done;
case $kernel in
*Image*)
if [ ! "$magisk_patched" -a ! "$NO_MAGISK_CHECK" ]; then
magiskboot cpio ramdisk.cpio test;
magisk_patched=$?;
fi;
if [ "$magisk_patched" == 1 ]; then
ui_print " " "Magisk detected! Patching kernel so reflashing Magisk is not necessary...";
comp=$(magiskboot decompress kernel 2>&1 | grep -vE 'raw|zimage' | sed -n 's;.*\[\(.*\)\];\1;p');
(magiskboot split $kernel || magiskboot decompress $kernel kernel) >&2;
if [ $? != 0 -a "$comp" ] && $comp --help 2>/dev/null; then
echo "Attempting kernel unpack with busybox $comp..." >&2;
$comp -dc $kernel > kernel;
fi;
# legacy SAR kernel string skip_initramfs -> want_initramfs
magiskboot hexpatch kernel 736B69705F696E697472616D6673 77616E745F696E697472616D6673 && needskernelpatch=1;
if [ "$(file_getprop $AKHOME/anykernel.sh do.modules)" == 1 ] && [ "$(file_getprop $AKHOME/anykernel.sh do.systemless)" == 1 ]; then
strings kernel 2>/dev/null | grep -E -m1 'Linux version.*#' > $AKHOME/vertmp;
fi;
if [ "$needskernelpatch" ]; then
if [ "$comp" ]; then
magiskboot compress=$comp kernel kernel.$comp;
if [ $? != 0 ] && $comp --help 2>/dev/null; then
echo "Attempting kernel repack with busybox $comp..." >&2;
$comp -9c kernel > kernel.$comp;
fi;
mv -f kernel.$comp kernel;
fi;
else
echo "Restoring untouched new kernel since no patching required..." >&2;
(magiskboot split -n $kernel || cp -f $kernel kernel) >&2;
fi;
[ ! -f .magisk ] && magiskboot cpio ramdisk.cpio "extract .backup/.magisk .magisk";
export $(cat .magisk);
for fdt in dtb extra kernel_dtb recovery_dtbo; do
[ -f $fdt ] && magiskboot dtb $fdt patch; # remove dtb verity/avb
done;
elif [ -d /data/data/me.weishu.kernelsu ] && [ "$(file_getprop $AKHOME/anykernel.sh do.modules)" == 1 ] && [ "$(file_getprop $AKHOME/anykernel.sh do.systemless)" == 1 ]; then
ui_print " " "KernelSU detected! Setting up for kernel helper module...";
comp=$(magiskboot decompress kernel 2>&1 | grep -vE 'raw|zimage' | sed -n 's;.*\[\(.*\)\];\1;p');
(magiskboot split $kernel || magiskboot decompress $kernel kernel) >&2;
if [ $? != 0 -a "$comp" ] && $comp --help 2>/dev/null; then
echo "Attempting kernel unpack with busybox $comp..." >&2;
$comp -dc $kernel > kernel;
fi;
strings kernel > stringstmp 2>/dev/null;
if grep -q -E '^/data/adb/ksud$' stringstmp; then
touch $AKHOME/kernelsu_patched;
grep -E -m1 'Linux version.*#' stringstmp > $AKHOME/vertmp;
[ -d $RAMDISK/overlay.d ] && ui_print " " "Warning: overlay.d detected in ramdisk but not currently supported by KernelSU!";
else
ui_print " " "Warning: No KernelSU support detected in kernel!";
fi;
rm -f stringstmp;
if [ "$comp" ]; then
magiskboot compress=$comp kernel kernel.$comp;
if [ $? != 0 ] && $comp --help 2>/dev/null; then
echo "Attempting kernel repack with busybox $comp..." >&2;
$comp -9c kernel > kernel.$comp;
fi;
mv -f kernel.$comp kernel;
fi;
else
case $kernel in
*-dtb) rm -f kernel_dtb;;
esac;
fi;
unset magisk_patched KEEPVERITY KEEPFORCEENCRYPT RECOVERYMODE VENDORBOOT PREINITDEVICE SHA1 RANDOMSEED; # leave PATCHVBMETAFLAG set for repack
;;
esac;
case $RAMDISK_COMPRESSION in
none|cpio) nocompflag="-n";;
esac;
case $PATCH_VBMETA_FLAG in
auto|"") [ "$PATCHVBMETAFLAG" ] || export PATCHVBMETAFLAG=false;;
1) export PATCHVBMETAFLAG=true;;
*) export PATCHVBMETAFLAG=false;;
esac;
magiskboot repack $nocompflag $BOOTIMG $AKHOME/boot-new.img;
fi;
if [ $? != 0 ]; then
abort "Repacking image failed. Aborting...";
fi;
[ "$PATCHVBMETAFLAG" ] && unset PATCHVBMETAFLAG;
[ -f .magisk ] && touch $AKHOME/magisk_patched;
cd $AKHOME;
if [ -f "$BIN/futility" -a -d "$BIN/chromeos" ]; then
if [ -f "$SPLITIMG/chromeos" ]; then
echo "Signing with CHROMEOS..." >&2;
futility vbutil_kernel --pack boot-new-signed.img --keyblock $BIN/chromeos/kernel.keyblock --signprivate $BIN/chromeos/kernel_data_key.vbprivk --version 1 --vmlinuz boot-new.img --bootloader $BIN/chromeos/empty --config $BIN/chromeos/empty --arch arm --flags 0x1;
fi;
[ $? != 0 ] && signfail=1;
fi;
if [ -d "$BIN/avb" ]; then
pk8=$(ls $BIN/avb/*.pk8);
cert=$(ls $BIN/avb/*.x509.*);
case $BLOCK in
*recovery*|*RECOVERY*|*SOS*) avbtype=recovery;;
*) avbtype=boot;;
esac;
if [ -f "$BIN/boot_signer-dexed.jar" ]; then
if [ -f /system/bin/dalvikvm ] && [ "$(/system/bin/dalvikvm -Xnoimage-dex2oat -cp $BIN/boot_signer-dexed.jar com.android.verity.BootSignature -verify boot.img 2>&1 | grep VALID)" ]; then
echo "Signing with AVBv1 /$avbtype..." >&2;
/system/bin/dalvikvm -Xnoimage-dex2oat -cp $BIN/boot_signer-dexed.jar com.android.verity.BootSignature /$avbtype boot-new.img $pk8 $cert boot-new-signed.img;
fi;
else
if magiskboot verify boot.img; then
echo "Signing with AVBv1 /$avbtype..." >&2;
magiskboot sign /$avbtype boot-new.img $cert $pk8;
fi;
fi;
fi;
if [ $? != 0 -o "$signfail" ]; then
abort "Signing image failed. Aborting...";
fi;
mv -f boot-new-signed.img boot-new.img 2>/dev/null;
if [ ! -f boot-new.img ]; then
abort "No repacked image found to flash. Aborting...";
elif [ "$(wc -c < boot-new.img)" -gt "$(wc -c < boot.img)" ]; then
abort "New image larger than target partition. Aborting...";
fi;
blockdev --setrw $BLOCK 2>/dev/null;
if [ -f "$BIN/flash_erase" -a -f "$BIN/nandwrite" ]; then
flash_erase $BLOCK 0 0;
nandwrite -p $BLOCK boot-new.img;
elif [ "$CUSTOMDD" ]; then
dd if=/dev/zero of=$BLOCK $CUSTOMDD 2>/dev/null;
dd if=boot-new.img of=$BLOCK $CUSTOMDD;
else
cat boot-new.img /dev/zero > $BLOCK 2>/dev/null || true;
fi;
if [ $? != 0 ]; then
abort "Flashing image failed. Aborting...";
fi;
}
# flash_generic <name>
flash_generic() {
local avb avbblock avbpath file flags img imgblock imgsz isro isunmounted path;
cd $AKHOME;
for file in $1 $1.img; do
if [ -f $file ]; then
img=$file;
break;
fi;
done;
if [ "$img" -a ! -f ${1}_flashed ]; then
for path in /dev/block/mapper /dev/block/by-name /dev/block/bootdevice/by-name; do
for file in $1 $1$SLOT; do
if [ -e $path/$file ]; then
imgblock=$path/$file;
break 2;
fi;
done;
done;
if [ ! "$imgblock" ]; then
abort "$1 partition could not be found. Aborting...";
fi;
if [ ! "$NO_BLOCK_DISPLAY" ]; then
ui_print " " "$imgblock";
fi;
if [ "$path" == "/dev/block/mapper" ]; then
avb=$(httools_static avb $1);
[ $? == 0 ] || abort "Failed to parse fstab entry for $1. Aborting...";
if [ "$avb" ] && [ ! "$NO_VBMETA_PARTITION_PATCH" ]; then
flags=$(httools_static disable-flags);
[ $? == 0 ] || abort "Failed to parse top-level vbmeta. Aborting...";
if [ "$flags" == "enabled" ]; then
ui_print " " "dm-verity detected! Patching $avb...";
for avbpath in /dev/block/mapper /dev/block/by-name /dev/block/bootdevice/by-name; do
for file in $avb $avb$SLOT; do
if [ -e $avbpath/$file ]; then
avbblock=$avbpath/$file;
break 2;
fi;
done;
done;
cd $BIN;
httools_static patch $1 $AKHOME/$img $avbblock || abort "Failed to patch $1 on $avb. Aborting...";
cd $AKHOME;
fi
fi
imgsz=$(wc -c < $img);
if [ "$imgsz" != "$(wc -c < $imgblock)" ]; then
if [ -d /postinstall/tmp -a "$SLOT_SELECT" == "inactive" ]; then
echo "Resizing $1$SLOT snapshot..." >&2;
snapshotupdater_static update $1 $imgsz || abort "Resizing $1$SLOT snapshot failed. Aborting...";
else
echo "Removing any existing $1_ak3..." >&2;
lptools_static remove $1_ak3;
echo "Clearing any merged cow partitions..." >&2;
lptools_static clear-cow;
echo "Attempting to create $1_ak3..." >&2;
if lptools_static create $1_ak3 $imgsz; then
echo "Replacing $1$SLOT with $1_ak3..." >&2;
lptools_static unmap $1_ak3 || abort "Unmapping $1_ak3 failed. Aborting...";
lptools_static map $1_ak3 || abort "Mapping $1_ak3 failed. Aborting...";
lptools_static replace $1_ak3 $1$SLOT || abort "Replacing $1$SLOT failed. Aborting...";
imgblock=/dev/block/mapper/$1_ak3;
ui_print " " "Warning: $1$SLOT replaced in super. Reboot before further logical partition operations.";
else
echo "Creating $1_ak3 failed. Attempting to resize $1$SLOT..." >&2;
httools_static umount $1 || abort "Unmounting $1 failed. Aborting...";
if [ -e $path/$1-verity ]; then
lptools_static unmap $1-verity || abort "Unmapping $1-verity failed. Aborting...";
fi
lptools_static unmap $1$SLOT || abort "Unmapping $1$SLOT failed. Aborting...";
lptools_static resize $1$SLOT $imgsz || abort "Resizing $1$SLOT failed. Aborting...";
lptools_static map $1$SLOT || abort "Mapping $1$SLOT failed. Aborting...";
isunmounted=1;
fi
fi
fi
elif [ "$(wc -c < $img)" -gt "$(wc -c < $imgblock)" ]; then
abort "New $1 image larger than $1 partition. Aborting...";
fi;
isro=$(blockdev --getro $imgblock 2>/dev/null);
blockdev --setrw $imgblock 2>/dev/null;
if [ -f "$BIN/flash_erase" -a -f "$BIN/nandwrite" ]; then
flash_erase $imgblock 0 0;
nandwrite -p $imgblock $img;
elif [ "$CUSTOMDD" ]; then
dd if=/dev/zero of=$imgblock 2>/dev/null;
dd if=$img of=$imgblock;
else
cat $img /dev/zero > $imgblock 2>/dev/null || true;
fi;
if [ $? != 0 ]; then
abort "Flashing $1 failed. Aborting...";
fi;
if [ "$isro" != 0 ]; then
blockdev --setro $imgblock 2>/dev/null;
fi;
if [ "$isunmounted" -a "$path" == "/dev/block/mapper" ]; then
httools_static mount $1 || abort "Mounting $1 failed. Aborting...";
fi
touch ${1}_flashed;
fi;
}
### write_boot (repack ramdisk then build, sign and write image, vendor_dlkm and dtbo)
write_boot() {
repack_ramdisk;
flash_boot;
flash_generic vendor_dlkm;
flash_generic system_dlkm;
flash_generic dtbo;
}
###
### file editing functions:
# backup_file <file>
backup_file() { [ ! -f $1~ ] && cp -fp $1 $1~; }
# restore_file <file>
restore_file() { [ -f $1~ ] && cp -fp $1~ $1; rm -f $1~; }
# replace_string <file> <if search string> <original string> <replacement string> <scope>
replace_string() {
[ "$5" == "global" ] && local scope=g;
if ! grep -q "$2" $1; then
sed -i "s;${3};${4};${scope}" $1;
fi;
}
# replace_section <file> <begin search string> <end search string> <replacement string>
replace_section() {
local begin endstr last end;
begin=$(grep -n -m1 "$2" $1 | cut -d: -f1);
if [ "$begin" ]; then
if [ "$3" == " " -o ! "$3" ]; then
endstr='^[[:space:]]*$';
last=$(wc -l $1 | cut -d\ -f1);
else
endstr="$3";
fi;
for end in $(grep -n "$endstr" $1 | cut -d: -f1) $last; do
if [ "$end" ] && [ "$begin" -lt "$end" ]; then
sed -i "${begin},${end}d" $1;
[ "$end" == "$last" ] && echo >> $1;
sed -i "${begin}s;^;${4}\n;" $1;
break;
fi;
done;
fi;
}
# remove_section <file> <begin search string> <end search string>
remove_section() {
local begin endstr last end;
begin=$(grep -n -m1 "$2" $1 | cut -d: -f1);
if [ "$begin" ]; then
if [ "$3" == " " -o ! "$3" ]; then
endstr='^[[:space:]]*$';
last=$(wc -l $1 | cut -d\ -f1);
else
endstr="$3";
fi;
for end in $(grep -n "$endstr" $1 | cut -d: -f1) $last; do
if [ "$end" ] && [ "$begin" -lt "$end" ]; then
sed -i "${begin},${end}d" $1;
break;
fi;
done;
fi;
}
# insert_line <file> <if search string> <before|after> <line match string> <inserted line>
insert_line() {
local offset line;
if ! grep -q "$2" $1; then
case $3 in
before) offset=0;;
after) offset=1;;
esac;
line=$((`grep -n -m1 "$4" $1 | cut -d: -f1` + offset));
if [ -f $1 -a "$line" ] && [ "$(wc -l $1 | cut -d\ -f1)" -lt "$line" ]; then
echo "$5" >> $1;
else
sed -i "${line}s;^;${5}\n;" $1;
fi;
fi;
}
# replace_line <file> <line replace string> <replacement line> <scope>
replace_line() {
local lines line;
if grep -q "$2" $1; then
lines=$(grep -n "$2" $1 | cut -d: -f1 | sort -nr);
[ "$4" == "global" ] || lines=$(echo "$lines" | tail -n1);
for line in $lines; do
sed -i "${line}s;.*;${3};" $1;
done;
fi;
}
# remove_line <file> <line match string> <scope>
remove_line() {
local lines line;
if grep -q "$2" $1; then
lines=$(grep -n "$2" $1 | cut -d: -f1 | sort -nr);
[ "$3" == "global" ] || lines=$(echo "$lines" | tail -n1);
for line in $lines; do
sed -i "${line}d" $1;
done;
fi;
}
# prepend_file <file> <if search string> <patch file>
prepend_file() {
if ! grep -q "$2" $1; then
echo "$(cat $PATCH/$3 $1)" > $1;
fi;
}
# insert_file <file> <if search string> <before|after> <line match string> <patch file>
insert_file() {
local offset line;
if ! grep -q "$2" $1; then
case $3 in
before) offset=0;;
after) offset=1;;
esac;
line=$((`grep -n -m1 "$4" $1 | cut -d: -f1` + offset));
sed -i "${line}s;^;\n;" $1;
sed -i "$((line - 1))r $PATCH/$5" $1;
fi;
}
# append_file <file> <if search string> <patch file>
append_file() {
if ! grep -q "$2" $1; then
echo -ne "\n" >> $1;
cat $PATCH/$3 >> $1;
echo -ne "\n" >> $1;
fi;
}
# replace_file <file> <permissions> <patch file>
replace_file() {
cp -pf $PATCH/$3 $1;
chmod $2 $1;
}
# patch_fstab <fstab file> <mount match name> <fs match type> block|mount|fstype|options|flags <original string> <replacement string>
patch_fstab() {
local entry part newpart newentry;
entry=$(grep "$2[[:space:]]" $1 | grep "$3");
if [ ! "$(echo "$entry" | grep "$6")" -o "$6" == " " -o ! "$6" ]; then
case $4 in
block) part=$(echo "$entry" | awk '{ print $1 }');;
mount) part=$(echo "$entry" | awk '{ print $2 }');;
fstype) part=$(echo "$entry" | awk '{ print $3 }');;
options) part=$(echo "$entry" | awk '{ print $4 }');;
flags) part=$(echo "$entry" | awk '{ print $5 }');;
esac;
newpart=$(echo "$part" | sed -e "s;${5};${6};" -e "s; ;;g" -e 's;,\{2,\};,;g' -e 's;,*$;;g' -e 's;^,;;g');
newentry=$(echo "$entry" | sed "s;${part};${newpart};");
sed -i "s;${entry};${newentry};" $1;
fi;
}
# patch_cmdline <cmdline entry name> <replacement string>
patch_cmdline() {
local cmdfile cmdtmp match;
if [ -f "$SPLITIMG/cmdline.txt" ]; then
cmdfile=$SPLITIMG/cmdline.txt;
else
cmdfile=$AKHOME/cmdtmp;
grep "^cmdline=" $SPLITIMG/header | cut -d= -f2- > $cmdfile;
fi;
if ! grep -q "$1" $cmdfile; then
cmdtmp=$(cat $cmdfile);
echo "$cmdtmp $2" > $cmdfile;
sed -i -e 's;^[ \t]*;;' -e 's; *; ;g' -e 's;[ \t]*$;;' $cmdfile;
else
match=$(grep -o "$1.*$" $cmdfile | cut -d\ -f1);
sed -i -e "s;${match};${2};" -e 's;^[ \t]*;;' -e 's; *; ;g' -e 's;[ \t]*$;;' $cmdfile;
fi;
if [ -f "$AKHOME/cmdtmp" ]; then
sed -i "s|^cmdline=.*|cmdline=$(cat $cmdfile)|" $SPLITIMG/header;
rm -f $cmdfile;
fi;
}
# patch_prop <prop file> <prop name> <new prop value>
patch_prop() {
if ! grep -q "^$2=" $1; then
echo -ne "\n$2=$3\n" >> $1;
else
local line=$(grep -n -m1 "^$2=" $1 | cut -d: -f1);
sed -i "${line}s;.*;${2}=${3};" $1;
fi;
}
# patch_ueventd <ueventd file> <device node> <permissions> <chown> <chgrp>
patch_ueventd() {
local file dev perm user group newentry line;
file=$1; dev=$2; perm=$3; user=$4;
shift 4;
group="$@";
newentry=$(printf "%-23s %-4s %-8s %s\n" "$dev" "$perm" "$user" "$group");
line=$(grep -n -m1 "$dev" $file | cut -d: -f1);
if [ "$line" ]; then
sed -i "${line}s;.*;${newentry};" $file;
else
echo -ne "\n$newentry\n" >> $file;
fi;
}
###
### configuration/setup functions:
# reset_ak [keep]
reset_ak() {
local current i;
current=$(dirname $AKHOME/*-files/current);
if [ -d "$current" ]; then
for i in $BOOTIMG $AKHOME/boot-new.img; do
[ -e $i ] && cp -af $i $current;
done;
for i in $current/*; do
[ -f $i ] && rm -f $AKHOME/$(basename $i);
done;
fi;
[ -d $SPLITIMG ] && rm -rf $RAMDISK $VENDORRD;
rm -rf $BOOTIMG $SPLITIMG $AKHOME/*-new* $AKHOME/*-files/current;
if [ "$1" == "keep" ]; then
[ -d $AKHOME/rdtmp ] && mv -f $AKHOME/rdtmp $RAMDISK;
[ -d $AKHOME/vrdtmp ] && mv -f $AKHOME/vrdtmp $VENDORRD;
else
rm -rf $PATCH $AKHOME/rdtmp $AKHOME/vrdtmp;
fi;
if [ ! "$NO_BLOCK_DISPLAY" ]; then
ui_print " ";
fi;
setup_ak;
}
# setup_ak
setup_ak() {
local blockfiles plistboot plistinit plistreco parttype name part mtdmount mtdpart mtdname target;
# slot detection enabled by IS_SLOT_DEVICE=1 or auto (from anykernel.sh)
case $IS_SLOT_DEVICE in
1|auto)
SLOT=$(getprop ro.boot.slot_suffix 2>/dev/null);
[ "$SLOT" ] || SLOT=$(grep -o 'androidboot.slot_suffix=.*$' /proc/cmdline | cut -d\ -f1 | cut -d= -f2);
if [ ! "$SLOT" ]; then
SLOT=$(getprop ro.boot.slot 2>/dev/null);
[ "$SLOT" ] || SLOT=$(grep -o 'androidboot.slot=.*$' /proc/cmdline | cut -d\ -f1 | cut -d= -f2);
[ "$SLOT" ] && SLOT=_$SLOT;
fi;
[ "$SLOT" == "normal" ] && unset SLOT;
if [ "$SLOT" ]; then
if [ -d /postinstall/tmp -a ! "$SLOT_SELECT" ]; then
SLOT_SELECT=inactive;
fi;
case $SLOT_SELECT in
inactive)
case $SLOT in
_a) SLOT=_b;;
_b) SLOT=_a;;
esac;
;;
esac;
fi;
if [ ! "$SLOT" -a "$IS_SLOT_DEVICE" == 1 ]; then
abort "Unable to determine active slot. Aborting...";
fi;
;;
esac;
# clean up any template placeholder files
cd $AKHOME;
rm -f modules/system/lib/modules/placeholder patch/placeholder ramdisk/placeholder;
rmdir -p modules patch ramdisk 2>/dev/null;
# automate simple multi-partition setup for hdr_v4 boot + init_boot + vendor_kernel_boot
if [ -e "/dev/block/bootdevice/by-name/init_boot$SLOT" -a -e "/dev/block/bootdevice/by-name/vendor_kernel_boot$SLOT" -a ! -f init_v4_setup ] && [ -f dtb -o -d vendor_ramdisk -o -d vendor_patch ]; then
echo "Setting up for simple automatic init_boot flashing..." >&2;
(mkdir boot-files;
mv -f Image* boot-files;
mkdir init_boot-files;
mv -f ramdisk patch init_boot-files;
mkdir vendor_kernel_boot-files;
mv -f dtb vendor_kernel_boot-files;
mv -f vendor_ramdisk vendor_kernel_boot-files/vendor_ramdisk;
mv -f vendor_patch vendor_kernel_boot-files/patch) 2>/dev/null;
touch init_v4_setup;
# automate simple multi-partition setup for hdr_v3+ boot + vendor_boot with dtb/dlkm
elif [ -e "/dev/block/bootdevice/by-name/vendor_boot$SLOT" -a ! -f vendor_v3_setup ] && [ -f dtb -o -d vendor_ramdisk -o -d vendor_patch ]; then
echo "Setting up for simple automatic vendor_boot flashing..." >&2;
(mkdir boot-files;
mv -f Image* ramdisk patch boot-files;
mkdir vendor_boot-files;
mv -f dtb vendor_boot-files;
mv -f vendor_ramdisk vendor_boot-files/vendor_ramdisk;
mv -f vendor_patch vendor_boot-files/patch) 2>/dev/null;
touch vendor_v3_setup;
fi;
# target block partition detection enabled by BLOCK=<partition filename> or auto (from anykernel.sh)
case $BLOCK in
/dev/*)
if [ "$SLOT" ] && [ -e "$BLOCK$SLOT" ]; then
target=$BLOCK$SLOT;
elif [ -e "$BLOCK" ]; then
target=$BLOCK;
fi;
;;
*)
# maintain brief lists of historic matching partition type names for boot, recovery and init_boot/ramdisk
plistboot="boot BOOT LNX android_boot bootimg KERN-A kernel KERNEL";
plistreco="recovery RECOVERY SOS android_recovery recovery_ramdisk";
plistinit="init_boot ramdisk";
case $BLOCK in
auto) parttype="$plistinit $plistboot";;
boot|kernel) parttype=$plistboot;;
recovery|recovery_ramdisk) parttype=$plistreco;;
init_boot|ramdisk) parttype=$plistinit;;
*) parttype=$BLOCK;;
esac;
for name in $parttype; do
for part in $name$SLOT $name; do
if [ "$(grep -w "$part" /proc/mtd 2>/dev/null)" ]; then
mtdmount=$(grep -w "$part" /proc/mtd);
mtdpart=$(echo "$mtdmount" | cut -d\" -f2);
if [ "$mtdpart" == "$part" ]; then
mtdname=$(echo "$mtdmount" | cut -d: -f1);
else
abort "Unable to determine mtd $BLOCK partition. Aborting...";
fi;
[ -e /dev/mtd/$mtdname ] && target=/dev/mtd/$mtdname;
elif [ -e /dev/block/by-name/$part ]; then
target=/dev/block/by-name/$part;
elif [ -e /dev/block/bootdevice/by-name/$part ]; then
target=/dev/block/bootdevice/by-name/$part;
elif [ -e /dev/block/platform/*/by-name/$part ]; then
target=/dev/block/platform/*/by-name/$part;
elif [ -e /dev/block/platform/*/*/by-name/$part ]; then
target=/dev/block/platform/*/*/by-name/$part;
elif [ -e /dev/$part ]; then
target=/dev/$part;
fi;
[ "$target" ] && break 2;
done;
done;
;;
esac;
if [ "$target" ]; then
BLOCK=$(ls $target 2>/dev/null);
else
abort "Unable to determine $BLOCK partition. Aborting...";
fi;
if [ ! "$NO_BLOCK_DISPLAY" ]; then
ui_print "$BLOCK";
fi;
# allow multi-partition ramdisk modifying configurations (using reset_ak)
name=$(basename $BLOCK | sed -e 's/_a$//' -e 's/_b$//');
if [ "$BLOCK" ] && [ ! -d "$RAMDISK" -a ! -d "$PATCH" ]; then
blockfiles=$AKHOME/$name-files;
if [ "$(ls $blockfiles 2>/dev/null)" ]; then
cp -af $blockfiles/* $AKHOME;
else
mkdir $blockfiles;
fi;
touch $blockfiles/current;
fi;
# run attributes function for current block if it exists
type ${name}_attributes >/dev/null 2>&1 && ${name}_attributes;
}
###
### end methods
setup_ak;