54 lines
1.4 KiB
Go
54 lines
1.4 KiB
Go
/*
|
|
Copyright 2015 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 metrics
|
|
|
|
import "testing"
|
|
|
|
func TestCleanUserAgent(t *testing.T) {
|
|
panicBuf := []byte{198, 73, 129, 133, 90, 216, 104, 29, 13, 134, 209, 233, 30, 0, 22}
|
|
|
|
for _, tc := range []struct {
|
|
In string
|
|
Out string
|
|
}{
|
|
{
|
|
In: "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_11_6) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/52.0.2743.116 Safari/537.36",
|
|
Out: "Browser",
|
|
},
|
|
{
|
|
In: "kubectl/v1.2.4",
|
|
Out: "kubectl/v1.2.4",
|
|
},
|
|
{
|
|
In: `C:\Users\Kubernetes\kubectl.exe/v1.5.4`,
|
|
Out: "kubectl.exe/v1.5.4",
|
|
},
|
|
{
|
|
In: `C:\Program Files\kubectl.exe/v1.5.4`,
|
|
Out: "kubectl.exe/v1.5.4",
|
|
},
|
|
{
|
|
// This malicious input courtesy of enisoc.
|
|
In: string(panicBuf) + "kubectl.exe",
|
|
Out: "kubectl.exe",
|
|
},
|
|
} {
|
|
if cleanUserAgent(tc.In) != tc.Out {
|
|
t.Errorf("Failed to clean User-Agent: %s", tc.In)
|
|
}
|
|
}
|
|
}
|