diff options
Diffstat (limited to 'lib/yahns/max_body/rewindable_wrapper.rb')
-rw-r--r-- | lib/yahns/max_body/rewindable_wrapper.rb | 19 |
1 files changed, 19 insertions, 0 deletions
diff --git a/lib/yahns/max_body/rewindable_wrapper.rb b/lib/yahns/max_body/rewindable_wrapper.rb new file mode 100644 index 0000000..5888def --- /dev/null +++ b/lib/yahns/max_body/rewindable_wrapper.rb @@ -0,0 +1,19 @@ +# -*- encoding: binary -*- +# Copyright (C) 2009-2013, Eric Wong <normalperson@yhbt.net> et. al. +# License: GPLv2 or later (https://www.gnu.org/licenses/gpl-2.0.txt) +class Yahns::MaxBody::RewindableWrapper < Yahns::MaxBody::Wrapper # :nodoc: + def initialize(rack_input, limit) + @orig_limit = limit + super + end + + def rewind + @limit = @orig_limit + @rbuf = '' + @input.rewind + end + + def size + @input.size + end +end |