Contents

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

cs__scoped_require 'contrast/components/interface'

module Contrast
  module Agent
    # This module is used to apply instrumentation to classes as they are required
    module TracePointHook
      include Contrast::Components::Interface

      access_component :logging, :scope

      class << self
        def active?
          instance_variable_defined?(:@require_hook) && @require_hook.enabled?
        end

        def enable!
          @require_hook ||= TracePoint.new(:end) do |tracepoint_event|
            process(tracepoint_event)
          end

          @require_hook.enable
          logger.debug('Enabled :end event tracing')
        end

        def disable
          instance_variable_defined?(:@require_hook) && @require_hook.disable
        end

        private

        def process tracepoint_event
          with_contrast_scope do
            logger.debug("Received end event for #{ tracepoint_event.self }")

            loaded_module = tracepoint_event.self
            path = tracepoint_event.path
            return if path&.include?('contrast')

            Contrast::Utils::InventoryUtil.inventory_class(path) if path
            Contrast::Agent::Patching::Policy::Patcher.patch_specific_module(loaded_module)
            Contrast::Agent::Assess::Policy::RewriterPatch.rewrite_interpolation(loaded_module)
            Contrast::Agent::Assess::Policy::PolicyScanner.scan(tracepoint_event)
          end
        end
      end
    end
  end
end

Version data entries

7 entries across 7 versions & 1 rubygems

Version	Path
contrast-agent-3.10.2	lib/contrast/agent/tracepoint_hook.rb
contrast-agent-3.10.1	lib/contrast/agent/tracepoint_hook.rb
contrast-agent-3.10.0	lib/contrast/agent/tracepoint_hook.rb
contrast-agent-3.9.1	lib/contrast/agent/tracepoint_hook.rb
contrast-agent-3.9.0	lib/contrast/agent/tracepoint_hook.rb
contrast-agent-3.8.5	lib/contrast/agent/tracepoint_hook.rb
contrast-agent-3.8.4	lib/contrast/agent/tracepoint_hook.rb