ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/Compress-LZF/LZF.xs
(Generate patch)

Comparing Compress-LZF/LZF.xs (file contents):
Revision 1.3 by root, Thu Sep 27 19:51:26 2001 UTC vs.
Revision 1.18 by pcg, Tue Aug 3 15:04:05 2004 UTC

… …
12#endif 12#endif
13#if PATCHLEVEL < 6 13#if PATCHLEVEL < 6
14# define call_sv perl_call_sv 14# define call_sv perl_call_sv
15#endif 15#endif
16 16
17#define HLOG 15
18#define ULTRA_FAST 0
19
17#include "lzf_c.c" 20#include "lzf_c.c"
18#include "lzf_d.c" 21#include "lzf_d.c"
19 22
20/* we re-use the storable header for our purposes */ 23/* we re-use the storable header for our purposes */
21#define MAGIC_LO 0 24#define MAGIC_LO 0
22#define MAGIC_U 0 /* uncompressed data follows */ 25#define MAGIC_U 0 /* uncompressed data follows */
23#define MAGIC_C 1 /* compressed data follows */ 26#define MAGIC_C 1 /* compressed data follows */
27#define MAGIC_undef 2 /* the special value undef */
28#define MAGIC_CR 3 /* storable (reference, freeze), compressed */
24#define MAGIC_R 4 /* storable (reference, freeze) */ 29#define MAGIC_R 4 /* storable (reference, freeze) */
25#define MAGIC_CR 3 /* storable (reference, freeze), compressed */ 30#define MAGIC_CR_deref 5 /* storable (NO reference, freeze), compressed */
31#define MAGIC_R_deref 6 /* storable (NO reference, freeze) */
26#define MAGIC_HI 7 /* room for one higher storable major */ 32#define MAGIC_HI 7 /* room for one higher storable major */
27 33
28#define IN_RANGE(v,l,h) ((unsigned int)((unsigned)(v) - (unsigned)(l)) <= (unsigned)(h) - (unsigned)(l)) 34#define IN_RANGE(v,l,h) ((unsigned int)((unsigned)(v) - (unsigned)(l)) <= (unsigned)(h) - (unsigned)(l))
29 35
36static SV *serializer_package, *serializer_mstore, *serializer_mretrieve;
30static CV *storable_mstore, *storable_mretrieve; 37static CV *storable_mstore, *storable_mretrieve;
31 38
32static SV * 39static SV *
33compress_sv (SV *data, char cprepend, char uprepend) 40compress_sv (SV *data, char cprepend, int uprepend)
34{ 41{
35 STRLEN usize, csize; 42 STRLEN usize, csize;
36 char *src = (char *)SvPV (data, usize); 43 char *src = (char *)SvPV (data, usize);
37 44
38 if (usize) 45 if (usize)
… …
88 95
89 if (csize) 96 if (csize)
90 { 97 {
91 SvCUR_set (ret, csize + skip); 98 SvCUR_set (ret, csize + skip);
92 } 99 }
93 else if (!uprepend) 100 else if (uprepend < 0)
94 { 101 {
95 SvREFCNT_dec (ret); 102 SvREFCNT_dec (ret);
96 ret = SvREFCNT_inc (data); 103 ret = SvREFCNT_inc (data);
97 } 104 }
98 else 105 else
99 { 106 {
100 *dst++ = 0; 107 *dst++ = uprepend;
101 108
102 Move ((void *)src, (void *)dst, usize, unsigned char); 109 Move ((void *)src, (void *)dst, usize, unsigned char);
103 110
104 SvCUR_set (ret, usize + 1); 111 SvCUR_set (ret, usize + 1);
105 } 112 }
… …
159 usize = (usize << 6) | (*src++ & 0x3f); 166 usize = (usize << 6) | (*src++ & 0x3f);
160 usize = (usize << 6) | (*src++ & 0x3f); 167 usize = (usize << 6) | (*src++ & 0x3f);
161 usize = (usize << 6) | (*src++ & 0x3f); 168 usize = (usize << 6) | (*src++ & 0x3f);
162 } 169 }
163 else 170 else
164 croak ("compressed data corrupted"); 171 croak ("compressed data corrupted (invalid length)");
165 172
166 ret = NEWSV (0, usize); 173 ret = NEWSV (0, usize);
167 SvPOK_only (ret); 174 SvPOK_only (ret);
168 dst = SvPVX (ret); 175 dst = SvPVX (ret);
169 176
170 if (lzf_decompress (src, csize, dst, usize) != usize) 177 if (lzf_decompress (src, csize, dst, usize) != usize)
171 croak ("compressed data corrupted", csize, skip, usize); 178 croak ("compressed data corrupted (size mismatch)", csize, skip, usize);
172 } 179 }
173 else 180 else
174 { 181 {
175 usize = csize - 1; 182 usize = csize - 1;
176 ret = NEWSV (0, usize); 183 ret = NEWSV (0, usize);
… …
189 196
190static void 197static void
191need_storable(void) 198need_storable(void)
192{ 199{
193#if PATCHLEVEL < 6 200#if PATCHLEVEL < 6
201 char req[8192];
202 sprintf (req, "require %s;", SvPV_nolen (serializer_package));
194 perl_eval_pv ("require Storable;", 1); 203 perl_eval_pv (req, 1);
195#else 204#else
196 load_module (PERL_LOADMOD_NOIMPORT, newSVpv ("Storable", 0), Nullsv); 205 load_module (PERL_LOADMOD_NOIMPORT, serializer_package, Nullsv);
197#endif 206#endif
198 207
199 storable_mstore = GvCV (gv_fetchpv ("Storable::mstore" , TRUE, SVt_PVCV)); 208 storable_mstore = GvCV (gv_fetchpv (SvPV_nolen (serializer_mstore ), TRUE, SVt_PVCV));
200 storable_mretrieve = GvCV (gv_fetchpv ("Storable::mretrieve", TRUE, SVt_PVCV)); 209 storable_mretrieve = GvCV (gv_fetchpv (SvPV_nolen (serializer_mretrieve), TRUE, SVt_PVCV));
201} 210}
202 211
203MODULE = Compress::LZF PACKAGE = Compress::LZF 212MODULE = Compress::LZF PACKAGE = Compress::LZF
213
214BOOT:
215 serializer_package = newSVpv ("Storable", 0);
216 serializer_mstore = newSVpv ("Storable::mstore", 0);
217 serializer_mretrieve = newSVpv ("Storable::mretrieve", 0);
218
219void
220set_serializer(package, mstore, mretrieve)
221 SV * package
222 SV * mstore
223 SV * mretrieve
224 PROTOTYPE: $$$
225 PPCODE:
226 SvSetSV (serializer_package , package );
227 SvSetSV (serializer_mstore , mstore );
228 SvSetSV (serializer_mretrieve, mretrieve);
229 storable_mstore =
230 storable_mretrieve = 0;
204 231
205void 232void
206compress(data) 233compress(data)
207 SV * data 234 SV * data
208 PROTOTYPE: $ 235 PROTOTYPE: $
209 PPCODE: 236 PPCODE:
210 XPUSHs (sv_2mortal (compress_sv (data, 0, 1))); 237 XPUSHs (sv_2mortal (compress_sv (data, 0, MAGIC_U)));
211 238
212void 239void
213decompress(data) 240decompress(data)
214 SV * data 241 SV * data
215 PROTOTYPE: $ 242 PROTOTYPE: $
… …
223 sfreeze_cr = 1 250 sfreeze_cr = 1
224 sfreeze_c = 2 251 sfreeze_c = 2
225 PROTOTYPE: $ 252 PROTOTYPE: $
226 PPCODE: 253 PPCODE:
227 254
255 SvGETMAGIC (sv);
256
257 if (!SvOK (sv))
258 XPUSHs (sv_2mortal (newSVpvn ("\02", 1))); /* 02 == MAGIC_undef */
259 else if (SvROK (sv)
228 if (SvTYPE(sv) != SVt_IV 260 || (SvTYPE(sv) != SVt_IV
229 && SvTYPE(sv) != SVt_NV 261 && SvTYPE(sv) != SVt_NV
262 && SvTYPE(sv) != SVt_PV
263 && SvTYPE(sv) != SVt_PVIV
264 && SvTYPE(sv) != SVt_PVNV
230 && SvTYPE(sv) != SVt_PV) /* mstore */ 265 && SvTYPE(sv) != SVt_PVMG)) /* mstore */
231 { 266 {
267 int deref = !SvROK (sv);
268
232 if (!storable_mstore) 269 if (!storable_mstore)
233 need_storable (); 270 need_storable ();
271
272 if (deref)
273 sv = newRV_noinc (sv);
234 274
235 PUSHMARK (SP); 275 PUSHMARK (SP);
236 XPUSHs (sv); 276 XPUSHs (sv);
237 PUTBACK; 277 PUTBACK;
238 278
… …
244 sv = POPs; 284 sv = POPs;
245 285
246 if (SvPVX (sv)[0] != MAGIC_R) 286 if (SvPVX (sv)[0] != MAGIC_R)
247 croak ("Storable format changed, need newer version of Compress::LZF"); 287 croak ("Storable format changed, need newer version of Compress::LZF");
248 288
289 if (deref)
290 SvPVX (sv)[0] = MAGIC_R_deref;
291
249 if (ix) /* compress */ 292 if (ix) /* compress */
250 XPUSHs (sv_2mortal (compress_sv (sv, MAGIC_CR, 0))); 293 sv = sv_2mortal (compress_sv (sv, deref ? MAGIC_CR_deref : MAGIC_CR, -1));
251 else 294
252 XPUSHs (sv); 295 XPUSHs (sv);
253 } 296 }
254 else if (IN_RANGE (SvPVX (sv)[0], MAGIC_LO, MAGIC_HI)) 297 else if (SvPOKp (sv) && IN_RANGE (SvPVX (sv)[0], MAGIC_LO, MAGIC_HI))
255 XPUSHs (sv_2mortal (compress_sv (sv, MAGIC_C, 1))); /* need to prefix only */ 298 XPUSHs (sv_2mortal (compress_sv (sv, MAGIC_C, MAGIC_U))); /* need to prefix only */
256 else if (ix == 2) /* compress always */ 299 else if (ix == 2) /* compress always */
257 XPUSHs (sv_2mortal (compress_sv (sv, MAGIC_C, 0))); 300 XPUSHs (sv_2mortal (compress_sv (sv, MAGIC_C, -1)));
301 else if (SvNIOK (sv)) /* don't compress */
302 {
303 STRLEN len;
304 char *s = SvPV (sv, len);
305 XPUSHs (sv_2mortal (newSVpvn (s, len)));
306 }
258 else /* don't compress */ 307 else /* don't compress */
259 XPUSHs (sv_2mortal (SvREFCNT_inc (sv))); 308 XPUSHs (sv_2mortal (newSVsv (sv)));
260 309
261void 310void
262sthaw(sv) 311sthaw(sv)
263 SV * sv 312 SV * sv
264 PROTOTYPE: $ 313 PROTOTYPE: $
265 PPCODE: 314 PPCODE:
266 char hdr = SvPVX (sv)[0];
267 315
268 if (IN_RANGE (hdr, MAGIC_LO, MAGIC_HI)) 316 int deref = 0;
317
318 SvGETMAGIC (sv);
319 if (SvPOK (sv) && IN_RANGE (SvPV_nolen (sv)[0], MAGIC_LO, MAGIC_HI))
269 { 320 {
270 switch (SvPVX (sv)[0]) 321 switch (SvPVX (sv)[0])
271 { 322 {
323 case MAGIC_undef:
324 XPUSHs (sv_2mortal (NEWSV (0, 0)));
325 break;
326
272 case MAGIC_U: 327 case MAGIC_U:
273 XPUSHs (sv_2mortal (decompress_sv (sv, 0))); 328 XPUSHs (sv_2mortal (decompress_sv (sv, 0)));
274 break; 329 break;
275 330
276 case MAGIC_C: 331 case MAGIC_C:
277 XPUSHs (sv_2mortal (decompress_sv (sv, 1))); 332 XPUSHs (sv_2mortal (decompress_sv (sv, 1)));
278 break; 333 break;
279 334
335 case MAGIC_R_deref:
336 deref = 1;
337 SvPVX (sv)[0] = MAGIC_R;
338 goto handle_MAGIC_R;
339
340 case MAGIC_CR_deref:
341 deref = 1;
280 case MAGIC_CR: 342 case MAGIC_CR:
281 sv = sv_2mortal (decompress_sv (sv, 1)); /* mortal could be optimized */ 343 sv = sv_2mortal (decompress_sv (sv, 1)); /* mortal could be optimized */
344 if (deref)
345 if (SvPVX (sv)[0] == MAGIC_R_deref)
346 SvPVX (sv)[0] = MAGIC_R;
347 else
348 croak ("Compress::LZF::sthaw(): invalid data, maybe you need a newer version of Compress::LZF?");
349
282 case MAGIC_R: 350 case MAGIC_R:
351 handle_MAGIC_R:
283 if (!storable_mstore) 352 if (!storable_mstore)
284 need_storable (); 353 need_storable ();
285 354
286 PUSHMARK (SP); 355 PUSHMARK (SP);
287 XPUSHs (sv); 356 XPUSHs (sv);
… …
290 if (1 != call_sv ((SV *)storable_mretrieve, G_SCALAR)) 359 if (1 != call_sv ((SV *)storable_mretrieve, G_SCALAR))
291 croak ("Storable::mstore didn't return a single scalar"); 360 croak ("Storable::mstore didn't return a single scalar");
292 361
293 SPAGAIN; 362 SPAGAIN;
294 363
295 /*XPUSHs (POPs);*/ 364 if (deref)
365 {
366 SETs (sv_2mortal (SvREFCNT_inc (SvRV (TOPs))));
367
368 if (SvPVX (sv)[0] == MAGIC_R)
369 SvPVX (sv)[0] = MAGIC_R_deref;
370 }
371 else
372 SETs (sv_2mortal (newSVsv (TOPs)));
296 373
297 break; 374 break;
298 375
299 default: 376 default:
300 croak ("Compress::LZF::sthaw(): invalid data, maybe you need a newer version of Compress::LZF?"); 377 croak ("Compress::LZF::sthaw(): invalid data, maybe you need a newer version of Compress::LZF?");
301 } 378 }
302 } 379 }
303 else 380 else
304 XPUSHs (sv_2mortal (SvREFCNT_inc (sv))); 381 XPUSHs (sv_2mortal (newSVsv (sv)));
382

Diff Legend

– Removed lines
+ Added lines
< Changed lines
> Changed lines