1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65
use cargo_metadata::MetadataCommand; use std::{any::type_name, env, path::PathBuf}; #[must_use] pub fn corpus_directory_from_args_type<T>() -> PathBuf { corpus_directory().join(path_from_args_type::<T>()) } #[must_use] pub fn corpus_directory_from_target(krate: &str, target: &str) -> PathBuf { corpus_directory().join(path_from_target(krate, target)) } #[must_use] pub fn crashes_directory_from_target(krate: &str, target: &str) -> PathBuf { output_directory_from_target(krate, target) .join("default") .join("crashes") } #[must_use] pub fn queue_directory_from_target(krate: &str, target: &str) -> PathBuf { output_directory_from_target(krate, target) .join("default") .join("queue") } #[must_use] pub fn output_directory_from_target(krate: &str, target: &str) -> PathBuf { output_directory().join(path_from_target(krate, target)) } #[must_use] fn corpus_directory() -> PathBuf { target_directory(false).join("corpus") } #[must_use] fn output_directory() -> PathBuf { target_directory(true).join("output") } #[must_use] pub fn target_directory(instrumented: bool) -> PathBuf { let mut command = MetadataCommand::new(); if let Ok(path) = env::var("TEST_FUZZ_MANIFEST_PATH") { command.manifest_path(path); } let mut target_dir = command.exec().unwrap().target_directory; if instrumented { target_dir = target_dir.join("afl"); } target_dir.into() } #[must_use] fn path_from_args_type<T>() -> String { let type_name = type_name::<T>(); type_name[..type_name.find("_fuzz::write_args::Args").unwrap()].to_owned() } #[must_use] fn path_from_target(krate: &str, target: &str) -> String { krate.replace("-", "_") + "::" + target }