1 files changed, 155 insertions, 0 deletions
diff --git a/bsp/meta-arm/meta-arm-bsp/classes/image_types_disk_img.bbclass b/bsp/meta-arm/meta-arm-bsp/classes/image_types_disk_img.bbclass
new file mode 100644
index 00000000..ea7ba86c
--- /dev/null
+++ b/bsp/meta-arm/meta-arm-bsp/classes/image_types_disk_img.bbclass
@@ -0,0 +1,155 @@
+# Defines the disk.img image type
+
+#
+# Add an image type 'disk.img' which creates a disk image
+# with up to 4 partitions
+#
+# For partition 1 (replace 1 by 2 for partition 2, and so on for 3 and 4):
+#
+# * DISK_IMG_PARTITION1_SIZE is the partition size in MB (1024 is 1 GB)
+#
+# * DISK_IMG_PARTITION1_FSTYPE is the file system to format the partition with.
+#   We support only ext files systems (ext2, ext3 and ext4)
+#   If this is empty, the partition will not be formated.
+#
+# * DISK_IMG_PARTITION1_CONTENT is the content to put in the filesystem.
+#   Only 'rootfs' is supported and will create a partition with the Yocto
+#   root filesystem.
+#
+
+# Default values for partition 1
+DISK_IMG_PARTITION1_SIZE ??= "2048"
+DISK_IMG_PARTITION1_FSTYPE ??= "ext4"
+DISK_IMG_PARTITION1_CONTENT ??= "rootfs"
+
+# Default values for partition 2
+DISK_IMG_PARTITION2_SIZE ??= "0"
+DISK_IMG_PARTITION2_FSTYPE ??= "ext2"
+DISK_IMG_PARTITION2_CONTENT ??= ""
+
+# Default values for partition 3
+DISK_IMG_PARTITION3_SIZE ??= "0"
+DISK_IMG_PARTITION3_FSTYPE ??= "ext4"
+DISK_IMG_PARTITION3_CONTENT ??= ""
+
+# Default values for partition 4
+DISK_IMG_PARTITION4_SIZE ??= "0"
+DISK_IMG_PARTITION4_FSTYPE ??= "ext4"
+DISK_IMG_PARTITION4_CONTENT ??= ""
+
+# Default disk sector size
+DISK_IMG_SECTOR_SIZE ??= "512"
+
+# We need mkfs.ext and parted tools to create our image (dd is always there)
+do_image_disk_img[depends] += "e2fsprogs-native:do_populate_sysroot \
+    parted-native:do_populate_sysroot"
+
+DISK_IMG_FILE = "${IMGDEPLOYDIR}/${IMAGE_NAME}${IMAGE_NAME_SUFFIX}.disk.img"
+
+# Create one disk partition
+disk_img_createpart() {
+    local imagefile="$1"
+    local start="$2"
+    local size="$3"
+    local fstype="${4:-ext4}"
+    local content="${5:-}"
+    local formatargs=""
+
+    set -x
+
+    rm -f $imagefile
+
+    # Create the partition image
+    dd if=/dev/zero of=$imagefile bs=${DISK_IMG_SECTOR_SIZE} count=0 \
+        seek=$(expr $size / ${DISK_IMG_SECTOR_SIZE})
+
+    if [ -n "$fstype" ]; then
+        case $content in
+            rootfs)
+                formatargs=" -d ${IMAGE_ROOTFS}"
+                ;;
+            boot)
+                echo "Unsupported"
+                exit 1
+                ;;
+            *)
+        esac
+
+        # Create the file system (with content if needed)
+        mkfs.$fstype -F $imagefile $formatargs
+    fi
+
+    cat $imagefile >> ${DISK_IMG_FILE}
+
+    # Add the partition to the partition table
+    parted -s ${DISK_IMG_FILE} unit B mkpart primary $start \
+        $(expr $start + $realsize - 1)
+}
+
+disk_img_create () {
+    local currpos
+    local realsize
+
+    set -x
+
+    currpos=${DISK_IMG_SECTOR_SIZE}
+
+    # Create reserved part for partition table (1MB)
+    dd if=/dev/zero of=${DISK_IMG_FILE} bs=${DISK_IMG_SECTOR_SIZE} count=0 \
+        seek=1
+
+    parted -s ${DISK_IMG_FILE} mklabel msdos
+
+    if [ ${DISK_IMG_PARTITION1_SIZE} -ne 0 ]; then
+
+        # Reduce the first block size of one sector to make space
+        # for the partition table
+        realsize=$(expr ${DISK_IMG_PARTITION1_SIZE} \* 1024 \* 1024 \
+            - ${DISK_IMG_SECTOR_SIZE})
+
+        # Create the partition
+        disk_img_createpart ${WORKDIR}/part1.img $currpos $realsize \
+            "${DISK_IMG_PARTITION1_FSTYPE}" "${DISK_IMG_PARTITION1_CONTENT}"
+
+        currpos=$(expr $currpos + $realsize)
+    fi
+
+    if [ ${DISK_IMG_PARTITION2_SIZE} -ne 0 ]; then
+        # Partition size
+        realsize=$(expr ${DISK_IMG_PARTITION2_SIZE} \* 1024 \* 1024)
+
+        # Create the partition
+        disk_img_createpart ${WORKDIR}/part2.img $currpos $realsize \
+            "${DISK_IMG_PARTITION2_FSTYPE}" "${DISK_IMG_PARTITION2_CONTENT}"
+
+        currpos=$(expr $currpos + $realsize)
+
+    fi
+
+    if [ ${DISK_IMG_PARTITION3_SIZE} -ne 0 ]; then
+        # Partition size
+        realsize=$(expr ${DISK_IMG_PARTITION3_SIZE} \* 1024 \* 1024)
+
+        # Create the partition
+        disk_img_createpart ${WORKDIR}/part3.img $currpos $realsize \
+            "${DISK_IMG_PARTITION3_FSTYPE}" "${DISK_IMG_PARTITION3_CONTENT}"
+
+        currpos=$(expr $currpos + $realsize)
+
+    fi
+    if [ ${DISK_IMG_PARTITION4_SIZE} -ne 0 ]; then
+        # Partition size
+        realsize=$(expr ${DISK_IMG_PARTITION4_SIZE} \* 1024 \* 1024)
+
+        # Create the partition
+        disk_img_createpart ${WORKDIR}/part4.img $currpos $realsize \
+            "${DISK_IMG_PARTITION4_FSTYPE}" "${DISK_IMG_PARTITION4_CONTENT}"
+
+        currpos=$(expr $currpos + $realsize)
+
+    fi
+}
+
+IMAGE_CMD_disk.img = "disk_img_create"
+IMAGE_TYPES += "disk.img"
+