From 69dbafeed5fc2618bee9e06ec2e348bb8979f154 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?S=C3=A9bastien=20Deronne?= <sebastien.deronne@gmail.com>
Date: Mon, 12 Dec 2016 19:22:02 +0100
Subject: [PATCH] wifi: Add some asserts in MacLow

---
 src/wifi/model/mac-low.cc | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/src/wifi/model/mac-low.cc b/src/wifi/model/mac-low.cc
index b1ee9cb9c..4fc7adacf 100644
--- a/src/wifi/model/mac-low.cc
+++ b/src/wifi/model/mac-low.cc
@@ -1879,7 +1879,11 @@ MacLow::SendDataPacket (void)
     }
   m_currentHdr.SetDuration (duration);
   Ptr <Packet> packet = m_currentPacket->Copy ();
-  if (!m_ampdu)
+  if (m_ampdu)
+    {
+      NS_ASSERT (m_currentHdr.IsQosData ());
+    }
+  else
     {
       packet->AddHeader (m_currentHdr);
       WifiMacTrailer fcs;
@@ -2081,7 +2085,11 @@ MacLow::SendDataAfterCts (Mac48Address source, Time duration)
   NS_ASSERT (duration >= MicroSeconds (0));
   m_currentHdr.SetDuration (duration);
   Ptr <Packet> packet = m_currentPacket->Copy ();
-  if (!m_ampdu)
+  if (m_ampdu)
+    {
+      NS_ASSERT (m_currentHdr.IsQosData ());
+    }
+  else
     {
       packet->AddHeader (m_currentHdr);
       WifiMacTrailer fcs;