1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
|
// 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
|
