|
| 1 | +// This Source Code Form is subject to the terms of the Mozilla Public |
| 2 | +// License, v. 2.0. If a copy of the MPL was not distributed with this |
| 3 | +// file, You can obtain one at https://mozilla.org/MPL/2.0/. |
| 4 | + |
| 5 | +use object::{write, Object, ObjectSection, RelocationTarget, SectionKind, SymbolKind}; |
| 6 | +use slog::{info, warn}; |
| 7 | +use std::collections::HashMap; |
| 8 | +use std::error::Error; |
| 9 | +use std::fmt; |
| 10 | + |
| 11 | +#[derive(Debug, Clone)] |
| 12 | +pub struct NoRewriteError; |
| 13 | + |
| 14 | +impl fmt::Display for NoRewriteError { |
| 15 | + fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result { |
| 16 | + write!(f, "no object rewriting was performed") |
| 17 | + } |
| 18 | +} |
| 19 | + |
| 20 | +impl Error for NoRewriteError { |
| 21 | + fn source(&self) -> Option<&(dyn Error + 'static)> { |
| 22 | + // Generic error, underlying cause isn't tracked. |
| 23 | + None |
| 24 | + } |
| 25 | +} |
| 26 | + |
| 27 | +/// Rename object syn PyInit_foo to PyInit_<full_name> to avoid clashes |
| 28 | +pub fn rename_init( |
| 29 | + logger: &slog::Logger, |
| 30 | + name: &String, |
| 31 | + object_data: &[u8], |
| 32 | +) -> Result<Vec<u8>, NoRewriteError> { |
| 33 | + let mut rewritten = false; |
| 34 | + |
| 35 | + let name_prefix = name.split('.').next().unwrap(); |
| 36 | + |
| 37 | + let in_object = match object::File::parse(object_data) { |
| 38 | + Ok(object) => object, |
| 39 | + Err(err) => { |
| 40 | + warn!(logger, "Failed to parse compiled object for {}: {}", name, err); |
| 41 | + return Err(NoRewriteError); |
| 42 | + } |
| 43 | + }; |
| 44 | + |
| 45 | + let mut out_object = write::Object::new(in_object.format(), in_object.architecture()); |
| 46 | + |
| 47 | + let mut out_sections = HashMap::new(); |
| 48 | + for in_section in in_object.sections() { |
| 49 | + if in_section.kind() == SectionKind::Metadata { |
| 50 | + continue; |
| 51 | + } |
| 52 | + let section_id = out_object.add_section( |
| 53 | + in_section.segment_name().unwrap_or("").as_bytes().to_vec(), |
| 54 | + in_section.name().unwrap_or("").as_bytes().to_vec(), |
| 55 | + in_section.kind(), |
| 56 | + ); |
| 57 | + let out_section = out_object.section_mut(section_id); |
| 58 | + if out_section.is_bss() { |
| 59 | + out_section.append_bss(in_section.size(), in_section.align()); |
| 60 | + } else { |
| 61 | + out_section.set_data(in_section.uncompressed_data().into(), in_section.align()); |
| 62 | + } |
| 63 | + out_sections.insert(in_section.index(), section_id); |
| 64 | + } |
| 65 | + |
| 66 | + let mut out_symbols = HashMap::new(); |
| 67 | + for (symbol_index, in_symbol) in in_object.symbols() { |
| 68 | + if in_symbol.kind() == SymbolKind::Null { |
| 69 | + continue; |
| 70 | + } |
| 71 | + let (section, value) = match in_symbol.section_index() { |
| 72 | + Some(index) => ( |
| 73 | + Some(*out_sections.get(&index).unwrap()), |
| 74 | + in_symbol.address() - in_object.section_by_index(index).unwrap().address(), |
| 75 | + ), |
| 76 | + None => (None, in_symbol.address()), |
| 77 | + }; |
| 78 | + let in_sym_name = in_symbol.name().unwrap_or(""); |
| 79 | + let sym_name = if in_sym_name.contains("PyInit_") && !in_sym_name.contains(name_prefix) { |
| 80 | + match out_object.mangling.global_prefix() { |
| 81 | + Some(prefix) => format!("{}PyInit_{}", prefix as char, name.replace(".", "_")), |
| 82 | + None => format!("PyInit_{}", name.replace(".", "_")) |
| 83 | + } |
| 84 | + } else { |
| 85 | + String::from(in_sym_name) |
| 86 | + }; |
| 87 | + if sym_name != in_sym_name { |
| 88 | + warn!( |
| 89 | + logger, |
| 90 | + "rewrote object symbol name {} to {}", in_sym_name, sym_name, |
| 91 | + ); |
| 92 | + |
| 93 | + rewritten = true; |
| 94 | + } |
| 95 | + |
| 96 | + let out_symbol = write::Symbol { |
| 97 | + name: sym_name.as_bytes().to_vec(), |
| 98 | + value, |
| 99 | + size: in_symbol.size(), |
| 100 | + kind: in_symbol.kind(), |
| 101 | + scope: in_symbol.scope(), |
| 102 | + weak: in_symbol.is_weak(), |
| 103 | + section, |
| 104 | + }; |
| 105 | + let symbol_id = out_object.add_symbol(out_symbol); |
| 106 | + out_symbols.insert(symbol_index, symbol_id); |
| 107 | + } |
| 108 | + |
| 109 | + if !rewritten { |
| 110 | + info!(logger, "no symbol name rewriting occurred for {}", name,); |
| 111 | + return Err(NoRewriteError); |
| 112 | + } |
| 113 | + |
| 114 | + for in_section in in_object.sections() { |
| 115 | + if in_section.kind() == SectionKind::Metadata { |
| 116 | + continue; |
| 117 | + } |
| 118 | + let out_section = *out_sections.get(&in_section.index()).unwrap(); |
| 119 | + for (offset, in_relocation) in in_section.relocations() { |
| 120 | + let symbol = match in_relocation.target() { |
| 121 | + RelocationTarget::Symbol(symbol) => *out_symbols.get(&symbol).unwrap(), |
| 122 | + RelocationTarget::Section(section) => { |
| 123 | + out_object.section_symbol(*out_sections.get(§ion).unwrap()) |
| 124 | + } |
| 125 | + }; |
| 126 | + let out_relocation = write::Relocation { |
| 127 | + offset, |
| 128 | + size: in_relocation.size(), |
| 129 | + kind: in_relocation.kind(), |
| 130 | + encoding: in_relocation.encoding(), |
| 131 | + symbol, |
| 132 | + addend: in_relocation.addend(), |
| 133 | + }; |
| 134 | + out_object |
| 135 | + .add_relocation(out_section, out_relocation) |
| 136 | + .unwrap(); |
| 137 | + } |
| 138 | + } |
| 139 | + |
| 140 | + Ok(out_object.write().unwrap()) |
| 141 | +} |
0 commit comments