mirror of https://github.com/PentHertz/srsLTE.git
141 lines
4.3 KiB
C++
141 lines
4.3 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 SRSUE_USIM_BASE_H
|
|
#define SRSUE_USIM_BASE_H
|
|
|
|
#include "srslte/common/common.h"
|
|
#include "srslte/common/log.h"
|
|
#include "srslte/common/security.h"
|
|
#include "srslte/interfaces/ue_interfaces.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) {}
|
|
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_base(srslte::log* log_);
|
|
virtual ~usim_base();
|
|
static std::unique_ptr<usim_base> get_instance(usim_args_t* args, srslte::log* log_);
|
|
|
|
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_imei_vec(uint8_t* imei_, uint32_t n) final;
|
|
bool get_home_plmn_id(srslte::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,
|
|
srslte::CIPHERING_ALGORITHM_ID_ENUM cipher_algo,
|
|
srslte::INTEGRITY_ALGORITHM_ID_ENUM integ_algo) final;
|
|
|
|
// RRC interface
|
|
void generate_as_keys(uint8_t* k_asme, uint32_t count_ul, srslte::as_security_config_t* sec_cfg) final;
|
|
void generate_as_keys_ho(uint32_t pci, uint32_t earfcn, int ncc, srslte::as_security_config_t* sec_cfg) final;
|
|
void store_keys_before_ho(const srslte::as_security_config_t& as_ctx) final;
|
|
void restore_keys_from_failed_ho(srslte::as_security_config_t* as_ctx) 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
|
|
srslte::log* log = nullptr;
|
|
|
|
// 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)
|
|
srslte::k_enb_context_t k_enb_ctx = {};
|
|
|
|
// Helpers to restore security context if HO fails
|
|
srslte::k_enb_context_t old_k_enb_ctx = {};
|
|
srslte::as_security_config_t old_as_ctx = {};
|
|
};
|
|
|
|
} // namespace srsue
|
|
|
|
#endif // SRSUE_USIM_BASE_H
|