datadog 2.12.0 → 2.22.0

data/lib/datadog/appsec/contrib/devise/event.rb

@@ -1,54 +0,0 @@
-# frozen_string_literal: true
-
-module Datadog
-  module AppSec
-    module Contrib
-      module Devise
-        # Class to extract event information from the resource
-        class Event
-          UUID_REGEX = /^\h{8}-\h{4}-\h{4}-\h{4}-\h{12}$/.freeze
-
-          attr_reader :user_id
-
-          def initialize(resource, mode)
-            @resource = resource
-            @mode = mode
-            @user_id = nil
-            @email = nil
-            @username = nil
-
-            extract if @resource
-          end
-
-          def to_h
-            return @event if defined?(@event)
-
-            @event = {}
-            @event[:email] = @email if @email
-            @event[:username] = @username if @username
-            @event
-          end
-
-          private
-
-          def extract
-            @user_id = @resource.id
-
-            case @mode
-            when AppSec::Configuration::Settings::IDENTIFICATION_AUTO_USER_INSTRUMENTATION_MODE
-              @email = @resource.email
-              @username = @resource.username
-            when AppSec::Configuration::Settings::ANONYMIZATION_AUTO_USER_INSTRUMENTATION_MODE
-              @user_id = nil unless @user_id && @user_id.to_s =~ UUID_REGEX
-            else
-              Datadog.logger.warn(
-                "Invalid auto_user_instrumentation.mode: `#{@mode}`. " \
-                "Supported modes are: #{AppSec::Configuration::Settings::AUTO_USER_INSTRUMENTATION_MODES.join(' | ')}."
-              )
-            end
-          end
-        end
-      end
-    end
-  end
-end

data/lib/datadog/appsec/contrib/devise/patcher/authenticatable_patch.rb

@@ -1,72 +0,0 @@
-# frozen_string_literal: true
-
-require_relative '../configuration'
-require_relative '../tracking'
-require_relative '../resource'
-require_relative '../event'
-
-module Datadog
-  module AppSec
-    module Contrib
-      module Devise
-        module Patcher
-          # Hook in devise validate method
-          module AuthenticatablePatch
-            # rubocop:disable Metrics/MethodLength
-            def validate(resource, &block)
-              result = super
-
-              return result unless AppSec.enabled?
-              return result if @_datadog_appsec_skip_track_login_event
-              return result unless Configuration.auto_user_instrumentation_enabled?
-              return result unless AppSec.active_context
-
-              devise_resource = resource ? Resource.new(resource) : nil
-              event_information = Event.new(devise_resource, Configuration.auto_user_instrumentation_mode)
-
-              if result
-                if event_information.user_id
-                  Datadog.logger.debug { 'AppSec: User successful login event' }
-                else
-                  Datadog.logger.debug do
-                    "AppSec: User successful login event, but can't extract user ID. Tracking empty event"
-                  end
-                end
-
-                Tracking.track_login_success(
-                  AppSec.active_context.trace,
-                  AppSec.active_context.span,
-                  user_id: event_information.user_id,
-                  **event_information.to_h
-                )
-
-                return result
-              end
-
-              user_exists = nil
-
-              if resource
-                user_exists = true
-                Datadog.logger.debug { 'AppSec: User failed login event, but user exists' }
-              else
-                user_exists = false
-                Datadog.logger.debug { 'AppSec: User failed login event and user does not exist' }
-              end
-
-              Tracking.track_login_failure(
-                AppSec.active_context.trace,
-                AppSec.active_context.span,
-                user_id: event_information.user_id,
-                user_exists: user_exists,
-                **event_information.to_h
-              )
-
-              result
-            end
-            # rubocop:enable Metrics/MethodLength
-          end
-        end
-      end
-    end
-  end
-end

data/lib/datadog/appsec/contrib/devise/patcher/registration_controller_patch.rb

