You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
83 lines
2.9 KiB
83 lines
2.9 KiB
// 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 "discovery/dnssd/impl/service_key.h"
|
|
|
|
#include "absl/strings/str_join.h"
|
|
#include "absl/strings/str_split.h"
|
|
#include "discovery/dnssd/impl/conversion_layer.h"
|
|
#include "discovery/mdns/mdns_records.h"
|
|
#include "discovery/mdns/public/mdns_constants.h"
|
|
|
|
namespace openscreen {
|
|
namespace discovery {
|
|
|
|
// The InstanceKey ctor used below cares about the Instance ID of the
|
|
// MdnsRecord, while this class doesn't, so it's possible that the InstanceKey
|
|
// ctor will fail for this reason. That's not a problem though - A failure when
|
|
// creating an InstanceKey would mean that the record we received was invalid,
|
|
// so there is no reason to continue processing.
|
|
ServiceKey::ServiceKey(const MdnsRecord& record) {
|
|
ErrorOr<ServiceKey> key = TryCreate(record);
|
|
*this = std::move(key.value());
|
|
}
|
|
|
|
ServiceKey::ServiceKey(const DomainName& domain) {
|
|
ErrorOr<ServiceKey> key = TryCreate(domain);
|
|
*this = std::move(key.value());
|
|
}
|
|
|
|
ServiceKey::ServiceKey(absl::string_view service, absl::string_view domain)
|
|
: service_id_(service.data(), service.size()),
|
|
domain_id_(domain.data(), domain.size()) {
|
|
OSP_DCHECK(IsServiceValid(service_id_)) << "invalid service id: " << service;
|
|
OSP_DCHECK(IsDomainValid(domain_id_)) << "invalid domain id: " << domain;
|
|
}
|
|
|
|
ServiceKey::ServiceKey(const ServiceKey& other) = default;
|
|
ServiceKey::ServiceKey(ServiceKey&& other) = default;
|
|
|
|
ServiceKey& ServiceKey::operator=(const ServiceKey& rhs) = default;
|
|
ServiceKey& ServiceKey::operator=(ServiceKey&& rhs) = default;
|
|
|
|
DomainName ServiceKey::GetName() const {
|
|
std::string service_type = service_id().substr(0, service_id().size() - 5);
|
|
std::string protocol = service_id().substr(service_id().size() - 4);
|
|
return DomainName{std::move(service_type), std::move(protocol), domain_id_};
|
|
}
|
|
|
|
// static
|
|
ErrorOr<ServiceKey> ServiceKey::TryCreate(const MdnsRecord& record) {
|
|
return TryCreate(GetDomainName(record));
|
|
}
|
|
|
|
// static
|
|
ErrorOr<ServiceKey> ServiceKey::TryCreate(const DomainName& names) {
|
|
// Size must be at least 4, because the minimum valid label is of the form
|
|
// <instance>.<service type>.<protocol>.<domain>
|
|
if (names.labels().size() < 4) {
|
|
return Error::Code::kParameterInvalid;
|
|
}
|
|
|
|
// Skip the InstanceId.
|
|
auto it = ++names.labels().begin();
|
|
|
|
std::string service_name = *it++;
|
|
const std::string protocol = *it++;
|
|
const std::string service_id = service_name.append(".").append(protocol);
|
|
if (!IsServiceValid(service_id)) {
|
|
return Error::Code::kParameterInvalid;
|
|
}
|
|
|
|
const std::string domain_id = absl::StrJoin(it, names.labels().end(), ".");
|
|
if (!IsDomainValid(domain_id)) {
|
|
return Error::Code::kParameterInvalid;
|
|
}
|
|
|
|
return ServiceKey(service_id, domain_id);
|
|
}
|
|
|
|
} // namespace discovery
|
|
} // namespace openscreen
|