Contents

# Copyright (c) 2015 Sqreen. All Rights Reserved.
# Please refer to our terms for more information: https://www.sqreen.com/terms.html

require 'sqreen/attack_blocked'

module Sqreen
  class ErrorHandlingMiddleware
    def initialize(app)
      @app = app
    end

    def call(env)
      @app.call(env)
    rescue StandardError => 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 }, ['']]
      end

      raise
    end
  end
end

Version data entries

9 entries across 9 versions & 1 rubygems

Version	Path
sqreen-1.18.6-java	lib/sqreen/error_handling_middleware.rb
sqreen-1.18.6	lib/sqreen/error_handling_middleware.rb
sqreen-1.18.5-java	lib/sqreen/error_handling_middleware.rb
sqreen-1.18.5	lib/sqreen/error_handling_middleware.rb
sqreen-1.18.4-java	lib/sqreen/error_handling_middleware.rb
sqreen-1.18.4	lib/sqreen/error_handling_middleware.rb
sqreen-1.18.3-java	lib/sqreen/error_handling_middleware.rb
sqreen-1.18.3	lib/sqreen/error_handling_middleware.rb
sqreen-1.18.3.beta2	lib/sqreen/error_handling_middleware.rb