wamp_rawsocket_transport.hpp

//
// Copyright (c) Crossbar.io Technologies GmbH and contributors
//
// Boost Software License - Version 1.0 - August 17th, 2003
//
// Permission is hereby granted, free of charge, to any person or organization
// obtaining a copy of the software and accompanying documentation covered by
// this license (the "Software") to use, reproduce, display, distribute,
// execute, and transmit the Software, and to prepare derivative works of the
// Software, and to permit third-parties to whom the Software is furnished to
// do so, all subject to the following:
//
// The copyright notices in the Software and this entire statement, including
// the above license grant, this restriction and the following disclaimer,
// must be included in all copies of the Software, in whole or in part, and
// all derivative works of the Software, unless such copies or derivative
// works are solely in the form of machine-executable object code generated by
// a source language processor.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE, TITLE AND NON-INFRINGEMENT. IN NO EVENT
// SHALL THE COPYRIGHT HOLDERS OR ANYONE DISTRIBUTING THE SOFTWARE BE LIABLE
// FOR ANY DAMAGES OR OTHER LIABILITY, WHETHER IN CONTRACT, TORT OR OTHERWISE,
// ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
// DEALINGS IN THE SOFTWARE.
//

#ifndef AUTOBAHN_WAMP_NETWORK_TRANSPORT_HPP
#define AUTOBAHN_WAMP_NETWORK_TRANSPORT_HPP

#include "boost_config.hpp"
#include "wamp_transport.hpp"

#include <boost/asio/io_service.hpp>
#include <cstddef>
#include <memory>
#include <msgpack.hpp>

namespace autobahn {

class wamp_message;
class wamp_transport_handler;

template <typename Socket>
class wamp_rawsocket_transport :
        public wamp_transport,
        public std::enable_shared_from_this<wamp_rawsocket_transport<Socket>>
{
public:
    typedef Socket socket_type;

    typedef typename Socket::endpoint_type endpoint_type;

public:
    wamp_rawsocket_transport(
            boost::asio::io_service& io_service,
            const endpoint_type& remote_endpoint,
            bool debug_enabled=false);

    virtual ~wamp_rawsocket_transport() override = default;

    /*
     * CONNECTION INTERFACE
     */
    virtual boost::future<void> connect() override;

    virtual boost::future<void> disconnect() override;

    virtual bool is_connected() const override;

    /*
     * SENDER INTERFACE
     */
    virtual void send_message(wamp_message&& message) override;

    virtual void set_pause_handler(pause_handler&& handler) override;

    virtual void set_resume_handler(resume_handler&& handler) override;

    /*
     * RECEIVER INTERFACE
     */
    virtual void pause() override;

    virtual void resume() override;

    virtual void attach(
            const std::shared_ptr<wamp_transport_handler>& handler) override;

    virtual void detach() override;

    virtual bool has_handler() const override;

protected:
    socket_type& socket();

private:

    void handshake_reply_handler(
            const boost::system::error_code& error_code,
            std::size_t /* bytes_transferred */);

    void receive_message();

    void receive_message_header(
            const boost::system::error_code& error,
            std::size_t /* bytes transferred */);

    void receive_message_body(
            const boost::system::error_code& error,
            std::size_t /* bytes transferred */);

private:
    socket_type m_socket;

    endpoint_type m_remote_endpoint;

    boost::promise<void> m_connect;

    boost::promise<void> m_disconnect;

    pause_handler m_pause_handler;

    resume_handler m_resume_handler;

    std::shared_ptr<wamp_transport_handler> m_handler;

    uint8_t m_handshake_buffer[4];

    uint32_t m_message_length;

    msgpack::unpacker m_message_unpacker;

    bool m_debug_enabled;
};

} // namespace autobahn

#include "wamp_rawsocket_transport.ipp"

#endif // AUTOBAHN_WAMP_NETWORK_TRANSPORT_HPP