Contents

# Copyright (c) 2021 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
# frozen_string_literal: true

require 'contrast/components/logger'
require 'contrast/utils/inventory_util'

module Contrast
  module Agent
    module Inventory
      module Policy
        # This Module is how we apply the Data Store detection required for the
        # FlowMap feature. It is called from our patches of the targeted methods
        # in which database operations occur. It is responsible for deciding if
        # the given invocation is worth reporting or not.
        module DataStores
          class << self
            extend Contrast::Components::Logger::InstanceMethods

            # The key used in policy.json to indicate the database type to
            # report.
            DATA_STORE_MARKER = 'data_store'

            def report_data_store _method, _exception, properties, object, _args
              return unless ::Contrast::INVENTORY.enabled?

              marker = properties[DATA_STORE_MARKER]
              return unless marker

              file_report(marker)
            rescue StandardError => e
              logger.error('Error reporting database call', e, object: object)
            end

            private

            def file_report data_store
              return unless data_store

              context = Contrast::Agent::REQUEST_TRACKER.current
              return unless context&.activity

              context.activity.query_count += 1
              return unless context.activity.query_count == 1

              Contrast::Utils::InventoryUtil.append_db_config(context.activity)
            end
          end
        end
      end
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
contrast-agent-4.9.1	lib/contrast/agent/inventory/policy/datastores.rb
contrast-agent-4.9.0	lib/contrast/agent/inventory/policy/datastores.rb