slowproxy 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 0f7ea6418d70a1dd7b806e74b654902b6c62e175
-  data.tar.gz: dacd479083a700b4dbb63ba98559cb445688cbe4
+  metadata.gz: 3dcec1ec81a38744cef9f2ce07661d8f90ad969f
+  data.tar.gz: 8d1fdd0a2fad34b56c41b490980a20faa3df3c40
 SHA512:
-  metadata.gz: b0e6842cac9bf037cf1c836150a8509f27ac60fb4dbbe458242770675bb66b22085b7081603192e0da3977ac2b9279f313b2e59cc170df7e2f16bf8232d424bf
-  data.tar.gz: 22ff37cdb2b0941d2df6b64b047861f0c0f225df1ab9f59cf81cb185ab101d2fddeec2b45744c760cbcd4682b368bd2ab1c7330e95ef1d090a4947f0ecea14ab
+  metadata.gz: 4567ec6977d45b0c6d1696b288c926f75bf770a7ae94101b1ef22cea84e3cfb4a09a86ac1d71264313fd7296fc555a3ba1feb6b2c2360fa69d9f73bb0dd5dc5d
+  data.tar.gz: 34b81cdf1e48c444bbcde0295deea629a1d202f551df9cc62ccc6ca8f15015ecf15bd304b0a22076fd311e7ef4e7e57a74ad5c589c6104f89b8f186d4bbbb4e9

data/lib/slowproxy/server.rb

@@ -56,5 +56,97 @@ module Slowproxy
       set_via(res)
       res.body = response.body
     end
+
+    def wait_for_connect
+      @wait ||= 1 / ((SlowBufferedIO.bps / 8.0) / 1024)
+    end
+
+    def do_CONNECT(req, res)
+      # Proxy Authentication
+      proxy_auth(req, res)
+
+      ua = Thread.current[:WEBrickSocket]  # User-Agent
+      raise WEBrick::HTTPStatus::InternalServerError,
+        "[BUG] cannot get socket" unless ua
+
+      host, port = req.unparsed_uri.split(":", 2)
+      # Proxy authentication for upstream proxy server
+      if proxy = proxy_uri(req, res)
+        proxy_request_line = "CONNECT #{host}:#{port} HTTP/1.0"
+        if proxy.userinfo
+          credentials = "Basic " + [proxy.userinfo].pack("m").delete("\n")
+        end
+        host, port = proxy.host, proxy.port
+      end
+
+      begin
+        @logger.debug("CONNECT: upstream proxy is `#{host}:#{port}'.")
+        os = TCPSocket.new(host, port)     # origin server
+
+        if proxy
+          @logger.debug("CONNECT: sending a Request-Line")
+          os << proxy_request_line << CRLF
+          @logger.debug("CONNECT: > #{proxy_request_line}")
+          if credentials
+            @logger.debug("CONNECT: sending a credentials")
+            os << "Proxy-Authorization: " << credentials << CRLF
+          end
+          os << CRLF
+          proxy_status_line = os.gets(LF)
+          @logger.debug("CONNECT: read a Status-Line form the upstream server")
+          @logger.debug("CONNECT: < #{proxy_status_line}")
+          if %r{^HTTP/\d+\.\d+\s+200\s*} =~ proxy_status_line
+            while line = os.gets(LF)
+              break if /\A(#{CRLF}|#{LF})\z/om =~ line
+            end
+          else
+            raise WEBrick::HTTPStatus::BadGateway
+          end
+        end
+        @logger.debug("CONNECT #{host}:#{port}: succeeded")
+        res.status = WEBrick::HTTPStatus::RC_OK
+      rescue => ex
+        @logger.debug("CONNECT #{host}:#{port}: failed `#{ex.message}'")
+        res.set_error(ex)
+        raise WEBrick::HTTPStatus::EOFError
+      ensure
+        if handler = @config[:ProxyContentHandler]
+          handler.call(req, res)
+        end
+        res.send_response(ua)
+        access_log(@config, req, res)
+
+        # Should clear request-line not to send the response twice.
+        # see: HTTPServer#run
+        req.parse(NullReader) rescue nil
+      end
+
+      begin
+        while fds = IO::select([ua, os])
+          if fds[0].member?(ua)
+            buf = ua.sysread(1024);
+            @logger.debug("CONNECT: #{buf.bytesize} byte from User-Agent")
+            # Write slowly
+            @logger.debug "wait for write (#{wait_for_connect}s)"
+            sleep wait_for_connect
+            # /Write slowly
+            os.syswrite(buf)
+          elsif fds[0].member?(os)
+            # Read slowly
+            @logger.debug "wait for read (#{wait_for_connect}s)"
+            sleep wait_for_connect
+            # /Read slowly
+            buf = os.sysread(1024);
+            @logger.debug("CONNECT: #{buf.bytesize} byte from #{host}:#{port}")
+            ua.syswrite(buf)
+          end
+        end
+      rescue => ex
+        os.close
+        @logger.debug("CONNECT #{host}:#{port}: closed")
+      end
+
+      raise WEBrick::HTTPStatus::EOFError
+    end
   end
 end

data/lib/slowproxy/version.rb

@@ -1,3 +1,3 @@
 module Slowproxy
-  VERSION = "0.0.1"
+  VERSION = "0.0.2"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: slowproxy
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - labocho
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-05-18 00:00:00.000000000 Z
+date: 2014-05-20 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler