From 5c2a8cf8b97dc02875050c3d17df29dfd63cf8cd Mon Sep 17 00:00:00 2001
From: Mathieu Tortuyaux <mtortuyaux@microsoft.com>
Date: Wed, 24 Apr 2024 10:55:56 +0200
Subject: [PATCH] vm_image_util: add IMAGE_DISK_EXTENSION variable

This variable allows to override the disk extension which is initially
based on the DISK_FORMAT.

Signed-off-by: Mathieu Tortuyaux <mtortuyaux@microsoft.com>
---
 build_library/vm_image_util.sh | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/build_library/vm_image_util.sh b/build_library/vm_image_util.sh
index 214bc2d179..e9bc4d31c0 100644
--- a/build_library/vm_image_util.sh
+++ b/build_library/vm_image_util.sh
@@ -110,6 +110,9 @@ IMG_DEFAULT_FS_HOOK=
 # May be raw, qcow2 (qemu), or vmdk (vmware, virtualbox)
 IMG_DEFAULT_DISK_FORMAT=raw
 
+# Extension to set before the compression extension.
+IMG_DEFAULT_DISK_EXTENSION=
+
 # Name of the partition layout from disk_layout.json
 IMG_DEFAULT_DISK_LAYOUT=base
 
@@ -421,6 +424,11 @@ _dst_path() {
 # Get the proper disk format extension.
 _disk_ext() {
     local disk_format=$(_get_vm_opt DISK_FORMAT)
+    local disk_extension=$(_get_vm_opt DISK_EXTENSION)
+    if [[ -n ${disk_extension} ]]; then
+	echo "${disk_extension}"
+	return 0
+    fi
     case ${disk_format} in
         raw) echo bin;;
         qcow2) echo img;;