alexb-raygun-apm 1.0.57-x64-mingw32

data/lib/raygun/apm/blacklist/parser.rb

@@ -0,0 +1,47 @@
+module Raygun
+  module Apm
+    module Blacklist
+      class Parser
+        COMMENT = /^#[^<].*/
+        ANONYMOUS = /^#<.*:.*>?/
+
+        def initialize(tracer)
+          @tracer = tracer
+          @translator = Blacklist::Translator.new
+        end
+
+        def add_filters(filters)
+          filters.each do |filter|
+            filter.strip!
+            add_filter(filter)
+          end
+          show_filters
+        end
+
+        private
+        def add_filter(filter)
+          if filter =~ COMMENT && filter !~ ANONYMOUS
+            return 
+          end
+          if filter.start_with?('+')
+            @tracer.add_whitelist *translate(filter[1..-1])
+          elsif filter.start_with?('-')
+            @tracer.add_blacklist *translate(filter[1..-1])
+          elsif filter.size > 0
+            @tracer.add_blacklist *translate(filter)
+          end
+        rescue => e
+          puts "Failed to add line '#{filter}' to the blacklist (#{e}) #{e.backtrace.join("\n")}"
+        end
+
+        def show_filters
+          @tracer.show_filters if @tracer.config.loglevel == Tracer::LOG_BLACKLIST
+        end
+
+        def translate(filter)
+          @translator.translate(filter)
+        end
+      end
+    end
+  end
+end

data/lib/raygun/apm/blacklist/translator.rb

@@ -0,0 +1,73 @@
+module Raygun
+  module Apm
+    module Blacklist
+      class Translator
+        class RubyTranslator
+          # Foo::Bar#baz
+          # Foo::Bar.baz
+          COMMENT = /^#.*/
+          ANONYMOUS = /^#<.*>?/
+          NAMESPACE = /::/
+          NAMESPACE_ONLY = /::$/
+          METHOD = /#|\./
+          LETTER_CASE = /^[A-Z]/
+
+          def translate(filter)
+            path, method = nil, nil
+            if filter !~ COMMENT && filter !~ ANONYMOUS
+              if filter.end_with?("#")
+                path = filter
+              else
+                path, method = filter.split(METHOD)
+              end
+              # .NET fallback
+              return if method =~ LETTER_CASE && !method.start_with?("Ruby")
+              if path == path.downcase
+                method = path
+                path = nil
+              end
+
+              # .NET fallback
+              return if method =~ NAMESPACE
+              [path, method]
+            elsif filter =~ ANONYMOUS
+              _, klass, method = filter.split(METHOD)
+              ["##{klass}", method]
+            else
+              nil
+            end
+          end
+        end
+
+        # References https://raygun.com/documentation/product-guides/apm/blacklist/
+        class DotnetTranslator
+          # Foo.Bar::Baz
+          COMMENT = /^#/
+          NAMESPACE = /\./
+          METHOD = /::/
+  
+          def translate(filter)
+            if filter !~ COMMENT
+              path, method = nil, nil
+              path, method = filter.split(METHOD)
+              path.gsub!(NAMESPACE, "::")
+              [path, method]
+            else
+              nil
+            end
+          end
+        end
+
+        def initialize
+          @ruby = RubyTranslator.new
+          @dotnet = DotnetTranslator.new
+        end
+
+        def translate(filter)
+          translated = @ruby.translate(filter)
+          translated ? translated : @dotnet.translate(filter)
+        end
+      end
+    end
+  end
+end

data/lib/raygun/apm/config.rb

