generate: add windows-resources-storage option

Signed-off-by: Zhou Hao <[email protected]>

Author: Zhou Hao

cmd/oci-runtime-tool/generate.go

@@ -128,6 +128,7 @@ var generateFlags = []cli.Flag{
 	cli.StringFlag{Name: "windows-network", Usage: "specifies network for container"},
 	cli.StringFlag{Name: "windows-resources-cpu", Usage: "specifies CPU for container"},
 	cli.Uint64Flag{Name: "windows-resources-memory-limit", Usage: "specifies limit of memory"},
+	cli.StringFlag{Name: "windows-resources-storage", Usage: "specifies storage for container"},
 }
 
 var generateCommand = cli.Command{
@@ -861,6 +862,14 @@ func setupSpec(g *generate.Generator, context *cli.Context) error {
 		g.SetWindowsResourcesMemoryLimit(limit)
 	}
 
+	if context.IsSet("windows-resources-storage") {
+		storage := context.String("windows-resources-storage")
+		err := g.SetWindowsResourcesStorage(storage)
+		if err != nil {
+			return err
+		}
+	}
+
 	err := addSeccomp(context, g)
 	return err
 }

completions/bash/oci-runtime-tool

@@ -396,6 +396,7 @@ _oci-runtime-tool_generate() {
 		--windows-network
 		--windows-resources-cpu
 		--windows-resources-memory-limit
+		--windows-resources-storage
 	"
 
 	local boolean_options="

generate/generate.go

@@ -1538,3 +1538,15 @@ func (g *Generator) SetWindowsResourcesMemoryLimit(limit uint64) {
 	g.initSpecWindowsResourcesMemory()
 	g.spec.Windows.Resources.Memory.Limit = &limit
 }
+
+// SetWindowsResourcesStorage sets g.spec.Windows.Resources.Storage.
+func (g *Generator) SetWindowsResourcesStorage(storage string) error {
+	g.initSpecWindowsResources()
+	tmpStorage := &rspec.WindowsStorageResources{}
+	err := json.Unmarshal([]byte(storage), tmpStorage)
+	if err != nil {
+		return err
+	}
+	g.spec.Windows.Resources.Storage = tmpStorage
+	return nil
+}

man/oci-runtime-tool-generate.1.md

@@ -495,6 +495,10 @@ read the configuration from `config.json`.
 **--windows-resources-memory-limit**=MEMORYLIMIT
   Set limit of memory.
 
+**--windows-resources-storage**=""
+  Specifies storage for container.
+  e.g. --windows-resources-storage '{"iops": 50, "bps": 20}'
+
 # EXAMPLES
 
 ## Generating container in read-only mode