mirror of
https://github.com/XRPLF/rippled.git
synced 2025-11-19 18:45:52 +00:00
90 lines
2.6 KiB
C++
90 lines
2.6 KiB
C++
//------------------------------------------------------------------------------
|
|
/*
|
|
This file is part of rippled: https://github.com/ripple/rippled
|
|
Copyright (c) 2012, 2013 Ripple Labs Inc.
|
|
|
|
Permission to use, copy, modify, and/or distribute this software for any
|
|
purpose with or without fee is hereby granted, provided that the above
|
|
copyright notice and this permission notice appear in all copies.
|
|
|
|
THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
|
|
WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
|
|
MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
|
|
ANY SPECIAL , DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
|
|
WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
|
|
ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
|
|
OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
|
|
*/
|
|
//==============================================================================
|
|
|
|
#ifndef RIPPLE_RPC_WSINFOSUB_H
|
|
#define RIPPLE_RPC_WSINFOSUB_H
|
|
|
|
#include <xrpld/rpc/InfoSub.h>
|
|
#include <xrpld/rpc/Role.h>
|
|
|
|
#include <xrpl/beast/net/IPAddressConversion.h>
|
|
#include <xrpl/json/json_writer.h>
|
|
#include <xrpl/server/WSSession.h>
|
|
|
|
#include <memory>
|
|
#include <string>
|
|
|
|
namespace ripple {
|
|
|
|
class WSInfoSub : public InfoSub
|
|
{
|
|
std::weak_ptr<WSSession> ws_;
|
|
std::string user_;
|
|
std::string fwdfor_;
|
|
|
|
public:
|
|
WSInfoSub(Source& source, std::shared_ptr<WSSession> const& ws)
|
|
: InfoSub(source), ws_(ws)
|
|
{
|
|
auto const& h = ws->request();
|
|
if (ipAllowed(
|
|
beast::IPAddressConversion::from_asio(ws->remote_endpoint())
|
|
.address(),
|
|
ws->port().secure_gateway_nets_v4,
|
|
ws->port().secure_gateway_nets_v6))
|
|
{
|
|
auto it = h.find("X-User");
|
|
if (it != h.end())
|
|
user_ = it->value();
|
|
fwdfor_ = std::string(forwardedFor(h));
|
|
}
|
|
}
|
|
|
|
std::string_view
|
|
user() const
|
|
{
|
|
return user_;
|
|
}
|
|
|
|
std::string_view
|
|
forwarded_for() const
|
|
{
|
|
return fwdfor_;
|
|
}
|
|
|
|
void
|
|
send(Json::Value const& jv, bool) override
|
|
{
|
|
auto sp = ws_.lock();
|
|
if (!sp)
|
|
return;
|
|
boost::beast::multi_buffer sb;
|
|
Json::stream(jv, [&](void const* data, std::size_t n) {
|
|
sb.commit(boost::asio::buffer_copy(
|
|
sb.prepare(n), boost::asio::buffer(data, n)));
|
|
});
|
|
auto m = std::make_shared<StreambufWSMsg<decltype(sb)>>(std::move(sb));
|
|
sp->send(m);
|
|
}
|
|
};
|
|
|
|
} // namespace ripple
|
|
|
|
#endif
|