mirror of https://github.com/PentHertz/srsLTE.git
131 lines
4.1 KiB
C++
131 lines
4.1 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 SRSRAN_RLC_TEST_COMMON_H
|
|
#define SRSRAN_RLC_TEST_COMMON_H
|
|
|
|
#include "srsran/common/byte_buffer.h"
|
|
#include "srsran/common/rlc_pcap.h"
|
|
#include "srsran/interfaces/ue_pdcp_interfaces.h"
|
|
#include "srsran/interfaces/ue_rrc_interfaces.h"
|
|
#include "srsran/rlc/rlc_metrics.h"
|
|
#include <vector>
|
|
|
|
namespace srsran {
|
|
|
|
class rlc_um_tester : public srsue::pdcp_interface_rlc, public srsue::rrc_interface_rlc
|
|
{
|
|
public:
|
|
rlc_um_tester() = default;
|
|
|
|
// PDCP interface
|
|
void write_pdu(uint32_t lcid, unique_byte_buffer_t sdu) final
|
|
{
|
|
// check length
|
|
if (lcid != 3 && sdu->N_bytes != expected_sdu_len) {
|
|
printf("Received PDU with size %d, expected %d. Exiting.\n", sdu->N_bytes, expected_sdu_len);
|
|
exit(-1);
|
|
}
|
|
|
|
// check content
|
|
uint8_t first_byte = *sdu->msg;
|
|
for (uint32_t i = 0; i < sdu->N_bytes; i++) {
|
|
if (sdu->msg[i] != first_byte) {
|
|
printf("Received corrupted SDU with size %d. Exiting.\n", sdu->N_bytes);
|
|
srsran_vec_fprint_byte(stdout, sdu->msg, sdu->N_bytes);
|
|
exit(-1);
|
|
}
|
|
}
|
|
|
|
// srsran_vec_fprint_byte(stdout, sdu->msg, sdu->N_bytes);
|
|
sdus.push_back(std::move(sdu));
|
|
}
|
|
void write_pdu_bcch_bch(unique_byte_buffer_t sdu) final {}
|
|
void write_pdu_bcch_dlsch(unique_byte_buffer_t sdu) final {}
|
|
void write_pdu_pcch(unique_byte_buffer_t sdu) final {}
|
|
void write_pdu_mch(uint32_t lcid, srsran::unique_byte_buffer_t sdu) final { sdus.push_back(std::move(sdu)); }
|
|
void notify_delivery(uint32_t lcid, const srsran::pdcp_sn_vector_t& pdcp_sns) final {}
|
|
void notify_failure(uint32_t lcid, const srsran::pdcp_sn_vector_t& pdcp_sns) final {}
|
|
|
|
// RRC interface
|
|
void max_retx_attempted() final {}
|
|
void protocol_failure() final {}
|
|
const char* get_rb_name(uint32_t lcid) { return ""; }
|
|
void set_expected_sdu_len(uint32_t len) { expected_sdu_len = len; }
|
|
|
|
uint32_t get_num_sdus() { return sdus.size(); }
|
|
|
|
// TODO: this should be private
|
|
std::vector<unique_byte_buffer_t> sdus;
|
|
uint32_t expected_sdu_len = 0;
|
|
};
|
|
|
|
class rlc_am_tester : public srsue::pdcp_interface_rlc, public srsue::rrc_interface_rlc
|
|
{
|
|
public:
|
|
rlc_am_tester(rlc_pcap* pcap_ = NULL) : pcap(pcap_) {}
|
|
|
|
// PDCP interface
|
|
void write_pdu(uint32_t lcid, unique_byte_buffer_t sdu)
|
|
{
|
|
assert(lcid == 1);
|
|
sdus.push_back(std::move(sdu));
|
|
}
|
|
void write_pdu_bcch_bch(unique_byte_buffer_t sdu) {}
|
|
void write_pdu_bcch_dlsch(unique_byte_buffer_t sdu) {}
|
|
void write_pdu_pcch(unique_byte_buffer_t sdu) {}
|
|
void write_pdu_mch(uint32_t lcid, srsran::unique_byte_buffer_t pdu) {}
|
|
void notify_delivery(uint32_t lcid, const srsran::pdcp_sn_vector_t& pdcp_sn_vec)
|
|
{
|
|
assert(lcid == 1);
|
|
for (uint32_t pdcp_sn : pdcp_sn_vec) {
|
|
if (notified_counts.find(pdcp_sn) == notified_counts.end()) {
|
|
notified_counts[pdcp_sn] = 0;
|
|
}
|
|
notified_counts[pdcp_sn] += 1;
|
|
}
|
|
}
|
|
void notify_failure(uint32_t lcid, const srsran::pdcp_sn_vector_t& pdcp_sn_vec)
|
|
{
|
|
assert(lcid == 1);
|
|
// TODO
|
|
}
|
|
|
|
// RRC interface
|
|
void max_retx_attempted() { max_retx_triggered = true; }
|
|
void protocol_failure() { protocol_failure_triggered = true; }
|
|
|
|
const char* get_rb_name(uint32_t lcid) { return ""; }
|
|
|
|
std::vector<unique_byte_buffer_t> sdus;
|
|
rlc_pcap* pcap = nullptr;
|
|
bool max_retx_triggered = false;
|
|
bool protocol_failure_triggered = false;
|
|
|
|
std::map<uint32_t, uint32_t> notified_counts; // Map of PDCP SNs to number of notifications
|
|
};
|
|
|
|
bool rx_is_tx(const rlc_bearer_metrics_t& rlc1_metrics, const rlc_bearer_metrics_t& rlc2_metrics)
|
|
{
|
|
if (rlc1_metrics.num_tx_pdu_bytes != rlc2_metrics.num_rx_pdu_bytes) {
|
|
return false;
|
|
}
|
|
|
|
if (rlc2_metrics.num_tx_pdu_bytes != rlc1_metrics.num_rx_pdu_bytes) {
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
} // namespace srsran
|
|
|
|
#endif // SRSRAN_RLC_TEST_COMMON_H
|