mirror of https://github.com/k3s-io/k3s
551 lines
20 KiB
Go
551 lines
20 KiB
Go
/*
|
|
Copyright 2014 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 auth
|
|
|
|
import (
|
|
"fmt"
|
|
"path"
|
|
"regexp"
|
|
"strings"
|
|
"time"
|
|
|
|
authenticationv1 "k8s.io/api/authentication/v1"
|
|
v1 "k8s.io/api/core/v1"
|
|
apierrors "k8s.io/apimachinery/pkg/api/errors"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/apimachinery/pkg/util/sets"
|
|
"k8s.io/apimachinery/pkg/util/uuid"
|
|
"k8s.io/apimachinery/pkg/util/wait"
|
|
"k8s.io/kubernetes/plugin/pkg/admission/serviceaccount"
|
|
"k8s.io/kubernetes/test/e2e/framework"
|
|
e2elog "k8s.io/kubernetes/test/e2e/framework/log"
|
|
imageutils "k8s.io/kubernetes/test/utils/image"
|
|
|
|
"github.com/onsi/ginkgo"
|
|
"github.com/onsi/gomega"
|
|
)
|
|
|
|
var mountImage = imageutils.GetE2EImage(imageutils.Mounttest)
|
|
var inClusterClientImage = imageutils.GetE2EImage(imageutils.InClusterClient)
|
|
|
|
var _ = SIGDescribe("ServiceAccounts", func() {
|
|
f := framework.NewDefaultFramework("svcaccounts")
|
|
|
|
ginkgo.It("should ensure a single API token exists", func() {
|
|
// wait for the service account to reference a single secret
|
|
var secrets []v1.ObjectReference
|
|
framework.ExpectNoError(wait.Poll(time.Millisecond*500, time.Second*10, func() (bool, error) {
|
|
ginkgo.By("waiting for a single token reference")
|
|
sa, err := f.ClientSet.CoreV1().ServiceAccounts(f.Namespace.Name).Get("default", metav1.GetOptions{})
|
|
if apierrors.IsNotFound(err) {
|
|
e2elog.Logf("default service account was not found")
|
|
return false, nil
|
|
}
|
|
if err != nil {
|
|
e2elog.Logf("error getting default service account: %v", err)
|
|
return false, err
|
|
}
|
|
switch len(sa.Secrets) {
|
|
case 0:
|
|
e2elog.Logf("default service account has no secret references")
|
|
return false, nil
|
|
case 1:
|
|
e2elog.Logf("default service account has a single secret reference")
|
|
secrets = sa.Secrets
|
|
return true, nil
|
|
default:
|
|
return false, fmt.Errorf("default service account has too many secret references: %#v", sa.Secrets)
|
|
}
|
|
}))
|
|
|
|
// make sure the reference doesn't flutter
|
|
{
|
|
ginkgo.By("ensuring the single token reference persists")
|
|
time.Sleep(2 * time.Second)
|
|
sa, err := f.ClientSet.CoreV1().ServiceAccounts(f.Namespace.Name).Get("default", metav1.GetOptions{})
|
|
framework.ExpectNoError(err)
|
|
gomega.Expect(sa.Secrets).To(gomega.Equal(secrets))
|
|
}
|
|
|
|
// delete the referenced secret
|
|
ginkgo.By("deleting the service account token")
|
|
framework.ExpectNoError(f.ClientSet.CoreV1().Secrets(f.Namespace.Name).Delete(secrets[0].Name, nil))
|
|
|
|
// wait for the referenced secret to be removed, and another one autocreated
|
|
framework.ExpectNoError(wait.Poll(time.Millisecond*500, framework.ServiceAccountProvisionTimeout, func() (bool, error) {
|
|
ginkgo.By("waiting for a new token reference")
|
|
sa, err := f.ClientSet.CoreV1().ServiceAccounts(f.Namespace.Name).Get("default", metav1.GetOptions{})
|
|
if err != nil {
|
|
e2elog.Logf("error getting default service account: %v", err)
|
|
return false, err
|
|
}
|
|
switch len(sa.Secrets) {
|
|
case 0:
|
|
e2elog.Logf("default service account has no secret references")
|
|
return false, nil
|
|
case 1:
|
|
if sa.Secrets[0] == secrets[0] {
|
|
e2elog.Logf("default service account still has the deleted secret reference")
|
|
return false, nil
|
|
}
|
|
e2elog.Logf("default service account has a new single secret reference")
|
|
secrets = sa.Secrets
|
|
return true, nil
|
|
default:
|
|
return false, fmt.Errorf("default service account has too many secret references: %#v", sa.Secrets)
|
|
}
|
|
}))
|
|
|
|
// make sure the reference doesn't flutter
|
|
{
|
|
ginkgo.By("ensuring the single token reference persists")
|
|
time.Sleep(2 * time.Second)
|
|
sa, err := f.ClientSet.CoreV1().ServiceAccounts(f.Namespace.Name).Get("default", metav1.GetOptions{})
|
|
framework.ExpectNoError(err)
|
|
gomega.Expect(sa.Secrets).To(gomega.Equal(secrets))
|
|
}
|
|
|
|
// delete the reference from the service account
|
|
ginkgo.By("deleting the reference to the service account token")
|
|
{
|
|
sa, err := f.ClientSet.CoreV1().ServiceAccounts(f.Namespace.Name).Get("default", metav1.GetOptions{})
|
|
framework.ExpectNoError(err)
|
|
sa.Secrets = nil
|
|
_, updateErr := f.ClientSet.CoreV1().ServiceAccounts(f.Namespace.Name).Update(sa)
|
|
framework.ExpectNoError(updateErr)
|
|
}
|
|
|
|
// wait for another one to be autocreated
|
|
framework.ExpectNoError(wait.Poll(time.Millisecond*500, framework.ServiceAccountProvisionTimeout, func() (bool, error) {
|
|
ginkgo.By("waiting for a new token to be created and added")
|
|
sa, err := f.ClientSet.CoreV1().ServiceAccounts(f.Namespace.Name).Get("default", metav1.GetOptions{})
|
|
if err != nil {
|
|
e2elog.Logf("error getting default service account: %v", err)
|
|
return false, err
|
|
}
|
|
switch len(sa.Secrets) {
|
|
case 0:
|
|
e2elog.Logf("default service account has no secret references")
|
|
return false, nil
|
|
case 1:
|
|
e2elog.Logf("default service account has a new single secret reference")
|
|
secrets = sa.Secrets
|
|
return true, nil
|
|
default:
|
|
return false, fmt.Errorf("default service account has too many secret references: %#v", sa.Secrets)
|
|
}
|
|
}))
|
|
|
|
// make sure the reference doesn't flutter
|
|
{
|
|
ginkgo.By("ensuring the single token reference persists")
|
|
time.Sleep(2 * time.Second)
|
|
sa, err := f.ClientSet.CoreV1().ServiceAccounts(f.Namespace.Name).Get("default", metav1.GetOptions{})
|
|
framework.ExpectNoError(err)
|
|
gomega.Expect(sa.Secrets).To(gomega.Equal(secrets))
|
|
}
|
|
})
|
|
|
|
/*
|
|
Release: v1.9
|
|
Testname: Service Account Tokens Must AutoMount
|
|
Description: Ensure that Service Account keys are mounted into the Container. Pod
|
|
contains three containers each will read Service Account token,
|
|
root CA and default namespace respectively from the default API
|
|
Token Mount path. All these three files MUST exist and the Service
|
|
Account mount path MUST be auto mounted to the Container.
|
|
*/
|
|
framework.ConformanceIt("should mount an API token into pods ", func() {
|
|
var rootCAContent string
|
|
|
|
sa, err := f.ClientSet.CoreV1().ServiceAccounts(f.Namespace.Name).Create(&v1.ServiceAccount{ObjectMeta: metav1.ObjectMeta{Name: "mount-test"}})
|
|
framework.ExpectNoError(err)
|
|
|
|
// Standard get, update retry loop
|
|
framework.ExpectNoError(wait.Poll(time.Millisecond*500, framework.ServiceAccountProvisionTimeout, func() (bool, error) {
|
|
ginkgo.By("getting the auto-created API token")
|
|
sa, err := f.ClientSet.CoreV1().ServiceAccounts(f.Namespace.Name).Get("mount-test", metav1.GetOptions{})
|
|
if apierrors.IsNotFound(err) {
|
|
e2elog.Logf("mount-test service account was not found")
|
|
return false, nil
|
|
}
|
|
if err != nil {
|
|
e2elog.Logf("error getting mount-test service account: %v", err)
|
|
return false, err
|
|
}
|
|
if len(sa.Secrets) == 0 {
|
|
e2elog.Logf("mount-test service account has no secret references")
|
|
return false, nil
|
|
}
|
|
for _, secretRef := range sa.Secrets {
|
|
secret, err := f.ClientSet.CoreV1().Secrets(f.Namespace.Name).Get(secretRef.Name, metav1.GetOptions{})
|
|
if err != nil {
|
|
e2elog.Logf("Error getting secret %s: %v", secretRef.Name, err)
|
|
continue
|
|
}
|
|
if secret.Type == v1.SecretTypeServiceAccountToken {
|
|
rootCAContent = string(secret.Data[v1.ServiceAccountRootCAKey])
|
|
return true, nil
|
|
}
|
|
}
|
|
|
|
e2elog.Logf("default service account has no secret references to valid service account tokens")
|
|
return false, nil
|
|
}))
|
|
|
|
zero := int64(0)
|
|
pod, err := f.ClientSet.CoreV1().Pods(f.Namespace.Name).Create(&v1.Pod{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "pod-service-account-" + string(uuid.NewUUID()),
|
|
},
|
|
Spec: v1.PodSpec{
|
|
ServiceAccountName: sa.Name,
|
|
Containers: []v1.Container{{
|
|
Name: "test",
|
|
Image: imageutils.GetE2EImage(imageutils.BusyBox),
|
|
Command: []string{"sleep", "100000"}, // run and pause
|
|
}},
|
|
TerminationGracePeriodSeconds: &zero, // terminate quickly when deleted
|
|
RestartPolicy: v1.RestartPolicyNever, // never restart
|
|
},
|
|
})
|
|
framework.ExpectNoError(err)
|
|
framework.ExpectNoError(framework.WaitForPodRunningInNamespace(f.ClientSet, pod))
|
|
|
|
mountedToken, err := f.ReadFileViaContainer(pod.Name, pod.Spec.Containers[0].Name, path.Join(serviceaccount.DefaultAPITokenMountPath, v1.ServiceAccountTokenKey))
|
|
framework.ExpectNoError(err)
|
|
mountedCA, err := f.ReadFileViaContainer(pod.Name, pod.Spec.Containers[0].Name, path.Join(serviceaccount.DefaultAPITokenMountPath, v1.ServiceAccountRootCAKey))
|
|
framework.ExpectNoError(err)
|
|
mountedNamespace, err := f.ReadFileViaContainer(pod.Name, pod.Spec.Containers[0].Name, path.Join(serviceaccount.DefaultAPITokenMountPath, v1.ServiceAccountNamespaceKey))
|
|
framework.ExpectNoError(err)
|
|
|
|
// CA and namespace should be identical
|
|
gomega.Expect(mountedCA).To(gomega.Equal(rootCAContent))
|
|
gomega.Expect(mountedNamespace).To(gomega.Equal(f.Namespace.Name))
|
|
// Token should be a valid credential that identifies the pod's service account
|
|
tokenReview := &authenticationv1.TokenReview{Spec: authenticationv1.TokenReviewSpec{Token: mountedToken}}
|
|
tokenReview, err = f.ClientSet.AuthenticationV1().TokenReviews().Create(tokenReview)
|
|
framework.ExpectNoError(err)
|
|
gomega.Expect(tokenReview.Status.Authenticated).To(gomega.Equal(true))
|
|
gomega.Expect(tokenReview.Status.Error).To(gomega.Equal(""))
|
|
gomega.Expect(tokenReview.Status.User.Username).To(gomega.Equal("system:serviceaccount:" + f.Namespace.Name + ":" + sa.Name))
|
|
groups := sets.NewString(tokenReview.Status.User.Groups...)
|
|
gomega.Expect(groups.Has("system:authenticated")).To(gomega.Equal(true), fmt.Sprintf("expected system:authenticated group, had %v", groups.List()))
|
|
gomega.Expect(groups.Has("system:serviceaccounts")).To(gomega.Equal(true), fmt.Sprintf("expected system:serviceaccounts group, had %v", groups.List()))
|
|
gomega.Expect(groups.Has("system:serviceaccounts:"+f.Namespace.Name)).To(gomega.Equal(true), fmt.Sprintf("expected system:serviceaccounts:"+f.Namespace.Name+" group, had %v", groups.List()))
|
|
})
|
|
|
|
/*
|
|
Release: v1.9
|
|
Testname: Service account tokens auto mount optionally
|
|
Description: Ensure that Service Account keys are mounted into the Pod only
|
|
when AutoMountServiceToken is not set to false. We test the
|
|
following scenarios here.
|
|
1. Create Pod, Pod Spec has AutomountServiceAccountToken set to nil
|
|
a) Service Account with default value,
|
|
b) Service Account is an configured AutomountServiceAccountToken set to true,
|
|
c) Service Account is an configured AutomountServiceAccountToken set to false
|
|
2. Create Pod, Pod Spec has AutomountServiceAccountToken set to true
|
|
a) Service Account with default value,
|
|
b) Service Account is configured with AutomountServiceAccountToken set to true,
|
|
c) Service Account is configured with AutomountServiceAccountToken set to false
|
|
3. Create Pod, Pod Spec has AutomountServiceAccountToken set to false
|
|
a) Service Account with default value,
|
|
b) Service Account is configured with AutomountServiceAccountToken set to true,
|
|
c) Service Account is configured with AutomountServiceAccountToken set to false
|
|
|
|
The Containers running in these pods MUST verify that the ServiceTokenVolume path is
|
|
auto mounted only when Pod Spec has AutomountServiceAccountToken not set to false
|
|
and ServiceAccount object has AutomountServiceAccountToken not set to false, this
|
|
include test cases 1a,1b,2a,2b and 2c.
|
|
In the test cases 1c,3a,3b and 3c the ServiceTokenVolume MUST not be auto mounted.
|
|
*/
|
|
framework.ConformanceIt("should allow opting out of API token automount ", func() {
|
|
|
|
var err error
|
|
trueValue := true
|
|
falseValue := false
|
|
mountSA := &v1.ServiceAccount{ObjectMeta: metav1.ObjectMeta{Name: "mount"}, AutomountServiceAccountToken: &trueValue}
|
|
nomountSA := &v1.ServiceAccount{ObjectMeta: metav1.ObjectMeta{Name: "nomount"}, AutomountServiceAccountToken: &falseValue}
|
|
mountSA, err = f.ClientSet.CoreV1().ServiceAccounts(f.Namespace.Name).Create(mountSA)
|
|
framework.ExpectNoError(err)
|
|
nomountSA, err = f.ClientSet.CoreV1().ServiceAccounts(f.Namespace.Name).Create(nomountSA)
|
|
framework.ExpectNoError(err)
|
|
|
|
// Standard get, update retry loop
|
|
framework.ExpectNoError(wait.Poll(time.Millisecond*500, framework.ServiceAccountProvisionTimeout, func() (bool, error) {
|
|
ginkgo.By("getting the auto-created API token")
|
|
sa, err := f.ClientSet.CoreV1().ServiceAccounts(f.Namespace.Name).Get(mountSA.Name, metav1.GetOptions{})
|
|
if apierrors.IsNotFound(err) {
|
|
e2elog.Logf("mount service account was not found")
|
|
return false, nil
|
|
}
|
|
if err != nil {
|
|
e2elog.Logf("error getting mount service account: %v", err)
|
|
return false, err
|
|
}
|
|
if len(sa.Secrets) == 0 {
|
|
e2elog.Logf("mount service account has no secret references")
|
|
return false, nil
|
|
}
|
|
for _, secretRef := range sa.Secrets {
|
|
secret, err := f.ClientSet.CoreV1().Secrets(f.Namespace.Name).Get(secretRef.Name, metav1.GetOptions{})
|
|
if err != nil {
|
|
e2elog.Logf("Error getting secret %s: %v", secretRef.Name, err)
|
|
continue
|
|
}
|
|
if secret.Type == v1.SecretTypeServiceAccountToken {
|
|
return true, nil
|
|
}
|
|
}
|
|
|
|
e2elog.Logf("default service account has no secret references to valid service account tokens")
|
|
return false, nil
|
|
}))
|
|
|
|
testcases := []struct {
|
|
PodName string
|
|
ServiceAccountName string
|
|
AutomountPodSpec *bool
|
|
ExpectTokenVolume bool
|
|
}{
|
|
{
|
|
PodName: "pod-service-account-defaultsa",
|
|
ServiceAccountName: "default",
|
|
AutomountPodSpec: nil,
|
|
ExpectTokenVolume: true, // default is true
|
|
},
|
|
{
|
|
PodName: "pod-service-account-mountsa",
|
|
ServiceAccountName: mountSA.Name,
|
|
AutomountPodSpec: nil,
|
|
ExpectTokenVolume: true,
|
|
},
|
|
{
|
|
PodName: "pod-service-account-nomountsa",
|
|
ServiceAccountName: nomountSA.Name,
|
|
AutomountPodSpec: nil,
|
|
ExpectTokenVolume: false,
|
|
},
|
|
|
|
// Make sure pod spec trumps when opting in
|
|
{
|
|
PodName: "pod-service-account-defaultsa-mountspec",
|
|
ServiceAccountName: "default",
|
|
AutomountPodSpec: &trueValue,
|
|
ExpectTokenVolume: true,
|
|
},
|
|
{
|
|
PodName: "pod-service-account-mountsa-mountspec",
|
|
ServiceAccountName: mountSA.Name,
|
|
AutomountPodSpec: &trueValue,
|
|
ExpectTokenVolume: true,
|
|
},
|
|
{
|
|
PodName: "pod-service-account-nomountsa-mountspec",
|
|
ServiceAccountName: nomountSA.Name,
|
|
AutomountPodSpec: &trueValue,
|
|
ExpectTokenVolume: true, // pod spec trumps
|
|
},
|
|
|
|
// Make sure pod spec trumps when opting out
|
|
{
|
|
PodName: "pod-service-account-defaultsa-nomountspec",
|
|
ServiceAccountName: "default",
|
|
AutomountPodSpec: &falseValue,
|
|
ExpectTokenVolume: false, // pod spec trumps
|
|
},
|
|
{
|
|
PodName: "pod-service-account-mountsa-nomountspec",
|
|
ServiceAccountName: mountSA.Name,
|
|
AutomountPodSpec: &falseValue,
|
|
ExpectTokenVolume: false, // pod spec trumps
|
|
},
|
|
{
|
|
PodName: "pod-service-account-nomountsa-nomountspec",
|
|
ServiceAccountName: nomountSA.Name,
|
|
AutomountPodSpec: &falseValue,
|
|
ExpectTokenVolume: false, // pod spec trumps
|
|
},
|
|
}
|
|
|
|
for _, tc := range testcases {
|
|
pod := &v1.Pod{
|
|
ObjectMeta: metav1.ObjectMeta{Name: tc.PodName},
|
|
Spec: v1.PodSpec{
|
|
Containers: []v1.Container{{Name: "token-test", Image: mountImage}},
|
|
RestartPolicy: v1.RestartPolicyNever,
|
|
ServiceAccountName: tc.ServiceAccountName,
|
|
AutomountServiceAccountToken: tc.AutomountPodSpec,
|
|
},
|
|
}
|
|
createdPod, err := f.ClientSet.CoreV1().Pods(f.Namespace.Name).Create(pod)
|
|
framework.ExpectNoError(err)
|
|
e2elog.Logf("created pod %s", tc.PodName)
|
|
|
|
hasServiceAccountTokenVolume := false
|
|
for _, c := range createdPod.Spec.Containers {
|
|
for _, vm := range c.VolumeMounts {
|
|
if vm.MountPath == serviceaccount.DefaultAPITokenMountPath {
|
|
hasServiceAccountTokenVolume = true
|
|
}
|
|
}
|
|
}
|
|
|
|
if hasServiceAccountTokenVolume != tc.ExpectTokenVolume {
|
|
framework.Failf("%s: expected volume=%v, got %v (%#v)", tc.PodName, tc.ExpectTokenVolume, hasServiceAccountTokenVolume, createdPod)
|
|
} else {
|
|
e2elog.Logf("pod %s service account token volume mount: %v", tc.PodName, hasServiceAccountTokenVolume)
|
|
}
|
|
}
|
|
})
|
|
|
|
ginkgo.It("should support InClusterConfig with token rotation [Slow] [Feature:TokenRequestProjection]", func() {
|
|
cfg, err := framework.LoadConfig()
|
|
framework.ExpectNoError(err)
|
|
|
|
if _, err := f.ClientSet.CoreV1().ConfigMaps(f.Namespace.Name).Create(&v1.ConfigMap{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "kube-root-ca.crt",
|
|
},
|
|
Data: map[string]string{
|
|
"ca.crt": string(cfg.TLSClientConfig.CAData),
|
|
},
|
|
}); err != nil && !apierrors.IsAlreadyExists(err) {
|
|
framework.Failf("Unexpected err creating kube-ca-crt: %v", err)
|
|
}
|
|
|
|
tenMin := int64(10 * 60)
|
|
pod := &v1.Pod{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "inclusterclient"},
|
|
Spec: v1.PodSpec{
|
|
Containers: []v1.Container{{
|
|
Name: "inclusterclient",
|
|
Image: inClusterClientImage,
|
|
VolumeMounts: []v1.VolumeMount{{
|
|
MountPath: "/var/run/secrets/kubernetes.io/serviceaccount",
|
|
Name: "kube-api-access-e2e",
|
|
ReadOnly: true,
|
|
}},
|
|
}},
|
|
RestartPolicy: v1.RestartPolicyNever,
|
|
ServiceAccountName: "default",
|
|
Volumes: []v1.Volume{{
|
|
Name: "kube-api-access-e2e",
|
|
VolumeSource: v1.VolumeSource{
|
|
Projected: &v1.ProjectedVolumeSource{
|
|
Sources: []v1.VolumeProjection{
|
|
{
|
|
ServiceAccountToken: &v1.ServiceAccountTokenProjection{
|
|
Path: "token",
|
|
ExpirationSeconds: &tenMin,
|
|
},
|
|
},
|
|
{
|
|
ConfigMap: &v1.ConfigMapProjection{
|
|
LocalObjectReference: v1.LocalObjectReference{
|
|
Name: "kube-root-ca.crt",
|
|
},
|
|
Items: []v1.KeyToPath{
|
|
{
|
|
Key: "ca.crt",
|
|
Path: "ca.crt",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
DownwardAPI: &v1.DownwardAPIProjection{
|
|
Items: []v1.DownwardAPIVolumeFile{
|
|
{
|
|
Path: "namespace",
|
|
FieldRef: &v1.ObjectFieldSelector{
|
|
APIVersion: "v1",
|
|
FieldPath: "metadata.namespace",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}},
|
|
},
|
|
}
|
|
pod, err = f.ClientSet.CoreV1().Pods(f.Namespace.Name).Create(pod)
|
|
framework.ExpectNoError(err)
|
|
|
|
framework.Logf("created pod")
|
|
if !framework.CheckPodsRunningReady(f.ClientSet, f.Namespace.Name, []string{pod.Name}, time.Minute) {
|
|
framework.Failf("pod %q in ns %q never became ready", pod.Name, f.Namespace.Name)
|
|
}
|
|
|
|
framework.Logf("pod is ready")
|
|
|
|
var logs string
|
|
if err := wait.Poll(1*time.Minute, 20*time.Minute, func() (done bool, err error) {
|
|
framework.Logf("polling logs")
|
|
logs, err = framework.GetPodLogs(f.ClientSet, f.Namespace.Name, "inclusterclient", "inclusterclient")
|
|
if err != nil {
|
|
framework.Logf("Error pulling logs: %v", err)
|
|
return false, nil
|
|
}
|
|
tokenCount, err := parseInClusterClientLogs(logs)
|
|
if err != nil {
|
|
return false, fmt.Errorf("inclusterclient reported an error: %v", err)
|
|
}
|
|
if tokenCount < 2 {
|
|
framework.Logf("Retrying. Still waiting to see more unique tokens: got=%d, want=2", tokenCount)
|
|
return false, nil
|
|
}
|
|
return true, nil
|
|
}); err != nil {
|
|
framework.Failf("Unexpected error: %v\n%s", err, logs)
|
|
}
|
|
})
|
|
})
|
|
|
|
var reportLogsParser = regexp.MustCompile("([a-zA-Z0-9-_]*)=([a-zA-Z0-9-_]*)$")
|
|
|
|
func parseInClusterClientLogs(logs string) (int, error) {
|
|
seenTokens := map[string]struct{}{}
|
|
|
|
lines := strings.Split(logs, "\n")
|
|
for _, line := range lines {
|
|
parts := reportLogsParser.FindStringSubmatch(line)
|
|
if len(parts) != 3 {
|
|
continue
|
|
}
|
|
|
|
key, value := parts[1], parts[2]
|
|
switch key {
|
|
case "authz_header":
|
|
if value == "<empty>" {
|
|
return 0, fmt.Errorf("saw empty Authorization header")
|
|
}
|
|
seenTokens[value] = struct{}{}
|
|
case "status":
|
|
if value == "failed" {
|
|
return 0, fmt.Errorf("saw status=failed")
|
|
}
|
|
}
|
|
}
|
|
|
|
return len(seenTokens), nil
|
|
}
|