3rdparty/modules/aviator/lib/puppet/feature/faraday/adapter/net_http.rb

   1 begin
   2   require 'net/https'
   3 rescue LoadError
   4   warn "Warning: no such file to load -- net/https. Make sure openssl is installed if you want ssl support"
   5   require 'net/http'
   6 end
   7 require 'zlib'
   8
   9 module Faraday
  10   class Adapter
  11     class NetHttp < Faraday::Adapter
  12       NET_HTTP_EXCEPTIONS = [
  13         EOFError,
  14         Errno::ECONNABORTED,
  15         Errno::ECONNREFUSED,
  16         Errno::ECONNRESET,
  17         Errno::EHOSTUNREACH,
  18         Errno::EINVAL,
  19         Errno::ENETUNREACH,
  20         Net::HTTPBadResponse,
  21         Net::HTTPHeaderSyntaxError,
  22         Net::ProtocolError,
  23         SocketError,
  24         Zlib::GzipFile::Error,
  25       ]
  26
  27       NET_HTTP_EXCEPTIONS << OpenSSL::SSL::SSLError if defined?(OpenSSL)
  28
  29       def call(env)
  30         super
  31         http = net_http_connection(env)
  32         configure_ssl(http, env[:ssl]) if env[:url].scheme == 'https' and env[:ssl]
  33
  34         req = env[:request]
  35         http.read_timeout = http.open_timeout = req[:timeout] if req[:timeout]
  36         http.open_timeout = req[:open_timeout]                if req[:open_timeout]
  37
  38         begin
  39           http_response = perform_request(http, env)
  40         rescue *NET_HTTP_EXCEPTIONS => err
  41           if defined?(OpenSSL) && OpenSSL::SSL::SSLError === err
  42             raise Faraday::SSLError, err
  43           else
  44             raise Error::ConnectionFailed, err
  45           end
  46         end
  47
  48         save_response(env, http_response.code.to_i, http_response.body || '') do |response_headers|
  49           http_response.each_header do |key, value|
  50             response_headers[key] = value
  51           end
  52         end
  53
  54         @app.call env
  55       rescue Timeout::Error => err
  56         raise Faraday::Error::TimeoutError, err
  57       end
  58
  59       def create_request(env)
  60         request = Net::HTTPGenericRequest.new \
  61           env[:method].to_s.upcase,    # request method
  62           !!env[:body],                # is there request body
  63           :head != env[:method],       # is there response body
  64           env[:url].request_uri,       # request uri path
  65           env[:request_headers]        # request headers
  66
  67         if env[:body].respond_to?(:read)
  68           request.body_stream = env[:body]
  69         else
  70           request.body = env[:body]
  71         end
  72         request
  73       end
  74
  75       def perform_request(http, env)
  76         if :get == env[:method] and !env[:body]
  77           # prefer `get` to `request` because the former handles gzip (ruby 1.9)
  78           http.get env[:url].request_uri, env[:request_headers]
  79         else
  80           http.request create_request(env)
  81         end
  82       end
  83
  84       def net_http_connection(env)
  85         if proxy = env[:request][:proxy]
  86           Net::HTTP::Proxy(proxy[:uri].host, proxy[:uri].port, proxy[:user], proxy[:password])
  87         else
  88           Net::HTTP
  89         end.new(env[:url].host, env[:url].port)
  90       end
  91
  92       def configure_ssl(http, ssl)
  93         http.use_ssl      = true
  94         http.verify_mode  = ssl_verify_mode(ssl)
  95         http.cert_store   = ssl_cert_store(ssl)
  96
  97         http.cert         = ssl[:client_cert]  if ssl[:client_cert]
  98         http.key          = ssl[:client_key]   if ssl[:client_key]
  99         http.ca_file      = ssl[:ca_file]      if ssl[:ca_file]
 100         http.ca_path      = ssl[:ca_path]      if ssl[:ca_path]
 101         http.verify_depth = ssl[:verify_depth] if ssl[:verify_depth]
 102         http.ssl_version  = ssl[:version]      if ssl[:version]
 103       end
 104
 105       def ssl_cert_store(ssl)
 106         return ssl[:cert_store] if ssl[:cert_store]
 107         # Use the default cert store by default, i.e. system ca certs
 108         cert_store = OpenSSL::X509::Store.new
 109         cert_store.set_default_paths
 110         cert_store
 111       end
 112
 113       def ssl_verify_mode(ssl)
 114         ssl[:verify_mode] || begin
 115           if ssl.fetch(:verify, true)
 116             OpenSSL::SSL::VERIFY_PEER
 117           else
 118             OpenSSL::SSL::VERIFY_NONE
 119           end
 120         end
 121       end
 122     end
 123   end
 124 end