Class: Yahns::MaxBody::Wrapper

Inherits:
Object
  • Object
show all
Defined in:
lib/yahns/max_body/wrapper.rb

Overview

Copyright © 2009-2013, Eric Wong <[email protected]> et. al. License: GPLv2 or later (www.gnu.org/licenses/gpl-2.0.txt)

This is only used for chunked request bodies, which are rare

Direct Known Subclasses

RewindableWrapper

Instance Method Summary collapse

Constructor Details

#initialize(rack_input, limit) ⇒ Wrapper

:nodoc:



7
8
9
 
# File 'lib/yahns/max_body/wrapper.rb', line 7

def initialize(rack_input, limit)
  @input, @limit, @rbuf = rack_input, limit, ''
end

Instance Method Details

#checked_read(length = 16384, buf = '') ⇒ Object 



56
57
58
59
60
61
 
# File 'lib/yahns/max_body/wrapper.rb', line 56

def checked_read(length = 16384, buf = '')
  if @input.read(length, buf)
    throw :yahns_EFBIG if ((@limit -= buf.size) < 0)
    return buf
  end
end

#each ⇒ Object 



11
12
13
14
15
 
# File 'lib/yahns/max_body/wrapper.rb', line 11

def each
  while line = gets
    yield line
  end
end

#gets ⇒ Object 



35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
 
# File 'lib/yahns/max_body/wrapper.rb', line 35

def gets
  sep = $/
  if sep.nil?
    rv = read_all
    return rv.empty? ? nil : rv
  end
  re = /\A(.*?#{Regexp.escape(sep)})/

  begin
    @rbuf.sub!(re, '') and return $1

    if tmp = checked_read(16384)
      @rbuf << tmp
    elsif @rbuf.empty? # EOF
      return nil
    else # EOF, return whatever is left
      return @rbuf.slice!(0, @rbuf.size)
    end
  end while true
end

#read(length = nil, rv = '') ⇒ Object

chunked encoding means this method behaves more like readpartial, since Rack does not support a method named “readpartial”



19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
 
# File 'lib/yahns/max_body/wrapper.rb', line 19

def read(length = nil, rv = '')
  if length
    if length <= @rbuf.size
      length < 0 and raise ArgumentError, "negative length #{length} given"
      rv.replace(@rbuf.slice!(0, length))
    elsif @rbuf.empty?
      checked_read(length, rv) or return
    else
      rv.replace(@rbuf.slice!(0, @rbuf.size))
    end
    rv.empty? && length != 0 ? nil : rv
  else
    rv.replace(read_all)
  end
end

#read_all ⇒ Object 



63
64
65
66
67
68
69
70
 
# File 'lib/yahns/max_body/wrapper.rb', line 63

def read_all
  rv = @rbuf.slice!(0, @rbuf.size)
  tmp = ''
  while checked_read(16384, tmp)
    rv << tmp
  end
  rv
end