kawaiifi 0.2.1

Wi-Fi scanning library for Linux, macOS, and Windows.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120

use std::{collections::HashSet, convert::From, fmt::Display, hash::Hash};

use derive_more::{Deref, DerefMut, From};
use num_enum::TryFromPrimitive;

#[cfg(target_os = "linux")]
use crate::nl80211::ChanWidth;
use crate::{Ie, IeData};

/// A Wi-Fi channel width.
#[derive(Debug, Copy, Clone, PartialEq, Eq, Hash, Ord, PartialOrd, TryFromPrimitive)]
#[repr(u8)]
pub enum ChannelWidth {
    /// A 20 MHz channel.
    TwentyMhz,
    /// A 40 MHz channel.
    FortyMhz,
    /// An 80 MHz channel.
    EightyMhz,
    /// An 80+80 MHz non-contiguous channel.
    EightyPlusEightyMhz,
    /// A 160 MHz channel.
    OneSixtyMhz,
    /// A 320 MHz channel.
    ThreeHundredTwentyMhz,
}

#[cfg(target_os = "linux")]
impl From<ChanWidth> for ChannelWidth {
    fn from(chan_width: ChanWidth) -> Self {
        match chan_width {
            ChanWidth::TwentyMhz => ChannelWidth::TwentyMhz,
            ChanWidth::FortyMhz => ChannelWidth::FortyMhz,
            ChanWidth::EightyMhz => ChannelWidth::EightyMhz,
            ChanWidth::EightyPlusEightyMhz => ChannelWidth::EightyPlusEightyMhz,
            ChanWidth::OneSixtyMhz => ChannelWidth::OneSixtyMhz,
            ChanWidth::ThreeHundredTwentyMhz => ChannelWidth::ThreeHundredTwentyMhz,
            _ => ChannelWidth::TwentyMhz,
        }
    }
}

impl From<&[Ie]> for ChannelWidth {
    // From Table 11-24 in IEEE Std 802.11-2016
    fn from(ies: &[Ie]) -> Self {
        // Iterate once through the IEs and find the IEs that report the BSSs channel width
        let (mut eht_op, mut he_op, mut vht_op, mut ht_op) = (None, None, None, None);

        for ie in ies {
            match &ie.data {
                IeData::EhtOperation(op) => eht_op = Some(op),
                IeData::HeOperation(op) => he_op = Some(op),
                IeData::VhtOperation(op) => vht_op = Some(op),
                IeData::HtOperation(op) => ht_op = Some(op),
                _ => continue,
            }
        }

        // Check for an EHT Operation element
        if let Some(eht_op) = eht_op
            && let Some(channel_width) = eht_op.channel_width()
        {
            return channel_width;
        }

        // Check for an HE Operation element
        if let Some(he_op) = he_op
            && let Some(channel_width) = he_op.channel_width()
        {
            return channel_width;
        }

        // Check for a VHT Operation element
        if let Some(vht_op) = vht_op
            && let Some(channel_width) = vht_op.channel_width()
        {
            return channel_width;
        }

        // Check for an HT Operation element
        if let Some(ht_op) = ht_op {
            return ht_op.channel_width();
        }

        // If there is no EHT/HE/VHT/HT Information
        ChannelWidth::TwentyMhz
    }
}

impl Display for ChannelWidth {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            ChannelWidth::TwentyMhz => write!(f, "20 MHz"),
            ChannelWidth::FortyMhz => write!(f, "40 MHz"),
            ChannelWidth::EightyMhz => write!(f, "80 MHz"),
            ChannelWidth::EightyPlusEightyMhz => write!(f, "80+80 MHz"),
            ChannelWidth::OneSixtyMhz => write!(f, "160 MHz"),
            ChannelWidth::ThreeHundredTwentyMhz => write!(f, "320 MHz"),
        }
    }
}

/// A set of Wi-Fi channel widths.
#[derive(Debug, Default, Clone, PartialEq, Eq, Deref, DerefMut, From)]
#[from(forward)]
pub struct ChannelWidths(HashSet<ChannelWidth>);

impl Display for ChannelWidths {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(
            f,
            "{}",
            self.0
                .iter()
                .map(|channel_width| channel_width.to_string())
                .collect::<Vec<String>>()
                .join(", ")
        )
    }
}