ristretto_classfile 0.31.0

A library for reading, writing and verifying Java classfiles.
Documentation 
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

use crate::class_file::ClassFile;
use crate::constant::Constant;
use crate::verifiers::error::Result;
use crate::verifiers::error::VerifyError::{
    InvalidConstantPoolIndex, InvalidConstantPoolIndexType,
};

/// Verify the `ClassFile` interfaces.
///
/// # Errors
/// Returns `VerificationError` if the interfaces are invalid.
pub(crate) fn verify(class_file: &ClassFile<'_>) -> Result<()> {
    for interface in &class_file.interfaces {
        let constant_pool = &class_file.constant_pool;
        match constant_pool.get(*interface) {
            Some(Constant::Class { .. }) => {} // valid constant
            None => return Err(InvalidConstantPoolIndex(*interface)),
            _ => return Err(InvalidConstantPoolIndexType(*interface)),
        }
    }
    Ok(())
}

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

    #[test]
    fn test_verify_success() -> Result<()> {
        let mut class_file = ClassFile::default();
        let constant_pool = &mut class_file.constant_pool;
        let index = constant_pool.add_class("Foo")?;
        class_file.interfaces.push(index);

        assert_eq!(Ok(()), verify(&class_file));
        Ok(())
    }

    #[test]
    fn test_verify_invalid_index() {
        let mut class_file = ClassFile::default();
        let index = 1;
        class_file.interfaces.push(index);

        assert_eq!(Err(InvalidConstantPoolIndex(index)), verify(&class_file));
    }

    #[test]
    fn test_verify_invalid_index_type() -> Result<()> {
        let mut class_file = ClassFile::default();
        let constant_pool = &mut class_file.constant_pool;
        let index = constant_pool.add_integer(42)?;
        class_file.interfaces.push(index);

        assert_eq!(
            Err(InvalidConstantPoolIndexType(index)),
            verify(&class_file)
        );
        Ok(())
    }
}