Class: FinalDestination::HTTP

Inherits:
Net::HTTP
  • Object
show all
Defined in:
lib/final_destination/http.rb

Instance Method Summary collapse

Instance Method Details

#connect ⇒ Object 



4
5
6
7
8
9
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/final_destination/http.rb', line 4

def connect
  original_open_timeout = @open_timeout
  return super if @ipaddr

  timeout_at = current_time + @open_timeout

  # This iteration through addresses would normally happen in Socket#tcp
  # We do it here because we're tightly controlling addresses rather than
  # handing Socket#tcp a hostname
  ips = FinalDestination::SSRFDetector.lookup_and_filter_ips(@address, timeout: @connect_timeout)

  ips.each_with_index do |ip, index|
    debug "[FinalDestination] Attempting connection to #{ip}..."
    self.ipaddr = ip

    remaining_time = timeout_at - current_time
    if remaining_time <= 0
      raise Net::OpenTimeout.new("Operation timed out - FinalDestination::HTTP")
    end

    @open_timeout = remaining_time
    return super
  rescue OpenSSL::SSL::SSLError, SystemCallError, Net::OpenTimeout => e
    debug "[FinalDestination] Error connecting to #{ip}... #{e.message}"
    was_last_attempt = index == ips.length - 1
    raise if was_last_attempt
  end
ensure
  @open_timeout = original_open_timeout
end