diff options
Diffstat (limited to 'hss/HSS_Tests.ttcn')
-rw-r--r-- | hss/HSS_Tests.ttcn | 295 |
1 files changed, 295 insertions, 0 deletions
diff --git a/hss/HSS_Tests.ttcn b/hss/HSS_Tests.ttcn new file mode 100644 index 00000000..1838619b --- /dev/null +++ b/hss/HSS_Tests.ttcn @@ -0,0 +1,295 @@ +module HSS_Tests { + +import from General_Types all; +import from Osmocom_Types all; +import from Native_Functions all; +import from Misc_Helpers all; + +import from DIAMETER_Types all; +import from DIAMETER_Templates all; +import from DIAMETER_ts29_272_Templates all; +import from DIAMETER_Emulation all; + +import from Prometheus_Checker all; + +type record of hexstring SubscriberConfigs; + +modulepar { + charstring mp_hss_hostname := "127.0.0.4"; + integer mp_hss_port := 3868; + charstring mp_hss_prometheus_hostname := "127.0.0.5"; + integer mp_hss_prometheus_port := c_prometheus_default_http_port; + charstring mp_diam_local_hostname := "127.0.0.1"; + integer mp_diam_local_port := 3868; + charstring mp_diam_orig_realm := "localdomain"; + charstring mp_diam_orig_host := "mme.localdomain"; + charstring mp_diam_dest_realm := "localdomain"; + charstring mp_diam_dest_host := "hss.localdomain"; + SubscriberConfigs subscribers := { + /* Existing subscriber, ULA returns SERVICE_GRANTED */ + '001010000000000'H, + '001010000000001'H + }; +} + +/* main component, we typically have one per testcase */ +type component MTC_CT { + + /* emulated MME/SGSN */ + var DIAMETER_Emulation_CT vc_S6a; + port DIAMETER_PT S6a_UNIT; + port DIAMETEREM_PROC_PT S6a_PROC; + /* global test case guard timer (actual timeout value is set in f_init()) */ + timer T_guard; +} + +/* global altstep for global guard timer; */ +altstep as_Tguard() runs on MTC_CT { + [] T_guard.timeout { + setverdict(fail, "Timeout of T_guard"); + mtc.stop; + } +} + +type component DIAMETER_ConnHdlr_CT extends DIAMETER_ConnHdlr { + port DIAMETER_Conn_PT DIAMETER_CLIENT; + port DIAMETEREM_PROC_PT DIAMETER_PROC_CLIENT; +} + +function f_diam_connhldr_ct_main(hexstring imsi) runs on DIAMETER_ConnHdlr_CT { + var DIAMETER_ConnHdlr vc_conn_unused; + var PDU_DIAMETER msg; + var UINT32 ete_id; + + f_diameter_expect_imsi(imsi); + + while (true) { + alt { + [] DIAMETER_CLIENT.receive(PDU_DIAMETER:?) -> value msg { + DIAMETER.send(msg); + } + [] DIAMETER.receive(PDU_DIAMETER:?) -> value msg { + DIAMETER_CLIENT.send(msg); + } + [] DIAMETER_PROC_CLIENT.getcall(DIAMETEREM_register_eteid:{?,?}) -> param(ete_id, vc_conn_unused) { + DIAMETER_PROC.call(DIAMETEREM_register_eteid:{ete_id, self}) { + [] DIAMETER_PROC.getreply(DIAMETEREM_register_eteid:{?,?}) {}; + } + DIAMETER_PROC_CLIENT.reply(DIAMETEREM_register_eteid:{ete_id, vc_conn_unused}); + } + } + } +} + +/* per-session component; we typically have 1..N per testcase */ +type component Cli_Session_CT extends Prometheus_Checker_CT { + var SessionPars g_pars; + + port DIAMETER_Conn_PT S6a; + port DIAMETEREM_PROC_PT S6a_PROC; + +} +function f_diam_connhldr_expect_eteid(UINT32 ete_id) runs on Cli_Session_CT { + S6a_PROC.call(DIAMETEREM_register_eteid:{ete_id, null}) { + [] S6a_PROC.getreply(DIAMETEREM_register_eteid:{?,?}) {}; + } +} + +/* configuration data for a given Session */ +type record SessionPars { + hexstring imsi, + uint32_t s6a_next_hbh_id, + uint32_t s6a_next_ete_id +} + +template (value) SessionPars +t_SessionPars(hexstring imsi, uint32_t s6a_next_hbh_id := 1000, uint32_t s6a_next_ete_id := 22220) := { + imsi := imsi, + s6a_next_hbh_id := s6a_next_hbh_id, + s6a_next_ete_id := s6a_next_ete_id +} + +type function void_fn() runs on Cli_Session_CT; + +friend function DiameterForwardUnitdataCallback(PDU_DIAMETER msg) +runs on DIAMETER_Emulation_CT return template PDU_DIAMETER { + DIAMETER_UNIT.send(msg); + return omit; +} + +friend function f_init_diameter(charstring id) runs on MTC_CT { + var DIAMETEROps ops := { + create_cb := refers(DIAMETER_Emulation.ExpectedCreateCallback), + unitdata_cb := refers(DiameterForwardUnitdataCallback), + raw := false /* handler mode (IMSI based routing) */ + }; + var DIAMETER_conn_parameters pars; + + /* S6a setup: */ + pars := { + remote_ip := mp_hss_hostname, + remote_sctp_port := mp_hss_port, + local_ip := mp_diam_local_hostname, + local_sctp_port := mp_diam_local_port, + origin_host := mp_diam_orig_host, + origin_realm := mp_diam_orig_realm, + auth_app_id := omit, + vendor_app_id := c_DIAMETER_3GPP_S6_AID + }; + vc_S6a := DIAMETER_Emulation_CT.create(id); + map(vc_S6a:DIAMETER, system:DIAMETER_CODEC_PT); + connect(vc_S6a:DIAMETER_UNIT, self:S6a_UNIT); + connect(vc_S6a:DIAMETER_PROC, self:S6a_PROC); + vc_S6a.start(DIAMETER_Emulation.main(ops, pars, id)); + + f_diameter_wait_capability(S6a_UNIT); + /* Give some time for our emulation to get out of SUSPECT list of SUT (3 watchdog ping-pongs): + * RFC6733 sec 5.1 + * RFC3539 sec 3.4.1 [5] + * https://github.com/freeDiameter/freeDiameter/blob/master/libfdcore/p_psm.c#L49 + */ + f_sleep(1.0); +} + +private function f_init(float guard_timeout := 60.0) runs on MTC_CT { + T_guard.start(guard_timeout); + activate(as_Tguard()); + f_init_diameter(testcasename()); +} + +function f_start_handler(void_fn fn, template (omit) SessionPars pars_tmpl := omit) +runs on MTC_CT return Cli_Session_CT { + var charstring id := testcasename(); + var DIAMETER_ConnHdlr_CT vc_conn_s6a; + var Cli_Session_CT vc_conn; + var SessionPars pars; + + if (isvalue(pars_tmpl)) { + pars := valueof(pars_tmpl); + } else { + /*TODO: set default values */ + } + + vc_conn := Cli_Session_CT.create(id); + + vc_conn_s6a := DIAMETER_ConnHdlr_CT.create(id); + connect(vc_conn_s6a:DIAMETER, vc_S6a:DIAMETER_CLIENT); + connect(vc_conn_s6a:DIAMETER_PROC, vc_S6a:DIAMETER_PROC); + connect(vc_conn:S6a, vc_conn_s6a:DIAMETER_CLIENT); + connect(vc_conn:S6a_PROC, vc_conn_s6a:DIAMETER_PROC_CLIENT); + vc_conn_s6a.start(f_diam_connhldr_ct_main(pars.imsi)); + + vc_conn.start(f_handler_init(fn, pars)); + return vc_conn; +} + +private function f_handler_init(void_fn fn, SessionPars pars) +runs on Cli_Session_CT { + g_pars := valueof(pars); + f_prometheus_init(mp_hss_prometheus_hostname, mp_hss_prometheus_port); + fn.apply(); +} + +/* ULR + ULA against HSS */ +private function f_dia_ulr_ula(template (present) AVP_list ula_sub_data) runs on Cli_Session_CT { + var octetstring sess_id := char2oct("foobar"); + var PDU_DIAMETER rx_dia; + var UINT32 hbh_id := int2oct(g_pars.s6a_next_hbh_id, 4); + var UINT32 ete_id := int2oct(g_pars.s6a_next_ete_id, 4); + + /* Unlike ULR, ULA contains no IMSI. Register ete_id in DIAMETER_Emulation, + * so AIA is forwarded back to us in DIAMETER port instead of MTC_CT.DIAMETER_UNIT. + */ + f_diam_connhldr_expect_eteid(ete_id); + + /* TODO: change this into a ts_DIA_ULR */ + S6a.send(ts_DIA_ULR(g_pars.imsi, '111F11'O, sess_id, + mp_diam_orig_host, mp_diam_orig_realm, + mp_diam_dest_realm, hbh_id, ete_id)); + g_pars.s6a_next_hbh_id := g_pars.s6a_next_hbh_id + 1; + g_pars.s6a_next_ete_id := g_pars.s6a_next_ete_id + 1; + + alt { + [] S6a.receive(tr_DIA_ULA(ula_sub_data, sess_id, ?, ?, hbh_id, ete_id)) -> value rx_dia { + setverdict(pass); + } + [] S6a.receive(PDU_DIAMETER:?) -> value rx_dia { + Misc_Helpers.f_shutdown(__BFILE__, __LINE__, fail, + log2str("Received unexpected DIAMETER ", rx_dia)); + } + } +} + +/* create a session, expect it to succeed */ +private function f_TC_ulr_ula() runs on Cli_Session_CT { + var PrometheusExpects expects := valueof({ + ts_PrometheusExpect("s6a_rx_ulr", COUNTER, min := 1, max := 1), + ts_PrometheusExpect("s6a_tx_ula", COUNTER, min := 1, max := 1) + }); + var PrometheusMetrics prom_snapshot := f_prometheus_snapshot(f_prometheus_keys_from_expect(expects)); + + var template (present) AVP_list sub_data := superset( + tr_AVP_3GPP_SubscriberStatus(SERVICE_GRANTED), + tr_AVP_3GPP_SubscrRauTauTmr(?), + tr_AVP_3GPP_AMBR(?, ?), + tr_AVP_3GPP_ApnConfigProfile(superset( + tr_AVP_3GPP_ContextId(?), + tr_AVP_3GPP_AllApnConfigsIncl, + tr_AVP_3GPP_ApnConfig(?, ?, ?) + )) + ); + f_dia_ulr_ula(sub_data); + + f_prometheus_expect_from_snapshot(expects, wait_converge := true, snapshot := prom_snapshot); + setverdict(pass); +} +testcase TC_ulr_ula() runs on MTC_CT { + var Cli_Session_CT vc_conn; + var SessionPars pars := valueof(t_SessionPars(subscribers[0])); + f_init(); + vc_conn := f_start_handler(refers(f_TC_ulr_ula), pars); + vc_conn.done; +} + +/* Same as TC_ulr_ula, but done on a subscriber configured with +Subscriber-Status=1 (OPERATOR_DETERMINED_BARRING) and +Operator-Determined-Barring=7. */ +private function f_TC_ulr_ula_subscr_op_det_barring_7() runs on Cli_Session_CT { + var PrometheusExpects expects := valueof({ + ts_PrometheusExpect("s6a_rx_ulr", COUNTER, min := 1, max := 1), + ts_PrometheusExpect("s6a_tx_ula", COUNTER, min := 1, max := 1) + }); + var PrometheusMetrics prom_snapshot := f_prometheus_snapshot(f_prometheus_keys_from_expect(expects)); + + var template (present) AVP_list sub_data := superset( + tr_AVP_3GPP_SubscriberStatus(OPERATOR_DETERMINED_BARRING), + tr_AVP_3GPP_OperatorDeterminedBarring(7), + tr_AVP_3GPP_SubscrRauTauTmr(?), + tr_AVP_3GPP_AMBR(?, ?), + tr_AVP_3GPP_ApnConfigProfile(superset( + tr_AVP_3GPP_ContextId(?), + tr_AVP_3GPP_AllApnConfigsIncl, + tr_AVP_3GPP_ApnConfig(?, ?, ?) + )) + ); + f_dia_ulr_ula(sub_data); + + f_prometheus_expect_from_snapshot(expects, wait_converge := true, snapshot := prom_snapshot); + setverdict(pass); +} +testcase TC_ulr_ula_subscr_op_det_barring_7() runs on MTC_CT { + var Cli_Session_CT vc_conn; + var SessionPars pars := valueof(t_SessionPars(subscribers[1])); + f_init(); + vc_conn := f_start_handler(refers(f_TC_ulr_ula_subscr_op_det_barring_7), pars); + vc_conn.done; +} + + +control { + execute( TC_ulr_ula() ); + execute( TC_ulr_ula_subscr_op_det_barring_7() ); +} + + +} |