diff options
Diffstat (limited to 'stages/build.sh')
| -rwxr-xr-x | stages/build.sh | 334 |
1 files changed, 334 insertions, 0 deletions
diff --git a/stages/build.sh b/stages/build.sh new file mode 100755 index 0000000..6e1c02c --- /dev/null +++ b/stages/build.sh @@ -0,0 +1,334 @@ +#!/bin/bash + +if [ ! -f .config ]; then + echo "[x] Configuration file not found!" + echo " Please configure build with 'make config' first." + exit 1 +fi + +source .config + +SUPER_IMAGE="$STOCK_FIRMWARE_PATH/super.img" +FIRMWARE_OUT_DIR="./firmware" +UNPACKED_DIR="./unpacked" +MNT_BASE="$UNPACKED_DIR/mnt" +TMP_DIR="/tmp/driedlamu" +MNT_PARTITIONS=( + "system_a" + "product_a" + "system_ext_a" +) + +if [[ -d $TMP_DIR || -d $UNPACKED_DIR ]]; then + echo "[x] Build directories detected" + echo " Please clean build environment with 'make clean' first." + exit 1 +fi + +prepare_erofs_utils() { + local erofs_utils_repo="https://api.github.com/repos/sekaiacg/erofs-utils/releases/latest" + local aarch=$(lscpu | sed -n 's/Architecture:[ \t]*//p') + local erofs_utils_url=$(curl -s "$erofs_utils_repo" | jq --arg aarch_jq $aarch -c '[ .assets[] | select( .browser_download_url | test("Linux_" + $aarch_jq + ".zip$")) ]' | jq -r ".[] | .browser_download_url") + local download_location="$TMP_DIR/erofs-utils.zip" + + wget -O "$download_location" "$erofs_utils_url" + unzip "$download_location" -d "./utils/bin/" +} + +prepare_apatch_utils() { + local apatch_repo="https://api.github.com/repos/bmax121/KernelPatch/releases/latest" + local apatch_urls=$(curl -s "$apatch_repo" | jq -r '[ .assets[] | select( .browser_download_url | test("kptools-linux|kpimg-android") ) ]' | jq -r ".[] | .browser_download_url") + + for l in $apatch_urls; do + wget -P "./utils/bin/" "$l" + chmod +x ./utils/bin/kp* + done +} + +simg() { + local super_image=$1 + local unpacked_dir=$2 + local super_dir=$(dirname "$super_image") + local super_raw="$super_dir/super_raw.img" + + echo "[i] Converting a sparse image to raw" + if [ ! -f "$super_image" ]; then + echo "[x] super.img not found" + exit 1 + fi + + # converting from sparse + if ! simg2img "$super_image" "$super_raw"; then + echo "[x] Failed to convert super image" + exit 1 + fi + + echo "[i] Unpacking super raw image" + if ! lpunpack "$super_raw" "$unpacked_dir"; then + echo "[x] Failed to unpack super image" + exit 1 + fi + + echo "[i] Super image converting successful" +} + +extract_erofs() { + local DIR=$1 + local MNT_BASE="$2" + + mkdir -p "${MNT_BASE}/${DIR}" + + ./utils/bin/extract.erofs -i "${UNPACKED_DIR}/${DIR}.img" -o "${MNT_BASE}" -x + echo "[STAGE 2] ${DIR} EROFS partition extracted" +} + +make_erofs() { + local part_name=$1 + local target_dir=$2 + local fs_config_dir="unpacked/mnt/config/" + local fs_uuid=$(cat ./${fs_config_dir}/${part_name}_fs_options | awk 'NR==2 {print $3}') + + ./utils/bin/mkfs.erofs --quiet -zlz4hc -T 0 -U "$fs_uuid" --mount-point="/${part_name}" --fs-config-file="${fs_config_dir}/${part_name}_fs_config" --file-contexts="${fs_config_dir}/${part_name}_file_contexts" "${target_dir}/${part_name}.img" "unpacked/mnt/${part_name}" + echo "[STAGE 3] ${part_name} EROFS partition repacked" +} + +unpack_images() { + local FS=$1 + + if [ ! -d "$MNT_BASE" ]; then + mkdir -p "$MNT_BASE" + echo "[+] Directory $MNT_BASE created" + else + echo "[i] Directory $MNT_BASE already exists" + fi + + for dir in "${MNT_PARTITIONS[@]}"; do + if [ ! -d "${MNT_BASE}/$dir" ]; then + mkdir -p "${MNT_BASE}/$dir" + echo "[+] Directory $dir created" + else + echo "[i] Directory $dir already exists" + fi + + if [[ "${FS}" = "erofs" ]]; then + extract_erofs "${dir}" "${MNT_BASE}" + mkdir -p "unpacked/original" + mv "unpacked/${dir}.img" "unpacked/original/${dir}.img" + fi + done +} + +pack_images() { + for i in "${MNT_PARTITIONS[@]}"; do + if [ ! -f "unpacked/${i}.img" ]; then + make_erofs "$i" "${UNPACKED_DIR}" + echo "[+] Image 'unpacked/${i}.img' created" + else + echo "[?] Image 'unpacked/${i}.img' already exists" + fi + done + + echo "[i] All images built" +} + +apatch_patch_boot() { + local apatch_key=$1 + local workdir="$TMP_DIR/root" + local previous_dir="$(pwd)" + + cd "$workdir" + + "$previous_dir"/utils/bin/kptools-linux unpack boot.img + "$previous_dir"/utils/bin/kptools-linux -p --image kernel --skey "$apatch_key" --kpimg "$previous_dir/utils/bin/kpimg-android" --out kernel + "$previous_dir"/utils/bin//kptools-linux repack boot.img + + cd "$previous_dir" +} + +vbmeta_add_hashtree() { + local partition="$1" + local properties=$(avbtool info_image --image ${UNPACKED_DIR}/original/${partition}.img | grep -E "Prop: com.android.build." | sed -e "s/[[:space:]]//g;s/Prop://g;s/->/:/g;s/^/--prop /g") + local cmd="avbtool add_hashtree_footer \ + --image '$UNPACKED_DIR/${partition}.img' \ + --partition_name $(sed -e "s/_a//g" <<< $partition) \ + --block_size 4096 \ + --hash_algorithm sha256 \ + ${properties[@]} \ + --key '$CUSTOM_AVB_KEY'" + + echo -e "[i] Adding hashtree footer to $partition" + + eval $cmd +} + +make_vbmeta() { + local CUSTOM_AVB_KEY_PATH="${AVB_KEY_PATH/#\~/$HOME}" + local CUSTOM_AVB_KEY="$CUSTOM_AVB_KEY_PATH/driedlamu_rsa2048_private.pem" + local CUSTOM_AVB_KEY_PUB="$CUSTOM_AVB_KEY_PATH/driedlamu_rsa2048_public.pem" + + local ORIGINAL_AVB_KEY_PATH="$TMP_DIR/vbmeta/keys/" + + mkdir -p $TMP_DIR \ + $TMP_DIR/vbmeta \ + $TMP_DIR/vbmeta/keys + + if [[ ! -f "$CUSTOM_AVB_KEY" ]]; then + echo "[i] Generating RSA2048 private key" + + openssl genpkey -algorithm RSA \ + -pkeyopt rsa_keygen_bits:2048 \ + -outform PEM \ + -out "$CUSTOM_AVB_KEY" + fi + + if [[ ! -f "$CUSTOM_AVB_KEY_PUB" ]]; then + echo "[i] Extracting RSA2048 public key with AVBTool" + + avbtool extract_public_key \ + --key "$CUSTOM_AVB_KEY" \ + --output "$CUSTOM_AVB_KEY_PUB" + fi + + # While all keys are the same, It's not known whether Motorola is going to change + # them between partition. + dd if="$STOCK_FIRMWARE_PATH/vbmeta.img" of="$ORIGINAL_AVB_KEY_PATH/vbmeta_system_key.pem" bs=1 skip="681" count="520" + dd if="$STOCK_FIRMWARE_PATH/vbmeta.img" of="$ORIGINAL_AVB_KEY_PATH/vbmeta_vendor_key.pem" bs=1 skip="1313" count="520" + dd if="$STOCK_FIRMWARE_PATH/vbmeta.img" of="$ORIGINAL_AVB_KEY_PATH/vbmeta_unknown_key.pem" bs=1 skip="6080" count="520" + + for i in "${MNT_PARTITIONS[@]}"; do + vbmeta_add_hashtree "$i" + done + + echo -e "[i] Making vbmeta_system" + + avbtool make_vbmeta_image \ + --key "$CUSTOM_AVB_KEY" \ + --algorithm 'SHA256_RSA2048' \ + --flags 0 --padding_size '4096' \ + --include_descriptors_from_image $UNPACKED_DIR/system_a.img \ + --output "$TMP_DIR/vbmeta/vbmeta_system.img" + + echo -e "[i] Making vbmeta" + + avbtool make_vbmeta_image --key "$CUSTOM_AVB_KEY" --algorithm 'SHA256_RSA2048' --flags 0 \ + --chain_partition vbmeta_system:2:"$CUSTOM_AVB_KEY_PUB" \ + --chain_partition vbmeta_vendor:4:"$ORIGINAL_AVB_KEY_PATH/vbmeta_vendor_key.pem" \ + --include_descriptors_from_image "$STOCK_FIRMWARE_PATH/boot.img" \ + --include_descriptors_from_image "$STOCK_FIRMWARE_PATH/dtbo.img" \ + --include_descriptors_from_image "$STOCK_FIRMWARE_PATH/init_boot.img" \ + --include_descriptors_from_image "$STOCK_FIRMWARE_PATH/vendor_boot.img" \ + --include_descriptors_from_image "$UNPACKED_DIR/odm_dlkm_a.img" \ + --include_descriptors_from_image "$UNPACKED_DIR/product_a.img" \ + --include_descriptors_from_image "$UNPACKED_DIR/system_dlkm_a.img" \ + --include_descriptors_from_image "$UNPACKED_DIR/system_ext_a.img" \ + --include_descriptors_from_image "$UNPACKED_DIR/vendor_dlkm_a.img" \ + --output "$TMP_DIR/vbmeta/vbmeta.img" + + echo -e "[i] VBMeta created" +} + +make_super() { + local output_path="$1" + local size=0 + local params="" + + mkdir -p "$UNPACKED_DIR/sparse" + + for file in `ls -1 $UNPACKED_DIR/*.img`; do + size=$(($size + $(stat -c %s $file))) + img2simg "$file" "$UNPACKED_DIR/sparse/$(basename $file)" + params+="--partition $(basename $file .img):readonly:$(stat -c %s $file):main_$(echo $(basename $file .img) | tail -c 2) " + if [[ $(stat -c %s $file) != 0 ]]; then + params+="--image $(basename $file .img)=${UNPACKED_DIR}/sparse/$(basename $file) " + fi + done + + size=$(($size * 2)) + + local cmd="lpmake --metadata-size 65536 --metadata-slots 3 --block-size 4096 --virtual-ab --device super:$size --group main_a:$(($size - 2097152)) --group main_b:$(($size - 2097152)) " + + cmd+="$params" + cmd+="--sparse --output $output_path/super.img" + + eval $cmd +} + +echo -e "[i] Prepare" + +# mkdir -p $FIRMWARE_OUT_DIR \ +# $TMP_DIR + +mkdir -p $TMP_DIR + +if [[ ! -f "./utils/bin/mkfs.erofs" && ! -f "./utils/bin/extract.erofs" ]]; then + echo -e "[i] Downloading erofs-utils" + prepare_erofs_utils +fi + +if [[ $INSTALL_SU == 1 ]]; then + if [[ $SU_METHOD == "apatch" ]]; then + if [[ ! -f "./utils/bin/kpimg-android" && ! -f "./utils/bin/kptools-linux" ]]; then + echo -e "[i] Downloading APatch tools" + prepare_apatch_utils + fi + fi +fi + +echo -e "[i] Stage 1" + +simg "${SUPER_IMAGE}" "${UNPACKED_DIR}" + +find ./res/* -type d -exec mkdir -p /tmp/driedlamu/{} \; +find res/* -type f -exec ln -sf "$(pwd)/{}" /tmp/driedlamu/{} \; + +echo -e "[i] Stage 2\n" +unpack_images "erofs" + +echo -e "[i] Running scripts" +echo " " +echo "[-] - Removed" +echo "[+] - Added" +echo "[=] - Modified" +echo " " + +./scripts/bloatware/main.sh "$MNT_BASE" +./scripts/props.sh "$MNT_BASE" +./scripts/configs/main.sh "$MNT_BASE" "$TMP_DIR" +./scripts/download_apks.sh "$TMP_DIR" +./scripts/apk_obtain_permissions.sh "$TMP_DIR" + +if [[ $PRELOAD_MICROG == 1 ]]; then + ./scripts/microg.sh "$MNT_BASE" "$TMP_DIR" +fi + +./scripts/install_apks.sh "$MNT_BASE" "$TMP_DIR/res" +./scripts/permissions/main.sh "$MNT_BASE" "$TMP_DIR" + +if [[ $ADD_TONES == 1 ]]; then + ./scripts/add_media.sh "$MNT_BASE" "$TMP_DIR" +fi + +echo -e "[i] Stage 3\n" +pack_images + +if [[ $INSTALL_SU == 1 ]]; then + echo -e "[i] Installing superuser" + mkdir -p "$TMP_DIR/root" + + cp -r "$STOCK_FIRMWARE_PATH/boot.img" "$TMP_DIR/root" + + if [[ $SU_METHOD == "apatch" ]]; then + echo -e "[i] Patching with APatch" + apatch_patch_boot "$SU_PASSWORD" + cp -r "$TMP_DIR/root/new-boot.img" "$UNPACKED_DIR/boot_su.img" + fi +fi + +if [[ $ENABLE_VBMETA == 1 ]]; then + echo -e "[i] Stage 4 (vbmeta)" + + make_vbmeta + make_super "." +fi + +echo -e "DriedLamu firmware was built!" |