StreamReceiver.hxx

Go to the documentation of this file.

 
#ifndef _OPENLCB_STREAMRECEIVER_HXX_
#define _OPENLCB_STREAMRECEIVER_HXX_
 
#include "openlcb/StreamReceiverInterface.hxx"
 
#include "openlcb/IfCan.hxx"
#include "openlcb/StreamDefs.hxx"
#include "utils/ByteBuffer.hxx"
#include "utils/LimitedPool.hxx"
 
namespace openlcb
{
 
class StreamReceiverCan : public StreamReceiverInterface
{
public:
    StreamReceiverCan(IfCan *interface, uint8_t local_stream_id);
 
    ~StreamReceiverCan();
 
    void send(Buffer<StreamReceiveRequest> *msg, unsigned prio = 0) override;
 
    void cancel_request() override;
    
private:
    void announced_stream();
 
    Action entry() override
    {
        return return_ok();
    }
 
    Action wait_for_wakeup()
    {
        if (pendingCancel_)
        {
            return call_immediately(STATE(wakeup));
        }
        isWaiting_ = 1;
        return wait_and_call(STATE(wakeup));
    }
 
    Action wakeup();
 
    Action init_reply();
    Action init_buffer_ready();
 
    Action window_reached();
    Action have_raw_buffer();
 
    void handle_stream_initiate(Buffer<GenMessage> *message);
 
    inline void handle_bytes_received(const uint8_t *data, size_t len);
 
    void handle_stream_complete(Buffer<GenMessage> *message);
 
    void unregister_handlers();
    
    IfCan *if_can()
    {
        return static_cast<IfCan *>(service());
    }
 
    Node *node()
    {
        return request()->dst_;
    }
 
    MessageHandler::GenericHandler streamInitiateHandler_ {
        this, &StreamReceiverCan::handle_stream_initiate};
 
    class StreamDataHandler;
    friend class StreamDataHandler;
 
    MessageHandler::GenericHandler streamCompleteHandler_ {
        this, &StreamReceiverCan::handle_stream_complete};
 
    LimitedPool lastBufferPool_ {sizeof(RawBuffer), 2, rawBufferPool};
 
    ByteBufferPtr currentBuffer_;
 
    RawBufferPtr lastBuffer_;
 
    std::unique_ptr<StreamDataHandler> dataHandler_;
 
    size_t totalByteCount_;
 
    uint16_t streamWindowRemaining_;
 
    const uint8_t assignedStreamId_;
 
    uint8_t streamClosed_ : 1;
    uint8_t pendingInit_ : 1;
    uint8_t pendingCancel_ : 1;
    uint8_t isWaiting_ : 1;
}; // class StreamReceiver
 
} // namespace openlcb
 
#endif // _OPENLCB_STREAMRECEIVER_HXX_