[Feature]add MT2731_MP2_MR2_SVN388 baseline version

Change-Id: Ief04314834b31e27effab435d3ca8ba33b499059
diff --git a/meta/meta-mediatek/classes/kernel-fitimage-extension.bbclass b/meta/meta-mediatek/classes/kernel-fitimage-extension.bbclass
new file mode 100644
index 0000000..95ea635
--- /dev/null
+++ b/meta/meta-mediatek/classes/kernel-fitimage-extension.bbclass
@@ -0,0 +1,275 @@
+inherit kernel-uboot-extension hsm-sign-env
+
+python __anonymous () {
+    kerneltype = d.getVar('KERNEL_IMAGETYPE', True)
+    if kerneltype == 'fitImage':
+        depends = d.getVar("DEPENDS", True)
+        depends = "%s u-boot-mkimage-native lz4-native dtc-native" % depends
+        d.setVar("DEPENDS", depends)
+
+        # Override KERNEL_IMAGETYPE_FOR_MAKE variable, which is internal
+        # to kernel.bbclass . We have to override it, since we pack zImage
+        # (at least for now) into the fitImage .
+        kernelarch = d.getVar('KERNEL_ARCH', True)
+        if kernelarch == 'arm':
+                d.setVar("KERNEL_IMAGETYPE_FOR_MAKE", "zImage")
+        elif kernelarch == 'arm64':
+                d.setVar("KERNEL_IMAGETYPE_FOR_MAKE", "Image")
+        else:
+                print("please set KERNEL_ARCH variable.")
+                sys.exit(1)
+        image = d.getVar('INITRAMFS_IMAGE', True)
+        if image:
+            d.appendVarFlag('do_assemble_fitimage', 'depends', ' ${INITRAMFS_IMAGE}:do_image_complete')
+}
+
+#
+# Emit the fitImage ITS header
+#
+fitimage_emit_fit_header() {
+        cat << EOF >> fit-image.its
+/dts-v1/;
+
+/ {
+        description = "U-Boot fitImage for ${DISTRO_NAME}/${PV}/${MACHINE}";
+        #address-cells = <1>;
+EOF
+}
+
+#
+# Emit the fitImage section bits
+#
+# $1 ... Section bit type: imagestart - image section start
+#                          confstart  - configuration section start
+#                          sectend    - section end
+#                          fitend     - fitimage end
+#
+fitimage_emit_section_maint() {
+        case $1 in
+        imagestart)
+                cat << EOF >> fit-image.its
+
+        images {
+EOF
+        ;;
+        confstart)
+                cat << EOF >> fit-image.its
+
+        configurations {
+EOF
+        ;;
+        sectend)
+                cat << EOF >> fit-image.its
+        };
+EOF
+        ;;
+        fitend)
+                cat << EOF >> fit-image.its
+};
+EOF
+        ;;
+        esac
+}
+
+#
+# Emit the fitImage ITS kernel section
+#
+# $1 ... Image counter
+# $2 ... Path to kernel image
+# $3 ... Compression type
+fitimage_emit_section_kernel() {
+
+        if [ -n "${IMAGE_HASH_ALGO}" ] ; then
+            kernel_csum="${IMAGE_HASH_ALGO}"
+        else
+            kernel_csum="sha256"
+        fi
+
+        ENTRYPOINT=${UBOOT_ENTRYPOINT}
+        if [ -n "${UBOOT_ENTRYSYMBOL}" ] ; then
+                ENTRYPOINT=`${HOST_PREFIX}nm ${S}/vmlinux | \
+                        awk '$3=="${UBOOT_ENTRYSYMBOL}" {print $1}'`
+        fi
+
+        cat << EOF >> fit-image.its
+                kernel@${1} {
+                        description = "Linux kernel";
+                        data = /incbin/("${2}");
+                        type = "kernel";
+                        arch = "${UBOOT_ARCH}";
+                        os = "linux";
+                        compression = "${3}";
+                        load = <${UBOOT_LOADADDRESS}>;
+                        entry = <${ENTRYPOINT}>;
+                        hash@1 {
+                                algo = "${kernel_csum}";
+                        };
+                };
+EOF
+}
+
+#
+# Emit the fitImage ITS DTB section
+#
+# $1 ... Image counter
+# $2 ... Path to DTB image
+fitimage_emit_section_dtb() {
+
+        if [ -n "${IMAGE_HASH_ALGO}" ] ; then
+            dtb_csum="${IMAGE_HASH_ALGO}"
+        else
+            dtb_csum="sha256"
+        fi
+
+        cat << EOF >> fit-image.its
+                fdt@${1} {
+                        description = "Flattened Device Tree blob";
+                        data = /incbin/("${2}");
+                        type = "flat_dt";
+                        arch = "${UBOOT_ARCH}";
+                        compression = "none";
+                        load = <${DTB_LOADADDRESS}>;
+                        hash@1 {
+                                algo = "${dtb_csum}";
+                        };
+                };
+EOF
+}
+
+#
+# Emit the fitImage ITS configuration section
+#
+# $1 ... Linux kernel ID
+# $2 ... DTB image ID
+fitimage_emit_section_config() {
+
+        if [ -n "${VB_HASH_ALGO}" -a -n "${VB_RSA_ALGO}" ] ; then
+            conf_csum="${VB_HASH_ALGO},${VB_RSA_ALGO}"
+        else
+            conf_csum="sha256,rsa2048"
+        fi
+        conf_key_name="dev"
+
+        # Test if we have any DTBs at all
+        if [ -z "${2}" ] ; then
+                conf_desc="Boot Linux kernel"
+                fdt_line=""
+        else
+                conf_desc="Boot Linux kernel with FDT blob"
+                fdt_line="fdt = \"fdt@${2}\";"
+        fi
+        kernel_line="kernel = \"kernel@${1}\";"
+
+        cat << EOF >> fit-image.its
+                default = "conf@1";
+                conf@1 {
+                        description = "${conf_desc}";
+                        ${kernel_line}
+                        ${fdt_line}
+                        signature@1 {
+                                algo = "${conf_csum}";
+                                key-name-hint="${conf_key_name}";
+                                sign-images="fdt","kernel";
+                        };
+                };
+EOF
+}
+
+do_assemble_fitimage() {
+        cd ${B}
+        if [ "x${KERNEL_IMAGETYPE}" = "xfitImage" ] ; then
+                kernelcount=1
+                dtbcount=""
+                rm -f fit-image.its
+
+                fitimage_emit_fit_header
+
+                #
+                # Step 1: Prepare a kernel image section.
+                #
+                fitimage_emit_section_maint imagestart
+
+                uboot_prep_kimage
+                fitimage_emit_section_kernel "${kernelcount}" linux.bin "${linux_comp}"
+
+                #
+                # Step 2: Prepare a DTB image section
+                #
+                if [ -n "${KERNEL_DEVICETREE}" ] ; then
+                        dtbcount=1
+                        for DTB in ${KERNEL_DEVICETREE}; do
+                                if echo ${DTB} | grep -q '/dts/'; then
+                                        bbwarn "${DTB} contains the full path to the the dts file, but only the dtb name should be used."
+                                        DTB=`basename ${DTB} | sed 's,\.dts$,.dtb,g'`
+                                fi
+
+                                # go through device tree blob dir for 32/64 bits kernel
+                                DTB_PATH="arch/${ARCH}/boot/dts/mediatek/${DTB}"
+                                if [ ! -e "${DTB_PATH}" ]; then
+                                        DTB_PATH="arch/${ARCH}/boot/dts/${DTB}"
+                                        if [ ! -e "${DTB_PATH}" ]; then
+                                                DTB_PATH="arch/${ARCH}/boot/${DTB}"
+                                        fi
+                                fi
+
+                                fitimage_emit_section_dtb ${dtbcount} ${DTB_PATH}
+                                dtbcount=`expr ${dtbcount} + 1`
+                        done
+                fi
+
+                fitimage_emit_section_maint sectend
+
+                # Force the first Kernel and DTB in the default config
+                kernelcount=1
+                dtbcount=1
+
+                #
+                # Step 3: Prepare a configurations section
+                #
+                fitimage_emit_section_maint confstart
+
+                fitimage_emit_section_config ${kernelcount} ${dtbcount}
+
+                fitimage_emit_section_maint sectend
+
+                fitimage_emit_section_maint fitend
+
+                #
+                # Step 4: Assemble the image
+                #
+
+                ${HSM_ENV} HSM_KEY_NAME=${VERIFIED_KEY} uboot-mkimage -f fit-image.its arch/${ARCH}/boot/fitImage
+
+                if [ "${SECURE_BOOT_ENABLE}" = "yes" -a "${STANDALONE_SIGN_PREPARE}" != "yes" ]; then
+                        mkdir -p ./mykeys
+                        cp ${MTK_KEY_DIR}/${VERIFIED_KEY}.crt ./mykeys/dev.crt
+                        cp ${MTK_KEY_DIR}/${VERIFIED_KEY}.pem ./mykeys/dev.key
+                        ${HSM_ENV} HSM_KEY_NAME=${VERIFIED_KEY} uboot-mkimage -D "-I dts -O dtb -p 1024" -k ./mykeys -f fit-image.its -r arch/${ARCH}/boot/fitImage
+                fi
+        fi
+}
+
+addtask assemble_fitimage before do_install after do_compile_kernelmodules
+
+kernel_do_deploy[vardepsexclude] = "DATETIME"
+kernel_do_deploy_append() {
+        # Update deploy directory
+        if [ "x${KERNEL_IMAGETYPE}" = "xfitImage" ] ; then
+                cd ${B}
+                echo "Copying fit-image.its source file..."
+                its_base_name="${KERNEL_IMAGETYPE}-its-${PV}-${PR}-${MACHINE}-${DATETIME}"
+                its_symlink_name=${KERNEL_IMAGETYPE}-its-${MACHINE}
+                install -m 0644 fit-image.its ${DEPLOYDIR}/${its_base_name}.its
+                linux_bin_base_name="${KERNEL_IMAGETYPE}-linux.bin-${PV}-${PR}-${MACHINE}-${DATETIME}"
+                linux_bin_symlink_name=${KERNEL_IMAGETYPE}-linux.bin-${MACHINE}
+                install -m 0644 linux.bin ${DEPLOYDIR}/${linux_bin_base_name}.bin
+                if [ -n "${RECOVERY_KERNEL_DEVICETREE}" ] ; then
+                        find arch/${ARCH}/boot/dts -name "${RECOVERY_KERNEL_DEVICETREE}" -exec install -m 0644 {} ${DEPLOYDIR}/${RECOVERY_KERNEL_DEVICETREE} \;
+                fi
+
+                cd ${DEPLOYDIR}
+                ln -sf ${its_base_name}.its ${its_symlink_name}.its
+                ln -sf ${linux_bin_base_name}.bin ${linux_bin_symlink_name}.bin
+                ln -nfs ${KERNEL_IMAGETYPE}-${KERNEL_IMAGE_NAME}.bin boot.img
+        fi
+}