Contents

require 'socket'

module Thin
  # Connection between the server and client.
  class Connection < EventMachine::Connection
    include Logging
    
    # Rack application served by this connection.
    attr_accessor :app
    
    # Connector to the server
    attr_accessor :connector
    
    attr_accessor :request, :response
    
    def post_init
      @request  = Request.new
      @response = Response.new
    end
    
    def receive_data(data)
      trace { data }
      process if @request.parse(data)
    rescue InvalidRequest => e
      log "Invalid request"
      log_error e
      close_connection
    end
    
    def process
      # Add client info to the request env
      @request.env[Request::REMOTE_ADDR] = remote_address
      
      # Process the request
      @response.status, @response.headers, @response.body = @app.call(@request.env)
      
      # Send the response
      @response.each do |chunk|
        trace { chunk }
        send_data chunk
      end
      
      close_connection_after_writing
      
    rescue Object => e
      log "Unexpected error while processing request: #{e.message}"
      log_error e
      close_connection rescue nil
    ensure
      @request.close  rescue nil
      @response.close rescue nil
    end
    
    def unbind
      @connector.connection_finished(self)
    end
    
    def remote_address
      @request.env[Request::FORWARDED_FOR] || (has_peername? ? socket_address : nil)
    rescue
      log_error($!)
      nil
    end
    
    protected
      def has_peername?
        !get_peername.nil? && !get_peername.empty?
      end
      
      def socket_address
        Socket.unpack_sockaddr_in(get_peername)[1]
      end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
thin-0.6.4	lib/thin/connection.rb