mirror of https://github.com/PentHertz/srsLTE.git
141 lines
4.3 KiB
C++
141 lines
4.3 KiB
C++
/**
|
|
*
|
|
* \section COPYRIGHT
|
|
*
|
|
* Copyright 2013-2021 Software Radio Systems Limited
|
|
*
|
|
* By using this file, you agree to the terms and conditions set
|
|
* forth in the LICENSE file which can be found at the top level of
|
|
* the distribution.
|
|
*
|
|
*/
|
|
|
|
#ifndef SRSUE_USIM_BASE_H
|
|
#define SRSUE_USIM_BASE_H
|
|
|
|
#include "srsran/common/common.h"
|
|
#include "srsran/common/security.h"
|
|
#include "srsran/interfaces/ue_usim_interfaces.h"
|
|
#include "srsran/srslog/srslog.h"
|
|
#include <string>
|
|
|
|
namespace srsue {
|
|
|
|
#define AKA_RAND_LEN 16
|
|
#define AKA_AUTN_LEN 16
|
|
#define AKA_AUTS_LEN 14
|
|
#define RES_MAX_LEN 16
|
|
#define MAC_LEN 8
|
|
#define IK_LEN 16
|
|
#define CK_LEN 16
|
|
#define AK_LEN 6
|
|
#define SQN_LEN 6
|
|
|
|
#define KEY_LEN 32
|
|
|
|
typedef enum {
|
|
auth_algo_milenage = 0,
|
|
auth_algo_xor,
|
|
} auth_algo_t;
|
|
|
|
class usim_args_t
|
|
{
|
|
public:
|
|
usim_args_t() : using_op(false) {}
|
|
~usim_args_t() = default;
|
|
std::string mode;
|
|
std::string algo;
|
|
bool using_op;
|
|
std::string op;
|
|
std::string opc;
|
|
std::string imsi;
|
|
std::string imei;
|
|
std::string k;
|
|
std::string pin;
|
|
std::string reader;
|
|
};
|
|
|
|
class usim_base : public usim_interface_nas, public usim_interface_rrc, public usim_interface_rrc_nr
|
|
{
|
|
public:
|
|
explicit usim_base(srslog::basic_logger& logger);
|
|
virtual ~usim_base();
|
|
static std::unique_ptr<usim_base> get_instance(usim_args_t* args, srslog::basic_logger& logger);
|
|
|
|
virtual int init(usim_args_t* args) = 0;
|
|
virtual void stop() = 0;
|
|
|
|
// NAS interface
|
|
std::string get_imsi_str() final;
|
|
std::string get_imei_str() final;
|
|
|
|
bool get_imsi_vec(uint8_t* imsi_, uint32_t n) final;
|
|
bool get_home_mcc_bytes(uint8_t* mcc_, uint32_t n) final;
|
|
bool get_home_mnc_bytes(uint8_t* mnc_, uint32_t n) final;
|
|
bool get_home_msin_bcd(uint8_t* msin_, uint32_t n) final;
|
|
bool get_imei_vec(uint8_t* imei_, uint32_t n) final;
|
|
bool get_home_plmn_id(srsran::plmn_id_t* home_plmn_id) final;
|
|
|
|
virtual auth_result_t generate_authentication_response(uint8_t* rand,
|
|
uint8_t* autn_enb,
|
|
uint16_t mcc,
|
|
uint16_t mnc,
|
|
uint8_t* res,
|
|
int* res_len,
|
|
uint8_t* k_asme) = 0;
|
|
|
|
void generate_nas_keys(uint8_t* k_asme,
|
|
uint8_t* k_nas_enc,
|
|
uint8_t* k_nas_int,
|
|
srsran::CIPHERING_ALGORITHM_ID_ENUM cipher_algo,
|
|
srsran::INTEGRITY_ALGORITHM_ID_ENUM integ_algo) final;
|
|
|
|
// RRC interface
|
|
void generate_as_keys(uint8_t* k_asme, uint32_t count_ul, srsran::as_security_config_t* sec_cfg) final;
|
|
void generate_as_keys_ho(uint32_t pci, uint32_t earfcn, int ncc, srsran::as_security_config_t* sec_cfg) final;
|
|
void store_keys_before_ho(const srsran::as_security_config_t& as_ctx) final;
|
|
void restore_keys_from_failed_ho(srsran::as_security_config_t* as_ctx) final;
|
|
|
|
// NR RRC interface
|
|
bool generate_nr_context(uint16_t sk_counter, srsran::as_security_config_t* sec_cfg) final;
|
|
bool update_nr_context(srsran::as_security_config_t* sec_cfg) final;
|
|
|
|
// Helpers
|
|
std::string get_mcc_str(const uint8_t* imsi_vec);
|
|
virtual std::string get_mnc_str(const uint8_t* imsi_vec, std::string mcc_str) = 0;
|
|
|
|
protected:
|
|
bool initiated = false;
|
|
|
|
// Logging
|
|
srslog::basic_logger& logger;
|
|
|
|
// User data
|
|
// 3GPP 33.102 v10.0.0 Annex H
|
|
uint64_t imsi = 0;
|
|
uint64_t imei = 0;
|
|
std::string imsi_str;
|
|
std::string imei_str;
|
|
|
|
// Security variables
|
|
uint8_t ck[CK_LEN] = {};
|
|
uint8_t ik[IK_LEN] = {};
|
|
uint8_t ak[AK_LEN] = {};
|
|
uint8_t k_asme[KEY_LEN] = {};
|
|
uint8_t k_enb_star[KEY_LEN] = {};
|
|
uint8_t k_enb_initial[KEY_LEN] = {};
|
|
uint8_t auts[AKA_AUTS_LEN] = {};
|
|
|
|
// Current K_eNB context (K_eNB, NH and NCC)
|
|
srsran::k_enb_context_t k_enb_ctx = {};
|
|
srsran::k_gnb_context_t k_gnb_ctx = {};
|
|
|
|
// Helpers to restore security context if HO fails
|
|
srsran::k_enb_context_t old_k_enb_ctx = {};
|
|
srsran::as_security_config_t old_as_ctx = {};
|
|
};
|
|
|
|
} // namespace srsue
|
|
|
|
#endif // SRSUE_USIM_BASE_H
|