diff options
Diffstat (limited to 'fluent-bit/plugins/out_azure/azure_conf.c')
| -rw-r--r-- | fluent-bit/plugins/out_azure/azure_conf.c | 219 |
1 files changed, 219 insertions, 0 deletions
diff --git a/fluent-bit/plugins/out_azure/azure_conf.c b/fluent-bit/plugins/out_azure/azure_conf.c new file mode 100644 index 00000000..9f8f8a05 --- /dev/null +++ b/fluent-bit/plugins/out_azure/azure_conf.c @@ -0,0 +1,219 @@ +/* -*- 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_output_plugin.h> +#include <fluent-bit/flb_base64.h> +#include <fluent-bit/flb_utils.h> +#include <fluent-bit/flb_record_accessor.h> +#include <fluent-bit/flb_pack.h> + +#include "azure.h" +#include "azure_conf.h" + +struct flb_azure *flb_azure_conf_create(struct flb_output_instance *ins, + struct flb_config *config) +{ + int ret; + size_t size; + size_t olen; + const char *tmp; + struct flb_upstream *upstream; + struct flb_azure *ctx; + struct flb_record_accessor *ra_prefix_key = NULL; + + /* Allocate config context */ + ctx = flb_calloc(1, sizeof(struct flb_azure)); + if (!ctx) { + flb_errno(); + return NULL; + } + ctx->ins = ins; + + /* Set context */ + flb_output_set_context(ins, ctx); + + /* Load config map */ + ret = flb_output_config_map_set(ins, (void *) ctx); + if (ret == -1) { + return NULL; + } + + if (!ctx->customer_id) { + flb_plg_error(ctx->ins, "property 'customer_id' is not defined"); + flb_azure_conf_destroy(ctx); + return NULL; + } + + /* config: 'shared_key' */ + if (!ctx->shared_key) { + flb_plg_error(ctx->ins, "property 'shared_key' is not defined"); + flb_azure_conf_destroy(ctx); + return NULL; + } + + /* decode shared key */ + size = flb_sds_len(ctx->shared_key) * 1.2; + ctx->dec_shared_key = flb_sds_create_size(size); + if (!ctx->dec_shared_key) { + flb_errno(); + flb_azure_conf_destroy(ctx); + return NULL; + } + + ret = flb_base64_decode((unsigned char *) ctx->dec_shared_key, size, + &olen, + (unsigned char *) ctx->shared_key, + flb_sds_len(ctx->shared_key)); + if (ret != 0) { + flb_plg_error(ctx->ins, "error decoding shared_key"); + flb_azure_conf_destroy(ctx); + return NULL; + } + flb_sds_len_set(ctx->dec_shared_key, olen); + + /* config: 'log_type_key' */ + if (ctx->log_type_key) { + ra_prefix_key = flb_ra_create(ctx->log_type_key, FLB_TRUE); + + if (!ra_prefix_key) { + flb_plg_error(ctx->ins, "invalid log_type_key pattern '%s'", ctx->log_type_key); + flb_azure_conf_destroy(ctx); + return NULL; + } + else { + ctx->ra_prefix_key = ra_prefix_key; + } + } + + /* Validate hostname given by command line or 'Host' property */ + if (!ins->host.name && !ctx->customer_id) { + flb_plg_error(ctx->ins, "property 'customer_id' is not defined"); + flb_free(ctx); + return NULL; + } + + /* Lookup customer id from given host name */ + if (!ctx->customer_id) { + tmp = strchr(ins->host.name, '.'); + if (!tmp) { + flb_plg_error(ctx->ins, "invalid hostname"); + flb_free(ctx); + return NULL; + } + else { + ctx->customer_id = flb_sds_create_len(ins->host.name, + tmp - ins->host.name); + if (!ctx->customer_id) { + flb_errno(); + flb_free(ctx); + return NULL; + } + } + } + + /* Compose real host */ + ctx->host = flb_sds_create_size(256); + if (!ctx->host) { + flb_errno(); + flb_free(ctx); + return NULL; + } + + if (!ins->host.name) { + flb_sds_cat(ctx->host, ctx->customer_id, + flb_sds_len(ctx->customer_id)); + flb_sds_cat(ctx->host, FLB_AZURE_HOST, sizeof(FLB_AZURE_HOST) - 1); + } + else { + if (!strstr(ins->host.name, ctx->customer_id)) { + flb_sds_cat(ctx->host, ctx->customer_id, + flb_sds_len(ctx->customer_id)); + if (ins->host.name[0] != '.') { + flb_sds_cat(ctx->host, ".", 1); + } + } + flb_sds_cat(ctx->host, ins->host.name, strlen(ins->host.name)); + } + + + /* TCP Port */ + if (ins->host.port == 0) { + ctx->port = FLB_AZURE_PORT; + } + else { + ctx->port = ins->host.port; + } + + /* Prepare an upstream handler */ + upstream = flb_upstream_create(config, + ctx->host, + ctx->port, + FLB_IO_TLS, + ins->tls); + if (!upstream) { + flb_plg_error(ctx->ins, "cannot create Upstream context"); + flb_azure_conf_destroy(ctx); + return NULL; + } + ctx->u = upstream; + flb_output_upstream_set(ctx->u, ins); + + /* Compose uri */ + ctx->uri = flb_sds_create_size(1024); + if (!ctx->uri) { + flb_errno(); + flb_azure_conf_destroy(ctx); + return NULL; + } + flb_sds_cat(ctx->uri, FLB_AZURE_RESOURCE, sizeof(FLB_AZURE_RESOURCE) - 1); + flb_sds_cat(ctx->uri, FLB_AZURE_API_VERSION, + sizeof(FLB_AZURE_API_VERSION) - 1); + + flb_plg_info(ctx->ins, "customer_id='%s' host='%s:%i'", + ctx->customer_id, ctx->host, ctx->port); + + return ctx; +} + +int flb_azure_conf_destroy(struct flb_azure *ctx) +{ + if (!ctx) { + return -1; + } + + if (ctx->dec_shared_key) { + flb_sds_destroy(ctx->dec_shared_key); + } + + if (ctx->host) { + flb_sds_destroy(ctx->host); + } + if (ctx->uri) { + flb_sds_destroy(ctx->uri); + } + if (ctx->ra_prefix_key) { + flb_ra_destroy(ctx->ra_prefix_key); + } + if (ctx->u) { + flb_upstream_destroy(ctx->u); + } + flb_free(ctx); + + return 0; +} |