datadog 2.8.0 → 2.39.0

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

@@ -1,57 +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
-
-          SAFE_MODE = 'safe'
-          EXTENDED_MODE = 'extended'
-
-          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 EXTENDED_MODE
-              @email = @resource.email
-              @username = @resource.username
-            when SAFE_MODE
-              @user_id = nil unless @user_id && @user_id.to_s =~ UUID_REGEX
-            else
-              Datadog.logger.warn(
-                "Invalid automated user evenst mode: `#{@mode}`. "\
-                              'Supported modes are: `safe` and `extended`.'
-              )
-            end
-          end
-        end
-      end
-    end
-  end
-end

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

@@ -1,77 +0,0 @@
-# frozen_string_literal: true
-
-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_skip_track_login_event
-
-              track_user_events_configuration = Datadog.configuration.appsec.track_user_events
-
-              return result unless track_user_events_configuration.enabled
-
-              automated_track_user_events_mode = track_user_events_configuration.mode
-
-              appsec_scope = Datadog::AppSec.active_scope
-
-              return result unless appsec_scope
-
-              devise_resource = resource ? Resource.new(resource) : nil
-
-              event_information = Event.new(devise_resource, automated_track_user_events_mode)
-
-              if result
-                if event_information.user_id
-                  Datadog.logger.debug { 'User Login Event success' }
-                else
-                  Datadog.logger.debug { 'User Login Event success, but can\'t extract user ID. Tracking empty event' }
-                end
-
-                Tracking.track_login_success(
-                  appsec_scope.trace,
-                  appsec_scope.service_entry_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 { 'User Login Event failure users exists' }
-              else
-                user_exists = false
-                Datadog.logger.debug { 'User Login Event failure user do not exists' }
-              end
-
-              Tracking.track_login_failure(
-                appsec_scope.trace,
-                appsec_scope.service_entry_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,54 +0,0 @@
-# frozen_string_literal: true
-
-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?
-
-              track_user_events_configuration = Datadog.configuration.appsec.track_user_events
-
-              return super unless track_user_events_configuration.enabled
-
-              automated_track_user_events_mode = track_user_events_configuration.mode
-
-              appsec_scope = Datadog::AppSec.active_scope
-              return super unless appsec_scope
-
-              super do |resource|
-                if resource.persisted?
-                  devise_resource = Resource.new(resource)
-
-                  event_information = Event.new(devise_resource, automated_track_user_events_mode)
-
-                  if event_information.user_id
-                    Datadog.logger.debug { 'User Signup Event' }
-                  else
-                    Datadog.logger.warn { 'User Signup Event, but can\'t extract user ID. Tracking empty event' }
-                  end
-
-                  Tracking.track_signup(
-                    appsec_scope.trace,
-                    appsec_scope.service_entry_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", Datadog.configuration.appsec.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/contrib/graphql/reactive/multiplex.rb

@@ -1,46 +0,0 @@
-# frozen_string_literal: true
-
-module Datadog
-  module AppSec
-    module Contrib
-      module GraphQL
-        module Reactive
-          # Dispatch data from a GraphQL resolve query to the WAF context
-          module Multiplex
-            ADDRESSES = [
-              'graphql.server.all_resolvers'
-            ].freeze
-            private_constant :ADDRESSES
-
-            def self.publish(op, gateway_multiplex)
-              catch(:block) do
-                op.publish('graphql.server.all_resolvers', gateway_multiplex.arguments)
-
-                nil
-              end
-            end
-
-            def self.subscribe(op, waf_context)
-              op.subscribe(*ADDRESSES) do |*values|
-                Datadog.logger.debug { "reacted to #{ADDRESSES.inspect}: #{values.inspect}" }
-                arguments = values[0]
-
-                persistent_data = {
-                  'graphql.server.all_resolvers' => arguments
-                }
-
-                waf_timeout = Datadog.configuration.appsec.waf_timeout
-                result = waf_context.run(persistent_data, {}, waf_timeout)
-
-                next if result.status != :match
-
-                yield result
-                throw(:block, true) unless result.actions.empty?
-              end
-            end
-          end
-        end
-      end
-    end
-  end
-end

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

@@ -1,12 +0,0 @@
-# frozen_string_literal: true
-
-module Datadog
-  module AppSec
-    module Contrib
-      # Common patcher code for AppSec integrations
-      # TODO: empty implementation, check with tracer shareable code
-      module Patcher
-      end
-    end
-  end
-end

data/lib/datadog/appsec/contrib/rack/reactive/request.rb

@@ -1,69 +0,0 @@
-# frozen_string_literal: true
-
-module Datadog
-  module AppSec
-    module Contrib
-      module Rack
-        module Reactive
-          # Dispatch data from a Rack request to the WAF context
-          module Request
-            ADDRESSES = [
-              'request.headers',
-              'request.uri.raw',
-              'request.query',
-              'request.cookies',
-              'request.client_ip',
-              'server.request.method'
-            ].freeze
-            private_constant :ADDRESSES
-
-            def self.publish(op, gateway_request)
-              catch(:block) do
-                op.publish('request.query', gateway_request.query)
-                op.publish('request.headers', gateway_request.headers)
-                op.publish('request.uri.raw', gateway_request.fullpath)
-                op.publish('request.cookies', gateway_request.cookies)
-                op.publish('request.client_ip', gateway_request.client_ip)
-                op.publish('server.request.method', gateway_request.method)
-
-                nil
-              end
-            end
-
-            def self.subscribe(op, waf_context)
-              op.subscribe(*ADDRESSES) do |*values|
-                Datadog.logger.debug { "reacted to #{ADDRESSES.inspect}: #{values.inspect}" }
-
-                headers = values[0]
-                headers_no_cookies = headers.dup.tap { |h| h.delete('cookie') }
-                uri_raw = values[1]
-                query = values[2]
-                cookies = values[3]
-                client_ip = values[4]
-                request_method = values[5]
-
-                persistent_data = {
-                  'server.request.cookies' => cookies,
-                  'server.request.query' => query,
-                  'server.request.uri.raw' => uri_raw,
-                  'server.request.headers' => headers,
-                  'server.request.headers.no_cookies' => headers_no_cookies,
-                  'http.client_ip' => client_ip,
-                  'server.request.method' => request_method,
-                }
-
-                waf_timeout = Datadog.configuration.appsec.waf_timeout
-                result = waf_context.run(persistent_data, {}, waf_timeout)
-
-                next if result.status != :match
-
-                yield result
-                throw(:block, true) unless result.actions.empty?
-              end
-            end
-          end
-        end
-      end
-    end
-  end
-end

data/lib/datadog/appsec/contrib/rack/reactive/request_body.rb

@@ -1,47 +0,0 @@
-# frozen_string_literal: true
-
-module Datadog
-  module AppSec
-    module Contrib
-      module Rack
-        module Reactive
-          # Dispatch data from a Rack request to the WAF context
-          module RequestBody
-            ADDRESSES = [
-              'request.body',
-            ].freeze
-            private_constant :ADDRESSES
-
-            def self.publish(op, gateway_request)
-              catch(:block) do
-                # params have been parsed from the request body
-                op.publish('request.body', gateway_request.form_hash)
-
-                nil
-              end
-            end
-
-            def self.subscribe(op, waf_context)
-              op.subscribe(*ADDRESSES) do |*values|
-                Datadog.logger.debug { "reacted to #{ADDRESSES.inspect}: #{values.inspect}" }
-                body = values[0]
-
-                persistent_data = {
-                  'server.request.body' => body,
-                }
-
-                waf_timeout = Datadog.configuration.appsec.waf_timeout
-                result = waf_context.run(persistent_data, {}, waf_timeout)
-
-                next if result.status != :match
-
-                yield result
-                throw(:block, true) unless result.actions.empty?
-              end
-            end
-          end
-        end
-      end
-    end
-  end
-end

data/lib/datadog/appsec/contrib/rack/reactive/response.rb

@@ -1,53 +0,0 @@
-# frozen_string_literal: true
-
-module Datadog
-  module AppSec
-    module Contrib
-      module Rack
-        module Reactive
-          # Dispatch data from a Rack response to the WAF context
-          module Response
-            ADDRESSES = [
-              'response.status',
-              'response.headers',
-            ].freeze
-            private_constant :ADDRESSES
-
-            def self.publish(op, gateway_response)
-              catch(:block) do
-                op.publish('response.status', gateway_response.status)
-                op.publish('response.headers', gateway_response.headers)
-
-                nil
-              end
-            end
-
-            def self.subscribe(op, waf_context)
-              op.subscribe(*ADDRESSES) do |*values|
-                Datadog.logger.debug { "reacted to #{ADDRESSES.inspect}: #{values.inspect}" }
-
-                response_status = values[0]
-                response_headers = values[1]
-                response_headers_no_cookies = response_headers.dup.tap { |h| h.delete('set-cookie') }
-
-                persistent_data = {
-                  'server.response.status' => response_status.to_s,
-                  'server.response.headers' => response_headers,
-                  'server.response.headers.no_cookies' => response_headers_no_cookies,
-                }
-
-                waf_timeout = Datadog.configuration.appsec.waf_timeout
-                result = waf_context.run(persistent_data, {}, waf_timeout)
-
-                next if result.status != :match
-
-                yield result
-                throw(:block, true) unless result.actions.empty?
-              end
-            end
-          end
-        end
-      end
-    end
-  end
-end

data/lib/datadog/appsec/contrib/rails/ext.rb

@@ -1,13 +0,0 @@
-# frozen_string_literal: true
-
-module Datadog
-  module AppSec
-    module Contrib
-      module Rails
-        # Rails integration constants
-        module Ext
-        end
-      end
-    end
-  end
-end

data/lib/datadog/appsec/contrib/rails/reactive/action.rb

@@ -1,53 +0,0 @@
-# frozen_string_literal: true
-
-require_relative '../request'
-
-module Datadog
-  module AppSec
-    module Contrib
-      module Rails
-        module Reactive
-          # Dispatch data from a Rails request to the WAF context
-          module Action
-            ADDRESSES = [
-              'rails.request.body',
-              'rails.request.route_params',
-            ].freeze
-            private_constant :ADDRESSES
-
-            def self.publish(op, gateway_request)
-              catch(:block) do
-                # params have been parsed from the request body
-                op.publish('rails.request.body', gateway_request.parsed_body)
-                op.publish('rails.request.route_params', gateway_request.route_params)
-
-                nil
-              end
-            end
-
-            def self.subscribe(op, waf_context)
-              op.subscribe(*ADDRESSES) do |*values|
-                Datadog.logger.debug { "reacted to #{ADDRESSES.inspect}: #{values.inspect}" }
-                body = values[0]
-                path_params = values[1]
-
-                persistent_data = {
-                  'server.request.body' => body,
-                  'server.request.path_params' => path_params,
-                }
-
-                waf_timeout = Datadog.configuration.appsec.waf_timeout
-                result = waf_context.run(persistent_data, {}, waf_timeout)
-
-                next if result.status != :match
-
-                yield result
-                throw(:block, true) unless result.actions.empty?
-              end
-            end
-          end
-        end
-      end
-    end
-  end
-end

data/lib/datadog/appsec/contrib/sinatra/ext.rb

@@ -1,14 +0,0 @@
-# frozen_string_literal: true
-
-module Datadog
-  module AppSec
-    module Contrib
-      module Sinatra
-        # Sinatra integration constants
-        module Ext
-          ROUTE_INTERRUPT = :datadog_appsec_contrib_sinatra_route_interrupt
-        end
-      end
-    end
-  end
-end

data/lib/datadog/appsec/contrib/sinatra/reactive/routed.rb

@@ -1,48 +0,0 @@
-# frozen_string_literal: true
-
-module Datadog
-  module AppSec
-    module Contrib
-      module Sinatra
-        module Reactive
-          # Dispatch data from a Sinatra request to the WAF context
-          module Routed
-            ADDRESSES = [
-              'sinatra.request.route_params',
-            ].freeze
-            private_constant :ADDRESSES
-
-            def self.publish(op, data)
-              _request, route_params = data
-
-              catch(:block) do
-                op.publish('sinatra.request.route_params', route_params.params)
-
-                nil
-              end
-            end
-
-            def self.subscribe(op, waf_context)
-              op.subscribe(*ADDRESSES) do |*values|
-                Datadog.logger.debug { "reacted to #{ADDRESSES.inspect}: #{values.inspect}" }
-                path_params = values[0]
-
-                persistent_data = {
-                  'server.request.path_params' => path_params,
-                }
-
-                waf_timeout = Datadog.configuration.appsec.waf_timeout
-                result = waf_context.run(persistent_data, {}, waf_timeout)
-
-                next if result.status != :match
-
-                yield result
-                throw(:block, true) unless result.actions.empty?
-              end
-            end
-          end
-        end
-      end
-    end
-  end
-end

data/lib/datadog/appsec/monitor/reactive/set_user.rb

@@ -1,45 +0,0 @@
-# frozen_string_literal: true
-
-module Datadog
-  module AppSec
-    module Monitor
-      module Reactive
-        # Dispatch data from Datadog::Kit::Identity.set_user to the WAF context
-        module SetUser
-          ADDRESSES = [
-            'usr.id',
-          ].freeze
-          private_constant :ADDRESSES
-
-          def self.publish(op, user)
-            catch(:block) do
-              op.publish('usr.id', user.id)
-
-              nil
-            end
-          end
-
-          def self.subscribe(op, waf_context)
-            op.subscribe(*ADDRESSES) do |*values|
-              Datadog.logger.debug { "reacted to #{ADDRESSES.inspect}: #{values.inspect}" }
-
-              user_id = values[0]
-
-              persistent_data = {
-                'usr.id' => user_id,
-              }
-
-              waf_timeout = Datadog.configuration.appsec.waf_timeout
-              result = waf_context.run(persistent_data, {}, waf_timeout)
-
-              next if result.status != :match
-
-              yield result
-              throw(:block, true) unless result.actions.empty?
-            end
-          end
-        end
-      end
-    end
-  end
-end