android-doctor 0.3.1

Extract and audit Android OTA and firmware images (payload.bin, super.img, ext4, erofs) without running them
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

//! Helpers shared by the test suites.
use std::path::PathBuf;

/// A temporary directory that is deleted when the value is dropped, so a
/// failed or panicking test cannot leak it (earlier helpers returned a bare
/// `PathBuf` and every test run left its directories behind).
pub struct Scratch(PathBuf);

impl Scratch {
    pub fn new(tag: &str) -> Self {
        let dir = std::env::temp_dir().join(format!("ad-test-{tag}-{}", std::process::id()));
        let _ = std::fs::remove_dir_all(&dir);
        std::fs::create_dir_all(&dir).unwrap();
        Self(dir)
    }
}

impl std::ops::Deref for Scratch {
    type Target = PathBuf;
    fn deref(&self) -> &PathBuf {
        &self.0
    }
}

impl AsRef<std::path::Path> for Scratch {
    fn as_ref(&self) -> &std::path::Path {
        &self.0
    }
}

impl Drop for Scratch {
    fn drop(&mut self) {
        let _ = std::fs::remove_dir_all(&self.0);
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn the_directory_is_removed_on_drop() {
        let p;
        {
            let s = Scratch::new("selfcheck");
            p = s.join("x");
            std::fs::write(&p, b"x").unwrap();
            assert!(p.exists());
        }
        assert!(!p.exists(), "the file goes with the directory");
        assert!(
            !p.parent().unwrap().exists(),
            "the directory itself is gone"
        );
    }
}