mirror of https://github.com/PentHertz/srsLTE.git
103 lines
2.6 KiB
C++
103 lines
2.6 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/.
|
|
*
|
|
*/
|
|
|
|
#include "srsran/common/mac_pcap.h"
|
|
#include "srsran/common/standard_streams.h"
|
|
#include "srsran/common/threads.h"
|
|
|
|
namespace srsran {
|
|
mac_pcap::mac_pcap() : mac_pcap_base() {}
|
|
|
|
mac_pcap::~mac_pcap()
|
|
{
|
|
close();
|
|
}
|
|
|
|
uint32_t mac_pcap::open(std::string filename_, uint32_t ue_id_)
|
|
{
|
|
std::lock_guard<std::mutex> lock(mutex);
|
|
if (pcap_file != nullptr) {
|
|
logger.error("PCAP writer for %s already running. Close first.", filename_.c_str());
|
|
return SRSRAN_ERROR;
|
|
}
|
|
|
|
// set UDP DLT
|
|
dlt = UDP_DLT;
|
|
pcap_file = DLT_PCAP_Open(dlt, filename_.c_str());
|
|
if (pcap_file == nullptr) {
|
|
logger.error("Couldn't open %s to write PCAP", filename_.c_str());
|
|
return SRSRAN_ERROR;
|
|
}
|
|
|
|
filename = filename_;
|
|
ue_id = ue_id_;
|
|
running = true;
|
|
|
|
// start writer thread
|
|
start();
|
|
|
|
return SRSRAN_SUCCESS;
|
|
}
|
|
|
|
uint32_t mac_pcap::close()
|
|
{
|
|
{
|
|
std::lock_guard<std::mutex> lock(mutex);
|
|
if (running == false || pcap_file == nullptr) {
|
|
return SRSRAN_ERROR;
|
|
}
|
|
|
|
// tell writer thread to stop
|
|
running = false;
|
|
pcap_pdu_t pdu = {};
|
|
queue.push_blocking(std::move(pdu));
|
|
}
|
|
|
|
wait_thread_finish();
|
|
|
|
// close file handle
|
|
{
|
|
std::lock_guard<std::mutex> lock(mutex);
|
|
srsran::console("Saving MAC PCAP (DLT=%d) to %s\n", dlt, filename.c_str());
|
|
DLT_PCAP_Close(pcap_file);
|
|
pcap_file = nullptr;
|
|
}
|
|
|
|
return SRSRAN_SUCCESS;
|
|
}
|
|
|
|
void mac_pcap::write_pdu(srsran::mac_pcap_base::pcap_pdu_t& pdu)
|
|
{
|
|
if (pdu.pdu != nullptr) {
|
|
switch (pdu.rat) {
|
|
case srsran_rat_t::lte:
|
|
LTE_PCAP_MAC_UDP_WritePDU(pcap_file, &pdu.context, pdu.pdu->msg, pdu.pdu->N_bytes);
|
|
break;
|
|
case srsran_rat_t::nr:
|
|
NR_PCAP_MAC_UDP_WritePDU(pcap_file, &pdu.context_nr, pdu.pdu->msg, pdu.pdu->N_bytes);
|
|
break;
|
|
default:
|
|
logger.error("Error writing PDU to PCAP. Unsupported RAT selected.");
|
|
}
|
|
}
|
|
}
|
|
|
|
} // namespace srsran
|