Image Customizer: Add Initial MIC release file (#6594)

Author: elainezhao1

toolkit/tools/pkg/imagecustomizerlib/customizeutils.go

@@ -9,6 +9,7 @@ import (
 	"os"
 	"path/filepath"
 	"strconv"
+	"time"
 
 	"github.com/microsoft/CBL-Mariner/toolkit/tools/imagecustomizerapi"
 	"github.com/microsoft/CBL-Mariner/toolkit/tools/imagegen/installutils"
@@ -34,6 +35,8 @@ func doCustomizations(buildDir string, baseConfigPath string, config *imagecusto
 	// Note: The ordering of the customization steps here should try to mirror the order of the equivalent steps in imager
 	// tool as closely as possible.
 
+	buildTime := time.Now().Format("2006-01-02T15:04:05Z")
+
 	err = overrideResolvConf(imageChroot)
 	if err != nil {
 		return err
@@ -69,6 +72,11 @@ func doCustomizations(buildDir string, baseConfigPath string, config *imagecusto
 		return err
 	}
 
+	err = addCustomizerRelease(imageChroot, ToolVersion, buildTime)
+	if err != nil {
+		return err
+	}
+
 	err = runScripts(baseConfigPath, config.SystemConfig.PostInstallScripts, imageChroot)
 	if err != nil {
 		return err
@@ -347,3 +355,23 @@ func loadOrDisableModules(modules imagecustomizerapi.Modules, imageChroot *safec
 
 	return nil
 }
+
+func addCustomizerRelease(imageChroot *safechroot.Chroot, toolVersion string, buildTime string) error {
+	var err error
+
+	logger.Log.Infof("Creating image customizer release file")
+
+	customizerReleaseFilePath := filepath.Join(imageChroot.RootDir(), "/etc/mariner-customizer-release")
+	lines := []string{
+		fmt.Sprintf("%s=\"%s\"", "TOOL_VERSION", toolVersion),
+		fmt.Sprintf("%s=\"%s\"", "BUILD_DATE", buildTime),
+		"",
+	}
+
+	err = file.WriteLines(lines, customizerReleaseFilePath)
+	if err != nil {
+		return fmt.Errorf("error writing customizer release file (%s): %w", customizerReleaseFilePath, err)
+	}
+
+	return nil
+}

toolkit/tools/pkg/imagecustomizerlib/customizeutils_test.go

@@ -4,9 +4,12 @@
 package imagecustomizerlib
 
 import (
+	"bufio"
 	"os"
 	"path/filepath"
+	"strings"
 	"testing"
+	"time"
 
 	"github.com/microsoft/CBL-Mariner/toolkit/tools/imagecustomizerapi"
 	"github.com/microsoft/CBL-Mariner/toolkit/tools/internal/ptrutils"
@@ -98,3 +101,47 @@ func TestCopyAdditionalFiles(t *testing.T) {
 	assert.Equal(t, orig_contents, copy_1_contents)
 	assert.Equal(t, orig_contents, copy_2_contents)
 }
+
+func TestAddCustomizerRelease(t *testing.T) {
+	if os.Geteuid() != 0 {
+		t.Skip("Test must be run as root because it uses a chroot")
+	}
+
+	proposedDir := filepath.Join(tmpDir, "TestAddCustomizerRelease")
+	chroot := safechroot.NewChroot(proposedDir, false)
+	err := chroot.Initialize("", []string{}, []*safechroot.MountPoint{})
+	assert.NoError(t, err)
+	defer chroot.Close(false)
+
+	err = os.MkdirAll(filepath.Join(chroot.RootDir(), "etc"), os.ModePerm)
+	assert.NoError(t, err)
+
+	expectedVersion := "0.1.0"
+	expectedDate := time.Now().Format("2006-01-02T15:04:05Z")
+	err = addCustomizerRelease(chroot, expectedVersion, expectedDate)
+	assert.NoError(t, err)
+
+	releaseFilePath := filepath.Join(chroot.RootDir(), "etc/mariner-customizer-release")
+
+	file, err := os.Open(releaseFilePath)
+	if err != nil {
+		t.Fatalf("Failed to open file: %v", err)
+	}
+	defer file.Close()
+
+	scanner := bufio.NewScanner(file)
+	config := make(map[string]string)
+	for scanner.Scan() {
+		line := scanner.Text()
+		if line == "" {
+			continue
+		}
+		parts := strings.Split(line, "=")
+		key := parts[0]
+		value := strings.Trim(parts[1], "\"")
+		config[key] = value
+	}
+
+	assert.Equal(t, expectedVersion, config["TOOL_VERSION"])
+	assert.Equal(t, expectedDate, config["BUILD_DATE"])
+}