class Net::HTTP

# File lib/net/http.rb, line 973
def HTTP.Proxy(p_addr = :ENV, p_port = nil, p_user = nil, p_pass = nil)
  return self unless p_addr

  Class.new(self) {
    @is_proxy_class = true

    if p_addr == :ENV then
      @proxy_from_env = true
      @proxy_address = nil
      @proxy_port    = nil
    else
      @proxy_from_env = false
      @proxy_address = p_addr
      @proxy_port    = p_port || default_port
    end

    @proxy_user = p_user
    @proxy_pass = p_pass
  }
end

# File lib/net/http.rb, line 517
def HTTP.default_port
  http_default_port()
end

# File lib/net/http.rb, line 453
def HTTP.get(uri_or_host, path = nil, port = nil)
  get_response(uri_or_host, path, port).body
end

# File lib/net/http.rb, line 434
def HTTP.get_print(uri_or_host, path = nil, port = nil)
  get_response(uri_or_host, path, port) {|res|
    res.read_body do |chunk|
      $stdout.print chunk
    end
  }
  nil
end

# File lib/net/http.rb, line 469
def HTTP.get_response(uri_or_host, path = nil, port = nil, &block)
  if path
    host = uri_or_host
    new(host, port || HTTP.default_port).start {|http|
      return http.request_get(path, &block)
    }
  else
    uri = uri_or_host
    start(uri.hostname, uri.port,
          :use_ssl => uri.scheme == 'https') {|http|
      return http.request_get(uri, &block)
    }
  end
end

# File lib/net/http.rb, line 522
def HTTP.http_default_port
  80
end

# File lib/net/http.rb, line 527
def HTTP.https_default_port
  443
end

# File lib/net/http.rb, line 608
def HTTP.new(address, port = nil, p_addr = :ENV, p_port = nil, p_user = nil, p_pass = nil)
  http = super address, port

  if proxy_class? then # from Net::HTTP::Proxy()
    http.proxy_from_env = @proxy_from_env
    http.proxy_address  = @proxy_address
    http.proxy_port     = @proxy_port
    http.proxy_user     = @proxy_user
    http.proxy_pass     = @proxy_pass
  elsif p_addr == :ENV then
    http.proxy_from_env = true
  else
    http.proxy_address = p_addr
    http.proxy_port    = p_port || default_port
    http.proxy_user    = p_user
    http.proxy_pass    = p_pass
  end

  http
end

# File lib/net/http.rb, line 632
def initialize(address, port = nil)
  @address = address
  @port    = (port || HTTP.default_port)
  @local_host = nil
  @local_port = nil
  @curr_http_version = HTTPVersion
  @keep_alive_timeout = 2
  @last_communicated = nil
  @close_on_empty_response = false
  @socket  = nil
  @started = false
  @open_timeout = nil
  @read_timeout = 60
  @continue_timeout = nil
  @debug_output = nil

  @proxy_from_env = false
  @proxy_uri      = nil
  @proxy_address  = nil
  @proxy_port     = nil
  @proxy_user     = nil
  @proxy_pass     = nil

  @use_ssl = false
  @ssl_context = nil
  @ssl_session = nil
  @enable_post_connection_check = true
  @sspi_enabled = false
  SSL_IVNAMES.each do |ivname|
    instance_variable_set ivname, nil
  end
end

# File lib/net/http.rb, line 502
def HTTP.post_form(url, params)
  req = Post.new(url)
  req.form_data = params
  req.basic_auth url.user, url.password if url.user
  start(url.hostname, url.port,
        :use_ssl => url.scheme == 'https' ) {|http|
    http.request(req)
  }
end

# File lib/net/http.rb, line 996
def proxy_class?
  defined?(@is_proxy_class) ? @is_proxy_class : false
end

# File lib/net/http.rb, line 565
def HTTP.start(address, *arg, &block) # :yield: +http+
  arg.pop if opt = Hash.try_convert(arg[-1])
  port, p_addr, p_port, p_user, p_pass = *arg
  port = https_default_port if !port && opt && opt[:use_ssl]
  http = new(address, port, p_addr, p_port, p_user, p_pass)

  if opt
    if opt[:use_ssl]
      opt = {verify_mode: OpenSSL::SSL::VERIFY_PEER}.update(opt)
    end
    http.methods.grep(/\A(\w+)=\z/) do |meth|
      key = $1.to_sym
      opt.key?(key) or next
      http.__send__(meth, opt[key])
    end
  end

  http.start(&block)
