IOWA 1.0.0

data/src/iowa/request/ENV.rb

@@ -0,0 +1,80 @@
+require 'iowa/Request'
+require 'iowa/Util'
+
+module Iowa
+
+	class Request
+
+		class ENV < Iowa::Request
+
+			def setup
+				@hostname = ::ENV['SERVER_NAME']
+				@unparsed_uri = @uri = ::ENV['REQUEST_URI']
+				@filename = ::ENV['SCRIPT_FILENAME']
+				@request_time = Time.now.asctime
+				@request_method = ::ENV['REQUEST_METHOD']
+				#@remote_host = ::ENV['REMOTE_HOST']
+				#@remote_addr = ::ENV['REMOTE_ADDR']
+				@headers_only = (@request_method.to_s == 'HEAD') ? true : false
+				@params = {}
+
+				# CGI will give us a hash of the params, but @args is supposed
+				# to be a String.
+				@params = parse_params
+				if @request_method == CPOST
+					@params.each_pair do |k,v|
+						if (v.class.name == 'StringIO') or (v.class.name == 'Tempfile')
+							@params[k] = Iowa::FileData.new(v.original_filename,v.content_type,v.read)
+						end
+					end
+				end
+				@args = @request_method == CPOST ? '' : @params.keys.collect {|i| "#{i}=#{@params[i]}"}.join('&')
+
+				@headers_in = @headers = Iowa::DataTable.new	
+				::ENV.each do |k,v|
+					next unless k.index(CHTTP_) == 0
+					key = k.dup
+					key.sub!(/^HTTP_/,'')
+					key.gsub!(/([^_]+_?)/) do |s|
+						s.capitalize
+					end
+					@headers[key] = v
+				end
+				@content_type = nil
+				@content_encoding = nil
+				@content_languages = nil
+			end
+
+			def initialize(request=nil)
+				setup
+			end
+
+			def parse_params
+				req_meth = @request_method
+
+# Add code supporting multipart.
+
+				if req_meth == CGET or req_meth == CHEAD
+					query = ::ENV[CQUERY_STRING] or C_empty
+				else
+					$stdin.binmode if defined? $stdin.binmode
+					query = $stdin.read(Integer(::ENV[Iowa::CCONTENT_LENGTH])) or C_empty
+				end
+
+				params = Hash.new([].freeze)
+				query.split(/[&;]/n).each do |pairs|
+					key, value = pairs.split(C_equal,2).collect{|v| Util::unescape(v) }
+					if params.has_key?(key)
+						params[key].push(value)
+					else
+						params[key] = [value]
+					end
+				end
+				params
+			end
+
+		end
+	end
+end
+
+Iowa::Request.Type = Iowa::Request::ENV

data/src/iowa/request/FCGI.rb

@@ -0,0 +1,68 @@
+require 'iowa/Request'
+
+module Iowa
+
+	class Request
+
+		class FCGI < Iowa::Request
+
+			def setup(request)
+				env = request.env
+				@hostname = env['SERVER_NAME']
+				@unparsed_uri = @uri = env['REQUEST_URI']
+				@filename = env['SCRIPT_FILENAME']
+				@request_time = Time.now.asctime
+				@request_method = env['REQUEST_METHOD']
+				@remote_host = env['REMOTE_HOST']
+				@remote_addr = env['REMOTE_ADDR']
+				@headers_only = (@request_method.to_s == 'HEAD') ? true : false
+				@params = {}
+
+				if @args = env['QUERY_STRING'] != ''
+					@args.split(/[&;]/).each do |a|
+						k,v = a.split('=',2).collect{|x| Iowa::Util.unescape(x)}
+						if @params.has_key? k
+							@params[k] += "\0" + (v or "")
+						else
+							@params[k] = (v or "")
+						end
+					end
+				end
+
+				if m = MIMERegexp.match(env['CONTENT_TYPE'])
+					boundary = m[1]
+					@params = read_multipart(boundary,env['CONTENT_LENGTH'],request.in,env['USER_AGENT'])
+				else
+					@content = request.in.read
+					@content = '' if @content.nil?
+					@content.split(/[&;]/).each do |x|
+						key, val = x.split(/=/,2).collect{|x| Iowa::Util.unescape(x)}
+						if @params.include?(key)
+							@params[key] += "\0" + (val or "")
+						else
+							@params[key] = (val or "")
+						end
+					end
+				end
+				@headers = Iowa::DataTable.new
+				@params.each do |k,v|
+					next unless k =~ /^HTTP_/
+					key = k.dup
+					key.sub!(/^HTTP_/,'')
+					key.gsub!(/([^_]+_?)/) do |s|
+						s.gsub('_','-').capitalize
+					end
+					@headers.set(key,@params[key])
+				end
+				@content_type = nil
+				@content_encoding = nil
+				@content_languages = nil
+			end
+
+			def initialize(request=nil)
+				setup(request)
+			end
+
+		end
+	end
+end

