srsLTE/lib/include/srsran/asn1/nas_5g_utils.h

96 lines
2.7 KiB
C++

/**
* Copyright 2013-2022 Software Radio Systems Limited
*
* This file is part of srsRAN.
*
* srsRAN 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.
*
* srsRAN 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 SRSRAN_NAS_5G_UTILS_H
#define SRSRAN_NAS_5G_UTILS_H
#include "srsran/asn1/asn1_utils.h"
#include "srsran/common/byte_buffer.h"
#include "srsran/config.h"
using namespace asn1;
namespace srsran {
namespace nas_5g {
struct ecies_scheme_profile_a_out {
uint8_t ecc_ephemeral_key[33];
std::vector<uint8_t> ciphertext;
uint8_t mac_tag[8];
};
struct ecies_scheme_profile_b_out {
uint8_t ecc_ephemeral_key[32];
std::vector<uint8_t> ciphertext;
uint8_t mac_tag[8];
};
template <typename Enum, int bl>
SRSASN_CODE unpack_enum(asn1::cbit_ref& bref, Enum* e)
{
uint32_t tmp = {};
HANDLE_CODE(bref.unpack(tmp, bl));
*e = static_cast<Enum>(tmp);
return SRSASN_SUCCESS;
}
template <typename Enum, int bl>
SRSASN_CODE pack_enum(asn1::bit_ref& bref, Enum e)
{
uint32_t tmp = static_cast<uint32_t>(e);
HANDLE_CODE(bref.pack(tmp, bl));
return SRSASN_SUCCESS;
}
template <class EnumType, uint32_t bit_length_>
class nas_enumerated : public EnumType
{
public:
static const uint32_t bit_length = bit_length_;
nas_enumerated() {}
nas_enumerated(typename EnumType::options o) { EnumType::value = o; }
SRSASN_CODE pack(asn1::bit_ref& bref) const
{
uint32_t tmp = static_cast<uint32_t>(EnumType::value);
HANDLE_CODE(bref.pack(tmp, bit_length));
return SRSASN_SUCCESS;
}
SRSASN_CODE unpack(asn1::cbit_ref& bref)
{
uint32_t tmp = {};
HANDLE_CODE(bref.unpack(tmp, bit_length));
*this = static_cast<typename EnumType::options>(tmp);
return SRSASN_SUCCESS;
}
EnumType& operator=(EnumType v)
{
EnumType::value = v;
return *this;
}
operator typename EnumType::options() const { return EnumType::value; }
};
SRSASN_CODE unpack_mcc_mnc(uint8_t* mcc_bytes, uint8_t* mnc_bytes, asn1::cbit_ref& bref);
SRSASN_CODE pack_mcc_mnc(uint8_t* mcc_bytes, uint8_t* mnc_bytes, asn1::bit_ref& bref);
} // namespace nas_5g
} // namespace srsran
#endif // MANUAL_H