/*
 * Licensed to Open Source Radio Access Network(OS-RAN) Alliance and OAI
 * Software Alliance under one or more contributor license agreements. The
 * initial OpenXG series projects are derided from OAI projects, the files from
 * OAI projects are all in compliance with OAI Public License, Version 1.1.
 * codes and files developed from OpenXG projects and from OS-RAN Alliance
 * are all under OS-RAN licenses; you may not use this file except in compliance
 * with the license.  You may get a copy of the license at:
 *	http://www.openxg.org.cn/?falu_69.html
 * For more information about OpenXG, please contact:
 * contact@openxg.org.cn
 */



#include "pfcp.hpp"

#include <cstdlib>

using namespace pfcp;
using namespace std;

extern itti_mw* itti_inst;

//------------------------------------------------------------------------------
pfcp_l4_stack::pfcp_l4_stack(
    const string& ip_address, const unsigned short port_num,
    const util::thread_sched_params& sched_params)
    : udp_s_8805(ip_address.c_str(), port_num),
      udp_s_allocated(ip_address.c_str(), 0) {
  Logger::pfcp().info(
      "pfcp_l4_stack created listening to %s:%d", ip_address.c_str(), port_num);
  trxn_id2seq_num      = {};
  proc_cleanup_timers  = {};
  msg_out_retry_timers = {};
  pending_procedures   = {};

  id = 0;

  timespec ts;
  clock_gettime(CLOCK_REALTIME, &ts);
  seq_num         = 0;  //(uint32_t)ts.tv_nsec & 0x7FFFFFFF;
  restart_counter = 0;
  udp_s_8805.start_receive(this, sched_params);
  udp_s_allocated.start_receive(this, sched_params);
}
//------------------------------------------------------------------------------
uint32_t pfcp_l4_stack::get_next_seq_num() {
  seq_num++;
  if (seq_num & 0x80000000) {
    seq_num = 0;
  }
  return seq_num;
}
//------------------------------------------------------------------------------
void pfcp_l4_stack::handle_receive(
    char* recv_buffer, const std::size_t bytes_transferred,
    endpoint& remote_endpoint) {
  Logger::pfcp().error("TODO implement in derived class");
}
//------------------------------------------------------------------------------
bool pfcp_l4_stack::check_request_type(const uint8_t initial) {
  switch (initial) {
    case PFCP_HEARTBEAT_REQUEST:
    case PFCP_PFCP_PFD_MANAGEMENT_REQUEST:
    case PFCP_ASSOCIATION_SETUP_REQUEST:
    case PFCP_ASSOCIATION_UPDATE_REQUEST:
    case PFCP_ASSOCIATION_RELEASE_REQUEST:
    case PFCP_NODE_REPORT_REQUEST:
    case PFCP_NODE_REPORT_RESPONSE:  // TODO
    case PFCP_SESSION_ESTABLISHMENT_REQUEST:
    case PFCP_SESSION_MODIFICATION_REQUEST:
    case PFCP_SESSION_DELETION_REQUEST:
    case PFCP_SESSION_REPORT_REQUEST:
    case PFCP_SESSION_REPORT_RESPONSE:  // TODO
      return true;
      break;
    default:
      return false;
  }
}
//------------------------------------------------------------------------------
bool pfcp_l4_stack::check_response_type(
    const uint8_t initial, const uint8_t triggered) {
  // Logger::pfcp().info( "check_response_type PFCP msg type %d/%d",
  // (int)initial, (int)triggered);
  switch (initial) {
    case PFCP_HEARTBEAT_REQUEST:
    case PFCP_PFCP_PFD_MANAGEMENT_REQUEST:
    case PFCP_ASSOCIATION_SETUP_REQUEST:
    case PFCP_ASSOCIATION_UPDATE_REQUEST:
    case PFCP_ASSOCIATION_RELEASE_REQUEST:
    case PFCP_NODE_REPORT_REQUEST:
    case PFCP_SESSION_ESTABLISHMENT_REQUEST:
    case PFCP_SESSION_MODIFICATION_REQUEST:
    case PFCP_SESSION_DELETION_REQUEST:
    case PFCP_SESSION_REPORT_REQUEST:
      if (triggered == (initial + 1)) return true;
      break;
    default:
      if (triggered == PFCP_VERSION_NOT_SUPPORTED_RESPONSE) return true;
  }
  return false;
}
//------------------------------------------------------------------------------
void pfcp_l4_stack::start_msg_retry_timer(
    pfcp_procedure& p, uint32_t time_out_milli_seconds,
    const task_id_t& task_id, const uint32_t& seq_num) {
  p.retry_timer_id = itti_inst->timer_setup(
      time_out_milli_seconds / 1000, time_out_milli_seconds % 1000, task_id);
  msg_out_retry_timers.insert(
      std::pair<timer_id_t, uint32_t>(p.retry_timer_id, seq_num));
  // Logger::pfcp().trace( "Started Msg retry timer %d, proc %" PRId64", seq
  // %d",p.retry_timer_id, p.trxn_id, seq_num);
}
//------------------------------------------------------------------------------
void pfcp_l4_stack::stop_msg_retry_timer(pfcp_procedure& p) {
  if (p.retry_timer_id) {
    itti_inst->timer_remove(p.retry_timer_id);
    msg_out_retry_timers.erase(p.retry_timer_id);
    // Logger::pfcp().trace( "Stopped Msg retry timer %d, proc %" PRId64", seq
    // %d",p.retry_timer_id, p.trxn_id, p.retry_msg->get_sequence_number());
    p.retry_timer_id = 0;
  }
}
//------------------------------------------------------------------------------
void pfcp_l4_stack::stop_msg_retry_timer(timer_id_t& t) {
  itti_inst->timer_remove(t);
  msg_out_retry_timers.erase(t);
  // Logger::pfcp().trace( "Stopped Msg retry timer %d",t);
}
//------------------------------------------------------------------------------
void pfcp_l4_stack::start_proc_cleanup_timer(
    pfcp_procedure& p, uint32_t time_out_milli_seconds,
    const task_id_t& task_id, const uint32_t& seq_num) {
  p.proc_cleanup_timer_id = itti_inst->timer_setup(
      time_out_milli_seconds / 1000, time_out_milli_seconds % 1000, task_id);
  proc_cleanup_timers.insert(
      std::pair<timer_id_t, uint32_t>(p.proc_cleanup_timer_id, seq_num));
  // Logger::pfcp().trace( "Started proc cleanup timer %d, proc %" PRId64" t-out
  // %" PRIu32" ms",p.proc_cleanup_timer_id,p.trxn_id, time_out_milli_seconds);
}
//------------------------------------------------------------------------------
void pfcp_l4_stack::stop_proc_cleanup_timer(pfcp_procedure& p) {
  itti_inst->timer_remove(p.proc_cleanup_timer_id);
  // Logger::pfcp().trace( "Stopped proc cleanup timer %d, proc %"
  // PRId64"",p.proc_cleanup_timer_id, p.trxn_id);
  msg_out_retry_timers.erase(p.proc_cleanup_timer_id);
  p.proc_cleanup_timer_id = 0;
}
//------------------------------------------------------------------------------
void pfcp_l4_stack::handle_receive_message_cb(
    const pfcp_msg& msg, const endpoint& remote_endpoint,
    const task_id_t& task_id, bool& error, uint64_t& trxn_id) {
  trxn_id = 0;
  error   = true;
  std::map<uint32_t, pfcp_procedure>::iterator it;
  it = pending_procedures.find(msg.get_sequence_number());
  if (it == pending_procedures.end()) {
    if (pfcp_l4_stack::check_request_type(msg.get_message_type())) {
      pfcp_procedure proc   = {};
      proc.trxn_id          = generate_trxn_id();
      proc.initial_msg_type = msg.get_message_type();
      // TODO later 13.3 Detection and handling of requests which have timed out
      // at the originating entity if (msg_has_timestamp()) {
      // start_proc_cleanup_timer(proc, (N3+1) x T3, task_id,
      // msg.get_sequence_number()); } else
      start_proc_cleanup_timer(
          proc, PFCP_PROC_TIME_OUT_MS, task_id, msg.get_sequence_number());
      pending_procedures.insert(
          std::pair<uint32_t, pfcp_procedure>(msg.get_sequence_number(), proc));
      trxn_id2seq_num.insert(std::pair<uint64_t, uint32_t>(
          proc.trxn_id, msg.get_sequence_number()));
      error   = false;
      trxn_id = proc.trxn_id;
      // Logger::pfcp().info( "Received Initial PFCP msg type %d, seq %d, proc
      // %" PRId64"", msg.get_message_type(), msg.get_sequence_number(),
      // proc.trxn_id);
    } else {
      Logger::pfcp().info(
          "Failed to check Initial message type, Silently discarding PFCP msg "
          "type %d, seq %d",
          msg.get_message_type(), msg.get_sequence_number());
      error = true;
    }
    return;
  } else {
    //    Logger::pfcp().info( "pfcp_procedure retry_timer_id        %d",
    //    it->second.retry_timer_id); Logger::pfcp().info( "pfcp_procedure
    //    proc_cleanup_timer_id %d", it->second.proc_cleanup_timer_id);
    //    Logger::pfcp().info( "pfcp_procedure trxn_id            %ld",
    //    it->second.trxn_id); Logger::pfcp().info( "pfcp_procedure
    //    initial_msg_type      %d", it->second.initial_msg_type);
    //    Logger::pfcp().info( "pfcp_procedure triggered_msg_type    %d",
    //    it->second.triggered_msg_type); Logger::pfcp().info( "pfcp_procedure
    //    retry_count           %d", it->second.retry_count);

    uint8_t check_initial_msg_type = it->second.triggered_msg_type;
    if (!it->second.triggered_msg_type) {
      check_initial_msg_type = it->second.initial_msg_type;
    }
    if (pfcp_l4_stack::check_response_type(
            check_initial_msg_type, msg.get_message_type())) {
      if (!it->second.triggered_msg_type) {
        it->second.triggered_msg_type = msg.get_message_type();
      }
      error   = false;
      trxn_id = it->second.trxn_id;
      if (it->second.retry_timer_id) {
        stop_msg_retry_timer(it->second);
      }
      // Logger::pfcp().info( "Received Triggered PFCP msg type %d, seq %d, proc
      // %" PRId64"", msg.get_message_type(), msg.get_sequence_number(),
      // trxn_id);
    } else {
      Logger::pfcp().info(
          "Failed to check Triggered message type, Silently discarding PFCP "
          "msg type %d, seq %d",
          msg.get_message_type(), msg.get_sequence_number());
      error = true;
    }
  }
}

