57 lines
1.3 KiB
Ruby
57 lines
1.3 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module Zip
|
|
class Inflater < Decompressor # :nodoc:all
|
|
def initialize(*args)
|
|
super
|
|
|
|
@buffer = +''
|
|
@zlib_inflater = ::Zlib::Inflate.new(-Zlib::MAX_WBITS)
|
|
end
|
|
|
|
def read(length = nil, outbuf = +'')
|
|
return (length.nil? || length.zero? ? '' : nil) if eof
|
|
|
|
while length.nil? || (@buffer.bytesize < length)
|
|
break if input_finished?
|
|
|
|
@buffer << produce_input
|
|
end
|
|
|
|
outbuf.replace(@buffer.slice!(0...(length || @buffer.bytesize)))
|
|
end
|
|
|
|
def eof
|
|
@buffer.empty? && input_finished?
|
|
end
|
|
|
|
alias eof? eof
|
|
|
|
private
|
|
|
|
def produce_input
|
|
retried = 0
|
|
begin
|
|
@zlib_inflater.inflate(input_stream.read(Decompressor::CHUNK_SIZE))
|
|
rescue Zlib::BufError
|
|
raise if retried >= 5 # how many times should we retry?
|
|
|
|
retried += 1
|
|
retry
|
|
end
|
|
rescue Zlib::Error => e
|
|
raise ::Zip::DecompressionError, e
|
|
end
|
|
|
|
def input_finished?
|
|
@zlib_inflater.finished?
|
|
end
|
|
end
|
|
|
|
::Zip::Decompressor.register(::Zip::COMPRESSION_METHOD_DEFLATE, ::Zip::Inflater)
|
|
end
|
|
|
|
# Copyright (C) 2002, 2003 Thomas Sondergaard
|
|
# rubyzip is free software; you can redistribute it and/or
|
|
# modify it under the terms of the ruby license.
|