hermes_rs 0.1.2

A dependency-free disassembler and assembler for the Hermes bytecode
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
62

#[macro_use]
pub mod hermes;

use hermes::{HermesHeader, HermesStruct};

use std::{fs::File, io};

// This is mostly just a test function as of right now.
// TODO: Scrap it and only export the lib
fn main() {
    let filename = "./input_data/test_file.hbc";
    let f = File::open(filename).expect("no file found");
    let mut reader = io::BufReader::new(f);
    let header: HermesHeader = HermesStruct::deserialize(&mut reader, 0);

    println!("HBC Version: {:?}", header.version);

    // works
    // header.function_headers.iter().for_each(|fh| {
    // println!("function header: {:?}", fh);
    // });

    // works
    // header.parse_bytecode(&mut reader);

    // works
    //for name in get_all_function_names(&header) {
    //    println!("Function name: {}", name);
    //}

    header.parse_bytecode_for_fn(0, &mut reader);

    // if writer == vec![115, 0, 2, 0, 98, 92, 0] {
    //     println!("Bytecode is correct!");
    // } else {
    //     println!("Bytecode is incorrect!");
    // }
}

#[allow(dead_code)]
fn get_all_function_names(header: &HermesHeader) -> Vec<String> {
    let mut function_names = Vec::new();
    for (index, function_header) in header.function_headers.iter().enumerate() {
        // let string_id = function_header.func_name().clone();
        let myfunc = &header
            .string_storage
            .get(function_header.func_name() as usize)
            .unwrap();
        // println!("------------------------------------------------");
        let func_start = myfunc.offset;
        let mut func_name = String::from_utf8(
            header.string_storage_bytes[func_start as usize..(func_start + myfunc.length) as usize]
                .to_vec(),
        )
        .unwrap();
        if func_name.is_empty() {
            func_name = format!("$FUNC_{}", index);
        }
        function_names.push(func_name);
    }
    function_names
}