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
129
|
/*
* 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.
*/
#pragma once
#include <hicn/transport/http/default_values.h>
#include <hicn/transport/http/request.h>
#include <hicn/transport/http/response.h>
#include <hicn/transport/interfaces/socket_consumer.h>
#include <hicn/transport/interfaces/socket_producer.h>
#include <hicn/transport/interfaces/verification_policy.h>
#include <hicn/transport/utils/uri.h>
#include <vector>
namespace transport {
namespace http {
using namespace interface;
using namespace core;
class HTTPClientConnection : public ConsumerSocket::ReadCallback {
static constexpr uint32_t max_buffer_capacity = 64 * 1024;
public:
class ReadBytesCallback {
public:
virtual void onBytesReceived(std::unique_ptr<utils::MemBuf> &&buffer) = 0;
virtual void onSuccess(std::size_t bytes) = 0;
virtual void onError(const std::error_code ec) = 0;
};
enum class RC : uint32_t { DOWNLOAD_FAILED, DOWNLOAD_SUCCESS };
HTTPClientConnection();
RC get(const std::string &url, HTTPHeaders headers = {},
HTTPPayload payload = {},
std::shared_ptr<HTTPResponse> response = nullptr,
ReadBytesCallback *callback = nullptr,
std::string ipv6_first_word = "b001");
RC sendRequest(const std::string &url, HTTPMethod method,
HTTPHeaders headers = {}, HTTPPayload payload = {},
std::shared_ptr<HTTPResponse> response = nullptr,
ReadBytesCallback *callback = nullptr,
std::string ipv6_first_word = "b001");
HTTPResponse response();
HTTPClientConnection &stop();
interface::ConsumerSocket &getConsumer();
HTTPClientConnection &setTimeout(const std::chrono::seconds &timeout);
HTTPClientConnection &setCertificate(const std::string &cert_path);
void verifyPacketSignature(bool verify);
private:
void sendRequestGetReply(const HTTPRequest &request,
std::shared_ptr<HTTPResponse> &response,
std::string &ipv6_first_word);
bool verifyData(interface::ConsumerSocket &c,
const core::ContentObject &contentObject);
void processLeavingInterest(interface::ConsumerSocket &c,
const core::Interest &interest,
std::string &payload);
VerificationPolicy onSignatureVerificationFailed(
ConsumerSocket &consumer, const core::ContentObject &content_object,
std::error_code reason);
// Read callback
bool isBufferMovable() noexcept override { return true; }
void getReadBuffer(uint8_t **application_buffer,
size_t *max_length) override {}
void readDataAvailable(size_t length) noexcept override {}
size_t maxBufferSize() const override { return max_buffer_capacity; }
void readBufferAvailable(
std::unique_ptr<utils::MemBuf> &&buffer) noexcept override;
void readError(const std::error_code ec) noexcept override;
void readSuccess(std::size_t total_size) noexcept override;
// The consumer socket
ConsumerSocket consumer_;
// The current url provided by the application
std::string current_url_;
// The current hICN name used for downloading
std::stringstream name_;
// Function to be called when the read is successful
std::function<std::shared_ptr<HTTPResponse>(std::size_t)> success_callback_;
// Return code for current download
RC return_code_;
// Application provided callback for saving the received content during
// the download. If this callback is used, the HTTPClient will NOT save
// any byte internally.
ReadBytesCallback *read_bytes_callback_;
// Internal read buffer and HTTP response, to be used if the application does
// not provide any read_bytes_callback
std::unique_ptr<utils::MemBuf> read_buffer_;
std::shared_ptr<HTTPResponse> response_;
// Timer
std::unique_ptr<asio::steady_timer> timer_;
};
} // end namespace http
} // end namespace transport
|