language-servers/roc
1
0
Fork 0
mirror of https://github.com/roc-lang/roc.git synced 2025-09-27 13:59:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

refactor section header writing

Browse source
This commit is contained in:
Folkert 2022-09-21 12:32:43 +02:00
parent 20cd50f712
commit 8f5b005f10
No known key found for this signature in database
GPG key ID: 1F17F6FFD112B97C
1 changed files with 12 additions and 40 deletions
Download patch file Download diff file Expand all files Collapse all files

52
crates/linker/src/pe.rs
View file

@ -676,53 +676,21 @@ fn update_optional_header(
} }
#[allow(dead_code)] #[allow(dead_code)]
fn write_app_text_section_header( #[allow(clippy::too_many_arguments)]
fn write_section_header(
data: &mut [u8], data: &mut [u8],
name: [u8; 8],
characteristics: u32,
section_header_start: usize, section_header_start: usize,
section_file_offset: usize, section_file_offset: usize,
virtual_size: u32, virtual_size: u32,
virtual_address: u32, virtual_address: u32,
size_of_raw_data: u32, size_of_raw_data: u32,
) { ) {
use object::{pe, U32}; use object::U32;
let characteristics =
pe::IMAGE_SCN_MEM_READ | pe::IMAGE_SCN_CNT_CODE | pe::IMAGE_SCN_MEM_EXECUTE;
let header = ImageSectionHeader { let header = ImageSectionHeader {
name: *b".text1\0\0", // unclear whether we can/should call this text name,
virtual_size: U32::new(LE, virtual_size),
virtual_address: U32::new(LE, virtual_address),
size_of_raw_data: U32::new(LE, size_of_raw_data),
pointer_to_raw_data: U32::new(LE, section_file_offset as u32),
pointer_to_relocations: Default::default(),
pointer_to_linenumbers: Default::default(),
number_of_relocations: Default::default(),
number_of_linenumbers: Default::default(),
characteristics: U32::new(LE, characteristics),
};
let header_array: [u8; std::mem::size_of::<ImageSectionHeader>()] =
unsafe { std::mem::transmute(header) };
data[section_header_start..][..header_array.len()].copy_from_slice(&header_array);
}
#[allow(dead_code)]
fn write_app_rdata_section_header(
data: &mut [u8],
section_header_start: usize,
section_file_offset: usize,
virtual_size: u32,
virtual_address: u32,
size_of_raw_data: u32,
) {
use object::{pe, U32};
let characteristics = pe::IMAGE_SCN_MEM_READ | pe::IMAGE_SCN_CNT_INITIALIZED_DATA;
let header = ImageSectionHeader {
name: *b".rdata1\0",
virtual_size: U32::new(LE, virtual_size), virtual_size: U32::new(LE, virtual_size),
virtual_address: U32::new(LE, virtual_address), virtual_address: U32::new(LE, virtual_address),
size_of_raw_data: U32::new(LE, size_of_raw_data), size_of_raw_data: U32::new(LE, size_of_raw_data),
@ -1175,8 +1143,10 @@ mod test {
SectionKind::Text => { SectionKind::Text => {
code_bytes_added += size_of_raw_data; code_bytes_added += size_of_raw_data;
write_app_text_section_header( write_section_header(
&mut app, &mut app,
*b".text1\0\0",
pe::IMAGE_SCN_MEM_READ | pe::IMAGE_SCN_CNT_CODE | pe::IMAGE_SCN_MEM_EXECUTE,
section_header_start, section_header_start,
section_file_offset, section_file_offset,
virtual_size, virtual_size,
@ -1187,8 +1157,10 @@ mod test {
SectionKind::ReadOnlyData => { SectionKind::ReadOnlyData => {
data_bytes_added += size_of_raw_data; data_bytes_added += size_of_raw_data;
write_app_rdata_section_header( write_section_header(
&mut app, &mut app,
*b".rdata1\0",
pe::IMAGE_SCN_MEM_READ | pe::IMAGE_SCN_CNT_INITIALIZED_DATA,
section_header_start, section_header_start,
section_file_offset, section_file_offset,
virtual_size, virtual_size,