/* * Copyright (C) 2019 by Sukchan Lee * * 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 . */ #include "ogs-gtp.h" #include "ogs-diameter-rx.h" #include "test-common.h" #include "test-fd-path.h" static struct session_handler *pcscf_rx_reg = NULL; static struct disp_hdl *hdl_rx_fb = NULL; static struct disp_hdl *hdl_rx_asr = NULL; static ogs_diam_config_t diam_config; struct sess_state { os0_t sid; /* Rx Session-Id */ struct timespec ts; /* Time of sending the message */ }; static void pcscf_rx_aaa_cb(void *data, struct msg **msg); static void pcscf_rx_sta_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)); ogs_expect_or_return_val(new, NULL); new->sid = (os0_t)ogs_strdup((char *)sid); ogs_expect_or_return_val(new->sid, NULL); return new; } static void state_cleanup(struct sess_state *sess_data, os0_t sid, void *opaque) { if (sess_data->sid) ogs_free(sess_data->sid); ogs_free(sess_data); } static int pcscf_rx_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; } void test_rx_send_aar_audio(uint8_t **rx_sid, test_sess_t *sess, int id_type, int qos_type, int flow_type) { int rv; 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; test_ue_t *test_ue = NULL; char *ipstr = NULL; char *sip_uri = NULL; ogs_paa_t paa; ogs_ipsubnet_t ipsub; ogs_assert(sess); test_ue = sess->test_ue; ogs_assert(test_ue); ogs_assert(rx_sid); ogs_assert(sess->ue_ip.ipv4); ipstr = ogs_ipv4_to_string(sess->ue_ip.addr); ogs_assert(ipstr); sip_uri = ogs_msprintf("sip:%s:40491", ipstr); ogs_assert(sip_uri); rv = ogs_ipsubnet(&ipsub, ipstr, NULL); ogs_assert(rv == OGS_OK); ogs_free(ipstr); /* Create the request */ ret = fd_msg_new(ogs_diam_rx_cmd_aar, MSGFL_ALLOC_ETEID, &req); ogs_assert(ret == 0); { struct msg_hdr * h; ret = fd_msg_hdr( req, &h ); ogs_assert(ret == 0); h->msg_appl = OGS_DIAM_RX_APPLICATION_ID; } /* Find Diameter Rx Session */ if (*rx_sid) { /* Retrieve session by Session-Id */ size_t sidlen = strlen((char *)*rx_sid); ret = fd_sess_fromsid_msg(*rx_sid, sidlen, &session, &new); ogs_assert(ret == 0); ogs_assert(new == 0); /* Add Session-Id to the message */ ret = ogs_diam_message_session_id_set(req, *rx_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 OGS_DIAM_RX_APP_SID_OPT "app_rx" ret = fd_msg_new_session(req, (os0_t)OGS_DIAM_RX_APP_SID_OPT, CONSTSTRLEN(OGS_DIAM_RX_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(pcscf_rx_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 */ *rx_sid = sess_data->sid; } /* Set Origin-Host & Origin-Realm */ ret = fd_msg_add_origin(req, 0); ogs_assert(ret == 0); /* Set the Destination-Host AVP */ ret = fd_msg_avp_new(ogs_diam_destination_host, 0, &avp); ogs_assert(ret == 0); val.os.data = TEST_PCRF_IDENTITY; val.os.len = strlen(TEST_PCRF_IDENTITY); 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 Destination-Realm AVP */ ret = fd_msg_avp_new(ogs_diam_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(ogs_diam_auth_application_id, 0, &avp); ogs_assert(ret == 0); val.i32 = OGS_DIAM_RX_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 Vendor-Specific-Application-Id AVP */ ret = ogs_diam_message_vendor_specific_appid_set( req, OGS_DIAM_RX_APPLICATION_ID); ogs_assert(ret == 0); /* Set the AF-Application-Identifier AVP */ ret = fd_msg_avp_new(ogs_diam_rx_af_application_identifier, 0, &avp); ogs_assert(ret == 0); val.os.data = (unsigned char *)("IMS Services"); val.os.len = strlen("IMS Services"); 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 Authroization-Lifetime AVP */ ret = fd_msg_avp_new(ogs_diam_authorization_lifetime, 0, &avp); ogs_assert(ret == 0); val.i32 = 7200; 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(ogs_diam_rx_subscription_id, 0, &avp); ogs_assert(ret == 0); ret = fd_msg_avp_new(ogs_diam_rx_subscription_id_type, 0, &avpch1); ogs_assert(ret == 0); val.i32 = id_type; 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(ogs_diam_rx_subscription_id_data, 0, &avpch1); ogs_assert(ret == 0); if (id_type == OGS_DIAM_RX_SUBSCRIPTION_ID_TYPE_END_USER_IMSI) { val.os.data = (uint8_t *)test_ue->imsi; val.os.len = strlen(test_ue->imsi); } else if (id_type == OGS_DIAM_RX_SUBSCRIPTION_ID_TYPE_END_USER_SIP_URI) { val.os.data = (uint8_t *)sip_uri; val.os.len = strlen(sip_uri); } 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 the Reservation-Priority AVP */ ret = fd_msg_avp_new(ogs_diam_rx_reservation_priority, 0, &avp); ogs_assert(ret == 0); val.i32 = 0; 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); if (ipsub.family == AF_INET) { /* Set Framed-IP-Address */ ret = fd_msg_avp_new(ogs_diam_rx_framed_ip_address, 0, &avp); ogs_assert(ret == 0); val.os.data = (uint8_t*)ipsub.sub; val.os.len = OGS_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); } else if (ipsub.family == AF_INET6) { /* Set Framed-IPv6-Prefix */ ret = fd_msg_avp_new(ogs_diam_rx_framed_ipv6_prefix, 0, &avp); ogs_assert(ret == 0); memset(&paa, 0, sizeof(ogs_paa_t)); memcpy(paa.addr6, ipsub.sub, OGS_IPV6_LEN); paa.session_type = 0x03; #define FRAMED_IPV6_PREFIX_LENGTH 128 /* from spec document */ paa.len = FRAMED_IPV6_PREFIX_LENGTH; val.os.data = (uint8_t*)&paa; val.os.len = OGS_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 Media-Component-Description */ ret = fd_msg_avp_new(ogs_diam_rx_media_component_description, 0, &avp); ogs_assert(ret == 0); ret = fd_msg_avp_new(ogs_diam_rx_media_component_number, 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(ogs_diam_rx_media_type, 0, &avpch1); ogs_assert(ret == 0); val.i32 = OGS_DIAM_RX_MEDIA_TYPE_AUDIO; 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 (qos_type == 0) { /* Nothing */ } else if (qos_type == 1) { ret = fd_msg_avp_new( ogs_diam_rx_max_requested_bandwidth_dl, 0, &avpch1); ogs_assert(ret == 0); val.i32 = 96000; 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( ogs_diam_rx_max_requested_bandwidth_ul, 0, &avpch1); ogs_assert(ret == 0); val.i32 = 96000; 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(ogs_diam_rx_rr_bandwidth, 0, &avpch1); ogs_assert(ret == 0); val.i32 = 2400; 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(ogs_diam_rx_rs_bandwidth, 0, &avpch1); ogs_assert(ret == 0); val.i32 = 2400; 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); } else if (qos_type == 2) { ret = fd_msg_avp_new( ogs_diam_rx_max_requested_bandwidth_dl, 0, &avpch1); ogs_assert(ret == 0); val.i32 = 96000; 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( ogs_diam_rx_max_requested_bandwidth_ul, 0, &avpch1); ogs_assert(ret == 0); val.i32 = 96000; 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( ogs_diam_rx_min_requested_bandwidth_dl, 0, &avpch1); ogs_assert(ret == 0); val.i32 = 88000; 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( ogs_diam_rx_min_requested_bandwidth_ul, 0, &avpch1); ogs_assert(ret == 0); val.i32 = 88000; 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); } else { ogs_fatal("Unknown qos-type[%d]", qos_type); ogs_assert_if_reached(); } if (flow_type == 1) { /* Set Media-Sub-Component #1 */ ret = fd_msg_avp_new(ogs_diam_rx_media_sub_component, 0, &avpch1); ret = fd_msg_avp_new(ogs_diam_rx_flow_number, 0, &avpch2); ogs_assert(ret == 0); val.i32 = 1; 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(ogs_diam_rx_flow_description, 0, &avpch2); ogs_assert(ret == 0); #define TEST_OGS_DIAM_RX_FLOW_DESC1 \ "permit out icmp from any to assigned" val.os.data = (uint8_t *)TEST_OGS_DIAM_RX_FLOW_DESC1; val.os.len = strlen(TEST_OGS_DIAM_RX_FLOW_DESC1); 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(ogs_diam_rx_flow_description, 0, &avpch2); ogs_assert(ret == 0); #define TEST_OGS_DIAM_RX_FLOW_DESC2 \ "permit in icmp from assigned to any" val.os.data = (uint8_t *)TEST_OGS_DIAM_RX_FLOW_DESC2; val.os.len = strlen(TEST_OGS_DIAM_RX_FLOW_DESC2); 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); /* Set Media-Sub-Component #2 */ ret = fd_msg_avp_new(ogs_diam_rx_media_sub_component, 0, &avpch1); ret = fd_msg_avp_new(ogs_diam_rx_flow_number, 0, &avpch2); ogs_assert(ret == 0); val.i32 = 2; 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(ogs_diam_rx_flow_usage, 0, &avpch2); ogs_assert(ret == 0); val.i32 = OGS_DIAM_RX_FLOW_USAGE_RTCP; 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(ogs_diam_rx_flow_description, 0, &avpch2); ogs_assert(ret == 0); #define TEST_OGS_DIAM_RX_FLOW_DESC3 \ "permit out 17 from 172.20.166.84 to 10.45.0.2 20002" val.os.data = (uint8_t *)TEST_OGS_DIAM_RX_FLOW_DESC3; val.os.len = strlen(TEST_OGS_DIAM_RX_FLOW_DESC3); 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(ogs_diam_rx_flow_description, 0, &avpch2); ogs_assert(ret == 0); #define TEST_OGS_DIAM_RX_FLOW_DESC4 \ "permit in 17 from 10.45.0.2 to 172.20.166.84 20361" val.os.data = (uint8_t *)TEST_OGS_DIAM_RX_FLOW_DESC4; val.os.len = strlen(TEST_OGS_DIAM_RX_FLOW_DESC4); 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); } else if (flow_type == 2) { /* Set Media-Sub-Component #1 */ ret = fd_msg_avp_new(ogs_diam_rx_media_sub_component, 0, &avpch1); ret = fd_msg_avp_new(ogs_diam_rx_flow_number, 0, &avpch2); ogs_assert(ret == 0); val.i32 = 1; 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(ogs_diam_rx_flow_description, 0, &avpch2); ogs_assert(ret == 0); #define TEST_OGS_DIAM_RX_FLOW_DESC17 \ "permit out 17 from 172.20.166.84 to 10.45.0.2 30001" val.os.data = (uint8_t *)TEST_OGS_DIAM_RX_FLOW_DESC17; val.os.len = strlen(TEST_OGS_DIAM_RX_FLOW_DESC17); 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(ogs_diam_rx_flow_description, 0, &avpch2); ogs_assert(ret == 0); #define TEST_OGS_DIAM_RX_FLOW_DESC18 \ "permit in 17 from 10.45.0.2 to 172.20.166.84 30360" val.os.data = (uint8_t *)TEST_OGS_DIAM_RX_FLOW_DESC18; val.os.len = strlen(TEST_OGS_DIAM_RX_FLOW_DESC18); 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); /* Set Media-Sub-Component #2 */ ret = fd_msg_avp_new(ogs_diam_rx_media_sub_component, 0, &avpch1); ret = fd_msg_avp_new(ogs_diam_rx_flow_number, 0, &avpch2); ogs_assert(ret == 0); val.i32 = 2; 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(ogs_diam_rx_flow_usage, 0, &avpch2); ogs_assert(ret == 0); val.i32 = OGS_DIAM_RX_FLOW_USAGE_RTCP; 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(ogs_diam_rx_flow_description, 0, &avpch2); ogs_assert(ret == 0); #define TEST_OGS_DIAM_RX_FLOW_DESC19 \ "permit out 17 from 172.20.166.84 to 10.45.0.2 30002" val.os.data = (uint8_t *)TEST_OGS_DIAM_RX_FLOW_DESC19; val.os.len = strlen(TEST_OGS_DIAM_RX_FLOW_DESC19); 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(ogs_diam_rx_flow_description, 0, &avpch2); ogs_assert(ret == 0); #define TEST_OGS_DIAM_RX_FLOW_DESC20 \ "permit in 17 from 10.45.0.2 to 172.20.166.84 30361" val.os.data = (uint8_t *)TEST_OGS_DIAM_RX_FLOW_DESC20; val.os.len = strlen(TEST_OGS_DIAM_RX_FLOW_DESC20); 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); } else { ogs_fatal("Unknown flow-type[%d]", flow_type); ogs_assert_if_reached(); } ret = fd_msg_avp_add(req, MSG_BRW_LAST_CHILD, avp); ogs_assert(ret == 0); ret = clock_gettime(CLOCK_REALTIME, &sess_data->ts); ogs_assert(ret == 0); /* 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(pcscf_rx_reg, session, &sess_data); ogs_assert(ret == 0); ogs_assert(sess_data == NULL); /* Send the request */ ret = fd_msg_send(&req, pcscf_rx_aaa_cb, svg); ogs_assert(ret == 0); /* Increment the counter */ ogs_assert(pthread_mutex_lock(&ogs_diam_logger_self()->stats_lock) == 0); ogs_diam_logger_self()->stats.nb_sent++; ogs_assert(pthread_mutex_unlock(&ogs_diam_logger_self()->stats_lock) == 0); /* Free string memory */ ogs_free(sip_uri); } void test_rx_send_aar_video(uint8_t **rx_sid, test_sess_t *sess, int id_type) { int rv; 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; test_ue_t *test_ue = NULL; char *ipstr = NULL; char *sip_uri = NULL; ogs_paa_t paa; ogs_ipsubnet_t ipsub; ogs_assert(sess); test_ue = sess->test_ue; ogs_assert(test_ue); ogs_assert(rx_sid); ogs_assert(sess->ue_ip.ipv4); ipstr = ogs_ipv4_to_string(sess->ue_ip.addr); ogs_assert(ipstr); sip_uri = ogs_msprintf("sip:%s:40491", ipstr); ogs_assert(sip_uri); rv = ogs_ipsubnet(&ipsub, ipstr, NULL); ogs_assert(rv == OGS_OK); ogs_free(ipstr); /* Create the request */ ret = fd_msg_new(ogs_diam_rx_cmd_aar, MSGFL_ALLOC_ETEID, &req); ogs_assert(ret == 0); { struct msg_hdr * h; ret = fd_msg_hdr( req, &h ); ogs_assert(ret == 0); h->msg_appl = OGS_DIAM_RX_APPLICATION_ID; } /* Find Diameter Rx Session */ if (*rx_sid) { /* Retrieve session by Session-Id */ size_t sidlen = strlen((char *)*rx_sid); ret = fd_sess_fromsid_msg(*rx_sid, sidlen, &session, &new); ogs_assert(ret == 0); ogs_assert(new == 0); /* Add Session-Id to the message */ ret = ogs_diam_message_session_id_set(req, *rx_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 OGS_DIAM_RX_APP_SID_OPT "app_rx" ret = fd_msg_new_session(req, (os0_t)OGS_DIAM_RX_APP_SID_OPT, CONSTSTRLEN(OGS_DIAM_RX_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(pcscf_rx_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 */ *rx_sid = sess_data->sid; } /* Set Origin-Host & Origin-Realm */ ret = fd_msg_add_origin(req, 0); ogs_assert(ret == 0); /* Set the Destination-Host AVP */ ret = fd_msg_avp_new(ogs_diam_destination_host, 0, &avp); ogs_assert(ret == 0); val.os.data = TEST_PCRF_IDENTITY; val.os.len = strlen(TEST_PCRF_IDENTITY); 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 Destination-Realm AVP */ ret = fd_msg_avp_new(ogs_diam_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(ogs_diam_auth_application_id, 0, &avp); ogs_assert(ret == 0); val.i32 = OGS_DIAM_RX_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 Vendor-Specific-Application-Id AVP */ ret = ogs_diam_message_vendor_specific_appid_set( req, OGS_DIAM_RX_APPLICATION_ID); ogs_assert(ret == 0); /* Set the AF-Application-Identifier AVP */ ret = fd_msg_avp_new(ogs_diam_rx_af_application_identifier, 0, &avp); ogs_assert(ret == 0); val.os.data = (unsigned char *)("IMS Services"); val.os.len = strlen("IMS Services"); 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 Authroization-Lifetime AVP */ ret = fd_msg_avp_new(ogs_diam_authorization_lifetime, 0, &avp); ogs_assert(ret == 0); val.i32 = 7200; 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(ogs_diam_rx_subscription_id, 0, &avp); ogs_assert(ret == 0); ret = fd_msg_avp_new(ogs_diam_rx_subscription_id_type, 0, &avpch1); ogs_assert(ret == 0); val.i32 = id_type; 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(ogs_diam_rx_subscription_id_data, 0, &avpch1); ogs_assert(ret == 0); if (id_type == OGS_DIAM_RX_SUBSCRIPTION_ID_TYPE_END_USER_IMSI) { val.os.data = (uint8_t *)test_ue->imsi; val.os.len = strlen(test_ue->imsi); } else if (id_type == OGS_DIAM_RX_SUBSCRIPTION_ID_TYPE_END_USER_SIP_URI) { val.os.data = (uint8_t *)sip_uri; val.os.len = strlen(sip_uri); } 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 the Reservation-Priority AVP */ ret = fd_msg_avp_new(ogs_diam_rx_reservation_priority, 0, &avp); ogs_assert(ret == 0); val.i32 = 0; 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 Media-Component-Description #1 */ ret = fd_msg_avp_new(ogs_diam_rx_media_component_description, 0, &avp); ogs_assert(ret == 0); ret = fd_msg_avp_new(ogs_diam_rx_media_component_number, 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); /* Set Media-Sub-Component #1 */ ret = fd_msg_avp_new(ogs_diam_rx_media_sub_component, 0, &avpch1); ret = fd_msg_avp_new(ogs_diam_rx_flow_number, 0, &avpch2); ogs_assert(ret == 0); val.i32 = 1; 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(ogs_diam_rx_flow_description, 0, &avpch2); ogs_assert(ret == 0); #define TEST_OGS_DIAM_RX_FLOW_DESC5 \ "permit out icmp from any to assigned" val.os.data = (uint8_t *)TEST_OGS_DIAM_RX_FLOW_DESC5; val.os.len = strlen(TEST_OGS_DIAM_RX_FLOW_DESC5); 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(ogs_diam_rx_flow_description, 0, &avpch2); ogs_assert(ret == 0); #define TEST_OGS_DIAM_RX_FLOW_DESC6 \ "permit in icmp from assigned to any" val.os.data = (uint8_t *)TEST_OGS_DIAM_RX_FLOW_DESC6; val.os.len = strlen(TEST_OGS_DIAM_RX_FLOW_DESC6); 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(ogs_diam_rx_flow_description, 0, &avpch2); ogs_assert(ret == 0); #define TEST_OGS_DIAM_RX_FLOW_DESC7 \ "permit out 17 from 10.45.0.5 50027 to 10.45.0.3 50023" val.os.data = (uint8_t *)TEST_OGS_DIAM_RX_FLOW_DESC7; val.os.len = strlen(TEST_OGS_DIAM_RX_FLOW_DESC7); 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(ogs_diam_rx_flow_description, 0, &avpch2); ogs_assert(ret == 0); #define TEST_OGS_DIAM_RX_FLOW_DESC8 \ "permit in 17 from 10.45.0.3 50023 to 10.45.0.5 50027" val.os.data = (uint8_t *)TEST_OGS_DIAM_RX_FLOW_DESC8; val.os.len = strlen(TEST_OGS_DIAM_RX_FLOW_DESC8); 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(ogs_diam_rx_flow_usage, 0, &avpch2); ogs_assert(ret == 0); val.i32 = OGS_DIAM_RX_FLOW_USAGE_NO_INFORMATION; 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_new(ogs_diam_rx_media_type, 0, &avpch1); ogs_assert(ret == 0); val.i32 = OGS_DIAM_RX_MEDIA_TYPE_AUDIO; 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(ogs_diam_rx_max_requested_bandwidth_dl, 0, &avpch1); ogs_assert(ret == 0); val.i32 = 41000; 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(ogs_diam_rx_max_requested_bandwidth_ul, 0, &avpch1); ogs_assert(ret == 0); val.i32 = 41000; 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(ogs_diam_rx_rs_bandwidth, 0, &avpch1); ogs_assert(ret == 0); val.i32 = 600; 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(ogs_diam_rx_rr_bandwidth, 0, &avpch1); ogs_assert(ret == 0); val.i32 = 2000; 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(ogs_diam_rx_codec_data, 0, &avpch1); ogs_assert(ret == 0); #define TEST_OGS_DIAM_RX_CODEC_DATA1 \ "downlink\noffer\nm=audio 50026 RTP/AVP 104 102 96 97\r\n" \ "b=AS:41\r\n" \ "b=RS:600\r\n" \ "b=RR:2000\r\n" \ "a=rtpmap:104 AMR-WB/16000/1\r\n" \ "a=fmtp:104 mode-change-capability=2;max-red=0\r\n" \ "a=rtpmap:102 AMR/8000/1\r\n" \ "a=fmtp:102 mode-change-capability=2;max-red=0\r\n" \ "a=rtpmap:96 telephone-event/16000\r\n" \ "a=fmtp:96 0-15\r\n" \ "a=rtpmap:97 telephone-event/8000\r\n" \ "a=fmtp:97 0-15\r\n" \ "a=curr:qos local none\r\n" \ "a=curr:qos remote none\r\n" \ "a=des:qos mandatory local sendrecv\r\n" \ "a=des:qos optional remote sendrecv\r\n" \ "a=sendrecv\r\n" \ "a=maxptime:240\r\n" \ "a=ptime:20\r\n" val.os.data = (uint8_t *)TEST_OGS_DIAM_RX_CODEC_DATA1; val.os.len = strlen(TEST_OGS_DIAM_RX_CODEC_DATA1) + 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(ogs_diam_rx_codec_data, 0, &avpch1); ogs_assert(ret == 0); #define TEST_OGS_DIAM_RX_CODEC_DATA2 \ "uplink\nanswer\nm=audio 50022 RTP/AVP 104 96\r\n" \ "b=AS:41\r\n" \ "b=RS:600\r\n" \ "b=RR:2000\r\n" \ "a=rtpmap:104 AMR-WB/16000/1\r\n" \ "a=fmtp:104 mode-change-capability=2;max-red=0\r\n" \ "a=rtpmap:96 telephone-event/16000\r\n" \ "a=fmtp:96 0-15\r\n" \ "a=curr:qos local none\r\n" \ "a=curr:qos remote none\r\n" \ "a=des:qos mandatory local sendrecv\r\n" \ "a=des:qos mandatory remote sendrecv\r\n" \ "a=conf:qos remote sendrecv\r\n" \ "a=sendrecv\r\n" \ "a=maxptime:240\r\n" \ "a=ptime:20\r\n" val.os.data = (uint8_t *)TEST_OGS_DIAM_RX_CODEC_DATA2; val.os.len = strlen(TEST_OGS_DIAM_RX_CODEC_DATA2) + 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(ogs_diam_rx_flow_status, 0, &avpch1); ogs_assert(ret == 0); val.i32 = OGS_DIAM_RX_FLOW_STATUS_ENABLED; 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 Media-Component-Description #2 */ ret = fd_msg_avp_new(ogs_diam_rx_media_component_description, 0, &avp); ogs_assert(ret == 0); ret = fd_msg_avp_new(ogs_diam_rx_media_component_number, 0, &avpch1); ogs_assert(ret == 0); val.i32 = 2; 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); /* Set Media-Sub-Component #1 */ ret = fd_msg_avp_new(ogs_diam_rx_media_sub_component, 0, &avpch1); ret = fd_msg_avp_new(ogs_diam_rx_flow_number, 0, &avpch2); ogs_assert(ret == 0); val.i32 = 2; 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(ogs_diam_rx_flow_description, 0, &avpch2); ogs_assert(ret == 0); #define TEST_OGS_DIAM_RX_FLOW_DESC9 \ "permit out icmp from any to assigned" val.os.data = (uint8_t *)TEST_OGS_DIAM_RX_FLOW_DESC9; val.os.len = strlen(TEST_OGS_DIAM_RX_FLOW_DESC9); 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(ogs_diam_rx_flow_description, 0, &avpch2); ogs_assert(ret == 0); #define TEST_OGS_DIAM_RX_FLOW_DESC10 \ "permit in icmp from assigned to any" val.os.data = (uint8_t *)TEST_OGS_DIAM_RX_FLOW_DESC10; val.os.len = strlen(TEST_OGS_DIAM_RX_FLOW_DESC10); 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(ogs_diam_rx_flow_description, 0, &avpch2); ogs_assert(ret == 0); #define TEST_OGS_DIAM_RX_FLOW_DESC11 \ "permit out 17 from 10.45.0.5 60011 to 10.45.0.3 60011" val.os.data = (uint8_t *)TEST_OGS_DIAM_RX_FLOW_DESC11; val.os.len = strlen(TEST_OGS_DIAM_RX_FLOW_DESC11); 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(ogs_diam_rx_flow_description, 0, &avpch2); ogs_assert(ret == 0); #define TEST_OGS_DIAM_RX_FLOW_DESC12 \ "permit in 17 from 10.45.0.3 60011 to 10.45.0.5 60011" val.os.data = (uint8_t *)TEST_OGS_DIAM_RX_FLOW_DESC12; val.os.len = strlen(TEST_OGS_DIAM_RX_FLOW_DESC12); 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(ogs_diam_rx_flow_usage, 0, &avpch2); ogs_assert(ret == 0); val.i32 = OGS_DIAM_RX_FLOW_USAGE_NO_INFORMATION; 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_new(ogs_diam_rx_media_type, 0, &avpch1); ogs_assert(ret == 0); val.i32 = OGS_DIAM_RX_MEDIA_TYPE_VIDEO; 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(ogs_diam_rx_max_requested_bandwidth_dl, 0, &avpch1); ogs_assert(ret == 0); val.i32 = 401000; 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(ogs_diam_rx_max_requested_bandwidth_ul, 0, &avpch1); ogs_assert(ret == 0); val.i32 = 401000; 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(ogs_diam_rx_rs_bandwidth, 0, &avpch1); ogs_assert(ret == 0); val.i32 = 600; 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(ogs_diam_rx_rr_bandwidth, 0, &avpch1); ogs_assert(ret == 0); val.i32 = 2000; 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(ogs_diam_rx_codec_data, 0, &avpch1); ogs_assert(ret == 0); #define TEST_OGS_DIAM_RX_CODEC_DATA3 \ "downlink\noffer\nm=video 60010 RTP/AVP 114 113\r\n" \ "b=AS:401\r\n" \ "b=RS:600\r\n" \ "b=RR:2000\r\n" \ "a=rtpmap:114 H264/90000\r\n" \ "a=fmtp:114 profile-level-id=42C00C;packetization-mode=1;sar-understood=16;sar-supported=1;sprop-parameter-sets=Z0LADNoPCmgG0KE1,aM4G4g==\r\n" \ "a=rtpmap:113 H264/90000\r\n" \ "a=fmtp:113 profile-level-id=42C00C;packetization-mode=0;sar-understood=16;sar-supported=1;sprop-parameter-sets=Z0LADNoPCmgG0KE1,aM4G4g==\r\n" \ "a=curr:qos local none\r\n" \ "a=curr:qos remote none\r\n" \ "a=des:qos mandatory local sendrecv\r\n" \ "a=des:qos optional remote sendrecv\r\n" \ "a=rtcp-fb:* nack\r\n" \ "a=rtcp-fb:* nack pli\r\n" \ "a=rtcp-fb:* ccm fir\r\n" \ "a=rtcp-fb:* ccm tmmbr\r\n" \ "a=sendrecv\r\n" \ "a=tcap:1 RTP/AVPF\r\n" \ "a=pcfg:1 t=1\r\n" \ "a=extmap:2 urn:3gpp:video-orientation\r\n" val.os.data = (uint8_t *)TEST_OGS_DIAM_RX_CODEC_DATA3; val.os.len = strlen(TEST_OGS_DIAM_RX_CODEC_DATA3) + 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(ogs_diam_rx_codec_data, 0, &avpch1); ogs_assert(ret == 0); #define TEST_OGS_DIAM_RX_CODEC_DATA4 \ "uplink\nanswer\nm=video 60010 RTP/AVPF 114\r\n" \ "b=AS:401\r\n" \ "b=RS:600\r\n" \ "b=RR:2000\r\n" \ "a=rtpmap:114 H264/90000\r\n" \ "a=fmtp:114 profile-level-id=42C00C;packetization-mode=1;sar-understood=16;sar-supported=1;sprop-parameter-sets=Z0LADNoPCmgG0KE1,aM4G4g==\r\n" \ "a=curr:qos local none\r\n" \ "a=curr:qos remote none\r\n" \ "a=des:qos mandatory local sendrecv\r\n" \ "a=des:qos mandatory remote sendrecv\r\n" \ "a=conf:qos remote sendrecv\r\n" \ "a=rtcp-fb:* nack\r\n" \ "a=rtcp-fb:* nack pli\r\n" \ "a=rtcp-fb:* ccm fir\r\n" \ "a=rtcp-fb:* ccm tmmbr\r\n" \ "a=sendrecv\r\n" \ "a=extmap:2 urn:3gpp:video-orientation\r\n" \ "a=acfg:1 t=1\r\n" val.os.data = (uint8_t *)TEST_OGS_DIAM_RX_CODEC_DATA4; val.os.len = strlen(TEST_OGS_DIAM_RX_CODEC_DATA4) + 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(ogs_diam_rx_flow_status, 0, &avpch1); ogs_assert(ret == 0); val.i32 = OGS_DIAM_RX_FLOW_STATUS_ENABLED; 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 (ipsub.family == AF_INET) { /* Set Framed-IP-Address */ ret = fd_msg_avp_new(ogs_diam_rx_framed_ip_address, 0, &avp); ogs_assert(ret == 0); val.os.data = (uint8_t*)ipsub.sub; val.os.len = OGS_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); } else if (ipsub.family == AF_INET6) { /* Set Framed-IPv6-Prefix */ ret = fd_msg_avp_new(ogs_diam_rx_framed_ipv6_prefix, 0, &avp); ogs_assert(ret == 0); memset(&paa, 0, sizeof(ogs_paa_t)); memcpy(paa.addr6, ipsub.sub, OGS_IPV6_LEN); paa.session_type = 0x03; #define FRAMED_IPV6_PREFIX_LENGTH 128 /* from spec document */ paa.len = FRAMED_IPV6_PREFIX_LENGTH; val.os.data = (uint8_t*)&paa; val.os.len = OGS_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 the Specific-Action AVP */ ret = fd_msg_avp_new(ogs_diam_rx_specific_action, 0, &avp); ogs_assert(ret == 0); val.i32 = OGS_DIAM_RX_SPECIFIC_ACTION_CHARGING_CORRELATION_EXCHANGE; 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(ogs_diam_rx_specific_action, 0, &avp); ogs_assert(ret == 0); val.i32 = OGS_DIAM_RX_SPECIFIC_ACTION_INDICATION_OF_LOSS_OF_BEARER; 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(ogs_diam_rx_specific_action, 0, &avp); ogs_assert(ret == 0); val.i32 = OGS_DIAM_RX_SPECIFIC_ACTION_INDICATION_OF_RECOVERY_OF_BEARER; 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(ogs_diam_rx_specific_action, 0, &avp); ogs_assert(ret == 0); val.i32 = OGS_DIAM_RX_SPECIFIC_ACTION_INDICATION_OF_RELEASE_OF_BEARER; 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(ogs_diam_rx_specific_action, 0, &avp); ogs_assert(ret == 0); val.i32 = OGS_DIAM_RX_SPECIFIC_ACTION_INDICATION_OF_ESTABLISHMENT_OF_BEARER; 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(ogs_diam_rx_specific_action, 0, &avp); ogs_assert(ret == 0); val.i32 = OGS_DIAM_RX_SPECIFIC_ACTION_IP_CAN_CHANGE; 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(ogs_diam_rx_specific_action, 0, &avp); ogs_assert(ret == 0); val.i32 = OGS_DIAM_RX_SPECIFIC_ACTION_ACCESS_NETWORK_INFO_REPORT; 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-Grace-Period AVP */ ret = fd_msg_avp_new(ogs_diam_auth_grace_period, 0, &avp); ogs_assert(ret == 0); val.i32 = 0; 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 Session-Timeout AVP */ ret = fd_msg_avp_new(ogs_diam_session_timeout, 0, &avp); ogs_assert(ret == 0); val.i32 = 7200; 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); ogs_assert(ret == 0); /* 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(pcscf_rx_reg, session, &sess_data); ogs_assert(ret == 0); ogs_assert(sess_data == NULL); /* Send the request */ ret = fd_msg_send(&req, pcscf_rx_aaa_cb, svg); ogs_assert(ret == 0); /* Increment the counter */ ogs_assert(pthread_mutex_lock(&ogs_diam_logger_self()->stats_lock) == 0); ogs_diam_logger_self()->stats.nb_sent++; ogs_assert(pthread_mutex_unlock(&ogs_diam_logger_self()->stats_lock) == 0); /* Free string memory */ ogs_free(sip_uri); } void test_rx_send_aar_ctrl(uint8_t **rx_sid, test_sess_t *sess, int id_type) { int rv; 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; test_ue_t *test_ue = NULL; char *ipstr = NULL; char *sip_uri = NULL; ogs_paa_t paa; ogs_ipsubnet_t ipsub; ogs_assert(sess); test_ue = sess->test_ue; ogs_assert(test_ue); ogs_assert(rx_sid); ogs_assert(sess->ue_ip.ipv4); ipstr = ogs_ipv4_to_string(sess->ue_ip.addr); ogs_assert(ipstr); sip_uri = ogs_msprintf("sip:%s@ims.mnc0970.mcc999.3gppnetwork.org", ipstr); ogs_assert(sip_uri); rv = ogs_ipsubnet(&ipsub, ipstr, NULL); ogs_assert(rv == OGS_OK); ogs_free(ipstr); /* Create the request */ ret = fd_msg_new(ogs_diam_rx_cmd_aar, MSGFL_ALLOC_ETEID, &req); ogs_assert(ret == 0); { struct msg_hdr * h; ret = fd_msg_hdr( req, &h ); ogs_assert(ret == 0); h->msg_appl = OGS_DIAM_RX_APPLICATION_ID; } /* Find Diameter Rx Session */ if (*rx_sid) { /* Retrieve session by Session-Id */ size_t sidlen = strlen((char *)*rx_sid); ret = fd_sess_fromsid_msg(*rx_sid, sidlen, &session, &new); ogs_assert(ret == 0); ogs_assert(new == 0); /* Add Session-Id to the message */ ret = ogs_diam_message_session_id_set(req, *rx_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 OGS_DIAM_RX_APP_SID_OPT "app_rx" ret = fd_msg_new_session(req, (os0_t)OGS_DIAM_RX_APP_SID_OPT, CONSTSTRLEN(OGS_DIAM_RX_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(pcscf_rx_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 */ *rx_sid = sess_data->sid; } /* Set Origin-Host & Origin-Realm */ ret = fd_msg_add_origin(req, 0); ogs_assert(ret == 0); /* Set the Destination-Host AVP */ ret = fd_msg_avp_new(ogs_diam_destination_host, 0, &avp); ogs_assert(ret == 0); val.os.data = TEST_PCRF_IDENTITY; val.os.len = strlen(TEST_PCRF_IDENTITY); 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 Destination-Realm AVP */ ret = fd_msg_avp_new(ogs_diam_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(ogs_diam_auth_application_id, 0, &avp); ogs_assert(ret == 0); val.i32 = OGS_DIAM_RX_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 Vendor-Specific-Application-Id AVP */ ret = ogs_diam_message_vendor_specific_appid_set( req, OGS_DIAM_RX_APPLICATION_ID); ogs_assert(ret == 0); /* Set Subscription-Id */ ret = fd_msg_avp_new(ogs_diam_rx_subscription_id, 0, &avp); ogs_assert(ret == 0); ret = fd_msg_avp_new(ogs_diam_rx_subscription_id_type, 0, &avpch1); ogs_assert(ret == 0); val.i32 = id_type; 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(ogs_diam_rx_subscription_id_data, 0, &avpch1); ogs_assert(ret == 0); if (id_type == OGS_DIAM_RX_SUBSCRIPTION_ID_TYPE_END_USER_IMSI) { val.os.data = (uint8_t *)test_ue->imsi; val.os.len = strlen(test_ue->imsi); } else if (id_type == OGS_DIAM_RX_SUBSCRIPTION_ID_TYPE_END_USER_SIP_URI) { val.os.data = (uint8_t *)sip_uri; val.os.len = strlen(sip_uri); } 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 Media-Component-Description #1 */ ret = fd_msg_avp_new(ogs_diam_rx_media_component_description, 0, &avp); ogs_assert(ret == 0); ret = fd_msg_avp_new(ogs_diam_rx_media_component_number, 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); /* Set Media-Sub-Component #1 */ ret = fd_msg_avp_new(ogs_diam_rx_media_sub_component, 0, &avpch1); ret = fd_msg_avp_new(ogs_diam_rx_flow_number, 0, &avpch2); ogs_assert(ret == 0); val.i32 = 1; 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(ogs_diam_rx_flow_description, 0, &avpch2); ogs_assert(ret == 0); #define TEST_OGS_DIAM_RX_FLOW_DESC13 \ "permit out ip from 172.30.0.144 43810 to 10.45.0.4 43810" val.os.data = (uint8_t *)TEST_OGS_DIAM_RX_FLOW_DESC13; val.os.len = strlen(TEST_OGS_DIAM_RX_FLOW_DESC13); 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(ogs_diam_rx_flow_description, 0, &avpch2); ogs_assert(ret == 0); #define TEST_OGS_DIAM_RX_FLOW_DESC14 \ "permit in ip from 10.45.0.4 43810 to 172.30.0.144 43810" val.os.data = (uint8_t *)TEST_OGS_DIAM_RX_FLOW_DESC14; val.os.len = strlen(TEST_OGS_DIAM_RX_FLOW_DESC14); 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(ogs_diam_rx_flow_description, 0, &avpch2); ogs_assert(ret == 0); #define TEST_OGS_DIAM_RX_FLOW_DESC15 \ "permit out ip from 172.30.0.144 43811 to 10.45.0.4 43811" val.os.data = (uint8_t *)TEST_OGS_DIAM_RX_FLOW_DESC15; val.os.len = strlen(TEST_OGS_DIAM_RX_FLOW_DESC15); 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(ogs_diam_rx_flow_description, 0, &avpch2); ogs_assert(ret == 0); #define TEST_OGS_DIAM_RX_FLOW_DESC16 \ "permit in ip from 10.45.0.4 43811 to 172.30.0.144 43811" val.os.data = (uint8_t *)TEST_OGS_DIAM_RX_FLOW_DESC16; val.os.len = strlen(TEST_OGS_DIAM_RX_FLOW_DESC16); 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(ogs_diam_rx_flow_usage, 0, &avpch2); ogs_assert(ret == 0); val.i32 = OGS_DIAM_RX_FLOW_USAGE_AF_SIGNALLING; 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_new(ogs_diam_rx_media_type, 0, &avpch1); ogs_assert(ret == 0); val.i32 = OGS_DIAM_RX_MEDIA_TYPE_CONTROL; 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(ogs_diam_rx_codec_data, 0, &avpch1); ogs_assert(ret == 0); #define TEST_OGS_DIAM_RX_CODEC_DATA5 \ "uplink\noffer\n" val.os.data = (uint8_t *)TEST_OGS_DIAM_RX_CODEC_DATA5; val.os.len = strlen(TEST_OGS_DIAM_RX_CODEC_DATA5) + 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(ogs_diam_rx_codec_data, 0, &avpch1); ogs_assert(ret == 0); #define TEST_OGS_DIAM_RX_CODEC_DATA6 \ "downlink\nanswer\n" val.os.data = (uint8_t *)TEST_OGS_DIAM_RX_CODEC_DATA6; val.os.len = strlen(TEST_OGS_DIAM_RX_CODEC_DATA6) + 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(ogs_diam_rx_flow_status, 0, &avpch1); ogs_assert(ret == 0); val.i32 = OGS_DIAM_RX_FLOW_STATUS_ENABLED; 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 the Specific-Action AVP */ ret = fd_msg_avp_new(ogs_diam_rx_specific_action, 0, &avp); ogs_assert(ret == 0); val.i32 = OGS_DIAM_RX_SPECIFIC_ACTION_CHARGING_CORRELATION_EXCHANGE; 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(ogs_diam_rx_specific_action, 0, &avp); ogs_assert(ret == 0); val.i32 = OGS_DIAM_RX_SPECIFIC_ACTION_INDICATION_OF_LOSS_OF_BEARER; 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(ogs_diam_rx_specific_action, 0, &avp); ogs_assert(ret == 0); val.i32 = OGS_DIAM_RX_SPECIFIC_ACTION_INDICATION_OF_RECOVERY_OF_BEARER; 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(ogs_diam_rx_specific_action, 0, &avp); ogs_assert(ret == 0); val.i32 = OGS_DIAM_RX_SPECIFIC_ACTION_INDICATION_OF_RELEASE_OF_BEARER; 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(ogs_diam_rx_specific_action, 0, &avp); ogs_assert(ret == 0); val.i32 = OGS_DIAM_RX_SPECIFIC_ACTION_INDICATION_OF_ESTABLISHMENT_OF_BEARER; 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(ogs_diam_rx_specific_action, 0, &avp); ogs_assert(ret == 0); val.i32 = OGS_DIAM_RX_SPECIFIC_ACTION_IP_CAN_CHANGE; 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(ogs_diam_rx_specific_action, 0, &avp); ogs_assert(ret == 0); val.i32 = OGS_DIAM_RX_SPECIFIC_ACTION_ACCESS_NETWORK_INFO_REPORT; 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); if (ipsub.family == AF_INET) { /* Set Framed-IP-Address */ ret = fd_msg_avp_new(ogs_diam_rx_framed_ip_address, 0, &avp); ogs_assert(ret == 0); val.os.data = (uint8_t*)ipsub.sub; val.os.len = OGS_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); } else if (ipsub.family == AF_INET6) { /* Set Framed-IPv6-Prefix */ ret = fd_msg_avp_new(ogs_diam_rx_framed_ipv6_prefix, 0, &avp); ogs_assert(ret == 0); memset(&paa, 0, sizeof(ogs_paa_t)); memcpy(paa.addr6, ipsub.sub, OGS_IPV6_LEN); paa.session_type = 0x03; #define FRAMED_IPV6_PREFIX_LENGTH 128 /* from spec document */ paa.len = FRAMED_IPV6_PREFIX_LENGTH; val.os.data = (uint8_t*)&paa; val.os.len = OGS_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 the Authroization-Lifetime AVP */ ret = fd_msg_avp_new(ogs_diam_authorization_lifetime, 0, &avp); ogs_assert(ret == 0); val.i32 = 7200; 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-Grace-Period AVP */ ret = fd_msg_avp_new(ogs_diam_auth_grace_period, 0, &avp); ogs_assert(ret == 0); val.i32 = 0; 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 Session-Timeout AVP */ ret = fd_msg_avp_new(ogs_diam_session_timeout, 0, &avp); ogs_assert(ret == 0); val.i32 = 7200; 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); ogs_assert(ret == 0); /* 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(pcscf_rx_reg, session, &sess_data); ogs_assert(ret == 0); ogs_assert(sess_data == NULL); /* Send the request */ ret = fd_msg_send(&req, pcscf_rx_aaa_cb, svg); ogs_assert(ret == 0); /* Increment the counter */ ogs_assert(pthread_mutex_lock(&ogs_diam_logger_self()->stats_lock) == 0); ogs_diam_logger_self()->stats.nb_sent++; ogs_assert(pthread_mutex_unlock(&ogs_diam_logger_self()->stats_lock) == 0); /* Free string memory */ ogs_free(sip_uri); } static void pcscf_rx_aaa_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; int result_code = 0; 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(pcscf_rx_reg, session, &sess_data); ogs_assert(ret == 0); ogs_assert(sess_data); ogs_assert((void *)sess_data == data); /* Value of Result Code */ ret = fd_msg_search_avp(*msg, ogs_diam_result_code, &avp); ogs_assert(ret == 0); if (avp) { ret = fd_msg_avp_hdr(avp, &hdr); ogs_assert(ret == 0); result_code = hdr->avp_value->i32; ogs_debug("Result Code: %d\n", hdr->avp_value->i32); } else { ret = fd_msg_search_avp(*msg, ogs_diam_experimental_result, &avp); ogs_assert(ret == 0); if (avp) { ret = fd_avp_search_avp(avp, ogs_diam_experimental_result_code, &avpch1); ogs_assert(ret == 0); if (avpch1) { ret = fd_msg_avp_hdr(avpch1, &hdr); ogs_assert(ret == 0); result_code = hdr->avp_value->i32; ogs_debug("Experimental Result Code: %d\n", result_code); } } else { ogs_error("no Result-Code"); error++; } } /* Value of Origin-Host */ ret = fd_msg_search_avp(*msg, ogs_diam_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, ogs_diam_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 (result_code != ER_DIAMETER_SUCCESS) { ogs_warn("ERROR DIAMETER Result Code(%d)", result_code); error++; goto out; } out: /* Free the message */ ogs_assert(pthread_mutex_lock(&ogs_diam_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 (ogs_diam_logger_self()->stats.nb_recv) { /* Ponderate in the avg */ ogs_diam_logger_self()->stats.avg = (ogs_diam_logger_self()->stats.avg * ogs_diam_logger_self()->stats.nb_recv + dur) / (ogs_diam_logger_self()->stats.nb_recv + 1); /* Min, max */ if (dur < ogs_diam_logger_self()->stats.shortest) ogs_diam_logger_self()->stats.shortest = dur; if (dur > ogs_diam_logger_self()->stats.longest) ogs_diam_logger_self()->stats.longest = dur; } else { ogs_diam_logger_self()->stats.shortest = dur; ogs_diam_logger_self()->stats.longest = dur; ogs_diam_logger_self()->stats.avg = dur; } if (error) ogs_diam_logger_self()->stats.nb_errs++; else ogs_diam_logger_self()->stats.nb_recv++; ogs_assert(pthread_mutex_unlock(&ogs_diam_logger_self()->stats_lock) == 0); /* Display how long it took */ if (ts.tv_nsec > sess_data->ts.tv_nsec) ogs_debug("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 ogs_debug("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(pcscf_rx_reg, session, &sess_data); ogs_assert(ret == 0); ogs_assert(sess_data == NULL); ret = fd_msg_free(*msg); ogs_assert(ret == 0); *msg = NULL; return; } static int pcscf_rx_asr_cb( struct msg **msg, struct avp *avp, struct session *sess, void *opaque, enum disp_action *act) { int rv; int ret; struct msg *ans, *qry; struct avp *avpch1, *avpch2, *avpch3; struct avp_hdr *hdr; union avp_value val; struct sess_state *sess_data = NULL; os0_t sid; size_t sidlen; ogs_assert(msg); ogs_assert(sess); ret = fd_sess_state_retrieve(pcscf_rx_reg, sess, &sess_data); ogs_assert(ret == 0); ogs_assert(sess_data); /* 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; /* Set the Auth-Application-Id AVP */ ret = fd_msg_avp_new(ogs_diam_auth_application_id, 0, &avp); ogs_assert(ret == 0); val.i32 = OGS_DIAM_RX_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 Auth-Request-Type AVP */ ret = fd_msg_avp_new(ogs_diam_auth_request_type, 0, &avp); ogs_assert(ret == 0); val.i32 = OGS_DIAM_AUTH_REQUEST_TYPE_AUTHENTICATE_ONLY; 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); /* Get Abort-Cause */ ret = fd_msg_search_avp(qry, ogs_diam_rx_abort_cause, &avp); ogs_assert(ret == 0); if (avp) { ret = fd_msg_avp_hdr(avp, &hdr); ogs_assert(ret == 0); } else { ogs_error("no_Abort-Cause "); } /* Set the Origin-Host, Origin-Realm, andResult-Code AVPs */ ret = fd_msg_rescode_set(ans, (char*)"DIAMETER_SUCCESS", NULL, NULL, 1); ogs_assert(ret == 0); /* Store this value in the session */ sid = sess_data->sid; ogs_assert(sid); ret = fd_sess_state_store(pcscf_rx_reg, sess, &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); /* Add this value to the stats */ ogs_assert(pthread_mutex_lock(&ogs_diam_logger_self()->stats_lock) == 0); ogs_diam_logger_self()->stats.nb_echoed++; ogs_assert(pthread_mutex_unlock(&ogs_diam_logger_self()->stats_lock) == 0); test_rx_send_str(sid); return 0; } void test_rx_send_str(uint8_t *rx_sid) { int rv; 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; size_t sidlen; ogs_assert(rx_sid); /* Create the request */ ret = fd_msg_new(ogs_diam_rx_cmd_str, MSGFL_ALLOC_ETEID, &req); ogs_assert(ret == 0); { struct msg_hdr * h; ret = fd_msg_hdr( req, &h ); ogs_assert(ret == 0); h->msg_appl = OGS_DIAM_RX_APPLICATION_ID; } /* Retrieve session by Session-Id */ sidlen = strlen((char*)rx_sid); ret = fd_sess_fromsid_msg(rx_sid, sidlen, &session, &new); ogs_assert(ret == 0); ogs_assert(new == 0); /* Add Session-Id to the message */ ret = ogs_diam_message_session_id_set(req, rx_sid, sidlen); ogs_assert(ret == 0); /* Save the session associated with the message */ ret = fd_msg_sess_set(req, session); /* Retrieve session state in this session */ ret = fd_sess_state_retrieve(pcscf_rx_reg, session, &sess_data); ogs_assert(ret == 0); ogs_assert(sess_data); /* Set Origin-Host & Origin-Realm */ ret = fd_msg_add_origin(req, 0); ogs_assert(ret == 0); /* Set the Destination-Host AVP */ ret = fd_msg_avp_new(ogs_diam_destination_host, 0, &avp); ogs_assert(ret == 0); val.os.data = TEST_PCRF_IDENTITY; val.os.len = strlen(TEST_PCRF_IDENTITY); 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 Destination-Realm AVP */ ret = fd_msg_avp_new(ogs_diam_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(ogs_diam_auth_application_id, 0, &avp); ogs_assert(ret == 0); val.i32 = OGS_DIAM_RX_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 the Termination-Cause AVP */ ret = fd_msg_avp_new(ogs_diam_termination_cause, 0, &avp); ogs_assert(ret == 0); val.i32 = OGS_DIAM_TERMINATION_CAUSE_DIAMETER_LOGOUT; 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); /* 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(pcscf_rx_reg, session, &sess_data); ogs_assert(ret == 0); ogs_assert(sess_data == NULL); /* Send the request */ ret = fd_msg_send(&req, pcscf_rx_sta_cb, svg); ogs_assert(ret == 0); /* Increment the counter */ ogs_assert(pthread_mutex_lock(&ogs_diam_logger_self()->stats_lock) == 0); ogs_diam_logger_self()->stats.nb_sent++; ogs_assert(pthread_mutex_unlock(&ogs_diam_logger_self()->stats_lock) == 0); } static void pcscf_rx_sta_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; int result_code = 0; 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(pcscf_rx_reg, session, &sess_data); ogs_assert(ret == 0); ogs_assert(sess_data && (void *)sess_data == data); /* Value of Result Code */ ret = fd_msg_search_avp(*msg, ogs_diam_result_code, &avp); ogs_assert(ret == 0); if (avp) { ret = fd_msg_avp_hdr(avp, &hdr); ogs_assert(ret == 0); result_code = hdr->avp_value->i32; ogs_debug("Result Code: %d\n", hdr->avp_value->i32); } else { ret = fd_msg_search_avp(*msg, ogs_diam_experimental_result, &avp); ogs_assert(ret == 0); if (avp) { ret = fd_avp_search_avp(avp, ogs_diam_experimental_result_code, &avpch1); ogs_assert(ret == 0); if (avpch1) { ret = fd_msg_avp_hdr(avpch1, &hdr); ogs_assert(ret == 0); result_code = hdr->avp_value->i32; ogs_debug("Experimental Result Code: %d\n", result_code); } } else { ogs_error("no Result-Code"); error++; } } /* Value of Origin-Host */ ret = fd_msg_search_avp(*msg, ogs_diam_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, ogs_diam_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 (result_code != ER_DIAMETER_SUCCESS) { ogs_warn("ERROR DIAMETER Result Code(%d)", result_code); error++; goto out; } out: /* Free the message */ ogs_assert(pthread_mutex_lock(&ogs_diam_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 (ogs_diam_logger_self()->stats.nb_recv) { /* Ponderate in the avg */ ogs_diam_logger_self()->stats.avg = (ogs_diam_logger_self()->stats.avg * ogs_diam_logger_self()->stats.nb_recv + dur) / (ogs_diam_logger_self()->stats.nb_recv + 1); /* Min, max */ if (dur < ogs_diam_logger_self()->stats.shortest) ogs_diam_logger_self()->stats.shortest = dur; if (dur > ogs_diam_logger_self()->stats.longest) ogs_diam_logger_self()->stats.longest = dur; } else { ogs_diam_logger_self()->stats.shortest = dur; ogs_diam_logger_self()->stats.longest = dur; ogs_diam_logger_self()->stats.avg = dur; } if (error) ogs_diam_logger_self()->stats.nb_errs++; else ogs_diam_logger_self()->stats.nb_recv++; ogs_assert(pthread_mutex_unlock(&ogs_diam_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\n", (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\n", (int)(ts.tv_sec + 1 - sess_data->ts.tv_sec), (long)(1000000000 + ts.tv_nsec - sess_data->ts.tv_nsec) / 1000); state_cleanup(sess_data, NULL, NULL); ret = fd_msg_free(*msg); ogs_assert(ret == 0); *msg = NULL; return; } int test_rx_init(void) { int ret; struct disp_when data; test_cx_init(); /* Install objects definitions for this application */ ret = ogs_diam_rx_init(); ogs_assert(ret == 0); /* Create handler for sessions */ ret = fd_sess_handler_create(&pcscf_rx_reg, state_cleanup, NULL, NULL); ogs_assert(ret == 0); /* Fallback CB if command != unexpected message received */ memset(&data, 0, sizeof(data)); data.app = ogs_diam_rx_application; ret = fd_disp_register(pcscf_rx_fb_cb, DISP_HOW_APPID, &data, NULL, &hdl_rx_fb); ogs_assert(ret == 0); /* Specific handler for Abort-Session-Request */ data.command = ogs_diam_rx_cmd_asr; ret = fd_disp_register(pcscf_rx_asr_cb, DISP_HOW_CC, &data, NULL, &hdl_rx_asr); ogs_assert(ret == 0); /* Advertise the support for the application in the peer */ ret = fd_disp_app_support(ogs_diam_rx_application, ogs_diam_vendor, 1, 0); ogs_assert(ret == 0); return 0; } void test_rx_final(void) { int ret; ret = fd_sess_handler_destroy(&pcscf_rx_reg, NULL); ogs_assert(ret == 0); if (hdl_rx_fb) (void) fd_disp_unregister(&hdl_rx_fb, NULL); if (hdl_rx_asr) (void) fd_disp_unregister(&hdl_rx_asr, NULL); }