diff options
Diffstat (limited to 'osp/impl/discovery')
-rw-r--r-- | osp/impl/discovery/mdns/BUILD.gn | 67 | ||||
-rw-r--r-- | osp/impl/discovery/mdns/DEPS | 7 | ||||
-rw-r--r-- | osp/impl/discovery/mdns/domain_name.cc | 132 | ||||
-rw-r--r-- | osp/impl/discovery/mdns/domain_name.h | 91 | ||||
-rw-r--r-- | osp/impl/discovery/mdns/domain_name_unittest.cc | 194 | ||||
-rw-r--r-- | osp/impl/discovery/mdns/mdns_demo.cc | 374 | ||||
-rw-r--r-- | osp/impl/discovery/mdns/mdns_responder_adapter.cc | 77 | ||||
-rw-r--r-- | osp/impl/discovery/mdns/mdns_responder_adapter.h | 258 | ||||
-rw-r--r-- | osp/impl/discovery/mdns/mdns_responder_adapter_impl.cc | 1044 | ||||
-rw-r--r-- | osp/impl/discovery/mdns/mdns_responder_adapter_impl.h | 159 | ||||
-rw-r--r-- | osp/impl/discovery/mdns/mdns_responder_adapter_impl_unittest.cc | 94 | ||||
-rw-r--r-- | osp/impl/discovery/mdns/mdns_responder_platform.cc | 290 | ||||
-rw-r--r-- | osp/impl/discovery/mdns/mdns_responder_platform.h | 16 |
13 files changed, 0 insertions, 2803 deletions
diff --git a/osp/impl/discovery/mdns/BUILD.gn b/osp/impl/discovery/mdns/BUILD.gn deleted file mode 100644 index cd051d17..00000000 --- a/osp/impl/discovery/mdns/BUILD.gn +++ /dev/null @@ -1,67 +0,0 @@ -# Copyright 2018 The Chromium Authors. All rights reserved. -# Use of this source code is governed by a BSD-style license that can be -# found in the LICENSE file. - -import("../../../build/config/services.gni") -assert(use_mdns_responder) - -source_set("mdns_interface") { - sources = [ - "domain_name.cc", - "domain_name.h", - "mdns_responder_adapter.cc", - "mdns_responder_adapter.h", - ] - - public_deps = [ - "../../../../platform", - "../../../../third_party/abseil", - "../../../../util", - ] -} - -source_set("unittests") { - testonly = true - - sources = [ - "domain_name_unittest.cc", - ] - - deps = [ - ":mdns_interface", - "../../../../third_party/googletest:gmock", - "../../../../third_party/googletest:gtest", - ] - - sources += [ "mdns_responder_adapter_impl_unittest.cc" ] - deps += [ ":mdns" ] -} - -executable("mdns_demo") { - sources = [ - "mdns_demo.cc", - ] - - deps = [ - ":mdns", - ] -} - -source_set("mdns") { - sources = [ - "mdns_responder_adapter_impl.cc", - "mdns_responder_adapter_impl.h", - "mdns_responder_platform.cc", - "mdns_responder_platform.h", - ] - - public_deps = [ - ":mdns_interface", - "../../../../platform", - "../../../../util", - ] - - deps = [ - "../../../../third_party/mDNSResponder:core", - ] -} diff --git a/osp/impl/discovery/mdns/DEPS b/osp/impl/discovery/mdns/DEPS deleted file mode 100644 index 96a7209c..00000000 --- a/osp/impl/discovery/mdns/DEPS +++ /dev/null @@ -1,7 +0,0 @@ -# Copyright 2019 The Chromium 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_rules = [ - '+platform/impl', # Needed by embedder_demo.cc -] diff --git a/osp/impl/discovery/mdns/domain_name.cc b/osp/impl/discovery/mdns/domain_name.cc deleted file mode 100644 index c574793a..00000000 --- a/osp/impl/discovery/mdns/domain_name.cc +++ /dev/null @@ -1,132 +0,0 @@ -// Copyright 2018 The Chromium 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 "osp/impl/discovery/mdns/domain_name.h" - -#include <algorithm> -#include <iterator> - -#include "util/stringprintf.h" - -namespace openscreen { -namespace osp { - -// static -DomainName DomainName::GetLocalDomain() { - return DomainName{{5, 'l', 'o', 'c', 'a', 'l', 0}}; -} - -// static -ErrorOr<DomainName> DomainName::Append(const DomainName& first, - const DomainName& second) { - OSP_CHECK(first.domain_name_.size()); - OSP_CHECK(second.domain_name_.size()); - - // Both vectors should represent null terminated domain names. - OSP_DCHECK_EQ(first.domain_name_.back(), '\0'); - OSP_DCHECK_EQ(second.domain_name_.back(), '\0'); - if ((first.domain_name_.size() + second.domain_name_.size() - 1) > - kDomainNameMaxLength) { - return Error::Code::kDomainNameTooLong; - } - - DomainName result; - result.domain_name_.clear(); - result.domain_name_.insert(result.domain_name_.begin(), - first.domain_name_.begin(), - first.domain_name_.end()); - result.domain_name_.insert(result.domain_name_.end() - 1, - second.domain_name_.begin(), - second.domain_name_.end() - 1); - return result; -} - -DomainName::DomainName() : domain_name_{0u} {} -DomainName::DomainName(std::vector<uint8_t>&& domain_name) - : domain_name_(std::move(domain_name)) { - OSP_CHECK_LE(domain_name_.size(), kDomainNameMaxLength); -} -DomainName::DomainName(const DomainName&) = default; -DomainName::DomainName(DomainName&&) noexcept = default; -DomainName::~DomainName() = default; -DomainName& DomainName::operator=(const DomainName&) = default; -DomainName& DomainName::operator=(DomainName&&) noexcept = default; - -bool DomainName::operator==(const DomainName& other) const { - if (domain_name_.size() != other.domain_name_.size()) { - return false; - } - for (size_t i = 0; i < domain_name_.size(); ++i) { - if (tolower(domain_name_[i]) != tolower(other.domain_name_[i])) { - return false; - } - } - return true; -} - -bool DomainName::operator!=(const DomainName& other) const { - return !(*this == other); -} - -bool DomainName::EndsWithLocalDomain() const { - const DomainName local_domain = GetLocalDomain(); - if (domain_name_.size() < local_domain.domain_name_.size()) - return false; - - return std::equal(local_domain.domain_name_.begin(), - local_domain.domain_name_.end(), - domain_name_.end() - local_domain.domain_name_.size()); -} - -Error DomainName::Append(const DomainName& after) { - OSP_CHECK(after.domain_name_.size()); - OSP_DCHECK_EQ(after.domain_name_.back(), 0u); - - if ((domain_name_.size() + after.domain_name_.size() - 1) > - kDomainNameMaxLength) { - return Error::Code::kDomainNameTooLong; - } - - domain_name_.insert(domain_name_.end() - 1, after.domain_name_.begin(), - after.domain_name_.end() - 1); - return Error::None(); -} - -std::vector<absl::string_view> DomainName::GetLabels() const { - OSP_DCHECK_GT(domain_name_.size(), 0u); - OSP_DCHECK_LT(domain_name_.size(), kDomainNameMaxLength); - - std::vector<absl::string_view> result; - const uint8_t* data = domain_name_.data(); - while (*data != 0) { - const size_t label_length = *data; - OSP_DCHECK_LT(label_length, kDomainNameMaxLabelLength); - - ++data; - result.emplace_back(reinterpret_cast<const char*>(data), label_length); - data += label_length; - } - return result; -} - -bool DomainNameComparator::operator()(const DomainName& a, - const DomainName& b) const { - return a.domain_name() < b.domain_name(); -} - -std::ostream& operator<<(std::ostream& os, const DomainName& domain_name) { - const auto& data = domain_name.domain_name(); - OSP_DCHECK_GT(data.size(), 0u); - auto it = data.begin(); - while (*it != 0) { - size_t length = *it++; - PrettyPrintAsciiHex(os, it, it + length); - it += length; - os << "."; - } - return os; -} - -} // namespace osp -} // namespace openscreen diff --git a/osp/impl/discovery/mdns/domain_name.h b/osp/impl/discovery/mdns/domain_name.h deleted file mode 100644 index c29ef9db..00000000 --- a/osp/impl/discovery/mdns/domain_name.h +++ /dev/null @@ -1,91 +0,0 @@ -// Copyright 2018 The Chromium Authors. All rights reserved. -// Use of this source code is governed by a BSD-style license that can be -// found in the LICENSE file. - -#ifndef OSP_IMPL_DISCOVERY_MDNS_DOMAIN_NAME_H_ -#define OSP_IMPL_DISCOVERY_MDNS_DOMAIN_NAME_H_ - -#include <cstdint> -#include <ostream> -#include <string> -#include <vector> - -#include "absl/strings/string_view.h" -#include "platform/base/error.h" -#include "util/osp_logging.h" - -namespace openscreen { -namespace osp { - -struct DomainName { - static ErrorOr<DomainName> Append(const DomainName& first, - const DomainName& second); - - template <typename It> - static ErrorOr<DomainName> FromLabels(It first, It last) { - size_t total_length = 1; - for (auto label = first; label != last; ++label) { - if (label->size() > kDomainNameMaxLabelLength) - return Error::Code::kDomainNameLabelTooLong; - - total_length += label->size() + 1; - } - if (total_length > kDomainNameMaxLength) - return Error::Code::kDomainNameTooLong; - - DomainName result; - result.domain_name_.resize(total_length); - auto result_it = result.domain_name_.begin(); - for (auto label = first; label != last; ++label) { - *result_it++ = static_cast<uint8_t>(label->size()); - result_it = std::copy(label->begin(), label->end(), result_it); - } - *result_it = 0; - return std::move(result); - } - - static DomainName GetLocalDomain(); - - static constexpr uint8_t kDomainNameMaxLabelLength = 63u; - static constexpr uint16_t kDomainNameMaxLength = 256u; - - DomainName(); - explicit DomainName(std::vector<uint8_t>&& domain_name); - DomainName(const DomainName&); - DomainName(DomainName&&) noexcept; - ~DomainName(); - DomainName& operator=(const DomainName&); - DomainName& operator=(DomainName&&) noexcept; - - bool operator==(const DomainName& other) const; - bool operator!=(const DomainName& other) const; - - bool EndsWithLocalDomain() const; - bool IsEmpty() const { return domain_name_.size() == 1 && !domain_name_[0]; } - - Error Append(const DomainName& after); - std::vector<absl::string_view> GetLabels() const; - - const std::vector<uint8_t>& domain_name() const { return domain_name_; } - - private: - // RFC 1035 domain name format: sequence of 1 octet label length followed by - // label data, ending with a 0 octet. May not exceed 256 bytes (including - // terminating 0). - // For example, openscreen.org would be encoded as: - // {10, 'o', 'p', 'e', 'n', 's', 'c', 'r', 'e', 'e', 'n', - // 3, 'o', 'r', 'g', 0} - std::vector<uint8_t> domain_name_; -}; - -class DomainNameComparator { - public: - bool operator()(const DomainName& a, const DomainName& b) const; -}; - -std::ostream& operator<<(std::ostream& os, const DomainName& domain_name); - -} // namespace osp -} // namespace openscreen - -#endif // OSP_IMPL_DISCOVERY_MDNS_DOMAIN_NAME_H_ diff --git a/osp/impl/discovery/mdns/domain_name_unittest.cc b/osp/impl/discovery/mdns/domain_name_unittest.cc deleted file mode 100644 index 76f31003..00000000 --- a/osp/impl/discovery/mdns/domain_name_unittest.cc +++ /dev/null @@ -1,194 +0,0 @@ -// Copyright 2018 The Chromium 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 "osp/impl/discovery/mdns/domain_name.h" - -#include <sstream> - -#include "gtest/gtest.h" -#include "platform/base/error.h" - -namespace openscreen { -namespace osp { - -namespace { - -ErrorOr<DomainName> FromLabels(const std::vector<std::string>& labels) { - return DomainName::FromLabels(labels.begin(), labels.end()); -} - -template <typename T> -T UnpackErrorOr(ErrorOr<T> error_or) { - EXPECT_TRUE(error_or); - return std::move(error_or.value()); -} - -} // namespace - -TEST(DomainNameTest, Constructors) { - DomainName empty; - - ASSERT_EQ(1u, empty.domain_name().size()); - EXPECT_EQ(0, empty.domain_name()[0]); - - DomainName original({10, 'o', 'p', 'e', 'n', 's', 'c', 'r', 'e', 'e', 'n', 3, - 'o', 'r', 'g', 0}); - ASSERT_EQ(16u, original.domain_name().size()); - - auto data_copy = original.domain_name(); - DomainName direct_ctor(std::move(data_copy)); - EXPECT_EQ(direct_ctor.domain_name(), original.domain_name()); - - DomainName copy_ctor(original); - EXPECT_EQ(copy_ctor.domain_name(), original.domain_name()); - - DomainName move_ctor(std::move(copy_ctor)); - EXPECT_EQ(move_ctor.domain_name(), original.domain_name()); - - DomainName copy_assign; - copy_assign = move_ctor; - EXPECT_EQ(copy_assign.domain_name(), original.domain_name()); - - DomainName move_assign; - move_assign = std::move(move_ctor); - EXPECT_EQ(move_assign.domain_name(), original.domain_name()); -} - -TEST(DomainNameTest, FromLabels) { - const auto typical = - std::vector<uint8_t>{10, 'o', 'p', 'e', 'n', 's', 'c', 'r', - 'e', 'e', 'n', 3, 'o', 'r', 'g', 0}; - DomainName result = UnpackErrorOr(FromLabels({"openscreen", "org"})); - EXPECT_EQ(result.domain_name(), typical); - - const auto includes_dot = - std::vector<uint8_t>{11, 'o', 'p', 'e', 'n', '.', 's', 'c', 'r', - 'e', 'e', 'n', 3, 'o', 'r', 'g', 0}; - result = UnpackErrorOr(FromLabels({"open.screen", "org"})); - EXPECT_EQ(result.domain_name(), includes_dot); - - const auto includes_non_ascii = - std::vector<uint8_t>{11, 'o', 'p', 'e', 'n', 7, 's', 'c', 'r', - 'e', 'e', 'n', 3, 'o', 'r', 'g', 0}; - result = UnpackErrorOr(FromLabels({"open\7screen", "org"})); - EXPECT_EQ(result.domain_name(), includes_non_ascii); - - ASSERT_FALSE( - FromLabels({"extremely-long-label-that-is-actually-too-long-" - "for-rfc-1034-and-will-not-generate"})); - - ASSERT_FALSE(FromLabels({ - "extremely-long-domain-name-that-is-made-of", - "valid-labels", - "however-overall-it-is-too-long-for-rfc-1034", - "so-it-should-fail-to-generate", - "filler-filler-filler-filler-filler", - "filler-filler-filler-filler-filler", - "filler-filler-filler-filler-filler", - "filler-filler-filler-filler-filler", - })); -} - -TEST(DomainNameTest, Equality) { - DomainName alpha = UnpackErrorOr(FromLabels({"alpha", "openscreen", "org"})); - DomainName beta = UnpackErrorOr(FromLabels({"beta", "openscreen", "org"})); - - const DomainName alpha_copy = alpha; - - EXPECT_TRUE(alpha == alpha); - EXPECT_FALSE(alpha != alpha); - EXPECT_TRUE(alpha == alpha_copy); - EXPECT_FALSE(alpha != alpha_copy); - EXPECT_FALSE(alpha == beta); - EXPECT_TRUE(alpha != beta); -} - -TEST(DomainNameTest, EndsWithLocalDomain) { - DomainName alpha; - EXPECT_FALSE(alpha.EndsWithLocalDomain()); - - alpha = UnpackErrorOr(FromLabels({"alpha", "openscreen", "org"})); - DomainName beta = UnpackErrorOr(FromLabels({"beta", "local"})); - - EXPECT_FALSE(alpha.EndsWithLocalDomain()); - EXPECT_TRUE(beta.EndsWithLocalDomain()); -} - -TEST(DomainNameTest, IsEmpty) { - DomainName alpha; - DomainName beta(std::vector<uint8_t>{0}); - - EXPECT_TRUE(alpha.IsEmpty()); - EXPECT_TRUE(beta.IsEmpty()); - - alpha = UnpackErrorOr(FromLabels({"alpha", "openscreen", "org"})); - EXPECT_FALSE(alpha.IsEmpty()); -} - -TEST(DomainNameTest, Append) { - const auto expected_service_name = - std::vector<uint8_t>{5, 'a', 'l', 'p', 'h', 'a', '\0'}; - const auto expected_service_type_initial = std::vector<uint8_t>{ - 11, '_', 'o', 'p', 'e', 'n', 's', 'c', 'r', 'e', 'e', 'n', '\0'}; - const auto expected_protocol = - std::vector<uint8_t>{5, '_', 'q', 'u', 'i', 'c', '\0'}; - const auto expected_service_type = - std::vector<uint8_t>{11, '_', 'o', 'p', 'e', 'n', 's', 'c', 'r', 'e', - 'e', 'n', 5, '_', 'q', 'u', 'i', 'c', '\0'}; - const auto total_expected = std::vector<uint8_t>{ - 5, 'a', 'l', 'p', 'h', 'a', 11, '_', 'o', 'p', 'e', 'n', 's', - 'c', 'r', 'e', 'e', 'n', 5, '_', 'q', 'u', 'i', 'c', '\0'}; - - DomainName service_name = UnpackErrorOr(FromLabels({"alpha"})); - EXPECT_EQ(service_name.domain_name(), expected_service_name); - - DomainName service_type = UnpackErrorOr(FromLabels({"_openscreen"})); - EXPECT_EQ(service_type.domain_name(), expected_service_type_initial); - - DomainName protocol = UnpackErrorOr(FromLabels({"_quic"})); - EXPECT_EQ(protocol.domain_name(), expected_protocol); - - EXPECT_TRUE(service_type.Append(protocol).ok()); - EXPECT_EQ(service_type.domain_name(), expected_service_type); - - DomainName result = - UnpackErrorOr(DomainName::Append(service_name, service_type)); - EXPECT_EQ(result.domain_name(), total_expected); -} - -TEST(DomainNameTest, GetLabels) { - const auto labels = std::vector<std::string>{"alpha", "beta", "gamma", "org"}; - DomainName domain_name = UnpackErrorOr(FromLabels(labels)); - - const auto actual_labels = domain_name.GetLabels(); - for (size_t i = 0; i < labels.size(); ++i) { - EXPECT_EQ(labels[i], actual_labels[i]); - } -} - -TEST(DomainNameTest, StreamEscaping) { - { - std::stringstream ss; - ss << DomainName(std::vector<uint8_t>{1, 0, 0}); - EXPECT_EQ(ss.str(), "\\x00."); - } - { - std::stringstream ss; - ss << DomainName(std::vector<uint8_t>{1, 1, 0}); - EXPECT_EQ(ss.str(), "\\x01."); - } - { - std::stringstream ss; - ss << DomainName(std::vector<uint8_t>{1, 18, 0}); - EXPECT_EQ(ss.str(), "\\x12."); - } - { - std::stringstream ss; - ss << DomainName(std::vector<uint8_t>{1, 255, 0}); - EXPECT_EQ(ss.str(), "\\xff."); - } -} - -} // namespace osp -} // namespace openscreen diff --git a/osp/impl/discovery/mdns/mdns_demo.cc b/osp/impl/discovery/mdns/mdns_demo.cc deleted file mode 100644 index 1fc1513e..00000000 --- a/osp/impl/discovery/mdns/mdns_demo.cc +++ /dev/null @@ -1,374 +0,0 @@ -// Copyright 2018 The Chromium 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 <signal.h> -#include <unistd.h> - -#include <algorithm> -#include <map> -#include <memory> -#include <vector> - -// TODO(rwkeane): Remove references to platform/impl -#include "osp/impl/discovery/mdns/mdns_responder_adapter_impl.h" -#include "platform/api/network_interface.h" -#include "platform/api/time.h" -#include "platform/base/error.h" -#include "platform/impl/logging.h" -#include "platform/impl/platform_client_posix.h" -#include "platform/impl/task_runner.h" -#include "platform/impl/udp_socket_reader_posix.h" - -// This file contains a demo of our mDNSResponder wrapper code. It can both -// listen for mDNS services and advertise an mDNS service. The command-line -// usage is: -// mdns_demo [service_type] [service_instance_name] -// service_type defaults to '_openscreen._udp' and service_instance_name -// defaults to ''. service_type determines services the program listens for and -// when service_instance_name is not empty, a service of -// 'service_instance_name.service_type' is also advertised. -// -// The program will print a list of discovered services when it receives a USR1 -// or INT signal. The pid is printed at the beginning of the program to -// facilitate this. -// -// There are a few known bugs around the handling of record events, so this -// shouldn't be expected to be a source of truth, nor should it be expected to -// be correct after running for a long time. - -namespace openscreen { -namespace osp { -namespace { - -bool g_done = false; -bool g_dump_services = false; - -struct Service { - explicit Service(DomainName service_instance) - : service_instance(std::move(service_instance)) {} - ~Service() = default; - - DomainName service_instance; - DomainName domain_name; - IPAddress address; - uint16_t port; - std::vector<std::string> txt; -}; - -class DemoSocketClient : public UdpSocket::Client { - public: - explicit DemoSocketClient(MdnsResponderAdapterImpl* mdns) : mdns_(mdns) {} - - void OnError(UdpSocket* socket, Error error) override { - // TODO(crbug.com/openscreen/66): Change to OSP_LOG_FATAL. - OSP_LOG_ERROR << "configuration failed for interface " << error.message(); - OSP_CHECK(false); - } - - void OnSendError(UdpSocket* socket, Error error) override { - OSP_UNIMPLEMENTED(); - } - - void OnRead(UdpSocket* socket, ErrorOr<UdpPacket> packet) override { - mdns_->OnRead(socket, std::move(packet)); - } - - private: - MdnsResponderAdapterImpl* mdns_; -}; - -using ServiceMap = std::map<DomainName, Service, DomainNameComparator>; -ServiceMap* g_services = nullptr; - -void sigusr1_dump_services(int) { - g_dump_services = true; -} - -void sigint_stop(int) { - OSP_LOG_INFO << "caught SIGINT, exiting..."; - g_done = true; -} - -std::vector<std::string> SplitByDot(const std::string& domain_part) { - std::vector<std::string> result; - auto copy_it = domain_part.begin(); - for (auto it = domain_part.begin(); it != domain_part.end(); ++it) { - if (*it == '.') { - result.emplace_back(copy_it, it); - copy_it = it + 1; - } - } - if (copy_it != domain_part.end()) - result.emplace_back(copy_it, domain_part.end()); - - return result; -} - -void SignalThings() { - struct sigaction usr1_sa; - struct sigaction int_sa; - struct sigaction unused; - - usr1_sa.sa_handler = &sigusr1_dump_services; - sigemptyset(&usr1_sa.sa_mask); - usr1_sa.sa_flags = 0; - - int_sa.sa_handler = &sigint_stop; - sigemptyset(&int_sa.sa_mask); - int_sa.sa_flags = 0; - - sigaction(SIGUSR1, &usr1_sa, &unused); - sigaction(SIGINT, &int_sa, &unused); - - OSP_LOG_INFO << "signal handlers setup" << std::endl << "pid: " << getpid(); -} - -std::vector<std::unique_ptr<UdpSocket>> SetUpMulticastSockets( - TaskRunner* task_runner, - const std::vector<NetworkInterfaceIndex>& index_list, - UdpSocket::Client* client) { - std::vector<std::unique_ptr<UdpSocket>> sockets; - for (const auto ifindex : index_list) { - auto create_result = - UdpSocket::Create(task_runner, client, IPEndpoint{{}, 5353}); - if (!create_result) { - OSP_LOG_ERROR << "failed to create IPv4 socket for interface " << ifindex - << ": " << create_result.error().message(); - continue; - } - std::unique_ptr<UdpSocket> socket = std::move(create_result.value()); - - socket->JoinMulticastGroup(IPAddress{224, 0, 0, 251}, ifindex); - socket->SetMulticastOutboundInterface(ifindex); - socket->Bind(); - - OSP_LOG_INFO << "listening on interface " << ifindex; - sockets.emplace_back(std::move(socket)); - } - return sockets; -} - -void LogService(const Service& s) { - OSP_LOG_INFO << "PTR: (" << s.service_instance << ")" << std::endl - << "SRV: " << s.domain_name << ":" << s.port << std::endl - << "TXT:"; - - for (const auto& l : s.txt) { - OSP_LOG_INFO << " | " << l; - } - OSP_LOG_INFO << "A: " << s.address; -} - -void HandleEvents(MdnsResponderAdapterImpl* mdns_adapter) { - for (auto& ptr_event : mdns_adapter->TakePtrResponses()) { - auto it = g_services->find(ptr_event.service_instance); - switch (ptr_event.header.response_type) { - case QueryEventHeader::Type::kAdded: - case QueryEventHeader::Type::kAddedNoCache: - mdns_adapter->StartSrvQuery(ptr_event.header.socket, - ptr_event.service_instance); - mdns_adapter->StartTxtQuery(ptr_event.header.socket, - ptr_event.service_instance); - if (it == g_services->end()) { - g_services->emplace(ptr_event.service_instance, - Service(ptr_event.service_instance)); - } - break; - case QueryEventHeader::Type::kRemoved: - // PTR may be removed and added without updating related entries (SRV - // and friends) so this simple logic is actually broken, but I don't - // want to do a better design or pointer hell for just a demo. - OSP_LOG_WARN << "ptr-remove: " << ptr_event.service_instance; - if (it != g_services->end()) - g_services->erase(it); - - break; - } - } - for (auto& srv_event : mdns_adapter->TakeSrvResponses()) { - auto it = g_services->find(srv_event.service_instance); - if (it == g_services->end()) - continue; - - switch (srv_event.header.response_type) { - case QueryEventHeader::Type::kAdded: - case QueryEventHeader::Type::kAddedNoCache: - mdns_adapter->StartAQuery(srv_event.header.socket, - srv_event.domain_name); - it->second.domain_name = std::move(srv_event.domain_name); - it->second.port = srv_event.port; - break; - case QueryEventHeader::Type::kRemoved: - OSP_LOG_WARN << "srv-remove: " << srv_event.service_instance; - it->second.domain_name = DomainName(); - it->second.port = 0; - break; - } - } - for (auto& txt_event : mdns_adapter->TakeTxtResponses()) { - auto it = g_services->find(txt_event.service_instance); - if (it == g_services->end()) - continue; - - switch (txt_event.header.response_type) { - case QueryEventHeader::Type::kAdded: - case QueryEventHeader::Type::kAddedNoCache: - it->second.txt = std::move(txt_event.txt_info); - break; - case QueryEventHeader::Type::kRemoved: - OSP_LOG_WARN << "txt-remove: " << txt_event.service_instance; - it->second.txt.clear(); - break; - } - } - for (const auto& a_event : mdns_adapter->TakeAResponses()) { - // TODO(btolsch): If multiple SRV records specify the same domain, the A - // will only update the first. I didn't think this would happen but I - // noticed this happens for cast groups. - auto it = std::find_if(g_services->begin(), g_services->end(), - [&a_event](const std::pair<DomainName, Service>& s) { - return s.second.domain_name == a_event.domain_name; - }); - if (it == g_services->end()) - continue; - - switch (a_event.header.response_type) { - case QueryEventHeader::Type::kAdded: - case QueryEventHeader::Type::kAddedNoCache: - it->second.address = a_event.address; - break; - case QueryEventHeader::Type::kRemoved: - OSP_LOG_WARN << "a-remove: " << a_event.domain_name; - it->second.address = IPAddress(0, 0, 0, 0); - break; - } - } -} - -void BrowseDemo(TaskRunner* task_runner, - const std::string& service_name, - const std::string& service_protocol, - const std::string& service_instance) { - SignalThings(); - - std::vector<std::string> labels{service_name, service_protocol}; - ErrorOr<DomainName> service_type = - DomainName::FromLabels(labels.begin(), labels.end()); - - if (!service_type) { - OSP_LOG_ERROR << "bad domain labels: " << service_name << ", " - << service_protocol; - return; - } - - auto mdns_adapter = std::make_unique<MdnsResponderAdapterImpl>(); - mdns_adapter->Init(); - mdns_adapter->SetHostLabel("gigliorononomicon"); - const std::vector<InterfaceInfo> interfaces = GetNetworkInterfaces(); - std::vector<NetworkInterfaceIndex> index_list; - for (const auto& interface : interfaces) { - OSP_LOG_INFO << "Found interface: " << interface; - if (!interface.addresses.empty()) { - index_list.push_back(interface.index); - } - } - OSP_LOG_IF(WARN, index_list.empty()) - << "No network interfaces had usable addresses for mDNS."; - - DemoSocketClient client(mdns_adapter.get()); - auto sockets = SetUpMulticastSockets(task_runner, index_list, &client); - // The code below assumes the elements in |sockets| is in exact 1:1 - // correspondence with the elements in |index_list|. Crash the demo if any - // sockets are missing (i.e., failed to be set up). - OSP_CHECK_EQ(sockets.size(), index_list.size()); - - // Listen on all interfaces. - auto socket_it = sockets.begin(); - for (NetworkInterfaceIndex index : index_list) { - const auto& interface = - *std::find_if(interfaces.begin(), interfaces.end(), - [index](const openscreen::InterfaceInfo& info) { - return info.index == index; - }); - // Pick any address for the given interface. - mdns_adapter->RegisterInterface(interface, interface.addresses.front(), - socket_it->get()); - ++socket_it; - } - - if (!service_instance.empty()) { - mdns_adapter->RegisterService(service_instance, service_name, - service_protocol, DomainName(), 12345, - {{"k1", "yurtle"}, {"k2", "turtle"}}); - } - - for (const std::unique_ptr<UdpSocket>& socket : sockets) { - mdns_adapter->StartPtrQuery(socket.get(), service_type.value()); - } - - while (!g_done) { - HandleEvents(mdns_adapter.get()); - if (g_dump_services) { - OSP_LOG_INFO << "num services: " << g_services->size(); - for (const auto& s : *g_services) { - LogService(s.second); - } - if (!service_instance.empty()) { - mdns_adapter->UpdateTxtData( - service_instance, service_name, service_protocol, - {{"k1", "oogley"}, {"k2", "moogley"}, {"k3", "googley"}}); - } - g_dump_services = false; - } - mdns_adapter->RunTasks(); - } - OSP_LOG_INFO << "num services: " << g_services->size(); - for (const auto& s : *g_services) { - LogService(s.second); - } - for (const std::unique_ptr<UdpSocket>& socket : sockets) { - mdns_adapter->DeregisterInterface(socket.get()); - } - mdns_adapter->Close(); -} - -} // namespace -} // namespace osp -} // namespace openscreen - -int main(int argc, char** argv) { - using openscreen::Clock; - using openscreen::PlatformClientPosix; - - openscreen::SetLogLevel(openscreen::LogLevel::kVerbose); - - std::string service_instance; - std::string service_type("_openscreen._udp"); - if (argc >= 2) - service_type = argv[1]; - - if (argc >= 3) - service_instance = argv[2]; - - if (service_type.size() && service_type[0] == '.') - return 1; - - auto labels = openscreen::osp::SplitByDot(service_type); - if (labels.size() != 2) - return 1; - - openscreen::osp::ServiceMap services; - openscreen::osp::g_services = &services; - - PlatformClientPosix::Create(std::chrono::milliseconds(50)); - - openscreen::osp::BrowseDemo( - PlatformClientPosix::GetInstance()->GetTaskRunner(), labels[0], labels[1], - service_instance); - - PlatformClientPosix::ShutDown(); - - openscreen::osp::g_services = nullptr; - return 0; -} diff --git a/osp/impl/discovery/mdns/mdns_responder_adapter.cc b/osp/impl/discovery/mdns/mdns_responder_adapter.cc deleted file mode 100644 index edf25503..00000000 --- a/osp/impl/discovery/mdns/mdns_responder_adapter.cc +++ /dev/null @@ -1,77 +0,0 @@ -// Copyright 2018 The Chromium 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 "osp/impl/discovery/mdns/mdns_responder_adapter.h" - -namespace openscreen { -namespace osp { - -QueryEventHeader::QueryEventHeader() = default; -QueryEventHeader::QueryEventHeader(QueryEventHeader::Type response_type, - UdpSocket* socket) - : response_type(response_type), socket(socket) {} -QueryEventHeader::QueryEventHeader(QueryEventHeader&&) noexcept = default; -QueryEventHeader::~QueryEventHeader() = default; -QueryEventHeader& QueryEventHeader::operator=(QueryEventHeader&&) noexcept = - default; - -AEvent::AEvent() = default; -AEvent::AEvent(QueryEventHeader header, - DomainName domain_name, - IPAddress address) - : header(std::move(header)), - domain_name(std::move(domain_name)), - address(std::move(address)) {} -AEvent::AEvent(AEvent&&) noexcept = default; -AEvent::~AEvent() = default; -AEvent& AEvent::operator=(AEvent&&) noexcept = default; - -AaaaEvent::AaaaEvent() = default; -AaaaEvent::AaaaEvent(QueryEventHeader header, - DomainName domain_name, - IPAddress address) - : header(std::move(header)), - domain_name(std::move(domain_name)), - address(std::move(address)) {} -AaaaEvent::AaaaEvent(AaaaEvent&&) noexcept = default; -AaaaEvent::~AaaaEvent() = default; -AaaaEvent& AaaaEvent::operator=(AaaaEvent&&) noexcept = default; - -PtrEvent::PtrEvent() = default; -PtrEvent::PtrEvent(QueryEventHeader header, DomainName service_instance) - : header(std::move(header)), - service_instance(std::move(service_instance)) {} -PtrEvent::PtrEvent(PtrEvent&&) noexcept = default; -PtrEvent::~PtrEvent() = default; -PtrEvent& PtrEvent::operator=(PtrEvent&&) noexcept = default; - -SrvEvent::SrvEvent() = default; -SrvEvent::SrvEvent(QueryEventHeader header, - DomainName service_instance, - DomainName domain_name, - uint16_t port) - : header(std::move(header)), - service_instance(std::move(service_instance)), - domain_name(std::move(domain_name)), - port(port) {} -SrvEvent::SrvEvent(SrvEvent&&) noexcept = default; -SrvEvent::~SrvEvent() = default; -SrvEvent& SrvEvent::operator=(SrvEvent&&) noexcept = default; - -TxtEvent::TxtEvent() = default; -TxtEvent::TxtEvent(QueryEventHeader header, - DomainName service_instance, - std::vector<std::string> txt_info) - : header(std::move(header)), - service_instance(std::move(service_instance)), - txt_info(std::move(txt_info)) {} -TxtEvent::TxtEvent(TxtEvent&&) noexcept = default; -TxtEvent::~TxtEvent() = default; -TxtEvent& TxtEvent::operator=(TxtEvent&&) noexcept = default; - -MdnsResponderAdapter::MdnsResponderAdapter() = default; -MdnsResponderAdapter::~MdnsResponderAdapter() = default; - -} // namespace osp -} // namespace openscreen diff --git a/osp/impl/discovery/mdns/mdns_responder_adapter.h b/osp/impl/discovery/mdns/mdns_responder_adapter.h deleted file mode 100644 index 66083d57..00000000 --- a/osp/impl/discovery/mdns/mdns_responder_adapter.h +++ /dev/null @@ -1,258 +0,0 @@ -// Copyright 2018 The Chromium Authors. All rights reserved. -// Use of this source code is governed by a BSD-style license that can be -// found in the LICENSE file. - -#ifndef OSP_IMPL_DISCOVERY_MDNS_MDNS_RESPONDER_ADAPTER_H_ -#define OSP_IMPL_DISCOVERY_MDNS_MDNS_RESPONDER_ADAPTER_H_ - -#include <cstdint> -#include <map> -#include <string> -#include <vector> - -#include "osp/impl/discovery/mdns/domain_name.h" -#include "osp/impl/discovery/mdns/mdns_responder_platform.h" -#include "platform/api/network_interface.h" -#include "platform/api/time.h" -#include "platform/api/udp_socket.h" -#include "platform/base/error.h" -#include "platform/base/ip_address.h" - -namespace openscreen { -namespace osp { - -struct QueryEventHeader { - enum class Type { - kAdded = 0, - kAddedNoCache, - kRemoved, - }; - - QueryEventHeader(); - QueryEventHeader(Type response_type, UdpSocket* socket); - QueryEventHeader(QueryEventHeader&&) noexcept; - ~QueryEventHeader(); - QueryEventHeader& operator=(QueryEventHeader&&) noexcept; - - Type response_type; - UdpSocket* socket; -}; - -struct PtrEvent { - PtrEvent(); - PtrEvent(QueryEventHeader header, DomainName service_instance); - PtrEvent(PtrEvent&&) noexcept; - ~PtrEvent(); - PtrEvent& operator=(PtrEvent&&) noexcept; - - QueryEventHeader header; - DomainName service_instance; -}; - -struct SrvEvent { - SrvEvent(); - SrvEvent(QueryEventHeader header, - DomainName service_instance, - DomainName domain_name, - uint16_t port); - SrvEvent(SrvEvent&&) noexcept; - ~SrvEvent(); - SrvEvent& operator=(SrvEvent&&) noexcept; - - QueryEventHeader header; - DomainName service_instance; - DomainName domain_name; - uint16_t port; -}; - -struct TxtEvent { - TxtEvent(); - TxtEvent(QueryEventHeader header, - DomainName service_instance, - std::vector<std::string> txt_info); - TxtEvent(TxtEvent&&) noexcept; - ~TxtEvent(); - TxtEvent& operator=(TxtEvent&&) noexcept; - - QueryEventHeader header; - DomainName service_instance; - - // NOTE: mDNS does not specify a character encoding for the data in TXT - // records. - std::vector<std::string> txt_info; -}; - -struct AEvent { - AEvent(); - AEvent(QueryEventHeader header, DomainName domain_name, IPAddress address); - AEvent(AEvent&&) noexcept; - ~AEvent(); - AEvent& operator=(AEvent&&) noexcept; - - QueryEventHeader header; - DomainName domain_name; - IPAddress address; -}; - -struct AaaaEvent { - AaaaEvent(); - AaaaEvent(QueryEventHeader header, DomainName domain_name, IPAddress address); - AaaaEvent(AaaaEvent&&) noexcept; - ~AaaaEvent(); - AaaaEvent& operator=(AaaaEvent&&) noexcept; - - QueryEventHeader header; - DomainName domain_name; - IPAddress address; -}; - -enum class MdnsResponderErrorCode { - kNoError = 0, - kUnsupportedError, - kDomainOverflowError, - kInvalidParameters, - kUnknownError, -}; - -// This interface wraps all the functionality of mDNSResponder, which includes -// both listening and publishing. As a result, some methods are only used by -// listeners, some are only used by publishers, and some are used by both. -// -// Listening for records might look like this: -// adapter->Init(); -// -// // Once for each interface, the meaning of false is described below. -// adapter->RegisterInterface(..., false); -// -// adapter->StartPtrQuery("_openscreen._udp"); -// adapter->RunTasks(); -// -// // When receiving multicast UDP traffic from port 5353. -// adapter->OnDataReceived(...); -// adapter->RunTasks(); -// -// // Check |ptrs| for responses after pulling. -// auto ptrs = adapter->TakePtrResponses(); -// -// // Eventually... -// adapter->StopPtrQuery("_openscreen._udp"); -// -// Publishing a service might look like this: -// adapter->Init(); -// -// // Once for each interface, the meaning of true is described below. -// adapter->RegisterInterface(..., true); -// -// adapter->SetHostLabel("deadbeef"); -// adapter->RegisterService("living-room", "_openscreen._udp", ...); -// adapter->RunTasks(); -// -// // When receiving multicast UDP traffic from port 5353. -// adapter->OnDataReceived(...); -// adapter->RunTasks(); -// -// // Eventually... -// adapter->DeregisterService("living-room", "_openscreen", "_udp"); -// -// Additionally, it's important to understand that mDNSResponder may defer some -// tasks (e.g. parsing responses, sending queries, etc.) and those deferred -// tasks are only run when RunTasks is called. Therefore, RunTasks should be -// called after any sequence of calls to mDNSResponder. It also returns a -// timeout value, after which it must be called again (e.g. for maintaining its -// cache). -class MdnsResponderAdapter : public UdpSocket::Client { - public: - MdnsResponderAdapter(); - virtual ~MdnsResponderAdapter() = 0; - - // Initializes mDNSResponder. This should be called before any queries or - // service registrations are made. - virtual Error Init() = 0; - - // Stops all open queries and service registrations. If this is not called - // before destruction, any registered services will not send their goodbye - // messages. - virtual void Close() = 0; - - // Called to change the name published by the A and AAAA records for the host - // when any service is active (via RegisterService). Returns true if the - // label was set successfully, false otherwise (e.g. the label did not meet - // DNS name requirements). - virtual Error SetHostLabel(const std::string& host_label) = 0; - - // The following methods register and deregister a network interface with - // mDNSResponder. |socket| will be used to identify which interface received - // the data in OnDataReceived and will be used to send data via the platform - // layer. - virtual Error RegisterInterface(const InterfaceInfo& interface_info, - const IPSubnet& interface_address, - UdpSocket* socket) = 0; - virtual Error DeregisterInterface(UdpSocket* socket) = 0; - - // Returns the time period after which this method must be called again, if - // any. - virtual Clock::duration RunTasks() = 0; - - virtual std::vector<PtrEvent> TakePtrResponses() = 0; - virtual std::vector<SrvEvent> TakeSrvResponses() = 0; - virtual std::vector<TxtEvent> TakeTxtResponses() = 0; - virtual std::vector<AEvent> TakeAResponses() = 0; - virtual std::vector<AaaaEvent> TakeAaaaResponses() = 0; - - virtual MdnsResponderErrorCode StartPtrQuery( - UdpSocket* socket, - const DomainName& service_type) = 0; - virtual MdnsResponderErrorCode StartSrvQuery( - UdpSocket* socket, - const DomainName& service_instance) = 0; - virtual MdnsResponderErrorCode StartTxtQuery( - UdpSocket* socket, - const DomainName& service_instance) = 0; - virtual MdnsResponderErrorCode StartAQuery(UdpSocket* socket, - const DomainName& domain_name) = 0; - virtual MdnsResponderErrorCode StartAaaaQuery( - UdpSocket* socket, - const DomainName& domain_name) = 0; - - virtual MdnsResponderErrorCode StopPtrQuery( - UdpSocket* socket, - const DomainName& service_type) = 0; - virtual MdnsResponderErrorCode StopSrvQuery( - UdpSocket* socket, - const DomainName& service_instance) = 0; - virtual MdnsResponderErrorCode StopTxtQuery( - UdpSocket* socket, - const DomainName& service_instance) = 0; - virtual MdnsResponderErrorCode StopAQuery(UdpSocket* socket, - const DomainName& domain_name) = 0; - virtual MdnsResponderErrorCode StopAaaaQuery( - UdpSocket* socket, - const DomainName& domain_name) = 0; - - // The following methods concern advertising a service via mDNS. The - // arguments correspond to values needed in the PTR, SRV, and TXT records that - // will be published for the service. An A or AAAA record will also be - // published with the service for each active interface known to mDNSResponder - // via RegisterInterface. - virtual MdnsResponderErrorCode RegisterService( - const std::string& service_instance, - const std::string& service_name, - const std::string& service_protocol, - const DomainName& target_host, - uint16_t target_port, - const std::map<std::string, std::string>& txt_data) = 0; - virtual MdnsResponderErrorCode DeregisterService( - const std::string& service_instance, - const std::string& service_name, - const std::string& service_protocol) = 0; - virtual MdnsResponderErrorCode UpdateTxtData( - const std::string& service_instance, - const std::string& service_name, - const std::string& service_protocol, - const std::map<std::string, std::string>& txt_data) = 0; -}; - -} // namespace osp -} // namespace openscreen - -#endif // OSP_IMPL_DISCOVERY_MDNS_MDNS_RESPONDER_ADAPTER_H_ diff --git a/osp/impl/discovery/mdns/mdns_responder_adapter_impl.cc b/osp/impl/discovery/mdns/mdns_responder_adapter_impl.cc deleted file mode 100644 index 205e125b..00000000 --- a/osp/impl/discovery/mdns/mdns_responder_adapter_impl.cc +++ /dev/null @@ -1,1044 +0,0 @@ -// Copyright 2018 The Chromium 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 "osp/impl/discovery/mdns/mdns_responder_adapter_impl.h" - -#include <algorithm> -#include <cctype> -#include <cstring> -#include <iostream> -#include <memory> -#include <string> -#include <utility> - -#include "util/osp_logging.h" -#include "util/trace_logging.h" - -namespace openscreen { -namespace osp { -namespace { - -// RFC 1035 specifies a max string length of 256, including the leading length -// octet. -constexpr size_t kMaxDnsStringLength = 255; - -// RFC 6763 recommends a maximum key length of 9 characters. -constexpr size_t kMaxTxtKeyLength = 9; - -constexpr size_t kMaxStaticTxtDataSize = 256; - -static_assert(sizeof(std::declval<RData>().u.txt) == kMaxStaticTxtDataSize, - "mDNSResponder static TXT data size expected to be 256 bytes"); - -static_assert(sizeof(mDNSAddr::ip.v4.b) == 4u, - "mDNSResponder IPv4 address must be 4 bytes"); -static_assert(sizeof(mDNSAddr::ip.v6.b) == 16u, - "mDNSResponder IPv6 address must be 16 bytes"); - -void AssignMdnsPort(mDNSIPPort* mdns_port, uint16_t port) { - mdns_port->b[0] = (port >> 8) & 0xff; - mdns_port->b[1] = port & 0xff; -} - -uint16_t GetNetworkOrderPort(const mDNSOpaque16& port) { - return port.b[0] << 8 | port.b[1]; -} - -bool IsValidServiceName(const std::string& service_name) { - // Service name requirements come from RFC 6335: - // - No more than 16 characters. - // - Begin with '_'. - // - Next is a letter or digit and end with a letter or digit. - // - May contain hyphens, but no consecutive hyphens. - // - Must contain at least one letter. - if (service_name.size() <= 1 || service_name.size() > 16) - return false; - - if (service_name[0] != '_' || !std::isalnum(service_name[1]) || - !std::isalnum(service_name.back())) { - return false; - } - bool has_alpha = false; - bool previous_hyphen = false; - for (auto it = service_name.begin() + 1; it != service_name.end(); ++it) { - if (*it == '-' && previous_hyphen) - return false; - - previous_hyphen = *it == '-'; - has_alpha = has_alpha || std::isalpha(*it); - } - return has_alpha && !previous_hyphen; -} - -bool IsValidServiceProtocol(const std::string& protocol) { - // RFC 6763 requires _tcp be used for TCP services and _udp for all others. - return protocol == "_tcp" || protocol == "_udp"; -} - -void MakeLocalServiceNameParts(const std::string& service_instance, - const std::string& service_name, - const std::string& service_protocol, - domainlabel* instance, - domainlabel* name, - domainlabel* protocol, - domainname* type, - domainname* domain) { - MakeDomainLabelFromLiteralString(instance, service_instance.c_str()); - MakeDomainLabelFromLiteralString(name, service_name.c_str()); - MakeDomainLabelFromLiteralString(protocol, service_protocol.c_str()); - type->c[0] = 0; - AppendDomainLabel(type, name); - AppendDomainLabel(type, protocol); - const DomainName local_domain = DomainName::GetLocalDomain(); - std::copy(local_domain.domain_name().begin(), - local_domain.domain_name().end(), domain->c); -} - -void MakeSubnetMaskFromPrefixLengthV4(uint8_t mask[4], uint8_t prefix_length) { - for (int i = 0; i < 4; prefix_length -= 8, ++i) { - if (prefix_length >= 8) { - mask[i] = 0xff; - } else if (prefix_length > 0) { - mask[i] = 0xff << (8 - prefix_length); - } else { - mask[i] = 0; - } - } -} - -void MakeSubnetMaskFromPrefixLengthV6(uint8_t mask[16], uint8_t prefix_length) { - for (int i = 0; i < 16; prefix_length -= 8, ++i) { - if (prefix_length >= 8) { - mask[i] = 0xff; - } else if (prefix_length > 0) { - mask[i] = 0xff << (8 - prefix_length); - } else { - mask[i] = 0; - } - } -} - -bool IsValidTxtDataKey(const std::string& s) { - if (s.size() > kMaxTxtKeyLength) - return false; - for (unsigned char c : s) - if (c < 0x20 || c > 0x7e || c == '=') - return false; - return true; -} - -std::string MakeTxtData(const std::map<std::string, std::string>& txt_data) { - std::string txt; - txt.reserve(kMaxStaticTxtDataSize); - for (const auto& line : txt_data) { - const auto key_size = line.first.size(); - const auto value_size = line.second.size(); - const auto line_size = value_size ? (key_size + 1 + value_size) : key_size; - if (!IsValidTxtDataKey(line.first) || line_size > kMaxDnsStringLength || - (txt.size() + 1 + line_size) > kMaxStaticTxtDataSize) { - return {}; - } - txt.push_back(line_size); - txt += line.first; - if (value_size) { - txt.push_back('='); - txt += line.second; - } - } - return txt; -} - -MdnsResponderErrorCode MapMdnsError(int err) { - switch (err) { - case mStatus_NoError: - return MdnsResponderErrorCode::kNoError; - case mStatus_UnsupportedErr: - return MdnsResponderErrorCode::kUnsupportedError; - case mStatus_UnknownErr: - return MdnsResponderErrorCode::kUnknownError; - default: - break; - } - OSP_DLOG_WARN << "unmapped mDNSResponder error: " << err; - return MdnsResponderErrorCode::kUnknownError; -} - -std::vector<std::string> ParseTxtResponse( - const uint8_t data[kMaxStaticTxtDataSize], - uint16_t length) { - OSP_DCHECK(length <= kMaxStaticTxtDataSize); - if (length == 0) - return {}; - - std::vector<std::string> lines; - int total_pos = 0; - while (total_pos < length) { - uint8_t line_length = data[total_pos]; - if ((line_length > kMaxDnsStringLength) || - (total_pos + line_length >= length)) { - return {}; - } - lines.emplace_back(&data[total_pos + 1], - &data[total_pos + line_length + 1]); - total_pos += line_length + 1; - } - return lines; -} - -void MdnsStatusCallback(mDNS* mdns, mStatus result) { - OSP_LOG_INFO << "status good? " << (result == mStatus_NoError); -} - -} // namespace - -MdnsResponderAdapterImpl::MdnsResponderAdapterImpl() = default; -MdnsResponderAdapterImpl::~MdnsResponderAdapterImpl() = default; - -Error MdnsResponderAdapterImpl::Init() { - const auto err = - mDNS_Init(&mdns_, &platform_storage_, rr_cache_, kRrCacheSize, - mDNS_Init_DontAdvertiseLocalAddresses, &MdnsStatusCallback, - mDNS_Init_NoInitCallbackContext); - - return (err == mStatus_NoError) ? Error::None() - : Error::Code::kInitializationFailure; -} - -void MdnsResponderAdapterImpl::Close() { - TRACE_SCOPED(TraceCategory::kMdns, "MdnsResponderAdapterImpl::Close"); - mDNS_StartExit(&mdns_); - // Let all services send goodbyes. - while (!service_records_.empty()) { - RunTasks(); - } - mDNS_FinalExit(&mdns_); - - socket_to_questions_.clear(); - - responder_interface_info_.clear(); - - a_responses_.clear(); - aaaa_responses_.clear(); - ptr_responses_.clear(); - srv_responses_.clear(); - txt_responses_.clear(); - - service_records_.clear(); -} - -Error MdnsResponderAdapterImpl::SetHostLabel(const std::string& host_label) { - TRACE_SCOPED(TraceCategory::kMdns, "MdnsResponderAdapterImpl::SetHostLabel"); - if (host_label.size() > DomainName::kDomainNameMaxLabelLength) - return Error::Code::kDomainNameTooLong; - - MakeDomainLabelFromLiteralString(&mdns_.hostlabel, host_label.c_str()); - mDNS_SetFQDN(&mdns_); - if (!service_records_.empty()) { - DeadvertiseInterfaces(); - AdvertiseInterfaces(); - } - return Error::None(); -} - -Error MdnsResponderAdapterImpl::RegisterInterface( - const InterfaceInfo& interface_info, - const IPSubnet& interface_address, - UdpSocket* socket) { - TRACE_SCOPED(TraceCategory::kMdns, - "MdnsResponderAdapterImpl::RegisterInterface"); - OSP_DCHECK(socket); - - const auto info_it = responder_interface_info_.find(socket); - if (info_it != responder_interface_info_.end()) - return Error::None(); - - NetworkInterfaceInfo& info = responder_interface_info_[socket]; - std::memset(&info, 0, sizeof(NetworkInterfaceInfo)); - info.InterfaceID = reinterpret_cast<decltype(info.InterfaceID)>(socket); - info.Advertise = mDNSfalse; - if (interface_address.address.IsV4()) { - info.ip.type = mDNSAddrType_IPv4; - interface_address.address.CopyToV4(info.ip.ip.v4.b); - info.mask.type = mDNSAddrType_IPv4; - MakeSubnetMaskFromPrefixLengthV4(info.mask.ip.v4.b, - interface_address.prefix_length); - } else { - info.ip.type = mDNSAddrType_IPv6; - interface_address.address.CopyToV6(info.ip.ip.v6.b); - info.mask.type = mDNSAddrType_IPv6; - MakeSubnetMaskFromPrefixLengthV6(info.mask.ip.v6.b, - interface_address.prefix_length); - } - - static_assert(sizeof(info.MAC.b) == sizeof(interface_info.hardware_address), - "MAC address size mismatch."); - memcpy(info.MAC.b, interface_info.hardware_address.data(), - sizeof(info.MAC.b)); - info.McastTxRx = 1; - platform_storage_.sockets.push_back(socket); - auto result = mDNS_RegisterInterface(&mdns_, &info, mDNSfalse); - OSP_LOG_IF(WARN, result != mStatus_NoError) - << "mDNS_RegisterInterface failed: " << result; - - return (result == mStatus_NoError) ? Error::None() - : Error::Code::kMdnsRegisterFailure; -} - -Error MdnsResponderAdapterImpl::DeregisterInterface(UdpSocket* socket) { - TRACE_SCOPED(TraceCategory::kMdns, - "MdnsResponderAdapterImpl::DeregisterInterface"); - const auto info_it = responder_interface_info_.find(socket); - if (info_it == responder_interface_info_.end()) - return Error::Code::kItemNotFound; - - const auto it = std::find(platform_storage_.sockets.begin(), - platform_storage_.sockets.end(), socket); - OSP_DCHECK(it != platform_storage_.sockets.end()); - platform_storage_.sockets.erase(it); - if (info_it->second.RR_A.namestorage.c[0]) { - mDNS_Deregister(&mdns_, &info_it->second.RR_A); - info_it->second.RR_A.namestorage.c[0] = 0; - } - mDNS_DeregisterInterface(&mdns_, &info_it->second, mDNSfalse); - responder_interface_info_.erase(info_it); - return Error::None(); -} -void MdnsResponderAdapterImpl::OnRead(UdpSocket* socket, - ErrorOr<UdpPacket> packet_or_error) { - if (packet_or_error.is_error()) { - return; - } - - UdpPacket packet = std::move(packet_or_error.value()); - TRACE_SCOPED(TraceCategory::kMdns, "MdnsResponderAdapterImpl::OnRead"); - mDNSAddr src; - if (packet.source().address.IsV4()) { - src.type = mDNSAddrType_IPv4; - packet.source().address.CopyToV4(src.ip.v4.b); - } else { - src.type = mDNSAddrType_IPv6; - packet.source().address.CopyToV6(src.ip.v6.b); - } - mDNSIPPort srcport; - AssignMdnsPort(&srcport, packet.source().port); - - mDNSAddr dst; - if (packet.source().address.IsV4()) { - dst.type = mDNSAddrType_IPv4; - packet.destination().address.CopyToV4(dst.ip.v4.b); - } else { - dst.type = mDNSAddrType_IPv6; - packet.destination().address.CopyToV6(dst.ip.v6.b); - } - mDNSIPPort dstport; - AssignMdnsPort(&dstport, packet.destination().port); - - auto* packet_data = packet.data(); - mDNSCoreReceive(&mdns_, const_cast<uint8_t*>(packet_data), - packet_data + packet.size(), &src, srcport, &dst, dstport, - reinterpret_cast<mDNSInterfaceID>(packet.socket())); -} - -void MdnsResponderAdapterImpl::OnSendError(UdpSocket* socket, Error error) { - // TODO(crbug.com/openscreen/67): Implement this method. - OSP_UNIMPLEMENTED(); -} - -void MdnsResponderAdapterImpl::OnError(UdpSocket* socket, Error error) { - // TODO(crbug.com/openscreen/67): Implement this method. - OSP_UNIMPLEMENTED(); -} - -void MdnsResponderAdapterImpl::OnBound(UdpSocket* socket) { - // TODO(crbug.com/openscreen/67): Implement this method. - OSP_UNIMPLEMENTED(); -} - -Clock::duration MdnsResponderAdapterImpl::RunTasks() { - TRACE_SCOPED(TraceCategory::kMdns, "MdnsResponderAdapterImpl::RunTasks"); - - mDNS_Execute(&mdns_); - - // Using mDNS_Execute's response to determine the correct timespan before - // re-running this method doesn't work as expected. In the demo, under some - // cases (about 25% of demo runs), the response is set to an unreasonably - // large number (in the order of multiple days). - // - // From the mDNS documentation: "it is the responsibility [...] to set the - // timer according to the m->NextScheduledEvent value, and then when the timer - // fires, the timer callback function should call mDNS_Execute()" - for more - // details see third_party/mDNSResponder/src/mDNSCore/mDNS.c : 3390 - // - // Together, I understand these to mean that the mdns library code doesn't - // expect we need mDNS_Execute called again by the task runner, only in the - // other special cases it calls out in documentation (which we currently do - // correctly). In our code, when we call mDNS_Execute again outside of the - // task runner, the result is currently discarded. What we would need to do is - // reach into the Task Runner's task and update how long before the task runs - // again. That would require some large refactoring and changes. - // - // Additionally, beyond this, the mDNS code documents that there are cases - // where the return value for mDNS_Execute should be ignored because it may be - // stale. - // - // TODO(rwkeane): More accurately determine when the next run of this method - // should be. - constexpr auto seconds_before_next_run = 1; - - // Return as a duration. - return std::chrono::seconds(seconds_before_next_run); -} - -std::vector<PtrEvent> MdnsResponderAdapterImpl::TakePtrResponses() { - return std::move(ptr_responses_); -} - -std::vector<SrvEvent> MdnsResponderAdapterImpl::TakeSrvResponses() { - return std::move(srv_responses_); -} - -std::vector<TxtEvent> MdnsResponderAdapterImpl::TakeTxtResponses() { - return std::move(txt_responses_); -} - -std::vector<AEvent> MdnsResponderAdapterImpl::TakeAResponses() { - return std::move(a_responses_); -} - -std::vector<AaaaEvent> MdnsResponderAdapterImpl::TakeAaaaResponses() { - return std::move(aaaa_responses_); -} - -MdnsResponderErrorCode MdnsResponderAdapterImpl::StartPtrQuery( - UdpSocket* socket, - const DomainName& service_type) { - TRACE_SCOPED(TraceCategory::kMdns, "MdnsResponderAdapterImpl::StartPtrQuery"); - auto& ptr_questions = socket_to_questions_[socket].ptr; - if (ptr_questions.find(service_type) != ptr_questions.end()) - return MdnsResponderErrorCode::kNoError; - - auto& question = ptr_questions[service_type]; - - question.InterfaceID = reinterpret_cast<mDNSInterfaceID>(socket); - question.Target = {0}; - if (service_type.EndsWithLocalDomain()) { - std::copy(service_type.domain_name().begin(), - service_type.domain_name().end(), question.qname.c); - } else { - const DomainName local_domain = DomainName::GetLocalDomain(); - ErrorOr<DomainName> service_type_with_local = - DomainName::Append(service_type, local_domain); - if (!service_type_with_local) { - return MdnsResponderErrorCode::kDomainOverflowError; - } - std::copy(service_type_with_local.value().domain_name().begin(), - service_type_with_local.value().domain_name().end(), - question.qname.c); - } - question.qtype = kDNSType_PTR; - question.qclass = kDNSClass_IN; - question.LongLived = mDNStrue; - question.ExpectUnique = mDNSfalse; - question.ForceMCast = mDNStrue; - question.ReturnIntermed = mDNSfalse; - question.SuppressUnusable = mDNSfalse; - question.RetryWithSearchDomains = mDNSfalse; - question.TimeoutQuestion = 0; - question.WakeOnResolve = 0; - question.SearchListIndex = 0; - question.AppendSearchDomains = 0; - question.AppendLocalSearchDomains = 0; - question.qnameOrig = nullptr; - question.QuestionCallback = &MdnsResponderAdapterImpl::PtrQueryCallback; - question.QuestionContext = this; - const auto err = mDNS_StartQuery(&mdns_, &question); - OSP_LOG_IF(WARN, err != mStatus_NoError) << "mDNS_StartQuery failed: " << err; - return MapMdnsError(err); -} - -MdnsResponderErrorCode MdnsResponderAdapterImpl::StartSrvQuery( - UdpSocket* socket, - const DomainName& service_instance) { - TRACE_SCOPED(TraceCategory::kMdns, "MdnsResponderAdapterImpl::StartSrvQuery"); - if (!service_instance.EndsWithLocalDomain()) - return MdnsResponderErrorCode::kInvalidParameters; - - auto& srv_questions = socket_to_questions_[socket].srv; - if (srv_questions.find(service_instance) != srv_questions.end()) - return MdnsResponderErrorCode::kNoError; - - auto& question = srv_questions[service_instance]; - - question.InterfaceID = reinterpret_cast<mDNSInterfaceID>(socket); - question.Target = {0}; - std::copy(service_instance.domain_name().begin(), - service_instance.domain_name().end(), question.qname.c); - question.qtype = kDNSType_SRV; - question.qclass = kDNSClass_IN; - question.LongLived = mDNStrue; - question.ExpectUnique = mDNSfalse; - question.ForceMCast = mDNStrue; - question.ReturnIntermed = mDNSfalse; - question.SuppressUnusable = mDNSfalse; - question.RetryWithSearchDomains = mDNSfalse; - question.TimeoutQuestion = 0; - question.WakeOnResolve = 0; - question.SearchListIndex = 0; - question.AppendSearchDomains = 0; - question.AppendLocalSearchDomains = 0; - question.qnameOrig = nullptr; - question.QuestionCallback = &MdnsResponderAdapterImpl::SrvQueryCallback; - question.QuestionContext = this; - const auto err = mDNS_StartQuery(&mdns_, &question); - OSP_LOG_IF(WARN, err != mStatus_NoError) << "mDNS_StartQuery failed: " << err; - return MapMdnsError(err); -} - -MdnsResponderErrorCode MdnsResponderAdapterImpl::StartTxtQuery( - UdpSocket* socket, - const DomainName& service_instance) { - TRACE_SCOPED(TraceCategory::kMdns, "MdnsResponderAdapterImpl::StartTxtQuery"); - if (!service_instance.EndsWithLocalDomain()) - return MdnsResponderErrorCode::kInvalidParameters; - - auto& txt_questions = socket_to_questions_[socket].txt; - if (txt_questions.find(service_instance) != txt_questions.end()) - return MdnsResponderErrorCode::kNoError; - - auto& question = txt_questions[service_instance]; - - question.InterfaceID = reinterpret_cast<mDNSInterfaceID>(socket); - question.Target = {0}; - std::copy(service_instance.domain_name().begin(), - service_instance.domain_name().end(), question.qname.c); - question.qtype = kDNSType_TXT; - question.qclass = kDNSClass_IN; - question.LongLived = mDNStrue; - question.ExpectUnique = mDNSfalse; - question.ForceMCast = mDNStrue; - question.ReturnIntermed = mDNSfalse; - question.SuppressUnusable = mDNSfalse; - question.RetryWithSearchDomains = mDNSfalse; - question.TimeoutQuestion = 0; - question.WakeOnResolve = 0; - question.SearchListIndex = 0; - question.AppendSearchDomains = 0; - question.AppendLocalSearchDomains = 0; - question.qnameOrig = nullptr; - question.QuestionCallback = &MdnsResponderAdapterImpl::TxtQueryCallback; - question.QuestionContext = this; - const auto err = mDNS_StartQuery(&mdns_, &question); - OSP_LOG_IF(WARN, err != mStatus_NoError) << "mDNS_StartQuery failed: " << err; - return MapMdnsError(err); -} - -MdnsResponderErrorCode MdnsResponderAdapterImpl::StartAQuery( - UdpSocket* socket, - const DomainName& domain_name) { - TRACE_SCOPED(TraceCategory::kMdns, "MdnsResponderAdapterImpl::StartAQuery"); - if (!domain_name.EndsWithLocalDomain()) - return MdnsResponderErrorCode::kInvalidParameters; - - auto& a_questions = socket_to_questions_[socket].a; - if (a_questions.find(domain_name) != a_questions.end()) - return MdnsResponderErrorCode::kNoError; - - auto& question = a_questions[domain_name]; - std::copy(domain_name.domain_name().begin(), domain_name.domain_name().end(), - question.qname.c); - - question.InterfaceID = reinterpret_cast<mDNSInterfaceID>(socket); - question.Target = {0}; - question.qtype = kDNSType_A; - question.qclass = kDNSClass_IN; - question.LongLived = mDNStrue; - question.ExpectUnique = mDNSfalse; - question.ForceMCast = mDNStrue; - question.ReturnIntermed = mDNSfalse; - question.SuppressUnusable = mDNSfalse; - question.RetryWithSearchDomains = mDNSfalse; - question.TimeoutQuestion = 0; - question.WakeOnResolve = 0; - question.SearchListIndex = 0; - question.AppendSearchDomains = 0; - question.AppendLocalSearchDomains = 0; - question.qnameOrig = nullptr; - question.QuestionCallback = &MdnsResponderAdapterImpl::AQueryCallback; - question.QuestionContext = this; - const auto err = mDNS_StartQuery(&mdns_, &question); - OSP_LOG_IF(WARN, err != mStatus_NoError) << "mDNS_StartQuery failed: " << err; - return MapMdnsError(err); -} - -MdnsResponderErrorCode MdnsResponderAdapterImpl::StartAaaaQuery( - UdpSocket* socket, - const DomainName& domain_name) { - TRACE_SCOPED(TraceCategory::kMdns, - "MdnsResponderAdapterImpl::StartAaaaQuery"); - if (!domain_name.EndsWithLocalDomain()) - return MdnsResponderErrorCode::kInvalidParameters; - - auto& aaaa_questions = socket_to_questions_[socket].aaaa; - if (aaaa_questions.find(domain_name) != aaaa_questions.end()) - return MdnsResponderErrorCode::kNoError; - - auto& question = aaaa_questions[domain_name]; - std::copy(domain_name.domain_name().begin(), domain_name.domain_name().end(), - question.qname.c); - - question.InterfaceID = reinterpret_cast<mDNSInterfaceID>(socket); - question.Target = {0}; - question.qtype = kDNSType_AAAA; - question.qclass = kDNSClass_IN; - question.LongLived = mDNStrue; - question.ExpectUnique = mDNSfalse; - question.ForceMCast = mDNStrue; - question.ReturnIntermed = mDNSfalse; - question.SuppressUnusable = mDNSfalse; - question.RetryWithSearchDomains = mDNSfalse; - question.TimeoutQuestion = 0; - question.WakeOnResolve = 0; - question.SearchListIndex = 0; - question.AppendSearchDomains = 0; - question.AppendLocalSearchDomains = 0; - question.qnameOrig = nullptr; - question.QuestionCallback = &MdnsResponderAdapterImpl::AaaaQueryCallback; - question.QuestionContext = this; - const auto err = mDNS_StartQuery(&mdns_, &question); - OSP_LOG_IF(WARN, err != mStatus_NoError) << "mDNS_StartQuery failed: " << err; - return MapMdnsError(err); -} - -MdnsResponderErrorCode MdnsResponderAdapterImpl::StopPtrQuery( - UdpSocket* socket, - const DomainName& service_type) { - TRACE_SCOPED(TraceCategory::kMdns, "MdnsResponderAdapterImpl::StopPtrQuery"); - auto interface_entry = socket_to_questions_.find(socket); - if (interface_entry == socket_to_questions_.end()) - return MdnsResponderErrorCode::kNoError; - auto entry = interface_entry->second.ptr.find(service_type); - if (entry == interface_entry->second.ptr.end()) - return MdnsResponderErrorCode::kNoError; - - const auto err = mDNS_StopQuery(&mdns_, &entry->second); - interface_entry->second.ptr.erase(entry); - OSP_LOG_IF(WARN, err != mStatus_NoError) << "mDNS_StopQuery failed: " << err; - RemoveQuestionsIfEmpty(socket); - return MapMdnsError(err); -} - -MdnsResponderErrorCode MdnsResponderAdapterImpl::StopSrvQuery( - UdpSocket* socket, - const DomainName& service_instance) { - TRACE_SCOPED(TraceCategory::kMdns, "MdnsResponderAdapterImpl::StopSrvQuery"); - auto interface_entry = socket_to_questions_.find(socket); - if (interface_entry == socket_to_questions_.end()) - return MdnsResponderErrorCode::kNoError; - auto entry = interface_entry->second.srv.find(service_instance); - if (entry == interface_entry->second.srv.end()) - return MdnsResponderErrorCode::kNoError; - - const auto err = mDNS_StopQuery(&mdns_, &entry->second); - interface_entry->second.srv.erase(entry); - OSP_LOG_IF(WARN, err != mStatus_NoError) << "mDNS_StopQuery failed: " << err; - RemoveQuestionsIfEmpty(socket); - return MapMdnsError(err); -} - -MdnsResponderErrorCode MdnsResponderAdapterImpl::StopTxtQuery( - UdpSocket* socket, - const DomainName& service_instance) { - TRACE_SCOPED(TraceCategory::kMdns, "MdnsResponderAdapterImpl::StopTxtQuery"); - auto interface_entry = socket_to_questions_.find(socket); - if (interface_entry == socket_to_questions_.end()) - return MdnsResponderErrorCode::kNoError; - auto entry = interface_entry->second.txt.find(service_instance); - if (entry == interface_entry->second.txt.end()) - return MdnsResponderErrorCode::kNoError; - - const auto err = mDNS_StopQuery(&mdns_, &entry->second); - interface_entry->second.txt.erase(entry); - OSP_LOG_IF(WARN, err != mStatus_NoError) << "mDNS_StopQuery failed: " << err; - RemoveQuestionsIfEmpty(socket); - return MapMdnsError(err); -} - -MdnsResponderErrorCode MdnsResponderAdapterImpl::StopAQuery( - UdpSocket* socket, - const DomainName& domain_name) { - TRACE_SCOPED(TraceCategory::kMdns, "MdnsResponderAdapterImpl::StopAQuery"); - auto interface_entry = socket_to_questions_.find(socket); - if (interface_entry == socket_to_questions_.end()) - return MdnsResponderErrorCode::kNoError; - auto entry = interface_entry->second.a.find(domain_name); - if (entry == interface_entry->second.a.end()) - return MdnsResponderErrorCode::kNoError; - - const auto err = mDNS_StopQuery(&mdns_, &entry->second); - interface_entry->second.a.erase(entry); - OSP_LOG_IF(WARN, err != mStatus_NoError) << "mDNS_StopQuery failed: " << err; - RemoveQuestionsIfEmpty(socket); - return MapMdnsError(err); -} - -MdnsResponderErrorCode MdnsResponderAdapterImpl::StopAaaaQuery( - UdpSocket* socket, - const DomainName& domain_name) { - TRACE_SCOPED(TraceCategory::kMdns, "MdnsResponderAdapterImpl::StopAaaaQuery"); - auto interface_entry = socket_to_questions_.find(socket); - if (interface_entry == socket_to_questions_.end()) - return MdnsResponderErrorCode::kNoError; - auto entry = interface_entry->second.aaaa.find(domain_name); - if (entry == interface_entry->second.aaaa.end()) - return MdnsResponderErrorCode::kNoError; - - const auto err = mDNS_StopQuery(&mdns_, &entry->second); - interface_entry->second.aaaa.erase(entry); - OSP_LOG_IF(WARN, err != mStatus_NoError) << "mDNS_StopQuery failed: " << err; - RemoveQuestionsIfEmpty(socket); - return MapMdnsError(err); -} - -MdnsResponderErrorCode MdnsResponderAdapterImpl::RegisterService( - const std::string& service_instance, - const std::string& service_name, - const std::string& service_protocol, - const DomainName& target_host, - uint16_t target_port, - const std::map<std::string, std::string>& txt_data) { - TRACE_SCOPED(TraceCategory::kMdns, - "MdnsResponderAdapterImpl::RegisterService"); - OSP_DCHECK(IsValidServiceName(service_name)); - OSP_DCHECK(IsValidServiceProtocol(service_protocol)); - service_records_.push_back(std::make_unique<ServiceRecordSet>()); - auto* service_record = service_records_.back().get(); - domainlabel instance; - domainlabel name; - domainlabel protocol; - domainname type; - domainname domain; - domainname host; - mDNSIPPort port; - - MakeLocalServiceNameParts(service_instance, service_name, service_protocol, - &instance, &name, &protocol, &type, &domain); - std::copy(target_host.domain_name().begin(), target_host.domain_name().end(), - host.c); - AssignMdnsPort(&port, target_port); - auto txt = MakeTxtData(txt_data); - if (txt.size() > kMaxStaticTxtDataSize) { - // Not handling oversized TXT records. - return MdnsResponderErrorCode::kUnsupportedError; - } - - if (service_records_.size() == 1) - AdvertiseInterfaces(); - - auto result = mDNS_RegisterService( - &mdns_, service_record, &instance, &type, &domain, &host, port, - reinterpret_cast<const uint8_t*>(txt.data()), txt.size(), nullptr, 0, - mDNSInterface_Any, &MdnsResponderAdapterImpl::ServiceCallback, this, 0); - - if (result != mStatus_NoError) { - service_records_.pop_back(); - if (service_records_.empty()) - DeadvertiseInterfaces(); - } - return MapMdnsError(result); -} - -MdnsResponderErrorCode MdnsResponderAdapterImpl::DeregisterService( - const std::string& service_instance, - const std::string& service_name, - const std::string& service_protocol) { - TRACE_SCOPED(TraceCategory::kMdns, - "MdnsResponderAdapterImpl::DeregisterService"); - domainlabel instance; - domainlabel name; - domainlabel protocol; - domainname type; - domainname domain; - domainname full_instance_name; - - MakeLocalServiceNameParts(service_instance, service_name, service_protocol, - &instance, &name, &protocol, &type, &domain); - if (!ConstructServiceName(&full_instance_name, &instance, &type, &domain)) - return MdnsResponderErrorCode::kInvalidParameters; - - for (auto it = service_records_.begin(); it != service_records_.end(); ++it) { - if (SameDomainName(&full_instance_name, &(*it)->RR_SRV.namestorage)) { - // |it| will be removed from |service_records_| in ServiceCallback, when - // mDNSResponder is done with the memory. - mDNS_DeregisterService(&mdns_, it->get()); - return MdnsResponderErrorCode::kNoError; - } - } - return MdnsResponderErrorCode::kNoError; -} - -MdnsResponderErrorCode MdnsResponderAdapterImpl::UpdateTxtData( - const std::string& service_instance, - const std::string& service_name, - const std::string& service_protocol, - const std::map<std::string, std::string>& txt_data) { - TRACE_SCOPED(TraceCategory::kMdns, "MdnsResponderAdapterImpl::UpdateTxtData"); - domainlabel instance; - domainlabel name; - domainlabel protocol; - domainname type; - domainname domain; - domainname full_instance_name; - - MakeLocalServiceNameParts(service_instance, service_name, service_protocol, - &instance, &name, &protocol, &type, &domain); - if (!ConstructServiceName(&full_instance_name, &instance, &type, &domain)) - return MdnsResponderErrorCode::kInvalidParameters; - std::string txt = MakeTxtData(txt_data); - if (txt.size() > kMaxStaticTxtDataSize) { - // Not handling oversized TXT records. - return MdnsResponderErrorCode::kUnsupportedError; - } - - for (std::unique_ptr<ServiceRecordSet>& record : service_records_) { - if (SameDomainName(&full_instance_name, &record->RR_SRV.namestorage)) { - std::copy(txt.begin(), txt.end(), record->RR_TXT.rdatastorage.u.txt.c); - mDNS_Update(&mdns_, &record->RR_TXT, 0, txt.size(), - &record->RR_TXT.rdatastorage, nullptr); - return MdnsResponderErrorCode::kNoError; - } - } - return MdnsResponderErrorCode::kNoError; -} - -// static -void MdnsResponderAdapterImpl::AQueryCallback(mDNS* m, - DNSQuestion* question, - const ResourceRecord* answer, - QC_result added) { - TRACE_SCOPED(TraceCategory::kMdns, - "MdnsResponderAdapterImpl::AQueryCallback"); - OSP_DCHECK(question); - OSP_DCHECK(answer); - OSP_DCHECK_EQ(answer->rrtype, kDNSType_A); - DomainName domain(std::vector<uint8_t>( - question->qname.c, - question->qname.c + DomainNameLength(&question->qname))); - IPAddress address(answer->rdata->u.ipv4.b); - - auto* adapter = - reinterpret_cast<MdnsResponderAdapterImpl*>(question->QuestionContext); - OSP_DCHECK(adapter); - auto event_type = QueryEventHeader::Type::kAddedNoCache; - if (added == QC_add) { - event_type = QueryEventHeader::Type::kAdded; - } else if (added == QC_rmv) { - event_type = QueryEventHeader::Type::kRemoved; - } else { - OSP_DCHECK_EQ(added, QC_addnocache); - } - adapter->a_responses_.emplace_back( - QueryEventHeader{event_type, - reinterpret_cast<UdpSocket*>(answer->InterfaceID)}, - std::move(domain), address); -} - -// static -void MdnsResponderAdapterImpl::AaaaQueryCallback(mDNS* m, - DNSQuestion* question, - const ResourceRecord* answer, - QC_result added) { - TRACE_SCOPED(TraceCategory::kMdns, - "MdnsResponderAdapterImpl::AaaaQueryCallback"); - OSP_DCHECK(question); - OSP_DCHECK(answer); - OSP_DCHECK_EQ(answer->rrtype, kDNSType_A); - DomainName domain(std::vector<uint8_t>( - question->qname.c, - question->qname.c + DomainNameLength(&question->qname))); - IPAddress address(IPAddress::Version::kV6, answer->rdata->u.ipv6.b); - - auto* adapter = - reinterpret_cast<MdnsResponderAdapterImpl*>(question->QuestionContext); - OSP_DCHECK(adapter); - auto event_type = QueryEventHeader::Type::kAddedNoCache; - if (added == QC_add) { - event_type = QueryEventHeader::Type::kAdded; - } else if (added == QC_rmv) { - event_type = QueryEventHeader::Type::kRemoved; - } else { - OSP_DCHECK_EQ(added, QC_addnocache); - } - adapter->aaaa_responses_.emplace_back( - QueryEventHeader{event_type, - reinterpret_cast<UdpSocket*>(answer->InterfaceID)}, - std::move(domain), address); -} - -// static -void MdnsResponderAdapterImpl::PtrQueryCallback(mDNS* m, - DNSQuestion* question, - const ResourceRecord* answer, - QC_result added) { - TRACE_SCOPED(TraceCategory::kMdns, - "MdnsResponderAdapterImpl::PtrQueryCallback"); - OSP_DCHECK(question); - OSP_DCHECK(answer); - OSP_DCHECK_EQ(answer->rrtype, kDNSType_PTR); - DomainName result(std::vector<uint8_t>( - answer->rdata->u.name.c, - answer->rdata->u.name.c + DomainNameLength(&answer->rdata->u.name))); - - auto* adapter = - reinterpret_cast<MdnsResponderAdapterImpl*>(question->QuestionContext); - OSP_DCHECK(adapter); - auto event_type = QueryEventHeader::Type::kAddedNoCache; - if (added == QC_add) { - event_type = QueryEventHeader::Type::kAdded; - } else if (added == QC_rmv) { - event_type = QueryEventHeader::Type::kRemoved; - } else { - OSP_DCHECK_EQ(added, QC_addnocache); - } - adapter->ptr_responses_.emplace_back( - QueryEventHeader{event_type, - reinterpret_cast<UdpSocket*>(answer->InterfaceID)}, - std::move(result)); -} - -// static -void MdnsResponderAdapterImpl::SrvQueryCallback(mDNS* m, - DNSQuestion* question, - const ResourceRecord* answer, - QC_result added) { - TRACE_SCOPED(TraceCategory::kMdns, - "MdnsResponderAdapterImpl::SrvQueryCallback"); - OSP_DCHECK(question); - OSP_DCHECK(answer); - OSP_DCHECK_EQ(answer->rrtype, kDNSType_SRV); - DomainName service(std::vector<uint8_t>( - question->qname.c, - question->qname.c + DomainNameLength(&question->qname))); - DomainName result( - std::vector<uint8_t>(answer->rdata->u.srv.target.c, - answer->rdata->u.srv.target.c + - DomainNameLength(&answer->rdata->u.srv.target))); - - auto* adapter = - reinterpret_cast<MdnsResponderAdapterImpl*>(question->QuestionContext); - OSP_DCHECK(adapter); - auto event_type = QueryEventHeader::Type::kAddedNoCache; - if (added == QC_add) { - event_type = QueryEventHeader::Type::kAdded; - } else if (added == QC_rmv) { - event_type = QueryEventHeader::Type::kRemoved; - } else { - OSP_DCHECK_EQ(added, QC_addnocache); - } - adapter->srv_responses_.emplace_back( - QueryEventHeader{event_type, - reinterpret_cast<UdpSocket*>(answer->InterfaceID)}, - std::move(service), std::move(result), - GetNetworkOrderPort(answer->rdata->u.srv.port)); -} - -// static -void MdnsResponderAdapterImpl::TxtQueryCallback(mDNS* m, - DNSQuestion* question, - const ResourceRecord* answer, - QC_result added) { - OSP_DCHECK(question); - OSP_DCHECK(answer); - OSP_DCHECK_EQ(answer->rrtype, kDNSType_TXT); - DomainName service(std::vector<uint8_t>( - question->qname.c, - question->qname.c + DomainNameLength(&question->qname))); - auto lines = ParseTxtResponse(answer->rdata->u.txt.c, answer->rdlength); - - auto* adapter = - reinterpret_cast<MdnsResponderAdapterImpl*>(question->QuestionContext); - OSP_DCHECK(adapter); - auto event_type = QueryEventHeader::Type::kAddedNoCache; - if (added == QC_add) { - event_type = QueryEventHeader::Type::kAdded; - } else if (added == QC_rmv) { - event_type = QueryEventHeader::Type::kRemoved; - } else { - OSP_DCHECK_EQ(added, QC_addnocache); - } - adapter->txt_responses_.emplace_back( - QueryEventHeader{event_type, - reinterpret_cast<UdpSocket*>(answer->InterfaceID)}, - std::move(service), std::move(lines)); -} - -// static -void MdnsResponderAdapterImpl::ServiceCallback(mDNS* m, - ServiceRecordSet* service_record, - mStatus result) { - // TODO(btolsch): Handle mStatus_NameConflict. - if (result == mStatus_MemFree) { - OSP_DLOG_INFO << "free service record"; - auto* adapter = reinterpret_cast<MdnsResponderAdapterImpl*>( - service_record->ServiceContext); - auto& service_records = adapter->service_records_; - service_records.erase( - std::remove_if( - service_records.begin(), service_records.end(), - [service_record](const std::unique_ptr<ServiceRecordSet>& sr) { - return sr.get() == service_record; - }), - service_records.end()); - - if (service_records.empty()) - adapter->DeadvertiseInterfaces(); - } -} - -void MdnsResponderAdapterImpl::AdvertiseInterfaces() { - TRACE_SCOPED(TraceCategory::kMdns, - "MdnsResponderAdapterImpl::AdvertiseInterfaces"); - for (auto& info : responder_interface_info_) { - UdpSocket* socket = info.first; - NetworkInterfaceInfo& interface_info = info.second; - mDNS_SetupResourceRecord(&interface_info.RR_A, /** RDataStorage */ nullptr, - reinterpret_cast<mDNSInterfaceID>(socket), - kDNSType_A, kHostNameTTL, kDNSRecordTypeUnique, - AuthRecordAny, - /** Callback */ nullptr, /** Context */ nullptr); - AssignDomainName(&interface_info.RR_A.namestorage, - &mdns_.MulticastHostname); - if (interface_info.ip.type == mDNSAddrType_IPv4) { - interface_info.RR_A.resrec.rdata->u.ipv4 = interface_info.ip.ip.v4; - } else { - interface_info.RR_A.resrec.rdata->u.ipv6 = interface_info.ip.ip.v6; - } - mDNS_Register(&mdns_, &interface_info.RR_A); - } -} - -void MdnsResponderAdapterImpl::DeadvertiseInterfaces() { - // Both loops below use the A resource record's domain name to determine - // whether the record was advertised. AdvertiseInterfaces sets the domain - // name before registering the A record, and this clears it after - // deregistering. - for (auto& info : responder_interface_info_) { - NetworkInterfaceInfo& interface_info = info.second; - if (interface_info.RR_A.namestorage.c[0]) { - mDNS_Deregister(&mdns_, &interface_info.RR_A); - interface_info.RR_A.namestorage.c[0] = 0; - } - } -} - -void MdnsResponderAdapterImpl::RemoveQuestionsIfEmpty(UdpSocket* socket) { - auto entry = socket_to_questions_.find(socket); - bool empty = entry->second.a.empty() || entry->second.aaaa.empty() || - entry->second.ptr.empty() || entry->second.srv.empty() || - entry->second.txt.empty(); - if (empty) - socket_to_questions_.erase(entry); -} - -} // namespace osp -} // namespace openscreen diff --git a/osp/impl/discovery/mdns/mdns_responder_adapter_impl.h b/osp/impl/discovery/mdns/mdns_responder_adapter_impl.h deleted file mode 100644 index d0dd55a1..00000000 --- a/osp/impl/discovery/mdns/mdns_responder_adapter_impl.h +++ /dev/null @@ -1,159 +0,0 @@ -// Copyright 2018 The Chromium Authors. All rights reserved. -// Use of this source code is governed by a BSD-style license that can be -// found in the LICENSE file. - -#ifndef OSP_IMPL_DISCOVERY_MDNS_MDNS_RESPONDER_ADAPTER_IMPL_H_ -#define OSP_IMPL_DISCOVERY_MDNS_MDNS_RESPONDER_ADAPTER_IMPL_H_ - -#include <map> -#include <memory> -#include <string> -#include <vector> - -#include "osp/impl/discovery/mdns/mdns_responder_adapter.h" -#include "platform/api/udp_socket.h" -#include "platform/base/error.h" -#include "third_party/mDNSResponder/src/mDNSCore/mDNSEmbeddedAPI.h" - -namespace openscreen { -namespace osp { - -class MdnsResponderAdapterImpl final : public MdnsResponderAdapter { - public: - static constexpr int kRrCacheSize = 500; - - MdnsResponderAdapterImpl(); - ~MdnsResponderAdapterImpl() override; - - Error Init() override; - void Close() override; - - Error SetHostLabel(const std::string& host_label) override; - - Error RegisterInterface(const InterfaceInfo& interface_info, - const IPSubnet& interface_address, - UdpSocket* socket) override; - Error DeregisterInterface(UdpSocket* socket) override; - - void OnRead(UdpSocket* socket, ErrorOr<UdpPacket> packet) override; - void OnSendError(UdpSocket* socket, Error error) override; - void OnError(UdpSocket* socket, Error error) override; - void OnBound(UdpSocket* socket) override; - - Clock::duration RunTasks() override; - - std::vector<PtrEvent> TakePtrResponses() override; - std::vector<SrvEvent> TakeSrvResponses() override; - std::vector<TxtEvent> TakeTxtResponses() override; - std::vector<AEvent> TakeAResponses() override; - std::vector<AaaaEvent> TakeAaaaResponses() override; - - MdnsResponderErrorCode StartPtrQuery(UdpSocket* socket, - const DomainName& service_type) override; - MdnsResponderErrorCode StartSrvQuery( - UdpSocket* socket, - const DomainName& service_instance) override; - MdnsResponderErrorCode StartTxtQuery( - UdpSocket* socket, - const DomainName& service_instance) override; - MdnsResponderErrorCode StartAQuery(UdpSocket* socket, - const DomainName& domain_name) override; - MdnsResponderErrorCode StartAaaaQuery(UdpSocket* socket, - const DomainName& domain_name) override; - MdnsResponderErrorCode StopPtrQuery(UdpSocket* socket, - const DomainName& service_type) override; - MdnsResponderErrorCode StopSrvQuery( - UdpSocket* socket, - const DomainName& service_instance) override; - MdnsResponderErrorCode StopTxtQuery( - UdpSocket* socket, - const DomainName& service_instance) override; - MdnsResponderErrorCode StopAQuery(UdpSocket* socket, - const DomainName& domain_name) override; - MdnsResponderErrorCode StopAaaaQuery(UdpSocket* socket, - const DomainName& domain_name) override; - - MdnsResponderErrorCode RegisterService( - const std::string& service_instance, - const std::string& service_name, - const std::string& service_protocol, - const DomainName& target_host, - uint16_t target_port, - const std::map<std::string, std::string>& txt_data) override; - MdnsResponderErrorCode DeregisterService( - const std::string& service_instance, - const std::string& service_name, - const std::string& service_protocol) override; - MdnsResponderErrorCode UpdateTxtData( - const std::string& service_instance, - const std::string& service_name, - const std::string& service_protocol, - const std::map<std::string, std::string>& txt_data) override; - - private: - struct Questions { - std::map<DomainName, DNSQuestion, DomainNameComparator> a; - std::map<DomainName, DNSQuestion, DomainNameComparator> aaaa; - std::map<DomainName, DNSQuestion, DomainNameComparator> ptr; - std::map<DomainName, DNSQuestion, DomainNameComparator> srv; - std::map<DomainName, DNSQuestion, DomainNameComparator> txt; - }; - - static void AQueryCallback(mDNS* m, - DNSQuestion* question, - const ResourceRecord* answer, - QC_result added); - static void AaaaQueryCallback(mDNS* m, - DNSQuestion* question, - const ResourceRecord* answer, - QC_result added); - static void PtrQueryCallback(mDNS* m, - DNSQuestion* question, - const ResourceRecord* answer, - QC_result added); - static void SrvQueryCallback(mDNS* m, - DNSQuestion* question, - const ResourceRecord* answer, - QC_result added); - static void TxtQueryCallback(mDNS* m, - DNSQuestion* question, - const ResourceRecord* answer, - QC_result added); - static void ServiceCallback(mDNS* m, - ServiceRecordSet* service_record, - mStatus result); - - void AdvertiseInterfaces(); - void DeadvertiseInterfaces(); - void RemoveQuestionsIfEmpty(UdpSocket* socket); - - CacheEntity rr_cache_[kRrCacheSize]; - - // The main context structure for mDNSResponder. - mDNS mdns_; - - // Our own storage that is placed inside |mdns_|. The intent in C is to allow - // us access to our own state during callbacks. Here we just use it to group - // platform sockets. - mDNS_PlatformSupport platform_storage_; - - std::map<UdpSocket*, Questions> socket_to_questions_; - - std::map<UdpSocket*, NetworkInterfaceInfo> responder_interface_info_; - - std::vector<AEvent> a_responses_; - std::vector<AaaaEvent> aaaa_responses_; - std::vector<PtrEvent> ptr_responses_; - std::vector<SrvEvent> srv_responses_; - std::vector<TxtEvent> txt_responses_; - - // A list of services we are advertising. ServiceRecordSet is an - // mDNSResponder structure which holds all the resource record data - // (PTR/SRV/TXT/A and misc.) that is necessary to advertise a service. - std::vector<std::unique_ptr<ServiceRecordSet>> service_records_; -}; - -} // namespace osp -} // namespace openscreen - -#endif // OSP_IMPL_DISCOVERY_MDNS_MDNS_RESPONDER_ADAPTER_IMPL_H_ diff --git a/osp/impl/discovery/mdns/mdns_responder_adapter_impl_unittest.cc b/osp/impl/discovery/mdns/mdns_responder_adapter_impl_unittest.cc deleted file mode 100644 index 29b76679..00000000 --- a/osp/impl/discovery/mdns/mdns_responder_adapter_impl_unittest.cc +++ /dev/null @@ -1,94 +0,0 @@ -// Copyright 2018 The Chromium 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 "osp/impl/discovery/mdns/mdns_responder_adapter_impl.h" - -#include <memory> -#include <string> - -#include "gmock/gmock.h" -#include "gtest/gtest.h" - -namespace openscreen { -namespace osp { -namespace { - -using ::testing::ElementsAre; -using ::testing::ElementsAreArray; - -// Example response for _openscreen._udp. Contains PTR, SRV, TXT, A records. -uint8_t data[] = { - 0x00, 0x00, 0x84, 0x00, 0x00, 0x00, 0x00, 0x05, 0x00, 0x00, 0x00, 0x03, - 0x06, 0x74, 0x75, 0x72, 0x74, 0x6c, 0x65, 0x0b, 0x5f, 0x6f, 0x70, 0x65, - 0x6e, 0x73, 0x63, 0x72, 0x65, 0x65, 0x6e, 0x04, 0x5f, 0x75, 0x64, 0x70, - 0x05, 0x6c, 0x6f, 0x63, 0x61, 0x6c, 0x00, 0x00, 0x10, 0x80, 0x01, 0x00, - 0x00, 0x11, 0x94, 0x00, 0x0e, 0x06, 0x79, 0x75, 0x72, 0x74, 0x6c, 0x65, - 0x06, 0x74, 0x75, 0x72, 0x74, 0x6c, 0x65, 0x09, 0x5f, 0x73, 0x65, 0x72, - 0x76, 0x69, 0x63, 0x65, 0x73, 0x07, 0x5f, 0x64, 0x6e, 0x73, 0x2d, 0x73, - 0x64, 0xc0, 0x1f, 0x00, 0x0c, 0x00, 0x01, 0x00, 0x00, 0x11, 0x94, 0x00, - 0x02, 0xc0, 0x13, 0xc0, 0x13, 0x00, 0x0c, 0x00, 0x01, 0x00, 0x00, 0x11, - 0x94, 0x00, 0x02, 0xc0, 0x0c, 0x11, 0x67, 0x69, 0x67, 0x6c, 0x69, 0x6f, - 0x72, 0x6f, 0x6e, 0x6f, 0x6e, 0x6f, 0x6d, 0x69, 0x63, 0x6f, 0x6e, 0xc0, - 0x24, 0x00, 0x01, 0x80, 0x01, 0x00, 0x00, 0x00, 0x78, 0x00, 0x04, 0xac, - 0x11, 0x20, 0x96, 0xc0, 0x0c, 0x00, 0x21, 0x80, 0x01, 0x00, 0x00, 0x00, - 0x78, 0x00, 0x08, 0x00, 0x00, 0x00, 0x00, 0x30, 0x39, 0xc0, 0x71, 0xc0, - 0x0c, 0x00, 0x2f, 0x80, 0x01, 0x00, 0x00, 0x11, 0x94, 0x00, 0x09, 0xc0, - 0x0c, 0x00, 0x05, 0x00, 0x00, 0x80, 0x00, 0x40, 0xc0, 0x71, 0x00, 0x2f, - 0x80, 0x01, 0x00, 0x00, 0x00, 0x78, 0x00, 0x05, 0xc0, 0x71, 0x00, 0x01, - 0x40, 0x00, 0x00, 0x29, 0x05, 0xa0, 0x00, 0x00, 0x11, 0x94, 0x00, 0x12, - 0x00, 0x04, 0x00, 0x0e, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, - 0x50, 0x65, 0xf3, 0x41, 0x27, 0x01, -}; - -} // namespace - -TEST(MdnsResponderAdapterImplTest, ExampleData) { - const DomainName openscreen_service{{11, '_', 'o', 'p', 'e', 'n', 's', 'c', - 'r', 'e', 'e', 'n', 4, '_', 'u', 'd', - 'p', 5, 'l', 'o', 'c', 'a', 'l', 0}}; - const IPEndpoint mdns_endpoint{{224, 0, 0, 251}, 5353}; - - UdpPacket packet(std::begin(data), std::end(data)); - packet.set_source({{192, 168, 0, 2}, 6556}); - packet.set_destination(mdns_endpoint); - packet.set_socket(nullptr); - - auto mdns_adapter = - std::unique_ptr<MdnsResponderAdapter>(new MdnsResponderAdapterImpl); - mdns_adapter->Init(); - mdns_adapter->StartPtrQuery(0, openscreen_service); - mdns_adapter->OnRead(nullptr, std::move(packet)); - mdns_adapter->RunTasks(); - - auto ptr = mdns_adapter->TakePtrResponses(); - ASSERT_EQ(1u, ptr.size()); - ASSERT_THAT(ptr[0].service_instance.GetLabels(), - ElementsAre("turtle", "_openscreen", "_udp", "local")); - mdns_adapter->StartSrvQuery(0, ptr[0].service_instance); - mdns_adapter->StartTxtQuery(0, ptr[0].service_instance); - mdns_adapter->RunTasks(); - - auto srv = mdns_adapter->TakeSrvResponses(); - ASSERT_EQ(1u, srv.size()); - ASSERT_THAT(srv[0].domain_name.GetLabels(), - ElementsAre("gigliorononomicon", "local")); - EXPECT_EQ(12345, srv[0].port); - - auto txt = mdns_adapter->TakeTxtResponses(); - ASSERT_EQ(1u, txt.size()); - const std::string expected_txt[] = {"yurtle", "turtle"}; - EXPECT_THAT(txt[0].txt_info, ElementsAreArray(expected_txt)); - - mdns_adapter->StartAQuery(0, srv[0].domain_name); - mdns_adapter->RunTasks(); - - auto a = mdns_adapter->TakeAResponses(); - ASSERT_EQ(1u, a.size()); - EXPECT_EQ((IPAddress{172, 17, 32, 150}), a[0].address); - - mdns_adapter->Close(); -} - -} // namespace osp -} // namespace openscreen diff --git a/osp/impl/discovery/mdns/mdns_responder_platform.cc b/osp/impl/discovery/mdns/mdns_responder_platform.cc deleted file mode 100644 index 14204ff5..00000000 --- a/osp/impl/discovery/mdns/mdns_responder_platform.cc +++ /dev/null @@ -1,290 +0,0 @@ -// Copyright 2018 The Chromium 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 "osp/impl/discovery/mdns/mdns_responder_platform.h" - -#include <algorithm> -#include <chrono> -#include <cstring> -#include <limits> -#include <vector> - -#include "platform/api/network_interface.h" -#include "platform/api/time.h" -#include "platform/api/udp_socket.h" -#include "platform/base/error.h" -#include "platform/base/ip_address.h" -#include "third_party/mDNSResponder/src/mDNSCore/mDNSEmbeddedAPI.h" -#include "util/osp_logging.h" - -namespace { - -using std::chrono::duration_cast; -using std::chrono::hours; -using std::chrono::milliseconds; -using std::chrono::seconds; - -} // namespace - -extern "C" { - -const char ProgramName[] = "openscreen"; - -mDNSs32 mDNSPlatformOneSecond = 1000; - -mStatus mDNSPlatformInit(mDNS* m) { - mDNSCoreInitComplete(m, mStatus_NoError); - return mStatus_NoError; -} - -void mDNSPlatformClose(mDNS* m) {} - -mStatus mDNSPlatformSendUDP(const mDNS* m, - const void* msg, - const mDNSu8* last, - mDNSInterfaceID InterfaceID, - UDPSocket* src, - const mDNSAddr* dst, - mDNSIPPort dstport) { - auto* const socket = reinterpret_cast<openscreen::UdpSocket*>(InterfaceID); - const auto socket_it = - std::find(m->p->sockets.begin(), m->p->sockets.end(), socket); - if (socket_it == m->p->sockets.end()) - return mStatus_BadInterfaceErr; - - openscreen::IPEndpoint dest{ - openscreen::IPAddress{dst->type == mDNSAddrType_IPv4 - ? openscreen::IPAddress::Version::kV4 - : openscreen::IPAddress::Version::kV6, - dst->ip.v4.b}, - static_cast<uint16_t>((dstport.b[0] << 8) | dstport.b[1])}; - const int64_t length = last - static_cast<const uint8_t*>(msg); - if (length < 0 || length > std::numeric_limits<ssize_t>::max()) { - return mStatus_BadParamErr; - } - - // UDP is inherently lossy, so don't worry about async failures and let the - // underlying protocol handle it. - (*socket_it)->SendMessage(msg, length, dest); - return mStatus_NoError; -} - -void mDNSPlatformLock(const mDNS* m) { - // We're single threaded. -} - -void mDNSPlatformUnlock(const mDNS* m) {} - -void mDNSPlatformStrCopy(void* dst, const void* src) { - // Unfortunately, the caller is responsible for making sure that dst - // if of sufficient length to store the src string. Otherwise we may - // cause an access violation. - std::strcpy(static_cast<char*>(dst), // NOLINT - static_cast<const char*>(src)); -} - -mDNSu32 mDNSPlatformStrLen(const void* src) { - return std::strlen(static_cast<const char*>(src)); -} - -void mDNSPlatformMemCopy(void* dst, const void* src, mDNSu32 len) { - std::memcpy(dst, src, len); -} - -mDNSBool mDNSPlatformMemSame(const void* dst, const void* src, mDNSu32 len) { - return std::memcmp(dst, src, len) == 0 ? mDNStrue : mDNSfalse; -} - -void mDNSPlatformMemZero(void* dst, mDNSu32 len) { - std::memset(dst, 0, len); -} - -void* mDNSPlatformMemAllocate(mDNSu32 len) { - return malloc(len); -} - -void mDNSPlatformMemFree(void* mem) { - free(mem); -} - -mDNSu32 mDNSPlatformRandomSeed() { - return std::chrono::steady_clock::now().time_since_epoch().count(); -} - -mStatus mDNSPlatformTimeInit() { - return mStatus_NoError; -} - -mDNSs32 mDNSPlatformRawTime() { - using openscreen::Clock; - - const Clock::time_point now = Clock::now(); - - // A signed 32-bit integer counting milliseconds only gives ~24.8 days of - // range. Thus, the first time this function is called, record a new origin - // timestamp to subtract from the raw monotonic clock values. The "one hour - // before now" value is used to keep the results well-ahead of zero because - // the mDNS library assumes this is the time since kernel boot and has hacks - // to disable certain things in the first few minutes. :-/ - static const Clock::time_point origin = now - hours(1); - - const int64_t millis_since_origin = - duration_cast<milliseconds>(now - origin).count(); - OSP_CHECK_LE(millis_since_origin, std::numeric_limits<mDNSs32>::max()); - return static_cast<mDNSs32>(millis_since_origin); -} - -mDNSs32 mDNSPlatformUTC() { - const auto seconds_since_epoch = - duration_cast<seconds>(openscreen::GetWallTimeSinceUnixEpoch()).count(); - - // The return type will cause overflow in early 2038. Warn future developers - // a year ahead of time. - constexpr mDNSs32 a_year_before_overflow = - std::numeric_limits<mDNSs32>::max() - - duration_cast<seconds>(365 * hours(24)).count(); - OSP_DCHECK_LE(seconds_since_epoch, a_year_before_overflow); - - return static_cast<mDNSs32>(seconds_since_epoch); -} - -void mDNSPlatformWriteDebugMsg(const char* msg) { - OSP_DVLOG << __func__ << ": " << msg; -} - -void mDNSPlatformWriteLogMsg(const char* ident, - const char* msg, - mDNSLogLevel_t loglevel) { - OSP_VLOG << __func__ << ": " << msg; -} - -TCPSocket* mDNSPlatformTCPSocket(mDNS* const m, - TCPSocketFlags flags, - mDNSIPPort* port) { - OSP_UNIMPLEMENTED(); - return nullptr; -} - -TCPSocket* mDNSPlatformTCPAccept(TCPSocketFlags flags, int sd) { - OSP_UNIMPLEMENTED(); - return nullptr; -} - -int mDNSPlatformTCPGetFD(TCPSocket* sock) { - OSP_UNIMPLEMENTED(); - return 0; -} - -mStatus mDNSPlatformTCPConnect(TCPSocket* sock, - const mDNSAddr* dst, - mDNSOpaque16 dstport, - domainname* hostname, - mDNSInterfaceID InterfaceID, - TCPConnectionCallback callback, - void* context) { - OSP_UNIMPLEMENTED(); - return mStatus_NoError; -} - -void mDNSPlatformTCPCloseConnection(TCPSocket* sock) { - OSP_UNIMPLEMENTED(); -} - -long mDNSPlatformReadTCP(TCPSocket* sock, // NOLINT - void* buf, - unsigned long buflen, // NOLINT - mDNSBool* closed) { - OSP_UNIMPLEMENTED(); - return 0; -} - -long mDNSPlatformWriteTCP(TCPSocket* sock, // NOLINT - const char* msg, - unsigned long len) { // NOLINT - OSP_UNIMPLEMENTED(); - return 0; -} - -UDPSocket* mDNSPlatformUDPSocket(mDNS* const m, - const mDNSIPPort requestedport) { - OSP_UNIMPLEMENTED(); - return nullptr; -} - -void mDNSPlatformUDPClose(UDPSocket* sock) { - OSP_UNIMPLEMENTED(); -} - -void mDNSPlatformReceiveBPF_fd(mDNS* const m, int fd) { - OSP_UNIMPLEMENTED(); -} - -void mDNSPlatformUpdateProxyList(mDNS* const m, - const mDNSInterfaceID InterfaceID) { - OSP_UNIMPLEMENTED(); -} - -void mDNSPlatformSendRawPacket(const void* const msg, - const mDNSu8* const end, - mDNSInterfaceID InterfaceID) { - OSP_UNIMPLEMENTED(); -} - -void mDNSPlatformSetLocalAddressCacheEntry(mDNS* const m, - const mDNSAddr* const tpa, - const mDNSEthAddr* const tha, - mDNSInterfaceID InterfaceID) {} - -void mDNSPlatformSourceAddrForDest(mDNSAddr* const src, - const mDNSAddr* const dst) {} - -mStatus mDNSPlatformTLSSetupCerts(void) { - OSP_UNIMPLEMENTED(); - return mStatus_NoError; -} - -void mDNSPlatformTLSTearDownCerts(void) { - OSP_UNIMPLEMENTED(); -} - -void mDNSPlatformSetDNSConfig(mDNS* const m, - mDNSBool setservers, - mDNSBool setsearch, - domainname* const fqdn, - DNameListElem** RegDomains, - DNameListElem** BrowseDomains) { - if (fqdn) { - std::memset(fqdn, 0, sizeof(*fqdn)); - } -} - -mStatus mDNSPlatformGetPrimaryInterface(mDNS* const m, - mDNSAddr* v4, - mDNSAddr* v6, - mDNSAddr* router) { - return mStatus_NoError; -} - -void mDNSPlatformDynDNSHostNameStatusChanged(const domainname* const dname, - const mStatus status) {} - -void mDNSPlatformSetAllowSleep(mDNS* const m, - mDNSBool allowSleep, - const char* reason) {} - -void mDNSPlatformSendWakeupPacket(mDNS* const m, - mDNSInterfaceID InterfaceID, - char* EthAddr, - char* IPAddr, - int iteration) { - OSP_UNIMPLEMENTED(); -} - -mDNSBool mDNSPlatformValidRecordForInterface(AuthRecord* rr, - const NetworkInterfaceInfo* intf) { - OSP_UNIMPLEMENTED(); - return mDNStrue; -} - -} // extern "C" diff --git a/osp/impl/discovery/mdns/mdns_responder_platform.h b/osp/impl/discovery/mdns/mdns_responder_platform.h deleted file mode 100644 index 342913fe..00000000 --- a/osp/impl/discovery/mdns/mdns_responder_platform.h +++ /dev/null @@ -1,16 +0,0 @@ -// Copyright 2018 The Chromium Authors. All rights reserved. -// Use of this source code is governed by a BSD-style license that can be -// found in the LICENSE file. - -#ifndef OSP_IMPL_DISCOVERY_MDNS_MDNS_RESPONDER_PLATFORM_H_ -#define OSP_IMPL_DISCOVERY_MDNS_MDNS_RESPONDER_PLATFORM_H_ - -#include <vector> - -#include "platform/api/udp_socket.h" - -struct mDNS_PlatformSupport_struct { - std::vector<openscreen::UdpSocket*> sockets; -}; - -#endif // OSP_IMPL_DISCOVERY_MDNS_MDNS_RESPONDER_PLATFORM_H_ |