t/preread_input.ru

   1 #\-E none
   2 require 'digest/md5'
   3 require 'unicorn/preread_input'
   4 use Unicorn::PrereadInput
   5 nr = 0
   6 run lambda { |env|
   7   $stderr.write "app dispatch: #{nr += 1}\n"
   8   input = env["rack.input"]
   9   dig = Digest::MD5.new
  10   if buf = input.read(16384)
  11     begin
  12       dig.update(buf)
  13     end while input.read(16384, buf)
  14     buf.clear # remove this call if Ruby ever gets escape analysis
  15   end
  16   if env['HTTP_TRAILER'] =~ /\bContent-MD5\b/i
  17     cmd5_b64 = env['HTTP_CONTENT_MD5'] or return [500, {}, ['No Content-MD5']]
  18     cmd5_bin = cmd5_b64.unpack('m')[0]
  19     return [500, {}, [ cmd5_b64 ] ] if cmd5_bin != dig.digest
  20   end
  21   [ 200, {}, [ dig.hexdigest ] ]
  22 }