aboutsummaryrefslogtreecommitdiff
path: root/examples/src
diff options
context:
space:
mode:
authorkalkyl <[email protected]>2023-05-09 01:51:08 +0200
committerkalkyl <[email protected]>2023-05-09 01:51:08 +0200
commit72b0379125b87bcd274bdb81127dd5f0ab29d661 (patch)
treeb620d67bc9ea930051668d63190dd6de62485b0a /examples/src
:rainbow:
Diffstat (limited to 'examples/src')
-rw-r--r--examples/src/bin/multisocket.rs148
-rw-r--r--examples/src/bin/tcp-client.rs128
-rw-r--r--examples/src/bin/tcp-server.rs136
-rw-r--r--examples/src/bin/udp.rs123
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
5use defmt::*;
6use embassy_executor::Spawner;
7use embassy_futures::yield_now;
8use embassy_net::{Stack, StackResources};
9use embassy_net_w5500::*;
10use embassy_rp::clocks::RoscRng;
11use embassy_rp::gpio::{Input, Level, Output, Pull};
12use embassy_rp::peripherals::{PIN_17, PIN_20, PIN_21, SPI0};
13use embassy_rp::spi::{Async, Config as SpiConfig, Spi};
14use embedded_hal_async::spi::ExclusiveDevice;
15use embedded_io::asynch::Write;
16use rand::RngCore;
17use static_cell::StaticCell;
18use {defmt_rtt as _, panic_probe as _};
19
20macro_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]
30async 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]
42async fn net_task(stack: &'static Stack<Device<'static>>) -> ! {
43 stack.run().await
44}
45
46#[embassy_executor::main]
47async 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)]
96async 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
141async 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
5use core::str::FromStr;
6use defmt::*;
7use embassy_executor::Spawner;
8use embassy_futures::yield_now;
9use embassy_net::{Stack, StackResources};
10use embassy_net_w5500::*;
11use embassy_rp::clocks::RoscRng;
12use embassy_rp::gpio::{Input, Level, Output, Pull};
13use embassy_rp::peripherals::{PIN_17, PIN_20, PIN_21, SPI0};
14use embassy_rp::spi::{Async, Config as SpiConfig, Spi};
15use embassy_time::{Duration, Timer};
16use embedded_hal_async::spi::ExclusiveDevice;
17use embedded_io::asynch::Write;
18use rand::RngCore;
19use static_cell::StaticCell;
20use {defmt_rtt as _, panic_probe as _};
21
22macro_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]
32async 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]
44async fn net_task(stack: &'static Stack<Device<'static>>) -> ! {
45 stack.run().await
46}
47
48#[embassy_executor::main]
49async 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
121async 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
5use defmt::*;
6use embassy_executor::Spawner;
7use embassy_futures::yield_now;
8use embassy_net::{Stack, StackResources};
9use embassy_net_w5500::*;
10use embassy_rp::clocks::RoscRng;
11use embassy_rp::gpio::{Input, Level, Output, Pull};
12use embassy_rp::peripherals::{PIN_17, PIN_20, PIN_21, SPI0};
13use embassy_rp::spi::{Async, Config as SpiConfig, Spi};
14use embedded_hal_async::spi::ExclusiveDevice;
15use embedded_io::asynch::Write;
16use rand::RngCore;
17use static_cell::StaticCell;
18use {defmt_rtt as _, panic_probe as _};
19
20macro_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]
30async 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]
42async fn net_task(stack: &'static Stack<Device<'static>>) -> ! {
43 stack.run().await
44}
45
46#[embassy_executor::main]
47async 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
129async 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
5use defmt::*;
6use embassy_executor::Spawner;
7use embassy_futures::yield_now;
8use embassy_net::udp::UdpSocket;
9use embassy_net::{PacketMetadata, Stack, StackResources};
10use embassy_net_w5500::*;
11use embassy_rp::clocks::RoscRng;
12use embassy_rp::gpio::{Input, Level, Output, Pull};
13use embassy_rp::peripherals::{PIN_17, PIN_20, PIN_21, SPI0};
14use embassy_rp::spi::{Async, Config as SpiConfig, Spi};
15use embedded_hal_async::spi::ExclusiveDevice;
16use rand::RngCore;
17use static_cell::StaticCell;
18use {defmt_rtt as _, panic_probe as _};
19
20macro_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]
30async 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]
42async fn net_task(stack: &'static Stack<Device<'static>>) -> ! {
43 stack.run().await
44}
45
46#[embassy_executor::main]
47async 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
116async 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}