contrast-agent 6.7.0 → 6.8.0

data/lib/contrast/agent/protect/rule/sql_sample_builder.rb

@@ -15,11 +15,11 @@ module Contrast
           # by TeamServer
           #
           # @param context [Contrast::Agent::RequestContext] the context for the current request
-          # @param input_analysis_result [Contrast::Api::Dtm::AttackResult, nil] previous attack result for this rule,
+          # @param input_analysis_result [Contrast::Agent::Reporting, nil] previous attack result for this rule,
           #   if one exists, in the case of multiple inputs being found to violate the protection criteria
           # @candidate_string [String] the value of the input which may be an attack
           # @kwargs [Hash] key - value pairs of context individual rules need to build out details
-          #   to send to the Service to tell the story of the attack
+          #   to send to TeamServer to tell the story of the attack
           # @return [Contrast::Agent::Reporting::RaspRuleSample] the sample from this attack
           module SqliSample
             def build_sample context, input_analysis_result, candidate_string, **kwargs
@@ -38,11 +38,11 @@ module Contrast
           # by TeamServer
           #
           # @param context [Contrast::Agent::RequestContext] the context for the current request
-          # @param input_analysis_result [Contrast::Api::Dtm::AttackResult, nil] previous attack result for this rule,
+          # @param input_analysis_result [Contrast::Agent::Reporting, nil] previous attack result for this rule,
           #   if one exists, in the case of multiple inputs being found to violate the protection criteria
           # @candidate_string [String] the value of the input which may be an attack
           # @kwargs [Hash] key - value pairs of context individual rules need to build out details
-          #   to send to the Service to tell the story of the attack
+          #   to send to TeamServer to tell the story of the attack
           # @return [Contrast::Agent::Reporting::RaspRuleSample] the sample from this attack
           module NoSqliSample
             def build_sample context, input_analysis_result, candidate_string, **kwargs
@@ -63,24 +63,21 @@ module Contrast
             # Set up an attack result and assigns Database scanner for the No-SQL and SQLI injection detection rules
             #
             # @param context [Contrast::Agent::RequestContext] the context for the current request
-            # @param input_analysis_result [Contrast::Api::Dtm::AttackResult, nil] previous attack result for this rule,
+            # @param input_analysis_result [Contrast::Agent::Reporting, nil] previous attack result for this rule,
             #   if one exists, in the case of multiple inputs being found to violate the protection criteria
-            # @param result [Contrast::Api::Dtm::AttackResult, nil] previous attack result for this rule, if one exists,
+            # @param result [Contrast::Agent::Reporting, nil] previous attack result for this rule, if one exists,
             #   in the case of multiple inputs being found to violate the protection criteria
             # @param query_string [String] the value of the input which may be an attack
             # @param kwargs [Hash] key - value pairs of context individual rules need to build out details to send
-            #   to the Service to tell the story of the attack
-            # @return [Contrast::Api::Dtm::AttackResult] the result from this attack
+            #   to TeamServer to tell the story of the attack
+            # @return [Contrast::Agent::Reporting] the result from this attack
             def build_attack_with_match context, input_analysis_result, result, query_string, **kwargs
-              if mode == Contrast::Api::Settings::ProtectionRule::Mode::NO_ACTION ||
-                    mode == Contrast::Api::Settings::ProtectionRule::Mode::PERMIT
-
-                return result
-              end
+              return result if mode == :NO_ACTION || mode == :PERMIT
 
               attack_string = input_analysis_result.value
               regexp = Regexp.new(Regexp.escape(attack_string), Regexp::IGNORECASE)
-
+              # extract struct result from kwargs
+              agent_lib_struct_result = kwargs[:result_struct]
               return unless query_string.match?(regexp)
 
               database = kwargs[:database]
@@ -94,7 +91,14 @@ module Contrast
                 next unless last_boundary && boundary
 
                 result ||= build_attack_result(context)
-                record_match(idx, length, boundary, last_boundary, kwargs)
+
+                # if the struct actually has the needed data in it - use it
+                if agent_lib_struct_result.cs__is_a?(Hash)
+                  record_agent_lib_match(agent_lib_struct_result, length, kwargs)
+                else
+                  record_match(idx, length, boundary, last_boundary, kwargs)
+                end
+
                 append_match(context, input_analysis_result, result, query_string, **kwargs)
               end
 
