mirror of https://github.com/PentHertz/srsLTE.git
295 lines
14 KiB
C++
295 lines
14 KiB
C++
/*
|
|
* Copyright 2013-2020 Software Radio Systems Limited
|
|
*
|
|
* This file is part of srsLTE.
|
|
*
|
|
* srsLTE 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.
|
|
*
|
|
* srsLTE 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 Affero General Public License for more details.
|
|
*
|
|
* A copy of the GNU Affero General Public License can be found in
|
|
* the LICENSE file in the top-level directory of this distribution
|
|
* and at http://www.gnu.org/licenses/.
|
|
*
|
|
*/
|
|
|
|
#ifndef SRSEPC_NAS_H
|
|
#define SRSEPC_NAS_H
|
|
|
|
#include "srslte/asn1/gtpc_ies.h"
|
|
#include "srslte/asn1/liblte_mme.h"
|
|
#include "srslte/common/buffer_pool.h"
|
|
#include "srslte/common/security.h"
|
|
#include "srslte/interfaces/epc_interfaces.h"
|
|
#include <netinet/sctp.h>
|
|
|
|
namespace srsepc {
|
|
|
|
static const uint8_t MAX_ERABS_PER_UE = 16;
|
|
|
|
// MME EMM states (3GPP 24.301 v10.0.0, section 5.1.3.4)
|
|
typedef enum {
|
|
EMM_STATE_DEREGISTERED = 0,
|
|
EMM_STATE_COMMON_PROCEDURE_INITIATED,
|
|
EMM_STATE_REGISTERED,
|
|
EMM_STATE_DEREGISTERED_INITIATED,
|
|
EMM_STATE_N_ITEMS,
|
|
} emm_state_t;
|
|
static const char emm_state_text[EMM_STATE_N_ITEMS][100] = {"DEREGISTERED",
|
|
"COMMON PROCEDURE INITIATED",
|
|
"REGISTERED",
|
|
"DEREGISTERED INITIATED"};
|
|
|
|
// MME ECM states (3GPP 23.401 v10.0.0, section 4.6.3)
|
|
typedef enum {
|
|
ECM_STATE_IDLE = 0,
|
|
ECM_STATE_CONNECTED,
|
|
ECM_STATE_N_ITEMS,
|
|
} ecm_state_t;
|
|
static const char ecm_state_text[ECM_STATE_N_ITEMS][100] = {"IDLE", "CONNECTED"};
|
|
|
|
/*
|
|
// MME ESM states (3GPP 23.401 v10.0.0, section 4.6.3)
|
|
typedef enum {
|
|
ESM_BEARER_CONTEXT_INACTIVE = 0,
|
|
ESM_BEARER_CONTEXT_ACTIVE_PENDING,
|
|
ESM_BEARER_CONTEXT_ACTIVE,
|
|
ESM_BEARER_CONTEXT_INACTIVE_PENDING,
|
|
ESM_BEARER_CONTEXT_MODIFY_PENDING,
|
|
ESM_BEARER_PROCEDURE_TRANSACTION_INACTIVE,
|
|
ESM_BEARER_PROCEDURE_TRANSACTION_PENDING,
|
|
ESM_STATE_N_ITEMS,
|
|
} esm_state_t;
|
|
static const char esm_state_text[ESM_STATE_N_ITEMS][100] = {"CONTEXT INACTIVE",
|
|
"CONTEXT ACTIVE PENDING",
|
|
"CONTEXT ACTIVE",
|
|
"CONTEXT_INACTIVE_PENDING",
|
|
"CONTEXT_MODIFY_PENDING",
|
|
"PROCEDURE_TRANSACTION_INACTIVE"
|
|
"PROCEDURE_TRANSACTION_PENDING"};
|
|
*/
|
|
typedef enum { ERAB_DEACTIVATED, ERAB_CTX_REQUESTED, ERAB_CTX_SETUP, ERAB_ACTIVE } esm_state_t;
|
|
|
|
/*
|
|
* EMM, ECM, ESM and EPS Security context definitions
|
|
*/
|
|
typedef struct {
|
|
uint64_t imsi;
|
|
emm_state_t state;
|
|
uint8_t procedure_transaction_id;
|
|
uint8_t attach_type;
|
|
struct in_addr ue_ip;
|
|
srslte::gtpc_f_teid_ie sgw_ctrl_fteid;
|
|
} emm_ctx_t;
|
|
|
|
typedef struct {
|
|
ecm_state_t state;
|
|
uint32_t enb_ue_s1ap_id;
|
|
uint32_t mme_ue_s1ap_id;
|
|
struct sctp_sndrcvinfo enb_sri;
|
|
bool eit;
|
|
} ecm_ctx_t;
|
|
|
|
typedef struct {
|
|
uint8_t erab_id;
|
|
esm_state_t state;
|
|
uint8_t qci;
|
|
srslte::gtpc_f_teid_ie enb_fteid;
|
|
srslte::gtpc_f_teid_ie sgw_s1u_fteid;
|
|
srslte::gtpc_pdn_address_allocation_ie pdn_addr_alloc;
|
|
} esm_ctx_t;
|
|
|
|
typedef struct {
|
|
uint8_t eksi;
|
|
uint8_t k_asme[32];
|
|
uint8_t autn[16];
|
|
uint8_t rand[16];
|
|
uint8_t xres[16]; // minimum 6, maximum 16
|
|
uint32_t dl_nas_count;
|
|
uint32_t ul_nas_count;
|
|
srslte::CIPHERING_ALGORITHM_ID_ENUM cipher_algo;
|
|
srslte::INTEGRITY_ALGORITHM_ID_ENUM integ_algo;
|
|
uint8_t k_nas_enc[32];
|
|
uint8_t k_nas_int[32];
|
|
uint8_t k_enb[32];
|
|
LIBLTE_MME_UE_NETWORK_CAPABILITY_STRUCT ue_network_cap;
|
|
bool ms_network_cap_present;
|
|
LIBLTE_MME_MS_NETWORK_CAPABILITY_STRUCT ms_network_cap;
|
|
LIBLTE_MME_EPS_MOBILE_ID_GUTI_STRUCT guti;
|
|
} sec_ctx_t;
|
|
|
|
/*
|
|
* NAS Initialization Arguments
|
|
*/
|
|
typedef struct {
|
|
uint16_t mcc;
|
|
uint16_t mnc;
|
|
uint8_t mme_code;
|
|
uint16_t mme_group;
|
|
uint16_t tac;
|
|
uint16_t paging_timer;
|
|
std::string apn;
|
|
std::string dns;
|
|
srslte::CIPHERING_ALGORITHM_ID_ENUM cipher_algo;
|
|
srslte::INTEGRITY_ALGORITHM_ID_ENUM integ_algo;
|
|
} nas_init_t;
|
|
|
|
typedef struct {
|
|
s1ap_interface_nas* s1ap;
|
|
gtpc_interface_nas* gtpc;
|
|
hss_interface_nas* hss;
|
|
mme_interface_nas* mme;
|
|
} nas_if_t;
|
|
|
|
class nas
|
|
{
|
|
public:
|
|
nas(const nas_init_t& args, const nas_if_t& itf, srslte::log* nas_log);
|
|
void reset();
|
|
|
|
/***********************
|
|
* Initial UE messages *
|
|
***********************/
|
|
// Attach request messages
|
|
static bool handle_attach_request(uint32_t enb_ue_s1ap_id,
|
|
struct sctp_sndrcvinfo* enb_sri,
|
|
srslte::byte_buffer_t* nas_rx,
|
|
const nas_init_t& args,
|
|
const nas_if_t& itf,
|
|
srslte::log* nas_log);
|
|
|
|
static bool handle_imsi_attach_request_unknown_ue(uint32_t enb_ue_s1ap_id,
|
|
struct sctp_sndrcvinfo* enb_sri,
|
|
const LIBLTE_MME_ATTACH_REQUEST_MSG_STRUCT& attach_req,
|
|
const LIBLTE_MME_PDN_CONNECTIVITY_REQUEST_MSG_STRUCT& pdn_con_req,
|
|
const nas_init_t& args,
|
|
const nas_if_t& itf,
|
|
srslte::log* nas_log);
|
|
|
|
static bool handle_imsi_attach_request_known_ue(nas* nas_ctx,
|
|
uint32_t enb_ue_s1ap_id,
|
|
struct sctp_sndrcvinfo* enb_sri,
|
|
const LIBLTE_MME_ATTACH_REQUEST_MSG_STRUCT& attach_req,
|
|
const LIBLTE_MME_PDN_CONNECTIVITY_REQUEST_MSG_STRUCT& pdn_con_req,
|
|
srslte::byte_buffer_t* nas_rx,
|
|
const nas_init_t& args,
|
|
const nas_if_t& itf,
|
|
srslte::log* nas_log);
|
|
|
|
static bool handle_guti_attach_request_unknown_ue(uint32_t enb_ue_s1ap_id,
|
|
struct sctp_sndrcvinfo* enb_sri,
|
|
const LIBLTE_MME_ATTACH_REQUEST_MSG_STRUCT& attach_req,
|
|
const LIBLTE_MME_PDN_CONNECTIVITY_REQUEST_MSG_STRUCT& pdn_con_req,
|
|
const nas_init_t& args,
|
|
const nas_if_t& itf,
|
|
srslte::log* nas_log);
|
|
|
|
static bool handle_guti_attach_request_known_ue(nas* nas_ctx,
|
|
uint32_t enb_ue_s1ap_id,
|
|
struct sctp_sndrcvinfo* enb_sri,
|
|
const LIBLTE_MME_ATTACH_REQUEST_MSG_STRUCT& attach_req,
|
|
const LIBLTE_MME_PDN_CONNECTIVITY_REQUEST_MSG_STRUCT& pdn_con_req,
|
|
srslte::byte_buffer_t* nas_rx,
|
|
const nas_init_t& args,
|
|
const nas_if_t& itf,
|
|
srslte::log* nas_log);
|
|
|
|
// Service request messages
|
|
static bool handle_service_request(uint32_t m_tmsi,
|
|
uint32_t enb_ue_s1ap_id,
|
|
struct sctp_sndrcvinfo* enb_sri,
|
|
srslte::byte_buffer_t* nas_rx,
|
|
const nas_init_t& args,
|
|
const nas_if_t& itf,
|
|
srslte::log* nas_log);
|
|
|
|
// Dettach request messages
|
|
static bool handle_detach_request(uint32_t m_tmsi,
|
|
uint32_t enb_ue_s1ap_id,
|
|
struct sctp_sndrcvinfo* enb_sri,
|
|
srslte::byte_buffer_t* nas_rx,
|
|
const nas_init_t& args,
|
|
const nas_if_t& itf,
|
|
srslte::log* nas_log);
|
|
|
|
// Tracking area update request messages
|
|
static bool handle_tracking_area_update_request(uint32_t m_tmsi,
|
|
uint32_t enb_ue_s1ap_id,
|
|
struct sctp_sndrcvinfo* enb_sri,
|
|
srslte::byte_buffer_t* nas_rx,
|
|
const nas_init_t& args,
|
|
const nas_if_t& itf,
|
|
srslte::log* nas_log);
|
|
|
|
/* Uplink NAS messages handling */
|
|
bool handle_attach_request(srslte::byte_buffer_t* nas_rx);
|
|
bool handle_authentication_response(srslte::byte_buffer_t* nas_rx);
|
|
bool handle_security_mode_complete(srslte::byte_buffer_t* nas_rx);
|
|
bool handle_attach_complete(srslte::byte_buffer_t* nas_rx);
|
|
bool handle_esm_information_response(srslte::byte_buffer_t* nas_rx);
|
|
bool handle_identity_response(srslte::byte_buffer_t* nas_rx);
|
|
bool handle_tracking_area_update_request(srslte::byte_buffer_t* nas_rx);
|
|
bool handle_authentication_failure(srslte::byte_buffer_t* nas_rx);
|
|
bool handle_detach_request(srslte::byte_buffer_t* nas_rx);
|
|
|
|
/* Downlink NAS messages packing */
|
|
bool pack_authentication_request(srslte::byte_buffer_t* nas_buffer);
|
|
bool pack_authentication_reject(srslte::byte_buffer_t* nas_buffer);
|
|
bool pack_security_mode_command(srslte::byte_buffer_t* nas_buffer);
|
|
bool pack_esm_information_request(srslte::byte_buffer_t* nas_buffer);
|
|
bool pack_identity_request(srslte::byte_buffer_t* nas_buffer);
|
|
bool pack_emm_information(srslte::byte_buffer_t* nas_buffer);
|
|
bool pack_service_reject(srslte::byte_buffer_t* nas_buffer, uint8_t emm_cause);
|
|
bool pack_tracking_area_update_reject(srslte::byte_buffer_t* nas_buffer, uint8_t emm_cause);
|
|
bool pack_attach_accept(srslte::byte_buffer_t* nas_buffer);
|
|
|
|
/* Security functions */
|
|
bool integrity_check(srslte::byte_buffer_t* pdu);
|
|
bool short_integrity_check(srslte::byte_buffer_t* pdu);
|
|
void integrity_generate(srslte::byte_buffer_t* pdu, uint8_t* mac);
|
|
void cipher_decrypt(srslte::byte_buffer_t* pdu);
|
|
void cipher_encrypt(srslte::byte_buffer_t* pdu);
|
|
|
|
/*Timer functions*/
|
|
bool start_timer(enum nas_timer_type type);
|
|
bool expire_timer(enum nas_timer_type type);
|
|
|
|
/* UE Context */
|
|
emm_ctx_t m_emm_ctx = {};
|
|
ecm_ctx_t m_ecm_ctx = {};
|
|
esm_ctx_t m_esm_ctx[MAX_ERABS_PER_UE] = {};
|
|
sec_ctx_t m_sec_ctx = {};
|
|
|
|
private:
|
|
srslte::byte_buffer_pool* m_pool = nullptr;
|
|
srslte::log* m_nas_log = nullptr;
|
|
gtpc_interface_nas* m_gtpc = nullptr;
|
|
s1ap_interface_nas* m_s1ap = nullptr;
|
|
hss_interface_nas* m_hss = nullptr;
|
|
mme_interface_nas* m_mme = nullptr;
|
|
|
|
uint16_t m_mcc = 0;
|
|
uint16_t m_mnc = 0;
|
|
uint16_t m_mme_group = 0;
|
|
uint16_t m_mme_code = 0;
|
|
uint16_t m_tac = 0;
|
|
std::string m_apn;
|
|
std::string m_dns;
|
|
|
|
// Timers timeout values
|
|
uint16_t m_t3413 = 0;
|
|
|
|
// Timer functions
|
|
bool start_t3413();
|
|
bool expire_t3413();
|
|
};
|
|
|
|
} // namespace srsepc
|
|
#endif // SRSEPC_NAS_H
|