1 // Copyright 2024 Google LLC 2 // 3 // Licensed under the Apache License, Version 2.0 (the "License"); 4 // you may not use this file except in compliance with the License. 5 // You may obtain a copy of the License at 6 // 7 // http://www.apache.org/licenses/LICENSE-2.0 8 // 9 // Unless required by applicable law or agreed to in writing, software 10 // distributed under the License is distributed on an "AS IS" BASIS, 11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 // See the License for the specific language governing permissions and 13 // limitations under the License. 14 15 #![cfg_attr(fuzzing, no_main)] 16 17 use arbitrary::{Arbitrary, Unstructured}; 18 use derive_fuzztest::fuzztest; 19 20 #[derive(Debug, Clone)] 21 pub struct SmallU8(u8); 22 23 impl<'a> Arbitrary<'a> for SmallU8 { arbitrary(u: &mut Unstructured<'a>) -> arbitrary::Result<Self>24 fn arbitrary(u: &mut Unstructured<'a>) -> arbitrary::Result<Self> { 25 Ok(SmallU8(u.int_in_range(0..=127)?)) 26 } 27 } 28 29 #[fuzztest] test(a: SmallU8, b: SmallU8)30pub fn test(a: SmallU8, b: SmallU8) { 31 let _ = a.0 + b.0; // Succeeds because our custom arbitrary impl only generates 0-127 so it never overflows 32 } 33