@@ -124,6 +128,20 @@ module Contrast
               kwargs[:input_boundary_idx] = last_boundary
             end
 
+            # all the agent_lib checks methods needed
+
+            # @param struct[ResultingStruct] The struct including all the data from the agent_lib scan
+            def record_agent_lib_match struct, length, kwargs
+              kwargs[:start_idx] = struct[:start_index]
+              kwargs[:end_idx] = if (struct[:end_index]).zero?
+                                   struct[:start_index] + length
+                                 else
+                                   struct[:end_index]
+                                 end
+              kwargs[:boundary_overrun_idx] = struct[:boundary_overrun_index]
+              kwargs[:input_boundary_idx] = struct[:input_boundary_index]
+            end
+
             def append_match context, input_analysis_result, result, query_string, **kwargs
               input_analysis_result.attack_count = input_analysis_result.attack_count + 1
               update_successful_attack_response(context, input_analysis_result, result, query_string)

data/lib/contrast/agent/protect/rule/sqli/sqli_base_rule.rb

@@ -19,20 +19,6 @@ module Contrast
             XML_VALUE, DWR_VALUE
           ].cs__freeze
 
-          class << self
-            # @param attack_sample [Contrast::Api::Dtm::RaspRuleSample]
-            # @return [Hash] the details for this specific rule
-            def extract_details attack_sample
-              {
-                  start: attack_sample.sqli.start_idx,
-                  end: attack_sample.sqli.end_idx,
-                  boundaryOverrunIndex: attack_sample.sqli.boundary_overrun_idx,
-                  inputBoundaryIndex: attack_sample.sqli.input_boundary_idx,
-                  query: attack_sample.sqli.query
-              }
-            end
-          end
-
           def infilter context, database, query_string
             return unless infilter?(context)
 

data/lib/contrast/agent/protect/rule/sqli/sqli_input_classification.rb

@@ -5,10 +5,8 @@ require 'contrast/utils/object_share'
 require 'contrast/agent/reporting/input_analysis/input_type'
 require 'contrast/agent/protect/rule/sqli'
 require 'contrast/agent/reporting/input_analysis/score_level'
-require 'contrast/agent/protect/rule/sqli/sqli_worth_watching'
 require 'contrast/agent/protect/input_analyzer/input_analyzer'
-require 'contrast/utils/input_classification'
-require 'contrast/components/logger'
+require 'contrast/utils/input_classification_base'
 
 module Contrast
   module Agent
@@ -18,68 +16,10 @@ module Contrast
         # as a result input would be marked as WORTHWATCHING or IGNORE,
         # to be analyzed at the sink level.
         module SqliInputClassification
+          WORTHWATCHING_MATCH = 'sqli-worth-watching-v2'.cs__freeze
           class << self
             include InputClassificationBase
-            include Contrast::Agent::Protect::Rule::SqliWorthWatching
             include Contrast::Components::Logger::InstanceMethods
-
-            WORTHWATCHING_MATCH = 'sqli-worth-watching-v2'
-            SQLI_KEYS_NEEDED = [
-              COOKIE_VALUE, PARAMETER_VALUE, JSON_VALUE, MULTIPART_VALUE, XML_VALUE, DWR_VALUE
-            ].cs__freeze
-
-            # Input Classification stage is done to determine if an user input is
-            # WORTHWATCHING or to be ignored.
-            #
-            # @param input_type [Contrast::Agent::Reporting::InputType] The type of the user input.
-            # @param value [String, Array<String>] the value of the input.
-            # @param input_analysis [Contrast::Agent::Reporting::InputAnalysis] Holds all the results from the
-            #                                                       agent analysis from the current
-            #                                                       Request.
-            # @return ia [Contrast::Agent::Reporting::InputAnalysis] with updated results.
-            def classify input_type, value, input_analysis
-              return unless Contrast::Agent::Protect::Rule::Sqli::APPLICABLE_USER_INPUTS.include?(input_type)
-              return unless super
-
-              rule_id = Contrast::Agent::Protect::Rule::Sqli::NAME
-
-              # double check the input to avoid calling match? on array
-              Array(value).each do |val|
-                Array(val).each do |v|
-                  input_analysis.results << sqli_create_new_input_result(input_analysis.request, rule_id, input_type, v)
-                end
-              end
-
-              input_analysis
-            rescue StandardError => e
-              logger.debug('An Error was recorded in the input classification of the sqli.')
-              logger.debug(e)
-            end
-
-            private
-
-            # This methods checks if input is tagged WORTHWATCHING or IGNORE matches value with it's
-            # key if needed and Creates new isntance of InputAnalysisResult.
-            #
-            # @param request [Contrast::Agent::Request] the current request context.
-            # @param rule_id [String] The name of the Protect Rule.
-            # @param input_type [Contrast::Agent::Reporting::InputType] The type of the user input.
-            # @param value [String, Array<String>] the value of the input.
-            #
-            # @return res [Contrast::Agent::Reporting::InputAnalysisResult]
-            def sqli_create_new_input_result request, rule_id, input_type, value
-              ia_result = new_ia_result(rule_id, input_type, request.path, value)
-              if sqli_worth_watching?(value)
-                ia_result.ids << WORTHWATCHING_MATCH
-                ia_result.score_level = WORTHWATCHING
-                ia_result
-              else
-                ia_result.score_level = IGNORE
-              end
-
-              add_needed_key(request, ia_result, input_type, value) if SQLI_KEYS_NEEDED.include?(input_type)
-              ia_result
-            end
           end
         end
       end