@@ -0,0 +1,88 @@
+module Raygun
+  module Apm
+    class Config
+      LOGLEVELS = {
+        "None" => Tracer::LOG_NONE,
+        "Info" => Tracer::LOG_INFO,
+        "Warning" => Tracer::LOG_WARNING,
+        "Error" => Tracer::LOG_ERROR,
+        "Verbose" => Tracer::LOG_VERBOSE,
+        "Debug" => Tracer::LOG_DEBUG,
+        "Everything" => Tracer::LOG_EVERYTHING,
+        # ruby profiler specific
+        "Blacklist" => Tracer::LOG_BLACKLIST
+      }
+
+      ENVIRONMENTS = {
+        "development" => Tracer::ENV_DEVELOPMENT,
+        "production" => Tracer::ENV_PRODUCTION
+      }
+
+      attr_accessor :env
+      def initialize(env=ENV)
+        @env = env
+      end
+
+      def self.cast_to_boolean(x)
+        case x
+        when true, 'true', 'True', 1, '1' then true
+        else
+          false
+        end
+      end
+
+      def self.config_var(attr, opts={}, &blk)
+        define_method attr.downcase do
+          val = if x = env[attr]
+            if opts[:as] == Integer
+              Integer(x)
+            elsif opts[:as] == String
+              x.to_s
+            elsif opts[:as] == :boolean
+              self.class.cast_to_boolean(x)
+            end
+          else
+            opts[:default]
+          end
+          blk ? blk.call(val) : val
+        end
+      end
+
+      # Initial constants for ProtonAgentTail.exe
+      UDP_SINK_HOST = TCP_MANAGEMENT_HOST = '127.0.0.1'
+      UDP_SINK_MULTICAST_HOST = '239.100.15.215'
+      UDP_SINK_PORT = 2799
+      TCP_MANAGEMENT_PORT = 2790
+
+      ## Enumerate all PROTON_ constants
+      config_var 'PROTON_API_KEY', as: String, default: ''
+      config_var 'PROTON_DEBUG_LOGLEVEL', as: String, default: 'None'
+      config_var 'PROTON_USER_OVERRIDES_FILE', as: String
+      config_var 'PROTON_NETWORK_MODE', as: String, default: 'Udp'
+      config_var 'PROTON_FILE_IPC_FOLDER', as: String
+      config_var 'PROTON_USE_MULTICAST', as: String, default: 'False'
+      config_var 'PROTON_BATCH_IDLE_COUNTER', as: Integer, default: 500
+
+      def udp_host
+        if proton_use_multicast == 'True'
+          UDP_SINK_MULTICAST_HOST
+        else
+          UDP_SINK_HOST
+        end
+      end
+
+      def udp_port
+        UDP_SINK_PORT
+      end
+
+      def loglevel
+        LOGLEVELS[proton_debug_loglevel] || raise(ArgumentError, "invalid log level")
+      end
+
+      def environment
+        environment = env['RACK_ENV'] || env['RAILS_ENV'] || 'production'
+        ENVIRONMENTS[environment] || Tracer::ENV_PRODUCTION
+      end
+    end
+  end
+end

data/lib/raygun/apm/diagnostics.rb

@@ -0,0 +1,52 @@
+require "socket"
+require "json"
+
+module Raygun
+  module Apm
+    class Diagnostics
+      AGENT_STATE_DOWN = "\nThe Raygun APM Agent appears to not be running on the current host.\nIf not already installed, please consult https://raygun.com/documentation/product-guides/apm/agent/downloads/\nOtherwise refer to https://raygun.com/documentation/product-guides/apm/agent/installation/ for starting the Agent."
+      AGENT_STATE_UNKNOWN = "\nUnable to determine the state of the Raygun APM Agent."
+      AGENT_STATE_UP_MISCONFIGURED = "\nThe Raygun APM Agent is running, but misconfigured.\nThe API Key needs to be set through the Raygun_ApiKey environment variable.\nThe API key can be found under 'Application Settings' in the Raygun UI"
+      AGENT_STATE_UP_CONFIGURED = "\nThe Raygun APM Agent is configured properly!"
+      AGENT_MINIMUM_VERSION_NOT_MET = "\nVersion #{Raygun::Apm::VERSION} of the Raygun APM Profiler requires a minimum Agent version #{Raygun::Apm::MINIMUM_AGENT_VERSION}\nPlease download the latest Agent from https://raygun.com/documentation/product-guides/apm/agent/downloads/"
+
+      def initialize(host: Apm::Config::TCP_MANAGEMENT_HOST, port: Apm::Config::TCP_MANAGEMENT_PORT)
+        @host = host
+        @port = port
+      end
+
+      def verify_agent(tracer)
+        socket.write "GetAgentInfo"
+        response = JSON.parse(socket.gets)
+        if minimum_agent_version_not_met?(response['Version'])
+          puts AGENT_MINIMUM_VERSION_NOT_MET
+          tracer.noop!
+        else
+          if response['Status'] == 1
+            puts AGENT_STATE_UP_CONFIGURED
+          elsif response['Status'] == 0
+            puts AGENT_STATE_UP_MISCONFIGURED
+          end
+        end
+      rescue Errno::ECONNREFUSED
+        puts AGENT_STATE_DOWN
+      rescue
+        puts AGENT_STATE_UNKNOWN
+      end
+
+      private
+      def socket
+        @socket ||= s = TCPSocket.new(@host, @port)
+      end
+
+      def minimum_agent_version_not_met?(version)
+        # Legacy path
+        if String === version
+          version < Raygun::Apm::MINIMUM_AGENT_VERSION
+        else
+          "#{version['Major']}.#{version['Minor']}.#{version['Build']}.#{version['Revision']}" < Raygun::Apm::MINIMUM_AGENT_VERSION
+        end
+      end
+    end
+  end
+end

data/lib/raygun/apm/event.rb

