class WEBrick::CGI::Socket

def <<(data)
  @out_port << data
end

def add_header(hdrname, value)
  unless value.empty?
    @header_part << hdrname << ": " << value << CRLF
  end
end

def addr
  [nil, @server_port, @server_name, @server_addr]
end

def cert
  return nil unless defined?(OpenSSL)
  if pem = @env["SSL_SERVER_CERT"]
    OpenSSL::X509::Certificate.new(pem) unless pem.empty?
  end
end

def cipher
  return nil unless defined?(OpenSSL)
  if cipher = @env["SSL_CIPHER"]
    ret = [ cipher ]
    ret << @env["SSL_PROTOCOL"]
    ret << @env["SSL_CIPHER_USEKEYSIZE"]
    ret << @env["SSL_CIPHER_ALGKEYSIZE"]
    ret
  end
end

def each
  input.each{|line| yield(line) }
end

def eof?
  input.eof?
end

def gets(eol=LF, size=nil)
  input.gets(eol, size)
end

def initialize(config, env, stdin, stdout)
  @config = config
  @env = env
  @header_part = StringIO.new
  @body_part = stdin
  @out_port = stdout
  @out_port.binmode
  @server_addr = @env["SERVER_ADDR"] || "0.0.0.0"
  @server_name = @env["SERVER_NAME"]
  @server_port = @env["SERVER_PORT"]
  @remote_addr = @env["REMOTE_ADDR"]
  @remote_host = @env["REMOTE_HOST"] || @remote_addr
  @remote_port = @env["REMOTE_PORT"] || 0
  begin
    @header_part << request_line << CRLF
    setup_header
    @header_part << CRLF
    @header_part.rewind
  rescue Exception
    raise CGIError, "invalid CGI environment"
  end
end

def input
  @header_part.eof? ? @body_part : @header_part
end

def peer_cert
  return nil unless defined?(OpenSSL)
  if pem = @env["SSL_CLIENT_CERT"]
    OpenSSL::X509::Certificate.new(pem) unless pem.empty?
  end
end

def peer_cert_chain
  return nil unless defined?(OpenSSL)
  if @env["SSL_CLIENT_CERT_CHAIN_0"]
    keys = @env.keys
    certs = keys.sort.collect{|k|
      if /^SSL_CLIENT_CERT_CHAIN_\d+$/ =~ k
        if pem = @env[k]
          OpenSSL::X509::Certificate.new(pem) unless pem.empty?
        end
      end
    }
    certs.compact
  end
end

def peeraddr
  [nil, @remote_port, @remote_host, @remote_addr]
end

def read(size=nil)
  input.read(size)
end

def request_line
  meth = @env["REQUEST_METHOD"] || "GET"
  unless url = @env["REQUEST_URI"]
    url = (@env["SCRIPT_NAME"] || File.expand_path($0)).dup
    url << @env["PATH_INFO"].to_s
    url = WEBrick::HTTPUtils.escape_path(url)
    if query_string = @env["QUERY_STRING"]
      unless query_string.empty?
        url << "?" << query_string
      end
    end
  end
  # we cannot get real HTTP version of client ;)
  httpv = @config[:HTTPVersion]
  return "#{meth} #{url} HTTP/#{httpv}"
end

def setup_header
  @env.each{|key, value|
    case key
    when "CONTENT_TYPE", "CONTENT_LENGTH"
      add_header(key.gsub(/_/, "-"), value)
    when /^HTTP_(.*)/
      add_header($1.gsub(/_/, "-"), value)
    end
  }
end

def write(data)
  @out_port.write(data)
end