about summary refs log tree commit homepage
path: root/lib/yahns/max_body/rewindable_wrapper.rb
diff options
context:
space:
mode:
Diffstat (limited to 'lib/yahns/max_body/rewindable_wrapper.rb')
-rw-r--r--lib/yahns/max_body/rewindable_wrapper.rb19
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