src/manager/lib/session.c

   1 /**
   2  * @file session.c
   3  *
   4  * @brief Implementation of session_t.
   5  *
   6  */
   7
   8 /*
   9  * Copyright (C) 2007 Martin Willi
  10  * Hochschule fuer Technik Rapperswil
  11  *
  12  * This program is free software; you can redistribute it and/or modify it
  13  * under the terms of the GNU General Public License as published by the
  14  * Free Software Foundation; either version 2 of the License, or (at your
  15  * option) any later version.  See <http://www.fsf.org/copyleft/gpl.txt>.
  16  *
  17  * This program is distributed in the hope that it will be useful, but
  18  * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
  19  * or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  20  * for more details.
  21  */
  22
  23 #define _GNU_SOURCE
  24
  25 #include "session.h"
  26
  27 #include <string.h>
  28 #include <fcgiapp.h>
  29 #include <stdio.h>
  30
  31 #include <utils/linked_list.h>
  32 #include <utils/randomizer.h>
  33
  34 typedef struct private_session_t private_session_t;
  35
  36 /**
  37  * private data of the task manager
  38  */
  39 struct private_session_t {
  40
  41         /**
  42          * public functions
  43          */
  44         session_t public;
  45
  46         /**
  47          * session ID
  48          */
  49         char *sid;
  50
  51         /**
  52          * list of controller instances controller_t
  53          */
  54         linked_list_t *controllers;
  55
  56         /**
  57          * user defined session context
  58          */
  59         context_t *context;
  60 };
  61
  62 /**
  63  * Implementation of session_t.load_controller.
  64  */
  65 static void add_controller(private_session_t *this, controller_t *controller)
  66 {
  67         this->controllers->insert_last(this->controllers, controller);
  68 }
  69
  70 /**
  71  * Create a session ID and a cookie
  72  */
  73 static void create_sid(private_session_t *this, request_t *request)
  74 {
  75         char buf[16];
  76         chunk_t chunk = chunk_from_buf(buf);
  77         randomizer_t *randomizer = randomizer_create();
  78
  79         randomizer->get_pseudo_random_bytes(randomizer, sizeof(buf), buf);
  80         this->sid = chunk_to_hex(chunk, FALSE);
  81         request->add_cookie(request, "SID", this->sid);
  82         randomizer->destroy(randomizer);
  83 }
  84
  85 /**
  86  * Implementation of session_t.process.
  87  */
  88 static void process(private_session_t *this, request_t *request)
  89 {
  90         char *pos, *start, *param[6] = {NULL, NULL, NULL, NULL, NULL, NULL};
  91         iterator_t *iterator;
  92         bool handled = FALSE;
  93         controller_t *current;
  94         int i = 0;
  95
  96         if (this->sid == NULL)
  97         {
  98                 create_sid(this, request);
  99         }
 100
 101         start = request->get_path(request);
 102         if (start)
 103         {
 104                 if (*start == '/') start++;
 105                 while ((pos = strchr(start, '/')) != NULL && i < 5)
 106                 {
 107                         param[i++] = strndup(start, pos - start);
 108                         start = pos + 1;
 109                 }
 110                 param[i] = strdup(start);
 111                 iterator = this->controllers->create_iterator(this->controllers, TRUE);
 112                 while (iterator->iterate(iterator, (void**)&current))
 113                 {
 114                         if (streq(current->get_name(current), param[0]))
 115                         {
 116                                 current->handle(current, request, param[1], param[2], param[3],
 117                                                                 param[4], param[5]);
 118                                 handled = TRUE;
 119                                 break;
 120                         }
 121                 }
 122                 iterator->destroy(iterator);
 123                 for (i = 0; i < 6; i++)
 124                 {
 125                         free(param[i]);
 126                 }
 127         }
 128         if (!handled)
 129         {
 130                 if (this->controllers->get_first(this->controllers,
 131                                                                                  (void**)&current) == SUCCESS)
 132                 {
 133                         request->redirect(request, current->get_name(current));
 134                 }
 135         }
 136 }
 137
 138 /**
 139  * Implementation of session_t.get_sid.
 140  */
 141 static char* get_sid(private_session_t *this)
 142 {
 143         return this->sid;
 144 }
 145
 146 /**
 147  * Implementation of session_t.destroy
 148  */
 149 static void destroy(private_session_t *this)
 150 {
 151         this->controllers->destroy_offset(this->controllers, offsetof(controller_t, destroy));
 152         if (this->context) this->context->destroy(this->context);
 153         free(this->sid);
 154         free(this);
 155 }
 156
 157 /*
 158  * see header file
 159  */
 160 session_t *session_create(context_t *context)
 161 {
 162         private_session_t *this = malloc_thing(private_session_t);
 163
 164         this->public.add_controller = (void(*)(session_t*, controller_t*))add_controller;
 165         this->public.process = (void(*)(session_t*,request_t*))process;
 166         this->public.get_sid = (char*(*)(session_t*))get_sid;
 167         this->public.destroy = (void(*)(session_t*))destroy;
 168
 169         this->sid = NULL;
 170         this->controllers = linked_list_create();
 171         this->context = context;
 172
 173         return &this->public;
 174 }
 175