contrast-agent 4.14.1 → 5.0.0

data/lib/contrast/agent/request_context_extend.rb

@@ -1,6 +1,8 @@
-# Copyright (c) 2021 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
+# Copyright (c) 2022 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
 # frozen_string_literal: true
 
+require 'contrast/agent/assess/rule/response/autocomplete_rule'
+
 module Contrast
   module Agent
     # This class extends RequestContexts: this class acts to encapsulate information about the currently
@@ -30,6 +32,18 @@ module Contrast
         @request.observed_route = @observed_route
       end
 
+      # Convert the discovered route for this request to appropriate forms and disseminate it to those locations
+      # where it is necessary for our route coverage and finding vulnerability discovery features to function.
+      #
+      def append_to_new_observed_route route
+        return unless route
+
+        @new_observed_route.signature  = route.route
+        @new_observed_route.verb       = route.verb
+        @new_observed_route.url        = route.url if route.url
+        @request.new_observed_route = @new_observed_route
+      end
+
       # Collect the results for the given rule with the given action
       #
       # @param rule [String] the id of the rule to which the results apply
@@ -96,6 +110,9 @@ module Contrast
       def extract_after rack_response
         @response = Contrast::Agent::Response.new(rack_response)
         activity.http_response = @response.dtm if @sample_res
+        return unless Contrast::Agent::Reporter.enabled?
+
+        Contrast::Agent::Assess::Rule::Response::Autocomplete.new.analyze(@response)
       rescue StandardError => e
         logger.error('Unable to extract information after request', e)
       end
@@ -124,6 +141,14 @@ module Contrast
         end
       end
 
+      # Generates the attack result
+      #
+      # @param rule [Contrast::Agent::Protect::Rule, Contrast::Agent::Assess::Rule]
+      # @param rule_id [String] String name of the rule
+      # @param ia_result [Contrast::Api::Settings::InputAnalysisResult] the
+      #   analysis of the input that was determined to be an attack
+      # @param results_by_rule [Hash] attack results from any evaluations on the agent settings object.
+      # @return [Contrast::Api::Dtm::AttackResult] the attack result from this input
       def attack_result rule, rule_id, ia_result, results_by_rule
         @_attack_result = if rule.mode == :BLOCK
                             # special case for rules (like reflected xss) that used to have an infilter / block mode

data/lib/contrast/agent/request_handler.rb

@@ -1,8 +1,10 @@
-# Copyright (c) 2021 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
+# Copyright (c) 2022 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
 # frozen_string_literal: true
 
 require 'contrast/components/logger'
 require 'contrast/components/scope'
+require 'contrast/agent/reporting/reporter'
+require 'contrast/agent/reporting/reporting_utilities/dtm_message'
 
 module Contrast
   module Agent
@@ -19,15 +21,49 @@ module Contrast
         @ruleset = ::Contrast::AGENT.ruleset
       end
 
-      # TODO: RUBY-1353
+      # Send Activities messages to TS [Contrast::Agent::Reporting::ServerActivity,
+      #                                 Contrast::Api::Dtm::ServerActivity,
+      #                                 Contrast::Api::Dtm::Activity,
+      #                                 Contrast::Api::Dtm::ObservedRoute]
+      # If bypass is enabled use the reporting service if not than the messages are
+      # send with speedracer
       # TODO: RUBY-1355
-      # TODO: RUBY-1357
       # TODO: RUBY-1358
+      # TODO: RUBY-1438 -- remove
       def send_activity_messages
+        events = [context.activity]
+        unless Contrast::Agent::Reporter.enabled?
+          Contrast::Agent::Inventory::DependencyUsageAnalysis.instance.generate_library_usage(context.activity)
+          events << context.server_activity
+          events << context.observed_route
+        end
+        events.each do |message|
+          Contrast::Agent.messaging_queue&.send_event_eventually(message)
+        end
+      end
+
+      # reports events[Contrast::Agent::Reporting::ReporterEvent] to TS
+      # This method is used to send our JSON messages directly to TeamServer at the end of each request. As we move
+      # more endpoints over, this method will take the messages originally sent by #send_actiivty_messages. At the end,
+      # that method should be removed.
+      # TODO: RUBY-1358
+      def report_activity
+        return unless Contrast::Agent::Reporter.enabled?
+
+        reporter = Contrast::Agent.reporter
+        return unless reporter
+
         Contrast::Agent::Inventory::DependencyUsageAnalysis.instance.generate_library_usage(context.activity)
