Add a getter for the HTTP referer
[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.add_cookie.
208 */
209 static void add_cookie(private_request_t *this, char *name, char *value)
210 {
211 thread_this->set(thread_this, this);
212 cgi_cookie_set (this->cgi, name, value,
213 FCGX_GetParam("SCRIPT_NAME", this->req.envp),
214 NULL, NULL, 0, 0);
215 }
216
217 /**
218 * Implementation of request_t.redirect.
219 */
220 static void redirect(private_request_t *this, char *fmt, ...)
221 {
222 va_list args;
223
224 FCGX_FPrintF(this->req.out, "Status: 303 See Other\n");
225 FCGX_FPrintF(this->req.out, "Location: %s%s",
226 FCGX_GetParam("SCRIPT_NAME", this->req.envp),
227 *fmt == '/' ? "" : "/");
228 va_start(args, fmt);
229 FCGX_VFPrintF(this->req.out, fmt, args);
230 va_end(args);
231 FCGX_FPrintF(this->req.out, "\n\n");
232 }
233
234 /**
235 * Implementation of request_t.get_referer.
236 */
237 static char* get_referer(private_request_t *this)
238 {
239 return FCGX_GetParam("HTTP_REFERER", this->req.envp);
240 }
241
242 /**
243 * Implementation of request_t.to_referer.
244 */
245 static void to_referer(private_request_t *this)
246 {
247 FCGX_FPrintF(this->req.out, "Status: 303 See Other\n");
248 FCGX_FPrintF(this->req.out, "Location: %s\n\n",
249 FCGX_GetParam("HTTP_REFERER", this->req.envp));
250 }
251
252 /**
253 * Implementation of request_t.get_base.
254 */
255 static char* get_base(private_request_t *this)
256 {
257 return FCGX_GetParam("SCRIPT_NAME", this->req.envp);
258 }
259
260 /**
261 * Implementation of request_t.session_closed.
262 */
263 static bool session_closed(private_request_t *this)
264 {
265 return this->closed;
266 }
267
268 /**
269 * Implementation of request_t.close_session.
270 */
271 static void close_session(private_request_t *this)
272 {
273 this->closed = TRUE;
274 }
275
276 /**
277 * Implementation of request_t.serve.
278 */
279 static void serve(private_request_t *this, char *headers, chunk_t chunk)
280 {
281 FCGX_FPrintF(this->req.out, "%s\n\n", headers);
282
283 FCGX_PutStr(chunk.ptr, chunk.len, this->req.out);
284 }
285
286 /**
287 * Implementation of request_t.render.
288 */
289 static void render(private_request_t *this, char *template)
290 {
291 NEOERR* err;
292
293 thread_this->set(thread_this, this);
294 err = cgi_display(this->cgi, template);
295 if (err)
296 {
297 cgi_neo_error(this->cgi, err);
298 nerr_log_error(err);
299 }
300 return;
301 }
302
303 /**
304 * Implementation of request_t.streamf.
305 */
306 static int streamf(private_request_t *this, char *format, ...)
307 {
308 va_list args;
309 int written;
310
311 va_start(args, format);
312 written = FCGX_VFPrintF(this->req.out, format, args);
313 va_end(args);
314 if (written >= 0 &&
315 FCGX_FFlush(this->req.out) == -1)
316 {
317 return -1;
318 }
319 return written;
320 }
321
322 /**
323 * Implementation of request_t.set.
324 */
325 static void set(private_request_t *this, char *key, char *value)
326 {
327 hdf_set_value(this->hdf, key, value);
328 }
329
330 /**
331 * Implementation of request_t.setf.
332 */
333 static void setf(private_request_t *this, char *format, ...)
334 {
335 va_list args;
336
337 va_start(args, format);
338 hdf_set_valuevf(this->hdf, format, args);
339 va_end(args);
340 }
341
342 /**
343 * Implementation of request_t.get_ref.
344 */
345 static request_t* get_ref(private_request_t *this)
346 {
347 ref_get(&this->ref);
348 return &this->public;
349 }
350
351 /**
352 * Implementation of request_t.destroy
353 */
354 static void destroy(private_request_t *this)
355 {
356 if (ref_put(&this->ref))
357 {
358 thread_this->set(thread_this, this);
359 cgi_destroy(&this->cgi);
360 FCGX_Finish_r(&this->req);
361 free(this);
362 }
363 }
364
365 /**
366 * This initialization method is guaranteed to run only once
367 * for all threads.
368 */
369 static void init(void)
370 {
371 cgiwrap_init_emu(NULL, read_cb, writef_cb, write_cb,
372 getenv_cb, putenv_cb, iterenv_cb);
373 thread_this = thread_value_create(NULL);
374 }
375
376 /*
377 * see header file
378 */
379 request_t *request_create(int fd, bool debug)
380 {
381 NEOERR* err;
382 private_request_t *this = malloc_thing(private_request_t);
383 bool failed = FALSE;
384
385 thread_cleanup_push(free, this);
386 if (FCGX_InitRequest(&this->req, fd, 0) != 0 ||
387 FCGX_Accept_r(&this->req) != 0)
388 {
389 failed = TRUE;
390 }
391 thread_cleanup_pop(failed);
392 if (failed)
393 {
394 return NULL;
395 }
396
397 this->public.get_path = (char*(*)(request_t*))get_path;
398 this->public.get_base = (char*(*)(request_t*))get_base;
399 this->public.get_host = (char*(*)(request_t*))get_host;
400 this->public.get_user_agent = (char*(*)(request_t*))get_user_agent;
401 this->public.add_cookie = (void(*)(request_t*, char *name, char *value))add_cookie;
402 this->public.get_cookie = (char*(*)(request_t*,char*))get_cookie;
403 this->public.get_query_data = (char*(*)(request_t*, char *name))get_query_data;
404 this->public.session_closed = (bool(*)(request_t*))session_closed;
405 this->public.close_session = (void(*)(request_t*))close_session;
406 this->public.redirect = (void(*)(request_t*, char *fmt,...))redirect;
407 this->public.get_referer = (char*(*)(request_t*))get_referer;
408 this->public.to_referer = (void(*)(request_t*))to_referer;
409 this->public.render = (void(*)(request_t*,char*))render;
410 this->public.streamf = (int(*)(request_t*, char *format, ...))streamf;
411 this->public.serve = (void(*)(request_t*,char*,chunk_t))serve;
412 this->public.set = (void(*)(request_t*, char *, char*))set;
413 this->public.setf = (void(*)(request_t*, char *format, ...))setf;
414 this->public.get_ref = (request_t*(*)(request_t*))get_ref;
415 this->public.destroy = (void(*)(request_t*))destroy;
416
417 pthread_once(&once, init);
418 thread_this->set(thread_this, this);
419
420 this->ref = 1;
421 this->closed = FALSE;
422 this->req_env_len = 0;
423 while (this->req.envp[this->req_env_len] != NULL)
424 {
425 this->req_env_len++;
426 }
427
428 err = hdf_init(&this->hdf);
429 if (!err)
430 {
431 hdf_set_value(this->hdf, "base", get_base(this));
432 hdf_set_value(this->hdf, "Config.NoCache", "true");
433 if (!debug)
434 {
435 hdf_set_value(this->hdf, "Config.TimeFooter", "0");
436 hdf_set_value(this->hdf, "Config.CompressionEnabled", "1");
437 hdf_set_value(this->hdf, "Config.WhiteSpaceStrip", "2");
438 }
439
440 err = cgi_init(&this->cgi, this->hdf);
441 if (!err)
442 {
443 err = cgi_parse(this->cgi);
444 if (!err)
445 {
446 return &this->public;
447 }
448 cgi_destroy(&this->cgi);
449 }
450 }
451 nerr_log_error(err);
452 FCGX_Finish_r(&this->req);
453 free(this);
454 return NULL;
455 }
456