tests/fuzz/kube_ingress_fuzzer.go

// Copyright Istio 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 fuzz

import (
	fuzz "github.com/AdaLogics/go-fuzz-headers"
	meshconfig "istio.io/api/mesh/v1alpha1"
	coreV1 "k8s.io/api/core/v1"
	knetworking "k8s.io/api/networking/v1"
	networkingV1beta1 "k8s.io/api/networking/v1beta1"
	"k8s.io/apimachinery/pkg/runtime"
	listerv1 "k8s.io/client-go/listers/core/v1"
)

import (
	kubeIngress "github.com/apache/dubbo-go-pixiu/pilot/pkg/config/kube/ingress"
	ingressv1 "github.com/apache/dubbo-go-pixiu/pilot/pkg/config/kube/ingressv1"
	"github.com/apache/dubbo-go-pixiu/pkg/config"
	"github.com/apache/dubbo-go-pixiu/pkg/kube"
)

func FuzzConvertIngressVirtualService(data []byte) int {
	f := fuzz.NewConsumer(data)
	ingress := knetworking.Ingress{}
	err := f.GenerateStruct(&ingress)
	if err != nil {
		return 0
	}
	service := &coreV1.Service{}
	cfgs := map[string]*config.Config{}
	serviceLister, teardown := newServiceLister(service)
	defer teardown()
	ingressv1.ConvertIngressVirtualService(ingress, "mydomain", cfgs, serviceLister)
	return 1
}

func FuzzConvertIngressVirtualService2(data []byte) int {
	f := fuzz.NewConsumer(data)
	ingress := networkingV1beta1.Ingress{}
	err := f.GenerateStruct(&ingress)
	if err != nil {
		return 0
	}
	service := &coreV1.Service{}
	cfgs := map[string]*config.Config{}
	serviceLister, teardown := newServiceLister(service)
	defer teardown()
	kubeIngress.ConvertIngressVirtualService(ingress, "mydomain", cfgs, serviceLister)
	return 1
}

func FuzzConvertIngressV1alpha3(data []byte) int {
	f := fuzz.NewConsumer(data)
	ingress := knetworking.Ingress{}
	err := f.GenerateStruct(&ingress)
	if err != nil {
		return 0
	}
	m := &meshconfig.MeshConfig{}
	err = f.GenerateStruct(m)
	if err != nil {
		return 0
	}
	ingressv1.ConvertIngressV1alpha3(ingress, m, "mydomain")
	return 1
}

func FuzzConvertIngressV1alpha32(data []byte) int {
	f := fuzz.NewConsumer(data)
	ingress := networkingV1beta1.Ingress{}
	err := f.GenerateStruct(&ingress)
	if err != nil {
		return 0
	}
	m := &meshconfig.MeshConfig{}
	err = f.GenerateStruct(m)
	if err != nil {
		return 0
	}
	kubeIngress.ConvertIngressV1alpha3(ingress, m, "mydomain")
	return 1
}

func newServiceLister(objects ...runtime.Object) (listerv1.ServiceLister, func()) {
	client := kube.NewFakeClient(objects...)
	stop := make(chan struct{})
	client.RunAndWait(stop)
	teardown := func() {
		close(stop)
	}
	return client.KubeInformer().Core().V1().Services().Lister(), teardown
}