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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
/*
* Copyright (c) 2021 Cisco and/or its affiliates.
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at:
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#pragma once
#include <core/manifest.h>
#include <core/manifest_format.h>
#include <hicn/transport/portability/portability.h>
#include <set>
namespace transport {
namespace core {
template <typename Base, typename FormatTraits>
class ManifestInline
: public Manifest<Base, FormatTraits, ManifestInline<Base, FormatTraits>> {
using ManifestBase =
Manifest<Base, FormatTraits, ManifestInline<Base, FormatTraits>>;
using Hash = typename FormatTraits::Hash;
using HashType = typename FormatTraits::HashType;
using Suffix = typename FormatTraits::Suffix;
using SuffixList = typename FormatTraits::SuffixList;
using HashEntry = std::pair<auth::CryptoHashType, std::vector<uint8_t>>;
public:
ManifestInline() : ManifestBase() {}
ManifestInline(Packet::Format format, const core::Name &name,
std::size_t signature_size = 0)
: ManifestBase(format, name, signature_size) {}
template <typename T>
ManifestInline(T &&base) : ManifestBase(std::forward<T &&>(base)) {}
template <typename T>
ManifestInline(T &base) : ManifestBase(base) {}
static TRANSPORT_ALWAYS_INLINE ManifestInline *createManifest(
Packet::Format format, const core::Name &manifest_name,
ManifestVersion version, ManifestType type, bool is_last,
const Name &base_name, HashType hash_algo, std::size_t signature_size) {
auto manifest = new ManifestInline(format, manifest_name, signature_size);
manifest->setVersion(version);
manifest->setType(type);
manifest->setHashAlgorithm(hash_algo);
manifest->setIsLast(is_last);
manifest->setBaseName(base_name);
return manifest;
}
ManifestInline &encodeImpl() {
ManifestBase::encoder_.encode();
return *this;
}
ManifestInline &decodeImpl() {
base_name_ = ManifestBase::decoder_.getBaseName();
suffix_hash_map_ = ManifestBase::decoder_.getSuffixHashList();
return *this;
}
std::size_t estimateManifestSizeImpl(std::size_t additional_entries = 0) {
return ManifestBase::encoder_.estimateSerializedLength(additional_entries);
}
ManifestInline &setBaseName(const Name &name) {
base_name_ = name;
ManifestBase::encoder_.setBaseName(base_name_);
return *this;
}
const Name &getBaseName() { return base_name_; }
ManifestInline &addSuffixHash(Suffix suffix, const Hash &hash) {
ManifestBase::encoder_.addSuffixAndHash(suffix, hash);
return *this;
}
// Call this function only after the decode function!
const SuffixList &getSuffixList() { return suffix_hash_map_; }
// Convert several manifests into a single map from suffixes to packet hashes.
// All manifests must have been decoded beforehand.
static std::unordered_map<Suffix, Hash> getSuffixMap(
const std::vector<ManifestInline *> &manifests) {
std::unordered_map<Suffix, Hash> suffix_map;
for (auto manifest_ptr : manifests) {
HashType hash_type = manifest_ptr->getHashAlgorithm();
SuffixList suffix_list = manifest_ptr->getSuffixList();
for (auto it = suffix_list.begin(); it != suffix_list.end(); ++it) {
Hash hash(it->second, Hash::getSize(hash_type), hash_type);
suffix_map[it->first] = hash;
}
}
return suffix_map;
}
static std::unordered_map<Suffix, Hash> getSuffixMap(
ManifestInline *manifest) {
return getSuffixMap(std::vector<ManifestInline *>{manifest});
}
private:
core::Name base_name_;
SuffixList suffix_hash_map_;
};
} // namespace core
} // namespace transport
|