out_azure: Add support for the x-ms-AzureResourceId Header #2184

Signed-off-by: Florian Koch <flo@ctrl.wtf>

Author: Florian Koch

plugins/out_azure/azure.c

@@ -206,7 +206,11 @@ static int build_headers(struct flb_http_client *c,
     flb_http_add_header(c, "Content-Type", 12, "application/json", 16);
     flb_http_add_header(c, "x-ms-date", 9, rfc1123date,
                         flb_sds_len(rfc1123date));
-
+    /* Header resource_id is optional */
+    if (ctx->resource_id) {
+        flb_http_add_header(c, "x-ms-AzureResourceId", 20, ctx->resource_id ,flb_sds_len(ctx->resource_id));
+        flb_plg_debug(ctx->ins, "resource_id=%s", ctx->resource_id);
+    }
     size = 32 + flb_sds_len(ctx->customer_id) + olen;
     auth = flb_malloc(size);
     if (!auth) {

plugins/out_azure/azure.h

@@ -36,6 +36,7 @@ struct flb_azure {
     /* account setup */
     flb_sds_t customer_id;
     flb_sds_t log_type;
+    flb_sds_t resource_id;
     flb_sds_t shared_key;
     flb_sds_t dec_shared_key;
 

plugins/out_azure/azure_conf.c

@@ -31,6 +31,7 @@ struct flb_azure *flb_azure_conf_create(struct flb_output_instance *ins,
     size_t size;
     size_t olen;
     const char *tmp;
+    const char *rid;
     const char *cid = NULL;
     struct flb_upstream *upstream;
     struct flb_azure *ctx;
@@ -110,6 +111,12 @@ struct flb_azure *flb_azure_conf_create(struct flb_output_instance *ins,
         return NULL;
     }
 
+    /* config: 'resource_id' */
+    rid = flb_output_get_property("resource_id", ins);
+    if (rid) {
+        ctx->resource_id = flb_sds_create(rid);
+    }
+
     /* Validate hostname given by command line or 'Host' property */
     if (!ins->host.name && !cid) {
         flb_plg_error(ctx->ins, "property 'customer_id' is not defined");
@@ -218,6 +225,9 @@ int flb_azure_conf_destroy(struct flb_azure *ctx)
     if (ctx->log_type) {
         flb_sds_destroy(ctx->log_type);
     }
+    if (ctx->resource_id) {
+        flb_sds_destroy(ctx->resource_id);
+    }
     if (ctx->time_key) {
         flb_sds_destroy(ctx->time_key);
     }