Method: Nitro::Cgi.parse_multipart

Defined in:

lib/nitro/cgi.rb

.parse_multipart(context, boundary) ⇒ Object

Parse a multipart request. Adapted from Ruby’s cgi.rb – TODO: optimize and rationalize this. ++

Raises:

• (EOFError)

# File 'lib/nitro/cgi.rb', line 210

def self.parse_multipart(context, boundary)
  input = context.in
  content_length = context.content_length
  env_table = context.env
  
  params = Hash.new()
  boundary = "--" + boundary
  quoted_boundary = Regexp.quote(boundary, "n")
  buf = ""
  boundary_end=""
  
  # start multipart/form-data
  input.binmode if defined? input.binmode
  boundary_size = boundary.size + EOL.size
  content_length -= boundary_size
  status = input.read(boundary_size)

  if nil == status
    raise EOFError, "no content body"
  elsif boundary + EOL != status
    raise EOFError, "bad content body"
  end

  loop do
    head = nil

    if 10240 < content_length
      body = Tempfile.new("CGI")
    else
      begin
        require "stringio"
        body = StringIO.new
      rescue LoadError
        body = Tempfile.new("CGI")
      end
    end
    body.binmode if defined? body.binmode

    until head and /#{quoted_boundary}(?:#{EOL}|--)/n.match(buf)

      if (not head) and /#{EOL}#{EOL}/n.match(buf)
        buf = buf.sub(/\A((?:.|\n)*?#{EOL})#{EOL}/n) do
          head = $1.dup
          ""
        end
        next
      end

      if head and ( (EOL + boundary + EOL).size < buf.size )
        body.print buf[0 ... (buf.size - (EOL + boundary + EOL).size)]
        buf[0 ... (buf.size - (EOL + boundary + EOL).size)] = ""
      end

      c = if Cgi.buffer_size < content_length
            input.read(Cgi.buffer_size)
          else
            input.read(content_length)
          end
      if c.nil? || c.empty?
        raise EOFError, "bad content body"
      end
      buf.concat(c)
      content_length -= c.size
    end

    buf = buf.sub(/\A((?:.|\n)*?)(?:[\r\n]{1,2})?#{quoted_boundary}([\r\n]{1,2}|--)/n) do
      body.print $1
      if "--" == $2
        content_length = -1
      end
      boundary_end = $2.dup
      ""
    end

    body.rewind

    /Content-Disposition:.* filename="?([^\";]*)"?/ni.match(head)
    
    filename = ($1 or "")

    if /Mac/ni.match(env_table['HTTP_USER_AGENT']) and
        /Mozilla/ni.match(env_table['HTTP_USER_AGENT']) and
        (not /MSIE/ni.match(env_table['HTTP_USER_AGENT']))
      filename = CGI::unescape(filename)
    end
    
    /Content-Type: (.*)/ni.match(head)
    content_type = ($1 or "")

    (class << body; self; end).class_eval do
      alias local_path path
      define_method(:original_filename) { filename.dup.taint }
      define_method(:content_type) { content_type.dup.taint }

      # gmosx: this hides the performance hit!!
      define_method(:to_s) { str = read; rewind; return str}
    end

    /Content-Disposition:.* name="?([^\";]*)"?/ni.match(head)
    name = $1.dup
    
    params = self.structure_param(params, name, body)

    break if buf.size == 0
    break if content_length === -1
  end
  raise EOFError, "bad boundary end of body" unless boundary_end =~ /--/

  return params
end