/**
 *
 * \section COPYRIGHT
 *
 * Copyright 2013-2020 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 SRSLTE_MAC_PCAP_H
#define SRSLTE_MAC_PCAP_H

#include "srslte/common/block_queue.h"
#include "srslte/common/buffer_pool.h"
#include "srslte/common/common.h"
#include "srslte/common/pcap.h"
#include "srslte/common/threads.h"
#include "srslte/srslog/srslog.h"
#include <mutex>
#include <stdint.h>
#include <thread>

namespace srslte {
class mac_pcap : srslte::thread
{
public:
  mac_pcap();
  ~mac_pcap();
  void     enable(bool enable);
  uint32_t open(std::string filename, uint32_t ue_id = 0);
  uint32_t close();

  void set_ue_id(uint16_t ue_id);

  // EUTRA
  void
       write_ul_crnti(uint8_t* pdu, uint32_t pdu_len_bytes, uint16_t crnti, uint32_t reTX, uint32_t tti, uint8_t cc_idx);
  void write_dl_crnti(uint8_t* pdu, uint32_t pdu_len_bytes, uint16_t crnti, bool crc_ok, uint32_t tti, uint8_t cc_idx);
  void write_dl_ranti(uint8_t* pdu, uint32_t pdu_len_bytes, uint16_t ranti, bool crc_ok, uint32_t tti, uint8_t cc_idx);

  void write_ul_crnti(uint8_t* pdu,
                      uint32_t pdu_len_bytes,
                      uint16_t crnti,
                      uint16_t ue_id,
                      uint32_t reTX,
                      uint32_t tti,
                      uint8_t  cc_idx);

  void write_dl_crnti(uint8_t* pdu,
                      uint32_t pdu_len_bytes,
                      uint16_t crnti,
                      uint16_t ue_id,
                      bool     crc_ok,
                      uint32_t tti,
                      uint8_t  cc_idx);

  // SI and BCH only for DL
  void write_dl_sirnti(uint8_t* pdu, uint32_t pdu_len_bytes, bool crc_ok, uint32_t tti, uint8_t cc_idx);
  void write_dl_bch(uint8_t* pdu, uint32_t pdu_len_bytes, bool crc_ok, uint32_t tti, uint8_t cc_idx);
  void write_dl_pch(uint8_t* pdu, uint32_t pdu_len_bytes, bool crc_ok, uint32_t tti, uint8_t cc_idx);
  void write_dl_mch(uint8_t* pdu, uint32_t pdu_len_bytes, bool crc_ok, uint32_t tti, uint8_t cc_idx);

  void write_ul_rrc_pdu(const uint8_t* input, const int32_t input_len);

  // Sidelink
  void write_sl_crnti(uint8_t* pdu, uint32_t pdu_len_bytes, uint16_t rnti, uint32_t reTX, uint32_t tti, uint8_t cc_idx);

  // NR
  void write_dl_crnti_nr(uint8_t* pdu, uint32_t pdu_len_bytes, uint16_t crnti, uint8_t harqid, uint32_t tti);
  void write_ul_crnti_nr(uint8_t* pdu, uint32_t pdu_len_bytes, uint16_t rnti, uint8_t harqid, uint32_t tti);
  void write_dl_ra_rnti_nr(uint8_t* pdu, uint32_t pdu_len_bytes, uint16_t rnti, uint8_t harqid, uint32_t tti);
  void write_dl_bch_nr(uint8_t* pdu, uint32_t pdu_len_bytes, uint16_t rnti, uint8_t harqid, uint32_t tti);
  void write_dl_pch_nr(uint8_t* pdu, uint32_t pdu_len_bytes, uint16_t rnti, uint8_t harqid, uint32_t tti);
  void write_dl_si_rnti_nr(uint8_t* pdu, uint32_t pdu_len_bytes, uint16_t rnti, uint8_t harqid, uint32_t tti);

  // NR for enb with different ue_id
  // clang-format off
  void write_dl_crnti_nr(uint8_t* pdu, uint32_t pdu_len_bytes, uint16_t crnti, uint16_t ue_id, uint8_t harqid, uint32_t tti);
  void write_ul_crnti_nr(uint8_t* pdu, uint32_t pdu_len_bytes, uint16_t rnti, uint16_t ue_id, uint8_t harqid, uint32_t tti);
  // clang-format on

private:
  srslog::basic_logger& logger;
  bool                  running = false;
  uint32_t              dlt     = 0; // The DLT used for the PCAP file
  std::string           filename;
  FILE*                 pcap_file = nullptr;
  uint16_t              ue_id     = 0;
  void                  pack_and_queue(uint8_t* payload,
                                       uint32_t payload_len,
                                       uint16_t ue_id,
                                       uint32_t reTX,
                                       bool     crc_ok,
                                       uint8_t  cc_idx,
                                       uint32_t tti,
                                       uint16_t crnti_,
                                       uint8_t  direction,
                                       uint8_t  rnti_type);
  void                  pack_and_queue_nr(uint8_t* payload,
                                          uint32_t payload_len,
                                          uint32_t tti,
                                          uint16_t crnti,
                                          uint16_t ue_id,
                                          uint8_t  harqid,
                                          uint8_t  direction,
                                          uint8_t  rnti_type);

  typedef struct {
    // Different PCAP context for both RATs
    srslte::srslte_rat_t  rat;
    MAC_Context_Info_t    context;
    mac_nr_context_info_t context_nr;
    unique_byte_buffer_t  pdu;
  } pcap_pdu_t;
  block_queue<pcap_pdu_t> queue;
  std::mutex              mutex;

  void write_pdu(pcap_pdu_t& pdu);
  void run_thread() final;
};

} // namespace srslte

#endif // SRSLTE_MAC_PCAP_H