//------------------------------------------------------------------------------
uint32_t pfcp_l4_stack::send_request(
    const endpoint& dest, const pfcp_heartbeat_request& pfcp_ies,
    const task_id_t& task_id, const uint64_t trxn_id) {
  std::ostringstream oss(std::ostringstream::binary);
  pfcp_msg msg(pfcp_ies);
  msg.set_sequence_number(get_next_seq_num());
  msg.dump_to(oss);
  // std::cout << string_to_hex(oss.str()) << std::endl;
  // std::cout << std::hex << "msg length 0x" << msg.get_message_length() <<
  // "msg seqnum 0x" << msg.get_sequence_number() << std::endl;
  std::string bstream = oss.str();

  Logger::pfcp().trace(
      "Sending %s, seq %d", pfcp_ies.get_msg_name(), msg.get_sequence_number());
  pfcp_procedure proc   = {};
  proc.initial_msg_type = msg.get_message_type();
  proc.trxn_id          = trxn_id;
  proc.retry_msg        = std::make_shared<pfcp_msg>(msg);
  proc.remote_endpoint  = dest;
  start_msg_retry_timer(
      proc, PFCP_T1_RESPONSE_MS, task_id, msg.get_sequence_number());
  start_proc_cleanup_timer(
      proc, PFCP_PROC_TIME_OUT_MS, task_id, msg.get_sequence_number());
  pending_procedures.insert(
      std::pair<uint32_t, pfcp_procedure>(msg.get_sequence_number(), proc));
  trxn_id2seq_num.insert(
      std::pair<uint64_t, uint32_t>(proc.trxn_id, msg.get_sequence_number()));

  // TTN: temporary fix to be able to work with UPF from dsTester (should be
  // removed later) udp_s_allocated.async_send_to(reinterpret_cast<const
  // char*>(bstream.c_str()), bstream.length(), dest);
  udp_s_8805.async_send_to(
      reinterpret_cast<const char*>(bstream.c_str()), bstream.length(), dest);
  return msg.get_sequence_number();
}
//------------------------------------------------------------------------------
uint32_t pfcp_l4_stack::send_request(
    const endpoint& dest, const pfcp_association_setup_request& pfcp_ies,
    const task_id_t& task_id, const uint64_t trxn_id) {
  std::ostringstream oss(std::ostringstream::binary);
  pfcp_msg msg(pfcp_ies);
  msg.set_sequence_number(get_next_seq_num());
  msg.dump_to(oss);
  // std::cout << string_to_hex(oss.str()) << std::endl;
  // std::cout << std::hex << "msg length 0x" << msg.get_message_length() <<
  // "msg seqnum 0x" << msg.get_sequence_number() << std::endl;
  std::string bstream = oss.str();

  Logger::pfcp().trace(
      "Sending %s, seq %d", pfcp_ies.get_msg_name(), msg.get_sequence_number());
  pfcp_procedure proc   = {};
  proc.initial_msg_type = msg.get_message_type();
  proc.trxn_id          = trxn_id;
  proc.retry_msg        = std::make_shared<pfcp_msg>(msg);
  proc.remote_endpoint  = dest;
  start_msg_retry_timer(
      proc, PFCP_T1_RESPONSE_MS, task_id, msg.get_sequence_number());
  start_proc_cleanup_timer(
      proc, PFCP_PROC_TIME_OUT_MS, task_id, msg.get_sequence_number());
  pending_procedures.insert(
      std::pair<uint32_t, pfcp_procedure>(msg.get_sequence_number(), proc));
  trxn_id2seq_num.insert(
      std::pair<uint64_t, uint32_t>(proc.trxn_id, msg.get_sequence_number()));

  // TTN: temporary fix to be able to work with UPF from dsTester (should be
  // removed later)
  // udp_s_allocated.async_send_to(reinterpret_cast<const
  // char*>(bstream.c_str()), bstream.length(), dest);
  udp_s_8805.async_send_to(
      reinterpret_cast<const char*>(bstream.c_str()), bstream.length(), dest);

  return msg.get_sequence_number();
}
//------------------------------------------------------------------------------
uint32_t pfcp_l4_stack::send_request(
    const endpoint& dest, const pfcp_association_release_request& pfcp_ies,
    const task_id_t& task_id, const uint64_t trxn_id) {
  std::ostringstream oss(std::ostringstream::binary);
  pfcp_msg msg(pfcp_ies);
  msg.set_sequence_number(get_next_seq_num());
  msg.dump_to(oss);
  // std::cout << string_to_hex(oss.str()) << std::endl;
  // std::cout << std::hex << "msg length 0x" << msg.get_message_length() <<
  // "msg seqnum 0x" << msg.get_sequence_number() << std::endl;
  std::string bstream = oss.str();

  Logger::pfcp().trace(
      "Sending %s, seq %d", pfcp_ies.get_msg_name(), msg.get_sequence_number());
  pfcp_procedure proc   = {};
  proc.initial_msg_type = msg.get_message_type();
  proc.trxn_id          = trxn_id;
  proc.retry_msg        = std::make_shared<pfcp_msg>(msg);
  proc.remote_endpoint  = dest;
  start_msg_retry_timer(
      proc, PFCP_T1_RESPONSE_MS, task_id, msg.get_sequence_number());
  start_proc_cleanup_timer(
      proc, PFCP_PROC_TIME_OUT_MS, task_id, msg.get_sequence_number());
  pending_procedures.insert(
      std::pair<uint32_t, pfcp_procedure>(msg.get_sequence_number(), proc));
  trxn_id2seq_num.insert(
      std::pair<uint64_t, uint32_t>(proc.trxn_id, msg.get_sequence_number()));

  // TTN: temporary fix to be able to work with UPF from dsTester (should be
  // removed later) udp_s_allocated.async_send_to(reinterpret_cast<const
  // char*>(bstream.c_str()), bstream.length(), dest);
  udp_s_8805.async_send_to(
      reinterpret_cast<const char*>(bstream.c_str()), bstream.length(), dest);

  return msg.get_sequence_number();
}
////------------------------------------------------------------------------------
// uint32_t pfcp_l4_stack::send_request(const endpoint& dest, const uint64_t
// seid, const pfcp_association_update_request& pfcp_ies, const task_id_t&
// task_id, const uint64_t trxn_id)
//{
//  std::ostringstream oss(std::ostringstream::binary);
//  pfcp_msg msg(pfcp_ies);
//  msg.set_seid(seid);
//  msg.set_sequence_number(get_next_seq_num());
//  msg.dump_to(oss);
//  //std::cout << string_to_hex(oss.str()) << std::endl;
//  //std::cout << std::hex << "msg length 0x" << msg.get_message_length() <<
//  "msg seqnum 0x" << msg.get_sequence_number() << std::endl; std::string
//  bstream = oss.str();
//
//  Logger::pfcp().trace( "Sending %s, seq %d", pfcp_ies.get_msg_name(),
//  msg.get_sequence_number()); pfcp_procedure proc = {}; proc.initial_msg_type
//  = msg.get_message_type(); proc.trxn_id = trxn_id; proc.retry_msg =
//  std::make_shared<pfcp_msg>(msg); proc.remote_endpoint = dest;
//  start_msg_retry_timer(proc, PFCP_T1_RESPONSE_MS, task_id,
//  msg.get_sequence_number()); start_proc_cleanup_timer(proc,
//  PFCP_PROC_TIME_OUT_MS, task_id, msg.get_sequence_number());
//  pending_procedures.insert(std::pair<uint32_t,
//  pfcp_procedure>(msg.get_sequence_number(), proc));
//  trxn_id2seq_num.insert(std::pair<uint64_t, uint32_t>(proc.trxn_id,
//  msg.get_sequence_number()));
//
//  udp_s.async_send_to(sm, dest);
//  return msg.get_sequence_number();
//}
////------------------------------------------------------------------------------
// uint32_t pfcp_l4_stack::send_request(const endpoint& dest, const uint64_t
// seid, const pfcp_association_release_request& pfcp_ies, const task_id_t&
// task_id, const uint64_t trxn_id)
//{
//  std::ostringstream oss(std::ostringstream::binary);
//  pfcp_msg msg(pfcp_ies);
//  msg.set_seid(seid);
//  msg.set_sequence_number(get_next_seq_num());
//  msg.dump_to(oss);
//  //std::cout << string_to_hex(oss.str()) << std::endl;
//  //std::cout << std::hex << "msg length 0x" << msg.get_message_length() <<
//  "msg seqnum 0x" << msg.get_sequence_number() << std::endl; std::string
//  bstream = oss.str();
//
//  Logger::pfcp().trace( "Sending %s, seq %d", pfcp_ies.get_msg_name(),
//  msg.get_sequence_number()); pfcp_procedure proc = {}; proc.initial_msg_type
//  = msg.get_message_type(); proc.trxn_id = trxn_id; proc.retry_msg =
//  std::make_shared<pfcp_msg>(msg); proc.remote_endpoint = dest;
//  start_msg_retry_timer(proc, PFCP_T1_RESPONSE_MS, task_id,
//  msg.get_sequence_number()); start_proc_cleanup_timer(proc,
//  PFCP_PROC_TIME_OUT_MS, task_id, msg.get_sequence_number());
//  pending_procedures.insert(std::pair<uint32_t,
//  pfcp_procedure>(msg.get_sequence_number(), proc));
//  trxn_id2seq_num.insert(std::pair<uint64_t, uint32_t>(proc.trxn_id,
//  msg.get_sequence_number()));
//
//  udp_s.async_send_to(sm, dest);
//  return msg.get_sequence_number();
//}
//------------------------------------------------------------------------------
uint32_t pfcp_l4_stack::send_request(
    const endpoint& dest, const uint64_t seid,
    const pfcp_node_report_request& pfcp_ies, const task_id_t& task_id,
    const uint64_t trxn_id) {
  std::ostringstream oss(std::ostringstream::binary);
  pfcp_msg msg(pfcp_ies);
  msg.set_sequence_number(get_next_seq_num());
  msg.dump_to(oss);
  // std::cout << string_to_hex(oss.str()) << std::endl;
  // std::cout << std::hex << "msg length 0x" << msg.get_message_length() <<
  // "msg seqnum 0x" << msg.get_sequence_number() << std::endl;
  std::string bstream = oss.str();

  Logger::pfcp().trace(
      "Sending %s, seq %d", pfcp_ies.get_msg_name(), msg.get_sequence_number());
  pfcp_procedure proc   = {};
  proc.initial_msg_type = msg.get_message_type();
  proc.trxn_id          = trxn_id;
  proc.retry_msg        = std::make_shared<pfcp_msg>(msg);
  proc.remote_endpoint  = dest;
  start_msg_retry_timer(
      proc, PFCP_T1_RESPONSE_MS, task_id, msg.get_sequence_number());
  start_proc_cleanup_timer(
      proc, PFCP_PROC_TIME_OUT_MS, task_id, msg.get_sequence_number());
  pending_procedures.insert(
      std::pair<uint32_t, pfcp_procedure>(msg.get_sequence_number(), proc));
  trxn_id2seq_num.insert(
      std::pair<uint64_t, uint32_t>(proc.trxn_id, msg.get_sequence_number()));

  // TTN: temporary fix to be able to work with UPF from dsTester (should be
  // removed later) udp_s_allocated.async_send_to(reinterpret_cast<const
  // char*>(bstream.c_str()), bstream.length(), dest);
  udp_s_8805.async_send_to(
      reinterpret_cast<const char*>(bstream.c_str()), bstream.length(), dest);
  return msg.get_sequence_number();
}
//------------------------------------------------------------------------------
uint32_t pfcp_l4_stack::send_request(
    const endpoint& dest, const uint64_t seid,
    const pfcp_session_establishment_request& pfcp_ies,
    const task_id_t& task_id, const uint64_t trxn_id) {
  std::ostringstream oss(std::ostringstream::binary);
  pfcp_msg msg(pfcp_ies);
  msg.set_seid(seid);
  msg.set_sequence_number(get_next_seq_num());
  msg.dump_to(oss);
  // std::cout << string_to_hex(oss.str()) << std::endl;
  // std::cout << std::hex << "msg length 0x" << msg.get_message_length() <<
  // "msg seqnum 0x" << msg.get_sequence_number() << std::endl;
  std::string bstream = oss.str();

  Logger::pfcp().trace(
      "Sending %s, seq %d seid " SEID_FMT " ", pfcp_ies.get_msg_name(),
      msg.get_sequence_number(), seid);
  pfcp_procedure proc   = {};
  proc.initial_msg_type = msg.get_message_type();
  proc.trxn_id          = trxn_id;
  proc.retry_msg        = std::make_shared<pfcp_msg>(msg);
  proc.remote_endpoint  = dest;
  start_msg_retry_timer(
      proc, PFCP_T1_RESPONSE_MS, task_id, msg.get_sequence_number());
  start_proc_cleanup_timer(
      proc, PFCP_PROC_TIME_OUT_MS, task_id, msg.get_sequence_number());
  pending_procedures.insert(
      std::pair<uint32_t, pfcp_procedure>(msg.get_sequence_number(), proc));
  trxn_id2seq_num.insert(
      std::pair<uint64_t, uint32_t>(proc.trxn_id, msg.get_sequence_number()));

  // TTN: temporary fix to be able to work with UPF from dsTester (should be
  // removed later) udp_s_allocated.async_send_to(reinterpret_cast<const
  // char*>(bstream.c_str()), bstream.length(), dest);
  udp_s_8805.async_send_to(
      reinterpret_cast<const char*>(bstream.c_str()), bstream.length(), dest);
  return msg.get_sequence_number();
}
//------------------------------------------------------------------------------
uint32_t pfcp_l4_stack::send_request(
    const endpoint& dest, const uint64_t seid,
    const pfcp_session_modification_request& pfcp_ies, const task_id_t& task_id,
    const uint64_t trxn_id) {
  std::ostringstream oss(std::ostringstream::binary);
  pfcp_msg msg(pfcp_ies);
  msg.set_seid(seid);
  msg.set_sequence_number(get_next_seq_num());
  msg.dump_to(oss);
  // std::cout << string_to_hex(oss.str()) << std::endl;
  // std::cout << std::hex << "msg length 0x" << msg.get_message_length() <<
  // "msg seqnum 0x" << msg.get_sequence_number() << std::endl;
  std::string bstream = oss.str();

  Logger::pfcp().trace(
      "Sending %s, seq %d seid " SEID_FMT " ", pfcp_ies.get_msg_name(),
      msg.get_sequence_number(), seid);
  pfcp_procedure proc   = {};
  proc.initial_msg_type = msg.get_message_type();
  proc.trxn_id          = trxn_id;
  proc.retry_msg        = std::make_shared<pfcp_msg>(msg);
  proc.remote_endpoint  = dest;
  start_msg_retry_timer(
      proc, PFCP_T1_RESPONSE_MS, task_id, msg.get_sequence_number());
  start_proc_cleanup_timer(
      proc, PFCP_PROC_TIME_OUT_MS, task_id, msg.get_sequence_number());
  pending_procedures.insert(
      std::pair<uint32_t, pfcp_procedure>(msg.get_sequence_number(), proc));
  trxn_id2seq_num.insert(
      std::pair<uint64_t, uint32_t>(proc.trxn_id, msg.get_sequence_number()));

  // TTN: temporary fix to be able to work with UPF from dsTester (should be
  // removed later) udp_s_allocated.async_send_to(reinterpret_cast<const
  // char*>(bstream.c_str()), bstream.length(), dest);
  udp_s_8805.async_send_to(
      reinterpret_cast<const char*>(bstream.c_str()), bstream.length(), dest);
  return msg.get_sequence_number();
}
////------------------------------------------------------------------------------
// uint32_t pfcp_l4_stack::send_request(const endpoint& dest, const uint64_t
// seid, const pfcp_session_modification_request& pfcp_ies, const task_id_t&
// task_id, const uint64_t trxn_id)
//{
//  std::ostringstream oss(std::ostringstream::binary);
//  pfcp_msg msg(pfcp_ies);
//  msg.set_seid(seid);
//  msg.set_sequence_number(get_next_seq_num());
//  msg.dump_to(oss);
//  //std::cout << string_to_hex(oss.str()) << std::endl;
//  //std::cout << std::hex << "msg length 0x" << msg.get_message_length() <<
//  "msg seqnum 0x" << msg.get_sequence_number() << std::endl; std::string
//  bstream = oss.str();
//
//  Logger::pfcp().trace( "Sending %s, seq %d", pfcp_ies.get_msg_name(),
//  msg.get_sequence_number()); pfcp_procedure proc = {}; proc.initial_msg_type
//  = msg.get_message_type(); proc.trxn_id = trxn_id; proc.retry_msg =
//  std::make_shared<pfcp_msg>(msg); proc.remote_endpoint = dest;
//  start_msg_retry_timer(proc, PFCP_T1_RESPONSE_MS, task_id,
//  msg.get_sequence_number()); start_proc_cleanup_timer(proc,
//  PFCP_PROC_TIME_OUT_MS, task_id, msg.get_sequence_number());
//  pending_procedures.insert(std::pair<uint32_t,
//  pfcp_procedure>(msg.get_sequence_number(), proc));
//  trxn_id2seq_num.insert(std::pair<uint64_t, uint32_t>(proc.trxn_id,
//  msg.get_sequence_number()));
//
//  udp_s.async_send_to(sm, dest);
//  return msg.get_sequence_number();
//}
//------------------------------------------------------------------------------
uint32_t pfcp_l4_stack::send_request(
    const endpoint& dest, const uint64_t seid,
    const pfcp_session_deletion_request& pfcp_ies, const task_id_t& task_id,
    const uint64_t trxn_id) {
  std::ostringstream oss(std::ostringstream::binary);
  pfcp_msg msg(pfcp_ies);
  msg.set_seid(seid);
  msg.set_sequence_number(get_next_seq_num());
  msg.dump_to(oss);
  // std::cout << string_to_hex(oss.str()) << std::endl;
  // std::cout << std::hex << "msg length 0x" << msg.get_message_length() <<
  // "msg seqnum 0x" << msg.get_sequence_number() << std::endl;
  std::string bstream = oss.str();

  Logger::pfcp().trace(
      "Sending %s, seq %d seid " SEID_FMT " ", pfcp_ies.get_msg_name(),
      msg.get_sequence_number(), seid);
  pfcp_procedure proc   = {};
  proc.initial_msg_type = msg.get_message_type();
  proc.trxn_id          = trxn_id;
  proc.retry_msg        = std::make_shared<pfcp_msg>(msg);
  proc.remote_endpoint  = dest;
  start_msg_retry_timer(
      proc, PFCP_T1_RESPONSE_MS, task_id, msg.get_sequence_number());
  start_proc_cleanup_timer(
      proc, PFCP_PROC_TIME_OUT_MS, task_id, msg.get_sequence_number());
  pending_procedures.insert(
      std::pair<uint32_t, pfcp_procedure>(msg.get_sequence_number(), proc));
  trxn_id2seq_num.insert(
      std::pair<uint64_t, uint32_t>(proc.trxn_id, msg.get_sequence_number()));

  // TTN: temporary fix to be able to work with UPF from dsTester (should be
  // removed later) udp_s_allocated.async_send_to(reinterpret_cast<const
  // char*>(bstream.c_str()), bstream.length(), dest);
  udp_s_8805.async_send_to(
      reinterpret_cast<const char*>(bstream.c_str()), bstream.length(), dest);
  return msg.get_sequence_number();
}
//------------------------------------------------------------------------------
uint32_t pfcp_l4_stack::send_request(
    const endpoint& dest, const uint64_t seid,
    const pfcp_session_report_request& pfcp_ies, const task_id_t& task_id,
    const uint64_t trxn_id) {
  std::ostringstream oss(std::ostringstream::binary);
  pfcp_msg msg(pfcp_ies);
  msg.set_seid(seid);
  msg.set_sequence_number(get_next_seq_num());
  msg.dump_to(oss);
  // std::cout << string_to_hex(oss.str()) << std::endl;
  // std::cout << std::hex << "msg length 0x" << msg.get_message_length() <<
  // "msg seqnum 0x" << msg.get_sequence_number() << std::endl;
  std::string bstream = oss.str();

  Logger::pfcp().trace(
      "Sending %s, seq %d seid " SEID_FMT " ", pfcp_ies.get_msg_name(),
      msg.get_sequence_number(), seid);
  pfcp_procedure proc   = {};
  proc.initial_msg_type = msg.get_message_type();
  proc.trxn_id          = trxn_id;
  proc.retry_msg        = std::make_shared<pfcp_msg>(msg);
  proc.remote_endpoint  = dest;
  start_msg_retry_timer(
      proc, PFCP_T1_RESPONSE_MS, task_id, msg.get_sequence_number());
  start_proc_cleanup_timer(
      proc, PFCP_PROC_TIME_OUT_MS, task_id, msg.get_sequence_number());
  pending_procedures.insert(
      std::pair<uint32_t, pfcp_procedure>(msg.get_sequence_number(), proc));
  trxn_id2seq_num.insert(
      std::pair<uint64_t, uint32_t>(proc.trxn_id, msg.get_sequence_number()));

  // TTN: temporary fix to be able to work with UPF from dsTester (should be
  // removed later) udp_s_allocated.async_send_to(reinterpret_cast<const
  // char*>(bstream.c_str()), bstream.length(), dest);
  udp_s_8805.async_send_to(
      reinterpret_cast<const char*>(bstream.c_str()), bstream.length(), dest);
  return msg.get_sequence_number();
}
//------------------------------------------------------------------------------
void pfcp_l4_stack::send_response(
    const endpoint& dest, const pfcp_heartbeat_response& pfcp_ies,
    const uint64_t trxn_id, const pfcp_transaction_action& a) {
  std::map<uint64_t, uint32_t>::iterator it;
  it = trxn_id2seq_num.find(trxn_id);
  if (it != trxn_id2seq_num.end()) {
    std::ostringstream oss(std::ostringstream::binary);
    pfcp_msg msg(pfcp_ies);
    msg.set_sequence_number(it->second);
    msg.dump_to(oss);
    std::string bstream = oss.str();
    Logger::pfcp().trace(
        "Sending %s, seq %d", pfcp_ies.get_msg_name(),
        msg.get_sequence_number());
    udp_s_8805.async_send_to(
        reinterpret_cast<const char*>(bstream.c_str()), bstream.length(), dest);

    if (a == DELETE_TX) {
      std::map<uint32_t, pfcp_procedure>::iterator it_proc =
          pending_procedures.find(it->second);
      if (it_proc != pending_procedures.end()) {
        stop_proc_cleanup_timer(it_proc->second);
        pending_procedures.erase(it_proc);
      }
      trxn_id2seq_num.erase(it);
    }
  } else {
    Logger::pfcp().error(
        "Sending %s, trxn_id %ld proc not found, discarded!",
        pfcp_ies.get_msg_name(), trxn_id);
  }
}
//------------------------------------------------------------------------------
void pfcp_l4_stack::send_response(
    const endpoint& dest, const pfcp_association_setup_response& pfcp_ies,
    const uint64_t trxn_id, const pfcp_transaction_action& a) {
  std::map<uint64_t, uint32_t>::iterator it;
  it = trxn_id2seq_num.find(trxn_id);
  if (it != trxn_id2seq_num.end()) {
    std::ostringstream oss(std::ostringstream::binary);
    pfcp_msg msg(pfcp_ies);
    msg.set_sequence_number(it->second);
    msg.dump_to(oss);
    std::string bstream = oss.str();
    Logger::pfcp().trace(
        "Sending %s, seq %d", pfcp_ies.get_msg_name(),
        msg.get_sequence_number());
    udp_s_8805.async_send_to(
        reinterpret_cast<const char*>(bstream.c_str()), bstream.length(), dest);

    if (a == DELETE_TX) {
      std::map<uint32_t, pfcp_procedure>::iterator it_proc =
          pending_procedures.find(it->second);
      if (it_proc != pending_procedures.end()) {
        stop_proc_cleanup_timer(it_proc->second);
        pending_procedures.erase(it_proc);
      }
      trxn_id2seq_num.erase(it);
    }
  } else {
    Logger::pfcp().error(
        "Sending %s, trxn_id %ld proc not found, discarded!",
        pfcp_ies.get_msg_name(), trxn_id);
  }
}

