diff options
Diffstat (limited to 'examples/stm32h7/src')
| -rw-r--r-- | examples/stm32h7/src/bin/eth.rs | 167 |
1 files changed, 167 insertions, 0 deletions
diff --git a/examples/stm32h7/src/bin/eth.rs b/examples/stm32h7/src/bin/eth.rs new file mode 100644 index 000000000..57997da0a --- /dev/null +++ b/examples/stm32h7/src/bin/eth.rs | |||
| @@ -0,0 +1,167 @@ | |||
| 1 | #![no_std] | ||
| 2 | #![no_main] | ||
| 3 | #![feature(trait_alias)] | ||
| 4 | #![feature(min_type_alias_impl_trait)] | ||
| 5 | #![feature(impl_trait_in_bindings)] | ||
| 6 | #![feature(type_alias_impl_trait)] | ||
| 7 | |||
| 8 | use core::pin::Pin; | ||
| 9 | use core::sync::atomic::{AtomicUsize, Ordering}; | ||
| 10 | |||
| 11 | use cortex_m_rt::entry; | ||
| 12 | use defmt::{info, unwrap}; | ||
| 13 | use defmt_rtt as _; // global logger | ||
| 14 | use embassy::executor::{Executor, Spawner}; | ||
| 15 | use embassy::io::AsyncWriteExt; | ||
| 16 | use embassy::time::{Duration, Timer}; | ||
| 17 | use embassy::util::Forever; | ||
| 18 | use embassy_macros::interrupt_take; | ||
| 19 | use embassy_net::{Config as NetConfig, Ipv4Address, Ipv4Cidr, StaticConfigurator, TcpSocket}; | ||
| 20 | use embassy_stm32::clock::{Alarm, Clock}; | ||
| 21 | use embassy_stm32::eth::lan8742a::LAN8742A; | ||
| 22 | use embassy_stm32::eth::Ethernet; | ||
| 23 | use embassy_stm32::rcc::{Config as RccConfig, Rcc}; | ||
| 24 | use embassy_stm32::rng::Random; | ||
| 25 | use embassy_stm32::time::Hertz; | ||
| 26 | use embassy_stm32::{interrupt, peripherals, Config}; | ||
| 27 | use heapless::Vec; | ||
| 28 | use panic_probe as _; | ||
| 29 | use peripherals::{RNG, TIM2}; | ||
| 30 | |||
| 31 | defmt::timestamp! {"{=u64}", { | ||
| 32 | static COUNT: AtomicUsize = AtomicUsize::new(0); | ||
| 33 | // NOTE(no-CAS) `timestamps` runs with interrupts disabled | ||
| 34 | let n = COUNT.load(Ordering::Relaxed); | ||
| 35 | COUNT.store(n + 1, Ordering::Relaxed); | ||
| 36 | n as u64 | ||
| 37 | } | ||
| 38 | } | ||
| 39 | |||
| 40 | #[embassy::task] | ||
| 41 | async fn main_task( | ||
| 42 | device: &'static mut Pin<&'static mut Ethernet<'static, LAN8742A, 4, 4>>, | ||
| 43 | config: &'static mut StaticConfigurator, | ||
| 44 | spawner: Spawner, | ||
| 45 | ) { | ||
| 46 | // Init network stack | ||
| 47 | embassy_net::init(device, config); | ||
| 48 | |||
| 49 | // Launch network task | ||
| 50 | unwrap!(spawner.spawn(net_task())); | ||
| 51 | |||
| 52 | info!("Network task initialized"); | ||
| 53 | |||
| 54 | // Then we can use it! | ||
| 55 | let mut rx_buffer = [0; 1024]; | ||
| 56 | let mut tx_buffer = [0; 1024]; | ||
| 57 | let mut socket = TcpSocket::new(&mut rx_buffer, &mut tx_buffer); | ||
| 58 | |||
| 59 | socket.set_timeout(Some(embassy_net::SmolDuration::from_secs(10))); | ||
| 60 | |||
| 61 | let remote_endpoint = (Ipv4Address::new(192, 168, 0, 10), 8000); | ||
| 62 | let r = socket.connect(remote_endpoint).await; | ||
| 63 | if let Err(e) = r { | ||
| 64 | info!("connect error: {:?}", e); | ||
| 65 | return; | ||
| 66 | } | ||
| 67 | info!("connected!"); | ||
| 68 | loop { | ||
| 69 | let r = socket.write_all(b"Hello\n").await; | ||
| 70 | if let Err(e) = r { | ||
| 71 | info!("write error: {:?}", e); | ||
| 72 | return; | ||
| 73 | } | ||
| 74 | Timer::after(Duration::from_secs(1)).await; | ||
| 75 | } | ||
| 76 | } | ||
| 77 | |||
| 78 | #[embassy::task] | ||
| 79 | async fn net_task() { | ||
| 80 | embassy_net::run().await | ||
| 81 | } | ||
| 82 | |||
| 83 | #[no_mangle] | ||
| 84 | fn _embassy_rand(buf: &mut [u8]) { | ||
| 85 | use rand_core::RngCore; | ||
| 86 | |||
| 87 | critical_section::with(|_| unsafe { | ||
| 88 | unwrap!(RNG_INST.as_mut()).fill_bytes(buf); | ||
| 89 | }); | ||
| 90 | } | ||
| 91 | |||
| 92 | static mut RNG_INST: Option<Random<RNG>> = None; | ||
| 93 | |||
| 94 | static EXECUTOR: Forever<Executor> = Forever::new(); | ||
| 95 | static TIMER_RTC: Forever<Clock<TIM2>> = Forever::new(); | ||
| 96 | static ALARM: Forever<Alarm<TIM2>> = Forever::new(); | ||
| 97 | static ETH: Forever<Ethernet<'static, LAN8742A, 4, 4>> = Forever::new(); | ||
| 98 | static DEVICE: Forever<Pin<&'static mut Ethernet<'static, LAN8742A, 4, 4>>> = Forever::new(); | ||
| 99 | static CONFIG: Forever<StaticConfigurator> = Forever::new(); | ||
| 100 | |||
| 101 | #[entry] | ||
| 102 | fn main() -> ! { | ||
| 103 | use stm32_metapac::RCC; | ||
| 104 | |||
| 105 | info!("Hello World!"); | ||
| 106 | |||
| 107 | info!("Setup RCC..."); | ||
| 108 | let mut rcc_config = RccConfig::default(); | ||
| 109 | rcc_config.sys_ck = Some(Hertz(400_000_000)); | ||
| 110 | rcc_config.pll1.q_ck = Some(Hertz(100_000_000)); | ||
| 111 | let config = Config::default().rcc(rcc_config); | ||
| 112 | |||
| 113 | let mut p = embassy_stm32::init(config); | ||
| 114 | |||
| 115 | // Constrain and Freeze clock | ||
| 116 | |||
| 117 | let mut rcc = Rcc::new(&mut p.RCC, RccConfig::default()); | ||
| 118 | rcc.enable_debug_wfe(&mut p.DBGMCU, true); | ||
| 119 | |||
| 120 | unsafe { | ||
| 121 | RCC.ahb4enr().modify(|w| { | ||
| 122 | w.set_gpioaen(true); | ||
| 123 | w.set_gpioben(true); | ||
| 124 | w.set_gpiocen(true); | ||
| 125 | w.set_gpioden(true); | ||
| 126 | w.set_gpioien(true); | ||
| 127 | }); | ||
| 128 | } | ||
| 129 | |||
| 130 | let rtc_int = interrupt_take!(TIM2); | ||
| 131 | let rtc = TIMER_RTC.put(Clock::new(p.TIM2, rtc_int)); | ||
| 132 | rtc.start(); | ||
| 133 | let alarm = ALARM.put(rtc.alarm1()); | ||
| 134 | |||
| 135 | unsafe { embassy::time::set_clock(rtc) }; | ||
| 136 | |||
| 137 | let rng = Random::new(p.RNG); | ||
| 138 | unsafe { | ||
| 139 | RNG_INST.replace(rng); | ||
| 140 | } | ||
| 141 | |||
| 142 | let eth_int = interrupt_take!(ETH); | ||
| 143 | let mac_addr = [0x10; 6]; | ||
| 144 | let eth = ETH.put(Ethernet::new( | ||
| 145 | p.ETH, eth_int, p.PA1, p.PA2, p.PC1, p.PA7, p.PC4, p.PC5, p.PB12, p.PB13, p.PB11, LAN8742A, | ||
| 146 | mac_addr, 1, | ||
| 147 | )); | ||
| 148 | |||
| 149 | // NOTE(unsafe) This thing is a &'static | ||
| 150 | let net_device = DEVICE.put(unsafe { Pin::new_unchecked(eth) }); | ||
| 151 | net_device.as_mut().init(); | ||
| 152 | |||
| 153 | let config = StaticConfigurator::new(NetConfig { | ||
| 154 | address: Ipv4Cidr::new(Ipv4Address::new(192, 168, 0, 61), 24), | ||
| 155 | dns_servers: Vec::new(), | ||
| 156 | gateway: Some(Ipv4Address::new(192, 168, 0, 1)), | ||
| 157 | }); | ||
| 158 | |||
| 159 | let config = CONFIG.put(config); | ||
| 160 | |||
| 161 | let executor = EXECUTOR.put(Executor::new()); | ||
| 162 | executor.set_alarm(alarm); | ||
| 163 | |||
| 164 | executor.run(move |spawner| { | ||
| 165 | unwrap!(spawner.spawn(main_task(net_device, config, spawner))); | ||
| 166 | }) | ||
| 167 | } | ||
