sqreen 1.18.2-java → 1.19.0-java

data/lib/sqreen/log/loggable.rb

@@ -0,0 +1,28 @@
+# typed: false
+
+# Copyright (c) 2015 Sqreen. All Rights Reserved.
+# Please refer to our terms for more information: https://www.sqreen.com/terms.html
+
+require 'logger'
+
+module Sqreen; end
+module Sqreen::Log; end
+
+module Sqreen::Log::Loggable
+  def self.included(klass)
+    klass.extend(ClassMethods)
+  end
+
+  module ClassMethods
+    def logger
+      # TODO: use progname for qualified log messages
+      # TODO: qualified logger proxies for selectable levels per component
+      # @logger ||= ::Logger.new(STDOUT, progname: name, level: Logger::DEBUG)
+      Sqreen.log
+    end
+  end
+
+  def logger
+    @logger || self.class.logger
+  end
+end

data/lib/sqreen/logger.rb

@@ -0,0 +1,85 @@
+# typed: true
+
+# Copyright (c) 2015 Sqreen. All Rights Reserved.
+# Please refer to our terms for more information: https://www.sqreen.com/terms.html
+
+require 'logger'
+require 'sqreen/log'
+require 'sqreen/formatter_with_tid'
+require 'sqreen/null_logger'
+
+# TODO: inherit from ::Logger
+
+module Sqreen
+  # Wrapper class for sqreen logging
+  class Logger
+    SEVERITY_TO_METHOD = ::Logger::Severity.constants.each_with_object({}) do |s, h|
+      h[::Logger::Severity.const_get(s)] = s.downcase
+    end
+
+    def initialize(desired_level, log_location, force_logger = nil)
+      if force_logger
+        @logger = force_logger
+      else
+        init_logger_output(log_location)
+      end
+      init_log_level(desired_level)
+      enforce_log_format(@logger)
+      create_error_logger
+    end
+
+    def debug(msg = nil, &block)
+      @logger.debug(msg, &block)
+    end
+
+    def info(msg = nil, &block)
+      @logger.info(msg, &block)
+    end
+
+    def warn(msg = nil, &block)
+      @logger.warn(msg, &block)
+    end
+
+    def error(msg = nil, &block)
+      @error_logger.error(msg, &block)
+      @logger.error(msg, &block)
+    end
+
+    def add(severity, msg = nil, &block)
+      send(SEVERITY_TO_METHOD[severity], msg, &block)
+    end
+
+    protected
+
+    def init_logger_output(path)
+      path = File.expand_path(path)
+      if File.writable?(path) || File.writable?(File.dirname(path))
+        @logger = ::Logger.new(path)
+      else
+        @logger = ::Logger.new(STDOUT)
+        @logger.info("Cannot access #{path} for writing. Defaulting to stdout")
+      end
+    rescue StandardError => e
+      @logger = ::Logger.new(STDOUT)
+      @logger.error('Got error while trying to setting logger up, '\
+                    "falling back to stdout #{e.inspect}")
+    end
+
+    def init_log_level(level)
+      log_level = ::Logger.const_get(level)
+      @logger.level = log_level
+      Sqreen::PerformanceNotifications::Log.enable if level == 'DEBUG'
+      return if level != 'DEBUG' && !Sqreen.config_get(:report_perf)
+      Sqreen::PerformanceNotifications::LogPerformance.enable
+    end
+
+    def create_error_logger
+      @error_logger = Kernel.const_defined?('MiniTest') ? NullLogger.instance : ::Logger.new(STDERR)
+      enforce_log_format(@error_logger)
+    end
+
+    def enforce_log_format(logger)
+      logger.formatter = Sqreen::FormatterWithTid.new
+    end
+  end
+end

data/lib/sqreen/metrics.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/metrics/average.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/metrics/base.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/metrics/binning.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/metrics/collect.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/metrics/sum.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/metrics_store.rb

