CppCMS C++ Web Framework Code

#include "worker_thread.h"
#include "session_interface.h"
#include "session_api.h"
#include "manager.h"

namespace cppcms {

session_interface::session_interface(worker_thread &w,boost::shared_ptr<session_api> api_ptr) :
	worker(w)
{
	timeout_val_def=w.app.config.ival("session.timeout",24*3600);
	string s_how=w.app.config.sval("session.expire","browser");
	if(s_how=="fixed") {
		how_def=fixed;
	}
	else if(s_how=="renew") {
		how_def=renew;
	}
	else if(s_how=="browser") {
		how_def=browser;
	}
	else {
		throw cppcms_error("Unsupported `session.expire' type `"+s_how+"'");
	}
	string api=w.app.config.sval("session.api","none");
	if(api_ptr.get()!=NULL) {
		storage=api_ptr;
	}
	else {
		if(api=="none")
			return;
		// TODO
	}
	w.on_start.connect(on_start());
	w.on_end.connect(on_end());
}

bool session_interface::load()
{
	check();
	data.clear();
	data_copy.clear();
	timeout_val=timeout_val_def;
	how=how_def;
	archive ar;
	if(!storage->load(this,ar.set(),timeout_in)) {
		return false;
	}
	int i,num;
	ar>>num;
	for(i=0;i<num;i++) {
		string key,val;
		ar>>key>>val;
		data[key].swap(val);
	}
	data_copy=data;
	return true;
}

int session_interface::cookie_age()
{
	if(how==browser)
		return 0;
	if(how==renew || ( how==fixed && new_session )) 
		return timeout_val;
	return timeout_in - time(NULL);
}

time_t session_interface::session_age()
{
	if(how==browser || how==renew || (how==fixed && new_session))
		return timeout_val + time(NULL);
	return timeout_in;
}

void session_interface::save()
{
	check();
	new_session  = data_copy.empty() && !data.empty();
	if(data.empty()) {
		if(session_cookie()!="")
			storage->clear(this);
		return;
	}	

	time_t now = time(NULL);

	if(data==data_copy) {
		if(how==fixed) {
			return;
		}
		if(how==renew || how==browser) {
			int64_t delta=now + timeout_val - timeout_in;
			if(delta < timeout_val * 0.1) {// Less then 10% -- no renew need
				return;
			}
		}
	}
	int num=data.size();
	archive ar;
	ar<<num;
	for(map<string,string>::iterator p=data.begin(),e=data.end();p!=e;++p) {
		a<<p->first<<p->second;
	}
	storage->save(this,session_age(),ar.get());
}

void session_interface::on_start()
{
	if(storage.get()!=NULL)
		load();
	}
}

void session_interface::on_end()
{
	try{
		if(storage.get()!=NULL)
			save();
	}
	catch(std::error const &e) {
		worker.log(0,e.what());
	}
	catch(...)
	{
		worker.log(0,"Catched unknown exception!!!");
	}
}

void session_interface::check()
{
	if(storage.get()==NULL)
		throw cppcms_error("Session storage backend is not loaded\n");
}

string &session_interface::operator[](string const &key)
{
	return data[key];
}

void session_interface::del(string const &key)
{
	data.erase(key);
}

bool session_interface::is_set(string const &key)
{
	return data.find(key)!=data.end();
}

void session_interface::clear()
{
	data.clear();
}

void session_interface::clear_session_cookie()
{
	if(get_session_cookie()!="")
		set_session_cookie(-1,"");
}
void session_interface::set_session_cookie(int64_t age,string const &data)
{
	cgicc::HTTPCookie
		cookie(	worker.app.config.sval("session.cookies_prefix","cppcms_session"), // name
			(age < 0 ? data : ""), // value
			"",   // comment
			worker.app.config.sval("session.cookies_domain",""), // domain
			( age < 0 ? 0 : age ),
			worker.app.config.sval("session.cookies_path","/"),
			worker.app.config.ival("session.cookies_secure",0));
	worker.set_cookie(cookie);
}
void session_interface::set_session_cookie(string const &data)
{
	set_session_cookie(cookie_age(),data);
}

string session_interface::get_session_cookie()
{
	string name=worker.app.config.sval("session.cookies_prefix","cppcms_session");
	vector<cgicc::HTTPCookie> const &cookies=worker.env->getCookiesList();
	for(i=0;i<cookies.size;i++) {
		if(cookies[i].getName()==name)
			return cookies[i].getValue();
	}
	return string("");
}


};