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