/* * Copyright (C) 2019,2020 by Sukchan Lee * * This file is part of Open5GS. * * This program is free software: you can redistribute it and/or modify * it under the terms of the GNU Affero General Public License as published by * the Free Software Foundation, either version 3 of the License, or * (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program. If not, see . */ #include "ngap-path.h" static amf_context_t self; int __amf_log_domain; int __gmm_log_domain; static OGS_POOL(amf_gnb_pool, amf_gnb_t); static OGS_POOL(amf_ue_pool, amf_ue_t); static OGS_POOL(ran_ue_pool, ran_ue_t); static OGS_POOL(amf_sess_pool, amf_sess_t); static int context_initialized = 0; static int num_of_ran_ue = 0; static int num_of_amf_sess = 0; static void stats_add_ran_ue(void); static void stats_remove_ran_ue(void); static void stats_add_amf_session(void); static void stats_remove_amf_session(void); void amf_context_init(void) { ogs_assert(context_initialized == 0); /* Initialize AMF context */ memset(&self, 0, sizeof(amf_context_t)); ogs_log_install_domain(&__ogs_sctp_domain, "sctp", ogs_core()->log.level); ogs_log_install_domain(&__ogs_ngap_domain, "ngap", ogs_core()->log.level); ogs_log_install_domain(&__ogs_nas_domain, "nas", ogs_core()->log.level); ogs_log_install_domain(&__amf_log_domain, "amf", ogs_core()->log.level); ogs_log_install_domain(&__gmm_log_domain, "gmm", ogs_core()->log.level); ogs_list_init(&self.ngap_list); ogs_list_init(&self.ngap_list6); /* Allocate TWICE the pool to check if maximum number of gNBs is reached */ ogs_pool_init(&amf_gnb_pool, ogs_app()->max.gnb*2); ogs_pool_init(&amf_ue_pool, ogs_app()->max.ue); ogs_pool_init(&ran_ue_pool, ogs_app()->max.ue); ogs_pool_init(&amf_sess_pool, ogs_app()->pool.sess); ogs_pool_init(&self.m_tmsi, ogs_app()->max.ue); ogs_list_init(&self.gnb_list); ogs_list_init(&self.amf_ue_list); self.gnb_addr_hash = ogs_hash_make(); self.gnb_id_hash = ogs_hash_make(); self.guti_ue_hash = ogs_hash_make(); self.suci_hash = ogs_hash_make(); self.supi_hash = ogs_hash_make(); context_initialized = 1; } void amf_context_final(void) { ogs_assert(context_initialized == 1); amf_gnb_remove_all(); amf_ue_remove_all(); ogs_assert(self.gnb_addr_hash); ogs_hash_destroy(self.gnb_addr_hash); ogs_assert(self.gnb_id_hash); ogs_hash_destroy(self.gnb_id_hash); ogs_assert(self.guti_ue_hash); ogs_hash_destroy(self.guti_ue_hash); ogs_assert(self.suci_hash); ogs_hash_destroy(self.suci_hash); ogs_assert(self.supi_hash); ogs_hash_destroy(self.supi_hash); ogs_pool_final(&self.m_tmsi); ogs_pool_final(&amf_sess_pool); ogs_pool_final(&amf_ue_pool); ogs_pool_final(&ran_ue_pool); ogs_pool_final(&amf_gnb_pool); context_initialized = 0; } amf_context_t *amf_self(void) { return &self; } static int amf_context_prepare(void) { self.nf_type = OpenAPI_nf_type_AMF; self.relative_capacity = 0xff; self.ngap_port = OGS_NGAP_SCTP_PORT; return OGS_OK; } static int amf_context_validation(void) { if (ogs_list_first(&self.ngap_list) == NULL && ogs_list_first(&self.ngap_list6) == NULL) { ogs_error("No amf.ngap in '%s'", ogs_app()->file); return OGS_RETRY; } if (self.num_of_served_guami == 0) { ogs_error("No amf.guami in '%s'", ogs_app()->file); return OGS_ERROR; } if (self.num_of_served_tai == 0) { ogs_error("No amf.tai in '%s'", ogs_app()->file); return OGS_ERROR; } if (self.num_of_plmn_support == 0) { ogs_error("No amf.plmn_support in '%s'", ogs_app()->file); return OGS_ERROR; } if (self.plmn_support[0].num_of_s_nssai == 0) { ogs_error("No amf.plmn_support.s_nssai in '%s'", ogs_app()->file); return OGS_ERROR; } if (self.served_tai[0].list0.tai[0].num == 0 && self.served_tai[0].list2.num == 0) { ogs_error("No amf.tai.plmn_id|tac in '%s'", ogs_app()->file); return OGS_ERROR; } if (self.amf_name == NULL) { ogs_error("No amf.amf_name in '%s'", ogs_app()->file); return OGS_ERROR; } if (self.num_of_integrity_order == 0) { ogs_error("No amf.security.integrity_order in '%s'", ogs_app()->file); return OGS_ERROR; } if (self.num_of_ciphering_order == 0) { ogs_error("no amf.security.ciphering_order in '%s'", ogs_app()->file); return OGS_ERROR; } return OGS_OK; } int amf_context_parse_config(void) { int rv; yaml_document_t *document = NULL; ogs_yaml_iter_t root_iter; document = ogs_app()->document; ogs_assert(document); rv = amf_context_prepare(); if (rv != OGS_OK) return rv; ogs_yaml_iter_init(&root_iter, document); while (ogs_yaml_iter_next(&root_iter)) { const char *root_key = ogs_yaml_iter_key(&root_iter); ogs_assert(root_key); if (!strcmp(root_key, "amf")) { ogs_yaml_iter_t amf_iter; ogs_yaml_iter_recurse(&root_iter, &amf_iter); while (ogs_yaml_iter_next(&amf_iter)) { const char *amf_key = ogs_yaml_iter_key(&amf_iter); ogs_assert(amf_key); if (!strcmp(amf_key, "relative_capacity")) { const char *v = ogs_yaml_iter_value(&amf_iter); if (v) self.relative_capacity = atoi(v); } else if (!strcmp(amf_key, "ngap")) { ogs_yaml_iter_t ngap_array, ngap_iter; ogs_yaml_iter_recurse(&amf_iter, &ngap_array); do { int family = AF_UNSPEC; int i, num = 0; const char *hostname[OGS_MAX_NUM_OF_HOSTNAME]; uint16_t port = self.ngap_port; const char *dev = NULL; ogs_sockaddr_t *addr = NULL; if (ogs_yaml_iter_type(&ngap_array) == YAML_MAPPING_NODE) { memcpy(&ngap_iter, &ngap_array, sizeof(ogs_yaml_iter_t)); } else if (ogs_yaml_iter_type(&ngap_array) == YAML_SEQUENCE_NODE) { if (!ogs_yaml_iter_next(&ngap_array)) break; ogs_yaml_iter_recurse(&ngap_array, &ngap_iter); } else if (ogs_yaml_iter_type(&ngap_array) == YAML_SCALAR_NODE) { break; } else ogs_assert_if_reached(); while (ogs_yaml_iter_next(&ngap_iter)) { const char *ngap_key = ogs_yaml_iter_key(&ngap_iter); ogs_assert(ngap_key); if (!strcmp(ngap_key, "family")) { const char *v = ogs_yaml_iter_value(&ngap_iter); if (v) family = atoi(v); if (family != AF_UNSPEC && family != AF_INET && family != AF_INET6) { ogs_warn("Ignore family(%d) : " "AF_UNSPEC(%d), " "AF_INET(%d), AF_INET6(%d) ", family, AF_UNSPEC, AF_INET, AF_INET6); family = AF_UNSPEC; } } else if (!strcmp(ngap_key, "addr") || !strcmp(ngap_key, "name")) { ogs_yaml_iter_t hostname_iter; ogs_yaml_iter_recurse( &ngap_iter, &hostname_iter); ogs_assert(ogs_yaml_iter_type(&hostname_iter) != YAML_MAPPING_NODE); do { if (ogs_yaml_iter_type(&hostname_iter) == YAML_SEQUENCE_NODE) { if (!ogs_yaml_iter_next(&hostname_iter)) break; } ogs_assert(num <= OGS_MAX_NUM_OF_HOSTNAME); hostname[num++] = ogs_yaml_iter_value(&hostname_iter); } while ( ogs_yaml_iter_type(&hostname_iter) == YAML_SEQUENCE_NODE); } else if (!strcmp(ngap_key, "port")) { const char *v = ogs_yaml_iter_value(&ngap_iter); if (v) port = atoi(v); } else if (!strcmp(ngap_key, "dev")) { dev = ogs_yaml_iter_value(&ngap_iter); } else ogs_warn("unknown key `%s`", ngap_key); } addr = NULL; for (i = 0; i < num; i++) { rv = ogs_addaddrinfo(&addr, family, hostname[i], port, 0); ogs_assert(rv == OGS_OK); } if (addr) { if (ogs_app()->parameter.no_ipv4 == 0) ogs_socknode_add( &self.ngap_list, AF_INET, addr); if (ogs_app()->parameter.no_ipv6 == 0) ogs_socknode_add( &self.ngap_list6, AF_INET6, addr); ogs_freeaddrinfo(addr); } if (dev) { rv = ogs_socknode_probe( ogs_app()->parameter.no_ipv4 ? NULL : &self.ngap_list, ogs_app()->parameter.no_ipv6 ? NULL : &self.ngap_list6, dev, port); ogs_assert(rv == OGS_OK); } } while (ogs_yaml_iter_type(&ngap_array) == YAML_SEQUENCE_NODE); if (ogs_list_first(&self.ngap_list) == NULL && ogs_list_first(&self.ngap_list6) == NULL) { rv = ogs_socknode_probe( ogs_app()->parameter.no_ipv4 ? NULL : &self.ngap_list, ogs_app()->parameter.no_ipv6 ? NULL : &self.ngap_list6, NULL, self.ngap_port); ogs_assert(rv == OGS_OK); } } else if (!strcmp(amf_key, "guami")) { ogs_yaml_iter_t guami_array, guami_iter; ogs_yaml_iter_recurse(&amf_iter, &guami_array); do { const char *mcc = NULL, *mnc = NULL; const char *region = NULL, *set = NULL; const char *pointer = NULL; ogs_assert(self.num_of_served_guami <= MAX_NUM_OF_SERVED_GUAMI); if (ogs_yaml_iter_type(&guami_array) == YAML_MAPPING_NODE) { memcpy(&guami_iter, &guami_array, sizeof(ogs_yaml_iter_t)); } else if (ogs_yaml_iter_type(&guami_array) == YAML_SEQUENCE_NODE) { if (!ogs_yaml_iter_next(&guami_array)) break; ogs_yaml_iter_recurse(&guami_array, &guami_iter); } else if (ogs_yaml_iter_type(&guami_array) == YAML_SCALAR_NODE) { break; } else ogs_assert_if_reached(); while (ogs_yaml_iter_next(&guami_iter)) { const char *guami_key = ogs_yaml_iter_key(&guami_iter); ogs_assert(guami_key); if (!strcmp(guami_key, "plmn_id")) { ogs_yaml_iter_t plmn_id_iter; ogs_yaml_iter_recurse(&guami_iter, &plmn_id_iter); while (ogs_yaml_iter_next(&plmn_id_iter)) { const char *plmn_id_key = ogs_yaml_iter_key(&plmn_id_iter); ogs_assert(plmn_id_key); if (!strcmp(plmn_id_key, "mcc")) { mcc = ogs_yaml_iter_value( &plmn_id_iter); } else if (!strcmp(plmn_id_key, "mnc")) { mnc = ogs_yaml_iter_value( &plmn_id_iter); } } if (mcc && mnc) { ogs_plmn_id_build( &self.served_guami[ self.num_of_served_guami]. plmn_id, atoi(mcc), atoi(mnc), strlen(mnc)); } } else if (!strcmp(guami_key, "amf_id")) { ogs_yaml_iter_t amf_id_iter; ogs_yaml_iter_recurse(&guami_iter, &amf_id_iter); while (ogs_yaml_iter_next(&amf_id_iter)) { const char *amf_id_key = ogs_yaml_iter_key(&amf_id_iter); ogs_assert(amf_id_key); if (!strcmp(amf_id_key, "region")) { region = ogs_yaml_iter_value( &amf_id_iter); } else if (!strcmp(amf_id_key, "set")) { set = ogs_yaml_iter_value( &amf_id_iter); } else if (!strcmp(amf_id_key, "pointer")) { pointer = ogs_yaml_iter_value( &amf_id_iter); } } if (region && set) { ogs_amf_id_build( &self.served_guami[ self.num_of_served_guami]. amf_id, atoi(region), atoi(set), pointer ? atoi(pointer) : 0); } } else ogs_warn("unknown key `%s`", guami_key); } if (mnc && mcc && region && set) { self.num_of_served_guami++; } else { ogs_warn("Ignore guami : " "mcc(%s), mnc(%s), region(%s), set(%s)", mcc, mnc, region, set); } } while (ogs_yaml_iter_type(&guami_array) == YAML_SEQUENCE_NODE); } else if (!strcmp(amf_key, "tai")) { int num_of_list0 = 0; ogs_5gs_tai0_list_t *list0 = NULL; ogs_5gs_tai2_list_t *list2 = NULL; ogs_assert(self.num_of_served_tai <= OGS_MAX_NUM_OF_SERVED_TAI); list0 = &self.served_tai[self.num_of_served_tai].list0; ogs_assert(list0); list2 = &self.served_tai[self.num_of_served_tai].list2; ogs_assert(list2); ogs_yaml_iter_t tai_array, tai_iter; ogs_yaml_iter_recurse(&amf_iter, &tai_array); do { const char *mcc = NULL, *mnc = NULL; ogs_uint24_t tac[OGS_MAX_NUM_OF_TAI]; int num_of_tac = 0; if (ogs_yaml_iter_type(&tai_array) == YAML_MAPPING_NODE) { memcpy(&tai_iter, &tai_array, sizeof(ogs_yaml_iter_t)); } else if (ogs_yaml_iter_type(&tai_array) == YAML_SEQUENCE_NODE) { if (!ogs_yaml_iter_next(&tai_array)) break; ogs_yaml_iter_recurse(&tai_array, &tai_iter); } else if (ogs_yaml_iter_type(&tai_array) == YAML_SCALAR_NODE) { break; } else ogs_assert_if_reached(); while (ogs_yaml_iter_next(&tai_iter)) { const char *tai_key = ogs_yaml_iter_key(&tai_iter); ogs_assert(tai_key); if (!strcmp(tai_key, "plmn_id")) { ogs_yaml_iter_t plmn_id_iter; ogs_yaml_iter_recurse(&tai_iter, &plmn_id_iter); while (ogs_yaml_iter_next(&plmn_id_iter)) { const char *plmn_id_key = ogs_yaml_iter_key(&plmn_id_iter); ogs_assert(plmn_id_key); if (!strcmp(plmn_id_key, "mcc")) { mcc = ogs_yaml_iter_value( &plmn_id_iter); } else if (!strcmp(plmn_id_key, "mnc")) { mnc = ogs_yaml_iter_value( &plmn_id_iter); } } } else if (!strcmp(tai_key, "tac")) { ogs_yaml_iter_t tac_iter; ogs_yaml_iter_recurse(&tai_iter, &tac_iter); ogs_assert(ogs_yaml_iter_type(&tac_iter) != YAML_MAPPING_NODE); do { const char *v = NULL; ogs_assert(num_of_tac <= OGS_MAX_NUM_OF_TAI); if (ogs_yaml_iter_type(&tac_iter) == YAML_SEQUENCE_NODE) { if (!ogs_yaml_iter_next(&tac_iter)) break; } v = ogs_yaml_iter_value(&tac_iter); if (v) { tac[num_of_tac].v = atoi(v); num_of_tac++; } } while ( ogs_yaml_iter_type(&tac_iter) == YAML_SEQUENCE_NODE); } else ogs_warn("unknown key `%s`", tai_key); } if (mcc && mnc && num_of_tac) { if (num_of_tac == 1) { ogs_plmn_id_build( &list2->tai[list2->num].plmn_id, atoi(mcc), atoi(mnc), strlen(mnc)); list2->tai[list2->num].tac.v = tac[0].v; list2->num++; if (list2->num > 1) list2->type = OGS_TAI2_TYPE; else list2->type = OGS_TAI1_TYPE; } else if (num_of_tac > 1) { int i; ogs_plmn_id_build( &list0->tai[num_of_list0].plmn_id, atoi(mcc), atoi(mnc), strlen(mnc)); for (i = 0; i < num_of_tac; i++) { list0->tai[num_of_list0].tac[i].v = tac[i].v; } list0->tai[num_of_list0].num = num_of_tac; list0->tai[num_of_list0].type = OGS_TAI0_TYPE; num_of_list0++; } } else { ogs_warn("Ignore tai : mcc(%p), mnc(%p), " "num_of_tac(%d)", mcc, mnc, num_of_tac); } } while (ogs_yaml_iter_type(&tai_array) == YAML_SEQUENCE_NODE); if (list2->num || num_of_list0) { self.num_of_served_tai++; } } else if (!strcmp(amf_key, "plmn_support")) { ogs_yaml_iter_t plmn_support_array, plmn_support_iter; ogs_yaml_iter_recurse(&amf_iter, &plmn_support_array); do { const char *mnc = NULL, *mcc = NULL; ogs_assert(self.num_of_plmn_support <= OGS_MAX_NUM_OF_PLMN); if (ogs_yaml_iter_type(&plmn_support_array) == YAML_MAPPING_NODE) { memcpy(&plmn_support_iter, &plmn_support_array, sizeof(ogs_yaml_iter_t)); } else if (ogs_yaml_iter_type(&plmn_support_array) == YAML_SEQUENCE_NODE) { if (!ogs_yaml_iter_next(&plmn_support_array)) break; ogs_yaml_iter_recurse(&plmn_support_array, &plmn_support_iter); } else if (ogs_yaml_iter_type(&plmn_support_array) == YAML_SCALAR_NODE) { break; } else ogs_assert_if_reached(); while (ogs_yaml_iter_next(&plmn_support_iter)) { const char *plmn_support_key = ogs_yaml_iter_key(&plmn_support_iter); ogs_assert(plmn_support_key); if (!strcmp(plmn_support_key, "plmn_id")) { ogs_yaml_iter_t plmn_id_iter; ogs_yaml_iter_recurse(&plmn_support_iter, &plmn_id_iter); while (ogs_yaml_iter_next(&plmn_id_iter)) { const char *plmn_id_key = ogs_yaml_iter_key(&plmn_id_iter); ogs_assert(plmn_id_key); if (!strcmp(plmn_id_key, "mcc")) { mcc = ogs_yaml_iter_value( &plmn_id_iter); } else if (!strcmp(plmn_id_key, "mnc")) { mnc = ogs_yaml_iter_value( &plmn_id_iter); } } if (mcc && mnc) { ogs_plmn_id_build( &self.plmn_support[ self.num_of_plmn_support]. plmn_id, atoi(mcc), atoi(mnc), strlen(mnc)); } } else if (!strcmp(plmn_support_key, "s_nssai")) { ogs_yaml_iter_t s_nssai_array, s_nssai_iter; ogs_yaml_iter_recurse(&plmn_support_iter, &s_nssai_array); do { ogs_s_nssai_t *s_nssai = NULL; const char *sst = NULL, *sd = NULL; ogs_assert( self.plmn_support[ self.num_of_plmn_support]. num_of_s_nssai <= OGS_MAX_NUM_OF_S_NSSAI); s_nssai = &self.plmn_support[ self.num_of_plmn_support].s_nssai[ self.plmn_support[ self.num_of_plmn_support]. num_of_s_nssai]; ogs_assert(s_nssai); if (ogs_yaml_iter_type(&s_nssai_array) == YAML_MAPPING_NODE) { memcpy(&s_nssai_iter, &s_nssai_array, sizeof(ogs_yaml_iter_t)); } else if (ogs_yaml_iter_type( &s_nssai_array) == YAML_SEQUENCE_NODE) { if (!ogs_yaml_iter_next(&s_nssai_array)) break; ogs_yaml_iter_recurse(&s_nssai_array, &s_nssai_iter); } else if (ogs_yaml_iter_type( &s_nssai_array) == YAML_SCALAR_NODE) { break; } else ogs_assert_if_reached(); while (ogs_yaml_iter_next(&s_nssai_iter)) { const char *s_nssai_key = ogs_yaml_iter_key(&s_nssai_iter); ogs_assert(s_nssai_key); if (!strcmp(s_nssai_key, "sst")) { sst = ogs_yaml_iter_value( &s_nssai_iter); } else if (!strcmp( s_nssai_key, "sd")) { sd = ogs_yaml_iter_value( &s_nssai_iter); } } if (sst) { s_nssai->sst = atoi(sst); if (sd) s_nssai->sd = ogs_uint24_from_string( (char*)sd); else s_nssai->sd.v = OGS_S_NSSAI_NO_SD_VALUE; s_nssai->mapped_hplmn_sst = 0; s_nssai->mapped_hplmn_sd.v = OGS_S_NSSAI_NO_SD_VALUE; self.plmn_support[ self.num_of_plmn_support]. num_of_s_nssai++; } } while (ogs_yaml_iter_type(&s_nssai_array) == YAML_SEQUENCE_NODE); } else ogs_warn("unknown key `%s`", plmn_support_key); } if (self.plmn_support[ self.num_of_plmn_support].num_of_s_nssai && mcc && mnc) { self.num_of_plmn_support++; } else { ogs_warn("Ignore plmn : " "s_nssai(%d) mcc(%s), mnc(%s)", self.plmn_support[ self.num_of_plmn_support].num_of_s_nssai, mcc, mnc); self.plmn_support[ self.num_of_plmn_support].num_of_s_nssai = 0; } } while (ogs_yaml_iter_type(&plmn_support_array) == YAML_SEQUENCE_NODE); } else if (!strcmp(amf_key, "security")) { ogs_yaml_iter_t security_iter; ogs_yaml_iter_recurse(&amf_iter, &security_iter); while (ogs_yaml_iter_next(&security_iter)) { const char *security_key = ogs_yaml_iter_key(&security_iter); ogs_assert(security_key); if (!strcmp(security_key, "integrity_order")) { ogs_yaml_iter_t integrity_order_iter; ogs_yaml_iter_recurse(&security_iter, &integrity_order_iter); ogs_assert(ogs_yaml_iter_type( &integrity_order_iter) != YAML_MAPPING_NODE); do { const char *v = NULL; if (ogs_yaml_iter_type(&integrity_order_iter) == YAML_SEQUENCE_NODE) { if (!ogs_yaml_iter_next( &integrity_order_iter)) break; } v = ogs_yaml_iter_value(&integrity_order_iter); if (v) { int integrity_index = self.num_of_integrity_order; if (strcmp(v, "NIA0") == 0) { self.integrity_order[integrity_index] = OGS_NAS_SECURITY_ALGORITHMS_NIA0; self.num_of_integrity_order++; } else if (strcmp(v, "NIA1") == 0) { self.integrity_order[integrity_index] = OGS_NAS_SECURITY_ALGORITHMS_128_NIA1; self.num_of_integrity_order++; } else if (strcmp(v, "NIA2") == 0) { self.integrity_order[integrity_index] = OGS_NAS_SECURITY_ALGORITHMS_128_NIA2; self.num_of_integrity_order++; } else if (strcmp(v, "NIA3") == 0) { self.integrity_order[integrity_index] = OGS_NAS_SECURITY_ALGORITHMS_128_NIA3; self.num_of_integrity_order++; } } } while ( ogs_yaml_iter_type(&integrity_order_iter) == YAML_SEQUENCE_NODE); } else if (!strcmp(security_key, "ciphering_order")) { ogs_yaml_iter_t ciphering_order_iter; ogs_yaml_iter_recurse(&security_iter, &ciphering_order_iter); ogs_assert(ogs_yaml_iter_type( &ciphering_order_iter) != YAML_MAPPING_NODE); do { const char *v = NULL; if (ogs_yaml_iter_type(&ciphering_order_iter) == YAML_SEQUENCE_NODE) { if (!ogs_yaml_iter_next( &ciphering_order_iter)) break; } v = ogs_yaml_iter_value(&ciphering_order_iter); if (v) { int ciphering_index = self.num_of_ciphering_order; if (strcmp(v, "NEA0") == 0) { self.ciphering_order[ciphering_index] = OGS_NAS_SECURITY_ALGORITHMS_NEA0; self.num_of_ciphering_order++; } else if (strcmp(v, "NEA1") == 0) { self.ciphering_order[ciphering_index] = OGS_NAS_SECURITY_ALGORITHMS_128_NEA1; self.num_of_ciphering_order++; } else if (strcmp(v, "NEA2") == 0) { self.ciphering_order[ciphering_index] = OGS_NAS_SECURITY_ALGORITHMS_128_NEA2; self.num_of_ciphering_order++; } else if (strcmp(v, "NEA3") == 0) { self.ciphering_order[ciphering_index] = OGS_NAS_SECURITY_ALGORITHMS_128_NEA3; self.num_of_ciphering_order++; } } } while ( ogs_yaml_iter_type(&ciphering_order_iter) == YAML_SEQUENCE_NODE); } } } else if (!strcmp(amf_key, "network_name")) { ogs_yaml_iter_t network_name_iter; ogs_yaml_iter_recurse(&amf_iter, &network_name_iter); while (ogs_yaml_iter_next(&network_name_iter)) { const char *network_name_key = ogs_yaml_iter_key(&network_name_iter); ogs_assert(network_name_key); if (!strcmp(network_name_key, "full")) { ogs_nas_network_name_t *network_full_name = &self.full_name; const char *c_network_name = ogs_yaml_iter_value(&network_name_iter); uint8_t size = strlen(c_network_name); uint8_t i; for (i = 0;iname[i*2] = 0; network_full_name->name[(i*2)+1] = c_network_name[i]; } network_full_name->length = size*2+1; network_full_name->coding_scheme = 1; } else if (!strcmp(network_name_key, "short")) { ogs_nas_network_name_t *network_short_name = &self.short_name; const char *c_network_name = ogs_yaml_iter_value(&network_name_iter); uint8_t size = strlen(c_network_name); uint8_t i; for (i = 0;iname[i*2] = 0; network_short_name->name[(i*2)+1] = c_network_name[i]; } network_short_name->length = size*2+1; network_short_name->coding_scheme = 1; } } } else if (!strcmp(amf_key, "amf_name")) { self.amf_name = ogs_yaml_iter_value(&amf_iter); } else if (!strcmp(amf_key, "sbi")) { /* handle config in sbi library */ } else ogs_warn("unknown key `%s`", amf_key); } } } rv = amf_context_validation(); if (rv != OGS_OK) return rv; return OGS_OK; } amf_gnb_t *amf_gnb_add(ogs_sock_t *sock, ogs_sockaddr_t *addr) { amf_gnb_t *gnb = NULL; amf_event_t e; ogs_assert(sock); ogs_assert(addr); ogs_pool_alloc(&amf_gnb_pool, &gnb); ogs_assert(gnb); memset(gnb, 0, sizeof *gnb); /* Defaut RAT-Type */ gnb->rat_type = OpenAPI_rat_type_NR; gnb->sctp.sock = sock; gnb->sctp.addr = addr; gnb->sctp.type = amf_gnb_sock_type(gnb->sctp.sock); if (gnb->sctp.type == SOCK_STREAM) { gnb->sctp.poll.read = ogs_pollset_add(ogs_app()->pollset, OGS_POLLIN, sock->fd, ngap_recv_upcall, sock); ogs_assert(gnb->sctp.poll.read); } gnb->max_num_of_ostreams = DEFAULT_SCTP_MAX_NUM_OF_OSTREAMS; gnb->ostream_id = 0; if (ogs_app()->sockopt.sctp.max_num_of_ostreams) { gnb->max_num_of_ostreams = ogs_app()->sockopt.sctp.max_num_of_ostreams; ogs_info("[GNB] max_num_of_ostreams : %d", gnb->max_num_of_ostreams); } ogs_list_init(&gnb->ran_ue_list); ogs_hash_set(self.gnb_addr_hash, gnb->sctp.addr, sizeof(ogs_sockaddr_t), gnb); memset(&e, 0, sizeof(e)); e.gnb = gnb; ogs_fsm_create(&gnb->sm, ngap_state_initial, ngap_state_final); ogs_fsm_init(&gnb->sm, &e); ogs_list_add(&self.gnb_list, gnb); ogs_info("[Added] Number of gNBs is now %d", ogs_list_count(&self.gnb_list)); return gnb; } int amf_gnb_remove(amf_gnb_t *gnb) { amf_event_t e; ogs_assert(gnb); ogs_assert(gnb->sctp.sock); ogs_list_remove(&self.gnb_list, gnb); memset(&e, 0, sizeof(e)); e.gnb = gnb; ogs_fsm_fini(&gnb->sm, &e); ogs_fsm_delete(&gnb->sm); ogs_hash_set(self.gnb_addr_hash, gnb->sctp.addr, sizeof(ogs_sockaddr_t), NULL); ogs_hash_set(self.gnb_id_hash, &gnb->gnb_id, sizeof(gnb->gnb_id), NULL); ogs_sctp_flush_and_destroy(&gnb->sctp); ogs_pool_free(&amf_gnb_pool, gnb); ogs_info("[Removed] Number of gNBs is now %d", ogs_list_count(&self.gnb_list)); return OGS_OK; } int amf_gnb_remove_all() { amf_gnb_t *gnb = NULL, *next_gnb = NULL; ogs_list_for_each_safe(&self.gnb_list, next_gnb, gnb) amf_gnb_remove(gnb); return OGS_OK; } amf_gnb_t *amf_gnb_find_by_addr(ogs_sockaddr_t *addr) { ogs_assert(addr); return (amf_gnb_t *)ogs_hash_get(self.gnb_addr_hash, addr, sizeof(ogs_sockaddr_t)); return NULL; } amf_gnb_t *amf_gnb_find_by_gnb_id(uint32_t gnb_id) { return (amf_gnb_t *)ogs_hash_get(self.gnb_id_hash, &gnb_id, sizeof(gnb_id)); } int amf_gnb_set_gnb_id(amf_gnb_t *gnb, uint32_t gnb_id) { ogs_assert(gnb); gnb->gnb_id = gnb_id; ogs_hash_set(self.gnb_id_hash, &gnb->gnb_id, sizeof(gnb->gnb_id), gnb); return OGS_OK; } int amf_gnb_sock_type(ogs_sock_t *sock) { ogs_socknode_t *snode = NULL; ogs_assert(sock); ogs_list_for_each(&amf_self()->ngap_list, snode) if (snode->sock == sock) return SOCK_SEQPACKET; ogs_list_for_each(&amf_self()->ngap_list6, snode) if (snode->sock == sock) return SOCK_SEQPACKET; return SOCK_STREAM; } /** ran_ue_context handling function */ ran_ue_t *ran_ue_add(amf_gnb_t *gnb, uint32_t ran_ue_ngap_id) { ran_ue_t *ran_ue = NULL; ogs_assert(gnb); ogs_pool_alloc(&ran_ue_pool, &ran_ue); ogs_assert(ran_ue); memset(ran_ue, 0, sizeof *ran_ue); ran_ue->index = ogs_pool_index(&ran_ue_pool, ran_ue); ogs_assert(ran_ue->index > 0 && ran_ue->index <= ogs_app()->max.ue); ran_ue->ran_ue_ngap_id = ran_ue_ngap_id; ran_ue->amf_ue_ngap_id = ran_ue->index; /* * SCTP output stream identification * Default ogs_app()->parameter.sctp_streams : 30 * 0 : Non UE signalling * 1-29 : UE specific association */ ran_ue->gnb_ostream_id = OGS_NEXT_ID(gnb->ostream_id, 1, gnb->max_num_of_ostreams-1); ran_ue->gnb = gnb; ogs_list_add(&gnb->ran_ue_list, ran_ue); stats_add_ran_ue(); return ran_ue; } void ran_ue_remove(ran_ue_t *ran_ue) { ogs_assert(ran_ue); ogs_assert(ran_ue->gnb); ogs_list_remove(&ran_ue->gnb->ran_ue_list, ran_ue); if (ran_ue->t_ng_holding) ogs_timer_delete(ran_ue->t_ng_holding); ogs_pool_free(&ran_ue_pool, ran_ue); stats_remove_ran_ue(); } void ran_ue_switch_to_gnb(ran_ue_t *ran_ue, amf_gnb_t *new_gnb) { ogs_assert(ran_ue); ogs_assert(ran_ue->gnb); ogs_assert(new_gnb); /* Remove from the old gnb */ ogs_list_remove(&ran_ue->gnb->ran_ue_list, ran_ue); /* Add to the new gnb */ ogs_list_add(&new_gnb->ran_ue_list, ran_ue); /* Switch to gnb */ ran_ue->gnb = new_gnb; } ran_ue_t *ran_ue_find_by_ran_ue_ngap_id( amf_gnb_t *gnb, uint32_t ran_ue_ngap_id) { ran_ue_t *ran_ue = NULL; ogs_list_for_each(&gnb->ran_ue_list, ran_ue) { if (ran_ue_ngap_id == ran_ue->ran_ue_ngap_id) break; } return ran_ue; } ran_ue_t *ran_ue_find(uint32_t index) { ogs_assert(index); return ogs_pool_find(&ran_ue_pool, index); } ran_ue_t *ran_ue_find_by_amf_ue_ngap_id(uint64_t amf_ue_ngap_id) { return ran_ue_find(amf_ue_ngap_id); } ran_ue_t *ran_ue_cycle(ran_ue_t *ran_ue) { return ogs_pool_cycle(&ran_ue_pool, ran_ue); } static int amf_ue_new_guti(amf_ue_t *amf_ue) { if (amf_ue->m_tmsi) { /* AMF has a VALID GUTI * As such, we need to remove previous GUTI in hash table */ ogs_hash_set(self.guti_ue_hash, &amf_ue->guti, sizeof(ogs_nas_5gs_guti_t), NULL); ogs_assert(amf_m_tmsi_free(amf_ue->m_tmsi) == OGS_OK); } memset(&amf_ue->guti, 0, sizeof(ogs_nas_5gs_guti_t)); ogs_assert(amf_ue->guami); ogs_nas_from_plmn_id(&amf_ue->guti.nas_plmn_id, &amf_ue->guami->plmn_id); memcpy(&amf_ue->guti.amf_id, &amf_ue->guami->amf_id, sizeof(ogs_amf_id_t)); amf_ue->m_tmsi = amf_m_tmsi_alloc(); ogs_assert(amf_ue->m_tmsi); amf_ue->guti.m_tmsi = *(amf_ue->m_tmsi); ogs_hash_set(self.guti_ue_hash, &amf_ue->guti, sizeof(ogs_nas_5gs_guti_t), amf_ue); return OGS_OK; } amf_ue_t *amf_ue_add(ran_ue_t *ran_ue) { amf_gnb_t *gnb = NULL; amf_ue_t *amf_ue = NULL; ogs_assert(ran_ue); gnb = ran_ue->gnb; ogs_assert(gnb); ogs_pool_alloc(&amf_ue_pool, &amf_ue); ogs_assert(amf_ue); memset(amf_ue, 0, sizeof *amf_ue); /* SBI Type */ amf_ue->sbi.type = OGS_SBI_OBJ_UE_TYPE; /* SBI Features */ OGS_SBI_FEATURES_SET(amf_ue->am_policy_control_features, OGS_SBI_NPCF_AM_POLICY_CONTROL_UE_AMBR_AUTHORIZATION); ogs_list_init(&amf_ue->sess_list); /* TODO : Hard-coded */ amf_ue->guami = &amf_self()->served_guami[0]; amf_ue->nas.access_type = OGS_ACCESS_TYPE_3GPP; amf_ue->abba_len = 2; /* Create New GUTI */ amf_ue_new_guti(amf_ue); /* Add All Timers */ amf_ue->t3513.timer = ogs_timer_add( ogs_app()->timer_mgr, amf_timer_t3513_expire, amf_ue); amf_ue->t3513.pkbuf = NULL; amf_ue->t3522.timer = ogs_timer_add( ogs_app()->timer_mgr, amf_timer_t3522_expire, amf_ue); amf_ue->t3522.pkbuf = NULL; amf_ue->t3550.timer = ogs_timer_add( ogs_app()->timer_mgr, amf_timer_t3550_expire, amf_ue); amf_ue->t3550.pkbuf = NULL; amf_ue->t3555.timer = ogs_timer_add( ogs_app()->timer_mgr, amf_timer_t3555_expire, amf_ue); amf_ue->t3555.pkbuf = NULL; amf_ue->t3560.timer = ogs_timer_add( ogs_app()->timer_mgr, amf_timer_t3560_expire, amf_ue); amf_ue->t3560.pkbuf = NULL; amf_ue->t3570.timer = ogs_timer_add( ogs_app()->timer_mgr, amf_timer_t3570_expire, amf_ue); amf_ue->t3570.pkbuf = NULL; amf_ue_fsm_init(amf_ue); ogs_list_add(&self.amf_ue_list, amf_ue); ogs_info("[Added] Number of AMF-UEs is now %d", ogs_list_count(&self.amf_ue_list)); return amf_ue; } void amf_ue_remove(amf_ue_t *amf_ue) { int i; ogs_assert(amf_ue); ogs_list_remove(&self.amf_ue_list, amf_ue); amf_ue_fsm_fini(amf_ue); /* Clear hash table */ if (amf_ue->m_tmsi) { ogs_hash_set(self.guti_ue_hash, &amf_ue->guti, sizeof(ogs_nas_5gs_guti_t), NULL); ogs_assert(amf_m_tmsi_free(amf_ue->m_tmsi) == OGS_OK); } if (amf_ue->suci) { ogs_hash_set(self.suci_hash, amf_ue->suci, strlen(amf_ue->suci), NULL); ogs_free(amf_ue->suci); } if (amf_ue->supi) { ogs_hash_set(self.supi_hash, amf_ue->supi, strlen(amf_ue->supi), NULL); ogs_free(amf_ue->supi); } if (amf_ue->pei) ogs_free(amf_ue->pei); for (i = 0; i < amf_ue->num_of_msisdn; i++) { ogs_assert(amf_ue->msisdn[i]); ogs_free(amf_ue->msisdn[i]); } for (i = 0; i < amf_ue->num_of_subscribed_dnn; i++) { ogs_assert(amf_ue->subscribed_dnn[i]); ogs_free(amf_ue->subscribed_dnn[i]); } if (amf_ue->policy_association_id) ogs_free(amf_ue->policy_association_id); if (amf_ue->confirmation_url_for_5g_aka) ogs_free(amf_ue->confirmation_url_for_5g_aka); /* Free UeRadioCapability */ OGS_ASN_CLEAR_DATA(&amf_ue->ueRadioCapability); /* Clear Transparent Container */ OGS_ASN_CLEAR_DATA(&amf_ue->container); /* Clear Paging Info */ AMF_UE_CLEAR_PAGING_INFO(amf_ue); /* Clear N2 Transfer */ AMF_UE_CLEAR_N2_TRANSFER(amf_ue, pdu_session_resource_setup_request); /* Delete All Timers */ CLEAR_AMF_UE_ALL_TIMERS(amf_ue); ogs_timer_delete(amf_ue->t3513.timer); ogs_timer_delete(amf_ue->t3522.timer); ogs_timer_delete(amf_ue->t3550.timer); ogs_timer_delete(amf_ue->t3555.timer); ogs_timer_delete(amf_ue->t3560.timer); ogs_timer_delete(amf_ue->t3570.timer); /* Free SBI object memory */ ogs_sbi_object_free(&amf_ue->sbi); amf_ue_deassociate(amf_ue); amf_sess_remove_all(amf_ue); ogs_pool_free(&amf_ue_pool, amf_ue); ogs_info("[Removed] Number of AMF-UEs is now %d", ogs_list_count(&self.amf_ue_list)); } void amf_ue_remove_all() { amf_ue_t *amf_ue = NULL, *next = NULL;; ogs_list_for_each_safe(&self.amf_ue_list, next, amf_ue) { ran_ue_t *ran_ue = ran_ue_cycle(amf_ue->ran_ue); if (ran_ue) ran_ue_remove(ran_ue); amf_ue_remove(amf_ue); } } void amf_ue_fsm_init(amf_ue_t *amf_ue) { amf_event_t e; ogs_assert(amf_ue); memset(&e, 0, sizeof(e)); e.amf_ue = amf_ue; ogs_fsm_create(&amf_ue->sm, gmm_state_initial, gmm_state_final); ogs_fsm_init(&amf_ue->sm, &e); } void amf_ue_fsm_fini(amf_ue_t *amf_ue) { amf_event_t e; ogs_assert(amf_ue); memset(&e, 0, sizeof(e)); e.amf_ue = amf_ue; ogs_fsm_fini(&amf_ue->sm, &e); ogs_fsm_delete(&amf_ue->sm); } amf_ue_t *amf_ue_find_by_guti(ogs_nas_5gs_guti_t *guti) { ogs_assert(guti); return (amf_ue_t *)ogs_hash_get( self.guti_ue_hash, guti, sizeof(ogs_nas_5gs_guti_t)); } amf_ue_t *amf_ue_find_by_teid(uint32_t teid) { return ogs_pool_find(&amf_ue_pool, teid); } amf_ue_t *amf_ue_find_by_suci(char *suci) { ogs_assert(suci); return (amf_ue_t *)ogs_hash_get(self.suci_hash, suci, strlen(suci)); } amf_ue_t *amf_ue_find_by_supi(char *supi) { ogs_assert(supi); return (amf_ue_t *)ogs_hash_get(self.supi_hash, supi, strlen(supi)); } amf_ue_t *amf_ue_find_by_message(ogs_nas_5gs_message_t *message) { amf_ue_t *amf_ue = NULL; ogs_nas_5gs_registration_request_t *registration_request = NULL; #if 0 ogs_nas_5gs_tracking_area_update_request_t *tau_request = NULL; #endif ogs_nas_5gs_mobile_identity_t *mobile_identity = NULL; ogs_nas_5gs_mobile_identity_header_t *mobile_identity_header = NULL; ogs_nas_5gs_mobile_identity_guti_t *mobile_identity_guti = NULL; ogs_nas_5gs_guti_t nas_guti; char *suci = NULL; ogs_assert(message); switch (message->gmm.h.message_type) { case OGS_NAS_5GS_REGISTRATION_REQUEST: registration_request = &message->gmm.registration_request; ogs_assert(registration_request); mobile_identity = ®istration_request->mobile_identity; ogs_assert(mobile_identity); if (!mobile_identity->length || !mobile_identity->buffer) return NULL; mobile_identity_header = (ogs_nas_5gs_mobile_identity_header_t *)mobile_identity->buffer; switch (mobile_identity_header->type) { case OGS_NAS_5GS_MOBILE_IDENTITY_SUCI: suci = ogs_nas_5gs_suci_from_mobile_identity(mobile_identity); ogs_assert(suci); amf_ue = amf_ue_find_by_suci(suci); if (amf_ue) { ogs_info("[%s] known UE by SUCI", suci); } else { ogs_info("[%s] Unknown UE by SUCI", suci); } ogs_free(suci); break; case OGS_NAS_5GS_MOBILE_IDENTITY_GUTI: mobile_identity_guti = (ogs_nas_5gs_mobile_identity_guti_t *)mobile_identity->buffer; if (!mobile_identity_guti) { ogs_error("No mobile identity"); return NULL; } ogs_nas_5gs_mobile_identity_guti_to_nas_guti( mobile_identity_guti, &nas_guti); amf_ue = amf_ue_find_by_guti(&nas_guti); if (amf_ue) { ogs_info("[%s] Known UE by 5G-S_TMSI[AMF_ID:0x%x,M_TMSI:0x%x]", amf_ue->suci ? amf_ue->suci : "Unknown", ogs_amf_id_hexdump(&nas_guti.amf_id), nas_guti.m_tmsi); } else { ogs_info("Unknown UE by 5G-S_TMSI[AMF_ID:0x%x,M_TMSI:0x%x]", ogs_amf_id_hexdump(&nas_guti.amf_id), nas_guti.m_tmsi); } break; default: ogs_error("Unknown SUCI type [%d]", mobile_identity_header->type); } break; default: break; } return amf_ue; } void amf_ue_set_suci(amf_ue_t *amf_ue, ogs_nas_5gs_mobile_identity_t *mobile_identity) { amf_ue_t *old_amf_ue = NULL; char *suci = NULL; ogs_assert(amf_ue); ogs_assert(mobile_identity); suci = ogs_nas_5gs_suci_from_mobile_identity(mobile_identity); ogs_assert(suci); /* Check if OLD amf_ue_t is existed */ old_amf_ue = amf_ue_find_by_suci(suci); if (old_amf_ue) { /* Check if OLD amf_ue_t is different with NEW amf_ue_t */ if (ogs_pool_index(&amf_ue_pool, amf_ue) != ogs_pool_index(&amf_ue_pool, old_amf_ue)) { ogs_warn("[%s] OLD UE Context Release", suci); if (CM_CONNECTED(old_amf_ue)) { /* Implcit NG release */ ogs_info("[%s] Implicit NG release", suci); ogs_info("[%s] RAN_UE_NGAP_ID[%d] AMF_UE_NGAP_ID[%lld]", old_amf_ue->suci, old_amf_ue->ran_ue->ran_ue_ngap_id, (long long)old_amf_ue->ran_ue->amf_ue_ngap_id); ran_ue_remove(old_amf_ue->ran_ue); } amf_ue_remove(old_amf_ue); } } if (amf_ue->suci) { ogs_hash_set(self.suci_hash, amf_ue->suci, strlen(amf_ue->suci), NULL); ogs_free(amf_ue->suci); } amf_ue->suci = suci; ogs_hash_set(self.suci_hash, amf_ue->suci, strlen(amf_ue->suci), amf_ue); amf_ue->guti_present = 1; } void amf_ue_set_supi(amf_ue_t *amf_ue, char *supi) { ogs_assert(supi); if (amf_ue->supi) { ogs_hash_set(self.supi_hash, amf_ue->supi, strlen(amf_ue->supi), NULL); ogs_free(amf_ue->supi); } amf_ue->supi = ogs_strdup(supi); ogs_assert(amf_ue->supi); ogs_hash_set(self.supi_hash, amf_ue->supi, strlen(amf_ue->supi), amf_ue); } OpenAPI_rat_type_e amf_ue_rat_type(amf_ue_t *amf_ue) { amf_gnb_t *gnb = NULL; ran_ue_t *ran_ue = NULL; ran_ue = amf_ue->ran_ue; ogs_assert(ran_ue); gnb = ran_ue->gnb; ogs_assert(gnb); return gnb->rat_type; } void amf_ue_associate_ran_ue(amf_ue_t *amf_ue, ran_ue_t *ran_ue) { ogs_assert(amf_ue); ogs_assert(ran_ue); amf_ue->ran_ue = ran_ue; ran_ue->amf_ue = amf_ue; } void ran_ue_deassociate(ran_ue_t *ran_ue) { ogs_assert(ran_ue); ran_ue->amf_ue = NULL; } void amf_ue_deassociate(amf_ue_t *amf_ue) { ogs_assert(amf_ue); amf_ue->ran_ue = NULL; } void source_ue_associate_target_ue( ran_ue_t *source_ue, ran_ue_t *target_ue) { amf_ue_t *amf_ue = NULL; ogs_assert(source_ue); ogs_assert(target_ue); amf_ue = source_ue->amf_ue; ogs_assert(amf_ue); target_ue->amf_ue = amf_ue; target_ue->source_ue = source_ue; source_ue->target_ue = target_ue; } void source_ue_deassociate_target_ue(ran_ue_t *ran_ue) { ran_ue_t *source_ue = NULL; ran_ue_t *target_ue = NULL; ogs_assert(ran_ue); if (ran_ue->target_ue) { source_ue = ran_ue; target_ue = ran_ue->target_ue; ogs_assert(source_ue->target_ue); ogs_assert(target_ue->source_ue); source_ue->target_ue = NULL; target_ue->source_ue = NULL; } else if (ran_ue->source_ue) { target_ue = ran_ue; source_ue = ran_ue->source_ue; ogs_assert(source_ue->target_ue); ogs_assert(target_ue->source_ue); source_ue->target_ue = NULL; target_ue->source_ue = NULL; } } amf_sess_t *amf_sess_add(amf_ue_t *amf_ue, uint8_t psi) { amf_sess_t *sess = NULL; ogs_assert(amf_ue); ogs_assert(psi != OGS_NAS_PDU_SESSION_IDENTITY_UNASSIGNED); ogs_pool_alloc(&amf_sess_pool, &sess); ogs_assert(sess); memset(sess, 0, sizeof *sess); sess->sbi.type = OGS_SBI_OBJ_SESS_TYPE; sess->amf_ue = amf_ue; sess->psi = psi; sess->s_nssai.sst = 0; sess->s_nssai.sd.v = OGS_S_NSSAI_NO_SD_VALUE; sess->s_nssai.mapped_hplmn_sst = 0; sess->s_nssai.mapped_hplmn_sd.v = OGS_S_NSSAI_NO_SD_VALUE; ogs_list_add(&amf_ue->sess_list, sess); stats_add_amf_session(); return sess; } void amf_sess_remove(amf_sess_t *sess) { ogs_assert(sess); ogs_assert(sess->amf_ue); ogs_list_remove(&sess->amf_ue->sess_list, sess); /* Free SBI object memory */ ogs_sbi_object_free(&sess->sbi); if (sess->sm_context_ref) ogs_free(sess->sm_context_ref); if (sess->payload_container) ogs_pkbuf_free(sess->payload_container); if (sess->dnn) ogs_free(sess->dnn); if (sess->pdu_session_establishment_accept) ogs_pkbuf_free(sess->pdu_session_establishment_accept); if (sess->transfer.pdu_session_resource_setup_request) ogs_pkbuf_free(sess->transfer.pdu_session_resource_setup_request); sess->transfer.pdu_session_resource_setup_request = NULL; if (sess->transfer.path_switch_request_ack) ogs_pkbuf_free(sess->transfer.path_switch_request_ack); sess->transfer.path_switch_request_ack = NULL; if (sess->transfer.pdu_session_resource_release_command) ogs_pkbuf_free(sess->transfer.pdu_session_resource_release_command); sess->transfer.pdu_session_resource_release_command = NULL; if (sess->paging.client) ogs_sbi_client_remove(sess->paging.client); OGS_NAS_CLEAR_DATA(&sess->ue_pco); OGS_TLV_CLEAR_DATA(&sess->pgw_pco); ogs_pool_free(&amf_sess_pool, sess); stats_remove_amf_session(); } void amf_sess_remove_all(amf_ue_t *amf_ue) { amf_sess_t *sess = NULL, *next_sess = NULL; ogs_assert(amf_ue); ogs_list_for_each_safe(&amf_ue->sess_list, next_sess, sess) amf_sess_remove(sess); } amf_sess_t *amf_sess_find_by_psi(amf_ue_t *amf_ue, uint8_t psi) { amf_sess_t *sess = NULL; ogs_list_for_each(&amf_ue->sess_list, sess) if (psi == sess->psi) return sess; return NULL; } amf_ue_t *amf_ue_cycle(amf_ue_t *amf_ue) { return ogs_pool_cycle(&amf_ue_pool, amf_ue); } amf_sess_t *amf_sess_cycle(amf_sess_t *sess) { return ogs_pool_cycle(&amf_sess_pool, sess); } int amf_sess_xact_count(amf_ue_t *amf_ue) { amf_sess_t *sess = NULL; int xact_count = 0; ogs_assert(amf_ue); ogs_list_for_each(&amf_ue->sess_list, sess) xact_count += ogs_list_count(&sess->sbi.xact_list); return xact_count; } int amf_sess_xact_state_count(amf_ue_t *amf_ue, int state) { amf_sess_t *sess = NULL; ogs_sbi_xact_t *xact = NULL; int xact_count = 0; ogs_assert(amf_ue); ogs_assert(state); ogs_list_for_each(&amf_ue->sess_list, sess) { ogs_list_for_each(&sess->sbi.xact_list, xact) if (xact->state == state) xact_count++; } return xact_count; } bool amf_pdu_res_setup_req_transfer_needed(amf_ue_t *amf_ue) { amf_sess_t *sess = NULL; ogs_assert(amf_ue); ogs_list_for_each(&amf_ue->sess_list, sess) if (sess->transfer.pdu_session_resource_setup_request) return true; return false; } int amf_find_served_tai(ogs_5gs_tai_t *tai) { int i = 0, j = 0, k = 0; ogs_assert(tai); for (i = 0; i < self.num_of_served_tai; i++) { ogs_5gs_tai0_list_t *list0 = &self.served_tai[i].list0; ogs_assert(list0); ogs_5gs_tai2_list_t *list2 = &self.served_tai[i].list2; ogs_assert(list2); for (j = 0; list0->tai[j].num; j++) { ogs_assert(list0->tai[j].type == OGS_TAI0_TYPE); ogs_assert(list0->tai[j].num < OGS_MAX_NUM_OF_TAI); for (k = 0; k < list0->tai[j].num; k++) { if (memcmp(&list0->tai[j].plmn_id, &tai->plmn_id, OGS_PLMN_ID_LEN) == 0 && list0->tai[j].tac[k].v == tai->tac.v) { return i; } } } if (list2->num) { ogs_assert(list2->type == OGS_TAI1_TYPE || list2->type == OGS_TAI2_TYPE); ogs_assert(list2->num < OGS_MAX_NUM_OF_TAI); for (j = 0; j < list2->num; j++) { if (memcmp(&list2->tai[j].plmn_id, &tai->plmn_id, OGS_PLMN_ID_LEN) == 0 && list2->tai[j].tac.v == tai->tac.v) { return i; } } } } return -1; } ogs_s_nssai_t *amf_find_s_nssai( ogs_plmn_id_t *served_plmn_id, ogs_s_nssai_t *s_nssai) { int i, j; ogs_assert(served_plmn_id); ogs_assert(s_nssai); for (i = 0; i < amf_self()->num_of_plmn_support; i++) { if (memcmp(&amf_self()->plmn_support[i].plmn_id, served_plmn_id, OGS_PLMN_ID_LEN) != 0) continue; for (j = 0; j < amf_self()->plmn_support[i].num_of_s_nssai; j++) { if (amf_self()->plmn_support[i].s_nssai[j].sst != s_nssai->sst) continue; if (amf_self()->plmn_support[i].s_nssai[j].sd.v != OGS_S_NSSAI_NO_SD_VALUE && s_nssai->sd.v != OGS_S_NSSAI_NO_SD_VALUE) { if (amf_self()->plmn_support[i].s_nssai[j].sd.v != s_nssai->sd.v) continue; } return &amf_self()->plmn_support[i].s_nssai[j]; } } return NULL; } int amf_m_tmsi_pool_generate() { int i, j; int index = 0; ogs_trace("M-TMSI Pool try to generate..."); for (i = 0; index < ogs_app()->max.ue; i++) { amf_m_tmsi_t *m_tmsi = NULL; int conflict = 0; m_tmsi = &self.m_tmsi.array[index]; ogs_assert(m_tmsi); *m_tmsi = ogs_random32(); /* for mapped-GUTI */ *m_tmsi |= 0xc0000000; *m_tmsi &= 0xff00ffff; for (j = 0; j < index; j++) { if (*m_tmsi == self.m_tmsi.array[j]) { conflict = 1; ogs_trace("[M-TMSI CONFLICT] %d:0x%x == %d:0x%x", index, *m_tmsi, j, self.m_tmsi.array[j]); break; } } if (conflict == 1) { continue; } index++; } self.m_tmsi.size = index; ogs_trace("M-TMSI Pool generate...done"); return OGS_OK; } amf_m_tmsi_t *amf_m_tmsi_alloc() { amf_m_tmsi_t *m_tmsi = NULL; ogs_pool_alloc(&self.m_tmsi, &m_tmsi); ogs_assert(m_tmsi); return m_tmsi; } int amf_m_tmsi_free(amf_m_tmsi_t *m_tmsi) { ogs_assert(m_tmsi); ogs_pool_free(&self.m_tmsi, m_tmsi); return OGS_OK; } uint8_t amf_selected_int_algorithm(amf_ue_t *amf_ue) { int i; ogs_assert(amf_ue); for (i = 0; i < amf_self()->num_of_integrity_order; i++) { if (amf_ue->ue_security_capability.nr_ia & (0x80 >> amf_self()->integrity_order[i])) { return amf_self()->integrity_order[i]; } } return 0; } uint8_t amf_selected_enc_algorithm(amf_ue_t *amf_ue) { int i; ogs_assert(amf_ue); for (i = 0; i < amf_self()->num_of_ciphering_order; i++) { if (amf_ue->ue_security_capability.nr_ea & (0x80 >> amf_self()->ciphering_order[i])) { return amf_self()->ciphering_order[i]; } } return 0; } static void stats_add_ran_ue(void) { num_of_ran_ue = num_of_ran_ue + 1; ogs_info("[Added] Number of gNB-UEs is now %d", num_of_ran_ue); } static void stats_remove_ran_ue(void) { num_of_ran_ue = num_of_ran_ue - 1; ogs_info("[Removed] Number of gNB-UEs is now %d", num_of_ran_ue); } static void stats_add_amf_session(void) { num_of_amf_sess = num_of_amf_sess + 1; ogs_info("[Added] Number of AMF-Sessions is now %d", num_of_amf_sess); } static void stats_remove_amf_session(void) { num_of_amf_sess = num_of_amf_sess - 1; ogs_info("[Removed] Number of AMF-Sessions is now %d", num_of_amf_sess); }