mirror of https://github.com/PentHertz/srsLTE.git
115 lines
4.4 KiB
C++
115 lines
4.4 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.
|
|
*
|
|
*/
|
|
|
|
#include "srsran/common/common.h"
|
|
#include "srsran/common/mac_pcap_net.h"
|
|
#include "srsran/common/test_common.h"
|
|
#include <iostream>
|
|
#include <thread>
|
|
|
|
// Write #num_pdus UL MAC PDUs using PCAP handle
|
|
void write_pcap_eutra_thread_function(srsran::mac_pcap_net* pcap_handle,
|
|
const std::array<uint8_t, 150>& pdu,
|
|
uint32_t num_pdus)
|
|
{
|
|
for (uint32_t i = 0; i < num_pdus; i++) {
|
|
pcap_handle->write_ul_crnti(const_cast<uint8_t*>(pdu.data()), pdu.size(), 0x1001, true, 1, 0);
|
|
}
|
|
|
|
std::cout << "Finished thread " << std::this_thread::get_id() << "\n";
|
|
}
|
|
|
|
// Write #num_pdus DL MAC NR PDUs using PCAP handle
|
|
void write_pcap_nr_thread_function(srsran::mac_pcap_net* pcap_handle,
|
|
const std::array<uint8_t, 11>& pdu,
|
|
uint32_t num_pdus)
|
|
{
|
|
for (uint32_t i = 0; i < num_pdus; i++) {
|
|
pcap_handle->write_dl_crnti_nr(const_cast<uint8_t*>(pdu.data()), pdu.size(), 0x1001, 0, 1);
|
|
}
|
|
|
|
std::cout << "Finished thread " << std::this_thread::get_id() << "\n";
|
|
}
|
|
|
|
int lte_mac_pcap_net_test()
|
|
{
|
|
std::array<uint8_t, 150> tv = {
|
|
0x21, 0x08, 0x22, 0x80, 0x82, 0x1f, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x02, 0x02, 0x02, 0x02, 0x02,
|
|
0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02,
|
|
0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02,
|
|
0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02,
|
|
0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02,
|
|
0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02,
|
|
0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02,
|
|
0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00};
|
|
|
|
uint32_t num_threads = 10;
|
|
uint32_t num_pdus_per_thread = 100;
|
|
|
|
std::unique_ptr<srsran::mac_pcap_net> pcap_handle = std::unique_ptr<srsran::mac_pcap_net>(new srsran::mac_pcap_net());
|
|
TESTASSERT(pcap_handle->open("127.0.0.1") == SRSRAN_SUCCESS);
|
|
TESTASSERT(pcap_handle->open("127.0.0.1") != SRSRAN_SUCCESS); // open again will fail
|
|
|
|
std::vector<std::thread> writer_threads;
|
|
|
|
for (uint32_t i = 0; i < num_threads; i++) {
|
|
writer_threads.push_back(std::thread(write_pcap_eutra_thread_function, pcap_handle.get(), tv, num_pdus_per_thread));
|
|
}
|
|
|
|
// wait for threads to finish
|
|
for (std::thread& thread : writer_threads) {
|
|
thread.join();
|
|
}
|
|
TESTASSERT(pcap_handle->close() == SRSRAN_SUCCESS);
|
|
TESTASSERT(pcap_handle->close() != SRSRAN_SUCCESS); // closing twice will fail
|
|
|
|
return SRSRAN_SUCCESS;
|
|
}
|
|
|
|
int nr_mac_pcap_net_test()
|
|
{
|
|
std::array<uint8_t, 11> tv = {0x42, 0x00, 0x08, 0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0x88};
|
|
|
|
uint32_t num_threads = 10;
|
|
uint32_t num_pdus_per_thread = 100;
|
|
|
|
std::unique_ptr<srsran::mac_pcap_net> pcap_handle = std::unique_ptr<srsran::mac_pcap_net>(new srsran::mac_pcap_net());
|
|
TESTASSERT(pcap_handle->open("127.0.0.1") == SRSRAN_SUCCESS);
|
|
TESTASSERT(pcap_handle->open("127.0.0.1") != SRSRAN_SUCCESS); // open again will fail
|
|
|
|
std::vector<std::thread> writer_threads;
|
|
|
|
for (uint32_t i = 0; i < num_threads; i++) {
|
|
writer_threads.push_back(std::thread(write_pcap_nr_thread_function, pcap_handle.get(), tv, num_pdus_per_thread));
|
|
}
|
|
|
|
// wait for threads to finish
|
|
for (std::thread& thread : writer_threads) {
|
|
thread.join();
|
|
}
|
|
|
|
TESTASSERT(pcap_handle->close() == SRSRAN_SUCCESS);
|
|
TESTASSERT(pcap_handle->close() != SRSRAN_SUCCESS); // closing twice will fail
|
|
|
|
return SRSRAN_SUCCESS;
|
|
}
|
|
|
|
int main(int argc, char** argv)
|
|
{
|
|
auto& mac_logger = srslog::fetch_basic_logger("MAC", false);
|
|
mac_logger.set_level(srslog::basic_levels::debug);
|
|
mac_logger.set_hex_dump_max_size(-1);
|
|
srslog::init();
|
|
|
|
TESTASSERT(lte_mac_pcap_net_test() == SRSRAN_SUCCESS);
|
|
TESTASSERT(nr_mac_pcap_net_test() == SRSRAN_SUCCESS);
|
|
} |