Quick tidy-up

thbar · Oct 5, 2012 · 00ea714 · 00ea714
1 parent b2cdc06
commit 00ea714
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/lib/puffing-billy/proxy_connection.rb b/lib/puffing-billy/proxy_connection.rb
@@ -18,10 +18,13 @@ def receive_data(data)
         @parser << data
       rescue HTTP::Parser::Error
         if @parser.http_method == 'CONNECT'
+          # work-around for CONNECT requests until https://github.com/tmm1/http_parser.rb/pull/15 gets merged
           @ssl = @header_data.split("\r\n").first.split(/\s+/)[1]
           @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 => 'node/mitm.key', :cert_chain_file => 'node/mitm.crt')
+        else
+          close_connection
         end
       end
     end