From 6cb61da4a1a84dad54f4c30fa9d42da84b7701e3 Mon Sep 17 00:00:00 2001
From: Mathy Vanhoef <Mathy.Vanhoef@kuleuven.be>
Date: Sun, 16 May 2021 22:07:13 +0400
Subject: [PATCH] libwifi: add remove_dot11qos function

---
 research/libwifi/wifi.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/research/libwifi/wifi.py b/research/libwifi/wifi.py
index 7fa3e0338..9df4aacaf 100644
--- a/research/libwifi/wifi.py
+++ b/research/libwifi/wifi.py
@@ -146,6 +146,15 @@ def is_amsdu(p):
 	else:
 		return p.Reserved == 1
 
+def remove_dot11qos(p):
+	if not Dot11QoS in p: return
+	p = p.copy()
+	payload = p[Dot11QoS].payload
+	p.remove_payload()
+	p /= payload
+	p.subtype = 0
+	return p
+
 #### Packet Processing Functions ####
 
 class DHCP_sock(DHCP_am):