Class: Slowproxy::Server

Inherits:
WEBrick::HTTPProxyServer
  • Object
show all
Defined in:
lib/slowproxy/server.rb

Instance Method Summary collapse

Constructor Details

#initialize(config, default = WEBrick::Config::HTTP) ⇒ Server

Returns a new instance of Server.



5
6
7
8
9
10
11
 
# File 'lib/slowproxy/server.rb', line 5

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

Instance Method Details

#do_CONNECT(req, res) ⇒ Object

Raises:

  • (WEBrick::HTTPStatus::InternalServerError) 


64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
 
# File 'lib/slowproxy/server.rb', line 64

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

#number_to_human_size(n) ⇒ Object 



13
14
15
16
17
18
19
 
# File 'lib/slowproxy/server.rb', line 13

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

#perform_proxy_request(req, res) ⇒ Object 



21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
 
# File 'lib/slowproxy/server.rb', line 21

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

#wait_for_connect ⇒ Object 



60
61
62
 
# File 'lib/slowproxy/server.rb', line 60

def wait_for_connect
  @wait ||= 1 / ((SlowBufferedIO.bps / 8.0) / 1024)
end