data/src/iowa/request/HTTPMachine.rb

@@ -0,0 +1,75 @@
+require 'iowa/Request'
+require 'iowa/Util'
+
+module Iowa
+
+	class Request
+
+		class HTTPMachine < Iowa::Request
+
+			C_nul = "\000"
+
+			def setup(rqdata)
+				raw_headers = rqdata.last
+				@unparsed_uri = @uri = rqdata[1]
+				@request_time = Time.now.asctime
+				@request_method = rqdata.first
+				@headers_only = (@request_method.to_s == 'HEAD') ? true : false
+				@params = {}
+
+				# CGI will give us a hash of the params, but @args is supposed
+				# to be a String.
+				@params = parse_params
+				if @request_method == CPOST
+					@params.each_pair do |k,v|
+						if (StringIO === v) or (Tempfile === v)
+							@params[k] = Iowa::FileData.new(v.original_filename,v.content_type,v.read)
+						end
+					end
+				end
+				@args = @request_method == CPOST ? '' : @params.keys.collect {|i| "#{i}=#{@params[i]}"}.join('&')
+
+				@headers_in = @headers = Iowa::DataTable.new	
+				raw_headers.split(C_nul).each do |hdr|
+					k,v = hdr.split( /:\s*/, 2)
+					@headers[k.upcase] = v
+				end
+
+				@content_type = nil
+				@content_encoding = nil
+				@content_languages = nil
+			end
+
+			def initialize(raw_headers = '')
+				setup(raw_headers)
+			end
+
+			def parse_params
+				req_meth = @request_method
+
+# Add code supporting multipart.
+
+				if req_meth == CGET or req_meth == CHEAD
+					query = ::ENV[CQUERY_STRING] or C_empty
+				else
+					$stdin.binmode if defined? $stdin.binmode
+					query = $stdin.read(Integer(::ENV[Iowa::CCONTENT_LENGTH])) or C_empty
+				end
+
+				params = Hash.new([].freeze)
+				query.split(/[&;]/n).each do |pairs|
+					key, value = pairs.split(C_equal,2).collect{|v| Util::unescape(v) }
+					if params.has_key?(key)
+						params[key].push(value)
+					else
+						params[key] = [value]
+					end
+				end
+				params
+			end
+
+		end
+	end
+end
+
+Iowa::Request.Type = Iowa::Request::ENV

data/src/iowa/request/Mongrel.rb

