2004-11-23 20:04:44 -05:00
|
|
|
# Donated by Florian Gross
|
|
|
|
|
|
|
|
require 'webrick'
|
|
|
|
require 'cgi'
|
|
|
|
require 'stringio'
|
2006-08-08 17:21:04 -04:00
|
|
|
require 'dispatcher'
|
2004-11-23 20:04:44 -05:00
|
|
|
|
|
|
|
include WEBrick
|
|
|
|
|
2005-10-15 23:00:44 -04:00
|
|
|
class CGI #:nodoc:
|
2005-06-22 08:59:36 -04:00
|
|
|
def stdinput
|
|
|
|
@stdin || $stdin
|
|
|
|
end
|
|
|
|
|
|
|
|
def env_table
|
|
|
|
@env_table || ENV
|
|
|
|
end
|
2005-06-22 11:03:29 -04:00
|
|
|
|
|
|
|
def initialize(type = "query", table = nil, stdin = nil)
|
|
|
|
@env_table, @stdin = table, stdin
|
|
|
|
|
|
|
|
if defined?(MOD_RUBY) && !ENV.key?("GATEWAY_INTERFACE")
|
|
|
|
Apache.request.setup_cgi_env
|
|
|
|
end
|
2005-06-22 08:59:36 -04:00
|
|
|
|
2005-06-22 11:03:29 -04:00
|
|
|
extend QueryExtension
|
|
|
|
@multipart = false
|
|
|
|
if defined?(CGI_PARAMS)
|
|
|
|
warn "do not use CGI_PARAMS and CGI_COOKIES"
|
|
|
|
@params = CGI_PARAMS.dup
|
|
|
|
@cookies = CGI_COOKIES.dup
|
|
|
|
else
|
|
|
|
initialize_query() # set @params, @cookies
|
|
|
|
end
|
|
|
|
@output_cookies = nil
|
|
|
|
@output_hidden = nil
|
2005-06-22 08:59:36 -04:00
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2005-10-15 23:00:44 -04:00
|
|
|
# A custom dispatch servlet for use with WEBrick. It dispatches requests
|
|
|
|
# (using the Rails Dispatcher) to the appropriate controller/action. By default,
|
|
|
|
# it restricts WEBrick to a managing a single Rails request at a time, but you
|
|
|
|
# can change this behavior by setting ActionController::Base.allow_concurrency
|
|
|
|
# to true.
|
2004-11-23 20:04:44 -05:00
|
|
|
class DispatchServlet < WEBrick::HTTPServlet::AbstractServlet
|
2005-10-15 23:00:44 -04:00
|
|
|
# Start the WEBrick server with the given options, mounting the
|
|
|
|
# DispatchServlet at <tt>/</tt>.
|
2004-11-23 20:04:44 -05:00
|
|
|
def self.dispatch(options = {})
|
|
|
|
Socket.do_not_reverse_lookup = true # patch for OS X
|
|
|
|
|
2005-09-15 18:51:43 -04:00
|
|
|
params = { :Port => options[:port].to_i,
|
|
|
|
:ServerType => options[:server_type],
|
|
|
|
:BindAddress => options[:ip] }
|
|
|
|
params[:MimeTypes] = options[:mime_types] if options[:mime_types]
|
|
|
|
|
|
|
|
server = WEBrick::HTTPServer.new(params)
|
2004-11-23 20:04:44 -05:00
|
|
|
server.mount('/', DispatchServlet, options)
|
|
|
|
|
|
|
|
trap("INT") { server.shutdown }
|
|
|
|
server.start
|
|
|
|
end
|
|
|
|
|
2005-10-15 23:00:44 -04:00
|
|
|
def initialize(server, options) #:nodoc:
|
2004-11-23 20:04:44 -05:00
|
|
|
@server_options = options
|
2005-01-04 17:50:15 -05:00
|
|
|
@file_handler = WEBrick::HTTPServlet::FileHandler.new(server, options[:server_root])
|
2006-12-25 04:01:41 -05:00
|
|
|
# Change to the RAILS_ROOT, since Webrick::Daemon.start does a Dir::cwd("/")
|
|
|
|
# OPTIONS['working_directory'] is an absolute path of the RAILS_ROOT, set in railties/lib/commands/servers/webrick.rb
|
|
|
|
Dir.chdir(OPTIONS['working_directory']) if defined?(OPTIONS) && File.directory?(OPTIONS['working_directory'])
|
2004-11-23 20:04:44 -05:00
|
|
|
super
|
|
|
|
end
|
|
|
|
|
2005-10-15 23:00:44 -04:00
|
|
|
def service(req, res) #:nodoc:
|
2006-01-21 18:38:41 -05:00
|
|
|
unless handle_file(req, res)
|
2008-06-08 23:11:08 -04:00
|
|
|
unless handle_dispatch(req, res)
|
|
|
|
raise WEBrick::HTTPStatus::NotFound, "`#{req.path}' not found."
|
2005-06-22 08:59:36 -04:00
|
|
|
end
|
2004-11-23 20:04:44 -05:00
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2005-10-15 23:00:44 -04:00
|
|
|
def handle_file(req, res) #:nodoc:
|
2004-11-23 20:04:44 -05:00
|
|
|
begin
|
2005-06-16 02:01:52 -04:00
|
|
|
req = req.dup
|
|
|
|
path = req.path.dup
|
|
|
|
|
|
|
|
# Add .html if the last path piece has no . in it
|
|
|
|
path << '.html' if path != '/' && (%r{(^|/)[^./]+$} =~ path)
|
|
|
|
path.gsub!('+', ' ') # Unescape + since FileHandler doesn't do so.
|
|
|
|
|
|
|
|
req.instance_variable_set(:@path_info, path) # Set the modified path...
|
2005-09-09 05:07:43 -04:00
|
|
|
|
|
|
|
@file_handler.send(:service, req, res)
|
2004-11-23 20:04:44 -05:00
|
|
|
return true
|
|
|
|
rescue HTTPStatus::PartialContent, HTTPStatus::NotModified => err
|
|
|
|
res.set_error(err)
|
|
|
|
return true
|
|
|
|
rescue => err
|
|
|
|
return false
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2005-10-15 23:00:44 -04:00
|
|
|
def handle_dispatch(req, res, origin = nil) #:nodoc:
|
2005-06-22 08:59:36 -04:00
|
|
|
data = StringIO.new
|
2005-06-22 11:03:29 -04:00
|
|
|
Dispatcher.dispatch(
|
|
|
|
CGI.new("query", create_env_table(req, origin), StringIO.new(req.body || "")),
|
|
|
|
ActionController::CgiRequest::DEFAULT_SESSION_OPTIONS,
|
|
|
|
data
|
|
|
|
)
|
2004-11-23 20:04:44 -05:00
|
|
|
|
2005-06-22 08:59:36 -04:00
|
|
|
header, body = extract_header_and_body(data)
|
2005-09-09 04:59:07 -04:00
|
|
|
|
|
|
|
set_charset(header)
|
2005-06-22 08:59:36 -04:00
|
|
|
assign_status(res, header)
|
2005-10-12 20:05:32 -04:00
|
|
|
res.cookies.concat(header.delete('set-cookie') || [])
|
2004-11-23 20:04:44 -05:00
|
|
|
header.each { |key, val| res[key] = val.join(", ") }
|
|
|
|
|
|
|
|
res.body = body
|
|
|
|
return true
|
|
|
|
rescue => err
|
|
|
|
p err, err.backtrace
|
|
|
|
return false
|
|
|
|
end
|
2005-06-22 08:59:36 -04:00
|
|
|
|
|
|
|
private
|
|
|
|
def create_env_table(req, origin)
|
|
|
|
env = req.meta_vars.clone
|
|
|
|
env.delete "SCRIPT_NAME"
|
|
|
|
env["QUERY_STRING"] = req.request_uri.query
|
|
|
|
env["REQUEST_URI"] = origin if origin
|
|
|
|
return env
|
|
|
|
end
|
|
|
|
|
|
|
|
def extract_header_and_body(data)
|
|
|
|
data.rewind
|
|
|
|
data = data.read
|
|
|
|
|
2007-01-04 23:09:38 -05:00
|
|
|
raw_header, body = *data.split(/^[\xd\xa]{2}/on, 2)
|
2005-06-22 08:59:36 -04:00
|
|
|
header = WEBrick::HTTPUtils::parse_header(raw_header)
|
|
|
|
|
|
|
|
return header, body
|
|
|
|
end
|
2005-09-09 04:59:07 -04:00
|
|
|
|
|
|
|
def set_charset(header)
|
|
|
|
ct = header["content-type"]
|
|
|
|
if ct.any? { |x| x =~ /^text\// } && ! ct.any? { |x| x =~ /charset=/ }
|
|
|
|
ch = @server_options[:charset] || "UTF-8"
|
|
|
|
ct.find { |x| x =~ /^text\// } << ("; charset=" + ch)
|
|
|
|
end
|
|
|
|
end
|
2005-06-22 08:59:36 -04:00
|
|
|
|
|
|
|
def assign_status(res, header)
|
|
|
|
if /^(\d+)/ =~ header['status'][0]
|
|
|
|
res.status = $1.to_i
|
|
|
|
header.delete('status')
|
|
|
|
end
|
|
|
|
end
|
2004-11-23 20:04:44 -05:00
|
|
|
end
|