wxtla 0.3.1

Wired eXploring Target Layer Accessor
Documentation
//! MBR volume-system driver and probe registration.

mod boot_record;
mod constants;
mod driver;
mod entry;
mod parser;
mod system;
mod validation;

pub use driver::MbrDriver;
pub use entry::{MbrPartitionEntry, MbrPartitionInfo, MbrPartitionOrigin};
pub use system::MbrVolumeSystem;

use crate::{
  FormatDescriptor, FormatKind, FormatProbe, ProbeConfidence, ProbeContext, ProbeMatch,
  ProbeRegistry, ProbeResult, Result,
};

/// MBR volume-system descriptor.
pub const DESCRIPTOR: FormatDescriptor =
  FormatDescriptor::new("volume.mbr", FormatKind::VolumeSystem);

inventory::submit! {
  crate::formats::FormatInventoryEntry::new(DESCRIPTOR, register_probes)
}

fn register_probes(registry: &mut ProbeRegistry) {
  registry.register(MbrProbe);
}

struct MbrProbe;

impl FormatProbe for MbrProbe {
  fn descriptor(&self) -> FormatDescriptor {
    DESCRIPTOR
  }

  fn probe(&self, context: &ProbeContext<'_>) -> Result<ProbeResult> {
    let Ok(signature) = context.read_bytes_at(510, 2) else {
      return Ok(ProbeResult::rejected());
    };

    if signature == constants::BOOT_SIGNATURE {
      Ok(ProbeResult::matched(ProbeMatch::new(
        DESCRIPTOR,
        ProbeConfidence::Weak,
        "mbr boot signature found",
      )))
    } else {
      Ok(ProbeResult::rejected())
    }
  }
}