74 lines
2 KiB
Rust
74 lines
2 KiB
Rust
//! ELF executables
|
|
|
|
use collections::String;
|
|
|
|
use core::str;
|
|
|
|
#[cfg(target_arch = "x86")]
|
|
pub use goblin::elf32::{header, program_header};
|
|
|
|
#[cfg(target_arch = "x86_64")]
|
|
pub use goblin::elf64::{header, program_header};
|
|
|
|
/// An ELF executable
|
|
pub struct Elf<'a> {
|
|
pub data: &'a [u8],
|
|
header: &'a header::Header
|
|
}
|
|
|
|
impl<'a> Elf<'a> {
|
|
/// Create a ELF executable from data
|
|
pub fn from(data: &'a [u8]) -> Result<Elf<'a>, String> {
|
|
if data.len() < header::SIZEOF_EHDR {
|
|
Err(format!("Elf: Not enough data: {} < {}", data.len(), header::SIZEOF_EHDR))
|
|
} else if &data[..header::SELFMAG] != header::ELFMAG {
|
|
Err(format!("Elf: Invalid magic: {:?} != {:?}", &data[..4], header::ELFMAG))
|
|
} else if data.get(header::EI_CLASS) != Some(&header::ELFCLASS) {
|
|
Err(format!("Elf: Invalid architecture: {:?} != {:?}", data.get(header::EI_CLASS), header::ELFCLASS))
|
|
} else {
|
|
Ok(Elf {
|
|
data: data,
|
|
header: unsafe { &*(data.as_ptr() as usize as *const header::Header) }
|
|
})
|
|
}
|
|
}
|
|
|
|
pub fn segments(&'a self) -> ElfSegments<'a> {
|
|
ElfSegments {
|
|
data: self.data,
|
|
header: self.header,
|
|
i: 0
|
|
}
|
|
}
|
|
|
|
/// Get the entry field of the header
|
|
pub fn entry(&self) -> usize {
|
|
self.header.e_entry as usize
|
|
}
|
|
}
|
|
|
|
pub struct ElfSegments<'a> {
|
|
data: &'a [u8],
|
|
header: &'a header::Header,
|
|
i: usize
|
|
}
|
|
|
|
impl<'a> Iterator for ElfSegments<'a> {
|
|
type Item = &'a program_header::ProgramHeader;
|
|
fn next(&mut self) -> Option<Self::Item> {
|
|
if self.i < self.header.e_phnum as usize {
|
|
let item = unsafe {
|
|
&* ((
|
|
self.data.as_ptr() as usize
|
|
+ self.header.e_phoff as usize
|
|
+ self.i * self.header.e_phentsize as usize
|
|
) as *const program_header::ProgramHeader)
|
|
};
|
|
self.i += 1;
|
|
Some(item)
|
|
} else {
|
|
None
|
|
}
|
|
}
|
|
}
|