1b0661ebb1
Allow capturing (and distinguishing) non-SOP packets as well. The default configuration will just configure SOP packets. For ease of use the default receive function signature is unchanged as for PD sinks (which is likely the common usage) just SOP is enough so no need to differentiate.
87 lines
2.8 KiB
Rust
87 lines
2.8 KiB
Rust
#![no_std]
|
|
#![no_main]
|
|
|
|
use defmt::{error, info, Format};
|
|
use embassy_executor::Spawner;
|
|
use embassy_stm32::ucpd::{self, CcPhy, CcPull, CcSel, CcVState, Ucpd};
|
|
use embassy_stm32::{bind_interrupts, peripherals, Config};
|
|
use embassy_time::{with_timeout, Duration};
|
|
use {defmt_rtt as _, panic_probe as _};
|
|
|
|
bind_interrupts!(struct Irqs {
|
|
UCPD1 => ucpd::InterruptHandler<peripherals::UCPD1>;
|
|
});
|
|
|
|
#[derive(Debug, Format)]
|
|
enum CableOrientation {
|
|
Normal,
|
|
Flipped,
|
|
DebugAccessoryMode,
|
|
}
|
|
|
|
// Returns true when the cable
|
|
async fn wait_attached<T: ucpd::Instance>(cc_phy: &mut CcPhy<'_, T>) -> CableOrientation {
|
|
loop {
|
|
let (cc1, cc2) = cc_phy.vstate();
|
|
if cc1 == CcVState::LOWEST && cc2 == CcVState::LOWEST {
|
|
// Detached, wait until attached by monitoring the CC lines.
|
|
cc_phy.wait_for_vstate_change().await;
|
|
continue;
|
|
}
|
|
|
|
// Attached, wait for CC lines to be stable for tCCDebounce (100..200ms).
|
|
if with_timeout(Duration::from_millis(100), cc_phy.wait_for_vstate_change())
|
|
.await
|
|
.is_ok()
|
|
{
|
|
// State has changed, restart detection procedure.
|
|
continue;
|
|
};
|
|
|
|
// State was stable for the complete debounce period, check orientation.
|
|
return match (cc1, cc2) {
|
|
(_, CcVState::LOWEST) => CableOrientation::Normal, // CC1 connected
|
|
(CcVState::LOWEST, _) => CableOrientation::Flipped, // CC2 connected
|
|
_ => CableOrientation::DebugAccessoryMode, // Both connected (special cable)
|
|
};
|
|
}
|
|
}
|
|
|
|
#[embassy_executor::main]
|
|
async fn main(_spawner: Spawner) {
|
|
let mut config = Config::default();
|
|
config.enable_ucpd1_dead_battery = true;
|
|
let p = embassy_stm32::init(config);
|
|
|
|
info!("Hello World!");
|
|
|
|
let mut ucpd = Ucpd::new(p.UCPD1, Irqs {}, p.PB6, p.PB4, Default::default());
|
|
ucpd.cc_phy().set_pull(CcPull::Sink);
|
|
|
|
info!("Waiting for USB connection...");
|
|
let cable_orientation = wait_attached(ucpd.cc_phy()).await;
|
|
info!("USB cable connected, orientation: {}", cable_orientation);
|
|
|
|
let cc_sel = match cable_orientation {
|
|
CableOrientation::Normal => {
|
|
info!("Starting PD communication on CC1 pin");
|
|
CcSel::CC1
|
|
}
|
|
CableOrientation::Flipped => {
|
|
info!("Starting PD communication on CC2 pin");
|
|
CcSel::CC2
|
|
}
|
|
CableOrientation::DebugAccessoryMode => panic!("No PD communication in DAM"),
|
|
};
|
|
let (_cc_phy, mut pd_phy) = ucpd.split_pd_phy(p.DMA1_CH1, p.DMA1_CH2, cc_sel);
|
|
|
|
loop {
|
|
// Enough space for the longest non-extended data message.
|
|
let mut buf = [0_u8; 30];
|
|
match pd_phy.receive(buf.as_mut()).await {
|
|
Ok(n) => info!("USB PD RX: {=[u8]:?}", &buf[..n]),
|
|
Err(e) => error!("USB PD RX: {}", e),
|
|
}
|
|
}
|
|
}
|