forked from acouzens/open5gs
315 lines
11 KiB
C
315 lines
11 KiB
C
/*
|
|
* Copyright (C) 2019,2020 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/>.
|
|
*/
|
|
|
|
#include "test-common.h"
|
|
|
|
static void test1_func(abts_case *tc, void *data)
|
|
{
|
|
int rv;
|
|
ogs_socknode_t *ngap;
|
|
ogs_socknode_t *gtpu;
|
|
ogs_pkbuf_t *gmmbuf;
|
|
ogs_pkbuf_t *gsmbuf;
|
|
ogs_pkbuf_t *nasbuf;
|
|
ogs_pkbuf_t *sendbuf;
|
|
ogs_pkbuf_t *recvbuf;
|
|
ogs_ngap_message_t message;
|
|
int i;
|
|
|
|
ogs_nas_5gs_mobile_identity_suci_t mobile_identity_suci;
|
|
test_ue_t *test_ue = NULL;
|
|
test_sess_t *sess = NULL;
|
|
test_bearer_t *qos_flow = NULL;
|
|
|
|
const char *_k_string = "5122250214c33e723a5dd523fc145fc0";
|
|
uint8_t k[OGS_KEY_LEN];
|
|
const char *_opc_string = "981d464c7c52eb6e5036234984ad0bcf";
|
|
uint8_t opc[OGS_KEY_LEN];
|
|
|
|
mongoc_collection_t *collection = NULL;
|
|
bson_t *doc = NULL;
|
|
int64_t count = 0;
|
|
bson_error_t error;
|
|
const char *json =
|
|
"{"
|
|
"\"_id\" : { \"$oid\" : \"597223158b8861d7605378c6\" }, "
|
|
"\"imsi\" : \"2089300007487\", "
|
|
"\"pdn\" : ["
|
|
"{"
|
|
"\"apn\" : \"internet\", "
|
|
"\"_id\" : { \"$oid\" : \"597223158b8861d7605378c7\" }, "
|
|
"\"ambr\" : {"
|
|
"\"uplink\" : { \"$numberLong\" : \"1024000\" }, "
|
|
"\"downlink\" : { \"$numberLong\" : \"1024000\" } "
|
|
"},"
|
|
"\"qos\" : { "
|
|
"\"qci\" : 9, "
|
|
"\"arp\" : { "
|
|
"\"priority_level\" : 8,"
|
|
"\"pre_emption_vulnerability\" : 1, "
|
|
"\"pre_emption_capability\" : 1"
|
|
"} "
|
|
"}, "
|
|
"\"type\" : 2"
|
|
"}"
|
|
"],"
|
|
"\"ambr\" : { "
|
|
"\"uplink\" : { \"$numberLong\" : \"1024000\" }, "
|
|
"\"downlink\" : { \"$numberLong\" : \"1024000\" } "
|
|
"},"
|
|
"\"subscribed_rau_tau_timer\" : 12,"
|
|
"\"network_access_mode\" : 2, "
|
|
"\"subscriber_status\" : 0, "
|
|
"\"access_restriction_data\" : 32, "
|
|
"\"security\" : { "
|
|
"\"k\" : \"5122250214c33e723a5dd523fc145fc0\", "
|
|
"\"opc\" : \"981d464c7c52eb6e5036234984ad0bcf\", "
|
|
"\"amf\" : \"8000\", "
|
|
"\"sqn\" : { \"$numberLong\" : \"25235952177090\" }, "
|
|
"\"rand\" : \"20080C38 18183B52 2614162C 07601D0D\" "
|
|
"}, "
|
|
"\"__v\" : 0 "
|
|
"}";
|
|
|
|
/* Setup Test UE & Session Context */
|
|
memset(&mobile_identity_suci, 0, sizeof(mobile_identity_suci));
|
|
|
|
mobile_identity_suci.h.supi_format = OGS_NAS_5GS_SUPI_FORMAT_IMSI;
|
|
mobile_identity_suci.h.type = OGS_NAS_5GS_MOBILE_IDENTITY_SUCI;
|
|
mobile_identity_suci.routing_indicator1 = 0;
|
|
mobile_identity_suci.routing_indicator2 = 0xf;
|
|
mobile_identity_suci.routing_indicator3 = 0xf;
|
|
mobile_identity_suci.routing_indicator4 = 0xf;
|
|
mobile_identity_suci.protection_scheme_id = OGS_NAS_5GS_NULL_SCHEME;
|
|
mobile_identity_suci.home_network_pki_value = 0;
|
|
mobile_identity_suci.scheme_output[0] = 0;
|
|
mobile_identity_suci.scheme_output[1] = 0;
|
|
mobile_identity_suci.scheme_output[2] = 0x47;
|
|
mobile_identity_suci.scheme_output[3] = 0x78;
|
|
|
|
test_ue = test_ue_add_by_suci(&mobile_identity_suci, 12);
|
|
ogs_assert(test_ue);
|
|
|
|
test_ue->nr_cgi.cell_id = 0x40001;
|
|
|
|
test_ue->nas.registration.type = 1;
|
|
test_ue->nas.registration.follow_on_request = 1;
|
|
test_ue->nas.registration.value = OGS_NAS_5GS_REGISTRATION_TYPE_INITIAL;
|
|
|
|
OGS_HEX(_k_string, strlen(_k_string), test_ue->k);
|
|
OGS_HEX(_opc_string, strlen(_opc_string), test_ue->opc);
|
|
|
|
sess = test_sess_add_by_dnn_and_psi(test_ue, "internet", 10);
|
|
ogs_assert(sess);
|
|
|
|
sess->pti = 0;
|
|
sess->pdu_session_type = OGS_PDU_SESSION_TYPE_IPV4;
|
|
|
|
/* gNB connects to AMF */
|
|
ngap = testngap_client(AF_INET);
|
|
ABTS_PTR_NOTNULL(tc, ngap);
|
|
|
|
/* gNB connects to UPF */
|
|
gtpu = test_gtpu_server(1, AF_INET);
|
|
ABTS_PTR_NOTNULL(tc, gtpu);
|
|
|
|
/* Send NG-Setup Reqeust */
|
|
sendbuf = testngap_build_ng_setup_request(0x102, 32);
|
|
ABTS_PTR_NOTNULL(tc, sendbuf);
|
|
rv = testgnb_ngap_send(ngap, sendbuf);
|
|
ABTS_INT_EQUAL(tc, OGS_OK, rv);
|
|
|
|
/* Receive NG-Setup Response */
|
|
recvbuf = testgnb_ngap_read(ngap);
|
|
ABTS_PTR_NOTNULL(tc, recvbuf);
|
|
testngap_recv(test_ue, recvbuf);
|
|
|
|
/********** Insert Subscriber in Database */
|
|
collection = mongoc_client_get_collection(
|
|
ogs_mongoc()->client, ogs_mongoc()->name, "subscribers");
|
|
ABTS_PTR_NOTNULL(tc, collection);
|
|
doc = BCON_NEW("imsi", BCON_UTF8(test_ue->imsi));
|
|
ABTS_PTR_NOTNULL(tc, doc);
|
|
|
|
count = mongoc_collection_count (
|
|
collection, MONGOC_QUERY_NONE, doc, 0, 0, NULL, &error);
|
|
if (count) {
|
|
ABTS_TRUE(tc, mongoc_collection_remove(collection,
|
|
MONGOC_REMOVE_SINGLE_REMOVE, doc, NULL, &error))
|
|
}
|
|
bson_destroy(doc);
|
|
|
|
doc = bson_new_from_json((const uint8_t *)json, -1, &error);;
|
|
ABTS_PTR_NOTNULL(tc, doc);
|
|
ABTS_TRUE(tc, mongoc_collection_insert(collection,
|
|
MONGOC_INSERT_NONE, doc, NULL, &error));
|
|
bson_destroy(doc);
|
|
|
|
doc = BCON_NEW("imsi", BCON_UTF8(test_ue->imsi));
|
|
ABTS_PTR_NOTNULL(tc, doc);
|
|
do {
|
|
count = mongoc_collection_count (
|
|
collection, MONGOC_QUERY_NONE, doc, 0, 0, NULL, &error);
|
|
} while (count == 0);
|
|
bson_destroy(doc);
|
|
|
|
/* Send Registration request */
|
|
test_ue->registration_request_param.gmm_capability = 1;
|
|
gmmbuf = testgmm_build_registration_request(test_ue, NULL);
|
|
ABTS_PTR_NOTNULL(tc, gmmbuf);
|
|
sendbuf = testngap_build_initial_ue_message(test_ue, gmmbuf, false);
|
|
ABTS_PTR_NOTNULL(tc, sendbuf);
|
|
rv = testgnb_ngap_send(ngap, sendbuf);
|
|
ABTS_INT_EQUAL(tc, OGS_OK, rv);
|
|
|
|
/* Receive Authentication request */
|
|
recvbuf = testgnb_ngap_read(ngap);
|
|
ABTS_PTR_NOTNULL(tc, recvbuf);
|
|
testngap_recv(test_ue, recvbuf);
|
|
|
|
/* Send Authentication response */
|
|
gmmbuf = testgmm_build_authentication_response(test_ue);
|
|
ABTS_PTR_NOTNULL(tc, gmmbuf);
|
|
sendbuf = testngap_build_uplink_nas_transport(test_ue, gmmbuf);
|
|
ABTS_PTR_NOTNULL(tc, sendbuf);
|
|
rv = testgnb_ngap_send(ngap, sendbuf);
|
|
ABTS_INT_EQUAL(tc, OGS_OK, rv);
|
|
|
|
/* Receive Security mode command */
|
|
recvbuf = testgnb_ngap_read(ngap);
|
|
ABTS_PTR_NOTNULL(tc, recvbuf);
|
|
testngap_recv(test_ue, recvbuf);
|
|
|
|
/* Send Security mode complete */
|
|
nasbuf = testgmm_build_registration_request(test_ue, NULL);
|
|
ABTS_PTR_NOTNULL(tc, nasbuf);
|
|
gmmbuf = testgmm_build_security_mode_complete(test_ue, nasbuf);
|
|
ABTS_PTR_NOTNULL(tc, gmmbuf);
|
|
sendbuf = testngap_build_uplink_nas_transport(test_ue, gmmbuf);
|
|
ABTS_PTR_NOTNULL(tc, sendbuf);
|
|
rv = testgnb_ngap_send(ngap, sendbuf);
|
|
ABTS_INT_EQUAL(tc, OGS_OK, rv);
|
|
|
|
/* Receive Initial context setup request */
|
|
recvbuf = testgnb_ngap_read(ngap);
|
|
ABTS_PTR_NOTNULL(tc, recvbuf);
|
|
testngap_recv(test_ue, recvbuf);
|
|
|
|
/* Send UE radio capability info indication */
|
|
sendbuf = testngap_build_ue_radio_capability_info_indication(test_ue);
|
|
ABTS_PTR_NOTNULL(tc, sendbuf);
|
|
rv = testgnb_ngap_send(ngap, sendbuf);
|
|
ABTS_INT_EQUAL(tc, OGS_OK, rv);
|
|
|
|
/* Send Initial context setup response */
|
|
sendbuf = testngap_build_initial_context_setup_response(test_ue, NULL);
|
|
ABTS_PTR_NOTNULL(tc, sendbuf);
|
|
rv = testgnb_ngap_send(ngap, sendbuf);
|
|
ABTS_INT_EQUAL(tc, OGS_OK, rv);
|
|
|
|
/* Send Registration complete */
|
|
gmmbuf = testgmm_build_registration_complete(test_ue);
|
|
ABTS_PTR_NOTNULL(tc, gmmbuf);
|
|
sendbuf = testngap_build_uplink_nas_transport(test_ue, gmmbuf);
|
|
ABTS_PTR_NOTNULL(tc, sendbuf);
|
|
rv = testgnb_ngap_send(ngap, sendbuf);
|
|
ABTS_INT_EQUAL(tc, OGS_OK, rv);
|
|
|
|
/* Receive Configuration update command */
|
|
recvbuf = testgnb_ngap_read(ngap);
|
|
ABTS_PTR_NOTNULL(tc, recvbuf);
|
|
testngap_recv(test_ue, recvbuf);
|
|
|
|
/* Send PDU session establishment request */
|
|
sess->ul_nas_transport_param.request_type =
|
|
OGS_NAS_5GS_REQUEST_TYPE_INITIAL;
|
|
sess->ul_nas_transport_param.dnn = 1;
|
|
sess->ul_nas_transport_param.s_nssai = 1;
|
|
|
|
gsmbuf = testgsm_build_pdu_session_establishment_request(sess);
|
|
ABTS_PTR_NOTNULL(tc, gsmbuf);
|
|
gmmbuf = testgmm_build_ul_nas_transport(sess,
|
|
OGS_NAS_PAYLOAD_CONTAINER_N1_SM_INFORMATION, gsmbuf);
|
|
ABTS_PTR_NOTNULL(tc, gmmbuf);
|
|
sendbuf = testngap_build_uplink_nas_transport(test_ue, gmmbuf);
|
|
ABTS_PTR_NOTNULL(tc, sendbuf);
|
|
rv = testgnb_ngap_send(ngap, sendbuf);
|
|
ABTS_INT_EQUAL(tc, OGS_OK, rv);
|
|
|
|
/* Receive PDU session establishment accept */
|
|
recvbuf = testgnb_ngap_read(ngap);
|
|
ABTS_PTR_NOTNULL(tc, recvbuf);
|
|
testngap_recv(test_ue, recvbuf);
|
|
|
|
/* Send GTP-U ICMP Packet */
|
|
qos_flow = test_qos_flow_find_by_ue_qfi(test_ue, 1);
|
|
ogs_assert(qos_flow);
|
|
rv = test_gtpu_send_ping(gtpu, qos_flow, TEST_PING_IPV4);
|
|
ABTS_INT_EQUAL(tc, OGS_OK, rv);
|
|
|
|
/* Send PDU session resource setup response */
|
|
sendbuf = testngap_build_pdu_session_resource_setup_response(sess);
|
|
ABTS_PTR_NOTNULL(tc, sendbuf);
|
|
rv = testgnb_ngap_send(ngap, sendbuf);
|
|
ABTS_INT_EQUAL(tc, OGS_OK, rv);
|
|
|
|
/* Receive GTP-U ICMP Packet */
|
|
recvbuf = testgnb_gtpu_read(gtpu);
|
|
ABTS_PTR_NOTNULL(tc, recvbuf);
|
|
ogs_pkbuf_free(recvbuf);
|
|
|
|
/* Send GTP-U ICMP Packet */
|
|
rv = test_gtpu_send_ping(gtpu, qos_flow, TEST_PING_IPV4);
|
|
ABTS_INT_EQUAL(tc, OGS_OK, rv);
|
|
|
|
/* Receive GTP-U ICMP Packet */
|
|
recvbuf = testgnb_gtpu_read(gtpu);
|
|
ABTS_PTR_NOTNULL(tc, recvbuf);
|
|
ogs_pkbuf_free(recvbuf);
|
|
|
|
ogs_msleep(300);
|
|
|
|
/********** Remove Subscriber in Database */
|
|
doc = BCON_NEW("imsi", BCON_UTF8(test_ue->imsi));
|
|
ABTS_PTR_NOTNULL(tc, doc);
|
|
ABTS_TRUE(tc, mongoc_collection_remove(collection,
|
|
MONGOC_REMOVE_SINGLE_REMOVE, doc, NULL, &error))
|
|
bson_destroy(doc);
|
|
|
|
mongoc_collection_destroy(collection);
|
|
|
|
/* Clear Test UE Context */
|
|
test_ue_remove(test_ue);
|
|
|
|
/* gNB disonncect from UPF */
|
|
testgnb_gtpu_close(gtpu);
|
|
|
|
/* gNB disonncect from AMF */
|
|
testgnb_ngap_close(ngap);
|
|
}
|
|
|
|
abts_suite *test_minimal(abts_suite *suite)
|
|
{
|
|
suite = ADD_SUITE(suite)
|
|
|
|
abts_run_test(suite, test1_func, NULL);
|
|
|
|
return suite;
|
|
}
|