@@ -0,0 +1,68 @@
+require 'iowa/Request'
+require 'time'
+
+module Iowa
+
+	class Request
+
+		# This class encapsulates a Mongrel request.  Mongrel just dumps all
+		# of the HTTP headers into the environment, CGI-style, so the ones
+		# that are necessary are extracted from the environment, as are the
+		# HTTP headers received from the web server.  The params are parsed
+		# out of the request body.  Will handle mime multipart.
+		class Mongrel < Iowa::Request
+
+			def setup(request)
+				params = request.params
+				hn = params[CX_FORWARDED_HOST] || params[CHTTP_HOST]
+				@hostname,@port = hn.split(/:/)
+				@hostname = params[CSERVER_NAME]
+				@unparsed_uri = params[::Mongrel::Const::REQUEST_URI]
+				@uri = params[::Mongrel::Const::PATH_INFO]
+				@request_time = Time.now.httpdate
+				@request_method = params[::Mongrel::Const::REQUEST_METHOD]
+				#@remote_host = params[::Mongrel::Const::REMOTE_HOST]
+				#@remote_addr = params[::Mongrel::Const::REMOTE_ADDR]
+				@header_only =  (@request_method == CHEAD) ? true : false
+				@args = params[CQUERY_STRING]
+
+				@headers_in = @headers = Iowa::DataTable.new	
+				params.each do |k,v|
+					next unless k =~ /^HTTP_/
+					key = k.dup
+					key.sub!(/^HTTP_/,'')
+					key.gsub!(/([^_]+_?)/) do |s|
+						s.gsub('_','-').capitalize
+					end
+					@headers[key] = v
+				end
+				
+				@params = {}
+				if m = MIMERegexp.match(@headers[CContent_type])
+					boundary = m[1]
+					@params = read_multipart(boundary,@headers[CContent_length],request.body,@headers[CUser_agent])
+				else
+					request.body.each {|line| parse_params(line)}
+				end
+				parse_params(@args) if @args
+				
+				@status_line = nil
+				@headers_out = Iowa::DataTable.new
+				@content_type = nil
+				@content_encoding = nil
+				@content_languages = nil
+				@content = ''
+			end
+
+			def initialize(request=nil)
+
+				@calculate_runtime = false
+
+				setup(request)
+			end
+
+		end
+	end
+end
+
+Iowa::Request.Type = Iowa::Request::Mongrel

data/src/iowa/request/WEBrick.rb

@@ -0,0 +1,48 @@
+require 'iowa/Request'
+
+module Iowa
+
+	class Request
+
+		class WEBrick < Iowa::Request
+
+			def setup(request)
+				@hostname = request.host
+				@unparsed_uri = request.unparsed_uri
+				#@uri = request.path.to_s
+				@uri = request.path_info
+				@request_time = request.request_time
+				@request_method = request.request_method
+				@remote_host = request.peeraddr[2]
+				@remote_addr = request.peeraddr[3].split(':')[3]
+				@filename = request.path
+				@header_only =  (@request_method.to_s == 'HEAD') ? true : false
+				@args = request.query_string
+				@params = {}
+				request.query.each do |k,v|
+					@params[ k ] = v.list.length > 1 ? v.list.join("\0") : v
+				end
+				@headers = @headers_in = Iowa::DataTable.new	
+				request.header.each {|k,v| @headers_in[k] = v}
+
+				@status_line = nil
+				@headers_out = Iowa::DataTable.new
+				@content_type = nil
+				@content_encoding = nil
+				@content_languages = nil
+				@content = ''
+			end
+
+			def initialize(request)
+				@calculate_runtime = false
+
+				setup(request)
+			end
+
+		end
+
+	end
+
+end
+
+Iowa::Request.Type = Iowa::Request::WEBrick

data/src/iowa/version.rb

@@ -0,0 +1,3 @@
+module Iowa
+  VERSION = "1.0.0"
+end

data/src/iowa/webrick/HTTPServer.rb

@@ -0,0 +1,43 @@
+require 'iowa'
+require 'webrick'
+require 'iowa/webrick/WEBrickServlet'
+
+module Iowa
+	class HTTPServer < WEBrick::HTTPServer
+	
+		def initialize(options)
+			@mapfile = options.has_key?(:MapFile) ? options[:MapFile] : nil
+			@mapfile_mtime = 0
+			@mappings = {}
+			Iowa.app.initialLoad
+			@iowa_servlet = Iowa::WEBrickServlet
+			super
+		end
+
+		def service(req,res)
+			if req.unparsed_uri == "*"
+				if req.request_method == "OPTIONS"
+					do_OPTIONS(req, res)
+					raise ::HTTPStatus::OK
+				end
+				raise ::HTTPStatus::NotFound, "`#{req.unparsed_uri}' not found."
+			end
+
+			servlet = options = script_name = path_info = nil
+			if Iowa.app.class.Dispatcher.handleRequest?(req.path)
+				servlet = @iowa_servlet
+				script_name = 'Iowa'
+				path_info = req.path
+			else
+				servlet, options, script_name, path_info = search_servlet(req.path)
+			end
+			raise ::HTTPStatus::NotFound, "`#{req.path}' not found." unless servlet
+			req.script_name = script_name
+			req.path_info = path_info
+			si = servlet.get_instance(self, *options)
+			@logger.debug(format("%s is invoked.", si.class.name))
+			si.service(req, res)
+		end
+
+	end
+end

