// Clippy doesn't like the lifetimes, but rustc wants them. 🤷
#![allow(clippy::needless_lifetimes)]
// Clippy doesn't understand that we exit with ? from the closure in
// and_then below and this can't be expressed with map.
#![allow(clippy::bind_instead_of_map)]

use alloc::{borrow::ToOwned, string::String};
use goblin::pe::section_table::SectionTable;

/// Extracts the data of a section in a loaded PE file
/// based on the section table.
pub fn pe_section_data<'a>(pe_data: &'a [u8], section: &SectionTable) -> Option<&'a [u8]> {
    let section_start: usize = section.virtual_address.try_into().ok()?;

    assert!(section.virtual_size <= section.size_of_raw_data);
    let section_end: usize = section_start + usize::try_from(section.virtual_size).ok()?;

    Some(&pe_data[section_start..section_end])
}

/// Extracts the data of a section of a loaded PE file
/// based on the section name.
pub fn pe_section<'a>(pe_data: &'a [u8], section_name: &str) -> Option<&'a [u8]> {
    let pe_binary = goblin::pe::PE::parse(pe_data).ok()?;

    pe_binary
        .sections
        .iter()
        .find(|s| s.name().map(|n| n == section_name).unwrap_or(false))
        .and_then(|s| pe_section_data(pe_data, s))
}

/// Extracts the data of a section of a loaded PE image and returns it as a string.
pub fn pe_section_as_string<'a>(pe_data: &'a [u8], section_name: &str) -> Option<String> {
    pe_section(pe_data, section_name).map(|data| core::str::from_utf8(data).unwrap().to_owned())
}