Module: Datadog::AppSec::Contrib::RestClient::RequestSSRFDetectionPatch

Defined in:
lib/datadog/appsec/contrib/rest_client/request_ssrf_detection_patch.rb

Overview

Module that adds SSRF detection to RestClient::Request#execute

Instance Method Summary collapse

Instance Method Details

#execute(&block) ⇒ Object 



10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
 
# File 'lib/datadog/appsec/contrib/rest_client/request_ssrf_detection_patch.rb', line 10

def execute(&block)
  return super unless AppSec.rasp_enabled? && AppSec.active_context

  context = AppSec.active_context

  ephemeral_data = { 'server.io.net.url' => url }
  result = context.run_rasp(Ext::RASP_SSRF, {}, ephemeral_data, Datadog.configuration.appsec.waf_timeout)

  if result.match?
    Datadog::AppSec::Event.tag_and_keep!(context, result)

    context.events << {
      waf_result: result,
      trace: context.trace,
      span: context.span,
      request_url: url,
      actions: result.actions
    }

    ActionsHandler.handle(result.actions)
  end

  super(&block)
end