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

adjusted how valid hosts are stored

Browse Source
This commit is contained in:
Peter Thorson
2011-09-08 08:28:56 -05:00
parent 882d243bd5
commit 792b30f291
7 changed files with 71 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
src/websocket_session.cpp
View File

@@ -39,6 +39,13 @@
using websocketpp::session;
session::session (boost::asio::io_service& io_service,
connection_handler_ptr defc)
: m_socket(io_service),
m_status(CONNECTING),
m_http_error_code(0),
m_local_interface(defc) {}
tcp::socket& session::socket() {
return m_socket;
}
@@ -68,6 +75,13 @@ void session::set_handler(connection_handler_ptr new_con) {
m_local_interface->connect(shared_from_this());
}
void session::add_host(std::string host) {
m_hosts.insert(host);
}
void session::remove_host(std::string host) {
m_hosts.erase(host);
}
std::string session::get_header(const std::string& key) const {
std::map<std::string,std::string>::const_iterator h = m_headers.find(key);
@@ -255,9 +269,10 @@ void session::handle_read_handshake(const boost::system::error_code& e,
// TODO: use exceptions or a helper function or something better here
// verify the presence of required headers
if (get_header("Host") != m_host) {
std::cerr << "Invalid or missing Host header. " << get_header("Host") << "!=" << m_host << std::endl;
// verify the presence of required headers
if (m_hosts.find(get_header("Host")) == m_hosts.end()) {
std::cerr << "Invalid or missing Host header: "
<< get_header("Host") << std::endl;
this->set_http_error(400);
}
if (!boost::iequals(get_header("Upgrade"),"websocket")) {