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