Sha256: 165a4015fe5ff3a82bac74a603d29fe227362256532400ea64b1fa44361dfb8f
Contents?: true
Size: 1.72 KB
Versions: 11
Compression:
Stored size: 1.72 KB
Contents
# Copyright (c) 2020 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
# frozen_string_literal: true
require 'contrast/components/interface'
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
include Contrast::Components::Interface
access_component :analysis, :logging
# 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 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
11 entries across 11 versions & 1 rubygems