From a0d727166ff5007db9c8b88695ff7ada1617e80e Mon Sep 17 00:00:00 2001
From: Pedro Alvarez <pedro.alvarez@softwareradiosystems.com>
Date: Mon, 5 Feb 2018 13:11:03 +0000
Subject: [PATCH] Starting to split EMM and ECM context.

---
 srsepc/hdr/mme/s1ap_common.h | 19 +++++++++++++
 srsepc/src/mme/s1ap.cc       | 55 ++++++++++++++++++++++++------------
 2 files changed, 56 insertions(+), 18 deletions(-)

diff --git a/srsepc/hdr/mme/s1ap_common.h b/srsepc/hdr/mme/s1ap_common.h
index 2d693eba9..b252df586 100644
--- a/srsepc/hdr/mme/s1ap_common.h
+++ b/srsepc/hdr/mme/s1ap_common.h
@@ -127,6 +127,25 @@ typedef struct{
     srslte::gtpc_f_teid_ie sgw_ctrl_fteid;
 } erab_ctx_t;
 
+typedef struct{
+    uint64_t imsi;
+    LIBLTE_MME_EPS_MOBILE_ID_GUTI_STRUCT guti;
+    emm_state_t emm_state;
+    eps_security_ctx_t security_ctxt;
+    uint8_t procedure_transaction_id;
+} ue_emm_ctx_t;
+
+typedef struct{
+  uint32_t enb_ue_s1ap_id;
+  uint32_t mme_ue_s1ap_id;
+  uint16_t enb_id;
+  struct   sctp_sndrcvinfo enb_sri;
+  ecm_state_t ecm_state; 
+  bool eit;
+  erab_ctx_t erabs_ctx[MAX_ERABS_PER_UE]; 
+} ue_esm_ctx_t;
+
+
 typedef struct{
   uint64_t imsi;
   uint32_t enb_ue_s1ap_id;
diff --git a/srsepc/src/mme/s1ap.cc b/srsepc/src/mme/s1ap.cc
index 877912e8d..0db79a8fb 100644
--- a/srsepc/src/mme/s1ap.cc
+++ b/srsepc/src/mme/s1ap.cc
@@ -399,26 +399,9 @@ s1ap::find_ue_ctx_from_mme_ue_s1ap_id(uint32_t mme_ue_s1ap_id)
     return find_ue_ctx_from_imsi(imsi_it->second);
   }
 }
-void
-s1ap::delete_ues_in_enb(uint16_t enb_id)
-{
-  //delete UEs ctx
-  std::map<uint16_t,std::set<uint32_t> >::iterator ues_in_enb = m_enb_id_to_ue_ids.find(enb_id);
-  std::set<uint32_t>::iterator ue_id = ues_in_enb->second.begin();
-  while(ue_id != ues_in_enb->second.end() )
-  {
-    std::map<uint32_t, ue_ctx_t*>::iterator ue_ctx = m_active_ues.find(*ue_id);
-    m_s1ap_log->info("Deleting UE context. UE IMSI: %lu\n", ue_ctx->second->imsi);
-    m_s1ap_log->console("Deleting UE context. UE IMSI: %lu\n", ue_ctx->second->imsi);
-    delete ue_ctx->second;             //delete UE context
-    m_active_ues.erase(ue_ctx);        //remove from general MME map 
-    ues_in_enb->second.erase(ue_id++); //erase from the eNB's UE set
-  }
-
-}
 
 bool
-s1ap::delete_ue_ctx(ue_ctx_t *ue_ctx)
+s1ap::delete_ue_esm_ctx(uint32_t mme_ue_s1ap_id)
 {
   uint32_t mme_ue_s1ap_id = ue_ctx->mme_ue_s1ap_id;
   std::map<uint32_t, ue_ctx_t*>::iterator ue_ctx_it = m_active_ues.find(mme_ue_s1ap_id);
@@ -451,6 +434,42 @@ s1ap::delete_ue_ctx(ue_ctx_t *ue_ctx)
 
   return true;
 }
+void
+s1ap::delete_ues_esm_ctx_in_enb(uint16_t enb_id)
+{
+  //delete UEs ctx
+  std::map<uint16_t,std::set<uint32_t> >::iterator ues_in_enb = m_enb_id_to_ue_ids.find(enb_id);
+  std::set<uint32_t>::iterator ue_id = ues_in_enb->second.begin();
+  while(ue_id != ues_in_enb->second.end() )
+  {
+    std::map<uint32_t, ue_ctx_t*>::iterator ue_ctx = m_active_ues.find(*ue_id);
+    m_s1ap_log->info("Deleting UE context. UE IMSI: %lu\n", ue_ctx->second->imsi);
+    m_s1ap_log->console("Deleting UE context. UE IMSI: %lu\n", ue_ctx->second->imsi);
+    delete ue_ctx->second;             //delete UE context
+    m_active_ues.erase(ue_ctx);        //remove from general MME map 
+    ues_in_enb->second.erase(ue_id++); //erase from the eNB's UE set
+  }
+}
+
+void
+s1ap::delete_ues_in_enb(uint16_t enb_id)
+{
+  //delete UEs ctx
+  std::map<uint16_t,std::set<uint32_t> >::iterator ues_in_enb = m_enb_id_to_ue_ids.find(enb_id);
+  std::set<uint32_t>::iterator ue_id = ues_in_enb->second.begin();
+  while(ue_id != ues_in_enb->second.end() )
+  {
+    std::map<uint32_t, ue_ctx_t*>::iterator ue_ctx = m_active_ues.find(*ue_id);
+    m_s1ap_log->info("Deleting UE context. UE IMSI: %lu\n", ue_ctx->second->imsi);
+    m_s1ap_log->console("Deleting UE context. UE IMSI: %lu\n", ue_ctx->second->imsi);
+    delete ue_ctx->second;             //delete UE context
+    m_active_ues.erase(ue_ctx);        //remove from general MME map 
+    ues_in_enb->second.erase(ue_id++); //erase from the eNB's UE set
+  }
+
+}
+
+