github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2025-07-29 14:37:00 +00:00
Code Issues Projects Releases Wiki Activity

Add fuzz targets for Duration, MicroTime, and Time

Browse Source
This commit is contained in:
Brendan Chang 2019-10-21 18:54:28 -04:00
parent b717be8269
commit d2074fcdb8
2 changed files with 61 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
test/fuzz/yaml/BUILD
View File

@ -6,6 +6,7 @@ go_library(
importpath = "k8s.io/kubernetes/test/fuzz/yaml",
visibility = ["//visibility:private"],
deps = [
"//staging/src/k8s.io/apimachinery/pkg/apis/meta/v1:go_default_library",
"//vendor/gopkg.in/yaml.v2:go_default_library",
"//vendor/sigs.k8s.io/yaml:go_default_library",
],

60
test/fuzz/yaml/yaml.go
View File

@ -20,10 +20,51 @@ limitations under the License.
package yaml
import (
"bytes"
"gopkg.in/yaml.v2"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
sigyaml "sigs.k8s.io/yaml"
)
// FuzzDuration is a fuzz target for unmarshaling Duration defined in "k8s.io/apimachinery/pkg/apis/meta/v1".
// This target also checks that the unmarshaled result can be marshaled back to the input.
func FuzzDuration(b []byte) int {
var unmarshalResult struct {
D metav1.Duration `json:"d"`
}
if err := yaml.Unmarshal(b, &unmarshalResult); err != nil {
return 0
}
marshalResult, err := yaml.Marshal(&unmarshalResult)
if err != nil {
panic(err)
}
if !bytes.Equal(marshalResult, b) {
panic("marshalResult != input")
}
return 1
}
// FuzzMicroTime is a fuzz target for unmarshaling MicroTime defined in "k8s.io/apimachinery/pkg/apis/meta/v1".
// This target also checks that the unmarshaled result can be marshaled back to the input.
func FuzzMicroTime(b []byte) int {
var unmarshalResult struct {
T metav1.MicroTime `json:"t"`
}
if err := yaml.Unmarshal(b, &unmarshalResult); err != nil {
return 0
}
marshalResult, err := yaml.Marshal(&unmarshalResult)
if err != nil {
panic(err)
}
if !bytes.Equal(marshalResult, b) {
panic("marshalResult != input")
}
return 1
}
// FuzzSigYaml is a fuzz target for "sigs.k8s.io/yaml" unmarshaling.
func FuzzSigYaml(b []byte) int {
t := struct{}{}
@ -38,6 +79,25 @@ func FuzzSigYaml(b []byte) int {
return out
}
// FuzzTime is a fuzz target for unmarshaling Time defined in "k8s.io/apimachinery/pkg/apis/meta/v1".
// This target also checks that the unmarshaled result can be marshaled back to the input.
func FuzzTime(b []byte) int {
var unmarshalResult struct {
T metav1.Time `json:"t"`
}
if err := yaml.Unmarshal(b, &unmarshalResult); err != nil {
return 0
}
marshalResult, err := yaml.Marshal(&unmarshalResult)
if err != nil {
panic(err)
}
if !bytes.Equal(marshalResult, b) {
panic("marshalResult != input")
}
return 1
}
// FuzzYamlV2 is a fuzz target for "gopkg.in/yaml.v2" unmarshaling.
func FuzzYamlV2(b []byte) int {
t := struct{}{}