Class: Sqreen::ErrorHandlingMiddleware

Inherits:
Object
  • Object
show all
Defined in:
lib/sqreen/error_handling_middleware.rb

Instance Method Summary collapse

Constructor Details

#initialize(app) ⇒ ErrorHandlingMiddleware

Returns a new instance of ErrorHandlingMiddleware.



10
11
12
 
# File 'lib/sqreen/error_handling_middleware.rb', line 10

def initialize(app)
  @app = app
end

Instance Method Details

#call(env) ⇒ Object 



14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
 
# File 'lib/sqreen/error_handling_middleware.rb', line 14

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