| // Copyright 2024 Google LLC | |
| // | |
| // 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. | |
| #![cfg_attr(fuzzing, no_main)] | |
| use arbitrary::{Arbitrary, Unstructured}; | |
| use derive_fuzztest::fuzztest; | |
| #[derive(Debug, Clone)] | |
| pub struct SmallU8(u8); | |
| impl<'a> Arbitrary<'a> for SmallU8 { | |
| fn arbitrary(u: &mut Unstructured<'a>) -> arbitrary::Result<Self> { | |
| Ok(SmallU8(u.int_in_range(0..=127)?)) | |
| } | |
| } | |
| #[fuzztest] | |
| pub fn test(a: SmallU8, b: SmallU8) { | |
| let _ = a.0 + b.0; // Succeeds because our custom arbitrary impl only generates 0-127 so it never overflows | |
| } |