end

# File lib/net/http.rb, line 400
def HTTP.version_1_2
  true
end

# File lib/net/http.rb, line 406
def HTTP.version_1_2?
  true
end

# File lib/net/http.rb, line 725
def continue_timeout=(sec)
  @socket.continue_timeout = sec if @socket
  @continue_timeout = sec
end

# File lib/net/http.rb, line 1241
def copy(path, initheader = nil)
  request(Copy.new(path, initheader))
end

# File lib/net/http.rb, line 1229
def delete(path, initheader = {'Depth' => 'Infinity'})
  request(Delete.new(path, initheader))
end

# File lib/net/http.rb, line 942
def finish
  raise IOError, 'HTTP session not yet started' unless started?
  do_finish
end

# File lib/net/http.rb, line 1129
def get(path, initheader = {}, dest = nil, &block) # :yield: +body_segment+
  res = nil
  request(Get.new(path, initheader)) {|r|
    r.read_body dest, &block
    res = r
  }
  res
end

# File lib/net/http.rb, line 1151
def head(path, initheader = nil)
  request(Head.new(path, initheader))
end

# File lib/net/http.rb, line 665
def inspect
  "#<#{self.class} #{@address}:#{@port} open=#{started?}>"
end

# File lib/net/http.rb, line 1205
def lock(path, body, initheader = nil)
  request(Lock.new(path, initheader), body)
end

# File lib/net/http.rb, line 1247
def mkcol(path, body = nil, initheader = nil)
  request(Mkcol.new(path, initheader), body)
end

# File lib/net/http.rb, line 1235
def move(path, initheader = nil)
  request(Move.new(path, initheader))
end

# File lib/net/http.rb, line 1217
def options(path, initheader = nil)
  request(Options.new(path, initheader))
end

# File lib/net/http.rb, line 1189
def patch(path, data, initheader = nil, dest = nil, &block) # :yield: +body_segment+
  send_entity(path, data, initheader, dest, Patch, &block)
end

# File lib/net/http.rb, line 831
def peer_cert
  if not use_ssl? or not @socket
    return nil
  end
  @socket.io.peer_cert
end

# File lib/net/http.rb, line 1183
def post(path, data, initheader = nil, dest = nil, &block) # :yield: +body_segment+
  send_entity(path, data, initheader, dest, Post, &block)
end

# File lib/net/http.rb, line 1223
def propfind(path, body = nil, initheader = {'Depth' => '0'})
  request(Propfind.new(path, initheader), body)
end

# File lib/net/http.rb, line 1199
def proppatch(path, body, initheader = nil)
  request(Proppatch.new(path, initheader), body)
end

# File lib/net/http.rb, line 1015
def proxy?
  !!if @proxy_from_env then
    proxy_uri
  else
    @proxy_address
  end
end

# File lib/net/http.rb, line 1036
def proxy_address
  if @proxy_from_env then
    proxy_uri && proxy_uri.hostname
  else
    @proxy_address
  end
end

# File lib/net/http.rb, line 1024
def proxy_from_env?
  @proxy_from_env
end

# File lib/net/http.rb, line 1059
def proxy_pass
  @proxy_pass
end

# File lib/net/http.rb, line 1045
def proxy_port
  if @proxy_from_env then
    proxy_uri && proxy_uri.port
  else
    @proxy_port
  end
end

# File lib/net/http.rb, line 1054
def proxy_user
  @proxy_user
end

# File lib/net/http.rb, line 714
def read_timeout=(sec)
  @socket.read_timeout = sec if @socket
  @read_timeout = sec
end

# File lib/net/http.rb, line 1371
def request(req, body = nil, &block)  # :yield: +response+
  unless started?
    start {
      req['connection'] ||= 'close'
      return request(req, body, &block)
    }
  end
  if proxy_user()
    req.proxy_basic_auth proxy_user(), proxy_pass() unless use_ssl?
  end
  req.set_body_internal body
  res = transport_request(req, &block)
  if sspi_auth?(res)
    sspi_auth(req)
    res = transport_request(req, &block)
  end
  res
end

# File lib/net/http.rb, line 1282
def request_get(path, initheader = nil, &block) # :yield: +response+
  request(Get.new(path, initheader), &block)
end

# File lib/net/http.rb, line 1296
def request_head(path, initheader = nil, &block)
  request(Head.new(path, initheader), &block)
