2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
42#include "web/http/client/Request.h"
44#include "core/file/FileReader.h"
45#include "core/socket/stream/SocketConnection.h"
46#include "web/http/MimeTypes.h"
47#include "web/http/client/SocketContext.h"
48#include "web/http/client/SocketContextUpgradeFactorySelector.h"
52#include "commands/EndCommand.h"
53#include "commands/SendFileCommand.h"
54#include "commands/SendFragmentCommand.h"
55#include "commands/SendHeaderCommand.h"
56#include "commands/UpgradeCommand.h"
58#ifndef DOXYGEN_SHOULD_SKIP_THIS
60#include "log/Logger.h"
61#include "web/http/http_utils.h"
66#include <system_error>
71#define to_hex_str(int_val) (static_cast<std::ostringstream const&>(std::ostringstream() << std::uppercase << std::hex << int_val)).str()
73namespace web::http::client {
105 delete requestCommand;
127 delete requestCommand;
138 set("X-Powered-By", "snode.c");
142 set("Host", hostFieldValue
);
148 const std::map<std::string, std::string>::iterator it =
headers.find(field);
151 set(field
, it->second.append(
", ").append(value)
);
160 if (!value.empty()) {
162 headers.insert_or_assign(field, value);
164 headers.insert({field, value});
176 headers.erase(
"Transfer-Encoding");
180 headers.erase(
"Content-Length");
206 for (
const auto& [field, value] : headers) {
214 headers.insert({
"Content-Type", type});
226 for (
const auto& [name, value] : cookies) {
240 if (!value.empty()) {
242 trailer.insert_or_assign(field, value);
244 trailer.insert({field, value});
246 if (!
headers.contains(
"Trailer")) {
249 headers[
"Trailer"].append(
"," + field);
260 <<
" HTTP: Response parse error: " << request->
method <<
" " << request->
url <<
" "
265 std::size_t chunkLen,
266 const std::function<
void(
const std::shared_ptr<
Request>&,
const std::shared_ptr<
Response>&)>& onResponseReceived,
267 const std::function<
void(
const std::shared_ptr<
Request>&,
const std::string&)>& onResponseParseError) {
275 set("Content-Type", "application/octet-stream", false);
292 const std::function<
void(
const std::shared_ptr<
Request>&,
const std::shared_ptr<
Response>&)>& onResponseReceived,
293 const std::function<
void(
const std::shared_ptr<
Request>&,
const std::string&)>& onResponseParseError) {
294 if (!chunk.empty()) {
295 set("Content-Type", "text/html; charset=utf-8", false);
298 return send(chunk.data()
, chunk.size()
, onResponseReceived
, onResponseParseError
);
303 const std::string& protocols,
304 const std::function<
void(
const std::shared_ptr<
Request>&,
bool)>& onUpgradeInitiate,
305 const std::function<
void(
const std::shared_ptr<
Request>&,
const std::shared_ptr<
Response>&,
bool)>& onResponseReceived,
306 const std::function<
void(
const std::shared_ptr<
Request>&,
const std::string&)>& onResponseParseError) {
315 LOG(DEBUG) << connectionName <<
" HTTP: Response to upgrade request: " << req->
method <<
" " << req->
url <<
" " <<
"HTTP/"
319 req->
upgrade(res
, [req, res, connectionName, &onResponseReceived](
const std::string& name) {
320 LOG(DEBUG) << connectionName <<
" HTTP: Upgrade bootstrap " << (!name.empty() ?
"success" :
"failed");
321 LOG(DEBUG) <<
" Protocol selected: " << name;
322 LOG(DEBUG) <<
" requested: " << req->
header("upgrade");
323 LOG(DEBUG) <<
" Subprotocol selected: " << res->
get("Sec-WebSocket-Protocol");
324 LOG(DEBUG) <<
" requested: " << req->
header("Sec-WebSocket-Protocol");
326 onResponseReceived(req, res, !name.empty());
338 void Request::
upgrade(
const std::shared_ptr<
Response>& response,
const std::function<
void(
const std::string&)>& status) {
344 if (response !=
nullptr) {
349 if (socketContextUpgradeFactory !=
nullptr) {
350 name = socketContextUpgradeFactory
->name();
352 LOG(DEBUG) << connectionName <<
" HTTP upgrade: SocketContextUpgradeFactory create success for: " << name;
357 if (socketContextUpgrade !=
nullptr) {
358 LOG(DEBUG) << connectionName <<
" HTTP upgrade: SocketContextUpgrade create success for: " << name;
362 LOG(DEBUG) << connectionName <<
" HTTP upgrade: SocketContextUpgrade create failed for: " << name;
367 LOG(DEBUG) << connectionName
368 <<
" HTTP upgrade: SocketContextUpgradeFactory not supported by server: " <<
header("upgrade");
373 LOG(DEBUG) << connectionName <<
" HTTP upgrade: No upgrade requested";
378 LOG(ERROR) << connectionName <<
" HTTP upgrade: Response has gone away";
383 LOG(ERROR) << connectionName <<
" HTTP upgrade: Unexpected disconnect";
390 const std::function<
void(
int)>& onStatus,
391 const std::function<
void(
const std::shared_ptr<
Request>&,
const std::shared_ptr<
Response>&)>& onResponseReceived,
392 const std::function<
void(
const std::shared_ptr<
Request>&,
const std::string&)>& onResponseParseError) {
432 const std::function<
void(
const std::shared_ptr<
Request>&,
const std::string&)>& onResponseParseError) {
457 const bool atomarCommand = requestCommand
->execute(request
);
459 atomar = atomarCommand;
465 delete requestCommand;
479 std::string absolutFileName = file;
481 if (std::filesystem::exists(absolutFileName)) {
483 absolutFileName = std::filesystem::canonical(absolutFileName);
485 if (std::filesystem::is_regular_file(absolutFileName, ec) && !ec) {
497 set("Transfer-Encoding", "chunked");
499 set("Content-Length", std::to_string(std::filesystem::file_size(absolutFileName) +
contentLength)
);
518 bool Request::
executeUpgrade(
const std::string& url,
const std::string& protocols,
const std::function<
void(
bool)>& onStatus) {
522 set("Connection", "Upgrade", true);
523 set("Upgrade", protocols
, true);
528 if (socketContextUpgradeFactory !=
nullptr) {
529 LOG(DEBUG) << connectionName <<
" HTTP: "
530 <<
"SocketContextUpgradeFactory create success: " << socketContextUpgradeFactory
->name();
531 LOG(DEBUG) << connectionName <<
" HTTP: Initiating upgrade: " <<
method <<
" " << url
535 LOG(DEBUG) << connectionName <<
" HTTP: "
536 <<
"SocketContextUpgradeFactory create failed: " << protocols;
537 LOG(DEBUG) << connectionName <<
" HTTP: Not initiating upgrade " <<
method <<
" " << url
541 LOG(DEBUG) << connectionName <<
" HTTP: Upgrade request:\n"
548 std::vector<
char>()
);
550 onStatus(socketContextUpgradeFactory !=
nullptr);
552 if (socketContextUpgradeFactory !=
nullptr) {
568 const std::string httpVersion =
"HTTP/" + std::to_string(
httpMajor) +
"." + std::to_string(
httpMinor);
570 std::string queryString;
573 for (
auto& [key, value] :
queries) {
576 queryString.pop_back();
586 for (
const auto& [field, value] :
headers) {
590 for (
const auto& [name, value] :
cookies) {
634 for (
auto& [field, value] :
trailer) {
static FileReader * open(const std::string &path)
void pipe(Sink *sink, const std::function< void(int)> &callback)
void sendToPeer(const std::string &data) const
const std::string & getConnectionName() const
SocketConnection * getSocketConnection() const
virtual void switchSocketContext(SocketContext *newSocketContext)
void sendToPeer(const char *chunk, std::size_t chunkLen) const final
bool streamToPeer(core::pipe::Source *source) const
static std::string contentType(const std::string &file)
core::socket::stream::SocketContext * create(core::socket::stream::SocketConnection *socketConnection) final
virtual std::string name()=0
virtual bool execute(const std::shared_ptr< Request > &request)=0
CiStringMap< std::string > headers
Request & set(const std::string &field, const std::string &value, bool overwrite=true)
TransferEncoding transferEncoding
ConnectionState connectionState
std::size_t contentLengthSent
void onSourceEof() override
CiStringMap< std::string > cookies
Request & cookie(const std::string &name, const std::string &value)
bool upgrade(const std::string &url, const std::string &protocols, const std::function< void(const std::shared_ptr< Request > &, bool)> &onUpgradeInitiate, const std::function< void(const std::shared_ptr< Request > &, const std::shared_ptr< Response > &, bool)> &onResponseReceived, const std::function< void(const std::shared_ptr< Request > &, const std::string &)> &onResponseParseError=responseParseError)
Request(Request &&) noexcept
CiStringMap< std::string > trailer
void onSourceError(int errnum) override
std::string header(const std::string &field)
std::function< void(const std::shared_ptr< Request > &, const std::shared_ptr< Response > &)> onResponseReceived
Request & sendFragment(const std::string &data)
bool sendFile(const std::string &file, const std::function< void(int errnum)> &onStatus, const std::function< void(const std::shared_ptr< Request > &, const std::shared_ptr< Response > &)> &onResponseReceived, const std::function< void(const std::shared_ptr< Request > &, const std::string &)> &onResponseParseError=responseParseError)
Request & query(const std::string &key, const std::string &value)
void onSourceData(const char *chunk, std::size_t chunkLen) override
Request(SocketContext *socketContext, const std::string &host)
bool executeSendFragment(const char *chunk, std::size_t chunkLen)
const CiStringMap< std::string > & getQueries() const
const CiStringMap< std::string > & getCookies() const
Request & setTrailer(const std::string &field, const std::string &value, bool overwrite=true)
Request & cookie(const std::map< std::string, std::string > &cookies)
SocketContext * getSocketContext() const
bool initiate(const std::shared_ptr< Request > &request)
static void responseParseError(const std::shared_ptr< Request > &request, const std::string &message)
void setMasterRequest(const std::shared_ptr< Request > &masterRequest)
std::function< void(const std::shared_ptr< Request > &request, const std::string &message)> onResponseParseError
CiStringMap< std::string > queries
Request & set(const std::map< std::string, std::string > &headers, bool overwrite=true)
std::size_t contentLength
web::http::client::SocketContext * socketContext
std::string hostFieldValue
Request & append(const std::string &field, const std::string &value)
Request & sendFragment(const char *chunk, std::size_t chunkLen)
std::list< RequestCommand * > requestCommands
void deliverResponse(const std::shared_ptr< Request > &request, const std::shared_ptr< Response > &response)
bool send(const char *chunk, std::size_t chunkLen, const std::function< void(const std::shared_ptr< Request > &, const std::shared_ptr< Response > &)> &onResponseReceived, const std::function< void(const std::shared_ptr< Request > &, const std::string &)> &onResponseParseError=responseParseError)
const CiStringMap< std::string > & getHeaders() const
Request & host(const std::string &hostFieldValue)
bool send(const std::string &chunk, const std::function< void(const std::shared_ptr< Request > &, const std::shared_ptr< Response > &)> &onResponseReceived, const std::function< void(const std::shared_ptr< Request > &, const std::string &)> &onResponseParseError=responseParseError)
std::weak_ptr< Request > masterRequest
Request & type(const std::string &type)
bool end(const std::function< void(const std::shared_ptr< Request > &, const std::shared_ptr< Response > &)> &onResponseReceived, const std::function< void(const std::shared_ptr< Request > &, const std::string &)> &onResponseParseError=responseParseError)
bool executeUpgrade(const std::string &url, const std::string &protocols, const std::function< void(bool)> &onStatus)
void deliverResponseParseError(const std::shared_ptr< Request > &request, const std::string &message)
void upgrade(const std::shared_ptr< Response > &response, const std::function< void(const std::string &)> &status)
bool executeSendFile(const std::string &file, const std::function< void(int)> &onStatus)
void onSourceConnect(core::pipe::Source *source) override
CiStringMap< std::string > headers
CiStringMap< CookieOptions > cookies
const std::string & get(const std::string &key, int i=0) const
SocketContextUpgradeFactory * select(Request &req, Response &res) override
static SocketContextUpgradeFactorySelector * instance()
SocketContextUpgradeFactory * select(const std::string &protocols, Request &req)
void checkRefCount() final
void requestPrepared(Request &&request)
void requestDelivered(Request &&request, bool success)
SendFileCommand(const std::string &file, const std::function< void(int)> &onStatus)
SendFragmentCommand(const char *chunk, std::size_t chunkLen)
SendHeaderCommand()=default
UpgradeCommand(const std::string &url, const std::string &protocols, const std::function< void(const std::shared_ptr< Request > &, bool)> &onUpgradeInitiate)
std::string to_http_date(struct tm *tm)
std::string toString(const std::string &version, const std::string &statusCode, const std::string &reason, const web::http::CiStringMap< std::string > &header, const web::http::CiStringMap< web::http::CookieOptions > &cookies, const std::vector< char > &body)
std::string url_encode(const std::string &text)
std::string file_mod_http_date(const std::string &filePath)
std::string toString(const std::string &method, const std::string &url, const std::string &version, const web::http::CiStringMap< std::string > &queries, const web::http::CiStringMap< std::string > &header, const web::http::CiStringMap< std::string > &cookies, const std::vector< char > &body)
bool ciEquals(const std::string &str1, const std::string &str2)
bool ciContains(const std::string &str1, const std::string &str2)
#define to_hex_str(int_val)