Sha256: 625cda06f8050ea99b3016160c450827f233d2822aa0fe565ae494512f7f23c9

Contents?: true

Size: 1.15 KB

Versions: 9

Compression:

Stored size: 1.15 KB

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'

module Contrast
  module Agent
    # This module adds an at_exit hook for us to send messages that may be lost at process exit
    module AtExitHook
      extend Contrast::Components::Logger::InstanceMethods

      def self.exit_hook
        @_exit_hook ||= begin
          at_exit do
            on_exit
          end
          true
        end
      end

      # Actions to take when a process exits. Typically called from our
      # exit_hook, but exposed here for other process terminations, like those
      # in Kernel#exec
      def self.on_exit
        logger.debug('at_exit invoked, host application terminating',
                     p_id: @pid,
                     pp_id: @ppid,
                     process_pid: Process.pid,
                     process_pp_id: Process.ppid)

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

        Contrast::Agent.messaging_queue.send_event_immediately(context.activity)
      end
    end
  end
end

Version data entries

9 entries across 9 versions & 1 rubygems

Version Path
contrast-agent-4.14.1 lib/contrast/agent/at_exit_hook.rb
contrast-agent-4.14.0 lib/contrast/agent/at_exit_hook.rb
contrast-agent-4.13.1 lib/contrast/agent/at_exit_hook.rb
contrast-agent-4.13.0 lib/contrast/agent/at_exit_hook.rb
contrast-agent-4.12.0 lib/contrast/agent/at_exit_hook.rb
contrast-agent-4.11.0 lib/contrast/agent/at_exit_hook.rb
contrast-agent-4.10.0 lib/contrast/agent/at_exit_hook.rb
contrast-agent-4.9.1 lib/contrast/agent/at_exit_hook.rb
contrast-agent-4.9.0 lib/contrast/agent/at_exit_hook.rb