http_request_reader.hpp

Go to the documentation of this file.

/*
 * Copyright 2015, alex at staticlibs.net
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

// ---------------------------------------------------------------------
// pion:  a Boost C++ framework for building lightweight HTTP interfaces
// ---------------------------------------------------------------------
// Copyright (C) 2007-2014 Splunk Inc.  (https://github.com/splunk/pion)
//
// Distributed under the Boost Software License, Version 1.0.
// See http://www.boost.org/LICENSE_1_0.txt
//

#ifndef STATICLIB_PION_HTTP_REQUEST_READER_HPP
#define STATICLIB_PION_HTTP_REQUEST_READER_HPP

#include <chrono>
#include <functional>
#include <memory>

#include "asio.hpp"

#include "staticlib/config.hpp"

#include "staticlib/pion/http_parser.hpp"
#include "staticlib/pion/http_request.hpp"
#include "staticlib/pion/tcp_connection.hpp"

namespace staticlib { 
namespace pion {

class http_server;

class http_request_reader : public http_parser {
private:
    http_server& server;

    tcp_connection_ptr tcp_conn;

    uint32_t read_timeout_millis;

    http_request_ptr request;

public:

    http_request_reader(http_server& srv, tcp_connection_ptr& tcp_conn,
            uint32_t read_timeout) :
    http_parser(true),
    server(srv),
    tcp_conn(tcp_conn),
    read_timeout_millis(read_timeout),
    request(new http_request()) {
        request->set_remote_ip(tcp_conn->get_remote_ip());
        request->set_request_reader(this);
    }

    http_request_reader(const http_request_reader&) = delete;

    http_request_reader& operator=(const http_request_reader&) = delete;

    static void receive(std::unique_ptr<http_request_reader> self);

private:

    static void consume_bytes(std::unique_ptr<http_request_reader> self, 
            const std::error_code& read_error, std::size_t bytes_read);

    static void consume_bytes(std::unique_ptr<http_request_reader> self);
    
    static void read_bytes_with_timeout(std::unique_ptr<http_request_reader> self);

    void handle_read_error(const std::error_code& read_error);

    void finished_parsing_headers(const std::error_code& ec, sl::support::tribool& rc);

    void finished_reading(const std::error_code& ec);

};

} // namespace
}

#endif // STATICLIB_PION_HTTP_REQUEST_READER_HPP