data/lib/contrast/agent/protect/rule/sqli.rb

@@ -39,6 +39,76 @@ module Contrast
           def sub_rules
             SUB_RULES
           end
+
+          def applicable_user_inputs
+            APPLICABLE_USER_INPUTS
+          end
+
+          def infilter context, database, query_string
+            return unless infilter?(context)
+
+            result = find_attacker(context, query_string, database: database)
+            return unless result
+
+            append_to_activity(context, result)
+
+            cef_logging(result, :successful_attack, value: query_string)
+            raise(Contrast::SecurityException.new(self, BLOCK_MESSAGE)) if blocked?
+          end
+
+          # Allows for the InputAnalysis from Agent Library to be extracted early
+          # @param context [Contrast::Agent::RequestContext]
+          # @param potential_attack_string [String, nil]
+          # @param ia_results [Array<Contrast::Agent::Reporting::InputAnalysis>]
+          # @param **kwargs
+          # @return [Contrast::Agent::Reporting, nil]
+          def find_attacker_with_results context, potential_attack_string, ia_results, **kwargs
+            logger.trace('Checking vectors for attacks', rule: rule_name, input: potential_attack_string)
+
+            result = nil
+            ia_results.each do |ia_result|
+              if potential_attack_string
+                idx = potential_attack_string.index(ia_result.value)
+                next unless idx
+
+                database_type = kwargs[:database].to_sym
+                input_length = ia_result.value.length
+                lib_result = check_sql_input_with_agent(potential_attack_string, database_type, idx, input_length)
+
+                kwargs[:result_struct] = lib_result
+                result = build_attack_with_match(context, ia_result, result, potential_attack_string, **kwargs)
+              else
+                result = build_attack_without_match(context, ia_result, result, **kwargs)
+              end
+            end
+            result
+          end
+
+          # We'll need a second place, where we need to check the token boundaries if are being crossed and
+          # worth-watching.
+          #
+          # @param sql[String] SQL coming from parameter
+          # @param database[String] Type of database
+          # @param input_index[String] index in the sqlQuery string where user input was found
+          # @param input_length[Number] length of the input value
+          # @return [Hash, Boolean]
+          def check_sql_input_with_agent sql, database, input_index, input_length
+            return false unless (agent_lib = Contrast::AGENT_LIB) && sql && database
+
+            agent_lib.check_sql_query(input_index, input_length, database, sql)
+          end
+
+          # @param context [Contrast::Agent::RequestContext]
+          def infilter? context
+            return false unless enabled?
+            return false unless context&.agent_input_analysis&.results&.any? do |result|
+              result.rule_id == rule_name
+            end
+
+            return false if protect_excluded_by_code?
+
+            true
+          end
         end
       end
     end

data/lib/contrast/agent/protect/rule/unsafe_file_upload/unsafe_file_upload_input_classification.rb

@@ -3,9 +3,8 @@
 
 require 'contrast/utils/object_share'
 require 'contrast/agent/protect/rule/unsafe_file_upload'
-require 'contrast/agent/protect/rule/unsafe_file_upload/unsafe_file_upload_matcher'
 require 'contrast/agent/protect/input_analyzer/input_analyzer'
-require 'contrast/utils/input_classification'
+require 'contrast/utils/input_classification_base'
 
 module Contrast
   module Agent
