acouzens
/
open5gs
1
0
Fork 1
Code Issues Pull Requests 1 Projects Releases Wiki Activity

use freeDiameter Library CHECK_FCT instead of d_assert

This commit is contained in:
Sukchan Lee 2017-08-05 17:29:35 +09:00
parent 9782363f2b
commit 72aa1b7eef
2 changed files with 458 additions and 265 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

717
src/mme/mme_s6a_handler.c
View File

@ -36,70 +36,135 @@ static void mme_s6a_aia_cb(void *data, struct msg **msg)
event_t e;
mme_ue_t *mme_ue = NULL;
CHECK_SYS_DO(clock_gettime(CLOCK_REALTIME, &ts), return);
CHECK_SYS_DO( clock_gettime(CLOCK_REALTIME, &ts), return );
/* Search the session, retrieve its data */
d_assert(fd_msg_sess_get(fd_g_config->cnf_dict, *msg, &sess, &new) == 0 &&
new == 0, return,);
CHECK_FCT_DO( fd_msg_sess_get(fd_g_config->cnf_dict, *msg, &sess, &new),
return );
d_assert(new == 0, return, );
d_assert(fd_sess_state_retrieve(mme_s6a_reg, sess, &mi) == 0 &&
mi && (void *)mi == data, fd_msg_free(*msg); *msg = NULL; return,);
CHECK_FCT_DO( fd_sess_state_retrieve(mme_s6a_reg, sess, &mi), return );
d_assert(mi && (void *)mi == data, return, );
mme_ue = mi->mme_ue;
d_assert(mme_ue, error++; goto out,);
d_assert(mme_ue, return, );
d_trace(3, "[S6A] Authentication-Information-Response : UE[%s] <-- HSS\n",
mme_ue->imsi_bcd);
/* Value of Result Code */
d_assert(fd_msg_search_avp(*msg, s6a_result_code, &avp) == 0 && avp,
error++; goto out,);
d_assert(fd_msg_avp_hdr(avp, &hdr) == 0 && hdr, error++; goto out,);
if (hdr->avp_value->i32 != ER_DIAMETER_SUCCESS)
CHECK_FCT_DO( fd_msg_search_avp(*msg, s6a_result_code, &avp), return );
if (avp)
{
d_error("ERROR DIAMETER Result Code(%d)", hdr->avp_value->i32);
error++;
goto out;
CHECK_FCT_DO( fd_msg_avp_hdr(avp, &hdr), return);
d_trace(3, "Status: %d\n", hdr->avp_value->i32);
if (hdr->avp_value->i32 != ER_DIAMETER_SUCCESS)
{
d_error("ERROR DIAMETER Result Code(%d)", hdr->avp_value->i32);
error++;
}
}
else
{
d_error("no Result-Code");
error++;
}
d_assert(fd_msg_search_avp(*msg, s6a_authentication_info, &avp) == 0 &&
avp, error++; goto out,);
d_assert(fd_msg_avp_hdr(avp, &hdr) == 0 && hdr, error++; goto out,);
d_assert(fd_avp_search_avp(avp, s6a_e_utran_vector, &avp_e_utran_vector)
== 0 && avp_e_utran_vector, error++; goto out,);
d_assert(fd_msg_avp_hdr(avp_e_utran_vector, &hdr) == 0 && hdr, error++;
goto out,);
d_assert(fd_avp_search_avp(avp_e_utran_vector, s6a_xres, &avp_xres) == 0 &&
avp_xres, error++; goto out,);
d_assert(fd_msg_avp_hdr(avp_xres, &hdr) == 0 && hdr, error++; goto out,);
memcpy(mme_ue->xres, hdr->avp_value->os.data, hdr->avp_value->os.len);
mme_ue->xres_len = hdr->avp_value->os.len;
/* Value of Origin-Host */
CHECK_FCT_DO( fd_msg_search_avp(*msg, s6a_origin_host, &avp), return );
if (avp)
{
CHECK_FCT_DO( fd_msg_avp_hdr(avp, &hdr), return );
d_trace(3, "From '%.*s' ",
(int)hdr->avp_value->os.len, hdr->avp_value->os.data);
}
else
{
d_error("no_Origin-Host ");
error++;
}
d_assert(
fd_avp_search_avp(avp_e_utran_vector, s6a_kasme, &avp_kasme) == 0 &&
avp_kasme, error++; goto out,);
d_assert(fd_msg_avp_hdr(avp_kasme, &hdr) == 0 && hdr, error++; goto out,);
memcpy(mme_ue->kasme, hdr->avp_value->os.data, hdr->avp_value->os.len);
/* Value of Origin-Realm */
CHECK_FCT_DO( fd_msg_search_avp(*msg, s6a_origin_realm, &avp), return );
if (avp)
{
CHECK_FCT_DO( fd_msg_avp_hdr(avp, &hdr), return );
d_trace(3, "('%.*s') ",
(int)hdr->avp_value->os.len, hdr->avp_value->os.data);
}
else
{
d_error("no_Origin-Realm ");
error++;
}
d_assert(fd_avp_search_avp(avp_e_utran_vector, s6a_rand, &avp_rand) == 0 &&
avp_rand, error++; goto out,);
d_assert(fd_msg_avp_hdr(avp_rand, &hdr) == 0 && hdr, error++; goto out,);
memcpy(mme_ue->rand, hdr->avp_value->os.data, hdr->avp_value->os.len);
CHECK_FCT_DO(
fd_msg_search_avp(*msg, s6a_authentication_info, &avp), return );
if (avp)
{
CHECK_FCT_DO( fd_msg_avp_hdr(avp, &hdr), return );
}
else
error++;
d_assert(fd_avp_search_avp(avp_e_utran_vector, s6a_autn, &avp_autn) == 0 &&
avp_autn, error++; goto out,);
d_assert(fd_msg_avp_hdr(avp_autn, &hdr) == 0 && hdr, error++; goto out,);
memcpy(mme_ue->autn, hdr->avp_value->os.data, hdr->avp_value->os.len);
CHECK_FCT_DO( fd_avp_search_avp(avp, s6a_e_utran_vector,
&avp_e_utran_vector), return );
if (avp)
{
CHECK_FCT_DO( fd_msg_avp_hdr(avp_e_utran_vector, &hdr), return );
}
else
error++;
CHECK_FCT_DO(
fd_avp_search_avp(avp_e_utran_vector, s6a_xres, &avp_xres), return );
if (avp)
{
CHECK_FCT_DO( fd_msg_avp_hdr(avp_xres, &hdr), return );
memcpy(mme_ue->xres, hdr->avp_value->os.data, hdr->avp_value->os.len);
mme_ue->xres_len = hdr->avp_value->os.len;
}
else
error++;
CHECK_FCT_DO(
fd_avp_search_avp(avp_e_utran_vector, s6a_kasme, &avp_kasme), return );
if (avp)
{
CHECK_FCT_DO( fd_msg_avp_hdr(avp_kasme, &hdr), return );
memcpy(mme_ue->kasme, hdr->avp_value->os.data, hdr->avp_value->os.len);
}
else
error++;
CHECK_FCT_DO(
fd_avp_search_avp(avp_e_utran_vector, s6a_rand, &avp_rand), return );
if (avp)
{
CHECK_FCT_DO( fd_msg_avp_hdr(avp_rand, &hdr), return );
memcpy(mme_ue->rand, hdr->avp_value->os.data, hdr->avp_value->os.len);
}
else
error++;
CHECK_FCT_DO(
fd_avp_search_avp(avp_e_utran_vector, s6a_autn, &avp_autn), return );
if (avp)
{
CHECK_FCT_DO( fd_msg_avp_hdr(avp_autn, &hdr), return );
memcpy(mme_ue->autn, hdr->avp_value->os.data, hdr->avp_value->os.len);
}
else
error++;
event_set(&e, MME_EVT_EMM_UE_FROM_S6A);
event_set_param1(&e, (c_uintptr_t)mme_ue->index);
event_set_param2(&e, (c_uintptr_t)S6A_CMD_AUTHENTICATION_INFORMATION);
mme_event_send(&e);
out:
/* Free the message */
d_assert(pthread_mutex_lock(&fd_self()->stats_lock) == 0,,);
CHECK_POSIX_DO( pthread_mutex_lock(&fd_self()->stats_lock), );
dur = ((ts.tv_sec - mi->ts.tv_sec) * 1000000) +
((ts.tv_nsec - mi->ts.tv_nsec) / 1000);
if (fd_self()->stats.nb_recv)
@ -123,7 +188,8 @@ out:
fd_self()->stats.nb_errs++;
else
fd_self()->stats.nb_recv++;
d_assert(pthread_mutex_unlock(&fd_self()->stats_lock) == 0,,);
CHECK_POSIX_DO( pthread_mutex_unlock(&fd_self()->stats_lock), );
/* Display how long it took */
if (ts.tv_nsec > mi->ts.tv_nsec)
@ -135,13 +201,16 @@ out:
(int)(ts.tv_sec + 1 - mi->ts.tv_sec),
(long)(1000000000 + ts.tv_nsec - mi->ts.tv_nsec) / 1000);
d_assert(fd_msg_free(*msg) == 0,,);
CHECK_FCT_DO( fd_msg_free(*msg), return );
*msg = NULL;
pool_free_node(&sess_state_pool, mi);
return;
}
/* MME Sends Authentication Information Request to HSS */
int mme_s6a_send_air(mme_ue_t *mme_ue)
void mme_s6a_send_air(mme_ue_t *mme_ue)
{
struct msg *req = NULL;
struct avp *avp;
@ -151,103 +220,98 @@ int mme_s6a_send_air(mme_ue_t *mme_ue)
struct session *sess = NULL;
enb_ue_t *enb_ue = NULL;
d_assert(mme_ue, return -1, "Null Param");
d_assert(mme_ue, return, "Null Param");
enb_ue = mme_ue->enb_ue;
d_assert(enb_ue, return -1, "Null Param");
d_assert(enb_ue, return, "Null Param");
/* Clear Security Context */
CLEAR_SECURITY_CONTEXT(mme_ue);
/* Create the random value to store with the session */
pool_alloc_node(&sess_state_pool, &mi);
d_assert(mi, return -1, "malloc failed: %s", strerror(errno));
d_assert(mi, return, "malloc failed: %s", strerror(errno));
mi->mme_ue = mme_ue;
/* Create the request */
d_assert(fd_msg_new(s6a_cmd_air, MSGFL_ALLOC_ETEID, &req) == 0,
pool_free_node(&sess_state_pool, mi); return -1,);
CHECK_FCT_DO( fd_msg_new(s6a_cmd_air, MSGFL_ALLOC_ETEID, &req), goto out );
/* Create a new session */
#define S6A_APP_SID_OPT "app_s6a"
d_assert(fd_msg_new_session(req, (os0_t)S6A_APP_SID_OPT,
CONSTSTRLEN(S6A_APP_SID_OPT)) == 0, goto out,);
d_assert(fd_msg_sess_get(fd_g_config->cnf_dict, req, &sess, NULL) == 0,
goto out, );
CHECK_FCT_DO( fd_msg_new_session(req, (os0_t)S6A_APP_SID_OPT,
CONSTSTRLEN(S6A_APP_SID_OPT)), goto out );
CHECK_FCT_DO( fd_msg_sess_get(fd_g_config->cnf_dict, req, &sess, NULL),
goto out );
/* Set the Auth-Session-State AVP */
d_assert(fd_msg_avp_new(s6a_auth_session_state, 0, &avp) == 0, goto out,);
CHECK_FCT_DO( fd_msg_avp_new(s6a_auth_session_state, 0, &avp), goto out );
val.i32 = 1;
d_assert(fd_msg_avp_setvalue(avp, &val) == 0, goto out,);
d_assert(fd_msg_avp_add(req, MSG_BRW_LAST_CHILD, avp) == 0, goto out,);
CHECK_FCT_DO( fd_msg_avp_setvalue(avp, &val), goto out );
CHECK_FCT_DO( fd_msg_avp_add(req, MSG_BRW_LAST_CHILD, avp), goto out );
/* Set Origin-Host & Origin-Realm */
d_assert(fd_msg_add_origin(req, 0) == 0, goto out, );
CHECK_FCT_DO( fd_msg_add_origin(req, 0), goto out );
/* Set the Destination-Realm AVP */
d_assert(fd_msg_avp_new(s6a_destination_realm, 0, &avp) == 0, goto out,);
CHECK_FCT_DO( fd_msg_avp_new(s6a_destination_realm, 0, &avp), goto out );
val.os.data = (unsigned char *)(fd_g_config->cnf_diamrlm);
val.os.len = strlen(fd_g_config->cnf_diamrlm);
d_assert(fd_msg_avp_setvalue(avp, &val) == 0, goto out, );
d_assert(fd_msg_avp_add(req, MSG_BRW_LAST_CHILD, avp) == 0, goto out, );
CHECK_FCT_DO( fd_msg_avp_setvalue(avp, &val), goto out );
CHECK_FCT_DO( fd_msg_avp_add(req, MSG_BRW_LAST_CHILD, avp), goto out );
/* Set the User-Name AVP */
d_assert(fd_msg_avp_new(s6a_user_name, 0, &avp) == 0, goto out,);
CHECK_FCT_DO( fd_msg_avp_new(s6a_user_name, 0, &avp), goto out );
val.os.data = (c_uint8_t *)mme_ue->imsi_bcd;
val.os.len = strlen(mme_ue->imsi_bcd);
d_assert(fd_msg_avp_setvalue(avp, &val) == 0, goto out, );
d_assert(fd_msg_avp_add(req, MSG_BRW_LAST_CHILD, avp) == 0, goto out,);
CHECK_FCT_DO( fd_msg_avp_setvalue(avp, &val), goto out );
CHECK_FCT_DO( fd_msg_avp_add(req, MSG_BRW_LAST_CHILD, avp), goto out );
/* Add the Authentication-Info */
d_assert(fd_msg_avp_new(s6a_req_eutran_auth_info, 0, &avp) == 0, goto out,);
d_assert(fd_msg_avp_new(s6a_number_of_requested_vectors, 0,
&avpch) == 0, goto out,);
CHECK_FCT_DO( fd_msg_avp_new(s6a_req_eutran_auth_info, 0, &avp), goto out );
CHECK_FCT_DO( fd_msg_avp_new(s6a_number_of_requested_vectors, 0, &avpch),
goto out );
val.u32 = 1;
d_assert(fd_msg_avp_setvalue (avpch, &val) == 0, goto out,);
d_assert(fd_msg_avp_add (avp, MSG_BRW_LAST_CHILD, avpch) == 0, goto out,);
CHECK_FCT_DO( fd_msg_avp_setvalue (avpch, &val), goto out );
CHECK_FCT_DO( fd_msg_avp_add (avp, MSG_BRW_LAST_CHILD, avpch), goto out );
d_assert(fd_msg_avp_new(s6a_immediate_response_preferred, 0,
&avpch) == 0, goto out,);
CHECK_FCT_DO( fd_msg_avp_new(s6a_immediate_response_preferred, 0, &avpch),
goto out );
val.u32 = 1;
d_assert(fd_msg_avp_setvalue(avpch, &val) == 0, goto out,);
d_assert(fd_msg_avp_add(avp, MSG_BRW_LAST_CHILD, avpch) == 0, goto out,);
CHECK_FCT_DO( fd_msg_avp_setvalue(avpch, &val), goto out );
CHECK_FCT_DO( fd_msg_avp_add(avp, MSG_BRW_LAST_CHILD, avpch), goto out );
d_assert(fd_msg_avp_add(req, MSG_BRW_LAST_CHILD, avp) == 0, goto out,);
CHECK_FCT_DO( fd_msg_avp_add(req, MSG_BRW_LAST_CHILD, avp), goto out );
/* Set the Visited-PLMN-Id AVP */
d_assert(fd_msg_avp_new(s6a_visited_plmn_id, 0, &avp) == 0, goto out,);
CHECK_FCT_DO( fd_msg_avp_new(s6a_visited_plmn_id, 0, &avp), goto out );
val.os.data = (c_uint8_t *)&mme_ue->visited_plmn_id;
val.os.len = PLMN_ID_LEN;
d_assert(fd_msg_avp_setvalue(avp, &val) == 0, goto out,);
d_assert(fd_msg_avp_add(req, MSG_BRW_LAST_CHILD, avp) == 0, goto out,);
CHECK_FCT_DO( fd_msg_avp_setvalue(avp, &val), goto out );
CHECK_FCT_DO( fd_msg_avp_add(req, MSG_BRW_LAST_CHILD, avp), goto out );
d_assert(clock_gettime(CLOCK_REALTIME, &mi->ts) == 0, goto out,);
CHECK_SYS_DO( clock_gettime(CLOCK_REALTIME, &mi->ts), goto out );
/* Keep a pointer to the session data for debug purpose,
* in real life we would not need it */
svg = mi;
/* Store this value in the session */
d_assert(fd_sess_state_store(mme_s6a_reg, sess, &mi) == 0, goto out,);
CHECK_FCT_DO( fd_sess_state_store(mme_s6a_reg, sess, &mi), goto out );
/* Send the request */
d_assert(fd_msg_send(&req, mme_s6a_aia_cb, svg) == 0, goto out,);
CHECK_FCT_DO( fd_msg_send(&req, mme_s6a_aia_cb, svg), goto out );
/* Increment the counter */
d_assert(pthread_mutex_lock(&fd_self()->stats_lock) == 0,,);
CHECK_POSIX_DO( pthread_mutex_lock(&fd_self()->stats_lock), );
fd_self()->stats.nb_sent++;
d_assert(pthread_mutex_unlock(&fd_self()->stats_lock) == 0,, );
CHECK_POSIX_DO( pthread_mutex_unlock(&fd_self()->stats_lock), );
d_trace(3, "[S6A] Authentication-Information-Request : UE[%s] --> HSS\n",
mme_ue->imsi_bcd);
return 0;
out:
d_assert(fd_msg_free(req) == 0,,);
pool_free_node(&sess_state_pool, mi);
return -1;
return;
}
/* MME received Update Location Answer from HSS */
@ -267,164 +331,295 @@ static void mme_s6a_ula_cb(void *data, struct msg **msg)
mme_ue_t *mme_ue = NULL;
pdn_t *pdn = NULL;
CHECK_SYS_DO(clock_gettime(CLOCK_REALTIME, &ts), return);
CHECK_SYS_DO( clock_gettime(CLOCK_REALTIME, &ts), return );
/* Search the session, retrieve its data */
d_assert(fd_msg_sess_get(fd_g_config->cnf_dict, *msg, &sess, &new) == 0 &&
new == 0, return,);
CHECK_FCT_DO( fd_msg_sess_get(fd_g_config->cnf_dict, *msg, &sess, &new),
return );
d_assert(new == 0, return, );
d_assert(fd_sess_state_retrieve(mme_s6a_reg, sess, &mi) == 0 &&
mi && (void *)mi == data, fd_msg_free(*msg); *msg = NULL; return,);
CHECK_FCT_DO( fd_sess_state_retrieve(mme_s6a_reg, sess, &mi), return );
d_assert(mi && (void *)mi == data, return, );
mme_ue = mi->mme_ue;
d_assert(mme_ue, error++; goto out,);
d_assert(mme_ue, return, );
d_trace(3, "[S6A] Update-Location-Response : UE[%s] <-- HSS\n",
mme_ue->imsi_bcd);
/* Value of Result Code */
d_assert(fd_msg_search_avp(*msg, s6a_result_code, &avp) == 0 && avp,
error++; goto out,);
d_assert(fd_msg_avp_hdr(avp, &hdr) == 0 && hdr, error++; goto out,);
if (hdr->avp_value->i32 != ER_DIAMETER_SUCCESS)
CHECK_FCT_DO( fd_msg_search_avp(*msg, s6a_result_code, &avp), return );
if (avp)
{
d_error("ERROR DIAMETER Result Code(%d)", hdr->avp_value->i32);
CHECK_FCT_DO( fd_msg_avp_hdr(avp, &hdr), return);
d_trace(3, "Status: %d\n", hdr->avp_value->i32);
if (hdr->avp_value->i32 != ER_DIAMETER_SUCCESS)
{
d_error("ERROR DIAMETER Result Code(%d)", hdr->avp_value->i32);
error++;
}
}
else
{
d_error("no Result-Code");
error++;
goto out;
}
d_assert(fd_msg_search_avp(*msg, s6a_ula_flags, &avp) == 0
&& avp, error++; goto out,);
d_assert(fd_msg_avp_hdr(avp, &hdr) == 0 && hdr, error++; goto out,);
mme_ue->ula_flags = hdr->avp_value->i32;
d_assert(fd_msg_search_avp(*msg, s6a_subscription_data, &avp) == 0 && avp,
error++; goto out,);
d_assert(fd_avp_search_avp(avp, s6a_ambr, &avpch1) == 0 && avpch1,
error++; goto out,);
d_assert(fd_avp_search_avp(avpch1, s6a_max_bandwidth_ul, &avpch2) == 0 &&
avpch2, error++; goto out,);
d_assert(fd_msg_avp_hdr(avpch2, &hdr) == 0 && hdr, error++; goto out,);
mme_ue->max_bandwidth_ul = hdr->avp_value->i32;
d_assert(fd_avp_search_avp(avpch1, s6a_max_bandwidth_dl, &avpch2) == 0 &&
avpch2, error++; goto out,);
d_assert(fd_msg_avp_hdr(avpch2, &hdr) == 0 && hdr, error++; goto out,);
mme_ue->max_bandwidth_dl = hdr->avp_value->i32;
d_assert(fd_avp_search_avp(avp, s6a_apn_configuration_profile, &avpch1)
== 0 && avpch1, error++; goto out,);
d_assert(fd_msg_browse(avpch1, MSG_BRW_FIRST_CHILD, &avpch2, NULL) == 0 &&
avpch2, error++;; goto out,);
while(avpch2)
/* Value of Origin-Host */
CHECK_FCT_DO( fd_msg_search_avp(*msg, s6a_origin_host, &avp), return );
if (avp)
{
d_assert(fd_msg_avp_hdr(avpch2, &hdr) == 0 && hdr, error++; goto out,);
switch(hdr->avp_code)
CHECK_FCT_DO( fd_msg_avp_hdr(avp, &hdr), return );
d_trace(3, "From '%.*s' ",
(int)hdr->avp_value->os.len, hdr->avp_value->os.data);
}
else
{
d_error("no_Origin-Host ");
error++;
}
/* Value of Origin-Realm */
CHECK_FCT_DO( fd_msg_search_avp(*msg, s6a_origin_realm, &avp), return );
if (avp)
{
CHECK_FCT_DO( fd_msg_avp_hdr(avp, &hdr), return );
d_trace(3, "('%.*s') ",
(int)hdr->avp_value->os.len, hdr->avp_value->os.data);
}
else
{
d_error("no_Origin-Realm ");
error++;
}
CHECK_FCT_DO( fd_msg_search_avp(*msg, s6a_ula_flags, &avp), return);
if (avp)
{
CHECK_FCT_DO( fd_msg_avp_hdr(avp, &hdr), return );
mme_ue->ula_flags = hdr->avp_value->i32;
}
else
error++;
CHECK_FCT_DO(
fd_msg_search_avp(*msg, s6a_subscription_data, &avp), return );
if (avp)
{
CHECK_FCT_DO( fd_avp_search_avp(avp, s6a_ambr, &avpch1), return );
if (avpch1)
{
case AVP_CODE_CONTEXT_IDENTIFIER:
break;
case AVP_CODE_ALL_APN_CONFIG_INC_IND:
break;
case AVP_CODE_APN_CONFIGURATION:
CHECK_FCT_DO( fd_avp_search_avp(
avpch1, s6a_max_bandwidth_ul, &avpch2), return);
if (avpch2)
{
c_int8_t apn[MAX_APN_LEN];
d_assert(fd_avp_search_avp(avpch2, s6a_service_selection,
&avpch3) == 0 && avpch3, error++; goto out,);
d_assert(fd_msg_avp_hdr(avpch3, &hdr) == 0 && hdr,
error++; goto out,);
memcpy(apn, hdr->avp_value->os.data, hdr->avp_value->os.len);
apn[hdr->avp_value->os.len] = 0;
pdn = mme_pdn_find_by_apn(mme_ue, apn);
if (!pdn)
pdn = mme_pdn_add(mme_ue, apn);
else
d_warn("Duplicate APN:[%s]", apn);
d_assert(pdn, error++; goto out,);
d_assert(fd_avp_search_avp(avpch2, s6a_context_identifier,
&avpch3) == 0 && avpch3, error++; goto out,);
d_assert(fd_msg_avp_hdr(avpch3, &hdr) == 0 && hdr,
error++; goto out,);
pdn->id = hdr->avp_value->i32;
d_assert(fd_avp_search_avp(avpch2, s6a_pdn_type,
&avpch3) == 0 && avpch3, error++; goto out,);
d_assert(fd_msg_avp_hdr(avpch3, &hdr) == 0 && hdr,
error++; goto out,);
pdn->s6a_type = hdr->avp_value->i32;
d_assert(fd_avp_search_avp(avpch2,
s6a_eps_subscribed_qos_profile, &avpch3) == 0 &&
avpch3, error++; goto out,);
d_assert(fd_avp_search_avp(avpch3, s6a_qos_class_identifier,
&avpch4) == 0 && avpch4, error++; goto out,);
d_assert(fd_msg_avp_hdr(avpch4, &hdr) == 0 && hdr,
error++; goto out,);
pdn->qci = hdr->avp_value->i32;
d_assert(fd_avp_search_avp(avpch3,
s6a_allocation_retention_priority, &avpch4) == 0 &&
avpch4, error++; goto out,);
d_assert(fd_avp_search_avp(avpch4, s6a_priority_level,
&avpch5) == 0 && avpch5, error++; goto out,);
d_assert(fd_msg_avp_hdr(avpch5, &hdr) == 0 && hdr,
error++; goto out,);
pdn->priority_level = hdr->avp_value->i32;
d_assert(fd_avp_search_avp(avpch4, s6a_pre_emption_capability,
&avpch5) == 0 && avpch5, error++; goto out,);
d_assert(fd_msg_avp_hdr(avpch5, &hdr) == 0 && hdr,
error++; goto out,);
pdn->pre_emption_capability = hdr->avp_value->i32;
d_assert(fd_avp_search_avp(avpch4,
s6a_pre_emption_vulnerability, &avpch5) == 0 &&
avpch5, error++; goto out,);
d_assert(fd_msg_avp_hdr(avpch5, &hdr) == 0 && hdr,
error++; goto out,);
pdn->pre_emption_vulnerability = hdr->avp_value->i32;
d_assert(fd_avp_search_avp(avpch2, s6a_ambr,
&avpch3) == 0 && avpch3, error++; goto out,);
d_assert(fd_avp_search_avp(avpch3, s6a_max_bandwidth_ul,
&avpch4) == 0 && avpch4, error++; goto out,);
d_assert(fd_msg_avp_hdr(avpch4, &hdr) == 0 && hdr,
error++; goto out,);
pdn->max_bandwidth_ul = hdr->avp_value->i32;
d_assert(fd_avp_search_avp(avpch3,
s6a_max_bandwidth_dl, &avpch4) == 0 && avpch4,
error++; goto out,);
d_assert(fd_msg_avp_hdr(avpch4, &hdr) == 0 && hdr,
error++; goto out,);
pdn->max_bandwidth_dl = hdr->avp_value->i32;
break;
CHECK_FCT_DO( fd_msg_avp_hdr(avpch2, &hdr), return );
mme_ue->max_bandwidth_ul = hdr->avp_value->i32;
}
default:
else
error++;
CHECK_FCT_DO(fd_avp_search_avp(
avpch1, s6a_max_bandwidth_dl, &avpch2), return );
if (avpch2)
{
d_warn("Unknown AVP-code:%d", hdr->avp_code);
break;
CHECK_FCT_DO( fd_msg_avp_hdr(avpch2, &hdr), return );
mme_ue->max_bandwidth_dl = hdr->avp_value->i32;
}
else
error++;
}
else
error++;
CHECK_FCT_DO( fd_avp_search_avp(
avp, s6a_apn_configuration_profile, &avpch1), return );
if (avpch1)
{
CHECK_FCT_DO( fd_msg_browse(
avpch1, MSG_BRW_FIRST_CHILD, &avpch2, NULL), return );
while(avpch2)
{
CHECK_FCT_DO( fd_msg_avp_hdr(avpch2, &hdr), return );
switch(hdr->avp_code)
{
case AVP_CODE_CONTEXT_IDENTIFIER:
break;
case AVP_CODE_ALL_APN_CONFIG_INC_IND:
break;
case AVP_CODE_APN_CONFIGURATION:
{
c_int8_t apn[MAX_APN_LEN];
CHECK_FCT_DO( fd_avp_search_avp(
avpch2, s6a_service_selection, &avpch3), return );
if (avpch3)
{
CHECK_FCT_DO(
fd_msg_avp_hdr(avpch3, &hdr), return );
memcpy(apn, hdr->avp_value->os.data,
hdr->avp_value->os.len);
apn[hdr->avp_value->os.len] = 0;
pdn = mme_pdn_find_by_apn(mme_ue, apn);
if (!pdn)
pdn = mme_pdn_add(mme_ue, apn);
else
d_warn("Duplicate APN:[%s]", apn);
}
else
error++;
d_assert(pdn, return, );
CHECK_FCT_DO( fd_avp_search_avp(avpch2,
s6a_context_identifier, &avpch3), return );
if (avpch3)
{
CHECK_FCT_DO(
fd_msg_avp_hdr(avpch3, &hdr), return );
pdn->id = hdr->avp_value->i32;
}
else
error++;
CHECK_FCT_DO( fd_avp_search_avp(avpch2,
s6a_pdn_type, &avpch3), return );
if (avpch3)
{
CHECK_FCT_DO(
fd_msg_avp_hdr(avpch3, &hdr), return );
pdn->s6a_type = hdr->avp_value->i32;
}
else
error++;
CHECK_FCT_DO( fd_avp_search_avp(avpch2,
s6a_eps_subscribed_qos_profile, &avpch3), return );
if (avpch3)
{
CHECK_FCT_DO(fd_avp_search_avp(avpch3,
s6a_qos_class_identifier, &avpch4), return );
if (avpch4)
{
CHECK_FCT_DO(
fd_msg_avp_hdr(avpch4, &hdr), return );
pdn->qci = hdr->avp_value->i32;
}
else
error++;
CHECK_FCT_DO(fd_avp_search_avp(avpch3,
s6a_allocation_retention_priority, &avpch4),
return );
if (avpch4)
{
CHECK_FCT_DO( fd_avp_search_avp(avpch4,
s6a_priority_level, &avpch5), return );
if (avpch5)
{
CHECK_FCT_DO( fd_msg_avp_hdr(avpch5, &hdr),
return );
pdn->priority_level = hdr->avp_value->i32;
}
else
error++;
CHECK_FCT_DO( fd_avp_search_avp(avpch4,
s6a_pre_emption_capability, &avpch5),
return );
if (avpch5)
{
CHECK_FCT_DO( fd_msg_avp_hdr(avpch5, &hdr),
return );
pdn->pre_emption_capability =
hdr->avp_value->i32;
}
else
error++;
CHECK_FCT_DO( fd_avp_search_avp(avpch4,
s6a_pre_emption_vulnerability, &avpch5),
return );
if (avpch5)
{
CHECK_FCT_DO( fd_msg_avp_hdr(avpch5, &hdr),
return );
pdn->pre_emption_vulnerability =
hdr->avp_value->i32;
}
else
error++;
}
else
error++;
}
else
error++;
CHECK_FCT_DO( fd_avp_search_avp(avpch2, s6a_ambr,
&avpch3), return );
if (avpch3)
{
CHECK_FCT_DO( fd_avp_search_avp(avpch3,
s6a_max_bandwidth_ul, &avpch4), return );
if (avpch4)
{
CHECK_FCT_DO(
fd_msg_avp_hdr(avpch4, &hdr), return );
pdn->max_bandwidth_ul = hdr->avp_value->i32;
}
else
error++;
CHECK_FCT_DO( fd_avp_search_avp(avpch3,
s6a_max_bandwidth_dl, &avpch4), return );
if (avpch4)
{
CHECK_FCT_DO(
fd_msg_avp_hdr(avpch4, &hdr), return );
pdn->max_bandwidth_dl = hdr->avp_value->i32;
}
else
error++;
}
break;
}
default:
{
d_warn("Unknown AVP-code:%d", hdr->avp_code);
break;
}
}
fd_msg_browse(avpch2, MSG_BRW_NEXT, &avpch2, NULL);
}
}
fd_msg_browse(avpch2, MSG_BRW_NEXT, &avpch2, NULL);
else
error++;
}
else
error++;
d_assert(fd_msg_search_avp(*msg, s6a_subscribed_rau_tau_timer, &avp) == 0
&& avp, error++; goto out,);
d_assert(fd_msg_avp_hdr(avp, &hdr) == 0 && hdr, error++; goto out,);
mme_ue->subscribed_rau_tau_timer = hdr->avp_value->i32;
CHECK_FCT_DO(
fd_msg_search_avp(*msg, s6a_subscribed_rau_tau_timer, &avp), return );
if (avp)
{
CHECK_FCT_DO( fd_msg_avp_hdr(avp, &hdr), return );
mme_ue->subscribed_rau_tau_timer = hdr->avp_value->i32;
}
else
error++;
event_set(&e, MME_EVT_EMM_UE_FROM_S6A);
event_set_param1(&e, (c_uintptr_t)mme_ue->index);
event_set_param2(&e, (c_uintptr_t)S6A_CMD_UPDATE_LOCATION);
mme_event_send(&e);
out:
/* Free the message */
d_assert(pthread_mutex_lock(&fd_self()->stats_lock) == 0,,);
CHECK_POSIX_DO( pthread_mutex_lock(&fd_self()->stats_lock), );
dur = ((ts.tv_sec - mi->ts.tv_sec) * 1000000) +
((ts.tv_nsec - mi->ts.tv_nsec) / 1000);
if (fd_self()->stats.nb_recv)
@ -448,7 +643,8 @@ out:
fd_self()->stats.nb_errs++;
else
fd_self()->stats.nb_recv++;
d_assert(pthread_mutex_unlock(&fd_self()->stats_lock) == 0,,);
CHECK_POSIX_DO( pthread_mutex_unlock(&fd_self()->stats_lock), );
/* Display how long it took */
if (ts.tv_nsec > mi->ts.tv_nsec)
@ -460,13 +656,16 @@ out:
(int)(ts.tv_sec + 1 - mi->ts.tv_sec),
(long)(1000000000 + ts.tv_nsec - mi->ts.tv_nsec) / 1000);
d_assert(fd_msg_free(*msg) == 0,,);
CHECK_FCT_DO( fd_msg_free(*msg), return );
*msg = NULL;
pool_free_node(&sess_state_pool, mi);
return;
}
/* MME Sends Update Location Request to HSS */
int mme_s6a_send_ulr(mme_ue_t *mme_ue)
void mme_s6a_send_ulr(mme_ue_t *mme_ue)
{
struct msg *req = NULL;
struct avp *avp;
@ -474,111 +673,105 @@ int mme_s6a_send_ulr(mme_ue_t *mme_ue)
struct sess_state *mi = NULL, *svg;
struct session *sess = NULL;
d_assert(mme_ue, return -1, "Null Param");
d_assert(mme_ue, return, "Null Param");
/* Create the random value to store with the session */
pool_alloc_node(&sess_state_pool, &mi);
d_assert(mi, return -1, "malloc failed: %s", strerror(errno));
d_assert(mi, return, "malloc failed: %s", strerror(errno));
mi->mme_ue = mme_ue;
/* Create the request */
d_assert(fd_msg_new(s6a_cmd_ulr, MSGFL_ALLOC_ETEID, &req) == 0,
pool_free_node(&sess_state_pool, mi); return -1,);
CHECK_FCT_DO( fd_msg_new(s6a_cmd_ulr, MSGFL_ALLOC_ETEID, &req), goto out );
/* Create a new session */
#define S6A_APP_SID_OPT "app_s6a"
d_assert(fd_msg_new_session(req, (os0_t)S6A_APP_SID_OPT,
CONSTSTRLEN(S6A_APP_SID_OPT)) == 0, goto out,);
d_assert(fd_msg_sess_get(fd_g_config->cnf_dict, req, &sess, NULL) == 0,
goto out, );
CHECK_FCT_DO( fd_msg_new_session(req, (os0_t)S6A_APP_SID_OPT,
CONSTSTRLEN(S6A_APP_SID_OPT)), goto out );
CHECK_FCT_DO( fd_msg_sess_get(fd_g_config->cnf_dict, req, &sess, NULL),
goto out );
/* Set the Auth-Session-State AVP */
d_assert(fd_msg_avp_new(s6a_auth_session_state, 0, &avp) == 0, goto out,);
CHECK_FCT_DO( fd_msg_avp_new(s6a_auth_session_state, 0, &avp), goto out );
val.i32 = 1;
d_assert(fd_msg_avp_setvalue(avp, &val) == 0, goto out,);
d_assert(fd_msg_avp_add(req, MSG_BRW_LAST_CHILD, avp) == 0, goto out,);
CHECK_FCT_DO( fd_msg_avp_setvalue(avp, &val), goto out );
CHECK_FCT_DO( fd_msg_avp_add(req, MSG_BRW_LAST_CHILD, avp), goto out );
/* Set Origin-Host & Origin-Realm */
d_assert(fd_msg_add_origin(req, 0) == 0, goto out, );
CHECK_FCT_DO( fd_msg_add_origin(req, 0), goto out );
/* Set the Destination-Realm AVP */
d_assert(fd_msg_avp_new(s6a_destination_realm, 0, &avp) == 0, goto out,);
CHECK_FCT_DO( fd_msg_avp_new(s6a_destination_realm, 0, &avp), goto out );
val.os.data = (unsigned char *)(fd_g_config->cnf_diamrlm);
val.os.len = strlen(fd_g_config->cnf_diamrlm);
d_assert(fd_msg_avp_setvalue(avp, &val) == 0, goto out, );
d_assert(fd_msg_avp_add(req, MSG_BRW_LAST_CHILD, avp) == 0, goto out, );
CHECK_FCT_DO( fd_msg_avp_setvalue(avp, &val), goto out );
CHECK_FCT_DO( fd_msg_avp_add(req, MSG_BRW_LAST_CHILD, avp), goto out );
/* Set the User-Name AVP */
d_assert(fd_msg_avp_new(s6a_user_name, 0, &avp) == 0, goto out,);
CHECK_FCT_DO( fd_msg_avp_new(s6a_user_name, 0, &avp), goto out );
val.os.data = (c_uint8_t *)mme_ue->imsi_bcd;
val.os.len = strlen(mme_ue->imsi_bcd);
d_assert(fd_msg_avp_setvalue(avp, &val) == 0, goto out, );
d_assert(fd_msg_avp_add(req, MSG_BRW_LAST_CHILD, avp) == 0, goto out,);
CHECK_FCT_DO( fd_msg_avp_setvalue(avp, &val), goto out );
CHECK_FCT_DO( fd_msg_avp_add(req, MSG_BRW_LAST_CHILD, avp), goto out );
/* Set the RAT-Type */
d_assert(fd_msg_avp_new(s6a_rat_type, 0, &avp) == 0, goto out,);
CHECK_FCT_DO( fd_msg_avp_new(s6a_rat_type, 0, &avp), goto out );
val.u32 = S6A_RAT_TYPE_EUTRAN;
d_assert(fd_msg_avp_setvalue(avp, &val) == 0, goto out,);
d_assert(fd_msg_avp_add(req, MSG_BRW_LAST_CHILD, avp) == 0, goto out,);
CHECK_FCT_DO( fd_msg_avp_setvalue(avp, &val), goto out );
CHECK_FCT_DO( fd_msg_avp_add(req, MSG_BRW_LAST_CHILD, avp), goto out );
/* Set the ULR-Flags */
d_assert(fd_msg_avp_new(s6a_ulr_flags, 0, &avp) == 0, goto out,);
CHECK_FCT_DO( fd_msg_avp_new(s6a_ulr_flags, 0, &avp), goto out );
val.u32 = S6A_ULR_S6A_S6D_INDICATOR;
d_assert(fd_msg_avp_setvalue(avp, &val) == 0, goto out,);
d_assert(fd_msg_avp_add(req, MSG_BRW_LAST_CHILD, avp) == 0, goto out,);
CHECK_FCT_DO( fd_msg_avp_setvalue(avp, &val), goto out );
CHECK_FCT_DO( fd_msg_avp_add(req, MSG_BRW_LAST_CHILD, avp), goto out );
/* Set the Visited-PLMN-Id */
d_assert(fd_msg_avp_new(s6a_visited_plmn_id, 0, &avp) == 0, goto out,);
CHECK_FCT_DO( fd_msg_avp_new(s6a_visited_plmn_id, 0, &avp), goto out );
val.os.data = (c_uint8_t *)&mme_ue->visited_plmn_id;
val.os.len = PLMN_ID_LEN;
d_assert(fd_msg_avp_setvalue(avp, &val) == 0, goto out,);
d_assert(fd_msg_avp_add(req, MSG_BRW_LAST_CHILD, avp) == 0, goto out,);
CHECK_FCT_DO( fd_msg_avp_setvalue(avp, &val), goto out );
CHECK_FCT_DO( fd_msg_avp_add(req, MSG_BRW_LAST_CHILD, avp), goto out );
/* Set the UE-SRVCC Capability */
d_assert(fd_msg_avp_new(s6a_ue_srvcc_cap, 0, &avp) == 0, goto out,);
CHECK_FCT_DO( fd_msg_avp_new(s6a_ue_srvcc_cap, 0, &avp), goto out );
val.u32 = S6A_UE_SRVCC_NOT_SUPPORTED;
d_assert(fd_msg_avp_setvalue(avp, &val) == 0, goto out,);
d_assert(fd_msg_avp_add(req, MSG_BRW_LAST_CHILD, avp) == 0, goto out,);
CHECK_FCT_DO( fd_msg_avp_setvalue(avp, &val), goto out );
CHECK_FCT_DO( fd_msg_avp_add(req, MSG_BRW_LAST_CHILD, avp), goto out );
d_assert(clock_gettime(CLOCK_REALTIME, &mi->ts) == 0, goto out,);
CHECK_SYS_DO( clock_gettime(CLOCK_REALTIME, &mi->ts), goto out );
/* Keep a pointer to the session data for debug purpose,
* in real life we would not need it */
svg = mi;
/* Store this value in the session */
d_assert(fd_sess_state_store(mme_s6a_reg, sess, &mi) == 0, goto out,);
CHECK_FCT_DO( fd_sess_state_store(mme_s6a_reg, sess, &mi), goto out );
/* Send the request */
d_assert(fd_msg_send(&req, mme_s6a_ula_cb, svg) == 0, goto out,);
CHECK_FCT_DO( fd_msg_send(&req, mme_s6a_ula_cb, svg), goto out );
/* Increment the counter */
d_assert(pthread_mutex_lock(&fd_self()->stats_lock) == 0,,);
CHECK_POSIX_DO( pthread_mutex_lock(&fd_self()->stats_lock), );
fd_self()->stats.nb_sent++;
d_assert(pthread_mutex_unlock(&fd_self()->stats_lock) == 0,, );
CHECK_POSIX_DO( pthread_mutex_unlock(&fd_self()->stats_lock), );
d_trace(3, "[S6A] Update-Location-Request : UE[%s] --> HSS\n",
mme_ue->imsi_bcd);
return 0;
out:
d_assert(fd_msg_free(req) == 0,,);
pool_free_node(&sess_state_pool, mi);
return -1;
return;
}
status_t mme_s6a_init(void)
int mme_s6a_init(void)
{
pool_init(&sess_state_pool, MAX_NUM_SESSION_STATE);
/* Install objects definitions for this application */
CHECK_FCT( s6a_dict_init() );
d_assert(fd_sess_handler_create(&mme_s6a_reg,
(void *)free, NULL, NULL) == 0, return -1,);
CHECK_FCT( fd_sess_handler_create(&mme_s6a_reg, (void *)free, NULL, NULL) );
/* Advertise the support for the application in the peer */
CHECK_FCT( fd_disp_app_support ( s6a_appli, s6a_vendor, 1, 0 ) );
@ -588,7 +781,7 @@ status_t mme_s6a_init(void)
void mme_s6a_final(void)
{
d_assert(fd_sess_handler_destroy(&mme_s6a_reg, NULL) == 0,,);
CHECK_FCT_DO( fd_sess_handler_destroy(&mme_s6a_reg, NULL), );
if (pool_size(&sess_state_pool) != pool_avail(&sess_state_pool))
d_warn("%d not freed in sess_state_pool[%d] of S6A-SM",

6
src/mme/mme_s6a_handler.h
View File

@ -9,13 +9,13 @@
extern "C" {
#endif /* __cplusplus */
CORE_DECLARE(status_t) mme_s6a_init(void);
CORE_DECLARE(int) mme_s6a_init(void);
CORE_DECLARE(void) mme_s6a_final(void);
/* MME Sends Authentication Information Request to HSS */
CORE_DECLARE(int) mme_s6a_send_air(mme_ue_t *mme_ue);
CORE_DECLARE(void) mme_s6a_send_air(mme_ue_t *mme_ue);
/* MME Sends Update Location Request to HSS */
CORE_DECLARE(int) mme_s6a_send_ulr(mme_ue_t *mme_ue);
CORE_DECLARE(void) mme_s6a_send_ulr(mme_ue_t *mme_ue);
#ifdef __cplusplus
}