open5gs/src/hss/hss_context.c

#define TRACE_MODULE _hss_ctx

#include "core_debug.h"
#include "core_pool.h"
#include "core_lib.h"

#include "hss_context.h"

#define OP "5F1D289C5D354D0A140C2548F5F3E3BA"
#define OPc "E8ED289DEBA952E4283B54E88E6183CA"
#define AMF "8000"

static hss_ctx_t self;

pool_declare(user_pool, user_ctx_t, MAX_NUM_OF_UE);

static list_t g_user_list;

hss_ctx_t* hss_self()
{
    return &self;
}

status_t hss_ctx_init(void)
{
    char buf[HSS_KEY_LEN];
    user_ctx_t *user;

    pool_init(&user_pool, MAX_NUM_OF_UE);

    memset(&self, 0, sizeof(hss_ctx_t));

    memcpy(self.op, CORE_HEX(OP, strlen(OP), buf), HSS_KEY_LEN);
    memcpy(self.amf, CORE_HEX(AMF, strlen(AMF), buf), HSS_AMF_LEN);

    #define K "465B5CE8B199B49FAA5F0A2EE238A6BC"
    #define UE1_IMSI "001010123456800"
    #define UE2_IMSI "001010123456796"

    #define UE3_IMSI "001010123456819"
    #define UE3_RAND "20080c3818183b52 2614162c07601d0d"

    user = hss_ctx_user_add();
    d_assert(user, return -1, "UE context add failed");

    strcpy((char*)user->imsi, UE1_IMSI);
    user->imsi_len = strlen(UE1_IMSI);
    memcpy(user->k, CORE_HEX(K, strlen(K), buf), HSS_KEY_LEN);
    core_generate_random_bytes(user->rand, RAND_LEN);
    user->sqn = 64;

    user = hss_ctx_user_add();
    d_assert(user, return -1, "UE context add failed");

    strcpy((char*)user->imsi, UE2_IMSI);
    user->imsi_len = strlen(UE2_IMSI);
    memcpy(user->k, CORE_HEX(K, strlen(K), buf), HSS_KEY_LEN);
    core_generate_random_bytes(user->rand, RAND_LEN);
    user->sqn = 64;

    user = hss_ctx_user_add();
    d_assert(user, return -1, "UE context add failed");

    strcpy((char*)user->imsi, UE3_IMSI);
    user->imsi_len = strlen(UE3_IMSI);
    memcpy(user->k, CORE_HEX(K, strlen(K), buf), HSS_KEY_LEN);
    memcpy(user->rand, CORE_HEX(UE3_RAND, strlen(UE3_RAND), buf), 
            RAND_LEN);
    user->sqn = 64;

	return CORE_OK;
}

void hss_ctx_final(void)
{
    hss_ctx_user_remove_all();

    pool_final(&user_pool);
	
	return;
}

user_ctx_t* hss_ctx_user_add()
{
    user_ctx_t *user = NULL;

    /* Allocate new eNB context */
    pool_alloc_node(&user_pool, &user);
    d_assert(user, return NULL, "HSS-UE context allocation failed");

    /* Initialize eNB context */
    memset(user, 0, sizeof(user_ctx_t));

    /* Add new eNB context to list */
    list_append(&g_user_list, user);

    memcpy(user->op, self.op, HSS_KEY_LEN);
    memcpy(user->amf, self.amf, HSS_AMF_LEN);
    
    return user;
}

status_t hss_ctx_user_remove(user_ctx_t *user)
{
    d_assert(user, return CORE_ERROR, "Null param");

    list_remove(&g_user_list, user);
    pool_free_node(&user_pool, user);

    return CORE_OK;
}

status_t hss_ctx_user_remove_all()
{
    user_ctx_t *user = NULL, *next_user = NULL;
    
    user = list_first(&g_user_list);
    while (user)
    {
        next_user = list_next(user);

        hss_ctx_user_remove(user);

        user = next_user;
    }

    return CORE_OK;
}

user_ctx_t* hss_ctx_user_find_by_imsi(c_uint8_t *imsi, c_uint8_t imsi_len)
{
    user_ctx_t *user = NULL;
    
    user = list_first(&g_user_list);
    while (user)
    {
        if (memcmp(user->imsi, imsi, imsi_len) == 0)
            break;

        user = list_next(user);
    }

    return user;
}

user_ctx_t* hss_ctx_user_first()
{
    return list_first(&g_user_list);
}

user_ctx_t* hss_ctx_user_next(user_ctx_t *user)
{
    return list_next(user);
}