//------------------------------------------------------------------------------
void pfcp_l4_stack::send_response(
    const endpoint& dest, const pfcp_association_release_response& pfcp_ies,
    const uint64_t trxn_id, const pfcp_transaction_action& a) {
  std::map<uint64_t, uint32_t>::iterator it;
  it = trxn_id2seq_num.find(trxn_id);
  if (it != trxn_id2seq_num.end()) {
    std::ostringstream oss(std::ostringstream::binary);
    pfcp_msg msg(pfcp_ies);
    msg.set_sequence_number(it->second);
    msg.dump_to(oss);
    std::string bstream = oss.str();
    Logger::pfcp().trace(
        "Sending %s, seq %d", pfcp_ies.get_msg_name(),
        msg.get_sequence_number());
    udp_s_8805.async_send_to(
        reinterpret_cast<const char*>(bstream.c_str()), bstream.length(), dest);

    if (a == DELETE_TX) {
      std::map<uint32_t, pfcp_procedure>::iterator it_proc =
          pending_procedures.find(it->second);
      if (it_proc != pending_procedures.end()) {
        stop_proc_cleanup_timer(it_proc->second);
        pending_procedures.erase(it_proc);
      }
      trxn_id2seq_num.erase(it);
    }
  } else {
    Logger::pfcp().error(
        "Sending %s, trxn_id %ld proc not found, discarded!",
        pfcp_ies.get_msg_name(), trxn_id);
  }
}

