Sha256: 8b736995f60b318ee0575d7440db6dfcc9eef0ca7e04d77c56ec947c3d3b921e

Contents?: true

Size: 1.52 KB

Versions: 7

Compression:

Stored size: 1.52 KB

Contents

# encoding: utf-8
require "stud/temporary"
require "socket"
require "fileutils"

module LogStash
  module Outputs
    class S3
      class WriteBucketPermissionValidator
        attr_reader :logger

        def initialize(logger)
          @logger = logger
        end

        def valid?(bucket_resource, upload_options = {})
          begin
            upload_test_file(bucket_resource)
            true
          rescue StandardError => e
            logger.error("Error validating bucket write permissions!",
              :message => e.message,
              :class => e.class.name,
              :backtrace => e.backtrace
              )
            false
          end
        end

        private
        def upload_test_file(bucket_resource, upload_options = {})
          generated_at = Time.now

          key = "logstash-programmatic-access-test-object-#{generated_at}"
          content = "Logstash permission check on #{generated_at}, by #{Socket.gethostname}"

          begin
            f = Stud::Temporary.file
            f.write(content)
            f.fsync

            obj = bucket_resource.object(key)
            obj.upload_file(f, upload_options)

            begin
              obj.delete
            rescue
              # Try to remove the files on the remote bucket,
              # but don't raise any errors if that doesn't work.
              # since we only really need `putobject`.
            end
          ensure
            f.close
            FileUtils.rm_rf(f.path)
          end
        end
      end
    end
  end
end

Version data entries

7 entries across 7 versions & 1 rubygems

Version Path
logstash-output-s3-4.1.4 lib/logstash/outputs/s3/write_bucket_permission_validator.rb
logstash-output-s3-4.1.3 lib/logstash/outputs/s3/write_bucket_permission_validator.rb
logstash-output-s3-4.1.2 lib/logstash/outputs/s3/write_bucket_permission_validator.rb
logstash-output-s3-4.1.1 lib/logstash/outputs/s3/write_bucket_permission_validator.rb
logstash-output-s3-4.1.0 lib/logstash/outputs/s3/write_bucket_permission_validator.rb
logstash-output-s3-4.0.13 lib/logstash/outputs/s3/write_bucket_permission_validator.rb
logstash-output-s3-4.0.12 lib/logstash/outputs/s3/write_bucket_permission_validator.rb