mirror of https://github.com/k3s-io/k3s
326 lines
11 KiB
Go
326 lines
11 KiB
Go
/*
|
|
Copyright 2018 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 serving
|
|
|
|
import (
|
|
"crypto/tls"
|
|
"crypto/x509"
|
|
"fmt"
|
|
"io"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"os"
|
|
"path"
|
|
"strings"
|
|
"testing"
|
|
|
|
"k8s.io/apiserver/pkg/server"
|
|
"k8s.io/apiserver/pkg/server/options"
|
|
"k8s.io/cloud-provider"
|
|
cloudctrlmgrtesting "k8s.io/kubernetes/cmd/cloud-controller-manager/app/testing"
|
|
kubeapiservertesting "k8s.io/kubernetes/cmd/kube-apiserver/app/testing"
|
|
kubectrlmgrtesting "k8s.io/kubernetes/cmd/kube-controller-manager/app/testing"
|
|
kubeschedulertesting "k8s.io/kubernetes/cmd/kube-scheduler/app/testing"
|
|
"k8s.io/kubernetes/pkg/cloudprovider/providers/fake"
|
|
"k8s.io/kubernetes/test/integration/framework"
|
|
)
|
|
|
|
type componentTester interface {
|
|
StartTestServer(t kubectrlmgrtesting.Logger, customFlags []string) (*options.SecureServingOptionsWithLoopback, *server.SecureServingInfo, *server.DeprecatedInsecureServingInfo, func(), error)
|
|
}
|
|
|
|
type kubeControllerManagerTester struct{}
|
|
|
|
func (kubeControllerManagerTester) StartTestServer(t kubectrlmgrtesting.Logger, customFlags []string) (*options.SecureServingOptionsWithLoopback, *server.SecureServingInfo, *server.DeprecatedInsecureServingInfo, func(), error) {
|
|
// avoid starting any controller loops, we're just testing serving
|
|
customFlags = append([]string{"--controllers="}, customFlags...)
|
|
gotResult, err := kubectrlmgrtesting.StartTestServer(t, customFlags)
|
|
if err != nil {
|
|
return nil, nil, nil, nil, err
|
|
}
|
|
return gotResult.Options.SecureServing, gotResult.Config.SecureServing, gotResult.Config.InsecureServing, gotResult.TearDownFn, err
|
|
}
|
|
|
|
type cloudControllerManagerTester struct{}
|
|
|
|
func (cloudControllerManagerTester) StartTestServer(t kubectrlmgrtesting.Logger, customFlags []string) (*options.SecureServingOptionsWithLoopback, *server.SecureServingInfo, *server.DeprecatedInsecureServingInfo, func(), error) {
|
|
gotResult, err := cloudctrlmgrtesting.StartTestServer(t, customFlags)
|
|
if err != nil {
|
|
return nil, nil, nil, nil, err
|
|
}
|
|
return gotResult.Options.SecureServing, gotResult.Config.SecureServing, gotResult.Config.InsecureServing, gotResult.TearDownFn, err
|
|
}
|
|
|
|
type kubeSchedulerTester struct{}
|
|
|
|
func (kubeSchedulerTester) StartTestServer(t kubectrlmgrtesting.Logger, customFlags []string) (*options.SecureServingOptionsWithLoopback, *server.SecureServingInfo, *server.DeprecatedInsecureServingInfo, func(), error) {
|
|
gotResult, err := kubeschedulertesting.StartTestServer(t, customFlags)
|
|
if err != nil {
|
|
return nil, nil, nil, nil, err
|
|
}
|
|
return gotResult.Options.SecureServing, gotResult.Config.SecureServing, gotResult.Config.InsecureServing, gotResult.TearDownFn, err
|
|
}
|
|
|
|
func TestComponentSecureServingAndAuth(t *testing.T) {
|
|
if !cloudprovider.IsCloudProvider("fake") {
|
|
cloudprovider.RegisterCloudProvider("fake", fakeCloudProviderFactory)
|
|
}
|
|
|
|
// Insulate this test from picking up in-cluster config when run inside a pod
|
|
// We can't assume we have permissions to write to /var/run/secrets/... from a unit test to mock in-cluster config for testing
|
|
originalHost := os.Getenv("KUBERNETES_SERVICE_HOST")
|
|
if len(originalHost) > 0 {
|
|
os.Setenv("KUBERNETES_SERVICE_HOST", "")
|
|
defer os.Setenv("KUBERNETES_SERVICE_HOST", originalHost)
|
|
}
|
|
|
|
// authenticate to apiserver via bearer token
|
|
token := "flwqkenfjasasdfmwerasd"
|
|
tokenFile, err := ioutil.TempFile("", "kubeconfig")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
tokenFile.WriteString(fmt.Sprintf(`
|
|
%s,system:kube-controller-manager,system:kube-controller-manager,""
|
|
`, token))
|
|
tokenFile.Close()
|
|
|
|
// start apiserver
|
|
server := kubeapiservertesting.StartTestServerOrDie(t, nil, []string{
|
|
"--token-auth-file", tokenFile.Name(),
|
|
"--authorization-mode", "RBAC",
|
|
}, framework.SharedEtcd())
|
|
defer server.TearDownFn()
|
|
|
|
// create kubeconfig for the apiserver
|
|
apiserverConfig, err := ioutil.TempFile("", "kubeconfig")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
apiserverConfig.WriteString(fmt.Sprintf(`
|
|
apiVersion: v1
|
|
kind: Config
|
|
clusters:
|
|
- cluster:
|
|
server: %s
|
|
certificate-authority: %s
|
|
name: integration
|
|
contexts:
|
|
- context:
|
|
cluster: integration
|
|
user: controller-manager
|
|
name: default-context
|
|
current-context: default-context
|
|
users:
|
|
- name: controller-manager
|
|
user:
|
|
token: %s
|
|
`, server.ClientConfig.Host, server.ServerOpts.SecureServing.ServerCert.CertKey.CertFile, token))
|
|
apiserverConfig.Close()
|
|
|
|
// create BROKEN kubeconfig for the apiserver
|
|
brokenApiserverConfig, err := ioutil.TempFile("", "kubeconfig")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
brokenApiserverConfig.WriteString(fmt.Sprintf(`
|
|
apiVersion: v1
|
|
kind: Config
|
|
clusters:
|
|
- cluster:
|
|
server: %s
|
|
certificate-authority: %s
|
|
name: integration
|
|
contexts:
|
|
- context:
|
|
cluster: integration
|
|
user: controller-manager
|
|
name: default-context
|
|
current-context: default-context
|
|
users:
|
|
- name: controller-manager
|
|
user:
|
|
token: WRONGTOKEN
|
|
`, server.ClientConfig.Host, server.ServerOpts.SecureServing.ServerCert.CertKey.CertFile))
|
|
brokenApiserverConfig.Close()
|
|
|
|
tests := []struct {
|
|
name string
|
|
tester componentTester
|
|
extraFlags []string
|
|
}{
|
|
{"kube-controller-manager", kubeControllerManagerTester{}, nil},
|
|
{"cloud-controller-manager", cloudControllerManagerTester{}, []string{"--cloud-provider=fake"}},
|
|
{"kube-scheduler", kubeSchedulerTester{}, nil},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
testComponent(t, tt.tester, apiserverConfig.Name(), brokenApiserverConfig.Name(), token, tt.extraFlags)
|
|
})
|
|
}
|
|
}
|
|
|
|
func testComponent(t *testing.T, tester componentTester, kubeconfig, brokenKubeconfig, token string, extraFlags []string) {
|
|
tests := []struct {
|
|
name string
|
|
flags []string
|
|
path string
|
|
anonymous bool // to use the token or not
|
|
wantErr bool
|
|
wantSecureCode, wantInsecureCode *int
|
|
}{
|
|
{"no-flags", nil, "/healthz", false, true, nil, nil},
|
|
{"insecurely /healthz", []string{
|
|
"--secure-port=0",
|
|
"--port=10253",
|
|
"--kubeconfig", kubeconfig,
|
|
"--leader-elect=false",
|
|
}, "/healthz", true, false, nil, intPtr(http.StatusOK)},
|
|
{"insecurely /metrics", []string{
|
|
"--secure-port=0",
|
|
"--port=10253",
|
|
"--kubeconfig", kubeconfig,
|
|
"--leader-elect=false",
|
|
}, "/metrics", true, false, nil, intPtr(http.StatusOK)},
|
|
{"/healthz without authn/authz", []string{
|
|
"--port=0",
|
|
"--kubeconfig", kubeconfig,
|
|
"--leader-elect=false",
|
|
}, "/healthz", true, false, intPtr(http.StatusOK), nil},
|
|
{"/metrics without authn/authz", []string{
|
|
"--kubeconfig", kubeconfig,
|
|
"--leader-elect=false",
|
|
"--port=10253",
|
|
}, "/metrics", true, false, intPtr(http.StatusForbidden), intPtr(http.StatusOK)},
|
|
{"authorization skipped for /healthz with authn/authz", []string{
|
|
"--port=0",
|
|
"--authentication-kubeconfig", kubeconfig,
|
|
"--authorization-kubeconfig", kubeconfig,
|
|
"--kubeconfig", kubeconfig,
|
|
"--leader-elect=false",
|
|
}, "/healthz", false, false, intPtr(http.StatusOK), nil},
|
|
{"authorization skipped for /healthz with BROKEN authn/authz", []string{
|
|
"--port=0",
|
|
"--authentication-skip-lookup", // to survive unaccessible extensions-apiserver-authentication configmap
|
|
"--authentication-kubeconfig", brokenKubeconfig,
|
|
"--authorization-kubeconfig", brokenKubeconfig,
|
|
"--kubeconfig", kubeconfig,
|
|
"--leader-elect=false",
|
|
}, "/healthz", false, false, intPtr(http.StatusOK), nil},
|
|
{"not authorized /metrics", []string{
|
|
"--port=0",
|
|
"--authentication-kubeconfig", kubeconfig,
|
|
"--authorization-kubeconfig", kubeconfig,
|
|
"--kubeconfig", kubeconfig,
|
|
"--leader-elect=false",
|
|
}, "/metrics", false, false, intPtr(http.StatusForbidden), nil},
|
|
{"not authorized /metrics with BROKEN authn/authz", []string{
|
|
"--port=10253",
|
|
"--authentication-kubeconfig", kubeconfig,
|
|
"--authorization-kubeconfig", brokenKubeconfig,
|
|
"--kubeconfig", kubeconfig,
|
|
"--leader-elect=false",
|
|
}, "/metrics", false, false, intPtr(http.StatusInternalServerError), intPtr(http.StatusOK)},
|
|
{"always-allowed /metrics with BROKEN authn/authz", []string{
|
|
"--port=0",
|
|
"--authentication-skip-lookup", // to survive unaccessible extensions-apiserver-authentication configmap
|
|
"--authentication-kubeconfig", kubeconfig,
|
|
"--authorization-kubeconfig", kubeconfig,
|
|
"--authorization-always-allow-paths", "/healthz,/metrics",
|
|
"--kubeconfig", kubeconfig,
|
|
"--leader-elect=false",
|
|
}, "/metrics", false, false, intPtr(http.StatusOK), nil},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
secureOptions, secureInfo, insecureInfo, tearDownFn, err := tester.StartTestServer(t, append(append([]string{}, tt.flags...), extraFlags...))
|
|
if tearDownFn != nil {
|
|
defer tearDownFn()
|
|
}
|
|
if (err != nil) != tt.wantErr {
|
|
t.Fatalf("StartTestServer() error = %v, wantErr %v", err, tt.wantErr)
|
|
}
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
if want, got := tt.wantSecureCode != nil, secureInfo != nil; want != got {
|
|
t.Errorf("SecureServing enabled: expected=%v got=%v", want, got)
|
|
} else if want {
|
|
url := fmt.Sprintf("https://%s%s", secureInfo.Listener.Addr().String(), tt.path)
|
|
url = strings.Replace(url, "[::]", "127.0.0.1", -1) // switch to IPv4 because the self-signed cert does not support [::]
|
|
|
|
// read self-signed server cert disk
|
|
pool := x509.NewCertPool()
|
|
serverCertPath := path.Join(secureOptions.ServerCert.CertDirectory, secureOptions.ServerCert.PairName+".crt")
|
|
serverCert, err := ioutil.ReadFile(serverCertPath)
|
|
if err != nil {
|
|
t.Fatalf("Failed to read component server cert %q: %v", serverCertPath, err)
|
|
}
|
|
pool.AppendCertsFromPEM(serverCert)
|
|
tr := &http.Transport{
|
|
TLSClientConfig: &tls.Config{
|
|
RootCAs: pool,
|
|
},
|
|
}
|
|
|
|
client := &http.Client{Transport: tr}
|
|
req, err := http.NewRequest("GET", url, nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !tt.anonymous {
|
|
req.Header.Add("Authorization", fmt.Sprintf("Token %s", token))
|
|
}
|
|
r, err := client.Do(req)
|
|
if err != nil {
|
|
t.Fatalf("failed to GET %s from component: %v", tt.path, err)
|
|
}
|
|
|
|
body, err := ioutil.ReadAll(r.Body)
|
|
defer r.Body.Close()
|
|
if got, expected := r.StatusCode, *tt.wantSecureCode; got != expected {
|
|
t.Fatalf("expected http %d at %s of component, got: %d %q", expected, tt.path, got, string(body))
|
|
}
|
|
}
|
|
|
|
if want, got := tt.wantInsecureCode != nil, insecureInfo != nil; want != got {
|
|
t.Errorf("InsecureServing enabled: expected=%v got=%v", want, got)
|
|
} else if want {
|
|
url := fmt.Sprintf("http://%s%s", insecureInfo.Listener.Addr().String(), tt.path)
|
|
r, err := http.Get(url)
|
|
if err != nil {
|
|
t.Fatalf("failed to GET %s from component: %v", tt.path, err)
|
|
}
|
|
body, err := ioutil.ReadAll(r.Body)
|
|
defer r.Body.Close()
|
|
if got, expected := r.StatusCode, *tt.wantInsecureCode; got != expected {
|
|
t.Fatalf("expected http %d at %s of component, got: %d %q", expected, tt.path, got, string(body))
|
|
}
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func intPtr(x int) *int {
|
|
return &x
|
|
}
|
|
|
|
func fakeCloudProviderFactory(io.Reader) (cloudprovider.Interface, error) {
|
|
return &fake.FakeCloud{}, nil
|
|
}
|