@@ -1,47 +0,0 @@
-# frozen_string_literal: true
-
-require_relative '../configuration'
-require_relative '../tracking'
-require_relative '../resource'
-require_relative '../event'
-
-module Datadog
-  module AppSec
-    module Contrib
-      module Devise
-        module Patcher
-          # Hook in devise registration controller
-          module RegistrationControllerPatch
-            def create
-              return super unless AppSec.enabled?
-              return super unless Configuration.auto_user_instrumentation_enabled?
-              return super unless AppSec.active_context
-
-              super do |resource|
-                if resource.persisted?
-                  devise_resource = Resource.new(resource)
-                  event_information = Event.new(devise_resource, Configuration.auto_user_instrumentation_mode)
-
-                  if event_information.user_id
-                    Datadog.logger.debug { 'AppSec: User signup event' }
-                  else
-                    Datadog.logger.warn { "AppSec: User signup event, but can't extract user ID. Tracking empty event" }
-                  end
-
-                  Tracking.track_signup(
-                    AppSec.active_context.trace,
-                    AppSec.active_context.span,
-                    user_id: event_information.user_id,
-                    **event_information.to_h
-                  )
-                end
-
-                yield resource if block_given?
-              end
-            end
-          end
-        end
-      end
-    end
-  end
-end

data/lib/datadog/appsec/contrib/devise/resource.rb

@@ -1,35 +0,0 @@
-# frozen_string_literal: true
-
-module Datadog
-  module AppSec
-    module Contrib
-      module Devise
-        # Class to encpasulate extracting information from a Devise resource
-        # Normally a devise resource would be an Active::Record instance
-        class Resource
-          def initialize(resource)
-            @resource = resource
-          end
-
-          def id
-            extract(:id) || extract(:uuid)
-          end
-
-          def email
-            extract(:email)
-          end
-
-          def username
-            extract(:username)
-          end
-
-          private
-
-          def extract(method)
-            @resource.send(method) if @resource.respond_to?(method)
-          end
-        end
-      end
-    end
-  end
-end

data/lib/datadog/appsec/contrib/devise/tracking.rb

@@ -1,57 +0,0 @@
-# frozen_string_literal: true
-
-require_relative '../../../kit/identity'
-
-module Datadog
-  module AppSec
-    module Contrib
-      module Devise
-        # Internal module to track user events
-        module Tracking
-          LOGIN_SUCCESS_EVENT = 'users.login.success'
-          LOGIN_FAILURE_EVENT = 'users.login.failure'
-          SIGNUP_EVENT = 'users.signup'
-
-          def self.track_login_success(trace, span, user_id:, **others)
-            return if trace.nil? || span.nil?
-
-            track(LOGIN_SUCCESS_EVENT, trace, span, **others)
-
-            Kit::Identity.set_user(trace, span, id: user_id.to_s, **others) if user_id
-          end
-
-          def self.track_login_failure(trace, span, user_id:, user_exists:, **others)
-            return if trace.nil? || span.nil?
-
-            track(LOGIN_FAILURE_EVENT, trace, span, **others)
-
-            span.set_tag('appsec.events.users.login.failure.usr.id', user_id) if user_id
-            span.set_tag('appsec.events.users.login.failure.usr.exists', user_exists)
-          end
-
-          def self.track_signup(trace, span, user_id:, **others)
-            return if trace.nil? || span.nil?
-
-            track(SIGNUP_EVENT, trace, span, **others)
-            Kit::Identity.set_user(trace, id: user_id.to_s, **others) if user_id
-          end
-
-          def self.track(event, trace, span, **others)
-            return if trace.nil? || span.nil?
-
-            span.set_tag("appsec.events.#{event}.track", 'true')
-            span.set_tag("_dd.appsec.events.#{event}.auto.mode", Configuration.track_user_events_mode)
-
-            others.each do |k, v|
-              raise ArgumentError, 'key cannot be :track' if k.to_sym == :track
-
-              span.set_tag("appsec.events.#{event}.#{k}", v) unless v.nil?
-            end
-
-            trace.keep!
-          end
-        end
-      end
-    end
-  end
-end

data/lib/datadog/appsec/processor/rule_merger.rb

