forked from acouzens/open5gs
957 lines
28 KiB
C
957 lines
28 KiB
C
/*
|
|
* Copyright (C) 2019 by Sukchan Lee <acetcom@gmail.com>
|
|
*
|
|
* 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 <https://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include "ogs-dbi.h"
|
|
#include "hss-context.h"
|
|
|
|
typedef struct hss_impi_s hss_impi_t;
|
|
|
|
typedef struct hss_imsi_s {
|
|
ogs_lnode_t lnode;
|
|
|
|
char *id;
|
|
ogs_plmn_id_t visited_plmn_id;
|
|
} hss_imsi_t;
|
|
|
|
typedef struct hss_impi_s {
|
|
ogs_lnode_t lnode;
|
|
|
|
char *id;
|
|
|
|
hss_imsi_t *imsi;
|
|
ogs_list_t impu_list;
|
|
} hss_impi_t;
|
|
|
|
typedef struct hss_impu_s {
|
|
ogs_lnode_t lnode;
|
|
|
|
char *id;
|
|
char *server_name;
|
|
|
|
hss_impi_t *impi;
|
|
} hss_impu_t;
|
|
|
|
static hss_context_t self;
|
|
static ogs_diam_config_t g_diam_conf;
|
|
|
|
int __hss_log_domain;
|
|
|
|
static int context_initialized = 0;
|
|
|
|
static OGS_POOL(imsi_pool, hss_imsi_t);
|
|
static OGS_POOL(impi_pool, hss_impi_t);
|
|
static OGS_POOL(impu_pool, hss_impu_t);
|
|
|
|
static hss_imsi_t *imsi_add(char *id);
|
|
static void imsi_remove(hss_imsi_t *imsi);
|
|
static void imsi_remove_all(void);
|
|
static hss_imsi_t *imsi_find_by_id(char *id);
|
|
|
|
static hss_impi_t *impi_add(char *id);
|
|
static void impi_remove(hss_impi_t *impi);
|
|
static void impi_remove_all(void);
|
|
|
|
static hss_impi_t *impi_find_by_id(char *id);
|
|
static char *impi_get_server_name(hss_impi_t *impi);
|
|
|
|
static hss_impu_t *impu_add(hss_impi_t *impi, char *id);
|
|
static void impu_remove(hss_impu_t *impu);
|
|
static void impu_remove_all(hss_impi_t *impi);
|
|
static hss_impu_t *impu_find_by_id(char *id);
|
|
static hss_impu_t *impu_find_by_impi_and_id(hss_impi_t *impi, char *id);
|
|
|
|
hss_context_t* hss_self(void)
|
|
{
|
|
return &self;
|
|
}
|
|
|
|
void hss_context_init(void)
|
|
{
|
|
ogs_assert(context_initialized == 0);
|
|
|
|
/* Initial FreeDiameter Config */
|
|
memset(&g_diam_conf, 0, sizeof(ogs_diam_config_t));
|
|
|
|
/* Initialize HSS context */
|
|
memset(&self, 0, sizeof(hss_context_t));
|
|
self.diam_config = &g_diam_conf;
|
|
|
|
ogs_log_install_domain(&__ogs_diam_domain, "diam", ogs_core()->log.level);
|
|
ogs_log_install_domain(&__ogs_dbi_domain, "dbi", ogs_core()->log.level);
|
|
ogs_log_install_domain(&__hss_log_domain, "hss", ogs_core()->log.level);
|
|
|
|
ogs_pool_init(&imsi_pool, ogs_app()->pool.impi);
|
|
ogs_pool_init(&impi_pool, ogs_app()->pool.impi);
|
|
ogs_pool_init(&impu_pool, ogs_app()->pool.impu);
|
|
|
|
self.imsi_hash = ogs_hash_make();
|
|
ogs_assert(self.imsi_hash);
|
|
self.impi_hash = ogs_hash_make();
|
|
ogs_assert(self.impi_hash);
|
|
self.impu_hash = ogs_hash_make();
|
|
ogs_assert(self.impu_hash);
|
|
|
|
ogs_thread_mutex_init(&self.db_lock);
|
|
ogs_thread_mutex_init(&self.cx_lock);
|
|
|
|
context_initialized = 1;
|
|
}
|
|
|
|
void hss_context_final(void)
|
|
{
|
|
ogs_assert(context_initialized == 1);
|
|
|
|
imsi_remove_all();
|
|
impi_remove_all();
|
|
|
|
ogs_assert(self.imsi_hash);
|
|
ogs_hash_destroy(self.imsi_hash);
|
|
ogs_assert(self.impi_hash);
|
|
ogs_hash_destroy(self.impi_hash);
|
|
ogs_assert(self.impu_hash);
|
|
ogs_hash_destroy(self.impu_hash);
|
|
|
|
ogs_pool_final(&imsi_pool);
|
|
ogs_pool_final(&impi_pool);
|
|
ogs_pool_final(&impu_pool);
|
|
|
|
ogs_thread_mutex_destroy(&self.db_lock);
|
|
ogs_thread_mutex_destroy(&self.cx_lock);
|
|
|
|
context_initialized = 0;
|
|
}
|
|
|
|
static int hss_context_prepare(void)
|
|
{
|
|
self.diam_config->cnf_port = DIAMETER_PORT;
|
|
self.diam_config->cnf_port_tls = DIAMETER_SECURE_PORT;
|
|
|
|
return OGS_OK;
|
|
}
|
|
|
|
static int hss_context_validation(void)
|
|
{
|
|
if (self.diam_conf_path == NULL &&
|
|
(self.diam_config->cnf_diamid == NULL ||
|
|
self.diam_config->cnf_diamrlm == NULL ||
|
|
self.diam_config->cnf_addr == NULL)) {
|
|
ogs_error("No hss.freeDiameter in '%s'", ogs_app()->file);
|
|
return OGS_ERROR;
|
|
}
|
|
|
|
return OGS_OK;
|
|
}
|
|
|
|
int hss_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 = hss_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, "hss")) {
|
|
ogs_yaml_iter_t hss_iter;
|
|
ogs_yaml_iter_recurse(&root_iter, &hss_iter);
|
|
while (ogs_yaml_iter_next(&hss_iter)) {
|
|
const char *hss_key = ogs_yaml_iter_key(&hss_iter);
|
|
ogs_assert(hss_key);
|
|
if (!strcmp(hss_key, "freeDiameter")) {
|
|
yaml_node_t *node =
|
|
yaml_document_get_node(document, hss_iter.pair->value);
|
|
ogs_assert(node);
|
|
if (node->type == YAML_SCALAR_NODE) {
|
|
self.diam_conf_path = ogs_yaml_iter_value(&hss_iter);
|
|
} else if (node->type == YAML_MAPPING_NODE) {
|
|
ogs_yaml_iter_t fd_iter;
|
|
ogs_yaml_iter_recurse(&hss_iter, &fd_iter);
|
|
|
|
while (ogs_yaml_iter_next(&fd_iter)) {
|
|
const char *fd_key = ogs_yaml_iter_key(&fd_iter);
|
|
ogs_assert(fd_key);
|
|
if (!strcmp(fd_key, "identity")) {
|
|
self.diam_config->cnf_diamid =
|
|
ogs_yaml_iter_value(&fd_iter);
|
|
} else if (!strcmp(fd_key, "realm")) {
|
|
self.diam_config->cnf_diamrlm =
|
|
ogs_yaml_iter_value(&fd_iter);
|
|
} else if (!strcmp(fd_key, "port")) {
|
|
const char *v = ogs_yaml_iter_value(&fd_iter);
|
|
if (v) self.diam_config->cnf_port = atoi(v);
|
|
} else if (!strcmp(fd_key, "sec_port")) {
|
|
const char *v = ogs_yaml_iter_value(&fd_iter);
|
|
if (v) self.diam_config->cnf_port_tls = atoi(v);
|
|
} else if (!strcmp(fd_key, "listen_on")) {
|
|
self.diam_config->cnf_addr =
|
|
ogs_yaml_iter_value(&fd_iter);
|
|
} else if (!strcmp(fd_key, "no_fwd")) {
|
|
self.diam_config->cnf_flags.no_fwd =
|
|
ogs_yaml_iter_bool(&fd_iter);
|
|
} else if (!strcmp(fd_key, "load_extension")) {
|
|
ogs_yaml_iter_t ext_array, ext_iter;
|
|
ogs_yaml_iter_recurse(&fd_iter, &ext_array);
|
|
do {
|
|
const char *module = NULL;
|
|
const char *conf = NULL;
|
|
|
|
if (ogs_yaml_iter_type(&ext_array) ==
|
|
YAML_MAPPING_NODE) {
|
|
memcpy(&ext_iter, &ext_array,
|
|
sizeof(ogs_yaml_iter_t));
|
|
} else if (ogs_yaml_iter_type(&ext_array) ==
|
|
YAML_SEQUENCE_NODE) {
|
|
if (!ogs_yaml_iter_next(&ext_array))
|
|
break;
|
|
ogs_yaml_iter_recurse(
|
|
&ext_array, &ext_iter);
|
|
} else if (ogs_yaml_iter_type(&ext_array) ==
|
|
YAML_SCALAR_NODE) {
|
|
break;
|
|
} else
|
|
ogs_assert_if_reached();
|
|
|
|
while (ogs_yaml_iter_next(&ext_iter)) {
|
|
const char *ext_key =
|
|
ogs_yaml_iter_key(&ext_iter);
|
|
ogs_assert(ext_key);
|
|
if (!strcmp(ext_key, "module")) {
|
|
module = ogs_yaml_iter_value(
|
|
&ext_iter);
|
|
} else if (!strcmp(ext_key, "conf")) {
|
|
conf = ogs_yaml_iter_value(
|
|
&ext_iter);
|
|
} else
|
|
ogs_warn("unknown key `%s`",
|
|
ext_key);
|
|
}
|
|
|
|
if (module) {
|
|
self.diam_config->
|
|
ext[self.diam_config->num_of_ext].
|
|
module = module;
|
|
self.diam_config->
|
|
ext[self.diam_config->num_of_ext].
|
|
conf = conf;
|
|
self.diam_config->num_of_ext++;
|
|
}
|
|
} while (ogs_yaml_iter_type(&ext_array) ==
|
|
YAML_SEQUENCE_NODE);
|
|
} else if (!strcmp(fd_key, "connect")) {
|
|
ogs_yaml_iter_t conn_array, conn_iter;
|
|
ogs_yaml_iter_recurse(&fd_iter, &conn_array);
|
|
do {
|
|
const char *identity = NULL;
|
|
const char *addr = NULL;
|
|
uint16_t port = 0;
|
|
|
|
if (ogs_yaml_iter_type(&conn_array) ==
|
|
YAML_MAPPING_NODE) {
|
|
memcpy(&conn_iter, &conn_array,
|
|
sizeof(ogs_yaml_iter_t));
|
|
} else if (ogs_yaml_iter_type(
|
|
&conn_array) ==
|
|
YAML_SEQUENCE_NODE) {
|
|
if (!ogs_yaml_iter_next(&conn_array))
|
|
break;
|
|
ogs_yaml_iter_recurse(
|
|
&conn_array, &conn_iter);
|
|
} else if (ogs_yaml_iter_type(
|
|
&conn_array) ==
|
|
YAML_SCALAR_NODE) {
|
|
break;
|
|
} else
|
|
ogs_assert_if_reached();
|
|
|
|
while (ogs_yaml_iter_next(&conn_iter)) {
|
|
const char *conn_key =
|
|
ogs_yaml_iter_key(&conn_iter);
|
|
ogs_assert(conn_key);
|
|
if (!strcmp(conn_key, "identity")) {
|
|
identity = ogs_yaml_iter_value(
|
|
&conn_iter);
|
|
} else if (!strcmp(conn_key, "addr")) {
|
|
addr = ogs_yaml_iter_value(
|
|
&conn_iter);
|
|
} else if (!strcmp(conn_key, "port")) {
|
|
const char *v =
|
|
ogs_yaml_iter_value(&conn_iter);
|
|
if (v) port = atoi(v);
|
|
} else
|
|
ogs_warn("unknown key `%s`",
|
|
conn_key);
|
|
}
|
|
|
|
if (identity && addr) {
|
|
self.diam_config->
|
|
conn[self.diam_config->num_of_conn].
|
|
identity = identity;
|
|
self.diam_config->
|
|
conn[self.diam_config->num_of_conn].
|
|
addr = addr;
|
|
self.diam_config->
|
|
conn[self.diam_config->num_of_conn].
|
|
port = port;
|
|
self.diam_config->num_of_conn++;
|
|
}
|
|
} while (ogs_yaml_iter_type(&conn_array) ==
|
|
YAML_SEQUENCE_NODE);
|
|
} else
|
|
ogs_warn("unknown key `%s`", fd_key);
|
|
}
|
|
}
|
|
} else
|
|
ogs_warn("unknown key `%s`", hss_key);
|
|
}
|
|
}
|
|
}
|
|
|
|
rv = hss_context_validation();
|
|
if (rv != OGS_OK) return rv;
|
|
|
|
return OGS_OK;
|
|
}
|
|
|
|
int hss_db_auth_info(char *imsi_bcd, ogs_dbi_auth_info_t *auth_info)
|
|
{
|
|
int rv;
|
|
char *supi = NULL;
|
|
|
|
ogs_assert(imsi_bcd);
|
|
ogs_assert(auth_info);
|
|
|
|
ogs_thread_mutex_lock(&self.db_lock);
|
|
supi = ogs_msprintf("%s-%s", OGS_ID_SUPI_TYPE_IMSI, imsi_bcd);
|
|
ogs_assert(supi);
|
|
|
|
rv = ogs_dbi_auth_info(supi, auth_info);
|
|
|
|
ogs_free(supi);
|
|
ogs_thread_mutex_unlock(&self.db_lock);
|
|
|
|
return rv;
|
|
}
|
|
|
|
int hss_db_update_sqn(char *imsi_bcd, uint8_t *rand, uint64_t sqn)
|
|
{
|
|
int rv;
|
|
char *supi = NULL;
|
|
|
|
ogs_assert(imsi_bcd);
|
|
|
|
ogs_thread_mutex_lock(&self.db_lock);
|
|
supi = ogs_msprintf("%s-%s", OGS_ID_SUPI_TYPE_IMSI, imsi_bcd);
|
|
ogs_assert(supi);
|
|
|
|
rv = ogs_dbi_update_sqn(supi, sqn);
|
|
|
|
ogs_free(supi);
|
|
ogs_thread_mutex_unlock(&self.db_lock);
|
|
|
|
return rv;
|
|
}
|
|
|
|
int hss_db_increment_sqn(char *imsi_bcd)
|
|
{
|
|
int rv;
|
|
char *supi = NULL;
|
|
|
|
ogs_assert(imsi_bcd);
|
|
|
|
ogs_thread_mutex_lock(&self.db_lock);
|
|
supi = ogs_msprintf("%s-%s", OGS_ID_SUPI_TYPE_IMSI, imsi_bcd);
|
|
ogs_assert(supi);
|
|
|
|
rv = ogs_dbi_increment_sqn(supi);
|
|
|
|
ogs_free(supi);
|
|
ogs_thread_mutex_unlock(&self.db_lock);
|
|
|
|
return rv;
|
|
}
|
|
|
|
int hss_db_subscription_data(
|
|
char *imsi_bcd, ogs_subscription_data_t *subscription_data)
|
|
{
|
|
int rv;
|
|
char *supi = NULL;
|
|
|
|
ogs_assert(imsi_bcd);
|
|
ogs_assert(subscription_data);
|
|
|
|
ogs_thread_mutex_lock(&self.db_lock);
|
|
supi = ogs_msprintf("%s-%s", OGS_ID_SUPI_TYPE_IMSI, imsi_bcd);
|
|
ogs_assert(supi);
|
|
|
|
rv = ogs_dbi_subscription_data(supi, subscription_data);
|
|
|
|
ogs_free(supi);
|
|
ogs_thread_mutex_unlock(&self.db_lock);
|
|
|
|
return rv;
|
|
}
|
|
|
|
int hss_db_msisdn_data(char *imsi_or_msisdn_bcd, ogs_msisdn_data_t *msisdn_data)
|
|
{
|
|
int rv;
|
|
|
|
ogs_assert(imsi_or_msisdn_bcd);
|
|
ogs_assert(msisdn_data);
|
|
|
|
ogs_thread_mutex_lock(&self.db_lock);
|
|
|
|
rv = ogs_dbi_msisdn_data(imsi_or_msisdn_bcd, msisdn_data);
|
|
|
|
ogs_thread_mutex_unlock(&self.db_lock);
|
|
|
|
return rv;
|
|
}
|
|
|
|
int hss_db_ims_data(char *imsi_bcd, ogs_ims_data_t *ims_data)
|
|
{
|
|
int rv;
|
|
char *supi = NULL;
|
|
|
|
ogs_assert(imsi_bcd);
|
|
ogs_assert(ims_data);
|
|
|
|
ogs_thread_mutex_lock(&self.db_lock);
|
|
supi = ogs_msprintf("%s-%s", OGS_ID_SUPI_TYPE_IMSI, imsi_bcd);
|
|
ogs_assert(supi);
|
|
|
|
rv = ogs_dbi_ims_data(supi, ims_data);
|
|
|
|
ogs_free(supi);
|
|
ogs_thread_mutex_unlock(&self.db_lock);
|
|
|
|
return rv;
|
|
}
|
|
|
|
static hss_imsi_t *imsi_add(char *id)
|
|
{
|
|
hss_imsi_t *imsi = NULL;
|
|
|
|
ogs_assert(id);
|
|
|
|
ogs_pool_alloc(&imsi_pool, &imsi);
|
|
ogs_assert(imsi);
|
|
memset(imsi, 0, sizeof *imsi);
|
|
|
|
imsi->id = ogs_strdup(id);
|
|
ogs_assert(imsi->id);
|
|
|
|
ogs_hash_set(self.imsi_hash, imsi->id, strlen(imsi->id), imsi);
|
|
|
|
ogs_list_add(&self.imsi_list, imsi);
|
|
|
|
return imsi;
|
|
}
|
|
|
|
static void imsi_remove(hss_imsi_t *imsi)
|
|
{
|
|
ogs_assert(imsi);
|
|
|
|
ogs_list_remove(&self.imsi_list, imsi);
|
|
|
|
ogs_assert(imsi->id);
|
|
ogs_hash_set(self.imsi_hash, imsi->id, strlen(imsi->id), NULL);
|
|
ogs_free(imsi->id);
|
|
|
|
ogs_pool_free(&imsi_pool, imsi);
|
|
}
|
|
|
|
static void imsi_remove_all(void)
|
|
{
|
|
hss_imsi_t *imsi = NULL, *next = NULL;
|
|
|
|
ogs_list_for_each_safe(&self.imsi_list, next, imsi)
|
|
imsi_remove(imsi);
|
|
}
|
|
|
|
static hss_imsi_t *imsi_find_by_id(char *id)
|
|
{
|
|
ogs_assert(id);
|
|
return (hss_imsi_t *)ogs_hash_get(self.imsi_hash, id, strlen(id));
|
|
}
|
|
|
|
static hss_impi_t *impi_add(char *id)
|
|
{
|
|
hss_impi_t *impi = NULL;
|
|
|
|
ogs_assert(id);
|
|
|
|
ogs_pool_alloc(&impi_pool, &impi);
|
|
ogs_assert(impi);
|
|
memset(impi, 0, sizeof *impi);
|
|
|
|
impi->id = ogs_strdup(id);
|
|
ogs_assert(impi->id);
|
|
|
|
ogs_hash_set(self.impi_hash, impi->id, strlen(impi->id), impi);
|
|
|
|
ogs_list_add(&self.impi_list, impi);
|
|
|
|
return impi;
|
|
}
|
|
|
|
static void impi_remove(hss_impi_t *impi)
|
|
{
|
|
ogs_assert(impi);
|
|
|
|
ogs_list_remove(&self.impi_list, impi);
|
|
|
|
impu_remove_all(impi);
|
|
|
|
ogs_assert(impi->id);
|
|
ogs_hash_set(self.impi_hash, impi->id, strlen(impi->id), NULL);
|
|
ogs_free(impi->id);
|
|
|
|
ogs_pool_free(&impi_pool, impi);
|
|
}
|
|
|
|
static void impi_remove_all(void)
|
|
{
|
|
hss_impi_t *impi = NULL, *next = NULL;
|
|
|
|
ogs_list_for_each_safe(&self.impi_list, next, impi)
|
|
impi_remove(impi);
|
|
}
|
|
|
|
static hss_impi_t *impi_find_by_id(char *id)
|
|
{
|
|
ogs_assert(id);
|
|
return (hss_impi_t *)ogs_hash_get(self.impi_hash, id, strlen(id));
|
|
}
|
|
|
|
static char *impi_get_server_name(hss_impi_t *impi)
|
|
{
|
|
hss_impu_t *impu = NULL;
|
|
|
|
ogs_assert(impi);
|
|
|
|
ogs_list_for_each(&impi->impu_list, impu) {
|
|
if (impu->server_name)
|
|
return impu->server_name;
|
|
}
|
|
|
|
return NULL;
|
|
}
|
|
|
|
static hss_impu_t *impu_add(hss_impi_t *impi, char *id)
|
|
{
|
|
hss_impu_t *impu = NULL;
|
|
|
|
ogs_assert(impi);
|
|
ogs_assert(id);
|
|
|
|
ogs_pool_alloc(&impu_pool, &impu);
|
|
ogs_assert(impu);
|
|
memset(impu, 0, sizeof *impu);
|
|
|
|
impu->id = ogs_strdup(id);
|
|
ogs_assert(impu->id);
|
|
|
|
ogs_hash_set(self.impu_hash, impu->id, strlen(impu->id), impu);
|
|
|
|
impu->impi = impi;
|
|
ogs_list_add(&impi->impu_list, impu);
|
|
|
|
return impu;
|
|
}
|
|
|
|
static void impu_remove(hss_impu_t *impu)
|
|
{
|
|
hss_impi_t *impi = NULL;
|
|
|
|
ogs_assert(impu);
|
|
impi = impu->impi;
|
|
ogs_assert(impi);
|
|
|
|
ogs_list_remove(&impi->impu_list, impu);
|
|
|
|
ogs_assert(impu->id);
|
|
ogs_hash_set(self.impu_hash, impu->id, strlen(impu->id), NULL);
|
|
ogs_free(impu->id);
|
|
|
|
if (impu->server_name)
|
|
ogs_free(impu->server_name);
|
|
|
|
ogs_pool_free(&impu_pool, impu);
|
|
}
|
|
|
|
static void impu_remove_all(hss_impi_t *impi)
|
|
{
|
|
hss_impu_t *impu = NULL, *next = NULL;
|
|
|
|
ogs_list_for_each_safe(&impi->impu_list, next, impu)
|
|
impu_remove(impu);
|
|
}
|
|
|
|
static hss_impu_t *impu_find_by_id(char *id)
|
|
{
|
|
ogs_assert(id);
|
|
return (hss_impu_t *)ogs_hash_get(self.impu_hash, id, strlen(id));
|
|
}
|
|
|
|
static hss_impu_t *impu_find_by_impi_and_id(hss_impi_t *impi, char *id)
|
|
{
|
|
hss_impu_t *impu = NULL;
|
|
|
|
ogs_assert(impi);
|
|
ogs_assert(id);
|
|
|
|
ogs_list_for_each(&impi->impu_list, impu) {
|
|
if (!strcmp(impu->id, id))
|
|
return impu;
|
|
}
|
|
|
|
return NULL;
|
|
}
|
|
|
|
void hss_s6a_set_visited_plmn_id(char *imsi_bcd, ogs_plmn_id_t *visited_plmn_id)
|
|
{
|
|
hss_imsi_t *imsi = NULL;
|
|
|
|
ogs_assert(imsi_bcd);
|
|
ogs_assert(visited_plmn_id);
|
|
|
|
ogs_thread_mutex_lock(&self.cx_lock);
|
|
|
|
imsi = imsi_find_by_id(imsi_bcd);
|
|
if (!imsi) {
|
|
imsi = imsi_add(imsi_bcd);
|
|
ogs_assert(imsi);
|
|
}
|
|
|
|
memcpy(&imsi->visited_plmn_id, visited_plmn_id, OGS_PLMN_ID_LEN);
|
|
|
|
ogs_thread_mutex_unlock(&self.cx_lock);
|
|
}
|
|
|
|
void hss_cx_associate_identity(char *user_name, char *public_identity)
|
|
{
|
|
hss_impi_t *impi = NULL;
|
|
hss_impu_t *impu = NULL;
|
|
|
|
ogs_assert(user_name);
|
|
ogs_assert(public_identity);
|
|
|
|
ogs_thread_mutex_lock(&self.cx_lock);
|
|
|
|
impi = impi_find_by_id(user_name);
|
|
if (!impi) {
|
|
impi = impi_add(user_name);
|
|
ogs_assert(impi);
|
|
}
|
|
|
|
impu = impu_find_by_id(public_identity);
|
|
if (!impu) {
|
|
impu = impu_add(impi, public_identity);
|
|
ogs_assert(impu);
|
|
}
|
|
|
|
ogs_thread_mutex_unlock(&self.cx_lock);
|
|
}
|
|
|
|
bool hss_cx_identity_is_associated(char *user_name, char *public_identity)
|
|
{
|
|
bool match_result = false;
|
|
|
|
hss_impi_t *impi = NULL;
|
|
hss_impu_t *impu = NULL;
|
|
|
|
ogs_thread_mutex_lock(&self.cx_lock);
|
|
|
|
impi = impi_find_by_id(user_name);
|
|
if (impi) {
|
|
impu = impu_find_by_impi_and_id(impi, public_identity);
|
|
if (impu) {
|
|
match_result = true;
|
|
}
|
|
}
|
|
|
|
ogs_thread_mutex_unlock(&self.cx_lock);
|
|
|
|
return match_result;
|
|
}
|
|
|
|
void hss_cx_set_imsi_bcd(char *user_name, char *imsi_bcd)
|
|
{
|
|
hss_imsi_t *imsi = NULL;
|
|
hss_impi_t *impi = NULL;
|
|
|
|
ogs_assert(user_name);
|
|
ogs_assert(imsi_bcd);
|
|
|
|
ogs_thread_mutex_lock(&self.cx_lock);
|
|
|
|
impi = impi_find_by_id(user_name);
|
|
ogs_assert(impi);
|
|
|
|
imsi = imsi_find_by_id(imsi_bcd);
|
|
ogs_assert(imsi);
|
|
|
|
impi->imsi = imsi;
|
|
|
|
ogs_thread_mutex_unlock(&self.cx_lock);
|
|
}
|
|
|
|
char *hss_cx_get_imsi_bcd(char *public_identity)
|
|
{
|
|
hss_impi_t *impi = NULL;
|
|
hss_impu_t *impu = NULL;
|
|
|
|
char *imsi_bcd = NULL;
|
|
|
|
ogs_thread_mutex_lock(&self.cx_lock);
|
|
|
|
impu = impu_find_by_id(public_identity);
|
|
if (impu) {
|
|
impi = impu->impi;
|
|
ogs_assert(impi);
|
|
|
|
if (impi->imsi)
|
|
imsi_bcd = impi->imsi->id;
|
|
}
|
|
|
|
ogs_thread_mutex_unlock(&self.cx_lock);
|
|
|
|
return imsi_bcd;
|
|
}
|
|
|
|
ogs_plmn_id_t *hss_cx_get_visited_plmn_id(char *public_identity)
|
|
{
|
|
hss_impi_t *impi = NULL;
|
|
hss_impu_t *impu = NULL;
|
|
|
|
ogs_plmn_id_t *visited_plmn_id = NULL;
|
|
|
|
ogs_thread_mutex_lock(&self.cx_lock);
|
|
|
|
impu = impu_find_by_id(public_identity);
|
|
if (impu) {
|
|
impi = impu->impi;
|
|
ogs_assert(impi);
|
|
|
|
if (impi->imsi)
|
|
visited_plmn_id = &impi->imsi->visited_plmn_id;
|
|
}
|
|
|
|
ogs_thread_mutex_unlock(&self.cx_lock);
|
|
|
|
return visited_plmn_id;
|
|
}
|
|
|
|
char *hss_cx_get_user_name(char *public_identity)
|
|
{
|
|
hss_impi_t *impi = NULL;
|
|
hss_impu_t *impu = NULL;
|
|
|
|
char *user_name = NULL;
|
|
|
|
ogs_thread_mutex_lock(&self.cx_lock);
|
|
|
|
impu = impu_find_by_id(public_identity);
|
|
if (impu) {
|
|
impi = impu->impi;
|
|
ogs_assert(impi);
|
|
|
|
user_name = impi->id;
|
|
}
|
|
|
|
ogs_thread_mutex_unlock(&self.cx_lock);
|
|
|
|
return user_name;
|
|
}
|
|
|
|
char *hss_cx_get_server_name(char *public_identity)
|
|
{
|
|
char *server_name = NULL;
|
|
|
|
hss_impi_t *impi = NULL;
|
|
hss_impu_t *impu = NULL;
|
|
|
|
ogs_thread_mutex_lock(&self.cx_lock);
|
|
|
|
impu = impu_find_by_id(public_identity);
|
|
if (impu) {
|
|
server_name = impu->server_name;
|
|
if (!server_name) {
|
|
impi = impu->impi;
|
|
ogs_assert(impi);
|
|
|
|
server_name = impi_get_server_name(impi);
|
|
}
|
|
}
|
|
|
|
ogs_thread_mutex_unlock(&self.cx_lock);
|
|
|
|
return server_name;
|
|
}
|
|
|
|
void hss_cx_set_server_name(
|
|
char *public_identity, char *server_name, bool overwrite)
|
|
{
|
|
hss_impu_t *impu = NULL;
|
|
|
|
ogs_thread_mutex_lock(&self.cx_lock);
|
|
|
|
impu = impu_find_by_id(public_identity);
|
|
ogs_assert(impu);
|
|
|
|
if (!impu->server_name) {
|
|
impu->server_name = ogs_strdup(server_name);
|
|
ogs_assert(impu->server_name);
|
|
} else {
|
|
if (strcmp(impu->server_name, server_name) != 0) {
|
|
if (overwrite == true) {
|
|
ogs_warn("S-CSCF reassignment[%s->%s]",
|
|
impu->server_name, server_name);
|
|
ogs_free(impu->server_name);
|
|
impu->server_name = ogs_strdup(server_name);
|
|
ogs_assert(impu->server_name);
|
|
} else {
|
|
ogs_error("Use Old S-CSCF[%s!=%s]",
|
|
server_name, impu->server_name);
|
|
}
|
|
}
|
|
}
|
|
|
|
ogs_thread_mutex_unlock(&self.cx_lock);
|
|
}
|
|
|
|
char *hss_cx_download_user_data(
|
|
char *user_name, ogs_plmn_id_t *plmn_id, ogs_ims_data_t *ims_data)
|
|
{
|
|
char *user_data = NULL;
|
|
|
|
hss_impi_t *impi = NULL;
|
|
hss_impu_t *impu = NULL;
|
|
|
|
bool barring_indication_presence = true;
|
|
int i;
|
|
|
|
ogs_assert(user_name);
|
|
ogs_assert(plmn_id);
|
|
ogs_assert(ims_data);
|
|
|
|
/* Download User-Data */
|
|
for (i = 0; i < ims_data->num_of_msisdn; i++) {
|
|
char *public_identity = NULL;
|
|
|
|
public_identity = ogs_msprintf(
|
|
"sip:%s@ims.mnc%03d.mcc%03d.3gppnetwork.org",
|
|
ims_data->msisdn[i].bcd,
|
|
ogs_plmn_id_mnc(plmn_id), ogs_plmn_id_mcc(plmn_id));
|
|
ogs_assert(public_identity);
|
|
hss_cx_associate_identity(user_name, public_identity);
|
|
ogs_free(public_identity);
|
|
|
|
public_identity = ogs_msprintf("tel:%s", ims_data->msisdn[i].bcd);
|
|
ogs_assert(public_identity);
|
|
hss_cx_associate_identity(user_name, public_identity);
|
|
ogs_free(public_identity);
|
|
}
|
|
|
|
|
|
/* Generate XML Data */
|
|
|
|
ogs_thread_mutex_lock(&self.cx_lock);
|
|
|
|
user_data = ogs_strdup(ogs_diam_cx_xml_version);
|
|
ogs_assert(user_data);
|
|
|
|
user_data = ogs_mstrcatf(user_data, "%s",
|
|
ogs_diam_cx_xml_ims_subscription_s);
|
|
ogs_assert(user_data);
|
|
|
|
user_data = ogs_mstrcatf(user_data, "%s%s%s",
|
|
ogs_diam_cx_xml_private_id_s,
|
|
user_name,
|
|
ogs_diam_cx_xml_private_id_e);
|
|
ogs_assert(user_data);
|
|
|
|
user_data = ogs_mstrcatf(user_data, "%s",
|
|
ogs_diam_cx_xml_service_profile_s);
|
|
ogs_assert(user_data);
|
|
|
|
impi = impi_find_by_id(user_name);
|
|
ogs_assert(impi);
|
|
|
|
barring_indication_presence = true;
|
|
|
|
ogs_list_for_each(&impi->impu_list, impu) {
|
|
user_data = ogs_mstrcatf(user_data, "%s",
|
|
ogs_diam_cx_xml_public_id_s);
|
|
ogs_assert(user_data);
|
|
|
|
if (barring_indication_presence == true) {
|
|
user_data = ogs_mstrcatf(user_data, "%s%s%s",
|
|
ogs_diam_cx_xml_barring_indication_s,
|
|
"1",
|
|
ogs_diam_cx_xml_barring_indication_e);
|
|
ogs_assert(user_data);
|
|
barring_indication_presence = false;
|
|
}
|
|
|
|
user_data = ogs_mstrcatf(user_data, "%s%s%s",
|
|
ogs_diam_cx_xml_identity_s,
|
|
impu->id,
|
|
ogs_diam_cx_xml_identity_e);
|
|
ogs_assert(user_data);
|
|
|
|
user_data = ogs_mstrcatf(user_data, "%s",
|
|
ogs_diam_cx_xml_extension_s);
|
|
ogs_assert(user_data);
|
|
|
|
user_data = ogs_mstrcatf(user_data, "%s%s%s",
|
|
ogs_diam_cx_xml_identity_type_s,
|
|
"0",
|
|
ogs_diam_cx_xml_identity_type_e);
|
|
ogs_assert(user_data);
|
|
|
|
user_data = ogs_mstrcatf(user_data, "%s",
|
|
ogs_diam_cx_xml_extension_e);
|
|
ogs_assert(user_data);
|
|
|
|
user_data = ogs_mstrcatf(user_data, "%s",
|
|
ogs_diam_cx_xml_public_id_e);
|
|
ogs_assert(user_data);
|
|
}
|
|
|
|
user_data = ogs_mstrcatf(user_data, "%s",
|
|
ogs_diam_cx_xml_service_profile_e);
|
|
ogs_assert(user_data);
|
|
|
|
user_data = ogs_mstrcatf(user_data,
|
|
"%s", ogs_diam_cx_xml_ims_subscription_e);
|
|
ogs_assert(user_data);
|
|
|
|
ogs_thread_mutex_unlock(&self.cx_lock);
|
|
|
|
return user_data;
|
|
}
|