rvdasm 0.2.3

A RISC-V disassembler written in Rust
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

use rvdasm::disassembler::*;
use std::fs::File;
use std::io::Read;
use object::{Object, ObjectSection};
use clap::Parser;

#[derive(Parser)]
struct Args {
    #[clap(short, long)]
    file: String,
    #[clap(short, long, default_value = "false")]
    canonical: bool,
}

fn main() {
    let args = Args::parse();
    let mut elf_file = File::open(args.file.clone()).unwrap();
    let mut elf_buffer = Vec::new();
    elf_file.read_to_end(&mut elf_buffer).unwrap();
    let elf = object::File::parse(&*elf_buffer).unwrap();

    let elf_arch = elf.architecture();
    // elf.architecture() == object::Architecture::Riscv64
    let xlen = if elf_arch == object::Architecture::Riscv64 {
        Xlen::XLEN64
    } else if elf_arch == object::Architecture::Riscv32 {
        Xlen::XLEN32
    } else {
        panic!("Unsupported architecture: {:?}", elf_arch);
    };
    let disassembler = Disassembler::new(xlen);

    let text_section = elf.section_by_name(".text").unwrap();
    let text_data = text_section.data().unwrap();
    let entry_point = text_section.address();

    println!("entry point: 0x{:08x}", entry_point);

    let decoded_insns = disassembler.disassemble_all(&text_data, entry_point);

    // sort keys by address 
    let mut keys: Vec<u64> = decoded_insns.keys().cloned().collect();
    keys.sort();
    
    // write to file with extension .dump
    // let mut dump_file = File::create(format!("{}.dump", args.file)).unwrap();
    for key in keys {
        if args.canonical {
            println!("{}", decoded_insns[&key].to_canonical());
        } else {
            println!("0x{:08x}: {:08x}     {}", key, decoded_insns[&key].get_raw(), decoded_insns[&key].to_string());
        }
    }
}