diff options
| author | kalkyl <[email protected]> | 2023-05-09 01:51:08 +0200 |
|---|---|---|
| committer | kalkyl <[email protected]> | 2023-05-09 01:51:08 +0200 |
| commit | 72b0379125b87bcd274bdb81127dd5f0ab29d661 (patch) | |
| tree | b620d67bc9ea930051668d63190dd6de62485b0a /examples/src/bin | |
:rainbow:
Diffstat (limited to 'examples/src/bin')
| -rw-r--r-- | examples/src/bin/multisocket.rs | 148 | ||||
| -rw-r--r-- | examples/src/bin/tcp-client.rs | 128 | ||||
| -rw-r--r-- | examples/src/bin/tcp-server.rs | 136 | ||||
| -rw-r--r-- | examples/src/bin/udp.rs | 123 |
4 files changed, 535 insertions, 0 deletions
diff --git a/examples/src/bin/multisocket.rs b/examples/src/bin/multisocket.rs new file mode 100644 index 000000000..49bcbdbb0 --- /dev/null +++ b/examples/src/bin/multisocket.rs | |||
| @@ -0,0 +1,148 @@ | |||
| 1 | #![no_std] | ||
| 2 | #![no_main] | ||
| 3 | #![feature(type_alias_impl_trait)] | ||
| 4 | |||
| 5 | use defmt::*; | ||
| 6 | use embassy_executor::Spawner; | ||
| 7 | use embassy_futures::yield_now; | ||
| 8 | use embassy_net::{Stack, StackResources}; | ||
| 9 | use embassy_net_w5500::*; | ||
| 10 | use embassy_rp::clocks::RoscRng; | ||
| 11 | use embassy_rp::gpio::{Input, Level, Output, Pull}; | ||
| 12 | use embassy_rp::peripherals::{PIN_17, PIN_20, PIN_21, SPI0}; | ||
| 13 | use embassy_rp::spi::{Async, Config as SpiConfig, Spi}; | ||
| 14 | use embedded_hal_async::spi::ExclusiveDevice; | ||
| 15 | use embedded_io::asynch::Write; | ||
| 16 | use rand::RngCore; | ||
| 17 | use static_cell::StaticCell; | ||
| 18 | use {defmt_rtt as _, panic_probe as _}; | ||
| 19 | |||
| 20 | macro_rules! singleton { | ||
| 21 | ($val:expr) => {{ | ||
| 22 | type T = impl Sized; | ||
| 23 | static STATIC_CELL: StaticCell<T> = StaticCell::new(); | ||
| 24 | let (x,) = STATIC_CELL.init(($val,)); | ||
| 25 | x | ||
| 26 | }}; | ||
| 27 | } | ||
| 28 | |||
| 29 | #[embassy_executor::task] | ||
| 30 | async fn ethernet_task( | ||
| 31 | runner: Runner< | ||
| 32 | 'static, | ||
| 33 | ExclusiveDevice<Spi<'static, SPI0, Async>, Output<'static, PIN_17>>, | ||
| 34 | Input<'static, PIN_21>, | ||
| 35 | Output<'static, PIN_20>, | ||
| 36 | >, | ||
| 37 | ) -> ! { | ||
| 38 | runner.run().await | ||
| 39 | } | ||
| 40 | |||
| 41 | #[embassy_executor::task] | ||
| 42 | async fn net_task(stack: &'static Stack<Device<'static>>) -> ! { | ||
| 43 | stack.run().await | ||
| 44 | } | ||
| 45 | |||
| 46 | #[embassy_executor::main] | ||
| 47 | async fn main(spawner: Spawner) { | ||
| 48 | let p = embassy_rp::init(Default::default()); | ||
| 49 | let mut rng = RoscRng; | ||
| 50 | |||
| 51 | let mut spi_cfg = SpiConfig::default(); | ||
| 52 | spi_cfg.frequency = 50_000_000; | ||
| 53 | let (miso, mosi, clk) = (p.PIN_16, p.PIN_19, p.PIN_18); | ||
| 54 | let spi = Spi::new(p.SPI0, clk, mosi, miso, p.DMA_CH0, p.DMA_CH1, spi_cfg); | ||
| 55 | let cs = Output::new(p.PIN_17, Level::High); | ||
| 56 | let w5500_int = Input::new(p.PIN_21, Pull::Up); | ||
| 57 | let w5500_reset = Output::new(p.PIN_20, Level::High); | ||
| 58 | |||
| 59 | let mac_addr = [0x02, 0x00, 0x00, 0x00, 0x00, 0x00]; | ||
| 60 | let state = singleton!(State::<8, 8>::new()); | ||
| 61 | let (device, runner) = embassy_net_w5500::new( | ||
| 62 | mac_addr, | ||
| 63 | state, | ||
| 64 | ExclusiveDevice::new(spi, cs), | ||
| 65 | w5500_int, | ||
| 66 | w5500_reset, | ||
| 67 | ) | ||
| 68 | .await; | ||
| 69 | unwrap!(spawner.spawn(ethernet_task(runner))); | ||
| 70 | |||
| 71 | // Generate random seed | ||
| 72 | let seed = rng.next_u64(); | ||
| 73 | |||
| 74 | // Init network stack | ||
| 75 | let stack = &*singleton!(Stack::new( | ||
| 76 | device, | ||
| 77 | embassy_net::Config::Dhcp(Default::default()), | ||
| 78 | singleton!(StackResources::<3>::new()), | ||
| 79 | seed | ||
| 80 | )); | ||
| 81 | |||
| 82 | // Launch network task | ||
| 83 | unwrap!(spawner.spawn(net_task(&stack))); | ||
| 84 | |||
| 85 | info!("Waiting for DHCP..."); | ||
| 86 | let cfg = wait_for_config(stack).await; | ||
| 87 | let local_addr = cfg.address.address(); | ||
| 88 | info!("IP address: {:?}", local_addr); | ||
| 89 | |||
| 90 | // Create two sockets listening to the same port, to handle simultaneous connections | ||
| 91 | unwrap!(spawner.spawn(listen_task(&stack, 0, 1234))); | ||
| 92 | unwrap!(spawner.spawn(listen_task(&stack, 1, 1234))); | ||
| 93 | } | ||
| 94 | |||
| 95 | #[embassy_executor::task(pool_size = 2)] | ||
| 96 | async fn listen_task(stack: &'static Stack<Device<'static>>, id: u8, port: u16) { | ||
| 97 | let mut rx_buffer = [0; 4096]; | ||
| 98 | let mut tx_buffer = [0; 4096]; | ||
| 99 | let mut buf = [0; 4096]; | ||
| 100 | loop { | ||
| 101 | let mut socket = embassy_net::tcp::TcpSocket::new(stack, &mut rx_buffer, &mut tx_buffer); | ||
| 102 | socket.set_timeout(Some(embassy_net::SmolDuration::from_secs(10))); | ||
| 103 | |||
| 104 | info!("SOCKET {}: Listening on TCP:{}...", id, port); | ||
| 105 | if let Err(e) = socket.accept(port).await { | ||
| 106 | warn!("accept error: {:?}", e); | ||
| 107 | continue; | ||
| 108 | } | ||
| 109 | info!( | ||
| 110 | "SOCKET {}: Received connection from {:?}", | ||
| 111 | id, | ||
| 112 | socket.remote_endpoint() | ||
| 113 | ); | ||
| 114 | |||
| 115 | loop { | ||
| 116 | let n = match socket.read(&mut buf).await { | ||
| 117 | Ok(0) => { | ||
| 118 | warn!("read EOF"); | ||
| 119 | break; | ||
| 120 | } | ||
| 121 | Ok(n) => n, | ||
| 122 | Err(e) => { | ||
| 123 | warn!("SOCKET {}: {:?}", id, e); | ||
| 124 | break; | ||
| 125 | } | ||
| 126 | }; | ||
| 127 | info!( | ||
| 128 | "SOCKET {}: rxd {}", | ||
| 129 | id, | ||
| 130 | core::str::from_utf8(&buf[..n]).unwrap() | ||
| 131 | ); | ||
| 132 | |||
| 133 | if let Err(e) = socket.write_all(&buf[..n]).await { | ||
| 134 | warn!("write error: {:?}", e); | ||
| 135 | break; | ||
| 136 | } | ||
| 137 | } | ||
| 138 | } | ||
| 139 | } | ||
| 140 | |||
| 141 | async fn wait_for_config(stack: &'static Stack<Device<'static>>) -> embassy_net::StaticConfig { | ||
| 142 | loop { | ||
| 143 | if let Some(config) = stack.config() { | ||
| 144 | return config.clone(); | ||
| 145 | } | ||
| 146 | yield_now().await; | ||
| 147 | } | ||
| 148 | } | ||
diff --git a/examples/src/bin/tcp-client.rs b/examples/src/bin/tcp-client.rs new file mode 100644 index 000000000..32dfb6a68 --- /dev/null +++ b/examples/src/bin/tcp-client.rs | |||
| @@ -0,0 +1,128 @@ | |||
| 1 | #![no_std] | ||
| 2 | #![no_main] | ||
| 3 | #![feature(type_alias_impl_trait)] | ||
| 4 | |||
| 5 | use core::str::FromStr; | ||
| 6 | use defmt::*; | ||
| 7 | use embassy_executor::Spawner; | ||
| 8 | use embassy_futures::yield_now; | ||
| 9 | use embassy_net::{Stack, StackResources}; | ||
| 10 | use embassy_net_w5500::*; | ||
| 11 | use embassy_rp::clocks::RoscRng; | ||
| 12 | use embassy_rp::gpio::{Input, Level, Output, Pull}; | ||
| 13 | use embassy_rp::peripherals::{PIN_17, PIN_20, PIN_21, SPI0}; | ||
| 14 | use embassy_rp::spi::{Async, Config as SpiConfig, Spi}; | ||
| 15 | use embassy_time::{Duration, Timer}; | ||
| 16 | use embedded_hal_async::spi::ExclusiveDevice; | ||
| 17 | use embedded_io::asynch::Write; | ||
| 18 | use rand::RngCore; | ||
| 19 | use static_cell::StaticCell; | ||
| 20 | use {defmt_rtt as _, panic_probe as _}; | ||
| 21 | |||
| 22 | macro_rules! singleton { | ||
| 23 | ($val:expr) => {{ | ||
| 24 | type T = impl Sized; | ||
| 25 | static STATIC_CELL: StaticCell<T> = StaticCell::new(); | ||
| 26 | let (x,) = STATIC_CELL.init(($val,)); | ||
| 27 | x | ||
| 28 | }}; | ||
| 29 | } | ||
| 30 | |||
| 31 | #[embassy_executor::task] | ||
| 32 | async fn ethernet_task( | ||
| 33 | runner: Runner< | ||
| 34 | 'static, | ||
| 35 | ExclusiveDevice<Spi<'static, SPI0, Async>, Output<'static, PIN_17>>, | ||
| 36 | Input<'static, PIN_21>, | ||
| 37 | Output<'static, PIN_20>, | ||
| 38 | >, | ||
| 39 | ) -> ! { | ||
| 40 | runner.run().await | ||
| 41 | } | ||
| 42 | |||
| 43 | #[embassy_executor::task] | ||
| 44 | async fn net_task(stack: &'static Stack<Device<'static>>) -> ! { | ||
| 45 | stack.run().await | ||
| 46 | } | ||
| 47 | |||
| 48 | #[embassy_executor::main] | ||
| 49 | async fn main(spawner: Spawner) { | ||
| 50 | let p = embassy_rp::init(Default::default()); | ||
| 51 | let mut rng = RoscRng; | ||
| 52 | let mut led = Output::new(p.PIN_25, Level::Low); | ||
| 53 | |||
| 54 | let mut spi_cfg = SpiConfig::default(); | ||
| 55 | spi_cfg.frequency = 50_000_000; | ||
| 56 | let (miso, mosi, clk) = (p.PIN_16, p.PIN_19, p.PIN_18); | ||
| 57 | let spi = Spi::new(p.SPI0, clk, mosi, miso, p.DMA_CH0, p.DMA_CH1, spi_cfg); | ||
| 58 | let cs = Output::new(p.PIN_17, Level::High); | ||
| 59 | let w5500_int = Input::new(p.PIN_21, Pull::Up); | ||
| 60 | let w5500_reset = Output::new(p.PIN_20, Level::High); | ||
| 61 | |||
| 62 | let mac_addr = [0x02, 0x00, 0x00, 0x00, 0x00, 0x00]; | ||
| 63 | let state = singleton!(State::<8, 8>::new()); | ||
| 64 | let (device, runner) = embassy_net_w5500::new( | ||
| 65 | mac_addr, | ||
| 66 | state, | ||
| 67 | ExclusiveDevice::new(spi, cs), | ||
| 68 | w5500_int, | ||
| 69 | w5500_reset, | ||
| 70 | ) | ||
| 71 | .await; | ||
| 72 | unwrap!(spawner.spawn(ethernet_task(runner))); | ||
| 73 | |||
| 74 | // Generate random seed | ||
| 75 | let seed = rng.next_u64(); | ||
| 76 | |||
| 77 | // Init network stack | ||
| 78 | let stack = &*singleton!(Stack::new( | ||
| 79 | device, | ||
| 80 | embassy_net::Config::Dhcp(Default::default()), | ||
| 81 | singleton!(StackResources::<2>::new()), | ||
| 82 | seed | ||
| 83 | )); | ||
| 84 | |||
| 85 | // Launch network task | ||
| 86 | unwrap!(spawner.spawn(net_task(&stack))); | ||
| 87 | |||
| 88 | info!("Waiting for DHCP..."); | ||
| 89 | let cfg = wait_for_config(stack).await; | ||
| 90 | let local_addr = cfg.address.address(); | ||
| 91 | info!("IP address: {:?}", local_addr); | ||
| 92 | |||
| 93 | let mut rx_buffer = [0; 4096]; | ||
| 94 | let mut tx_buffer = [0; 4096]; | ||
| 95 | loop { | ||
| 96 | let mut socket = embassy_net::tcp::TcpSocket::new(stack, &mut rx_buffer, &mut tx_buffer); | ||
| 97 | socket.set_timeout(Some(embassy_net::SmolDuration::from_secs(10))); | ||
| 98 | |||
| 99 | led.set_low(); | ||
| 100 | info!("Connecting..."); | ||
| 101 | let host_addr = embassy_net::Ipv4Address::from_str("192.168.1.110").unwrap(); | ||
| 102 | if let Err(e) = socket.connect((host_addr, 1234)).await { | ||
| 103 | warn!("connect error: {:?}", e); | ||
| 104 | continue; | ||
| 105 | } | ||
| 106 | info!("Connected to {:?}", socket.remote_endpoint()); | ||
| 107 | led.set_high(); | ||
| 108 | |||
| 109 | let msg = b"Hello world!\n"; | ||
| 110 | loop { | ||
| 111 | if let Err(e) = socket.write_all(msg).await { | ||
| 112 | warn!("write error: {:?}", e); | ||
| 113 | break; | ||
| 114 | } | ||
| 115 | info!("txd: {}", core::str::from_utf8(msg).unwrap()); | ||
| 116 | Timer::after(Duration::from_secs(1)).await; | ||
| 117 | } | ||
| 118 | } | ||
| 119 | } | ||
| 120 | |||
| 121 | async fn wait_for_config(stack: &'static Stack<Device<'static>>) -> embassy_net::StaticConfig { | ||
| 122 | loop { | ||
| 123 | if let Some(config) = stack.config() { | ||
| 124 | return config.clone(); | ||
| 125 | } | ||
| 126 | yield_now().await; | ||
| 127 | } | ||
| 128 | } | ||
diff --git a/examples/src/bin/tcp-server.rs b/examples/src/bin/tcp-server.rs new file mode 100644 index 000000000..04b220146 --- /dev/null +++ b/examples/src/bin/tcp-server.rs | |||
| @@ -0,0 +1,136 @@ | |||
| 1 | #![no_std] | ||
| 2 | #![no_main] | ||
| 3 | #![feature(type_alias_impl_trait)] | ||
| 4 | |||
| 5 | use defmt::*; | ||
| 6 | use embassy_executor::Spawner; | ||
| 7 | use embassy_futures::yield_now; | ||
| 8 | use embassy_net::{Stack, StackResources}; | ||
| 9 | use embassy_net_w5500::*; | ||
| 10 | use embassy_rp::clocks::RoscRng; | ||
| 11 | use embassy_rp::gpio::{Input, Level, Output, Pull}; | ||
| 12 | use embassy_rp::peripherals::{PIN_17, PIN_20, PIN_21, SPI0}; | ||
| 13 | use embassy_rp::spi::{Async, Config as SpiConfig, Spi}; | ||
| 14 | use embedded_hal_async::spi::ExclusiveDevice; | ||
| 15 | use embedded_io::asynch::Write; | ||
| 16 | use rand::RngCore; | ||
| 17 | use static_cell::StaticCell; | ||
| 18 | use {defmt_rtt as _, panic_probe as _}; | ||
| 19 | |||
| 20 | macro_rules! singleton { | ||
| 21 | ($val:expr) => {{ | ||
| 22 | type T = impl Sized; | ||
| 23 | static STATIC_CELL: StaticCell<T> = StaticCell::new(); | ||
| 24 | let (x,) = STATIC_CELL.init(($val,)); | ||
| 25 | x | ||
| 26 | }}; | ||
| 27 | } | ||
| 28 | |||
| 29 | #[embassy_executor::task] | ||
| 30 | async fn ethernet_task( | ||
| 31 | runner: Runner< | ||
| 32 | 'static, | ||
| 33 | ExclusiveDevice<Spi<'static, SPI0, Async>, Output<'static, PIN_17>>, | ||
| 34 | Input<'static, PIN_21>, | ||
| 35 | Output<'static, PIN_20>, | ||
| 36 | >, | ||
| 37 | ) -> ! { | ||
| 38 | runner.run().await | ||
| 39 | } | ||
| 40 | |||
| 41 | #[embassy_executor::task] | ||
| 42 | async fn net_task(stack: &'static Stack<Device<'static>>) -> ! { | ||
| 43 | stack.run().await | ||
| 44 | } | ||
| 45 | |||
| 46 | #[embassy_executor::main] | ||
| 47 | async fn main(spawner: Spawner) { | ||
| 48 | let p = embassy_rp::init(Default::default()); | ||
| 49 | let mut rng = RoscRng; | ||
| 50 | let mut led = Output::new(p.PIN_25, Level::Low); | ||
| 51 | |||
| 52 | let mut spi_cfg = SpiConfig::default(); | ||
| 53 | spi_cfg.frequency = 50_000_000; | ||
| 54 | let (miso, mosi, clk) = (p.PIN_16, p.PIN_19, p.PIN_18); | ||
| 55 | let spi = Spi::new(p.SPI0, clk, mosi, miso, p.DMA_CH0, p.DMA_CH1, spi_cfg); | ||
| 56 | let cs = Output::new(p.PIN_17, Level::High); | ||
| 57 | let w5500_int = Input::new(p.PIN_21, Pull::Up); | ||
| 58 | let w5500_reset = Output::new(p.PIN_20, Level::High); | ||
| 59 | |||
| 60 | let mac_addr = [0x02, 0x00, 0x00, 0x00, 0x00, 0x00]; | ||
| 61 | let state = singleton!(State::<8, 8>::new()); | ||
| 62 | let (device, runner) = embassy_net_w5500::new( | ||
| 63 | mac_addr, | ||
| 64 | state, | ||
| 65 | ExclusiveDevice::new(spi, cs), | ||
| 66 | w5500_int, | ||
| 67 | w5500_reset, | ||
| 68 | ) | ||
| 69 | .await; | ||
| 70 | unwrap!(spawner.spawn(ethernet_task(runner))); | ||
| 71 | |||
| 72 | // Generate random seed | ||
| 73 | let seed = rng.next_u64(); | ||
| 74 | |||
| 75 | // Init network stack | ||
| 76 | let stack = &*singleton!(Stack::new( | ||
| 77 | device, | ||
| 78 | embassy_net::Config::Dhcp(Default::default()), | ||
| 79 | singleton!(StackResources::<2>::new()), | ||
| 80 | seed | ||
| 81 | )); | ||
| 82 | |||
| 83 | // Launch network task | ||
| 84 | unwrap!(spawner.spawn(net_task(&stack))); | ||
| 85 | |||
| 86 | info!("Waiting for DHCP..."); | ||
| 87 | let cfg = wait_for_config(stack).await; | ||
| 88 | let local_addr = cfg.address.address(); | ||
| 89 | info!("IP address: {:?}", local_addr); | ||
| 90 | |||
| 91 | let mut rx_buffer = [0; 4096]; | ||
| 92 | let mut tx_buffer = [0; 4096]; | ||
| 93 | let mut buf = [0; 4096]; | ||
| 94 | loop { | ||
| 95 | let mut socket = embassy_net::tcp::TcpSocket::new(stack, &mut rx_buffer, &mut tx_buffer); | ||
| 96 | socket.set_timeout(Some(embassy_net::SmolDuration::from_secs(10))); | ||
| 97 | |||
| 98 | led.set_low(); | ||
| 99 | info!("Listening on TCP:1234..."); | ||
| 100 | if let Err(e) = socket.accept(1234).await { | ||
| 101 | warn!("accept error: {:?}", e); | ||
| 102 | continue; | ||
| 103 | } | ||
| 104 | info!("Received connection from {:?}", socket.remote_endpoint()); | ||
| 105 | led.set_high(); | ||
| 106 | |||
| 107 | loop { | ||
| 108 | let n = match socket.read(&mut buf).await { | ||
| 109 | Ok(0) => { | ||
| 110 | warn!("read EOF"); | ||
| 111 | break; | ||
| 112 | } | ||
| 113 | Ok(n) => n, | ||
| 114 | Err(e) => { | ||
| 115 | warn!("{:?}", e); | ||
| 116 | break; | ||
| 117 | } | ||
| 118 | }; | ||
| 119 | info!("rxd {}", core::str::from_utf8(&buf[..n]).unwrap()); | ||
| 120 | |||
| 121 | if let Err(e) = socket.write_all(&buf[..n]).await { | ||
| 122 | warn!("write error: {:?}", e); | ||
| 123 | break; | ||
| 124 | } | ||
| 125 | } | ||
| 126 | } | ||
| 127 | } | ||
| 128 | |||
| 129 | async fn wait_for_config(stack: &'static Stack<Device<'static>>) -> embassy_net::StaticConfig { | ||
| 130 | loop { | ||
| 131 | if let Some(config) = stack.config() { | ||
| 132 | return config.clone(); | ||
| 133 | } | ||
| 134 | yield_now().await; | ||
| 135 | } | ||
| 136 | } | ||
diff --git a/examples/src/bin/udp.rs b/examples/src/bin/udp.rs new file mode 100644 index 000000000..4dc5e1f2f --- /dev/null +++ b/examples/src/bin/udp.rs | |||
| @@ -0,0 +1,123 @@ | |||
| 1 | #![no_std] | ||
| 2 | #![no_main] | ||
| 3 | #![feature(type_alias_impl_trait)] | ||
| 4 | |||
| 5 | use defmt::*; | ||
| 6 | use embassy_executor::Spawner; | ||
| 7 | use embassy_futures::yield_now; | ||
| 8 | use embassy_net::udp::UdpSocket; | ||
| 9 | use embassy_net::{PacketMetadata, Stack, StackResources}; | ||
| 10 | use embassy_net_w5500::*; | ||
| 11 | use embassy_rp::clocks::RoscRng; | ||
| 12 | use embassy_rp::gpio::{Input, Level, Output, Pull}; | ||
| 13 | use embassy_rp::peripherals::{PIN_17, PIN_20, PIN_21, SPI0}; | ||
| 14 | use embassy_rp::spi::{Async, Config as SpiConfig, Spi}; | ||
| 15 | use embedded_hal_async::spi::ExclusiveDevice; | ||
| 16 | use rand::RngCore; | ||
| 17 | use static_cell::StaticCell; | ||
| 18 | use {defmt_rtt as _, panic_probe as _}; | ||
| 19 | |||
| 20 | macro_rules! singleton { | ||
| 21 | ($val:expr) => {{ | ||
| 22 | type T = impl Sized; | ||
| 23 | static STATIC_CELL: StaticCell<T> = StaticCell::new(); | ||
| 24 | let (x,) = STATIC_CELL.init(($val,)); | ||
| 25 | x | ||
| 26 | }}; | ||
| 27 | } | ||
| 28 | |||
| 29 | #[embassy_executor::task] | ||
| 30 | async fn ethernet_task( | ||
| 31 | runner: Runner< | ||
| 32 | 'static, | ||
| 33 | ExclusiveDevice<Spi<'static, SPI0, Async>, Output<'static, PIN_17>>, | ||
| 34 | Input<'static, PIN_21>, | ||
| 35 | Output<'static, PIN_20>, | ||
| 36 | >, | ||
| 37 | ) -> ! { | ||
| 38 | runner.run().await | ||
| 39 | } | ||
| 40 | |||
| 41 | #[embassy_executor::task] | ||
| 42 | async fn net_task(stack: &'static Stack<Device<'static>>) -> ! { | ||
| 43 | stack.run().await | ||
| 44 | } | ||
| 45 | |||
| 46 | #[embassy_executor::main] | ||
| 47 | async fn main(spawner: Spawner) { | ||
| 48 | let p = embassy_rp::init(Default::default()); | ||
| 49 | let mut rng = RoscRng; | ||
| 50 | |||
| 51 | let mut spi_cfg = SpiConfig::default(); | ||
| 52 | spi_cfg.frequency = 50_000_000; | ||
| 53 | let (miso, mosi, clk) = (p.PIN_16, p.PIN_19, p.PIN_18); | ||
| 54 | let spi = Spi::new(p.SPI0, clk, mosi, miso, p.DMA_CH0, p.DMA_CH1, spi_cfg); | ||
| 55 | let cs = Output::new(p.PIN_17, Level::High); | ||
| 56 | let w5500_int = Input::new(p.PIN_21, Pull::Up); | ||
| 57 | let w5500_reset = Output::new(p.PIN_20, Level::High); | ||
| 58 | |||
| 59 | let mac_addr = [0x02, 0x00, 0x00, 0x00, 0x00, 0x00]; | ||
| 60 | let state = singleton!(State::<8, 8>::new()); | ||
| 61 | let (device, runner) = embassy_net_w5500::new( | ||
| 62 | mac_addr, | ||
| 63 | state, | ||
| 64 | ExclusiveDevice::new(spi, cs), | ||
| 65 | w5500_int, | ||
| 66 | w5500_reset, | ||
| 67 | ) | ||
| 68 | .await; | ||
| 69 | unwrap!(spawner.spawn(ethernet_task(runner))); | ||
| 70 | |||
| 71 | // Generate random seed | ||
| 72 | let seed = rng.next_u64(); | ||
| 73 | |||
| 74 | // Init network stack | ||
| 75 | let stack = &*singleton!(Stack::new( | ||
| 76 | device, | ||
| 77 | embassy_net::Config::Dhcp(Default::default()), | ||
| 78 | singleton!(StackResources::<2>::new()), | ||
| 79 | seed | ||
| 80 | )); | ||
| 81 | |||
| 82 | // Launch network task | ||
| 83 | unwrap!(spawner.spawn(net_task(&stack))); | ||
| 84 | |||
| 85 | info!("Waiting for DHCP..."); | ||
| 86 | let cfg = wait_for_config(stack).await; | ||
| 87 | let local_addr = cfg.address.address(); | ||
| 88 | info!("IP address: {:?}", local_addr); | ||
| 89 | |||
| 90 | // Then we can use it! | ||
| 91 | let mut rx_buffer = [0; 4096]; | ||
| 92 | let mut tx_buffer = [0; 4096]; | ||
| 93 | let mut rx_meta = [PacketMetadata::EMPTY; 16]; | ||
| 94 | let mut tx_meta = [PacketMetadata::EMPTY; 16]; | ||
| 95 | let mut buf = [0; 4096]; | ||
| 96 | loop { | ||
| 97 | let mut socket = UdpSocket::new( | ||
| 98 | stack, | ||
| 99 | &mut rx_meta, | ||
| 100 | &mut rx_buffer, | ||
| 101 | &mut tx_meta, | ||
| 102 | &mut tx_buffer, | ||
| 103 | ); | ||
| 104 | socket.bind(1234).unwrap(); | ||
| 105 | |||
| 106 | loop { | ||
| 107 | let (n, ep) = socket.recv_from(&mut buf).await.unwrap(); | ||
| 108 | if let Ok(s) = core::str::from_utf8(&buf[..n]) { | ||
| 109 | info!("rxd from {}: {}", ep, s); | ||
| 110 | } | ||
| 111 | socket.send_to(&buf[..n], ep).await.unwrap(); | ||
| 112 | } | ||
| 113 | } | ||
| 114 | } | ||
| 115 | |||
| 116 | async fn wait_for_config(stack: &'static Stack<Device<'static>>) -> embassy_net::StaticConfig { | ||
| 117 | loop { | ||
| 118 | if let Some(config) = stack.config() { | ||
| 119 | return config.clone(); | ||
| 120 | } | ||
| 121 | yield_now().await; | ||
| 122 | } | ||
| 123 | } | ||