end

# File lib/net/http.rb, line 1326
def request_post(path, data, initheader = nil, &block) # :yield: +response+
  request Post.new(path, initheader), data, &block
end

# File lib/net/http.rb, line 1350
def send_request(name, path, data = nil, header = nil)
  has_response_body = name != 'HEAD'
  r = HTTPGenericRequest.new(name,(data ? true : false),has_response_body,path,header)
  request r, data
end

# File lib/net/http.rb, line 678
def set_debug_output(output)
  warn 'Net::HTTP#set_debug_output called after HTTP started' if started?
  @debug_output = output
end

# File lib/net/http.rb, line 847
def start  # :yield: http
  raise IOError, 'HTTP session already opened' if @started
  if block_given?
    begin
      do_start
      return yield(self)
    ensure
      do_finish
    end
  end
  do_start
  self
end

# File lib/net/http.rb, line 737
def started?
  @started
end

# File lib/net/http.rb, line 1253
def trace(path, initheader = nil)
  request(Trace.new(path, initheader))
end

# File lib/net/http.rb, line 1211
def unlock(path, body, initheader = nil)
  request(Unlock.new(path, initheader), body)
end

# File lib/net/http.rb, line 754
def use_ssl=(flag)
  flag = flag ? true : false
  if started? and @use_ssl != flag
    raise IOError, "use_ssl value changed, but session already started"
  end
  @use_ssl = flag
end

# File lib/net/http.rb, line 746
def use_ssl?
  @use_ssl
end

# File lib/net/http.rb, line 1537
def D(msg)
  return unless @debug_output
  @debug_output << msg
  @debug_output << "\n"
end

# File lib/net/http.rb, line 1529
def addr_port
  if use_ssl?
    address() + (port == HTTP.https_default_port ? '' : ":#{port()}")
  else
    address() + (port == HTTP.http_default_port ? '' : ":#{port()}")
  end
end

# File lib/net/http.rb, line 1449
def begin_transport(req)
  if @socket.closed?
    connect
  elsif @last_communicated && @last_communicated + @keep_alive_timeout < Time.now
    D 'Conn close because of keep_alive_timeout'
    @socket.close
    connect
  end

  if not req.response_body_permitted? and @close_on_empty_response
    req['connection'] ||= 'close'
  end

  host = req['host'] || address
  host = $1 if host =~ /(.*):\d+$/
  req.update_uri host, port, use_ssl?

  req['host'] ||= addr_port()
end

# File lib/net/http.rb, line 867
def connect
  if proxy? then
    conn_address = proxy_address
    conn_port    = proxy_port
  else
    conn_address = address
    conn_port    = port
  end

  D "opening connection to #{conn_address}:#{conn_port}..."
  s = Timeout.timeout(@open_timeout, Net::OpenTimeout) {
    TCPSocket.open(conn_address, conn_port, @local_host, @local_port)
  }
  D "opened"
  if use_ssl?
    ssl_parameters = Hash.new
    iv_list = instance_variables
    SSL_IVNAMES.each_with_index do |ivname, i|
      if iv_list.include?(ivname) and
        value = instance_variable_get(ivname)
        ssl_parameters[SSL_ATTRIBUTES[i]] = value if value
      end
    end
    @ssl_context = OpenSSL::SSL::SSLContext.new
    @ssl_context.set_params(ssl_parameters)
    D "starting SSL for #{conn_address}:#{conn_port}..."
    s = OpenSSL::SSL::SSLSocket.new(s, @ssl_context)
    s.sync_close = true
    D "SSL established"
  end
  @socket = BufferedIO.new(s)
  @socket.read_timeout = @read_timeout
  @socket.continue_timeout = @continue_timeout
  @socket.debug_output = @debug_output
  if use_ssl?
    begin
      if proxy?
        buf = "CONNECT #{@address}:#{@port} HTTP/#{HTTPVersion}\r\n"
        buf << "Host: #{@address}:#{@port}\r\n"
        if proxy_user
          credential = ["#{proxy_user}:#{proxy_pass}"].pack('m')
          credential.delete!("\r\n")
          buf << "Proxy-Authorization: Basic #{credential}\r\n"
        end
        buf << "\r\n"
        @socket.write(buf)
        HTTPResponse.read_new(@socket).value
      end
      if @ssl_session and
         Time.now < @ssl_session.time + @ssl_session.timeout
        s.session = @ssl_session if @ssl_session
      end
      # Server Name Indication (SNI) RFC 3546
      s.hostname = @address if s.respond_to? :hostname=
      Timeout.timeout(@open_timeout, Net::OpenTimeout) { s.connect }
      if @ssl_context.verify_mode != OpenSSL::SSL::VERIFY_NONE
        s.post_connection_check(@address)
      end
      @ssl_session = s.session
    rescue => exception
      D "Conn close because of connect error #{exception}"
      @socket.close if @socket and not @socket.closed?
      raise exception
    end
  end
  on_connect
