AMCProtocolClient.hpp

Go to the documentation of this file.

#ifndef __DUNEDAQ_TDEMODULES_INCLUDE_AMCPROTOCOLCLIENT_HPP___
#define __DUNEDAQ_TDEMODULES_INCLUDE_AMCPROTOCOLCLIENT_HPP___
 
#include <cstdint>
#include <vector>
 
#include <boost/asio.hpp>
#include <boost/endian/arithmetic.hpp>
 
#include "logging/Logging.hpp"
 
namespace dunedaq {
 
ERS_DECLARE_ISSUE(tdemodules,
                  AMCProtocolIssue,
                  prefix << "AMCProtocolResponse: " << text,
                  ((std::string)prefix)((std::string)text));
 
ERS_DECLARE_ISSUE(tdemodules,
                  AMCResponseErr,
                  prefix << "Received Error packet:" << "\n"
                         << "Error code:" << code << "\nMessage: " << text,
                  ((std::string)prefix)((uint16_t)code)((std::string)text));
 
ERS_DECLARE_ISSUE(tdemodules,
                  AMCUnknownOpCode,
                  prefix << "Unknown opcode recived: " << opcode,
                  ((std::string)prefix)((uint16_t)opcode));
 
namespace tdemodules {
 
class AMCProtocolClient
{
public:
  enum TFTPOpCode : uint16_t
  {
    RRQ = 1,
    WRQ = 2,
    DAT = 3,
    ACK = 4,
    ERR = 5,
  };
 
#pragma pack(push, 1)
  struct TFTP_Data_Header
  {
    boost::endian::big_uint16_t opcode;
    boost::endian::big_uint16_t block;
  };
 
  struct TFTP_Error_Header
  {
    boost::endian::big_uint16_t opcode;
    boost::endian::big_uint16_t error_code;
  };
 
  struct TFTP_Ack_Header
  {
    boost::endian::big_uint16_t opcode;
    boost::endian::big_uint16_t block;
  };
#pragma pack(pop)
 
  AMCProtocolClient(const std::string& server_ip, uint16_t port);
 
  std::vector<uint8_t> send_request(TFTPOpCode opcode, const std::vector<uint8_t>& payload);
 
  std::vector<uint8_t> send_rrq(const std::vector<uint8_t>& payload) { return send_request(RRQ, payload); }
 
  std::vector<uint8_t> send_wrq(const std::vector<uint8_t>& payload) { return send_request(WRQ, payload); }
 
private:
  std::string m_host;
  uint16_t m_port;
  uint16_t m_timeout;
 
  const std::string m_log_prefix = "AMC with ip:" + m_host + " and port: " + std::to_string(m_port) + " | ";
 
  boost::asio::io_context m_io_context;
  boost::asio::ip::udp::socket m_socket;
  boost::asio::ip::udp::endpoint m_server_endpoint;
};
 
} // namespace tdemodules
} // namespace dunedaq
 
#endif // __DUNEDAQ_TDEMODULES_INCLUDE_AMCPROTOCOLCLIENT_HPP___