Incrementing DL NAS count

srsepc/src/mme/s1ap_nas_transport.cc

@@ -176,6 +176,15 @@ s1ap_nas_transport::handle_uplink_nas_transport(LIBLTE_S1AP_MESSAGE_UPLINKNASTRA
       m_pool->deallocate(nas_msg);
       return false;
     }
+    //Increment UL NAS count.
+    ue_emm_ctx = m_s1ap->find_ue_emm_ctx_from_imsi(ue_ecm_ctx->imsi);
+    if(ue_emm_ctx == NULL)
+    {
+      m_s1ap_log->warning("Could not find UE EMM context in ", msg_type );
+      m_pool->deallocate(nas_msg);
+      return false;
+    }
+    ue_emm_ctx->security_ctxt.ul_nas_count++;
   }
   else if(sec_hdr_type == LIBLTE_MME_SECURITY_HDR_TYPE_INTEGRITY_WITH_NEW_EPS_SECURITY_CONTEXT || sec_hdr_type == LIBLTE_MME_SECURITY_HDR_TYPE_INTEGRITY_AND_CIPHERED_WITH_NEW_EPS_SECURITY_CONTEXT)
   {
@@ -244,8 +253,12 @@ s1ap_nas_transport::handle_uplink_nas_transport(LIBLTE_S1AP_MESSAGE_UPLINKNASTRA
 
   if(*reply_flag == true)
   {
-    m_s1ap_log->info("DL NAS: Sent Downlink NAS message\n");
-    m_s1ap_log->console("DL NAS: Sent Downlink NAs Message\n");
+    if(ue_emm_ctx != NULL)
+    {
+      ue_emm_ctx->security_ctxt.dl_nas_count++;
+      m_s1ap_log->console("DL NAS: Sent Downlink NAs Message. DL NAS Count=%d\n",ue_emm_ctx->security_ctxt.dl_nas_count);
+      m_s1ap_log->info("DL NAS: Sent Downlink NAS message. DL NAS Count=%d\n",ue_emm_ctx->security_ctxt.dl_nas_count);
+    }
   }
   m_pool->deallocate(nas_msg);