phink_lib/cli/
env.rs

1use std::fmt::Display;
2
3#[derive(Clone, Copy)]
4pub enum PhinkEnv {
5    FuzzingWithConfig,
6    FromZiggy,
7    CargoManifestDir,
8    AflForkServerTimeout,
9    AflDebug,
10    AllowList,
11}
12
13impl Display for PhinkEnv {
14    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
15        f.write_str(match self {
16            PhinkEnv::FuzzingWithConfig => "PHINK_START_FUZZING_WITH_CONFIG",
17            PhinkEnv::FromZiggy => "PHINK_FROM_ZIGGY",
18            PhinkEnv::CargoManifestDir => "CARGO_MANIFEST_DIR",
19            PhinkEnv::AflForkServerTimeout => "AFL_FORKSRV_INIT_TMOUT",
20            PhinkEnv::AflDebug => "AFL_DEBUG",
21            PhinkEnv::AllowList => "AFL_LLVM_ALLOWLIST",
22        })
23    }
24}
25
26#[cfg(test)]
27mod tests {
28    use super::*;
29
30    #[test]
31    fn test_phink_env_display() {
32        assert_eq!(
33            PhinkEnv::FuzzingWithConfig.to_string(),
34            "PHINK_START_FUZZING_WITH_CONFIG"
35        );
36        assert_eq!(PhinkEnv::FromZiggy.to_string(), "PHINK_FROM_ZIGGY");
37        assert_eq!(PhinkEnv::CargoManifestDir.to_string(), "CARGO_MANIFEST_DIR");
38    }
39}