-
Notifications
You must be signed in to change notification settings - Fork 157
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #1588 from ibihim/AUTH-442_psa_cluster-fleet-evalu…
…ation AUTH-442: psa cluster fleet evaluation
- Loading branch information
Showing
109 changed files
with
9,568 additions
and
557 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,80 @@ | ||
| package podsecurityreadinesscontroller | ||
|
|
||
| import ( | ||
| "fmt" | ||
| "sort" | ||
| "strings" | ||
|
|
||
| metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" | ||
| "k8s.io/apimachinery/pkg/util/sets" | ||
|
|
||
| operatorv1 "github.com/openshift/api/operator/v1" | ||
| "github.com/openshift/library-go/pkg/operator/v1helpers" | ||
| ) | ||
|
|
||
| const ( | ||
| PodSecurityCustomerType = "PodSecurityCustomerEvaluationConditionsDetected" | ||
| PodSecurityOpenshiftType = "PodSecurityOpenshiftEvaluationConditionsDetected" | ||
| PodSecurityRunLevelZeroType = "PodSecurityRunLevelZeroEvaluationConditionsDetected" | ||
| ) | ||
|
|
||
| var ( | ||
| // run-level zero namespaces, shouldn't avoid openshift namespaces | ||
| runLevelZeroNamespaces = sets.New[string]( | ||
| "default", | ||
| "kube-system", | ||
| "kube-public", | ||
| ) | ||
| ) | ||
|
|
||
| type podSecurityOperatorConditions struct { | ||
| violatingOpenShiftNamespaces []string | ||
| violatingRunLevelZeroNamespaces []string | ||
| violatingCustomerNamespaces []string | ||
| } | ||
|
|
||
| func (c *podSecurityOperatorConditions) addViolation(name string) { | ||
| if runLevelZeroNamespaces.Has(name) { | ||
| c.violatingRunLevelZeroNamespaces = append(c.violatingRunLevelZeroNamespaces, name) | ||
| return | ||
| } | ||
|
|
||
| isOpenShift := strings.HasPrefix(name, "openshift") | ||
| if isOpenShift { | ||
| c.violatingOpenShiftNamespaces = append(c.violatingOpenShiftNamespaces, name) | ||
| return | ||
| } | ||
|
|
||
| c.violatingCustomerNamespaces = append(c.violatingCustomerNamespaces, name) | ||
| } | ||
|
|
||
| func makeCondition(conditionType string, namespaces []string) operatorv1.OperatorCondition { | ||
| if len(namespaces) > 0 { | ||
| sort.Strings(namespaces) | ||
| return operatorv1.OperatorCondition{ | ||
| Type: conditionType, | ||
| Status: operatorv1.ConditionTrue, | ||
| LastTransitionTime: metav1.Now(), | ||
| Reason: "PSViolationsDetected", | ||
| Message: fmt.Sprintf( | ||
| "Violations detected in namespaces: %v", | ||
| namespaces, | ||
| ), | ||
| } | ||
| } | ||
|
|
||
| return operatorv1.OperatorCondition{ | ||
| Type: conditionType, | ||
| Status: operatorv1.ConditionFalse, | ||
| LastTransitionTime: metav1.Now(), | ||
| Reason: "ExpectedReason", | ||
| } | ||
| } | ||
|
|
||
| func (c *podSecurityOperatorConditions) toConditionFuncs() []v1helpers.UpdateStatusFunc { | ||
| return []v1helpers.UpdateStatusFunc{ | ||
| v1helpers.UpdateConditionFn(makeCondition(PodSecurityCustomerType, c.violatingCustomerNamespaces)), | ||
| v1helpers.UpdateConditionFn(makeCondition(PodSecurityOpenshiftType, c.violatingOpenShiftNamespaces)), | ||
| v1helpers.UpdateConditionFn(makeCondition(PodSecurityRunLevelZeroType, c.violatingRunLevelZeroNamespaces)), | ||
| } | ||
| } |
69 changes: 69 additions & 0 deletions
69
pkg/operator/podsecurityreadinesscontroller/conditions_test.go
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,69 @@ | ||
| package podsecurityreadinesscontroller | ||
|
|
||
| import ( | ||
| "testing" | ||
|
|
||
| operatorv1 "github.com/openshift/api/operator/v1" | ||
| ) | ||
|
|
||
| func TestCondition(t *testing.T) { | ||
| t.Run("with namespaces", func(t *testing.T) { | ||
| namespaces := []string{"namespace1", "namespace2"} | ||
| expectedCondition := operatorv1.OperatorCondition{ | ||
| Type: PodSecurityCustomerType, | ||
| Status: operatorv1.ConditionTrue, | ||
| Reason: "PSViolationsDetected", | ||
| Message: "Violations detected in namespaces: [namespace1 namespace2]", | ||
| } | ||
|
|
||
| condition := makeCondition(PodSecurityCustomerType, namespaces) | ||
|
|
||
| if condition.Type != expectedCondition.Type { | ||
| t.Errorf("expected condition type %s, got %s", expectedCondition.Type, condition.Type) | ||
| } | ||
|
|
||
| if condition.Status != expectedCondition.Status { | ||
| t.Errorf("expected condition status %s, got %s", expectedCondition.Status, condition.Status) | ||
| } | ||
|
|
||
| if condition.Reason != expectedCondition.Reason { | ||
| t.Errorf("expected condition reason %s, got %s", expectedCondition.Reason, condition.Reason) | ||
| } | ||
|
|
||
| if condition.Message != expectedCondition.Message { | ||
| t.Errorf("expected condition message %s, got %s", expectedCondition.Message, condition.Message) | ||
| } | ||
| }) | ||
|
|
||
| t.Run("without namespaces", func(t *testing.T) { | ||
| namespaces := []string{} | ||
| expectedCondition := operatorv1.OperatorCondition{ | ||
| Type: PodSecurityCustomerType, | ||
| Status: operatorv1.ConditionFalse, | ||
| Reason: "ExpectedReason", | ||
| } | ||
|
|
||
| condition := makeCondition(PodSecurityCustomerType, namespaces) | ||
|
|
||
| if condition.Type != expectedCondition.Type { | ||
| t.Errorf("expected condition type %s, got %s", expectedCondition.Type, condition.Type) | ||
| } | ||
|
|
||
| if condition.Status != expectedCondition.Status { | ||
| t.Errorf("expected condition status %s, got %s", expectedCondition.Status, condition.Status) | ||
| } | ||
|
|
||
| if condition.Reason != expectedCondition.Reason { | ||
| t.Errorf("expected condition reason %s, got %s", expectedCondition.Reason, condition.Reason) | ||
| } | ||
|
|
||
| if condition.Message != expectedCondition.Message { | ||
| t.Errorf("expected condition message %s, got %s", expectedCondition.Message, condition.Message) | ||
| } | ||
| }) | ||
|
|
||
| t.Run("without anything", func(t *testing.T) { | ||
| cond := podSecurityOperatorConditions{} | ||
| cond.addViolation("hello world") | ||
| }) | ||
| } |
171 changes: 171 additions & 0 deletions
171
pkg/operator/podsecurityreadinesscontroller/podsecurityreadinesscontroller.go
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,171 @@ | ||
| package podsecurityreadinesscontroller | ||
|
|
||
| import ( | ||
| "context" | ||
| "time" | ||
|
|
||
| corev1 "k8s.io/api/core/v1" | ||
| apierrors "k8s.io/apimachinery/pkg/api/errors" | ||
| metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" | ||
| "k8s.io/apimachinery/pkg/labels" | ||
| "k8s.io/apimachinery/pkg/selection" | ||
| applyconfiguration "k8s.io/client-go/applyconfigurations/core/v1" | ||
| "k8s.io/client-go/kubernetes" | ||
| "k8s.io/client-go/rest" | ||
| "k8s.io/client-go/util/retry" | ||
| "k8s.io/klog/v2" | ||
| psapi "k8s.io/pod-security-admission/api" | ||
|
|
||
| "github.com/openshift/library-go/pkg/controller/factory" | ||
| "github.com/openshift/library-go/pkg/operator/events" | ||
| "github.com/openshift/library-go/pkg/operator/v1helpers" | ||
| ) | ||
|
|
||
| const ( | ||
| checkInterval = 240 * time.Minute // Adjust the interval as needed. | ||
| ) | ||
|
|
||
| var podSecurityAlertLabels = []string{ | ||
| psapi.AuditLevelLabel, | ||
| psapi.WarnLevelLabel, | ||
| } | ||
|
|
||
| // PodSecurityReadinessController checks if namespaces are ready for Pod Security Admission enforcement. | ||
| type PodSecurityReadinessController struct { | ||
| kubeClient kubernetes.Interface | ||
| operatorClient v1helpers.OperatorClient | ||
|
|
||
| warningsHandler *warningsHandler | ||
| namespaceSelector string | ||
| } | ||
|
|
||
| func NewPodSecurityReadinessController( | ||
| kubeConfig *rest.Config, | ||
| operatorClient v1helpers.OperatorClient, | ||
| recorder events.Recorder, | ||
| ) (factory.Controller, error) { | ||
| warningsHandler := &warningsHandler{} | ||
|
|
||
| kubeClientCopy := rest.CopyConfig(kubeConfig) | ||
| kubeClientCopy.WarningHandler = warningsHandler | ||
| // We don't want to overwhelm the apiserver with requests. On a cluster with | ||
| // 10k namespaces, we would send 10k + 1 requests to the apiserver. | ||
| kubeClientCopy.QPS = 2 | ||
| kubeClientCopy.Burst = 2 | ||
| kubeClient, err := kubernetes.NewForConfig(kubeClientCopy) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| selector := labels.NewSelector() | ||
| labelsRequirement, err := labels.NewRequirement(psapi.EnforceLevelLabel, selection.DoesNotExist, []string{}) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| c := &PodSecurityReadinessController{ | ||
| operatorClient: operatorClient, | ||
| kubeClient: kubeClient, | ||
| warningsHandler: warningsHandler, | ||
| namespaceSelector: selector.Add(*labelsRequirement).String(), | ||
| } | ||
|
|
||
| return factory.New(). | ||
| WithSync(c.sync). | ||
| ResyncEvery(checkInterval). | ||
| ToController("PodSecurityReadinessController", recorder), nil | ||
| } | ||
|
|
||
| func (c *PodSecurityReadinessController) sync(ctx context.Context, syncCtx factory.SyncContext) error { | ||
| nsList, err := c.kubeClient.CoreV1().Namespaces().List(ctx, metav1.ListOptions{LabelSelector: c.namespaceSelector}) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| conditions := podSecurityOperatorConditions{} | ||
| for _, ns := range nsList.Items { | ||
| err := retry.RetryOnConflict(retry.DefaultBackoff, func() error { | ||
| isViolating, err := c.isNamespaceViolating(ctx, &ns) | ||
| if apierrors.IsNotFound(err) { | ||
| return nil | ||
| } | ||
| if err != nil { | ||
| return err | ||
| } | ||
| if isViolating { | ||
| conditions.addViolation(ns.Name) | ||
| } | ||
|
|
||
| return nil | ||
| }) | ||
| if err != nil { | ||
| klog.V(2).ErrorS(err, "namespace:", ns.Name) | ||
|
|
||
| // We don't want to sync more often than the resync interval. | ||
| return nil | ||
|
|
||
| } | ||
| } | ||
|
|
||
| // We expect the Cluster's status conditions to be picked up by the status | ||
| // controller and push it into the ClusterOperator's status, where it will | ||
| // be evaluated by the ClusterFleetMechanic. | ||
| _, _, err = v1helpers.UpdateStatus(ctx, c.operatorClient, conditions.toConditionFuncs()...) | ||
| return err | ||
| } | ||
|
|
||
| func (c *PodSecurityReadinessController) isNamespaceViolating(ctx context.Context, ns *corev1.Namespace) (bool, error) { | ||
| if ns.Labels[psapi.EnforceLevelLabel] != "" { | ||
| // If someone has taken care of the enforce label, we don't need to | ||
| // check for violations. Global Config nor PS-Label-Syncer will modify | ||
| // it. | ||
| return false, nil | ||
| } | ||
|
|
||
| targetLevel := "" | ||
| for _, label := range podSecurityAlertLabels { | ||
| levelStr, ok := ns.Labels[label] | ||
| if !ok { | ||
| continue | ||
| } | ||
|
|
||
| level, err := psapi.ParseLevel(levelStr) | ||
| if err != nil { | ||
| klog.V(4).InfoS("invalid level", "namespace", ns.Name, "level", levelStr) | ||
| continue | ||
| } | ||
|
|
||
| if targetLevel == "" { | ||
| targetLevel = levelStr | ||
| continue | ||
| } | ||
|
|
||
| if psapi.CompareLevels(psapi.Level(targetLevel), level) < 0 { | ||
| targetLevel = levelStr | ||
| } | ||
| } | ||
|
|
||
| if targetLevel == "" { | ||
| // Global Config will set it to "restricted". | ||
| targetLevel = string(psapi.LevelRestricted) | ||
| } | ||
|
|
||
| nsApply := applyconfiguration.Namespace(ns.Name).WithLabels(map[string]string{ | ||
| psapi.EnforceLevelLabel: string(targetLevel), | ||
| }) | ||
|
|
||
| _, err := c.kubeClient.CoreV1(). | ||
| Namespaces(). | ||
| Apply(ctx, nsApply, metav1.ApplyOptions{ | ||
| DryRun: []string{metav1.DryRunAll}, | ||
| FieldManager: "pod-security-readiness-controller", | ||
| }) | ||
| if err != nil { | ||
| return false, err | ||
| } | ||
|
|
||
| // The information we want is in the warnings. It collects violations. | ||
| warnings := c.warningsHandler.PopAll() | ||
|
|
||
| return len(warnings) > 0, nil | ||
| } |
Oops, something went wrong.