/*
  Copyright 2013-2017 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/.
*/

#include "srslte/asn1/rrc_asn1.h"
#include <cstdio>
#include <iostream>

#define TESTASSERT(cond)                                                                                               \
  {                                                                                                                    \
    if (!(cond)) {                                                                                                     \
      std::cout << "[" << __FUNCTION__ << "][Line " << __LINE__ << "]: FAIL at " << (#cond) << std::endl;              \
      return -1;                                                                                                       \
    }                                                                                                                  \
  }

using namespace asn1;
using namespace asn1::rrc;

struct ConsoleLogger {
  ConsoleLogger(const std::string& layer_) : layer(layer_) {}

  void log(srsasn_logger_level_t log_level, const char* str)
  {
    switch (log_level) {
      case LOG_LEVEL_DEBUG:
        printf("[%s][D] %s", layer.c_str(), str);
        break;
      case LOG_LEVEL_INFO:
        printf("[%s][I] %s", layer.c_str(), str);
        break;
      case LOG_LEVEL_WARN:
        printf("[%s][W] %s", layer.c_str(), str);
        break;
      case LOG_LEVEL_ERROR:
        printf("[%s][E] %s", layer.c_str(), str);
        break;
      default:
        break;
    }
  }

private:
  std::string layer;
};
void print_console(srsasn_logger_level_t log_level, void* ctx, const char* str)
{
  ConsoleLogger* logger = (ConsoleLogger*)ctx;
  logger->log(log_level, str);
}

struct TestLogger {
  TestLogger(const std::string& layer_) : layer(layer_) {}
  void log(srsasn_logger_level_t log_level, const char* str)
  {
    last_level = log_level;
    last_str   = str;
  }
  std::string           layer;
  srsasn_logger_level_t last_level;
  std::string           last_str;
};
void test_print(srsasn_logger_level_t log_level, void* ctx, const char* str)
{
  TestLogger* logger = (TestLogger*)ctx;
  logger->log(log_level, str);
}

ConsoleLogger asn_logger("ASN");
ConsoleLogger rrc_logger("RRC");
TestLogger    test_logger("TEST");

// TESTS

int test_generic()
{
  pusch_enhance_cfg_r14_c choice_type1;

  // make suce a choice is always started in null mode
  TESTASSERT(choice_type1.type() == pusch_enhance_cfg_r14_c::types::nulltype);

  // test logger handler
  rrc_log_register_handler(&test_logger, test_print);
  std::string test_str = "This is a console test to see if the RRC logger is working fine\n";
  rrc_log_print(LOG_LEVEL_INFO, test_str.c_str());
  TESTASSERT(test_logger.last_str == test_str);
  TESTASSERT(test_logger.last_level == LOG_LEVEL_INFO);
  // go back to original logger
  rrc_log_register_handler(&rrc_logger, print_console);

  // Test deep copy of choice types
  sib_type14_r11_s::eab_param_r11_c_ choice2;
  choice2.set_eab_per_plmn_list_r11();
  choice2.eab_per_plmn_list_r11().resize(5);
  for (uint32_t i = 0; i < 5; ++i) {
    choice2.eab_per_plmn_list_r11()[i].eab_cfg_r11_present = true;
  }
  {
    sib_type14_r11_s::eab_param_r11_c_ choice2_cpy(choice2);
    TESTASSERT(choice2_cpy.type() == sib_type14_r11_s::eab_param_r11_c_::types::eab_per_plmn_list_r11);
    TESTASSERT(choice2_cpy.eab_per_plmn_list_r11()[3].eab_cfg_r11_present);
    TESTASSERT(choice2_cpy.eab_per_plmn_list_r11().data() != choice2.eab_per_plmn_list_r11().data());

    choice2_cpy = choice2;
    TESTASSERT(choice2_cpy.type() == sib_type14_r11_s::eab_param_r11_c_::types::eab_per_plmn_list_r11);
    TESTASSERT(choice2_cpy.eab_per_plmn_list_r11()[3].eab_cfg_r11_present);
    TESTASSERT(choice2_cpy.eab_per_plmn_list_r11().data() != choice2.eab_per_plmn_list_r11().data());
  }

  asn1::rrc::setup_e e;
  e.value = setup_opts::setup;
  TESTASSERT(e.value == setup_e::setup);

  //
  // srb_to_add_mod_s::rlc_cfg_c_ choice_type2;
  // choice_type2.set(srb_to_add_mod_s::rlc_cfg_c_::types::explicit_value);
  // printf("%s\n", choice_type2.explicit_value().type().to_string().c_str());

  return 0;
}

int test_json_printer()
{
  as_cfg_s cfg;
  //  cfg.source_other_cfg_r9 = asn1::make_copy_ptr(other_cfg_r9_s());
  // To avoid errors with unitialized enums
  cfg.source_mib.phich_cfg.phich_res.value = phich_cfg_s::phich_res_e_::half;
  cfg.source_mib.phich_cfg.phich_dur.value = phich_cfg_s::phich_dur_e_::normal;

  json_writer jw;
  //  cfg.to_json(jw);
  //  printf("%s\n", jw.to_string().c_str());
  return 0;
}

int test_mib_msg()
{
  uint8_t  rrc_msg[]   = {0x94, 0x64, 0xC0};
  uint32_t rrc_msg_len = sizeof(rrc_msg);
  bit_ref  bref(&rrc_msg[0], sizeof(rrc_msg));
  bit_ref  bref0(&rrc_msg[0], sizeof(rrc_msg));

  bcch_bch_msg_s bcch_bch_msg;
  bcch_bch_msg.unpack(bref);

  bcch_bch_msg_type_s& bcch_msg = bcch_bch_msg.msg;
  TESTASSERT(bcch_msg.dl_bw == mib_s::dl_bw_e_::n75);
  TESTASSERT(bcch_msg.phich_cfg.phich_dur == phich_cfg_s::phich_dur_e_::extended);
  TESTASSERT(bcch_msg.phich_cfg.phich_res == phich_cfg_s::phich_res_e_::half);
  TESTASSERT(bcch_msg.sys_frame_num == "00011001");
  TESTASSERT(bcch_msg.sched_info_sib1_br_r13 == 6);
  TESTASSERT(not bcch_msg.sys_info_unchanged_br_r15);
  TESTASSERT(bcch_msg.spare == "0000");

  TESTASSERT((uint32_t)bref.distance(bref0) / 8 == rrc_msg_len);

  uint8_t rrc_msg2[rrc_msg_len];
  bit_ref bref2(&rrc_msg2[0], sizeof(rrc_msg2));
  bcch_bch_msg.pack(bref2);
  TESTASSERT(bref.distance(bref0) == bref2.distance(bit_ref(&rrc_msg2[0], sizeof(rrc_msg2))));
  TESTASSERT(memcmp(rrc_msg2, rrc_msg, rrc_msg_len) == 0);

  //  // test print
  //  json_writer j;
  //  bcch_bch_msg.to_json(j);
  //  std::cout << j.to_string() << std::endl;

  return 0;
}

int test_bcch_dl_sch_msg()
{
  // 000149001250400800094000A03F01000A7FC9800104286C000C
  uint8_t  rrc_msg[]   = {0x00, 0x01, 0x49, 0x00, 0x12, 0x50, 0x40, 0x08, 0x00, 0x09, 0x40, 0x00, 0xA0,
                       0x3F, 0x01, 0x00, 0x0A, 0x7F, 0xC9, 0x80, 0x01, 0x04, 0x28, 0x6C, 0x00, 0x0C};
  uint32_t rrc_msg_len = sizeof(rrc_msg);
  bit_ref  bref(&rrc_msg[0], sizeof(rrc_msg));
  bit_ref  bref0(&rrc_msg[0], sizeof(rrc_msg));

  bcch_dl_sch_msg_s bcch_msg;
  bcch_msg.unpack(bref);

  TESTASSERT(bcch_msg.msg.type() == bcch_dl_sch_msg_type_c::types::c1);
  TESTASSERT(bcch_msg.msg.c1().type() == bcch_dl_sch_msg_type_c::c1_c_::types::sys_info);
  sys_info_s::crit_exts_c_* sinfo = &bcch_msg.msg.c1().sys_info().crit_exts;
  TESTASSERT(sinfo->type() == sys_info_s::crit_exts_c_::types::sys_info_r8);
  sys_info_r8_ies_s& sys_r8 = sinfo->sys_info_r8();
  TESTASSERT(not sys_r8.non_crit_ext_present);
  TESTASSERT(sys_r8.sib_type_and_info.size() == 1);
  TESTASSERT(sys_r8.sib_type_and_info[0].type() == sib_info_item_c::types::sib2);
  sib_type2_s* sib2 = &(sys_r8.sib_type_and_info[0].sib2());
  TESTASSERT(sib2->ac_barr_info_present);
  TESTASSERT(not sib2->mbsfn_sf_cfg_list_present);
  TESTASSERT(sib2->ac_barr_info.ac_barr_for_mo_sig_present);
  TESTASSERT(not sib2->ac_barr_info.ac_barr_for_mo_data_present);
  TESTASSERT(not sib2->ac_barr_info.ac_barr_for_emergency);
  TESTASSERT(sib2->ac_barr_info.ac_barr_for_mo_sig.ac_barr_factor == ac_barr_cfg_s::ac_barr_factor_e_::p60);
  TESTASSERT(sib2->ac_barr_info.ac_barr_for_mo_sig.ac_barr_time == ac_barr_cfg_s::ac_barr_time_e_::s4);
  TESTASSERT(sib2->ac_barr_info.ac_barr_for_mo_sig.ac_barr_for_special_ac == "00000");
  rr_cfg_common_sib_s* res_sib = &sib2->rr_cfg_common;
  TESTASSERT(not res_sib->rach_cfg_common.preamb_info.preambs_group_a_cfg_present);
  TESTASSERT(res_sib->rach_cfg_common.preamb_info.nof_ra_preambs ==
             rach_cfg_common_s::preamb_info_s_::nof_ra_preambs_e_::n40);
  TESTASSERT(res_sib->rach_cfg_common.pwr_ramp_params.pwr_ramp_step == pwr_ramp_params_s::pwr_ramp_step_e_::db0);
  TESTASSERT(res_sib->rach_cfg_common.pwr_ramp_params.preamb_init_rx_target_pwr ==
             pwr_ramp_params_s::preamb_init_rx_target_pwr_e_::dbm_minus100);
  TESTASSERT(res_sib->rach_cfg_common.ra_supervision_info.preamb_trans_max == preamb_trans_max_e::n3);
  TESTASSERT(res_sib->rach_cfg_common.ra_supervision_info.ra_resp_win_size ==
             rach_cfg_common_s::ra_supervision_info_s_::ra_resp_win_size_e_::sf6);
  TESTASSERT(res_sib->rach_cfg_common.ra_supervision_info.mac_contention_resolution_timer ==
             rach_cfg_common_s::ra_supervision_info_s_::mac_contention_resolution_timer_e_::sf8);
  TESTASSERT(res_sib->rach_cfg_common.max_harq_msg3_tx == 1);

  TESTASSERT(res_sib->bcch_cfg.mod_period_coeff == bcch_cfg_s::mod_period_coeff_e_::n2);
  TESTASSERT(res_sib->pcch_cfg.default_paging_cycle == pcch_cfg_s::default_paging_cycle_e_::rf128);
  TESTASSERT(res_sib->pcch_cfg.nb == pcch_cfg_s::nb_e_::four_t);
  TESTASSERT(res_sib->prach_cfg.root_seq_idx == 0);
  TESTASSERT(res_sib->prach_cfg.prach_cfg_info.prach_cfg_idx == 18);
  TESTASSERT(res_sib->prach_cfg.prach_cfg_info.high_speed_flag);
  TESTASSERT(res_sib->prach_cfg.prach_cfg_info.zero_correlation_zone_cfg == 0);
  TESTASSERT(res_sib->prach_cfg.prach_cfg_info.prach_freq_offset == 0);
  TESTASSERT(res_sib->pdsch_cfg_common.ref_sig_pwr == -50);
  TESTASSERT(res_sib->pdsch_cfg_common.p_b == 0);
  //...
  TESTASSERT(res_sib->pucch_cfg_common.n1_pucch_an == 511);
  TESTASSERT(res_sib->srs_ul_cfg_common.type() == srs_ul_cfg_common_c::types::release);
  TESTASSERT(res_sib->ul_pwr_ctrl_common.p0_nominal_pusch == -50);
  //...
  TESTASSERT(sib2->freq_info.add_spec_emission == 1);
  TESTASSERT(sib2->time_align_timer_common == time_align_timer_e::sf1920);

  TESTASSERT(ceil(bref.distance(bref0) / 8.0) == rrc_msg_len);

  uint8_t rrc_msg2[rrc_msg_len];
  bit_ref bref2(&rrc_msg2[0], sizeof(rrc_msg2));
  bcch_msg.pack(bref2);
  TESTASSERT(bref.distance(bref0) == bref2.distance(bit_ref(&rrc_msg2[0], sizeof(rrc_msg2))));
  TESTASSERT(memcmp(rrc_msg2, rrc_msg, rrc_msg_len) == 0);

  //  // test print
  //  json_writer j;
  //  bcch_msg.to_json(j);
  //  std::cout << j.to_string() << std::endl;

  return 0;
}

int test_bcch_dl_sch_msg2()
{
  // 406404e100070019b018c06010A940
  uint8_t  rrc_msg[]   = {0x40, 0x64, 0x04, 0xe1, 0x00, 0x07, 0x00, 0x19, 0xb0, 0x18, 0xc0, 0x60, 0x10, 0xA9, 0x40};
  uint32_t rrc_msg_len = sizeof(rrc_msg);
  bit_ref  bref(&rrc_msg[0], sizeof(rrc_msg));
  bit_ref  bref0(&rrc_msg[0], sizeof(rrc_msg));

  bcch_dl_sch_msg_s bcch_msg;
  bcch_msg.unpack(bref);

  TESTASSERT(bcch_msg.msg.type() == bcch_dl_sch_msg_type_c::types::c1);
  TESTASSERT(bcch_msg.msg.c1().type() == bcch_dl_sch_msg_type_c::c1_c_::types::sib_type1);
  sib_type1_s& sib1 = bcch_msg.msg.c1().sib_type1();
  TESTASSERT(not sib1.p_max_present);
  TESTASSERT(sib1.sched_info_list.size() == 1);
  TESTASSERT(sib1.sched_info_list[0].si_periodicity.value == sched_info_s::si_periodicity_e_::rf16);
  TESTASSERT(sib1.sched_info_list[0].sib_map_info.size() == 1);
  TESTASSERT(sib1.sched_info_list[0].sib_map_info[0] == sib_type_e::sib_type13_v920);

  TESTASSERT(ceil(bref.distance(bref0) / 8.0) == rrc_msg_len);

  uint8_t rrc_msg2[rrc_msg_len];
  bit_ref bref2(&rrc_msg2[0], sizeof(rrc_msg2));
  bcch_msg.pack(bref2);
  TESTASSERT(bref.distance(bref0) == bref2.distance(bit_ref(&rrc_msg2[0], sizeof(rrc_msg2))));
  TESTASSERT(memcmp(rrc_msg2, rrc_msg, rrc_msg_len) == 0);

  return 0;
}

int test_bcch_dl_sch_msg3()
{
  // 00830992B7EC9300A3424B000C000500205D6AAAF04200C01DDC801C4880030010A713228500
  uint8_t  rrc_msg[]   = {0x00, 0x83, 0x09, 0x92, 0xB7, 0xEC, 0x93, 0x00, 0xA3, 0x42, 0x4B, 0x00, 0x0C,
                       0x00, 0x05, 0x00, 0x20, 0x5D, 0x6A, 0xAA, 0xF0, 0x42, 0x00, 0xC0, 0x1D, 0xDC,
                       0x80, 0x1C, 0x48, 0x80, 0x03, 0x00, 0x10, 0xA7, 0x13, 0x22, 0x85, 0x00};
  uint32_t rrc_msg_len = sizeof(rrc_msg);
  bit_ref  bref(&rrc_msg[0], sizeof(rrc_msg));

  bcch_dl_sch_msg_s bcch_msg;
  bcch_msg.unpack(bref);

  TESTASSERT(bcch_msg.msg.type() == bcch_dl_sch_msg_type_c::types::c1);

  TESTASSERT(ceil(bref.distance(rrc_msg) / 8.0) == rrc_msg_len);

  uint8_t rrc_msg2[rrc_msg_len];
  bit_ref bref2(&rrc_msg2[0], sizeof(rrc_msg2));
  bcch_msg.pack(bref2);

  bref = bit_ref(&rrc_msg2[0], sizeof(rrc_msg2));
  bcch_msg.unpack(bref);
  TESTASSERT(bref.distance(rrc_msg2) == bref2.distance(rrc_msg2));

  bref = bit_ref(&rrc_msg[0], sizeof(rrc_msg));
  bcch_msg.pack(bref);

  TESTASSERT(bref.distance(rrc_msg) == bref2.distance(rrc_msg2));
  TESTASSERT(memcmp(rrc_msg2, rrc_msg, bref.distance_bytes(rrc_msg)) == 0);

  return 0;
}

int test_dl_dcch_msg()
{
  // 20021008000C406000
  uint8_t  rrc_msg[]   = {0x20, 0x02, 0x10, 0x08, 0x00, 0x0C, 0x40, 0x60, 0x00};
  uint32_t rrc_msg_len = sizeof(rrc_msg);
  bit_ref  bref(&rrc_msg[0], sizeof(rrc_msg));
  bit_ref  bref0(&rrc_msg[0], sizeof(rrc_msg));

  dl_dcch_msg_s dl_dcch_msg;
  dl_dcch_msg.unpack(bref);

  TESTASSERT(ceil(bref.distance(bref0) / 8.0) == rrc_msg_len);

  TESTASSERT(dl_dcch_msg.msg.type() == dl_dcch_msg_type_c::types::c1);
  TESTASSERT(dl_dcch_msg.msg.c1().type() == dl_dcch_msg_type_c::c1_c_::types::rrc_conn_recfg);
  rrc_conn_recfg_s* rrc_recfg = &dl_dcch_msg.msg.c1().rrc_conn_recfg();
  TESTASSERT(rrc_recfg->rrc_transaction_id == 0);
  TESTASSERT(rrc_recfg->crit_exts.type() == rrc_conn_recfg_s::crit_exts_c_::types::c1);
  TESTASSERT(rrc_recfg->crit_exts.c1().type() == rrc_conn_recfg_s::crit_exts_c_::c1_c_::types::rrc_conn_recfg_r8);
  rrc_conn_recfg_r8_ies_s* rrc_recfg_r8 = &rrc_recfg->crit_exts.c1().rrc_conn_recfg_r8();
  TESTASSERT(not rrc_recfg_r8->meas_cfg_present);
  TESTASSERT(not rrc_recfg_r8->mob_ctrl_info_present);
  TESTASSERT(not rrc_recfg_r8->ded_info_nas_list_present);
  TESTASSERT(rrc_recfg_r8->rr_cfg_ded_present);
  TESTASSERT(not rrc_recfg_r8->security_cfg_ho_present);
  TESTASSERT(not rrc_recfg_r8->non_crit_ext_present);

  //...
  TESTASSERT(rrc_recfg_r8->rr_cfg_ded.drb_to_add_mod_list_present);
  TESTASSERT(rrc_recfg_r8->rr_cfg_ded.drb_to_add_mod_list.size() == 1);
  drb_to_add_mod_s* drb = &rrc_recfg_r8->rr_cfg_ded.drb_to_add_mod_list[0];
  TESTASSERT(drb->drb_id == 1);
  //...
  TESTASSERT(drb->rlc_cfg_v1510.is_present());

  uint8_t rrc_msg2[rrc_msg_len];
  bit_ref bref2(&rrc_msg2[0], sizeof(rrc_msg2)), bref2_0(&rrc_msg2[0], sizeof(rrc_msg2));
  dl_dcch_msg.pack(bref2); // FIXME: Should I generate a pack/unpack method for RLC-Config-v1510???
  TESTASSERT(bref.distance(bref0) == bref2.distance(bref2_0));
  TESTASSERT(memcmp(rrc_msg2, rrc_msg, rrc_msg_len) == 0);

  //  // test print
  //  json_writer j;
  //  dl_dcch_msg.to_json(j);
  //  std::cout << j.to_string() << std::endl;

  return 0;
}

int ue_rrc_conn_recfg_r15_v10_test()
{
  // 201695a8000005143a0002900878b0000046625a03593800000000083a100a48aa1a2780280002a782800002a783000002a78400000001c2900e080848e0434b73a32b93732ba0336b73198181b0336b1b19a1a980233b8393982808c8005332f037f7f7d7d7f7f2f83027a12027a122805fb2a7830400000f38900f78b962ca4f5380dfb9c0327002ea03a03b1793400f40010800d9809016cda8141a0020c8287000b001efb00024a082120205024a04e3f0d00000
  uint8_t rrc_msg[] = {
      0x20, 0x16, 0x95, 0xa8, 0x00, 0x00, 0x05, 0x14, 0x3a, 0x00, 0x02, 0x90, 0x08, 0x78, 0xb0, 0x00, 0x00, 0x46, 0x62,
      0x5a, 0x03, 0x59, 0x38, 0x00, 0x00, 0x00, 0x00, 0x08, 0x3a, 0x10, 0x0a, 0x48, 0xaa, 0x1a, 0x27, 0x80, 0x28, 0x00,
      0x02, 0xa7, 0x82, 0x80, 0x00, 0x02, 0xa7, 0x83, 0x00, 0x00, 0x02, 0xa7, 0x84, 0x00, 0x00, 0x00, 0x01, 0xc2, 0x90,
      0x0e, 0x08, 0x08, 0x48, 0xe0, 0x43, 0x4b, 0x73, 0xa3, 0x2b, 0x93, 0x73, 0x2b, 0xa0, 0x33, 0x6b, 0x73, 0x19, 0x81,
      0x81, 0xb0, 0x33, 0x6b, 0x1b, 0x19, 0xa1, 0xa9, 0x80, 0x23, 0x3b, 0x83, 0x93, 0x98, 0x28, 0x08, 0xc8, 0x00, 0x53,
      0x32, 0xf0, 0x37, 0xf7, 0xf7, 0xd7, 0xd7, 0xf7, 0xf2, 0xf8, 0x30, 0x27, 0xa1, 0x20, 0x27, 0xa1, 0x22, 0x80, 0x5f,
      0xb2, 0xa7, 0x83, 0x04, 0x00, 0x00, 0x0f, 0x38, 0x90, 0x0f, 0x78, 0xb9, 0x62, 0xca, 0x4f, 0x53, 0x80, 0xdf, 0xb9,
      0xc0, 0x32, 0x70, 0x02, 0xea, 0x03, 0xa0, 0x3b, 0x17, 0x93, 0x40, 0x0f, 0x40, 0x01, 0x08, 0x00, 0xd9, 0x80, 0x90,
      0x16, 0xcd, 0xa8, 0x14, 0x1a, 0x00, 0x20, 0xc8, 0x28, 0x70, 0x00, 0xb0, 0x01, 0xef, 0xb0, 0x00, 0x24, 0xa0, 0x82,
      0x12, 0x02, 0x05, 0x02, 0x4a, 0x04, 0xe3, 0xf0, 0xd0, 0x00, 0x00};
  uint32_t rrc_msg_len = sizeof(rrc_msg);
  bit_ref  bref(&rrc_msg[0], sizeof(rrc_msg));
  bit_ref  bref0(&rrc_msg[0], sizeof(rrc_msg));

  dl_dcch_msg_s dl_dcch_msg;
  dl_dcch_msg.unpack(bref);

  TESTASSERT(ceil(bref.distance(bref0) / 8.0) == rrc_msg_len);

  TESTASSERT(dl_dcch_msg.msg.type() == dl_dcch_msg_type_c::types::c1);
  TESTASSERT(dl_dcch_msg.msg.c1().type() == dl_dcch_msg_type_c::c1_c_::types::rrc_conn_recfg);
  rrc_conn_recfg_s* rrc_recfg = &dl_dcch_msg.msg.c1().rrc_conn_recfg();
  TESTASSERT(rrc_recfg->rrc_transaction_id == 0);
  TESTASSERT(rrc_recfg->crit_exts.type() == rrc_conn_recfg_s::crit_exts_c_::types::c1);
  TESTASSERT(rrc_recfg->crit_exts.c1().type() == rrc_conn_recfg_s::crit_exts_c_::c1_c_::types::rrc_conn_recfg_r8);
  rrc_conn_recfg_r8_ies_s* rrc_recfg_r8 = &rrc_recfg->crit_exts.c1().rrc_conn_recfg_r8();
  TESTASSERT(rrc_recfg_r8->meas_cfg_present);
  TESTASSERT(not rrc_recfg_r8->mob_ctrl_info_present);
  TESTASSERT(rrc_recfg_r8->ded_info_nas_list_present);
  TESTASSERT(rrc_recfg_r8->rr_cfg_ded_present);
  TESTASSERT(not rrc_recfg_r8->security_cfg_ho_present);
  TESTASSERT(rrc_recfg_r8->non_crit_ext_present);

  // MeasConfig
  TESTASSERT(not rrc_recfg_r8->meas_cfg.meas_obj_to_rem_list_present);
  TESTASSERT(rrc_recfg_r8->meas_cfg.meas_obj_to_add_mod_list_present);
  TESTASSERT(not rrc_recfg_r8->meas_cfg.report_cfg_to_rem_list_present);
  TESTASSERT(rrc_recfg_r8->meas_cfg.report_cfg_to_add_mod_list_present);
  TESTASSERT(not rrc_recfg_r8->meas_cfg.meas_id_to_rem_list_present);
  TESTASSERT(rrc_recfg_r8->meas_cfg.meas_id_to_add_mod_list_present);
  TESTASSERT(rrc_recfg_r8->meas_cfg.quant_cfg_present);
  TESTASSERT(rrc_recfg_r8->meas_cfg.s_measure_present);
  TESTASSERT(rrc_recfg_r8->meas_cfg.speed_state_pars_present);

  TESTASSERT(rrc_recfg_r8->meas_cfg.quant_cfg.quant_cfg_eutra_present);
  TESTASSERT(not rrc_recfg_r8->meas_cfg.quant_cfg.quant_cfg_utra_present);
  TESTASSERT(not rrc_recfg_r8->meas_cfg.quant_cfg.quant_cfg_geran_present);
  TESTASSERT(not rrc_recfg_r8->meas_cfg.quant_cfg.quant_cfg_cdma2000_present);
  TESTASSERT(rrc_recfg_r8->meas_cfg.quant_cfg.quant_cfg_eutra.filt_coef_rsrp_present);
  TESTASSERT(rrc_recfg_r8->meas_cfg.quant_cfg.quant_cfg_eutra.filt_coef_rsrp == filt_coef_e::fc6);
  TESTASSERT(rrc_recfg_r8->meas_cfg.quant_cfg.quant_cfg_eutra.filt_coef_rsrq_present);
  TESTASSERT(rrc_recfg_r8->meas_cfg.quant_cfg.quant_cfg_eutra.filt_coef_rsrq == filt_coef_e::fc4);
  TESTASSERT(rrc_recfg_r8->meas_cfg.s_measure == 90);
  TESTASSERT(rrc_recfg_r8->meas_cfg.speed_state_pars.type() == meas_cfg_s::speed_state_pars_c_::types::release);

  TESTASSERT(rrc_recfg_r8->ded_info_nas_list.size() == 1);
  TESTASSERT(rrc_recfg_r8->ded_info_nas_list[0].size() == 107);

  // RR Config
  TESTASSERT(rrc_recfg_r8->rr_cfg_ded.srb_to_add_mod_list_present);
  TESTASSERT(rrc_recfg_r8->rr_cfg_ded.drb_to_add_mod_list_present);

  TESTASSERT(rrc_recfg_r8->rr_cfg_ded.srb_to_add_mod_list.size() == 1);
  srb_to_add_mod_s* srb = &rrc_recfg_r8->rr_cfg_ded.srb_to_add_mod_list[0];
  TESTASSERT(srb->rlc_cfg_present);
  TESTASSERT(srb->lc_ch_cfg_present);
  TESTASSERT(srb->srb_id == 2);
  TESTASSERT(srb->rlc_cfg.type() == srb_to_add_mod_s::rlc_cfg_c_::types::explicit_value);
  rlc_cfg_c& explicit_value = srb->rlc_cfg.explicit_value();
  TESTASSERT(explicit_value.type() == rlc_cfg_c::types::am);
  rlc_cfg_c::am_s_& am = explicit_value.am();
  TESTASSERT(am.ul_am_rlc.t_poll_retx == t_poll_retx_e::ms35);
  TESTASSERT(am.ul_am_rlc.poll_pdu == poll_pdu_e::p_infinity);
  TESTASSERT(am.ul_am_rlc.poll_byte == poll_byte_e::kbinfinity);
  TESTASSERT(am.ul_am_rlc.max_retx_thres == ul_am_rlc_s::max_retx_thres_e_::t32);
  TESTASSERT(am.dl_am_rlc.t_reordering == t_reordering_e::ms35);
  TESTASSERT(am.dl_am_rlc.t_status_prohibit == t_status_prohibit_e::ms0);
  TESTASSERT(srb->lc_ch_cfg.type() == srb_to_add_mod_s::lc_ch_cfg_c_::types::explicit_value);
  TESTASSERT(srb->lc_ch_cfg.explicit_value().ul_specific_params_present);
  lc_ch_cfg_s::ul_specific_params_s_& ulparams = srb->lc_ch_cfg.explicit_value().ul_specific_params;
  TESTASSERT(ulparams.lc_ch_group_present);
  TESTASSERT(ulparams.prio == 3);
  TESTASSERT(ulparams.prioritised_bit_rate == lc_ch_cfg_s::ul_specific_params_s_::prioritised_bit_rate_e_::infinity);
  TESTASSERT(ulparams.bucket_size_dur == lc_ch_cfg_s::ul_specific_params_s_::bucket_size_dur_e_::ms50);
  TESTASSERT(ulparams.lc_ch_group == 0);

  TESTASSERT(rrc_recfg_r8->rr_cfg_ded.drb_to_add_mod_list.size() == 1);
  drb_to_add_mod_s* drb = &rrc_recfg_r8->rr_cfg_ded.drb_to_add_mod_list[0];
  TESTASSERT(drb->eps_bearer_id_present);
  TESTASSERT(not drb->pdcp_cfg_present);
  TESTASSERT(drb->rlc_cfg_present);
  TESTASSERT(drb->lc_ch_id_present);
  TESTASSERT(drb->lc_ch_cfg_present);
  TESTASSERT(drb->eps_bearer_id == 5);
  TESTASSERT(drb->drb_id == 1);
  TESTASSERT(drb->rlc_cfg.type() == rlc_cfg_c::types::um_bi_dir);
  TESTASSERT(drb->rlc_cfg.um_bi_dir().ul_um_rlc.sn_field_len == sn_field_len_e::size10);
  TESTASSERT(drb->rlc_cfg.um_bi_dir().dl_um_rlc.sn_field_len == sn_field_len_e::size10);
  TESTASSERT(drb->rlc_cfg.um_bi_dir().dl_um_rlc.t_reordering == t_reordering_e::ms40);
  TESTASSERT(drb->lc_ch_id == 3);
  TESTASSERT(drb->lc_ch_cfg.ul_specific_params_present);
  TESTASSERT(drb->lc_ch_cfg.ul_specific_params.lc_ch_group_present);
  TESTASSERT(drb->lc_ch_cfg.ul_specific_params.prio == 12);
  TESTASSERT(drb->lc_ch_cfg.ul_specific_params.prioritised_bit_rate ==
             lc_ch_cfg_s::ul_specific_params_s_::prioritised_bit_rate_e_::kbps8);
  TESTASSERT(drb->lc_ch_cfg.ul_specific_params.bucket_size_dur ==
             lc_ch_cfg_s::ul_specific_params_s_::bucket_size_dur_e_::ms300);
  TESTASSERT(drb->lc_ch_cfg.ul_specific_params.lc_ch_group == 3);
  // ...
  TESTASSERT(not rrc_recfg_r8->rr_cfg_ded.phys_cfg_ded.pdsch_cfg_ded_present);
  TESTASSERT(rrc_recfg_r8->rr_cfg_ded.phys_cfg_ded.pucch_cfg_ded_present);
  TESTASSERT(not rrc_recfg_r8->rr_cfg_ded.phys_cfg_ded.pucch_cfg_ded.tdd_ack_nack_feedback_mode_present);
  TESTASSERT(rrc_recfg_r8->rr_cfg_ded.phys_cfg_ded.pucch_cfg_ded.ack_nack_repeat.type() ==
             pucch_cfg_ded_s::ack_nack_repeat_c_::types::release);
  TESTASSERT(rrc_recfg_r8->rr_cfg_ded.phys_cfg_ded.ant_info_r10.is_present());
  TESTASSERT(not rrc_recfg_r8->rr_cfg_ded.phys_cfg_ded.ant_info_ul_r10.is_present());
  TESTASSERT(not rrc_recfg_r8->rr_cfg_ded.phys_cfg_ded.cif_presence_r10_present);
  TESTASSERT(rrc_recfg_r8->rr_cfg_ded.phys_cfg_ded.cqi_report_cfg_r10.is_present());
  TESTASSERT(rrc_recfg_r8->rr_cfg_ded.phys_cfg_ded.csi_rs_cfg_r10.is_present());
  // ...
  TESTASSERT(rrc_recfg_r8->rr_cfg_ded.phys_cfg_ded.ant_info_r10->type() ==
             phys_cfg_ded_s::ant_info_r10_c_::types::explicit_value_r10);
  ant_info_ded_r10_s& ant_r10 = rrc_recfg_r8->rr_cfg_ded.phys_cfg_ded.ant_info_r10->explicit_value_r10();
  TESTASSERT(ant_r10.codebook_subset_restrict_r10_present);
  TESTASSERT(ant_r10.tx_mode_r10 == ant_info_ded_r10_s::tx_mode_r10_e_::tm3);
  TESTASSERT(ant_r10.codebook_subset_restrict_r10.length() == 2);
  TESTASSERT(ant_r10.codebook_subset_restrict_r10 == "11");
  TESTASSERT(ant_r10.ue_tx_ant_sel.type() == ant_info_ded_r10_s::ue_tx_ant_sel_c_::types::release);
  // ...
  TESTASSERT(not rrc_recfg_r8->non_crit_ext.late_non_crit_ext_present);
  TESTASSERT(rrc_recfg_r8->non_crit_ext.non_crit_ext_present);
  TESTASSERT(not rrc_recfg_r8->non_crit_ext.non_crit_ext.other_cfg_r9_present);
  TESTASSERT(not rrc_recfg_r8->non_crit_ext.non_crit_ext.full_cfg_r9_present);
  TESTASSERT(not rrc_recfg_r8->non_crit_ext.non_crit_ext.non_crit_ext.s_cell_to_release_list_r10_present);
  TESTASSERT(not rrc_recfg_r8->non_crit_ext.non_crit_ext.non_crit_ext.s_cell_to_add_mod_list_r10_present);
  TESTASSERT(rrc_recfg_r8->non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext_present);
  TESTASSERT(not rrc_recfg_r8->non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext.sib_type1_ded_r11_present);
  TESTASSERT(rrc_recfg_r8->non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext_present);
  TESTASSERT(
      not rrc_recfg_r8->non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext.wlan_offload_info_r12_present);
  TESTASSERT(not rrc_recfg_r8->non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext.scg_cfg_r12_present);
  TESTASSERT(
      not rrc_recfg_r8->non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext.sl_sync_tx_ctrl_r12_present);
  TESTASSERT(
      not rrc_recfg_r8->non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext.sl_disc_cfg_r12_present);
  TESTASSERT(
      not rrc_recfg_r8->non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext.sl_comm_cfg_r12_present);
  TESTASSERT(rrc_recfg_r8->non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext_present);
  TESTASSERT(not rrc_recfg_r8->non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext
                     .s_cell_to_release_list_ext_r13_present);
  TESTASSERT(not rrc_recfg_r8->non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext
                     .s_cell_to_add_mod_list_ext_r13_present);
  TESTASSERT(not rrc_recfg_r8->non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext
                     .lwa_cfg_r13_present);
  TESTASSERT(not rrc_recfg_r8->non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext
                     .lwip_cfg_r13_present);
  TESTASSERT(not rrc_recfg_r8->non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext
                     .rclwi_cfg_r13_present);
  TESTASSERT(
      rrc_recfg_r8->non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext_present);
  TESTASSERT(not rrc_recfg_r8->non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext
                     .non_crit_ext.sl_v2x_cfg_ded_r14_present);
  TESTASSERT(not rrc_recfg_r8->non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext
                     .non_crit_ext.s_cell_to_add_mod_list_ext_v1430_present);
  TESTASSERT(not rrc_recfg_r8->non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext
                     .non_crit_ext.per_cc_gap_ind_request_r14_present);
  TESTASSERT(not rrc_recfg_r8->non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext
                     .non_crit_ext.sib_type2_ded_r14_present);
  TESTASSERT(rrc_recfg_r8->non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext
                 .non_crit_ext_present);
  rrc_conn_recfg_v1510_ies_s* v1510 = &rrc_recfg_r8->non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext.non_crit_ext
                                           .non_crit_ext.non_crit_ext.non_crit_ext;
  TESTASSERT(not v1510->nr_cfg_r15_present);
  TESTASSERT(not v1510->sk_counter_r15_present);
  TESTASSERT(v1510->nr_radio_bearer_cfg1_r15_present);
  TESTASSERT(not v1510->nr_radio_bearer_cfg2_r15_present);
  TESTASSERT(not v1510->tdm_pattern_cfg_r15_present);
  TESTASSERT(not v1510->non_crit_ext_present);
  TESTASSERT(v1510->nr_radio_bearer_cfg1_r15.size() == 8); // FIXME: Get a better way for oct_string comparison
  TESTASSERT(v1510->nr_radio_bearer_cfg1_r15[0] == 20);
  TESTASSERT(v1510->nr_radio_bearer_cfg1_r15[7] == 0);

  return 0;
}

int failed_dl_ccch_unpack()
{
  uint8_t rrc_msg[] = {0xa5, 0xa8, 0xd8, 0x10, 0x0e, 0xc8, 0x02};
  bit_ref bref(&rrc_msg[0], sizeof(rrc_msg));

  asn1::rrc::dl_ccch_msg_s msg;

  TESTASSERT(msg.unpack(bref) == SRSASN_SUCCESS);

  return 0;
}

int unrecognized_ext_group_test()
{
  uint8_t rrc_msg[] = {0x00, 0x81, 0x19, 0x8c, 0x37, 0x91, 0x90, 0x10, 0x22, 0xc1, 0x29, 0x40, 0x48,
                       0x00, 0x82, 0x00, 0x32, 0x67, 0x29, 0x8a, 0x5a, 0xa8, 0x31, 0x00, 0x18, 0x01,
                       0x2e, 0x38, 0x03, 0x84, 0x28, 0xc5, 0xb0, 0x9d, 0x4b, 0x48, 0x00};
  // 0081198c3791901022c12940480082003267298a5aa8310018012e38038428c5b09d4b4800
  uint32_t rrc_msg_len = sizeof(rrc_msg);

  bit_ref bref(&rrc_msg[0], sizeof(rrc_msg));

  bcch_dl_sch_msg_s dl_sch_msg;
  dl_sch_msg.unpack(bref);

  TESTASSERT(dl_sch_msg.msg.type() == bcch_dl_sch_msg_type_c::types::c1);
  TESTASSERT(dl_sch_msg.msg.c1().type() == bcch_dl_sch_msg_type_c::c1_c_::types::sys_info);

  return 0;
}

int main()
{
  srsasn_log_register_handler(&asn_logger, print_console);
  rrc_log_register_handler(&rrc_logger, print_console);

  TESTASSERT(test_generic() == 0);
  TESTASSERT(test_json_printer() == 0);
  TESTASSERT(test_mib_msg() == 0);
  TESTASSERT(test_bcch_dl_sch_msg() == 0);
  TESTASSERT(test_bcch_dl_sch_msg2() == 0);
  TESTASSERT(test_bcch_dl_sch_msg3() == 0);
  TESTASSERT(test_dl_dcch_msg() == 0);
  TESTASSERT(ue_rrc_conn_recfg_r15_v10_test() == 0);
  TESTASSERT(failed_dl_ccch_unpack() == 0);
  TESTASSERT(unrecognized_ext_group_test() == 0);

  printf("Success\n");
  return 0;
}