diff options
| author | Raul Alimbekov <[email protected]> | 2025-12-16 09:05:22 +0300 |
|---|---|---|
| committer | GitHub <[email protected]> | 2025-12-16 09:05:22 +0300 |
| commit | c9a04b4b732b7a3b696eb8223664c1a7942b1875 (patch) | |
| tree | 6dbe5c02e66eed8d8762f13f95afd24f8db2b38c /examples/stm32wb/src/bin/mac_ffd_net.rs | |
| parent | cde24a3ef1117653ba5ed4184102b33f745782fb (diff) | |
| parent | 5ae6e060ec1c90561719aabdc29d5b6e7b8b0a82 (diff) | |
Merge branch 'main' into main
Diffstat (limited to 'examples/stm32wb/src/bin/mac_ffd_net.rs')
| -rw-r--r-- | examples/stm32wb/src/bin/mac_ffd_net.rs | 194 |
1 files changed, 82 insertions, 112 deletions
diff --git a/examples/stm32wb/src/bin/mac_ffd_net.rs b/examples/stm32wb/src/bin/mac_ffd_net.rs index cc3b21e2e..b4789e3ee 100644 --- a/examples/stm32wb/src/bin/mac_ffd_net.rs +++ b/examples/stm32wb/src/bin/mac_ffd_net.rs | |||
| @@ -1,32 +1,44 @@ | |||
| 1 | #![no_std] | 1 | #![no_std] |
| 2 | #![no_main] | 2 | #![no_main] |
| 3 | 3 | ||
| 4 | use core::net::Ipv6Addr; | ||
| 5 | |||
| 4 | use defmt::*; | 6 | use defmt::*; |
| 5 | use embassy_executor::Spawner; | 7 | use embassy_executor::Spawner; |
| 8 | use embassy_net::udp::{PacketMetadata, UdpSocket}; | ||
| 9 | use embassy_net::{Ipv6Cidr, StackResources, StaticConfigV6}; | ||
| 6 | use embassy_stm32::bind_interrupts; | 10 | use embassy_stm32::bind_interrupts; |
| 7 | use embassy_stm32::ipcc::{Config, ReceiveInterruptHandler, TransmitInterruptHandler}; | 11 | use embassy_stm32::ipcc::{Config, ReceiveInterruptHandler, TransmitInterruptHandler}; |
| 12 | use embassy_stm32::peripherals::RNG; | ||
| 8 | use embassy_stm32::rcc::WPAN_DEFAULT; | 13 | use embassy_stm32::rcc::WPAN_DEFAULT; |
| 9 | use embassy_stm32_wpan::mac::commands::{ResetRequest, SetRequest, StartRequest}; | 14 | use embassy_stm32::rng::InterruptHandler as RngInterruptHandler; |
| 10 | use embassy_stm32_wpan::mac::typedefs::{MacChannel, PanId, PibId}; | ||
| 11 | use embassy_stm32_wpan::mac::{self, Runner}; | ||
| 12 | use embassy_stm32_wpan::sub::mm; | ||
| 13 | use embassy_stm32_wpan::TlMbox; | 15 | use embassy_stm32_wpan::TlMbox; |
| 16 | use embassy_stm32_wpan::mac::{Driver, DriverState, Runner}; | ||
| 17 | use embassy_stm32_wpan::sub::mm; | ||
| 18 | use embassy_time::{Duration, Timer}; | ||
| 19 | use heapless::Vec; | ||
| 14 | use static_cell::StaticCell; | 20 | use static_cell::StaticCell; |
| 15 | use {defmt_rtt as _, panic_probe as _}; | 21 | use {defmt_rtt as _, panic_probe as _}; |
| 16 | 22 | ||
| 17 | bind_interrupts!(struct Irqs{ | 23 | bind_interrupts!(struct Irqs{ |
| 18 | IPCC_C1_RX => ReceiveInterruptHandler; | 24 | IPCC_C1_RX => ReceiveInterruptHandler; |
| 19 | IPCC_C1_TX => TransmitInterruptHandler; | 25 | IPCC_C1_TX => TransmitInterruptHandler; |
| 26 | RNG => RngInterruptHandler<RNG>; | ||
| 20 | }); | 27 | }); |
| 21 | 28 | ||
| 22 | #[embassy_executor::task] | 29 | #[embassy_executor::task] |
| 23 | async fn run_mm_queue(memory_manager: mm::MemoryManager) { | 30 | async fn run_mm_queue(mut memory_manager: mm::MemoryManager<'static>) -> ! { |
| 24 | memory_manager.run_queue().await; | 31 | memory_manager.run_queue().await |
| 25 | } | 32 | } |
| 26 | 33 | ||
| 27 | #[embassy_executor::task] | 34 | #[embassy_executor::task] |
| 28 | async fn run_mac(runner: &'static Runner<'static>) { | 35 | async fn run_mac(runner: &'static Runner<'static>) -> ! { |
| 29 | runner.run().await; | 36 | runner.run().await |
| 37 | } | ||
| 38 | |||
| 39 | #[embassy_executor::task] | ||
| 40 | async fn run_net(mut runner: embassy_net::Runner<'static, Driver<'static>>) -> ! { | ||
| 41 | runner.run().await | ||
| 30 | } | 42 | } |
| 31 | 43 | ||
| 32 | #[embassy_executor::main] | 44 | #[embassy_executor::main] |
| @@ -60,118 +72,76 @@ async fn main(spawner: Spawner) { | |||
| 60 | info!("Hello World!"); | 72 | info!("Hello World!"); |
| 61 | 73 | ||
| 62 | let config = Config::default(); | 74 | let config = Config::default(); |
| 63 | let mbox = TlMbox::init(p.IPCC, Irqs, config); | 75 | let mut mbox = TlMbox::init(p.IPCC, Irqs, config).await; |
| 64 | 76 | ||
| 65 | spawner.spawn(run_mm_queue(mbox.mm_subsystem).unwrap()); | 77 | spawner.spawn(run_mm_queue(mbox.mm_subsystem).unwrap()); |
| 66 | 78 | ||
| 67 | let sys_event = mbox.sys_subsystem.read().await; | ||
| 68 | info!("sys event: {}", sys_event.payload()); | ||
| 69 | |||
| 70 | core::mem::drop(sys_event); | ||
| 71 | |||
| 72 | let result = mbox.sys_subsystem.shci_c2_mac_802_15_4_init().await; | 79 | let result = mbox.sys_subsystem.shci_c2_mac_802_15_4_init().await; |
| 73 | info!("initialized mac: {}", result); | 80 | info!("initialized mac: {}", result); |
| 74 | 81 | ||
| 75 | info!("resetting"); | 82 | static DRIVER_STATE: StaticCell<DriverState> = StaticCell::new(); |
| 76 | mbox.mac_subsystem | ||
| 77 | .send_command(&ResetRequest { | ||
| 78 | set_default_pib: true, | ||
| 79 | ..Default::default() | ||
| 80 | }) | ||
| 81 | .await | ||
| 82 | .unwrap(); | ||
| 83 | defmt::info!("{:#x}", mbox.mac_subsystem.read().await.unwrap()); | ||
| 84 | |||
| 85 | info!("setting extended address"); | ||
| 86 | let extended_address: u64 = 0xACDE480000000001; | ||
| 87 | mbox.mac_subsystem | ||
| 88 | .send_command(&SetRequest { | ||
| 89 | pib_attribute_ptr: &extended_address as *const _ as *const u8, | ||
| 90 | pib_attribute: PibId::ExtendedAddress, | ||
| 91 | }) | ||
| 92 | .await | ||
| 93 | .unwrap(); | ||
| 94 | defmt::info!("{:#x}", mbox.mac_subsystem.read().await.unwrap()); | ||
| 95 | |||
| 96 | info!("setting short address"); | ||
| 97 | let short_address: u16 = 0x1122; | ||
| 98 | mbox.mac_subsystem | ||
| 99 | .send_command(&SetRequest { | ||
| 100 | pib_attribute_ptr: &short_address as *const _ as *const u8, | ||
| 101 | pib_attribute: PibId::ShortAddress, | ||
| 102 | }) | ||
| 103 | .await | ||
| 104 | .unwrap(); | ||
| 105 | defmt::info!("{:#x}", mbox.mac_subsystem.read().await.unwrap()); | ||
| 106 | |||
| 107 | info!("setting association permit"); | ||
| 108 | let association_permit: bool = true; | ||
| 109 | mbox.mac_subsystem | ||
| 110 | .send_command(&SetRequest { | ||
| 111 | pib_attribute_ptr: &association_permit as *const _ as *const u8, | ||
| 112 | pib_attribute: PibId::AssociationPermit, | ||
| 113 | }) | ||
| 114 | .await | ||
| 115 | .unwrap(); | ||
| 116 | defmt::info!("{:#x}", mbox.mac_subsystem.read().await.unwrap()); | ||
| 117 | |||
| 118 | info!("setting TX power"); | ||
| 119 | let transmit_power: i8 = 2; | ||
| 120 | mbox.mac_subsystem | ||
| 121 | .send_command(&SetRequest { | ||
| 122 | pib_attribute_ptr: &transmit_power as *const _ as *const u8, | ||
| 123 | pib_attribute: PibId::TransmitPower, | ||
| 124 | }) | ||
| 125 | .await | ||
| 126 | .unwrap(); | ||
| 127 | defmt::info!("{:#x}", mbox.mac_subsystem.read().await.unwrap()); | ||
| 128 | |||
| 129 | info!("starting FFD device"); | ||
| 130 | mbox.mac_subsystem | ||
| 131 | .send_command(&StartRequest { | ||
| 132 | pan_id: PanId([0x1A, 0xAA]), | ||
| 133 | channel_number: MacChannel::Channel16, | ||
| 134 | beacon_order: 0x0F, | ||
| 135 | superframe_order: 0x0F, | ||
| 136 | pan_coordinator: true, | ||
| 137 | battery_life_extension: false, | ||
| 138 | ..Default::default() | ||
| 139 | }) | ||
| 140 | .await | ||
| 141 | .unwrap(); | ||
| 142 | defmt::info!("{:#x}", mbox.mac_subsystem.read().await.unwrap()); | ||
| 143 | |||
| 144 | info!("setting RX on when idle"); | ||
| 145 | let rx_on_while_idle: bool = true; | ||
| 146 | mbox.mac_subsystem | ||
| 147 | .send_command(&SetRequest { | ||
| 148 | pib_attribute_ptr: &rx_on_while_idle as *const _ as *const u8, | ||
| 149 | pib_attribute: PibId::RxOnWhenIdle, | ||
| 150 | }) | ||
| 151 | .await | ||
| 152 | .unwrap(); | ||
| 153 | defmt::info!("{:#x}", mbox.mac_subsystem.read().await.unwrap()); | ||
| 154 | |||
| 155 | static TX1: StaticCell<[u8; 127]> = StaticCell::new(); | ||
| 156 | static TX2: StaticCell<[u8; 127]> = StaticCell::new(); | ||
| 157 | static TX3: StaticCell<[u8; 127]> = StaticCell::new(); | ||
| 158 | static TX4: StaticCell<[u8; 127]> = StaticCell::new(); | ||
| 159 | static TX5: StaticCell<[u8; 127]> = StaticCell::new(); | ||
| 160 | let tx_queue = [ | ||
| 161 | TX1.init([0u8; 127]), | ||
| 162 | TX2.init([0u8; 127]), | ||
| 163 | TX3.init([0u8; 127]), | ||
| 164 | TX4.init([0u8; 127]), | ||
| 165 | TX5.init([0u8; 127]), | ||
| 166 | ]; | ||
| 167 | |||
| 168 | static RUNNER: StaticCell<Runner> = StaticCell::new(); | 83 | static RUNNER: StaticCell<Runner> = StaticCell::new(); |
| 169 | let runner = RUNNER.init(Runner::new(mbox.mac_subsystem, tx_queue)); | 84 | static RESOURCES: StaticCell<StackResources<3>> = StaticCell::new(); |
| 85 | |||
| 86 | let driver_state = DRIVER_STATE.init(DriverState::new(mbox.mac_subsystem)); | ||
| 87 | |||
| 88 | let (driver, mac_runner, mut control) = Driver::new( | ||
| 89 | driver_state, | ||
| 90 | 0x1122u16.to_be_bytes().try_into().unwrap(), | ||
| 91 | 0xACDE480000000001u64.to_be_bytes().try_into().unwrap(), | ||
| 92 | ); | ||
| 93 | |||
| 94 | // TODO: rng does not work for some reason | ||
| 95 | // Generate random seed. | ||
| 96 | // let mut rng = Rng::new(p.RNG, Irqs); | ||
| 97 | let seed = [0; 8]; | ||
| 98 | // let _ = rng.async_fill_bytes(&mut seed).await; | ||
| 99 | let seed = u64::from_le_bytes(seed); | ||
| 100 | |||
| 101 | info!("seed generated"); | ||
| 102 | |||
| 103 | // Init network stack | ||
| 104 | let ipv6_addr = Ipv6Addr::new(0, 0, 0, 0, 0, 0xffff, 0xc00a, 0x2ff); | ||
| 105 | |||
| 106 | let config = embassy_net::Config::ipv6_static(StaticConfigV6 { | ||
| 107 | address: Ipv6Cidr::new(ipv6_addr, 104), | ||
| 108 | gateway: None, | ||
| 109 | dns_servers: Vec::new(), | ||
| 110 | }); | ||
| 111 | |||
| 112 | let (stack, eth_runner) = embassy_net::new(driver, config, RESOURCES.init(StackResources::new()), seed); | ||
| 113 | |||
| 114 | // wpan runner | ||
| 115 | spawner.spawn(run_mac(RUNNER.init(mac_runner)).unwrap()); | ||
| 116 | |||
| 117 | // Launch network task | ||
| 118 | spawner.spawn(unwrap!(run_net(eth_runner))); | ||
| 119 | |||
| 120 | info!("Network task initialized"); | ||
| 121 | |||
| 122 | control.init_link([0x1A, 0xAA]).await; | ||
| 123 | |||
| 124 | // Ensure DHCP configuration is up before trying connect | ||
| 125 | stack.wait_config_up().await; | ||
| 126 | |||
| 127 | info!("Network up"); | ||
| 128 | |||
| 129 | // Then we can use it! | ||
| 130 | let mut rx_meta = [PacketMetadata::EMPTY]; | ||
| 131 | let mut rx_buffer = [0; 4096]; | ||
| 132 | let mut tx_meta = [PacketMetadata::EMPTY]; | ||
| 133 | let mut tx_buffer = [0; 4096]; | ||
| 134 | |||
| 135 | let mut socket = UdpSocket::new(stack, &mut rx_meta, &mut rx_buffer, &mut tx_meta, &mut tx_buffer); | ||
| 136 | |||
| 137 | let remote_endpoint = (Ipv6Addr::new(0, 0, 0, 0, 0, 0xffff, 0xc00a, 0x2fb), 8000); | ||
| 138 | |||
| 139 | let send_buf = [0u8; 20]; | ||
| 170 | 140 | ||
| 171 | spawner.spawn(run_mac(runner).unwrap()); | 141 | socket.bind((ipv6_addr, 8000)).unwrap(); |
| 142 | socket.send_to(&send_buf, remote_endpoint).await.unwrap(); | ||
| 172 | 143 | ||
| 173 | let (driver, control) = mac::new(runner).await; | 144 | Timer::after(Duration::from_secs(2)).await; |
| 174 | 145 | ||
| 175 | let _ = driver; | 146 | cortex_m::asm::bkpt(); |
| 176 | let _ = control; | ||
| 177 | } | 147 | } |