//------------------------------------------------------------------------------
void pfcp_l4_stack::send_response(
    const endpoint& dest, const uint64_t seid,
    const pfcp_session_establishment_response& pfcp_ies, const uint64_t trxn_id,
    const pfcp_transaction_action& a) {
  std::map<uint64_t, uint32_t>::iterator it;
  it = trxn_id2seq_num.find(trxn_id);
  if (it != trxn_id2seq_num.end()) {
    std::ostringstream oss(std::ostringstream::binary);
    pfcp_msg msg(pfcp_ies);
    msg.set_seid(seid);
    msg.set_sequence_number(it->second);
    msg.dump_to(oss);
    std::string bstream = oss.str();
    Logger::pfcp().trace(
        "Sending %s, seq %d seid " SEID_FMT " ", pfcp_ies.get_msg_name(),
        msg.get_sequence_number(), seid);
    udp_s_8805.async_send_to(
        reinterpret_cast<const char*>(bstream.c_str()), bstream.length(), dest);

    if (a == DELETE_TX) {
      std::map<uint32_t, pfcp_procedure>::iterator it_proc =
          pending_procedures.find(it->second);
      if (it_proc != pending_procedures.end()) {
        stop_proc_cleanup_timer(it_proc->second);
        pending_procedures.erase(it_proc);
      }
      trxn_id2seq_num.erase(it);
    }
  } else {
    Logger::pfcp().error(
        "Sending %s, trxn_id %ld proc not found, discarded!",
        pfcp_ies.get_msg_name(), trxn_id);
  }
}
//------------------------------------------------------------------------------
void pfcp_l4_stack::send_response(
    const endpoint& dest, const uint64_t seid,
    const pfcp_session_modification_response& pfcp_ies, const uint64_t trxn_id,
    const pfcp_transaction_action& a) {
  std::map<uint64_t, uint32_t>::iterator it;
  it = trxn_id2seq_num.find(trxn_id);
  if (it != trxn_id2seq_num.end()) {
    std::ostringstream oss(std::ostringstream::binary);
    pfcp_msg msg(pfcp_ies);
    msg.set_seid(seid);
    msg.set_sequence_number(it->second);
    msg.dump_to(oss);
    std::string bstream = oss.str();
    Logger::pfcp().trace(
        "Sending %s, seq %d seid " SEID_FMT " ", pfcp_ies.get_msg_name(),
        msg.get_sequence_number(), seid);
    udp_s_8805.async_send_to(
        reinterpret_cast<const char*>(bstream.c_str()), bstream.length(), dest);

    if (a == DELETE_TX) {
      std::map<uint32_t, pfcp_procedure>::iterator it_proc =
          pending_procedures.find(it->second);
      if (it_proc != pending_procedures.end()) {
        stop_proc_cleanup_timer(it_proc->second);
        pending_procedures.erase(it_proc);
      }
      trxn_id2seq_num.erase(it);
    }
  } else {
    Logger::pfcp().error(
        "Sending %s, trxn_id %ld proc not found, discarded!",
        pfcp_ies.get_msg_name(), trxn_id);
  }
}
//------------------------------------------------------------------------------
void pfcp_l4_stack::send_response(
    const endpoint& dest, const uint64_t seid,
    const pfcp_session_deletion_response& pfcp_ies, const uint64_t trxn_id,
    const pfcp_transaction_action& a) {
  std::map<uint64_t, uint32_t>::iterator it;
  it = trxn_id2seq_num.find(trxn_id);
  if (it != trxn_id2seq_num.end()) {
    std::ostringstream oss(std::ostringstream::binary);
    pfcp_msg msg(pfcp_ies);
    msg.set_seid(seid);
    msg.set_sequence_number(it->second);
    msg.dump_to(oss);
    std::string bstream = oss.str();
    Logger::pfcp().trace(
        "Sending %s, seq %d seid " SEID_FMT " ", pfcp_ies.get_msg_name(),
        msg.get_sequence_number(), seid);
    udp_s_8805.async_send_to(
        reinterpret_cast<const char*>(bstream.c_str()), bstream.length(), dest);

    if (a == DELETE_TX) {
      std::map<uint32_t, pfcp_procedure>::iterator it_proc =
          pending_procedures.find(it->second);
      if (it_proc != pending_procedures.end()) {
        stop_proc_cleanup_timer(it_proc->second);
        pending_procedures.erase(it_proc);
      }
      trxn_id2seq_num.erase(it);
    }
  } else {
    Logger::pfcp().error(
        "Sending %s, trxn_id %ld proc not found, discarded!",
        pfcp_ies.get_msg_name(), trxn_id);
  }
}
//------------------------------------------------------------------------------
void pfcp_l4_stack::send_response(
    const endpoint& dest, const uint64_t seid,
    const pfcp_session_report_response& pfcp_ies, const uint64_t trxn_id,
    const pfcp_transaction_action& a) {
  std::map<uint64_t, uint32_t>::iterator it;
  it = trxn_id2seq_num.find(trxn_id);
  if (it != trxn_id2seq_num.end()) {
    std::ostringstream oss(std::ostringstream::binary);
    pfcp_msg msg(pfcp_ies);
    msg.set_seid(seid);
    msg.set_sequence_number(it->second);
    msg.dump_to(oss);
    std::string bstream = oss.str();
    Logger::pfcp().trace(
        "Sending %s, seq %d seid " SEID_FMT " to %s", pfcp_ies.get_msg_name(),
        msg.get_sequence_number(), seid, dest.toString().c_str());
    udp_s_8805.async_send_to(
        reinterpret_cast<const char*>(bstream.c_str()), bstream.length(), dest);

    if (a == DELETE_TX) {
      std::map<uint32_t, pfcp_procedure>::iterator it_proc =
          pending_procedures.find(it->second);
      if (it_proc != pending_procedures.end()) {
        stop_proc_cleanup_timer(it_proc->second);
        pending_procedures.erase(it_proc);
      }
      trxn_id2seq_num.erase(it);
    }
  } else {
    Logger::pfcp().error(
        "Sending %s, trxn_id %ld proc not found, discarded!",
        pfcp_ies.get_msg_name(), trxn_id);
  }
}

