Skip to main content

load/
load.rs

1//! Test the elf loader
2
3use neotron_loader as ldr;
4
5#[derive(Debug)]
6enum Error {
7    Io(std::io::Error),
8    Loader(ldr::Error<ldr::traits::SliceError>),
9    MissingArgument,
10}
11
12impl From<std::io::Error> for Error {
13    fn from(value: std::io::Error) -> Self {
14        Error::Io(value)
15    }
16}
17
18impl From<ldr::Error<ldr::traits::SliceError>> for Error {
19    fn from(value: ldr::Error<ldr::traits::SliceError>) -> Self {
20        Error::Loader(value)
21    }
22}
23
24fn main() -> Result<(), Error> {
25    let mut args = std::env::args_os();
26    let _example_name = args.next();
27    let filename = args.next().ok_or(Error::MissingArgument)?;
28    let data = std::fs::read(&filename)?;
29    let loader = ldr::Loader::new(&data[..])?;
30
31    println!("Loaded ELF {}", filename.to_string_lossy());
32    println!("Entry Point: 0x{:08x}", loader.e_entry());
33
34    let segment_start_addr = loader.segment_start_offset();
35    let mut total_ram_used = 0;
36    for (idx, ph) in loader.iter_program_headers().enumerate() {
37        let ph = ph.expect("PH loaded OK");
38        let p_type = match ph.p_type() {
39            ldr::ProgramHeader::PT_NULL => "PT_NULL",
40            ldr::ProgramHeader::PT_LOAD => "PT_LOAD",
41            ldr::ProgramHeader::PT_DYNAMIC => "PT_DYNAMIC",
42            ldr::ProgramHeader::PT_INTERP => "PT_INTERP",
43            ldr::ProgramHeader::PT_NOTE => "PT_NOTE",
44            ldr::ProgramHeader::PT_SHLIB => "PT_SHLIB",
45            ldr::ProgramHeader::PT_PHDR => "PT_PHDR",
46            ldr::ProgramHeader::PT_TLS => "PT_TLS",
47            ldr::ProgramHeader::PT_GNU_STACK => "PT_GNU_STACK",
48            _ => "PT_???",
49        };
50
51        let ignored = if ph.p_offset() >= segment_start_addr {
52            "OK"
53        } else {
54            "Ignored"
55        };
56
57        let data_bytes = ph.p_filesz();
58        let zero_bytes = ph.p_memsz() - data_bytes;
59        let load_addr = ph.p_paddr();
60
61        total_ram_used += ph.p_memsz();
62
63        println!("PH {idx:02}: p_type = {p_type:12}, data_bytes=0x{data_bytes:04x}, zero_bytes=0x{zero_bytes:04x}, load_addr=0x{load_addr:08x} ({ignored})");
64    }
65
66    println!("Total RAM used: {total_ram_used} bytes");
67
68    for (idx, sh) in loader.iter_section_headers().enumerate() {
69        let sh = sh.expect("SH loaded OK");
70        let sh_type = match sh.sh_type() {
71            ldr::SectionHeader::SHT_NULL => "SHT_NULL",
72            ldr::SectionHeader::SHT_PROGBITS => "SHT_PROGBITS",
73            ldr::SectionHeader::SHT_SYMTAB => "SHT_SYMTAB",
74            ldr::SectionHeader::SHT_STRTAB => "SHT_STRTAB",
75            ldr::SectionHeader::SHT_RELA => "SHT_RELA",
76            ldr::SectionHeader::SHT_HASH => "SHT_HASH",
77            ldr::SectionHeader::SHT_DYNAMIC => "SHT_DYNAMIC",
78            ldr::SectionHeader::SHT_NOTE => "SHT_NOTE",
79            ldr::SectionHeader::SHT_NOBITS => "SHT_NOBITS",
80            ldr::SectionHeader::SHT_REL => "SHT_REL",
81            ldr::SectionHeader::SHT_DYNSYM => "SHT_DYNSYM",
82            ldr::SectionHeader::SHT_INIT_ARRAY => "SHT_INIT_ARRAY",
83            ldr::SectionHeader::SHT_FINI_ARRAY => "SHT_FINI_ARRAY",
84            ldr::SectionHeader::SHT_PREINIT_ARRAY => "SHT_PREINIT_ARRAY",
85            ldr::SectionHeader::SHT_GROUP => "SHT_GROUP",
86            ldr::SectionHeader::SHT_SYMTAB_SHNDX => "SHT_SYMTAB_SHNDX",
87            _ => "???",
88        };
89
90        let mut buffer = [0u8; 64];
91        let name = sh.sh_name(&loader, &mut buffer).unwrap_or("E_TOO_LONG");
92        println!("SH {idx:02}: {sh:08x?} ({sh_type}, {name:?})");
93    }
94
95    Ok(())
96}