Deauthentication frame parsing

libwifi/src/frame/management/deauthentication.rs

@@ -0,0 +1,8 @@
+use crate::frame::components::{MacAddress, ManagementHeader};
+use libwifi_macros::AddressHeader;
+
+#[derive(Clone, Debug, AddressHeader)]
+pub struct Deauthentication {
+    pub header: ManagementHeader,
+    pub reason_code: u16,
+}

libwifi/src/frame/management/mod.rs

@@ -1,7 +1,9 @@
 mod association;
 mod beacon;
 mod probe;
+mod deauthentication;
 
 pub use association::{AssociationRequest, AssociationResponse};
+pub use deauthentication::{Deauthentication};
 pub use beacon::Beacon;
 pub use probe::{ProbeRequest, ProbeResponse};

libwifi/src/frame/mod.rs

@@ -27,6 +27,7 @@ pub enum Frame {
     ProbeResponse(ProbeResponse),
     AssociationRequest(AssociationRequest),
     AssociationResponse(AssociationResponse),
+    Deauthentication(Deauthentication),
 
     // Control Frames
     Rts(Rts),

libwifi/src/lib.rs

@@ -36,6 +36,7 @@ pub fn parse_frame(input: &[u8]) -> Result<Frame, Error> {
         FrameSubType::ProbeResponse => parse_probe_response(frame_control, input),
         FrameSubType::AssociationRequest => parse_association_request(frame_control, input),
         FrameSubType::AssociationResponse => parse_association_response(frame_control, input),
+        FrameSubType::Deauthentication => parse_deauthentication(frame_control, input),
 
         // Control
         FrameSubType::Rts => parse_rts(frame_control, input),

libwifi/src/parsers/frame_types/management.rs

@@ -4,7 +4,7 @@ use nom::sequence::tuple;
 use crate::error::Error;
 use crate::frame::components::FrameControl;
 use crate::frame::*;
-use crate::parsers::{parse_management_header, parse_station_info};
+use crate::parsers::{clone_slice, parse_management_header, parse_station_info};
 
 /// Parse an [AssociationRequest] frame.
 ///
@@ -110,3 +110,19 @@ pub fn parse_probe_response(frame_control: FrameControl, input: &[u8]) -> Result
         station_info,
     }))
 }
+
+/// Parse a [Deauthentication] frame.
+///
+/// The general structure is:
+/// - ManagementHeader
+/// - Reason code
+pub fn parse_deauthentication(frame_control: FrameControl, input: &[u8]) -> Result<Frame, Error> {
+    let (input, header) = parse_management_header(frame_control, input)?;
+    let reason_code_bytes = clone_slice::<2>(input);
+    let reason_code = u16::from_le_bytes(reason_code_bytes);
+
+    Ok(Frame::Deauthentication(Deauthentication {
+        header,
+        reason_code,
+    }))
+}

libwifi/tests/management_frames.rs

@@ -119,12 +119,20 @@ fn test_authentication() {
 
 #[test]
 fn test_deauthentication() {
-    let _payload = [
+    let payload = [
         192, 0, // FrameControl
         58, 1, // Duration id
         248, 50, 228, 173, 71, 184, // First Address
         192, 238, 251, 75, 207, 58, // Second Address
         248, 50, 228, 173, 71, 184, // Third Address
         224, 146, 3, 0,
     ];
+
+    let frame = parse_frame(&payload).expect("Payload should be valid");
+    println!("{frame:?}");
+    assert!(matches!(frame, Frame::Deauthentication(_)));
+
+    if let Frame::Deauthentication(deauthentication) = frame {
+        assert_eq!(3, deauthentication.reason_code);
+    }
 }