Contents

# 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/rules/rule_cb'
require 'sqreen/actions'
require 'sqreen/middleware'
require 'sqreen/rails_middleware'
require 'sqreen/sinatra_middleware'

module Sqreen
  module Rules
    # Runs actions concerned with whether the request ought to be served
    class RunReqStartActions < FrameworkCB
      PRIORITY = 95

      def initialize(framework)
        if defined?(Sqreen::Frameworks::SinatraFramework) &&
           framework.is_a?(Sqreen::Frameworks::SinatraFramework)
          super(Sqreen::SinatraMiddleware, :call)
        elsif defined?(Sqreen::Frameworks::RailsFramework) &&
              framework.is_a?(Sqreen::Frameworks::RailsFramework)
          super(Sqreen::RailsMiddleware, :call)
        else
          # last resort; we won't get nice errors
          super(Sqreen::Middleware, :call)
        end

        self.framework = framework
      end

      def whitelisted?
        whitelisted = SharedStorage.get(:whitelisted)
        return whitelisted unless whitelisted.nil?
        framework && !framework.whitelisted_match.nil?
      end

      def priority
        PRIORITY
      end

      def pre(_inst, _args, _budget = nil, &_block)
        return unless framework
        ip = framework.client_ip
        return unless ip

        actions = actions_repo.get(Sqreen::Actions::BlockIp, ip) +
                  actions_repo.get(Sqreen::Actions::RedirectIp, ip)

        actions.each do |act|
          res = run_client_ip_action(act, ip)
          return res unless res.nil?
        end
        nil
      end

      private

      # @param action [Sqreen::Actions::Base]
      def run_client_ip_action(action, client_ip)
        action.run client_ip
      end

      # @return [Sqreen::Actions::Repository]
      def actions_repo
        Sqreen::Actions::Repository.current
      end
    end
  end
end

Version data entries

39 entries across 39 versions & 1 rubygems

Version	Path
sqreen-1.20.4.beta1	lib/sqreen/rules/run_req_start_actions.rb
sqreen-1.19.4	lib/sqreen/rules/run_req_start_actions.rb
sqreen-1.20.3	lib/sqreen/rules/run_req_start_actions.rb
sqreen-1.20.2	lib/sqreen/rules/run_req_start_actions.rb
sqreen-1.21.0.beta2	lib/sqreen/rules/run_req_start_actions.rb
sqreen-1.21.0.beta1	lib/sqreen/rules/run_req_start_actions.rb
sqreen-1.20.1-java	lib/sqreen/rules/run_req_start_actions.rb
sqreen-1.20.1	lib/sqreen/rules/run_req_start_actions.rb
sqreen-1.20.0-java	lib/sqreen/rules/run_req_start_actions.rb
sqreen-1.20.0	lib/sqreen/rules/run_req_start_actions.rb
sqreen-1.19.3-java	lib/sqreen/rules/run_req_start_actions.rb
sqreen-1.19.3	lib/sqreen/rules/run_req_start_actions.rb
sqreen-1.19.2-java	lib/sqreen/rules/run_req_start_actions.rb
sqreen-1.19.2	lib/sqreen/rules/run_req_start_actions.rb
sqreen-1.19.1-java	lib/sqreen/rules/run_req_start_actions.rb
sqreen-1.19.1	lib/sqreen/rules/run_req_start_actions.rb
sqreen-1.19.0-java	lib/sqreen/rules/run_req_start_actions.rb
sqreen-1.19.0	lib/sqreen/rules/run_req_start_actions.rb
sqreen-1.19.0.beta1	lib/sqreen/rules/run_req_start_actions.rb