contrast-agent 4.9.0 → 4.12.0

data/ext/cs__protect_kernel/cs__protect_kernel.c

@@ -1,47 +0,0 @@
-/* Copyright (c) 2021 Contrast Security, Inc.  See
- * https://www.contrastsecurity.com/enduser-terms-0317a for more details. */
-
-#include "cs__protect_kernel.h"
-#include "../cs__common/cs__common.h"
-#include <ruby.h>
-
-static VALUE contrast_protect_fork(const int argc, const VALUE *argv,
-                                   const VALUE self) {
-    VALUE ret;
-    if (rb_block_given_p()) {
-        /* We call our hook, but it's a little complicated.
-         * We wrap the fork block with our own lambda in
-         * order to instrument it.  There are no public
-         * methods in the Ruby C API to set the prevailing
-         * block, so we have to use rb_funcall_with_block.
-         * Also, rb_funcall_with_block does a public call,
-         * and our method is private.
-         * So we (as a hack) temporarily set it to public.
-         */
-        VALUE wrapper;
-        wrapper =
-            rb_funcall_with_block(kernel_protect, rb_sym_protect_kernel_wrapper,
-                                  0, NULL, rb_block_proc());
-        rb_funcall(rb_mKernel, rb_intern("public"), 1,
-                   ID2SYM(rb_sym_protect_kernel_fork));
-        ret = rb_funcall_with_block(self, rb_sym_protect_kernel_fork, argc,
-                                    argv, wrapper);
-        rb_funcall(rb_mKernel, rb_intern("private"), 1,
-                   ID2SYM(rb_sym_protect_kernel_fork));
-    } else {
-        ret = rb_funcall2(self, rb_sym_protect_kernel_fork, argc, argv);
-    }
-    return ret;
-}
-
-void Init_cs__protect_kernel(void) {
-    VALUE core_protect = rb_define_module_under(core_extensions, "Protect");
-    kernel_protect = rb_define_module_under(core_protect, "Kernel");
-    rb_sym_protect_kernel_wrapper = rb_intern("build_wrapper");
-
-    rb_sym_protect_kernel_fork =
-        contrast_register_patch("Kernel", "fork", &contrast_protect_fork);
-
-    rb_sym_protect_kernel_fork = contrast_register_singleton_patch(
-        "Kernel", "fork", &contrast_protect_fork);
-}

data/ext/cs__protect_kernel/cs__protect_kernel.h

@@ -1,12 +0,0 @@
-#include <ruby.h>
-
-extern VALUE rb_vm_top_self(void);
-
-static VALUE kernel_protect;
-static VALUE rb_sym_protect_kernel_fork;
-static VALUE rb_sym_protect_kernel_wrapper;
-
-static VALUE contrast_protect_fork(const int argc, const VALUE *argv,
-                                   const VALUE self);
-
-void Init_cs__protect_kernel(void);

data/ext/cs__protect_kernel/extconf.rb

@@ -1,5 +0,0 @@
-# Copyright (c) 2021 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
-# frozen_string_literal: true
-
-$TO_MAKE = File.basename(__dir__)
-require_relative '../extconf_common'

data/lib/contrast/extension/protect/kernel.rb

@@ -1,39 +0,0 @@
-# Copyright (c) 2021 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
-# frozen_string_literal: true
-
-module Contrast
-  module Extension
-    module Protect
-      # This Module functions as our patch into the Kernel class for Protect,
-      # allowing us to track activity as it crosses spawned processes.
-      module Kernel
-        class << self
-          def build_wrapper
-            lambda {
-              proc_start
-              yield
-              # AtExitHook handles sending any messages generated in the new forked process
-            }
-          end
-
-          def proc_start
-            context = Contrast::Agent::REQUEST_TRACKER.current
-            return unless context
-
-            context.reset_activity
-          end
-
-          def instrument
-            @_instrument ||= begin
-              require 'cs__protect_kernel/cs__protect_kernel'
-              true
-            end
-          rescue StandardError, LoadError => e
-            logger.error('Error loading kernel protect patch', e)
-            false
-          end
-        end
-      end
-    end
-  end
-end

data/lib/contrast/utils/inventory_util.rb

