--- Compress-LZF/LZF.xs 2002/02/27 20:51:21 1.8 +++ Compress-LZF/LZF.xs 2004/08/03 15:04:05 1.18 @@ -14,6 +14,9 @@ # define call_sv perl_call_sv #endif +#define HLOG 15 +#define ULTRA_FAST 0 + #include "lzf_c.c" #include "lzf_d.c" @@ -24,14 +27,17 @@ #define MAGIC_undef 2 /* the special value undef */ #define MAGIC_CR 3 /* storable (reference, freeze), compressed */ #define MAGIC_R 4 /* storable (reference, freeze) */ +#define MAGIC_CR_deref 5 /* storable (NO reference, freeze), compressed */ +#define MAGIC_R_deref 6 /* storable (NO reference, freeze) */ #define MAGIC_HI 7 /* room for one higher storable major */ #define IN_RANGE(v,l,h) ((unsigned int)((unsigned)(v) - (unsigned)(l)) <= (unsigned)(h) - (unsigned)(l)) +static SV *serializer_package, *serializer_mstore, *serializer_mretrieve; static CV *storable_mstore, *storable_mretrieve; static SV * -compress_sv (SV *data, char cprepend, char uprepend) +compress_sv (SV *data, char cprepend, int uprepend) { STRLEN usize, csize; char *src = (char *)SvPV (data, usize); @@ -91,14 +97,14 @@ { SvCUR_set (ret, csize + skip); } - else if (!uprepend) + else if (uprepend < 0) { SvREFCNT_dec (ret); ret = SvREFCNT_inc (data); } else { - *dst++ = 0; + *dst++ = uprepend; Move ((void *)src, (void *)dst, usize, unsigned char); @@ -162,14 +168,14 @@ usize = (usize << 6) | (*src++ & 0x3f); } else - croak ("compressed data corrupted"); + croak ("compressed data corrupted (invalid length)"); ret = NEWSV (0, usize); SvPOK_only (ret); dst = SvPVX (ret); if (lzf_decompress (src, csize, dst, usize) != usize) - croak ("compressed data corrupted", csize, skip, usize); + croak ("compressed data corrupted (size mismatch)", csize, skip, usize); } else { @@ -192,23 +198,43 @@ need_storable(void) { #if PATCHLEVEL < 6 - perl_eval_pv ("require Storable;", 1); + char req[8192]; + sprintf (req, "require %s;", SvPV_nolen (serializer_package)); + perl_eval_pv (req, 1); #else - load_module (PERL_LOADMOD_NOIMPORT, newSVpv ("Storable", 0), Nullsv); + load_module (PERL_LOADMOD_NOIMPORT, serializer_package, Nullsv); #endif - storable_mstore = GvCV (gv_fetchpv ("Storable::mstore" , TRUE, SVt_PVCV)); - storable_mretrieve = GvCV (gv_fetchpv ("Storable::mretrieve", TRUE, SVt_PVCV)); + storable_mstore = GvCV (gv_fetchpv (SvPV_nolen (serializer_mstore ), TRUE, SVt_PVCV)); + storable_mretrieve = GvCV (gv_fetchpv (SvPV_nolen (serializer_mretrieve), TRUE, SVt_PVCV)); } MODULE = Compress::LZF PACKAGE = Compress::LZF +BOOT: + serializer_package = newSVpv ("Storable", 0); + serializer_mstore = newSVpv ("Storable::mstore", 0); + serializer_mretrieve = newSVpv ("Storable::mretrieve", 0); + +void +set_serializer(package, mstore, mretrieve) + SV * package + SV * mstore + SV * mretrieve + PROTOTYPE: $$$ + PPCODE: + SvSetSV (serializer_package , package ); + SvSetSV (serializer_mstore , mstore ); + SvSetSV (serializer_mretrieve, mretrieve); + storable_mstore = + storable_mretrieve = 0; + void compress(data) SV * data PROTOTYPE: $ PPCODE: - XPUSHs (sv_2mortal (compress_sv (data, 0, 1))); + XPUSHs (sv_2mortal (compress_sv (data, 0, MAGIC_U))); void decompress(data) @@ -226,16 +252,26 @@ PROTOTYPE: $ PPCODE: + SvGETMAGIC (sv); + if (!SvOK (sv)) XPUSHs (sv_2mortal (newSVpvn ("\02", 1))); /* 02 == MAGIC_undef */ - else if (SvTYPE(sv) != SVt_IV - && SvTYPE(sv) != SVt_NV - && SvTYPE(sv) != SVt_PV - && SvTYPE(sv) != SVt_PVMG) /* mstore */ + else if (SvROK (sv) + || (SvTYPE(sv) != SVt_IV + && SvTYPE(sv) != SVt_NV + && SvTYPE(sv) != SVt_PV + && SvTYPE(sv) != SVt_PVIV + && SvTYPE(sv) != SVt_PVNV + && SvTYPE(sv) != SVt_PVMG)) /* mstore */ { + int deref = !SvROK (sv); + if (!storable_mstore) need_storable (); + if (deref) + sv = newRV_noinc (sv); + PUSHMARK (SP); XPUSHs (sv); PUTBACK; @@ -250,17 +286,26 @@ if (SvPVX (sv)[0] != MAGIC_R) croak ("Storable format changed, need newer version of Compress::LZF"); + if (deref) + SvPVX (sv)[0] = MAGIC_R_deref; + if (ix) /* compress */ - XPUSHs (sv_2mortal (compress_sv (sv, MAGIC_CR, 0))); - else - XPUSHs (sv); + sv = sv_2mortal (compress_sv (sv, deref ? MAGIC_CR_deref : MAGIC_CR, -1)); + + XPUSHs (sv); } - else if (sv && IN_RANGE (SvPVX (sv)[0], MAGIC_LO, MAGIC_HI)) - XPUSHs (sv_2mortal (compress_sv (sv, MAGIC_C, 1))); /* need to prefix only */ + else if (SvPOKp (sv) && IN_RANGE (SvPVX (sv)[0], MAGIC_LO, MAGIC_HI)) + XPUSHs (sv_2mortal (compress_sv (sv, MAGIC_C, MAGIC_U))); /* need to prefix only */ else if (ix == 2) /* compress always */ - XPUSHs (sv_2mortal (compress_sv (sv, MAGIC_C, 0))); + XPUSHs (sv_2mortal (compress_sv (sv, MAGIC_C, -1))); + else if (SvNIOK (sv)) /* don't compress */ + { + STRLEN len; + char *s = SvPV (sv, len); + XPUSHs (sv_2mortal (newSVpvn (s, len))); + } else /* don't compress */ - XPUSHs (sv_2mortal (SvREFCNT_inc (sv))); + XPUSHs (sv_2mortal (newSVsv (sv))); void sthaw(sv) @@ -268,6 +313,8 @@ PROTOTYPE: $ PPCODE: + int deref = 0; + SvGETMAGIC (sv); if (SvPOK (sv) && IN_RANGE (SvPV_nolen (sv)[0], MAGIC_LO, MAGIC_HI)) { @@ -285,9 +332,23 @@ XPUSHs (sv_2mortal (decompress_sv (sv, 1))); break; + case MAGIC_R_deref: + deref = 1; + SvPVX (sv)[0] = MAGIC_R; + goto handle_MAGIC_R; + + case MAGIC_CR_deref: + deref = 1; case MAGIC_CR: sv = sv_2mortal (decompress_sv (sv, 1)); /* mortal could be optimized */ + if (deref) + if (SvPVX (sv)[0] == MAGIC_R_deref) + SvPVX (sv)[0] = MAGIC_R; + else + croak ("Compress::LZF::sthaw(): invalid data, maybe you need a newer version of Compress::LZF?"); + case MAGIC_R: + handle_MAGIC_R: if (!storable_mstore) need_storable (); @@ -300,7 +361,15 @@ SPAGAIN; - /*XPUSHs (POPs); this is a nop, hopefully */ + if (deref) + { + SETs (sv_2mortal (SvREFCNT_inc (SvRV (TOPs)))); + + if (SvPVX (sv)[0] == MAGIC_R) + SvPVX (sv)[0] = MAGIC_R_deref; + } + else + SETs (sv_2mortal (newSVsv (TOPs))); break; @@ -309,4 +378,5 @@ } } else - XPUSHs (sv_2mortal (SvREFCNT_inc (sv))); + XPUSHs (sv_2mortal (newSVsv (sv))); +