1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37

// Copyright lowRISC contributors (OpenTitan project).
// Licensed under the Apache License, Version 2.0, see LICENSE for details.
// SPDX-License-Identifier: Apache-2.0

use anyhow::{Context, Result};
use object::{Object, ObjectSection, ObjectSymbol};

/// Load a symbol's data from an object file.
pub fn symbol_data(object: &object::read::File, name: &str) -> Result<Vec<u8>> {
    let mut symbols = object.symbols();
    let symbol = symbols
        .find(|symbol| symbol.name() == Ok(name))
        .with_context(|| format!("could not find symbol {name} in ELF"))?;

    let section_index = symbol
        .section_index()
        .with_context(|| format!("symbol {name} was did not have a section index"))?;
    let section = object
        .section_by_index(section_index)
        .with_context(|| format!("could not find section containing {name}"))?;

    let offset = (symbol.address() - section.address()) as usize;
    let data = section.uncompressed_data()?;
    let data = &data[offset..][..symbol.size() as usize];

    Ok(data.to_vec())
}

/// Get a symbol's address from an object file.
pub fn symbol_addr(object: &object::read::File, name: &str) -> Result<u32> {
    let addr = object
        .symbols()
        .find(|symbol| symbol.name() == Ok(name))
        .with_context(|| format!("failed to find {name} symbol"))?
        .address();
    Ok(addr as u32)
}