//------------------------------------------------------------------------------
void pfcp_l4_stack::notify_ul_error(
    const pfcp_procedure& p, const cause_value_e cause) {
  Logger::pfcp().trace(
      "notify_ul_error proc %" PRId64 " cause %d", p.trxn_id, cause);
}
//------------------------------------------------------------------------------
void pfcp_l4_stack::time_out_event(
    const uint32_t timer_id, const task_id_t& task_id, bool& handled) {
  handled = false;
  std::map<timer_id_t, uint32_t>::iterator it =
      msg_out_retry_timers.find(timer_id);
  if (it != msg_out_retry_timers.end()) {
    std::map<uint32_t, pfcp_procedure>::iterator it_proc =
        pending_procedures.find(it->second);
    msg_out_retry_timers.erase(it);
    handled = true;
    if (it_proc != pending_procedures.end()) {
      if (it_proc->second.retry_count < PFCP_N1_REQUESTS) {
        it_proc->second.retry_count++;
        start_msg_retry_timer(
            it_proc->second, PFCP_T1_RESPONSE_MS, task_id,
            it_proc->second.retry_msg->get_sequence_number());
        // send again message
        Logger::pfcp().trace(
            "Retry %d Sending msg type %d, seq %d", it_proc->second.retry_count,
            it_proc->second.retry_msg->get_message_type(),
            it_proc->second.retry_msg->get_sequence_number());
        std::ostringstream oss(std::ostringstream::binary);
        it_proc->second.retry_msg->dump_to(oss);
        std::string bstream = oss.str();
        udp_s_8805.async_send_to(
            reinterpret_cast<const char*>(bstream.c_str()), bstream.length(),
            it_proc->second.remote_endpoint);
      } else {
        // abort procedure
        notify_ul_error(
            it_proc->second,
            cause_value_e::CAUSE_VALUE_PFCP_ENTITY_IN_CONGESTION);
      }
    }
  } else {
    it = proc_cleanup_timers.find(timer_id);
    if (it != proc_cleanup_timers.end()) {
      std::map<uint32_t, pfcp_procedure>::iterator it_proc =
          pending_procedures.find(it->second);
      proc_cleanup_timers.erase(it);
      handled = true;
      if (it_proc != pending_procedures.end()) {
        it_proc->second.proc_cleanup_timer_id = 0;
        Logger::pfcp().trace(
            "Delete proc %" PRId64 " Retry %d seq %d timer id %u",
            it_proc->second.trxn_id, it_proc->second.retry_count,
            it_proc->first, timer_id);
        pending_procedures.erase(it_proc);
      }
    }
  }
}