@@ -14,67 +13,44 @@ module Contrast
         # This module will do the Input Classification stage of Unsafe File Upload
         # rule. As a result input would be marked as DEFINITEATTACK or IGNORE.
         module UnsafeFileUploadInputClassification
-          # UNSAFE_UPLOAD_MATCH = 'unsafe-file-upload-input-tracing-v1'
-          #
-          # class << self
-          #   include InputClassificationBase
-          #   include Contrast::Agent::Protect::Rule::UnsafeFileUploadMatcher
-          #
-          #   # Input Classification stage is done to determine if an user input is
-          #   # DEFINITEATTACK or to be ignored.
-          #   #
-          #   # @param input_type [Contrast::Agent::Reporting::InputType] The type of the user input.
-          #   # @param value [String, Array<String>] the value of the input.
-          #   # @param input_analysis [Contrast::Agent::Reporting::InputAnalysis] Holds all the results from the
-          #   #                                                       agent analysis from the current
-          #   #                                                       Request.
-          #   # @return ia [Contrast::Agent::Reporting::InputAnalysis] with updated results.
-          #   def classify input_type, value, input_analysis
-          #     unless Contrast::Agent::Protect::Rule::UnsafeFileUpload::APPLICABLE_USER_INPUTS.include?(input_type)
-          #       return
-          #     end
-          #     return unless input_analysis.request
-          #
-          #     rule_id = Contrast::Agent::Protect::Rule::UnsafeFileUpload::NAME
-          #     results = []
-          #
-          #     Array(value).each do |val|
-          #       Array(val).each do |v|
-          #         results << create_new_input_result(input_analysis.request, rule_id, input_type, v)
-          #       end
-          #     end
-          #
-          #     input_analysis.results = results
-          #     input_analysis
-          #   end
-          #
-          #   private
-          #
-          #   # This methods checks if input is tagged DEFINITEATTACK or IGNORE matches value with it's
-          #   # key if needed and Creates new isntance of InputAnalysisResult.
-          #   #
-          #   # @param request [Contrast::Agent::Request] the current request context.
-          #   # @param rule_id [String] The name of the Protect Rule.
-          #   # @param input_type [Contrast::Agent::Reporting::InputType] The type of the user input.
-          #   # @param value [String, Array<String>] the value of the input.
-          #   #
-          #   # @return res [Contrast::Agent::Reporting::InputAnalysisResult]
-          #   def create_new_input_result request, rule_id, input_type, value
-          #     ia_result = new_ia_result rule_id, input_type, request.path, value
-          #     if unsafe_match? value
-          #       ia_result.score_level = DEFINITEATTACK
-          #       ia_result.ids << UNSAFE_UPLOAD_MATCH
-          #     else
-          #       ia_result.score_level = IGNORE
-          #     end
-          #     ia_result.key = if input_type == MULTIPART_FIELD_NAME
-          #                       Contrast::Agent::Protect::InputAnalyzer::DISPOSITION_FILENAME
-          #                     else
-          #                       Contrast::Agent::Protect::InputAnalyzer::DISPOSITION_NAME
-          #                     end
-          #     ia_result
-          #   end
-          # end
+          UNSAFE_UPLOAD_MATCH = 'unsafe-file-upload-input-tracing-v1'
+
+          class << self
+            include InputClassificationBase
+
+            private
+
+            # This methods checks if input is tagged DEFINITEATTACK or IGNORE matches value with it's
+            # key if needed and Creates new isntance of InputAnalysisResult.
+            #
+            # @param request [Contrast::Agent::Request] the current request context.
+            # @param rule_id [String] The name of the Protect Rule.
+            # @param input_type [Contrast::Agent::Reporting::InputType] The type of the user input.
+            # @param value [String, Array<String>] the value of the input.
+            #
+            # @return res [Contrast::Agent::Reporting::InputAnalysisResult]
+            def create_new_input_result request, rule_id, input_type, value
+              return unless Contrast::AGENT_LIB
+              return unless (input_eval = Contrast::AGENT_LIB.eval_input(value,
+                                                                         Contrast::AGENT_LIB.input_set[:MULTIPART_NAME],
+                                                                         Contrast::AGENT_LIB.rule_set[rule_id],
+                                                                         Contrast::AGENT_LIB.eval_option[:NONE]))
+
+              ia_result = new_ia_result(rule_id, input_type, request.path, value)
+              if input_eval.score >= THRESHOLD
+                ia_result.score_level = DEFINITEATTACK
+                ia_result.ids << UNSAFE_UPLOAD_MATCH
+              else
+                ia_result.score_level = IGNORE
+              end
+              ia_result.key = if input_type == MULTIPART_FIELD_NAME
+                                Contrast::Agent::Protect::InputAnalyzer::DISPOSITION_FILENAME
+                              else
+                                Contrast::Agent::Protect::InputAnalyzer::DISPOSITION_NAME
+                              end
+              ia_result
+            end
+          end
         end
       end
     end

