sqreen 1.1.11481117869-java → 1.1.21481714484-java

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 0380f38a3af0fbb87e91fb92c71261300ac8e705
-  data.tar.gz: e26603f5e01edd06550f54bb66ae4b202b24e5fc
+  metadata.gz: f03f4a7d74cf483d5876f295673eab4441ebda4f
+  data.tar.gz: 5ef683e47e23b38df0ccc8974d5b596be9436370
 SHA512:
-  metadata.gz: b274e896a689f03dbc034ee62ffb06761dfa7361665606bd7d31261a8fdfcc978361e832ad221ab2ceaf767a99eeb7b2bd49b6727b6c578b0c0d545795b48e1e
-  data.tar.gz: 5de74d338c25967869f36a2f6a6da4700d342ea544b0c3db30d7738bc0d0f0cfd490322e12e4f1a35964712f2e6338a817266243888848156fe8946807aaebaa
+  metadata.gz: 8273d011709cc1edc9d1203e6c31f5d4beb70d43d3cc936ad2635534ded5f512ad4f7bf8b9059847b93d76fc9c80fbba4363ca83a08dd5d16c83c40933b4bc91
+  data.tar.gz: 97ff17c99a5f914f75db942d81a8b9f17fb63eea1d7e2e51c89f6d1021c4799688847f561458016cd3fd02bfc069f7b144ed8df19c0bf6eeb59db8d654bb3e19

data/lib/sqreen/call_countable.rb

@@ -0,0 +1,67 @@
+# Copyright (c) 2015 Sqreen. All Rights Reserved.
+# Please refer to our terms for more information: https://www.sqreen.io/terms.html
+
+module Sqreen
+  # A module that will dynamically had call_counts to the pre/post/failing
+  # callbacks
+  module CallCountable
+    # Hook the necessary callback function
+    # The module being decorated is expected to have a
+    # record_observation & rulespack_id & rule_name method available (like RuleCallback)
+    #
+    # @param count [Hash] hash of callback names => count
+    def count_callback_calls(count)
+      base = self.class
+      @call_count_interval = 0
+      return if count.to_i == 0
+      @call_counts = {}
+      @call_count_interval = count
+      @call_count_names = {}
+      %w(pre post failing).each do |cb|
+        next unless base.method_defined?(cb)
+        @call_counts[cb] = 0
+        @call_count_names[cb] = "#{rulespack_id}/#{rule_name}/#{cb}".freeze
+        defd = base.instance_variable_defined?("@call_count_hooked_#{cb}")
+        next if defd && base.instance_variable_get("@call_count_hooked_#{cb}")
+        base.send(:alias_method, "#{cb}_without_count", cb)
+        base.send(:alias_method, cb, "#{cb}_with_count")
+        base.instance_variable_set("@call_count_hooked_#{cb}", true)
+      end
+    end
+    PRE = 'pre'.freeze
+    POST = 'post'.freeze
+    FAILING = 'failing'.freeze
+    COUNT_CALLS = 'sqreen_calls_count'.freeze
+
+    def pre_with_count(inst, *args, &block)
+      ret = pre_without_count(inst, *args, &block)
+      count_calls('pre')
+      ret
+    end
+
+    def post_with_count(rv, inst, *args, &block)
+      ret = post_without_count(rv, inst, *args, &block)
+      count_calls('post')
+      ret
+    end
+
+    def failing_with_count(rv, inst, *args, &block)
+      ret = failing_without_count(rv, inst, *args, &block)
+      count_calls('failing')
+      ret
+    end
+
+    attr_reader :call_counts
+    attr_reader :call_count_interval
+
+    protected
+
+    def count_calls(what)
+      return unless @call_count_interval > 0
+      new_value = (@call_counts[what] += 1)
+      return unless new_value % call_count_interval == 0
+      @call_counts[what] = 0
+      record_observation(COUNT_CALLS, @call_count_names[what], new_value)
+    end
+  end
+end

data/lib/sqreen/conditionable.rb

@@ -11,13 +11,12 @@ module Sqreen
     #
     # @param conditions [Hash] hash of callback names => conditions
     def condition_callbacks(conditions)
+      conditions = {} if conditions.nil?
       base = self.class
       %w(pre post failing).each do |cb|
-        next unless conditions.key?(cb)
         conds = conditions[cb]
-        next if conds.respond_to?(:empty?) && conds.empty?
         next unless base.method_defined?(cb)
-        send("#{cb}_conditions=", ConditionEvaluator.new(conds))
+        send("#{cb}_conditions=", ConditionEvaluator.new(conds)) unless conds.nil?
         defd = base.instance_variable_defined?("@conditional_hooked_#{cb}")
         next if defd && base.instance_variable_get("@conditional_hooked_#{cb}")
         base.send(:alias_method, "#{cb}_without_conditions", cb)

data/lib/sqreen/instrumentation.rb

@@ -6,6 +6,7 @@ require 'sqreen/log'
 require 'sqreen/stats'
 require 'sqreen/exception'
 require 'sqreen/performance_notifications'
+require 'sqreen/call_countable'
 require 'sqreen/events/remote_exception'
 require 'sqreen/rules_signature'
 require 'set'
@@ -496,6 +497,11 @@ module Sqreen
 
     def initialize(metrics_engine = nil)
       self.metrics_engine = metrics_engine
