// Copyright 2017 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "src/connectivity/network/mdns/service/agents/host_name_resolver.h"

#include <lib/syslog/cpp/macros.h>
#include <lib/zx/time.h>

#include <iterator>

#include "src/connectivity/network/mdns/service/common/mdns_names.h"

namespace mdns {

HostNameResolver::HostNameResolver(MdnsAgent::Owner* owner, const std::string& host_name,
                                   Media media, IpVersions ip_versions, bool include_local,
                                   bool include_local_proxies, zx::duration timeout,
                                   Mdns::ResolveHostNameCallback callback)
    : MdnsAgent(owner),
      host_name_(host_name),
      host_full_name_(MdnsNames::HostFullName(host_name)),
      media_(media),
      ip_versions_(ip_versions),
      include_local_(include_local),
      include_local_proxies_(include_local_proxies),
      timeout_(timeout),
      callback_(std::move(callback)) {
  FX_DCHECK(callback_);
}

HostNameResolver::~HostNameResolver() {}

void HostNameResolver::Start(const std::string& local_host_full_name) {
  // Note that |host_full_name_| is the name we're trying to resolve, not the
  // name of the local host, which is the parameter to this method.

  MdnsAgent::Start(local_host_full_name);

  if (include_local_ && host_full_name_ == local_host_full_name) {
    auto addresses = local_host_addresses();
    std::copy(addresses.begin(), addresses.end(), std::inserter(addresses_, addresses_.end()));
    InvokeCallbackAndRemoveSelf();
    return;
  }

  SendQuestion(std::make_shared<DnsQuestion>(host_full_name_, DnsType::kA),
               ReplyAddress::Multicast(media_, ip_versions_));
  SendQuestion(std::make_shared<DnsQuestion>(host_full_name_, DnsType::kAaaa),
               ReplyAddress::Multicast(media_, ip_versions_));

  PostTaskForTime(
      [this]() {
        if (callback_) {
          InvokeCallbackAndRemoveSelf();
        }
      },
      now() + timeout_);
}

void HostNameResolver::ReceiveResource(const DnsResource& resource, MdnsResourceSection section,
                                       ReplyAddress sender_address) {
  if (!sender_address.Matches(media_) || !sender_address.Matches(ip_versions_) ||
      resource.name_.dotted_string_ != host_full_name_) {
    return;
  }

  if (resource.type_ == DnsType::kA) {
    addresses_.insert(HostAddress(resource.a_.address_.address_, sender_address.interface_id(),
                                  zx::sec(resource.time_to_live_)));
  } else if (resource.type_ == DnsType::kAaaa) {
    addresses_.insert(HostAddress(resource.aaaa_.address_.address_, sender_address.interface_id(),
                                  zx::sec(resource.time_to_live_)));
  }
}

void HostNameResolver::EndOfMessage() {
  if (!callback_) {
    // This can happen when a redundant response is received after the block below runs and before
    // the posted task runs, e.g. when two NICs are connected to the same LAN.
    return;
  }

  if (!addresses_.empty() && callback_) {
    InvokeCallbackAndRemoveSelf();
  }
}

void HostNameResolver::OnAddProxyHost(const std::string& host_full_name,
                                      const std::vector<HostAddress>& host_addresses) {
  if (!include_local_proxies_ || host_full_name != host_full_name_) {
    return;
  }

  addresses_.clear();
  for (auto& host_address : host_addresses) {
    addresses_.insert(host_address);
  }

  if (callback_) {
    InvokeCallbackAndRemoveSelf();
  }
}

void HostNameResolver::Quit() {
  if (callback_) {
    callback_(host_name_, addresses());
    callback_ = nullptr;
  }

  MdnsAgent::Quit();
}

void HostNameResolver::InvokeCallbackAndRemoveSelf() {
  if (callback_) {
    callback_(host_name_, addresses());
    callback_ = nullptr;
  }

  PostTaskForTime([this]() { RemoveSelf(); }, now());
}

}  // namespace mdns