use crate::{
architecture::arm::{
ap::valid_access_ports_allowlist,
dp::{
Abort, Ctrl, DebugPortError, DebugPortId, DebugPortVersion, DpAccess, Select, BASEPTR0,
BASEPTR1, DPIDR, DPIDR1,
},
memory::{adi_v5_memory_interface::ADIMemoryInterface, ArmMemoryInterface, Component},
sequences::{ArmDebugSequence, DefaultArmSequence},
ArmError, DapAccess, DpAddress, FullyQualifiedApAddress, PortType, RawDapAccess, SwoAccess,
SwoConfig,
},
probe::{DebugProbe, DebugProbeError, Probe},
CoreStatus, Error,
};
use jep106::JEP106Code;
use std::{
collections::{hash_map, BTreeSet, HashMap},
fmt::Debug,
sync::Arc,
time::Duration,
};
#[derive(Debug, thiserror::Error, Clone, PartialEq, Eq, Copy)]
pub enum DapError {
#[error("An error occurred in the SWD communication between probe and device.")]
SwdProtocol,
#[error("Target device did not respond to request.")]
NoAcknowledge,
#[error("Target device responded with a FAULT response to the request.")]
FaultResponse,
#[error("Target device responded with a WAIT response to the request.")]
WaitResponse,
#[error("Incorrect parity on READ request.")]
IncorrectParity,
}
pub trait Register:
Clone + TryFrom<u32, Error = RegisterParseError> + Into<u32> + Sized + Debug
{
const ADDRESS: u8;
const NAME: &'static str;
}
#[derive(Debug, thiserror::Error)]
#[error("Failed to parse register {name} from {value:#010x}")]
pub struct RegisterParseError {
name: &'static str,
value: u32,
}
impl RegisterParseError {
pub fn new(name: &'static str, value: u32) -> Self {
RegisterParseError { name, value }
}
}
pub trait ArmProbeInterface: DapAccess + SwdSequence + SwoAccess + Send {
fn access_ports(
&mut self,
dp: DpAddress,
) -> Result<BTreeSet<FullyQualifiedApAddress>, ArmError>;
fn close(self: Box<Self>) -> Probe;
fn current_debug_port(&self) -> DpAddress;
fn memory_interface(
&mut self,
access_port: &FullyQualifiedApAddress,
) -> Result<Box<dyn ArmMemoryInterface + '_>, ArmError>;
fn read_chip_info_from_rom_table(
&mut self,
dp: DpAddress,
) -> Result<Option<ArmChipInfo>, ArmError>;
}
pub trait SwdSequence {
fn swj_sequence(&mut self, bit_len: u8, bits: u64) -> Result<(), DebugProbeError>;
fn swj_pins(
&mut self,
pin_out: u32,
pin_select: u32,
pin_wait: u32,
) -> Result<u32, DebugProbeError>;
}
pub trait UninitializedArmProbe: SwdSequence + Debug {
fn initialize(
self: Box<Self>,
sequence: Arc<dyn ArmDebugSequence>,
dp: DpAddress,
) -> Result<Box<dyn ArmProbeInterface>, (Box<dyn UninitializedArmProbe>, Error)>;
fn initialize_unspecified(
self: Box<Self>,
dp: DpAddress,
) -> Result<Box<dyn ArmProbeInterface>, (Box<dyn UninitializedArmProbe>, Error)> {
self.initialize(DefaultArmSequence::create(), dp)
}
fn close(self: Box<Self>) -> Probe;
}
pub trait ArmDebugState {
fn disconnect(&mut self, _probe: &mut dyn DapProbe) {}
}
#[derive(Debug)]
pub struct Uninitialized {
pub(crate) use_overrun_detect: bool,
}
pub struct Initialized {
pub(crate) current_dp: DpAddress,
dps: HashMap<DpAddress, DpState>,
use_overrun_detect: bool,
sequence: Arc<dyn ArmDebugSequence>,
}
impl Initialized {
pub fn new(
sequence: Arc<dyn ArmDebugSequence>,
current_dp: DpAddress,
use_overrun_detect: bool,
) -> Self {
Self {
current_dp,
dps: HashMap::new(),
use_overrun_detect,
sequence,
}
}
}
impl ArmDebugState for Uninitialized {}
impl ArmDebugState for Initialized {
fn disconnect(&mut self, probe: &mut dyn DapProbe) {
let stop_span = tracing::debug_span!("debug_port_stop").entered();
self.sequence.debug_port_stop(probe, self.current_dp).ok();
for dp in self.dps.keys() {
if self.sequence.debug_port_connect(probe, *dp).is_ok() {
self.sequence.debug_port_stop(probe, *dp).ok();
} else {
tracing::warn!("Failed to stop DP {:x?}", dp);
}
}
probe.raw_flush().ok();
drop(stop_span);
}
}
#[derive(Debug)]
pub(crate) struct DpState {
pub _debug_port_version: DebugPortVersion,
pub current_dpbanksel: u8,
pub current_apsel: u8,
pub current_apbanksel: u8,
pub access_ports: BTreeSet<FullyQualifiedApAddress>,
}
impl DpState {
pub fn new() -> Self {
Self {
_debug_port_version: DebugPortVersion::Unsupported(0xFF),
current_dpbanksel: 0,
current_apsel: 0,
current_apbanksel: 0,
access_ports: BTreeSet::new(),
}
}
}
#[derive(Debug)]
pub struct ArmCommunicationInterface<S: ArmDebugState> {
probe: Option<Box<dyn DapProbe>>,
state: S,
}
impl<S: ArmDebugState> Drop for ArmCommunicationInterface<S> {
fn drop(&mut self) {
if let Some(mut probe) = self.probe.take() {
self.state.disconnect(&mut *probe);
}
}
}
impl<S: ArmDebugState> ArmCommunicationInterface<S> {
fn probe_mut(&mut self) -> &mut dyn DapProbe {
self.probe.as_deref_mut().expect("ArmCommunicationInterface is in an inconsistent state. This is a bug, please report it.")
}
fn close(mut self) -> Probe {
let mut probe = self.probe.take().expect("ArmCommunicationInterface is in an inconsistent state. This is a bug, please report it.");
self.state.disconnect(&mut *probe);
Probe::from_attached_probe(RawDapAccess::into_probe(probe))
}
}
pub trait DapProbe: RawDapAccess + DebugProbe {}
impl ArmProbeInterface for ArmCommunicationInterface<Initialized> {
fn memory_interface(
&mut self,
access_port_address: &FullyQualifiedApAddress,
) -> Result<Box<dyn ArmMemoryInterface + '_>, ArmError> {
ArmCommunicationInterface::memory_interface(self, access_port_address)
}
fn read_chip_info_from_rom_table(
&mut self,
dp: DpAddress,
) -> Result<Option<ArmChipInfo>, ArmError> {
ArmCommunicationInterface::read_chip_info_from_rom_table(self, dp)
}
fn current_debug_port(&self) -> DpAddress {
self.state.current_dp
}
fn close(self: Box<Self>) -> Probe {
ArmCommunicationInterface::close(*self)
}
fn access_ports(
&mut self,
dp: DpAddress,
) -> Result<BTreeSet<FullyQualifiedApAddress>, ArmError> {
self.select_dp(dp).map(|state| state.access_ports.clone())
}
}
impl<S: ArmDebugState> SwdSequence for ArmCommunicationInterface<S> {
fn swj_sequence(&mut self, bit_len: u8, bits: u64) -> Result<(), DebugProbeError> {
self.probe_mut().swj_sequence(bit_len, bits)?;
Ok(())
}
fn swj_pins(
&mut self,
pin_out: u32,
pin_select: u32,
pin_wait: u32,
) -> Result<u32, DebugProbeError> {
self.probe_mut().swj_pins(pin_out, pin_select, pin_wait)
}
}
impl ArmCommunicationInterface<Uninitialized> {
pub(crate) fn new(probe: Box<dyn DapProbe>, use_overrun_detect: bool) -> Self {
let state = Uninitialized { use_overrun_detect };
Self {
probe: Some(probe),
state,
}
}
}
impl UninitializedArmProbe for ArmCommunicationInterface<Uninitialized> {
fn initialize(
mut self: Box<Self>,
sequence: Arc<dyn ArmDebugSequence>,
dp: DpAddress,
) -> Result<Box<dyn ArmProbeInterface>, (Box<dyn UninitializedArmProbe>, Error)> {
let use_overrun_detect = self.state.use_overrun_detect;
let probe = self.probe.take().expect("ArmCommunicationInterface is in an inconsistent state. This is a bug, please report it.");
match ArmCommunicationInterface::<Initialized>::try_setup(
probe,
sequence,
dp,
use_overrun_detect,
) {
Ok(initialized) => Ok(Box::new(initialized)),
Err((probe, err)) => Err((
Box::new(ArmCommunicationInterface::new(probe, use_overrun_detect)),
Error::Arm(err),
)),
}
}
fn close(self: Box<Self>) -> Probe {
ArmCommunicationInterface::close(*self)
}
}
impl<'interface> ArmCommunicationInterface<Initialized> {
fn try_setup(
mut probe: Box<dyn DapProbe>,
sequence: Arc<dyn ArmDebugSequence>,
dp: DpAddress,
use_overrun_detect: bool,
) -> Result<Self, (Box<dyn DapProbe>, ArmError)> {
if let Err(err) = tracing::debug_span!("debug_port_setup")
.in_scope(|| sequence.debug_port_setup(&mut *probe, dp))
{
return Err((probe, err));
}
let mut initializing = Self {
probe: Some(probe),
state: Initialized::new(sequence, dp, use_overrun_detect),
};
if let Err(err) = initializing.select_dp(dp) {
return Err((initializing.probe.take().unwrap(), err));
}
Ok(initializing)
}
pub(crate) fn reinitialize(&mut self) -> Result<(), ArmError> {
let mut probe = self.probe.take().expect("ArmCommunicationInterface is in an inconsistent state. This is a bug, please report it.");
self.state.disconnect(&mut *probe);
match Self::try_setup(
probe,
self.state.sequence.clone(),
self.current_debug_port(),
self.state.use_overrun_detect,
) {
Ok(reinitialized) => {
let _ = std::mem::replace(self, reinitialized);
Ok(())
}
Err((probe, err)) => {
self.probe.replace(probe);
Err(err)
}
}
}
pub fn core_status_notification(&mut self, state: CoreStatus) {
self.probe_mut().core_status_notification(state).ok();
}
pub fn memory_interface(
&'interface mut self,
access_port_address: &FullyQualifiedApAddress,
) -> Result<Box<dyn ArmMemoryInterface + 'interface>, ArmError> {
if !self
.select_dp(access_port_address.dp())?
.access_ports
.contains(access_port_address)
{
return Err(ArmError::ApDoesNotExist(access_port_address.clone()));
}
let memory_interface = ADIMemoryInterface::new(self, access_port_address)?;
Ok(Box::new(memory_interface))
}
fn select_dp(&mut self, dp: DpAddress) -> Result<&mut DpState, ArmError> {
let mut switched_dp = false;
let sequence = self.state.sequence.clone();
if self.state.current_dp != dp {
tracing::debug!("Selecting DP {:x?}", dp);
switched_dp = true;
self.probe_mut().raw_flush()?;
if let Err(e) = sequence.debug_port_connect(&mut *self.probe_mut(), dp) {
tracing::warn!("Failed to switch to DP {:x?}: {}", dp, e);
sequence.debug_port_setup(&mut *self.probe_mut(), dp)?;
}
self.state.current_dp = dp;
}
if let hash_map::Entry::Vacant(entry) = self.state.dps.entry(dp) {
let sequence = self.state.sequence.clone();
entry.insert(DpState::new());
let start_span = tracing::debug_span!("debug_port_start").entered();
sequence.debug_port_start(self, dp)?;
drop(start_span);
let mut ctrl_reg: Ctrl = self.read_dp_register(dp)?;
if ctrl_reg.orun_detect() != self.state.use_overrun_detect {
tracing::debug!("Setting orun_detect: {}", self.state.use_overrun_detect);
ctrl_reg.set_orun_detect(self.state.use_overrun_detect);
self.write_dp_register(dp, ctrl_reg)?;
}
let idr: DebugPortId = self.read_dp_register::<DPIDR>(dp)?.into();
if idr.version == DebugPortVersion::DPv3 {
let idr1: DPIDR1 = self.read_dp_register(dp)?;
let base_ptr0: BASEPTR0 = self.read_dp_register(dp)?;
let base_ptr1: BASEPTR1 = self.read_dp_register(dp)?;
let base_ptr_str = base_ptr0.valid().then(|| {
format!(
"0x{:x}",
u64::from(base_ptr1.ptr()) | u64::from(base_ptr0.ptr() << 12)
)
});
tracing::info!(
"DPv3 detected: DPIDR1:{:?} BASE_PTR: {}",
idr1,
base_ptr_str.as_deref().unwrap_or("not valid")
);
return Err(ArmError::DebugPort(DebugPortError::Unsupported(
"Unsupported version (DPv3)".to_string(),
)));
}
tracing::trace!("Searching valid APs");
let ap_span = tracing::debug_span!("AP discovery").entered();
let allowed_aps = sequence.allowed_access_ports();
let access_ports = valid_access_ports_allowlist(self, dp, allowed_aps);
let state = self.state.dps.get_mut(&dp).unwrap();
state.access_ports = access_ports.into_iter().collect();
drop(ap_span);
} else if switched_dp {
let sequence = self.state.sequence.clone();
let start_span = tracing::debug_span!("debug_port_start").entered();
sequence.debug_port_start(self, dp)?;
drop(start_span);
}
Ok(self.state.dps.get_mut(&dp).unwrap())
}
fn select_dp_and_dp_bank(
&mut self,
dp: DpAddress,
dp_register_address: u8,
) -> Result<(), ArmError> {
let dp_state = self.select_dp(dp)?;
let bank = dp_register_address >> 4;
let addr = dp_register_address & 0xF;
if addr != 0 && addr != 4 {
return Ok(());
}
if bank != dp_state.current_dpbanksel {
dp_state.current_dpbanksel = bank;
let mut select = Select(0);
tracing::debug!("Changing DP_BANK_SEL to {}", dp_state.current_dpbanksel);
select.set_ap_sel(dp_state.current_apsel);
select.set_ap_bank_sel(dp_state.current_apbanksel);
select.set_dp_bank_sel(dp_state.current_dpbanksel);
self.write_dp_register(dp, select)?;
}
Ok(())
}
fn select_ap_and_ap_bank(
&mut self,
ap: &FullyQualifiedApAddress,
ap_register_address: u8,
) -> Result<(), ArmError> {
let dp_state = self.select_dp(ap.dp())?;
let port = ap.ap_v1()?;
let ap_bank = ap_register_address >> 4;
let mut cache_changed = if dp_state.current_apsel != port {
dp_state.current_apsel = port;
true
} else {
false
};
if dp_state.current_apbanksel != ap_bank {
dp_state.current_apbanksel = ap_bank;
cache_changed = true;
}
if cache_changed {
let mut select = Select(0);
tracing::debug!(
"Changing AP to {}, AP_BANK_SEL to {}",
dp_state.current_apsel,
dp_state.current_apbanksel
);
select.set_ap_sel(dp_state.current_apsel);
select.set_ap_bank_sel(dp_state.current_apbanksel);
select.set_dp_bank_sel(dp_state.current_dpbanksel);
self.write_dp_register(ap.dp(), select)?;
}
Ok(())
}
}
impl FlushableArmAccess for ArmCommunicationInterface<Initialized> {
fn flush(&mut self) -> Result<(), ArmError> {
self.probe_mut().raw_flush()
}
}
impl SwoAccess for ArmCommunicationInterface<Initialized> {
fn enable_swo(&mut self, config: &SwoConfig) -> Result<(), ArmError> {
match self.probe_mut().get_swo_interface_mut() {
Some(interface) => interface.enable_swo(config),
None => Err(ArmError::ArchitectureRequired(&["ARMv7", "ARMv8"])),
}
}
fn disable_swo(&mut self) -> Result<(), ArmError> {
match self.probe_mut().get_swo_interface_mut() {
Some(interface) => interface.disable_swo(),
None => Err(ArmError::ArchitectureRequired(&["ARMv7", "ARMv8"])),
}
}
fn read_swo_timeout(&mut self, timeout: Duration) -> Result<Vec<u8>, ArmError> {
match self.probe_mut().get_swo_interface_mut() {
Some(interface) => interface.read_swo_timeout(timeout),
None => Err(ArmError::ArchitectureRequired(&["ARMv7", "ARMv8"])),
}
}
}
impl DapAccess for ArmCommunicationInterface<Initialized> {
fn read_raw_dp_register(&mut self, dp: DpAddress, address: u8) -> Result<u32, ArmError> {
self.select_dp_and_dp_bank(dp, address)?;
let result = self
.probe_mut()
.raw_read_register(PortType::DebugPort, address & 0xf)?;
Ok(result)
}
fn write_raw_dp_register(
&mut self,
dp: DpAddress,
address: u8,
value: u32,
) -> Result<(), ArmError> {
self.select_dp_and_dp_bank(dp, address)?;
self.probe_mut()
.raw_write_register(PortType::DebugPort, address, value)?;
Ok(())
}
fn read_raw_ap_register(
&mut self,
ap: &FullyQualifiedApAddress,
address: u8,
) -> Result<u32, ArmError> {
self.select_ap_and_ap_bank(ap, address)?;
let result = self
.probe_mut()
.raw_read_register(PortType::AccessPort, address & 0xf)?;
Ok(result)
}
fn read_raw_ap_register_repeated(
&mut self,
ap: &FullyQualifiedApAddress,
address: u8,
values: &mut [u32],
) -> Result<(), ArmError> {
self.select_ap_and_ap_bank(ap, address)?;
self.probe_mut()
.raw_read_block(PortType::AccessPort, address, values)?;
Ok(())
}
fn write_raw_ap_register(
&mut self,
ap: &FullyQualifiedApAddress,
address: u8,
value: u32,
) -> Result<(), ArmError> {
self.select_ap_and_ap_bank(ap, address)?;
self.probe_mut()
.raw_write_register(PortType::AccessPort, address, value)?;
Ok(())
}
fn write_raw_ap_register_repeated(
&mut self,
ap: &FullyQualifiedApAddress,
address: u8,
values: &[u32],
) -> Result<(), ArmError> {
self.select_ap_and_ap_bank(ap, address)?;
self.probe_mut()
.raw_write_block(PortType::AccessPort, address, values)?;
Ok(())
}
}
#[derive(Debug, Clone, Copy)]
pub struct ArmChipInfo {
pub manufacturer: JEP106Code,
pub part: u16,
}
impl ArmCommunicationInterface<Initialized> {
pub fn read_chip_info_from_rom_table(
&mut self,
dp: DpAddress,
) -> Result<Option<ArmChipInfo>, ArmError> {
let ctrl_reg: crate::architecture::arm::dp::Ctrl = self.read_dp_register(dp)?;
if ctrl_reg.sticky_err() {
tracing::trace!("AP Search faulted. Cleaning up");
let mut abort = Abort::default();
abort.set_stkerrclr(true);
self.write_dp_register(dp, abort)?;
}
let state = self.select_dp(dp)?;
for access_port in state.access_ports.clone() {
if let Ok(mut memory) = self.memory_interface(&access_port) {
let base_addr = memory.base_address()?;
let component = Component::try_parse(&mut *memory, base_addr)?;
if let Component::Class1RomTable(component_id, _) = component {
if let Some(jep106) = component_id.peripheral_id().jep106() {
return Ok(Some(ArmChipInfo {
manufacturer: jep106,
part: component_id.peripheral_id().part(),
}));
}
}
}
}
Ok(None)
}
}
impl std::fmt::Display for ArmChipInfo {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
let manu = match self.manufacturer.get() {
Some(name) => name.to_string(),
None => format!(
"<unknown manufacturer (cc={:2x}, id={:2x})>",
self.manufacturer.cc, self.manufacturer.id
),
};
write!(f, "{} 0x{:04x}", manu, self.part)
}
}
pub trait FlushableArmAccess {
fn flush(&mut self) -> Result<(), ArmError>;
}