Merge pull request #1 from iamAzeem/feature-tls-support

Added TLS support.

Author: iamazeem

README.md

@@ -8,6 +8,7 @@
 * **Incoming Message Format**: Support for binary or JSON format (`Content-Type`: `application/octet-stream` or `application/json`)
 * **Outgoing Message Format**: Support for binary or JSON format
 * **Message Types**: Single or Batch
+* **TLS Support**: Use `<transport tls> ... </transport>` section in configuration and `https://` URL protocol prefix. See this [example](https://docs.fluentd.org/plugin-helper-overview/api-plugin-helper-server#configuration-example) for more details.
 
 ## Installation
 

fluent-plugin-protobuf-http.gemspec

@@ -8,7 +8,7 @@ Gem::Specification.new do |spec|
   spec.email          = ['[email protected]']
 
   spec.summary        = 'fluentd HTTP Input Plugin for Protocol Buffers'
-  spec.description    = 'fluentd HTTP Input Plugin for Protocol Buffers with Single and Batch Messages Support]'
+  spec.description    = 'fluentd HTTP Input Plugin for Protocol Buffers with Single and Batch Messages Support'
   spec.homepage       = 'https://github.com/iamAzeem/fluent-plugin-protobuf-http'
   spec.license        = 'Apache-2.0'
 

lib/fluent/plugin/in_protobuf_http.rb

@@ -42,6 +42,10 @@ class ProtobufHttpInput < Fluent::Plugin::Input
       desc 'The tag for the event.'
       config_param :tag, :string
 
+      config_section :transport, required: false, multi: false, init: true, param_name: :transport_config do
+        config_argument :protocol, :enum, list: %i[tcp tls], default: :tcp
+      end
+
       def initialize
         super
 
@@ -136,9 +140,18 @@ def start
         compile_protos
         populate_msgclass_lookup
 
-        log.info("Starting protobuf server [#{@bind}:#{@port}]...")
+        # TLS check
+        proto = :tcp
+        tls_opts = nil
+        if @transport_config && @transport_config.protocol == :tls
+          proto = :tls
+          tls_opts = @transport_config.to_h
+        end
+
+        log.info("Starting protobuf #{proto == :tcp ? 'HTTP' : 'HTTPS'} server [#{@bind}:#{@port}]...")
+        log.debug("TLS configuration:\n#{tls_opts}") if tls_opts
 
-        http_server_create_http_server(:protobuf_server, addr: @bind, port: @port, logger: log) do |server|
+        http_server_create_http_server(:protobuf_server, addr: @bind, port: @port, logger: log, proto: proto, tls_opts: tls_opts) do |server|
           server.post("/#{tag}") do |req|
             peeraddr = "#{req.peeraddr[2]}:#{req.peeraddr[1]}".freeze # ip:port
             serialized_msg = req.body.freeze