ip_format.hpp

Go to the documentation of this file.

 
#ifndef SNBMODULES_INCLUDE_SNBMODULES_IP_FORMAT_HPP_
#define SNBMODULES_INCLUDE_SNBMODULES_IP_FORMAT_HPP_
 
// #define _CRT_SECURE_NO_WARNINGS
 
#include <cstring>
#include <filesystem>
#include <iostream>
#include <stdexcept>
#include <string>
#include <utility>
#include <vector>
 
namespace dunedaq::snbmodules {
class IPFormat
{
 
public:
  // Constructors
  explicit IPFormat(const std::string& ip = "0.0.0.0", int port = 0)
  {
    set_port(port);
    set_ip(ip);
  }
 
  IPFormat(const IPFormat& o)
  {
    set_port(o.m_port);
    set_ip(o.m_ip);
  }
 
  IPFormat& operator=(IPFormat const&) = default;
 
  bool operator==(IPFormat const& o) const { return m_ip == o.m_ip && m_port == o.m_port; }
 
  bool operator<(IPFormat const& o) const { return m_ip.compare(o.m_ip) || (m_ip == o.m_ip && m_port < o.m_port); }
 
  std::string get_ip_port() const { return m_ip + ":" + std::to_string(m_port); }
 
  inline bool is_default() const { return m_ip == "0.0.0.0" && m_port == 0; }
 
  // Setters
  void set_port(int port)
  {
 
    if (port < 0 || port > 65535) {
      throw std::invalid_argument("Port must be between 0 and 65535");
    }
 
    m_port = port;
  }
 
#ifndef _MSC_VER
  inline char* strtok_s(char* s, const char* delim, char** context) { return strtok_r(s, delim, context); }
#endif // !_MSC_VER
 
  void set_ip(const std::string& ip)
  {
    if (ip.empty()) {
      throw std::invalid_argument("IP address cannot be empty");
    }
 
    // Splitting the string IP:PORT or IP
    std::vector<std::string> ip_port_pair;
    char* next_token = nullptr;
 
    char* ip_char = new char[ip.length() + 1];
    strcpy(ip_char, ip.c_str());
 
    char* token = strtok_s(ip_char, ":", &next_token);
    while (token != nullptr) {
      ip_port_pair.emplace_back(std::string(token));
      token = strtok_s(nullptr, ":", &next_token);
    }
 
    delete[] ip_char;
 
    if (ip_port_pair.size() != 1 && ip_port_pair.size() != 2) {
      throw std::invalid_argument("Invalid IP address format (IP:PORT or IP)");
    }
 
    // TODO Aug-14-2022 Leo Joly leo.vincent.andre.joly@cern.ch : Check if the IPv4 address is valid
 
    // set values
    if (ip_port_pair.size() == 2) {
      set_port(std::stoi(ip_port_pair[1]));
    }
 
    m_ip = ip_port_pair[0];
  }
 
  // Getters
  std::string get_ip() const { return m_ip; }
  int get_port() const { return m_port; }
 
private:
  std::string m_ip = "0.0.0.0";
  int m_port = 0;
};
} // namespace dunedaq::snbmodules
#endif // SNBMODULES_INCLUDE_SNBMODULES_IP_FORMAT_HPP_