@@ -1,170 +0,0 @@
-# frozen_string_literal: true
-
-require_relative '../assets'
-
-module Datadog
-  module AppSec
-    class Processor
-      # RuleMerger merge different sources of information
-      # into the rules payload
-      module RuleMerger
-        # RuleVersionMismatchError
-        class RuleVersionMismatchError < StandardError
-          def initialize(version1, version2)
-            msg = 'Merging rule files with different version could lead to unkown behaviour. '\
-              "We have receieve two rule files with versions: #{version1}, #{version2}. "\
-              'Please validate the configuration is correct and try again.'
-            super(msg)
-          end
-        end
-
-        class << self
-          # TODO: `processors` and `scanners` are not provided by the caller, consider removing them
-          def merge(
-            telemetry:,
-            rules:, data: [], overrides: [], exclusions: [], custom_rules: [],
-            processors: nil, scanners: nil
-          )
-            processors ||= begin
-              default_waf_processors
-            rescue StandardError => e
-              Datadog.logger.error("libddwaf rulemerger failed to parse default waf processors. Error: #{e.inspect}")
-              telemetry.report(
-                e,
-                description: 'libddwaf rulemerger failed to parse default waf processors'
-              )
-              []
-            end
-
-            scanners ||= begin
-              default_waf_scanners
-            rescue StandardError => e
-              Datadog.logger.error("libddwaf rulemerger failed to parse default waf scanners. Error: #{e.inspect}")
-              telemetry.report(
-                e,
-                description: 'libddwaf rulemerger failed to parse default waf scanners'
-              )
-              []
-            end
-
-            combined_rules = combine_rules(rules)
-
-            combined_data = combine_data(data) if data.any?
-            combined_overrides = combine_overrides(overrides) if overrides.any?
-            combined_exclusions = combine_exclusions(exclusions) if exclusions.any?
-            combined_custom_rules = combine_custom_rules(custom_rules) if custom_rules.any?
-
-            combined_rules['rules_data'] = combined_data if combined_data
-            combined_rules['rules_override'] = combined_overrides if combined_overrides
-            combined_rules['exclusions'] = combined_exclusions if combined_exclusions
-            combined_rules['custom_rules'] = combined_custom_rules if combined_custom_rules
-            combined_rules['processors'] = processors
-            combined_rules['scanners'] = scanners
-            combined_rules
-          end
-
-          def default_waf_processors
-            @default_waf_processors ||= JSON.parse(Datadog::AppSec::Assets.waf_processors)
-          end
-
-          def default_waf_scanners
-            @default_waf_scanners ||= JSON.parse(Datadog::AppSec::Assets.waf_scanners)
-          end
-
-          private
-
-          def combine_rules(rules)
-            return rules[0].dup if rules.size == 1
-
-            final_rules = []
-            # @type var final_version: ::String
-            final_version = (_ = nil)
-
-            rules.each do |rule_file|
-              version = rule_file['version']
-
-              if version && !final_version
-                final_version = version
-              elsif final_version != version
-                raise RuleVersionMismatchError.new(final_version, version)
-              end
-
-              final_rules.concat(rule_file['rules'])
-            end
-
-            {
-              'version' => final_version,
-              'rules' => final_rules
-            }
-          end
-
-          def combine_data(data)
-            result = []
-
-            data.each do |data_entry|
-              data_entry.each do |value|
-                existing_data = result.find { |x| x['id'] == value['id'] }
-
-                if existing_data && existing_data['type'] == value['type']
-                  # Duplicate entry base on type and id
-                  # We need to merge the existing data with the new one
-                  # and make sure to remove duplicates
-                  merged_data = merge_data_base_on_expiration(existing_data['data'], value['data'])
-                  existing_data['data'] = merged_data
-                else
-                  result << value
-                end
-              end
-            end
-
-            return unless result.any?
-
-            result
-          end
-
-          def merge_data_base_on_expiration(data1, data2)
-            result = data1.each_with_object({}) do |value, acc|
-              acc[value['value']] = value['expiration']
-            end
-
-            data2.each do |data|
-              if result.key?(data['value'])
-                # The value is duplicated so we need to keep
-                # the one with the highest expiration value
-                # We replace it if the expiration is higher than the current one
-                # or if no experiration
-                current_expiration = result[data['value']]
-                new_expiration = data['expiration']
-
-                if new_expiration.nil? || current_expiration && new_expiration > current_expiration
-                  result[data['value']] = new_expiration
-                end
-              else
-                result[data['value']] = data['expiration']
-              end
-            end
-
-            result.each_with_object([]) do |entry, acc|
-              value = { 'value' => entry[0] }
-              value['expiration'] = entry[1] if entry[1]
-
-              acc << value
-            end
-          end
-
-          def combine_overrides(overrides)
-            overrides.flatten
-          end
-
-          def combine_exclusions(exclusions)
-            exclusions.flatten
-          end
-
-          def combine_custom_rules(custom_rules)
-            custom_rules.flatten
-          end
-        end
-      end
-    end
-  end
-end

