From b120cee75c97fba702ebd56e677fdb482f813846 Mon Sep 17 00:00:00 2001
From: Rohit Jnagal <jnagal@google.com>
Date: Thu, 24 Jul 2014 17:53:38 +0000
Subject: [PATCH] Handle cpumask in raw driver for unified hierarchy.

Docker-DCO-1.1-Signed-off-by: Rohit Jnagal <jnagal@google.com> (github: rjnagal)
---
 container/raw/factory.go |  1 +
 container/raw/handler.go | 45 ++++++++++++++++++++++++++++------------
 2 files changed, 33 insertions(+), 13 deletions(-)

diff --git a/container/raw/factory.go b/container/raw/factory.go
index 976eecac..245fc3be 100644
--- a/container/raw/factory.go
+++ b/container/raw/factory.go
@@ -92,4 +92,5 @@ var supportedSubsystems map[string]struct{} = map[string]struct{}{
 	"cpu":     {},
 	"cpuacct": {},
 	"memory":  {},
+	"cpuset":  {},
 }
diff --git a/container/raw/handler.go b/container/raw/handler.go
index 9988bea5..32786cf5 100644
--- a/container/raw/handler.go
+++ b/container/raw/handler.go
@@ -59,23 +59,32 @@ func (self *rawContainerHandler) ContainerReference() (info.ContainerReference,
 	}, nil
 }
 
-func readInt64(path string, file string) uint64 {
+func readString(path string, file string) string {
 	cgroupFile := filepath.Join(path, file)
 
 	// Ignore non-existent files
 	if !utils.FileExists(cgroupFile) {
-		return 0
+		return ""
 	}
 
 	// Read
 	out, err := ioutil.ReadFile(cgroupFile)
 	if err != nil {
 		log.Printf("raw driver: Failed to read %q: %s", cgroupFile, err)
+		return ""
+	}
+	return string(out)
+}
+
+func readInt64(path string, file string) uint64 {
+	out := readString(path, file)
+	if out == "" {
 		return 0
 	}
-	val, err := strconv.ParseUint(strings.TrimSpace(string(out)), 10, 64)
+
+	val, err := strconv.ParseUint(strings.TrimSpace(out), 10, 64)
 	if err != nil {
-		log.Printf("raw driver: Failed to parse in %q from file %q: %s", string(out), cgroupFile, err)
+		log.Printf("raw driver: Failed to parse in %q from file %q: %s", out, filepath.Join(path, file), err)
 		return 0
 	}
 
@@ -87,22 +96,32 @@ func (self *rawContainerHandler) GetSpec() (*info.ContainerSpec, error) {
 
 	// The raw driver assumes unified hierarchy containers.
 
+	// Get machine info.
+	mi, err := self.machineInfoFactory.GetMachineInfo()
+	if err != nil {
+		return nil, err
+	}
+
 	// CPU.
 	cpuRoot, ok := self.cgroupSubsystems.mountPoints["cpu"]
 	if ok {
 		cpuRoot = filepath.Join(cpuRoot, self.name)
 		if utils.FileExists(cpuRoot) {
-			// Get machine info.
-			mi, err := self.machineInfoFactory.GetMachineInfo()
-			if err != nil {
-				return nil, err
-			}
-
 			spec.Cpu = new(info.CpuSpec)
 			spec.Cpu.Limit = readInt64(cpuRoot, "cpu.shares")
+		}
+	}
 
-			// TODO(vmarmol): Get CPUs from config.Cgroups.CpusetCpus
-			spec.Cpu.Mask = fmt.Sprintf("0-%d", mi.NumCores-1)
+	// Cpu Mask.
+	// This will fail for non-unified hierarchies. We'll return the whole machine mask in that case.
+	cpusetRoot, ok := self.cgroupSubsystems.mountPoints["cpuset"]
+	if ok {
+		cpusetRoot = filepath.Join(cpusetRoot, self.name)
+		if utils.FileExists(cpusetRoot) {
+			spec.Cpu.Mask = readString(cpusetRoot, "cpuset.cpus")
+			if spec.Cpu.Mask == "" {
+				spec.Cpu.Mask = fmt.Sprintf("0-%d", mi.NumCores-1)
+			}
 		}
 	}
 
@@ -113,7 +132,7 @@ func (self *rawContainerHandler) GetSpec() (*info.ContainerSpec, error) {
 		if utils.FileExists(memoryRoot) {
 			spec.Memory = new(info.MemorySpec)
 			spec.Memory.Limit = readInt64(memoryRoot, "memory.limit_in_bytes")
-			spec.Memory.SwapLimit = readInt64(memoryRoot, "memory.limit_in_bytes")
+			spec.Memory.SwapLimit = readInt64(memoryRoot, "memory.memsw.limit_in_bytes")
 		}
 	}