diff options
Diffstat (limited to 'examples/std/src/bin/net_dns.rs')
| -rw-r--r-- | examples/std/src/bin/net_dns.rs | 94 |
1 files changed, 94 insertions, 0 deletions
diff --git a/examples/std/src/bin/net_dns.rs b/examples/std/src/bin/net_dns.rs new file mode 100644 index 000000000..65b5a2cd9 --- /dev/null +++ b/examples/std/src/bin/net_dns.rs | |||
| @@ -0,0 +1,94 @@ | |||
| 1 | #![feature(type_alias_impl_trait)] | ||
| 2 | |||
| 3 | use std::default::Default; | ||
| 4 | |||
| 5 | use clap::Parser; | ||
| 6 | use embassy_executor::{Executor, Spawner}; | ||
| 7 | use embassy_net::dns::DnsQueryType; | ||
| 8 | use embassy_net::{Config, Ipv4Address, Ipv4Cidr, Stack, StackResources}; | ||
| 9 | use heapless::Vec; | ||
| 10 | use log::*; | ||
| 11 | use rand_core::{OsRng, RngCore}; | ||
| 12 | use static_cell::{make_static, StaticCell}; | ||
| 13 | |||
| 14 | #[path = "../tuntap.rs"] | ||
| 15 | mod tuntap; | ||
| 16 | |||
| 17 | use crate::tuntap::TunTapDevice; | ||
| 18 | #[derive(Parser)] | ||
| 19 | #[clap(version = "1.0")] | ||
| 20 | struct Opts { | ||
| 21 | /// TAP device name | ||
| 22 | #[clap(long, default_value = "tap0")] | ||
| 23 | tap: String, | ||
| 24 | /// use a static IP instead of DHCP | ||
| 25 | #[clap(long)] | ||
| 26 | static_ip: bool, | ||
| 27 | } | ||
| 28 | |||
| 29 | #[embassy_executor::task] | ||
| 30 | async fn net_task(stack: &'static Stack<TunTapDevice>) -> ! { | ||
| 31 | stack.run().await | ||
| 32 | } | ||
| 33 | |||
| 34 | #[embassy_executor::task] | ||
| 35 | async fn main_task(spawner: Spawner) { | ||
| 36 | let opts: Opts = Opts::parse(); | ||
| 37 | |||
| 38 | // Init network device | ||
| 39 | let device = TunTapDevice::new(&opts.tap).unwrap(); | ||
| 40 | |||
| 41 | // Choose between dhcp or static ip | ||
| 42 | let config = if opts.static_ip { | ||
| 43 | Config::ipv4_static(embassy_net::StaticConfigV4 { | ||
| 44 | address: Ipv4Cidr::new(Ipv4Address::new(192, 168, 69, 1), 24), | ||
| 45 | dns_servers: Vec::from_slice(&[Ipv4Address::new(8, 8, 4, 4).into(), Ipv4Address::new(8, 8, 8, 8).into()]) | ||
| 46 | .unwrap(), | ||
| 47 | gateway: Some(Ipv4Address::new(192, 168, 69, 100)), | ||
| 48 | }) | ||
| 49 | } else { | ||
| 50 | Config::dhcpv4(Default::default()) | ||
| 51 | }; | ||
| 52 | |||
| 53 | // Generate random seed | ||
| 54 | let mut seed = [0; 8]; | ||
| 55 | OsRng.fill_bytes(&mut seed); | ||
| 56 | let seed = u64::from_le_bytes(seed); | ||
| 57 | |||
| 58 | // Init network stack | ||
| 59 | let stack: &Stack<_> = &*make_static!(Stack::new( | ||
| 60 | device, | ||
| 61 | config, | ||
| 62 | make_static!(StackResources::<3>::new()), | ||
| 63 | seed | ||
| 64 | )); | ||
| 65 | |||
| 66 | // Launch network task | ||
| 67 | spawner.spawn(net_task(stack)).unwrap(); | ||
| 68 | |||
| 69 | let host = "example.com"; | ||
| 70 | info!("querying host {:?}...", host); | ||
| 71 | match stack.dns_query(host, DnsQueryType::A).await { | ||
| 72 | Ok(r) => { | ||
| 73 | info!("query response: {:?}", r); | ||
| 74 | } | ||
| 75 | Err(e) => { | ||
| 76 | warn!("query error: {:?}", e); | ||
| 77 | } | ||
| 78 | }; | ||
| 79 | } | ||
| 80 | |||
| 81 | static EXECUTOR: StaticCell<Executor> = StaticCell::new(); | ||
| 82 | |||
| 83 | fn main() { | ||
| 84 | env_logger::builder() | ||
| 85 | .filter_level(log::LevelFilter::Debug) | ||
| 86 | .filter_module("async_io", log::LevelFilter::Info) | ||
| 87 | .format_timestamp_nanos() | ||
| 88 | .init(); | ||
| 89 | |||
| 90 | let executor = EXECUTOR.init(Executor::new()); | ||
| 91 | executor.run(|spawner| { | ||
| 92 | spawner.spawn(main_task(spawner)).unwrap(); | ||
| 93 | }); | ||
| 94 | } | ||