data/lib/datadog/appsec/processor.rb

@@ -1,107 +0,0 @@
-# frozen_string_literal: true
-
-require_relative 'security_engine/runner'
-
-module Datadog
-  module AppSec
-    # Processor integrates libddwaf into datadog/appsec
-    # NOTE: This class will be moved under AppSec::SecurityEngine namespace
-    class Processor
-      attr_reader :diagnostics, :addresses
-
-      def initialize(ruleset:, telemetry:)
-        @telemetry = telemetry
-        @diagnostics = nil
-        @addresses = []
-
-        settings = Datadog.configuration.appsec
-
-        # TODO: Refactor to make it easier to test
-        unless require_libddwaf && libddwaf_provides_waf? && create_waf_handle(settings, ruleset)
-          Datadog.logger.warn('AppSec is disabled, see logged errors above')
-        end
-      end
-
-      def ready?
-        !@handle.nil?
-      end
-
-      def finalize
-        @handle.finalize
-      end
-
-      def new_runner
-        SecurityEngine::Runner.new(@handle, telemetry: @telemetry)
-      end
-
-      private
-
-      # libddwaf raises a LoadError on unsupported platforms; it may at some
-      # point succeed in being required yet not provide a specific needed feature.
-      def require_libddwaf
-        Datadog.logger.debug { "libddwaf platform: #{libddwaf_platform}" }
-
-        require 'libddwaf'
-
-        true
-      rescue LoadError => e
-        Datadog.logger.error do
-          'libddwaf failed to load,' \
-            "installed platform: #{libddwaf_platform} ruby platforms: #{ruby_platforms} error: #{e.inspect}"
-        end
-        @telemetry.report(e, description: 'libddwaf failed to load')
-
-        false
-      end
-
-      # check whether libddwaf is required *and* able to provide the needed feature
-      def libddwaf_provides_waf?
-        defined?(Datadog::AppSec::WAF) ? true : false
-      end
-
-      def create_waf_handle(settings, ruleset)
-        # TODO: this may need to be reset if the main Datadog logging level changes after initialization
-        Datadog::AppSec::WAF.logger = Datadog.logger if Datadog.logger.debug? && settings.waf_debug
-
-        obfuscator_config = {
-          key_regex: settings.obfuscator_key_regex,
-          value_regex: settings.obfuscator_value_regex,
-        }
-
-        @handle = Datadog::AppSec::WAF::Handle.new(ruleset, obfuscator: obfuscator_config)
-        @diagnostics = @handle.diagnostics
-        @addresses = @handle.required_addresses
-
-        true
-      rescue WAF::LibDDWAF::Error => e
-        Datadog.logger.error do
-          "libddwaf failed to initialize, error: #{e.inspect}"
-        end
-        @telemetry.report(e, description: 'libddwaf failed to initialize')
-
-        @diagnostics = e.diagnostics if e.diagnostics
-
-        false
-      rescue StandardError => e
-        Datadog.logger.error do
-          "libddwaf failed to initialize, error: #{e.inspect}"
-        end
-        @telemetry.report(e, description: 'libddwaf failed to initialize')
-
-        false
-      end
-
-      def libddwaf_platform
-        if Gem.loaded_specs['libddwaf']
-          Gem.loaded_specs['libddwaf'].platform.to_s
-        else
-          'unknown'
-        end
-      end
-
-      def ruby_platforms
-        Gem.platforms.map(&:to_s)
-      end
-    end
-  end
-end

