[SOL] Temporarily patch BPF ELF file header with SBF machine code

Author: dmakarov

compiler/rustc_codegen_ssa/src/back/link.rs

@@ -46,7 +46,7 @@ use itertools::Itertools;
 use std::collections::BTreeSet;
 use std::ffi::{OsStr, OsString};
 use std::fs::{read, File, OpenOptions};
-use std::io::{BufWriter, Write};
+use std::io::{prelude::*, BufWriter, SeekFrom, Write};
 use std::ops::Deref;
 use std::path::{Path, PathBuf};
 use std::process::{ExitStatus, Output, Stdio};
@@ -311,6 +311,9 @@ fn link_rlib<'a>(
                 codegen_results.metadata.raw_data(),
             );
             let metadata = emit_wrapper_file(sess, &metadata, tmpdir, METADATA_FILENAME);
+            if sess.target.arch == "sbf" {
+                patch_synthetic_object_file(sess, &metadata);
+            }
             match metadata_position {
                 MetadataPosition::First => {
                     // Most of the time metadata in rlib files is wrapped in a "dummy" object
@@ -2013,9 +2016,23 @@ fn add_linked_symbol_object(
     if let Err(error) = result {
         sess.dcx().emit_fatal(errors::FailedToWrite { path, error });
     }
+    if sess.target.arch == "sbf" {
+        patch_synthetic_object_file(sess, &path);
+    }
     cmd.add_object(&path);
 }
 
+fn patch_synthetic_object_file(sess: &Session, path: &PathBuf) {
+    const EM_SBF: [u8; 2] = [0x07, 0x01];
+    if let Ok(mut sf) = fs::OpenOptions::new().write(true).open(path) {
+        if let Ok(_) = sf.seek(SeekFrom::Start(0x12)) {
+            sf.write(&EM_SBF).unwrap();
+        }
+    } else {
+        sess.fatal(&format!("failed to patch {}", path.display()));
+    }
+}
+
 /// Add object files containing code from the current crate.
 fn add_local_crate_regular_objects(cmd: &mut dyn Linker, codegen_results: &CodegenResults) {
     for obj in codegen_results.modules.iter().filter_map(|m| m.object.as_ref()) {