+      unless metrics_engine.nil?
+        metrics_engine.create_metric('name' => CallCountable::COUNT_CALLS,
+                                     'period' => 60,
+                                     'kind' => 'Sum')
+      end
     end
   end
 end

data/lib/sqreen/metrics_store.rb

@@ -23,6 +23,8 @@ module Sqreen
 
     # Currently ready samples
     attr_reader :store
+    # All known metrics
+    attr_reader :metrics
 
     def initialize
       @store = []

data/lib/sqreen/rule_attributes.rb

@@ -18,6 +18,7 @@ module Sqreen
       CALLBACK_CLASS = 'callback_class'.freeze
       METRICS = 'metrics'.freeze
       CONDITIONS = 'conditions'.freeze
+      CALL_COUNT_INTERVAL = 'call_count_interval'.freeze
 
       freeze
     end

data/lib/sqreen/rule_callback.rb

@@ -3,6 +3,7 @@
 
 require 'sqreen/callbacks'
 require 'sqreen/conditionable'
+require 'sqreen/call_countable'
 require 'sqreen/rule_attributes'
 require 'sqreen/events/attack'
 require 'sqreen/events/remote_exception'
@@ -14,6 +15,7 @@ module Sqreen
     # Base class for callback that are initialized by rules from Sqreen
     class RuleCB < CB
       include Conditionable
+      include CallCountable
       # If nothing was asked by the rule we will ask for all sections available
       # These information will be pruned later when exporting in #to_hash
       DEFAULT_PAYLOAD = PayloadCreator::METHODS.keys.freeze
@@ -32,8 +34,8 @@ module Sqreen
         @rule = rule_hash
         payload_tpl = @rule[Attrs::PAYLOAD] || DEFAULT_PAYLOAD
         @payload_generator = PayloadCreator.new(payload_tpl)
-        conditions = @rule[Attrs::CONDITIONS]
-        condition_callbacks(conditions) unless conditions.nil?
+        condition_callbacks(@rule[Attrs::CONDITIONS])
+        count_callback_calls(@rule[Attrs::CALL_COUNT_INTERVAL])
       end
 
       def rule_name

data/lib/sqreen/rules_callbacks/crawler_user_agent_matches_metrics.rb

@@ -13,7 +13,6 @@ module Sqreen
       def pre(_inst, *_args, &_block)
         ua = framework.client_user_agent
         return unless ua
-        ua = ua.freeze
         found = match(ua)
         return unless found
         Sqreen.log.debug { "Found UA #{ua} - found: #{found}" }

data/lib/sqreen/runner.rb

@@ -17,6 +17,7 @@ require 'sqreen/deliveries/simple'
 require 'sqreen/deliveries/batch'
 require 'sqreen/performance_notifications/metrics'
 require 'sqreen/instrumentation'
+require 'sqreen/call_countable'
 
 module Sqreen
   @features = {}
@@ -108,7 +109,6 @@ module Sqreen
 
       # Ensure a deliverer is there unless features have set it first
       self.deliverer ||= Deliveries::Simple.new(session)
-
       context_infos = {}
       %w(rules pack_id).each do |p|
         context_infos[p] = response[p] unless response[p].nil?
@@ -160,6 +160,14 @@ module Sqreen
       [rulespack_id, rules]
     end
 
+    def call_counts_metrics_period=(value)
+      value = value.to_i
+      return unless value > 0 # else disable collection?
+      metrics_engine.create_metric('name' => CallCountable::COUNT_CALLS,
+                                   'period' => value,
+                                   'kind' => 'Sum')
+    end
+
     def performance_metrics_period=(value)
       value = value.to_i
       if value > 0
@@ -214,6 +222,7 @@ module Sqreen
       Sqreen.update_features(features)
       session.request_compression = features['request_compression'] if session
       self.performance_metrics_period = features['performance_metrics_period']
+      self.call_counts_metrics_period = features['call_counts_metrics_period']
       hd = features['heartbeat_delay'].to_i
       self.heartbeat_delay = hd if hd > 0
       return if features['batch_size'].nil?

data/lib/sqreen/version.rb

@@ -2,5 +2,5 @@
 # Please refer to our terms for more information: https://www.sqreen.io/terms.html
 # Warning This file is auto generated! DO NOT edit.
 module Sqreen
-  VERSION = "1.1.11481117869".freeze
+  VERSION = "1.1.21481714484".freeze
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: sqreen
 version: !ruby/object:Gem::Version
-  version: 1.1.11481117869
+  version: 1.1.21481714484
 platform: java
 authors:
 - Sqreen
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2016-12-07 00:00:00.000000000 Z
+date: 2016-12-14 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: execjs
@@ -50,6 +50,7 @@ files:
 - lib/sqreen.rb
 - lib/sqreen/binding_accessor.rb
 - lib/sqreen/ca.crt
+- lib/sqreen/call_countable.rb
 - lib/sqreen/callback_tree.rb
 - lib/sqreen/callbacks.rb
 - lib/sqreen/capped_queue.rb
@@ -129,7 +130,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project:
-rubygems_version: 2.6.4
+rubygems_version: 2.6.8
 signing_key:
 specification_version: 4
 summary: Sqreen Ruby agent