Adding protectionScheme into config file

obgm · May 9, 2023 · 4cdac7d · 4cdac7d
1 parent 88c99cb
commit 4cdac7d
Show file tree

Hide file tree

Showing 6 changed files with 21 additions and 4 deletions.
diff --git a/config/custom-ue.yaml b/config/custom-ue.yaml
@@ -4,6 +4,8 @@ supi: 'imsi-286010000000001'
 mcc: '286'
 # Mobile Network Code value of HPLMN (2 or 3 digits)
 mnc: '93'
+# SUCI Protection Scheme : 0 for Null-scheme, 1 for Profile A and 2 for Profile B
+protectionScheme: 0
 # Routing Indicator
 routingIndicator: '0000'
 

diff --git a/config/free5gc-ue.yaml b/config/free5gc-ue.yaml
@@ -4,6 +4,8 @@ supi: 'imsi-208930000000003'
 mcc: '208'
 # Mobile Network Code value of HPLMN (2 or 3 digits)
 mnc: '93'
+# SUCI Protection Scheme : 0 for Null-scheme, 1 for Profile A and 2 for Profile B
+protectionScheme: 0
 # Routing Indicator
 routingIndicator: '0000'
 

diff --git a/config/open5gs-ue.yaml b/config/open5gs-ue.yaml
@@ -4,6 +4,8 @@ supi: 'imsi-999700000000001'
 mcc: '999'
 # Mobile Network Code value of HPLMN (2 or 3 digits)
 mnc: '70'
+# SUCI Protection Scheme : 0 for Null-scheme, 1 for Profile A and 2 for Profile B
+protectionScheme: 0
 # Routing Indicator
 routingIndicator: '0000'
 

diff --git a/src/ue.cpp b/src/ue.cpp
@@ -150,6 +150,8 @@ static nr::ue::UeConfig *ReadConfigYaml()
 
     if (yaml::HasField(config, "supi"))
         result->supi = Supi::Parse(yaml::GetString(config, "supi"));
+    if (yaml::HasField(config, "protectionScheme"))
+        result->protectionScheme = yaml::GetInt32(config, "protectionScheme", 0, 255);     
     if (yaml::HasField(config, "imei"))
         result->imei = yaml::GetString(config, "imei", 15, 15);
     if (yaml::HasField(config, "imeiSv"))

diff --git a/src/ue/nas/mm/identity.cpp b/src/ue/nas/mm/identity.cpp
@@ -73,6 +73,7 @@ nas::IE5gsMobileIdentity NasMm::generateSuci()
 {
     auto &supi = m_base->config->supi;
     auto &plmn = m_base->config->hplmn;
+    auto &protectionScheme = m_base->config->protectionScheme;
 
     if (!supi.has_value())
         return {};
@@ -99,10 +100,17 @@ nas::IE5gsMobileIdentity NasMm::generateSuci()
     {
         ret.imsi.routingIndicator = "0000";
     }
-    ret.imsi.protectionSchemaId = 0;
-    ret.imsi.homeNetworkPublicKeyIdentifier = 0;
-    ret.imsi.schemeOutput = imsi.substr(plmn.isLongMnc ? 6 : 5);
-    return ret;
+    if (protectionScheme == 0) {
+        ret.imsi.protectionSchemaId = 0;
+        ret.imsi.homeNetworkPublicKeyIdentifier = 0;
+        ret.imsi.schemeOutput = imsi.substr(plmn.isLongMnc ? 6 : 5);
+        return ret;
+    }
+    else
+    {
+        m_logger->err("Protection Scheme %d not implemented", protectionScheme);
+        return {};
+    }
 }
 
 nas::IE5gsMobileIdentity NasMm::getOrGeneratePreferredId()

diff --git a/src/ue/types.hpp b/src/ue/types.hpp
@@ -93,6 +93,7 @@ struct UeConfig
 {
     /* Read from config file */
     std::optional<Supi> supi{};
+    int protectionScheme;
     std::optional<std::string> routingIndicator{};
     Plmn hplmn{};
     OctetString key{};