@@ -0,0 +1,39 @@
+module Raygun
+  module Apm
+    class Event
+      def inspect
+        "#<#{self.class.name}:#{self.object_id}> length:#{self.length} pid:#{self[:pid]} tid:#{self[:tid]} timestamp:#{self[:timestamp]}"
+      end
+      class ThreadStarted < Event
+        def inspect
+          super + " parent_tid:#{self[:parent_tid]}"
+        end
+      end
+      class Begin < Event
+        def inspect
+          super + " function_id:#{self[:function_id]} instance_id:#{self[:instance_id]}"
+        end
+      end
+      class End < Event
+        def inspect
+          super + " function_id:#{self[:function_id]}"
+        end
+      end
+      class Methodinfo < Event
+        def inspect
+          super + " function_id:#{self[:function_id]} class_name:#{self[:class_name]} method_name:#{self[:method_name]} method_source:#{self[:method_source]}"
+        end
+      end
+      class HttpOut < Event
+        def inspect
+          super + " url:#{self[:url]} verb:#{self[:verb]} status:#{self[:status]} duration:#{self[:duration]}"
+        end
+      end
+      class BeginTransaction < Event
+        def inspect
+          super + " api_key:#{self[:api_key]} technology_type:#{self[:technology_type]} process_type:#{self[:process_type]}"
+        end
+      end
+    end
+  end
+end

data/lib/raygun/apm/hooks/excon.rb

@@ -0,0 +1,36 @@
+require 'excon'
+
+module Raygun
+  module Apm
+    module Hooks
+      module Excon
+        def request(params={}, &block)
+          if tracer = Raygun::Apm::Tracer.instance
+            started = tracer.now
+            response = super
+            ended = tracer.now
+            event = raygun_apm_http_out_event
+            event[:pid] = Process.pid
+            event[:url] = "#{@data[:scheme]}://#{@data[:host]}/#{params[:path]}"
+            event[:verb] = params[:method].to_s.upcase
+            event[:status] = response.status
+            event[:duration] = ended - started
+            event[:timestamp] = started
+            event[:tid] = tracer.get_thread_id(Thread.current)
+            tracer.emit(event)
+            response
+          else
+            super
+          end
+        end
+
+        private
+        def raygun_apm_http_out_event
+          @_raygun_apm_http_out_event ||= Raygun::Apm::Event::HttpOut.new
+        end
+      end
+    end
+  end
+end
+
+Excon::Connection.prepend(Raygun::Apm::Hooks::Excon)

data/lib/raygun/apm/hooks/httpclient.rb

@@ -0,0 +1,36 @@
+require 'httpclient'
+
+module Raygun
+  module Apm
+    module Hooks
+      module HTTPClient
+        def do_request(method, uri, query, body, header, &filtered_block)
+          if tracer = Raygun::Apm::Tracer.instance
+            started = tracer.now
+            response = super
+            ended = tracer.now
+            event = raygun_apm_http_out_event
+            event[:pid] = Process.pid
+            event[:url] = query ? URI.join(uri, query).to_s : uri.to_s
+            event[:verb] = method.to_s.upcase
+            event[:status] = response.code.to_i
+            event[:duration] = ended - started
+            event[:timestamp] = started
+            event[:tid] = tracer.get_thread_id(Thread.current)
+            tracer.emit(event)
+            response
+          else
+            super
+          end
+        end
+
+        private
+        def raygun_apm_http_out_event
+          @_raygun_apm_http_out_event ||= Raygun::Apm::Event::HttpOut.new
+        end
+      end
+    end
+  end
+end
+
+HTTPClient.prepend(Raygun::Apm::Hooks::HTTPClient)

data/lib/raygun/apm/hooks/internals.rb

@@ -0,0 +1,74 @@
+module Raygun
+  module Apm
+    module Hooks
+      module Object
+        def system(*args)
+          super
+        end
+
+        def sleep(*args)
+          super
+        end
+
+        def exec(*args)
+          super
+        end
+
+        def spawn(*args)
+          super
+        end
+
+        def fork(*args)
+          super
+        end
+      end
+
+      module IO
+        def sycall(*args)
+          super
+        end
+
+        def open(*args)
+          super
+        end
+
+        def puts(*args)
+          super
+        end
+
+        def gets(*args)
+          super
+        end
+
+        def readline(*args)
+          super
+        end
+
+        def readlines(*args)
+          super
+        end
+      end
+
+      module Random
+        def srand(*args)
+          super
+        end
+
+        def rand(*args)
+          super
+        end
+      end
+
+      module Signal
+        def trap(*args)
+          super
+        end
+      end
+    end
+  end
+end
+
+Object.prepend Raygun::Apm::Hooks::Object
+IO.prepend Raygun::Apm::Hooks::IO
+Random.prepend Raygun::Apm::Hooks::Random
+Signal.prepend Raygun::Apm::Hooks::Signal