diff options
Diffstat (limited to 'examples/std/src/bin/net.rs')
| -rw-r--r-- | examples/std/src/bin/net.rs | 103 |
1 files changed, 103 insertions, 0 deletions
diff --git a/examples/std/src/bin/net.rs b/examples/std/src/bin/net.rs new file mode 100644 index 000000000..5a726e5d2 --- /dev/null +++ b/examples/std/src/bin/net.rs | |||
| @@ -0,0 +1,103 @@ | |||
| 1 | #![feature(type_alias_impl_trait)] | ||
| 2 | #![feature(min_type_alias_impl_trait)] | ||
| 3 | #![feature(impl_trait_in_bindings)] | ||
| 4 | #![allow(incomplete_features)] | ||
| 5 | |||
| 6 | use clap::{AppSettings, Clap}; | ||
| 7 | use embassy::executor::Spawner; | ||
| 8 | use embassy::io::AsyncWriteExt; | ||
| 9 | use embassy::util::Forever; | ||
| 10 | use embassy_net::*; | ||
| 11 | use embassy_std::Executor; | ||
| 12 | use heapless::Vec; | ||
| 13 | use log::*; | ||
| 14 | |||
| 15 | #[path = "../tuntap.rs"] | ||
| 16 | mod tuntap; | ||
| 17 | |||
| 18 | use crate::tuntap::TunTapDevice; | ||
| 19 | |||
| 20 | static DEVICE: Forever<TunTapDevice> = Forever::new(); | ||
| 21 | static CONFIG: Forever<DhcpConfigurator> = Forever::new(); | ||
| 22 | |||
| 23 | #[derive(Clap)] | ||
| 24 | #[clap(version = "1.0")] | ||
| 25 | #[clap(setting = AppSettings::ColoredHelp)] | ||
| 26 | struct Opts { | ||
| 27 | /// TAP device name | ||
| 28 | #[clap(long, default_value = "tap0")] | ||
| 29 | tap: String, | ||
| 30 | } | ||
| 31 | |||
| 32 | #[embassy::task] | ||
| 33 | async fn net_task() { | ||
| 34 | embassy_net::run().await | ||
| 35 | } | ||
| 36 | |||
| 37 | #[embassy::task] | ||
| 38 | async fn main_task(spawner: Spawner) { | ||
| 39 | let opts: Opts = Opts::parse(); | ||
| 40 | |||
| 41 | // Init network device | ||
| 42 | let device = TunTapDevice::new(&opts.tap).unwrap(); | ||
| 43 | |||
| 44 | // Static IP configuration | ||
| 45 | let config = StaticConfigurator::new(Config { | ||
| 46 | address: Ipv4Cidr::new(Ipv4Address::new(192, 168, 69, 2), 24), | ||
| 47 | dns_servers: Vec::new(), | ||
| 48 | gateway: Some(Ipv4Address::new(192, 168, 69, 1)), | ||
| 49 | }); | ||
| 50 | |||
| 51 | // DHCP configruation | ||
| 52 | let config = DhcpConfigurator::new(); | ||
| 53 | |||
| 54 | // Init network stack | ||
| 55 | embassy_net::init(DEVICE.put(device), CONFIG.put(config)); | ||
| 56 | |||
| 57 | // Launch network task | ||
| 58 | spawner.spawn(net_task()).unwrap(); | ||
| 59 | |||
| 60 | // Then we can use it! | ||
| 61 | let mut rx_buffer = [0; 4096]; | ||
| 62 | let mut tx_buffer = [0; 4096]; | ||
| 63 | let mut socket = TcpSocket::new(&mut rx_buffer, &mut tx_buffer); | ||
| 64 | |||
| 65 | socket.set_timeout(Some(embassy_net::SmolDuration::from_secs(10))); | ||
| 66 | |||
| 67 | let remote_endpoint = (Ipv4Address::new(192, 168, 69, 74), 8000); | ||
| 68 | info!("connecting to {:?}...", remote_endpoint); | ||
| 69 | let r = socket.connect(remote_endpoint).await; | ||
| 70 | if let Err(e) = r { | ||
| 71 | warn!("connect error: {:?}", e); | ||
| 72 | return; | ||
| 73 | } | ||
| 74 | info!("connected!"); | ||
| 75 | loop { | ||
| 76 | let r = socket.write_all(b"Hello!\n").await; | ||
| 77 | if let Err(e) = r { | ||
| 78 | warn!("write error: {:?}", e); | ||
| 79 | return; | ||
| 80 | } | ||
| 81 | } | ||
| 82 | } | ||
| 83 | |||
| 84 | #[no_mangle] | ||
| 85 | fn _embassy_rand(buf: &mut [u8]) { | ||
| 86 | use rand_core::{OsRng, RngCore}; | ||
| 87 | OsRng.fill_bytes(buf); | ||
| 88 | } | ||
| 89 | |||
| 90 | static EXECUTOR: Forever<Executor> = Forever::new(); | ||
| 91 | |||
| 92 | fn main() { | ||
| 93 | env_logger::builder() | ||
| 94 | .filter_level(log::LevelFilter::Debug) | ||
| 95 | .filter_module("async_io", log::LevelFilter::Info) | ||
| 96 | .format_timestamp_nanos() | ||
| 97 | .init(); | ||
| 98 | |||
| 99 | let executor = EXECUTOR.put(Executor::new()); | ||
| 100 | executor.run(|spawner| { | ||
| 101 | spawner.spawn(main_task(spawner)).unwrap(); | ||
| 102 | }); | ||
| 103 | } | ||
