DatagramHandlerDefault.hxx

Go to the documentation of this file.

 
#ifndef _OPENLCB_DATAGRAMHANDLERDEFAULT_HXX_
#define _OPENLCB_DATAGRAMHANDLERDEFAULT_HXX_
 
#include "openlcb/Datagram.hxx"
 
namespace openlcb
{
 
class DefaultDatagramHandler : public DatagramHandlerFlow
{
protected:
    DefaultDatagramHandler(DatagramService *if_datagram)
        : DatagramHandlerFlow(if_datagram)
    {
    }
 
public:
    DatagramService *dg_service()
    {
        return static_cast<DatagramService *>(service());
    }
 
protected:
    Action respond_ok(uint8_t flags)
    {
        prepare_respond_ok(flags);
        return allocate_and_call(
            dg_service()->iface()->addressed_message_write_flow(),
            STATE(send_ok_response));
    }
 
    
    void inline_respond_ok(uint8_t flags)
    {
        prepare_respond_ok(flags);
        auto *b =
            dg_service()->iface()->addressed_message_write_flow()->alloc();
        respond_ok_helper(b);
    }
 
private:
    void prepare_respond_ok(uint8_t flags)
    {
        responseMti_ = Defs::MTI_DATAGRAM_OK;
        responseErrorCode_ = flags;
    }
 
    void respond_ok_helper(Buffer<GenMessage> *b)
    {
        b->data()->reset(responseMti_, message()->data()->dst->node_id(),
            message()->data()->src,
            Payload(1, (char)(responseErrorCode_ & 0xff)));
        dg_service()->iface()->addressed_message_write_flow()->send(b);
    }
 
    Action send_ok_response()
    {
        auto *b = get_allocation_result(
            dg_service()->iface()->addressed_message_write_flow());
        respond_ok_helper(b);
        return call_immediately(STATE(ok_response_sent));
    }
 
protected:
    Action respond_reject(uint16_t error_code)
    {
        responseMti_ = Defs::MTI_DATAGRAM_REJECTED;
        responseErrorCode_ = error_code;
        return allocate_and_call(
            dg_service()->iface()->addressed_message_write_flow(),
            STATE(send_reject_response));
    }
 
private:
    Action send_reject_response()
    {
        auto *b = get_allocation_result(
            dg_service()->iface()->addressed_message_write_flow());
        b->data()->reset(responseMti_, message()->data()->dst->node_id(),
                         message()->data()->src,
                         error_to_buffer(responseErrorCode_));
        dg_service()->iface()->addressed_message_write_flow()->send(b);
        return release_and_exit();
    }
 
protected:
    virtual Action ok_response_sent()
    {
        return release_and_exit();
    }
 
    size_t size()
    {
        return message()->data()->payload.size();
    }
 
    const uint8_t *payload()
    {
        return reinterpret_cast<const uint8_t *>(
            message()->data()->payload.data());
    }
 
private:
    uint16_t responseErrorCode_;
    openlcb::Defs::MTI responseMti_;
};
 
} // namespace
 
#endif // _OPENLCB_DATAGRAMHANDLERDEFAULT_HXX_