-        [context.server_activity, context.activity, context.observed_route].each do |message|
-          Contrast::Agent.messaging_queue.send_event_eventually(message)
+        [
+          context.new_observed_route,
+          Contrast::Agent::Reporting::DtmMessage.dtm_to_event(context.server_activity),
+          Contrast::Agent::Reporting::DtmMessage.dtm_to_event(context.activity.library_usages)
+        ].each do |event|
+          reporter.send_event(event)
+        rescue StandardError => e
+          logger.warn('Unable to send Event Activity', e)
         end
+        context.activity.library_usages.clear # TODO: RUBY-1355 remove when no longer using activity
       end
 
       # If the response is streaming, we should only perform filtering on our stream safe rules

data/lib/contrast/agent/response.rb

@@ -1,4 +1,4 @@
-# Copyright (c) 2021 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
+# Copyright (c) 2022 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
 # frozen_string_literal: true
 
 require 'resolv'
@@ -29,6 +29,9 @@ module Contrast
         @is_array = !rack_response.is_a?(Rack::Response)
       end
 
+      # The document type of the response, based on its Content-Type. Can be one of :JSON, :NORMAL, :XML, or nothing.
+      #
+      # @return [Symbol<:XML, :JSON, :NORMAL>, nil]
       def document_type
         case content_type
         when /xml/i
@@ -40,7 +43,10 @@ module Contrast
         end
       end
 
-      # B/c the response can change, we can't memoize this :(
+      # A form of the Rack::Response which we can report to the Service to be sent to TeamServer for processing. B/c
+      # the response can change, we can't memoize this :(
+      #
+      # @return [Contrast::Api::Dtm::HttpResponse]
       def dtm
         context_response = Contrast::Api::Dtm::HttpResponse.new
         context_response.response_code = response_code.to_i
@@ -55,35 +61,37 @@ module Contrast
         context_response
       end
 
+      # The response code of this response
+      #
+      # @return [Integer]
       def response_code
         return unless rack_response
 
         @is_array ? rack_response[0].to_i : rack_response.status
       end
 
+      # The headers of this response
+      #
+      # @return [Hash, nil]
       def headers
         return unless rack_response
 
-        if @is_array
-          rack_response[1]
-        else
-          rack_response.headers
-        end
+        @is_array ? rack_response[1] : rack_response.headers
       end
 
+      # The Content-Type of this response, as set in the headers hash under the key Rack::CONTENT_TYPE
+      #
+      # @return [String, nil]
       def content_type
         return unless rack_response
 
-        if @is_array
-          headers[Rack::CONTENT_TYPE]
-        else
-          rack_response.content_type
-        end
+        @is_array ? headers[Rack::CONTENT_TYPE] : rack_response.content_type
       end
 
-      # The response body can change during the request lifecycle
-      # We should not extract it out as a variable here, or we'll miss those
-      # changes.
+      # The response body can change during the request lifecycle, so we have to look it up every time we need it.
+      # We should not extract it out as a variable here, or we'll miss those changes.
+      #
+      # @return [String, nil]
       def body
         return unless rack_response
 

data/lib/contrast/agent/rule_set.rb

@@ -1,4 +1,4 @@
-# Copyright (c) 2021 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
+# Copyright (c) 2022 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
 # frozen_string_literal: true
 
 require 'contrast/components/logger'

data/lib/contrast/agent/scope.rb

@@ -1,4 +1,4 @@
-# Copyright (c) 2021 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
+# Copyright (c) 2022 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
 # frozen_string_literal: true
 
 module Contrast

data/lib/contrast/agent/service_heartbeat.rb

@@ -1,4 +1,4 @@
-# Copyright (c) 2021 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
+# Copyright (c) 2022 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
 # frozen_string_literal: true
 
 require 'contrast/components/logger'

