mirror of
https://github.com/containers/podman.git
synced 2025-06-11 10:25:41 +08:00
vendor c/psgo@v1.3.2
Now supports cgroupsv2. Fixes: #4192 Signed-off-by: Valentin Rothberg <rothberg@redhat.com>
This commit is contained in:
6
vendor/github.com/containers/psgo/.travis.yml
generated
vendored
6
vendor/github.com/containers/psgo/.travis.yml
generated
vendored
@ -14,6 +14,6 @@ before_install:
|
||||
- sudo apt-get install -qq bats
|
||||
|
||||
script:
|
||||
- make validate || travis_terminate 1
|
||||
- make build || travis_terminate 1
|
||||
- make test || travis_terminate 1
|
||||
- make validate
|
||||
- make build
|
||||
- make test
|
||||
|
9
vendor/github.com/containers/psgo/Makefile
generated
vendored
9
vendor/github.com/containers/psgo/Makefile
generated
vendored
@ -1,4 +1,5 @@
|
||||
export GO111MODULE=off
|
||||
export GOPROXY=https://proxy.golang.org
|
||||
|
||||
SHELL= /bin/bash
|
||||
GO ?= go
|
||||
@ -9,11 +10,17 @@ PROJECT := github.com/containers/psgo
|
||||
BATS_TESTS := *.bats
|
||||
GO_SRC=$(shell find . -name \*.go)
|
||||
|
||||
GO_BUILD=$(GO) build
|
||||
# Go module support: set `-mod=vendor` to use the vendored sources
|
||||
ifeq ($(shell go help mod >/dev/null 2>&1 && echo true), true)
|
||||
GO_BUILD=GO111MODULE=on $(GO) build -mod=vendor
|
||||
endif
|
||||
|
||||
all: validate build
|
||||
|
||||
.PHONY: build
|
||||
build: $(GO_SRC)
|
||||
$(GO) build -buildmode=pie -o $(BUILD_DIR)/$(NAME) $(PROJECT)/sample
|
||||
$(GO_BUILD) -buildmode=pie -o $(BUILD_DIR)/$(NAME) $(PROJECT)/sample
|
||||
|
||||
.PHONY: clean
|
||||
clean:
|
||||
|
2
vendor/github.com/containers/psgo/go.mod
generated
vendored
2
vendor/github.com/containers/psgo/go.mod
generated
vendored
@ -6,6 +6,6 @@ require (
|
||||
github.com/opencontainers/runc v0.0.0-20190425234816-dae70e8efea4
|
||||
github.com/pkg/errors v0.0.0-20190227000051-27936f6d90f9
|
||||
github.com/sirupsen/logrus v0.0.0-20190403091019-9b3cdde74fbe
|
||||
github.com/stretchr/testify v1.3.0
|
||||
github.com/stretchr/testify v1.4.0
|
||||
golang.org/x/sys v0.0.0-20190425145619-16072639606e
|
||||
)
|
||||
|
8
vendor/github.com/containers/psgo/go.sum
generated
vendored
8
vendor/github.com/containers/psgo/go.sum
generated
vendored
@ -16,8 +16,12 @@ github.com/stretchr/objx v0.1.1 h1:2vfRuCMp5sSVIDSqO8oNnWJq7mPa6KVP3iPIwFBuy8A=
|
||||
github.com/stretchr/objx v0.1.1/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
|
||||
github.com/stretchr/testify v1.2.2 h1:bSDNvY7ZPG5RlJ8otE/7V6gMiyenm9RtJ7IUVIAoJ1w=
|
||||
github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs=
|
||||
github.com/stretchr/testify v1.3.0 h1:TivCn/peBQ7UY8ooIcPgZFpTNSz0Q2U6UrFlUfqbe0Q=
|
||||
github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
|
||||
github.com/stretchr/testify v1.4.0 h1:2E4SXV/wtOkTonXsotYi4li6zVWxYlZuYNCXe9XRJyk=
|
||||
github.com/stretchr/testify v1.4.0/go.mod h1:j7eGeouHqKxXV5pUuKE4zz7dFj8WfuZ+81PSLYec5m4=
|
||||
golang.org/x/sys v0.0.0-20180905080454-ebe1bf3edb33/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
||||
golang.org/x/sys v0.0.0-20190425145619-16072639606e h1:4ktJgTV34+N3qOZUc5fAaG3Pb11qzMm3PkAoTAgUZ2I=
|
||||
golang.org/x/sys v0.0.0-20190425145619-16072639606e/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=
|
||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||
gopkg.in/yaml.v2 v2.2.2 h1:ZCJp+EgiOT7lHqUV2J862kp8Qj64Jo6az82+3Td9dZw=
|
||||
gopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
|
||||
|
44
vendor/github.com/containers/psgo/internal/cgroups/cgroups.go
generated
vendored
Normal file
44
vendor/github.com/containers/psgo/internal/cgroups/cgroups.go
generated
vendored
Normal file
@ -0,0 +1,44 @@
|
||||
// Copyright 2019 psgo authors
|
||||
//
|
||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
// you may not use this file except in compliance with the License.
|
||||
// You may obtain a copy of the License at
|
||||
//
|
||||
// http://www.apache.org/licenses/LICENSE-2.0
|
||||
//
|
||||
// Unless required by applicable law or agreed to in writing, software
|
||||
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
// See the License for the specific language governing permissions and
|
||||
// limitations under the License.
|
||||
|
||||
package cgroups
|
||||
|
||||
import (
|
||||
"sync"
|
||||
"syscall"
|
||||
)
|
||||
|
||||
const (
|
||||
CgroupRoot = "/sys/fs/cgroup"
|
||||
cgroup2SuperMagic = 0x63677270
|
||||
)
|
||||
|
||||
var (
|
||||
isUnifiedOnce sync.Once
|
||||
isUnified bool
|
||||
isUnifiedErr error
|
||||
)
|
||||
|
||||
// IsCgroup2UnifiedMode returns whether we are running in cgroup or cgroupv2 mode.
|
||||
func IsCgroup2UnifiedMode() (bool, error) {
|
||||
isUnifiedOnce.Do(func() {
|
||||
var st syscall.Statfs_t
|
||||
if err := syscall.Statfs(CgroupRoot, &st); err != nil {
|
||||
isUnified, isUnifiedErr = false, err
|
||||
} else {
|
||||
isUnified, isUnifiedErr = st.Type == cgroup2SuperMagic, nil
|
||||
}
|
||||
})
|
||||
return isUnified, isUnifiedErr
|
||||
}
|
86
vendor/github.com/containers/psgo/internal/proc/pids.go
generated
vendored
86
vendor/github.com/containers/psgo/internal/proc/pids.go
generated
vendored
@ -1,4 +1,4 @@
|
||||
// Copyright 2018 psgo authors
|
||||
// Copyright 2018-2019 psgo authors
|
||||
//
|
||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
// you may not use this file except in compliance with the License.
|
||||
@ -18,8 +18,11 @@ import (
|
||||
"bufio"
|
||||
"fmt"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strconv"
|
||||
"strings"
|
||||
|
||||
"github.com/containers/psgo/internal/cgroups"
|
||||
)
|
||||
|
||||
// GetPIDs extracts and returns all PIDs from /proc.
|
||||
@ -49,45 +52,98 @@ func GetPIDs() ([]string, error) {
|
||||
return pids, nil
|
||||
}
|
||||
|
||||
// pidCgroupPath returns the path to the pid's pids cgroup.
|
||||
func pidCgroupPath(pid string) (string, error) {
|
||||
// GetPIDsFromCgroup returns a strings slice of all pids listesd in pid's pids
|
||||
// cgroup. It automatically detects if we're running in unified mode or not.
|
||||
func GetPIDsFromCgroup(pid string) ([]string, error) {
|
||||
unified, err := cgroups.IsCgroup2UnifiedMode()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if unified {
|
||||
return getPIDsFromCgroupV2(pid)
|
||||
}
|
||||
return getPIDsFromCgroupV1(pid)
|
||||
}
|
||||
|
||||
// getPIDsFromCgroupV1 returns a strings slice of all pids listesd in pid's pids
|
||||
// cgroup.
|
||||
func getPIDsFromCgroupV1(pid string) ([]string, error) {
|
||||
// First, find the corresponding path to the PID cgroup.
|
||||
f, err := os.Open(fmt.Sprintf("/proc/%s/cgroup", pid))
|
||||
if err != nil {
|
||||
return "", err
|
||||
return nil, err
|
||||
}
|
||||
defer f.Close()
|
||||
|
||||
scanner := bufio.NewScanner(f)
|
||||
cgroupPath := ""
|
||||
for scanner.Scan() {
|
||||
fields := strings.Split(scanner.Text(), ":")
|
||||
if len(fields) != 3 {
|
||||
continue
|
||||
}
|
||||
if fields[1] == "pids" {
|
||||
return fmt.Sprintf("/sys/fs/cgroup/pids/%s/cgroup.procs", fields[2]), nil
|
||||
cgroupPath = fmt.Sprintf("/sys/fs/cgroup/pids/%s/cgroup.procs", fields[2])
|
||||
}
|
||||
}
|
||||
return "", fmt.Errorf("couldn't find pids group for PID %s", pid)
|
||||
}
|
||||
|
||||
// GetPIDsFromCgroup returns a strings slice of all pids listesd in pid's pids
|
||||
// cgroup.
|
||||
func GetPIDsFromCgroup(pid string) ([]string, error) {
|
||||
cgroupPath, err := pidCgroupPath(pid)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
if cgroupPath == "" {
|
||||
return nil, fmt.Errorf("couldn't find v1 pids group for PID %s", pid)
|
||||
}
|
||||
|
||||
f, err := os.Open(cgroupPath)
|
||||
// Second, extract the PIDs inside the cgroup.
|
||||
f, err = os.Open(cgroupPath)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer f.Close()
|
||||
|
||||
pids := []string{}
|
||||
scanner := bufio.NewScanner(f)
|
||||
scanner = bufio.NewScanner(f)
|
||||
for scanner.Scan() {
|
||||
pids = append(pids, scanner.Text())
|
||||
}
|
||||
|
||||
return pids, nil
|
||||
}
|
||||
|
||||
// getPIDsFromCgroupV2 returns a strings slice of all pids listesd in pid's pids
|
||||
// cgroup.
|
||||
func getPIDsFromCgroupV2(pid string) ([]string, error) {
|
||||
// First, find the corresponding path to the PID cgroup.
|
||||
f, err := os.Open(fmt.Sprintf("/proc/%s/cgroup", pid))
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer f.Close()
|
||||
|
||||
scanner := bufio.NewScanner(f)
|
||||
cgroupSlice := ""
|
||||
for scanner.Scan() {
|
||||
fields := strings.Split(scanner.Text(), ":")
|
||||
if len(fields) != 3 {
|
||||
continue
|
||||
}
|
||||
cgroupSlice = fields[2]
|
||||
break
|
||||
}
|
||||
|
||||
if cgroupSlice == "" {
|
||||
return nil, fmt.Errorf("couldn't find v2 pids group for PID %s", pid)
|
||||
}
|
||||
|
||||
// Second, extract the PIDs inside the cgroup.
|
||||
f, err = os.Open(filepath.Join(cgroups.CgroupRoot, cgroupSlice, "cgroup.procs"))
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer f.Close()
|
||||
|
||||
pids := []string{}
|
||||
scanner = bufio.NewScanner(f)
|
||||
for scanner.Scan() {
|
||||
pids = append(pids, scanner.Text())
|
||||
}
|
||||
|
||||
return pids, nil
|
||||
}
|
||||
|
Reference in New Issue
Block a user