awunpack.sh

#!/usr/bin/sudo sh

echo "....................."
echo "AllWinner Kitchen"
echo "....................."
echo "....................."
echo "Select level 1,2,3 or q/Q to exit: "
read level
if [ $level = 1 ]; then
    if [ -d level1 ]; then
        echo "Deleting existing level1"
        rm -rf level1 && mkdir level1
    else
        mkdir level1
    fi

    echo "....................."
    echo "AllWinner Kitchen"
    echo "....................."
    if [ ! -d in ]; then
        echo "Can't find /in folder"
        echo "Creating /in folder"
        mkdir in
    fi
    count_file=$(ls -1 in/*.img 2>/dev/null | wc -l)
    if [ "$count_file" = 0 ]; then
        echo "No files found in /in"
        exit 0
    fi
    rename 's/ /_/g' in/*.img
    echo "Files in input dir (*.img)"
    count=0
    for entry in $(ls in/*.img); do
        count=$(($count + 1))
        name=$(basename in/$entry .img)
        echo $count - $name
    done
    echo "....................."
    echo "Enter a file name :"
    read projectname
    echo $projectname >level1/projectname.txt

    if [ ! -f in/$projectname.img ]; then
        echo "Can't find the file"
        exit 0
    fi

    filename=$(cat level1/projectname.txt)
    cp in/$filename.img level1/$filename.img
    bin/linux/imgrepacker level1/$filename.img
    rm level1/$filename.img
    echo "Done."
elif [ $level = 2 ]; then
    imgextractor="bin/common/imgextractor.py"

    version=$(python -V 2>&1 | grep -Po '(?<=Python )(.+)')
    if [ -z "$version" ]; then
        echo "No Python installed!"
    fi

    if [ ! -d level1 ]; then
        echo "Unpack level 1 first"
        exit 0
    fi

    if [ -d level2 ]; then
        echo "Deleting existing level2"
        rm -rf level2 && mkdir -p level2/config
    else
        mkdir -p level2/config
    fi

    foldername=$(cat level1/projectname.txt).img.dump

    for part in system system_ext vendor product odm oem odm_ext_a odm_ext_b; do
        if [ -f level1/$foldername/$part.fex ]; then
            echo "Extracting $part"
            if echo $(file level1/$foldername/$part.fex) | grep -q "sparse"; then
                bin/linux/simg2img level1/$foldername/$part.fex level2/$part.raw.img
                python3 $imgextractor "level2/$part.raw.img" "level2"
            else
                python3 $imgextractor "level1/$foldername/$part.fex" "level2"
            fi
            awk -i inplace '!seen[$0]++' level2/config/${part}_f*
        fi
    done

    rm -rf level2/*.raw.img

    if [ -f level1/$foldername/super.fex ]; then
        bin/linux/simg2img level1/$foldername/super.fex level2/super.img
        echo $(du -b level2/super.img | cut -f1) >level2/config/super_size.txt
        bin/linux/super/lpunpack -slot=0 level2/super.img level2/
        rm -rf level2/super.img

        if [ $(ls -1q level2/*_a.img 2>/dev/null | wc -l) -gt 0 ]; then
            echo "3" >level2/config/super_type.txt
        else
            echo "2" >level2/config/super_type.txt
        fi

        for part in system system_ext vendor product odm oem system_a system_ext_a vendor_a product_a odm_a system_b system_ext_b vendor_b product_b odm_b; do
            if [ -f level2/$part.img ]; then
                size=$(du -b level2/$part.img | cut -f1)
                if [ $size -ge 1024 ]; then
                    python3 $imgextractor "level2/$part.img" "level2"
                    awk -i inplace '!seen[$0]++' level2/config/${part}_f*
                fi
            fi
        done
    fi

    echo "Done."
elif [ $level = 3 ]; then
    if [ ! $(which dtc) ]; then
        echo "install dtc, please (apt-get install device-tree-compiler)"
        exit 0
    fi

    if [ ! -d level1 ]; then
        echo "Unpack level 1 first"
        exit 0
    fi

    if [ -d level3 ]; then
        rm -rf level3 && mkdir level3
    else
        mkdir level3
    fi

    foldername=$(cat level1/projectname.txt).img.dump

    for part in boot recovery vendor_boot boot_a recovery_a vendor_boot_a; do
        if [ -f level1/$foldername/${part}.fex ]; then
            mkdir level3/$part
            bin/linux/aik/unpackimg.sh level1/$foldername/${part}.fex
            mv -i bin/linux/aik/ramdisk level3/$part/
            mv -i bin/linux/aik/split_img level3/$part/
        fi
    done

    if [ -f "level1/$foldername/boot-resource.fex" ]; then
        echo "Extracting boot-resource in level3"
        tmp_mount_dir=$(mktemp -d)
        mount -o loop -t vfat "level1/$foldername/boot-resource.fex" "$tmp_mount_dir"
        cp -a "$tmp_mount_dir"/. "level3/boot-resource/"
        umount "$tmp_mount_dir"
        rmdir "$tmp_mount_dir"
        rm -f "level3/boot-resource/magic.bin"
    fi

    echo "Done."
elif [ $level = "q" -o $level = "Q" ]; then
    exit
fi

while true; do ./write_perm.sh && ./awunpack.sh && break; done