@@ -1,23 +1,17 @@
+# typed: true
+
 # Copyright (c) 2015 Sqreen. All Rights Reserved.
 # Please refer to our terms for more information: https://www.sqreen.com/terms.html
 
-require 'sqreen/exception'
 require 'sqreen/metrics'
 require 'sqreen/mono_time'
+require 'sqreen/metrics_store/unknown_metric'
+require 'sqreen/metrics_store/unregistered_metric'
+require 'sqreen/metrics_store/already_registered_metric'
 
 module Sqreen
   # This store and register metrics
   class MetricsStore
-    # When a metric is not yet created
-    class UnregisteredMetric < Sqreen::Exception
-    end
-    # When the metric is unknown
-    class UnknownMetric < Sqreen::Exception
-    end
-    # When this name as already been declared with another kind
-    class AlreadyRegisteredMetric < Sqreen::Exception
-    end
-
     # definition keys
     NAME_KEY = 'name'.freeze
     KIND_KEY = 'kind'.freeze

data/lib/sqreen/metrics_store/already_registered_metric.rb

@@ -0,0 +1,13 @@
+# typed: strong
+
+# Copyright (c) 2015 Sqreen. All Rights Reserved.
+# Please refer to our terms for more information: https://www.sqreen.com/terms.html
+
+require 'sqreen/exception'
+
+module Sqreen
+  class MetricsStore
+    class AlreadyRegisteredMetric < Sqreen::Exception
+    end
+  end
+end

data/lib/sqreen/metrics_store/unknown_metric.rb

@@ -0,0 +1,13 @@
+# typed: strong
+
+# Copyright (c) 2015 Sqreen. All Rights Reserved.
+# Please refer to our terms for more information: https://www.sqreen.com/terms.html
+
+require 'sqreen/exception'
+
+module Sqreen
+  class MetricsStore
+    class UnknownMetric < Sqreen::Exception
+    end
+  end
+end

data/lib/sqreen/metrics_store/unregistered_metric.rb

@@ -0,0 +1,13 @@
+# typed: strong
+
+# Copyright (c) 2015 Sqreen. All Rights Reserved.
+# Please refer to our terms for more information: https://www.sqreen.com/terms.html
+
+require 'sqreen/exception'
+
+module Sqreen
+  class MetricsStore
+    class UnregisteredMetric < Sqreen::Exception
+    end
+  end
+end

data/lib/sqreen/middleware.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
 
@@ -11,38 +13,4 @@ module Sqreen
       @app.call(env)
     end
   end
-
-  class ErrorHandlingMiddleware
-    def initialize(app)
-      @app = app
-    end
-
-    def call(env)
-      @app.call(env)
-    rescue => e
-      sqreen_attack = nil
-      if e.is_a?(Sqreen::AttackBlocked)
-        sqreen_attack = e
-      elsif e.respond_to?(:original_exception) &&
-                            e.original_exception.is_a?(Sqreen::AttackBlocked)
-        sqreen_attack = e.original_exception
-      end
-
-      if sqreen_attack && sqreen_attack.redirect_url
-        return [303, { 'Location' => sqreen_attack.redirect_url }, ['']]
-      else
-        raise
-      end
-    end
-  end
-
-  class RailsMiddleware
-    def initialize(app)
-      @app = app
-    end
-
-    def call(env)
-      @app.call(env)
-    end
-  end
 end

data/lib/sqreen/mono_time.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
 
+# TODO: move to Sqreen::Time
+
 module Sqreen
   has_mono_time = begin
     Process.clock_gettime Process::CLOCK_MONOTONIC

data/lib/sqreen/node.rb

