diff options
Diffstat (limited to 'src/fluent-bit/plugins/filter_lua/lua_config.c')
| -rw-r--r-- | src/fluent-bit/plugins/filter_lua/lua_config.c | 206 |
1 files changed, 206 insertions, 0 deletions
diff --git a/src/fluent-bit/plugins/filter_lua/lua_config.c b/src/fluent-bit/plugins/filter_lua/lua_config.c new file mode 100644 index 000000000..f0c154196 --- /dev/null +++ b/src/fluent-bit/plugins/filter_lua/lua_config.c @@ -0,0 +1,206 @@ +/* -*- Mode: C; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */ + +/* Fluent Bit + * ========== + * Copyright (C) 2015-2022 The Fluent Bit Authors + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +#include <fluent-bit/flb_info.h> +#include <fluent-bit/flb_compat.h> +#include <fluent-bit/flb_filter_plugin.h> +#include <fluent-bit/flb_mem.h> +#include <fluent-bit/flb_log.h> +#include <fluent-bit/flb_sds.h> +#include <fluent-bit/flb_str.h> +#include <fluent-bit/flb_utils.h> + +#include "lua_config.h" + +#include <sys/types.h> +#include <sys/stat.h> +#include <fcntl.h> + +struct lua_filter *lua_config_create(struct flb_filter_instance *ins, + struct flb_config *config) +{ + int ret; + char *tmp_key; + char buf[PATH_MAX]; + const char *script = NULL; + const char *tmp = NULL; + (void) config; + struct stat st; + struct lua_filter *lf; + struct mk_list *split = NULL; + struct mk_list *head = NULL; + struct mk_list *tmp_list= NULL; + struct flb_lua_l2c_type *l2c = NULL; + struct flb_split_entry *sentry = NULL; + + /* Allocate context */ + lf = flb_calloc(1, sizeof(struct lua_filter)); + if (!lf) { + flb_errno(); + return NULL; + } + ret = flb_filter_config_map_set(ins, (void*)lf); + if (ret < 0) { + flb_errno(); + flb_plg_error(ins, "configuration error"); + flb_free(lf); + return NULL; + } + + mk_list_init(&lf->l2cc.l2c_types); + lf->ins = ins; + lf->script = NULL; + + /* config: code */ + tmp = flb_filter_get_property("code", ins); + if (tmp) { + lf->code = flb_sds_create(tmp); + } + else { + /* Config: script */ + script = flb_filter_get_property("script", ins); + if (!script) { + flb_plg_error(lf->ins, "no script path defined"); + flb_free(lf); + return NULL; + } + + /* Compose path */ + ret = stat(script, &st); + if (ret == -1 && errno == ENOENT) { + if (script[0] == '/') { + flb_plg_error(lf->ins, "cannot access script '%s'", script); + flb_free(lf); + return NULL; + } + + if (config->conf_path) { + snprintf(buf, sizeof(buf) - 1, "%s%s", + config->conf_path, script); + script = buf; + } + } + + /* Validate script path */ + ret = access(script, R_OK); + if (ret == -1) { + flb_plg_error(lf->ins, "cannot access script '%s'", script); + flb_free(lf); + return NULL; + } + + lf->script = flb_sds_create(script); + if (!lf->script) { + flb_plg_error(lf->ins, "could not allocate string"); + flb_free(lf); + return NULL; + } + } + + if (!lf->call) { + flb_plg_error(lf->ins, "function name defined by 'call' is not set"); + lua_config_destroy(lf); + return NULL; + } + + lf->buffer = flb_sds_create_size(LUA_BUFFER_CHUNK); + if (!lf->buffer) { + flb_plg_error(lf->ins, "could not allocate decode buffer"); + lua_config_destroy(lf); + return NULL; + } + + lf->l2cc.l2c_types_num = 0; + tmp = flb_filter_get_property("type_int_key", ins); + if (tmp) { + split = flb_utils_split(tmp, ' ', FLB_LUA_L2C_TYPES_NUM_MAX); + mk_list_foreach_safe(head, tmp_list, split) { + l2c = flb_malloc(sizeof(struct flb_lua_l2c_type)); + + sentry = mk_list_entry(head, struct flb_split_entry, _head); + + tmp_key = flb_strndup(sentry->value, sentry->len); + l2c->key = flb_sds_create(tmp_key); + l2c->type = FLB_LUA_L2C_TYPE_INT; + flb_free(tmp_key); + + mk_list_add(&l2c->_head, &lf->l2cc.l2c_types); + lf->l2cc.l2c_types_num++; + } + flb_utils_split_free(split); + } + + tmp = flb_filter_get_property("type_array_key", ins); + if (tmp) { + split = flb_utils_split(tmp, ' ', FLB_LUA_L2C_TYPES_NUM_MAX); + mk_list_foreach_safe(head, tmp_list, split) { + l2c = flb_malloc(sizeof(struct flb_lua_l2c_type)); + + sentry = mk_list_entry(head, struct flb_split_entry, _head); + + tmp_key = flb_strndup(sentry->value, sentry->len); + l2c->key = flb_sds_create(tmp_key); + l2c->type = FLB_LUA_L2C_TYPE_ARRAY; + flb_free(tmp_key); + + mk_list_add(&l2c->_head, &lf->l2cc.l2c_types); + lf->l2cc.l2c_types_num++; + } + flb_utils_split_free(split); + } + + return lf; +} + +void lua_config_destroy(struct lua_filter *lf) +{ + struct mk_list *tmp_list = NULL; + struct mk_list *head = NULL; + struct flb_lua_l2c_type *l2c = NULL; + + if (!lf) { + return; + } + + if (lf->code) { + flb_sds_destroy(lf->code); + } + + if (lf->script) { + flb_sds_destroy(lf->script); + } + + if (lf->buffer) { + flb_sds_destroy(lf->buffer); + } + + mk_list_foreach_safe(head, tmp_list, &lf->l2cc.l2c_types) { + l2c = mk_list_entry(head, struct flb_lua_l2c_type, _head); + if (l2c) { + if (l2c->key) { + flb_sds_destroy(l2c->key); + } + mk_list_del(&l2c->_head); + flb_free(l2c); + } + } + + flb_sds_destroy(lf->packbuf); + flb_free(lf); +} |