cdefmt-decoder 0.8.0

//! Contains logic related to finding logs in the elf and parsing them.

use std::collections::HashMap;

use cdefmt_parser::{
    metadata::Metadata,
    r#type::{self, Type},
    Parser,
};
use gimli::Reader;
use object::ReadRef;

use crate::{log::Log, var::Var, Error, Result};

/// Responsible for parsing logs from the elf.
pub struct Decoder<'elf> {
    parser: Parser<'elf>,
    log_cache: HashMap<usize, (Metadata<'elf>, Option<Type>)>,
}

impl<'elf> Decoder<'elf> {
    /// Creates a new Parser from elf data.
    pub fn new<R: ReadRef<'elf>>(data: R) -> Result<Self> {
        Ok(Decoder {
            parser: Parser::new(data)?,
            log_cache: Default::default(),
        })
    }

    /// Decodes a raw log
    pub fn decode_log(&mut self, data: &[u8]) -> Result<Log<'elf>> {
        let mut data = gimli::EndianSlice::new(data, self.parser.endian());
        let id = data.read_address(self.parser.address_size().bytes())? as usize;

        if let std::collections::hash_map::Entry::Vacant(e) = self.log_cache.entry(id) {
            // Parse log metadata and type if we don't have it cached.
            let metadata = self.parser.get_log_metadata(id)?;
            let ty = self.parser.get_log_args_type(&metadata)?;
            e.insert((metadata, ty));
        };

        // Unwrap safety: made sure that the entry exists right above here.
        let (metadata, ty) = self.log_cache.get(&id).unwrap();

        let args = if let Some(ty) = ty {
            Self::decode_log_args(ty, data)?
        } else {
            vec![]
        };

        let log = Log::new(metadata.clone(), args);

        if id == 0 {
            self.validate_init(&log)?
        }

        Ok(log)
    }

    /// Parses all logs in the elf and pre-caches their metadata and type
    /// information, to speed up future log decoding.
    /// This can also serve as a way to validate that all the logs encoded into
    /// the file are valid and can be properly parsed.
    pub fn precache_log_metadata(&mut self) -> Result<usize> {
        self.log_cache = self
            .parser
            .iter_logs()
            .map(|l| {
                let (metadata, ty) = l?;
                Ok((metadata.id, (metadata, ty)))
            })
            .collect::<Result<_>>()?;

        Ok(self.log_cache.len())
    }

    pub fn get_endianness(&self) -> gimli::RunTimeEndian {
        self.parser.endian()
    }

    // Parses the log's arguments.
    fn decode_log_args<R: Reader>(ty: &Type, mut data: R) -> Result<Vec<Var>> {
        let members = if let Type::Structure { members, .. } = ty {
            members
        } else {
            return Err(Error::Custom("The log's args aren't a structure!").into());
        };

        // We already read the log_id from the data, skip it.
        let members = &members[1..];

        // Parse the raw data into `Var` representation.
        let mut decoded = members
            .iter()
            // Filter out the 'dynamic_data' member for now - this contains variable-length data
            // that needs to be processed after parsing the fixed-size fields, since we need
            // information from those fields to know how to interpret the dynamic data.
            // The dynamic_data field is positioned at the end of the structure when present,
            // but not all logs have it, so we filter by name rather than skipping the last element.
            .filter(|m| !matches!(m.name.as_str(), "dynamic_data"))
            .map(|m| Ok(Var::parse(&m.ty, &mut data)?.0))
            .collect::<Result<Vec<_>>>()?;

        // Decode dynamic members
        for (i, member) in members.iter().enumerate() {
            match member.name.as_str() {
                n if n.contains("dynamic_array") => {
                    decoded[i] = Self::decode_dynamic_array(member, &decoded[i], &mut data)?
                }
                _ => continue,
            }
        }

        Ok(decoded)
    }

    fn validate_init(&self, log: &Log) -> Result<()> {
        let args = log.get_args();

        if args.is_empty() {
            return Err(Error::Custom("No build ID argument information!").into());
        }

        if let Var::Array(build_id) = args.first().unwrap() {
            let build_id = build_id
                .iter()
                .map(|b| match b {
                    Var::U8(b) => Ok(*b),
                    _ => Err(Error::Custom("Build ID data contains non u8 element!").into()),
                })
                .collect::<Result<Vec<_>>>()?;
            if self.parser.build_id() != build_id {
                Err(Error::Custom("Build ID mismatch!").into())
            } else {
                Ok(())
            }
        } else {
            Err(Error::Custom("Build ID missing or not an array").into())
        }
    }

    fn decode_dynamic_array<R: Reader>(
        metadata: &r#type::StructureMember,
        value: &Var,
        data: &mut R,
    ) -> Result<Var> {
        // The dynamic_array is structured as:
        // [0] size
        // [1] type

        // Extract size from value that was previously decoded
        let size = match value {
            Var::Structure { members } => &members[0].value,
            _ => return Err(Error::Custom("Dynamic array metadata is not a struct!").into()),
        }
        .as_u64();

        // Extract type from metadata
        let arr_ty = match &metadata.ty {
            Type::Structure { members, .. } => &members[1].ty,
            _ => return Err(Error::Custom("Dynamic array metadata is not a struct!").into()),
        };

        let ty = match arr_ty {
            Type::Array { ty, .. } => ty,
            _ => {
                return Err(Error::Custom(
                    "Dynamic array type metadata is not an array!",
                ).into())
            }
        };

        let dyn_ty = Type::Array {
            ty: ty.clone(),
            lengths: vec![size / ty.size() as u64],
        };

        Ok(Var::parse(&dyn_ty, data)?.0)
    }
}