srsRAN_4G/lib/include/srslte/common/netsource_handler.h

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

/*! \brief Base class for implementing socket handlers using
 * the netsource object.
 *
 */

#ifndef SRSLTE_NETSOURE_HANDLER_H
#define SRSLTE_NETSOURE_HANDLER_H

#include "srslte/common/log.h"
#include "srslte/common/threads.h"
#include "srslte/phy/io/netsource.h"
#include <array>
#include <iostream>
#include <memory>

class netsource_handler : public srslte::thread
{
public:
  netsource_handler(const std::string name_) : thread(name_) { rx_buf = unique_byte_array_t(new byte_array_t); }

  ~netsource_handler(){};

  void stop()
  {
    run_enable = false;
    int cnt    = 0;
    while (running && cnt < 100) {
      usleep(10000);
      cnt++;
    }
    if (running) {
      thread_cancel();
    }
    wait_thread_finish();
  }

  virtual void run_thread() = 0;

  bool run_enable  = true;
  bool running     = false;
  bool initialized = false;

  std::string net_ip   = "0.0.0.0";
  uint32_t    net_port = 0;

  const static uint32_t                    RX_BUF_SIZE = 1024 * 1024;
  typedef std::array<uint8_t, RX_BUF_SIZE> byte_array_t;
  typedef std::unique_ptr<byte_array_t>    unique_byte_array_t;
  unique_byte_array_t                      rx_buf;

  srslte_netsource_t net_source;

  srslte::log* log = nullptr;
};

#endif // SRSLTE_NETSOURE_HANDLER_H