mirror of
				https://github.com/ruby/ruby.git
				synced 2022-11-09 12:17:21 -05:00 
			
		
		
		
	 d7cc39658c
			
		
	
	
		d7cc39658c
		
	
	
	
	
		
			
			(WEBrick::HTTPServlet::CGIHandler#do_GET): set binary mode for tempfile. http://pc12.2ch.net/test/read.cgi/tech/1265467681/286 git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@26887 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
		
			
				
	
	
		
			108 lines
		
	
	
	
		
			3.3 KiB
		
	
	
	
		
			Ruby
		
	
	
	
	
	
			
		
		
	
	
			108 lines
		
	
	
	
		
			3.3 KiB
		
	
	
	
		
			Ruby
		
	
	
	
	
	
| #
 | |
| # cgihandler.rb -- CGIHandler Class
 | |
| #
 | |
| # Author: IPR -- Internet Programming with Ruby -- writers
 | |
| # Copyright (c) 2001 TAKAHASHI Masayoshi, GOTOU Yuuzou
 | |
| # Copyright (c) 2002 Internet Programming with Ruby writers. All rights
 | |
| # reserved.
 | |
| #
 | |
| # $IPR: cgihandler.rb,v 1.27 2003/03/21 19:56:01 gotoyuzo Exp $
 | |
| 
 | |
| require 'rbconfig'
 | |
| require 'tempfile'
 | |
| require 'webrick/config'
 | |
| require 'webrick/httpservlet/abstract'
 | |
| 
 | |
| module WEBrick
 | |
|   module HTTPServlet
 | |
| 
 | |
|     class CGIHandler < AbstractServlet
 | |
|       Ruby = RbConfig.ruby
 | |
|       CGIRunner = "\"#{Ruby}\" \"#{WEBrick::Config::LIBDIR}/httpservlet/cgi_runner.rb\""
 | |
| 
 | |
|       def initialize(server, name)
 | |
|         super(server, name)
 | |
|         @script_filename = name
 | |
|         @tempdir = server[:TempDir]
 | |
|         @cgicmd = "#{CGIRunner} #{server[:CGIInterpreter]}"
 | |
|       end
 | |
| 
 | |
|       def do_GET(req, res)
 | |
|         data = nil
 | |
|         status = -1
 | |
| 
 | |
|         cgi_in = IO::popen(@cgicmd, "wb")
 | |
|         cgi_out = Tempfile.new("webrick.cgiout.", @tempdir, mode: IO::BINARY)
 | |
|         cgi_out.set_encoding("ASCII-8BIT")
 | |
|         cgi_err = Tempfile.new("webrick.cgierr.", @tempdir, mode: IO::BINARY)
 | |
|         cgi_err.set_encoding("ASCII-8BIT")
 | |
|         begin
 | |
|           cgi_in.sync = true
 | |
|           meta = req.meta_vars
 | |
|           meta["SCRIPT_FILENAME"] = @script_filename
 | |
|           meta["PATH"] = @config[:CGIPathEnv]
 | |
|           if /mswin|bccwin|mingw/ =~ RUBY_PLATFORM
 | |
|             meta["SystemRoot"] = ENV["SystemRoot"]
 | |
|           end
 | |
|           dump = Marshal.dump(meta)
 | |
| 
 | |
|           cgi_in.write("%8d" % cgi_out.path.bytesize)
 | |
|           cgi_in.write(cgi_out.path)
 | |
|           cgi_in.write("%8d" % cgi_err.path.bytesize)
 | |
|           cgi_in.write(cgi_err.path)
 | |
|           cgi_in.write("%8d" % dump.bytesize)
 | |
|           cgi_in.write(dump)
 | |
| 
 | |
|           if req.body and req.body.bytesize > 0
 | |
|             cgi_in.write(req.body)
 | |
|           end
 | |
|         ensure
 | |
|           cgi_in.close
 | |
|           status = $?.exitstatus
 | |
|           sleep 0.1 if /mswin|bccwin|mingw/ =~ RUBY_PLATFORM
 | |
|           data = cgi_out.read
 | |
|           cgi_out.close(true)
 | |
|           if errmsg = cgi_err.read
 | |
|             if errmsg.bytesize > 0
 | |
|               @logger.error("CGIHandler: #{@script_filename}:\n" + errmsg)
 | |
|             end
 | |
|           end
 | |
|           cgi_err.close(true)
 | |
|         end
 | |
| 
 | |
|         if status != 0
 | |
|           @logger.error("CGIHandler: #{@script_filename} exit with #{status}")
 | |
|         end
 | |
| 
 | |
|         data = "" unless data
 | |
|         raw_header, body = data.split(/^[\xd\xa]+/, 2)
 | |
|         raise HTTPStatus::InternalServerError,
 | |
|           "Premature end of script headers: #{@script_filename}" if body.nil?
 | |
| 
 | |
|         begin
 | |
|           header = HTTPUtils::parse_header(raw_header)
 | |
|           if /^(\d+)/ =~ header['status'][0]
 | |
|             res.status = $1.to_i
 | |
|             header.delete('status')
 | |
|           end
 | |
|           if header.has_key?('location')
 | |
|             # RFC 3875 6.2.3, 6.2.4
 | |
|             res.status = 302 unless (300...400) === res.status
 | |
|           end
 | |
|           if header.has_key?('set-cookie')
 | |
|             header['set-cookie'].each{|k|
 | |
|               res.cookies << Cookie.parse_set_cookie(k)
 | |
|             }
 | |
|             header.delete('set-cookie')
 | |
|           end
 | |
|           header.each{|key, val| res[key] = val.join(", ") }
 | |
|         rescue => ex
 | |
|           raise HTTPStatus::InternalServerError, ex.message
 | |
|         end
 | |
|         res.body = body
 | |
|       end
 | |
|       alias do_POST do_GET
 | |
|     end
 | |
| 
 | |
|   end
 | |
| end
 |