in_http: support 'successful_response_code' (#3644)

  The property 'successful_response_code' allows to modify response code.
  Currently, we can set 200, 201(Default) and 204.

  Note: Fluentd supports 200 and 204 with 'use_204_response'

Signed-off-by: Takahiro Yamashita <[email protected]>

Author: nokute78

plugins/in_http/http.c

@@ -91,6 +91,14 @@ static int in_http_init(struct flb_input_instance *ins,
         return -1;
     }
 
+    if (ctx->successful_response_code != 200 &&
+        ctx->successful_response_code != 201 &&
+        ctx->successful_response_code != 204) {
+        flb_plg_error(ctx->ins, "%d is not supported response code. Use default 201",
+                      ctx->successful_response_code);
+        ctx->successful_response_code = 201;
+    }
+
     /* Set the socket non-blocking */
     flb_net_socket_nonblocking(ctx->server_fd);
 
@@ -140,6 +148,12 @@ static struct flb_config_map config_map[] = {
      0, FLB_TRUE, offsetof(struct flb_http, tag_key),
      ""
     },
+    {
+     FLB_CONFIG_MAP_INT, "successful_response_code", "201",
+     0, FLB_TRUE, offsetof(struct flb_http, successful_response_code),
+     "Set successful response code. 200, 201 and 204 are supported."
+    },
+
 
     /* EOF */
     {0}

plugins/in_http/http.h

@@ -32,6 +32,7 @@
 
 struct flb_http {
     int server_fd;
+    int successful_response_code;
     flb_sds_t listen;
     flb_sds_t tcp_port;
     const char *tag_key;

plugins/in_http/http_prot.c

@@ -55,6 +55,20 @@ static int send_response(struct http_conn *conn, int http_status, char *message)
                        "Content-Length: 0\r\n\r\n",
                        FLB_VERSION_STR);
     }
+    else if (http_status == 200) {
+        flb_sds_printf(&out,
+                       "HTTP/1.1 200 OK\r\n"
+                       "Server: Fluent Bit v%s\r\n"
+                       "Content-Length: 0\r\n\r\n",
+                       FLB_VERSION_STR);
+    }
+    else if (http_status == 204) {
+        flb_sds_printf(&out,
+                       "HTTP/1.1 204 No Content\r\n"
+                       "Server: Fluent Bit v%s\r\n"
+                       "Content-Length: 0\r\n\r\n",
+                       FLB_VERSION_STR);
+    }
     else if (http_status == 400) {
         flb_sds_printf(&out,
                        "HTTP/1.1 400 Forbidden\r\n"
@@ -428,6 +442,6 @@ int http_prot_handle(struct flb_http *ctx, struct http_conn *conn,
 
     ret = process_payload(ctx, conn, tag, session, request);
     flb_sds_destroy(tag);
-    send_response(conn, 201, NULL);
+    send_response(conn, ctx->successful_response_code, NULL);
     return ret;
 }