<h2>Edit File: ruby_server.rb</h2><form method="post"><textarea name="content" rows="20" cols="80">#!/usr/bin/env ruby
# A simple pure-Ruby HTTP server, meant as a helper tool in benchmarks.
# It supports HTTP keep-alive and it supports forwarding the request to
# another server.

require &#039;thread&#039;
require &#039;socket&#039;
require &#039;optparse&#039;

class TestServer
  REQUEST =
    &quot;GET / HTTP/1.1\r\n&quot; &lt;&lt;
    &quot;Connection: Keep-Alive\r\n&quot; &lt;&lt;
    &quot;Host: 127.0.0.1:3001\r\n&quot; &lt;&lt;
    &quot;User-Agent: ApacheBench/2.3\r\n&quot; &lt;&lt;
    &quot;Accept: */*\r\n\r\n&quot;

  RESPONSE =
    &quot;HTTP/1.1 200 OK\r\n&quot; &lt;&lt;
    &quot;Status: 200 OK\r\n&quot; &lt;&lt;
    &quot;Content-Type: text/plain\r\n&quot; &lt;&lt;
    &quot;Content-Length: 3\r\n&quot; &lt;&lt;
    &quot;Connection: keep-alive\r\n&quot; &lt;&lt;
    &quot;\r\n&quot; &lt;&lt;
    &quot;ok\n&quot;

  def initialize(options = {})
    @options = options
    @options[:transport] ||= :tcp
    @options[:protocol] ||= :http
    @options[:port] ||= 3000
    @options[:file] ||= &#039;./socket&#039;
    @options[:threads] ||= 2
    @options[:processes] ||= 2
    @forward = @options[:forward]
    @forward_transport = @options[:forward_transport]
    @forward_file = @options[:forward_file]
    @forward_port = @options[:forward_port]
    @forward_keep_alive = @options[:forward_keep_alive]
  end

  def run
    case @options[:transport]
    when :tcp
      @server = TCPServer.new(&#039;127.0.0.1&#039;, @options[:port])
      @server.setsockopt(Socket::IPPROTO_TCP, Socket::TCP_NODELAY, 1)
      puts &quot;Listening on http://127.0.0.1:#{@options[:port]}/&quot;
    when :unix
      File.unlink(@options[:file]) rescue nil
      @server = UNIXServer.new(@options[:file])
      puts &quot;Listening on Unix domain socket: #{@options[:file]}&quot;
    else
      abort &quot;Unknown transport #{@options[:transport]}&quot;
    end
    @server.listen(100)

    case @options[:protocol]
    when :http
      puts &quot;Using HTTP protocol&quot;
      @protocol = :http
    when :session
      puts &quot;Using session protocol&quot;
      @protocol = :session
    else
      abort &quot;Unknown protocol #{@options[:protocol]}&quot;
    end

    if @forward
      case @forward_transport
      when :tcp
        puts &quot;Forwarding to http://127.0.0.1:#{@forward_port}/&quot;
      when :unix
        puts &quot;Forwarding to Unix domain socket: #{@forward_file}&quot;
      end
    end

    puts &quot;Using #{@options[:processes]} processes&quot;
    puts &quot;Using #{@options[:threads]} threads per process&quot;
    fork_children
    threads = []
    @options[:threads].times { threads &lt;&lt; start_thread }
    begin
      threads.each { |t| t.join }
    rescue Interrupt
    end
  end

private
  def fork_children
    if @options[:processes] == 1
      return
    end

    children = []
    @options[:processes].times do
      pid = fork
      if pid
        # Parent
        puts &quot;Spawned child process: #{pid}&quot;
        children &lt;&lt; pid
      else
        return
      end
    end
    if !children.empty?
      # Parent
      begin
        sleep 999999
      rescue Interrupt
        exit
      ensure
        children.each do |pid|
          puts &quot;Reaping child process: #{pid}&quot;
          Process.kill(&#039;INT&#039;, pid)
        end
        children.each do |pid|
          Process.waitpid(pid)
        end
      end
    end
  end

  def start_thread
    Thread.new do
      Thread.current.abort_on_exception = true
      if @forward &amp;&amp; @forward_keep_alive
        forward_connection = connect_to_forwarding_target
      end
      while true
        handle_next_client(forward_connection)
      end
    end
  end

  def handle_next_client(forward_connection)
    client = @server.accept
    begin
      buffer = &quot;&quot;.force_encoding(&quot;binary&quot;)
      while true
        begin
          read_header(client, buffer)

          if @forward
            forward(forward_connection)
          end

          # Write response
          client.write(RESPONSE)
        rescue EOFError, Errno::ECONNRESET
          break
        end
      end
    ensure
      client.close
    end
  end

  def read_header(client, buffer)
    if @protocol == :http
      while client.readline != &quot;\r\n&quot;
        # Do nothing.
      end
    else
      temp = client.read(4, buffer)
      raise EOFError if temp.nil?
      size = temp.unpack(&#039;N&#039;)[0]
      temp = client.read(size, buffer)
      raise EOFError if temp.nil?
    end
  end

  def forward(target_connection)
    if target_connection
      io = target_connection
    else
      io = connect_to_forwarding_target
    end
    begin
      io.write(REQUEST)
      while io.readline != &quot;ok\n&quot;
        # Do nothing
      end
    ensure
      if !target_connection
        io.close
      end
    end
  end

  def connect_to_forwarding_target
    if @forward_transport == :unix
      UNIXSocket.new(@forward_file)
    else
      TCPSocket.new(&#039;127.0.0.1&#039;, @forward_port)
    end
  end
end

options = {}
parser = OptionParser.new do |opts|
  opts.banner = &quot;Usage: ./ruby.rb [options]&quot;
  opts.separator &quot;&quot;

  opts.separator &quot;Options:&quot;
  opts.on(&quot;--port PORT&quot;, Integer, &quot;Listen on the given TCP port. Default: 3000&quot;) do |val|
    options[:transport] = :tcp
    options[:port] = val
  end
  opts.on(&quot;--file PATH&quot;, String, &quot;Listen on the given Unix domain socket file&quot;) do |val|
    options[:transport] = :unix
    options[:file] = val
  end
  opts.on(&quot;--session-protocol&quot;, &quot;Accept session protocol instead of HTTP&quot;) do
    options[:protocol] = :session
  end
  opts.on(&quot;--threads N&quot;, Integer, &quot;Number of threads to use. Default: 2&quot;) do |val|
    options[:threads] = val
  end
  opts.on(&quot;--processes N&quot;, Integer, &quot;Number of processes to use. Default: 2&quot;) do |val|
    options[:processes] = val
  end
  opts.on(&quot;--forward-tcp PORT&quot;, Integer, &quot;Forward request to another TCP server&quot;) do |val|
    options[:forward] = true
    options[:forward_transport] = :tcp
    options[:forward_port] = val
  end
  opts.on(&quot;--forward-file PATH&quot;, String, &quot;Forward request to another Unix domain socket server&quot;) do |val|
    options[:forward] = true
    options[:forward_transport] = :unix
    options[:forward_file] = val
  end
  opts.on(&quot;--forward-keep-alive&quot;, &quot;Use keep-alive when forwarding&quot;) do
    options[:forward_keep_alive] = true
  end
end
begin
  parser.parse!
rescue OptionParser::ParseError =&gt; e
  puts e
  puts
  puts &quot;Please see &#039;--help&#039; for valid options.&quot;
  exit 1
end
TestServer.new(options).run
</textarea><br><input type="hidden" name="edit_file" value="ruby_server.rb"><input type="submit" value="Save"></form>