nearby/internal/platform/implementation/linux/avahi.cc

#include "internal/platform/implementation/linux/avahi.h"
#include "internal/platform/implementation/linux/dbus.h"
#include "internal/platform/logging.h"
#include "internal/platform/nsd_service_info.h"

namespace nearby {
namespace linux {
namespace avahi {
void ServiceBrowser::onItemNew(const int32_t &interface,
                               const int32_t &protocol, const std::string &name,
                               const std::string &type,
                               const std::string &domain,
                               const uint32_t &flags) {
  NEARBY_LOGS(VERBOSE) << __func__ << ": " << getObjectPath()
                       << ": Found new item through the ServiceBrowser: "
                       << "interface: " << interface << ", protocol: "
                       << protocol << ", name: '" << name << "', type: '"
                       << type << "', domain: '" << domain
                       << "', flags: " << flags;
  if (flags & kAvahiLookupResultLocal) {
    NEARBY_LOGS(VERBOSE) << __func__ << ": Ignoring local service.";
    return;
  }

  NsdServiceInfo info;
  try {
    auto [r_iface, r_protocol, r_name, r_type, r_domain, r_host, r_aprotocol,
          r_address, r_port, r_txt, r_flags] =
        server_->ResolveService(interface, protocol, name, type, domain,
                                0, // AVAHI_PROTO_INET
                                0);
    info.SetServiceName(r_name);
    info.SetIPAddress(r_address);
    info.SetPort(r_port);
    info.SetServiceType(r_type);
    for (auto &attr : r_txt) {
      auto attr_str = std::string(attr.begin(), attr.end());
      size_t pos = attr_str.find('=');
      if (pos == 0 || pos == std::string::npos || pos == attr_str.size() - 1) {
        NEARBY_LOGS(WARNING) << " found invalid text attribute: " << attr_str;
      }

      info.SetTxtRecord(attr_str.substr(0, pos), attr_str.substr(pos + 1));
    }
  } catch (const sdbus::Error &e) {
    DBUS_LOG_METHOD_CALL_ERROR(server_, "ResolveService", e);
  }

  discovery_cb_.service_discovered_cb(std::move(info));
}

void ServiceBrowser::onItemRemove(
    const int32_t &interface, const int32_t &protocol, const std::string &name,
    const std::string &type, const std::string &domain, const uint32_t &flags) {
  // TODO: Can we even resolve removed items?
  NEARBY_LOGS(VERBOSE) << __func__ << ": " << getObjectPath()
                       << ": Item removed through the ServiceBrowser: "
                       << "interface: " << interface << ", protocol: "
                       << protocol << ", name: '" << name << "', type: '"
                       << type << "', domain: '" << domain
                       << "', flags: " << flags;
  if (flags & kAvahiLookupResultLocal) {
    NEARBY_LOGS(VERBOSE) << __func__ << ": Ignoring local service.";
    return;
  }

  NsdServiceInfo info;
  try {
    auto [r_iface, r_protocol, r_name, r_type, r_domain, r_host, r_aprotocol,
          r_address, r_port, r_txt, r_flags] =
        server_->ResolveService(interface, protocol, name, type, domain,
                                0, // AVAHI_PROTO_INET
                                flags);
    info.SetServiceName(r_name);
    info.SetIPAddress(r_address);
    info.SetPort(r_port);
    info.SetServiceType(r_type);
    for (auto &attr : r_txt) {
      auto attr_str = std::string(attr.begin(), attr.end());
      size_t pos = attr_str.find('=');
      if (pos == 0 || pos == std::string::npos || pos == attr_str.size() - 1) {
        NEARBY_LOGS(WARNING) << " found invalid text attribute: " << attr_str;
      }

      info.SetTxtRecord(attr_str.substr(0, pos), attr_str.substr(pos + 1));
    }
  } catch (const sdbus::Error &e) {
    DBUS_LOG_METHOD_CALL_ERROR(server_, "ResolveService", e);
  }

  discovery_cb_.service_lost_cb(std::move(info));
}

void ServiceBrowser::onFailure(const std::string &error) {
  NEARBY_LOGS(ERROR) << __func__ << ": " << getObjectPath()
                     << ": ServiceBrowser reported a failure: " << error;
}

void ServiceBrowser::onAllForNow() {
  NEARBY_LOGS(VERBOSE) << __func__ << ": " << getObjectPath()
                       << ": notified via ServiceBrowser that all records have "
                          "been added for now";
}

void ServiceBrowser::onCacheExhausted() {
  NEARBY_LOGS(VERBOSE) << __func__ << ": " << getObjectPath()
                       << ": notified via ServiceBrowser of cache exhaustion";
}

} // namespace avahi
} // namespace linux
} // namespace nearby