Fix issue for PLMN (MNC length=2, starting with 0)

src/amf-app/amf_n1.cpp

@@ -2075,42 +2075,64 @@ void amf_n1::security_mode_complete_handle(
     return;
   }
 
+  std::shared_ptr<nas_context> nc;
+  nc = amf_ue_id_2_nas_context(amf_ue_ngap_id);
+  Logger::amf_n1().info(
+      "UE (IMSI %s, GUTI %s, current RAN ID %d, current AMF ID %d) has been "
+      "registered to the network",
+      nc.get()->imsi.c_str(), guti.c_str(), ran_ue_ngap_id, amf_ue_ngap_id);
+  if (nc.get()->is_stacs_available) {
+    stacs.update_5gmm_state(nc.get()->imsi, "5GMM-REGISTERED");
+    // update_ue_information_statics(stacs.ues[index], "", "RM-REGISTRED",
+    // ran_ue_ngap_id, amf_ue_ngap_id, "", guti, "", "", 0);
+  } else {
+    // ue_infos ueItem;
+    // update_ue_information_statics(ueItem, "CM-CONNECTED",
+    // "REGISTRATION-REGISTRED", ran_ue_ngap_id, amf_ue_ngap_id,
+    // nc.get()->imsi, "", mcc, mnc, uc.get()->cgi.nrCellID);
+    // stacs.ues.push_back(ueItem);
+    nc.get()->is_stacs_available = true;
+  }
+  set_5gmm_state(nc, _5GMM_REGISTERED);
+
+  set_guti_2_nas_context(guti, nc);
+  nc.get()->is_common_procedure_for_security_mode_control_running = false;
+  nas_secu_ctx* secu = nc.get()->security_ctx;
+  // protect nas message
+  bstring protectedNas;
+  encode_nas_message_protected(
+      secu, false, INTEGRITY_PROTECTED_AND_CIPHERED, NAS_MESSAGE_DOWNLINK,
+      buffer, encoded_size, protectedNas);
+
   if (!uc.get()->isUeContextRequest) {
     Logger::amf_n1().debug(
         "UE Context is not requested, UE with ran_ue_ngap_id %d, "
         "amf_ue_ngap_id %d attached",
         ran_ue_ngap_id, amf_ue_ngap_id);
-    // send registration accept back
+
+    // TODO: Use DownlinkNasTransport to convey Registration Accept
+    // IE: UEAggregateMaximumBitRate
+    // AllowedNSSAI
+
+    /*   itti_dl_nas_transport* dnt =
+           new itti_dl_nas_transport(TASK_AMF_N1, TASK_AMF_N2);
+       dnt->nas            = ;
+       dnt->amf_ue_ngap_id = amf_ue_ngap_id;
+       dnt->ran_ue_ngap_id = ran_ue_ngap_id;
+       std::shared_ptr<itti_dl_nas_transport> i =
+           std::shared_ptr<itti_dl_nas_transport>(dnt);
+       int ret = itti_inst->send_msg(i);
+       if (0 != ret) {
+         Logger::amf_n1().error(
+             "Could not send ITTI message %s to task TASK_AMF_N2",
+             i->get_msg_name());
+       }
+      */
+
   } else {
-    // encoding InitialContextSetupRequest(NGAP message) back
-    std::shared_ptr<nas_context> nc;
-    nc = amf_ue_id_2_nas_context(amf_ue_ngap_id);
-    Logger::amf_n1().info(
-        "UE (IMSI %s, GUTI %s, current RAN ID %d, current AMF ID %d) has been "
-        "registered to the network",
-        nc.get()->imsi.c_str(), guti.c_str(), ran_ue_ngap_id, amf_ue_ngap_id);
-    if (nc.get()->is_stacs_available) {
-      stacs.update_5gmm_state(nc.get()->imsi, "5GMM-REGISTERED");
-      // update_ue_information_statics(stacs.ues[index], "", "RM-REGISTRED",
-      // ran_ue_ngap_id, amf_ue_ngap_id, "", guti, "", "", 0);
-    } else {
-      // ue_infos ueItem;
-      // update_ue_information_statics(ueItem, "CM-CONNECTED",
-      // "REGISTRATION-REGISTRED", ran_ue_ngap_id, amf_ue_ngap_id,
-      // nc.get()->imsi, "", mcc, mnc, uc.get()->cgi.nrCellID);
-      // stacs.ues.push_back(ueItem);
-      nc.get()->is_stacs_available = true;
-    }
-    set_5gmm_state(nc, _5GMM_REGISTERED);
+    // use InitialContextSetupRequest (NGAP message) to convey Registration
+    // Accept
 
-    set_guti_2_nas_context(guti, nc);
-    nc.get()->is_common_procedure_for_security_mode_control_running = false;
-    nas_secu_ctx* secu = nc.get()->security_ctx;
-    // protect nas message
-    bstring protectedNas;
-    encode_nas_message_protected(
-        secu, false, INTEGRITY_PROTECTED_AND_CIPHERED, NAS_MESSAGE_DOWNLINK,
-        buffer, encoded_size, protectedNas);
     uint8_t* kamf = nc.get()->kamf[secu->vector_pointer];
     uint8_t kgnb[32];
     uint32_t ulcount = secu->ul_count.seq_num | (secu->ul_count.overflow << 8);
@@ -2119,6 +2141,7 @@ void amf_n1::security_mode_complete_handle(
     print_buffer("amf_n1", "kamf", kamf, 32);
     // Authentication_5gaka::derive_kgnb(ulcount, 0x01, kamf, kgnb);
     bstring kgnb_bs = blk2bstr(kgnb, 32);
+
     itti_initial_context_setup_request* itti_msg =
         new itti_initial_context_setup_request(TASK_AMF_N1, TASK_AMF_N2);
     itti_msg->ran_ue_ngap_id = ran_ue_ngap_id;

src/nas/msgs/ULNASTransport.cpp

@@ -412,4 +412,5 @@ int ULNASTransport::decodefrombuffer(
   }
   Logger::nas_mm().debug(
       "decoded ULNASTransport message len(%d)", decoded_size);
+  return 0;
 }

src/ngap/ngapIEs/PlmnId.cpp

@@ -58,13 +58,17 @@ void PlmnId::setMccMnc(const std::string mcc, const std::string mnc) {
   int mncValue = fromString<int>(mnc);
 
   mcc_digit1 = mccValue / 100;
-  mcc_digit2 = (uint8_t) floor((double) (mccValue % 100) / 10);
+  mcc_digit2 = (mccValue - mcc_digit1 * 100) /
+               10;
   mcc_digit3 = mccValue % 10;
 
-  mnc_digit1 = (uint8_t) floor(((double) mncValue / 100));
-  if (!mnc_digit1) mnc_digit1 = 0xf;
-  mnc_digit2 = (uint8_t) floor((double) (mncValue % 100) / 10);
-  mnc_digit3 = mncValue % 10;
+  if (mncValue > 99) {
+    mnc_digit3 = mncValue / 100;
+  } else {
+    mnc_digit3 = 0xf;
+  }
+  mnc_digit1 = (uint8_t) floor((double) (mncValue % 100) / 10);
+  mnc_digit2 = mncValue % 10;
 }
 
 //------------------------------------------------------------------------------
@@ -76,10 +80,15 @@ void PlmnId::getMcc(std::string& mcc) {
 //------------------------------------------------------------------------------
 void PlmnId::getMnc(std::string& mnc) {
   int m_mnc = 0;
-  if (mnc_digit1 == 0xf)
-    m_mnc = mnc_digit2 * 10 + mnc_digit3;
-  else
-    m_mnc = mnc_digit1 * 100 + mnc_digit2 * 10 + mnc_digit3;
+  if (mnc_digit3 == 0xf) {
+    m_mnc = mnc_digit1 * 10 + mnc_digit2;
+    if (mnc_digit1 == 0) {
+      mnc = "0" + to_string(m_mnc);
+      return;
+    }
+  } else {
+    m_mnc = mnc_digit3 * 100 + mnc_digit1 * 10 + mnc_digit2;
+  }
   mnc = to_string(m_mnc);
 }
 
@@ -89,9 +98,9 @@ bool PlmnId::encode2octetstring(Ngap_PLMNIdentity_t& plmn) {
   uint8_t* buffer = (uint8_t*) calloc(1, 3 * sizeof(uint8_t));
   if (!buffer) return false;
 
-  buffer[0] = (mcc_digit1 & 0x0f) | ((mcc_digit2 & 0x0f) << 4);
-  buffer[1] = (mcc_digit3 & 0x0f) | ((mnc_digit1 & 0x0f) << 4);
-  buffer[2] = (mnc_digit2 & 0x0f) | ((mnc_digit3 & 0x0f) << 4);
+  buffer[0] = 0x00 | ((mcc_digit2 & 0x0f) << 4) | (mcc_digit1 & 0x0f);
+  buffer[1] = 0x00 | ((mnc_digit3 & 0x0f) << 4) | (mcc_digit3 & 0x0f);
+  buffer[2] = 0x00 | ((mnc_digit2 & 0x0f) << 4) | (mnc_digit1 & 0x0f);
   plmn.buf  = buffer;
 
   return true;
@@ -102,10 +111,11 @@ bool PlmnId::decodefromoctetstring(Ngap_PLMNIdentity_t& plmn) {
   if (!plmn.buf) return false;
   mcc_digit1 = plmn.buf[0] & 0x0f;
   mcc_digit2 = plmn.buf[0] >> 4;
+
   mcc_digit3 = plmn.buf[1] & 0x0f;
-  mnc_digit1 = plmn.buf[1] >> 4;
-  mnc_digit2 = plmn.buf[2] & 0x0f;
-  mnc_digit3 = plmn.buf[2] >> 4;
+  mnc_digit3 = plmn.buf[1] >> 4;
+  mnc_digit1 = plmn.buf[2] & 0x0f;
+  mnc_digit2 = plmn.buf[2] >> 4;
 
   return true;
 }

src/ngap/ngapIEs/PlmnId.hpp

@@ -49,12 +49,12 @@ class PlmnId {
   void getMnc(std::string& mnc);
 
  private:
-  uint8_t mcc_digit1;
   uint8_t mcc_digit2;
+  uint8_t mcc_digit1;
+  uint8_t mnc_digit3;  // in case of 2 digit MNC, it should be 0xf
   uint8_t mcc_digit3;
-  uint8_t mnc_digit1;  // in case of 2 digit MNC, it should be 0xf
   uint8_t mnc_digit2;
-  uint8_t mnc_digit3;
+  uint8_t mnc_digit1;
 };
 
 }  // namespace ngap

src/ngap/ngapIEs/ResetType.cpp

@@ -85,6 +85,7 @@ bool ResetType::decode(Ngap_ResetType_t* type) {
   } else {
     return false;
   }
+  return true;
 }
 
 //------------------------------------------------------------------------------

src/ngap/ngapIEs/UEAssociationLogicalNGConnectionItem.cpp

@@ -89,6 +89,7 @@ bool UEAssociationLogicalNGConnectionItem::encode(
   amfUeNgapId->encode2AMF_UE_NGAP_ID(*item.aMF_UE_NGAP_ID);
   item.rAN_UE_NGAP_ID = new Ngap_RAN_UE_NGAP_ID_t();
   ranUeNgapId->encode2RAN_UE_NGAP_ID(*item.rAN_UE_NGAP_ID);
+  return true;
 }
 
 //------------------------------------------------------------------------------
@@ -98,6 +99,7 @@ bool UEAssociationLogicalNGConnectionItem::encode(
   item.amfUeNgapId = amfUeNgapId;
   item.ranUeNgapId = new RAN_UE_NGAP_ID();
   item.ranUeNgapId = ranUeNgapId;
+  return true;
 }
 
 //------------------------------------------------------------------------------

src/ngap/ngapMsgs/NGReset.cpp

@@ -108,6 +108,7 @@ bool NGResetMsg::getResetType(ResetType& resetType) {
       this->resetType->getResetType() == Ngap_ResetType_PR_partOfNG_Interface) {
     // TODO
   }
+  return true;
 }
 
 //------------------------------------------------------------------------------