#define TRACE_MODULE _pcrf_gx_path #include "core_lib.h" #include "core_debug.h" #include "core_pool.h" #include "core_pkbuf.h" #include "fd/fd_lib.h" #include "fd/gx/gx_dict.h" #include "fd/gx/gx_message.h" #include "fd/rx/rx_message.h" #include "pcrf_context.h" #include "pcrf_fd_path.h" struct rx_sess_state { os0_t sid; /* Rx Session-Id */ c_int8_t *name[MAX_NUM_OF_PCC_RULE]; /* PCC Rule Name */ int num_of_name; }; struct sess_state { os0_t sid; /* Gx Session-Id */ c_uint32_t cc_request_type; /* CC-Request-Type */ os0_t peer_host; /* Peer Host */ c_int8_t *imsi_bcd; c_int8_t *apn; ED3(c_uint8_t ipv4:1;, c_uint8_t ipv6:1;, c_uint8_t reserved:6;) c_uint32_t addr; /* Framed-IPv4-Address */ c_uint8_t addr6[IPV6_LEN]; /* Framed-IPv6-Prefix */ #define MAX_NUM_OF_RX_SESSION_STATE 16 struct rx_sess_state rx[MAX_NUM_OF_RX_SESSION_STATE]; int num_of_rx; struct timespec ts; /* Time of sending the message */ }; static struct session_handler *pcrf_gx_reg = NULL; static struct disp_hdl *hdl_gx_fb = NULL; static struct disp_hdl *hdl_gx_ccr = NULL; pool_declare(pcrf_gx_sess_pool, struct sess_state, MAX_POOL_OF_DIAMETER_SESS); static status_t encode_pcc_rule_definition( struct avp *avp, pcc_rule_t *pcc_rule); static void pcrf_gx_raa_cb(void *data, struct msg **msg); static __inline__ struct sess_state *new_state(os0_t sid) { struct sess_state *new = NULL; pool_alloc_node(&pcrf_gx_sess_pool, &new); d_assert(new, return NULL,); memset(new, 0, sizeof *new); new->sid = (os0_t)core_strdup((char *)sid); d_assert(new->sid, return NULL,); return new; } static void state_cleanup(struct sess_state *sess_data, os0_t sid, void *opaque) { int i, j; d_assert(sess_data, return,); if (sess_data->peer_host) CORE_FREE(sess_data->peer_host); if (sess_data->imsi_bcd) CORE_FREE(sess_data->imsi_bcd); if (sess_data->apn) CORE_FREE(sess_data->apn); if (sess_data->ipv4) pcrf_sess_set_ipv4(&sess_data->addr, NULL); if (sess_data->ipv6) pcrf_sess_set_ipv6(sess_data->addr6, NULL); if (sess_data->sid) CORE_FREE(sess_data->sid); for (i = 0; i < sess_data->num_of_rx; i++) { struct rx_sess_state *rx_sess_data = &sess_data->rx[i]; for (j = 0; j < rx_sess_data->num_of_name; j++) { d_assert(rx_sess_data->name[j],,); CORE_FREE(rx_sess_data->name[j]); } if (rx_sess_data->sid) CORE_FREE(rx_sess_data->sid); } pool_free_node(&pcrf_gx_sess_pool, sess_data); } static int pcrf_gx_fb_cb(struct msg **msg, struct avp *avp, struct session *sess, void *opaque, enum disp_action *act) { /* This CB should never be called */ d_warn("Unexpected message received!"); return ENOTSUP; } static int pcrf_gx_ccr_cb( struct msg **msg, struct avp *avp, struct session *sess, void *opaque, enum disp_action *act) { status_t rv; int ret = 0, i; struct msg *ans, *qry; struct avp *avpch1, *avpch2; struct avp_hdr *hdr; union avp_value val; struct sess_state *sess_data = NULL; gx_message_t gx_message; c_uint32_t cc_request_number = 0; c_uint32_t result_code = FD_DIAMETER_MISSING_AVP; d_assert(msg, return EINVAL,); ret = fd_sess_state_retrieve(pcrf_gx_reg, sess, &sess_data); d_assert(ret == 0, return EINVAL,); if (!sess_data) { os0_t sid; size_t sidlen; ret = fd_sess_getsid(sess, &sid, &sidlen); d_assert(ret == 0, return EINVAL,); sess_data = new_state(sid); d_assert(sess_data, return EINVAL,); } /* Initialize Message */ memset(&gx_message, 0, sizeof(gx_message_t)); /* Create answer header */ qry = *msg; ret = fd_msg_new_answer_from_req(fd_g_config->cnf_dict, msg, 0); d_assert(ret == 0, return EINVAL,); ans = *msg; /* Get Origin-Host */ ret = fd_msg_search_avp(qry, fd_origin_host, &avp); d_assert(ret == 0, return EINVAL,); if (avp) { ret = fd_msg_avp_hdr(avp, &hdr); d_assert(ret == 0, return EINVAL,); if (sess_data->peer_host) CORE_FREE(sess_data->peer_host); sess_data->peer_host = (os0_t)core_strdup((char *)hdr->avp_value->os.data); d_assert(sess_data->peer_host, return CORE_ERROR,); } else { d_error("no_CC-Request-Type "); result_code = FD_DIAMETER_MISSING_AVP; goto out; } /* Get CC-Request-Type */ ret = fd_msg_search_avp(qry, gx_cc_request_type, &avp); d_assert(ret == 0, return EINVAL,); if (avp) { ret = fd_msg_avp_hdr(avp, &hdr); d_assert(ret == 0, return EINVAL,); sess_data->cc_request_type = hdr->avp_value->i32; } else { d_error("no_CC-Request-Type "); result_code = FD_DIAMETER_MISSING_AVP; goto out; } /* Get CC-Request-Number */ ret = fd_msg_search_avp(qry, gx_cc_request_number, &avp); d_assert(ret == 0, return EINVAL,); if (avp) { ret = fd_msg_avp_hdr(avp, &hdr); d_assert(ret == 0, return EINVAL,); cc_request_number = hdr->avp_value->i32; } /* Get Framed-IP-Address */ ret = fd_msg_search_avp(qry, gx_framed_ip_address, &avp); d_assert(ret == 0, return EINVAL,); if (avp) { ret = fd_msg_avp_hdr(avp, &hdr); d_assert(ret == 0, return EINVAL,); memcpy(&sess_data->addr, hdr->avp_value->os.data, sizeof sess_data->addr); pcrf_sess_set_ipv4(&sess_data->addr, sess_data->sid); sess_data->ipv4 = 1; } /* Get Framed-IPv6-Prefix */ ret = fd_msg_search_avp(qry, gx_framed_ipv6_prefix, &avp); d_assert(ret == 0, return EINVAL,); if (avp) { paa_t *paa = NULL; ret = fd_msg_avp_hdr(avp, &hdr); d_assert(ret == 0, return EINVAL,); paa = (paa_t *)hdr->avp_value->os.data; d_assert(paa, return EINVAL,); d_assert(paa->len == IPV6_LEN * 8 /* 128bit */, return EINVAL, "Invalid Framed-IPv6-Prefix Length:%d", paa->len); memcpy(sess_data->addr6, paa->addr6, sizeof sess_data->addr6); pcrf_sess_set_ipv6(sess_data->addr6, sess_data->sid); sess_data->ipv6 = 1; } /* Set the Auth-Application-Id AVP */ ret = fd_msg_avp_new(fd_auth_application_id, 0, &avp); d_assert(ret == 0, return EINVAL,); val.i32 = GX_APPLICATION_ID; ret = fd_msg_avp_setvalue(avp, &val); d_assert(ret == 0, return EINVAL,); ret = fd_msg_avp_add(ans, MSG_BRW_LAST_CHILD, avp); d_assert(ret == 0, return EINVAL,); /* Set CC-Request-Type */ ret = fd_msg_avp_new(gx_cc_request_type, 0, &avp); d_assert(ret == 0, return EINVAL,); val.i32 = sess_data->cc_request_type; ret = fd_msg_avp_setvalue(avp, &val); d_assert(ret == 0, return EINVAL,); ret = fd_msg_avp_add(ans, MSG_BRW_LAST_CHILD, avp); d_assert(ret == 0, return EINVAL,); /* Set CC-Request-Number */ ret = fd_msg_avp_new(gx_cc_request_number, 0, &avp); d_assert(ret == 0, return EINVAL,); val.i32 = cc_request_number; ret = fd_msg_avp_setvalue(avp, &val); d_assert(ret == 0, return EINVAL,); ret = fd_msg_avp_add(ans, MSG_BRW_LAST_CHILD, avp); d_assert(ret == 0, return EINVAL,); /* Get IMSI + APN */ ret = fd_msg_search_avp(qry, gx_subscription_id, &avp); d_assert(ret == 0, return EINVAL,); if (avp) { ret = fd_msg_avp_hdr(avp, &hdr); d_assert(ret == 0, return EINVAL,); ret = fd_avp_search_avp(avp, gx_subscription_id_type, &avpch1); d_assert(ret == 0, return EINVAL,); if (avpch1) { ret = fd_msg_avp_hdr(avpch1, &hdr); d_assert(ret == 0, return EINVAL,); if (hdr->avp_value->i32 != GX_SUBSCRIPTION_ID_TYPE_END_USER_IMSI) { d_error("Not implemented Subscription-Id-Type(%d)", hdr->avp_value->i32); result_code = FD_DIAMETER_AVP_UNSUPPORTED; goto out; } } else { d_error("no_Subscription-Id-Type"); result_code = FD_DIAMETER_MISSING_AVP; goto out; } ret = fd_avp_search_avp(avp, gx_subscription_id_data, &avpch1); d_assert(ret == 0, return EINVAL,); if (avpch1) { ret = fd_msg_avp_hdr(avpch1, &hdr); d_assert(ret == 0, return EINVAL,); if (sess_data->imsi_bcd) CORE_FREE(sess_data->imsi_bcd); sess_data->imsi_bcd = core_strdup((char *)hdr->avp_value->os.data); d_assert(sess_data->imsi_bcd, return CORE_ERROR,); } else { d_error("no_Subscription-Id-Data"); result_code = FD_DIAMETER_MISSING_AVP; goto out; } } if (sess_data->imsi_bcd == NULL) { d_error("no_Subscription-Id"); result_code = FD_DIAMETER_MISSING_AVP; goto out; } ret = fd_msg_search_avp(qry, gx_called_station_id, &avp); d_assert(ret == 0, return EINVAL,); if (avp) { ret = fd_msg_avp_hdr(avp, &hdr); d_assert(ret == 0, return EINVAL,); if (sess_data->apn) CORE_FREE(sess_data->apn); sess_data->apn = core_strdup((char *)hdr->avp_value->os.data); d_assert(sess_data->apn, return CORE_ERROR,); } if (sess_data->apn == NULL) { d_error("no_Called-Station-Id"); result_code = FD_DIAMETER_MISSING_AVP; goto out; } /* Retrieve QoS Data from Database */ rv = pcrf_db_qos_data(sess_data->imsi_bcd, sess_data->apn, &gx_message); if (rv != CORE_OK) { d_error("Cannot get data for IMSI(%s)+APN(%s)'\n", sess_data->imsi_bcd, sess_data->apn); result_code = FD_DIAMETER_UNKNOWN_SESSION_ID; goto out; } if (sess_data->cc_request_type != GX_CC_REQUEST_TYPE_TERMINATION_REQUEST) { int charging_rule_install = 0; for (i = 0; i < gx_message.num_of_pcc_rule; i++) { pcc_rule_t *pcc_rule = &gx_message.pcc_rule[i]; if (pcc_rule->num_of_flow) { if (charging_rule_install == 0) { ret = fd_msg_avp_new(gx_charging_rule_install, 0, &avp); d_assert(ret == 0, return CORE_ERROR,); charging_rule_install = 1; } rv = encode_pcc_rule_definition(avp, pcc_rule); d_assert(rv == CORE_OK, return EINVAL,); } } if (charging_rule_install) { ret = fd_msg_avp_add(ans, MSG_BRW_LAST_CHILD, avp); d_assert(ret == 0, return CORE_ERROR,); } /* Set QoS-Information */ if (gx_message.pdn.ambr.downlink || gx_message.pdn.ambr.uplink) { ret = fd_msg_avp_new(gx_qos_information, 0, &avp); d_assert(ret == 0, return EINVAL,); if (gx_message.pdn.ambr.uplink) { ret = fd_msg_avp_new(gx_apn_aggregate_max_bitrate_ul, 0, &avpch1); d_assert(ret == 0, return EINVAL,); val.u32 = gx_message.pdn.ambr.uplink; ret = fd_msg_avp_setvalue (avpch1, &val); d_assert(ret == 0, return EINVAL,); ret = fd_msg_avp_add (avp, MSG_BRW_LAST_CHILD, avpch1); d_assert(ret == 0, return EINVAL,); } if (gx_message.pdn.ambr.downlink) { ret = fd_msg_avp_new(gx_apn_aggregate_max_bitrate_dl, 0, &avpch1); d_assert(ret == 0, return EINVAL,); val.u32 = gx_message.pdn.ambr.downlink; ret = fd_msg_avp_setvalue (avpch1, &val); d_assert(ret == 0, return EINVAL,); ret = fd_msg_avp_add (avp, MSG_BRW_LAST_CHILD, avpch1); d_assert(ret == 0, return EINVAL,); } ret = fd_msg_avp_add(ans, MSG_BRW_LAST_CHILD, avp); d_assert(ret == 0, return EINVAL,); } /* Set Default-EPS-Bearer-QoS */ ret = fd_msg_avp_new(gx_default_eps_bearer_qos, 0, &avp); d_assert(ret == 0, return EINVAL,); ret = fd_msg_avp_new(gx_qos_class_identifier, 0, &avpch1); d_assert(ret == 0, return EINVAL,); val.u32 = gx_message.pdn.qos.qci; ret = fd_msg_avp_setvalue (avpch1, &val); d_assert(ret == 0, return EINVAL,); ret = fd_msg_avp_add (avp, MSG_BRW_LAST_CHILD, avpch1); d_assert(ret == 0, return EINVAL,); ret = fd_msg_avp_new(gx_allocation_retention_priority, 0, &avpch1); d_assert(ret == 0, return EINVAL,); ret = fd_msg_avp_new(gx_priority_level, 0, &avpch2); d_assert(ret == 0, return EINVAL,); val.u32 = gx_message.pdn.qos.arp.priority_level; ret = fd_msg_avp_setvalue (avpch2, &val); d_assert(ret == 0, return EINVAL,); ret = fd_msg_avp_add (avpch1, MSG_BRW_LAST_CHILD, avpch2); d_assert(ret == 0, return EINVAL,); ret = fd_msg_avp_new(gx_pre_emption_capability, 0, &avpch2); d_assert(ret == 0, return EINVAL,); val.u32 = gx_message.pdn.qos.arp.pre_emption_capability; ret = fd_msg_avp_setvalue (avpch2, &val); d_assert(ret == 0, return EINVAL,); ret = fd_msg_avp_add (avpch1, MSG_BRW_LAST_CHILD, avpch2); d_assert(ret == 0, return EINVAL,); ret = fd_msg_avp_new(gx_pre_emption_vulnerability, 0, &avpch2); d_assert(ret == 0, return EINVAL,); val.u32 = gx_message.pdn.qos.arp.pre_emption_vulnerability; ret = fd_msg_avp_setvalue (avpch2, &val); d_assert(ret == 0, return EINVAL,); ret = fd_msg_avp_add (avpch1, MSG_BRW_LAST_CHILD, avpch2); d_assert(ret == 0, return EINVAL,); ret = fd_msg_avp_add (avp, MSG_BRW_LAST_CHILD, avpch1); d_assert(ret == 0, return EINVAL,); ret = fd_msg_avp_add(ans, MSG_BRW_LAST_CHILD, avp); d_assert(ret == 0, return EINVAL,); /* Set Supported Features */ ret = fd_msg_avp_new(gx_supported_features, 0, &avp); d_assert(ret == 0, return EINVAL,); ret = fd_msg_avp_new(gx_feature_list_id, 0, &avpch1); d_assert(ret == 0, return EINVAL,); val.i32 = 1; ret = fd_msg_avp_setvalue (avpch1, &val); d_assert(ret == 0, return EINVAL,); ret = fd_msg_avp_add (avp, MSG_BRW_LAST_CHILD, avpch1); d_assert(ret == 0, return EINVAL,); ret = fd_msg_avp_new(gx_feature_list, 0, &avpch1); d_assert(ret == 0, return EINVAL,); val.u32 = 0x0000000b; ret = fd_msg_avp_setvalue (avpch1, &val); d_assert(ret == 0, return EINVAL,); ret = fd_msg_avp_add (avp, MSG_BRW_LAST_CHILD, avpch1); d_assert(ret == 0, return EINVAL,); ret = fd_msg_avp_add(ans, MSG_BRW_LAST_CHILD, avp); d_assert(ret == 0, return EINVAL,); } /* Set the Origin-Host, Origin-Realm, andResult-Code AVPs */ ret = fd_msg_rescode_set(ans, "DIAMETER_SUCCESS", NULL, NULL, 1); d_assert(ret == 0, return EINVAL,); if (sess_data->cc_request_type != GX_CC_REQUEST_TYPE_TERMINATION_REQUEST) { /* Store this value in the session */ ret = fd_sess_state_store(pcrf_gx_reg, sess, &sess_data); d_assert(ret == 0,,); d_assert(sess_data == NULL,,); } else { state_cleanup(sess_data, NULL, NULL); } /* Send the answer */ ret = fd_msg_send(msg, NULL, NULL); d_assert(ret == 0,,); /* Add this value to the stats */ d_assert(pthread_mutex_lock(&fd_logger_self()->stats_lock) == 0,,); fd_logger_self()->stats.nb_echoed++; d_assert(pthread_mutex_unlock(&fd_logger_self()->stats_lock) ==0,, ); gx_message_free(&gx_message); return 0; out: /* Set the Result-Code */ if (result_code == FD_DIAMETER_AVP_UNSUPPORTED) { ret = fd_msg_rescode_set(ans, "DIAMETER_AVP_UNSUPPORTED", NULL, NULL, 1); d_assert(ret == 0, return EINVAL,); } else if (result_code == FD_DIAMETER_UNKNOWN_SESSION_ID) { ret = fd_msg_rescode_set(ans, "DIAMETER_UNKNOWN_SESSION_ID", NULL, NULL, 1); d_assert(ret == 0, return EINVAL,); } else if (result_code == FD_DIAMETER_MISSING_AVP) { ret = fd_msg_rescode_set(ans, "DIAMETER_MISSING_AVP", NULL, NULL, 1); d_assert(ret == 0, return EINVAL,); } else { ret = fd_message_experimental_rescode_set(ans, result_code); d_assert(ret == 0, return EINVAL,); } if (sess_data->cc_request_type != GX_CC_REQUEST_TYPE_TERMINATION_REQUEST) { /* Store this value in the session */ ret = fd_sess_state_store(pcrf_gx_reg, sess, &sess_data); d_assert(sess_data == NULL,,); } else { state_cleanup(sess_data, NULL, NULL); } ret = fd_msg_send(msg, NULL, NULL); d_assert(ret == 0,,); gx_message_free(&gx_message); return 0; } status_t pcrf_gx_send_rar( c_uint8_t *gx_sid, c_uint8_t *rx_sid, rx_message_t *rx_message) { status_t rv; int ret = 0, i, j, k; struct msg *req = NULL; struct avp *avp, *avpch1; union avp_value val; struct sess_state *sess_data = NULL, *svg; struct rx_sess_state *rx_sess_data = NULL; struct session *session = NULL; int new; size_t sidlen; gx_message_t gx_message; int charging_rule = 0; d_assert(gx_sid, return CORE_ERROR,); d_assert(rx_sid, return CORE_ERROR,); d_assert(rx_message, return CORE_ERROR,); /* Initialize Message */ memset(&gx_message, 0, sizeof(gx_message_t)); /* Set default error result code */ rx_message->result_code = FD_DIAMETER_UNKNOWN_SESSION_ID; /* Create the request */ ret = fd_msg_new(gx_cmd_rar, MSGFL_ALLOC_ETEID, &req); d_assert(ret == 0, return CORE_ERROR,); { struct msg_hdr * h; ret = fd_msg_hdr( req, &h ); d_assert(ret == 0, return CORE_ERROR,); h->msg_appl = GX_APPLICATION_ID; } /* Retrieve session by Session-Id */ sidlen = strlen((char *)gx_sid); ret = fd_sess_fromsid_msg((os0_t)gx_sid, sidlen, &session, &new); d_assert(ret == 0, return CORE_ERROR,); if (new) { d_error("No session data"); ret = fd_msg_free(req); d_assert(ret == 0,,); rx_message->result_code = FD_DIAMETER_UNKNOWN_PEER; return CORE_ERROR; } /* Add Session-Id to the message */ ret = fd_message_session_id_set(req, (os0_t)gx_sid, sidlen); d_assert(ret == 0, return CORE_ERROR,); /* Save the session associated with the message */ ret = fd_msg_sess_set(req, session); d_assert(ret == 0, return CORE_ERROR,); /* Retrieve session state in this session */ ret = fd_sess_state_retrieve(pcrf_gx_reg, session, &sess_data); d_assert(ret == 0, return CORE_ERROR,); if (sess_data == NULL) { d_error("No session data"); ret = fd_msg_free(req); d_assert(ret == 0,,); rx_message->result_code = FD_DIAMETER_UNKNOWN_SESSION_ID; return CORE_ERROR; } /* Find RX session state */ rx_sess_data = &sess_data->rx[sess_data->num_of_rx]; for (i = 0; i < sess_data->num_of_rx; i++) { if (strcmp((char *)sess_data->rx[i].sid, (char*)rx_sid) == 0) { rx_sess_data = &sess_data->rx[i]; break; } } if (rx_message->cmd_code == RX_CMD_CODE_AA) { if (!rx_sess_data->sid) { rx_sess_data->sid = (os0_t)core_strdup((char *)rx_sid); sess_data->num_of_rx++; } d_assert(rx_sess_data->sid, return CORE_ERROR,); /* Retrieve QoS Data from Database */ rv = pcrf_db_qos_data(sess_data->imsi_bcd, sess_data->apn, &gx_message); if (rv != CORE_OK) { d_error("Cannot get data for IMSI(%s)+APN(%s)'\n", sess_data->imsi_bcd, sess_data->apn); rx_message->result_code = RX_DIAMETER_IP_CAN_SESSION_NOT_AVAILABLE; goto out; } /* Match Media-Component with PCC Rule */ for (i = 0; i < rx_message->num_of_media_component; i++) { pcc_rule_t *pcc_rule = NULL; c_uint8_t qci = 0; rx_media_component_t *media_component = &rx_message->media_component[i]; if (media_component->media_component_number == 0) { continue; } switch(media_component->media_type) { case RX_MEDIA_TYPE_AUDIO: { qci = PDN_QCI_1; break; } default: { d_error("Not implemented : [Media-Type:%d]", media_component->media_type); rx_message->result_code = FD_DIAMETER_INVALID_AVP_VALUE; goto out; } } for (j = 0; j < gx_message.num_of_pcc_rule; j++) { pcc_rule = &gx_message.pcc_rule[j]; if (pcc_rule->qos.qci == qci) { break; } } if (!pcc_rule) { d_error("CHECK WEBUI : No PCC Rule [QCI:%d]", qci); d_error("Please add PCC Rule using WEBUI"); rx_message->result_code = RX_DIAMETER_REQUESTED_SERVICE_NOT_AUTHORIZED; goto out; } if (pcc_rule->num_of_flow) { d_warn("STATIC dedicated bearer has already been activated"); d_warn("EPC will initiate modify dedicated bearer"); for (j = 0; j < pcc_rule->num_of_flow; j++) { flow_t *flow = &pcc_rule->flow[j]; if (flow->description) { CORE_FREE(flow->description); } else d_assert(0,, "Null param"); } pcc_rule->num_of_flow = 0; } for (j = 0; j < media_component->num_of_sub; j++) { rx_media_sub_component_t *sub = &media_component->sub[j]; if (sub->flow_number == 0) { continue; } for (k = 0; k < sub->num_of_flow; k++) { int len; flow_t *gx_flow = &pcc_rule->flow[pcc_rule->num_of_flow]; flow_t *rx_flow = &sub->flow[k]; if (!strncmp(rx_flow->description, "permit out", strlen("permit out"))) { gx_flow->direction = FLOW_DOWNLINK_ONLY; len = strlen(rx_flow->description)+1; gx_flow->description = core_malloc(len); core_cpystrn(gx_flow->description, rx_flow->description, len); } else if (!strncmp(rx_flow->description, "permit in", strlen("permit in"))) { gx_flow->direction = FLOW_UPLINK_ONLY; /* 'permit in' should be changed * 'permit out' in Gx Diameter */ len = strlen(rx_flow->description)+2; gx_flow->description = core_malloc(len); strcpy(gx_flow->description, "permit out"); strcat(gx_flow->description, &rx_flow->description[strlen("permit in")]); d_assert(len == strlen(gx_flow->description)+1, goto out,); } else { d_error("Invalid Flow Descripton : [%s]", rx_flow->description); rx_message->result_code = RX_DIAMETER_FILTER_RESTRICTIONS; goto out; } pcc_rule->num_of_flow++; } } if (charging_rule == 0) { ret = fd_msg_avp_new(gx_charging_rule_install, 0, &avp); d_assert(ret == 0, return CORE_ERROR,); charging_rule = 1; } rv = encode_pcc_rule_definition(avp, pcc_rule); d_assert(rv == CORE_OK, return CORE_ERROR,); /* Store PCC Rule Name in Rx session state */ rx_sess_data->name[rx_sess_data->num_of_name] = core_strdup(pcc_rule->name); d_assert(rx_sess_data->name[rx_sess_data->num_of_name], return CORE_ERROR,); rx_sess_data->num_of_name++; } if (charging_rule == 1) { ret = fd_msg_avp_add(req, MSG_BRW_LAST_CHILD, avp); d_assert(ret == 0, return CORE_ERROR,); } } else if (rx_message->cmd_code == RX_CMD_CODE_SESSION_TERMINATION) { d_assert(rx_sess_data->sid, return CORE_ERROR,); for (i = 0; i < rx_sess_data->num_of_name; i++) { d_assert(rx_sess_data->name[i],,); if (charging_rule == 0) { ret = fd_msg_avp_new(gx_charging_rule_remove, 0, &avp); d_assert(ret == 0, return CORE_ERROR,); charging_rule = 1; } ret = fd_msg_avp_new(gx_charging_rule_name, 0, &avpch1); d_assert(ret == 0, return CORE_ERROR,); val.os.data = (c_uint8_t *)rx_sess_data->name[i]; val.os.len = strlen(rx_sess_data->name[i]); ret = fd_msg_avp_setvalue(avpch1, &val); d_assert(ret == 0, return CORE_ERROR,); ret = fd_msg_avp_add(avp, MSG_BRW_LAST_CHILD, avpch1); d_assert(ret == 0, return CORE_ERROR,); } if (charging_rule == 1) { ret = fd_msg_avp_add(req, MSG_BRW_LAST_CHILD, avp); d_assert(ret == 0, return CORE_ERROR,); } } else d_assert(0, return CORE_ERROR, "Invalid Command Code(%d)", rx_message->cmd_code); if (charging_rule == 0) { rx_message->result_code = ER_DIAMETER_SUCCESS; goto out; } /* Set Origin-Host & Origin-Realm */ ret = fd_msg_add_origin(req, 0); d_assert(ret == 0, return CORE_ERROR,); /* Set the Destination-Realm AVP */ ret = fd_msg_avp_new(fd_destination_realm, 0, &avp); d_assert(ret == 0, return CORE_ERROR,); 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); d_assert(ret == 0, return CORE_ERROR,); ret = fd_msg_avp_add(req, MSG_BRW_LAST_CHILD, avp); d_assert(ret == 0, return CORE_ERROR,); /* Set the Destination-Host AVP */ ret = fd_msg_avp_new(fd_destination_host, 0, &avp); d_assert(ret == 0, return CORE_ERROR,); val.os.data = sess_data->peer_host; val.os.len = strlen((char *)sess_data->peer_host); ret = fd_msg_avp_setvalue(avp, &val); d_assert(ret == 0, return CORE_ERROR,); ret = fd_msg_avp_add(req, MSG_BRW_LAST_CHILD, avp); d_assert(ret == 0, return CORE_ERROR,); /* Set the Auth-Application-Id AVP */ ret = fd_msg_avp_new(fd_auth_application_id, 0, &avp); d_assert(ret == 0, return CORE_ERROR,); val.i32 = GX_APPLICATION_ID; ret = fd_msg_avp_setvalue(avp, &val); d_assert(ret == 0, return CORE_ERROR,); ret = fd_msg_avp_add(req, MSG_BRW_LAST_CHILD, avp); d_assert(ret == 0, return CORE_ERROR,); /* Set the Re-Auth-Request-Type */ ret = fd_msg_avp_new(fd_re_auth_request_type, 0, &avp); d_assert(ret == 0, return CORE_ERROR,); val.i32 = FD_RE_AUTH_REQUEST_TYPE_AUTHORIZE_ONLY; ret = fd_msg_avp_setvalue(avp, &val); d_assert(ret == 0, return CORE_ERROR,); ret = fd_msg_avp_add(req, MSG_BRW_LAST_CHILD, avp); d_assert(ret == 0, return CORE_ERROR,); ret = clock_gettime(CLOCK_REALTIME, &sess_data->ts); d_assert(ret == 0, return CORE_ERROR,); /* 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(pcrf_gx_reg, session, &sess_data); d_assert(ret == 0, return CORE_ERROR,); d_assert(sess_data == NULL, return CORE_ERROR,); /* Send the request */ ret = fd_msg_send(&req, pcrf_gx_raa_cb, svg); d_assert(ret == 0, return CORE_ERROR,); /* Increment the counter */ d_assert(pthread_mutex_lock(&fd_logger_self()->stats_lock) == 0,,); fd_logger_self()->stats.nb_sent++; d_assert(pthread_mutex_unlock(&fd_logger_self()->stats_lock) == 0,, ); d_trace(3, "[Gx] Re-Auth-Request : PCRF --> PGW\n"); /* Set no error */ rx_message->result_code = ER_DIAMETER_SUCCESS; gx_message_free(&gx_message); return CORE_OK; out: /* Store this value in the session */ ret = fd_sess_state_store(pcrf_gx_reg, session, &sess_data); d_assert(sess_data == NULL,,); gx_message_free(&gx_message); return CORE_ERROR; } static void pcrf_gx_raa_cb(void *data, struct msg **msg) { int ret; struct sess_state *sess_data = NULL; struct timespec ts; struct session *session; struct avp *avp, *avpch1; struct avp_hdr *hdr; unsigned long dur; int error = 0; int new; c_uint32_t result_code; ret = clock_gettime(CLOCK_REALTIME, &ts); d_assert(ret == 0, return,); /* Search the session, retrieve its data */ ret = fd_msg_sess_get(fd_g_config->cnf_dict, *msg, &session, &new); d_assert(ret == 0, return,); d_assert(new == 0, return, ); ret = fd_sess_state_retrieve(pcrf_gx_reg, session, &sess_data); d_assert(ret == 0, return,); d_assert(sess_data && (void *)sess_data == data, return, ); d_trace(3, "[Gx] Re-Auth-Answer : PGW --> PCRF\n"); /* Value of Result Code */ ret = fd_msg_search_avp(*msg, fd_result_code, &avp); d_assert(ret == 0, return,); if (avp) { ret = fd_msg_avp_hdr(avp, &hdr); d_assert(ret == 0, return,); result_code = hdr->avp_value->i32; d_trace(3, "Result Code: %d\n", hdr->avp_value->i32); } else { ret = fd_msg_search_avp(*msg, fd_experimental_result, &avp); d_assert(ret == 0, return,); if (avp) { ret = fd_avp_search_avp(avp, fd_experimental_result_code, &avpch1); d_assert(ret == 0, return,); if (avpch1) { ret = fd_msg_avp_hdr(avpch1, &hdr); d_assert(ret == 0, return,); result_code = hdr->avp_value->i32; d_trace(3, "Experimental Result Code: %d\n", result_code); } } else { d_error("no Result-Code"); error++; } } /* Value of Origin-Host */ ret = fd_msg_search_avp(*msg, fd_origin_host, &avp); d_assert(ret == 0, return,); if (avp) { ret = fd_msg_avp_hdr(avp, &hdr); d_assert(ret == 0, 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 */ ret = fd_msg_search_avp(*msg, fd_origin_realm, &avp); d_assert(ret == 0, return,); if (avp) { ret = fd_msg_avp_hdr(avp, &hdr); d_assert(ret == 0, return,); d_trace(3, "('%.*s') ", (int)hdr->avp_value->os.len, hdr->avp_value->os.data); } else { d_error("no_Origin-Realm"); error++; } /* Free the message */ d_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++; d_assert(pthread_mutex_unlock(&fd_logger_self()->stats_lock) == 0,,); /* Display how long it took */ if (ts.tv_nsec > sess_data->ts.tv_nsec) d_trace(3, "in %d.%06ld sec\n", (int)(ts.tv_sec - sess_data->ts.tv_sec), (long)(ts.tv_nsec - sess_data->ts.tv_nsec) / 1000); else d_trace(3, "in %d.%06ld sec\n", (int)(ts.tv_sec + 1 - sess_data->ts.tv_sec), (long)(1000000000 + ts.tv_nsec - sess_data->ts.tv_nsec) / 1000); ret = fd_sess_state_store(pcrf_gx_reg, session, &sess_data); d_assert(ret == 0, return,); d_assert(sess_data == NULL, return,); ret = fd_msg_free(*msg); d_assert(ret == 0,,); *msg = NULL; return; } status_t pcrf_gx_init(void) { int ret; struct disp_when data; pool_init(&pcrf_gx_sess_pool, MAX_POOL_OF_DIAMETER_SESS); /* Install objects definitions for this application */ ret = gx_dict_init(); d_assert(ret == 0, return CORE_ERROR,); /* Create handler for sessions */ ret = fd_sess_handler_create(&pcrf_gx_reg, state_cleanup, NULL, NULL); d_assert(ret == 0, return CORE_ERROR,); memset(&data, 0, sizeof(data)); data.app = gx_application; ret = fd_disp_register(pcrf_gx_fb_cb, DISP_HOW_APPID, &data, NULL, &hdl_gx_fb); d_assert(ret == 0, return CORE_ERROR,); data.command = gx_cmd_ccr; ret = fd_disp_register(pcrf_gx_ccr_cb, DISP_HOW_CC, &data, NULL, &hdl_gx_ccr); d_assert(ret == 0, return CORE_ERROR,); /* Advertise the support for the application in the peer */ ret = fd_disp_app_support(gx_application, fd_vendor, 1, 0); d_assert(ret == 0, return CORE_ERROR,); return CORE_OK; } void pcrf_gx_final(void) { int ret; ret = fd_sess_handler_destroy(&pcrf_gx_reg, NULL); d_assert(ret == 0,,); if (hdl_gx_fb) (void) fd_disp_unregister(&hdl_gx_fb, NULL); if (hdl_gx_ccr) (void) fd_disp_unregister(&hdl_gx_ccr, NULL); if (pool_used(&pcrf_gx_sess_pool)) d_error("%d not freed in pcrf_gx_sess_pool[%d] of GX-SM", pool_used(&pcrf_gx_sess_pool), pool_size(&pcrf_gx_sess_pool)); d_trace(3, "%d not freed in pcrf_gx_sess_pool[%d] of GX-SM\n", pool_used(&pcrf_gx_sess_pool), pool_size(&pcrf_gx_sess_pool)); pool_final(&pcrf_gx_sess_pool); } static status_t encode_pcc_rule_definition( struct avp *avp, pcc_rule_t *pcc_rule) { struct avp *avpch1, *avpch2, *avpch3, *avpch4; union avp_value val; int ret = 0, i; d_assert(avp, return CORE_ERROR,); d_assert(pcc_rule, return CORE_ERROR,); ret = fd_msg_avp_new(gx_charging_rule_definition, 0, &avpch1); d_assert(ret == 0, return CORE_ERROR,); ret = fd_msg_avp_new(gx_charging_rule_name, 0, &avpch2); d_assert(ret == 0, return CORE_ERROR,); val.os.data = (c_uint8_t *)pcc_rule->name; val.os.len = strlen(pcc_rule->name); ret = fd_msg_avp_setvalue(avpch2, &val); d_assert(ret == 0, return CORE_ERROR,); ret = fd_msg_avp_add(avpch1, MSG_BRW_LAST_CHILD, avpch2); d_assert(ret == 0, return CORE_ERROR,); for (i = 0; i < pcc_rule->num_of_flow; i++) { flow_t *flow = &pcc_rule->flow[i]; ret = fd_msg_avp_new(gx_flow_information, 0, &avpch2); d_assert(ret == 0, return CORE_ERROR,); ret = fd_msg_avp_new(gx_flow_direction, 0, &avpch3); d_assert(ret == 0, return CORE_ERROR,); val.i32 = flow->direction; ret = fd_msg_avp_setvalue(avpch3, &val); d_assert(ret == 0, return CORE_ERROR,); ret = fd_msg_avp_add(avpch2, MSG_BRW_LAST_CHILD, avpch3); d_assert(ret == 0, return CORE_ERROR,); ret = fd_msg_avp_new(gx_flow_description, 0, &avpch3); d_assert(ret == 0, return CORE_ERROR,); val.os.data = (c_uint8_t *)flow->description; val.os.len = strlen(flow->description); ret = fd_msg_avp_setvalue(avpch3, &val); d_assert(ret == 0, return CORE_ERROR,); ret = fd_msg_avp_add(avpch2, MSG_BRW_LAST_CHILD, avpch3); d_assert(ret == 0, return CORE_ERROR,); ret = fd_msg_avp_add(avpch1, MSG_BRW_LAST_CHILD, avpch2); d_assert(ret == 0, return CORE_ERROR,); } ret = fd_msg_avp_new(gx_flow_status, 0, &avpch2); d_assert(ret == 0, return CORE_ERROR,); val.i32 = pcc_rule->flow_status; ret = fd_msg_avp_setvalue(avpch2, &val); d_assert(ret == 0, return CORE_ERROR,); ret = fd_msg_avp_add(avpch1, MSG_BRW_LAST_CHILD, avpch2); d_assert(ret == 0, return CORE_ERROR,); ret = fd_msg_avp_new(gx_qos_information, 0, &avpch2); d_assert(ret == 0, return CORE_ERROR,); ret = fd_msg_avp_new(gx_qos_class_identifier, 0, &avpch3); d_assert(ret == 0, return CORE_ERROR,); val.u32 = pcc_rule->qos.qci; ret = fd_msg_avp_setvalue (avpch3, &val); d_assert(ret == 0, return CORE_ERROR,); ret = fd_msg_avp_add (avpch2, MSG_BRW_LAST_CHILD, avpch3); d_assert(ret == 0, return CORE_ERROR,); ret = fd_msg_avp_new(gx_allocation_retention_priority, 0, &avpch3); d_assert(ret == 0, return CORE_ERROR,); ret = fd_msg_avp_new(gx_priority_level, 0, &avpch4); d_assert(ret == 0, return CORE_ERROR,); val.u32 = pcc_rule->qos.arp.priority_level; ret = fd_msg_avp_setvalue (avpch4, &val); d_assert(ret == 0, return CORE_ERROR,); ret = fd_msg_avp_add (avpch3, MSG_BRW_LAST_CHILD, avpch4); d_assert(ret == 0, return CORE_ERROR,); ret = fd_msg_avp_new(gx_pre_emption_capability, 0, &avpch4); d_assert(ret == 0, return CORE_ERROR,); val.u32 = pcc_rule->qos.arp.pre_emption_capability; ret = fd_msg_avp_setvalue (avpch4, &val); d_assert(ret == 0, return CORE_ERROR,); ret = fd_msg_avp_add (avpch3, MSG_BRW_LAST_CHILD, avpch4); d_assert(ret == 0, return CORE_ERROR,); ret = fd_msg_avp_new(gx_pre_emption_vulnerability, 0, &avpch4); d_assert(ret == 0, return CORE_ERROR,); val.u32 = pcc_rule->qos.arp.pre_emption_vulnerability; ret = fd_msg_avp_setvalue (avpch4, &val); d_assert(ret == 0, return CORE_ERROR,); ret = fd_msg_avp_add (avpch3, MSG_BRW_LAST_CHILD, avpch4); d_assert(ret == 0, return CORE_ERROR,); ret = fd_msg_avp_add (avpch2, MSG_BRW_LAST_CHILD, avpch3); d_assert(ret == 0, return CORE_ERROR,); if (pcc_rule->qos.mbr.uplink) { ret = fd_msg_avp_new(gx_max_requested_bandwidth_ul, 0, &avpch3); d_assert(ret == 0, return CORE_ERROR,); val.u32 = pcc_rule->qos.mbr.uplink; ret = fd_msg_avp_setvalue (avpch3, &val); d_assert(ret == 0, return CORE_ERROR,); ret = fd_msg_avp_add (avpch2, MSG_BRW_LAST_CHILD, avpch3); d_assert(ret == 0, return CORE_ERROR,); } if (pcc_rule->qos.mbr.downlink) { ret = fd_msg_avp_new(gx_max_requested_bandwidth_dl, 0, &avpch3); d_assert(ret == 0, return CORE_ERROR,); val.u32 = pcc_rule->qos.mbr.downlink; ret = fd_msg_avp_setvalue (avpch3, &val); d_assert(ret == 0, return CORE_ERROR,); ret = fd_msg_avp_add (avpch2, MSG_BRW_LAST_CHILD, avpch3); d_assert(ret == 0, return CORE_ERROR,); } if (pcc_rule->qos.gbr.uplink) { ret = fd_msg_avp_new(gx_guaranteed_bitrate_ul, 0, &avpch3); d_assert(ret == 0, return CORE_ERROR,); val.u32 = pcc_rule->qos.gbr.uplink; ret = fd_msg_avp_setvalue (avpch3, &val); d_assert(ret == 0, return CORE_ERROR,); ret = fd_msg_avp_add (avpch2, MSG_BRW_LAST_CHILD, avpch3); d_assert(ret == 0, return CORE_ERROR,); } if (pcc_rule->qos.gbr.downlink) { ret = fd_msg_avp_new(gx_guaranteed_bitrate_dl, 0, &avpch3); d_assert(ret == 0, return CORE_ERROR,); val.u32 = pcc_rule->qos.gbr.downlink; ret = fd_msg_avp_setvalue (avpch3, &val); d_assert(ret == 0, return CORE_ERROR,); ret = fd_msg_avp_add (avpch2, MSG_BRW_LAST_CHILD, avpch3); d_assert(ret == 0, return CORE_ERROR,); } ret = fd_msg_avp_add(avpch1, MSG_BRW_LAST_CHILD, avpch2); d_assert(ret == 0, return CORE_ERROR,); ret = fd_msg_avp_new(gx_precedence, 0, &avpch2); d_assert(ret == 0, return CORE_ERROR,); val.u32 = pcc_rule->precedence; ret = fd_msg_avp_setvalue (avpch2, &val); d_assert(ret == 0, return CORE_ERROR,); ret = fd_msg_avp_add(avpch1, MSG_BRW_LAST_CHILD, avpch2); d_assert(ret == 0, return CORE_ERROR,); ret = fd_msg_avp_add(avp, MSG_BRW_LAST_CHILD, avpch1); d_assert(ret == 0, return CORE_ERROR,); return CORE_OK; }