sysmocom
/
open5gs
forked from acouzens/open5gs
11
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
open5gs/src/pgw/pgw-fd-path.c

1184 lines
38 KiB
C
Raw Blame History

/*
* 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 "gtp/gtp-xact.h"
#include "fd/fd-lib.h"
#include "fd/gx/gx-dict.h"
#include "fd/gx/gx-message.h"
#include "pgw-event.h"
#include "pgw-fd-path.h"
static struct session_handler *pgw_gx_reg = NULL;
static struct disp_hdl *hdl_gx_fb = NULL;
static struct disp_hdl *hdl_gx_rar = NULL;
struct sess_state {
os0_t gx_sid; /* Gx Session-Id */
pgw_sess_t *sess;
gtp_xact_t *xact;
ogs_pkbuf_t *gtpbuf;
uint32_t cc_request_type;
uint32_t cc_request_number;
struct timespec ts; /* Time of sending the message */
};
static int decode_pcc_rule_definition(
pcc_rule_t *pcc_rule, struct avp *avpch1, int *perror);
static void pgw_gx_cca_cb(void *data, struct msg **msg);
static __inline__ struct sess_state *new_state(os0_t sid)
{
struct sess_state *new = NULL;
new = ogs_calloc(1, sizeof(*new));
new->gx_sid = (os0_t)ogs_strdup((char *)sid);
ogs_assert(new->gx_sid);
return new;
}
static void state_cleanup(struct sess_state *sess_data, os0_t sid, void *opaque)
{
if (sess_data->gx_sid)
ogs_free(sess_data->gx_sid);
ogs_free(sess_data);
}
void pgw_gx_send_ccr(pgw_sess_t *sess, gtp_xact_t *xact,
ogs_pkbuf_t *gtpbuf, uint32_t cc_request_type)
{
int ret;
struct msg *req = NULL;
struct avp *avp;
struct avp *avpch1, *avpch2;
union avp_value val;
struct sess_state *sess_data = NULL, *svg;
struct session *session = NULL;
int new;
gtp_message_t *message = NULL;
paa_t paa; /* For changing Framed-IPv6-Prefix Length to 128 */
ogs_assert(sess);
ogs_assert(sess->ipv4 || sess->ipv6);
ogs_assert(gtpbuf);
message = gtpbuf->data;
ogs_assert(message);
ogs_debug("[PGW2] Credit-Control-Request");
/* Create the request */
ret = fd_msg_new(gx_cmd_ccr, MSGFL_ALLOC_ETEID, &req);
ogs_assert(ret == 0);
/* Find Diameter Gx Session */
if (sess->gx_sid) {
/* Retrieve session by Session-Id */
size_t sidlen = strlen(sess->gx_sid);
ret = fd_sess_fromsid_msg((os0_t)sess->gx_sid, sidlen, &session, &new);
ogs_assert(ret == 0);
ogs_assert(new == 0);
/* Add Session-Id to the message */
ret = fd_message_session_id_set(req, (os0_t)sess->gx_sid, sidlen);
ogs_assert(ret == 0);
/* Save the session associated with the message */
ret = fd_msg_sess_set(req, session);
} else {
/* Create a new session */
#define GX_APP_SID_OPT "app_gx"
ret = fd_msg_new_session(req, (os0_t)GX_APP_SID_OPT,
CONSTSTRLEN(GX_APP_SID_OPT));
ogs_assert(ret == 0);
ret = fd_msg_sess_get(fd_g_config->cnf_dict, req, &session, NULL);
ogs_assert(ret == 0);
}
/* Retrieve session state in this session */
ret = fd_sess_state_retrieve(pgw_gx_reg, session, &sess_data);
if (!sess_data) {
os0_t sid;
size_t sidlen;
ret = fd_sess_getsid(session, &sid, &sidlen);
ogs_assert(ret == 0);
/* Allocate new session state memory */
sess_data = new_state(sid);
ogs_assert(sess_data);
/* Save Session-Id to PGW Session Context */
sess->gx_sid = (char*)sess_data->gx_sid;
}
/* Update session state */
sess_data->sess = sess;
sess_data->xact = xact;
sess_data->gtpbuf = gtpbuf;
sess_data->cc_request_type = cc_request_type;
if (cc_request_type == GX_CC_REQUEST_TYPE_INITIAL_REQUEST ||
cc_request_type == GX_CC_REQUEST_TYPE_EVENT_REQUEST)
sess_data->cc_request_number = 0;
else
sess_data->cc_request_number++;
/* Set Origin-Host & Origin-Realm */
ret = fd_msg_add_origin(req, 0);
ogs_assert(ret == 0);
/* Set the Destination-Realm AVP */
ret = fd_msg_avp_new(fd_destination_realm, 0, &avp);
ogs_assert(ret == 0);
val.os.data = (unsigned char *)(fd_g_config->cnf_diamrlm);
val.os.len = strlen(fd_g_config->cnf_diamrlm);
ret = fd_msg_avp_setvalue(avp, &val);
ogs_assert(ret == 0);
ret = fd_msg_avp_add(req, MSG_BRW_LAST_CHILD, avp);
ogs_assert(ret == 0);
/* Set the Auth-Application-Id AVP */
ret = fd_msg_avp_new(fd_auth_application_id, 0, &avp);
ogs_assert(ret == 0);
val.i32 = GX_APPLICATION_ID;
ret = fd_msg_avp_setvalue(avp, &val);
ogs_assert(ret == 0);
ret = fd_msg_avp_add(req, MSG_BRW_LAST_CHILD, avp);
ogs_assert(ret == 0);
/* Set CC-Request-Type, CC-Request-Number */
ret = fd_msg_avp_new(gx_cc_request_type, 0, &avp);
ogs_assert(ret == 0);
val.i32 = sess_data->cc_request_type;
ret = fd_msg_avp_setvalue(avp, &val);
ogs_assert(ret == 0);
ret = fd_msg_avp_add(req, MSG_BRW_LAST_CHILD, avp);
ogs_assert(ret == 0);
ret = fd_msg_avp_new(gx_cc_request_number, 0, &avp);
ogs_assert(ret == 0);
val.i32 = sess_data->cc_request_number;
ret = fd_msg_avp_setvalue(avp, &val);
ogs_assert(ret == 0);
ret = fd_msg_avp_add(req, MSG_BRW_LAST_CHILD, avp);
ogs_assert(ret == 0);
/* Set Subscription-Id */
ret = fd_msg_avp_new(gx_subscription_id, 0, &avp);
ogs_assert(ret == 0);
ret = fd_msg_avp_new(gx_subscription_id_type, 0, &avpch1);
ogs_assert(ret == 0);
val.i32 = GX_SUBSCRIPTION_ID_TYPE_END_USER_IMSI;
ret = fd_msg_avp_setvalue (avpch1, &val);
ogs_assert(ret == 0);
ret = fd_msg_avp_add (avp, MSG_BRW_LAST_CHILD, avpch1);
ogs_assert(ret == 0);
ret = fd_msg_avp_new(gx_subscription_id_data, 0, &avpch1);
ogs_assert(ret == 0);
val.os.data = (uint8_t *)sess->imsi_bcd;
val.os.len = strlen(sess->imsi_bcd);
ret = fd_msg_avp_setvalue (avpch1, &val);
ogs_assert(ret == 0);
ret = fd_msg_avp_add (avp, MSG_BRW_LAST_CHILD, avpch1);
ogs_assert(ret == 0);
ret = fd_msg_avp_add(req, MSG_BRW_LAST_CHILD, avp);
ogs_assert(ret == 0);
if (cc_request_type != GX_CC_REQUEST_TYPE_TERMINATION_REQUEST) {
/* Set Supported-Features */
ret = fd_msg_avp_new(gx_supported_features, 0, &avp);
ogs_assert(ret == 0);
ret = fd_msg_avp_new(gx_feature_list_id, 0, &avpch1);
ogs_assert(ret == 0);
val.i32 = 1;
ret = fd_msg_avp_setvalue (avpch1, &val);
ogs_assert(ret == 0);
ret = fd_msg_avp_add (avp, MSG_BRW_LAST_CHILD, avpch1);
ogs_assert(ret == 0);
ret = fd_msg_avp_new(gx_feature_list, 0, &avpch1);
ogs_assert(ret == 0);
val.u32 = 0x0000000b;
ret = fd_msg_avp_setvalue (avpch1, &val);
ogs_assert(ret == 0);
ret = fd_msg_avp_add (avp, MSG_BRW_LAST_CHILD, avpch1);
ogs_assert(ret == 0);
ret = fd_msg_avp_add(req, MSG_BRW_LAST_CHILD, avp);
ogs_assert(ret == 0);
/* Set Network-Request-Support */
ret = fd_msg_avp_new(gx_network_request_support, 0, &avp);
ogs_assert(ret == 0);
val.i32 = 1;
ret = fd_msg_avp_setvalue(avp, &val);
ogs_assert(ret == 0);
ret = fd_msg_avp_add(req, MSG_BRW_LAST_CHILD, avp);
ogs_assert(ret == 0);
/* Set Framed-IP-Address */
if (sess->ipv4) {
ret = fd_msg_avp_new(gx_framed_ip_address, 0, &avp);
ogs_assert(ret == 0);
val.os.data = (uint8_t*)&sess->ipv4->addr;
val.os.len = IPV4_LEN;
ret = fd_msg_avp_setvalue(avp, &val);
ogs_assert(ret == 0);
ret = fd_msg_avp_add(req, MSG_BRW_LAST_CHILD, avp);
ogs_assert(ret == 0);
}
/* Set Framed-IPv6-Prefix */
if (sess->ipv6) {
ret = fd_msg_avp_new(gx_framed_ipv6_prefix, 0, &avp);
ogs_assert(ret == 0);
memcpy(&paa, &sess->pdn.paa, PAA_IPV6_LEN);
#define FRAMED_IPV6_PREFIX_LENGTH 128 /* from spec document */
paa.len = FRAMED_IPV6_PREFIX_LENGTH;
val.os.data = (uint8_t*)&paa;
val.os.len = PAA_IPV6_LEN;
ret = fd_msg_avp_setvalue(avp, &val);
ogs_assert(ret == 0);
ret = fd_msg_avp_add(req, MSG_BRW_LAST_CHILD, avp);
ogs_assert(ret == 0);
}
/* Set IP-Can-Type */
ret = fd_msg_avp_new(gx_ip_can_type, 0, &avp);
ogs_assert(ret == 0);
val.i32 = GX_IP_CAN_TYPE_3GPP_EPS;
ret = fd_msg_avp_setvalue(avp, &val);
ogs_assert(ret == 0);
ret = fd_msg_avp_add(req, MSG_BRW_LAST_CHILD, avp);
ogs_assert(ret == 0);
/* Set RAT-Type */
ret = fd_msg_avp_new(gx_rat_type, 0, &avp);
ogs_assert(ret == 0);
val.i32 = GX_RAT_TYPE_EUTRAN;
ret = fd_msg_avp_setvalue(avp, &val);
ogs_assert(ret == 0);
ret = fd_msg_avp_add(req, MSG_BRW_LAST_CHILD, avp);
ogs_assert(ret == 0);
/* Set QoS-Information */
if (sess->pdn.ambr.downlink || sess->pdn.ambr.uplink) {
ret = fd_msg_avp_new(gx_qos_information, 0, &avp);
ogs_assert(ret == 0);
if (sess->pdn.ambr.uplink) {
ret = fd_msg_avp_new(gx_apn_aggregate_max_bitrate_ul,
0, &avpch1);
ogs_assert(ret == 0);
val.u32 = sess->pdn.ambr.uplink;
ret = fd_msg_avp_setvalue (avpch1, &val);
ogs_assert(ret == 0);
ret = fd_msg_avp_add (avp, MSG_BRW_LAST_CHILD, avpch1);
ogs_assert(ret == 0);
}
if (sess->pdn.ambr.downlink) {
ret = fd_msg_avp_new(gx_apn_aggregate_max_bitrate_dl, 0,
&avpch1);
ogs_assert(ret == 0);
val.u32 = sess->pdn.ambr.downlink;
ret = fd_msg_avp_setvalue (avpch1, &val);
ogs_assert(ret == 0);
ret = fd_msg_avp_add (avp, MSG_BRW_LAST_CHILD, avpch1);
ogs_assert(ret == 0);
}
ret = fd_msg_avp_add(req, MSG_BRW_LAST_CHILD, avp);
ogs_assert(ret == 0);
}
/* Set Default-EPS-Bearer-QoS */
ret = fd_msg_avp_new(gx_default_eps_bearer_qos, 0, &avp);
ogs_assert(ret == 0);
ret = fd_msg_avp_new(gx_qos_class_identifier, 0, &avpch1);
ogs_assert(ret == 0);
val.u32 = sess->pdn.qos.qci;
ret = fd_msg_avp_setvalue (avpch1, &val);
ogs_assert(ret == 0);
ret = fd_msg_avp_add (avp, MSG_BRW_LAST_CHILD, avpch1);
ogs_assert(ret == 0);
ret = fd_msg_avp_new(gx_allocation_retention_priority, 0, &avpch1);
ogs_assert(ret == 0);
ret = fd_msg_avp_new(gx_priority_level, 0, &avpch2);
ogs_assert(ret == 0);
val.u32 = sess->pdn.qos.arp.priority_level;
ret = fd_msg_avp_setvalue (avpch2, &val);
ogs_assert(ret == 0);
ret = fd_msg_avp_add (avpch1, MSG_BRW_LAST_CHILD, avpch2);
ogs_assert(ret == 0);
ret = fd_msg_avp_new(gx_pre_emption_capability, 0, &avpch2);
ogs_assert(ret == 0);
val.u32 = sess->pdn.qos.arp.pre_emption_capability;
ret = fd_msg_avp_setvalue (avpch2, &val);
ogs_assert(ret == 0);
ret = fd_msg_avp_add (avpch1, MSG_BRW_LAST_CHILD, avpch2);
ogs_assert(ret == 0);
ret = fd_msg_avp_new(gx_pre_emption_vulnerability, 0, &avpch2);
ogs_assert(ret == 0);
val.u32 = sess->pdn.qos.arp.pre_emption_vulnerability;
ret = fd_msg_avp_setvalue (avpch2, &val);
ogs_assert(ret == 0);
ret = fd_msg_avp_add (avpch1, MSG_BRW_LAST_CHILD, avpch2);
ogs_assert(ret == 0);
ret = fd_msg_avp_add (avp, MSG_BRW_LAST_CHILD, avpch1);
ogs_assert(ret == 0);
ret = fd_msg_avp_add(req, MSG_BRW_LAST_CHILD, avp);
ogs_assert(ret == 0);
/* Set 3GPP-User-Location-Info */
{
struct gx_uli_t {
uint8_t type;
tai_t tai;
e_cgi_t e_cgi;
} gx_uli;
memset(&gx_uli, 0, sizeof(gx_uli));
gx_uli.type = GX_3GPP_USER_LOCATION_INFO_TYPE_TAI_AND_ECGI;
memcpy(&gx_uli.tai.plmn_id, &sess->tai.plmn_id,
sizeof(sess->tai.plmn_id));
gx_uli.tai.tac = htons(sess->tai.tac);
memcpy(&gx_uli.e_cgi.plmn_id, &sess->e_cgi.plmn_id,
sizeof(sess->e_cgi.plmn_id));
gx_uli.e_cgi.cell_id = htonl(sess->e_cgi.cell_id);
ret = fd_msg_avp_new(gx_3gpp_user_location_info, 0, &avp);
ogs_assert(ret == 0);
val.os.data = (uint8_t*)&gx_uli;
val.os.len = sizeof(gx_uli);
ret = fd_msg_avp_setvalue(avp, &val);
ogs_assert(ret == 0);
ret = fd_msg_avp_add(req, MSG_BRW_LAST_CHILD, avp);
ogs_assert(ret == 0);
}
/* Set 3GPP-MS-Timezone */
if (message->create_session_request.ue_time_zone.presence) {
ret = fd_msg_avp_new(gx_3gpp_ms_timezone, 0, &avp);
ogs_assert(ret == 0);
val.os.data = message->create_session_request.ue_time_zone.data;
val.os.len = message->create_session_request.ue_time_zone.len;
ret = fd_msg_avp_setvalue(avp, &val);
ogs_assert(ret == 0);
ret = fd_msg_avp_add(req, MSG_BRW_LAST_CHILD, avp);
ogs_assert(ret == 0);
}
}
/* Set Called-Station-Id */
ret = fd_msg_avp_new(gx_called_station_id, 0, &avp);
ogs_assert(ret == 0);
val.os.data = (uint8_t*)sess->pdn.apn;
val.os.len = strlen(sess->pdn.apn);
ret = fd_msg_avp_setvalue(avp, &val);
ogs_assert(ret == 0);
ret = fd_msg_avp_add(req, MSG_BRW_LAST_CHILD, avp);
ogs_assert(ret == 0);
ret = clock_gettime(CLOCK_REALTIME, &sess_data->ts);
/* Keep a pointer to the session data for debug purpose,
* in real life we would not need it */
svg = sess_data;
/* Store this value in the session */
ret = fd_sess_state_store(pgw_gx_reg, session, &sess_data);
ogs_assert(ret == 0);
ogs_assert(sess_data == NULL);
/* Send the request */
ret = fd_msg_send(&req, pgw_gx_cca_cb, svg);
ogs_assert(ret == 0);
/* Increment the counter */
ogs_assert(pthread_mutex_lock(&fd_logger_self()->stats_lock) == 0);
fd_logger_self()->stats.nb_sent++;
ogs_assert(pthread_mutex_unlock(&fd_logger_self()->stats_lock) == 0);
}
static void pgw_gx_cca_cb(void *data, struct msg **msg)
{
int rv;
int ret;
struct sess_state *sess_data = NULL;
struct timespec ts;
struct session *session;
struct avp *avp, *avpch1, *avpch2;
struct avp_hdr *hdr;
unsigned long dur;
int error = 0;
int new;
pgw_event_t *e = NULL;
gtp_xact_t *xact = NULL;
pgw_sess_t *sess = NULL;
ogs_pkbuf_t *gxbuf = NULL, *gtpbuf = NULL;
gx_message_t *gx_message = NULL;
uint16_t gxbuf_len = 0;
ogs_debug("[PGW] Credit-Control-Answer");
ret = clock_gettime(CLOCK_REALTIME, &ts);
ogs_assert(ret == 0);
/* Search the session, retrieve its data */
ret = fd_msg_sess_get(fd_g_config->cnf_dict, *msg, &session, &new);
ogs_assert(ret == 0);
ogs_assert(new == 0);
ret = fd_sess_state_retrieve(pgw_gx_reg, session, &sess_data);
ogs_assert(ret == 0);
ogs_assert(sess_data);
ogs_assert((void *)sess_data == data);
xact = sess_data->xact;
ogs_assert(xact);
sess = sess_data->sess;
ogs_assert(sess);
gtpbuf = sess_data->gtpbuf;
ogs_assert(gtpbuf);
gxbuf_len = sizeof(gx_message_t);
ogs_assert(gxbuf_len < 8192);
gxbuf = ogs_pkbuf_alloc(NULL, gxbuf_len);
ogs_pkbuf_put(gxbuf, gxbuf_len);
gx_message = gxbuf->data;
ogs_assert(gx_message);
/* Set Credit Control Command */
memset(gx_message, 0, gxbuf_len);
gx_message->cmd_code = GX_CMD_CODE_CREDIT_CONTROL;
/* Value of Result Code */
ret = fd_msg_search_avp(*msg, fd_result_code, &avp);
ogs_assert(ret == 0);
if (avp) {
ret = fd_msg_avp_hdr(avp, &hdr);
ogs_assert(ret == 0);
gx_message->result_code = hdr->avp_value->i32;
ogs_debug(" Result Code: %d", hdr->avp_value->i32);
} else {
ret = fd_msg_search_avp(*msg, fd_experimental_result, &avp);
ogs_assert(ret == 0);
if (avp) {
ret = fd_avp_search_avp(avp, fd_experimental_result_code, &avpch1);
ogs_assert(ret == 0);
if (avpch1) {
ret = fd_msg_avp_hdr(avpch1, &hdr);
ogs_assert(ret == 0);
gx_message->result_code = hdr->avp_value->i32;
ogs_debug(" Experimental Result Code: %d",
gx_message->result_code);
}
} else {
ogs_error("no Result-Code");
error++;
}
}
/* Value of Origin-Host */
ret = fd_msg_search_avp(*msg, fd_origin_host, &avp);
ogs_assert(ret == 0);
if (avp) {
ret = fd_msg_avp_hdr(avp, &hdr);
ogs_assert(ret == 0);
ogs_debug(" From '%.*s'",
(int)hdr->avp_value->os.len, hdr->avp_value->os.data);
} else {
ogs_error("no_Origin-Host");
error++;
}
/* Value of Origin-Realm */
ret = fd_msg_search_avp(*msg, fd_origin_realm, &avp);
ogs_assert(ret == 0);
if (avp) {
ret = fd_msg_avp_hdr(avp, &hdr);
ogs_assert(ret == 0);
ogs_debug(" ('%.*s')",
(int)hdr->avp_value->os.len, hdr->avp_value->os.data);
} else {
ogs_error("no_Origin-Realm");
error++;
}
if (gx_message->result_code != ER_DIAMETER_SUCCESS) {
ogs_warn("ERROR DIAMETER Result Code(%d)", gx_message->result_code);
goto out;
}
ret = fd_msg_search_avp(*msg, gx_cc_request_type, &avp);
ogs_assert(ret == 0);
if (avp) {
ret = fd_msg_avp_hdr(avp, &hdr);
ogs_assert(ret == 0);
gx_message->cc_request_type = hdr->avp_value->i32;
} else {
ogs_error("no_CC-Request-Type");
error++;
}
ret = fd_msg_search_avp(*msg, gx_qos_information, &avp);
ogs_assert(ret == 0);
if (avp) {
ret = fd_avp_search_avp(avp, gx_apn_aggregate_max_bitrate_ul, &avpch1);
ogs_assert(ret == 0);
if (avpch1) {
ret = fd_msg_avp_hdr(avpch1, &hdr);
ogs_assert(ret == 0);
gx_message->pdn.ambr.uplink = hdr->avp_value->u32;
}
ret = fd_avp_search_avp(avp, gx_apn_aggregate_max_bitrate_dl, &avpch1);
ogs_assert(ret == 0);
if (avpch1) {
ret = fd_msg_avp_hdr(avpch1, &hdr);
ogs_assert(ret == 0);
gx_message->pdn.ambr.downlink = hdr->avp_value->u32;
}
}
ret = fd_msg_search_avp(*msg, gx_default_eps_bearer_qos, &avp);
ogs_assert(ret == 0);
if (avp) {
ret = fd_avp_search_avp(avp, gx_qos_class_identifier, &avpch1);
ogs_assert(ret == 0);
if (avpch1) {
ret = fd_msg_avp_hdr(avpch1, &hdr);
ogs_assert(ret == 0);
gx_message->pdn.qos.qci = hdr->avp_value->u32;
}
ret = fd_avp_search_avp(avp, gx_allocation_retention_priority, &avpch1);
ogs_assert(ret == 0);
if (avpch1) {
ret = fd_avp_search_avp(avpch1, gx_priority_level, &avpch2);
ogs_assert(ret == 0);
if (avpch2) {
ret = fd_msg_avp_hdr(avpch2, &hdr);
ogs_assert(ret == 0);
gx_message->pdn.qos.arp.priority_level = hdr->avp_value->u32;
}
ret = fd_avp_search_avp(avpch1, gx_pre_emption_capability, &avpch2);
ogs_assert(ret == 0);
if (avpch2) {
ret = fd_msg_avp_hdr(avpch2, &hdr);
ogs_assert(ret == 0);
gx_message->pdn.qos.arp.pre_emption_capability =
hdr->avp_value->u32;
}
ret = fd_avp_search_avp(avpch1,
gx_pre_emption_vulnerability, &avpch2);
ogs_assert(ret == 0);
if (avpch2) {
ret = fd_msg_avp_hdr(avpch2, &hdr);
ogs_assert(ret == 0);
gx_message->pdn.qos.arp.pre_emption_vulnerability =
hdr->avp_value->u32;
}
}
}
ret = fd_msg_browse(*msg, MSG_BRW_FIRST_CHILD, &avp, NULL);
ogs_assert(ret == 0);
while (avp) {
ret = fd_msg_avp_hdr(avp, &hdr);
ogs_assert(ret == 0);
switch (hdr->avp_code) {
case AC_SESSION_ID:
case AC_ORIGIN_HOST:
case AC_ORIGIN_REALM:
case AC_DESTINATION_REALM:
case AC_RESULT_CODE:
case AC_ROUTE_RECORD:
case AC_PROXY_INFO:
case AC_AUTH_APPLICATION_ID:
break;
case GX_AVP_CODE_CC_REQUEST_TYPE:
case GX_AVP_CODE_CC_REQUEST_NUMBER:
case GX_AVP_CODE_SUPPORTED_FEATURES:
break;
case GX_AVP_CODE_QOS_INFORMATION:
case GX_AVP_CODE_DEFAULT_EPS_BEARER_QOS:
break;
case GX_AVP_CODE_CHARGING_RULE_INSTALL:
ret = fd_msg_browse(avp, MSG_BRW_FIRST_CHILD, &avpch1, NULL);
ogs_assert(ret == 0);
while (avpch1) {
ret = fd_msg_avp_hdr(avpch1, &hdr);
ogs_assert(ret == 0);
switch (hdr->avp_code) {
case GX_AVP_CODE_CHARGING_RULE_DEFINITION:
{
pcc_rule_t *pcc_rule = &gx_message->pcc_rule
[gx_message->num_of_pcc_rule];
rv = decode_pcc_rule_definition(
pcc_rule, avpch1, &error);
ogs_assert(rv == OGS_OK);
pcc_rule->type = PCC_RULE_TYPE_INSTALL;
gx_message->num_of_pcc_rule++;
break;
}
default:
ogs_error("Not supported(%d)", hdr->avp_code);
break;
}
fd_msg_browse(avpch1, MSG_BRW_NEXT, &avpch1, NULL);
}
break;
default:
ogs_warn("Not supported(%d)", hdr->avp_code);
break;
}
fd_msg_browse(avp, MSG_BRW_NEXT, &avp, NULL);
}
out:
if (!error) {
e = pgw_event_new(PGW_EVT_GX_MESSAGE);
ogs_assert(e);
e->sess_index = sess->index;
e->gxbuf = gxbuf;
e->xact_index = xact->index;
e->gtpbuf = gtpbuf;
rv = ogs_queue_push(pgw_self()->queue, e);
if (rv != OGS_OK) {
ogs_error("ogs_queue_push() failed:%d", (int)rv);
gx_message_free(gx_message);
ogs_pkbuf_free(e->gxbuf);
ogs_pkbuf_free(e->gtpbuf);
pgw_event_free(e);
} else {
ogs_pollset_notify(pgw_self()->pollset);
}
} else {
gx_message_free(gx_message);
ogs_pkbuf_free(gxbuf);
ogs_pkbuf_free(gtpbuf);
}
/* Free the message */
ogs_assert(pthread_mutex_lock(&fd_logger_self()->stats_lock) == 0);
dur = ((ts.tv_sec - sess_data->ts.tv_sec) * 1000000) +
((ts.tv_nsec - sess_data->ts.tv_nsec) / 1000);
if (fd_logger_self()->stats.nb_recv) {
/* Ponderate in the avg */
fd_logger_self()->stats.avg = (fd_logger_self()->stats.avg *
fd_logger_self()->stats.nb_recv + dur) /
(fd_logger_self()->stats.nb_recv + 1);
/* Min, max */
if (dur < fd_logger_self()->stats.shortest)
fd_logger_self()->stats.shortest = dur;
if (dur > fd_logger_self()->stats.longest)
fd_logger_self()->stats.longest = dur;
} else {
fd_logger_self()->stats.shortest = dur;
fd_logger_self()->stats.longest = dur;
fd_logger_self()->stats.avg = dur;
}
if (error)
fd_logger_self()->stats.nb_errs++;
else
fd_logger_self()->stats.nb_recv++;
ogs_assert(pthread_mutex_unlock(&fd_logger_self()->stats_lock) == 0);
/* Display how long it took */
if (ts.tv_nsec > sess_data->ts.tv_nsec)
ogs_trace("in %d.%06ld sec",
(int)(ts.tv_sec - sess_data->ts.tv_sec),
(long)(ts.tv_nsec - sess_data->ts.tv_nsec) / 1000);
else
ogs_trace("in %d.%06ld sec",
(int)(ts.tv_sec + 1 - sess_data->ts.tv_sec),
(long)(1000000000 + ts.tv_nsec - sess_data->ts.tv_nsec) / 1000);
if (sess_data->cc_request_type != GX_CC_REQUEST_TYPE_TERMINATION_REQUEST) {
ret = fd_sess_state_store(pgw_gx_reg, session, &sess_data);
ogs_assert(ret == 0);
ogs_assert(sess_data == NULL);
} else {
state_cleanup(sess_data, NULL, NULL);
}
ret = fd_msg_free(*msg);
ogs_assert(ret == 0);
*msg = NULL;
return;
}
static int pgw_gx_fb_cb(struct msg **msg, struct avp *avp,
struct session *sess, void *opaque, enum disp_action *act)
{
/* This CB should never be called */
ogs_warn("Unexpected message received!");
return ENOTSUP;
}
static int pgw_gx_rar_cb( struct msg **msg, struct avp *avp,
struct session *session, void *opaque, enum disp_action *act)
{
int rv;
int ret;
struct msg *ans, *qry;
struct avp *avpch1;
struct avp_hdr *hdr;
union avp_value val;
struct sess_state *sess_data = NULL;
pgw_event_t *e = NULL;
uint16_t gxbuf_len = 0;
ogs_pkbuf_t *gxbuf = NULL;
pgw_sess_t *sess = NULL;
gx_message_t *gx_message = NULL;
uint32_t result_code = FD_DIAMETER_UNKNOWN_SESSION_ID;
ogs_assert(msg);
ogs_debug("[PGW] Re-Auth-Request");
gxbuf_len = sizeof(gx_message_t);
ogs_assert(gxbuf_len < 8192);
gxbuf = ogs_pkbuf_alloc(NULL, gxbuf_len);
ogs_pkbuf_put(gxbuf, gxbuf_len);
gx_message = gxbuf->data;
ogs_assert(gx_message);
/* Set Credit Control Command */
memset(gx_message, 0, gxbuf_len);
gx_message->cmd_code = GX_CMD_RE_AUTH;
/* Create answer header */
qry = *msg;
ret = fd_msg_new_answer_from_req(fd_g_config->cnf_dict, msg, 0);
ogs_assert(ret == 0);
ans = *msg;
ret = fd_sess_state_retrieve(pgw_gx_reg, session, &sess_data);
ogs_assert(ret == 0);
if (!sess_data) {
ogs_error("No Session Data");
goto out;
}
/* Get Session Information */
sess = sess_data->sess;
ogs_assert(sess);
ret = fd_msg_browse(qry, MSG_BRW_FIRST_CHILD, &avp, NULL);
ogs_assert(ret == 0);
while (avp) {
ret = fd_msg_avp_hdr(avp, &hdr);
ogs_assert(ret == 0);
switch(hdr->avp_code) {
case AC_SESSION_ID:
case AC_ORIGIN_HOST:
case AC_ORIGIN_REALM:
case AC_DESTINATION_REALM:
case AC_DESTINATION_HOST:
case AC_ROUTE_RECORD:
case AC_PROXY_INFO:
case AC_AUTH_APPLICATION_ID:
break;
case GX_AVP_CODE_RE_AUTH_REQUEST_TYPE:
break;
case GX_AVP_CODE_CHARGING_RULE_INSTALL:
{
ret = fd_msg_browse(avp, MSG_BRW_FIRST_CHILD, &avpch1, NULL);
ogs_assert(ret == 0);
while(avpch1)
{
ret = fd_msg_avp_hdr(avpch1, &hdr);
ogs_assert(ret == 0);
switch(hdr->avp_code)
{
case GX_AVP_CODE_CHARGING_RULE_DEFINITION:
{
pcc_rule_t *pcc_rule = &gx_message->pcc_rule
[gx_message->num_of_pcc_rule];
rv = decode_pcc_rule_definition(
pcc_rule, avpch1, NULL);
ogs_assert(rv == OGS_OK);
pcc_rule->type = PCC_RULE_TYPE_INSTALL;
gx_message->num_of_pcc_rule++;
break;
}
default:
{
ogs_error("Not supported(%d)", hdr->avp_code);
break;
}
}
fd_msg_browse(avpch1, MSG_BRW_NEXT, &avpch1, NULL);
}
break;
}
case GX_AVP_CODE_CHARGING_RULE_REMOVE:
{
ret = fd_msg_browse(avp, MSG_BRW_FIRST_CHILD, &avpch1, NULL);
ogs_assert(ret == 0);
while (avpch1) {
ret = fd_msg_avp_hdr(avpch1, &hdr);
ogs_assert(ret == 0);
switch (hdr->avp_code) {
case GX_AVP_CODE_CHARGING_RULE_NAME:
{
pcc_rule_t *pcc_rule = &gx_message->pcc_rule
[gx_message->num_of_pcc_rule];
pcc_rule->name =
ogs_strdup((char*)hdr->avp_value->os.data);
ogs_assert(pcc_rule->name);
pcc_rule->type = PCC_RULE_TYPE_REMOVE;
gx_message->num_of_pcc_rule++;
break;
}
default:
ogs_error("Not supported(%d)", hdr->avp_code);
break;
}
fd_msg_browse(avpch1, MSG_BRW_NEXT, &avpch1, NULL);
}
break;
}
default:
ogs_warn("Not supported(%d)", hdr->avp_code);
break;
}
fd_msg_browse(avp, MSG_BRW_NEXT, &avp, NULL);
}
/* Send Gx Event to PGW State Machine */
e = pgw_event_new(PGW_EVT_GX_MESSAGE);
ogs_assert(e);
e->sess_index = sess->index;
e->gxbuf = gxbuf;
rv = ogs_queue_push(pgw_self()->queue, e);
if (rv != OGS_OK) {
ogs_error("ogs_queue_push() failed:%d", (int)rv);
gx_message_free(gx_message);
ogs_pkbuf_free(e->gxbuf);
pgw_event_free(e);
} else {
ogs_pollset_notify(pgw_self()->pollset);
}
/* Set the Auth-Application-Id AVP */
ret = fd_msg_avp_new(fd_auth_application_id, 0, &avp);
ogs_assert(ret == 0);
val.i32 = GX_APPLICATION_ID;
ret = fd_msg_avp_setvalue(avp, &val);
ogs_assert(ret == 0);
ret = fd_msg_avp_add(ans, MSG_BRW_LAST_CHILD, avp);
ogs_assert(ret == 0);
/* Set the Origin-Host, Origin-Realm, andResult-Code AVPs */
ret = fd_msg_rescode_set(ans, "DIAMETER_SUCCESS", NULL, NULL, 1);
ogs_assert(ret == 0);
/* Store this value in the session */
ret = fd_sess_state_store(pgw_gx_reg, session, &sess_data);
ogs_assert(ret == 0);
ogs_assert(sess_data == NULL);
/* Send the answer */
ret = fd_msg_send(msg, NULL, NULL);
ogs_assert(ret == 0);
ogs_debug("[PGW] Re-Auth-Answer");
/* Add this value to the stats */
ogs_assert(pthread_mutex_lock(&fd_logger_self()->stats_lock) == 0);
fd_logger_self()->stats.nb_echoed++;
ogs_assert(pthread_mutex_unlock(&fd_logger_self()->stats_lock) == 0);
return 0;
out:
if (result_code == FD_DIAMETER_UNKNOWN_SESSION_ID) {
ret = fd_msg_rescode_set(ans,
"DIAMETER_UNKNOWN_SESSION_ID", NULL, NULL, 1);
ogs_assert(ret == 0);
} else {
ret = fd_message_experimental_rescode_set(ans, result_code);
ogs_assert(ret == 0);
}
/* Store this value in the session */
ret = fd_sess_state_store(pgw_gx_reg, session, &sess_data);
ogs_assert(ret == 0);
ogs_assert(sess_data == NULL);
ret = fd_msg_send(msg, NULL, NULL);
ogs_assert(ret == 0);
gx_message_free(gx_message);
ogs_pkbuf_free(gxbuf);
return 0;
}
int pgw_fd_init(void)
{
int ret;
struct disp_when data;
ret = fd_init(FD_MODE_CLIENT|FD_MODE_SERVER,
pgw_self()->fd_conf_path, pgw_self()->fd_config);
ogs_assert(ret == 0);
/* Install objects definitions for this application */
ret = gx_dict_init();
ogs_assert(ret == 0);
/* Create handler for sessions */
ret = fd_sess_handler_create(&pgw_gx_reg, state_cleanup, NULL, NULL);
ogs_assert(ret == 0);
memset(&data, 0, sizeof(data));
data.app = gx_application;
ret = fd_disp_register(pgw_gx_fb_cb, DISP_HOW_APPID, &data, NULL,
&hdl_gx_fb);
ogs_assert(ret == 0);
data.command = gx_cmd_rar;
ret = fd_disp_register(pgw_gx_rar_cb, DISP_HOW_CC, &data, NULL,
&hdl_gx_rar);
ogs_assert(ret == 0);
/* Advertise the support for the application in the peer */
ret = fd_disp_app_support(gx_application, fd_vendor, 1, 0);
return OGS_OK;
}
void pgw_fd_final(void)
{
int ret;
ret = fd_sess_handler_destroy(&pgw_gx_reg, NULL);
ogs_assert(ret == 0);
if (hdl_gx_fb)
(void) fd_disp_unregister(&hdl_gx_fb, NULL);
if (hdl_gx_rar)
(void) fd_disp_unregister(&hdl_gx_rar, NULL);
fd_final();
}
static int decode_pcc_rule_definition(
pcc_rule_t *pcc_rule, struct avp *avpch1, int *perror)
{
int ret = 0, error = 0;
struct avp *avpch2, *avpch3, *avpch4;
struct avp_hdr *hdr;
ogs_assert(pcc_rule);
ogs_assert(avpch1);
ret = fd_msg_browse(avpch1, MSG_BRW_FIRST_CHILD, &avpch2, NULL);
ogs_assert(ret == 0);
while (avpch2) {
ret = fd_msg_avp_hdr(avpch2, &hdr);
ogs_assert(ret == 0);
switch (hdr->avp_code) {
case GX_AVP_CODE_CHARGING_RULE_NAME:
if (pcc_rule->name)
{
ogs_error("PCC Rule Name has already been defined");
ogs_free(pcc_rule->name);
}
pcc_rule->name = ogs_strdup((char*)hdr->avp_value->os.data);
ogs_assert(pcc_rule->name);
break;
case GX_AVP_CODE_FLOW_INFORMATION:
{
flow_t *flow =
&pcc_rule->flow[pcc_rule->num_of_flow];
ret = fd_avp_search_avp(avpch2, gx_flow_direction, &avpch3);
ogs_assert(ret == 0);
if (avpch3)
{
ret = fd_msg_avp_hdr( avpch3, &hdr);
ogs_assert(ret == 0);
flow->direction = hdr->avp_value->i32;
}
ret = fd_avp_search_avp(avpch2, gx_flow_description, &avpch3);
ogs_assert(ret == 0);
if (avpch3)
{
ret = fd_msg_avp_hdr(avpch3, &hdr);
ogs_assert(ret == 0);
flow->description = ogs_malloc(hdr->avp_value->os.len+1);
ogs_cpystrn(flow->description,
(char*)hdr->avp_value->os.data,
hdr->avp_value->os.len+1);
}
pcc_rule->num_of_flow++;
break;
}
case GX_AVP_CODE_FLOW_STATUS:
pcc_rule->flow_status = hdr->avp_value->i32;
break;
case GX_AVP_CODE_QOS_INFORMATION:
ret = fd_avp_search_avp(avpch2,
gx_qos_class_identifier, &avpch3);
ogs_assert(ret == 0);
if (avpch3) {
ret = fd_msg_avp_hdr(avpch3, &hdr);
ogs_assert(ret == 0);
pcc_rule->qos.qci = hdr->avp_value->u32;
} else {
ogs_error("no_QCI");
error++;
}
ret = fd_avp_search_avp(avpch2,
gx_allocation_retention_priority, &avpch3);
ogs_assert(ret == 0);
if (avpch3) {
ret = fd_avp_search_avp(avpch3, gx_priority_level, &avpch4);
ogs_assert(ret == 0);
if (avpch4) {
ret = fd_msg_avp_hdr(avpch4, &hdr);
ogs_assert(ret == 0);
pcc_rule->qos.arp.priority_level = hdr->avp_value->u32;
} else {
ogs_error("no_Priority-Level");
error++;
}
ret = fd_avp_search_avp(avpch3,
gx_pre_emption_capability, &avpch4);
ogs_assert(ret == 0);
if (avpch4) {
ret = fd_msg_avp_hdr(avpch4, &hdr);
ogs_assert(ret == 0);
pcc_rule->qos.arp.pre_emption_capability =
hdr->avp_value->u32;
} else {
ogs_error("no_Preemption-Capability");
error++;
}
ret = fd_avp_search_avp(avpch3,
gx_pre_emption_vulnerability, &avpch4);
ogs_assert(ret == 0);
if (avpch4) {
ret = fd_msg_avp_hdr(avpch4, &hdr);
ogs_assert(ret == 0);
pcc_rule->qos.arp.pre_emption_vulnerability =
hdr->avp_value->u32;
} else {
ogs_error("no_Preemption-Vulnerability");
error++;
}
} else {
ogs_error("no_ARP");
error++;
}
ret = fd_avp_search_avp(avpch2,
gx_max_requested_bandwidth_ul, &avpch3);
ogs_assert(ret == 0);
if (avpch3) {
ret = fd_msg_avp_hdr(avpch3, &hdr);
ogs_assert(ret == 0);
pcc_rule->qos.mbr.uplink = hdr->avp_value->u32;
}
ret = fd_avp_search_avp(avpch2,
gx_max_requested_bandwidth_dl, &avpch3);
ogs_assert(ret == 0);
if (avpch3) {
ret = fd_msg_avp_hdr(avpch3, &hdr);
ogs_assert(ret == 0);
pcc_rule->qos.mbr.downlink = hdr->avp_value->u32;
}
ret = fd_avp_search_avp(avpch2,
gx_guaranteed_bitrate_ul, &avpch3);
ogs_assert(ret == 0);
if (avpch3) {
ret = fd_msg_avp_hdr(avpch3, &hdr);
ogs_assert(ret == 0);
pcc_rule->qos.gbr.uplink = hdr->avp_value->u32;
}
ret = fd_avp_search_avp(avpch2,
gx_guaranteed_bitrate_dl, &avpch3);
ogs_assert(ret == 0);
if (avpch3) {
ret = fd_msg_avp_hdr(avpch3, &hdr);
ogs_assert(ret == 0);
pcc_rule->qos.gbr.downlink = hdr->avp_value->u32;
}
break;
case GX_AVP_CODE_PRECEDENCE:
pcc_rule->precedence = hdr->avp_value->i32;
break;
default:
ogs_error("Not implemented(%d)", hdr->avp_code);
break;
}
fd_msg_browse(avpch2, MSG_BRW_NEXT, &avpch2, NULL);
}
if (perror)
*perror = error;
return OGS_OK;
}
Reference in New Issue View Git Blame Copy Permalink