rippled/beast/http/parser.h

//------------------------------------------------------------------------------
/*
    This file is part of Beast: https://github.com/vinniefalco/Beast
    Copyright 2013, Vinnie Falco <vinnie.falco@gmail.com>

    Permission to use, copy, modify, and/or distribute this software for any
    purpose  with  or without fee is hereby granted, provided that the above
    copyright notice and this permission notice appear in all copies.

    THE  SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
    WITH  REGARD  TO  THIS  SOFTWARE  INCLUDING  ALL  IMPLIED  WARRANTIES  OF
    MERCHANTABILITY  AND  FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
    ANY  SPECIAL ,  DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
    WHATSOEVER  RESULTING  FROM  LOSS  OF USE, DATA OR PROFITS, WHETHER IN AN
    ACTION  OF  CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
    OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
*/
//==============================================================================

#ifndef BEAST_HTTP_PARSER_H_INCLUDED
#define BEAST_HTTP_PARSER_H_INCLUDED

#include <beast/http/message.h>
#include <beast/http/body.h>
#include <functional>
#include <string>
#include <utility>

namespace beast {
namespace http {

/** Parser for HTTP messages.
    The result is stored in a message object.
*/
class parser : public beast::http::basic_parser
{
private:
    std::reference_wrapper <message> message_;
    std::function<void(void const*, std::size_t)> write_body_;

public:
    /** Construct a parser for HTTP request or response.
        The headers plus request or status line are stored in message.
        The content-body, if any, is passed as a series of calls to
        the write_body function. Transfer encodings are applied before
        any data is passed to the write_body function.
    */
    parser (std::function<void(void const*, std::size_t)> write_body,
            message& m, bool request)
        : beast::http::basic_parser (request)
        , message_(m)
        , write_body_(std::move(write_body))
    {
        message_.get().request(request);
    }

    parser (message& m, body& b, bool request)
        : beast::http::basic_parser (request)
        , message_(m)
    {
        write_body_ = [&b](void const* data, std::size_t size)
            {
                b.write(data, size);
            };

        message_.get().request(request);
    }

    parser& operator= (parser&& other) = default;

private:
    template <class = void>
    void
    do_start ();

    template <class = void>
    bool
    do_request (method_t method, std::string const& url,
        int major, int minor, bool keep_alive, bool upgrade);

    template <class = void>
    bool
    do_response (int status, std::string const& text,
        int major, int minor, bool keep_alive, bool upgrade);

    template <class = void>
    void
    do_field (std::string const& field, std::string const& value);

    template <class = void>
    void
    do_body (void const* data, std::size_t bytes);

    template <class = void>
    void
    do_complete();

    void
    on_start () override
    {
        do_start();
    }

    bool
    on_request (method_t method, std::string const& url,
        int major, int minor, bool keep_alive, bool upgrade) override
    {
        return do_request (method, url, major, minor, keep_alive, upgrade);
    }

    bool
    on_response (int status, std::string const& text,
        int major, int minor, bool keep_alive, bool upgrade) override
    {
        return do_response (status, text, major, minor, keep_alive, upgrade);
    }

    void
    on_field (std::string const& field, std::string const& value) override
    {
        do_field (field, value);
    }

    void
    on_body (void const* data, std::size_t bytes) override
    {
        do_body (data, bytes);
    }

    void
    on_complete() override
    {
        do_complete();
    }
};

//------------------------------------------------------------------------------
template <class>
void
parser::do_start()
{
}

template <class>
bool
parser::do_request (method_t method, std::string const& url,
    int major, int minor, bool keep_alive, bool upgrade)
{
    message_.get().method (method);
    message_.get().url (url);
    message_.get().version (major, minor);
    message_.get().keep_alive (keep_alive);
    message_.get().upgrade (upgrade);
    return true;
}

template <class>
bool
parser::do_response (int status, std::string const& text,
    int major, int minor, bool keep_alive, bool upgrade)
{
    message_.get().status (status);
    message_.get().reason (text);
    message_.get().version (major, minor);
    message_.get().keep_alive (keep_alive);
    message_.get().upgrade (upgrade);
    return true;
}

template <class>
void
parser::do_field (std::string const& field, std::string const& value)
{
    message_.get().headers.append (field, value);
}

template <class>
void
parser::do_body (void const* data, std::size_t bytes)
{
    write_body_(data, bytes);
}

template <class>
void
parser::do_complete()
{
}

} // http
} // beast

#endif