sqreen 1.18.6 → 1.19.0.beta1

data/lib/sqreen/backport/original_name.rb

@@ -1,3 +1,5 @@
+# typed: false
+
 # Copyright (c) 2015 Sqreen. All Rights Reserved.
 # Please refer to our terms for more information: https://www.sqreen.com/terms.html
 

data/lib/sqreen/binding_accessor.rb

@@ -1,3 +1,5 @@
+# typed: false
+
 # Copyright (c) 2015 Sqreen. All Rights Reserved.
 # Please refer to our terms for more information: https://www.sqreen.com/terms.html
 

data/lib/sqreen/binding_accessor/path_elem.rb

@@ -1,3 +1,5 @@
+# typed: true
+
 # Copyright (c) 2015 Sqreen. All Rights Reserved.
 # Please refer to our terms for more information: https://www.sqreen.com/terms.html
 

data/lib/sqreen/binding_accessor/transforms.rb

@@ -1,3 +1,5 @@
+# typed: true
+
 # Copyright (c) 2015 Sqreen. All Rights Reserved.
 # Please refer to our terms for more information: https://www.sqreen.com/terms.html
 
@@ -19,7 +21,8 @@ module Sqreen
           next unless seen.add?(val.object_id)
           case val
           when Hash
-            keys.concat(val.keys)
+            # convert symbols are converted to strings (ARB-60)
+            keys.concat(val.keys.map { |x| sym_to_s(x) })
             look_into.concat(val.values)
           when Array
             look_into.concat(val)
@@ -75,6 +78,10 @@ module Sqreen
 
       private
 
+      def sym_to_s(value)
+        value.is_a?(Symbol) ? value.to_s : value
+      end
+
       def descend(value, max_iter = 1000)
         seen = Set.new
         look_into = [value]

data/lib/sqreen/call_countable.rb

@@ -1,3 +1,5 @@
+# typed: false
+
 # Copyright (c) 2015 Sqreen. All Rights Reserved.
 # Please refer to our terms for more information: https://www.sqreen.com/terms.html
 

data/lib/sqreen/capped_queue.rb

@@ -1,3 +1,5 @@
+# typed: true
+
 # Copyright (c) 2015 Sqreen. All Rights Reserved.
 # Please refer to our terms for more information: https://www.sqreen.com/terms.html
 

data/lib/sqreen/cb.rb

@@ -1,3 +1,5 @@
+# typed: true
+
 # Copyright (c) 2015 Sqreen. All Rights Reserved.
 # Please refer to our terms for more information: https://www.sqreen.com/terms.html
 

data/lib/sqreen/cb_tree.rb

@@ -1,3 +1,5 @@
+# typed: ignore
+
 # Copyright (c) 2015 Sqreen. All Rights Reserved.
 # Please refer to our terms for more information: https://www.sqreen.com/terms.html
 

data/lib/sqreen/condition_evaluator.rb

@@ -1,3 +1,5 @@
+# typed: false
+
 # Copyright (c) 2015 Sqreen. All Rights Reserved.
 # Please refer to our terms for more information: https://www.sqreen.com/terms.html
 

data/lib/sqreen/conditionable.rb

@@ -1,3 +1,5 @@
+# typed: false
+
 # Copyright (c) 2015 Sqreen. All Rights Reserved.
 # Please refer to our terms for more information: https://www.sqreen.com/terms.html
 

data/lib/sqreen/configuration.rb

@@ -1,3 +1,5 @@
+# typed: ignore
+
 # Copyright (c) 2015 Sqreen. All Rights Reserved.
 # Please refer to our terms for more information: https://www.sqreen.com/terms.html
 
@@ -36,6 +38,10 @@ module Sqreen
       :default => false, :convert => :to_bool },
     { :env => :SQREEN_LIBSQREEN, :name => :libsqreen,
       :default => true, :convert => :to_bool },
