agent: add memory hotplug probe interface check in GetGuestDetails

In order to support memory hotplug br probe interface, firstly, we need to verify whether guest kernel is capable of that. Typically we will check if the path /sys/devices/system/memory/probe exists Fixes: #442 Signed-off-by: Penny Zheng <[email protected]>
kata-containers · Jan 23, 2019 · 34b3caa · 34b3caa
1 parent 91a87bc
commit 34b3caa
Show file tree

Hide file tree

Showing 3 changed files with 276 additions and 171 deletions.
diff --git a/grpc.go b/grpc.go
@@ -49,11 +49,12 @@ const (
 )
 
 var (
-	sysfsCPUOnlinePath       = "/sys/devices/system/cpu"
-	sysfsMemOnlinePath       = "/sys/devices/system/memory"
-	sysfsMemoryBlockSizePath = "/sys/devices/system/memory/block_size_bytes"
-	sysfsConnectedCPUsPath   = filepath.Join(sysfsCPUOnlinePath, "online")
-	containersRootfsPath     = "/run"
+	sysfsCPUOnlinePath          = "/sys/devices/system/cpu"
+	sysfsMemOnlinePath          = "/sys/devices/system/memory"
+	sysfsMemoryBlockSizePath    = "/sys/devices/system/memory/block_size_bytes"
+	sysfsMemoryHotplugProbePath = "/sys/devices/system/memory/probe"
+	sysfsConnectedCPUsPath      = filepath.Join(sysfsCPUOnlinePath, "online")
+	containersRootfsPath        = "/run"
 )
 
 type onlineResource struct {
@@ -1414,6 +1415,16 @@ func (a *agentGRPC) GetGuestDetails(ctx context.Context, req *pb.GuestDetailsReq
 		}
 	}
 
+	if req.MemHotplugProbe {
+		if _, err := os.Stat(sysfsMemoryHotplugProbePath); os.IsNotExist(err) {
+			details.SupportMemHotplugProbe = false
+		} else if err != nil {
+			return nil, err
+		} else {
+			details.SupportMemHotplugProbe = true
+		}
+	}
+
 	details.AgentDetails = a.getAgentDetails(ctx)
 
 	return &details, nil