From 03ecb7d5b395f6bfc8d7b419b89a39874fdc7073 Mon Sep 17 00:00:00 2001
From: Tim Schubert <tim.schubert@tu-bs.de>
Date: Tue, 21 Jul 2020 16:11:32 +0200
Subject: [PATCH] Set missing protocol type

---
 model/mock-net-device.cc | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/model/mock-net-device.cc b/model/mock-net-device.cc
index 49dea76..ca1ea9a 100644
--- a/model/mock-net-device.cc
+++ b/model/mock-net-device.cc
@@ -192,6 +192,7 @@ MockNetDevice::AddHeader (Ptr<Packet> p,
 {
   NS_LOG_FUNCTION (this << p << protocolNumber);
   EthernetHeader ethernet;
+  ethernet.SetLengthType (protocolNumber);
   ethernet.SetSource (Mac48Address::ConvertFrom (src));
   ethernet.SetDestination (Mac48Address::ConvertFrom (dst));
   p->AddHeader (ethernet);
@@ -359,8 +360,6 @@ MockNetDevice::Receive (Ptr<Packet> packet, Ptr<MockNetDevice> senderDevice)
 {
   NS_LOG_FUNCTION (this << packet << senderDevice);
 
-  uint16_t protocol = 0;
-
   if (m_receiveErrorModel && m_receiveErrorModel->IsCorrupt (packet) )
     {
       //
@@ -388,6 +387,8 @@ MockNetDevice::Receive (Ptr<Packet> packet, Ptr<MockNetDevice> senderDevice)
 
       EthernetHeader header;
       packet->RemoveHeader (header);
+      uint16_t protocol = header.GetLengthType ();
+
       PacketType packetType;
       if (header.GetDestination ().IsBroadcast ())
       	{