From 194ea70abb376d63d0bd85cbe809b5d5c19034c5 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Pawe=C5=82=20Wa=C5=84czyk?= <Pawel.Wanczyk@nordicsemi.no>
Date: Thu, 19 Dec 2019 20:17:41 +0100
Subject: [PATCH] [mac] add retry flag support for direct messages (#4421)

---
 src/core/mac/mac.cpp     | 1 +
 src/core/mac/sub_mac.cpp | 1 +
 2 files changed, 2 insertions(+)

diff --git a/src/core/mac/mac.cpp b/src/core/mac/mac.cpp
index b8392905e..6cca1d7bc 100644
--- a/src/core/mac/mac.cpp
+++ b/src/core/mac/mac.cpp
@@ -956,6 +956,7 @@ void Mac::BeginTransmit(void)
     TxFrame &sendFrame             = mSubMac.GetTransmitFrame();
 
     VerifyOrExit(mEnabled, error = OT_ERROR_ABORT);
+    sendFrame.SetIsARetransmission(false);
 
     switch (mOperation)
     {
diff --git a/src/core/mac/sub_mac.cpp b/src/core/mac/sub_mac.cpp
index 747069ebc..c5c05ac1d 100644
--- a/src/core/mac/sub_mac.cpp
+++ b/src/core/mac/sub_mac.cpp
@@ -370,6 +370,7 @@ void SubMac::HandleTransmitDone(TxFrame &aFrame, RxFrame *aAckFrame, otError aEr
     if (shouldRetx)
     {
         mTransmitRetries++;
+        aFrame.SetIsARetransmission(true);
         StartCsmaBackoff();
         ExitNow();
     }