end

# File lib/net/http.rb, line 947
def do_finish
  @started = false
  @socket.close if @socket and not @socket.closed?
  @socket = nil
end

# File lib/net/http.rb, line 861
def do_start
  connect
  @started = true
end

# File lib/net/http.rb, line 1078
def edit_path(path)
  if proxy? and not use_ssl? then
    "http://#{addr_port}#{path}"
  else
    path
  end
end

# File lib/net/http.rb, line 1469
def end_transport(req, res)
  @curr_http_version = res.http_version
  @last_communicated = nil
  if @socket.closed?
    D 'Conn socket closed'
  elsif not res.body and @close_on_empty_response
    D 'Conn close'
    @socket.close
  elsif keep_alive?(req, res)
    D 'Conn keep-alive'
    @last_communicated = Time.now
  else
    D 'Conn close'
    @socket.close
  end
end

# File lib/net/http.rb, line 1486
def keep_alive?(req, res)
  return false if req.connection_close?
  if @curr_http_version <= '1.0'
    res.connection_keep_alive?
  else   # HTTP/1.1 or later
    not res.connection_close?
  end
end

# File lib/net/http.rb, line 936
def on_connect
end

# File lib/net/http.rb, line 1394
def send_entity(path, data, initheader, dest, type, &block)
  res = nil
  request(type.new(path, initheader), data) {|r|
    r.read_body dest, &block
    res = r
  }
  res
end

# File lib/net/http.rb, line 1510
def sspi_auth(req)
  n = Win32::SSPI::NegotiateAuth.new
  req["Proxy-Authorization"] = "Negotiate #{n.get_initial_token}"
  # Some versions of ISA will close the connection if this isn't present.
  req["Connection"] = "Keep-Alive"
  req["Proxy-Connection"] = "Keep-Alive"
  res = transport_request(req)
  authphrase = res["Proxy-Authenticate"]  or return res
  req["Proxy-Authorization"] = "Negotiate #{n.complete_authentication(authphrase)}"
rescue => err
  raise HTTPAuthenticationError.new('HTTP authentication failed', err)
end

# File lib/net/http.rb, line 1495
def sspi_auth?(res)
  return false unless @sspi_enabled
  if res.kind_of?(HTTPProxyAuthenticationRequired) and
      proxy? and res["Proxy-Authenticate"].include?("Negotiate")
    begin
      require 'win32/sspi'
      true
    rescue LoadError
      false
    end
  else
    false
  end
end

# File lib/net/http.rb, line 1405
def transport_request(req)
  count = 0
  begin
    begin_transport req
    res = catch(:response) {
      req.exec @socket, @curr_http_version, edit_path(req.path)
      begin
        res = HTTPResponse.read_new(@socket)
        res.decode_content = req.decode_content
      end while res.kind_of?(HTTPContinue)

      res.uri = req.uri

      res.reading_body(@socket, req.response_body_permitted?) {
        yield res if block_given?
      }
      res
    }
  rescue Net::OpenTimeout
    raise
  rescue Net::ReadTimeout, IOError, EOFError,
         Errno::ECONNRESET, Errno::ECONNABORTED, Errno::EPIPE,
         # avoid a dependency on OpenSSL
         defined?(OpenSSL::SSL) ? OpenSSL::SSL::SSLError : IOError,
         Timeout::Error => exception
    if count == 0 && IDEMPOTENT_METHODS_.include?(req.method)
      count += 1
      @socket.close if @socket and not @socket.closed?
      D "Conn close because of error #{exception}, and retry"
      retry
    end
    D "Conn close because of error #{exception}"
    @socket.close if @socket and not @socket.closed?
    raise
  end

  end_transport req, res
  res
rescue => exception
  D "Conn close because of error #{exception}"
  @socket.close if @socket and not @socket.closed?
  raise exception
end