Harald Welte | c98d9a9 | 2019-08-22 10:53:23 +0200 | [diff] [blame^] | 1 | module DIA2GSUP_Tests { |
| 2 | |
| 3 | import from General_Types all; |
| 4 | import from Osmocom_Types all; |
| 5 | import from L3_Common all; |
| 6 | |
| 7 | import from IPA_Emulation all; |
| 8 | import from GSUP_Emulation all; |
| 9 | import from GSUP_Types all; |
| 10 | |
| 11 | import from DIAMETER_Types all; |
| 12 | import from DIAMETER_Templates all; |
| 13 | import from DIAMETER_Emulation all; |
| 14 | |
| 15 | type component MTC_CT { |
| 16 | var DIAMETER_Emulation_CT vc_DIAMETER; |
| 17 | port DIAMETER_PT DIAMETER_UNIT; |
| 18 | port DIAMETEREM_PROC_PT DIAMETER_PROC; |
| 19 | |
| 20 | var GSUP_Emulation_CT vc_GSUP; |
| 21 | var IPA_Emulation_CT vc_GSUP_IPA; |
| 22 | port IPA_CTRL_PT GSUP_IPA_EVENT; |
| 23 | |
| 24 | }; |
| 25 | |
| 26 | type component D2G_ConnHdlr extends DIAMETER_ConnHdlr, GSUP_ConnHdlr { |
| 27 | var D2G_ConnHdlrPars g_pars; |
| 28 | timer g_Tguard := 30.0; |
| 29 | }; |
| 30 | |
| 31 | type record D2G_ConnHdlrPars { |
| 32 | hexstring imsi, |
| 33 | AuthVector vec optional |
| 34 | }; |
| 35 | |
| 36 | private function f_init_pars(integer imsi_suffix) |
| 37 | runs on MTC_CT return D2G_ConnHdlrPars { |
| 38 | var D2G_ConnHdlrPars pars := { |
| 39 | imsi := f_gen_imsi(imsi_suffix), |
| 40 | vec := f_gen_auth_vec_3g() |
| 41 | }; |
| 42 | return pars; |
| 43 | } |
| 44 | |
| 45 | |
| 46 | modulepar { |
| 47 | /* our emulated GSUP HLR */ |
| 48 | charstring mp_hlr_ip := "127.0.0.1"; |
| 49 | integer mp_hlr_port := 4222; |
| 50 | |
| 51 | /* our emulated MME */ |
| 52 | charstring mp_s6_local_ip := "127.0.0.100"; |
| 53 | integer mp_s6_local_port := 3868; |
| 54 | |
| 55 | /* IUT behaving as HSS */ |
| 56 | charstring mp_hss_ip := "127.0.0.4"; |
| 57 | integer mp_hss_port := 3868; |
| 58 | |
| 59 | charstring mp_diam_realm := "localdomain"; |
| 60 | } |
| 61 | |
| 62 | private altstep as_Tguard() runs on D2G_ConnHdlr { |
| 63 | [] g_Tguard.timeout { |
| 64 | setverdict(fail, "Tguard timeout"); |
| 65 | mtc.stop; |
| 66 | } |
| 67 | } |
| 68 | |
| 69 | private function f_init_gsup(charstring id) runs on MTC_CT { |
| 70 | id := id & "-GSUP"; |
| 71 | var GsupOps ops := { |
| 72 | create_cb := refers(GSUP_Emulation.ExpectedCreateCallback) |
| 73 | }; |
| 74 | |
| 75 | vc_GSUP_IPA := IPA_Emulation_CT.create(id & "-IPA"); |
| 76 | vc_GSUP := GSUP_Emulation_CT.create(id); |
| 77 | |
| 78 | map(vc_GSUP_IPA:IPA_PORT, system:IPA_CODEC_PT); |
| 79 | connect(vc_GSUP:GSUP, vc_GSUP_IPA:IPA_GSUP_PORT); |
| 80 | /* we use this hack to get events like ASP_IPA_EVENT_UP */ |
| 81 | connect(vc_GSUP_IPA:IPA_CTRL_PORT, self:GSUP_IPA_EVENT); |
| 82 | |
| 83 | vc_GSUP.start(GSUP_Emulation.main(ops, id)); |
| 84 | vc_GSUP_IPA.start(IPA_Emulation.main_server(mp_hlr_ip, mp_hlr_port)); |
| 85 | |
| 86 | /* wait for incoming connection to GSUP port before proceeding */ |
| 87 | timer T := 10.0; |
| 88 | T.start; |
| 89 | alt { |
| 90 | [] GSUP_IPA_EVENT.receive(tr_ASP_IPA_EV(ASP_IPA_EVENT_UP)) { } |
| 91 | [] T.timeout { |
| 92 | setverdict(fail, "No connection to GSUP Port"); |
| 93 | mtc.stop; |
| 94 | } |
| 95 | } |
| 96 | } |
| 97 | |
| 98 | private function f_init_handler(D2G_ConnHdlrPars pars, float t_guard := 20.0) runs on D2G_ConnHdlr { |
| 99 | g_pars := pars; |
| 100 | activate(as_Tguard()); |
| 101 | g_Tguard.start(t_guard); |
| 102 | |
| 103 | /* tell GSUP dispatcher to send this IMSI to us */ |
| 104 | f_create_gsup_expect(hex2str(g_pars.imsi)); |
| 105 | } |
| 106 | |
| 107 | private type function void_fn(charstring id, D2G_ConnHdlrPars pars) runs on D2G_ConnHdlr; |
| 108 | |
| 109 | private function f_start_handler_with_pars(void_fn fn, D2G_ConnHdlrPars pars) |
| 110 | runs on MTC_CT return D2G_ConnHdlr { |
| 111 | var D2G_ConnHdlr vc_conn; |
| 112 | var charstring id := testcasename(); |
| 113 | |
| 114 | vc_conn := D2G_ConnHdlr.create(id); |
| 115 | |
| 116 | connect(vc_conn:GSUP, vc_GSUP:GSUP_CLIENT); |
| 117 | connect(vc_conn:GSUP_PROC, vc_GSUP:GSUP_PROC); |
| 118 | |
| 119 | connect(vc_conn:DIAMETER, vc_DIAMETER:DIAMETER_CLIENT); |
| 120 | connect(vc_conn:DIAMETER_PROC, vc_DIAMETER:DIAMETER_PROC); |
| 121 | |
| 122 | DIAMETER_UNIT.receive(DiameterCapabilityExchgInd:?); |
| 123 | |
| 124 | vc_conn.start(derefers(fn)(id, pars)); |
| 125 | return vc_conn; |
| 126 | } |
| 127 | |
| 128 | private function DiameterForwardUnitdataCallback(PDU_DIAMETER msg) |
| 129 | runs on DIAMETER_Emulation_CT return template PDU_DIAMETER { |
| 130 | DIAMETER_UNIT.send(msg); |
| 131 | return omit; |
| 132 | } |
| 133 | |
| 134 | private function f_init_diameter(charstring id) runs on MTC_CT { |
| 135 | var DIAMETEROps ops := { |
| 136 | create_cb := refers(DIAMETER_Emulation.ExpectedCreateCallback), |
| 137 | unitdata_cb := refers(DiameterForwardUnitdataCallback), |
| 138 | raw := false /* handler mode (IMSI based routing) */ |
| 139 | }; |
| 140 | var DIAMETER_conn_parameters pars := { |
| 141 | remote_ip := mp_hss_ip, |
| 142 | remote_sctp_port := mp_hss_port, |
| 143 | local_ip := mp_s6_local_ip, |
| 144 | local_sctp_port := mp_s6_local_port, |
| 145 | origin_host := "hss." & mp_diam_realm, |
| 146 | origin_realm := mp_diam_realm, |
| 147 | auth_app_id := omit, |
| 148 | vendor_app_id := c_DIAMETER_3GPP_S6_AID |
| 149 | }; |
| 150 | vc_DIAMETER := DIAMETER_Emulation_CT.create(id); |
| 151 | map(vc_DIAMETER:DIAMETER, system:DIAMETER_CODEC_PT); |
| 152 | connect(vc_DIAMETER:DIAMETER_UNIT, self:DIAMETER_UNIT); |
| 153 | connect(vc_DIAMETER:DIAMETER_PROC, self:DIAMETER_PROC); |
| 154 | vc_DIAMETER.start(DIAMETER_Emulation.main(ops, pars, id)); |
| 155 | } |
| 156 | |
| 157 | private function f_init() runs on MTC_CT { |
| 158 | f_init_gsup(testcasename()); |
| 159 | f_init_diameter(testcasename()); |
| 160 | } |
| 161 | |
| 162 | |
| 163 | private function f_DIA_AI() runs on D2G_ConnHdlr { |
| 164 | var PDU_DIAMETER rx_dia; |
| 165 | var UINT32 hbh_id := f_rnd_octstring(4); |
| 166 | var UINT32 ete_id := f_rnd_octstring(4); |
| 167 | var octetstring sess_id := char2oct("foobar"); |
| 168 | var OCT3 vplmn := '00F110'O; |
| 169 | |
| 170 | /* Unlike AIR, AIA contains no IMSI. Register ete_id in DIAMETER_Emulation, |
| 171 | * so AIA is forwarded back to us in DIAMETER port instead of MTC_CT.DIAMETER_UNIT. |
| 172 | */ |
| 173 | f_diameter_expect_eteid(ete_id); |
| 174 | |
| 175 | /* Send AIR to translator; expect it to show up on GSUP side */ |
| 176 | DIAMETER.send(ts_DIA_AIR(g_pars.imsi, vplmn, sess_id, hbh_id := hbh_id, ete_id := ete_id)); |
| 177 | GSUP.receive(tr_GSUP_SAI_REQ(g_pars.imsi)); |
| 178 | |
| 179 | /* Send a positive response back to the translator; expect AIA */ |
| 180 | var GSUP_IE auth_tuple := valueof(ts_GSUP_IE_AuthTuple2G3G(g_pars.vec.rand, |
| 181 | g_pars.vec.sres, |
| 182 | g_pars.vec.kc, |
| 183 | g_pars.vec.ik, |
| 184 | g_pars.vec.ck, |
| 185 | g_pars.vec.autn, |
| 186 | g_pars.vec.res)); |
| 187 | GSUP.send(ts_GSUP_SAI_RES(g_pars.imsi, auth_tuple)); |
| 188 | |
| 189 | alt { |
| 190 | /* Unlike AIR, AIA contains no IMSI, hence it is received in DIAMETER_UNIT: */ |
| 191 | [] DIAMETER.receive(tr_DIA_AIA) { |
| 192 | setverdict(pass); |
| 193 | } |
| 194 | } |
| 195 | } |
| 196 | |
| 197 | private function f_TC_authinfo_normal(charstring id, D2G_ConnHdlrPars pars) runs on D2G_ConnHdlr { |
| 198 | f_init_handler(pars); |
| 199 | f_DIA_AI(); |
| 200 | } |
| 201 | |
| 202 | testcase TC_authinfo_normal() runs on MTC_CT { |
| 203 | var D2G_ConnHdlrPars pars := f_init_pars(1); |
| 204 | var D2G_ConnHdlr vc_conn; |
| 205 | f_init(); |
| 206 | f_sleep(10.0); |
| 207 | vc_conn := f_start_handler_with_pars(refers(f_TC_authinfo_normal), pars); |
| 208 | vc_conn.done; |
| 209 | setverdict(pass); |
| 210 | } |
| 211 | |
| 212 | |
| 213 | control { |
| 214 | execute ( TC_authinfo_normal() ); |
| 215 | } |
| 216 | |
| 217 | } |