RLPXSocketIO.h

Go to the documentation of this file.

/*
 This file is part of cpp-ethereum.
 
 cpp-ethereum is free software: you can redistribute it and/or modify
 it under the terms of the GNU General Public License as published by
 the Free Software Foundation, either version 3 of the License, or
 (at your option) any later version.
 
 cpp-ethereum is distributed in the hope that it will be useful,
 but WITHOUT ANY WARRANTY; without even the implied warranty of
 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 GNU General Public License for more details.
 
 You should have received a copy of the GNU General Public License
 along with cpp-ethereum.  If not, see <http://www.gnu.org/licenses/>.
 */
#pragma once

#include "RLPXFrameWriter.h"
namespace ba = boost::asio;
namespace bi = boost::asio::ip;

namespace dev
{
namespace p2p
{

class RLPXSocketIO: public std::enable_shared_from_this<RLPXSocketIO>
{
public:
        static uint32_t const MinFrameSize;
        static uint32_t const MaxPacketSize;
        static uint16_t const DefaultInitialCapacity;
        
        RLPXSocketIO(unsigned _protCount, RLPXFrameCoder& _coder, bi::tcp::socket& _socket, bool _flowControl = true, size_t _initialCapacity = DefaultInitialCapacity);

        void send(unsigned _protocolType, unsigned _type, RLPStream& _payload);
        
        void doWrite();
        
private:
        static std::vector<RLPXFrameWriter> writers(unsigned _capacity);
        
        void deferWrite();
        
        void write(size_t _dequed);
        
        bool const m_flowControl;               
        
        RLPXFrameCoder& m_coder;                
        bi::tcp::socket& m_socket;

        std::deque<bytes> m_toSend;     
        
        std::vector<RLPXFrameWriter> m_writers;                 
        std::unique_ptr<ba::deadline_timer> m_congestion;       

        Mutex x_queued;
        unsigned m_queued = 0;  
        uint32_t m_egressCapacity;
};

}
}