data/src/iowa/webrick/WEBrickServlet.rb

@@ -0,0 +1,28 @@
+require 'iowa'
+require 'webrick'
+
+# This is a _basic_ level of support.  I'm sure that I need to add things to
+# deal with multipart/form-data forms for file uploads, and for providing some
+# of the other information that is encapsulated in an Iowa::Request.
+
+class Iowa::Request
+	alias old_init initialize
+	def initialize
+		@headers_in = Iowa::R::Table.new
+		@headers_out = Iowa::R::Table.new
+		@params = {}
+	end
+end
+
+class Iowa::WEBrickServlet < WEBrick::HTTPServlet::AbstractServlet
+
+	#####
+	#// This is the only magic.  It takes the place of the
+	#// Iowa::handleConnection() method that is used when running in a
+	#// non-webrick mode.
+	#####
+	def do_GET request, response
+		Iowa::handleConnection(request,response)
+	end
+	alias do_POST do_GET
+end

data/src/iowa_fcgi_handler.rb

@@ -0,0 +1,101 @@
+#!/usr/bin/env ruby
+
+#FCGI_PURE_RUBY = true
+
+require 'yaml'
+require 'fcgi'
+require 'iowa/Client'
+require 'getoptlong'
+require 'logger'
+
+matchURI = Regexp.new('(/*(?:[^/]*/*)*?)(\w+\-\w+\-\w+\.\w+\.[\w\.]+)*(-\d+)?$')
+matchLoc = Regexp.new('^(..)(..)(..)(..)(....)')
+
+def read_configuration(file)
+	File.open(file) {|cf| YAML::load(cf)}
+end
+
+def parse_configuration
+	@config = read_configuration(@config_file)
+	@socket_list = []
+	@config.each do |k,v|
+		next unless v
+		@socket_list.push("#{k}:#{v}")
+	end
+	@config_length = @config.length
+end
+
+def open_logfile(lf)
+	Logger.new(lf,'daily')
+end
+
+# Command line switches:
+#  -f FILENAME
+#    Specifies a configuration file.
+#  -v Print the version and exit.
+#  -l The process should log to this file.
+
+options = GetoptLong.new(
+	['--file', '-f', GetoptLong::REQUIRED_ARGUMENT],
+	['--version', '-v', GetoptLong::NO_ARGUMENT],
+	['--logfile','-l', GetoptLong::REQUIRED_ARGUMENT])
+
+@logfile = nil
+@config = {}
+options.each do |opt, arg|
+	case opt
+	when '--file'
+		# Configuration file format:
+		#  IP/port: [true|false]
+		#  num: /path/to/socket
+		@config_file = arg
+		parse_configuration
+	when '--version'
+		puts "v0.1 IOWA FCGI Muxxer; 02 March 2005"
+		exit
+	when '--logfile'
+		@logfile = open_logfile(arg)
+	end
+end
+	
+FCGI.each do |request|
+	env = request.env
+	m = matchURI.match(env['REQUEST_URI'])
+	ip = nil
+	if m && m[2].to_s != ''
+		l = matchLoc.match(m[2])
+		if l
+			ip = "#{l[0].hex}.#{l[1].hex}.#{l[2].hex}.#{l[3].hex}"
+			port = l[4].hex
+		end
+	end
+	
+	if ip == '0.0.0.0' && @config.has_key?(port)
+		# unix socket
+		iowa_socket = @config[port]
+	elsif ip && @config.has_key?("#{ip}/#{port}")
+		# tcp socket
+		iowa_socket = "#{ip}:#{port}"
+	else
+		# pick a random socket.
+		id,sock = @socket_list[rand(@config_length)].split(/:/,2)
+		if m = /\//.match(id)
+			iowa_socket = id.sub(/\//,':')
+		else
+			iowa_socket = sock
+		end
+	end
+	
+	if @logfile
+		@logfile.info("Dispatching request to #{iowa_socket}")
+	end
+
+	begin
+		client = Iowa::Client.new(iowa_socket)
+		client.handle_request({:request => request, :ios => request.out})
+		request.finish
+	rescue Exception => e
+		@logfile.error("Error: #{e} -- #{e.backtrace.inspect}")
+		request.finish
+	end
+end