#include #include #include #include #include static liAction* core_list(liServer *srv, liPlugin* p, liValue *val) { liAction *a; guint i; UNUSED(p); if (!val) { ERROR(srv, "%s", "need parameter"); return NULL; } if (val->type == LI_VALUE_ACTION) { a = val->data.val_action.action; li_action_acquire(a); return a; } if (val->type != LI_VALUE_LIST) { ERROR(srv, "expected list, got %s", li_value_type_string(val->type)); return NULL; } a = li_action_new_list(); for (i = 0; i < val->data.list->len; i++) { liValue *oa = g_array_index(val->data.list, liValue*, i); if (oa->type != LI_VALUE_ACTION) { ERROR(srv, "expected action at entry %u of list, got %s", i, li_value_type_string(oa->type)); li_action_release(srv, a); return NULL; } assert(srv == oa->data.val_action.srv); li_action_acquire(oa->data.val_action.action); g_array_append_val(a->data.list, oa->data.val_action.action); } return a; } static liAction* core_when(liServer *srv, liPlugin* p, liValue *val) { liValue *val_cond, *val_act, *val_act_else; liAction *a, *act = NULL, *act_else = NULL; UNUSED(p); if (!val) { ERROR(srv, "%s", "need parameter"); return NULL; } if (val->type != LI_VALUE_LIST) { ERROR(srv, "expected list, got %s", li_value_type_string(val->type)); return NULL; } if (val->data.list->len == 2) { val_act_else = NULL; act_else = NULL; } else if (val->data.list->len == 3) { val_act_else = g_array_index(val->data.list, liValue*, 2); } else { ERROR(srv, "expected list with length 2 or 3, has length %u", val->data.list->len); return NULL; } val_cond = g_array_index(val->data.list, liValue*, 0); val_act = g_array_index(val->data.list, liValue*, 1); if (val_cond->type != LI_VALUE_CONDITION) { ERROR(srv, "expected condition as first parameter, got %s", li_value_type_string(val_cond->type)); return NULL; } if (val_act->type == LI_VALUE_NONE) { act = NULL; } else if (val_act->type == LI_VALUE_ACTION) { act = val_act->data.val_action.action; } else { ERROR(srv, "expected action as second parameter, got %s", li_value_type_string(val_act->type)); return NULL; } if (val_act_else) { if (val_act_else->type == LI_VALUE_NONE) { act_else = NULL; } else if (val_act_else->type == LI_VALUE_ACTION) { act_else = val_act_else->data.val_action.action; } else { ERROR(srv, "expected action as third parameter, got %s", li_value_type_string(val_act_else->type)); return NULL; } } li_condition_acquire(val_cond->data.val_cond.cond); if (act) li_action_acquire(act); if (act_else) li_action_acquire(act_else); a = li_action_new_condition(val_cond->data.val_cond.cond, act, act_else); return a; } static liAction* core_set(liServer *srv, liPlugin* p, liValue *val) { liValue *val_val, *val_name; liAction *a; UNUSED(p); if (!val) { ERROR(srv, "%s", "need parameter"); return NULL; } if (val->type != LI_VALUE_LIST) { ERROR(srv, "expected list, got %s", li_value_type_string(val->type)); return NULL; } if (val->data.list->len != 2) { ERROR(srv, "expected list with length 2, has length %u", val->data.list->len); return NULL; } val_name = g_array_index(val->data.list, liValue*, 0); val_val = g_array_index(val->data.list, liValue*, 1); if (val_name->type != LI_VALUE_STRING) { ERROR(srv, "expected string as first parameter, got %s", li_value_type_string(val_name->type)); return NULL; } a = li_option_action(srv, val_name->data.string->str, val_val); return a; } static gboolean core_setup_set(liServer *srv, liPlugin* p, liValue *val) { liValue *val_val, *val_name; UNUSED(p); if (!val) { ERROR(srv, "%s", "need parameter"); return FALSE; } if (val->type != LI_VALUE_LIST) { ERROR(srv, "expected list, got %s", li_value_type_string(val->type)); return FALSE; } if (val->data.list->len != 2) { ERROR(srv, "expected list with length 2, has length %u", val->data.list->len); return FALSE; } val_name = g_array_index(val->data.list, liValue*, 0); val_val = g_array_index(val->data.list, liValue*, 1); if (val_name->type != LI_VALUE_STRING) { ERROR(srv, "expected string as first parameter, got %s", li_value_type_string(val_name->type)); return FALSE; } return li_plugin_set_default_option(srv, val_name->data.string->str, val_val); } static liHandlerResult core_handle_docroot(liVRequest *vr, gpointer param, gpointer *context) { UNUSED(context); g_string_truncate(vr->physical.doc_root, 0); g_string_append_len(vr->physical.doc_root, GSTR_LEN((GString*) param)); /* reset stat info because path has changed */ vr->physical.have_stat = FALSE; vr->physical.have_errno = FALSE; VR_DEBUG(vr, "docroot: %s", vr->physical.doc_root->str); /* build physical path: docroot + uri.path */ g_string_truncate(vr->physical.path, 0); g_string_append_len(vr->physical.path, GSTR_LEN(vr->physical.doc_root)); g_string_append_len(vr->physical.path, GSTR_LEN(vr->request.uri.path)); VR_DEBUG(vr, "physical path: %s", vr->physical.path->str); return LI_HANDLER_GO_ON; } static void core_docroot_free(liServer *srv, gpointer param) { UNUSED(srv); g_string_free(param, TRUE); } static liAction* core_docroot(liServer *srv, liPlugin* p, liValue *val) { UNUSED(p); if (!val || val->type != LI_VALUE_STRING) { ERROR(srv, "%s", "docroot action expects a string parameter"); return NULL; } return li_action_new_function(core_handle_docroot, NULL, core_docroot_free, li_value_extract(val).string); } static liHandlerResult core_handle_index(liVRequest *vr, gpointer param, gpointer *context) { liHandlerResult res; guint i; struct stat st; gint err; GString *file, *tmp_docroot, *tmp_path; GArray *files = param; UNUSED(context); if (!vr->physical.doc_root->len) { VR_ERROR(vr, "%s", "no docroot specified yet but index action called"); return LI_HANDLER_ERROR; } res = li_stat_cache_get(vr, vr->physical.path, &st, &err, NULL); if (res == LI_HANDLER_WAIT_FOR_EVENT) return LI_HANDLER_WAIT_FOR_EVENT; if (res == LI_HANDLER_ERROR) { /* we ignore errors here so a later action can deal with it (e.g. 'static') */ return LI_HANDLER_GO_ON; } if (!S_ISDIR(st.st_mode)) return LI_HANDLER_GO_ON; /* we use two temporary strings here, one to append to docroot and one to append to physical path */ tmp_docroot = vr->wrk->tmp_str; g_string_truncate(tmp_docroot, 0); g_string_append_len(vr->wrk->tmp_str, GSTR_LEN(vr->physical.doc_root)); tmp_path = g_string_new_len(GSTR_LEN(vr->physical.path)); /* loop through the list to find a possible index file */ for (i = 0; i < files->len; i++) { file = g_array_index(files, liValue*, i)->data.string; if (file->str[0] == '/') { /* entries beginning with a slash shall be looked up directly at the docroot */ g_string_truncate(tmp_docroot, vr->physical.doc_root->len); g_string_append_len(tmp_docroot, GSTR_LEN(file)); res = li_stat_cache_get(vr, tmp_docroot, &st, &err, NULL); } else { g_string_truncate(tmp_path, vr->physical.path->len); g_string_append_len(tmp_path, GSTR_LEN(file)); res = li_stat_cache_get(vr, tmp_path, &st, &err, NULL); } if (res == LI_HANDLER_WAIT_FOR_EVENT) { g_string_free(tmp_path, TRUE); return LI_HANDLER_WAIT_FOR_EVENT; } if (res == LI_HANDLER_GO_ON) { /* file exists. change physical path */ if (file->str[0] == '/') g_string_truncate(vr->physical.path, vr->physical.doc_root->len); g_string_append_len(vr->physical.path, GSTR_LEN(file)); if (CORE_OPTION(LI_CORE_OPTION_DEBUG_REQUEST_HANDLING).boolean) { VR_DEBUG(vr, "using index file: '%s'", file->str); } g_string_free(tmp_path, TRUE); return LI_HANDLER_GO_ON; } } g_string_free(tmp_path, TRUE); return LI_HANDLER_GO_ON; } static void core_index_free(liServer *srv, gpointer param) { guint i; GArray *files = param; UNUSED(srv); for (i = 0; i < files->len; i++) li_value_free(g_array_index(files, liValue*, i)); g_array_free(files, TRUE); } static liAction* core_index(liServer *srv, liPlugin* p, liValue *val) { GArray *files; guint i; UNUSED(p); if (!val || val->type != LI_VALUE_LIST) { ERROR(srv, "%s", "index action expects a list of strings as parameter"); return NULL; } files = val->data.list; for (i = 0; i < files->len; i++) { if (g_array_index(files, liValue*, i)->type != LI_VALUE_STRING) { ERROR(srv, "%s", "index action expects a list of strings as parameter"); return NULL; } } return li_action_new_function(core_handle_index, NULL, core_index_free, li_value_extract(val).list); } static liHandlerResult core_handle_static(liVRequest *vr, gpointer param, gpointer *context) { int fd; struct stat st; int err; liHandlerResult res; UNUSED(param); UNUSED(context); switch (vr->request.http_method) { case LI_HTTP_METHOD_GET: case LI_HTTP_METHOD_HEAD: break; default: return LI_HANDLER_GO_ON; } if (li_vrequest_is_handled(vr)) return LI_HANDLER_GO_ON; if (vr->physical.path->len == 0) return LI_HANDLER_GO_ON; res = li_stat_cache_get(vr, vr->physical.path, &st, &err, &fd); if (res == LI_HANDLER_WAIT_FOR_EVENT) return res; if (CORE_OPTION(LI_CORE_OPTION_DEBUG_REQUEST_HANDLING).boolean) { VR_DEBUG(vr, "try serving static file: '%s'", vr->physical.path->str); } if (res == LI_HANDLER_ERROR) { /* open or fstat failed */ if (fd != -1) close(fd); if (!li_vrequest_handle_direct(vr)) { return LI_HANDLER_ERROR; } switch (err) { case ENOENT: case ENOTDIR: vr->response.http_status = 404; return LI_HANDLER_GO_ON; case EACCES: vr->response.http_status = 403; return LI_HANDLER_GO_ON; default: VR_ERROR(vr, "stat() or open() for '%s' failed: %s", vr->physical.path->str, g_strerror(err)); return LI_HANDLER_ERROR; } } else if (S_ISDIR(st.st_mode)) { if (fd != -1) close(fd); return LI_HANDLER_GO_ON; } else if (!S_ISREG(st.st_mode)) { if (CORE_OPTION(LI_CORE_OPTION_DEBUG_REQUEST_HANDLING).boolean) { VR_DEBUG(vr, "not a regular file: '%s'", vr->physical.path->str); } if (fd != -1) close(fd); if (!li_vrequest_handle_direct(vr)) { return LI_HANDLER_ERROR; } vr->response.http_status = 403; } else { GString *mime_str; gboolean cachable; #ifdef FD_CLOEXEC fcntl(fd, F_SETFD, FD_CLOEXEC); #endif if (!li_vrequest_handle_direct(vr)) { close(fd); return LI_HANDLER_ERROR; } li_etag_set_header(vr, &st, &cachable); if (cachable) { vr->response.http_status = 304; close(fd); return LI_HANDLER_GO_ON; } mime_str = li_mimetype_get(vr, vr->physical.path); vr->response.http_status = 200; if (mime_str) li_http_header_overwrite(vr->response.headers, CONST_STR_LEN("Content-Type"), GSTR_LEN(mime_str)); else li_http_header_overwrite(vr->response.headers, CONST_STR_LEN("Content-Type"), CONST_STR_LEN("application/octet-stream")); li_chunkqueue_append_file_fd(vr->out, NULL, 0, st.st_size, fd); } return LI_HANDLER_GO_ON; } static liAction* core_static(liServer *srv, liPlugin* p, liValue *val) { UNUSED(p); if (val) { ERROR(srv, "%s", "static action doesn't have parameters"); return NULL; } return li_action_new_function(core_handle_static, NULL, NULL, NULL); } static liHandlerResult core_handle_status(liVRequest *vr, gpointer param, gpointer *context) { UNUSED(param); UNUSED(context); vr->response.http_status = GPOINTER_TO_INT(param); return LI_HANDLER_GO_ON; } static liAction* core_status(liServer *srv, liPlugin* p, liValue *val) { gpointer ptr; UNUSED(p); if (!val || val->type != LI_VALUE_NUMBER) { ERROR(srv, "%s", "status action expects a number as parameter"); return NULL; } ptr = GINT_TO_POINTER((gint) li_value_extract(val).number); return li_action_new_function(core_handle_status, NULL, NULL, ptr); } static void core_log_write_free(liServer *srv, gpointer param) { UNUSED(srv); g_string_free(param, TRUE); } static liHandlerResult core_handle_log_write(liVRequest *vr, gpointer param, gpointer *context) { GString *msg = param; UNUSED(context); VR_INFO(vr, "%s", msg->str); return LI_HANDLER_GO_ON; } static liAction* core_log_write(liServer *srv, liPlugin* p, liValue *val) { UNUSED(p); if (!val || val->type != LI_VALUE_STRING) { ERROR(srv, "%s", "log.write expects a string parameter"); return NULL; } return li_action_new_function(core_handle_log_write, NULL, core_log_write_free, li_value_extract(val).string); } static liHandlerResult core_handle_blank(liVRequest *vr, gpointer param, gpointer *context) { UNUSED(param); UNUSED(context); if (!li_vrequest_handle_direct(vr)) return LI_HANDLER_GO_ON; vr->response.http_status = 200; return LI_HANDLER_GO_ON; } static liAction* core_blank(liServer *srv, liPlugin* p, liValue *val) { UNUSED(p); if (val) { ERROR(srv, "%s", "'blank' action doesn't have parameters"); return NULL; } return li_action_new_function(core_handle_blank, NULL, NULL, NULL); } static liHandlerResult core_handle_profile_mem(liVRequest *vr, gpointer param, gpointer *context) { UNUSED(vr); UNUSED(param); UNUSED(context); /*g_mem_profile();*/ profiler_dump(); return LI_HANDLER_GO_ON; } static liAction* core_profile_mem(liServer *srv, liPlugin* p, liValue *val) { UNUSED(p); if (val) { ERROR(srv, "%s", "'profile_mem' action doesn't have parameters"); return NULL; } return li_action_new_function(core_handle_profile_mem, NULL, NULL, NULL); } static gboolean core_listen(liServer *srv, liPlugin* p, liValue *val) { GString *ipstr; UNUSED(p); if (val->type != LI_VALUE_STRING) { ERROR(srv, "%s", "listen expects a string as parameter"); return FALSE; } ipstr = val->data.string; li_angel_listen(srv, ipstr); return TRUE; } static gboolean core_event_handler(liServer *srv, liPlugin* p, liValue *val) { guint backend; gchar *str; UNUSED(p); if (val->type != LI_VALUE_STRING) { ERROR(srv, "%s", "event_handler expects a string as parameter"); return FALSE; } str = val->data.string->str; if (g_str_equal(str, "select")) backend = EVBACKEND_SELECT; else if (g_str_equal(str, "poll")) backend = EVBACKEND_POLL; else if (g_str_equal(str, "epoll")) backend = EVBACKEND_EPOLL; else if (g_str_equal(str, "kqueue")) backend = EVBACKEND_KQUEUE; else if (g_str_equal(str, "devpoll")) backend = EVBACKEND_DEVPOLL; else if (g_str_equal(str, "port")) backend = EVBACKEND_PORT; else { ERROR(srv, "unkown event handler: '%s'", str); return FALSE; } if (backend) { if (!(ev_supported_backends() & backend)) { ERROR(srv, "unsupported event handler: '%s'", str); return FALSE; } if (!(ev_recommended_backends() & backend)) { DEBUG(srv, "warning: event handler '%s' not recommended for this platform!", str); } } srv->loop_flags |= backend; return TRUE; } static gboolean core_workers(liServer *srv, liPlugin* p, liValue *val) { gint workers; UNUSED(p); workers = val->data.number; if (val->type != LI_VALUE_NUMBER || workers < 1) { ERROR(srv, "%s", "workers expects a positive integer as parameter"); return FALSE; } if (srv->worker_count != 0) { ERROR(srv, "workers already called with '%i', overwriting", srv->worker_count); } srv->worker_count = workers; return TRUE; } static gboolean core_module_load(liServer *srv, liPlugin* p, liValue *val) { liValue *mods = li_value_new_list(); UNUSED(p); if (!g_module_supported()) { ERROR(srv, "%s", "module loading not supported on this platform"); li_value_free(mods); return FALSE; } if (val->type == LI_VALUE_STRING) { /* load only one module */ liValue *name = li_value_new_string(li_value_extract(val).string); g_array_append_val(mods->data.list, name); } else if (val->type == LI_VALUE_LIST) { /* load a list of modules */ for (guint i = 0; i < val->data.list->len; i++) { liValue *v = g_array_index(val->data.list, liValue*, i); if (v->type != LI_VALUE_STRING) { ERROR(srv, "module_load takes either a string or a list of strings as parameter, list with %s entry given", li_value_type_string(v->type)); li_value_free(mods); return FALSE; } } g_array_free(mods->data.list, TRUE); mods->data.list = li_value_extract(val).list; } else { ERROR(srv, "module_load takes either a string or a list of strings as parameter, %s given", li_value_type_string(val->type)); return FALSE; } /* parameter types ok, load modules */ for (guint i = 0; i < mods->data.list->len; i++) { GString *name = g_array_index(mods->data.list, liValue*, i)->data.string; if (!li_module_load(srv->modules, name->str)) { ERROR(srv, "could not load module '%s': %s", name->str, g_module_error()); li_value_free(mods); return FALSE; } DEBUG(srv, "loaded module '%s'", name->str); } li_value_free(mods); return TRUE; } static gboolean core_io_timeout(liServer *srv, liPlugin* p, liValue *val) { UNUSED(p); if (!val || val->type != LI_VALUE_NUMBER || val->data.number < 1) { ERROR(srv, "%s", "io_timeout expects a positive number as parameter"); return FALSE; } srv->io_timeout = li_value_extract(val).number; return TRUE; } static gboolean core_stat_cache_ttl(liServer *srv, liPlugin* p, liValue *val) { UNUSED(p); if (!val || val->type != LI_VALUE_NUMBER || val->data.number < 1) { ERROR(srv, "%s", "stat_cache.ttl expects a positive number as parameter"); return FALSE; } srv->stat_cache_ttl = (gdouble)li_value_extract(val).number; return TRUE; } /* * OPTIONS */ static gboolean core_option_log_parse(liServer *srv, liPlugin *p, size_t ndx, liValue *val, liOptionValue *oval) { GHashTableIter iter; gpointer k, v; liLogLevel level; GString *path; GString *level_str; GArray *arr = g_array_sized_new(FALSE, TRUE, sizeof(liLog*), 6); UNUSED(p); UNUSED(ndx); oval->list = arr; g_array_set_size(arr, 6); /* default value */ if (!val) { /* default: log LI_LOG_LEVEL_WARNING, LI_LOG_LEVEL_ERROR and LI_LOG_LEVEL_BACKEND to stderr */ liLog *log = srv->logs.stderr; log_ref(srv, log); g_array_index(arr, liLog*, LI_LOG_LEVEL_WARNING) = log; log_ref(srv, log); g_array_index(arr, liLog*, LI_LOG_LEVEL_ERROR) = log; log_ref(srv, log); g_array_index(arr, liLog*, LI_LOG_LEVEL_BACKEND) = log; return TRUE; } g_hash_table_iter_init(&iter, val->data.hash); while (g_hash_table_iter_next(&iter, &k, &v)) { if (((liValue*)v)->type != LI_VALUE_STRING) { ERROR(srv, "log expects a hashtable with string values, %s given", li_value_type_string(((liValue*)v)->type)); g_array_free(arr, TRUE); return FALSE; } path = ((liValue*)v)->data.string; level_str = (GString*)k; if (g_str_equal(level_str->str, "*")) { for (guint i = 0; i < arr->len; i++) { liLog *log; if (NULL != g_array_index(arr, liLog*, i)) continue; log = log_new(srv, log_type_from_path(path), path); g_array_index(arr, liLog*, i) = log; } } else { liLog *log = log_new(srv, log_type_from_path(path), path); level = log_level_from_string(level_str); g_array_index(arr, liLog*, level) = log; } } return TRUE; } static void core_option_log_free(liServer *srv, liPlugin *p, size_t ndx, liOptionValue oval) { GArray *arr = oval.list; UNUSED(p); UNUSED(ndx); if (!arr) return; for (guint i = 0; i < arr->len; i++) { if (NULL != g_array_index(arr, liLog*, i)) log_unref(srv, g_array_index(arr, liLog*, i)); } g_array_free(arr, TRUE); } static gboolean core_option_log_timestamp_parse(liServer *srv, liPlugin *p, size_t ndx, liValue *val, liOptionValue *oval) { UNUSED(p); UNUSED(ndx); if (!val) return TRUE; oval->ptr = li_log_timestamp_new(srv, li_value_extract(val).string); return TRUE; } static void core_option_log_timestamp_free(liServer *srv, liPlugin *p, size_t ndx, liOptionValue oval) { UNUSED(p); UNUSED(ndx); if (!oval.ptr) return; li_log_timestamp_free(srv, oval.ptr); } static gboolean core_option_mime_types_parse(liServer *srv, liPlugin *p, size_t ndx, liValue *val, liOptionValue *oval) { GArray *arr; UNUSED(srv); UNUSED(p); UNUSED(ndx); /* default value */ if (!val) { oval->list = g_array_new(FALSE, TRUE, sizeof(liValue)); return TRUE; } /* check if the passed val is of type (("a", "b"), ("x", y")) */ arr = val->data.list; for (guint i = 0; i < arr->len; i++) { liValue *v = g_array_index(arr, liValue*, i); liValue *v1, *v2; if (v->type != LI_VALUE_LIST) { ERROR(srv, "mime_types option expects a list of string tuples, entry #%u is of type %s", i, li_value_type_string(v->type)); return FALSE; } if (v->data.list->len != 2) { ERROR(srv, "mime_types option expects a list of string tuples, entry #%u is not a tuple", i); return FALSE; } v1 = g_array_index(v->data.list, liValue*, 0); v2 = g_array_index(v->data.list, liValue*, 1); if (v1->type != LI_VALUE_STRING || v2->type != LI_VALUE_STRING) { ERROR(srv, "mime_types option expects a list of string tuples, entry #%u is a (%s,%s) tuple", i, li_value_type_string(v1->type), li_value_type_string(v2->type)); return FALSE; } } /* everything ok */ oval->list = li_value_extract(val).list; return TRUE; } static void core_option_mime_types_free(liServer *srv, liPlugin *p, size_t ndx, liOptionValue oval) { UNUSED(srv); UNUSED(p); UNUSED(ndx); for (guint i = 0; i < oval.list->len; i++) li_value_free(g_array_index(oval.list, liValue*, i)); g_array_free(oval.list, TRUE); } static gboolean core_option_etag_use_parse(liServer *srv, liPlugin *p, size_t ndx, liValue *val, liOptionValue *oval) { GArray *arr; guint flags = 0; UNUSED(p); UNUSED(ndx); /* default value */ if (!val) { oval->number = LI_ETAG_USE_INODE | LI_ETAG_USE_MTIME | LI_ETAG_USE_SIZE; return TRUE; } if (val->type != LI_VALUE_LIST) { ERROR(srv, "etag.use option expects a list of strings, parameter is of type %s", li_value_type_string(val->type)); } arr = val->data.list; for (guint i = 0; i < arr->len; i++) { liValue *v = g_array_index(arr, liValue*, i); if (v->type != LI_VALUE_STRING) { ERROR(srv, "etag.use option expects a list of strings, entry #%u is of type %s", i, li_value_type_string(v->type)); return FALSE; } if (0 == strcmp(v->data.string->str, "inode")) { flags |= LI_ETAG_USE_INODE; } else if (0 == strcmp(v->data.string->str, "mtime")) { flags |= LI_ETAG_USE_MTIME; } else if (0 == strcmp(v->data.string->str, "size")) { flags |= LI_ETAG_USE_SIZE; } else { ERROR(srv, "unknown etag.use flag: %s", v->data.string->str); return FALSE; } } oval->number = (guint64) flags; return TRUE; } static liHandlerResult core_handle_header_add(liVRequest *vr, gpointer param, gpointer *context) { GArray *l = (GArray*)param; GString *k = g_array_index(l, liValue*, 0)->data.string; GString *v = g_array_index(l, liValue*, 1)->data.string; UNUSED(param); UNUSED(context); li_http_header_insert(vr->response.headers, GSTR_LEN(k), GSTR_LEN(v)); return LI_HANDLER_GO_ON; } static void core_header_free(liServer *srv, gpointer param) { UNUSED(srv); li_value_list_free(param); } static liAction* core_header_add(liServer *srv, liPlugin* p, liValue *val) { GArray *l; UNUSED(p); if (val->type != LI_VALUE_LIST) { ERROR(srv, "'core_header_add' action expects a string tuple as parameter, %s given", li_value_type_string(val->type)); return NULL; } l = val->data.list; if (l->len != 2) { ERROR(srv, "'core_header_add' action expects a string tuple as parameter, list has %u entries", l->len); return NULL; } if (g_array_index(l, liValue*, 0)->type != LI_VALUE_STRING || g_array_index(l, liValue*, 0)->type != LI_VALUE_STRING) { ERROR(srv, "%s", "'core_header_add' action expects a string tuple as parameter"); return NULL; } return li_action_new_function(core_handle_header_add, NULL, core_header_free, li_value_extract(val).list); } static liHandlerResult core_handle_header_append(liVRequest *vr, gpointer param, gpointer *context) { GArray *l = (GArray*)param; GString *k = g_array_index(l, liValue*, 0)->data.string; GString *v = g_array_index(l, liValue*, 1)->data.string; UNUSED(param); UNUSED(context); li_http_header_append(vr->response.headers, GSTR_LEN(k), GSTR_LEN(v)); return LI_HANDLER_GO_ON; } static liAction* core_header_append(liServer *srv, liPlugin* p, liValue *val) { GArray *l; UNUSED(p); if (val->type != LI_VALUE_LIST) { ERROR(srv, "'core_header_append' action expects a string tuple as parameter, %s given", li_value_type_string(val->type)); return NULL; } l = val->data.list; if (l->len != 2) { ERROR(srv, "'core_header_append' action expects a string tuple as parameter, list has %u entries", l->len); return NULL; } if (g_array_index(l, liValue*, 0)->type != LI_VALUE_STRING || g_array_index(l, liValue*, 0)->type != LI_VALUE_STRING) { ERROR(srv, "%s", "'core_header_append' action expects a string tuple as parameter"); return NULL; } return li_action_new_function(core_handle_header_append, NULL, core_header_free, li_value_extract(val).list); } static liHandlerResult core_handle_header_overwrite(liVRequest *vr, gpointer param, gpointer *context) { GArray *l = (GArray*)param; GString *k = g_array_index(l, liValue*, 0)->data.string; GString *v = g_array_index(l, liValue*, 1)->data.string; UNUSED(param); UNUSED(context); li_http_header_overwrite(vr->response.headers, GSTR_LEN(k), GSTR_LEN(v)); return LI_HANDLER_GO_ON; } static liAction* core_header_overwrite(liServer *srv, liPlugin* p, liValue *val) { GArray *l; UNUSED(p); if (val->type != LI_VALUE_LIST) { ERROR(srv, "'core_header_overwrite' action expects a string tuple as parameter, %s given", li_value_type_string(val->type)); return NULL; } l = val->data.list; if (l->len != 2) { ERROR(srv, "'core_header_overwrite' action expects a string tuple as parameter, list has %u entries", l->len); return NULL; } if (g_array_index(l, liValue*, 0)->type != LI_VALUE_STRING || g_array_index(l, liValue*, 0)->type != LI_VALUE_STRING) { ERROR(srv, "%s", "'core_header_overwrite' action expects a string tuple as parameter"); return NULL; } return li_action_new_function(core_handle_header_overwrite, NULL, core_header_free, li_value_extract(val).list); } typedef struct { liAction *target_true, *target_false; } core_conditional; static core_conditional* core_conditional_create(liServer *srv, liValue *val, guint idx) { core_conditional *cc; liValue *val_true = NULL, *val_false = NULL; liAction *act_true = NULL, *act_false = NULL; GArray *l; if (idx == 0 && val->type == LI_VALUE_ACTION) { cc = g_slice_new0(core_conditional); act_true = val->data.val_action.action; li_action_acquire(act_true); cc->target_true = act_true; return cc; } if (val->type != LI_VALUE_LIST) { ERROR(srv, "unexpected parameter of type %s, expected list", li_value_type_string(val->type)); return NULL; } l = val->data.list; if (idx >= l->len) { ERROR(srv, "expected at least %u parameters, %u given", idx+1, l->len); } val_true = g_array_index(l, liValue*, idx); if (idx + 1 < l->len) val_false = g_array_index(l, liValue*, idx+1); if (val_true->type == LI_VALUE_NONE) { act_true = NULL; } else if (val_true->type == LI_VALUE_ACTION) { act_true = val_true->data.val_action.action; } else { ERROR(srv, "expected action at entry %u of list, got %s", idx, li_value_type_string(val_true->type)); } if (val_false) { if (val_false->type == LI_VALUE_NONE) { act_false = NULL; } else if (val_false->type == LI_VALUE_ACTION) { act_false = val_false->data.val_action.action; } else { ERROR(srv, "expected action at entry %u of list, got %s", idx+1, li_value_type_string(val_false->type)); } } else { act_false = NULL; } cc = g_slice_new0(core_conditional); if (act_true) li_action_acquire(act_true); if (act_false) li_action_acquire(act_false); cc->target_true = act_true; cc->target_false = act_false; return cc; } static void core_conditional_free(liServer *srv, gpointer param) { core_conditional *cc = (core_conditional*) param; if (!cc) return; li_action_release(srv, cc->target_true); li_action_release(srv, cc->target_false); g_slice_free(core_conditional, cc); } static liHandlerResult core_conditional_do(liVRequest *vr, gpointer param, gboolean way) { core_conditional *cc = (core_conditional*)param; if (way) { if (cc->target_true) li_action_enter(vr, cc->target_true); } else { if (cc->target_false) li_action_enter(vr, cc->target_false); } return LI_HANDLER_GO_ON; } static liHandlerResult core_handle_physical_exists(liVRequest *vr, gpointer param, gpointer *context) { UNUSED(context); if (0 == vr->physical.path->len) return core_conditional_do(vr, param, FALSE); li_vrequest_stat(vr); return core_conditional_do(vr, param, vr->physical.have_stat); } static liAction* core_physical_exists(liServer *srv, liPlugin* p, liValue *val) { core_conditional *cc = core_conditional_create(srv, val, 0); UNUSED(p); if (!cc) return NULL; return li_action_new_function(core_handle_physical_exists, NULL, core_conditional_free, cc); } static liHandlerResult core_handle_physical_is_file(liVRequest *vr, gpointer param, gpointer *context) { UNUSED(context); if (0 == vr->physical.path->len) return core_conditional_do(vr, param, FALSE); li_vrequest_stat(vr); return core_conditional_do(vr, param, vr->physical.have_stat && S_ISREG(vr->physical.stat.st_mode)); } static liAction* core_physical_is_file(liServer *srv, liPlugin* p, liValue *val) { core_conditional *cc = core_conditional_create(srv, val, 0); UNUSED(p); if (!cc) return NULL; return li_action_new_function(core_handle_physical_is_file, NULL, core_conditional_free, cc); } static liHandlerResult core_handle_physical_is_dir(liVRequest *vr, gpointer param, gpointer *context) { UNUSED(context); if (0 == vr->physical.path->len) return core_conditional_do(vr, param, FALSE); li_vrequest_stat(vr); return core_conditional_do(vr, param, vr->physical.have_stat && S_ISDIR(vr->physical.stat.st_mode)); } static liAction* core_physical_is_dir(liServer *srv, liPlugin* p, liValue *val) { core_conditional *cc = core_conditional_create(srv, val, 0); UNUSED(p); if (!cc) return NULL; return li_action_new_function(core_handle_physical_is_dir, NULL, core_conditional_free, cc); } /* chunkqueue memory limits */ static liHandlerResult core_handle_buffer_out(liVRequest *vr, gpointer param, gpointer *context) { gint limit = GPOINTER_TO_INT(param); UNUSED(context); li_cqlimit_set_limit(vr->out->limit, limit); return LI_HANDLER_GO_ON; } static liAction* core_buffer_out(liServer *srv, liPlugin* p, liValue *val) { gint64 limit; UNUSED(p); if (val->type != LI_VALUE_NUMBER) { ERROR(srv, "'buffer.out' action expects an integer as parameter, %s given", li_value_type_string(val->type)); return NULL; } limit = val->data.number; if (limit < 0) { limit = 0; /* no limit */ } else if (limit < (16*1024)) { ERROR(srv, "buffer %"G_GINT64_FORMAT" is too low (need at least 16 kb)", limit); return NULL; } else if (limit > (1 << 30)) { ERROR(srv, "buffer %"G_GINT64_FORMAT" is too high (1GB is the maximum)", limit); return NULL; } return li_action_new_function(core_handle_buffer_out, NULL, NULL, GINT_TO_POINTER((gint) limit)); } static liHandlerResult core_handle_buffer_in(liVRequest *vr, gpointer param, gpointer *context) { gint limit = GPOINTER_TO_INT(param); UNUSED(context); li_cqlimit_set_limit(vr->out->limit, limit); return LI_HANDLER_GO_ON; } static liAction* core_buffer_in(liServer *srv, liPlugin* p, liValue *val) { gint64 limit; UNUSED(p); if (val->type != LI_VALUE_NUMBER) { ERROR(srv, "'buffer.in' action expects an integer as parameter, %s given", li_value_type_string(val->type)); return NULL; } limit = val->data.number; if (limit < 0) { limit = 0; /* no limit */ } if (limit > (1 << 30)) { ERROR(srv, "buffer %"G_GINT64_FORMAT" is too high (1GB is the maximum)", limit); return NULL; } return li_action_new_function(core_handle_buffer_in, NULL, NULL, GINT_TO_POINTER((gint) limit)); } static liHandlerResult core_handle_throttle_pool(liVRequest *vr, gpointer param, gpointer *context) { liThrottlePool *pool = param; gint magazine; UNUSED(context); if (vr->con->throttle.pool.ptr != pool) { if (vr->con->throttle.pool.ptr) { /* connection has been in a different pool, give back bandwidth */ g_atomic_int_add(&vr->con->throttle.pool.ptr->magazine, vr->con->throttle.pool.magazine); vr->con->throttle.pool.magazine = 0; if (vr->con->throttle.pool.queued) { liThrottlePool *p = vr->con->throttle.pool.ptr; g_queue_unlink(p->queues[vr->con->wrk->ndx+p->current_queue[vr->con->wrk->ndx]], &vr->con->throttle.ip.lnk); g_atomic_int_add(&p->num_cons, -1); } } /* try to steal some initial 4kbytes from the pool */ while ((magazine = g_atomic_int_get(&pool->magazine)) > (4*1024)) { if (g_atomic_int_compare_and_exchange(&pool->magazine, magazine, magazine - (4*1024))) { vr->con->throttle.pool.magazine = 4*1024; break; } } } vr->con->throttle.pool.ptr = pool; vr->con->throttled = TRUE; return LI_HANDLER_GO_ON; } static liAction* core_throttle_pool(liServer *srv, liPlugin* p, liValue *val) { GString *name; guint i; liThrottlePool *pool = NULL; gint64 rate; UNUSED(p); if (val->type != LI_VALUE_STRING && val->type != LI_VALUE_LIST) { ERROR(srv, "'throttle_pool' action expects a string or a string-number tuple as parameter, %s given", li_value_type_string(val->type)); return NULL; } if (val->type == LI_VALUE_LIST) { if (val->data.list->len != 2 || g_array_index(val->data.list, liValue*, 0)->type != LI_VALUE_STRING || g_array_index(val->data.list, liValue*, 1)->type != LI_VALUE_NUMBER) { ERROR(srv, "%s", "'throttle_pool' action expects a string or a string-number tuple as parameter"); return NULL; } name = g_array_index(val->data.list, liValue*, 0)->data.string; rate = g_array_index(val->data.list, liValue*, 1)->data.number; if (rate && rate < (32*1024)) { ERROR(srv, "throttle_pool: rate %"G_GINT64_FORMAT" is too low (32kbyte/s minimum or 0 for unlimited)", rate); return NULL; } if (rate > (0xFFFFFFFF)) { ERROR(srv, "throttle_pool: rate %"G_GINT64_FORMAT" is too high (4gbyte/s maximum)", rate); return NULL; } } else { name = val->data.string; rate = 0; } for (i = 0; i < srv->throttle_pools->len; i++) { if (g_string_equal(g_array_index(srv->throttle_pools, liThrottlePool*, i)->name, name)) { /* pool already defined */ if (val->type == LI_VALUE_LIST && g_array_index(srv->throttle_pools, liThrottlePool*, i)->rate != (guint)rate) { ERROR(srv, "throttle_pool: pool '%s' already defined but with different rate (%ukbyte/s)", name->str, g_array_index(srv->throttle_pools, liThrottlePool*, i)->rate); return NULL; } pool = g_array_index(srv->throttle_pools, liThrottlePool*, i); break; } } if (!pool) { /* pool not yet defined */ if (val->type == LI_VALUE_STRING) { ERROR(srv, "throttle_pool: rate for pool '%s' hasn't been defined", name->str); return NULL; } pool = throttle_pool_new(srv, li_value_extract(g_array_index(val->data.list, liValue*, 0)).string, (guint)rate); g_array_append_val(srv->throttle_pools, pool); } return li_action_new_function(core_handle_throttle_pool, NULL, NULL, pool); } static liHandlerResult core_handle_throttle_connection(liVRequest *vr, gpointer param, gpointer *context) { gint supply; liConnection *con = vr->con; guint rate = GPOINTER_TO_UINT(param); UNUSED(context); con->throttle.con.rate = rate; con->throttled = TRUE; if (con->throttle.pool.magazine) { supply = MAX(con->throttle.pool.magazine, rate * THROTTLE_GRANULARITY); con->throttle.con.magazine += supply; con->throttle.pool.magazine -= supply; } return LI_HANDLER_GO_ON; } static liAction* core_throttle_connection(liServer *srv, liPlugin* p, liValue *val) { gint64 rate; UNUSED(p); if (val->type != LI_VALUE_NUMBER) { ERROR(srv, "'throttle_connection' action expects a positiv integer as parameter, %s given", li_value_type_string(val->type)); return NULL; } rate = val->data.number; if (rate < 0) { rate = 0; /* no limit */ } if (rate && rate < (32*1024)) { ERROR(srv, "throttle_connection: rate %"G_GUINT64_FORMAT" is too low (32kbyte/s minimum or 0 for unlimited)", rate); return NULL; } if (rate > (0xFFFFFFFF)) { ERROR(srv, "throttle_connection: rate %"G_GINT64_FORMAT" is too high (4gbyte/s maximum)", rate); return NULL; } return li_action_new_function(core_handle_throttle_connection, NULL, NULL, GUINT_TO_POINTER((guint) rate)); } static void core_warmup(liServer *srv, liPlugin *p, gint32 id, GString *data) { UNUSED(p); UNUSED(id); UNUSED(data); li_server_goto_state(srv, LI_SERVER_WARMUP); } static void core_run(liServer *srv, liPlugin *p, gint32 id, GString *data) { UNUSED(p); UNUSED(id); UNUSED(data); li_server_goto_state(srv, LI_SERVER_RUNNING); } static void core_suspend(liServer *srv, liPlugin *p, gint32 id, GString *data) { UNUSED(p); UNUSED(id); UNUSED(data); li_server_goto_state(srv, LI_SERVER_SUSPENDED); } static const liPluginOption options[] = { { "debug.log_request_handling", LI_VALUE_BOOLEAN, GINT_TO_POINTER(FALSE), NULL, NULL }, { "log.timestamp", LI_VALUE_STRING, NULL, core_option_log_timestamp_parse, core_option_log_timestamp_free }, { "log", LI_VALUE_HASH, NULL, core_option_log_parse, core_option_log_free }, { "static-file.exclude", LI_VALUE_LIST, NULL, NULL, NULL }, { "server.name", LI_VALUE_STRING, NULL, NULL, NULL }, { "server.tag", LI_VALUE_STRING, "lighttpd-2.0~sandbox", NULL, NULL }, { "server.max_keep_alive_idle", LI_VALUE_NUMBER, GINT_TO_POINTER(5), NULL, NULL }, { "server.max_keep_alive_requests", LI_VALUE_NUMBER, GINT_TO_POINTER(15), NULL, NULL }, { "mime_types", LI_VALUE_LIST, NULL, core_option_mime_types_parse, core_option_mime_types_free }, { "throttle", LI_VALUE_NUMBER, GINT_TO_POINTER(0), NULL, NULL }, { "etag.use", LI_VALUE_NONE, NULL, core_option_etag_use_parse, NULL }, { NULL, 0, NULL, NULL, NULL } }; static const liPluginAction actions[] = { { "list", core_list }, { "when", core_when }, { "set", core_set }, { "docroot", core_docroot }, { "index", core_index }, { "static", core_static }, { "status", core_status }, { "log.write", core_log_write }, { "blank", core_blank }, { "profile_mem", core_profile_mem }, { "header_add", core_header_add }, { "header_append", core_header_append }, { "header_overwrite", core_header_overwrite }, { "physical.exists", core_physical_exists }, { "physical.is_file", core_physical_is_file }, { "physical.is_dir", core_physical_is_dir }, { "buffer.out", core_buffer_out }, { "buffer.in", core_buffer_in }, { "throttle_pool", core_throttle_pool }, /*{ "throttle.ip", core_throttle_ip },*/ { "throttle_connection", core_throttle_connection }, { NULL, NULL } }; static const liPluginSetup setups[] = { { "set_default", core_setup_set }, { "listen", core_listen }, { "event_handler", core_event_handler }, { "workers", core_workers }, { "module_load", core_module_load }, { "io_timeout", core_io_timeout }, { "stat_cache.ttl", core_stat_cache_ttl }, { NULL, NULL } }; static const liPluginAngel angelcbs[] = { { "warmup", core_warmup }, { "run", core_run }, { "suspend", core_suspend }, { NULL, NULL } }; void plugin_core_init(liServer *srv, liPlugin *p); void plugin_core_init(liServer *srv, liPlugin *p) { UNUSED(srv); p->options = options; p->actions = actions; p->setups = setups; p->angelcbs = angelcbs; }