Sha256: 1dbae543d552a543e41043cbf47b04e92e538be39af7b3f9c8166bf4756d4405
Contents?: true
Size: 1.71 KB
Versions: 8
Compression:
Stored size: 1.71 KB
Contents
# Copyright (c) 2022 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
# frozen_string_literal: true
require 'contrast/components/logger'
require 'contrast/agent/inventory/database_config'
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
extend Contrast::Components::Logger::InstanceMethods
class << self
# 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.enable
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::Agent::Inventory::DatabaseConfig.append_db_config(context.activity)
end
end
end
end
end
end
end
Version data entries
8 entries across 8 versions & 1 rubygems