ET9_WebGL_Dev/Egametang/Experimental/http_server/connection.cc

//
// connection.cpp
// ~~~~~~~~~~~~~~
//
// Copyright (c) 2003-2010 Christopher M. Kohlhoff (chris at kohlhoff dot com)
//
// Distributed under the Boost Software License, Version 1.0. (See accompanying
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
//

#include <vector>
#include <boost/bind.hpp>
#include "Experimental/http_server/connection.h"
#include "Experimental/http_server/connection_manager.h"
#include "Experimental/http_server/request_handler.h"

namespace http_server {

connection::connection(boost::asio::io_service& io_service,
		connection_manager& manager, request_handler& handler) :
	socket_(io_service), connection_manager_(manager),
			request_handler_(handler)
{
}

boost::asio::ip::tcp::socket& connection::socket()
{
	return socket_;
}

void connection::start()
{
	socket_.async_read_some(boost::asio::buffer(buffer_), boost::bind(
			&connection::handle_read, shared_from_this(),
			boost::asio::placeholders::error,
			boost::asio::placeholders::bytes_transferred));
}

void connection::stop()
{
	socket_.close();
}

void connection::handle_read(const boost::system::error_code& e,
		std::size_t bytes_transferred)
{
	if (!e)
	{
		boost::tribool result;
		boost::tie(result, boost::tuples::ignore) = request_parser_.parse(
				request_, buffer_.data(), buffer_.data() + bytes_transferred);

		if (result)
		{
			request_handler_.handle_request(request_, reply_);
			boost::asio::async_write(socket_, reply_.to_buffers(), boost::bind(
					&connection::handle_write, shared_from_this(),
					boost::asio::placeholders::error));
		}
		else if (!result)
		{
			reply_ = reply::stock_reply(reply::bad_request);
			boost::asio::async_write(socket_, reply_.to_buffers(), boost::bind(
					&connection::handle_write, shared_from_this(),
					boost::asio::placeholders::error));
		}
		else
		{
			socket_.async_read_some(boost::asio::buffer(buffer_), boost::bind(
					&connection::handle_read, shared_from_this(),
					boost::asio::placeholders::error,
					boost::asio::placeholders::bytes_transferred));
		}
	}
	else if (e != boost::asio::error::operation_aborted)
	{
		connection_manager_.stop(shared_from_this());
	}
}

void connection::handle_write(const boost::system::error_code& e)
{
	if (!e)
	{
		// Initiate graceful connection closure.
		boost::system::error_code ignored_ec;
		socket_.shutdown(boost::asio::ip::tcp::socket::shutdown_both,
				ignored_ec);
	}

	if (e != boost::asio::error::operation_aborted)
	{
		connection_manager_.stop(shared_from_this());
	}
}

} // namespace http_server