data/lib/datadog/appsec/utils/trace_operation.rb

@@ -1,15 +0,0 @@
-# frozen_string_literal: true
-
-module Datadog
-  module AppSec
-    module Utils
-      # Utility class to to AppSec-specific trace operations
-      class TraceOperation
-        def self.appsec_standalone_reject?(trace)
-          Datadog.configuration.appsec.standalone.enabled &&
-            (trace.nil? || trace.get_tag(Datadog::AppSec::Ext::TAG_DISTRIBUTED_APPSEC_EVENT) != '1')
-        end
-      end
-    end
-  end
-end

data/lib/datadog/core/telemetry/http/env.rb

@@ -1,20 +0,0 @@
-# frozen_string_literal: true
-
-module Datadog
-  module Core
-    module Telemetry
-      module Http
-        # Data structure for an HTTP request
-        class Env
-          attr_accessor :path, :body
-
-          attr_writer :headers
-
-          def headers
-            @headers ||= {}
-          end
-        end
-      end
-    end
-  end
-end

data/lib/datadog/core/telemetry/http/ext.rb

@@ -1,28 +0,0 @@
-# frozen_string_literal: true
-
-module Datadog
-  module Core
-    module Telemetry
-      module Http
-        module Ext
-          HEADER_DD_API_KEY = 'DD-API-KEY'
-          HEADER_CONTENT_TYPE = 'Content-Type'
-          HEADER_CONTENT_LENGTH = 'Content-Length'
-          HEADER_DD_TELEMETRY_API_VERSION = 'DD-Telemetry-API-Version'
-          HEADER_DD_TELEMETRY_REQUEST_TYPE = 'DD-Telemetry-Request-Type'
-          HEADER_TELEMETRY_DEBUG_ENABLED = 'DD-Telemetry-Debug-Enabled'
-          HEADER_CLIENT_LIBRARY_LANGUAGE = 'DD-Client-Library-Language'
-          HEADER_CLIENT_LIBRARY_VERSION = 'DD-Client-Library-Version'
-
-          CONTENT_TYPE_APPLICATION_JSON = 'application/json'
-          API_VERSION = 'v2'
-
-          AGENTLESS_HOST_PREFIX = 'instrumentation-telemetry-intake'
-
-          AGENT_ENDPOINT = '/telemetry/proxy/api/v2/apmtelemetry'
-          AGENTLESS_ENDPOINT = '/api/v2/apmtelemetry'
-        end
-      end
-    end
-  end
-end

data/lib/datadog/core/telemetry/http/response.rb

@@ -1,70 +0,0 @@
-# frozen_string_literal: true
-
-module Datadog
-  module Core
-    module Telemetry
-      module Http
-        # Module for base HTTP response
-        module Response
-          def payload
-            nil
-          end
-
-          def ok?
-            nil
-          end
-
-          def unsupported?
-            nil
-          end
-
-          def not_found?
-            nil
-          end
-
-          def client_error?
-            nil
-          end
-
-          def server_error?
-            nil
-          end
-
-          def internal_error?
-            nil
-          end
-
-          def code
-            nil
-          end
-
-          def inspect
-            "#{self.class} ok?:#{ok?} unsupported?:#{unsupported?}, " \
-            "not_found?:#{not_found?}, client_error?:#{client_error?}, " \
-            "server_error?:#{server_error?}, internal_error?:#{internal_error?}, " \
-            "payload:#{payload}"
-          end
-        end
-
-        # A generic error response for internal errors
-        class InternalErrorResponse
-          include Response
-
-          attr_reader :error
-
-          def initialize(error)
-            @error = error
-          end
-
-          def internal_error?
-            true
-          end
-
-          def inspect
-            "#{super}, error_type:#{error.class} error:#{error}"
-          end
-        end
-      end
-    end
-  end
-end