udp_comm.h

Go to the documentation of this file.

 
#ifndef UDP_COMM_H_
#define UDP_COMM_H_
 
#include <netdb.h>
 
#include <cstddef>
#include <map>
#include <mutex>
#include <string>
#include <vector>
 
// Basic UDP client class based on OS sockets that supports sending messages
// and caches remote addrinfo mappings
class UDPComm {
 public:
  static constexpr bool kDebugPrintUdpInit = false;
  static constexpr bool kDebugPrintUdpSend = false;
  static constexpr bool kDebugPrintUdpRecv = false;
  explicit UDPComm(std::string local_addr, uint16_t local_port,
                   size_t rx_buffer_size, size_t tx_buffer_size);
 
  explicit UDPComm(std::string local_addr, const std::string& local_port,
                   size_t rx_buffer_size, size_t tx_buffer_size);
 
  UDPComm& operator=(const UDPComm&) = delete;
  UDPComm(const UDPComm&) = delete;
  ~UDPComm();
 
  ssize_t Connect(const std::string& remote_address,
                  uint16_t remote_port) const;
  ssize_t Connect(const std::string& remote_address,
                  const std::string& remote_port) const;
 
  void Send(const std::string& rem_hostname, uint16_t rem_port,
            const std::byte* msg, size_t len);
 
  void Send(const std::byte* msg, size_t len);
 
  ssize_t Recv(std::byte* buf, size_t len) const;
  ssize_t Recv(const std::string& src_address, uint16_t src_port,
               std::byte* buf, size_t len);
 
  // Enable recording of all packets sent by this UDPComm object
  inline void EnableRecording() { enable_recording_flag_ = true; }
 
  void MakeBlocking(size_t rx_timeout_sec = 0) const;
 
 private:
  int sock_fd_ = -1;
 
  std::map<std::string, addrinfo*> addrinfo_map_;
  std::mutex map_insert_access_;
 
  std::vector<std::vector<uint8_t>> sent_vec_;
 
  bool enable_recording_flag_ = false;
};
 
#endif  // UDP_CLIENT_H_