@@ -1,113 +0,0 @@
-# Copyright (c) 2021 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
-# frozen_string_literal: true
-
-require 'contrast/utils/timer'
-require 'contrast/utils/object_share'
-require 'contrast/components/logger'
-
-module Contrast
-  module Utils
-    # Utilities for getting inventory information from the application
-    class InventoryUtil
-      extend Contrast::Components::Logger::InstanceMethods
-
-      # TeamServer only accepts certain values for ArchitectureComponents.
-      # DO NOT CHANGE THIS!
-      AC_TYPE_DB = 'db'
-      # TeamServer only accepts certain values for FlowMap Services.
-      # DO NOT CHANGE THIS
-      ADAPTER = 'adapter'
-      HOST = 'host'
-      PORT = 'port'
-      DATABASE = 'database'
-      DEFAULT = 'default'
-      LOCALHOST = 'localhost'
-
-      def self.active_record_config
-        return @_active_record_config if instance_variable_defined?(:@_active_record_config)
-
-        @_active_record_config = ActiveRecord::Base.connection_config rescue nil # rubocop:disable Style/RescueModifier
-      end
-
-      def self.append_db_config activity_or_update, hash_or_str = Contrast::Utils::InventoryUtil.active_record_config
-        arr = build_from_db_config(hash_or_str)
-        return unless arr&.any?
-
-        arr.each do |a|
-          next unless a
-
-          if activity_or_update.is_a?(Contrast::Api::Dtm::Activity)
-            activity_or_update.architectures << a
-          else
-            activity_or_update.components << a
-          end
-        end
-      rescue StandardError => e
-        logger.error('Unable to append db config', e)
-        nil
-      end
-
-      def self.build_from_db_config hash_or_str
-        return unless hash_or_str
-
-        if hash_or_str.is_a?(Hash)
-          build_from_db_hash(hash_or_str)
-        else
-          build_from_db_string(hash_or_str.to_s)
-        end
-      end
-
-      def self.build_from_db_hash hash
-        ac = Contrast::Api::Dtm::ArchitectureComponent.new
-        ac.vendor = hash[:adapter] || hash[ADAPTER] || Contrast::Utils::ObjectShare::EMPTY_STRING
-        ac.remote_host = host_from_hash(hash)
-        ac.remote_port = port_from_hash(hash)
-        ac.type = AC_TYPE_DB
-        ac.url = hash[:database] || hash[DATABASE] || DEFAULT
-        [ac]
-      end
-
-      def self.host_from_hash hash
-        hash[:host] || hash[HOST] || Contrast::Utils::ObjectShare::EMPTY_STRING
-      end
-
-      def self.port_from_hash hash
-        p = hash[:port] || hash[PORT] || Contrast::Utils::ObjectShare::EMPTY_STRING
-        p.to_i
-      end
-
-      # Examples:
-      # mongodb://[user:pass@]host1[:port1][,host2[:port2],[,hostN[:portN]]][/[database][?options]]
-      # postgresql://scott:tiger@localhost/mydatabase
-      # mysql+mysqlconnector://scott:tiger@localhost/foo
-      def self.build_from_db_string str
-        adapter, hosts, database = split_connection_str(str)
-        acs = []
-        hosts.split(Contrast::Utils::ObjectShare::COMMA).map do |s|
-          host, port = s.split(Contrast::Utils::ObjectShare::COLON)
-
-          ac = Contrast::Api::Dtm::ArchitectureComponent.new
-          ac.vendor = Contrast::Utils::StringUtils.force_utf8(adapter)
-          ac.remote_host = Contrast::Utils::StringUtils.force_utf8(host)
-          ac.remote_port = port.to_i
-          ac.type = AC_TYPE_DB
-          ac.url = Contrast::Utils::StringUtils.force_utf8(database)
-          acs << ac
-        end
-        acs
-      end
-
-      def self.split_connection_str str
-        adapter, str = str.split(Contrast::Utils::ObjectShare::COLON_SLASH_SLASH)
-        _auth, str = str.split(Contrast::Utils::ObjectShare::AT)
-        # Not currently used
-        # user, pass = auth.split(Contrast::Utils::ObjectShare::COLON)
-        hosts, db_and_options = str.split(Contrast::Utils::ObjectShare::SLASH)
-        hosts << LOCALHOST if hosts.empty?
-        database, _options = db_and_options.split(Contrast::Utils::ObjectShare::QUESTION_MARK)
-
-        [adapter, hosts, database]
-      end
-    end
-  end
-end