#define CPPCMS_SOURCE
#include "session_pool.h"
#include "service.h"
#include "thread_pool.h"
#include "aio_timer.h"
#include "session_cookies.h"
#include "session_sid.h"
#include "session_dual.h"
#include "hmac_encryptor.h"
#include "json.h"
#include "cppcms_error.h"
#include "config.h"
#ifdef CPPCMS_USE_EXTERNAL_BOOST
# include <boost/bind.hpp>
# include <boost/shared_ptr.hpp>
# include <boost/enable_shared_from_this.hpp>
#else // Internal Boost
# include <cppcms_boost/bind.hpp>
# include <cppcms_boost/shared_ptr.hpp>
# include <cppcms_boost/enable_shared_from_this.hpp>
namespace boost = cppcms_boost;
#endif
#ifndef HAVE_GCRYPT
#include "aes_encryptor.h"
#endif
#ifdef CPPCMS_WIN_NATIVE
#include "session_win32_file_storage.h"
#else
#include "session_posix_file_storage.h"
#endif
namespace cppcms {
struct session_pool::data
{
};
using namespace cppcms::sessions;
template<typename Encryptor>
struct session_pool::enc_factory : public encryptor_factory {
enc_factory(std::string const &key) : key_(key) {}
virtual std::auto_ptr<cppcms::sessions::encryptor> get()
{
std::auto_ptr<cppcms::sessions::encryptor> tmp(new Encryptor(key_));
return tmp;
}
private:
std::string key_;
};
struct session_pool::sid_factory : public session_api_factory
{
sid_factory(session_pool *pool) : pool_(pool) {}
bool requires_gc() {
if(pool_->storage_.get())
return pool_->storage_->requires_gc();
else
return false;
}
void gc()
{
if(pool_->storage_.get())
pool_->storage_->gc_job();
}
intrusive_ptr<session_api> get() {
if(pool_->storage_.get())
return new session_sid(pool_->storage_->get());
else
return 0;
}
private:
session_pool *pool_;
};
struct session_pool::cookies_factory : public session_api_factory
{
cookies_factory(session_pool *pool) : pool_(pool) {}
bool requires_gc() {
return false;
}
void gc() {}
intrusive_ptr<session_api> get() {
if(pool_->encryptor_.get())
return new session_cookies(pool_->encryptor_->get());
else
return 0;
}
private:
session_pool *pool_;
};
struct session_pool::dual_factory : public session_api_factory
{
dual_factory(unsigned limit,session_pool *pool) : limit_(limit), pool_(pool) {}
bool requires_gc() {
if(pool_->storage_.get())
return pool_->storage_->requires_gc();
else
return false;
}
void gc()
{
if(pool_->storage_.get())
pool_->storage_->gc_job();
}
intrusive_ptr<session_api> get() {
if(pool_->storage_.get() && pool_->encryptor_.get())
return new session_dual(pool_->encryptor_->get(),pool_->storage_->get(),limit_);
else
return 0;
}
private:
unsigned limit_;
session_pool *pool_;
};
class session_pool::gc_job : public boost::enable_shared_from_this<gc_job> {
public:
gc_job(service *ser,int freq,session_pool *pool) :
timer_(new aio::timer(*ser)),
service_(ser),
freq_(freq),
pool_(pool)
{
}
void async_run() const
{
service_->thread_pool().post(boost::bind(&gc_job::gc,shared_from_this()));
}
private:
void gc() const
{
pool_->backend_->gc();
timer_->expires_from_now(freq_);
timer_->async_wait(boost::bind(&gc_job::async_run,shared_from_this()));
}
boost::shared_ptr<aio::timer> timer_;
service *service_;
int freq_;
session_pool *pool_;
};
session_pool::session_pool(service &srv) :
service_(&srv)
{
std::string location=srv.settings().get("session.location","none");
if(location == "client" || location=="both") {
std::string enc=srv.settings().get("session.cookies_encryptor","");
std::auto_ptr<cppcms::sessions::encryptor_factory> factory;
if(enc=="hmac") {
std::string key = srv.settings().get<std::string>("session.cookies_key");
factory.reset(new enc_factory<cppcms::sessions::impl::hmac_cipher>(key));
}
#ifdef HAVE_GCRYPT
else if(enc=="aes") {
std::string key = srv.settings().get<std::string>("session.cookies_key");
factory.reset(new enc_factory<cppcms::sessions::impl::aes_cipher>(key));
}
#endif
else if(enc.empty())
; // Nothing to do
else
throw cppcms_error("Unknown encryptor: "+enc);
encryptor(factory);
}
if(location == "server" || location == "both" ) {
std::string storage=srv.settings().get("session.backend","");
if(storage == "files") {
std::auto_ptr<sessions::session_storage_factory> tmp;
std::string dir = srv.settings().get("session.files.dir","");
#ifdef CPPCMS_WIN_NATIVE
tmp.reset(new session_file_storage_factory(dir));
#else
bool sharing = srv.settings().get("session.files.shared",true);
int threads = srv.threads_no();
int procs = srv.procs_no();
if(procs == 0) procs=1;
tmp.reset(new session_file_storage_factory(dir,threads*procs,procs,sharing));
#endif
}
else if(storage.empty())
; /// Nothing to do
else
throw cppcms_error("Unknown server side storage:"+storage);
}
if(location == "server") {
std::auto_ptr<session_api_factory> f(new sid_factory(this));
backend(f);
}
else if(location == "client") {
std::auto_ptr<session_api_factory> f(new cookies_factory(this));
backend(f);
}
else if(location == "both") {
unsigned limit=srv.settings().get("session.client_size_limit",2048);
std::auto_ptr<session_api_factory> f(new dual_factory(limit,this));
backend(f);
}
else if(location == "none")
;
else
throw cppcms_error("Unknown location");
service_->after_fork(boost::bind(&session_pool::after_fork,this));
}
void session_pool::after_fork()
{
if(backend_.get() && backend_->requires_gc()) {
if(service_->process_id()!=1)
return;
int frequency = service_->settings().get("sessions.gc",0);
if(frequency > 0) {
boost::shared_ptr<gc_job> job(new gc_job(service_,frequency,this));
job->async_run();
}
}
}
session_pool::~session_pool()
{
}
intrusive_ptr<session_api> session_pool::get()
{
if(backend_.get())
return backend_->get();
else
return 0;
}
void session_pool::backend(std::auto_ptr<session_api_factory> b)
{
backend_=b;
}
void session_pool::encryptor(std::auto_ptr<sessions::encryptor_factory> e)
{
encryptor_=e;
}
void session_pool::storage(std::auto_ptr<sessions::session_storage_factory> s)
{
storage_=s;
}
} /// cppcms