data/lib/contrast/agent/protect/rule/unsafe_file_upload.rb

@@ -11,7 +11,7 @@ module Contrast
       module Rule
         # The Ruby implementation of the Protect Unsafe File Upload rule.
         # The unsafe-file-upload rule can trigger the following results:
-        # BLOCKED in Blocking mode na SUSPICIOUS in Monitor mode.
+        # BLOCKED in Blocking mode and SUSPICIOUS in Monitor mode.
         class UnsafeFileUpload < Contrast::Agent::Protect::Rule::BaseService
           include Contrast::Agent::Reporting::InputType
 
@@ -23,40 +23,13 @@ module Contrast
             NAME
           end
 
-          # This rule is solely based on input analysis, which the Service handles. When we move from the Service to the
-          # agent with protect library, we should re-enable these tests and that rule.
-          # TODO: RUBY-1574
-          def enabled?
-            super && false
+          def applicable_user_inputs
+            APPLICABLE_USER_INPUTS
           end
 
-          # def block_message
-          #   BLOCK_MESSAGE
-          # end
-          #
-          # def prefilter context
-          #   return unless prefilter?(context)
-          #
-          #   ia_results = gather_ia_results context
-          #
-          #   ia_results.each do |ia_result|
-          #     result = build_attack_result(context)
-          #     build_attack_without_match context, ia_result, result
-          #     append_to_activity context, result
-          #
-          #     cef_logging result, :successful_attack
-          #     raise Contrast::SecurityException.new(self, BLOCK_MESSAGE) if blocked?
-          #   end
-          # end
-          #
-          # private
-          #
-          # def prefilter? _context
-          #   return false unless enabled?
-          #   return false if protect_excluded_by_code?
-          #
-          #   true
-          # end
+          def block_message
+            BLOCK_MESSAGE
+          end
         end
       end
     end

data/lib/contrast/agent/protect/rule/xss/reflected_xss_input_classification.rb

@@ -0,0 +1,58 @@
+# Copyright (c) 2022 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
+# frozen_string_literal: true
+
+require 'contrast/utils/input_classification_base'
+
+module Contrast
+  module Agent
+    module Protect
+      module Rule
+        # The Ruby implementation of the Protect Reflected XSS rule
+        # Input classification
+        module ReflectedXssInputClassification
+          REFLECTED_XSS_MATCH = 'reflected-xss-input-tracing-v1'.cs__freeze
+          WORTHWATCHING_MATCH = 'xss-worth-watching-v2'.cs__freeze
+          class << self
+            include InputClassificationBase
+
+            private
+
+            # This methods checks if input is tagged WORTHWATCHING or IGNORE matches value with it's
+            # key if needed and Creates new isntance of InputAnalysisResult.
+            #
+            # @param request [Contrast::Agent::Request] the current request context.
+            # @param rule_id [String] The name of the Protect Rule.
+            # @param input_type [Contrast::Agent::Reporting::InputType] The type of the user input.
+            # @param value [String, Array<String>] the value of the input.
+            #
+            # @return res [Contrast::Agent::Reporting::InputAnalysisResult]
+            def create_new_input_result request, rule_id, input_type, value
+              return unless Contrast::AGENT_LIB
+
+              input_eval = Contrast::AGENT_LIB.eval_input(value,
+                                                          convert_input_type(input_type),
+                                                          Contrast::AGENT_LIB.rule_set[rule_id],
+                                                          Contrast::AGENT_LIB.
+                                                            eval_option[:WORTHWATCHING])
+
+              score = input_eval&.score || 0
+              ia_result = new_ia_result(rule_id, input_type, request.path, value)
+              if score >= THRESHOLD
+                ia_result.score_level = DEFINITEATTACK
+                ia_result.ids << REFLECTED_XSS_MATCH
+              elsif score >= WORTHWATCHING_THRESHOLD
+                ia_result.score_level = WORTHWATCHING
+                ia_result.ids << WORTHWATCHING_MATCH
+              else
+                ia_result.score_level = IGNORE
+              end
+
+              add_needed_key(request, ia_result, input_type, value)
+              ia_result
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/contrast/agent/protect/rule/xss.rb

