Contents

require 'addressable/uri'
require 'eventmachine'
require 'http/parser'
require 'em-http'
require 'evma_httpserver'
require 'em-synchrony'

module Billy
  class ProxyConnection < EventMachine::Connection
    attr_accessor :handler
    attr_accessor :cache

    def post_init
      @parser = Http::Parser.new(self)
    end

    def receive_data(data)
      @parser << data
    end

    def on_message_begin
      @headers = nil
      @body = ''
    end

    def on_headers_complete(headers)
      @headers = headers
    end

    def on_body(chunk)
      @body << chunk
    end

    def on_message_complete
      if @parser.http_method == 'CONNECT'
        restart_with_ssl(@parser.request_url)
      else
        if @ssl
          uri = Addressable::URI.parse(@parser.request_url)
          @url = "https://#{@ssl}#{[uri.path,uri.query].compact.join('?')}"
        else
          @url = @parser.request_url
        end
        handle_request
      end
    end

    protected

    def restart_with_ssl(url)
      @ssl = url
      @parser = Http::Parser.new(self)
      send_data("HTTP/1.0 200 Connection established\r\nProxy-agent: Puffing-Billy/0.0.0\r\n\r\n")
      start_tls(
        :private_key_file => File.expand_path('../mitm.key', __FILE__),
        :cert_chain_file => File.expand_path('../mitm.crt', __FILE__)
      )
    end

    def handle_request
      EM.synchrony do
        handler.handle_request(@parser.http_method, @url, @headers, @body).tap do |response|
          if response.has_key?(:error)
            close_connection
            raise "puffing-billy: #{response[:error]}"
          else
            send_response(response)
          end
        end
      end
    end

    private

    def send_response(response)
      res = EM::DelegatedHttpResponse.new(self)
      res.status = response[:status]
      res.headers = response[:headers]
      res.content = response[:content]
      res.send_response
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
puffing-billy-0.4.1	lib/billy/proxy_connection.rb