get.pme.sh/pnats@v0.0.0-20240304004023-26bb5a137ed0/server/fuzz.go (about)

     1  // Copyright 2020 The NATS Authors
     2  // Licensed under the Apache License, Version 2.0 (the "License");
     3  // you may not use this file except in compliance with the License.
     4  // You may obtain a copy of the License at
     5  //
     6  // http://www.apache.org/licenses/LICENSE-2.0
     7  //
     8  // Unless required by applicable law or agreed to in writing, software
     9  // distributed under the License is distributed on an "AS IS" BASIS,
    10  // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    11  // See the License for the specific language governing permissions and
    12  // limitations under the License.
    13  
    14  //go:build gofuzz
    15  // +build gofuzz
    16  
    17  package server
    18  
    19  var defaultFuzzServerOptions = Options{
    20  	Host:                  "127.0.0.1",
    21  	Trace:                 true,
    22  	Debug:                 true,
    23  	DisableShortFirstPing: true,
    24  	NoLog:                 true,
    25  	NoSigs:                true,
    26  }
    27  
    28  func dummyFuzzClient() *client {
    29  	return &client{srv: New(&defaultFuzzServerOptions), msubs: -1, mpay: MAX_PAYLOAD_SIZE, mcl: MAX_CONTROL_LINE_SIZE}
    30  }
    31  
    32  func FuzzClient(data []byte) int {
    33  	if len(data) < 100 {
    34  		return -1
    35  	}
    36  	c := dummyFuzzClient()
    37  
    38  	err := c.parse(data[:50])
    39  	if err != nil {
    40  		return 0
    41  	}
    42  
    43  	err = c.parse(data[50:])
    44  	if err != nil {
    45  		return 0
    46  	}
    47  	return 1
    48  }