@@ -9,37 +9,31 @@ module Contrast
       module Rule
         # The Ruby implementation of the Protect Cross-Site Scripting rule.
         class Xss < Contrast::Agent::Protect::Rule::BaseService
+          include Contrast::Agent::Reporting::InputType
           NAME = 'reflected-xss'
           BLOCK_MESSAGE = 'XSS rule triggered. Response blocked.'
 
-          class << self
-            # @param attack_sample [Contrast::Api::Dtm::RaspRuleSample]
-            # @return [Hash] the details for this specific rule
-            def extract_details attack_sample
-              # TODO: RUBY-1702 - figure out why xss isn't populated when reported; probably something to do w/
-              #   suspicious
-              return Contrast::Utils::ObjectShare::EMPTY_HASH unless attack_sample&.xss
-
-              {
-                  input: attack_sample.xss.input,
-                  matches: attack_sample.xss.matches.map do |match|
-                             {
-                                 evidenceStart: match.evidence_start_ms,
-                                 evidence: match.evidence,
-                                 offset: match.offset
-                             }
-                           end
-              }
-            end
-          end
+          APPLICABLE_USER_INPUTS = [
+            BODY, PARAMETER_NAME, PARAMETER_VALUE, JSON_VALUE,
+            MULTIPART_VALUE, MULTIPART_FIELD_NAME, XML_VALUE,
+            DWR_VALUE, URI, QUERYSTRING
+          ].cs__freeze
 
           def rule_name
             NAME
           end
 
+          def block_message
+            BLOCK_MESSAGE
+          end
+
           def stream_safe?
             false
           end
+
+          def applicable_user_inputs
+            APPLICABLE_USER_INPUTS
+          end
         end
       end
     end

data/lib/contrast/agent/protect/rule/xxe.rb

@@ -22,28 +22,6 @@ module Contrast
           BLOCK_MESSAGE = 'XXE rule triggered. Response blocked.'
           EXTERNAL_ENTITY_PATTERN = /<!ENTITY\s+[a-zA-Z0-f]+\s+(?:SYSTEM|PUBLIC)\s+(.*?)>/.cs__freeze
 
-          class << self
-            # @param attack_sample [Contrast::Api::Dtm::RaspRuleSample]
-            # @return [Hash] the details for this specific rule
-            def extract_details attack_sample
-              {
-                  xml: attack_sample.xxe.xml,
-                  declaredEntities: attack_sample.xxe.declared_entities.map do |entity|
-                                      {
-                                          start: entity.start_idx,
-                                          end: entity.end_idx
-                                      }
-                                    end,
-                  entitiesResolved: attack_sample.xxe.entities_resolved.map do |entity|
-                                      {
-                                          systemId: entity.system_id,
-                                          publicId: entity.public_id
-                                      }
-                                    end
-              }
-            end
-          end
-
           def rule_name
             NAME
           end
@@ -81,7 +59,7 @@ module Contrast
           # @param xml [String] the literal value of the XML being checked for
           #   external entity resolution
           # @param _kwargs [Hash]
-          # @return [Contrast::Api::Dtm::AttackResult, nil] the determination
+          # @return [Contrast::Agent::Reporting, nil] the determination
           #   as to whether or not this XML has an XXE attack in it.
           def find_attacker context, xml, **_kwargs
             return unless xml
@@ -142,8 +120,10 @@ module Contrast
           # We know that this attack happened, so the result is always matched
           # and the level is always critical. Only variable is the XML value
           # supplied by the attacker.
+          #
+          # @return [Contrast::Agent::Reporting::InputAnalysisResult]
           def build_evaluation xml
-            ia_result = Contrast::Api::Settings::InputAnalysisResult.new
+            ia_result = Contrast::Agent::Reporting::InputAnalysisResult.new
             ia_result.rule_id = rule_name
             ia_result.input_type = :UNKNOWN
             ia_result.value = Contrast::Utils::StringUtils.protobuf_safe_string(xml)