Initial vendor packages
Signed-off-by: Valentin Popov <valentin@popov.link>
This commit is contained in:
2
vendor/object/tests/integration.rs
vendored
Normal file
2
vendor/object/tests/integration.rs
vendored
Normal file
@ -0,0 +1,2 @@
|
||||
mod read;
|
||||
mod round_trip;
|
25
vendor/object/tests/parse_self.rs
vendored
Normal file
25
vendor/object/tests/parse_self.rs
vendored
Normal file
@ -0,0 +1,25 @@
|
||||
#![cfg(feature = "read")]
|
||||
use object::{File, Object};
|
||||
use std::{env, fs};
|
||||
|
||||
#[test]
|
||||
fn parse_self() {
|
||||
let exe = env::current_exe().unwrap();
|
||||
let data = fs::read(exe).unwrap();
|
||||
let object = File::parse(&*data).unwrap();
|
||||
assert!(object.entry() != 0);
|
||||
assert!(object.sections().count() != 0);
|
||||
}
|
||||
|
||||
#[cfg(feature = "std")]
|
||||
#[test]
|
||||
fn parse_self_cache() {
|
||||
use object::read::{ReadCache, ReadRef};
|
||||
let exe = env::current_exe().unwrap();
|
||||
let file = fs::File::open(exe).unwrap();
|
||||
let cache = ReadCache::new(file);
|
||||
let data = cache.range(0, cache.len().unwrap());
|
||||
let object = File::parse(data).unwrap();
|
||||
assert!(object.entry() != 0);
|
||||
assert!(object.sections().count() != 0);
|
||||
}
|
23
vendor/object/tests/read/coff.rs
vendored
Normal file
23
vendor/object/tests/read/coff.rs
vendored
Normal file
@ -0,0 +1,23 @@
|
||||
use object::{pe, read, Object, ObjectSection};
|
||||
use std::fs;
|
||||
use std::path::PathBuf;
|
||||
|
||||
#[cfg(feature = "coff")]
|
||||
#[test]
|
||||
fn coff_extended_relocations() {
|
||||
let path_to_obj: PathBuf = ["testfiles", "coff", "relocs_overflow.o"].iter().collect();
|
||||
let contents = fs::read(&path_to_obj).expect("Could not read relocs_overflow.o");
|
||||
let file =
|
||||
read::coff::CoffFile::<_>::parse(&contents[..]).expect("Could not parse relocs_overflow.o");
|
||||
let code_section = file
|
||||
.section_by_name(".text")
|
||||
.expect("Could not find .text section in relocs_overflow.o");
|
||||
match code_section.flags() {
|
||||
object::SectionFlags::Coff { characteristics } => {
|
||||
assert!(characteristics & pe::IMAGE_SCN_LNK_NRELOC_OVFL != 0)
|
||||
}
|
||||
_ => panic!("Invalid section flags flavour."),
|
||||
};
|
||||
let relocations = code_section.relocations().collect::<Vec<_>>();
|
||||
assert_eq!(relocations.len(), 65536);
|
||||
}
|
3
vendor/object/tests/read/mod.rs
vendored
Normal file
3
vendor/object/tests/read/mod.rs
vendored
Normal file
@ -0,0 +1,3 @@
|
||||
#![cfg(feature = "read")]
|
||||
|
||||
mod coff;
|
255
vendor/object/tests/round_trip/bss.rs
vendored
Normal file
255
vendor/object/tests/round_trip/bss.rs
vendored
Normal file
@ -0,0 +1,255 @@
|
||||
#![cfg(all(feature = "read", feature = "write"))]
|
||||
|
||||
use object::read::{Object, ObjectSection, ObjectSymbol};
|
||||
use object::{read, write};
|
||||
use object::{
|
||||
Architecture, BinaryFormat, Endianness, SectionKind, SymbolFlags, SymbolKind, SymbolScope,
|
||||
};
|
||||
|
||||
#[test]
|
||||
fn coff_x86_64_bss() {
|
||||
let mut object =
|
||||
write::Object::new(BinaryFormat::Coff, Architecture::X86_64, Endianness::Little);
|
||||
|
||||
let section = object.section_id(write::StandardSection::UninitializedData);
|
||||
|
||||
let symbol = object.add_symbol(write::Symbol {
|
||||
name: b"v1".to_vec(),
|
||||
value: 0,
|
||||
size: 0,
|
||||
kind: SymbolKind::Data,
|
||||
scope: SymbolScope::Linkage,
|
||||
weak: false,
|
||||
section: write::SymbolSection::Undefined,
|
||||
flags: SymbolFlags::None,
|
||||
});
|
||||
object.add_symbol_bss(symbol, section, 18, 4);
|
||||
|
||||
let symbol = object.add_symbol(write::Symbol {
|
||||
name: b"v2".to_vec(),
|
||||
value: 0,
|
||||
size: 0,
|
||||
kind: SymbolKind::Data,
|
||||
scope: SymbolScope::Linkage,
|
||||
weak: false,
|
||||
section: write::SymbolSection::Undefined,
|
||||
flags: SymbolFlags::None,
|
||||
});
|
||||
object.add_symbol_bss(symbol, section, 34, 8);
|
||||
|
||||
let bytes = object.write().unwrap();
|
||||
|
||||
//std::fs::write(&"bss.o", &bytes).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 bss = sections.next().unwrap();
|
||||
println!("{:?}", bss);
|
||||
let bss_index = bss.index();
|
||||
assert_eq!(bss.name(), Ok(".bss"));
|
||||
assert_eq!(bss.kind(), SectionKind::UninitializedData);
|
||||
assert_eq!(bss.size(), 58);
|
||||
assert_eq!(bss.data(), Ok(&[][..]));
|
||||
|
||||
let section = sections.next();
|
||||
assert!(section.is_none(), "unexpected section {:?}", section);
|
||||
|
||||
let mut symbols = object.symbols();
|
||||
|
||||
let symbol = symbols.next().unwrap();
|
||||
println!("{:?}", symbol);
|
||||
assert_eq!(symbol.name(), Ok("v1"));
|
||||
assert_eq!(symbol.kind(), SymbolKind::Data);
|
||||
assert_eq!(symbol.section_index(), Some(bss_index));
|
||||
assert_eq!(symbol.scope(), SymbolScope::Linkage);
|
||||
assert_eq!(symbol.is_weak(), false);
|
||||
assert_eq!(symbol.is_undefined(), false);
|
||||
assert_eq!(symbol.address(), 0);
|
||||
|
||||
let symbol = symbols.next().unwrap();
|
||||
println!("{:?}", symbol);
|
||||
assert_eq!(symbol.name(), Ok("v2"));
|
||||
assert_eq!(symbol.kind(), SymbolKind::Data);
|
||||
assert_eq!(symbol.section_index(), Some(bss_index));
|
||||
assert_eq!(symbol.scope(), SymbolScope::Linkage);
|
||||
assert_eq!(symbol.is_weak(), false);
|
||||
assert_eq!(symbol.is_undefined(), false);
|
||||
assert_eq!(symbol.address(), 24);
|
||||
|
||||
let symbol = symbols.next();
|
||||
assert!(symbol.is_none(), "unexpected symbol {:?}", symbol);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn elf_x86_64_bss() {
|
||||
let mut object =
|
||||
write::Object::new(BinaryFormat::Elf, Architecture::X86_64, Endianness::Little);
|
||||
|
||||
let section = object.section_id(write::StandardSection::UninitializedData);
|
||||
|
||||
let symbol = object.add_symbol(write::Symbol {
|
||||
name: b"v1".to_vec(),
|
||||
value: 0,
|
||||
size: 0,
|
||||
kind: SymbolKind::Data,
|
||||
scope: SymbolScope::Linkage,
|
||||
weak: false,
|
||||
section: write::SymbolSection::Undefined,
|
||||
flags: SymbolFlags::None,
|
||||
});
|
||||
object.add_symbol_bss(symbol, section, 18, 4);
|
||||
|
||||
let symbol = object.add_symbol(write::Symbol {
|
||||
name: b"v2".to_vec(),
|
||||
value: 0,
|
||||
size: 0,
|
||||
kind: SymbolKind::Data,
|
||||
scope: SymbolScope::Linkage,
|
||||
weak: false,
|
||||
section: write::SymbolSection::Undefined,
|
||||
flags: SymbolFlags::None,
|
||||
});
|
||||
object.add_symbol_bss(symbol, section, 34, 8);
|
||||
|
||||
let bytes = object.write().unwrap();
|
||||
|
||||
//std::fs::write(&"bss.o", &bytes).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();
|
||||
|
||||
let section = sections.next().unwrap();
|
||||
println!("{:?}", section);
|
||||
assert_eq!(section.name(), Ok(""));
|
||||
assert_eq!(section.kind(), SectionKind::Metadata);
|
||||
assert_eq!(section.address(), 0);
|
||||
assert_eq!(section.size(), 0);
|
||||
|
||||
let bss = sections.next().unwrap();
|
||||
println!("{:?}", bss);
|
||||
let bss_index = bss.index();
|
||||
assert_eq!(bss.name(), Ok(".bss"));
|
||||
assert_eq!(bss.kind(), SectionKind::UninitializedData);
|
||||
assert_eq!(bss.size(), 58);
|
||||
assert_eq!(bss.data(), Ok(&[][..]));
|
||||
|
||||
let mut symbols = object.symbols();
|
||||
|
||||
let symbol = symbols.next().unwrap();
|
||||
println!("{:?}", symbol);
|
||||
assert_eq!(symbol.name(), Ok(""));
|
||||
|
||||
let symbol = symbols.next().unwrap();
|
||||
println!("{:?}", symbol);
|
||||
assert_eq!(symbol.name(), Ok("v1"));
|
||||
assert_eq!(symbol.kind(), SymbolKind::Data);
|
||||
assert_eq!(symbol.section_index(), Some(bss_index));
|
||||
assert_eq!(symbol.scope(), SymbolScope::Linkage);
|
||||
assert_eq!(symbol.is_weak(), false);
|
||||
assert_eq!(symbol.is_undefined(), false);
|
||||
assert_eq!(symbol.address(), 0);
|
||||
assert_eq!(symbol.size(), 18);
|
||||
|
||||
let symbol = symbols.next().unwrap();
|
||||
println!("{:?}", symbol);
|
||||
assert_eq!(symbol.name(), Ok("v2"));
|
||||
assert_eq!(symbol.kind(), SymbolKind::Data);
|
||||
assert_eq!(symbol.section_index(), Some(bss_index));
|
||||
assert_eq!(symbol.scope(), SymbolScope::Linkage);
|
||||
assert_eq!(symbol.is_weak(), false);
|
||||
assert_eq!(symbol.is_undefined(), false);
|
||||
assert_eq!(symbol.address(), 24);
|
||||
assert_eq!(symbol.size(), 34);
|
||||
|
||||
let symbol = symbols.next();
|
||||
assert!(symbol.is_none(), "unexpected symbol {:?}", symbol);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn macho_x86_64_bss() {
|
||||
let mut object = write::Object::new(
|
||||
BinaryFormat::MachO,
|
||||
Architecture::X86_64,
|
||||
Endianness::Little,
|
||||
);
|
||||
|
||||
let section = object.section_id(write::StandardSection::UninitializedData);
|
||||
|
||||
let symbol = object.add_symbol(write::Symbol {
|
||||
name: b"v1".to_vec(),
|
||||
value: 0,
|
||||
size: 0,
|
||||
kind: SymbolKind::Data,
|
||||
scope: SymbolScope::Linkage,
|
||||
weak: false,
|
||||
section: write::SymbolSection::Undefined,
|
||||
flags: SymbolFlags::None,
|
||||
});
|
||||
object.add_symbol_bss(symbol, section, 18, 4);
|
||||
|
||||
let symbol = object.add_symbol(write::Symbol {
|
||||
name: b"v2".to_vec(),
|
||||
value: 0,
|
||||
size: 0,
|
||||
kind: SymbolKind::Data,
|
||||
scope: SymbolScope::Linkage,
|
||||
weak: false,
|
||||
section: write::SymbolSection::Undefined,
|
||||
flags: SymbolFlags::None,
|
||||
});
|
||||
object.add_symbol_bss(symbol, section, 34, 8);
|
||||
|
||||
let bytes = object.write().unwrap();
|
||||
|
||||
//std::fs::write(&"bss.o", &bytes).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 bss = sections.next().unwrap();
|
||||
println!("{:?}", bss);
|
||||
let bss_index = bss.index();
|
||||
assert_eq!(bss.name(), Ok("__bss"));
|
||||
assert_eq!(bss.segment_name(), Ok(Some("__DATA")));
|
||||
assert_eq!(bss.kind(), SectionKind::UninitializedData);
|
||||
assert_eq!(bss.size(), 58);
|
||||
assert_eq!(bss.data(), Ok(&[][..]));
|
||||
|
||||
let section = sections.next();
|
||||
assert!(section.is_none(), "unexpected section {:?}", section);
|
||||
|
||||
let mut symbols = object.symbols();
|
||||
|
||||
let symbol = symbols.next().unwrap();
|
||||
println!("{:?}", symbol);
|
||||
assert_eq!(symbol.name(), Ok("_v1"));
|
||||
assert_eq!(symbol.kind(), SymbolKind::Data);
|
||||
assert_eq!(symbol.section_index(), Some(bss_index));
|
||||
assert_eq!(symbol.scope(), SymbolScope::Linkage);
|
||||
assert_eq!(symbol.is_weak(), false);
|
||||
assert_eq!(symbol.is_undefined(), false);
|
||||
assert_eq!(symbol.address(), 0);
|
||||
|
||||
let symbol = symbols.next().unwrap();
|
||||
println!("{:?}", symbol);
|
||||
assert_eq!(symbol.name(), Ok("_v2"));
|
||||
assert_eq!(symbol.kind(), SymbolKind::Data);
|
||||
assert_eq!(symbol.section_index(), Some(bss_index));
|
||||
assert_eq!(symbol.scope(), SymbolScope::Linkage);
|
||||
assert_eq!(symbol.is_weak(), false);
|
||||
assert_eq!(symbol.is_undefined(), false);
|
||||
assert_eq!(symbol.address(), 24);
|
||||
|
||||
let symbol = symbols.next();
|
||||
assert!(symbol.is_none(), "unexpected symbol {:?}", symbol);
|
||||
}
|
56
vendor/object/tests/round_trip/coff.rs
vendored
Normal file
56
vendor/object/tests/round_trip/coff.rs
vendored
Normal file
@ -0,0 +1,56 @@
|
||||
use object::read::{Object, ObjectSection};
|
||||
use object::{read, write};
|
||||
use object::{
|
||||
Architecture, BinaryFormat, Endianness, RelocationEncoding, RelocationKind, SymbolFlags,
|
||||
SymbolKind, SymbolScope,
|
||||
};
|
||||
|
||||
#[test]
|
||||
fn reloc_overflow() {
|
||||
let mut object =
|
||||
write::Object::new(BinaryFormat::Coff, Architecture::X86_64, Endianness::Little);
|
||||
let text = object.section_id(write::StandardSection::Text);
|
||||
object.append_section_data(text, &[0; 4], 4);
|
||||
let symbol = object.add_symbol(write::Symbol {
|
||||
name: b"f".to_vec(),
|
||||
value: 0,
|
||||
size: 4,
|
||||
kind: SymbolKind::Text,
|
||||
scope: SymbolScope::Linkage,
|
||||
weak: false,
|
||||
section: write::SymbolSection::Section(text),
|
||||
flags: SymbolFlags::None,
|
||||
});
|
||||
for i in 0..0x10000 {
|
||||
object
|
||||
.add_relocation(
|
||||
text,
|
||||
write::Relocation {
|
||||
offset: i,
|
||||
size: 64,
|
||||
kind: RelocationKind::Absolute,
|
||||
encoding: RelocationEncoding::Generic,
|
||||
symbol,
|
||||
addend: 0,
|
||||
},
|
||||
)
|
||||
.unwrap();
|
||||
}
|
||||
let bytes = object.write().unwrap();
|
||||
|
||||
//std::fs::write(&"reloc_overflow.o", &bytes).unwrap();
|
||||
|
||||
let object = read::File::parse(&*bytes).unwrap();
|
||||
assert_eq!(object.format(), BinaryFormat::Coff);
|
||||
assert_eq!(object.architecture(), Architecture::X86_64);
|
||||
|
||||
let section = object.sections().next().unwrap();
|
||||
assert_eq!(section.name(), Ok(".text"));
|
||||
|
||||
let mut i = 0;
|
||||
for (offset, _relocation) in section.relocations() {
|
||||
assert_eq!(offset, i);
|
||||
i += 1;
|
||||
}
|
||||
assert_eq!(i, 0x10000);
|
||||
}
|
225
vendor/object/tests/round_trip/comdat.rs
vendored
Normal file
225
vendor/object/tests/round_trip/comdat.rs
vendored
Normal file
@ -0,0 +1,225 @@
|
||||
#![cfg(all(feature = "read", feature = "write"))]
|
||||
|
||||
use object::pe;
|
||||
use object::read::{Object, ObjectComdat, ObjectSection, ObjectSymbol};
|
||||
use object::{read, write};
|
||||
use object::{
|
||||
Architecture, BinaryFormat, ComdatKind, Endianness, SectionKind, SymbolFlags, SymbolKind,
|
||||
SymbolScope,
|
||||
};
|
||||
|
||||
#[test]
|
||||
fn coff_x86_64_comdat() {
|
||||
let mut object =
|
||||
write::Object::new(BinaryFormat::Coff, Architecture::X86_64, Endianness::Little);
|
||||
|
||||
let (section1, offset) =
|
||||
object.add_subsection(write::StandardSection::Text, b"s1", &[0, 1, 2, 3], 4);
|
||||
object.section_symbol(section1);
|
||||
let (section2, _) =
|
||||
object.add_subsection(write::StandardSection::Data, b"s1", &[0, 1, 2, 3], 4);
|
||||
object.section_symbol(section2);
|
||||
|
||||
let symbol = object.add_symbol(write::Symbol {
|
||||
name: b"s1".to_vec(),
|
||||
value: offset,
|
||||
size: 4,
|
||||
kind: SymbolKind::Data,
|
||||
scope: SymbolScope::Linkage,
|
||||
weak: false,
|
||||
section: write::SymbolSection::Section(section1),
|
||||
flags: SymbolFlags::None,
|
||||
});
|
||||
|
||||
object.add_comdat(write::Comdat {
|
||||
kind: ComdatKind::NoDuplicates,
|
||||
symbol,
|
||||
sections: vec![section1, section2],
|
||||
});
|
||||
|
||||
let bytes = object.write().unwrap();
|
||||
|
||||
//std::fs::write(&"comdat.o", &bytes).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 section1 = sections.next().unwrap();
|
||||
println!("{:?}", section1);
|
||||
let section1_index = section1.index();
|
||||
assert_eq!(section1.name(), Ok(".text$s1"));
|
||||
assert_eq!(section1.kind(), SectionKind::Text);
|
||||
assert_eq!(section1.address(), 0);
|
||||
assert_eq!(section1.size(), 4);
|
||||
|
||||
let section2 = sections.next().unwrap();
|
||||
println!("{:?}", section2);
|
||||
let section2_index = section2.index();
|
||||
assert_eq!(section2.name(), Ok(".data$s1"));
|
||||
assert_eq!(section2.kind(), SectionKind::Data);
|
||||
assert_eq!(section2.address(), 0);
|
||||
assert_eq!(section2.size(), 4);
|
||||
|
||||
let mut symbols = object.symbols();
|
||||
|
||||
let symbol = symbols.next().unwrap();
|
||||
println!("{:?}", symbol);
|
||||
assert_eq!(symbol.name(), Ok(".text$s1"));
|
||||
assert_eq!(symbol.kind(), SymbolKind::Section);
|
||||
assert_eq!(
|
||||
symbol.section(),
|
||||
read::SymbolSection::Section(section1.index())
|
||||
);
|
||||
assert_eq!(
|
||||
symbol.flags(),
|
||||
SymbolFlags::CoffSection {
|
||||
selection: pe::IMAGE_COMDAT_SELECT_NODUPLICATES,
|
||||
associative_section: None
|
||||
}
|
||||
);
|
||||
|
||||
let symbol = symbols.next().unwrap();
|
||||
println!("{:?}", symbol);
|
||||
assert_eq!(symbol.name(), Ok(".data$s1"));
|
||||
assert_eq!(symbol.kind(), SymbolKind::Section);
|
||||
assert_eq!(
|
||||
symbol.section(),
|
||||
read::SymbolSection::Section(section2.index())
|
||||
);
|
||||
assert_eq!(
|
||||
symbol.flags(),
|
||||
SymbolFlags::CoffSection {
|
||||
selection: pe::IMAGE_COMDAT_SELECT_ASSOCIATIVE,
|
||||
associative_section: Some(section1_index)
|
||||
}
|
||||
);
|
||||
|
||||
let symbol = symbols.next().unwrap();
|
||||
let symbol_index = symbol.index();
|
||||
println!("{:?}", symbol);
|
||||
assert_eq!(symbol.name(), Ok("s1"));
|
||||
assert_eq!(symbol.kind(), SymbolKind::Data);
|
||||
assert_eq!(
|
||||
symbol.section(),
|
||||
read::SymbolSection::Section(section1.index())
|
||||
);
|
||||
assert_eq!(symbol.scope(), SymbolScope::Linkage);
|
||||
assert_eq!(symbol.is_weak(), false);
|
||||
assert_eq!(symbol.is_undefined(), false);
|
||||
assert_eq!(symbol.address(), 0);
|
||||
|
||||
let symbol = symbols.next();
|
||||
assert!(symbol.is_none(), "unexpected symbol {:?}", symbol);
|
||||
|
||||
let mut comdats = object.comdats();
|
||||
|
||||
let comdat = comdats.next().unwrap();
|
||||
println!("{:?}", comdat);
|
||||
assert_eq!(comdat.kind(), ComdatKind::NoDuplicates);
|
||||
assert_eq!(comdat.symbol(), symbol_index);
|
||||
|
||||
let mut comdat_sections = comdat.sections();
|
||||
assert_eq!(comdat_sections.next(), Some(section1_index));
|
||||
assert_eq!(comdat_sections.next(), Some(section2_index));
|
||||
assert_eq!(comdat_sections.next(), None);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn elf_x86_64_comdat() {
|
||||
let mut object =
|
||||
write::Object::new(BinaryFormat::Elf, Architecture::X86_64, Endianness::Little);
|
||||
|
||||
let (section1, offset) =
|
||||
object.add_subsection(write::StandardSection::Text, b"s1", &[0, 1, 2, 3], 4);
|
||||
let (section2, _) =
|
||||
object.add_subsection(write::StandardSection::Data, b"s1", &[0, 1, 2, 3], 4);
|
||||
|
||||
let symbol = object.add_symbol(write::Symbol {
|
||||
name: b"s1".to_vec(),
|
||||
value: offset,
|
||||
size: 4,
|
||||
kind: SymbolKind::Data,
|
||||
scope: SymbolScope::Linkage,
|
||||
weak: false,
|
||||
section: write::SymbolSection::Section(section1),
|
||||
flags: SymbolFlags::None,
|
||||
});
|
||||
|
||||
object.add_comdat(write::Comdat {
|
||||
kind: ComdatKind::Any,
|
||||
symbol,
|
||||
sections: vec![section1, section2],
|
||||
});
|
||||
|
||||
let bytes = object.write().unwrap();
|
||||
|
||||
//std::fs::write(&"comdat.o", &bytes).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();
|
||||
|
||||
let section = sections.next().unwrap();
|
||||
println!("{:?}", section);
|
||||
assert_eq!(section.name(), Ok(""));
|
||||
|
||||
let section = sections.next().unwrap();
|
||||
println!("{:?}", section);
|
||||
assert_eq!(section.name(), Ok(".group"));
|
||||
|
||||
let section1 = sections.next().unwrap();
|
||||
println!("{:?}", section1);
|
||||
let section1_index = section1.index();
|
||||
assert_eq!(section1.name(), Ok(".text.s1"));
|
||||
assert_eq!(section1.kind(), SectionKind::Text);
|
||||
assert_eq!(section1.address(), 0);
|
||||
assert_eq!(section1.size(), 4);
|
||||
|
||||
let section2 = sections.next().unwrap();
|
||||
println!("{:?}", section2);
|
||||
let section2_index = section2.index();
|
||||
assert_eq!(section2.name(), Ok(".data.s1"));
|
||||
assert_eq!(section2.kind(), SectionKind::Data);
|
||||
assert_eq!(section2.address(), 0);
|
||||
assert_eq!(section2.size(), 4);
|
||||
|
||||
let mut symbols = object.symbols();
|
||||
|
||||
let symbol = symbols.next().unwrap();
|
||||
println!("{:?}", symbol);
|
||||
assert_eq!(symbol.name(), Ok(""));
|
||||
|
||||
let symbol = symbols.next().unwrap();
|
||||
let symbol_index = symbol.index();
|
||||
println!("{:?}", symbol);
|
||||
assert_eq!(symbol.name(), Ok("s1"));
|
||||
assert_eq!(symbol.kind(), SymbolKind::Data);
|
||||
assert_eq!(
|
||||
symbol.section(),
|
||||
read::SymbolSection::Section(section1.index())
|
||||
);
|
||||
assert_eq!(symbol.scope(), SymbolScope::Linkage);
|
||||
assert_eq!(symbol.is_weak(), false);
|
||||
assert_eq!(symbol.is_undefined(), false);
|
||||
assert_eq!(symbol.address(), 0);
|
||||
|
||||
let symbol = symbols.next();
|
||||
assert!(symbol.is_none(), "unexpected symbol {:?}", symbol);
|
||||
|
||||
let mut comdats = object.comdats();
|
||||
|
||||
let comdat = comdats.next().unwrap();
|
||||
println!("{:?}", comdat);
|
||||
assert_eq!(comdat.kind(), ComdatKind::Any);
|
||||
assert_eq!(comdat.symbol(), symbol_index);
|
||||
|
||||
let mut comdat_sections = comdat.sections();
|
||||
assert_eq!(comdat_sections.next(), Some(section1_index));
|
||||
assert_eq!(comdat_sections.next(), Some(section2_index));
|
||||
assert_eq!(comdat_sections.next(), None);
|
||||
}
|
245
vendor/object/tests/round_trip/common.rs
vendored
Normal file
245
vendor/object/tests/round_trip/common.rs
vendored
Normal file
@ -0,0 +1,245 @@
|
||||
#![cfg(all(feature = "read", feature = "write"))]
|
||||
|
||||
use object::read::{Object, ObjectSection, ObjectSymbol};
|
||||
use object::{read, write};
|
||||
use object::{
|
||||
Architecture, BinaryFormat, Endianness, SectionKind, SymbolFlags, SymbolKind, SymbolScope,
|
||||
};
|
||||
|
||||
#[test]
|
||||
fn coff_x86_64_common() {
|
||||
let mut object =
|
||||
write::Object::new(BinaryFormat::Coff, Architecture::X86_64, Endianness::Little);
|
||||
|
||||
let symbol = write::Symbol {
|
||||
name: b"v1".to_vec(),
|
||||
value: 0,
|
||||
size: 0,
|
||||
kind: SymbolKind::Data,
|
||||
scope: SymbolScope::Linkage,
|
||||
weak: false,
|
||||
section: write::SymbolSection::Undefined,
|
||||
flags: SymbolFlags::None,
|
||||
};
|
||||
object.add_common_symbol(symbol, 4, 4);
|
||||
|
||||
let symbol = write::Symbol {
|
||||
name: b"v2".to_vec(),
|
||||
value: 0,
|
||||
size: 0,
|
||||
kind: SymbolKind::Data,
|
||||
scope: SymbolScope::Linkage,
|
||||
weak: false,
|
||||
section: write::SymbolSection::Undefined,
|
||||
flags: SymbolFlags::None,
|
||||
};
|
||||
object.add_common_symbol(symbol, 8, 8);
|
||||
|
||||
// Also check undefined symbols, which are very similar.
|
||||
let symbol = write::Symbol {
|
||||
name: b"v3".to_vec(),
|
||||
value: 0,
|
||||
size: 0,
|
||||
kind: SymbolKind::Data,
|
||||
scope: SymbolScope::Linkage,
|
||||
weak: false,
|
||||
section: write::SymbolSection::Undefined,
|
||||
flags: SymbolFlags::None,
|
||||
};
|
||||
object.add_symbol(symbol);
|
||||
|
||||
let bytes = object.write().unwrap();
|
||||
|
||||
//std::fs::write(&"common.o", &bytes).unwrap();
|
||||
|
||||
let object = read::File::parse(&*bytes).unwrap();
|
||||
assert_eq!(object.format(), BinaryFormat::Coff);
|
||||
assert_eq!(object.architecture(), Architecture::X86_64);
|
||||
|
||||
let mut symbols = object.symbols();
|
||||
|
||||
let symbol = symbols.next().unwrap();
|
||||
println!("{:?}", symbol);
|
||||
assert_eq!(symbol.name(), Ok("v1"));
|
||||
assert_eq!(symbol.kind(), SymbolKind::Data);
|
||||
assert_eq!(symbol.section(), read::SymbolSection::Common);
|
||||
assert_eq!(symbol.scope(), SymbolScope::Linkage);
|
||||
assert_eq!(symbol.is_weak(), false);
|
||||
assert_eq!(symbol.is_undefined(), false);
|
||||
assert_eq!(symbol.address(), 0);
|
||||
assert_eq!(symbol.size(), 4);
|
||||
|
||||
let symbol = symbols.next().unwrap();
|
||||
println!("{:?}", symbol);
|
||||
assert_eq!(symbol.name(), Ok("v2"));
|
||||
assert_eq!(symbol.kind(), SymbolKind::Data);
|
||||
assert_eq!(symbol.section(), read::SymbolSection::Common);
|
||||
assert_eq!(symbol.scope(), SymbolScope::Linkage);
|
||||
assert_eq!(symbol.is_weak(), false);
|
||||
assert_eq!(symbol.is_undefined(), false);
|
||||
assert_eq!(symbol.address(), 0);
|
||||
assert_eq!(symbol.size(), 8);
|
||||
|
||||
let symbol = symbols.next().unwrap();
|
||||
println!("{:?}", symbol);
|
||||
assert_eq!(symbol.name(), Ok("v3"));
|
||||
assert_eq!(symbol.kind(), SymbolKind::Data);
|
||||
assert_eq!(symbol.section(), read::SymbolSection::Undefined);
|
||||
assert_eq!(symbol.scope(), SymbolScope::Linkage);
|
||||
assert_eq!(symbol.is_weak(), false);
|
||||
assert_eq!(symbol.is_undefined(), true);
|
||||
assert_eq!(symbol.address(), 0);
|
||||
assert_eq!(symbol.size(), 0);
|
||||
|
||||
let symbol = symbols.next();
|
||||
assert!(symbol.is_none(), "unexpected symbol {:?}", symbol);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn elf_x86_64_common() {
|
||||
let mut object =
|
||||
write::Object::new(BinaryFormat::Elf, Architecture::X86_64, Endianness::Little);
|
||||
|
||||
let symbol = write::Symbol {
|
||||
name: b"v1".to_vec(),
|
||||
value: 0,
|
||||
size: 0,
|
||||
kind: SymbolKind::Data,
|
||||
scope: SymbolScope::Linkage,
|
||||
weak: false,
|
||||
section: write::SymbolSection::Undefined,
|
||||
flags: SymbolFlags::None,
|
||||
};
|
||||
object.add_common_symbol(symbol, 4, 4);
|
||||
|
||||
let symbol = write::Symbol {
|
||||
name: b"v2".to_vec(),
|
||||
value: 0,
|
||||
size: 0,
|
||||
kind: SymbolKind::Data,
|
||||
scope: SymbolScope::Linkage,
|
||||
weak: false,
|
||||
section: write::SymbolSection::Undefined,
|
||||
flags: SymbolFlags::None,
|
||||
};
|
||||
object.add_common_symbol(symbol, 8, 8);
|
||||
|
||||
let bytes = object.write().unwrap();
|
||||
|
||||
//std::fs::write(&"common.o", &bytes).unwrap();
|
||||
|
||||
let object = read::File::parse(&*bytes).unwrap();
|
||||
assert_eq!(object.format(), BinaryFormat::Elf);
|
||||
assert_eq!(object.architecture(), Architecture::X86_64);
|
||||
|
||||
let mut symbols = object.symbols();
|
||||
|
||||
let symbol = symbols.next().unwrap();
|
||||
println!("{:?}", symbol);
|
||||
assert_eq!(symbol.name(), Ok(""));
|
||||
|
||||
let symbol = symbols.next().unwrap();
|
||||
println!("{:?}", symbol);
|
||||
assert_eq!(symbol.name(), Ok("v1"));
|
||||
assert_eq!(symbol.kind(), SymbolKind::Data);
|
||||
assert_eq!(symbol.section(), read::SymbolSection::Common);
|
||||
assert_eq!(symbol.scope(), SymbolScope::Linkage);
|
||||
assert_eq!(symbol.is_weak(), false);
|
||||
assert_eq!(symbol.is_undefined(), false);
|
||||
assert_eq!(symbol.address(), 0);
|
||||
assert_eq!(symbol.size(), 4);
|
||||
|
||||
let symbol = symbols.next().unwrap();
|
||||
println!("{:?}", symbol);
|
||||
assert_eq!(symbol.name(), Ok("v2"));
|
||||
assert_eq!(symbol.kind(), SymbolKind::Data);
|
||||
assert_eq!(symbol.section(), read::SymbolSection::Common);
|
||||
assert_eq!(symbol.scope(), SymbolScope::Linkage);
|
||||
assert_eq!(symbol.is_weak(), false);
|
||||
assert_eq!(symbol.is_undefined(), false);
|
||||
assert_eq!(symbol.address(), 0);
|
||||
assert_eq!(symbol.size(), 8);
|
||||
|
||||
let symbol = symbols.next();
|
||||
assert!(symbol.is_none(), "unexpected symbol {:?}", symbol);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn macho_x86_64_common() {
|
||||
let mut object = write::Object::new(
|
||||
BinaryFormat::MachO,
|
||||
Architecture::X86_64,
|
||||
Endianness::Little,
|
||||
);
|
||||
|
||||
let symbol = write::Symbol {
|
||||
name: b"v1".to_vec(),
|
||||
value: 0,
|
||||
size: 0,
|
||||
kind: SymbolKind::Data,
|
||||
scope: SymbolScope::Linkage,
|
||||
weak: false,
|
||||
section: write::SymbolSection::Undefined,
|
||||
flags: SymbolFlags::None,
|
||||
};
|
||||
object.add_common_symbol(symbol, 4, 4);
|
||||
|
||||
let symbol = write::Symbol {
|
||||
name: b"v2".to_vec(),
|
||||
value: 0,
|
||||
size: 0,
|
||||
kind: SymbolKind::Data,
|
||||
scope: SymbolScope::Linkage,
|
||||
weak: false,
|
||||
section: write::SymbolSection::Undefined,
|
||||
flags: SymbolFlags::None,
|
||||
};
|
||||
object.add_common_symbol(symbol, 8, 8);
|
||||
|
||||
let bytes = object.write().unwrap();
|
||||
|
||||
//std::fs::write(&"common.o", &bytes).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 common = sections.next().unwrap();
|
||||
println!("{:?}", common);
|
||||
let common_index = common.index();
|
||||
assert_eq!(common.name(), Ok("__common"));
|
||||
assert_eq!(common.segment_name(), Ok(Some("__DATA")));
|
||||
assert_eq!(common.kind(), SectionKind::Common);
|
||||
assert_eq!(common.size(), 16);
|
||||
assert_eq!(common.data(), Ok(&[][..]));
|
||||
|
||||
let section = sections.next();
|
||||
assert!(section.is_none(), "unexpected section {:?}", section);
|
||||
|
||||
let mut symbols = object.symbols();
|
||||
|
||||
let symbol = symbols.next().unwrap();
|
||||
println!("{:?}", symbol);
|
||||
assert_eq!(symbol.name(), Ok("_v1"));
|
||||
assert_eq!(symbol.kind(), SymbolKind::Data);
|
||||
assert_eq!(symbol.section_index(), Some(common_index));
|
||||
assert_eq!(symbol.scope(), SymbolScope::Linkage);
|
||||
assert_eq!(symbol.is_weak(), false);
|
||||
assert_eq!(symbol.is_undefined(), false);
|
||||
assert_eq!(symbol.address(), 0);
|
||||
|
||||
let symbol = symbols.next().unwrap();
|
||||
println!("{:?}", symbol);
|
||||
assert_eq!(symbol.name(), Ok("_v2"));
|
||||
assert_eq!(symbol.kind(), SymbolKind::Data);
|
||||
assert_eq!(symbol.section_index(), Some(common_index));
|
||||
assert_eq!(symbol.scope(), SymbolScope::Linkage);
|
||||
assert_eq!(symbol.is_weak(), false);
|
||||
assert_eq!(symbol.is_undefined(), false);
|
||||
assert_eq!(symbol.address(), 8);
|
||||
|
||||
let symbol = symbols.next();
|
||||
assert!(symbol.is_none(), "unexpected symbol {:?}", symbol);
|
||||
}
|
289
vendor/object/tests/round_trip/elf.rs
vendored
Normal file
289
vendor/object/tests/round_trip/elf.rs
vendored
Normal file
@ -0,0 +1,289 @@
|
||||
use object::read::elf::{FileHeader, SectionHeader};
|
||||
use object::read::{Object, ObjectSection, ObjectSymbol};
|
||||
use object::{
|
||||
elf, read, write, Architecture, BinaryFormat, Endianness, LittleEndian, SectionIndex,
|
||||
SectionKind, SymbolFlags, SymbolKind, SymbolScope, SymbolSection, U32,
|
||||
};
|
||||
use std::io::Write;
|
||||
|
||||
#[test]
|
||||
fn symtab_shndx() {
|
||||
let mut object =
|
||||
write::Object::new(BinaryFormat::Elf, Architecture::X86_64, Endianness::Little);
|
||||
|
||||
for i in 0..0x10000 {
|
||||
let name = format!("func{}", i).into_bytes();
|
||||
let (section, offset) =
|
||||
object.add_subsection(write::StandardSection::Text, &name, &[0xcc], 1);
|
||||
object.add_symbol(write::Symbol {
|
||||
name,
|
||||
value: offset,
|
||||
size: 1,
|
||||
kind: SymbolKind::Text,
|
||||
scope: SymbolScope::Linkage,
|
||||
weak: false,
|
||||
section: write::SymbolSection::Section(section),
|
||||
flags: SymbolFlags::None,
|
||||
});
|
||||
}
|
||||
let bytes = object.write().unwrap();
|
||||
|
||||
//std::fs::write(&"symtab_shndx.o", &bytes).unwrap();
|
||||
|
||||
let object = read::File::parse(&*bytes).unwrap();
|
||||
assert_eq!(object.format(), BinaryFormat::Elf);
|
||||
assert_eq!(object.architecture(), Architecture::X86_64);
|
||||
|
||||
for symbol in object.symbols().skip(1) {
|
||||
assert_eq!(
|
||||
symbol.section(),
|
||||
SymbolSection::Section(SectionIndex(symbol.index().0))
|
||||
);
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn aligned_sections() {
|
||||
let mut object =
|
||||
write::Object::new(BinaryFormat::Elf, Architecture::X86_64, Endianness::Little);
|
||||
|
||||
let text_section_id = object.add_section(vec![], b".text".to_vec(), SectionKind::Text);
|
||||
let text_section = object.section_mut(text_section_id);
|
||||
text_section.set_data(&[][..], 4096);
|
||||
|
||||
let data_section_id = object.add_section(vec![], b".data".to_vec(), SectionKind::Data);
|
||||
let data_section = object.section_mut(data_section_id);
|
||||
data_section.set_data(&b"1234"[..], 16);
|
||||
|
||||
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();
|
||||
let _ = sections.next().unwrap();
|
||||
|
||||
let section = sections.next().unwrap();
|
||||
assert_eq!(section.name(), Ok(".text"));
|
||||
assert_eq!(section.file_range(), Some((4096, 0)));
|
||||
|
||||
let section = sections.next().unwrap();
|
||||
assert_eq!(section.name(), Ok(".data"));
|
||||
assert_eq!(section.file_range(), Some((4096, 4)));
|
||||
}
|
||||
|
||||
#[cfg(feature = "compression")]
|
||||
#[test]
|
||||
fn compression_zlib() {
|
||||
use object::read::ObjectSection;
|
||||
use object::LittleEndian as LE;
|
||||
|
||||
let data = b"test data data data";
|
||||
let len = data.len() as u64;
|
||||
|
||||
let mut ch = object::elf::CompressionHeader64::<LE>::default();
|
||||
ch.ch_type.set(LE, object::elf::ELFCOMPRESS_ZLIB);
|
||||
ch.ch_size.set(LE, len);
|
||||
ch.ch_addralign.set(LE, 1);
|
||||
|
||||
let mut buf = Vec::new();
|
||||
buf.write(object::bytes_of(&ch)).unwrap();
|
||||
let mut encoder = flate2::write::ZlibEncoder::new(buf, flate2::Compression::default());
|
||||
encoder.write_all(data).unwrap();
|
||||
let compressed = encoder.finish().unwrap();
|
||||
|
||||
let mut object =
|
||||
write::Object::new(BinaryFormat::Elf, Architecture::X86_64, Endianness::Little);
|
||||
let section = object.add_section(
|
||||
Vec::new(),
|
||||
b".debug_info".to_vec(),
|
||||
object::SectionKind::Other,
|
||||
);
|
||||
object.section_mut(section).set_data(compressed, 1);
|
||||
object.section_mut(section).flags = object::SectionFlags::Elf {
|
||||
sh_flags: object::elf::SHF_COMPRESSED.into(),
|
||||
};
|
||||
let bytes = object.write().unwrap();
|
||||
|
||||
//std::fs::write(&"compression.o", &bytes).unwrap();
|
||||
|
||||
let object = read::File::parse(&*bytes).unwrap();
|
||||
assert_eq!(object.format(), BinaryFormat::Elf);
|
||||
assert_eq!(object.architecture(), Architecture::X86_64);
|
||||
|
||||
let section = object.section_by_name(".debug_info").unwrap();
|
||||
let uncompressed = section.uncompressed_data().unwrap();
|
||||
assert_eq!(data, &*uncompressed);
|
||||
}
|
||||
|
||||
#[cfg(feature = "compression")]
|
||||
#[test]
|
||||
fn compression_gnu() {
|
||||
use object::read::ObjectSection;
|
||||
use std::io::Write;
|
||||
|
||||
let data = b"test data data data";
|
||||
let len = data.len() as u32;
|
||||
|
||||
let mut buf = Vec::new();
|
||||
buf.write_all(b"ZLIB\0\0\0\0").unwrap();
|
||||
buf.write_all(&len.to_be_bytes()).unwrap();
|
||||
let mut encoder = flate2::write::ZlibEncoder::new(buf, flate2::Compression::default());
|
||||
encoder.write_all(data).unwrap();
|
||||
let compressed = encoder.finish().unwrap();
|
||||
|
||||
let mut object =
|
||||
write::Object::new(BinaryFormat::Elf, Architecture::X86_64, Endianness::Little);
|
||||
let section = object.add_section(
|
||||
Vec::new(),
|
||||
b".zdebug_info".to_vec(),
|
||||
object::SectionKind::Other,
|
||||
);
|
||||
object.section_mut(section).set_data(compressed, 1);
|
||||
let bytes = object.write().unwrap();
|
||||
|
||||
//std::fs::write(&"compression.o", &bytes).unwrap();
|
||||
|
||||
let object = read::File::parse(&*bytes).unwrap();
|
||||
assert_eq!(object.format(), BinaryFormat::Elf);
|
||||
assert_eq!(object.architecture(), Architecture::X86_64);
|
||||
|
||||
let section = object.section_by_name(".zdebug_info").unwrap();
|
||||
let uncompressed = section.uncompressed_data().unwrap();
|
||||
assert_eq!(data, &*uncompressed);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn note() {
|
||||
let endian = Endianness::Little;
|
||||
let mut object = write::Object::new(BinaryFormat::Elf, Architecture::X86_64, endian);
|
||||
|
||||
// Add note section with align = 4.
|
||||
let mut buffer = Vec::new();
|
||||
|
||||
buffer
|
||||
.write(object::bytes_of(&elf::NoteHeader32 {
|
||||
n_namesz: U32::new(endian, 6),
|
||||
n_descsz: U32::new(endian, 11),
|
||||
n_type: U32::new(endian, 1),
|
||||
}))
|
||||
.unwrap();
|
||||
buffer.write(b"name1\0\0\0").unwrap();
|
||||
buffer.write(b"descriptor\0\0").unwrap();
|
||||
|
||||
buffer
|
||||
.write(object::bytes_of(&elf::NoteHeader32 {
|
||||
n_namesz: U32::new(endian, 6),
|
||||
n_descsz: U32::new(endian, 11),
|
||||
n_type: U32::new(endian, 2),
|
||||
}))
|
||||
.unwrap();
|
||||
buffer.write(b"name2\0\0\0").unwrap();
|
||||
buffer.write(b"descriptor\0\0").unwrap();
|
||||
|
||||
let section = object.add_section(Vec::new(), b".note4".to_vec(), SectionKind::Note);
|
||||
object.section_mut(section).set_data(buffer, 4);
|
||||
|
||||
// Add note section with align = 8.
|
||||
let mut buffer = Vec::new();
|
||||
|
||||
buffer
|
||||
.write(object::bytes_of(&elf::NoteHeader32 {
|
||||
n_namesz: U32::new(endian, 6),
|
||||
n_descsz: U32::new(endian, 11),
|
||||
n_type: U32::new(endian, 1),
|
||||
}))
|
||||
.unwrap();
|
||||
buffer.write(b"name1\0\0\0\0\0\0\0").unwrap();
|
||||
buffer.write(b"descriptor\0\0\0\0\0\0").unwrap();
|
||||
|
||||
buffer
|
||||
.write(object::bytes_of(&elf::NoteHeader32 {
|
||||
n_namesz: U32::new(endian, 4),
|
||||
n_descsz: U32::new(endian, 11),
|
||||
n_type: U32::new(endian, 2),
|
||||
}))
|
||||
.unwrap();
|
||||
buffer.write(b"abc\0").unwrap();
|
||||
buffer.write(b"descriptor\0\0\0\0\0\0").unwrap();
|
||||
|
||||
let section = object.add_section(Vec::new(), b".note8".to_vec(), SectionKind::Note);
|
||||
object.section_mut(section).set_data(buffer, 8);
|
||||
|
||||
let bytes = &*object.write().unwrap();
|
||||
|
||||
//std::fs::write(&"note.o", &bytes).unwrap();
|
||||
|
||||
let header = elf::FileHeader64::parse(bytes).unwrap();
|
||||
let endian: LittleEndian = header.endian().unwrap();
|
||||
let sections = header.sections(endian, bytes).unwrap();
|
||||
|
||||
let section = sections.section(SectionIndex(1)).unwrap();
|
||||
assert_eq!(sections.section_name(endian, section).unwrap(), b".note4");
|
||||
assert_eq!(section.sh_addralign(endian), 4);
|
||||
let mut notes = section.notes(endian, bytes).unwrap().unwrap();
|
||||
let note = notes.next().unwrap().unwrap();
|
||||
assert_eq!(note.name(), b"name1");
|
||||
assert_eq!(note.desc(), b"descriptor\0");
|
||||
assert_eq!(note.n_type(endian), 1);
|
||||
let note = notes.next().unwrap().unwrap();
|
||||
assert_eq!(note.name(), b"name2");
|
||||
assert_eq!(note.desc(), b"descriptor\0");
|
||||
assert_eq!(note.n_type(endian), 2);
|
||||
assert!(notes.next().unwrap().is_none());
|
||||
|
||||
let section = sections.section(SectionIndex(2)).unwrap();
|
||||
assert_eq!(sections.section_name(endian, section).unwrap(), b".note8");
|
||||
assert_eq!(section.sh_addralign(endian), 8);
|
||||
let mut notes = section.notes(endian, bytes).unwrap().unwrap();
|
||||
let note = notes.next().unwrap().unwrap();
|
||||
assert_eq!(note.name(), b"name1");
|
||||
assert_eq!(note.desc(), b"descriptor\0");
|
||||
assert_eq!(note.n_type(endian), 1);
|
||||
let note = notes.next().unwrap().unwrap();
|
||||
assert_eq!(note.name(), b"abc");
|
||||
assert_eq!(note.desc(), b"descriptor\0");
|
||||
assert_eq!(note.n_type(endian), 2);
|
||||
assert!(notes.next().unwrap().is_none());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn gnu_property() {
|
||||
gnu_property_inner::<elf::FileHeader32<Endianness>>(Architecture::I386);
|
||||
gnu_property_inner::<elf::FileHeader64<Endianness>>(Architecture::X86_64);
|
||||
}
|
||||
|
||||
fn gnu_property_inner<Elf: FileHeader<Endian = Endianness>>(architecture: Architecture) {
|
||||
let endian = Endianness::Little;
|
||||
let mut object = write::Object::new(BinaryFormat::Elf, architecture, endian);
|
||||
object.add_elf_gnu_property_u32(
|
||||
elf::GNU_PROPERTY_X86_FEATURE_1_AND,
|
||||
elf::GNU_PROPERTY_X86_FEATURE_1_IBT | elf::GNU_PROPERTY_X86_FEATURE_1_SHSTK,
|
||||
);
|
||||
|
||||
let bytes = &*object.write().unwrap();
|
||||
|
||||
//std::fs::write(&"note.o", &bytes).unwrap();
|
||||
|
||||
let header = Elf::parse(bytes).unwrap();
|
||||
assert_eq!(header.endian().unwrap(), endian);
|
||||
let sections = header.sections(endian, bytes).unwrap();
|
||||
let section = sections.section(SectionIndex(1)).unwrap();
|
||||
assert_eq!(
|
||||
sections.section_name(endian, section).unwrap(),
|
||||
b".note.gnu.property"
|
||||
);
|
||||
assert_eq!(section.sh_flags(endian).into(), u64::from(elf::SHF_ALLOC));
|
||||
let mut notes = section.notes(endian, bytes).unwrap().unwrap();
|
||||
let note = notes.next().unwrap().unwrap();
|
||||
let mut props = note.gnu_properties(endian).unwrap();
|
||||
let prop = props.next().unwrap().unwrap();
|
||||
assert_eq!(prop.pr_type(), elf::GNU_PROPERTY_X86_FEATURE_1_AND);
|
||||
assert_eq!(
|
||||
prop.data_u32(endian).unwrap(),
|
||||
elf::GNU_PROPERTY_X86_FEATURE_1_IBT | elf::GNU_PROPERTY_X86_FEATURE_1_SHSTK
|
||||
);
|
||||
assert!(props.next().unwrap().is_none());
|
||||
assert!(notes.next().unwrap().is_none());
|
||||
}
|
64
vendor/object/tests/round_trip/macho.rs
vendored
Normal file
64
vendor/object/tests/round_trip/macho.rs
vendored
Normal file
@ -0,0 +1,64 @@
|
||||
use object::read::macho::MachHeader;
|
||||
use object::read::{Object, ObjectSection};
|
||||
use object::{macho, read, write, Architecture, BinaryFormat, Endianness};
|
||||
|
||||
// Test that segment size is valid when the first section needs alignment.
|
||||
#[test]
|
||||
fn issue_286_segment_file_size() {
|
||||
let mut object = write::Object::new(
|
||||
BinaryFormat::MachO,
|
||||
Architecture::X86_64,
|
||||
Endianness::Little,
|
||||
);
|
||||
|
||||
let text = object.section_id(write::StandardSection::Text);
|
||||
object.append_section_data(text, &[1; 30], 0x1000);
|
||||
|
||||
let bytes = &*object.write().unwrap();
|
||||
let header = macho::MachHeader64::parse(bytes, 0).unwrap();
|
||||
let endian: Endianness = header.endian().unwrap();
|
||||
let mut commands = header.load_commands(endian, bytes, 0).unwrap();
|
||||
let command = commands.next().unwrap().unwrap();
|
||||
let (segment, _) = command.segment_64().unwrap().unwrap();
|
||||
assert_eq!(segment.vmsize.get(endian), 30);
|
||||
assert_eq!(segment.filesize.get(endian), 30);
|
||||
}
|
||||
|
||||
// We were emitting section file alignment padding that didn't match the address alignment padding.
|
||||
#[test]
|
||||
fn issue_552_section_file_alignment() {
|
||||
let mut object = write::Object::new(
|
||||
BinaryFormat::MachO,
|
||||
Architecture::X86_64,
|
||||
Endianness::Little,
|
||||
);
|
||||
|
||||
// The starting file offset is not a multiple of 32 (checked later).
|
||||
// Length of 32 ensures that the file offset of the end of this section is still not a
|
||||
// multiple of 32.
|
||||
let section = object.add_section(vec![], vec![], object::SectionKind::ReadOnlyDataWithRel);
|
||||
object.append_section_data(section, &vec![0u8; 32], 1);
|
||||
|
||||
// Address is already aligned correctly, so there must not any padding,
|
||||
// even though file offset is not aligned.
|
||||
let section = object.add_section(vec![], vec![], object::SectionKind::ReadOnlyData);
|
||||
object.append_section_data(section, &vec![0u8; 1], 32);
|
||||
|
||||
let bytes = &*object.write().unwrap();
|
||||
//std::fs::write(&"align.o", &bytes).unwrap();
|
||||
let object = read::File::parse(bytes).unwrap();
|
||||
let mut sections = object.sections();
|
||||
|
||||
let section = sections.next().unwrap();
|
||||
let offset = section.file_range().unwrap().0;
|
||||
// Check file offset is not aligned to 32.
|
||||
assert_ne!(offset % 32, 0);
|
||||
assert_eq!(section.address(), 0);
|
||||
assert_eq!(section.size(), 32);
|
||||
|
||||
let section = sections.next().unwrap();
|
||||
// Check there is no padding.
|
||||
assert_eq!(section.file_range(), Some((offset + 32, 1)));
|
||||
assert_eq!(section.address(), 32);
|
||||
assert_eq!(section.size(), 1);
|
||||
}
|
686
vendor/object/tests/round_trip/mod.rs
vendored
Normal file
686
vendor/object/tests/round_trip/mod.rs
vendored
Normal file
@ -0,0 +1,686 @@
|
||||
#![cfg(all(feature = "read", feature = "write"))]
|
||||
|
||||
use object::read::{Object, ObjectSection, ObjectSymbol};
|
||||
use object::{read, write, SectionIndex, SubArchitecture};
|
||||
use object::{
|
||||
Architecture, BinaryFormat, Endianness, RelocationEncoding, RelocationKind, SectionKind,
|
||||
SymbolFlags, SymbolKind, SymbolScope, SymbolSection,
|
||||
};
|
||||
|
||||
mod bss;
|
||||
mod coff;
|
||||
mod comdat;
|
||||
mod common;
|
||||
mod elf;
|
||||
mod macho;
|
||||
mod section_flags;
|
||||
mod tls;
|
||||
|
||||
#[test]
|
||||
fn coff_any() {
|
||||
for (arch, sub_arch) in [
|
||||
(Architecture::Aarch64, None),
|
||||
(Architecture::Aarch64, Some(SubArchitecture::Arm64EC)),
|
||||
(Architecture::Arm, None),
|
||||
(Architecture::I386, None),
|
||||
(Architecture::X86_64, None),
|
||||
]
|
||||
.iter()
|
||||
.copied()
|
||||
{
|
||||
let mut object = write::Object::new(BinaryFormat::Coff, arch, Endianness::Little);
|
||||
object.set_sub_architecture(sub_arch);
|
||||
|
||||
object.add_file_symbol(b"file.c".to_vec());
|
||||
|
||||
let text = object.section_id(write::StandardSection::Text);
|
||||
object.append_section_data(text, &[1; 30], 4);
|
||||
|
||||
let func1_offset = object.append_section_data(text, &[1; 30], 4);
|
||||
assert_eq!(func1_offset, 32);
|
||||
let func1_symbol = object.add_symbol(write::Symbol {
|
||||
name: b"func1".to_vec(),
|
||||
value: func1_offset,
|
||||
size: 32,
|
||||
kind: SymbolKind::Text,
|
||||
scope: SymbolScope::Linkage,
|
||||
weak: false,
|
||||
section: write::SymbolSection::Section(text),
|
||||
flags: SymbolFlags::None,
|
||||
});
|
||||
let func2_offset = object.append_section_data(text, &[1; 30], 4);
|
||||
assert_eq!(func2_offset, 64);
|
||||
object.add_symbol(write::Symbol {
|
||||
name: b"func2_long".to_vec(),
|
||||
value: func2_offset,
|
||||
size: 32,
|
||||
kind: SymbolKind::Text,
|
||||
scope: SymbolScope::Linkage,
|
||||
weak: false,
|
||||
section: write::SymbolSection::Section(text),
|
||||
flags: SymbolFlags::None,
|
||||
});
|
||||
object
|
||||
.add_relocation(
|
||||
text,
|
||||
write::Relocation {
|
||||
offset: 8,
|
||||
size: arch.address_size().unwrap().bytes() * 8,
|
||||
kind: RelocationKind::Absolute,
|
||||
encoding: RelocationEncoding::Generic,
|
||||
symbol: func1_symbol,
|
||||
addend: 0,
|
||||
},
|
||||
)
|
||||
.unwrap();
|
||||
|
||||
let bytes = object.write().unwrap();
|
||||
let object = read::File::parse(&*bytes).unwrap();
|
||||
assert_eq!(object.format(), BinaryFormat::Coff);
|
||||
assert_eq!(object.architecture(), arch);
|
||||
assert_eq!(object.sub_architecture(), sub_arch);
|
||||
assert_eq!(object.endianness(), Endianness::Little);
|
||||
|
||||
let mut sections = object.sections();
|
||||
|
||||
let text = sections.next().unwrap();
|
||||
println!("{:?}", text);
|
||||
let text_index = text.index();
|
||||
assert_eq!(text.name(), Ok(".text"));
|
||||
assert_eq!(text.kind(), SectionKind::Text);
|
||||
assert_eq!(text.address(), 0);
|
||||
assert_eq!(text.size(), 94);
|
||||
assert_eq!(&text.data().unwrap()[..30], &[1; 30]);
|
||||
assert_eq!(&text.data().unwrap()[32..62], &[1; 30]);
|
||||
|
||||
let mut symbols = object.symbols();
|
||||
|
||||
let symbol = symbols.next().unwrap();
|
||||
println!("{:?}", symbol);
|
||||
assert_eq!(symbol.name(), Ok("file.c"));
|
||||
assert_eq!(symbol.address(), 0);
|
||||
assert_eq!(symbol.kind(), SymbolKind::File);
|
||||
assert_eq!(symbol.section(), SymbolSection::None);
|
||||
assert_eq!(symbol.scope(), SymbolScope::Compilation);
|
||||
assert_eq!(symbol.is_weak(), false);
|
||||
|
||||
let decorated_name = |name: &str| {
|
||||
if arch == Architecture::I386 {
|
||||
format!("_{name}")
|
||||
} else {
|
||||
name.to_owned()
|
||||
}
|
||||
};
|
||||
|
||||
let symbol = symbols.next().unwrap();
|
||||
println!("{:?}", symbol);
|
||||
let func1_symbol = symbol.index();
|
||||
assert_eq!(symbol.name(), Ok(decorated_name("func1").as_str()));
|
||||
assert_eq!(symbol.address(), func1_offset);
|
||||
assert_eq!(symbol.kind(), SymbolKind::Text);
|
||||
assert_eq!(symbol.section_index(), Some(text_index));
|
||||
assert_eq!(symbol.scope(), SymbolScope::Linkage);
|
||||
assert_eq!(symbol.is_weak(), false);
|
||||
assert_eq!(symbol.is_undefined(), false);
|
||||
|
||||
let symbol = symbols.next().unwrap();
|
||||
println!("{:?}", symbol);
|
||||
assert_eq!(symbol.name(), Ok(decorated_name("func2_long").as_str()));
|
||||
assert_eq!(symbol.address(), func2_offset);
|
||||
assert_eq!(symbol.kind(), SymbolKind::Text);
|
||||
assert_eq!(symbol.section_index(), Some(text_index));
|
||||
assert_eq!(symbol.scope(), SymbolScope::Linkage);
|
||||
assert_eq!(symbol.is_weak(), false);
|
||||
assert_eq!(symbol.is_undefined(), false);
|
||||
|
||||
let mut relocations = text.relocations();
|
||||
|
||||
let (offset, relocation) = relocations.next().unwrap();
|
||||
println!("{:?}", relocation);
|
||||
assert_eq!(offset, 8);
|
||||
assert_eq!(relocation.kind(), RelocationKind::Absolute);
|
||||
assert_eq!(relocation.encoding(), RelocationEncoding::Generic);
|
||||
assert_eq!(relocation.size(), arch.address_size().unwrap().bytes() * 8);
|
||||
assert_eq!(
|
||||
relocation.target(),
|
||||
read::RelocationTarget::Symbol(func1_symbol)
|
||||
);
|
||||
assert_eq!(relocation.addend(), 0);
|
||||
|
||||
let map = object.symbol_map();
|
||||
let symbol = map.get(func1_offset + 1).unwrap();
|
||||
assert_eq!(symbol.address(), func1_offset);
|
||||
assert_eq!(symbol.name(), decorated_name("func1"));
|
||||
assert_eq!(map.get(func1_offset - 1), None);
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn elf_x86_64() {
|
||||
let mut object =
|
||||
write::Object::new(BinaryFormat::Elf, Architecture::X86_64, Endianness::Little);
|
||||
|
||||
object.add_file_symbol(b"file.c".to_vec());
|
||||
|
||||
let text = object.section_id(write::StandardSection::Text);
|
||||
object.append_section_data(text, &[1; 30], 4);
|
||||
|
||||
let func1_offset = object.append_section_data(text, &[1; 30], 4);
|
||||
assert_eq!(func1_offset, 32);
|
||||
let func1_symbol = object.add_symbol(write::Symbol {
|
||||
name: b"func1".to_vec(),
|
||||
value: func1_offset,
|
||||
size: 32,
|
||||
kind: SymbolKind::Text,
|
||||
scope: SymbolScope::Linkage,
|
||||
weak: false,
|
||||
section: write::SymbolSection::Section(text),
|
||||
flags: SymbolFlags::None,
|
||||
});
|
||||
object
|
||||
.add_relocation(
|
||||
text,
|
||||
write::Relocation {
|
||||
offset: 8,
|
||||
size: 64,
|
||||
kind: RelocationKind::Absolute,
|
||||
encoding: RelocationEncoding::Generic,
|
||||
symbol: func1_symbol,
|
||||
addend: 0,
|
||||
},
|
||||
)
|
||||
.unwrap();
|
||||
|
||||
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);
|
||||
assert_eq!(object.endianness(), Endianness::Little);
|
||||
|
||||
let mut sections = object.sections();
|
||||
|
||||
let section = sections.next().unwrap();
|
||||
println!("{:?}", section);
|
||||
assert_eq!(section.name(), Ok(""));
|
||||
assert_eq!(section.kind(), SectionKind::Metadata);
|
||||
assert_eq!(section.address(), 0);
|
||||
assert_eq!(section.size(), 0);
|
||||
|
||||
let text = sections.next().unwrap();
|
||||
println!("{:?}", text);
|
||||
let text_index = text.index();
|
||||
assert_eq!(text.name(), Ok(".text"));
|
||||
assert_eq!(text.kind(), SectionKind::Text);
|
||||
assert_eq!(text.address(), 0);
|
||||
assert_eq!(text.size(), 62);
|
||||
assert_eq!(&text.data().unwrap()[..30], &[1; 30]);
|
||||
assert_eq!(&text.data().unwrap()[32..62], &[1; 30]);
|
||||
|
||||
let mut symbols = object.symbols();
|
||||
|
||||
let symbol = symbols.next().unwrap();
|
||||
println!("{:?}", symbol);
|
||||
assert_eq!(symbol.name(), Ok(""));
|
||||
assert_eq!(symbol.address(), 0);
|
||||
assert_eq!(symbol.kind(), SymbolKind::Null);
|
||||
assert_eq!(symbol.section_index(), None);
|
||||
assert_eq!(symbol.scope(), SymbolScope::Unknown);
|
||||
assert_eq!(symbol.is_weak(), false);
|
||||
assert_eq!(symbol.is_undefined(), true);
|
||||
|
||||
let symbol = symbols.next().unwrap();
|
||||
println!("{:?}", symbol);
|
||||
assert_eq!(symbol.name(), Ok("file.c"));
|
||||
assert_eq!(symbol.address(), 0);
|
||||
assert_eq!(symbol.kind(), SymbolKind::File);
|
||||
assert_eq!(symbol.section(), SymbolSection::None);
|
||||
assert_eq!(symbol.scope(), SymbolScope::Compilation);
|
||||
assert_eq!(symbol.is_weak(), false);
|
||||
|
||||
let symbol = symbols.next().unwrap();
|
||||
println!("{:?}", symbol);
|
||||
let func1_symbol = symbol.index();
|
||||
assert_eq!(symbol.name(), Ok("func1"));
|
||||
assert_eq!(symbol.address(), func1_offset);
|
||||
assert_eq!(symbol.kind(), SymbolKind::Text);
|
||||
assert_eq!(symbol.section_index(), Some(text_index));
|
||||
assert_eq!(symbol.scope(), SymbolScope::Linkage);
|
||||
assert_eq!(symbol.is_weak(), false);
|
||||
assert_eq!(symbol.is_undefined(), false);
|
||||
|
||||
let mut relocations = text.relocations();
|
||||
|
||||
let (offset, relocation) = relocations.next().unwrap();
|
||||
println!("{:?}", relocation);
|
||||
assert_eq!(offset, 8);
|
||||
assert_eq!(relocation.kind(), RelocationKind::Absolute);
|
||||
assert_eq!(relocation.encoding(), RelocationEncoding::Generic);
|
||||
assert_eq!(relocation.size(), 64);
|
||||
assert_eq!(
|
||||
relocation.target(),
|
||||
read::RelocationTarget::Symbol(func1_symbol)
|
||||
);
|
||||
assert_eq!(relocation.addend(), 0);
|
||||
|
||||
let map = object.symbol_map();
|
||||
let symbol = map.get(func1_offset + 1).unwrap();
|
||||
assert_eq!(symbol.address(), func1_offset);
|
||||
assert_eq!(symbol.name(), "func1");
|
||||
assert_eq!(map.get(func1_offset - 1), None);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn elf_any() {
|
||||
for (arch, endian) in [
|
||||
(Architecture::Aarch64, Endianness::Little),
|
||||
(Architecture::Aarch64_Ilp32, Endianness::Little),
|
||||
(Architecture::Arm, Endianness::Little),
|
||||
(Architecture::Avr, Endianness::Little),
|
||||
(Architecture::Bpf, Endianness::Little),
|
||||
(Architecture::Csky, Endianness::Little),
|
||||
(Architecture::I386, Endianness::Little),
|
||||
(Architecture::X86_64, Endianness::Little),
|
||||
(Architecture::X86_64_X32, Endianness::Little),
|
||||
(Architecture::Hexagon, Endianness::Little),
|
||||
(Architecture::LoongArch64, Endianness::Little),
|
||||
(Architecture::Mips, Endianness::Little),
|
||||
(Architecture::Mips64, Endianness::Little),
|
||||
(Architecture::Msp430, Endianness::Little),
|
||||
(Architecture::PowerPc, Endianness::Big),
|
||||
(Architecture::PowerPc64, Endianness::Big),
|
||||
(Architecture::Riscv32, Endianness::Little),
|
||||
(Architecture::Riscv64, Endianness::Little),
|
||||
(Architecture::S390x, Endianness::Big),
|
||||
(Architecture::Sbf, Endianness::Little),
|
||||
(Architecture::Sparc64, Endianness::Big),
|
||||
(Architecture::Xtensa, Endianness::Little),
|
||||
]
|
||||
.iter()
|
||||
.copied()
|
||||
{
|
||||
let mut object = write::Object::new(BinaryFormat::Elf, arch, endian);
|
||||
|
||||
let section = object.section_id(write::StandardSection::Data);
|
||||
object.append_section_data(section, &[1; 30], 4);
|
||||
let symbol = object.section_symbol(section);
|
||||
|
||||
object
|
||||
.add_relocation(
|
||||
section,
|
||||
write::Relocation {
|
||||
offset: 8,
|
||||
size: 32,
|
||||
kind: RelocationKind::Absolute,
|
||||
encoding: RelocationEncoding::Generic,
|
||||
symbol,
|
||||
addend: 0,
|
||||
},
|
||||
)
|
||||
.unwrap();
|
||||
if arch.address_size().unwrap().bytes() >= 8 {
|
||||
object
|
||||
.add_relocation(
|
||||
section,
|
||||
write::Relocation {
|
||||
offset: 16,
|
||||
size: 64,
|
||||
kind: RelocationKind::Absolute,
|
||||
encoding: RelocationEncoding::Generic,
|
||||
symbol,
|
||||
addend: 0,
|
||||
},
|
||||
)
|
||||
.unwrap();
|
||||
}
|
||||
|
||||
let bytes = object.write().unwrap();
|
||||
let object = read::File::parse(&*bytes).unwrap();
|
||||
println!("{:?}", object.architecture());
|
||||
assert_eq!(object.format(), BinaryFormat::Elf);
|
||||
assert_eq!(object.architecture(), arch);
|
||||
assert_eq!(object.endianness(), endian);
|
||||
|
||||
let mut sections = object.sections();
|
||||
|
||||
let section = sections.next().unwrap();
|
||||
println!("{:?}", section);
|
||||
assert_eq!(section.name(), Ok(""));
|
||||
assert_eq!(section.kind(), SectionKind::Metadata);
|
||||
assert_eq!(section.address(), 0);
|
||||
assert_eq!(section.size(), 0);
|
||||
|
||||
let data = sections.next().unwrap();
|
||||
println!("{:?}", data);
|
||||
assert_eq!(data.name(), Ok(".data"));
|
||||
assert_eq!(data.kind(), SectionKind::Data);
|
||||
|
||||
let mut relocations = data.relocations();
|
||||
|
||||
let (offset, relocation) = relocations.next().unwrap();
|
||||
println!("{:?}", relocation);
|
||||
assert_eq!(offset, 8);
|
||||
assert_eq!(relocation.kind(), RelocationKind::Absolute);
|
||||
assert_eq!(relocation.encoding(), RelocationEncoding::Generic);
|
||||
assert_eq!(relocation.size(), 32);
|
||||
assert_eq!(relocation.addend(), 0);
|
||||
|
||||
if arch.address_size().unwrap().bytes() >= 8 {
|
||||
let (offset, relocation) = relocations.next().unwrap();
|
||||
println!("{:?}", relocation);
|
||||
assert_eq!(offset, 16);
|
||||
assert_eq!(relocation.kind(), RelocationKind::Absolute);
|
||||
assert_eq!(relocation.encoding(), RelocationEncoding::Generic);
|
||||
assert_eq!(relocation.size(), 64);
|
||||
assert_eq!(relocation.addend(), 0);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn macho_x86_64() {
|
||||
let mut object = write::Object::new(
|
||||
BinaryFormat::MachO,
|
||||
Architecture::X86_64,
|
||||
Endianness::Little,
|
||||
);
|
||||
|
||||
object.add_file_symbol(b"file.c".to_vec());
|
||||
|
||||
let text = object.section_id(write::StandardSection::Text);
|
||||
object.append_section_data(text, &[1; 30], 4);
|
||||
|
||||
let func1_offset = object.append_section_data(text, &[1; 30], 4);
|
||||
assert_eq!(func1_offset, 32);
|
||||
let func1_symbol = object.add_symbol(write::Symbol {
|
||||
name: b"func1".to_vec(),
|
||||
value: func1_offset,
|
||||
size: 32,
|
||||
kind: SymbolKind::Text,
|
||||
scope: SymbolScope::Linkage,
|
||||
weak: false,
|
||||
section: write::SymbolSection::Section(text),
|
||||
flags: SymbolFlags::None,
|
||||
});
|
||||
object
|
||||
.add_relocation(
|
||||
text,
|
||||
write::Relocation {
|
||||
offset: 8,
|
||||
size: 64,
|
||||
kind: RelocationKind::Absolute,
|
||||
encoding: RelocationEncoding::Generic,
|
||||
symbol: func1_symbol,
|
||||
addend: 0,
|
||||
},
|
||||
)
|
||||
.unwrap();
|
||||
object
|
||||
.add_relocation(
|
||||
text,
|
||||
write::Relocation {
|
||||
offset: 16,
|
||||
size: 32,
|
||||
kind: RelocationKind::Relative,
|
||||
encoding: RelocationEncoding::Generic,
|
||||
symbol: func1_symbol,
|
||||
addend: -4,
|
||||
},
|
||||
)
|
||||
.unwrap();
|
||||
|
||||
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);
|
||||
assert_eq!(object.endianness(), Endianness::Little);
|
||||
|
||||
let mut sections = object.sections();
|
||||
|
||||
let text = sections.next().unwrap();
|
||||
println!("{:?}", text);
|
||||
let text_index = text.index();
|
||||
assert_eq!(text.name(), Ok("__text"));
|
||||
assert_eq!(text.segment_name(), Ok(Some("__TEXT")));
|
||||
assert_eq!(text.kind(), SectionKind::Text);
|
||||
assert_eq!(text.address(), 0);
|
||||
assert_eq!(text.size(), 62);
|
||||
assert_eq!(&text.data().unwrap()[..30], &[1; 30]);
|
||||
assert_eq!(&text.data().unwrap()[32..62], &[1; 30]);
|
||||
|
||||
let mut symbols = object.symbols();
|
||||
|
||||
let symbol = symbols.next().unwrap();
|
||||
println!("{:?}", symbol);
|
||||
let func1_symbol = symbol.index();
|
||||
assert_eq!(symbol.name(), Ok("_func1"));
|
||||
assert_eq!(symbol.address(), func1_offset);
|
||||
assert_eq!(symbol.kind(), SymbolKind::Text);
|
||||
assert_eq!(symbol.section_index(), Some(text_index));
|
||||
assert_eq!(symbol.scope(), SymbolScope::Linkage);
|
||||
assert_eq!(symbol.is_weak(), false);
|
||||
assert_eq!(symbol.is_undefined(), false);
|
||||
|
||||
let mut relocations = text.relocations();
|
||||
|
||||
let (offset, relocation) = relocations.next().unwrap();
|
||||
println!("{:?}", relocation);
|
||||
assert_eq!(offset, 8);
|
||||
assert_eq!(relocation.kind(), RelocationKind::Absolute);
|
||||
assert_eq!(relocation.encoding(), RelocationEncoding::Generic);
|
||||
assert_eq!(relocation.size(), 64);
|
||||
assert_eq!(
|
||||
relocation.target(),
|
||||
read::RelocationTarget::Symbol(func1_symbol)
|
||||
);
|
||||
assert_eq!(relocation.addend(), 0);
|
||||
|
||||
let (offset, relocation) = relocations.next().unwrap();
|
||||
println!("{:?}", relocation);
|
||||
assert_eq!(offset, 16);
|
||||
assert_eq!(relocation.kind(), RelocationKind::Relative);
|
||||
assert_eq!(relocation.encoding(), RelocationEncoding::X86RipRelative);
|
||||
assert_eq!(relocation.size(), 32);
|
||||
assert_eq!(
|
||||
relocation.target(),
|
||||
read::RelocationTarget::Symbol(func1_symbol)
|
||||
);
|
||||
assert_eq!(relocation.addend(), -4);
|
||||
|
||||
let map = object.symbol_map();
|
||||
let symbol = map.get(func1_offset + 1).unwrap();
|
||||
assert_eq!(symbol.address(), func1_offset);
|
||||
assert_eq!(symbol.name(), "_func1");
|
||||
assert_eq!(map.get(func1_offset - 1), None);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn macho_any() {
|
||||
for (arch, subarch, endian) in [
|
||||
(Architecture::Aarch64, None, Endianness::Little),
|
||||
(
|
||||
Architecture::Aarch64,
|
||||
Some(SubArchitecture::Arm64E),
|
||||
Endianness::Little,
|
||||
),
|
||||
(Architecture::Aarch64_Ilp32, None, Endianness::Little),
|
||||
/* TODO:
|
||||
(Architecture::Arm, None, Endianness::Little),
|
||||
*/
|
||||
(Architecture::I386, None, Endianness::Little),
|
||||
(Architecture::X86_64, None, Endianness::Little),
|
||||
/* TODO:
|
||||
(Architecture::PowerPc, None, Endianness::Big),
|
||||
(Architecture::PowerPc64, None, Endianness::Big),
|
||||
*/
|
||||
]
|
||||
.iter()
|
||||
.copied()
|
||||
{
|
||||
let mut object = write::Object::new(BinaryFormat::MachO, arch, endian);
|
||||
object.set_sub_architecture(subarch);
|
||||
|
||||
let section = object.section_id(write::StandardSection::Data);
|
||||
object.append_section_data(section, &[1; 30], 4);
|
||||
let symbol = object.section_symbol(section);
|
||||
|
||||
object
|
||||
.add_relocation(
|
||||
section,
|
||||
write::Relocation {
|
||||
offset: 8,
|
||||
size: 32,
|
||||
kind: RelocationKind::Absolute,
|
||||
encoding: RelocationEncoding::Generic,
|
||||
symbol,
|
||||
addend: 0,
|
||||
},
|
||||
)
|
||||
.unwrap();
|
||||
if arch.address_size().unwrap().bytes() >= 8 {
|
||||
object
|
||||
.add_relocation(
|
||||
section,
|
||||
write::Relocation {
|
||||
offset: 16,
|
||||
size: 64,
|
||||
kind: RelocationKind::Absolute,
|
||||
encoding: RelocationEncoding::Generic,
|
||||
symbol,
|
||||
addend: 0,
|
||||
},
|
||||
)
|
||||
.unwrap();
|
||||
}
|
||||
|
||||
let bytes = object.write().unwrap();
|
||||
let object = read::File::parse(&*bytes).unwrap();
|
||||
println!("{:?}", object.architecture());
|
||||
assert_eq!(object.format(), BinaryFormat::MachO);
|
||||
assert_eq!(object.architecture(), arch);
|
||||
assert_eq!(object.sub_architecture(), subarch);
|
||||
assert_eq!(object.endianness(), endian);
|
||||
|
||||
let mut sections = object.sections();
|
||||
|
||||
let data = sections.next().unwrap();
|
||||
println!("{:?}", data);
|
||||
assert_eq!(data.segment_name(), Ok(Some("__DATA")));
|
||||
assert_eq!(data.name(), Ok("__data"));
|
||||
assert_eq!(data.kind(), SectionKind::Data);
|
||||
|
||||
let mut relocations = data.relocations();
|
||||
|
||||
let (offset, relocation) = relocations.next().unwrap();
|
||||
println!("{:?}", relocation);
|
||||
assert_eq!(offset, 8);
|
||||
assert_eq!(relocation.kind(), RelocationKind::Absolute);
|
||||
assert_eq!(relocation.encoding(), RelocationEncoding::Generic);
|
||||
assert_eq!(relocation.size(), 32);
|
||||
assert_eq!(relocation.addend(), 0);
|
||||
|
||||
if arch.address_size().unwrap().bytes() >= 8 {
|
||||
let (offset, relocation) = relocations.next().unwrap();
|
||||
println!("{:?}", relocation);
|
||||
assert_eq!(offset, 16);
|
||||
assert_eq!(relocation.kind(), RelocationKind::Absolute);
|
||||
assert_eq!(relocation.encoding(), RelocationEncoding::Generic);
|
||||
assert_eq!(relocation.size(), 64);
|
||||
assert_eq!(relocation.addend(), 0);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(feature = "xcoff")]
|
||||
#[test]
|
||||
fn xcoff_powerpc() {
|
||||
for arch in [Architecture::PowerPc, Architecture::PowerPc64] {
|
||||
let mut object = write::Object::new(BinaryFormat::Xcoff, arch, Endianness::Big);
|
||||
|
||||
object.add_file_symbol(b"file.c".to_vec());
|
||||
|
||||
let text = object.section_id(write::StandardSection::Text);
|
||||
object.append_section_data(text, &[1; 30], 4);
|
||||
|
||||
let func1_offset = object.append_section_data(text, &[1; 30], 4);
|
||||
assert_eq!(func1_offset, 32);
|
||||
let func1_symbol = object.add_symbol(write::Symbol {
|
||||
name: b"func1".to_vec(),
|
||||
value: func1_offset,
|
||||
size: 32,
|
||||
kind: SymbolKind::Text,
|
||||
scope: SymbolScope::Linkage,
|
||||
weak: false,
|
||||
section: write::SymbolSection::Section(text),
|
||||
flags: SymbolFlags::None,
|
||||
});
|
||||
|
||||
object
|
||||
.add_relocation(
|
||||
text,
|
||||
write::Relocation {
|
||||
offset: 8,
|
||||
size: 64,
|
||||
kind: RelocationKind::Absolute,
|
||||
encoding: RelocationEncoding::Generic,
|
||||
symbol: func1_symbol,
|
||||
addend: 0,
|
||||
},
|
||||
)
|
||||
.unwrap();
|
||||
|
||||
let bytes = object.write().unwrap();
|
||||
let object = read::File::parse(&*bytes).unwrap();
|
||||
assert_eq!(object.format(), BinaryFormat::Xcoff);
|
||||
assert_eq!(object.architecture(), arch);
|
||||
assert_eq!(object.endianness(), Endianness::Big);
|
||||
|
||||
let mut sections = object.sections();
|
||||
|
||||
let text = sections.next().unwrap();
|
||||
println!("{:?}", text);
|
||||
let text_index = text.index().0;
|
||||
assert_eq!(text.name(), Ok(".text"));
|
||||
assert_eq!(text.kind(), SectionKind::Text);
|
||||
assert_eq!(text.address(), 0);
|
||||
assert_eq!(text.size(), 62);
|
||||
assert_eq!(&text.data().unwrap()[..30], &[1; 30]);
|
||||
assert_eq!(&text.data().unwrap()[32..62], &[1; 30]);
|
||||
|
||||
let mut symbols = object.symbols();
|
||||
|
||||
let mut symbol = symbols.next().unwrap();
|
||||
println!("{:?}", symbol);
|
||||
assert_eq!(symbol.name(), Ok("file.c"));
|
||||
assert_eq!(symbol.address(), 0);
|
||||
assert_eq!(symbol.kind(), SymbolKind::File);
|
||||
assert_eq!(symbol.section_index(), None);
|
||||
assert_eq!(symbol.scope(), SymbolScope::Compilation);
|
||||
assert_eq!(symbol.is_weak(), false);
|
||||
assert_eq!(symbol.is_undefined(), false);
|
||||
|
||||
symbol = symbols.next().unwrap();
|
||||
println!("{:?}", symbol);
|
||||
let func1_symbol = symbol.index();
|
||||
assert_eq!(symbol.name(), Ok("func1"));
|
||||
assert_eq!(symbol.address(), func1_offset);
|
||||
assert_eq!(symbol.kind(), SymbolKind::Text);
|
||||
assert_eq!(symbol.section_index(), Some(SectionIndex(text_index)));
|
||||
assert_eq!(symbol.scope(), SymbolScope::Linkage);
|
||||
assert_eq!(symbol.is_weak(), false);
|
||||
assert_eq!(symbol.is_undefined(), false);
|
||||
|
||||
let mut relocations = text.relocations();
|
||||
|
||||
let (offset, relocation) = relocations.next().unwrap();
|
||||
println!("{:?}", relocation);
|
||||
assert_eq!(offset, 8);
|
||||
assert_eq!(relocation.kind(), RelocationKind::Absolute);
|
||||
assert_eq!(relocation.encoding(), RelocationEncoding::Generic);
|
||||
assert_eq!(relocation.size(), 64);
|
||||
assert_eq!(
|
||||
relocation.target(),
|
||||
read::RelocationTarget::Symbol(func1_symbol)
|
||||
);
|
||||
assert_eq!(relocation.addend(), 0);
|
||||
}
|
||||
}
|
90
vendor/object/tests/round_trip/section_flags.rs
vendored
Normal file
90
vendor/object/tests/round_trip/section_flags.rs
vendored
Normal file
@ -0,0 +1,90 @@
|
||||
#![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,
|
||||
}
|
||||
);
|
||||
}
|
316
vendor/object/tests/round_trip/tls.rs
vendored
Normal file
316
vendor/object/tests/round_trip/tls.rs
vendored
Normal file
@ -0,0 +1,316 @@
|
||||
#![cfg(all(feature = "read", feature = "write"))]
|
||||
|
||||
use object::read::{Object, ObjectSection, ObjectSymbol};
|
||||
use object::{read, write};
|
||||
use object::{
|
||||
Architecture, BinaryFormat, Endianness, RelocationEncoding, RelocationKind, SectionKind,
|
||||
SymbolFlags, SymbolKind, SymbolScope,
|
||||
};
|
||||
|
||||
#[test]
|
||||
fn coff_x86_64_tls() {
|
||||
let mut object =
|
||||
write::Object::new(BinaryFormat::Coff, Architecture::X86_64, Endianness::Little);
|
||||
|
||||
let section = object.section_id(write::StandardSection::Tls);
|
||||
let symbol = object.add_symbol(write::Symbol {
|
||||
name: b"tls1".to_vec(),
|
||||
value: 0,
|
||||
size: 0,
|
||||
kind: SymbolKind::Tls,
|
||||
scope: SymbolScope::Linkage,
|
||||
weak: false,
|
||||
section: write::SymbolSection::Undefined,
|
||||
flags: SymbolFlags::None,
|
||||
});
|
||||
object.add_symbol_data(symbol, section, &[1; 30], 4);
|
||||
|
||||
let bytes = object.write().unwrap();
|
||||
|
||||
//std::fs::write(&"tls.o", &bytes).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();
|
||||
println!("{:?}", section);
|
||||
let tls_index = section.index();
|
||||
assert_eq!(section.name(), Ok(".tls$"));
|
||||
assert_eq!(section.kind(), SectionKind::Data);
|
||||
assert_eq!(section.size(), 30);
|
||||
assert_eq!(§ion.data().unwrap()[..], &[1; 30]);
|
||||
|
||||
let mut symbols = object.symbols();
|
||||
|
||||
let symbol = symbols.next().unwrap();
|
||||
println!("{:?}", symbol);
|
||||
assert_eq!(symbol.name(), Ok("tls1"));
|
||||
assert_eq!(symbol.kind(), SymbolKind::Data);
|
||||
assert_eq!(symbol.section_index(), Some(tls_index));
|
||||
assert_eq!(symbol.scope(), SymbolScope::Linkage);
|
||||
assert_eq!(symbol.is_weak(), false);
|
||||
assert_eq!(symbol.is_undefined(), false);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn elf_x86_64_tls() {
|
||||
let mut object =
|
||||
write::Object::new(BinaryFormat::Elf, Architecture::X86_64, Endianness::Little);
|
||||
|
||||
let section = object.section_id(write::StandardSection::Tls);
|
||||
let symbol = object.add_symbol(write::Symbol {
|
||||
name: b"tls1".to_vec(),
|
||||
value: 0,
|
||||
size: 0,
|
||||
kind: SymbolKind::Tls,
|
||||
scope: SymbolScope::Linkage,
|
||||
weak: false,
|
||||
section: write::SymbolSection::Undefined,
|
||||
flags: SymbolFlags::None,
|
||||
});
|
||||
object.add_symbol_data(symbol, section, &[1; 30], 4);
|
||||
|
||||
let section = object.section_id(write::StandardSection::UninitializedTls);
|
||||
let symbol = object.add_symbol(write::Symbol {
|
||||
name: b"tls2".to_vec(),
|
||||
value: 0,
|
||||
size: 0,
|
||||
kind: SymbolKind::Tls,
|
||||
scope: SymbolScope::Linkage,
|
||||
weak: false,
|
||||
section: write::SymbolSection::Undefined,
|
||||
flags: SymbolFlags::None,
|
||||
});
|
||||
object.add_symbol_bss(symbol, section, 31, 4);
|
||||
|
||||
let bytes = object.write().unwrap();
|
||||
|
||||
//std::fs::write(&"tls.o", &bytes).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();
|
||||
|
||||
let section = sections.next().unwrap();
|
||||
println!("{:?}", section);
|
||||
assert_eq!(section.name(), Ok(""));
|
||||
|
||||
let section = sections.next().unwrap();
|
||||
println!("{:?}", section);
|
||||
let tdata_index = section.index();
|
||||
assert_eq!(section.name(), Ok(".tdata"));
|
||||
assert_eq!(section.kind(), SectionKind::Tls);
|
||||
assert_eq!(section.size(), 30);
|
||||
assert_eq!(§ion.data().unwrap()[..], &[1; 30]);
|
||||
|
||||
let section = sections.next().unwrap();
|
||||
println!("{:?}", section);
|
||||
let tbss_index = section.index();
|
||||
assert_eq!(section.name(), Ok(".tbss"));
|
||||
assert_eq!(section.kind(), SectionKind::UninitializedTls);
|
||||
assert_eq!(section.size(), 31);
|
||||
assert_eq!(§ion.data().unwrap()[..], &[]);
|
||||
|
||||
let mut symbols = object.symbols();
|
||||
|
||||
let symbol = symbols.next().unwrap();
|
||||
println!("{:?}", symbol);
|
||||
assert_eq!(symbol.name(), Ok(""));
|
||||
|
||||
let symbol = symbols.next().unwrap();
|
||||
println!("{:?}", symbol);
|
||||
assert_eq!(symbol.name(), Ok("tls1"));
|
||||
assert_eq!(symbol.kind(), SymbolKind::Tls);
|
||||
assert_eq!(symbol.section_index(), Some(tdata_index));
|
||||
assert_eq!(symbol.scope(), SymbolScope::Linkage);
|
||||
assert_eq!(symbol.is_weak(), false);
|
||||
assert_eq!(symbol.is_undefined(), false);
|
||||
assert_eq!(symbol.size(), 30);
|
||||
|
||||
let symbol = symbols.next().unwrap();
|
||||
println!("{:?}", symbol);
|
||||
assert_eq!(symbol.name(), Ok("tls2"));
|
||||
assert_eq!(symbol.kind(), SymbolKind::Tls);
|
||||
assert_eq!(symbol.section_index(), Some(tbss_index));
|
||||
assert_eq!(symbol.scope(), SymbolScope::Linkage);
|
||||
assert_eq!(symbol.is_weak(), false);
|
||||
assert_eq!(symbol.is_undefined(), false);
|
||||
assert_eq!(symbol.size(), 31);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn macho_x86_64_tls() {
|
||||
let mut object = write::Object::new(
|
||||
BinaryFormat::MachO,
|
||||
Architecture::X86_64,
|
||||
Endianness::Little,
|
||||
);
|
||||
|
||||
let section = object.section_id(write::StandardSection::Tls);
|
||||
let symbol = object.add_symbol(write::Symbol {
|
||||
name: b"tls1".to_vec(),
|
||||
value: 0,
|
||||
size: 0,
|
||||
kind: SymbolKind::Tls,
|
||||
scope: SymbolScope::Linkage,
|
||||
weak: false,
|
||||
section: write::SymbolSection::Undefined,
|
||||
flags: SymbolFlags::None,
|
||||
});
|
||||
object.add_symbol_data(symbol, section, &[1; 30], 4);
|
||||
|
||||
let section = object.section_id(write::StandardSection::UninitializedTls);
|
||||
let symbol = object.add_symbol(write::Symbol {
|
||||
name: b"tls2".to_vec(),
|
||||
value: 0,
|
||||
size: 0,
|
||||
kind: SymbolKind::Tls,
|
||||
scope: SymbolScope::Linkage,
|
||||
weak: false,
|
||||
section: write::SymbolSection::Undefined,
|
||||
flags: SymbolFlags::None,
|
||||
});
|
||||
object.add_symbol_bss(symbol, section, 31, 4);
|
||||
|
||||
let bytes = object.write().unwrap();
|
||||
|
||||
//std::fs::write(&"tls.o", &bytes).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 thread_data = sections.next().unwrap();
|
||||
println!("{:?}", thread_data);
|
||||
let thread_data_index = thread_data.index();
|
||||
assert_eq!(thread_data.name(), Ok("__thread_data"));
|
||||
assert_eq!(thread_data.segment_name(), Ok(Some("__DATA")));
|
||||
assert_eq!(thread_data.kind(), SectionKind::Tls);
|
||||
assert_eq!(thread_data.size(), 30);
|
||||
assert_eq!(&thread_data.data().unwrap()[..], &[1; 30]);
|
||||
|
||||
let thread_vars = sections.next().unwrap();
|
||||
println!("{:?}", thread_vars);
|
||||
let thread_vars_index = thread_vars.index();
|
||||
assert_eq!(thread_vars.name(), Ok("__thread_vars"));
|
||||
assert_eq!(thread_vars.segment_name(), Ok(Some("__DATA")));
|
||||
assert_eq!(thread_vars.kind(), SectionKind::TlsVariables);
|
||||
assert_eq!(thread_vars.size(), 2 * 3 * 8);
|
||||
assert_eq!(&thread_vars.data().unwrap()[..], &[0; 48][..]);
|
||||
|
||||
let thread_bss = sections.next().unwrap();
|
||||
println!("{:?}", thread_bss);
|
||||
let thread_bss_index = thread_bss.index();
|
||||
assert_eq!(thread_bss.name(), Ok("__thread_bss"));
|
||||
assert_eq!(thread_bss.segment_name(), Ok(Some("__DATA")));
|
||||
assert_eq!(thread_bss.kind(), SectionKind::UninitializedTls);
|
||||
assert_eq!(thread_bss.size(), 31);
|
||||
assert_eq!(thread_bss.data(), Ok(&[][..]));
|
||||
|
||||
let mut symbols = object.symbols();
|
||||
|
||||
let symbol = symbols.next().unwrap();
|
||||
println!("{:?}", symbol);
|
||||
let tls1_init_symbol = symbol.index();
|
||||
assert_eq!(symbol.name(), Ok("_tls1$tlv$init"));
|
||||
assert_eq!(symbol.kind(), SymbolKind::Tls);
|
||||
assert_eq!(symbol.section_index(), Some(thread_data_index));
|
||||
assert_eq!(symbol.scope(), SymbolScope::Compilation);
|
||||
assert_eq!(symbol.is_weak(), false);
|
||||
assert_eq!(symbol.is_undefined(), false);
|
||||
|
||||
let symbol = symbols.next().unwrap();
|
||||
println!("{:?}", symbol);
|
||||
let tls2_init_symbol = symbol.index();
|
||||
assert_eq!(symbol.name(), Ok("_tls2$tlv$init"));
|
||||
assert_eq!(symbol.kind(), SymbolKind::Tls);
|
||||
assert_eq!(symbol.section_index(), Some(thread_bss_index));
|
||||
assert_eq!(symbol.scope(), SymbolScope::Compilation);
|
||||
assert_eq!(symbol.is_weak(), false);
|
||||
assert_eq!(symbol.is_undefined(), false);
|
||||
|
||||
let symbol = symbols.next().unwrap();
|
||||
println!("{:?}", symbol);
|
||||
assert_eq!(symbol.name(), Ok("_tls1"));
|
||||
assert_eq!(symbol.kind(), SymbolKind::Tls);
|
||||
assert_eq!(symbol.section_index(), Some(thread_vars_index));
|
||||
assert_eq!(symbol.scope(), SymbolScope::Linkage);
|
||||
assert_eq!(symbol.is_weak(), false);
|
||||
assert_eq!(symbol.is_undefined(), false);
|
||||
|
||||
let symbol = symbols.next().unwrap();
|
||||
println!("{:?}", symbol);
|
||||
assert_eq!(symbol.name(), Ok("_tls2"));
|
||||
assert_eq!(symbol.kind(), SymbolKind::Tls);
|
||||
assert_eq!(symbol.section_index(), Some(thread_vars_index));
|
||||
assert_eq!(symbol.scope(), SymbolScope::Linkage);
|
||||
assert_eq!(symbol.is_weak(), false);
|
||||
assert_eq!(symbol.is_undefined(), false);
|
||||
|
||||
let symbol = symbols.next().unwrap();
|
||||
println!("{:?}", symbol);
|
||||
let tlv_bootstrap_symbol = symbol.index();
|
||||
assert_eq!(symbol.name(), Ok("__tlv_bootstrap"));
|
||||
assert_eq!(symbol.kind(), SymbolKind::Unknown);
|
||||
assert_eq!(symbol.section_index(), None);
|
||||
assert_eq!(symbol.scope(), SymbolScope::Unknown);
|
||||
assert_eq!(symbol.is_weak(), false);
|
||||
assert_eq!(symbol.is_undefined(), true);
|
||||
|
||||
let mut relocations = thread_vars.relocations();
|
||||
|
||||
let (offset, relocation) = relocations.next().unwrap();
|
||||
println!("{:?}", relocation);
|
||||
assert_eq!(offset, 0);
|
||||
assert_eq!(relocation.kind(), RelocationKind::Absolute);
|
||||
assert_eq!(relocation.encoding(), RelocationEncoding::Generic);
|
||||
assert_eq!(relocation.size(), 64);
|
||||
assert_eq!(
|
||||
relocation.target(),
|
||||
read::RelocationTarget::Symbol(tlv_bootstrap_symbol)
|
||||
);
|
||||
assert_eq!(relocation.addend(), 0);
|
||||
|
||||
let (offset, relocation) = relocations.next().unwrap();
|
||||
println!("{:?}", relocation);
|
||||
assert_eq!(offset, 16);
|
||||
assert_eq!(relocation.kind(), RelocationKind::Absolute);
|
||||
assert_eq!(relocation.encoding(), RelocationEncoding::Generic);
|
||||
assert_eq!(relocation.size(), 64);
|
||||
assert_eq!(
|
||||
relocation.target(),
|
||||
read::RelocationTarget::Symbol(tls1_init_symbol)
|
||||
);
|
||||
assert_eq!(relocation.addend(), 0);
|
||||
|
||||
let (offset, relocation) = relocations.next().unwrap();
|
||||
println!("{:?}", relocation);
|
||||
assert_eq!(offset, 24);
|
||||
assert_eq!(relocation.kind(), RelocationKind::Absolute);
|
||||
assert_eq!(relocation.encoding(), RelocationEncoding::Generic);
|
||||
assert_eq!(relocation.size(), 64);
|
||||
assert_eq!(
|
||||
relocation.target(),
|
||||
read::RelocationTarget::Symbol(tlv_bootstrap_symbol)
|
||||
);
|
||||
assert_eq!(relocation.addend(), 0);
|
||||
|
||||
let (offset, relocation) = relocations.next().unwrap();
|
||||
println!("{:?}", relocation);
|
||||
assert_eq!(offset, 40);
|
||||
assert_eq!(relocation.kind(), RelocationKind::Absolute);
|
||||
assert_eq!(relocation.encoding(), RelocationEncoding::Generic);
|
||||
assert_eq!(relocation.size(), 64);
|
||||
assert_eq!(
|
||||
relocation.target(),
|
||||
read::RelocationTarget::Symbol(tls2_init_symbol)
|
||||
);
|
||||
assert_eq!(relocation.addend(), 0);
|
||||
}
|
Reference in New Issue
Block a user