data/lib/contrast/agent/startup_metrics_telemetry_event.rb

@@ -1,4 +1,4 @@
-# Copyright (c) 2021 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
+# Copyright (c) 2022 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
 # frozen_string_literal: true
 
 require 'contrast/utils/metrics_hash'
@@ -44,9 +44,14 @@ module Contrast
       FALLBACK_TENANTS = [SAAS_CUSTOM, SAAS_CUSTOM_JP].cs__freeze
       # Fallback for Custom, most likely self-hosted, Environments
       EOP = 'EOP'
+      REJECTED_VALUES = [:EMPTY_VALUE, 'NEEDS_TO_BE_SET', Contrast::Utils::ObjectShare::EMPTY_STRING].cs__freeze
 
       def initialize
         super
+        @settings = []
+        add_config_keys ::Contrast::CONFIG.root, 'root'
+        @settings << ENV.keys.select { |v| v.starts_with?('CONTRAST') }
+        @settings.flatten
         add_tags
       end
 
@@ -55,14 +60,25 @@ module Contrast
       end
 
       def add_tags
-        @tags['teamserver'] = teamserver_type
-        @tags['agent_version'] = VERSION
-        @tags['ruby_version'] = RUBY_VERSION
-        @tags['os_type'] = sys_info['os_type'] == 'Darwin' ? 'MacOS' : 'Linux'
-        @tags['os_arch'] = sys_info['os_arch']
-        @tags['os_version'] = sys_info['os_version']
+        add_system_tags
         @tags['app_framework_and_version'] = APP_AND_SERVER_DATA[:application_info].to_s
         @tags['server_framework_and_version'] = APP_AND_SERVER_DATA[:server_info].to_s
+        @tags['ASSESS'] = Contrast::ASSESS.enabled?.to_s
+        @tags['PROTECT'] = Contrast::PROTECT.enabled?.to_s
+        @tags['settings'] = @settings.join(',')
+      end
+
+      def add_config_keys config, nested_key
+        config.configuration_map.reject! { |_k, v| REJECTED_VALUES.include?(v) }
+
+        config.configuration_map.each do |k, v|
+          unless v.cs__class <= Contrast::Config::BaseConfiguration
+            @settings << "#{ nested_key }.#{ k }"
+            next
+          end
+
+          add_config_keys v, "#{ nested_key }.#{ k }"
+        end
       end
 
       def sys_info
@@ -89,6 +105,17 @@ module Contrast
           end
         end
       end
+
+      # Here we attach the key-value pairs of the system
+      #
+      def add_system_tags
+        @tags['teamserver'] = teamserver_type
+        @tags['agent_version'] = VERSION
+        @tags['ruby_version'] = RUBY_VERSION
+        @tags['os_type'] = sys_info['os_type'] == 'Darwin' ? 'MacOS' : 'Linux'
+        @tags['os_arch'] = sys_info['os_arch']
+        @tags['os_version'] = sys_info['os_version']
+      end
     end
   end
 end

data/lib/contrast/agent/static_analysis.rb

@@ -1,8 +1,9 @@
-# Copyright (c) 2021 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
+# Copyright (c) 2022 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
 # frozen_string_literal: true
 
 require 'contrast/components/logger'
 require 'contrast/components/scope'
+require 'contrast/agent/reporting/reporting_events/application_update'
 require 'contrast/api/decorators/application_update'
 
 module Contrast
@@ -23,7 +24,6 @@ module Contrast
           end
         end
 
-        # TODO: RUBY-1354
         # TODO: RUBY-1356
         def send_inventory_message
           return unless ::Contrast::INVENTORY.enabled?
@@ -31,7 +31,17 @@ module Contrast
           app_update_msg = Contrast::Api::Dtm::ApplicationUpdate.build
 
           Contrast::Agent::Inventory::DatabaseConfig.append_db_config(app_update_msg)
-          Contrast::Agent.messaging_queue.send_event_eventually(app_update_msg)
+          # TODO: RUBY-1438 -- remove and build ReportingEvent directly
+          if Contrast::Agent.reporter
+            report = Contrast::Agent::Reporting::DtmMessage.dtm_to_event(app_update_msg)
+            Contrast::Agent.reporter.send_event(report)
+
+            # This is being reported separately because we extract the data from the same message
+            inventory_report = Contrast::Agent::Reporting::ApplicationInventory.convert(app_update_msg)
+            Contrast::Agent.reporter.send_event(inventory_report)
+          else
+            Contrast::Agent.messaging_queue.send_event_eventually(app_update_msg)
+          end
         end
 
         private

data/lib/contrast/agent/telemetry.rb

@@ -1,4 +1,4 @@
-# Copyright (c) 2021 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
+# Copyright (c) 2022 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
 # frozen_string_literal: true
 
 require 'contrast/config/env_variables'

data/lib/contrast/agent/telemetry_event.rb

@@ -1,4 +1,4 @@
-# Copyright (c) 2021 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
+# Copyright (c) 2022 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
 # frozen_string_literal: true
 
 require 'contrast/utils/metrics_hash'

data/lib/contrast/agent/thread.rb

@@ -1,4 +1,4 @@
-# Copyright (c) 2021 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
+# Copyright (c) 2022 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
 # frozen_string_literal: true
 
 require 'contrast/components/scope'

data/lib/contrast/agent/thread_watcher.rb

@@ -1,4 +1,4 @@
-# Copyright (c) 2021 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
+# Copyright (c) 2022 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
 # frozen_string_literal: true
 
 require 'contrast/components/logger'
@@ -59,7 +59,7 @@ module Contrast
         messaging_queue.stop!
         heapdump_util.stop!
         telemetry_queue&.stop!
-        reporter_queue&.stop!
+        reporter&.stop!
       end
 
       def heartbeat_thread_init
@@ -108,7 +108,7 @@ module Contrast
       end
 
       # @return [Contrast::Agent::Reporter]
-      def reporter_queue
+      def reporter
         return if @reporter.nil?
 
         @reporter

data/lib/contrast/agent/tracepoint_hook.rb

@@ -1,4 +1,4 @@
-# Copyright (c) 2021 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
+# Copyright (c) 2022 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
 # frozen_string_literal: true
 
 require 'contrast/components/logger'
@@ -46,9 +46,6 @@ module Contrast
             Contrast::Agent.framework_manager.register_late_framework(loaded_module)
             Contrast::Agent::Inventory::DependencyUsageAnalysis.instance.associate_file(path) if path
             Contrast::Agent::Patching::Policy::Patcher.patch_specific_module(loaded_module)
-            if RUBY_VERSION < '2.6.0' # TODO: RUBY-714 remove guard w/ EOL of 2.5
-              Contrast::Agent::Assess::Policy::RewriterPatch.rewrite_interpolation(loaded_module)
-            end
             Contrast::Agent::Assess::Policy::PolicyScanner.scan(tracepoint_event)
           rescue StandardError => e
             logger.error('Unable to complete TracePoint analysis', e, module: loaded_module, path: path)

data/lib/contrast/agent/version.rb

@@ -1,8 +1,8 @@
-# Copyright (c) 2021 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
+# Copyright (c) 2022 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
 # frozen_string_literal: true
 
 module Contrast
   module Agent
-    VERSION = '4.14.1'
+    VERSION = '5.0.0'
   end
 end

data/lib/contrast/agent/worker_thread.rb

@@ -1,4 +1,4 @@
-# Copyright (c) 2021 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
+# Copyright (c) 2022 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
 # frozen_string_literal: true
 
 module Contrast

data/lib/contrast/agent.rb

@@ -1,4 +1,4 @@
-# Copyright (c) 2021 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
+# Copyright (c) 2022 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
 # frozen_string_literal: true
 
 require 'English'
@@ -76,10 +76,10 @@ module Contrast
     end
 
     # @return [nil, Contrast::Agent::Reporter]
-    def self.reporter_queue
-      return unless  thread_watcher.reporter_queue
+    def self.reporter
+      return unless  thread_watcher.reporter
 
-      thread_watcher.reporter_queue
+      thread_watcher.reporter
     end
 
     def self.thread_watcher

data/lib/contrast/api/communication/connection_status.rb

