def Decompress_decompress_beta(self, inp, out):
stream = self.stream
i_data = cast(inp.getattr(u"data"), Uint8Data, u"inp.data")
i_start = cast(inp.getattr(u"start"), Integer, u"inp.start").value
i_stop = cast(inp.getattr(u"stop"), Integer, u"inp.stop").value
if not (0 <= i_start <= i_stop <= i_data.length):
raise unwind(LTypeError(u"inp.start/stop invalid"))
o_data = cast(out.getattr(u"data"), Uint8Data, u"out.data")
o_start = cast(out.getattr(u"start"), Integer, u"out.start").value
o_stop = cast(out.getattr(u"stop"), Integer, u"out.stop").value
if not (0 <= o_start <= o_stop <= o_data.length):
raise unwind(LTypeError(u"out.start/stop invalid"))
stream.c_next_in = rffi.ptradd(i_data.uint8data, i_start)
stream.c_avail_in = rffi.r_uint(i_stop - i_start)
stream.c_next_out = rffi.ptradd(o_data.uint8data, o_stop)
stream.c_avail_out = rffi.r_uint(o_data.length - o_stop)
if i_start == i_stop:
err = rzlib._inflate(stream, rzlib.Z_FINISH)
else:
err = rzlib._inflate(stream, 0)
if err == rzlib.Z_NEED_DICT:
raise unwind(LError(u"Z_NEED_DICT"))
if err == rzlib.Z_BUF_ERROR:
raise unwind(LError(u"Z_BUF_ERROR"))
if err == rzlib.Z_OK or err == rzlib.Z_STREAM_END:
inp.setattr(u"start", Integer(i_stop - rffi.r_long(stream.c_avail_in)))
out.setattr(u"stop", Integer(rffi.r_long(stream.c_avail_out) + o_stop))
return null
def test_deflate_set_dictionary():
text = 'abcabc'
zdict = 'abc'
stream = rzlib.deflateInit()
rzlib.deflateSetDictionary(stream, zdict)
bytes = rzlib.compress(stream, text, rzlib.Z_FINISH)
rzlib.deflateEnd(stream)
stream2 = rzlib.inflateInit()
from rpython.rtyper.lltypesystem import lltype, rffi, rstr
from rpython.rtyper.annlowlevel import llstr
from rpython.rlib.rstring import StringBuilder
with lltype.scoped_alloc(rffi.CCHARP.TO, len(bytes)) as inbuf:
rstr.copy_string_to_raw(llstr(bytes), inbuf, 0, len(bytes))
stream2.c_next_in = rffi.cast(rzlib.Bytefp, inbuf)
rffi.setintfield(stream2, 'c_avail_in', len(bytes))
with lltype.scoped_alloc(rffi.CCHARP.TO, 100) as outbuf:
stream2.c_next_out = rffi.cast(rzlib.Bytefp, outbuf)
bufsize = 100
rffi.setintfield(stream2, 'c_avail_out', bufsize)
err = rzlib._inflate(stream2, rzlib.Z_SYNC_FLUSH)
assert err == rzlib.Z_NEED_DICT
rzlib.inflateSetDictionary(stream2, zdict)
rzlib._inflate(stream2, rzlib.Z_SYNC_FLUSH)
avail_out = rffi.cast(lltype.Signed, stream2.c_avail_out)
result = StringBuilder()
result.append_charpsize(outbuf, bufsize - avail_out)
rzlib.inflateEnd(stream2)
assert result.build() == text
def test_deflate_set_dictionary():
text = 'abcabc'
zdict = 'abc'
stream = rzlib.deflateInit()
rzlib.deflateSetDictionary(stream, zdict)
bytes = rzlib.compress(stream, text, rzlib.Z_FINISH)
rzlib.deflateEnd(stream)
stream2 = rzlib.inflateInit()
from rpython.rtyper.lltypesystem import lltype, rffi, rstr
from rpython.rtyper.annlowlevel import llstr
from rpython.rlib.rstring import StringBuilder
with lltype.scoped_alloc(rffi.CCHARP.TO, len(bytes)) as inbuf:
rstr.copy_string_to_raw(llstr(bytes), inbuf, 0, len(bytes))
stream2.c_next_in = rffi.cast(rzlib.Bytefp, inbuf)
rffi.setintfield(stream2, 'c_avail_in', len(bytes))
with lltype.scoped_alloc(rffi.CCHARP.TO, 100) as outbuf:
stream2.c_next_out = rffi.cast(rzlib.Bytefp, outbuf)
bufsize = 100
rffi.setintfield(stream2, 'c_avail_out', bufsize)
err = rzlib._inflate(stream2, rzlib.Z_SYNC_FLUSH)
assert err == rzlib.Z_NEED_DICT
rzlib.inflateSetDictionary(stream2, zdict)
rzlib._inflate(stream2, rzlib.Z_SYNC_FLUSH)
avail_out = rffi.cast(lltype.Signed, stream2.c_avail_out)
result = StringBuilder()
result.append_charpsize(outbuf, bufsize - avail_out)
rzlib.inflateEnd(stream2)
assert result.build() == text