@@ -0,0 +1,46 @@
+# typed: false
+
+# Copyright (c) 2015 Sqreen. All Rights Reserved.
+# Please refer to our terms for more information: https://www.sqreen.com/terms.html
+
+# TODO: move to Sqreen::IP::Trie
+
+module Sqreen
+  # bit starts at 0 (most significant)
+  Node = Struct.new(:bit, :prefix, :l, :r, :parent) do
+    def initialize(*args)
+      super
+      raise ArgumentError, 'no bit given' if bit.nil?
+    end
+
+    def empty?
+      prefix.nil?
+    end
+
+    # cover the whole tree
+    def walk(max_bits, empty_nodes = false)
+      xstack = Array.new(max_bits + 1)
+      sidx = 0 # stack index
+      xhead = self
+      xcur = xhead
+      until xcur.nil?
+        yield xcur unless xcur.empty? && !empty_nodes
+
+        if xcur.l
+          if xcur.r
+            xstack[sidx] = xcur.r
+            sidx += 1
+          end
+          xcur = xcur.l
+        elsif xcur.r
+          xcur = xcur.r
+        elsif sidx.nonzero?
+          sidx -= 1
+          xcur = xstack[sidx]
+        else
+          xcur = nil
+        end
+      end
+    end
+  end
+end

data/lib/sqreen/not_implemented_yet.rb

@@ -0,0 +1,10 @@
+# typed: strong
+
+# Copyright (c) 2015 Sqreen. All Rights Reserved.
+# Please refer to our terms for more information: https://www.sqreen.com/terms.html
+
+require 'sqreen/exception'
+
+module Sqreen
+  class NotImplementedYet < Sqreen::Exception; end
+end

data/lib/sqreen/null_logger.rb

@@ -0,0 +1,26 @@
+# typed: true
+
+# Copyright (c) 2015 Sqreen. All Rights Reserved.
+# Please refer to our terms for more information: https://www.sqreen.com/terms.html
+
+require 'singleton'
+
+module Sqreen
+  class NullLogger
+    include Singleton
+
+    def debug(_msg = nil); end
+
+    def info(_msg = nil); end
+
+    def warn(_msg = nil); end
+
+    def error(_msg = nil); end
+
+    def fatal(_msg = nil); end
+
+    def add(_severity, _msg = nil); end
+
+    def formatter=(_); end
+  end
+end

data/lib/sqreen/payload_creator.rb

@@ -1,8 +1,11 @@
+# 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/runtime_infos'
 require 'sqreen/events/remote_exception'
+require 'sqreen/payload_creator/header_section'
 
 module Sqreen
   # Create a payload from a given query
@@ -116,26 +119,8 @@ module Sqreen
       Sqreen::RemoteException.record(e)
     end
 
-    # object that default to call on framework header
-    class HeaderSection
-      def initialize(framework)
-        @framework = framework
-      end
-
-      def [](value)
-        if %w[rack_client_ip rails_client_ip ip_headers].include?(value)
-          return @framework.send(value)
-        end
-        @framework.header(value)
-      end
-
-      def ip_headers
-        @framework.ip_headers
-      end
-    end
-
     def section_headers(framework)
-      HeaderSection.new(framework)
+      Sqreen::PayloadCreator::HeaderSection.new(framework)
     end
   end
 end

data/lib/sqreen/payload_creator/header_section.rb

@@ -0,0 +1,30 @@
+# typed: true
+
+# Copyright (c) 2015 Sqreen. All Rights Reserved.
+# Please refer to our terms for more information: https://www.sqreen.com/terms.html
+
+require 'sqreen/runtime_infos'
+require 'sqreen/events/remote_exception'
+require 'sqreen/payload_creator/header_section'
+
+module Sqreen
+  class PayloadCreator
+    # object that default to call on framework header
+    class HeaderSection
+      def initialize(framework)
+        @framework = framework
+      end
+
+      def [](value)
+        if %w[rack_client_ip rails_client_ip ip_headers].include?(value)
+          return @framework.send(value)
+        end
+        @framework.header(value)
+      end
+
+      def ip_headers
+        @framework.ip_headers
+      end
+    end
+  end
+end

data/lib/sqreen/performance_notifications.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