source: roaraudio/plugins/universal/protocol-http.c @ 5726:4a15c13827a8

Last change on this file since 5726:4a15c13827a8 was 5726:4a15c13827a8, checked in by phi, 11 years ago

allow passing GET parameters thru uste

File size: 27.8 KB
Line 
1//protocol-http.c:
2
3/*
4 *      Copyright (C) Philipp 'ph3-der-loewe' Schafft - 2011-2012
5 *
6 *  This file is part of roard a part of RoarAudio,
7 *  a cross-platform sound system for both, home and professional use.
8 *  See README for details.
9 *
10 *  This file is free software; you can redistribute it and/or modify
11 *  it under the terms of the GNU General Public License version 3
12 *  as published by the Free Software Foundation.
13 *
14 *  RoarAudio is distributed in the hope that it will be useful,
15 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
16 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17 *  GNU General Public License for more details.
18 *
19 *  You should have received a copy of the GNU General Public License
20 *  along with this software; see the file COPYING.  If not, write to
21 *  the Free Software Foundation, 51 Franklin Street, Fifth Floor,
22 *  Boston, MA 02110-1301, USA.
23 *
24 */
25
26#include <roaraudio.h>
27#ifdef ROAR_HAVE_LIBUSTE
28#include <uste.h>
29#endif
30
31struct resource_handle;
32
33enum ctl_cmd {
34 CMD_SET_CLIENTNAME
35};
36
37enum status {
38 STATUS_WAITING_FOR_HEADERS = 0,
39 STATUS_END_OF_HEADER,
40 STATUS_BAD_HEADER,
41 STATUS_RUNNING_DATA,
42 STATUS_DONE
43};
44
45struct input_source {
46 int userdata_si;
47 void * userdata_vp;
48 ssize_t offset;
49 struct roar_vio_calls * vio;
50 struct roar_vio_calls vio_store;
51};
52
53struct http_client {
54 enum status status;
55 struct roar_buffer * input_buffer;
56 char * method;
57 char * proto;
58 char * uri;
59 char * path, * query_string;
60 void * headersstore;
61 struct roar_keyval * headers;
62 ssize_t headerslen;
63 const struct resource_handle * resource;
64 struct input_source input;
65 struct roar_dl_librarypara * pluginpara;
66 int clientid;
67};
68
69struct resource_handle {
70 const char  * uri;
71 const char  * rawdata;
72 const ssize_t rawlen;
73 const char  * content_type;
74 int (*header_send)(struct http_client * self, struct roar_buffer ** obuffer);
75 int (*body_send)(struct http_client * self, struct roar_buffer ** obuffer);
76};
77
78static struct {
79 enum {
80  HOSTTYPE_GENERIC = 0,
81  HOSTTYPE_ROARD
82 } hosttype;
83 union {
84  struct {
85   int (*clients_set_name)(int id, const char * name);
86  } roard;
87 } hostspec;
88} __host = {.hosttype = HOSTTYPE_GENERIC};
89
90static int header_send(struct roar_buffer ** obuffer, int status, const char * msg, const char * content_type, ssize_t len);
91static void send_errorpage(struct http_client * self, struct roar_buffer ** obuffer, int error, const char * msg);
92
93static int slow_zero_stream(struct http_client * self, struct roar_buffer ** obuffer) {
94 struct roar_buffer * buffer;
95 void * data;
96
97 self->status = STATUS_RUNNING_DATA;
98
99 if ( roar_buffer_new_data(&buffer, 1, &data) == -1 )
100  return -1;
101
102 *((char*)data) = 0;
103
104 if ( roar_buffer_moveintoqueue(obuffer, &buffer) == -1 ) {
105  roar_buffer_free(buffer);
106  return -1;
107 }
108
109 return 0;
110}
111
112static const char * __res_vio_content_type_get(struct http_client * self) {
113 const char * point = rindex(self->path, '.');
114
115 if ( point == NULL )
116  return NULL;
117
118 point++;
119
120 if ( !strcmp(point, "txt") || !strcmp(point, "text") ) {
121  return "text/plain";
122 } else if ( !strcmp(point, "htm") || !strcmp(point, "html") ) {
123  return "text/html";
124 } else if ( !strcmp(point, "css") ) {
125  return "text/css";
126 } else if ( !strcmp(point, "png") ) {
127  return "image/png";
128 } else if ( !strcmp(point, "jpg") || !strcmp(point, "jpeg") ) {
129  return "image/jpeg";
130 }
131
132 return NULL;
133}
134
135#ifdef ROAR_HAVE_LIBUSTE
136static uste_var_t __res_vio_handle_uste_build_headers(struct http_client * self) {
137 uste_var_t root = uste_var_new_kv();
138 uste_var_t var;
139 ssize_t i;
140
141 if ( root == NULL )
142  return NULL;
143
144 uste_var_set_key(root, "header");
145
146 for (i = 0; i < self->headerslen; i++) {
147  var = uste_var_new_str(self->headers[i].value == NULL ? "" : self->headers[i].value);
148  if ( self->headers[i].key != NULL )
149   uste_var_set_key(var, self->headers[i].key);
150  uste_var_push(root, var);
151  uste_var_unref(var);
152 }
153
154 return root;
155}
156
157static inline int __is_hex(char c) {
158 if ( c >= '0' && c <= '9' )
159  return 1;
160 if ( c >= 'a' && c <= 'f' )
161  return 1;
162 if ( c >= 'A' && c <= 'F' )
163  return 1;
164 return 0;
165}
166
167static inline int __hex2num(char c) {
168 if ( c >= '0' && c <= '9' )
169  return c - '0';
170 if ( c >= 'a' && c <= 'f' )
171  return c - 'a' + 10;
172 if ( c >= 'A' && c <= 'F' )
173  return c - 'A' + 10;
174 return 0;
175}
176
177static void __res_vio_handle_uste_uridecode(char * str) {
178 const char * c = str;
179
180 for (; *c; c++) {
181  if ( *c == '%' && __is_hex(c[1]) && __is_hex(c[2]) ) {
182   *str = __hex2num(c[1])*16 + __hex2num(c[2]);
183   str++;
184   c += 2;
185  } else {
186   *str = *c;
187   str++;
188  }
189 }
190 *str = *c;
191}
192
193static void __res_vio_handle_uste_keyify(char * str) {
194 if ( *str == '#' )
195  *str = '_';
196 for (;*str; str++)
197  if (*str == '.')
198   *str = '_';
199}
200
201static uste_var_t __res_vio_handle_uste_build_getparams(struct http_client * self) {
202 uste_var_t root = uste_var_new_kv();
203 uste_var_t subroot;
204 uste_var_t var;
205 char * qs_buffer;
206 char * cur;
207 char * state;
208 char * key, * value;
209
210 if ( root == NULL )
211  return NULL;
212
213 uste_var_set_key(root, "get");
214 if ( self->query_string == NULL )
215  return root;
216
217 qs_buffer = roar_mm_strdup(self->query_string);
218
219 // do better handling here.
220 if ( qs_buffer == NULL )
221  return root;
222
223 cur = roar_mm_strtok_r(qs_buffer, "&", &state);
224
225 while (cur != NULL) {
226  printf("cur='%s'\n", cur);
227  key = cur;
228  value = strstr(cur, "=");
229  if ( value != NULL ) {
230   *value = 0;
231   value++;
232   __res_vio_handle_uste_uridecode(value);
233  }
234  __res_vio_handle_uste_uridecode(key);
235  __res_vio_handle_uste_keyify(key);
236
237  if ( value == NULL ) {
238   var = uste_var_new_undef();
239  } else {
240   var = uste_var_new_str(value);
241  }
242
243  subroot = uste_var_pull(root, key);
244  if ( subroot == NULL ) {
245   subroot = uste_var_new_array();
246   if ( subroot != NULL ) {
247    uste_var_set_key(subroot, key);
248    uste_var_push(root, subroot);
249   }
250  }
251
252  if ( subroot != NULL ) {
253   uste_var_push(subroot, var);
254   uste_var_unref(subroot);
255  }
256  uste_var_unref(var);
257  cur = roar_mm_strtok_r(NULL, "&", &state);
258 }
259
260 roar_mm_free(qs_buffer);
261 return root;
262}
263
264static uste_var_t __res_vio_handle_uste_build_rootkv(struct http_client * self) {
265 uste_var_t root = uste_var_new_kv();
266 uste_var_t subroot, subsubroot;
267 uste_var_t var;
268
269 if ( root == NULL )
270  return NULL;
271
272#define __new_member_prefix(prefix,name) \
273   if ( prefix->name != NULL ) { \
274    var = uste_var_new_str(prefix->name); \
275    uste_var_set_key(var, #name); \
276    uste_var_push(subsubroot, var); \
277    uste_var_unref(var); \
278   }
279#define __new_member(name) __new_member_prefix(self,name)
280
281 subroot = uste_var_new_kv();
282 if ( subroot != NULL ) {
283  uste_var_set_key(subroot, "__client__");
284  uste_var_push(root, subroot);
285
286  var = uste_var_new_int(self->clientid);
287  uste_var_set_key(var, "clientid");
288  uste_var_push(subroot, var);
289  uste_var_unref(var);
290
291  subsubroot = uste_var_new_kv();
292  if ( subsubroot != NULL ) {
293   uste_var_set_key(subsubroot, "http");
294   uste_var_push(subroot, subsubroot);
295
296   __new_member(method);
297   __new_member(proto);
298   __new_member(uri);
299   __new_member(path);
300   __new_member(query_string);
301
302   var = __res_vio_handle_uste_build_headers(self);
303   uste_var_push(subsubroot, var);
304   uste_var_unref(var);
305
306   var = __res_vio_handle_uste_build_getparams(self);
307   uste_var_push(subsubroot, var);
308   uste_var_unref(var);
309
310   uste_var_unref(subsubroot);
311  }
312
313  uste_var_unref(subroot);
314 }
315
316 subroot = uste_var_new_kv();
317 if ( subroot != NULL ) {
318  uste_var_set_key(subroot, "__host__");
319  uste_var_push(root, subroot);
320
321  subsubroot = subroot;
322  __new_member_prefix(self->pluginpara, appname);
323  __new_member_prefix(self->pluginpara, abiversion);
324
325  uste_var_unref(subroot);
326 }
327
328 return root;
329}
330
331static void __res_vio_handle_uste(struct http_client * self, struct roar_buffer ** obuffer, const char * content_type) {
332 struct roar_buffer * buffer = NULL;
333 uste_renderer_t renderer;
334 uste_parser_t parser;
335 uste_var_t rootkv;
336 uste_node_t root;
337 int err;
338
339
340 parser = uste_parser_new(self->input.vio);
341 err = roar_error;
342 roar_vio_close(self->input.vio);
343 if ( parser == NULL ) {
344  send_errorpage(self, obuffer, err, NULL);
345  return;
346 }
347
348 if ( uste_parser_parse(parser) == -1 ) {
349  send_errorpage(self, obuffer, roar_error, NULL);
350  return;
351 }
352
353 root = uste_parser_get_rootnode(parser);
354 err = roar_error;
355 uste_parser_unref(parser);
356
357 if ( root == NULL ) {
358  send_errorpage(self, obuffer, err, NULL);
359  return;
360 }
361
362 renderer = uste_renderer_new();
363 err = roar_error;
364 if ( renderer == NULL ) {
365  uste_node_unref(root);
366  send_errorpage(self, obuffer, err, NULL);
367  return;
368 }
369
370 uste_stdfunc_register_all(renderer);
371
372 rootkv = __res_vio_handle_uste_build_rootkv(self);
373 err = roar_error;
374 if ( rootkv == NULL ) {
375  uste_node_unref(root);
376  uste_renderer_unref(renderer);
377  send_errorpage(self, obuffer, err, NULL);
378  return;
379 }
380
381 uste_renderer_set_rootvar(renderer, rootkv);
382 uste_var_unref(rootkv);
383
384 buffer = uste_node_render(root, renderer);
385 err = roar_error;
386 uste_node_unref(root);
387 uste_renderer_unref(renderer);
388
389 if ( buffer == NULL ) {
390  send_errorpage(self, obuffer, err, NULL);
391  return;
392 }
393
394 header_send(obuffer, 200, NULL, content_type, -1);
395
396 if ( buffer != NULL ) {
397  if ( roar_buffer_moveintoqueue(obuffer, &buffer) == -1 ) {
398   roar_buffer_free(buffer);
399   return;
400  }
401 }
402
403 self->status = STATUS_DONE;
404}
405#endif
406
407static int __res_vio_check_uste(struct http_client * self, struct roar_buffer ** obuffer, const char * content_type) {
408 const size_t len = 14;
409 struct roar_buffer * buffer;
410 void * data;
411 ssize_t ret;
412
413 // TODO: use content type for some basic filtering.
414 (void)content_type;
415
416 *obuffer = NULL;
417
418 if ( roar_buffer_new_data(&buffer, len, &data) == -1 )
419  return -1;
420
421 ret = roar_vio_read(self->input.vio, data, len);
422
423 if ( ret < 1 ) {
424  roar_buffer_free(buffer);
425  return 0;
426 }
427
428 *obuffer = buffer;
429
430 if ( ret == (ssize_t)len ) {
431  return !strncmp(data, "@@@TEMPLATE@@@", len);
432 } else {
433  if ( roar_buffer_set_len(buffer, ret) == -1 ) {
434   *obuffer = NULL;
435   roar_buffer_free(buffer);
436   return -1;
437  }
438  // is this really a nice idea?
439  return 0;
440 }
441}
442
443static int __res_vio_header_send(struct http_client * self, struct roar_buffer ** obuffer) {
444 struct roar_buffer * buffer;
445 struct roar_keyval * path;
446 char filename[1024];
447 const char * slash = "/";
448 const char * content_type = NULL;
449
450 if ( self->pluginpara == NULL ) {
451  roar_err_set(ROAR_ERROR_INVAL);
452  return -1;
453 }
454
455 path = roar_keyval_lookup(self->pluginpara->argv, "webroot", self->pluginpara->argc, 1);
456 if ( path == NULL ) {
457  if ( roar_error == ROAR_ERROR_NOENT )
458   roar_err_set(ROAR_ERROR_INVAL);
459  return -1;
460 }
461
462 if ( path->value == NULL ) {
463  roar_err_set(ROAR_ERROR_INVAL);
464  return -1;
465 }
466
467 if ( strstr(self->path, "..") != NULL || strstr(self->path, "#") != NULL ) {
468  roar_err_set(ROAR_ERROR_INVAL);
469  return -1;
470 }
471
472 if ( (roar_mm_strlen(path->value) + roar_mm_strlen(self->path) + 2) > sizeof(filename) ) {
473  roar_err_set(ROAR_ERROR_NAMETOOLONG);
474  return -1;
475 }
476
477 if ( self->path[0] == '/' || path->value[strlen(path->value)-1] == '/' )
478  slash = "";
479
480 snprintf(filename, sizeof(filename), "%s%s%s", path->value, slash, self->path);
481
482
483 self->input.vio = &(self->input.vio_store);
484 if ( roar_vio_open_dstr_simple(self->input.vio, filename, ROAR_VIOF_READ|ROAR_VIOF_NONBLOCK) == -1 ) {
485  return -1;
486 }
487
488 if ( roar_vio_ctl(self->input.vio, ROAR_VIO_CTL_GET_MIMETYPE, &content_type) == -1 )
489  content_type = NULL;
490
491 if ( content_type == NULL )
492  content_type = __res_vio_content_type_get(self);
493
494 if ( __res_vio_check_uste(self, &buffer, content_type) == 1 ) {
495  if ( buffer != NULL )
496   roar_buffer_free(buffer);
497#ifdef ROAR_HAVE_LIBUSTE
498  __res_vio_handle_uste(self, obuffer, content_type);
499  return 0;
500#else
501  roar_vio_close(self->input.vio);
502  send_errorpage(self, obuffer, ROAR_ERROR_NOSYS, NULL);
503  return 0;
504#endif
505 }
506
507 header_send(obuffer, 200, NULL, content_type, -1);
508
509 if ( buffer != NULL ) {
510  if ( roar_buffer_moveintoqueue(obuffer, &buffer) == -1 ) {
511   roar_buffer_free(buffer);
512   return -1;
513  }
514 }
515
516 self->status = STATUS_RUNNING_DATA;
517 return 0;
518}
519
520static int __res_vio_body_send(struct http_client * self, struct roar_buffer ** obuffer) {
521 const size_t len = 1024;
522 struct roar_buffer * buffer;
523 void * data;
524 ssize_t ret;
525
526 if ( self->status == STATUS_DONE )
527  return 0;
528
529 if ( roar_buffer_new_data(&buffer, len, &data) == -1 )
530  return -1;
531
532 ret = roar_vio_read(self->input.vio, data, len);
533
534 if ( ret == -1 && roar_error == ROAR_ERROR_AGAIN ) {
535  ret = 0;
536 } else if ( ret < 1 ) {
537  roar_buffer_free(buffer);
538  if ( roar_error != ROAR_ERROR_AGAIN ) {
539   self->status = STATUS_DONE;
540   roar_vio_close(self->input.vio);
541  }
542  return 0;
543 }
544
545 if ( roar_buffer_set_len(buffer, ret) == -1 ) {
546  roar_buffer_free(buffer);
547  return -1;
548 }
549
550 if ( roar_buffer_moveintoqueue(obuffer, &buffer) == -1 ) {
551  roar_buffer_free(buffer);
552  return -1;
553 }
554
555 return 0;
556}
557
558static const struct resource_handle _g_resources[] = {
559 {
560  .uri = "/test/*",
561  .rawdata = "Hello world!\n",
562  .rawlen  = 13,
563  .content_type = NULL,
564  .header_send = NULL,
565  .body_send = NULL
566 },
567 {
568  .uri = "/szs/*",
569  .rawdata = NULL,
570  .rawlen  = -1,
571  .content_type = NULL,
572  .header_send = NULL,
573  .body_send = slow_zero_stream
574 },
575 {
576  .uri = "*",
577  .rawdata = NULL,
578  .rawlen  = -1,
579  .content_type = NULL,
580  .header_send = __res_vio_header_send,
581  .body_send = __res_vio_body_send
582 }
583};
584
585
586static inline int __ret(struct http_client * self, struct roar_buffer ** obuffer) {
587 ROAR_DBG("__ret(self=%p, obuffer=%p{%p}): self->status=%i", self, obuffer, *obuffer, (int)self->status);
588 if ( *obuffer != NULL )
589  return 0;
590 if ( self->status == STATUS_DONE )
591  return -1;
592 return 0;
593}
594
595static int __ctl(struct http_client * self, enum ctl_cmd cmd, void * argp) {
596 switch (cmd) {
597  case CMD_SET_CLIENTNAME:
598    if ( argp == NULL ) {
599     roar_err_set(ROAR_ERROR_FAULT);
600     return -1;
601    }
602    ROAR_DBG("__ctl(self=%p, cmd=%i, argp='%s') = ?", self, (int)cmd, (const char *)argp);
603    switch (__host.hosttype) {
604     case HOSTTYPE_GENERIC:
605       roar_err_set(ROAR_ERROR_BADHOST);
606       return -1;
607      break;
608     case HOSTTYPE_ROARD:
609       __host.hostspec.roard.clients_set_name(self->clientid, argp);
610      break;
611    }
612   break;
613 }
614
615 roar_err_set(ROAR_ERROR_BADRQC);
616 return -1;
617}
618
619static void __init(struct roar_dl_librarypara * para) {
620 // TODO add some init code here.
621
622 if ( para == NULL )
623  return;
624
625 if ( !roar_dl_para_check_version(para, "roard <0/RoarAudio>", "1.0beta8") ) {
626  __host.hosttype = HOSTTYPE_ROARD;
627  __host.hostspec.roard.clients_set_name = roar_dl_getsym(ROAR_DL_HANDLE_APPLICATION, "clients_set_name", -1);
628
629  // check if *all* function have been found:
630  if ( __host.hostspec.roard.clients_set_name == NULL ) {
631   __host.hosttype = HOSTTYPE_GENERIC;
632  }
633 }
634}
635
636// this function is a wrapper for __init() with the code which should be inlined.
637static inline void _init(struct roar_dl_librarypara * para) {
638 static int inited = 0;
639 if (inited) return;
640 __init(para);
641 inited = 1;
642}
643
644static const struct resource_handle * resource_lookup_by_uri(const char * uri) {
645 size_t i;
646
647 for (i = 0; i < (sizeof(_g_resources)/sizeof(*_g_resources)); i++) {
648  if ( !roar_mm_strselcmp(_g_resources[i].uri, uri) ) {
649   return &(_g_resources[i]);
650  }
651 }
652
653 roar_err_set(ROAR_ERROR_NOENT);
654 return NULL;
655}
656
657static int resource_header_send(struct http_client * self, struct roar_buffer ** obuffer) {
658 if ( self->resource->header_send != NULL )
659  return self->resource->header_send(self, obuffer);
660
661 return header_send(obuffer, 200, NULL, self->resource->content_type, self->resource->rawlen);
662}
663
664static int resource_body_send(struct http_client * self, struct roar_buffer ** obuffer) {
665 struct roar_buffer * buffer;
666 void * data;
667 int ret;
668
669 if ( self->resource->body_send != NULL )
670  return self->resource->body_send(self, obuffer);
671
672 if ( self->resource->rawlen == -1 ) {
673  self->status = STATUS_DONE;
674  roar_err_set(ROAR_ERROR_INVAL);
675  return -1;
676 }
677
678 if ( roar_buffer_new_data(&buffer, self->resource->rawlen, &data) == -1 ) {
679  self->status = STATUS_DONE;
680  return -1;
681 }
682
683 memcpy(data, self->resource->rawdata, self->resource->rawlen);
684
685 ret = roar_buffer_moveintoqueue(obuffer, &buffer);
686 self->status = STATUS_DONE;
687
688 return ret;
689}
690
691static int header_parse(struct http_client * self) {
692 void * data;
693 size_t len;
694 size_t i;
695 char * start, * end;
696 char * start_uri;
697 char * start_headers, * end_headers = NULL;
698 char * tmp;
699 struct roar_keyval * c;
700
701 if ( roar_buffer_get_datalen(self->input_buffer, &data, &len) == -1 )
702  return -1;
703
704 ROAR_DBG("header_parse(self=%p): data='%*s'", self, (int)len, (const char*)data);
705
706 if ( len < 4 )
707  return 0;
708
709 for (i = 0; i < (len - 3); i++) {
710  if ( ((const char*)data)[i] == '\r' ) {
711   ROAR_DBG("header_parse(self=%p): i=%i", self, (int)i);
712   if ( !strncmp(data + i, "\r\n\r\n", 4) ) {
713    self->status = STATUS_END_OF_HEADER;
714    end_headers = data + i;
715    break;
716   }
717  }
718 }
719
720 if ( self->status != STATUS_END_OF_HEADER ) {
721  for (i = 0; i < (len - 1); i++) {
722   if ( ((const char*)data)[i] == '\n' ) {
723    ROAR_DBG("header_parse(self=%p): i=%i", self, (int)i);
724    if ( !strncmp(data + i, "\n\n", 2) ) {
725     self->status = STATUS_END_OF_HEADER;
726     end_headers = data + i;
727     break;
728    }
729   }
730  }
731 }
732
733 if ( self->status != STATUS_END_OF_HEADER )
734  return 0;
735
736 *end_headers = 0;
737
738 end = strstr(data, " ");
739 if ( end == NULL ) {
740  self->status = STATUS_BAD_HEADER;
741  return 1;
742 }
743
744 *end = 0;
745 end++;
746 self->method = roar_mm_strdup(data);
747
748 start = end;
749
750 end = strstr(start, " ");
751 if ( end == NULL ) {
752  self->status = STATUS_BAD_HEADER;
753  return 1;
754 }
755
756 *end = 0;
757 end++;
758 self->uri = roar_mm_strdup(start);
759 start_uri = start;
760
761 start = end;
762
763 end = strstr(start, "\r");
764 if ( end == NULL )
765  end = strstr(start, "\n");
766 if ( end == NULL ) {
767  self->status = STATUS_BAD_HEADER;
768  return 1;
769 }
770
771 *end = 0;
772 end++;
773 self->proto = roar_mm_strdup(start);
774
775 start_headers = end;
776
777 start = start_uri;
778
779 end = strstr(start, "?");
780 if ( end == NULL ) {
781  self->path = roar_mm_strdup(start);
782 } else {
783  *end = 0;
784  end++;
785  self->path = roar_mm_strdup(start);
786  self->query_string = roar_mm_strdup(end);
787 }
788
789 self->headerslen = 0;
790
791 i = end_headers - start_headers + 1;
792 self->headersstore = roar_mm_memdup(start_headers, i);
793 if ( self->headersstore == NULL ) {
794  self->status = STATUS_BAD_HEADER; // TODO: FIXME: not the right error, but works. ;)
795  return -1;
796 }
797
798 start_headers = self->headersstore;
799 end_headers   = start_headers + i;
800
801 for (tmp = start_headers; tmp < end_headers;) {
802  for (; tmp < end_headers && (*tmp == '\r' || *tmp == '\n'); tmp++);
803  if ( tmp == end_headers )
804   break;
805  self->headerslen++;
806  for (; tmp < end_headers && !(*tmp == '\r' || *tmp == '\n'); tmp++);
807 }
808
809 ROAR_DBG("header_parse(self=%p): self->headerslen=%i", self, (int)self->headerslen);
810
811 self->headers = roar_mm_malloc(sizeof(struct roar_keyval)*(self->headerslen+1));
812 if ( self->headers == NULL ) {
813  self->status = STATUS_BAD_HEADER; // TODO: FIXME: not the right error, but works. ;)
814  return -1;
815 }
816 memset(self->headers, 0, sizeof(struct roar_keyval)*(self->headerslen+1));
817 c = self->headers;
818
819 for (tmp = start_headers; tmp < end_headers;) {
820  ROAR_DBG("header_parse(self=%p): tmp='%s'", self, tmp);
821  for (; tmp < end_headers && (*tmp == '\r' || *tmp == '\n'); tmp++) *tmp = '\0';
822  if ( tmp == end_headers )
823   break;
824  ROAR_DBG("header_parse(self=%p): tmp='%s'", self, tmp);
825  c->key = tmp;
826  ROAR_DBG("header_parse(self=%p): c->key='%s'", self, c->key);
827  for (; tmp < end_headers && !(*tmp == '\r' || *tmp == '\n' || *tmp == ':'); tmp++);
828  c->value = tmp;
829  for (; tmp < end_headers && !(*tmp == '\r' || *tmp == '\n'); tmp++);
830  c++;
831 }
832
833 c->key = NULL;
834 c->value = NULL;
835
836 for (i = 0; i < (size_t)self->headerslen; i++) {
837  c = &(self->headers[i]);
838  if ( c->value[0] == '\0' ) {
839   c->value = NULL;
840   continue;
841  }
842
843  c->value[0] = '\0';
844  c->value++;
845
846  for (; c->value[0] == ' '; c->value++);
847
848  if ( c->value[0] == '\0' )
849   c->value = NULL;
850 }
851
852 return 1;
853}
854
855static int header_send(struct roar_buffer ** obuffer, int status, const char * msg, const char * content_type, ssize_t len) {
856 struct roar_buffer * buffer;
857 size_t bufferlen = 1024;
858 void * data;
859 char buffer_len[64];
860
861 if ( roar_buffer_new_data(&buffer, bufferlen, &data) == -1 )
862  return -1;
863
864 if ( msg == NULL ) {
865  switch (status) {
866   case 200: msg = "OK"; break;
867   case 400: msg = "Bad Request"; break;
868   case 404: msg = "File not found"; break;
869   case 500: msg = "Internal server error"; break;
870   default:  msg = "<<<unknown status code>>>"; break;
871  }
872 }
873
874 if ( content_type == NULL )
875  content_type = "text/plain";
876
877 if ( len == (ssize_t)-1 ) {
878  buffer_len[0] = 0;
879 } else {
880  snprintf(buffer_len, sizeof(buffer_len), "Content-Length: %lu\r\n", (long unsigned int)len);
881 }
882
883/*
884Date: Sun, 29 Jul 2012 01:08:15 GMT
885Cache-Control: no-cache,no-store
886Content-Type: text/html; charset=%s
887*/
888 snprintf(data, bufferlen, "HTTP/1.0 %i %s\r\n"
889                           "Server: protocol-http (libroar plugin)\r\n"
890                           "Connection: close\r\n"
891                           "Content-Type: %s\r\n"
892                           "%s"
893                           "\r\n",
894                           status, msg, content_type, buffer_len
895                           );
896
897 if ( roar_buffer_set_len(buffer, roar_mm_strlen(data)) == -1 ) {
898  roar_buffer_free(buffer);
899  return -1;
900 }
901
902 if ( roar_buffer_moveintoqueue(obuffer, &buffer) == -1 ) {
903  roar_buffer_free(buffer);
904  return -1;
905 }
906
907 return 0;
908}
909
910static void send_errorpage(struct http_client * self, struct roar_buffer ** obuffer, int error, const char * msg) {
911 struct roar_buffer * buffer;
912 const size_t bufferlen = 1024;
913 void * data;
914 int httperror;
915
916 if ( roar_err_convert(&httperror, ROAR_ERROR_TYPE_HTTP, error, ROAR_ERROR_TYPE_ROARAUDIO) == -1 )
917  httperror = 500;
918
919 if ( msg == NULL )
920  msg = roar_error2str(error);
921
922 // send header and mark as done early so we can just return in case some of the later calls fail.
923 header_send(obuffer, httperror, msg, "text/html", -1);
924 self->status = STATUS_DONE;
925
926 if ( roar_buffer_new_data(&buffer, bufferlen, &data) == -1 )
927  return;
928
929 snprintf(data, bufferlen, "<html>\n"
930                           " <head><title>%i - %s</title></head>\n"
931                           " <body>\n"
932                           "  <h1>%i - %s</h1><hr>\n"
933                           " </body>\n"
934                           "</html>",
935                           httperror, msg, httperror, msg);
936
937 _LIBROAR_IGNORE_RET(roar_buffer_set_len(buffer, roar_mm_strlen(data)));
938
939 if ( roar_buffer_moveintoqueue(obuffer, &buffer) == -1 )
940  roar_buffer_free(buffer);
941}
942
943static void handle_client(struct http_client * self, struct roar_buffer ** obuffer) {
944 struct roar_keyval * kv;
945
946 // meta stuff:
947 // try to set client name.
948 kv = roar_keyval_lookup(self->headers, "user-agent", self->headerslen, 0);
949 if ( kv != NULL && kv->value != NULL ) {
950  __ctl(self, CMD_SET_CLIENTNAME, kv->value);
951 }
952
953 // first try lookup including query string. if nothing is found
954 // retry search with only resource path.
955 self->resource = resource_lookup_by_uri(self->uri);
956 if ( self->resource == NULL )
957  self->resource = resource_lookup_by_uri(self->path);
958
959 if ( self->resource == NULL ) {
960  send_errorpage(self, obuffer, roar_error, NULL);
961  return;
962 }
963
964 if ( resource_header_send(self, obuffer) == -1 ) {
965  send_errorpage(self, obuffer, roar_error, NULL);
966  return;
967 }
968
969 resource_body_send(self, obuffer);
970
971 //send_errorpage(self, obuffer, ROAR_ERROR_CAUSALITY, NULL);
972// send_errorpage(self, obuffer, roar_random_uint16() % ROAR_ERROR_BADLICENSE, NULL);
973 return;
974}
975
976
977static int _set_proto(int client, struct roar_vio_calls * vio, struct roar_buffer ** obuffer, void ** userdata, const struct roar_keyval * protopara, ssize_t protoparalen, struct roar_dl_librarypara * pluginpara) {
978 struct http_client * self = roar_mm_malloc(sizeof(struct http_client));
979
980 (void)vio, (void)protopara, (void)protoparalen;
981
982 if ( self == NULL )
983  return -1;
984
985 _init(pluginpara);
986
987 memset(self, 0, sizeof(*self));
988 self->status = STATUS_WAITING_FOR_HEADERS;
989 self->headerslen = (ssize_t)-1;
990 self->clientid = client;
991
992 if ( pluginpara != NULL ) {
993  roar_dl_para_ref(pluginpara);
994  self->pluginpara = pluginpara;
995 }
996
997 *userdata = self;
998
999 return __ret(self, obuffer);
1000}
1001
1002static int _unset_proto(int client, struct roar_vio_calls * vio, struct roar_buffer ** obuffer, void ** userdata, const struct roar_keyval * protopara, ssize_t protoparalen, struct roar_dl_librarypara * pluginpara) {
1003 struct http_client * self = *userdata;
1004
1005 (void)client, (void)vio, (void)obuffer, (void)protopara, (void)protoparalen, (void)pluginpara;
1006
1007 if ( self->input_buffer != NULL )
1008  roar_buffer_free(self->input_buffer);
1009
1010 if ( self->method != NULL )
1011  roar_mm_free(self->method);
1012 if ( self->proto != NULL )
1013  roar_mm_free(self->proto);
1014 if ( self->uri != NULL )
1015  roar_mm_free(self->uri);
1016 if ( self->path != NULL )
1017  roar_mm_free(self->path);
1018 if ( self->query_string != NULL )
1019  roar_mm_free(self->query_string);
1020 if ( self->headersstore != NULL )
1021  roar_mm_free(self->headersstore);
1022 if ( self->headers != NULL )
1023  roar_mm_free(self->headers);
1024
1025 if ( self->pluginpara != NULL )
1026  roar_dl_para_unref(self->pluginpara);
1027
1028 roar_mm_free(self);
1029 *userdata = NULL;
1030
1031 return 0;
1032}
1033
1034#define _INCREMENT 256
1035static int _handle(int client, struct roar_vio_calls * vio, struct roar_buffer ** obuffer, void ** userdata, const struct roar_keyval * protopara, ssize_t protoparalen, struct roar_dl_librarypara * pluginpara) {
1036 struct http_client * self = *userdata;
1037 void * data;
1038 size_t oldlen;
1039 ssize_t ret;
1040
1041 (void)client, (void)protopara, (void)protoparalen, (void)pluginpara;
1042
1043 if ( self->status != STATUS_WAITING_FOR_HEADERS )
1044  return __ret(self, obuffer);
1045
1046 if ( self->input_buffer == NULL ) {
1047  if ( roar_buffer_new_data(&(self->input_buffer), _INCREMENT, &data) == -1 )
1048   return -1;
1049  oldlen = 0;
1050 } else {
1051  if ( roar_buffer_get_len(self->input_buffer, &oldlen) == -1 )
1052   return -1;
1053  if ( roar_buffer_set_len(self->input_buffer, oldlen + _INCREMENT) == -1 )
1054   return -1;
1055  if ( roar_buffer_get_data(self->input_buffer, &data) == -1 )
1056   return -1;
1057 }
1058
1059 data += oldlen;
1060
1061 ret = roar_vio_read(vio, data, _INCREMENT);
1062 if ( ret == (ssize_t)-1 ) {
1063  // we can safely ignore return value here as we return error anyway.
1064  _LIBROAR_IGNORE_RET(roar_buffer_set_len(self->input_buffer, oldlen));
1065  return -1;
1066 }
1067
1068 if ( roar_buffer_set_len(self->input_buffer, oldlen + ret) == -1 ) {
1069  ROAR_WARN("_handle(*): Can not reset buffer length. BAD. Error was: %s", roar_errorstring);
1070  return -1;
1071 }
1072
1073 header_parse(self);
1074
1075 if ( self->status == STATUS_BAD_HEADER ) {
1076  header_send(obuffer, 400, NULL, NULL, -1);
1077  self->status = STATUS_DONE;
1078 } else if ( self->status == STATUS_END_OF_HEADER ) {
1079  handle_client(self, obuffer);
1080 }
1081
1082 return __ret(self, obuffer);
1083}
1084
1085// this is a dummy function only used to kill the client after all data has been flushed.
1086static int _flushed(int client, struct roar_vio_calls * vio, struct roar_buffer ** obuffer, void ** userdata, const struct roar_keyval * protopara, ssize_t protoparalen, struct roar_dl_librarypara * pluginpara) {
1087 struct http_client * self = *userdata;
1088
1089 (void)client, (void)vio, (void)protopara, (void)protoparalen, (void)pluginpara;
1090
1091 ROAR_DBG("_flushed(*) = ?");
1092
1093 if ( self->status == STATUS_RUNNING_DATA )
1094  resource_body_send(self, obuffer);
1095
1096 return __ret(self, obuffer);
1097}
1098
1099static const struct roar_dl_proto proto = {
1100 .proto = ROAR_PROTO_HTTP,
1101 .description = "Hyper Text Transfer Protocol",
1102 .flags = ROAR_DL_PROTO_FLAGS_NONE,
1103 .set_proto = _set_proto,
1104 .unset_proto = _unset_proto,
1105 .handle = _handle,
1106 .flush = NULL,
1107 .flushed = _flushed,
1108 .status = NULL
1109};
1110
1111static int __reg_proto(struct roar_dl_librarypara * para, struct roar_dl_libraryinst * lib) {
1112 (void)para, (void)lib;
1113 ROAR_DL_PLUGIN_REG_FN(ROAR_DL_PROTO_SUBTYPE, proto, ROAR_DL_PROTO_VERSION);
1114 return 0;
1115}
1116
1117ROAR_DL_PLUGIN_START(protocol_http) {
1118 ROAR_DL_PLUGIN_META_PRODUCT_NIV("protocol-http", ROAR_VID_ROARAUDIO, ROAR_VNAME_ROARAUDIO);
1119 ROAR_DL_PLUGIN_META_VERSION(ROAR_VERSION_STRING);
1120 ROAR_DL_PLUGIN_META_LICENSE_TAG(GPLv3_0);
1121 ROAR_DL_PLUGIN_META_CONTACT_FLNE("Philipp", "Schafft", "ph3-der-loewe", "lion@lion.leolix.org");
1122 ROAR_DL_PLUGIN_META_DESC("Implementation of the HTTP Protocol");
1123
1124 ROAR_DL_PLUGIN_REG(ROAR_DL_FN_PROTO, __reg_proto);
1125} ROAR_DL_PLUGIN_END
1126
1127//ll
Note: See TracBrowser for help on using the repository browser.