amlpack.sh

#!/usr/bin/sudo bash

set -e

echo "....................."
echo "Amlogic 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 "Can't find level1 folder"
  elif [ "$(ls -1 level1/image.cfg 2>/dev/null | wc -l)" = 0 ]; then
    echo "Can't find image.cfg"
  else
    if [ ! -d out ]; then
      echo "Can't find out folder"
      echo "Created out folder"
      mkdir out
    fi
    file_name=$(cat level1/projectname.txt)
    bin/AmlImagePack -r level1/image.cfg level1 out/"$file_name.img"
    echo "Done."
  fi
elif [ $level = 2 ]; then
  if [ ! -d level2 ]; then
    echo "Unpack level 2 first"
    exit 0
  fi

  if [ ! -f level1/super.PARTITION ]; then
    for part in system system_ext vendor product odm oem oem_a; do
      if [ -d level2/$part ]; then
        echo "Creating $part image"
        size=$(cat level2/config/${part}_size.txt)
        fs=level2/config/${part}_fs_config
        fc=level2/config/${part}_file_contexts
        if [ ! -z "$size" ]; then
          bin/make_ext4fs -s -J -L $part -T -1 -S $fc -C $fs -l $size -a $part level1/$part.PARTITION level2/$part
        fi
        echo "Done."
      fi
    done
  fi

  for part in oem_a odm_ext_a odm_ext_b; do
    if [ -d level2/$part ]; then
      echo "Creating $part image"
      size=$(cat level2/config/${part}_size.txt)
      fs=level2/config/${part}_fs_config
      fc=level2/config/${part}_file_contexts
      if [ ! -z "$size" ]; then
        bin/make_ext4fs -s -J -L $part -T -1 -S $fc -C $fs -l $size -a $part level1/$part.PARTITION level2/$part
      fi
      echo "Done."
    fi
  done

  if [ -f level1/super.PARTITION ]; then
    for filename in level2/*.img; do
      part="$(basename "$filename" .img)"
      if [ -d level2/$part ]; then
        msize=$(du -sk level2/$part | cut -f1 | gawk '{$1*=1024;$1=int($1*1.08);printf $1}')
        fs=level2/config/${part}_fs_config
        fc=level2/config/${part}_file_contexts
        echo "Creating $part image"
        if [ $msize -lt 1048576 ]; then
          msize=1048576
        fi
        bin/make_ext4fs -J -L $part -T -1 -S $fc -C $fs -l $msize -a $part level2/$part.img level2/$part/
        bin/resize2fs -M level2/${part}.img
        echo "Done."
      fi
    done
  fi

  if [ -f level2/config/super_type.txt ]; then
    supertype=$(cat level2/config/super_type.txt)
  fi
  if [ ! -z "$supertype" ] && [ $supertype -eq "3" ]; then
    metadata_size=65536
    metadata_slot=3
    supername="super"
    supersize=$(cat level2/config/super_size.txt)
    superusage1=$(du -cb level2/*.img | grep total | cut -f1)
    command="bin/lpmake --metadata-size $metadata_size --super-name $supername --metadata-slots $metadata_slot"
    command="$command --device $supername:$supersize --group amlogic_dynamic_partitions_a:$superusage1"

    for filename in level2/*_a.img; do
      part="$(basename "$filename" .img)"
      if [ -f level2/$part.img ]; then
        asize=$(du -skb level2/$part.img | cut -f1)
        if [ $asize -gt 0 ]; then
          command="$command --partition $part:readonly:$asize:amlogic_dynamic_partitions_a --image $part=level2/$part.img"
        fi
      fi
    done

    superusage2=$(expr $supersize - $superusage1)
    command="$command --group amlogic_dynamic_partitions_b:$superusage2"

    for filename in level2/*_b.img; do
      part="$(basename "$filename" .img)"
      if [ -f level2/$part.img ]; then
        bsize=$(du -skb level2/$part.img | cut -f1)
        if [ $bsize -eq 0 ]; then
          command="$command --partition $part:readonly:$bsize:amlogic_dynamic_partitions_b"
        fi
      fi
    done

    if [ $superusage2 -ge $supersize ]; then
      echo "Unable to create super image, recreated images are too big."
      echo "Cleanup some files before retrying"
      echo "Needed space: $superusage1"
      echo "Available maximum space: $supersize"
      exit 0
    fi

    command="$command --virtual-ab --sparse --output level1/super.PARTITION"
    if [ -f level1/super.PARTITION ]; then
      $($command)
    fi
  elif [ ! -z "$supertype" ] && [ $supertype -eq "2" ]; then
    metadata_size=65536
    metadata_slot=2
    supername="super"
    supersize=$(cat level2/config/super_size.txt)
    superusage=$(du -cb level2/*.img | grep total | cut -f1)
    command="bin/lpmake --metadata-size $metadata_size --super-name $supername --metadata-slots $metadata_slot"
    command="$command --device $supername:$supersize --group amlogic_dynamic_partitions:$superusage"

    for part in system_ext system odm product vendor; do
      if [ -f level2/$part.img ]; then
        asize=$(du -skb level2/$part.img | cut -f1)
        if [ $asize -gt 0 ]; then
          command="$command --partition $part:readonly:$asize:amlogic_dynamic_partitions --image $part=level2/$part.img"
        fi
      fi
    done

    if [ $superusage -ge $supersize ]; then
      echo "Unable to create super image, recreated images are too big."
      echo "Cleanup some files before retrying"
      echo "Needed space: $superusage1"
      echo "Available maximum space: $supersize"
      exit 0
    fi

    command="$command --sparse --output level1/super.PARTITION"
    if [ -f level1/super.PARTITION ]; then
      $($command)
    fi
  fi

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

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

  if [ -d level3/logo ]; then
    bin/imgpack -r level3/logo level1/logo.PARTITION
  fi

  if [ -d level3/devtree ]; then
    DIR='level3/devtree/'
    count=$(ls -1U $DIR | wc -l)
    if [ $count -gt 1 ]; then
      for filename in level3/devtree/*.dts; do
        [ -e "$filename" ] || continue
        name=$(basename $filename .dts)
        dtc -I dts -O dtb level3/devtree/$name.dts -o "$(echo level3/devtree/$name.dtb | sed -e s'/\.dts/\.dtb/')"
        bin/dtbTool -o level1/_aml_dtb.PARTITION level3/devtree/
      done
    else
      dtc -I dts -O dtb level3/devtree/single.dts -o "$(echo level1/_aml_dtb.PARTITION | sed -e s'/\.dts/\.dtb/')"
    fi
  fi

  if [ -d level3/meson1 ]; then
    DIR='level3/meson1/'
    count=$(ls -1U $DIR | wc -l)
    if [ $count -gt 1 ]; then
      for filename in level3/meson1/*.dts; do
        [ -e "$filename" ] || continue
        name=$(basename $filename .dts)
        dtc -I dts -O dtb level3/meson1/$name.dts -o "$(echo level3/meson1/$name.dtb | sed -e s'/\.dts/\.dtb/')"
        bin/dtbTool -o level1/meson1.dtb level3/meson1/
      done
    else
      dtc -I dts -O dtb level3/meson1/single.dts -o "$(echo level1/meson1.dtb | sed -e s'/\.dts/\.dtb/')"
    fi
  fi

  if [ -f level1/_aml_dtb.PARTITION ]; then
    echo "Do you want to compress _aml_dtb.PARTITION? (y/n)"
    read answer
    if [[ "$answer" =~ ^[Yy]$ ]]; then
      size=$(du -b level1/_aml_dtb.PARTITION | cut -f1)
      if [ $size -gt 196607 ]; then
        gzip -nc level1/_aml_dtb.PARTITION >level1/_aml_dtb.PARTITION.gzip
        mv level1/_aml_dtb.PARTITION.gzip level1/_aml_dtb.PARTITION
      fi
      rm -rf level3/devtree/*.dtb
    fi
  fi

  if [ -f level1/meson1.dtb ]; then
    echo "Do you want to compress _aml_dtb.PARTITION? (y/n)"
    echo "Not recommeded if not supported"
    read answer
    if [[ "$answer" =~ ^[Yy]$ ]]; then
      msize=$(du -b level1/meson1.dtb | cut -f1)
      if [ $msize -gt 196607 ]; then
        gzip -nc level1/meson1.dtb >level1/meson1.dtb.gzip
        mv level1/meson1.dtb.gzip level1/meson1.dtb
      fi
      rm -rf level3/meson1/*.dtb
    fi
  fi

  for part in boot recovery vendor_boot boot_a recovery_a vendor_boot_a; do
    if [ -d level3/${part} ]; then
      if [ -f "level3/$part/split_img/$part.PARTITION-dtb" ]; then
        cp level1/_aml_dtb.PARTITION "level3/$part/split_img/$part.PARTITION-dtb"
      fi
      if [ -f "level3/$part/split_img/$part.PARTITION-second" ]; then
        cp level1/_aml_dtb.PARTITION "level3/$part/split_img/$part.PARTITION-second"
      fi
      bin/aik/cleanup.sh
      cp -r level3/$part/ramdisk bin/aik/
      cp -r level3/$part/split_img bin/aik/
      bin/aik/repackimg.sh
      mv bin/aik/image-new.img level1/${part}.PARTITION
      bin/aik/cleanup.sh
    fi
  done

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

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