Add GNU Property Note

1 files changed, 15 insertions, 11 deletions

diff --git a/compiler/rustc_codegen_ssa/src/back/metadata.rs b/compiler/rustc_codegen_ssa/src/back/metadata.rs
index baf674cf9eb..d4515a794ee 100644
--- a/compiler/rustc_codegen_ssa/src/back/metadata.rs
+++ b/compiler/rustc_codegen_ssa/src/back/metadata.rs
@@ -98,6 +98,7 @@ fn add_gnu_property_note(
     file: &mut write::Object<'static>,
     architecture: Architecture,
     binary_format: BinaryFormat,
+    endianness: Endianness,
 ) {
     // check bti protection
     if binary_format != BinaryFormat::Elf
@@ -115,12 +116,12 @@ fn add_gnu_property_note(
     let n_namsz: u32 = 4; // Size of the n_name field
     let n_descsz: u32 = 16; // Size of the n_desc field
     let n_type: u32 = NT_GNU_PROPERTY_TYPE_0; // Type of note descriptor
-    let values = [n_namsz, n_descsz, n_type];
-    values.map(|v| data.extend_from_slice(&(v.to_le_bytes())));
-    data.push(b'G'); // Owner of the program property note
-    data.push(b'N');
-    data.push(b'U');
-    data.push(0);
+    let header_values = [n_namsz, n_descsz, n_type];
+    match endianness {
+        Endianness::Little => header_values.map(|v| data.extend_from_slice(&(v.to_le_bytes()))),
+        Endianness::Big => header_values.map(|v| data.extend_from_slice(&(v.to_be_bytes()))),
+    };
+    data.extend_from_slice(b"GNU\0"); // Owner of the program property note
     let pr_type: u32 = match architecture {
         Architecture::X86_64 => 0xc0000002,
         Architecture::Aarch64 => 0xc0000000,
@@ -128,10 +129,13 @@ fn add_gnu_property_note(
     };
     let pr_datasz: u32 = 4; //size of the pr_data field
     let pr_data: u32 = 3; //program property descriptor
-    let pr_padding: u32 = 3;
-    let values = [pr_type, pr_datasz, pr_data, pr_padding];
-    values.map(|v| data.extend_from_slice(&(v.to_le_bytes())));
-    file.append_section_data(section, &data, 4);
+    let pr_padding: u32 = 0;
+    let property_values = [pr_type, pr_datasz, pr_data, pr_padding];
+    match endianness {
+        Endianness::Little => property_values.map(|v| data.extend_from_slice(&(v.to_le_bytes()))),
+        Endianness::Big => property_values.map(|v| data.extend_from_slice(&(v.to_be_bytes()))),
+    };
+    file.append_section_data(section, &data, 8);
 }
 
 pub(crate) fn create_object_file(sess: &Session) -> Option<write::Object<'static>> {
@@ -246,7 +250,7 @@ pub(crate) fn create_object_file(sess: &Session) -> Option<write::Object<'static
         _ => elf::ELFOSABI_NONE,
     };
     let abi_version = 0;
-    add_gnu_property_note(&mut file, architecture, binary_format);
+    add_gnu_property_note(&mut file, architecture, binary_format, endianness);
     file.flags = FileFlags::Elf { os_abi, abi_version, e_flags };
     Some(file)
 }