Further fix of #246 : corrected use of response streambuf after reconnect (wrong response object was used in Session::callback). Also closes connection when appropriate, and simplified Session::callback.
This commit is contained in:
eidheim
parent
a316310815
commit
dbd2287418
2 changed files with 76 additions and 72 deletions
|
|
@ -130,7 +130,7 @@ namespace SimpleWeb {
|
|||
std::shared_ptr<Connection> connection;
|
||||
std::unique_ptr<asio::streambuf> request_streambuf;
|
||||
std::shared_ptr<Response> response;
|
||||
std::function<void(const std::shared_ptr<Connection> &, const error_code &)> callback;
|
||||
std::function<void(const error_code &)> callback;
|
||||
};
|
||||
|
||||
public:
|
||||
|
|
@ -206,17 +206,18 @@ namespace SimpleWeb {
|
|||
void request(const std::string &method, const std::string &path, string_view content, const CaseInsensitiveMultimap &header,
|
||||
std::function<void(std::shared_ptr<Response>, const error_code &)> &&request_callback_) {
|
||||
auto session = std::make_shared<Session>(config.max_response_streambuf_size, get_connection(), create_request_header(method, path, header));
|
||||
auto response = session->response;
|
||||
std::weak_ptr<Session> session_weak(session); // To avoid keeping session alive longer than needed
|
||||
auto request_callback = std::make_shared<std::function<void(std::shared_ptr<Response>, const error_code &)>>(std::move(request_callback_));
|
||||
session->callback = [this, response, request_callback](const std::shared_ptr<Connection> &connection, const error_code &ec) {
|
||||
{
|
||||
session->callback = [this, session_weak, request_callback](const error_code &ec) {
|
||||
if(auto session = session_weak.lock()) {
|
||||
if(session->connection) {
|
||||
std::lock_guard<std::mutex> lock(this->connections_mutex);
|
||||
connection->in_use = false;
|
||||
session->connection->in_use = false;
|
||||
|
||||
// Remove unused connections, but keep one open for HTTP persistent connection:
|
||||
std::size_t unused_connections = 0;
|
||||
for(auto it = this->connections.begin(); it != this->connections.end();) {
|
||||
if(ec && connection == *it)
|
||||
if(ec && session->connection == *it)
|
||||
it = this->connections.erase(it);
|
||||
else if((*it)->in_use)
|
||||
++it;
|
||||
|
|
@ -231,7 +232,8 @@ namespace SimpleWeb {
|
|||
}
|
||||
|
||||
if(*request_callback)
|
||||
(*request_callback)(response, ec);
|
||||
(*request_callback)(session->response, ec);
|
||||
}
|
||||
};
|
||||
|
||||
std::ostream write_stream(session->request_streambuf.get());
|
||||
|
|
@ -271,17 +273,18 @@ namespace SimpleWeb {
|
|||
void request(const std::string &method, const std::string &path, std::istream &content, const CaseInsensitiveMultimap &header,
|
||||
std::function<void(std::shared_ptr<Response>, const error_code &)> &&request_callback_) {
|
||||
auto session = std::make_shared<Session>(config.max_response_streambuf_size, get_connection(), create_request_header(method, path, header));
|
||||
auto response = session->response;
|
||||
std::weak_ptr<Session> session_weak(session); // To avoid keeping session alive longer than needed
|
||||
auto request_callback = std::make_shared<std::function<void(std::shared_ptr<Response>, const error_code &)>>(std::move(request_callback_));
|
||||
session->callback = [this, response, request_callback](const std::shared_ptr<Connection> &connection, const error_code &ec) {
|
||||
{
|
||||
session->callback = [this, session_weak, request_callback](const error_code &ec) {
|
||||
if(auto session = session_weak.lock()) {
|
||||
if(session->connection) {
|
||||
std::lock_guard<std::mutex> lock(this->connections_mutex);
|
||||
connection->in_use = false;
|
||||
session->connection->in_use = false;
|
||||
|
||||
// Remove unused connections, but keep one open for HTTP persistent connection:
|
||||
std::size_t unused_connections = 0;
|
||||
for(auto it = this->connections.begin(); it != this->connections.end();) {
|
||||
if(ec && connection == *it)
|
||||
if(ec && session->connection == *it)
|
||||
it = this->connections.erase(it);
|
||||
else if((*it)->in_use)
|
||||
++it;
|
||||
|
|
@ -296,7 +299,8 @@ namespace SimpleWeb {
|
|||
}
|
||||
|
||||
if(*request_callback)
|
||||
(*request_callback)(response, ec);
|
||||
(*request_callback)(session->response, ec);
|
||||
}
|
||||
};
|
||||
|
||||
content.seekg(0, std::ios::end);
|
||||
|
|
@ -442,7 +446,7 @@ namespace SimpleWeb {
|
|||
if(!ec)
|
||||
this->read(session);
|
||||
else
|
||||
session->callback(session->connection, ec);
|
||||
session->callback(ec);
|
||||
});
|
||||
}
|
||||
|
||||
|
|
@ -454,7 +458,7 @@ namespace SimpleWeb {
|
|||
if(!lock)
|
||||
return;
|
||||
if((!ec || ec == asio::error::not_found) && session->response->streambuf.size() == session->response->streambuf.max_size()) {
|
||||
session->callback(session->connection, make_error_code::make_error_code(errc::message_size));
|
||||
session->callback(make_error_code::make_error_code(errc::message_size));
|
||||
return;
|
||||
}
|
||||
if(!ec) {
|
||||
|
|
@ -462,7 +466,7 @@ namespace SimpleWeb {
|
|||
std::size_t num_additional_bytes = session->response->streambuf.size() - bytes_transferred;
|
||||
|
||||
if(!ResponseMessage::parse(session->response->content, session->response->http_version, session->response->status_code, session->response->header)) {
|
||||
session->callback(session->connection, make_error_code::make_error_code(errc::protocol_error));
|
||||
session->callback(make_error_code::make_error_code(errc::protocol_error));
|
||||
return;
|
||||
}
|
||||
|
||||
|
|
@ -478,17 +482,17 @@ namespace SimpleWeb {
|
|||
return;
|
||||
if(!ec) {
|
||||
if(session->response->streambuf.size() == session->response->streambuf.max_size()) {
|
||||
session->callback(session->connection, make_error_code::make_error_code(errc::message_size));
|
||||
session->callback(make_error_code::make_error_code(errc::message_size));
|
||||
return;
|
||||
}
|
||||
session->callback(session->connection, ec);
|
||||
session->callback(ec);
|
||||
}
|
||||
else
|
||||
session->callback(session->connection, ec);
|
||||
session->callback(ec);
|
||||
});
|
||||
}
|
||||
else
|
||||
session->callback(session->connection, ec);
|
||||
session->callback(ec);
|
||||
}
|
||||
else if((header_it = session->response->header.find("Transfer-Encoding")) != session->response->header.end() && header_it->second == "chunked") {
|
||||
auto chunks_streambuf = std::make_shared<asio::streambuf>(this->config.max_response_streambuf_size);
|
||||
|
|
@ -501,19 +505,19 @@ namespace SimpleWeb {
|
|||
auto lock = session->connection->handler_runner->continue_lock();
|
||||
if(!lock)
|
||||
return;
|
||||
session->connection = nullptr; // Disconnect
|
||||
if(!ec) {
|
||||
if(session->response->streambuf.size() == session->response->streambuf.max_size()) {
|
||||
session->callback(session->connection, make_error_code::make_error_code(errc::message_size));
|
||||
return;
|
||||
}
|
||||
session->callback(session->connection, ec);
|
||||
if(session->response->streambuf.size() == session->response->streambuf.max_size())
|
||||
session->callback(make_error_code::make_error_code(errc::message_size));
|
||||
else
|
||||
session->callback(ec);
|
||||
}
|
||||
else
|
||||
session->callback(session->connection, ec == asio::error::eof ? error_code() : ec);
|
||||
session->callback(ec == asio::error::eof ? error_code() : ec);
|
||||
});
|
||||
}
|
||||
else
|
||||
session->callback(session->connection, ec);
|
||||
session->callback(ec);
|
||||
}
|
||||
else {
|
||||
if(session->connection->attempt_reconnect && ec != asio::error::operation_aborted) {
|
||||
|
|
@ -531,11 +535,11 @@ namespace SimpleWeb {
|
|||
}
|
||||
else {
|
||||
lock.unlock();
|
||||
session->callback(session->connection, ec);
|
||||
session->callback(ec);
|
||||
}
|
||||
}
|
||||
else
|
||||
session->callback(session->connection, ec);
|
||||
session->callback(ec);
|
||||
}
|
||||
});
|
||||
}
|
||||
|
|
@ -548,7 +552,7 @@ namespace SimpleWeb {
|
|||
if(!lock)
|
||||
return;
|
||||
if((!ec || ec == asio::error::not_found) && session->response->streambuf.size() == session->response->streambuf.max_size()) {
|
||||
session->callback(session->connection, make_error_code::make_error_code(errc::message_size));
|
||||
session->callback(make_error_code::make_error_code(errc::message_size));
|
||||
return;
|
||||
}
|
||||
if(!ec) {
|
||||
|
|
@ -561,7 +565,7 @@ namespace SimpleWeb {
|
|||
length = stoul(line, 0, 16);
|
||||
}
|
||||
catch(...) {
|
||||
session->callback(session->connection, make_error_code::make_error_code(errc::protocol_error));
|
||||
session->callback(make_error_code::make_error_code(errc::protocol_error));
|
||||
return;
|
||||
}
|
||||
|
||||
|
|
@ -576,20 +580,20 @@ namespace SimpleWeb {
|
|||
return;
|
||||
if(!ec) {
|
||||
if(session->response->streambuf.size() == session->response->streambuf.max_size()) {
|
||||
session->callback(session->connection, make_error_code::make_error_code(errc::message_size));
|
||||
session->callback(make_error_code::make_error_code(errc::message_size));
|
||||
return;
|
||||
}
|
||||
this->read_chunked_transfer_encoded_chunk(session, chunks_streambuf, length);
|
||||
}
|
||||
else
|
||||
session->callback(session->connection, ec);
|
||||
session->callback(ec);
|
||||
});
|
||||
}
|
||||
else
|
||||
this->read_chunked_transfer_encoded_chunk(session, chunks_streambuf, length);
|
||||
}
|
||||
else
|
||||
session->callback(session->connection, ec);
|
||||
session->callback(ec);
|
||||
});
|
||||
}
|
||||
|
||||
|
|
@ -600,7 +604,7 @@ namespace SimpleWeb {
|
|||
session->response->content.read(buffer.get(), static_cast<std::streamsize>(length));
|
||||
tmp_stream.write(buffer.get(), static_cast<std::streamsize>(length));
|
||||
if(chunks_streambuf->size() == chunks_streambuf->max_size()) {
|
||||
session->callback(session->connection, make_error_code::make_error_code(errc::message_size));
|
||||
session->callback(make_error_code::make_error_code(errc::message_size));
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
|
@ -617,7 +621,7 @@ namespace SimpleWeb {
|
|||
ostream << chunks_streambuf.get();
|
||||
}
|
||||
error_code ec;
|
||||
session->callback(session->connection, ec);
|
||||
session->callback(ec);
|
||||
}
|
||||
}
|
||||
};
|
||||
|
|
@ -660,11 +664,11 @@ namespace SimpleWeb {
|
|||
this->write(session);
|
||||
}
|
||||
else
|
||||
session->callback(session->connection, ec);
|
||||
session->callback(ec);
|
||||
});
|
||||
}
|
||||
else
|
||||
session->callback(session->connection, ec);
|
||||
session->callback(ec);
|
||||
});
|
||||
}
|
||||
else
|
||||
|
|
|
|||
|
|
@ -84,36 +84,36 @@ namespace SimpleWeb {
|
|||
if(!lock)
|
||||
return;
|
||||
if((!ec || ec == asio::error::not_found) && response->streambuf.size() == response->streambuf.max_size()) {
|
||||
session->callback(session->connection, make_error_code::make_error_code(errc::message_size));
|
||||
session->callback(make_error_code::make_error_code(errc::message_size));
|
||||
return;
|
||||
}
|
||||
if(!ec) {
|
||||
if(!ResponseMessage::parse(response->content, response->http_version, response->status_code, response->header))
|
||||
session->callback(session->connection, make_error_code::make_error_code(errc::protocol_error));
|
||||
session->callback(make_error_code::make_error_code(errc::protocol_error));
|
||||
else {
|
||||
if(response->status_code.compare(0, 3, "200") != 0)
|
||||
session->callback(session->connection, make_error_code::make_error_code(errc::permission_denied));
|
||||
session->callback(make_error_code::make_error_code(errc::permission_denied));
|
||||
else
|
||||
this->handshake(session);
|
||||
}
|
||||
}
|
||||
else
|
||||
session->callback(session->connection, ec);
|
||||
session->callback(ec);
|
||||
});
|
||||
}
|
||||
else
|
||||
session->callback(session->connection, ec);
|
||||
session->callback(ec);
|
||||
});
|
||||
}
|
||||
else
|
||||
this->handshake(session);
|
||||
}
|
||||
else
|
||||
session->callback(session->connection, ec);
|
||||
session->callback(ec);
|
||||
});
|
||||
}
|
||||
else
|
||||
session->callback(session->connection, ec);
|
||||
session->callback(ec);
|
||||
});
|
||||
}
|
||||
else
|
||||
|
|
@ -132,7 +132,7 @@ namespace SimpleWeb {
|
|||
if(!ec)
|
||||
this->write(session);
|
||||
else
|
||||
session->callback(session->connection, ec);
|
||||
session->callback(ec);
|
||||
});
|
||||
}
|
||||
};
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue