newrelic_security 0.1.0

data/lib/newrelic_security/instrumentation-security/ethon/instrumentation.rb

@@ -0,0 +1,122 @@
+require_relative 'prepend'
+require_relative 'chain'
+require 'uri'
+
+module NewRelic::Security
+  module Instrumentation
+    module Ethon
+      module Easy
+
+        def fabricate_on_enter(url, action_name, options)
+          NewRelic::Security::Agent.logger.debug "OnEnter : #{self.class}.#{__method__}"
+          NewRelic::Security::Agent::Control::HTTPContext.get_context.cache[self.object_id] = { :method => action_name } if NewRelic::Security::Agent::Control::HTTPContext.get_context
+          NewRelic::Security::Agent::Control::HTTPContext.get_context.cache[self.object_id][:body] = options[:body] if NewRelic::Security::Agent::Control::HTTPContext.get_context
+        rescue => exception
+          NewRelic::Security::Agent.logger.error "Exception in hook in #{self.class}.#{__method__}, #{exception.inspect}, #{exception.backtrace}"
+        ensure
+          yield
+        end
+
+        def headers_equals_on_enter(headers)
+          NewRelic::Security::Agent.logger.debug "OnEnter : #{self.class}.#{__method__}"
+          NewRelic::Security::Agent::Control::HTTPContext.get_context.cache[self.object_id][:headers] = headers if NewRelic::Security::Agent::Control::HTTPContext.get_context && NewRelic::Security::Agent::Control::HTTPContext.get_context.cache[self.object_id]
+        rescue => exception
+          NewRelic::Security::Agent.logger.error "Exception in hook in #{self.class}.#{__method__}, #{exception.inspect}, #{exception.backtrace}"
+        ensure
+          yield
+        end
+
+        def perform_on_enter(*args)
+          event = nil
+          NewRelic::Security::Agent.logger.debug "OnEnter : #{self.class}.#{__method__}"
+          context = NewRelic::Security::Agent::Control::HTTPContext.get_context.cache[self.object_id] if NewRelic::Security::Agent::Control::HTTPContext.get_context
+          uri = ::URI.parse(url)
+          ob = {}
+          ob[:Method] = context[:method] if context
+          ob[:scheme] = uri.scheme
+          ob[:host] = uri.host
+          ob[:port] = uri.port
+          ob[:URI] = uri.to_s
+          ob[:path] = uri.path
+          ob[:query] = uri.query
+          ob[:Body] = context[:body] if context
+          ob[:Headers] = context[:headers] if context
+          ob.each { |_, value| value.dup.force_encoding(ISO_8859_1).encode(UTF_8) if value.is_a?(String) }
+          event = NewRelic::Security::Agent::Control::Collector.collect(HTTP_REQUEST, [ob])
+          headers_copy = {}
+          headers_copy.merge!(context[:headers]) if context&.key?(:headers)
+          NewRelic::Security::Instrumentation::InstrumentationUtils.add_tracing_data(headers_copy, event) if event
+          self.headers = headers_copy if self.headers 
+          event
+        rescue => exception
+          NewRelic::Security::Agent.logger.error "Exception in hook in #{self.class}.#{__method__}, #{exception.inspect}, #{exception.backtrace}"
+        ensure
+          yield
+          return event
+        end
+
+        def perform_on_exit(event)
+          NewRelic::Security::Agent.logger.debug "OnExit :  #{self.class}.#{__method__}"
+          NewRelic::Security::Agent::Utils.create_exit_event(event)
+        rescue => exception
+          NewRelic::Security::Agent.logger.error "Exception in hook in #{self.class}.#{__method__}, #{exception.inspect}, #{exception.backtrace}"
+        ensure
+          yield
+        end
+      end
+
+      module Multi
+
+        def perform_on_enter(*args)
+          event = nil
+          NewRelic::Security::Agent.logger.debug "OnEnter : #{self.class}.#{__method__}"
+          ic_args = []
+          easy_handles.each do |easy|
+            context = NewRelic::Security::Agent::Control::HTTPContext.get_context.cache[easy.object_id] if NewRelic::Security::Agent::Control::HTTPContext.get_context
+            uri = NewRelic::Security::Instrumentation::InstrumentationUtils.parse_uri(easy.url)
+            if uri
+              ob = {}
+              ob[:Method] = context[:method] if context
+              ob[:scheme] = uri.scheme
+              ob[:host] = uri.host
+              ob[:port] = uri.port
+              ob[:URI] = easy.url.to_s
+              ob[:path] = uri.path
+              ob[:query] = uri.query
+              ob[:Body] = context[:body] if context
+              ob[:Headers] = context[:headers] if context
+              ob.each { |_, value| value.dup.force_encoding(ISO_8859_1).encode(UTF_8) if value.is_a?(String) }
+              ic_args << ob
+            end
+          end
+          event = NewRelic::Security::Agent::Control::Collector.collect(HTTP_REQUEST, ic_args) unless ic_args.empty?
+          easy_handles.each do |easy|
+            context = NewRelic::Security::Agent::Control::HTTPContext.get_context.cache[easy.object_id] if NewRelic::Security::Agent::Control::HTTPContext.get_context
+            headers_copy = {}
+            headers_copy.merge!(context[:headers]) if context&.key?(:headers)
+            NewRelic::Security::Instrumentation::InstrumentationUtils.add_tracing_data(headers_copy, event) if event
+            easy.headers = headers_copy
+          end
+          event
+        rescue => exception
+          NewRelic::Security::Agent.logger.error "Exception in hook in #{self.class}.#{__method__}, #{exception.inspect}, #{exception.backtrace}"
+        ensure
+          yield
+          return event
+        end
+
+        def perform_on_exit(event)
+          NewRelic::Security::Agent.logger.debug "OnExit :  #{self.class}.#{__method__}"
+          NewRelic::Security::Agent::Utils.create_exit_event(event)
+        rescue => exception
+          NewRelic::Security::Agent.logger.error "Exception in hook in #{self.class}.#{__method__}, #{exception.inspect}, #{exception.backtrace}"
+        ensure
+          yield
+        end
+      end
+    end
+  end
+end
+
+NewRelic::Security::Instrumentation::InstrumentationLoader.install_instrumentation(:ethon, ::Ethon::Easy, ::NewRelic::Security::Instrumentation::Ethon::Easy)
+NewRelic::Security::Instrumentation::InstrumentationLoader.install_instrumentation(:ethon, ::Ethon::Multi, ::NewRelic::Security::Instrumentation::Ethon::Multi)

data/lib/newrelic_security/instrumentation-security/ethon/prepend.rb

@@ -0,0 +1,39 @@
+module NewRelic::Security
+  module Instrumentation
+    module Ethon
+      module Easy
+        module Prepend
+          include NewRelic::Security::Instrumentation::Ethon::Easy
+
+          def fabricate(url, action_name, options)
+            fabricate_on_enter(url, action_name, options) { return super }
+          end
+  
+          def headers=(headers)
+            headers_equals_on_enter(headers) { return super }
+          end
+
+          def perform(*args)
+            retval = nil
+            event = perform_on_enter(*args) { retval = super }
+            perform_on_exit(event) { return retval }
+          end
+
+        end
+      end
+
+      module Multi
+        module Prepend
+          include NewRelic::Security::Instrumentation::Ethon::Multi
+
+          def perform(*args)
+            retval = nil
+            event = perform_on_enter(*args) { retval = super }
+            perform_on_exit(event) { return retval }
+          end
+
+        end
+      end
+    end
+  end
+end

data/lib/newrelic_security/instrumentation-security/excon/chain.rb

@@ -0,0 +1,23 @@
+module NewRelic::Security
+  module Instrumentation
+    module Excon
+      module Connection
+        module Chain
+          def self.instrument!
+            ::Excon::Connection.class_eval do
+              include NewRelic::Security::Instrumentation::Excon::Connection
+
+              alias_method :request_without_security, :request
+    
+              def request(params={}, &block)
+                retval = nil
+                event = request_on_enter(params) { retval = request_without_security(params, &block) }
+                request_on_exit(event) { return retval }
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/newrelic_security/instrumentation-security/excon/instrumentation.rb

@@ -0,0 +1,44 @@
+require_relative 'prepend'
+require_relative 'chain'
+
+module NewRelic::Security
+  module Instrumentation
+    module Excon::Connection
+
+      def request_on_enter(params)
+        event = nil
+        NewRelic::Security::Agent.logger.debug "OnEnter : #{self.class}.#{__method__}"
+        ob = {}
+        ob[:Method] = params[:method]
+        ob[:scheme]  = self.data[:scheme]
+        ob[:host]    = self.data[:host]
+        ob[:port]    = self.data[:port]
+        ob[:URI]     = self.data[:query].nil? ? "#{self.data[:host]}#{self.data[:path]}" : "#{self.data[:host]}#{self.data[:path]}?#{self.data[:query]}"
+        ob[:path]    = self.data[:path]
+        ob[:query]   = self.data[:query]
+        ob[:Body] = self.data[:body]
+        ob[:Headers] = self.data[:headers]
+        ob.each { |_, value| value.dup.force_encoding(ISO_8859_1).encode(UTF_8) if value.is_a?(String) }
+        event = NewRelic::Security::Agent::Control::Collector.collect(HTTP_REQUEST, [ob])
+        NewRelic::Security::Instrumentation::InstrumentationUtils.add_tracing_data(self.data[:headers], event) if event
+        event
+      rescue => exception
+        NewRelic::Security::Agent.logger.error "Exception in hook in #{self.class}.#{__method__}, #{exception.inspect}, #{exception.backtrace}"
+      ensure
+        yield
+        return event
+      end
+
+      def request_on_exit(event)
+        NewRelic::Security::Agent.logger.debug "OnExit :  #{self.class}.#{__method__}"
+        NewRelic::Security::Agent::Utils.create_exit_event(event)
+      rescue => exception
+        NewRelic::Security::Agent.logger.error "Exception in hook in #{self.class}.#{__method__}, #{exception.inspect}, #{exception.backtrace}"
+      ensure
+        yield
+      end
+    end
+  end
+end
+
+NewRelic::Security::Instrumentation::InstrumentationLoader.install_instrumentation(:excon, ::Excon::Connection, ::NewRelic::Security::Instrumentation::Excon::Connection)

data/lib/newrelic_security/instrumentation-security/excon/prepend.rb

@@ -0,0 +1,17 @@
+module NewRelic::Security
+  module Instrumentation
+    module Excon
+      module Connection
+        module Prepend
+          include NewRelic::Security::Instrumentation::Excon::Connection
+  
+          def request(params={}, &block)
+            retval = nil
+            event = request_on_enter(params) { retval = super }
+            request_on_exit(event) { return retval }
+          end
+        end
+      end
+    end
+  end
+end

data/lib/newrelic_security/instrumentation-security/file/chain.rb

@@ -0,0 +1,34 @@
+module NewRelic::Security
+  module Instrumentation
+    module File
+      module Chain
+
+        def self.instrument!
+          ::File.class_eval do
+            class << self
+              include ::NewRelic::Security::Instrumentation::File
+
+              alias_method :delete_without_security, :delete
+
+              def delete(*var)
+                retval = nil
+                event = delete_on_enter(*var) { retval = delete_without_security(*var) }
+                delete_on_exit(event, retval) { return retval }
+              end
+              
+              alias_method :unlink_without_security, :unlink
+
+              def unlink(*var)
+                retval = nil
+                event = unlink_on_enter(*var) { retval = unlink_without_security(*var) }
+                unlink_on_exit(event, retval) { return retval }
+              end
+
+            end
+          end
+        end
+        
+      end
+    end
+  end
+end

data/lib/newrelic_security/instrumentation-security/file/instrumentation.rb

@@ -0,0 +1,62 @@
+require_relative 'prepend'
+require_relative 'chain'
+
+module NewRelic::Security
+  module Instrumentation
+    module File
+
+      def delete_on_enter(*var)
+        event = nil
+        NewRelic::Security::Agent.logger.debug "OnEnter : #{self.class}.#{__method__}"
+        if NewRelic::Security::Instrumentation::InstrumentationUtils.notify_app_integrity_delete?(var)
+          event = NewRelic::Security::Agent::Control::Collector.collect(FILE_INTEGRITY, var, DELETE)
+        else
+          # TODO: Add handling read_filter for outside app root directory files, example ../abc.txt
+          NewRelic::Security::Agent.logger.debug "Filtered because File name exist in filtered list #{self.class}.#{__method__} Args:: #{var}"
+        end
+      rescue => exception
+        NewRelic::Security::Agent.logger.error "Exception in hook in #{self.class}.#{__method__}, #{exception.inspect}, #{exception.backtrace}"
+      ensure
+        yield
+        return event
+      end
+      
+      def delete_on_exit(event, retval)
+        NewRelic::Security::Agent.logger.debug "OnExit :  #{self.class}.#{__method__}"
+        NewRelic::Security::Agent::Utils.create_exit_event(event) if retval.is_a?(Integer)
+      rescue => exception
+        NewRelic::Security::Agent.logger.error "Exception in hook in #{self.class}.#{__method__}, #{exception.inspect}, #{exception.backtrace}"
+      ensure
+        yield
+      end
+
+      def unlink_on_enter(*var)
+        event = nil
+        NewRelic::Security::Agent.logger.debug "OnEnter : #{self.class}.#{__method__}"
+        if NewRelic::Security::Instrumentation::InstrumentationUtils.notify_app_integrity_delete?(var)
+          event = NewRelic::Security::Agent::Control::Collector.collect(FILE_INTEGRITY, var, DELETE)
+        else
+          # TODO: Add handling read_filter for outside app root directory files, example ../abc.txt
+          NewRelic::Security::Agent.logger.debug "Filtered because File name exist in filtered list #{self.class}.#{__method__} Args:: #{var}"
+        end
+      rescue => exception
+        NewRelic::Security::Agent.logger.error "Exception in hook in #{self.class}.#{__method__}, #{exception.inspect}, #{exception.backtrace}"
+      ensure
+        yield
+        return event
+      end
+      
+      def unlink_on_exit(event, retval)
+        NewRelic::Security::Agent.logger.debug "OnExit :  #{self.class}.#{__method__}"
+        NewRelic::Security::Agent::Utils.create_exit_event(event) if retval.is_a?(Integer)
+      rescue => exception
+        NewRelic::Security::Agent.logger.error "Exception in hook in #{self.class}.#{__method__}, #{exception.inspect}, #{exception.backtrace}"
+      ensure
+        yield
+      end
+
+    end
+  end
+end
+
+NewRelic::Security::Instrumentation::InstrumentationLoader.install_instrumentation(:file, ::File.singleton_class, ::NewRelic::Security::Instrumentation::File)

data/lib/newrelic_security/instrumentation-security/file/prepend.rb

@@ -0,0 +1,22 @@
+module NewRelic::Security
+  module Instrumentation
+    module File
+      module Prepend
+        include ::NewRelic::Security::Instrumentation::File
+
+        def delete(*var)
+          retval = nil
+          event = delete_on_enter(*var) { retval = super }
+          delete_on_exit(event, retval) { return retval }
+        end
+
+        def unlink(*var)
+          retval = nil
+          event = unlink_on_enter(*var) { retval = super }
+          unlink_on_exit(event, retval) { return retval }
+        end
+
+      end
+    end
+  end
+end

data/lib/newrelic_security/instrumentation-security/grape/chain.rb

@@ -0,0 +1,42 @@
+module NewRelic::Security
+  module Instrumentation
+    module Grape
+      module API
+        module Instance
+          module Chain
+            def self.instrument!
+              ::Grape::API::Instance.class_eval do 
+                include NewRelic::Security::Instrumentation::Grape::API::Instance
+                
+                alias_method :call_without_security, :call
+                
+                def call(env)
+                  retval = nil
+                  event = call_on_enter(env) { retval = call_without_security(env) }
+                  call_on_exit(event, retval) { return retval }
+                end
+
+              end
+            end
+          end
+        end
+      end
+
+      module Router
+        module Chain
+          def self.instrument!
+            ::Grape::Router.class_eval do
+              include NewRelic::Security::Instrumentation::Grape::Router
+
+              alias_method :prepare_env_from_route_without_security, :prepare_env_from_route
+
+              def prepare_env_from_route(env, route)
+                prepare_env_from_route_on_enter(route) { prepare_env_from_route_without_security(env, route) }
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/newrelic_security/instrumentation-security/grape/instrumentation.rb

@@ -0,0 +1,56 @@
+require_relative 'prepend'
+require_relative 'chain'
+
+module NewRelic::Security
+  module Instrumentation
+    module Grape::API::Instance
+
+      def call_on_enter(env)
+        event = nil
+        NewRelic::Security::Agent.logger.debug "OnEnter : #{self.class}.#{__method__}"
+        NewRelic::Security::Agent.config.update_port = NewRelic::Security::Agent::Utils.app_port(env) unless NewRelic::Security::Agent.config[:listen_port]
+        NewRelic::Security::Agent::Utils.get_app_routes(:grape) if NewRelic::Security::Agent.agent.route_map.empty?
+        NewRelic::Security::Agent::Control::HTTPContext.set_context(env)
+        NewRelic::Security::Agent::Utils.parse_fuzz_header(NewRelic::Security::Agent::Control::HTTPContext.get_context)
+      rescue => exception
+        NewRelic::Security::Agent.logger.error "Exception in hook in #{self.class}.#{__method__}, #{exception.inspect}, #{exception.backtrace}"
+      ensure
+        yield
+        return event
+      end
+      
+      def call_on_exit(event, retval)
+        NewRelic::Security::Agent.logger.debug "OnExit :  #{self.class}.#{__method__}"
+        # NewRelic::Security::Agent.logger.debug "\n\nHTTP Context : #{::NewRelic::Agent::Tracer.current_transaction.instance_variable_get(:@security_context_data).inspect}\n\n"
+        NewRelic::Security::Agent::Control::ReflectedXSS.check_xss(NewRelic::Security::Agent::Control::HTTPContext.get_context, retval) if NewRelic::Security::Agent.config[:'security.detection.rxss.enabled']
+        NewRelic::Security::Agent::Utils.delete_created_files(NewRelic::Security::Agent::Control::HTTPContext.get_context)
+        NewRelic::Security::Agent::Control::HTTPContext.reset_context
+        NewRelic::Security::Agent.logger.debug "Exit event : #{event}"
+      rescue => exception
+        NewRelic::Security::Agent.logger.error "Exception in hook in #{self.class}.#{__method__}, #{exception.inspect}, #{exception.backtrace}"
+      ensure
+        yield
+      end
+            
+    end
+
+    module Grape
+      module Router
+        
+        def prepare_env_from_route_on_enter(route)
+          NewRelic::Security::Agent.logger.debug "OnEnter : #{self.class}.#{__method__}"
+          ctxt = NewRelic::Security::Agent::Control::HTTPContext.get_context
+          http_method = route.instance_variable_get(:@request_method) ? route.instance_variable_get(:@request_method) : route.instance_variable_get(:@options)[:method]
+          ctxt.route = "#{http_method}@#{route.options[:namespace]}" unless ctxt.nil?
+        rescue => exception
+          NewRelic::Security::Agent.logger.error "Exception in hook in #{self.class}.#{__method__}, #{exception.inspect}, #{exception.backtrace}"
+        ensure
+          yield
+        end
+      end
+    end
+  end
+end
+
+NewRelic::Security::Instrumentation::InstrumentationLoader.install_instrumentation(:grape, ::Grape::API::Instance, ::NewRelic::Security::Instrumentation::Grape::API::Instance)
+NewRelic::Security::Instrumentation::InstrumentationLoader.install_instrumentation(:grape, ::Grape::Router, ::NewRelic::Security::Instrumentation::Grape::Router)

data/lib/newrelic_security/instrumentation-security/grape/prepend.rb

@@ -0,0 +1,30 @@
+module NewRelic::Security
+  module Instrumentation
+    module Grape
+      module API
+        module Instance
+          module Prepend
+            include NewRelic::Security::Instrumentation::Grape::API::Instance
+                        
+            def call(env)
+              retval = nil
+              event = call_on_enter(env) { retval = super(env) }
+              call_on_exit(event, retval) { return retval }
+            end
+
+          end
+        end
+      end
+
+      module Router
+        module Prepend
+          include NewRelic::Security::Instrumentation::Grape::Router
+
+          def prepare_env_from_route(env, route)
+            prepare_env_from_route_on_enter(route) { super }
+          end
+        end
+      end
+    end
+  end
+end

data/lib/newrelic_security/instrumentation-security/grpc/client/chain.rb

@@ -0,0 +1,47 @@
+module NewRelic::Security
+  module Instrumentation
+    module GRPC
+      module ClientStub
+        module Chain
+          def self.instrument!
+            ::GRPC::ClientStub.class_eval do
+              include NewRelic::Security::Instrumentation::GRPC::ClientStub
+
+              alias_method :request_response_without_security, :request_response
+    
+              def request_response(method, req, marshal, unmarshal, deadline: nil, return_op: false, parent: nil, credentials: nil, metadata: {}) # rubocop:disable Metrics/ParameterLists
+                retval = nil
+                event = grpc_client_on_enter(method, metadata) { retval = request_response_without_security(method, req, marshal, unmarshal, deadline, return_op, parent, credentials, metadata) }
+                grpc_client_on_exit(event) { return retval }
+              end
+
+              alias_method :server_streamer_without_security, :server_streamer
+
+              def server_streamer(method, req, marshal, unmarshal, deadline: nil, return_op: false, parent: nil, credentials: nil, metadata: {}, &blk) # rubocop:disable Metrics/ParameterLists
+                retval = nil
+                event = grpc_client_on_enter(method, metadata) { retval = server_streamer_without_security(method, req, marshal, unmarshal, deadline, return_op, parent, credentials, metadata, &blk) }
+                grpc_client_on_exit(event) { return retval }
+              end
+
+              alias_method :client_streamer_without_security, :client_streamer
+
+              def client_streamer(method, requests, marshal, unmarshal, deadline: nil, return_op: false, parent: nil, credentials: nil, metadata: {}) # rubocop:disable Metrics/ParameterLists
+                retval = nil
+                event = grpc_client_on_enter(method, metadata) { retval = client_streamer_without_security(method, requests, marshal, unmarshal, deadline, return_op, parent, credentials, metadata) }
+                grpc_client_on_exit(event) { return retval }
+              end
+
+              alias_method :bidi_streamer_without_security, :bidi_streamer
+
+              def bidi_streamer(method, requests, marshal, unmarshal, deadline: nil, return_op: false, parent: nil, credentials: nil, metadata: {}, &blk) # rubocop:disable Metrics/ParameterLists
+                retval = nil
+                event = grpc_client_on_enter(method, metadata) { retval = bidi_streamer_without_security(method, requests, marshal, unmarshal, deadline, return_op, parent, credentials, metadata, &blk) }
+                grpc_client_on_exit(event) { return retval }
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/newrelic_security/instrumentation-security/grpc/client/instrumentation.rb

@@ -0,0 +1,37 @@
+require_relative 'prepend'
+require_relative 'chain'
+
+module NewRelic::Security
+  module Instrumentation
+    module GRPC
+      module ClientStub
+        def grpc_client_on_enter(method, metadata)
+          event = nil
+          NewRelic::Security::Agent.logger.debug "OnEnter : #{self.class}.#{__method__}"
+          host = "grpc://#{instance_variable_get(:@host)}#{method}"
+          # TODO: Recheck this parameters, this is diff for other LCs
+          event = NewRelic::Security::Agent::Control::Collector.collect(HTTP_REQUEST, [{:host => host, :path => SLASH}])
+          # TODO: tracing data is required only in microservices, currently add_tracing_data data causing issues in gRPC.
+          # NewRelic::Security::Instrumentation::InstrumentationUtils.add_tracing_data(metadata, event) if event
+          event
+        rescue => exception
+          NewRelic::Security::Agent.logger.error "Exception in hook in #{self.class}.#{__method__}, #{exception.inspect}, #{exception.backtrace}"
+        ensure
+          yield
+          return event
+        end
+  
+        def grpc_client_on_exit(event)
+          NewRelic::Security::Agent.logger.debug "OnExit :  #{self.class}.#{__method__}"
+          NewRelic::Security::Agent::Utils.create_exit_event(event)
+        rescue => exception
+          NewRelic::Security::Agent.logger.error "Exception in hook in #{self.class}.#{__method__}, #{exception.inspect}, #{exception.backtrace}"
+        ensure
+          yield
+        end
+      end
+    end
+  end
+end
+
+NewRelic::Security::Instrumentation::InstrumentationLoader.install_instrumentation(:gRPC_Client, ::GRPC::ClientStub, ::NewRelic::Security::Instrumentation::GRPC::ClientStub)

data/lib/newrelic_security/instrumentation-security/grpc/client/prepend.rb

@@ -0,0 +1,36 @@
+module NewRelic::Security
+  module Instrumentation
+    module GRPC
+      module ClientStub
+        module Prepend
+          include NewRelic::Security::Instrumentation::GRPC::ClientStub
+
+          def request_response(method, req, marshal, unmarshal, deadline: nil, return_op: false, parent: nil, credentials: nil, metadata: {}) # rubocop:disable Metrics/ParameterLists
+            retval = nil
+            event = grpc_client_on_enter(method, metadata) { retval = super }
+            grpc_client_on_exit(event) { return retval }
+          end
+
+          def server_streamer(method, req, marshal, unmarshal, deadline: nil, return_op: false, parent: nil, credentials: nil, metadata: {}, &blk) # rubocop:disable Metrics/ParameterLists
+            retval = nil
+            event = grpc_client_on_enter(method, metadata) { retval = super }
+            grpc_client_on_exit(event) { return retval }
+          end
+
+          def client_streamer(method, requests, marshal, unmarshal, deadline: nil, return_op: false, parent: nil, credentials: nil, metadata: {}) # rubocop:disable Metrics/ParameterLists
+            retval = nil
+            event = grpc_client_on_enter(method, metadata) { retval = super }
+            grpc_client_on_exit(event) { return retval }
+          end
+
+          def bidi_streamer(method, requests, marshal, unmarshal, deadline: nil, return_op: false, parent: nil, credentials: nil, metadata: {}, &blk) # rubocop:disable Metrics/ParameterLists
+            retval = nil
+            event = grpc_client_on_enter(method, metadata) { retval = super }
+            grpc_client_on_exit(event) { return retval }
+          end
+
+        end
+      end
+    end
+  end
+end