Move ./modules to ./src

2025-12-06 17:27:52 +00:00 · 2013-09-11 11:17:22 -07:00
parent 6d828ae476
commit 45eccf2ccf
386 changed files with 1673 additions and 1674 deletions
--- a/src/ripple_app/websocket/WSConnection.cpp
+++ b/src/ripple_app/websocket/WSConnection.cpp
@@ -0,0 +1,180 @@
+//------------------------------------------------------------------------------
+/*
+    Copyright (c) 2011-2013, OpenCoin, Inc.
+*/
+//==============================================================================
+
+SETUP_LOGN (WSConnection, "WSConnection")
+
+//------------------------------------------------------------------------------
+
+WSConnection::WSConnection (InfoSub::Source& source, bool isPublic,
+    std::string const& remoteIP, boost::asio::io_service& io_service)
+    : InfoSub (source)
+    , m_isPublic (isPublic)
+    , m_remoteIP (remoteIP)
+    , m_receiveQueueMutex (this, "WSConnection", __FILE__, __LINE__)
+    , m_netOPs (getApp ().getOPs ())
+    , m_loadSource (m_remoteIP)
+    , m_pingTimer (io_service)
+    , m_sentPing (false)
+    , m_receiveQueueRunning (false)
+    , m_isDead (false)
+{
+    WriteLog (lsDEBUG, WSConnection) << "Websocket connection from " << m_remoteIP;
+}
+
+WSConnection::~WSConnection ()
+{
+}
+
+void WSConnection::onPong (const std::string&)
+{
+    m_sentPing = false;
+}
+
+void WSConnection::rcvMessage (message_ptr msg, bool& msgRejected, bool& runQueue)
+{
+    ScopedLockType sl (m_receiveQueueMutex, __FILE__, __LINE__);
+
+    if (m_isDead)
+    {
+        msgRejected = false;
+        runQueue = false;
+        return;
+    }
+
+    if (m_isDead || (m_receiveQueue.size () >= 1000))
+    {
+        msgRejected = !m_isDead;
+        runQueue = false;
+    }
+    else
+    {
+        msgRejected = false;
+        m_receiveQueue.push_back (msg);
+
+        if (m_receiveQueueRunning)
+            runQueue = false;
+        else
+        {
+            runQueue = true;
+            m_receiveQueueRunning = true;
+        }
+    }
+}
+
+WSConnection::message_ptr WSConnection::getMessage ()
+{
+    ScopedLockType sl (m_receiveQueueMutex, __FILE__, __LINE__);
+
+    if (m_isDead || m_receiveQueue.empty ())
+    {
+        m_receiveQueueRunning = false;
+        return message_ptr ();
+    }
+
+    message_ptr m = m_receiveQueue.front ();
+    m_receiveQueue.pop_front ();
+    return m;
+}
+
+void WSConnection::returnMessage (message_ptr ptr)
+{
+    ScopedLockType sl (m_receiveQueueMutex, __FILE__, __LINE__);
+
+    if (!m_isDead)
+        m_receiveQueue.push_front(ptr);
+}
+
+Json::Value WSConnection::invokeCommand (Json::Value& jvRequest)
+{
+    // VFALCO TODO Make LoadManager a ctor argument
+    if (getApp().getLoadManager ().shouldCutoff (m_loadSource))
+    {
+        // VFALCO TODO This must be implemented before open sourcing
+        #if BEAST_MSVC
+        # pragma message(BEAST_FILEANDLINE_ "Need to implement before open sourcing")
+        #else
+        # warning message("WSConnection.h: Need implementation before open sourcing.")
+        #endif
+
+#if SHOULD_DISCONNECT
+        disconnect ();
+
+        return rpcError (rpcSLOW_DOWN);
+#endif
+    }
+
+    // Requests without "command" are invalid.
+    //
+    if (!jvRequest.isMember ("command"))
+    {
+        Json::Value jvResult (Json::objectValue);
+
+        jvResult["type"]    = "response";
+        jvResult["status"]  = "error";
+        jvResult["error"]   = "missingCommand";
+        jvResult["request"] = jvRequest;
+
+        if (jvRequest.isMember ("id"))
+        {
+            jvResult["id"]  = jvRequest["id"];
+        }
+
+        getApp().getLoadManager ().applyLoadCharge (m_loadSource, LT_RPCInvalid);
+
+        return jvResult;
+    }
+
+    LoadType loadType = LT_RPCReference;
+    RPCHandler  mRPCHandler (&this->m_netOPs, boost::dynamic_pointer_cast<InfoSub> (this->shared_from_this ()));
+    Json::Value jvResult (Json::objectValue);
+
+    Config::Role const role = m_isPublic
+            ? Config::GUEST     // Don't check on the public interface.
+            : getConfig ().getAdminRole (jvRequest, m_remoteIP);
+        
+    if (Config::FORBID == role)
+    {
+        jvResult["result"]  = rpcError (rpcFORBIDDEN);
+    }
+    else
+    {
+        jvResult["result"] = mRPCHandler.doCommand (jvRequest, role, &loadType);
+    }
+
+    // Debit/credit the load and see if we should include a warning.
+    //
+    if (getApp().getLoadManager ().applyLoadCharge (m_loadSource, loadType) &&
+        getApp().getLoadManager ().shouldWarn (m_loadSource))
+    {
+        jvResult["warning"] = "load";
+    }
+
+    // Currently we will simply unwrap errors returned by the RPC
+    // API, in the future maybe we can make the responses
+    // consistent.
+    //
+    // Regularize result. This is duplicate code.
+    if (jvResult["result"].isMember ("error"))
+    {
+        jvResult            = jvResult["result"];
+        jvResult["status"]  = "error";
+        jvResult["request"] = jvRequest;
+
+    }
+    else
+    {
+        jvResult["status"]  = "success";
+    }
+
+    if (jvRequest.isMember ("id"))
+    {
+        jvResult["id"]      = jvRequest["id"];
+    }
+
+    jvResult["type"]        = "response";
+
+    return jvResult;
+}