543 lines
16 KiB
C
543 lines
16 KiB
C
/*
|
|
* Copyright (C) 2019-2023 by Sukchan Lee <acetcom@gmail.com>
|
|
*
|
|
* 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 <https://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#if !defined(OGS_TEST_INSIDE) && !defined(OGS_TEST_COMPILATION)
|
|
#error "This header cannot be included directly."
|
|
#endif
|
|
|
|
#ifndef TEST_COMMON_CONTEXT_H
|
|
#define TEST_COMMON_CONTEXT_H
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
#define TEST_MME_IPV4 "127.0.0.1"
|
|
#define TEST_ENB_IPV4 "127.0.0.5"
|
|
#define TEST_AMF_IPV4 TEST_MME_IPV4
|
|
#define TEST_GNB_IPV4 TEST_ENB_IPV4
|
|
|
|
#define TEST_PING_IPV4 "10.45.0.1"
|
|
#define TEST_PING_IPV6 "2001:db8:cafe::1"
|
|
|
|
#define TEST_MSISDN "491725670014"
|
|
#define TEST_ADDITIONAL_MSISDN "491725670015"
|
|
|
|
#define TEST_HSS_IDENTITY "hss.localdomain"
|
|
#define TEST_PCRF_IDENTITY "pcrf.localdomain"
|
|
#define TEST_SMF_IDENTITY "smf.localdomain"
|
|
|
|
typedef struct test_context_s {
|
|
uint16_t ngap_port; /* Default NGAP Port */
|
|
ogs_list_t ngap_list; /* AMF NGAP IPv4 Server List */
|
|
ogs_list_t ngap_list6; /* AMF NGAP IPv6 Server List */
|
|
ogs_sockaddr_t *ngap_addr; /* AMF NGAP IPv4 Address */
|
|
ogs_sockaddr_t *ngap_addr6; /* AMF NGAP IPv6 Address */
|
|
|
|
uint16_t s1ap_port; /* Default S1AP Port */
|
|
ogs_list_t s1ap_list; /* MME S1AP IPv4 Server List */
|
|
ogs_list_t s1ap_list6; /* MME S1AP IPv6 Server List */
|
|
ogs_sockaddr_t *s1ap_addr; /* MME S1AP IPv4 Address */
|
|
ogs_sockaddr_t *s1ap_addr6; /* MME S1AP IPv6 Address */
|
|
|
|
ogs_sockaddr_t *gnb1_addr;
|
|
ogs_sockaddr_t *gnb1_addr6;
|
|
ogs_sockaddr_t *gnb2_addr;
|
|
ogs_sockaddr_t *gnb2_addr6;
|
|
|
|
uint32_t gtpc_port; /* SMF GTPC local port */
|
|
ogs_list_t gtpc_list; /* SMF GTPC Client List */
|
|
|
|
/* 5G PLMN Support */
|
|
uint8_t num_of_plmn_support;
|
|
struct {
|
|
ogs_plmn_id_t plmn_id;
|
|
int num_of_s_nssai;
|
|
ogs_s_nssai_t s_nssai[OGS_MAX_NUM_OF_SLICE_SUPPORT];
|
|
} plmn_support[OGS_MAX_NUM_OF_PLMN];
|
|
|
|
/* Served EPC TAI */
|
|
uint8_t num_of_e_served_tai;
|
|
struct {
|
|
ogs_eps_tai0_list_t list0;
|
|
ogs_eps_tai1_list_t list1;
|
|
ogs_eps_tai2_list_t list2;
|
|
} e_served_tai[OGS_MAX_NUM_OF_SUPPORTED_TA];
|
|
|
|
ogs_eps_tai_t e_tai;
|
|
|
|
/* Served 5GC TAI */
|
|
uint8_t num_of_nr_served_tai;
|
|
struct {
|
|
ogs_5gs_tai0_list_t list0;
|
|
ogs_5gs_tai1_list_t list1;
|
|
ogs_5gs_tai2_list_t list2;
|
|
} nr_served_tai[OGS_MAX_NUM_OF_SUPPORTED_TA];
|
|
|
|
ogs_5gs_tai_t nr_tai;
|
|
ogs_nr_cgi_t nr_cgi;
|
|
|
|
ogs_list_t test_ue_list;
|
|
} test_context_t;
|
|
|
|
typedef struct test_sess_s test_sess_t;
|
|
|
|
typedef struct test_initial_ue_param_s {
|
|
struct {
|
|
ED8(uint8_t gummei_id:1;,
|
|
uint8_t gummei_type:1;,
|
|
uint8_t spare1:1;,
|
|
uint8_t spare2:1;,
|
|
uint8_t spare3:1;,
|
|
uint8_t spare4:1;,
|
|
uint8_t spare5:1;,
|
|
uint8_t spare6:1;)
|
|
};
|
|
struct {
|
|
uint16_t mme_gid;
|
|
uint8_t mme_code;
|
|
uint8_t type;
|
|
} gummei;
|
|
} __attribute__ ((packed)) test_initial_ue_param_t;
|
|
|
|
typedef struct test_registration_request_param_s {
|
|
struct {
|
|
ED7(uint8_t spare1:1;,
|
|
uint8_t guti:1;,
|
|
uint8_t gmm_capability:1;,
|
|
uint8_t requested_nssai:1;,
|
|
uint8_t last_visited_registered_tai:1;,
|
|
uint8_t ue_usage_setting:1;,
|
|
uint8_t update_type:1;)
|
|
};
|
|
struct {
|
|
ED5(uint8_t uplink_data_status:1;,
|
|
uint8_t pdu_session_status:1;,
|
|
uint8_t allowed_pdu_session_status:1;,
|
|
uint8_t s1_ue_network_capability:1;,
|
|
uint8_t spare2:4;)
|
|
};
|
|
struct {
|
|
uint16_t pdu_session_status;
|
|
uint16_t uplink_data_status;
|
|
uint16_t allowed_pdu_session_status;
|
|
} psimask;
|
|
} __attribute__ ((packed)) test_registration_request_param_t;
|
|
|
|
typedef struct test_attach_request_param_s {
|
|
struct {
|
|
ED7(uint8_t spare1:1;,
|
|
uint8_t guti:1;,
|
|
uint8_t last_visited_registered_tai:1;,
|
|
uint8_t drx_parameter:1;,
|
|
uint8_t location_area_identication:1;,
|
|
uint8_t mobile_station_classmark_2:1;,
|
|
uint8_t ue_usage_setting:1;)
|
|
};
|
|
struct {
|
|
ED8(uint8_t old_guti_type:1;,
|
|
uint8_t nri_container:1;,
|
|
uint8_t ue_additional_security_capability:1;,
|
|
uint8_t tmsi_status:1;,
|
|
uint8_t ms_network_feature_support:1;,
|
|
uint8_t ms_network_capability:1;,
|
|
uint8_t mobile_station_classmark_3:1;,
|
|
uint8_t supported_codecs:1;)
|
|
};
|
|
struct {
|
|
ED8(uint8_t additional_update_type:1;,
|
|
uint8_t spare2:1;,
|
|
uint8_t spare3:1;,
|
|
uint8_t spare4:1;,
|
|
uint8_t spare5:1;,
|
|
uint8_t spare6:1;,
|
|
uint8_t spare7:1;,
|
|
uint8_t spare8:1;)
|
|
};
|
|
} __attribute__ ((packed)) test_attach_request_param_t;
|
|
|
|
typedef struct test_tau_request_param_s {
|
|
struct {
|
|
ED8(uint8_t ue_network_capability:1;,
|
|
uint8_t eps_bearer_context_status:1;,
|
|
uint8_t guti:1;,
|
|
uint8_t last_visited_registered_tai:1;,
|
|
uint8_t drx_parameter:1;,
|
|
uint8_t location_area_identication:1;,
|
|
uint8_t mobile_station_classmark_2:1;,
|
|
uint8_t ue_usage_setting:1;)
|
|
};
|
|
struct {
|
|
ED8(uint8_t old_guti_type:1;,
|
|
uint8_t nri_container:1;,
|
|
uint8_t ue_additional_security_capability:1;,
|
|
uint8_t tmsi_status:1;,
|
|
uint8_t ms_network_feature_support:1;,
|
|
uint8_t ms_network_capability:1;,
|
|
uint8_t mobile_station_classmark_3:1;,
|
|
uint8_t supported_codecs:1;)
|
|
};
|
|
struct {
|
|
ED3(uint8_t additional_update_type:1;,
|
|
uint8_t device_properties:1;,
|
|
uint8_t spare2:6;)
|
|
};
|
|
} __attribute__ ((packed)) test_tau_request_param_t;
|
|
|
|
typedef struct test_service_request_param_s {
|
|
union {
|
|
struct {
|
|
ED4(uint8_t pdu_session_status:1;,
|
|
uint8_t uplink_data_status:1;,
|
|
uint8_t allowed_pdu_session_status:1;,
|
|
uint8_t reserved:5;)
|
|
};
|
|
uint8_t value;
|
|
};
|
|
struct {
|
|
uint16_t pdu_session_status;
|
|
uint16_t uplink_data_status;
|
|
uint16_t allowed_pdu_session_status;
|
|
} psimask;
|
|
} __attribute__ ((packed)) test_service_request_param_t;
|
|
|
|
typedef struct test_extended_service_request_param_s {
|
|
union {
|
|
struct {
|
|
ED3(uint8_t csfb_response:1;,
|
|
uint8_t eps_bearer_context_status:1;,
|
|
uint8_t spare:6;)
|
|
};
|
|
uint8_t value;
|
|
};
|
|
struct {
|
|
uint8_t response;
|
|
} csfb;
|
|
} __attribute__ ((packed)) test_extended_service_request_param_t;
|
|
|
|
typedef struct test_ul_nas_transport_param_s {
|
|
union {
|
|
struct {
|
|
ED4(uint8_t request_type:4;,
|
|
uint8_t dnn:1;,
|
|
uint8_t s_nssai:1;,
|
|
uint8_t reserved:2;)
|
|
};
|
|
uint8_t value;
|
|
};
|
|
} __attribute__ ((packed)) test_ul_nas_transport_param_t;
|
|
|
|
typedef struct test_pdu_session_establishment_param_s {
|
|
union {
|
|
struct {
|
|
ED3(uint8_t ssc_mode:1;,
|
|
uint8_t epco:1;,
|
|
uint8_t reserved:6;)
|
|
};
|
|
uint8_t value;
|
|
};
|
|
} __attribute__ ((packed)) test_pdu_session_establishment_param_t;
|
|
|
|
typedef struct test_pdn_connectivity_param_s {
|
|
union {
|
|
struct {
|
|
ED7(uint8_t eit:1;,
|
|
uint8_t eit_no_required:1;,
|
|
uint8_t apn:1;,
|
|
uint8_t pco:1;,
|
|
uint8_t epco:1;,
|
|
uint8_t spare:1;,
|
|
uint8_t request_type:3;)
|
|
};
|
|
uint8_t value;
|
|
};
|
|
} __attribute__ ((packed)) test_pdn_connectivity_param_t;
|
|
|
|
typedef struct test_esm_information_param_s {
|
|
union {
|
|
struct {
|
|
ED8(uint8_t pco:1;,
|
|
uint8_t epco:1;,
|
|
uint8_t spare2:1;,
|
|
uint8_t spare3:1;,
|
|
uint8_t spare4:1;,
|
|
uint8_t spare5:1;,
|
|
uint8_t spare6:1;,
|
|
uint8_t spare7:1;)
|
|
};
|
|
uint8_t value;
|
|
};
|
|
} __attribute__ ((packed)) test_esm_information_param_t;
|
|
|
|
typedef struct test_ue_s {
|
|
ogs_lnode_t lnode; /**< A node of list_t */
|
|
|
|
uint32_t ran_ue_ngap_id; /* gNB-UE-NGAP-ID received from gNB */
|
|
uint64_t amf_ue_ngap_id; /* AMF-UE-NGAP-ID received from AMF */
|
|
uint32_t enb_ue_s1ap_id; /* eNB-UE-S1AP-ID received from eNB */
|
|
uint32_t mme_ue_s1ap_id; /* MME-UE-S1AP-ID received from MME */
|
|
|
|
char *imsi;
|
|
char *suci; /* TS33.501 : SUCI */
|
|
char *supi; /* TS33.501 : SUPI */
|
|
|
|
uint8_t imsi_buf[OGS_MAX_IMSI_LEN];
|
|
int imsi_len;
|
|
|
|
ogs_nas_5gs_mobile_identity_t mobile_identity;
|
|
ogs_nas_mobile_identity_imeisv_t mobile_identity_imeisv;
|
|
bool mobile_identity_imeisv_presence;
|
|
uint16_t mobile_identity_suci_length;
|
|
ogs_nas_5gs_guti_t nas_5gs_guti;
|
|
ogs_nas_eps_guti_t nas_eps_guti;
|
|
ogs_nas_mobile_identity_imsi_t mobile_identity_imsi;
|
|
|
|
ogs_eps_tai_t e_tai;
|
|
ogs_e_cgi_t e_cgi;
|
|
ogs_5gs_tai_t nr_tai;
|
|
ogs_nr_cgi_t nr_cgi;
|
|
|
|
uint8_t k[OGS_KEY_LEN];
|
|
const char *k_string;
|
|
uint8_t opc[OGS_KEY_LEN];
|
|
const char *opc_string;
|
|
|
|
uint8_t rand[OGS_RAND_LEN];
|
|
uint8_t autn[OGS_AUTN_LEN];
|
|
uint8_t abba[OGS_NAS_MAX_ABBA_LEN];
|
|
uint8_t abba_len;
|
|
uint8_t kamf[OGS_SHA256_DIGEST_SIZE];
|
|
uint8_t kasme[OGS_SHA256_DIGEST_SIZE];
|
|
|
|
struct {
|
|
int access_type; /* 3GPP or Non-3GPP */
|
|
|
|
union {
|
|
struct {
|
|
ED3(uint8_t tsc:1;,
|
|
uint8_t ksi:3;,
|
|
uint8_t value:4;)
|
|
};
|
|
ogs_nas_5gs_registration_type_t registration;
|
|
ogs_nas_de_registration_type_t de_registration;
|
|
ogs_nas_eps_attach_type_t attach;
|
|
|
|
uint8_t data;
|
|
};
|
|
|
|
} __attribute__ ((packed)) nas;
|
|
|
|
uint8_t knas_int[OGS_SHA256_DIGEST_SIZE/2];
|
|
uint8_t knas_enc[OGS_SHA256_DIGEST_SIZE/2];
|
|
uint32_t ul_count;
|
|
union {
|
|
struct {
|
|
ED3(uint8_t spare;,
|
|
uint16_t overflow;,
|
|
uint8_t sqn;)
|
|
} __attribute__ ((packed));
|
|
uint32_t i32;
|
|
} dl_count;
|
|
uint8_t kgnb[OGS_SHA256_DIGEST_SIZE];
|
|
|
|
uint8_t selected_enc_algorithm;
|
|
uint8_t selected_int_algorithm;
|
|
|
|
#define TEST_SECURITY_CONTEXT_IS_VALID(__tEST) \
|
|
((__tEST) && \
|
|
((__tEST)->security_context_available == 1) && \
|
|
((__tEST)->mac_failed == 0) && \
|
|
((__tEST)->nas.ksi != OGS_NAS_KSI_NO_KEY_IS_AVAILABLE))
|
|
#define TEST_CLEAR_SECURITY_CONTEXT(__tEST) \
|
|
do { \
|
|
ogs_assert((__tEST)); \
|
|
(__tEST)->security_context_available = 0; \
|
|
(__tEST)->mac_failed = 0; \
|
|
(__tEST)->nas.ksi = 0; \
|
|
} while(0)
|
|
int security_context_available;
|
|
int mac_failed;
|
|
|
|
ogs_nas_ue_security_capability_t ue_security_capability;
|
|
ogs_nas_ue_network_capability_t ue_network_capability;
|
|
|
|
struct {
|
|
int num_of_s_nssai;
|
|
ogs_nas_s_nssai_ie_t s_nssai[OGS_MAX_NUM_OF_SLICE];
|
|
} requested_nssai, allowed_nssai;
|
|
|
|
struct {
|
|
int num_of_s_nssai;
|
|
ogs_nas_rejected_s_nssai_t s_nssai[OGS_MAX_NUM_OF_SLICE];
|
|
} rejected_nssai;
|
|
|
|
test_initial_ue_param_t initial_ue_param;
|
|
|
|
test_registration_request_param_t registration_request_param;
|
|
test_service_request_param_t service_request_param;
|
|
test_extended_service_request_param_t extended_service_request_param;
|
|
|
|
uint16_t pdu_session_status;
|
|
uint16_t pdu_session_reactivation_result;
|
|
|
|
test_attach_request_param_t attach_request_param;
|
|
test_tau_request_param_t tau_request_param;
|
|
|
|
/* 5GC: Last received message */
|
|
S1AP_ProcedureCode_t ngap_procedure_code;
|
|
uint8_t gmm_message_type;
|
|
uint8_t gsm_message_type;
|
|
|
|
/* EPC: Last received message */
|
|
S1AP_ProcedureCode_t s1ap_procedure_code;
|
|
uint8_t emm_message_type;
|
|
uint8_t esm_message_type;
|
|
|
|
test_sess_t *sess;
|
|
|
|
ogs_list_t sess_list;
|
|
} test_ue_t;
|
|
|
|
typedef struct test_sess_s {
|
|
ogs_lnode_t lnode; /**< A node of list_t */
|
|
uint32_t index;
|
|
|
|
uint8_t psi;
|
|
|
|
uint8_t pti;
|
|
uint8_t pdu_session_type;
|
|
union {
|
|
char *dnn;
|
|
char *apn;
|
|
};
|
|
|
|
/* RAT Type */
|
|
uint8_t gtp_rat_type;
|
|
OpenAPI_rat_type_e sbi_rat_type;
|
|
|
|
ogs_ip_t ue_ip;
|
|
|
|
ogs_ip_t upf_n3_ip; /* UPF-N3 IPv4/IPv6 */
|
|
uint32_t upf_n3_teid; /* UPF-N3 TEID */
|
|
|
|
uint32_t gnb_n3_teid; /* gNB-N3 TEID */
|
|
ogs_sockaddr_t *gnb_n3_addr; /* gNB-N3 IPv4 */
|
|
ogs_sockaddr_t *gnb_n3_addr6; /* gNB-N3 IPv6 */
|
|
|
|
test_ul_nas_transport_param_t ul_nas_transport_param;
|
|
test_pdu_session_establishment_param_t pdu_session_establishment_param;
|
|
test_pdn_connectivity_param_t pdn_connectivity_param;
|
|
test_esm_information_param_t esm_information_param;
|
|
|
|
struct {
|
|
bool data_forwarding_not_possible;
|
|
|
|
/* Indirect DL Forwarding */
|
|
uint32_t upf_dl_teid;
|
|
ogs_ip_t upf_dl_ip;
|
|
} handover;
|
|
|
|
/* ePDG */
|
|
uint32_t epdg_s2b_c_teid;
|
|
uint32_t smf_s2b_c_teid;
|
|
ogs_gtp_node_t *gnode;
|
|
|
|
ogs_list_t bearer_list;
|
|
|
|
ogs_list_t qos_flow_to_modify_list;
|
|
|
|
test_ue_t *test_ue;
|
|
} test_sess_t;
|
|
|
|
typedef struct test_bearer_s {
|
|
ogs_lnode_t lnode; /**< A node of list_t */
|
|
|
|
ogs_lnode_t to_modify_node;
|
|
|
|
uint32_t index;
|
|
|
|
uint8_t qfi; /* 5GC */
|
|
uint8_t ebi; /* EPC */
|
|
|
|
uint32_t sgw_s1u_teid; /* SGW-S1U TEID */
|
|
ogs_ip_t sgw_s1u_ip; /* SGW-S1U IPv4/IPv6 */
|
|
struct {
|
|
/* Indirect Forwarding */
|
|
uint32_t dl_teid;
|
|
ogs_ip_t dl_ip;
|
|
uint32_t ul_teid;
|
|
ogs_ip_t ul_ip;
|
|
} handover;
|
|
|
|
uint32_t enb_s1u_teid; /* eNB-S1U TEID */
|
|
ogs_sockaddr_t *enb_s1u_addr; /* eNB-S1U IPv4 */
|
|
ogs_sockaddr_t *enb_s1u_addr6; /* eNB-S1U IPv6 */
|
|
|
|
test_sess_t *sess;
|
|
} test_bearer_t;
|
|
|
|
void test_context_init(void);
|
|
void test_context_final(void);
|
|
test_context_t *test_self(void);
|
|
|
|
int test_context_parse_config(void);
|
|
|
|
test_ue_t *test_ue_add_by_suci(
|
|
ogs_nas_5gs_mobile_identity_suci_t *mobile_identity_suci,
|
|
const char *scheme_output);
|
|
void test_ue_remove(test_ue_t *test_ue);
|
|
void test_ue_remove_all(void);
|
|
|
|
test_sess_t *test_sess_add_by_dnn_and_psi(
|
|
test_ue_t *test_ue, char *dnn, uint8_t psi);
|
|
test_sess_t *test_sess_add_by_apn(
|
|
test_ue_t *test_ue, char *apn, uint8_t rat_type);
|
|
void test_sess_remove(test_sess_t *sess);
|
|
void test_sess_remove_all(test_ue_t *test_ue);
|
|
|
|
test_sess_t *test_sess_find_by_apn(
|
|
test_ue_t *test_ue, char *apn, uint8_t rat_type);
|
|
test_sess_t *test_sess_find_by_psi(test_ue_t *test_ue, uint8_t psi);
|
|
|
|
test_bearer_t *test_bearer_add(test_sess_t *sess, uint8_t ebi);
|
|
test_bearer_t *test_qos_flow_add(test_sess_t *sess);
|
|
void test_bearer_remove(test_bearer_t *bearer);
|
|
void test_bearer_remove_all(test_sess_t *sess);
|
|
|
|
test_bearer_t *test_bearer_find_by_sess_ebi(test_sess_t *sess, uint8_t ebi);
|
|
test_bearer_t *test_bearer_find_by_ue_ebi(test_ue_t *test_ue, uint8_t ebi);
|
|
|
|
test_bearer_t *test_qos_flow_find_by_qfi(test_sess_t *sess, uint8_t qfi);
|
|
|
|
int test_db_insert_ue(test_ue_t *test_ue, bson_t *doc);
|
|
int test_db_remove_ue(test_ue_t *test_ue);
|
|
|
|
bson_t *test_db_new_simple(test_ue_t *test_ue);
|
|
bson_t *test_db_new_qos_flow(test_ue_t *test_ue);
|
|
bson_t *test_db_new_session(test_ue_t *test_ue);
|
|
bson_t *test_db_new_ims(test_ue_t *test_ue);
|
|
bson_t *test_db_new_slice_with_same_dnn(test_ue_t *test_ue);
|
|
bson_t *test_db_new_slice_with_different_dnn(test_ue_t *test_ue);
|
|
bson_t *test_db_new_non3gpp(test_ue_t *test_ue);
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
|
|
#endif /* TEST_COMMON_CONTEXT_H */
|