From e73a7b0315f3722f9d161abb893800291918c6de Mon Sep 17 00:00:00 2001
From: Justin Cormack <justin.cormack@docker.com>
Date: Wed, 4 Oct 2017 16:54:41 +0100
Subject: [PATCH] Add a "tar-kernel-initrd" output format

This is a tarball of the kernel, initrd and cmdline files, suitable for
sending to the mkimage images that expect this format.

Note you can't currently stream this output format using `-o` will clean this
up in future commits.

Signed-off-by: Justin Cormack <justin.cormack@docker.com>
---
 src/moby/output.go | 55 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 55 insertions(+)

diff --git a/src/moby/output.go b/src/moby/output.go
index 46c5a3b8c..6d6dcee43 100644
--- a/src/moby/output.go
+++ b/src/moby/output.go
@@ -32,6 +32,16 @@ var outFuns = map[string]func(string, []byte, int) error{
 		}
 		return nil
 	},
+	"tar-kernel-initrd": func(base string, image []byte, size int) error {
+		kernel, initrd, cmdline, err := tarToInitrd(image)
+		if err != nil {
+			return fmt.Errorf("Error converting to initrd: %v", err)
+		}
+		if err := outputKernelInitrdTarball(base, kernel, initrd, cmdline); err != nil {
+			return fmt.Errorf("Error writing kernel+initrd tarball output: %v", err)
+		}
+		return nil
+	},
 	"iso-bios": func(base string, image []byte, size int) error {
 		err := outputIso(bios, base+".iso", image)
 		if err != nil {
@@ -293,3 +303,48 @@ func outputKernelInitrd(base string, kernel []byte, initrd []byte, cmdline strin
 	}
 	return nil
 }
+
+func outputKernelInitrdTarball(base string, kernel []byte, initrd []byte, cmdline string) error {
+	log.Debugf("output kernel/initrd tarball: %s %s", base, cmdline)
+	log.Infof("  %s", base+"-initrd.tar")
+	f, err := os.Create(base + "-initrd.tar")
+	if err != nil {
+		return err
+	}
+	defer f.Close()
+	tw := tar.NewWriter(f)
+	hdr := &tar.Header{
+		Name: "kernel",
+		Mode: 0644,
+		Size: int64(len(kernel)),
+	}
+	if err := tw.WriteHeader(hdr); err != nil {
+		return err
+	}
+	if _, err := tw.Write(kernel); err != nil {
+		return err
+	}
+	hdr = &tar.Header{
+		Name: "initrd.img",
+		Mode: 0644,
+		Size: int64(len(initrd)),
+	}
+	if err := tw.WriteHeader(hdr); err != nil {
+		return err
+	}
+	if _, err := tw.Write(initrd); err != nil {
+		return err
+	}
+	hdr = &tar.Header{
+		Name: "cmdline",
+		Mode: 0644,
+		Size: int64(len(cmdline)),
+	}
+	if err := tw.WriteHeader(hdr); err != nil {
+		return err
+	}
+	if _, err := tw.Write([]byte(cmdline)); err != nil {
+		return err
+	}
+	return tw.Close()
+}