ARCHIVE/rippled
1
0
Fork 0
mirror of https://github.com/XRPLF/rippled.git synced 2026-06-02 16:26:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

cleaned up session, added better protocol version identification

Browse Source
This commit is contained in:
Peter Thorson
2011-09-09 20:37:59 -05:00
parent 8129c30f69
commit 56485f543a
4 changed files with 41 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
src/websocket_session.cpp
View File

@@ -83,6 +83,13 @@ void session::remove_host(std::string host) {
m_hosts.erase(host);
}
void session::set_max_message_size(uint64_t val) {
if (val > frame::PAYLOAD_64BIT_LIMIT) {
throw "illegal maximum message size";
}
m_max_message_size = val;
}
std::string session::get_header(const std::string& key) const {
std::map<std::string,std::string>::const_iterator h = m_headers.find(key);
@@ -289,10 +296,15 @@ void session::handle_read_handshake(const boost::system::error_code& e,
this->set_http_error(400);
}
if (get_header("Sec-WebSocket-Version") != "8" &&
get_header("Sec-WebSocket-Version") != "7") {
std::cerr << "Invalid or missing Sec-Websocket-Version header." << std::endl;
if (get_header("Sec-WebSocket-Version") == "" ) {
std::cerr << "Missing Sec-Websocket-Version header." << std::endl;
this->set_http_error(400);
} else {
m_version = strtoul(get_header("Sec-WebSocket-Version").c_str(),NULL,10);
if (m_version != 7 && m_version != 8 && m_version != 13) {
this->set_http_error(400);
}
}
if (m_http_error_code != 0) {