@@ -1,4 +1,4 @@
-# Copyright (c) 2021 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
+# Copyright (c) 2022 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
 # frozen_string_literal: true
 
 module Contrast

data/lib/contrast/api/communication/messaging_queue.rb

@@ -1,4 +1,4 @@
-# Copyright (c) 2021 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
+# Copyright (c) 2022 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
 # frozen_string_literal: true
 
 require 'contrast/components/logger'
@@ -16,7 +16,7 @@ module Contrast
 
         def initialize
           @speedracer = Contrast::Api::Communication::Speedracer.new
-          @audit = Contrast::Agent::Reporting::Audit.new if ::Contrast::API.request_audit_enable
+          @audit = Contrast::Agent::Reporting::Audit.new if ::Contrast::API.request_audit_enable?
           super
         end
 
@@ -34,7 +34,7 @@ module Contrast
             return
           end
           response_data = speedracer.return_response(event)
-          return response_data unless ::Contrast::API.request_audit_enable
+          return response_data unless ::Contrast::API.request_audit_enable?
 
           @audit&.audit_event(event, response_data)
           response_data
@@ -63,7 +63,7 @@ module Contrast
           end
           logger.debug('Enqueued event for sending', event_type: event.cs__class)
           queue << event if event
-          return unless ::Contrast::API.request_audit_enable
+          return unless ::Contrast::API.request_audit_enable?
 
           @audit&.audit_event(event)
         end

data/lib/contrast/api/communication/response_processor.rb

@@ -1,4 +1,4 @@
-# Copyright (c) 2021 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
+# Copyright (c) 2022 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
 # frozen_string_literal: true
 
 require 'contrast/agent/reaction_processor'
@@ -8,8 +8,12 @@ module Contrast
   module Api
     module Communication
       # Handles processing deferred messages sent to SpeedRacer.
+      #
+      # @attr_reader last_app_update_ms [Integer] the time, in ms, that application settings last changed
+      # @attr_reader last_server_update_ms [Integer] the time, in ms, that server settings last changed
       class ResponseProcessor
         include Contrast::Components::Logger::InstanceMethods
+        attr_reader :last_app_update_ms, :last_server_update_ms
 
         # Use the given response to update the Agent's server features and application settings, allowing it to reflect
         # the latest options configured by the user in TeamServer
@@ -44,7 +48,7 @@ module Contrast
           logger.trace('Agent: Received updated server features')
 
           ::Contrast::SETTINGS.update_from_server_features(server_features)
-
+          @last_server_update_ms = Contrast::Utils::Timer.now_ms
           server_features
         end
 
@@ -60,15 +64,15 @@ module Contrast
           logger.debug('Agent: Received updated application settings')
 
           ::Contrast::SETTINGS.update_from_application_settings(app_settings)
-
+          @last_app_update_ms = Contrast::Utils::Timer.now_ms
           app_settings
         end
 
         # This can't go in the Settings component because protect and assess depend on settings
         # I don't think it should go into contrast_service because that only handles connection specific data.
         #
-        # @param server_features [Contrast::Api::Settings::AgentSettings]
-        # @param app_settings [Contrast::Api::Settings::ApplicationSettings]
+        # @param server_features [Contrast::Api::Settings::AgentSettings, Array]
+        # @param app_settings [Contrast::Api::Settings::ApplicationSettings, Array]
         def update_features server_features, app_settings
           return unless !!(server_features || app_settings)
           return unless ::Contrast::AGENT.enabled?

data/lib/contrast/api/communication/service_lifecycle.rb

@@ -1,4 +1,4 @@
-# Copyright (c) 2021 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
+# Copyright (c) 2022 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
 # frozen_string_literal: true
 
 require 'contrast/components/logger'

data/lib/contrast/api/communication/socket.rb

@@ -1,4 +1,4 @@
-# Copyright (c) 2021 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
+# Copyright (c) 2022 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
 # frozen_string_literal: true
 
 module Contrast

data/lib/contrast/api/communication/socket_client.rb

@@ -1,4 +1,4 @@
-# Copyright (c) 2021 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
+# Copyright (c) 2022 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
 # frozen_string_literal: true
 
 require 'socket'
