create: support oci spec memory limit

We should run VM(container) regarding oci config.json memory limit.

Fixes #381

Signed-off-by: WANG Chao <chao.wang@ucloud.cn>

config.go

@@ -18,6 +18,7 @@ import (
 	"errors"
 	"fmt"
 	"io/ioutil"
+	"math"
 	"os"
 	"path/filepath"
 	goruntime "runtime"
@@ -376,3 +377,31 @@ func loadConfiguration(configPath string, ignoreLogging bool) (resolvedConfigPat
 
 	return resolved, logfilePath, config, nil
 }
+
+func ociResourceUpdateRuntimeConfig(ocispec oci.CompatOCISpec, config *oci.RuntimeConfig) error {
+	if ocispec.Linux == nil {
+		return nil
+	}
+
+	if ocispec.Linux.Resources == nil {
+		return nil
+	}
+
+	if ocispec.Linux.Resources.Memory == nil {
+		return nil
+	}
+
+	if ocispec.Linux.Resources.Memory.Limit == nil {
+		return nil
+	}
+
+	memBytes := *ocispec.Linux.Resources.Memory.Limit
+
+	mem := int(math.Ceil(float64(memBytes) / 1024 / 1024))
+	if mem <= 0 {
+		return fmt.Errorf("Invalid OCI memory limit %d", memBytes)
+	}
+	config.VMConfig.Memory = uint(mem)
+
+	return nil
+}

create.go

@@ -98,6 +98,10 @@ func create(containerID, bundlePath, console, pidFilePath string, detach bool,
 		return err
 	}
 
+	if err := ociResourceUpdateRuntimeConfig(ociSpec, &runtimeConfig); err != nil {
+		return err
+	}
+
 	containerType, err := ociSpec.ContainerType()
 	if err != nil {
 		return err