Resolver.cpp

Go to the documentation of this file.

 
#include "utilities/Resolver.hpp"
 
#include <string>
#include <vector>
 
std::vector<std::string>
dunedaq::utilities::get_ips_from_hostname(std::string hostname)
{
  std::vector<std::string> output;
 
  TLOG_DEBUG(12) << "Name is " << hostname;
 
  struct addrinfo* addrinfo = nullptr;
  auto s = getaddrinfo(hostname.c_str(), nullptr, nullptr, &addrinfo);
 
  if (s != 0) {
    ers::error(NameNotFound(ERS_HERE, hostname, std::string(gai_strerror(s))));
    return output;
  }
 
  for (auto rp = addrinfo; rp != nullptr; rp = rp->ai_next) {
    char hbuf[NI_MAXHOST], sbuf[NI_MAXSERV]; // NOLINT
 
    // Let's skip all the IPv6 here
    if (rp->ai_family == AF_INET6)
      continue;
 
    getnameinfo(rp->ai_addr, rp->ai_addrlen, hbuf, sizeof(hbuf), sbuf, sizeof(sbuf), NI_NUMERICHOST | NI_NUMERICSERV);
    auto ipaddr = std::string(hbuf);
    bool duplicate = false;
    for (auto& ip : output) {
      if (ip == ipaddr) {
        duplicate = true;
        break;
      }
    }
    if (!duplicate) {
      TLOG_DEBUG(13) << "Found address " << ipaddr << " for hostname " << hostname;
      output.push_back(ipaddr);
    }
  }
 
  freeaddrinfo(addrinfo);
 
  return output;
}
 
std::vector<std::string>
dunedaq::utilities::resolve_uri_hostname(std::string connection_string)
{
  ZmqUri uri(connection_string);
 
  if (uri.scheme == "tcp") {
    auto output = get_ips_from_hostname(uri.host);
 
    for (size_t ii = 0; ii < output.size(); ++ii) {
      output[ii] = "tcp://" + output[ii] + ":" + uri.port;
    }
    return output;
  } else {
    return { connection_string };
  }
}
 
// ZMQ URIs are formatted as follows: tcp://{host}:{port}
dunedaq::utilities::ZmqUri::ZmqUri(std::string connection_string)
{
  if (connection_string.find("://") == std::string::npos) {
    throw InvalidUri(ERS_HERE, connection_string);
  }
 
  scheme = connection_string.substr(0, connection_string.find("://"));
  connection_string = connection_string.substr(connection_string.find("://") + 3);
 
  if (connection_string.find(":") != std::string::npos) {
    port = connection_string.substr(connection_string.find(":") + 1);
    connection_string = connection_string.substr(0, connection_string.find(":"));
  }
  host = connection_string;
}