forked from acouzens/open5gs
312 lines
9.7 KiB
C
312 lines
9.7 KiB
C
|
|
#include <mongoc.h>
|
|
|
|
#include "core/abts.h"
|
|
|
|
#include "app/context.h"
|
|
#include "mme/mme-context.h"
|
|
#include "mme/s1ap-build.h"
|
|
#include "asn1c/s1ap-message.h"
|
|
|
|
#include "test-packet.h"
|
|
|
|
extern ogs_socknode_t *sgsap;
|
|
|
|
static void test1_func(abts_case *tc, void *data)
|
|
{
|
|
int rv;
|
|
ogs_socknode_t *s1ap;
|
|
ogs_pkbuf_t *sendbuf;
|
|
ogs_pkbuf_t *recvbuf;
|
|
s1ap_message_t message;
|
|
int i;
|
|
int msgindex = 19;
|
|
enb_ue_t *enb_ue = NULL;
|
|
mme_ue_t *mme_ue = NULL;
|
|
uint32_t m_tmsi = 0;
|
|
|
|
mongoc_collection_t *collection = NULL;
|
|
bson_t *doc = NULL;
|
|
int64_t count = 0;
|
|
bson_error_t error;
|
|
const char *json =
|
|
"{"
|
|
"\"_id\" : { \"$oid\" : \"310014158b8861d7605378c6\" }, "
|
|
"\"imsi\" : \"724210000000003\", "
|
|
"\"pdn\" : ["
|
|
"{"
|
|
"\"apn\" : \"internet\", "
|
|
"\"_id\" : { \"$oid\" : \"310014158b8861d7605378c7\" }, "
|
|
"\"ambr\" : {"
|
|
"\"uplink\" : { \"$numberLong\" : \"1000000\" }, "
|
|
"\"downlink\" : { \"$numberLong\" : \"1000000\" } "
|
|
"},"
|
|
"\"qos\" : { "
|
|
"\"qci\" : 9, "
|
|
"\"arp\" : { "
|
|
"\"priority_level\" : 8,"
|
|
"\"pre_emption_vulnerability\" : 0, "
|
|
"\"pre_emption_capability\" : 0"
|
|
"} "
|
|
"}, "
|
|
"\"type\" : 2"
|
|
"}"
|
|
"],"
|
|
"\"ambr\" : { "
|
|
"\"uplink\" : { \"$numberLong\" : \"1000000\" }, "
|
|
"\"downlink\" : { \"$numberLong\" : \"1000000\" } "
|
|
"},"
|
|
"\"subscribed_rau_tau_timer\" : 12,"
|
|
"\"network_access_mode\" : 2, "
|
|
"\"subscriber_status\" : 0, "
|
|
"\"access_restriction_data\" : 32, "
|
|
"\"security\" : { "
|
|
"\"k\" : \"70D49A71DD1A2B806A25ABE0EF749F1E\", "
|
|
"\"opc\" : \"6F1BF53D624B3A43AF6592854E2444C7\", "
|
|
"\"amf\" : \"8000\", "
|
|
"\"sqn\" : { \"$numberLong\" : \"2374\" }, "
|
|
"\"rand\" : \"aa266700bc2887354e9f87368d5d0ae7\" "
|
|
"}, "
|
|
"\"__v\" : 0 "
|
|
"}";
|
|
|
|
/* eNB connects to MME */
|
|
s1ap = testenb_s1ap_client("127.0.0.1");
|
|
ABTS_PTR_NOTNULL(tc, s1ap);
|
|
|
|
/* Send S1-Setup Reqeust */
|
|
rv = tests1ap_build_setup_req(
|
|
&sendbuf, S1AP_ENB_ID_PR_macroENB_ID, 0x0019b0, 12345, 724, 21, 2);
|
|
ABTS_INT_EQUAL(tc, OGS_OK, rv);
|
|
rv = testenb_s1ap_send(s1ap, sendbuf);
|
|
ABTS_INT_EQUAL(tc, OGS_OK, rv);
|
|
|
|
/* Receive S1-Setup Response */
|
|
recvbuf = testenb_s1ap_read(s1ap);
|
|
ABTS_PTR_NOTNULL(tc, recvbuf);
|
|
rv = s1ap_decode_pdu(&message, recvbuf);
|
|
ABTS_INT_EQUAL(tc, OGS_OK, rv);
|
|
s1ap_free_pdu(&message);
|
|
ogs_pkbuf_free(recvbuf);
|
|
|
|
collection = mongoc_client_get_collection(
|
|
context_self()->db.client,
|
|
context_self()->db.name, "subscribers");
|
|
ABTS_PTR_NOTNULL(tc, collection);
|
|
|
|
/********** Insert Subscriber in Database */
|
|
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("724210000000003"));
|
|
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 Attach Request */
|
|
mme_self()->mme_ue_s1ap_id = 247;
|
|
rv = tests1ap_build_initial_ue_msg(&sendbuf, msgindex);
|
|
ABTS_INT_EQUAL(tc, OGS_OK, rv);
|
|
rv = testenb_s1ap_send(s1ap, sendbuf);
|
|
ABTS_INT_EQUAL(tc, OGS_OK, rv);
|
|
|
|
/* Receive Identity-Request */
|
|
recvbuf = testenb_s1ap_read(s1ap);
|
|
ABTS_PTR_NOTNULL(tc, recvbuf);
|
|
ogs_pkbuf_free(recvbuf);
|
|
|
|
/* Send Identity Response */
|
|
rv = tests1ap_build_identity_response(&sendbuf, msgindex);
|
|
ABTS_INT_EQUAL(tc, OGS_OK, rv);
|
|
rv = testenb_s1ap_send(s1ap, sendbuf);
|
|
ABTS_INT_EQUAL(tc, OGS_OK, rv);
|
|
|
|
/* Receive Authentication Request */
|
|
recvbuf = testenb_s1ap_read(s1ap);
|
|
ABTS_PTR_NOTNULL(tc, recvbuf);
|
|
ogs_pkbuf_free(recvbuf);
|
|
|
|
/* Send Authentication Response */
|
|
rv = tests1ap_build_authentication_response(&sendbuf, msgindex);
|
|
ABTS_INT_EQUAL(tc, OGS_OK, rv);
|
|
rv = testenb_s1ap_send(s1ap, sendbuf);
|
|
ABTS_INT_EQUAL(tc, OGS_OK, rv);
|
|
|
|
/* Receive Security mode Command */
|
|
recvbuf = testenb_s1ap_read(s1ap);
|
|
ABTS_PTR_NOTNULL(tc, recvbuf);
|
|
ogs_pkbuf_free(recvbuf);
|
|
|
|
/* Send Security mode Complete */
|
|
rv = tests1ap_build_security_mode_complete(&sendbuf, msgindex);
|
|
ABTS_INT_EQUAL(tc, OGS_OK, rv);
|
|
rv = testenb_s1ap_send(s1ap, sendbuf);
|
|
ABTS_INT_EQUAL(tc, OGS_OK, rv);
|
|
|
|
/* Receive ESM Information Request */
|
|
recvbuf = testenb_s1ap_read(s1ap);
|
|
ABTS_PTR_NOTNULL(tc, recvbuf);
|
|
ogs_pkbuf_free(recvbuf);
|
|
|
|
/* Send ESM Information Response */
|
|
rv = tests1ap_build_esm_information_response(&sendbuf, msgindex);
|
|
ABTS_INT_EQUAL(tc, OGS_OK, rv);
|
|
rv = testenb_s1ap_send(s1ap, sendbuf);
|
|
ABTS_INT_EQUAL(tc, OGS_OK, rv);
|
|
|
|
/* Receive SGsAP-Location-Update-Request */
|
|
recvbuf = testvlr_sgsap_read(sgsap);
|
|
ABTS_PTR_NOTNULL(tc, recvbuf);
|
|
ogs_pkbuf_free(recvbuf);
|
|
|
|
/* Send SGsAP-Location-Update-Accept */
|
|
rv = testsgsap_location_update_accept(&sendbuf, 1);
|
|
ABTS_INT_EQUAL(tc, OGS_OK, rv);
|
|
rv = testvlr_sgsap_send(sgsap, sendbuf);
|
|
ABTS_INT_EQUAL(tc, OGS_OK, rv);
|
|
|
|
/* Receive Initial Context Setup Request +
|
|
* Attach Accept +
|
|
* Activate Default Bearer Context Request */
|
|
recvbuf = testenb_s1ap_read(s1ap);
|
|
ABTS_PTR_NOTNULL(tc, recvbuf);
|
|
ogs_pkbuf_free(recvbuf);
|
|
|
|
/* Send UE Capability Info Indication */
|
|
rv = tests1ap_build_ue_capability_info_indication(&sendbuf, msgindex);
|
|
ABTS_INT_EQUAL(tc, OGS_OK, rv);
|
|
rv = testenb_s1ap_send(s1ap, sendbuf);
|
|
ABTS_INT_EQUAL(tc, OGS_OK, rv);
|
|
|
|
/* Send Initial Context Setup Response */
|
|
rv = tests1ap_build_initial_context_setup_response(&sendbuf,
|
|
248, 256403, 5, 0x00000615, "127.0.0.5");
|
|
ABTS_INT_EQUAL(tc, OGS_OK, rv);
|
|
rv = testenb_s1ap_send(s1ap, sendbuf);
|
|
ABTS_INT_EQUAL(tc, OGS_OK, rv);
|
|
|
|
/* Send Attach Complete + Activate default EPS bearer cotext accept */
|
|
/* Receive EMM information */
|
|
rv = tests1ap_build_attach_complete(&sendbuf, msgindex);
|
|
ABTS_INT_EQUAL(tc, OGS_OK, rv);
|
|
rv = testenb_s1ap_send(s1ap, sendbuf);
|
|
ABTS_INT_EQUAL(tc, OGS_OK, rv);
|
|
|
|
/* Receive EMM Information */
|
|
recvbuf = testenb_s1ap_read(s1ap);
|
|
ABTS_PTR_NOTNULL(tc, recvbuf);
|
|
ogs_pkbuf_free(recvbuf);
|
|
|
|
/* Receive SGsAP TMSI-REALLOCATION-COMPLETE */
|
|
recvbuf = testvlr_sgsap_read(sgsap);
|
|
ABTS_PTR_NOTNULL(tc, recvbuf);
|
|
ogs_pkbuf_free(recvbuf);
|
|
|
|
/* Retreive M-TMSI */
|
|
enb_ue = enb_ue_find_by_mme_ue_s1ap_id(248);
|
|
ogs_assert(enb_ue);
|
|
mme_ue = enb_ue->mme_ue;
|
|
ogs_assert(mme_ue);
|
|
m_tmsi = mme_ue->guti.m_tmsi;
|
|
|
|
/* Send SGsAP-Paging-Request */
|
|
rv = testsgsap_paging_request(&sendbuf, 3);
|
|
ABTS_INT_EQUAL(tc, OGS_OK, rv);
|
|
rv = testvlr_sgsap_send(sgsap, sendbuf);
|
|
ABTS_INT_EQUAL(tc, OGS_OK, rv);
|
|
|
|
/* Receive S1AP Paging */
|
|
recvbuf = testenb_s1ap_read(s1ap);
|
|
ABTS_PTR_NOTNULL(tc, recvbuf);
|
|
ogs_pkbuf_free(recvbuf);
|
|
|
|
/* Send Extended Service Request */
|
|
rv = tests1ap_build_extended_service_request(&sendbuf, 22,
|
|
1, m_tmsi, 3, mme_ue->knas_int);
|
|
ABTS_INT_EQUAL(tc, OGS_OK, rv);
|
|
rv = testenb_s1ap_send(s1ap, sendbuf);
|
|
ABTS_INT_EQUAL(tc, OGS_OK, rv);
|
|
|
|
/* Receive Service Request */
|
|
recvbuf = testvlr_sgsap_read(sgsap);
|
|
ABTS_PTR_NOTNULL(tc, recvbuf);
|
|
ogs_pkbuf_free(recvbuf);
|
|
|
|
/* Receive UE Context Modification Request */
|
|
recvbuf = testenb_s1ap_read(s1ap);
|
|
ABTS_PTR_NOTNULL(tc, recvbuf);
|
|
ogs_pkbuf_free(recvbuf);
|
|
|
|
/* Send UE Context Modification Response */
|
|
rv = tests1ap_build_ue_context_modification_response(&sendbuf, 248, 256403);
|
|
ABTS_INT_EQUAL(tc, OGS_OK, rv);
|
|
rv = testenb_s1ap_send(s1ap, sendbuf);
|
|
ABTS_INT_EQUAL(tc, OGS_OK, rv);
|
|
|
|
/* Send UE Context Release Request */
|
|
rv = tests1ap_build_ue_context_release_request(&sendbuf, 20);
|
|
ABTS_INT_EQUAL(tc, OGS_OK, rv);
|
|
rv = testenb_s1ap_send(s1ap, sendbuf);
|
|
ABTS_INT_EQUAL(tc, OGS_OK, rv);
|
|
|
|
/* Receive UE Context Release Command */
|
|
recvbuf = testenb_s1ap_read(s1ap);
|
|
ABTS_PTR_NOTNULL(tc, recvbuf);
|
|
ogs_pkbuf_free(recvbuf);
|
|
|
|
/* Send UE Context Release Complete */
|
|
rv = tests1ap_build_ue_context_release_complete(&sendbuf, 20);
|
|
ABTS_INT_EQUAL(tc, OGS_OK, rv);
|
|
rv = testenb_s1ap_send(s1ap, sendbuf);
|
|
ABTS_INT_EQUAL(tc, OGS_OK, rv);
|
|
|
|
/* Send SGsAP-Paging-Request */
|
|
rv = testsgsap_paging_request(&sendbuf, 3);
|
|
ABTS_INT_EQUAL(tc, OGS_OK, rv);
|
|
rv = testvlr_sgsap_send(sgsap, sendbuf);
|
|
ABTS_INT_EQUAL(tc, OGS_OK, rv);
|
|
|
|
/* Receive S1AP Paging */
|
|
recvbuf = testenb_s1ap_read(s1ap);
|
|
ABTS_PTR_NOTNULL(tc, recvbuf);
|
|
ogs_pkbuf_free(recvbuf);
|
|
|
|
#if 0
|
|
/* Send Extended Service Request */
|
|
rv = tests1ap_build_extended_service_request(&sendbuf, 21,
|
|
1, m_tmsi, 4, mme_ue->knas_int);
|
|
ABTS_INT_EQUAL(tc, OGS_OK, rv);
|
|
rv = testenb_s1ap_send(s1ap, sendbuf);
|
|
ABTS_INT_EQUAL(tc, OGS_OK, rv);
|
|
#endif
|
|
|
|
ogs_msleep(300);
|
|
|
|
/********** Remove Subscriber in Database */
|
|
doc = BCON_NEW("imsi", BCON_UTF8("724210000000003"));
|
|
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);
|
|
|
|
/* eNB disonncect from MME */
|
|
testenb_s1ap_close(s1ap);
|
|
}
|
|
|
|
abts_suite *test_crash(abts_suite *suite)
|
|
{
|
|
suite = ADD_SUITE(suite)
|
|
|
|
abts_run_test(suite, test1_func, NULL);
|
|
|
|
return suite;
|
|
}
|