6bf596fd88ed0a792ea62a3f656f53658df08eb4
[strongswan.git] / src / libfast / request.c
1 /*
2 * Copyright (C) 2007 Martin Willi
3 * Hochschule fuer Technik Rapperswil
4 *
5 * This program is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License as published by the
7 * Free Software Foundation; either version 2 of the License, or (at your
8 * option) any later version. See <http://www.fsf.org/copyleft/gpl.txt>.
9 *
10 * This program is distributed in the hope that it will be useful, but
11 * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
12 * or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
13 * for more details.
14 */
15
16 #define _GNU_SOURCE
17
18 #include "request.h"
19
20 #include <library.h>
21 #include <debug.h>
22 #include <stdlib.h>
23 #include <pthread.h>
24 #include <string.h>
25 #include <ClearSilver/ClearSilver.h>
26
27 #include <threading/thread.h>
28 #include <threading/thread_value.h>
29
30 typedef struct private_request_t private_request_t;
31
32 /**
33 * private data of the task manager
34 */
35 struct private_request_t {
36
37 /**
38 * public functions
39 */
40 request_t public;
41
42 /**
43 * FastCGI request object
44 */
45 FCGX_Request req;
46
47 /**
48 * length of the req.envp array
49 */
50 int req_env_len;
51
52 /**
53 * ClearSilver CGI Kit context
54 */
55 CGI *cgi;
56
57 /**
58 * ClearSilver HDF dataset for this request
59 */
60 HDF *hdf;
61
62 /**
63 * close the session?
64 */
65 bool closed;
66
67 /**
68 * reference count
69 */
70 refcount_t ref;
71 };
72
73 /**
74 * ClearSilver cgiwrap is not threadsave, so we use a private
75 * context for each thread.
76 */
77 static thread_value_t *thread_this;
78
79 /**
80 * control variable for pthread_once
81 */
82 pthread_once_t once = PTHREAD_ONCE_INIT;
83
84 /**
85 * fcgiwrap read callback
86 */
87 static int read_cb(void *null, char *buf, int size)
88 {
89 private_request_t *this = (private_request_t*)thread_this->get(thread_this);
90
91 return FCGX_GetStr(buf, size, this->req.in);
92 }
93
94 /**
95 * fcgiwrap writef callback
96 */
97 static int writef_cb(void *null, const char *format, va_list args)
98 {
99 private_request_t *this = (private_request_t*)thread_this->get(thread_this);
100
101 FCGX_VFPrintF(this->req.out, format, args);
102 return 0;
103 }
104 /**
105 * fcgiwrap write callback
106 */
107 static int write_cb(void *null, const char *buf, int size)
108 {
109 private_request_t *this = (private_request_t*)thread_this->get(thread_this);
110
111 return FCGX_PutStr(buf, size, this->req.out);
112 }
113
114 /**
115 * fcgiwrap getenv callback
116 */
117 static char *getenv_cb(void *null, const char *key)
118 {
119 char *value;
120 private_request_t *this = (private_request_t*)thread_this->get(thread_this);
121
122 value = FCGX_GetParam(key, this->req.envp);
123 return value ? strdup(value) : NULL;
124 }
125
126 /**
127 * fcgiwrap getenv callback
128 */
129 static int putenv_cb(void *null, const char *key, const char *value)
130 {
131 /* not supported */
132 return 1;
133 }
134
135 /**
136 * fcgiwrap iterenv callback
137 */
138 static int iterenv_cb(void *null, int num, char **key, char **value)
139 {
140 *key = NULL;
141 *value = NULL;
142 private_request_t *this = (private_request_t*)thread_this->get(thread_this);
143 if (num < this->req_env_len)
144 {
145 char *eq;
146
147 eq = strchr(this->req.envp[num], '=');
148 if (eq)
149 {
150 *key = strndup(this->req.envp[num], eq - this->req.envp[num]);
151 *value = strdup(eq + 1);
152 }
153 if (*key == NULL || *value == NULL)
154 {
155 free(*key);
156 free(*value);
157 return 1;
158 }
159 }
160 return 0;
161 }
162
163 /**
164 * Implementation of request_t.get_cookie.
165 */
166 static char* get_cookie(private_request_t *this, char *name)
167 {
168 return hdf_get_valuef(this->hdf, "Cookie.%s", name);
169 }
170
171 /**
172 * Implementation of request_t.get_path.
173 */
174 static char* get_path(private_request_t *this)
175 {
176 char * path = FCGX_GetParam("PATH_INFO", this->req.envp);
177 return path ? path : "";
178 }
179
180 /**
181 * Implementation of request_t.get_host.
182 */
183 static char* get_host(private_request_t *this)
184 {
185 char *addr = FCGX_GetParam("REMOTE_ADDR", this->req.envp);
186 return addr ? addr : "";
187 }
188
189 /**
190 * Implementation of request_t.get_user_agent.
191 */
192 static char* get_user_agent(private_request_t *this)
193 {
194 char *agent = FCGX_GetParam("HTTP_USER_AGENT", this->req.envp);
195 return agent ? agent : "";
196 }
197
198 /**
199 * Implementation of request_t.get_post_data.
200 */
201 static char* get_query_data(private_request_t *this, char *name)
202 {
203 return hdf_get_valuef(this->hdf, "Query.%s", name);
204 }
205
206 /**
207 * Implementation of request_t.get_base.
208 */
209 static char* get_base(private_request_t *this)
210 {
211 return FCGX_GetParam("SCRIPT_NAME", this->req.envp);
212 }
213
214 /**
215 * Implementation of request_t.add_cookie.
216 */
217 static void add_cookie(private_request_t *this, char *name, char *value)
218 {
219 thread_this->set(thread_this, this);
220 cgi_cookie_set (this->cgi, name, value, get_base(this), NULL, NULL, 0, 0);
221 }
222
223 /**
224 * Implementation of request_t.redirect.
225 */
226 static void redirect(private_request_t *this, char *fmt, ...)
227 {
228 va_list args;
229
230 FCGX_FPrintF(this->req.out, "Status: 303 See Other\n");
231 FCGX_FPrintF(this->req.out, "Location: %s%s", get_base(this),
232 *fmt == '/' ? "" : "/");
233 va_start(args, fmt);
234 FCGX_VFPrintF(this->req.out, fmt, args);
235 va_end(args);
236 FCGX_FPrintF(this->req.out, "\n\n");
237 }
238
239 /**
240 * Implementation of request_t.get_referer.
241 */
242 static char* get_referer(private_request_t *this)
243 {
244 return FCGX_GetParam("HTTP_REFERER", this->req.envp);
245 }
246
247 /**
248 * Implementation of request_t.to_referer.
249 */
250 static void to_referer(private_request_t *this)
251 {
252 char *referer;
253
254 referer = get_referer(this);
255 if (referer)
256 {
257 FCGX_FPrintF(this->req.out, "Status: 303 See Other\n");
258 FCGX_FPrintF(this->req.out, "Location: %s\n\n", referer);
259 }
260 else
261 {
262 redirect(this, "/");
263 }
264 }
265
266 /**
267 * Implementation of request_t.session_closed.
268 */
269 static bool session_closed(private_request_t *this)
270 {
271 return this->closed;
272 }
273
274 /**
275 * Implementation of request_t.close_session.
276 */
277 static void close_session(private_request_t *this)
278 {
279 this->closed = TRUE;
280 }
281
282 /**
283 * Implementation of request_t.serve.
284 */
285 static void serve(private_request_t *this, char *headers, chunk_t chunk)
286 {
287 FCGX_FPrintF(this->req.out, "%s\n\n", headers);
288
289 FCGX_PutStr(chunk.ptr, chunk.len, this->req.out);
290 }
291
292 /**
293 * Implementation of request_t.render.
294 */
295 static void render(private_request_t *this, char *template)
296 {
297 NEOERR* err;
298
299 thread_this->set(thread_this, this);
300 err = cgi_display(this->cgi, template);
301 if (err)
302 {
303 cgi_neo_error(this->cgi, err);
304 nerr_log_error(err);
305 }
306 return;
307 }
308
309 /**
310 * Implementation of request_t.streamf.
311 */
312 static int streamf(private_request_t *this, char *format, ...)
313 {
314 va_list args;
315 int written;
316
317 va_start(args, format);
318 written = FCGX_VFPrintF(this->req.out, format, args);
319 va_end(args);
320 if (written >= 0 &&
321 FCGX_FFlush(this->req.out) == -1)
322 {
323 return -1;
324 }
325 return written;
326 }
327
328 /**
329 * Implementation of request_t.set.
330 */
331 static void set(private_request_t *this, char *key, char *value)
332 {
333 hdf_set_value(this->hdf, key, value);
334 }
335
336 /**
337 * Implementation of request_t.setf.
338 */
339 static void setf(private_request_t *this, char *format, ...)
340 {
341 va_list args;
342
343 va_start(args, format);
344 hdf_set_valuevf(this->hdf, format, args);
345 va_end(args);
346 }
347
348 /**
349 * Implementation of request_t.get_ref.
350 */
351 static request_t* get_ref(private_request_t *this)
352 {
353 ref_get(&this->ref);
354 return &this->public;
355 }
356
357 /**
358 * Implementation of request_t.destroy
359 */
360 static void destroy(private_request_t *this)
361 {
362 if (ref_put(&this->ref))
363 {
364 thread_this->set(thread_this, this);
365 cgi_destroy(&this->cgi);
366 FCGX_Finish_r(&this->req);
367 free(this);
368 }
369 }
370
371 /**
372 * This initialization method is guaranteed to run only once
373 * for all threads.
374 */
375 static void init(void)
376 {
377 cgiwrap_init_emu(NULL, read_cb, writef_cb, write_cb,
378 getenv_cb, putenv_cb, iterenv_cb);
379 thread_this = thread_value_create(NULL);
380 }
381
382 /*
383 * see header file
384 */
385 request_t *request_create(int fd, bool debug)
386 {
387 NEOERR* err;
388 private_request_t *this = malloc_thing(private_request_t);
389 bool failed = FALSE;
390
391 thread_cleanup_push(free, this);
392 if (FCGX_InitRequest(&this->req, fd, 0) != 0 ||
393 FCGX_Accept_r(&this->req) != 0)
394 {
395 failed = TRUE;
396 }
397 thread_cleanup_pop(failed);
398 if (failed)
399 {
400 return NULL;
401 }
402
403 this->public.get_path = (char*(*)(request_t*))get_path;
404 this->public.get_base = (char*(*)(request_t*))get_base;
405 this->public.get_host = (char*(*)(request_t*))get_host;
406 this->public.get_user_agent = (char*(*)(request_t*))get_user_agent;
407 this->public.add_cookie = (void(*)(request_t*, char *name, char *value))add_cookie;
408 this->public.get_cookie = (char*(*)(request_t*,char*))get_cookie;
409 this->public.get_query_data = (char*(*)(request_t*, char *name))get_query_data;
410 this->public.session_closed = (bool(*)(request_t*))session_closed;
411 this->public.close_session = (void(*)(request_t*))close_session;
412 this->public.redirect = (void(*)(request_t*, char *fmt,...))redirect;
413 this->public.get_referer = (char*(*)(request_t*))get_referer;
414 this->public.to_referer = (void(*)(request_t*))to_referer;
415 this->public.render = (void(*)(request_t*,char*))render;
416 this->public.streamf = (int(*)(request_t*, char *format, ...))streamf;
417 this->public.serve = (void(*)(request_t*,char*,chunk_t))serve;
418 this->public.set = (void(*)(request_t*, char *, char*))set;
419 this->public.setf = (void(*)(request_t*, char *format, ...))setf;
420 this->public.get_ref = (request_t*(*)(request_t*))get_ref;
421 this->public.destroy = (void(*)(request_t*))destroy;
422
423 pthread_once(&once, init);
424 thread_this->set(thread_this, this);
425
426 this->ref = 1;
427 this->closed = FALSE;
428 this->req_env_len = 0;
429 while (this->req.envp[this->req_env_len] != NULL)
430 {
431 this->req_env_len++;
432 }
433
434 err = hdf_init(&this->hdf);
435 if (!err)
436 {
437 hdf_set_value(this->hdf, "base", get_base(this));
438 hdf_set_value(this->hdf, "Config.NoCache", "true");
439 if (!debug)
440 {
441 hdf_set_value(this->hdf, "Config.TimeFooter", "0");
442 hdf_set_value(this->hdf, "Config.CompressionEnabled", "1");
443 hdf_set_value(this->hdf, "Config.WhiteSpaceStrip", "2");
444 }
445
446 err = cgi_init(&this->cgi, this->hdf);
447 if (!err)
448 {
449 err = cgi_parse(this->cgi);
450 if (!err)
451 {
452 return &this->public;
453 }
454 cgi_destroy(&this->cgi);
455 }
456 }
457 nerr_log_error(err);
458 FCGX_Finish_r(&this->req);
459 free(this);
460 return NULL;
461 }
462