smda 0.2.5

use crate::{
    error::Error,
    label_providers::{
        elf_api_resolver::ElfApiResolver, elf_symbol_provider::ElfSymbolProvider,
        pdb_symbol_provider::PdbSymbolProvider, win_api_resolver::WinApiResolver,
    },
    BinaryInfo, Result,
};

#[derive(Debug)]
pub enum LabelProvider {
    WinApi(WinApiResolver),
    ElfApi(ElfApiResolver),
    ElfSymbol(ElfSymbolProvider),
    PdbSymbol(PdbSymbolProvider),
}

impl LabelProvider {
    pub fn get_api(
        &self,
        to_addr: u64,
        absolute_addr: u64,
    ) -> Result<(Option<String>, Option<String>)> {
        match self {
            LabelProvider::WinApi(w) => w.get_api(to_addr, absolute_addr),
            LabelProvider::ElfApi(w) => w.get_api(to_addr, absolute_addr),
            _ => Err(Error::InvalidRule(line!(), file!().to_string())),
        }
    }

    pub fn is_api_provider(&self) -> Result<bool> {
        match self {
            LabelProvider::WinApi(_) => Ok(true),
            LabelProvider::ElfApi(_) => Ok(true),
            _ => Ok(false),
        }
    }

    pub fn is_symbol_provider(&self) -> Result<bool> {
        match self {
            LabelProvider::PdbSymbol(_) => Ok(true),
            LabelProvider::ElfSymbol(_) => Ok(true),
            _ => Ok(false),
        }
    }

    pub fn get_functions_symbols(&self) -> Result<&std::collections::HashMap<u64, String>> {
        match self {
            LabelProvider::PdbSymbol(s) => s.get_functions_symbols(),
            LabelProvider::ElfSymbol(s) => s.get_functions_symbols(),
            _ => Err(Error::InvalidRule(line!(), file!().to_string())),
        }
    }

    pub fn get_symbol(&self, _address: u64) -> Result<String> {
        Err(Error::NotImplementedError)
    }

    pub fn update(&mut self, bin: &BinaryInfo) -> Result<()> {
        match self {
            LabelProvider::WinApi(w) => w.update(bin),
            LabelProvider::ElfSymbol(w) => w.update(bin),
            LabelProvider::ElfApi(w) => w.update(bin),
            LabelProvider::PdbSymbol(w) => w.update(bin),
        }
    }
}