91 lines
2.8 KiB
Rust
91 lines
2.8 KiB
Rust
#![cfg(all(feature = "read", feature = "write"))]
|
|
|
|
use object::read::{Object, ObjectSection};
|
|
use object::{read, write};
|
|
use object::{Architecture, BinaryFormat, Endianness, SectionFlags, SectionKind};
|
|
|
|
#[test]
|
|
fn coff_x86_64_section_flags() {
|
|
let mut object =
|
|
write::Object::new(BinaryFormat::Coff, Architecture::X86_64, Endianness::Little);
|
|
|
|
let section = object.add_section(Vec::new(), b".text".to_vec(), SectionKind::Text);
|
|
object.section_mut(section).flags = SectionFlags::Coff {
|
|
characteristics: object::pe::IMAGE_SCN_MEM_WRITE,
|
|
};
|
|
|
|
let bytes = object.write().unwrap();
|
|
|
|
let object = read::File::parse(&*bytes).unwrap();
|
|
assert_eq!(object.format(), BinaryFormat::Coff);
|
|
assert_eq!(object.architecture(), Architecture::X86_64);
|
|
|
|
let mut sections = object.sections();
|
|
let section = sections.next().unwrap();
|
|
assert_eq!(section.name(), Ok(".text"));
|
|
assert_eq!(
|
|
section.flags(),
|
|
SectionFlags::Coff {
|
|
characteristics: object::pe::IMAGE_SCN_MEM_WRITE | object::pe::IMAGE_SCN_ALIGN_1BYTES,
|
|
}
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn elf_x86_64_section_flags() {
|
|
let mut object =
|
|
write::Object::new(BinaryFormat::Elf, Architecture::X86_64, Endianness::Little);
|
|
|
|
let section = object.add_section(Vec::new(), b".text".to_vec(), SectionKind::Text);
|
|
object.section_mut(section).flags = SectionFlags::Elf {
|
|
sh_flags: object::elf::SHF_WRITE.into(),
|
|
};
|
|
|
|
let bytes = object.write().unwrap();
|
|
|
|
let object = read::File::parse(&*bytes).unwrap();
|
|
assert_eq!(object.format(), BinaryFormat::Elf);
|
|
assert_eq!(object.architecture(), Architecture::X86_64);
|
|
|
|
let mut sections = object.sections();
|
|
sections.next().unwrap();
|
|
let section = sections.next().unwrap();
|
|
assert_eq!(section.name(), Ok(".text"));
|
|
assert_eq!(
|
|
section.flags(),
|
|
SectionFlags::Elf {
|
|
sh_flags: object::elf::SHF_WRITE.into(),
|
|
}
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn macho_x86_64_section_flags() {
|
|
let mut object = write::Object::new(
|
|
BinaryFormat::MachO,
|
|
Architecture::X86_64,
|
|
Endianness::Little,
|
|
);
|
|
|
|
let section = object.add_section(Vec::new(), b".text".to_vec(), SectionKind::Text);
|
|
object.section_mut(section).flags = SectionFlags::MachO {
|
|
flags: object::macho::S_ATTR_SELF_MODIFYING_CODE,
|
|
};
|
|
|
|
let bytes = object.write().unwrap();
|
|
|
|
let object = read::File::parse(&*bytes).unwrap();
|
|
assert_eq!(object.format(), BinaryFormat::MachO);
|
|
assert_eq!(object.architecture(), Architecture::X86_64);
|
|
|
|
let mut sections = object.sections();
|
|
let section = sections.next().unwrap();
|
|
assert_eq!(section.name(), Ok(".text"));
|
|
assert_eq!(
|
|
section.flags(),
|
|
SectionFlags::MachO {
|
|
flags: object::macho::S_ATTR_SELF_MODIFYING_CODE,
|
|
}
|
|
);
|
|
}
|