add containerd support

Signed-off-by: tombokombo <[email protected]>
coroot · Mar 11, 2022 · cdd6f29 · cdd6f29
1 parent 5eac9ec
commit cdd6f29
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/cgroup/cgroup.go b/cgroup/cgroup.go
@@ -18,6 +18,7 @@ var (
 
 	dockerIdRegexp      = regexp.MustCompile(`([a-z0-9]{64})`)
 	crioIdRegexp        = regexp.MustCompile(`crio-([a-z0-9]{64})`)
+	containerdIdRegexp  = regexp.MustCompile(`cri-containerd-([a-z0-9]{64})`)
 	lxcIdRegexp         = regexp.MustCompile(`/lxc/([^/]+)`)
 	systemSliceIdRegexp = regexp.MustCompile(`(/system\.slice/([^/]+))`)
 )
@@ -29,6 +30,7 @@ const (
 	ContainerTypeStandaloneProcess
 	ContainerTypeDocker
 	ContainerTypeCrio
+	ContainerTypeContainerd
 	ContainerTypeLxc
 	ContainerTypeSystemdService
 )
@@ -41,6 +43,8 @@ func (t ContainerType) String() string {
 		return "docker"
 	case ContainerTypeCrio:
 		return "crio"
+	case ContainerTypeContainerd:
+		return "cri-containerd"
 	case ContainerTypeLxc:
 		return "lxc"
 	case ContainerTypeSystemdService:
@@ -184,11 +188,15 @@ func containerByCgroup(path string) (ContainerType, string, error) {
 			return ContainerTypeUnknown, "", fmt.Errorf("invalid docker cgroup %s", path)
 		}
 		return ContainerTypeDocker, matches[1], nil
-	case "kubepods":
+	case "kubepods", "kubepods.slice":
 		crioMatches := crioIdRegexp.FindStringSubmatch(path)
 		if crioMatches != nil {
 			return ContainerTypeCrio, crioMatches[1], nil
 		}
+		containerdMatches := containerdIdRegexp.FindStringSubmatch(path)
+		if containerdMatches != nil {
+			return ContainerTypeContainerd, containerdMatches[1], nil
+		}
 		matches := dockerIdRegexp.FindStringSubmatch(path)
 		if matches == nil {
 			return ContainerTypeUnknown, "", fmt.Errorf("invalid docker cgroup %s", path)

diff --git a/containers/registry.go b/containers/registry.go
@@ -251,7 +251,7 @@ func calcId(cg *cgroup.Cgroup, md *ContainerMetadata) ContainerID {
 	if cg.ContainerType == cgroup.ContainerTypeSystemdService {
 		return ContainerID(cg.ContainerId)
 	}
-	if cg.ContainerType != cgroup.ContainerTypeDocker {
+	if cg.ContainerType != cgroup.ContainerTypeDocker && cg.ContainerType != cgroup.ContainerTypeContainerd {
 		return ""
 	}
 	if md.labels["io.kubernetes.pod.name"] != "" {
@@ -271,7 +271,7 @@ func calcId(cg *cgroup.Cgroup, md *ContainerMetadata) ContainerID {
 }
 
 func getContainerMetadata(cg *cgroup.Cgroup) (*ContainerMetadata, error) {
-	if cg.ContainerType != cgroup.ContainerTypeDocker {
+	if cg.ContainerType != cgroup.ContainerTypeDocker && cg.ContainerType != cgroup.ContainerTypeContainerd {
 		return &ContainerMetadata{}, nil
 	}
 	var dockerdErr error