Pau Espin Pedrol | 965ac64 | 2023-10-16 18:12:45 +0200 | [diff] [blame] | 1 | module EPDG_Tests { |
| 2 | |
| 3 | import from Misc_Helpers all; |
| 4 | import from General_Types all; |
Pau Espin Pedrol | d19fdf5 | 2023-10-19 16:58:24 +0200 | [diff] [blame] | 5 | import from Native_Functions all; |
Pau Espin Pedrol | 965ac64 | 2023-10-16 18:12:45 +0200 | [diff] [blame] | 6 | import from Osmocom_Types all; |
| 7 | import from L3_Common all; |
| 8 | |
| 9 | import from IPA_Emulation all; |
| 10 | import from GSUP_Emulation all; |
| 11 | import from GSUP_Types all; |
| 12 | |
| 13 | import from DIAMETER_Types all; |
| 14 | import from DIAMETER_Templates all; |
Pau Espin Pedrol | 4669b61 | 2023-10-23 17:43:14 +0200 | [diff] [blame] | 15 | import from DIAMETER_ts29_273_Templates all; |
Pau Espin Pedrol | 965ac64 | 2023-10-16 18:12:45 +0200 | [diff] [blame] | 16 | import from DIAMETER_Emulation all; |
| 17 | |
Pau Espin Pedrol | d19fdf5 | 2023-10-19 16:58:24 +0200 | [diff] [blame] | 18 | import from GTPv2_Types all; |
| 19 | import from GTPv2_Templates all; |
Pau Espin Pedrol | 965ac64 | 2023-10-16 18:12:45 +0200 | [diff] [blame] | 20 | import from GTPv2_Emulation all; |
| 21 | |
| 22 | modulepar { |
| 23 | /* our emulated GSUP strongswan (CEAI iface) */ |
| 24 | charstring mp_gsup_local_ip := "127.0.0.100"; |
| 25 | integer mp_gsup_local_port := 0; |
| 26 | charstring mp_gsup_remote_ip := "127.0.0.1"; |
| 27 | integer mp_gsup_remote_port := 4222; |
| 28 | |
| 29 | /* our emulated HSS */ |
| 30 | charstring mp_swx_local_ip := "127.0.0.100"; |
| 31 | integer mp_swx_local_port := 3868; |
| 32 | |
| 33 | /* our emulated PGW (Diameter S6b) */ |
| 34 | charstring mp_s6b_local_ip := "127.0.0.100"; |
| 35 | integer mp_s6b_local_port := 3869; |
Pau Espin Pedrol | cdd4162 | 2023-10-23 14:38:20 +0200 | [diff] [blame] | 36 | charstring mp_s6b_remote_ip := "127.0.0.1"; |
| 37 | integer mp_s6b_remote_port := 3869; |
Pau Espin Pedrol | 965ac64 | 2023-10-16 18:12:45 +0200 | [diff] [blame] | 38 | |
| 39 | /* our emulated PGW (GTPv2C S2b) */ |
| 40 | charstring mp_s2b_local_ip := "127.0.0.100"; |
| 41 | integer mp_s2b_local_port := GTP2C_PORT; |
| 42 | charstring mp_s2b_remote_ip := "127.0.0.1"; |
| 43 | integer mp_s2b_remote_port := GTP2C_PORT; |
| 44 | |
| 45 | charstring mp_diam_realm := "localdomain"; |
| 46 | } |
| 47 | |
| 48 | |
| 49 | type component MTC_CT { |
| 50 | var DIAMETER_Emulation_CT vc_SWx; |
| 51 | port DIAMETER_PT SWx_UNIT; |
| 52 | port DIAMETEREM_PROC_PT SWx_PROC; |
| 53 | |
| 54 | var DIAMETER_Emulation_CT vc_S6b; |
| 55 | port DIAMETER_PT S6b_UNIT; |
| 56 | port DIAMETEREM_PROC_PT S6b_PROC; |
| 57 | |
| 58 | var GSUP_Emulation_CT vc_GSUP; |
| 59 | var IPA_Emulation_CT vc_GSUP_IPA; |
| 60 | port IPA_CTRL_PT GSUP_IPA_EVENT; |
| 61 | |
| 62 | var GTPv2_Emulation_CT vc_GTP2; |
| 63 | port GTP2EM_PT TEID0; |
| 64 | |
| 65 | timer g_Tguard; |
| 66 | }; |
| 67 | |
| 68 | private altstep as_Tguard() runs on MTC_CT { |
| 69 | [] g_Tguard.timeout { |
| 70 | Misc_Helpers.f_shutdown(__BFILE__, __LINE__, fail, "Tguard timeout"); |
| 71 | } |
| 72 | } |
| 73 | |
| 74 | type component DIAMETER_ConnHdlr_CT extends DIAMETER_ConnHdlr { |
| 75 | port DIAMETER_Conn_PT DIAMETER_CLIENT; |
| 76 | port DIAMETEREM_PROC_PT DIAMETER_PROC_CLIENT; |
| 77 | } |
| 78 | |
| 79 | function f_diam_connhldr_ct_main(hexstring imsi) runs on DIAMETER_ConnHdlr_CT { |
| 80 | var DIAMETER_ConnHdlr vc_conn_unused; |
| 81 | var PDU_DIAMETER msg; |
| 82 | var UINT32 ete_id; |
| 83 | |
| 84 | f_diameter_expect_imsi(imsi); |
| 85 | |
| 86 | |
| 87 | while (true) { |
| 88 | alt { |
| 89 | [] DIAMETER_CLIENT.receive(PDU_DIAMETER:?) -> value msg { |
| 90 | DIAMETER.send(msg); |
| 91 | } |
| 92 | [] DIAMETER.receive(PDU_DIAMETER:?) -> value msg { |
| 93 | DIAMETER_CLIENT.send(msg); |
| 94 | } |
| 95 | [] DIAMETER_PROC_CLIENT.getcall(DIAMETEREM_register_eteid:{?,?}) -> param(ete_id, vc_conn_unused) { |
| 96 | DIAMETER_PROC.call(DIAMETEREM_register_eteid:{ete_id, self}) { |
| 97 | [] DIAMETER_PROC.getreply(DIAMETEREM_register_eteid:{?,?}) {}; |
| 98 | } |
| 99 | DIAMETER_PROC_CLIENT.reply(DIAMETEREM_register_eteid:{ete_id, vc_conn_unused}); |
| 100 | } |
| 101 | } |
| 102 | } |
| 103 | } |
| 104 | |
| 105 | type component EPDG_ConnHdlr extends DIAMETER_ConnHdlr, GSUP_ConnHdlr, GTP2_ConnHdlr { |
| 106 | var EPDG_ConnHdlrPars g_pars; |
| 107 | |
| 108 | port DIAMETER_Conn_PT SWx; |
| 109 | port DIAMETEREM_PROC_PT SWx_PROC; |
| 110 | port DIAMETER_Conn_PT S6b; |
| 111 | port DIAMETEREM_PROC_PT S6b_PROC; |
| 112 | }; |
| 113 | |
| 114 | |
| 115 | type record EPDG_ConnHdlrPars { |
| 116 | hexstring imsi, |
Pau Espin Pedrol | 2c2e187 | 2023-10-23 18:31:03 +0200 | [diff] [blame^] | 117 | charstring apn, |
Pau Espin Pedrol | d19fdf5 | 2023-10-19 16:58:24 +0200 | [diff] [blame] | 118 | charstring ue_ip, |
| 119 | |
| 120 | /* TEI (Data) local side */ |
| 121 | OCT4 teid, |
| 122 | /* TEI (Control) local side */ |
| 123 | OCT4 teic, |
| 124 | /* TEI (Data) remote side */ |
| 125 | OCT4 teid_remote optional, |
| 126 | /* TEI (Control) remote side */ |
| 127 | OCT4 teic_remote optional, |
| 128 | |
Pau Espin Pedrol | 965ac64 | 2023-10-16 18:12:45 +0200 | [diff] [blame] | 129 | AuthVector vec optional |
| 130 | }; |
| 131 | |
| 132 | private function f_epdg_connhldr_SWx_expect_eteid(UINT32 ete_id) runs on EPDG_ConnHdlr { |
| 133 | SWx_PROC.call(DIAMETEREM_register_eteid:{ete_id, null}) { |
| 134 | [] SWx_PROC.getreply(DIAMETEREM_register_eteid:{?,?}) {}; |
| 135 | } |
| 136 | } |
| 137 | private function f_epdg_connhldr_S6b_expect_eteid(UINT32 ete_id) runs on EPDG_ConnHdlr { |
| 138 | S6b_PROC.call(DIAMETEREM_register_eteid:{ete_id, null}) { |
| 139 | [] S6b_PROC.getreply(DIAMETEREM_register_eteid:{?,?}) {}; |
| 140 | } |
| 141 | } |
| 142 | |
Pau Espin Pedrol | 965ac64 | 2023-10-16 18:12:45 +0200 | [diff] [blame] | 143 | private function f_init_pars(integer imsi_suffix := 1) |
| 144 | runs on MTC_CT return EPDG_ConnHdlrPars { |
| 145 | var EPDG_ConnHdlrPars pars := { |
| 146 | imsi := f_gen_imsi(imsi_suffix), |
Pau Espin Pedrol | 2c2e187 | 2023-10-23 18:31:03 +0200 | [diff] [blame^] | 147 | apn := "internet", |
Pau Espin Pedrol | d19fdf5 | 2023-10-19 16:58:24 +0200 | [diff] [blame] | 148 | ue_ip := "192.168.123.50", |
| 149 | teid := '00000000'O, |
| 150 | teic := '00000000'O, |
| 151 | teid_remote := omit, |
| 152 | teic_remote := omit, |
Pau Espin Pedrol | 965ac64 | 2023-10-16 18:12:45 +0200 | [diff] [blame] | 153 | vec := f_gen_auth_vec_3g() |
| 154 | }; |
| 155 | return pars; |
| 156 | } |
| 157 | |
| 158 | private function f_init_gsup(charstring id) runs on MTC_CT { |
| 159 | id := id & "-GSUP"; |
| 160 | var GsupOps ops := { |
| 161 | create_cb := refers(GSUP_Emulation.ExpectedCreateCallback) |
| 162 | }; |
| 163 | |
| 164 | vc_GSUP_IPA := IPA_Emulation_CT.create(id & "-IPA"); |
| 165 | vc_GSUP := GSUP_Emulation_CT.create(id); |
| 166 | |
| 167 | map(vc_GSUP_IPA:IPA_PORT, system:IPA_CODEC_PT); |
| 168 | connect(vc_GSUP:GSUP, vc_GSUP_IPA:IPA_GSUP_PORT); |
| 169 | /* we use this hack to get events like ASP_IPA_EVENT_UP */ |
| 170 | connect(vc_GSUP_IPA:IPA_CTRL_PORT, self:GSUP_IPA_EVENT); |
| 171 | |
| 172 | vc_GSUP.start(GSUP_Emulation.main(ops, id)); |
| 173 | vc_GSUP_IPA.start(IPA_Emulation.main_client(mp_gsup_remote_ip, mp_gsup_remote_port, |
| 174 | mp_gsup_local_ip, mp_gsup_local_port)); |
| 175 | |
| 176 | /* wait for incoming connection to GSUP port before proceeding */ |
| 177 | timer T := 10.0; |
| 178 | T.start; |
| 179 | alt { |
| 180 | [] GSUP_IPA_EVENT.receive(tr_ASP_IPA_EV(ASP_IPA_EVENT_UP)) { } |
| 181 | [] T.timeout { |
| 182 | Misc_Helpers.f_shutdown(__BFILE__, __LINE__, fail, "No connection to GSUP Port"); |
| 183 | } |
| 184 | } |
| 185 | } |
| 186 | |
| 187 | private function DiameterForwardUnitdataCallback(PDU_DIAMETER msg) |
| 188 | runs on DIAMETER_Emulation_CT return template PDU_DIAMETER { |
| 189 | DIAMETER_UNIT.send(msg); |
| 190 | return omit; |
| 191 | } |
| 192 | |
| 193 | private function f_init_diameter(charstring id) runs on MTC_CT { |
| 194 | var DIAMETEROps ops := { |
| 195 | create_cb := refers(DIAMETER_Emulation.ExpectedCreateCallback), |
| 196 | unitdata_cb := refers(DiameterForwardUnitdataCallback), |
| 197 | raw := false /* handler mode (IMSI based routing) */ |
| 198 | }; |
| 199 | var DIAMETER_conn_parameters pars; |
| 200 | |
| 201 | /* SWx setup: */ |
| 202 | pars := { |
| 203 | remote_ip := "", |
| 204 | remote_sctp_port := -1, /* server mode */ |
| 205 | local_ip := mp_swx_local_ip, |
| 206 | local_sctp_port := mp_swx_local_port, |
| 207 | origin_host := "hss." & mp_diam_realm, |
| 208 | origin_realm := mp_diam_realm, |
| 209 | auth_app_id := omit, |
| 210 | vendor_app_id := c_DIAMETER_3GPP_SWx_AID |
| 211 | }; |
| 212 | vc_SWx := DIAMETER_Emulation_CT.create(id); |
| 213 | map(vc_SWx:DIAMETER, system:DIAMETER_CODEC_PT); |
| 214 | connect(vc_SWx:DIAMETER_UNIT, self:SWx_UNIT); |
| 215 | connect(vc_SWx:DIAMETER_PROC, self:SWx_PROC); |
| 216 | vc_SWx.start(DIAMETER_Emulation.main(ops, pars, id)); |
| 217 | |
| 218 | /* S6b setup: */ |
| 219 | pars := { |
Pau Espin Pedrol | cdd4162 | 2023-10-23 14:38:20 +0200 | [diff] [blame] | 220 | remote_ip := mp_s6b_remote_ip, |
| 221 | remote_sctp_port := mp_s6b_remote_port, /* client mode */ |
Pau Espin Pedrol | 965ac64 | 2023-10-16 18:12:45 +0200 | [diff] [blame] | 222 | local_ip := mp_s6b_local_ip, |
| 223 | local_sctp_port := mp_s6b_local_port, |
Pau Espin Pedrol | cdd4162 | 2023-10-23 14:38:20 +0200 | [diff] [blame] | 224 | origin_host := "pgw." & mp_diam_realm, |
Pau Espin Pedrol | 965ac64 | 2023-10-16 18:12:45 +0200 | [diff] [blame] | 225 | origin_realm := mp_diam_realm, |
Pau Espin Pedrol | cdd4162 | 2023-10-23 14:38:20 +0200 | [diff] [blame] | 226 | auth_app_id := c_DIAMETER_3GPP_S6b_AID, |
| 227 | vendor_app_id := c_DIAMETER_3GPP_S6b_AID |
Pau Espin Pedrol | 965ac64 | 2023-10-16 18:12:45 +0200 | [diff] [blame] | 228 | }; |
| 229 | vc_S6b := DIAMETER_Emulation_CT.create(id); |
| 230 | map(vc_S6b:DIAMETER, system:DIAMETER_CODEC_PT); |
| 231 | connect(vc_S6b:DIAMETER_UNIT, self:S6b_UNIT); |
| 232 | connect(vc_S6b:DIAMETER_PROC, self:S6b_PROC); |
| 233 | vc_S6b.start(DIAMETER_Emulation.main(ops, pars, id)); |
| 234 | |
| 235 | f_diameter_wait_capability(SWx_UNIT); |
Pau Espin Pedrol | cdd4162 | 2023-10-23 14:38:20 +0200 | [diff] [blame] | 236 | f_diameter_wait_capability(S6b_UNIT); |
Pau Espin Pedrol | 965ac64 | 2023-10-16 18:12:45 +0200 | [diff] [blame] | 237 | |
| 238 | /* Give some time for our emulation to get out of SUSPECT list of SUT (3 watchdog ping-pongs): |
| 239 | * RFC6733 sec 5.1 |
| 240 | * RFC3539 sec 3.4.1 [5] |
| 241 | * https://github.com/freeDiameter/freeDiameter/blob/master/libfdcore/p_psm.c#L49 |
| 242 | */ |
| 243 | f_sleep(1.0); |
| 244 | } |
| 245 | |
| 246 | private function f_init_gtp(charstring id) runs on MTC_CT { |
| 247 | var Gtp2EmulationCfg cfg := { |
| 248 | gtpc_bind_ip := mp_s2b_local_ip, |
| 249 | gtpc_bind_port := mp_s2b_local_port, |
| 250 | gtpc_remote_ip := mp_s2b_remote_ip, |
| 251 | gtpc_remote_port := mp_s2b_remote_port, |
| 252 | sgw_role := false, |
| 253 | use_gtpu_daemon := false /* TODO: maybe use, set to true */ |
| 254 | }; |
| 255 | |
Pau Espin Pedrol | d19fdf5 | 2023-10-19 16:58:24 +0200 | [diff] [blame] | 256 | vc_GTP2 := GTPv2_Emulation_CT.create(id & "-GTPV2"); |
Pau Espin Pedrol | 965ac64 | 2023-10-16 18:12:45 +0200 | [diff] [blame] | 257 | map(vc_GTP2:GTP2C, system:GTP2C); |
| 258 | connect(vc_GTP2:TEID0, self:TEID0); |
| 259 | vc_GTP2.start(GTPv2_Emulation.main(cfg)); |
| 260 | } |
| 261 | |
| 262 | private function f_init(float t_guard := 40.0) runs on MTC_CT { |
| 263 | |
| 264 | g_Tguard.start(t_guard); |
| 265 | activate(as_Tguard()); |
| 266 | |
| 267 | f_init_gsup(testcasename()); |
| 268 | f_init_diameter(testcasename()); |
| 269 | f_init_gtp(testcasename()); |
| 270 | } |
| 271 | |
| 272 | private type function void_fn(charstring id) runs on EPDG_ConnHdlr; |
| 273 | |
| 274 | private function f_init_handler(void_fn fn, charstring id, EPDG_ConnHdlrPars pars) runs on EPDG_ConnHdlr { |
| 275 | g_pars := pars; |
| 276 | |
| 277 | /* tell GSUP dispatcher to send this IMSI to us */ |
| 278 | f_create_gsup_expect(hex2str(g_pars.imsi)); |
| 279 | |
Pau Espin Pedrol | d19fdf5 | 2023-10-19 16:58:24 +0200 | [diff] [blame] | 280 | /* tell GTPv2 dispatcher to send this IMSI to us */ |
| 281 | f_gtp2_register_imsi(g_pars.imsi); |
| 282 | |
Pau Espin Pedrol | 965ac64 | 2023-10-16 18:12:45 +0200 | [diff] [blame] | 283 | fn.apply(id); |
| 284 | } |
| 285 | |
| 286 | private function f_start_handler(void_fn fn, EPDG_ConnHdlrPars pars) |
| 287 | runs on MTC_CT return EPDG_ConnHdlr { |
| 288 | var EPDG_ConnHdlr vc_conn; |
| 289 | var charstring id := testcasename(); |
| 290 | var DIAMETER_ConnHdlr_CT vc_conn_swx, vc_conn_s6b; |
| 291 | |
| 292 | vc_conn := EPDG_ConnHdlr.create(id); |
| 293 | |
Pau Espin Pedrol | d19fdf5 | 2023-10-19 16:58:24 +0200 | [diff] [blame] | 294 | /* GSUP */ |
Pau Espin Pedrol | 965ac64 | 2023-10-16 18:12:45 +0200 | [diff] [blame] | 295 | connect(vc_conn:GSUP, vc_GSUP:GSUP_CLIENT); |
| 296 | connect(vc_conn:GSUP_PROC, vc_GSUP:GSUP_PROC); |
| 297 | |
Pau Espin Pedrol | d19fdf5 | 2023-10-19 16:58:24 +0200 | [diff] [blame] | 298 | /* GTP2 */ |
| 299 | connect(vc_conn:GTP2, vc_GTP2:CLIENT); |
| 300 | connect(vc_conn:GTP2_PROC, vc_GTP2:CLIENT_PROC); |
| 301 | |
Pau Espin Pedrol | 965ac64 | 2023-10-16 18:12:45 +0200 | [diff] [blame] | 302 | /* SWx */ |
| 303 | vc_conn_swx := DIAMETER_ConnHdlr_CT.create(id); |
| 304 | connect(vc_conn_swx:DIAMETER, vc_SWx:DIAMETER_CLIENT); |
| 305 | connect(vc_conn_swx:DIAMETER_PROC, vc_SWx:DIAMETER_PROC); |
| 306 | connect(vc_conn:SWx, vc_conn_swx:DIAMETER_CLIENT); |
| 307 | connect(vc_conn:SWx_PROC, vc_conn_swx:DIAMETER_PROC_CLIENT); |
| 308 | vc_conn_swx.start(f_diam_connhldr_ct_main(pars.imsi)); |
| 309 | |
| 310 | /* S6b */ |
| 311 | vc_conn_s6b := DIAMETER_ConnHdlr_CT.create(id); |
| 312 | connect(vc_conn_s6b:DIAMETER, vc_S6b:DIAMETER_CLIENT); |
| 313 | connect(vc_conn_s6b:DIAMETER_PROC, vc_S6b:DIAMETER_PROC); |
| 314 | connect(vc_conn:S6b, vc_conn_s6b:DIAMETER_CLIENT); |
| 315 | connect(vc_conn:S6b_PROC, vc_conn_s6b:DIAMETER_PROC_CLIENT); |
| 316 | vc_conn_s6b.start(f_diam_connhldr_ct_main(pars.imsi)); |
| 317 | |
| 318 | vc_conn.start(f_init_handler(fn, id, pars)); |
| 319 | return vc_conn; |
| 320 | } |
| 321 | |
| 322 | /* Diameter SWx MAR + MAA. */ |
| 323 | private altstep as_DIA_SWx_MA_success() runs on EPDG_ConnHdlr { |
| 324 | var PDU_DIAMETER rx_dia; |
| 325 | var template (omit) AVP avp; |
| 326 | var octetstring sess_id; |
| 327 | var template (value) GenericAVP sip_auth_data_item; |
| 328 | [] SWx.receive(tr_DIA_SWx_MAR(g_pars.imsi)) -> value rx_dia { |
| 329 | avp := f_DIAMETER_get_avp(rx_dia, c_AVP_Code_BASE_NONE_Session_Id); |
| 330 | sess_id := valueof(avp.avp_data.avp_BASE_NONE_Session_Id); |
| 331 | sip_auth_data_item := ts_AVP_3GPP_SIPAuthDataItem(0, |
| 332 | g_pars.vec.rand, |
| 333 | g_pars.vec.ik, |
| 334 | g_pars.vec.ck, |
| 335 | g_pars.vec.autn, |
| 336 | g_pars.vec.auts); |
| 337 | /* Send MAA to translator; expect it to show up on GSUP side */ |
| 338 | SWx.send(ts_DIA_SWx_MAA(g_pars.imsi, sip_auth_data_item, |
| 339 | sess_id := sess_id, |
| 340 | hbh_id := rx_dia.hop_by_hop_id, |
| 341 | ete_id := rx_dia.end_to_end_id)); |
| 342 | setverdict(pass); |
| 343 | } |
| 344 | [] SWx.receive(PDU_DIAMETER:?) -> value rx_dia { |
| 345 | Misc_Helpers.f_shutdown(__BFILE__, __LINE__, fail, log2str("Unexpected Diameter msg rx: ", rx_dia)); |
| 346 | } |
| 347 | } |
| 348 | |
| 349 | /* Diameter SWx SAR + SAA. */ |
| 350 | private altstep as_DIA_SWx_SA_success() runs on EPDG_ConnHdlr { |
| 351 | var PDU_DIAMETER rx_dia; |
| 352 | var template (omit) AVP avp; |
| 353 | var octetstring sess_id; |
| 354 | [] SWx.receive(tr_DIA_SWx_SAR(g_pars.imsi)) -> value rx_dia { |
| 355 | avp := f_DIAMETER_get_avp(rx_dia, c_AVP_Code_BASE_NONE_Session_Id); |
| 356 | sess_id := valueof(avp.avp_data.avp_BASE_NONE_Session_Id); |
| 357 | /* Send SAA to translator; expect it to show up on GSUP side */ |
| 358 | SWx.send(ts_DIA_SWx_SAA(g_pars.imsi, |
| 359 | sess_id := sess_id, |
| 360 | hbh_id := rx_dia.hop_by_hop_id, |
| 361 | ete_id := rx_dia.end_to_end_id)); |
| 362 | setverdict(pass); |
| 363 | } |
| 364 | [] SWx.receive(PDU_DIAMETER:?) -> value rx_dia { |
| 365 | Misc_Helpers.f_shutdown(__BFILE__, __LINE__, fail, log2str("Unexpected Diameter msg rx: ", rx_dia)); |
| 366 | } |
| 367 | } |
| 368 | |
Pau Espin Pedrol | cdd4162 | 2023-10-23 14:38:20 +0200 | [diff] [blame] | 369 | /* Send AAR as PGW to AAA server, expect back AAA */ |
| 370 | private function f_S6b_AA_success() runs on EPDG_ConnHdlr { |
| 371 | var PDU_DIAMETER rx_dia; |
| 372 | var UINT32 hbh_id := f_rnd_octstring(4); |
| 373 | var UINT32 ete_id := f_rnd_octstring(4); |
| 374 | |
| 375 | /* Unlike AAR, AAA contains no IMSI. Register ete_id in DIAMETER_Emulation, |
| 376 | * so AIA is forwarded back to us in DIAMETER port instead of MTC_CT.DIAMETER_UNIT. |
| 377 | */ |
| 378 | f_epdg_connhldr_S6b_expect_eteid(ete_id); |
| 379 | |
Pau Espin Pedrol | 2c2e187 | 2023-10-23 18:31:03 +0200 | [diff] [blame^] | 380 | S6b.send(ts_DIA_S6b_AAR(g_pars.imsi, |
| 381 | int2oct(DIA_TS29_373_MIP6_Feature_Vector_GTPv2_SUPPORTED, 8), |
| 382 | g_pars.apn, hbh_id := hbh_id, ete_id := ete_id)); |
Pau Espin Pedrol | cdd4162 | 2023-10-23 14:38:20 +0200 | [diff] [blame] | 383 | alt { |
| 384 | [] S6b.receive(tr_DIA_S6b_AAA) -> value rx_dia {} |
| 385 | [] S6b.receive(PDU_DIAMETER:?) -> value rx_dia { |
| 386 | Misc_Helpers.f_shutdown(__BFILE__, __LINE__, fail, log2str("Unexpected Diameter S6b msg rx: ", rx_dia)); |
| 387 | } |
| 388 | } |
| 389 | } |
| 390 | |
Pau Espin Pedrol | d19fdf5 | 2023-10-19 16:58:24 +0200 | [diff] [blame] | 391 | /* Diameter SWx SAR + SAA. */ |
| 392 | private altstep as_GTP2C_CreateSession_success() runs on EPDG_ConnHdlr { |
| 393 | var PDU_GTPCv2 rx_msg; |
| 394 | var template (value) FullyQualifiedTEID fteid_c_ie, fteid_u_ie; |
| 395 | var template (value) PDN_AddressAllocation paa; |
| 396 | |
| 397 | [] GTP2.receive(tr_GTP2C_CreateSessionReq(g_pars.imsi)) -> value rx_msg { |
| 398 | /* TODO: parse TEIC and TEID and store it in g_pars.remote_tei{c,d} */ |
| 399 | |
| 400 | /* allocate + register TEID-C on local side */ |
| 401 | g_pars.teic := f_gtp2_allocate_teid(); |
| 402 | g_pars.teid := g_pars.teic; |
| 403 | |
Pau Espin Pedrol | cdd4162 | 2023-10-23 14:38:20 +0200 | [diff] [blame] | 404 | /* Upon rx of CreateSession, emulate PGW asking the AAA server. */ |
| 405 | f_S6b_AA_success(); |
| 406 | |
Pau Espin Pedrol | d19fdf5 | 2023-10-19 16:58:24 +0200 | [diff] [blame] | 407 | fteid_c_ie := ts_GTP2C_FTEID(FTEID_IF_S2b_ePDG_GTPC, g_pars.teic, 0, |
| 408 | f_inet_addr(mp_s2b_local_ip), omit); |
| 409 | fteid_u_ie := ts_GTP2C_FTEID(FTEID_IF_S2bU_ePDG_GTPU, g_pars.teid, 2, |
| 410 | f_inet_addr(mp_s2b_local_ip), omit); |
| 411 | paa := ts_GTP2C_PdnAddrAlloc_v4(f_inet_addr(g_pars.ue_ip)); |
Pau Espin Pedrol | cdd4162 | 2023-10-23 14:38:20 +0200 | [diff] [blame] | 412 | |
Pau Espin Pedrol | d19fdf5 | 2023-10-19 16:58:24 +0200 | [diff] [blame] | 413 | GTP2.send(ts_GTP2C_CreateSessionResp({ fteid_c_ie }, paa)); |
| 414 | setverdict(pass); |
| 415 | } |
| 416 | [] GTP2.receive(PDU_GTPCv2:?) -> value rx_msg { |
| 417 | Misc_Helpers.f_shutdown(__BFILE__, __LINE__, fail, log2str("Unexpected GTP2C msg rx: ", rx_msg)); |
| 418 | } |
| 419 | } |
| 420 | private function f_GTP2C_CreateSession_success() runs on EPDG_ConnHdlr { |
| 421 | f_gtp2_register_udmsg('20'O); |
| 422 | as_GTP2C_CreateSession_success(); |
| 423 | } |
| 424 | |
Pau Espin Pedrol | 965ac64 | 2023-10-16 18:12:45 +0200 | [diff] [blame] | 425 | /* GSUP AuthInfo Req + Resp, triggers SWx MAR + MAA. */ |
| 426 | private function f_GSUP_AI_success() runs on EPDG_ConnHdlr { |
| 427 | var GSUP_PDU rx_gsup; |
| 428 | var template (present) GSUP_IE auth_tuple_ie := tr_GSUP_IE_AuthTuple3G(g_pars.vec.rand, |
| 429 | g_pars.vec.ik, |
| 430 | g_pars.vec.ck, |
| 431 | g_pars.vec.autn, |
| 432 | g_pars.vec.rand & g_pars.vec.auts); |
| 433 | |
| 434 | GSUP.send(ts_GSUP_SAI_REQ(g_pars.imsi)); |
| 435 | as_DIA_SWx_MA_success(); |
| 436 | /* Expect a positive response back to the translator; expect AIA */ |
| 437 | alt { |
| 438 | [] GSUP.receive(tr_GSUP_SAI_RES(g_pars.imsi, auth_tuple_ie)); |
| 439 | [] GSUP.receive(GSUP_PDU:?) -> value rx_gsup { |
| 440 | Misc_Helpers.f_shutdown(__BFILE__, __LINE__, fail, log2str("Unexpected GSUP msg rx: ", rx_gsup)); |
| 441 | } |
| 442 | } |
| 443 | setverdict(pass); |
| 444 | } |
| 445 | |
| 446 | /* GSUP LU Req + Resp, triggers SWx SAR + SAA (Server Assignment). */ |
| 447 | private function f_GSUP_LU_success() runs on EPDG_ConnHdlr { |
| 448 | var GSUP_PDU rx_gsup; |
| 449 | var template octetstring destination_name := *; |
| 450 | GSUP.send(ts_GSUP_UL_REQ(g_pars.imsi)); |
| 451 | as_DIA_SWx_SA_success(); |
Pau Espin Pedrol | d19fdf5 | 2023-10-19 16:58:24 +0200 | [diff] [blame] | 452 | /* Expect a positive response back to the translator */ |
Pau Espin Pedrol | 965ac64 | 2023-10-16 18:12:45 +0200 | [diff] [blame] | 453 | alt { |
| 454 | [] GSUP.receive(tr_GSUP_UL_RES(g_pars.imsi, destination_name)); |
| 455 | [] GSUP.receive(GSUP_PDU:?) -> value rx_gsup { |
| 456 | Misc_Helpers.f_shutdown(__BFILE__, __LINE__, fail, log2str("Unexpected GSUP msg rx: ", rx_gsup)); |
| 457 | } |
| 458 | } |
| 459 | setverdict(pass); |
| 460 | } |
| 461 | |
Pau Espin Pedrol | d19fdf5 | 2023-10-19 16:58:24 +0200 | [diff] [blame] | 462 | /* GSUP TunnelEPDG Tunnel Req + Resp, triggers S2b CreateSession Req + Response. */ |
| 463 | private function f_GSUP_EPDGTunnel_success() runs on EPDG_ConnHdlr { |
| 464 | var GSUP_PDU rx_gsup; |
| 465 | GSUP.send(ts_GSUP_EPDGTunnel_REQ(g_pars.imsi)); |
| 466 | f_GTP2C_CreateSession_success(); |
| 467 | /* Expect a positive response back to the translator; */ |
| 468 | alt { |
| 469 | [] GSUP.receive(tr_GSUP_EPDGTunnel_RES(g_pars.imsi)); |
| 470 | [] GSUP.receive(GSUP_PDU:?) -> value rx_gsup { |
| 471 | Misc_Helpers.f_shutdown(__BFILE__, __LINE__, fail, log2str("Unexpected GSUP msg rx: ", rx_gsup)); |
| 472 | } |
| 473 | } |
| 474 | setverdict(pass); |
| 475 | } |
| 476 | |
Pau Espin Pedrol | 965ac64 | 2023-10-16 18:12:45 +0200 | [diff] [blame] | 477 | private function f_TC_authinfo_normal(charstring id) runs on EPDG_ConnHdlr { |
| 478 | f_GSUP_AI_success(); |
| 479 | f_GSUP_LU_success(); |
Pau Espin Pedrol | d19fdf5 | 2023-10-19 16:58:24 +0200 | [diff] [blame] | 480 | f_GSUP_EPDGTunnel_success(); |
Pau Espin Pedrol | 965ac64 | 2023-10-16 18:12:45 +0200 | [diff] [blame] | 481 | } |
| 482 | |
| 483 | testcase TC_authinfo_normal() runs on MTC_CT { |
| 484 | var EPDG_ConnHdlrPars pars := f_init_pars(); |
| 485 | var EPDG_ConnHdlr vc_conn; |
| 486 | f_init(); |
| 487 | vc_conn := f_start_handler(refers(f_TC_authinfo_normal), pars); |
| 488 | vc_conn.done; |
| 489 | setverdict(pass); |
| 490 | } |
| 491 | |
| 492 | control { |
| 493 | execute ( TC_authinfo_normal() ); |
| 494 | } |
| 495 | |
| 496 | } |