@@ -105,6 +105,9 @@ module Contrast
           to_service = Contrast::Api::Dtm::Message.encode(msg)
           from_service = send_marshaled(to_service)
           response = Contrast::Api::Settings::AgentSettings.decode(from_service)
+          logger.debug('RESPONSE FROM TS')
+          logger.debug(from_service)
+          logger.debug('RESPONSE FROM TS')
           logger.debug('Received response.', msg_id: msg.__id__, p_id: msg.pid, msg_count: msg.message_count,
                                              response_id: response&.__id__)
           response

data/lib/contrast/api/communication/speedracer.rb

@@ -1,4 +1,4 @@
-# Copyright (c) 2021 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
+# Copyright (c) 2022 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
 # frozen_string_literal: true
 
 require 'contrast/components/logger'

data/lib/contrast/api/communication/tcp_socket.rb

@@ -1,4 +1,4 @@
-# Copyright (c) 2021 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
+# Copyright (c) 2022 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
 # frozen_string_literal: true
 
 require 'contrast/api/communication/socket'

data/lib/contrast/api/communication/unix_socket.rb

@@ -1,4 +1,4 @@
-# Copyright (c) 2021 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
+# Copyright (c) 2022 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
 # frozen_string_literal: true
 
 require 'contrast/api/communication/socket'

data/lib/contrast/api/communication.rb

@@ -1,4 +1,4 @@
-# Copyright (c) 2021 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
+# Copyright (c) 2022 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
 # frozen_string_literal: true
 
 module Contrast

data/lib/contrast/api/decorators/address.rb

@@ -1,4 +1,4 @@
-# Copyright (c) 2021 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
+# Copyright (c) 2022 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
 # frozen_string_literal: true
 
 require 'contrast/api/dtm.pb'

data/lib/contrast/api/decorators/agent_startup.rb

@@ -1,4 +1,4 @@
-# Copyright (c) 2021 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
+# Copyright (c) 2022 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
 # frozen_string_literal: true
 
 require 'contrast/api/dtm.pb'

data/lib/contrast/api/decorators/application_settings.rb

@@ -1,4 +1,4 @@
-# Copyright (c) 2021 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
+# Copyright (c) 2022 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
 # frozen_string_literal: true
 
 require 'contrast/utils/string_utils'

data/lib/contrast/api/decorators/application_startup.rb

@@ -1,4 +1,4 @@
-# Copyright (c) 2021 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
+# Copyright (c) 2022 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
 # frozen_string_literal: true
 
 require 'contrast/api/dtm.pb'

data/lib/contrast/api/decorators/application_update.rb

@@ -1,4 +1,4 @@
-# Copyright (c) 2021 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
+# Copyright (c) 2022 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
 # frozen_string_literal: true
 
 require 'contrast/utils/string_utils'

data/lib/contrast/api/decorators/architecture_component.rb

@@ -0,0 +1,36 @@
+# Copyright (c) 2022 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
+# frozen_string_literal: true
+
+require 'contrast/utils/string_utils'
+
+module Contrast
+  module Api
+    module Decorators
+      # Used to decorate the {Contrast::Api::Dtm::ArchitectureComponent} protobuf model so it can own some of the data
+      # massaging required for AppUpdate and Activity dtms.
+      module ArchitectureComponent
+        def self.included klass
+          klass.extend(ClassMethods)
+        end
+
+        # Used to add class methods to the ApplicationUpdate class on inclusion of the decorator
+        module ClassMethods
+          # TeamServer only accepts certain values for ArchitectureComponents.
+          # DO NOT CHANGE THIS!
+          AC_TYPE_DB = 'db'
+
+          # Create a Contrast::Api::Dtm::ArchitectureComponent of the database type.
+          #
+          # @return [Contrast::Api::Dtm::ArchitectureComponent]
+          def build_database
+            msg = Contrast::Api::Dtm::ArchitectureComponent.new
+            msg.type = AC_TYPE_DB
+            msg
+          end
+        end
+      end
+    end
+  end
+end
+
+Contrast::Api::Dtm::ArchitectureComponent.include(Contrast::Api::Decorators::ArchitectureComponent)