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
|
/*
* Copyright (c) 2017-2019 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.
*/
#include <hicn/transport/interfaces/socket_consumer.h>
#include <protocols/index_manager_bytestream.h>
#include <protocols/manifest_incremental_indexer_bytestream.h>
#include <protocols/transport_protocol.h>
namespace transport {
namespace protocol {
IndexManager::IndexManager(implementation::ConsumerSocket *icn_socket,
TransportProtocol *transport)
: IncrementalIndexer(icn_socket, transport),
indexer_(std::make_unique<IncrementalIndexer>(icn_socket, transport)),
first_segment_received_(false) {}
void IndexManager::onContentObject(core::Interest &interest,
core::ContentObject &content_object,
bool reassembly) {
if (first_segment_received_) {
return indexer_->onContentObject(interest, content_object, reassembly);
} else {
std::uint32_t segment_number = interest.getName().getSuffix();
if (segment_number == 0) {
// Check if manifest
if (content_object.getPayloadType() == core::PayloadType::MANIFEST) {
IncrementalIndexer *indexer =
static_cast<IncrementalIndexer *>(indexer_.release());
indexer_ =
std::make_unique<ManifestIncrementalIndexer>(std::move(*indexer));
delete indexer;
}
indexer_->onContentObject(interest, content_object);
auto it = interest_data_set_.begin();
while (it != interest_data_set_.end()) {
indexer_->onContentObject(*it->first, *it->second);
it = interest_data_set_.erase(it);
}
first_segment_received_ = true;
} else {
interest_data_set_.emplace(interest.shared_from_this(),
content_object.shared_from_this());
}
}
}
void IndexManager::reset() {
indexer_ = std::make_unique<IncrementalIndexer>(socket_, transport_);
indexer_->setReassembly(this->reassembly_);
indexer_->reset();
first_segment_received_ = false;
interest_data_set_.clear();
}
} // namespace protocol
} // namespace transport
|
