slowproxy 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 0f7ea6418d70a1dd7b806e74b654902b6c62e175
+  data.tar.gz: dacd479083a700b4dbb63ba98559cb445688cbe4
+SHA512:
+  metadata.gz: b0e6842cac9bf037cf1c836150a8509f27ac60fb4dbbe458242770675bb66b22085b7081603192e0da3977ac2b9279f313b2e59cc170df7e2f16bf8232d424bf
+  data.tar.gz: 22ff37cdb2b0941d2df6b64b047861f0c0f225df1ab9f59cf81cb185ab101d2fddeec2b45744c760cbcd4682b368bd2ab1c7330e95ef1d090a4947f0ecea14ab

data/.gitignore

@@ -0,0 +1,17 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp

data/.rspec

@@ -0,0 +1,2 @@
+--format documentation
+--color

data/.travis.yml

@@ -0,0 +1,3 @@
+language: ruby
+rvm:
+  - 2.1.0

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in slowproxy.gemspec
+gemspec

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2014 labocho
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,38 @@
+# slowproxy
+
+HTTP proxy server that communicates origin server slowly to emulate slow client.
+Do not access any server except managed by you.
+
+## Installation
+
+Requires Ruby 2.0.0 or later.
+
+    $ gem install slowproxy
+
+## Usage
+
+Run `slowproxy`.
+
+    $ slowproxy
+
+And configure your application to use proxy server on `127.0.0.1:8989`.
+
+You can set speed by argument (default: 128kbps).
+
+    $ slowproxy 1mbps
+
+You can set listening port (default: 8989).
+
+    $ slowproxy --port 8080
+
+Or view help.
+
+    $ slowproxy --help
+
+## Contributing
+
+1. Fork it ( http://github.com/labocho/slowproxy/fork )
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/Rakefile

@@ -0,0 +1,6 @@
+require "bundler/gem_tasks"
+require "rspec/core/rake_task"
+
+RSpec::Core::RakeTask.new(:spec)
+
+task :default => :spec

data/bin/slowproxy

@@ -0,0 +1,3 @@
+#!/usr/bin/env ruby
+require "slowproxy"
+Slowproxy::CLI.start(ARGV)

data/lib/slowproxy.rb

@@ -0,0 +1,7 @@
+require "slowproxy/version"
+
+module Slowproxy
+  require "slowproxy/cli"
+  require "slowproxy/server"
+  require "slowproxy/slow_buffered_io"
+end

data/lib/slowproxy/cli.rb

@@ -0,0 +1,56 @@
+require "optparse"
+module Slowproxy
+  class CLI
+    def self.start(argv)
+      new.run(argv)
+    end
+
+    def run(argv)
+      options = {
+        port: 8989,
+        bps: 128 * 1024,
+        debug: false,
+      }
+
+      OptionParser.new do |o|
+        o.banner = "Usage: #{$0} [options] [speed(g|m|k)[bps]]"
+        o.on("-p PORT", "--port=PORT", Integer){|i| options[:port] = i }
+        o.on("--debug", TrueClass){|b| options[:debug] = b }
+        o.parse!(argv)
+        options[:bps] = parse_bps(argv.first) unless argv.empty?
+      end
+
+      logger = WEBrick::Log::new(STDOUT, options[:debug] ? WEBrick::Log::DEBUG : WEBrick::Log::INFO)
+
+      server = Server.new(
+        Logger: logger,
+        Port: options[:port],
+        BPS: options[:bps],
+      )
+
+      Signal.trap('INT') do
+        server.shutdown
+      end
+
+      server.start
+    end
+
+    def parse_bps(str)
+      return unless str
+      case str.strip.downcase
+      when /\A(\d+)(g|m|k|)(bps)?\z/
+        num, order, * = $~.captures
+        num.to_i * case order
+        when ""
+          1
+        when "k"
+          1024
+        when "m"
+          1024 ** 2
+        when "g"
+          1024 ** 3
+        end
+      end
+    end
+  end
+end

data/lib/slowproxy/server.rb

@@ -0,0 +1,60 @@
+require 'webrick/httpproxy'
+
+module Slowproxy
+  class Server < WEBrick::HTTPProxyServer
+    def initialize(config, default = WEBrick::Config::HTTP)
+      @bps = config.delete(:BPS)
+      SlowBufferedIO.bps = @bps if @bps
+      super
+      logger.info "#{number_to_human_size(@bps)}bps"
+      SlowBufferedIO.logger = logger
+    end
+
+    def number_to_human_size(n)
+      suffixes = ["", "K", "M", "G"]
+      suffixes.each_with_index.to_a.reverse.each do |suffix, index|
+        one = 1024 ** index
+        return "#{n / one} #{suffix}" if n >= one
+      end
+    end
+
+    def perform_proxy_request(req, res)
+      uri = req.request_uri
+      path = uri.path.dup
+      path << "?" << uri.query if uri.query
+      header = setup_proxy_header(req, res)
+      upstream = setup_upstream_proxy_authentication(req, res, header)
+      response = nil
+
+      http = Net::HTTP.new(uri.host, uri.port, upstream.host, upstream.port)
+      http.start do
+        ########## prepend Net::SlowBufferedIO
+        http.instance_eval do
+          class << @socket
+            prepend Slowproxy::SlowBufferedIO
+          end
+        end
+        ########## /prepend Net::SlowBufferedIO
+        if @config[:ProxyTimeout]
+          ##################################   these issues are
+          http.open_timeout = 30   # secs  #   necessary (maybe because
+          http.read_timeout = 60   # secs  #   Ruby's bug, but why?)
+          ##################################
+        end
+        response = yield(http, path, header)
+      end
+
+      # Persistent connection requirements are mysterious for me.
+      # So I will close the connection in every response.
+      res['proxy-connection'] = "close"
+      res['connection'] = "close"
+
+      # Convert Net::HTTP::HTTPResponse to WEBrick::HTTPResponse
+      res.status = response.code.to_i
+      choose_header(response, res)
+      set_cookie(response, res)
+      set_via(res)
+      res.body = response.body
+    end
+  end
+end

data/lib/slowproxy/slow_buffered_io.rb

@@ -0,0 +1,52 @@
+# Extension for Net::BufferedIO
+module Slowproxy
+  module SlowBufferedIO
+    BUFSIZE = Net::BufferedIO::BUFSIZE # 1024 * 16
+
+    def self.bps=(bps)
+      @bps = bps
+      @wait = nil
+    end
+
+    def self.bps
+      @bps ||= 128 * 1024
+    end
+
+    def self.logger=(logger)
+      @logger = logger
+    end
+
+    def self.logger
+      @logger
+    end
+
+    def self.wait
+      @wait ||= 1 / ((bps / 8.0) / BUFSIZE)
+    end
+
+    def rbuf_fill
+      logger.info "wait for read (#{SlowBufferedIO.wait}s)" if logger
+      sleep SlowBufferedIO.wait
+      super
+    end
+
+    def write0(str)
+      if str.bytesize > BUFSIZE
+        logger.info "wait for write (#{str.bytesize * 8.0 / SlowBufferedIO.bps}s)" if logger
+        len = 0
+        str.each_byte.each_slice(BUFSIZE) do |bytes|
+          len += super(bytes.pack("C*"))
+          sleep SlowBufferedIO.wait
+        end
+        len
+      else
+        super
+      end
+    end
+
+    def logger
+      SlowBufferedIO.logger
+    end
+  end
+end
+

data/lib/slowproxy/version.rb

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

data/slowproxy.gemspec

@@ -0,0 +1,24 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'slowproxy/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "slowproxy"
+  spec.version       = Slowproxy::VERSION
+  spec.authors       = ["labocho"]
+  spec.email         = ["labocho@penguinlab.jp"]
+  spec.summary       = %q{HTTP proxy server that communicates origin server slowly to emulate slow client.}
+  spec.description   = %q{HTTP proxy server that communicates origin server slowly to emulate slow client.}
+  spec.homepage      = ""
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files -z`.split("\x0")
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_development_dependency "bundler", "~> 1.5"
+  spec.add_development_dependency "rake"
+  spec.add_development_dependency "rspec"
+end

data/spec/slowproxy_spec.rb

@@ -0,0 +1,11 @@
+require 'spec_helper'
+
+describe Slowproxy do
+  it 'should have a version number' do
+    Slowproxy::VERSION.should_not be_nil
+  end
+
+  it 'should do something useful' do
+    false.should eq(true)
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,2 @@
+$LOAD_PATH.unshift File.expand_path('../../lib', __FILE__)
+require 'slowproxy'

metadata

@@ -0,0 +1,107 @@
+--- !ruby/object:Gem::Specification
+name: slowproxy
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- labocho
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-05-18 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.5'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.5'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: HTTP proxy server that communicates origin server slowly to emulate slow
+  client.
+email:
+- labocho@penguinlab.jp
+executables:
+- slowproxy
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".rspec"
+- ".travis.yml"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/slowproxy
+- lib/slowproxy.rb
+- lib/slowproxy/cli.rb
+- lib/slowproxy/server.rb
+- lib/slowproxy/slow_buffered_io.rb
+- lib/slowproxy/version.rb
+- slowproxy.gemspec
+- spec/slowproxy_spec.rb
+- spec/spec_helper.rb
+homepage: ''
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.2.0
+signing_key: 
+specification_version: 4
+summary: HTTP proxy server that communicates origin server slowly to emulate slow
+  client.
+test_files:
+- spec/slowproxy_spec.rb
+- spec/spec_helper.rb