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
|
/*
* 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 <hicn/transport/protocols/indexing_manager.h>
#include <hicn/transport/protocols/reassembly.h>
#include <hicn/transport/utils/array.h>
namespace transport {
namespace protocol {
BaseReassembly::BaseReassembly(interface::ConsumerSocket *icn_socket,
ContentReassembledCallback *content_callback)
: reassembly_consumer_socket_(icn_socket),
incremental_index_manager_(
std::make_unique<IncrementalIndexManager>(icn_socket)),
manifest_index_manager_(
std::make_unique<ManifestIndexManager>(icn_socket)),
index_manager_(incremental_index_manager_.get()),
index_(0) {
setContentCallback(content_callback);
}
void BaseReassembly::reassemble(ContentObject::Ptr &&content_object) {
if (TRANSPORT_EXPECT_TRUE(content_object != nullptr)) {
received_packets_.emplace(std::make_pair(
content_object->getName().getSuffix(), std::move(content_object)));
}
auto it = received_packets_.find((const unsigned int)index_);
while (it != received_packets_.end()) {
if (it->second->getPayloadType() == PayloadType::CONTENT_OBJECT) {
copyContent(*it->second);
received_packets_.erase(it);
}
index_ = index_manager_->getNextReassemblySegment();
it = received_packets_.find((const unsigned int)index_);
}
}
void BaseReassembly::copyContent(const ContentObject &content_object) {
auto a = content_object.getPayload();
std::shared_ptr<std::vector<uint8_t>> content_buffer;
reassembly_consumer_socket_->getSocketOption(
interface::GeneralTransportOptions::APPLICATION_BUFFER, content_buffer);
content_buffer->insert(content_buffer->end(), (uint8_t *)a->data(),
(uint8_t *)a->data() + a->length());
bool download_completed =
index_manager_->getFinalSuffix() == content_object.getName().getSuffix();
if (TRANSPORT_EXPECT_FALSE(download_completed)) {
content_callback_->onContentReassembled(std::make_error_code(std::errc(0)));
}
}
void BaseReassembly::reset() {
manifest_index_manager_->reset();
incremental_index_manager_->reset();
index_ = index_manager_->getNextReassemblySegment();
received_packets_.clear();
}
} // namespace protocol
} // end namespace transport
|
