ARCHIVE/rippled
1
0
Fork 0
mirror of https://github.com/XRPLF/rippled.git synced 2025-12-06 17:27:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

Correctly handle connections that fail security checks (RIPD-1114):

Browse Source 
* Return error code 400 to the peer along with a descriptive message
* Release the slot and decrement IP connection counters.
This commit is contained in:
Nik Bougalis
2016-04-21 15:03:04 -07:00
parent 5e5d5fdee4
commit b5dbd7942f
5 changed files with 65 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
src/ripple/server/SimpleWriter.h
View File

@@ -21,30 +21,29 @@
#define RIPPLE_SERVER_SIMPLEWRITER_H_INCLUDED
#include <ripple/server/Writer.h>
#include <beast/asio/streambuf.h>
#include <beast/http/message.h>
#include <ripple/beast/deprecated_http.h>
#include <beast/streambuf.hpp>
#include <utility>
namespace ripple {
namespace HTTP {
/** Writer that sends a simple HTTP response with a message body. */
class SimpleWriter : public Writer
{
private:
beast::http::message message_;
beast::asio::streambuf streambuf_;
beast::deprecated_http::message message_;
beast::streambuf streambuf_;
std::string body_;
bool prepared_ = false;
public:
explicit
SimpleWriter(beast::http::message&& message)
: message_(std::forward<beast::http::message>(message))
SimpleWriter(beast::deprecated_http::message&& message)
: message_(std::forward<beast::deprecated_http::message>(message))
{
}
beast::http::message&
beast::deprecated_http::message&
message()
{
return message_;
@@ -95,14 +94,13 @@ private:
{
prepared_ = true;
message_.headers.erase("Content-Length");
message_.headers.append("Content-Length",
message_.headers.insert("Content-Length",
std::to_string(body_.size()));
write(streambuf_, message_);
write(streambuf_, body_;
beast::http::detail::write(streambuf_, body_);
}
};
}
}
#endif