///////////////////////////////////////////////////////////////////////////////
//
// Copyright (C) 2008-2012 Artyom Beilis (Tonkikh) <artyomtnk@yahoo.com>
//
// See accompanying file COPYING.TXT file for licensing details.
//
///////////////////////////////////////////////////////////////////////////////
#define CPPCMS_SOURCE
#include "cgi_api.h"
#include <cppcms/service.h>
#include <cppcms/http_context.h>
#include <cppcms/http_request.h>
#include <cppcms/http_response.h>
#include <cppcms/application.h>
#include <cppcms/applications_pool.h>
#include <cppcms/thread_pool.h>
#include <cppcms/views_pool.h>
#include <cppcms/cache_interface.h>
#include <cppcms/session_interface.h>
#include <cppcms/cppcms_error.h>
#include <booster/log.h>
#include <booster/backtrace.h>
#include "cached_settings.h"
#include <cppcms/config.h>
#include "binder.h"
namespace cppcms {
namespace http {
struct context::_data {
std::locale locale;
std::string skin;
http::request request;
std::auto_ptr<http::response> response;
std::auto_ptr<cache_interface> cache;
std::auto_ptr<session_interface> session;
_data(context &cntx) :
locale(cntx.connection().service().locale()),
request(cntx.connection())
{
}
};
context::context(booster::shared_ptr<impl::cgi::connection> conn) :
conn_(conn)
{
d.reset(new _data(*this));
d->response.reset(new http::response(*this));
skin(service().views_pool().default_skin());
}
std::string context::skin()
{
return d->skin;
}
cache_interface &context::cache()
{
if(!d->cache.get()) {
d->cache.reset(new cache_interface(*this));
}
return *d->cache;
}
void context::skin(std::string const &skin)
{
d->skin=skin;
}
namespace {
struct ct_to_bool {
void (context::*member)(bool r);
booster::shared_ptr<context> ctx;
void operator()(context::completion_type c)
{
((*ctx).*member)(c!=context::operation_completed);
}
};
}
void context::run()
{
ct_to_bool cb = { &context::on_request_ready, self() };
conn_->async_prepare_request(this,cb);
}
namespace {
struct dispatcher {
void (*func)(booster::shared_ptr<application_specific_pool> const &,booster::shared_ptr<context> const &,std::string const &);
booster::shared_ptr<application_specific_pool> pool;
booster::shared_ptr<context> ctx;
std::string url;
void operator()() {
func(pool,ctx,url);
}
};
}
void context::on_request_ready(bool error)
{
if(error) return;
char const *host = conn_->cgetenv("HTTP_HOST");
char const *path_info = conn_->cgetenv("PATH_INFO");
char const *script_name = conn_->cgetenv("SCRIPT_NAME");
std::string matched;
booster::shared_ptr<application_specific_pool> pool =
service().applications_pool().get_application_specific_pool(
host,
script_name,
path_info,
matched
);
if(!pool) {
response().io_mode(http::response::asynchronous);
response().make_error_response(http::response::not_found);
async_complete_response();
return;
}
if((pool->flags() & app::op_mode_mask)!=app::synchronous) {
// asynchronous
booster::intrusive_ptr<application> app = pool->get(service());
if(!app) {
BOOSTER_ERROR("cppcms") << "Cound fetch asynchronous application from pool";
response().io_mode(http::response::asynchronous);
response().make_error_response(http::response::internal_server_error);
async_complete_response();
return;
}
app->assign_context(self());
response().io_mode(http::response::asynchronous);
dispatch(app,matched,false);
return;
}
else {
dispatcher dt;
dt.func = &context::dispatch;
dt.pool = pool;
dt.ctx = self();
dt.url.swap(matched);
service().thread_pool().post(dt);
return;
}
}
namespace {
struct run_ctx {
booster::shared_ptr<context> ctx;
void operator()() {
ctx->run();
}
};
}
void context::complete_response()
{
response().finalize();
if(conn_->is_reuseable()) {
booster::shared_ptr<context> cont(new context(conn_));
run_ctx rn = { cont };
service().post(rn);
}
conn_.reset();
}
// static
void context::dispatch(booster::shared_ptr<application_specific_pool> const &pool,booster::shared_ptr<context> const &self,std::string const &url)
{
booster::intrusive_ptr<application> app = pool->get(self->service());
if(!app) {
BOOSTER_ERROR("cppcms") << "Cound fetch synchronous application from a pool";
self->response().make_error_response(http::response::internal_server_error);
self->complete_response();
return;
}
app->assign_context(self);
dispatch(app,url,true);
}
// static
void context::dispatch(booster::intrusive_ptr<application> const &app,std::string const &url,bool syncronous)
{
try {
if(syncronous && !app->context().service().cached_settings().session.disable_automatic_load)
app->context().session().load();
app->main(url);
}
catch(request_forgery_error const &e) {
if(app->get_context() && !app->response().some_output_was_written()) {
app->response().make_error_response(http::response::forbidden);
}
}
catch(std::exception const &e){
BOOSTER_ERROR("cppcms") << "Caught exception ["<<e.what()<<"]\n" << booster::trace(e) ;
if(app->get_context()) {
if(!app->response().some_output_was_written()) {
if(app->service().cached_settings().security.display_error_message) {
std::ostringstream ss;
ss << e.what() << '\n';
ss << booster::trace(e);
app->response().make_error_response(http::response::internal_server_error,ss.str());
}
else
app->response().make_error_response(http::response::internal_server_error);
}
}
}
if(app->get_context()) {
if(syncronous) {
app->context().complete_response();
}
else {
app->context().async_complete_response();
}
}
}
void context::async_flush_output(context::handler const &h)
{
if(response().io_mode() != http::response::asynchronous && response().io_mode()!=http::response::asynchronous_raw) {
throw cppcms_error("Can't use asynchronouse operations when I/O mode is synchronous");
}
conn_->async_write_response(
response(),
false,
h);
}
void context::async_complete_response()
{
response().finalize();
if(response().io_mode() == http::response::asynchronous || response().io_mode() == http::response::asynchronous_raw) {
ct_to_bool cb = { &context::try_restart, self() };
conn_->async_write_response(
response(),
true,
cb);
return;
}
complete_response();
}
void context::try_restart(bool e)
{
if(e) return;
if(conn_->is_reuseable()) {
booster::shared_ptr<context> cont(new context(conn_));
cont->run();
}
conn_.reset();
}
booster::shared_ptr<context> context::self()
{
return shared_from_this();
}
context::~context()
{
}
void context::async_on_peer_reset(booster::callback<void()> const &h)
{
conn_->aync_wait_for_close_by_peer(h);
}
impl::cgi::connection &context::connection()
{
return *conn_;
}
cppcms::service &context::service()
{
return conn_->service();
}
http::request &context::request()
{
return d->request;
}
http::response &context::response()
{
return *d->response;
}
json::value const &context::settings()
{
return conn_->service().settings();
}
std::locale context::locale()
{
return d->locale;
}
void context::locale(std::locale const &new_locale)
{
d->locale=new_locale;
if(response().some_output_was_written())
response().out().imbue(d->locale);
}
void context::locale(std::string const &name)
{
locale(service().locale(name));
}
session_interface &context::session()
{
if(!d->session.get())
d->session.reset(new session_interface(*this));
return *d->session;
}
} // http
} // cppcms
