/**
* 为当前访问者启用 session
*
* @param const char* datadir 用于保存 session 数据文件的目录
*
* @return int 成功返回0，失败返回非0值
*/
int session_start(const char* datadir) {
char cookie_session_id[34];
int result;
if (g_session_data != NULL) {
fprintf(cgiOut, "Warning: session_start() calls again.<br />\n");
return -1;
}
// 初始化 session_data
sess_init(datadir);
// 根据 cookie 状态确定是创建新 session 还是载入现有的 session
result = cgiCookieString("CWEB_SESSION_ID", cookie_session_id, 33);
#ifdef DEBUG
fprintf(cgiOut, "cgiCookieString() result: %u\n", result);
fprintf(cgiOut, "cookie_session_id: %s\n", cookie_session_id);
#endif
if (result != cgiFormSuccess) {
// 没有找到有效的 cookie session_id，创建一个新 session
result = sess_create();
} else {
result = sess_load(cookie_session_id, datadir);
}
sess_gc(); //处理过期session
return result;
}
Object *
object_new (Object_Type type)
{
struct free_list_el *el;
Event_Type event = 0;
size_t obj_size;
Any_Type arg;
Object *obj;
obj_size = type_size[type];
if (free_list[type])
{
el = free_list[type];
free_list[type] = el->next;
obj = (Object *) el;
}
else
{
obj = malloc (obj_size);
if (!obj)
{
fprintf (stderr, "%s.object_new: %s\n", prog_name, strerror (errno));
return 0;
}
}
memset (obj, 0, obj_size);
obj->ref_count = 1;
obj->type = type;
switch (type)
{
case OBJ_CALL:
call_init ((Call *) obj);
event = EV_CALL_NEW;
break;
case OBJ_CONN:
conn_init ((Conn *) obj);
event = EV_CONN_NEW;
break;
case OBJ_SESS:
sess_init ((Sess *) obj);
event = EV_SESS_NEW;
break;
default:
panic ("object_new: bad object type %d\n", type);
break;
}
arg.l = 0;
event_signal (event, obj, arg);
return obj;
}
static struct h_scalar *session_hash_cache_create_fn(struct h_table *ht, void *key, uint8_t *is_syn)
{
struct key_5tuple *k = (struct key_5tuple *)key;
struct sft_fdb_entry *dc;
if(k->proto == IPPROTO_TCP && !k->is_syn)
{
return NULL;
}
struct sess_hash *hash = get_session_hash_by_lcore(rte_lcore_id());
if(rte_mempool_mc_get(hash->mem_cache, (void **)&dc) <0)
return NULL;
hash->curr_entry_count++;
sess_init(dc, k);
return &dc->h_scalar;
}
/**
* 载入现有的 session
*
* @param const char* session_id 要读取的 session_id
* @param const char* datadir
*
* @return int 成功返回0，失败返回非0值
*/
int sess_load(const char* session_id, const char* datadir) {
FILE* fp = NULL;
char* buffer = NULL;
char* tmp = NULL;
SESSION_ITEM_LIST tmp_list =
NULL;
size_t buffer_length = 2048;
int index = 0;
if (g_session_data == NULL) {
return -1;
}
#ifdef DEBUG
fprintf(cgiOut, "sess_load() BEGIN -->\n");
#endif
// 构造 session_data
g_session_data->session_id = (char*) malloc(
(strlen(session_id) + 1) * sizeof(char));
strcpy(g_session_data->session_id, session_id);
g_session_data->session_filename = sess_get_filename(
g_session_data->session_datadir, g_session_data->session_id);
#ifdef DEBUG
fprintf(cgiOut, "sess_load(), session file:%s\n", g_session_data->session_filename);
#endif
fp = fopen(g_session_data->session_filename, "r");
if (fp == NULL) {
// 无效的 session_filename
sess_free_resource();
sess_init(datadir);
return sess_create();
}
buffer = (char*) malloc(buffer_length * sizeof(char));
tmp = buffer;
while (!feof(fp)) {
if (fgets(tmp, buffer_length, fp) == NULL) {
break;
}
/*
if (buffer[strlen(buffer) + 1] != '\n' && !feof(fp)) {
// 缓冲区太小，无法容纳该行字符串
buffer_length += buffer_length;
tmp = (char*)malloc(buffer_length * sizeof(char));
strcpy(tmp, buffer);
free(buffer);
buffer = tmp;
tmp = buffer += strlen(buffer);
continue;
}
*/
// 如果 count >= max_index，则需要增加列表容量
if (g_session_data->count >= g_session_data->max_index) {
g_session_data->max_index += 10;
#ifdef DEBUG
fprintf(cgiOut, "* items count: %ld, max_index grow up to: %ld\n",
g_session_data->count,
g_session_data->max_index);
#endif
tmp_list = (SESSION_ITEM_LIST) malloc(
g_session_data->max_index * sizeof(SESSION_ITEM));
memset(tmp_list, 0,
g_session_data->max_index * sizeof(SESSION_ITEM));
if (g_session_data->items != NULL) {
memcpy(tmp_list, g_session_data->items,
g_session_data->count * sizeof(SESSION_ITEM));
free(g_session_data->items);
}
g_session_data->items = tmp_list;
tmp_list = NULL;
}
// 添加一个新项目
if ((index % 2) == 0) {
tmp = g_session_data->items[g_session_data->count].name;
#ifdef DEBUG
fprintf(cgiOut, "read name:");
#endif
} else {
tmp = g_session_data->items[g_session_data->count].string;
#ifdef DEBUG
fprintf(cgiOut, "read string:");
#endif
}
if (buffer[strlen(buffer) - 1] == '\n') {
buffer[strlen(buffer) - 1] = '\0';
}
tmp = sess_unserialize(buffer);
#ifdef DEBUG
fprintf(cgiOut, "%s\n", tmp);
#endif
if ((index % 2) == 0) {
g_session_data->items[g_session_data->count].name = tmp;
} else {
g_session_data->items[g_session_data->count].string = tmp;
g_session_data->count++;
}
// 增加索引，准备读取下一行
tmp = buffer;
index++;
}
free(buffer);
fclose(fp);
if ((index % 2) != 0) {
free(g_session_data->items[g_session_data->count - 1].name);
g_session_data->items[g_session_data->count - 1].name =
NULL;
}
#ifdef DEBUG
session_dump(cgiOut);
#endif
#ifdef DEBUG
fprintf(cgiOut, "sess_load() END -->\n");
#endif
return 0;
}
