mirror of
				https://github.com/optim-enterprises-bv/kubernetes.git
				synced 2025-11-03 19:58:17 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			284 lines
		
	
	
		
			9.2 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			284 lines
		
	
	
		
			9.2 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
//go:build linux
 | 
						|
// +build linux
 | 
						|
 | 
						|
/*
 | 
						|
Copyright 2016 The Kubernetes 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 e2enode
 | 
						|
 | 
						|
import (
 | 
						|
	"context"
 | 
						|
	"fmt"
 | 
						|
	"os/exec"
 | 
						|
	"path"
 | 
						|
	"strconv"
 | 
						|
	"strings"
 | 
						|
	"time"
 | 
						|
 | 
						|
	v1 "k8s.io/api/core/v1"
 | 
						|
	"k8s.io/apimachinery/pkg/api/resource"
 | 
						|
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 | 
						|
	"k8s.io/apimachinery/pkg/util/sets"
 | 
						|
	"k8s.io/apimachinery/pkg/util/uuid"
 | 
						|
	runtimeapi "k8s.io/cri-api/pkg/apis/runtime/v1"
 | 
						|
	"k8s.io/kubernetes/test/e2e/feature"
 | 
						|
	"k8s.io/kubernetes/test/e2e/framework"
 | 
						|
	e2epod "k8s.io/kubernetes/test/e2e/framework/pod"
 | 
						|
	imageutils "k8s.io/kubernetes/test/utils/image"
 | 
						|
	admissionapi "k8s.io/pod-security-admission/api"
 | 
						|
 | 
						|
	"github.com/onsi/ginkgo/v2"
 | 
						|
	"github.com/onsi/gomega"
 | 
						|
)
 | 
						|
 | 
						|
func getOOMScoreForPid(pid int) (int, error) {
 | 
						|
	procfsPath := path.Join("/proc", strconv.Itoa(pid), "oom_score_adj")
 | 
						|
	out, err := exec.Command("sudo", "cat", procfsPath).CombinedOutput()
 | 
						|
	if err != nil {
 | 
						|
		return 0, err
 | 
						|
	}
 | 
						|
	return strconv.Atoi(strings.TrimSpace(string(out)))
 | 
						|
}
 | 
						|
 | 
						|
func validateOOMScoreAdjSetting(pid int, expectedOOMScoreAdj int) error {
 | 
						|
	oomScore, err := getOOMScoreForPid(pid)
 | 
						|
	if err != nil {
 | 
						|
		return fmt.Errorf("failed to get oom_score_adj for %d: %w", pid, err)
 | 
						|
	}
 | 
						|
	if expectedOOMScoreAdj != oomScore {
 | 
						|
		return fmt.Errorf("expected pid %d's oom_score_adj to be %d; found %d", pid, expectedOOMScoreAdj, oomScore)
 | 
						|
	}
 | 
						|
	return nil
 | 
						|
}
 | 
						|
 | 
						|
func validateOOMScoreAdjSettingIsInRange(pid int, expectedMinOOMScoreAdj, expectedMaxOOMScoreAdj int) error {
 | 
						|
	oomScore, err := getOOMScoreForPid(pid)
 | 
						|
	if err != nil {
 | 
						|
		return fmt.Errorf("failed to get oom_score_adj for %d", pid)
 | 
						|
	}
 | 
						|
	if oomScore < expectedMinOOMScoreAdj {
 | 
						|
		return fmt.Errorf("expected pid %d's oom_score_adj to be >= %d; found %d", pid, expectedMinOOMScoreAdj, oomScore)
 | 
						|
	}
 | 
						|
	if oomScore >= expectedMaxOOMScoreAdj {
 | 
						|
		return fmt.Errorf("expected pid %d's oom_score_adj to be < %d; found %d", pid, expectedMaxOOMScoreAdj, oomScore)
 | 
						|
	}
 | 
						|
	return nil
 | 
						|
}
 | 
						|
 | 
						|
func dumpRunningContainer(ctx context.Context) error {
 | 
						|
	runtime, _, err := getCRIClient()
 | 
						|
	if err != nil {
 | 
						|
		return err
 | 
						|
	}
 | 
						|
	containers, err := runtime.ListContainers(ctx, &runtimeapi.ContainerFilter{
 | 
						|
		State: &runtimeapi.ContainerStateValue{
 | 
						|
			State: runtimeapi.ContainerState_CONTAINER_RUNNING,
 | 
						|
		},
 | 
						|
	})
 | 
						|
	if err != nil {
 | 
						|
		return err
 | 
						|
	}
 | 
						|
	framework.Logf("Running containers:")
 | 
						|
	for _, c := range containers {
 | 
						|
		framework.Logf("%+v", c)
 | 
						|
	}
 | 
						|
	return nil
 | 
						|
}
 | 
						|
 | 
						|
var _ = SIGDescribe("Container Manager Misc", framework.WithSerial(), func() {
 | 
						|
	f := framework.NewDefaultFramework("kubelet-container-manager")
 | 
						|
	f.NamespacePodSecurityLevel = admissionapi.LevelPrivileged
 | 
						|
	f.Describe("Validate OOM score adjustments", feature.OOMScoreAdj, func() {
 | 
						|
		ginkgo.Context("once the node is setup", func() {
 | 
						|
			ginkgo.It("container runtime's oom-score-adj should be -999", func(ctx context.Context) {
 | 
						|
				runtimePids, err := getPidsForProcess(framework.TestContext.ContainerRuntimeProcessName, framework.TestContext.ContainerRuntimePidFile)
 | 
						|
				framework.ExpectNoError(err, "failed to get list of container runtime pids")
 | 
						|
				for _, pid := range runtimePids {
 | 
						|
					gomega.Eventually(ctx, func() error {
 | 
						|
						return validateOOMScoreAdjSetting(pid, -999)
 | 
						|
					}, 5*time.Minute, 30*time.Second).Should(gomega.BeNil())
 | 
						|
				}
 | 
						|
			})
 | 
						|
			ginkgo.It("Kubelet's oom-score-adj should be -999", func(ctx context.Context) {
 | 
						|
				kubeletPids, err := getPidsForProcess(kubeletProcessName, "")
 | 
						|
				framework.ExpectNoError(err, "failed to get list of kubelet pids")
 | 
						|
				gomega.Expect(kubeletPids).To(gomega.HaveLen(1), "expected only one kubelet process; found %d", len(kubeletPids))
 | 
						|
				gomega.Eventually(ctx, func() error {
 | 
						|
					return validateOOMScoreAdjSetting(kubeletPids[0], -999)
 | 
						|
				}, 5*time.Minute, 30*time.Second).Should(gomega.BeNil())
 | 
						|
			})
 | 
						|
 | 
						|
			ginkgo.Context("with test pods", func() {
 | 
						|
				var testPod *v1.Pod
 | 
						|
 | 
						|
				// Log the running containers here to help debugging.
 | 
						|
				ginkgo.AfterEach(func(ctx context.Context) {
 | 
						|
					if ginkgo.CurrentSpecReport().Failed() {
 | 
						|
						ginkgo.By("Dump all running containers")
 | 
						|
						_ = dumpRunningContainer(ctx)
 | 
						|
					}
 | 
						|
 | 
						|
					if testPod == nil {
 | 
						|
						return // nothing to do
 | 
						|
					}
 | 
						|
					deletePodSyncByName(ctx, f, testPod.Name)
 | 
						|
					waitForAllContainerRemoval(ctx, testPod.Name, testPod.Namespace)
 | 
						|
				})
 | 
						|
 | 
						|
				ginkgo.It("pod infra containers oom-score-adj should be -998 and best effort container's should be 1000", func(ctx context.Context) {
 | 
						|
					// Take a snapshot of existing pause processes. These were
 | 
						|
					// created before this test, and may not be infra
 | 
						|
					// containers. They should be excluded from the test.
 | 
						|
					existingPausePIDs, err := getPidsForProcess("pause", "")
 | 
						|
					framework.ExpectNoError(err, "failed to list all pause processes on the node")
 | 
						|
					existingPausePIDSet := sets.NewInt(existingPausePIDs...)
 | 
						|
 | 
						|
					podClient := e2epod.NewPodClient(f)
 | 
						|
					podName := "besteffort" + string(uuid.NewUUID())
 | 
						|
					testPod = podClient.Create(ctx, &v1.Pod{
 | 
						|
						ObjectMeta: metav1.ObjectMeta{
 | 
						|
							Name: podName,
 | 
						|
						},
 | 
						|
						Spec: v1.PodSpec{
 | 
						|
							Containers: []v1.Container{
 | 
						|
								{
 | 
						|
									Image: imageutils.GetE2EImage(imageutils.Agnhost),
 | 
						|
									Name:  podName,
 | 
						|
								},
 | 
						|
							},
 | 
						|
						},
 | 
						|
					})
 | 
						|
 | 
						|
					var pausePids []int
 | 
						|
					ginkgo.By("checking infra container's oom-score-adj")
 | 
						|
					gomega.Eventually(ctx, func() error {
 | 
						|
						pausePids, err = getPidsForProcess("pause", "")
 | 
						|
						if err != nil {
 | 
						|
							return fmt.Errorf("failed to get list of pause pids: %w", err)
 | 
						|
						}
 | 
						|
						for _, pid := range pausePids {
 | 
						|
							if existingPausePIDSet.Has(pid) {
 | 
						|
								// Not created by this test. Ignore it.
 | 
						|
								continue
 | 
						|
							}
 | 
						|
							if err := validateOOMScoreAdjSetting(pid, -998); err != nil {
 | 
						|
								return err
 | 
						|
							}
 | 
						|
						}
 | 
						|
						return nil
 | 
						|
					}, 2*time.Minute, time.Second*4).Should(gomega.BeNil())
 | 
						|
					var shPids []int
 | 
						|
					ginkgo.By("checking besteffort container's oom-score-adj")
 | 
						|
					gomega.Eventually(ctx, func() error {
 | 
						|
						shPids, err = getPidsForProcess("agnhost", "")
 | 
						|
						if err != nil {
 | 
						|
							return fmt.Errorf("failed to get list of serve hostname process pids: %w", err)
 | 
						|
						}
 | 
						|
						if len(shPids) != 1 {
 | 
						|
							return fmt.Errorf("expected only one agnhost process; found %d", len(shPids))
 | 
						|
						}
 | 
						|
						return validateOOMScoreAdjSetting(shPids[0], 1000)
 | 
						|
					}, 2*time.Minute, time.Second*4).Should(gomega.BeNil())
 | 
						|
				})
 | 
						|
 | 
						|
				ginkgo.It("guaranteed container's oom-score-adj should be -998", func(ctx context.Context) {
 | 
						|
					podClient := e2epod.NewPodClient(f)
 | 
						|
					podName := "guaranteed" + string(uuid.NewUUID())
 | 
						|
					testPod = podClient.Create(ctx, &v1.Pod{
 | 
						|
						ObjectMeta: metav1.ObjectMeta{
 | 
						|
							Name: podName,
 | 
						|
						},
 | 
						|
						Spec: v1.PodSpec{
 | 
						|
							Containers: []v1.Container{
 | 
						|
								{
 | 
						|
									Image: imageutils.GetE2EImage(imageutils.Nginx),
 | 
						|
									Name:  podName,
 | 
						|
									Resources: v1.ResourceRequirements{
 | 
						|
										Limits: v1.ResourceList{
 | 
						|
											v1.ResourceCPU:    resource.MustParse("100m"),
 | 
						|
											v1.ResourceMemory: resource.MustParse("50Mi"),
 | 
						|
										},
 | 
						|
									},
 | 
						|
								},
 | 
						|
							},
 | 
						|
						},
 | 
						|
					})
 | 
						|
					var (
 | 
						|
						ngPids []int
 | 
						|
						err    error
 | 
						|
					)
 | 
						|
					gomega.Eventually(ctx, func() error {
 | 
						|
						ngPids, err = getPidsForProcess("nginx", "")
 | 
						|
						if err != nil {
 | 
						|
							return fmt.Errorf("failed to get list of nginx process pids: %w", err)
 | 
						|
						}
 | 
						|
						for _, pid := range ngPids {
 | 
						|
							if err := validateOOMScoreAdjSetting(pid, -998); err != nil {
 | 
						|
								return err
 | 
						|
							}
 | 
						|
						}
 | 
						|
 | 
						|
						return nil
 | 
						|
					}, 2*time.Minute, time.Second*4).Should(gomega.BeNil())
 | 
						|
 | 
						|
				})
 | 
						|
				ginkgo.It("burstable container's oom-score-adj should be between [2, 1000)", func(ctx context.Context) {
 | 
						|
					podClient := e2epod.NewPodClient(f)
 | 
						|
					podName := "burstable" + string(uuid.NewUUID())
 | 
						|
					testPod = podClient.Create(ctx, &v1.Pod{
 | 
						|
						ObjectMeta: metav1.ObjectMeta{
 | 
						|
							Name: podName,
 | 
						|
						},
 | 
						|
						Spec: v1.PodSpec{
 | 
						|
							Containers: []v1.Container{
 | 
						|
								{
 | 
						|
									Image: imageutils.GetE2EImage(imageutils.Agnhost),
 | 
						|
									Args:  []string{"test-webserver"},
 | 
						|
									Name:  podName,
 | 
						|
									Resources: v1.ResourceRequirements{
 | 
						|
										Requests: v1.ResourceList{
 | 
						|
											v1.ResourceCPU:    resource.MustParse("100m"),
 | 
						|
											v1.ResourceMemory: resource.MustParse("50Mi"),
 | 
						|
										},
 | 
						|
									},
 | 
						|
								},
 | 
						|
							},
 | 
						|
						},
 | 
						|
					})
 | 
						|
					var (
 | 
						|
						wsPids []int
 | 
						|
						err    error
 | 
						|
					)
 | 
						|
					gomega.Eventually(ctx, func() error {
 | 
						|
						wsPids, err = getPidsForProcess("agnhost", "")
 | 
						|
						if err != nil {
 | 
						|
							return fmt.Errorf("failed to get list of test-webserver process pids: %w", err)
 | 
						|
						}
 | 
						|
						for _, pid := range wsPids {
 | 
						|
							if err := validateOOMScoreAdjSettingIsInRange(pid, 2, 1000); err != nil {
 | 
						|
								return err
 | 
						|
							}
 | 
						|
						}
 | 
						|
						return nil
 | 
						|
					}, 2*time.Minute, time.Second*4).Should(gomega.BeNil())
 | 
						|
 | 
						|
					// TODO: Test the oom-score-adj logic for burstable more accurately.
 | 
						|
				})
 | 
						|
			})
 | 
						|
		})
 | 
						|
	})
 | 
						|
})
 |