github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2025-07-30 06:54:01 +00:00
Code Issues Projects Releases Wiki Activity

Use strict unmarshaling for metav1 fuzz targets

Browse Source
This commit is contained in:
Brendan Chang 2019-10-23 19:16:50 -04:00
parent 862e814f82
commit 7aa9434811
1 changed files with 26 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -27,40 +27,42 @@ import (
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 {
// FuzzDuration is a fuzz target for strict-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 FuzzDurationStrict(b []byte) int {
var durationHolder struct {
D metav1.Duration `json:"d"`
}
if err := yaml.Unmarshal(b, &unmarshalResult); err != nil {
if err := sigyaml.UnmarshalStrict(b, &durationHolder); err != nil {
return 0
}
marshalResult, err := sigyaml.Marshal(&unmarshalResult)
result, err := sigyaml.Marshal(&durationHolder)
if err != nil {
panic(err)
}
if !bytes.Equal(marshalResult, b) {
panic("marshalResult != input")
if !bytes.Equal(result, b) {
panic("result != 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 {
// FuzzMicroTime is a fuzz target for strict-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 FuzzMicroTimeStrict(b []byte) int {
var microTimeHolder struct {
T metav1.MicroTime `json:"t"`
}
if err := yaml.Unmarshal(b, &unmarshalResult); err != nil {
if err := sigyaml.UnmarshalStrict(b, &microTimeHolder); err != nil {
return 0
}
marshalResult, err := sigyaml.Marshal(&unmarshalResult)
result, err := sigyaml.Marshal(&microTimeHolder)
if err != nil {
panic(err)
}
if !bytes.Equal(marshalResult, b) {
panic("marshalResult != input")
if !bytes.Equal(result, b) {
panic("result != input")
}
return 1
}
@ -79,21 +81,22 @@ 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.
// FuzzTime is a fuzz target for strict-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 {
var timeHolder struct {
T metav1.Time `json:"t"`
}
if err := sigyaml.Unmarshal(b, &unmarshalResult); err != nil {
if err := sigyaml.UnmarshalStrict(b, &timeHolder); err != nil {
return 0
}
marshalResult, err := yaml.Marshal(&unmarshalResult)
result, err := sigyaml.Marshal(&timeHolder)
if err != nil {
panic(err)
}
if !bytes.Equal(marshalResult, b) {
panic("marshalResult != input")
if !bytes.Equal(result, b) {
panic("result != input")
}
return 1
}