+    { :env => :SQREEN_WEAVE, :name => :weave,
+      :default => false, :convert => :to_bool },
+    { :env => :SQREEN_WEAVE_STRATEGY, :name => :weave_strategy,
+      :default => :chain, :convert => :to_sym },
     { :env => :SQREEN_URL,       :name => :url,
       :default => 'https://back.sqreen.io' },
     { :env => :SQREEN_TOKEN,     :name => :token,
@@ -83,6 +89,10 @@ module Sqreen
     str.to_i
   end
 
+  def self.to_sym(value)
+    value.to_sym
+  end
+
   # Class to access configurations variables
   # This try to load environment by different ways.
   # 1. By file:
@@ -206,5 +216,9 @@ module Sqreen
     def to_int(value)
       Sqreen::to_int(value)
     end
+
+    def to_sym(value)
+      Sqreen::to_sym(value)
+    end
   end
 end

data/lib/sqreen/context.rb

@@ -1,3 +1,5 @@
+# typed: true
+
 # Copyright (c) 2015 Sqreen. All Rights Reserved.
 # Please refer to our terms for more information: https://www.sqreen.com/terms.html
 

data/lib/sqreen/default_cb.rb

@@ -1,3 +1,5 @@
+# typed: true
+
 # Copyright (c) 2015 Sqreen. All Rights Reserved.
 # Please refer to our terms for more information: https://www.sqreen.com/terms.html
 

data/lib/sqreen/deferred_logger.rb

@@ -1,3 +1,5 @@
+# typed: false
+
 # Copyright (c) 2015 Sqreen. All Rights Reserved.
 # Please refer to our terms for more information: https://www.sqreen.com/terms.html
 

data/lib/sqreen/deliveries.rb

@@ -1,3 +1,5 @@
+# typed: strong
+
 # Copyright (c) 2015 Sqreen. All Rights Reserved.
 # Please refer to our terms for more information: https://www.sqreen.com/terms.html
 

data/lib/sqreen/deliveries/batch.rb

@@ -1,3 +1,5 @@
+# typed: true
+
 # Copyright (c) 2015 Sqreen. All Rights Reserved.
 # Please refer to our terms for more information: https://www.sqreen.com/terms.html
 

data/lib/sqreen/deliveries/simple.rb

@@ -1,3 +1,5 @@
+# typed: true
+
 # Copyright (c) 2015 Sqreen. All Rights Reserved.
 # Please refer to our terms for more information: https://www.sqreen.com/terms.html
 

data/lib/sqreen/dependency.rb

@@ -1,3 +1,5 @@
+# typed: true
+
 # Copyright (c) 2015 Sqreen. All Rights Reserved.
 # Please refer to our terms for more information: https://www.sqreen.com/terms.html
 
@@ -12,7 +14,7 @@ module Sqreen
     def self.resolve_const(name)
       raise ArgumentError if name.nil? || name.empty?
 
-      name.to_s.split('::').inject(Object) { |a, e| a.const_get(e) }
+      name.to_s.split('::').inject(Object) { |a, e| a.const_get(e, false) }
     end
   end
 end

data/lib/sqreen/dependency/detector.rb

@@ -1,7 +1,9 @@
+# typed: ignore
+
 # Copyright (c) 2015 Sqreen. All Rights Reserved.
 # Please refer to our terms for more information: https://www.sqreen.com/terms.html
 
-require 'sqreen/dependency/hook'
+require 'sqreen/graft/hook'
 require 'sqreen/dependency/rails'
 require 'sqreen/dependency/sinatra'
 require 'sqreen/dependency/rack'
@@ -32,28 +34,34 @@ module Sqreen
           Sqreen::Dependency::Rails.insert_sqreen_middlewares
         end if Sqreen::Dependency::Rails.required?
 
-        Sqreen::Dependency::Hook.add('Rack::Builder#to_app') do
+        Sqreen::Graft::Hook.add('Rack::Builder#to_app') do
           after do
             Sqreen::Dependency::Rails.inspect_middlewares
           end
         end if Sqreen::Dependency::Rails.required?
 
-        Sqreen::Dependency::Hook.add('Sinatra::Base.setup_middleware') do
-          after do |_, _, _, args|
+        Sqreen::Graft::Hook.add('Sinatra::Base.setup_middleware') do
+          after do |call|
+            args = call.args
+
             Sqreen::Dependency::Sinatra.insert_sqreen_middlewares(args.first)
           end
         end.install if Sqreen::Dependency::Sinatra.required?
 
-        Sqreen::Dependency::Hook.add('Rack::Builder#to_app') do
-          after do |_, builder|
+        Sqreen::Graft::Hook.add('Rack::Builder#to_app') do
+          after do |call|
+            builder = call.instance
+
             Sqreen::Dependency::Sinatra.inspect_middlewares(builder)
           end
         end if Sqreen::Dependency::Sinatra.required?
 
         # ensure startup of thread in request handling processes
 
-        Sqreen::Dependency::Hook.add('Rack::Builder#to_app') do
-          after do |callback, *|
+        Sqreen::Graft::Hook.add('Rack::Builder#to_app') do
+          after do |call|
+            callback = call.callback
+
             Sqreen.log.debug "[#{Process.pid}] Start mode #{Sqreen::Dependency::Detector.start_mode}"
             if Sqreen::Dependency::Detector.start_mode == :rails || Sqreen::Dependency::Detector.start_mode == :rackup
 
@@ -61,7 +69,7 @@ module Sqreen
                 Sqreen::Dependency::Rack.on_run(handler) do
                   case handler.name
                   when 'Rack::Handler::Puma'
-                    Sqreen::Dependency::Hook.add('Puma::Launcher#run') do
+                    Sqreen::Graft::Hook.add('Puma::Launcher#run') do
                       before do
                         # HACK: Puma master? hack falls apart when not preloading
                         # it would think master is not, triggering startup
@@ -73,12 +81,12 @@ module Sqreen
                         end
                       end
                     end
-                    Sqreen::Dependency::Hook['Puma::Launcher#run'].install
+                    Sqreen::Graft::Hook['Puma::Launcher#run'].install
                   when 'Rack::Handler::PhusionPassenger'
                     # noop, passenger will start his own separate process
                     Sqreen.log.debug "[#{Process.pid}] Passenger will start in standalone process"
                   when 'Rack::Handler::Unicorn' # unicorn-rails
-                    Sqreen::Dependency::Hook.add('Unicorn::HttpServer.new') do
+                    Sqreen::Graft::Hook.add('Unicorn::HttpServer.new') do
                       before do
                         # BUG: detects single process...
                       end
@@ -97,12 +105,12 @@ module Sqreen
           end
         end
 
-        Sqreen::Dependency::Hook['Rack::Builder#to_app'].install
+        Sqreen::Graft::Hook['Rack::Builder#to_app'].install
 
-        # Sqreen::Dependency::Hook.add('Rails::Server#start') do
+        # Sqreen::Graft::Hook.add('Rails::Server#start') do
         #   before { }
         # end
-        # Sqreen::Dependency::Hook['Rails::Server#start'].install
+        # Sqreen::Graft::Hook['Rails::Server#start'].install
         # /!\ double instrument Rails < Rack => Rails.start_with -> Rails.start_without -> super -> Rack.start_with -> Rails.start_without
       end
     end

data/lib/sqreen/dependency/libsqreen.rb

@@ -1,6 +1,10 @@
+# typed: ignore
+
 # Copyright (c) 2015 Sqreen. All Rights Reserved.
 # Please refer to our terms for more information: https://www.sqreen.com/terms.html
 
+require 'sqreen/dependency'
+
 module Sqreen
   module Dependency
     module LibSqreen

data/lib/sqreen/dependency/new_relic.rb

@@ -1,3 +1,5 @@
+# typed: ignore
+
 # Copyright (c) 2015 Sqreen. All Rights Reserved.
 # Please refer to our terms for more information: https://www.sqreen.com/terms.html
 

data/lib/sqreen/dependency/rack.rb

@@ -1,3 +1,5 @@
+# typed: ignore
+
 # Copyright (c) 2015 Sqreen. All Rights Reserved.
 # Please refer to our terms for more information: https://www.sqreen.com/terms.html
 
@@ -7,23 +9,26 @@ module Sqreen
       module_function
 
       def find_handler(&block)
-        Sqreen::Dependency::Hook.add('Rack::Server#server') do
-          after do |callback, _, server, _|
+        Sqreen::Graft::Hook.add('Rack::Server#server') do
+          after do |call|
+            callback = call.callback
+            server = call.returned
+
             block.call(server)
             callback.disable # do this once, :server is a lazy init accessor
           end
         end
-        Sqreen::Dependency::Hook['Rack::Server#server'].install
+        Sqreen::Graft::Hook['Rack::Server#server'].install
       end
 
       def on_run(handler, &block)
         Sqreen.log.debug "[#{Process.pid}] #{handler.inspect}"
         hookpoint_name = "#{handler.name}.run"
 
-        Sqreen::Dependency::Hook.add(hookpoint_name) do
+        Sqreen::Graft::Hook.add(hookpoint_name) do
           before { block.call(handler) }
         end
-        Sqreen::Dependency::Hook[hookpoint_name].install
+        Sqreen::Graft::Hook[hookpoint_name].install
       end
 
       def rackup?

data/lib/sqreen/dependency/rails.rb

@@ -1,6 +1,9 @@
+# typed: ignore
+
 # Copyright (c) 2015 Sqreen. All Rights Reserved.
 # Please refer to our terms for more information: https://www.sqreen.com/terms.html
 
+require 'sqreen/shrink_wrap'
 require 'sqreen/middleware'
 require 'sqreen/error_handling_middleware'
 require 'sqreen/rails_middleware'
@@ -27,6 +30,7 @@ module Sqreen
       def insert_sqreen_middlewares
         Sqreen.log.debug { 'Inserting Sqreen middlewares for Rails' }
         app = ::Rails.application
+        app.middleware.insert(0, Sqreen::ShrinkWrap)
         app.middleware.insert_after(::Rack::Runtime, Sqreen::Middleware)
         app.middleware.insert_after(::ActionDispatch::DebugExceptions, Sqreen::RailsMiddleware)
         app.middleware.insert_after(::ActionDispatch::DebugExceptions, Sqreen::ErrorHandlingMiddleware)

data/lib/sqreen/dependency/sentry.rb

@@ -1,3 +1,5 @@
+# typed: ignore
+
 # Copyright (c) 2015 Sqreen. All Rights Reserved.
 # Please refer to our terms for more information: https://www.sqreen.com/terms.html
 

data/lib/sqreen/dependency/sinatra.rb

@@ -1,6 +1,9 @@
+# typed: ignore
+
 # Copyright (c) 2015 Sqreen. All Rights Reserved.
 # Please refer to our terms for more information: https://www.sqreen.com/terms.html
 
+require 'sqreen/shrink_wrap'
 require 'sqreen/middleware'
 require 'sqreen/error_handling_middleware'
 require 'sqreen/sinatra_middleware'
@@ -33,7 +36,7 @@ module Sqreen
           end
         end
 
-        insert_middleware(builder, Sqreen::Middleware, args, block) do |p, u|
+        insert_middleware(builder, Sqreen::ShrinkWrap, args, block) do |p, u|
           if (i = middlewares(builder).index(::Sinatra::ExtendedRack))
             u.insert(i, p)
           else
@@ -41,6 +44,14 @@ module Sqreen
           end
         end
 
+        insert_middleware(builder, Sqreen::Middleware, args, block) do |p, u|
+          if (i = middlewares(builder).index(::Sinatra::ExtendedRack))
+            u.insert(i, p)
+          else
+            u.insert(1, p)
+          end
+        end
+
         insert_middleware(builder, Sqreen::SinatraMiddleware, args, block) do |p, u|
           if ::Sqreen::Dependency.const_exist?('Rack::PostBodyContentTypeParser') && (i = middlewares(builder).index(::Rack::PostBodyContentTypeParser))
             u.insert(i + 1, p)

data/lib/sqreen/encoding_sanitizer.rb

@@ -1,3 +1,5 @@
+# typed: true
+
 # Copyright (c) 2015 Sqreen. All Rights Reserved.
 # Please refer to our terms for more information: https://www.sqreen.com/terms.html
 

data/lib/sqreen/error_handling_middleware.rb

@@ -1,3 +1,5 @@
+# typed: false
+
 # Copyright (c) 2015 Sqreen. All Rights Reserved.
 # Please refer to our terms for more information: https://www.sqreen.com/terms.html
 

data/lib/sqreen/event.rb

@@ -1,3 +1,5 @@
+# typed: true
+
 # Copyright (c) 2015 Sqreen. All Rights Reserved.
 # Please refer to our terms for more information: https://www.sqreen.com/terms.html
 

data/lib/sqreen/events/attack.rb

@@ -1,3 +1,5 @@
+# typed: true
+
 # Copyright (c) 2015 Sqreen. All Rights Reserved.
 # Please refer to our terms for more information: https://www.sqreen.com/terms.html
 

data/lib/sqreen/events/remote_exception.rb

@@ -1,3 +1,5 @@
+# typed: true
+
 # Copyright (c) 2015 Sqreen. All Rights Reserved.
 # Please refer to our terms for more information: https://www.sqreen.com/terms.html
 

data/lib/sqreen/events/request_record.rb

@@ -1,3 +1,5 @@
+# typed: false
+
 # Copyright (c) 2015 Sqreen. All Rights Reserved.
 # Please refer to our terms for more information: https://www.sqreen.com/terms.html
 

data/lib/sqreen/exception.rb

@@ -1,3 +1,5 @@
+# typed: false
+
 # Copyright (c) 2015 Sqreen. All Rights Reserved.
 # Please refer to our terms for more information: https://www.sqreen.com/terms.html