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.9 by root, Sun Mar 3 04:45:18 2002 UTC vs.
Revision 1.21 by root, Wed Sep 27 15:52:47 2006 UTC

… …
23#define MAGIC_C 1 /* compressed data follows */ 23#define MAGIC_C 1 /* compressed data follows */
24#define MAGIC_undef 2 /* the special value undef */ 24#define MAGIC_undef 2 /* the special value undef */
25#define MAGIC_CR 3 /* storable (reference, freeze), compressed */ 25#define MAGIC_CR 3 /* storable (reference, freeze), compressed */
26#define MAGIC_R 4 /* storable (reference, freeze) */ 26#define MAGIC_R 4 /* storable (reference, freeze) */
27#define MAGIC_CR_deref 5 /* storable (NO reference, freeze), compressed */ 27#define MAGIC_CR_deref 5 /* storable (NO reference, freeze), compressed */
28#define MAGIC_R_deref 6 /* storable NO (reference, freeze) */ 28#define MAGIC_R_deref 6 /* storable (NO reference, freeze) */
29#define MAGIC_HI 7 /* room for one higher storable major */ 29#define MAGIC_HI 7 /* room for one higher storable major */
30/* for historical reasons, MAGIC_undef + MAGIC_R and MAGIC_undef + MAGIC_R_deref are used, too */
30 31
31#define IN_RANGE(v,l,h) ((unsigned int)((unsigned)(v) - (unsigned)(l)) <= (unsigned)(h) - (unsigned)(l)) 32#define IN_RANGE(v,l,h) ((unsigned int)((unsigned)(v) - (unsigned)(l)) <= (unsigned)(h) - (unsigned)(l))
32 33
34static SV *serializer_package, *serializer_mstore, *serializer_mretrieve;
33static CV *storable_mstore, *storable_mretrieve; 35static CV *storable_mstore, *storable_mretrieve;
34 36
37#if Size_t_size > 4
38# define MAX_LENGTH ((Size_t)0x80000000L)
39#else
40# define MAX_LENGTH ((Size_t) 0x8000000L)
41#endif
42
35static SV * 43static SV *
36compress_sv (SV *data, char cprepend, char uprepend) 44compress_sv (SV *data, char cprepend, int uprepend)
37{ 45{
38 STRLEN usize, csize; 46 STRLEN usize, csize;
39 char *src = (char *)SvPV (data, usize); 47 char *src = (char *)SvPV (data, usize);
40 48
41 if (usize) 49 if (usize)
… …
48 dst = (unsigned char *)SvPVX (ret); 56 dst = (unsigned char *)SvPVX (ret);
49 57
50 if (cprepend) 58 if (cprepend)
51 dst[skip++] = cprepend; 59 dst[skip++] = cprepend;
52 60
53 if (usize < 0x80) 61 if (usize <= 0x7f)
54 { 62 {
55 dst[skip++] = usize; 63 dst[skip++] = usize;
56 } 64 }
57 else if (usize < 0x800) 65 else if (usize <= 0x7ff)
58 { 66 {
59 dst[skip++] = (( usize >> 6) | 0xc0); 67 dst[skip++] = (( usize >> 6) | 0xc0);
60 dst[skip++] = (( usize & 0x3f) | 0x80); 68 dst[skip++] = (( usize & 0x3f) | 0x80);
61 } 69 }
62 else if (usize < 0x10000) 70 else if (usize <= 0xffff)
63 { 71 {
64 dst[skip++] = (( usize >> 12) | 0xe0); 72 dst[skip++] = (( usize >> 12) | 0xe0);
65 dst[skip++] = (((usize >> 6) & 0x3f) | 0x80); 73 dst[skip++] = (((usize >> 6) & 0x3f) | 0x80);
66 dst[skip++] = (( usize & 0x3f) | 0x80); 74 dst[skip++] = (( usize & 0x3f) | 0x80);
67 } 75 }
68 else if (usize < 0x200000) 76 else if (usize <= 0x1fffff)
69 { 77 {
70 dst[skip++] = (( usize >> 18) | 0xf0); 78 dst[skip++] = (( usize >> 18) | 0xf0);
71 dst[skip++] = (((usize >> 12) & 0x3f) | 0x80); 79 dst[skip++] = (((usize >> 12) & 0x3f) | 0x80);
72 dst[skip++] = (((usize >> 6) & 0x3f) | 0x80); 80 dst[skip++] = (((usize >> 6) & 0x3f) | 0x80);
73 dst[skip++] = (( usize & 0x3f) | 0x80); 81 dst[skip++] = (( usize & 0x3f) | 0x80);
74 } 82 }
75 else if (usize < 0x4000000) 83 else if (usize <= 0x3ffffff)
76 { 84 {
77 dst[skip++] = (( usize >> 24) | 0xf8); 85 dst[skip++] = (( usize >> 24) | 0xf8);
78 dst[skip++] = (((usize >> 18) & 0x3f) | 0x80); 86 dst[skip++] = (((usize >> 18) & 0x3f) | 0x80);
79 dst[skip++] = (((usize >> 12) & 0x3f) | 0x80); 87 dst[skip++] = (((usize >> 12) & 0x3f) | 0x80);
80 dst[skip++] = (((usize >> 6) & 0x3f) | 0x80); 88 dst[skip++] = (((usize >> 6) & 0x3f) | 0x80);
81 dst[skip++] = (( usize & 0x3f) | 0x80); 89 dst[skip++] = (( usize & 0x3f) | 0x80);
82 } 90 }
91 else if (usize <= 0x7fffffff)
92 {
93 dst[skip++] = (( usize >> 30) | 0xfc);
94 dst[skip++] = (((usize >> 24) & 0x3f) | 0x80);
95 dst[skip++] = (((usize >> 18) & 0x3f) | 0x80);
96 dst[skip++] = (((usize >> 12) & 0x3f) | 0x80);
97 dst[skip++] = (((usize >> 6) & 0x3f) | 0x80);
98 dst[skip++] = (( usize & 0x3f) | 0x80);
99 }
83 else 100 else
84 croak ("compress can only compress up to %ld bytes", 0x4000000-1); 101 croak ("compress can only compress up to %ld bytes", 0x7fffffffL);
85 102
86 /* 11 bytes is the smallest compressible string */ 103 /* 11 bytes is the smallest compressible string */
87 csize = usize < 11 ? 0 : 104 csize = usize < 11 ? 0 :
88 lzf_compress (src, usize, 105 lzf_compress (src, usize,
89 dst + skip, 106 dst + skip,
… …
91 108
92 if (csize) 109 if (csize)
93 { 110 {
94 SvCUR_set (ret, csize + skip); 111 SvCUR_set (ret, csize + skip);
95 } 112 }
96 else if (!uprepend) 113 else if (uprepend < 0)
97 { 114 {
98 SvREFCNT_dec (ret); 115 SvREFCNT_dec (ret);
99 ret = SvREFCNT_inc (data); 116 ret = SvREFCNT_inc (data);
100 } 117 }
101 else 118 else
102 { 119 {
103 *dst++ = 0; 120 *dst++ = uprepend;
104 121
105 Move ((void *)src, (void *)dst, usize, unsigned char); 122 Move ((void *)src, (void *)dst, usize, unsigned char);
106 123
107 SvCUR_set (ret, usize + 1); 124 SvCUR_set (ret, usize + 1);
108 } 125 }
… …
126 143
127 csize -= skip; 144 csize -= skip;
128 145
129 if (src[0]) 146 if (src[0])
130 { 147 {
131 if (!(src[0] & 0x80)) 148 if (!(src[0] & 0x80) && csize >= 1)
132 { 149 {
133 csize -= 1; 150 csize -= 1;
134 usize = *src++ & 0xff; 151 usize = *src++ & 0xff;
135 } 152 }
136 else if (!(src[0] & 0x20)) 153 else if (!(src[0] & 0x20) && csize >= 2)
137 { 154 {
138 csize -= 2; 155 csize -= 2;
139 usize = *src++ & 0x1f; 156 usize = *src++ & 0x1f;
140 usize = (usize << 6) | (*src++ & 0x3f); 157 usize = (usize << 6) | (*src++ & 0x3f);
141 } 158 }
142 else if (!(src[0] & 0x10)) 159 else if (!(src[0] & 0x10) && csize >= 3)
143 { 160 {
144 csize -= 3; 161 csize -= 3;
145 usize = *src++ & 0x0f; 162 usize = *src++ & 0x0f;
146 usize = (usize << 6) | (*src++ & 0x3f); 163 usize = (usize << 6) | (*src++ & 0x3f);
147 usize = (usize << 6) | (*src++ & 0x3f); 164 usize = (usize << 6) | (*src++ & 0x3f);
148 } 165 }
149 else if (!(src[0] & 0x08)) 166 else if (!(src[0] & 0x08) && csize >= 4)
150 { 167 {
151 csize -= 4; 168 csize -= 4;
152 usize = *src++ & 0x07; 169 usize = *src++ & 0x07;
153 usize = (usize << 6) | (*src++ & 0x3f); 170 usize = (usize << 6) | (*src++ & 0x3f);
154 usize = (usize << 6) | (*src++ & 0x3f); 171 usize = (usize << 6) | (*src++ & 0x3f);
155 usize = (usize << 6) | (*src++ & 0x3f); 172 usize = (usize << 6) | (*src++ & 0x3f);
156 } 173 }
157 else if (!(src[0] & 0x04)) 174 else if (!(src[0] & 0x04) && csize >= 5)
158 { 175 {
159 csize -= 5; 176 csize -= 5;
160 usize = *src++ & 0x03; 177 usize = *src++ & 0x03;
161 usize = (usize << 6) | (*src++ & 0x3f); 178 usize = (usize << 6) | (*src++ & 0x3f);
162 usize = (usize << 6) | (*src++ & 0x3f); 179 usize = (usize << 6) | (*src++ & 0x3f);
163 usize = (usize << 6) | (*src++ & 0x3f); 180 usize = (usize << 6) | (*src++ & 0x3f);
164 usize = (usize << 6) | (*src++ & 0x3f); 181 usize = (usize << 6) | (*src++ & 0x3f);
165 } 182 }
183 else if (!(src[0] & 0x02) && csize >= 6)
184 {
185 csize -= 6;
186 usize = *src++ & 0x01;
187 usize = (usize << 6) | (*src++ & 0x3f);
188 usize = (usize << 6) | (*src++ & 0x3f);
189 usize = (usize << 6) | (*src++ & 0x3f);
190 usize = (usize << 6) | (*src++ & 0x3f);
191 usize = (usize << 6) | (*src++ & 0x3f);
192 }
166 else 193 else
167 croak ("compressed data corrupted"); 194 croak ("compressed data corrupted (invalid length)");
195
196 if (!usize)
197 croak ("compressed data corrupted (invalid length)");
168 198
169 ret = NEWSV (0, usize); 199 ret = NEWSV (0, usize);
170 SvPOK_only (ret); 200 SvPOK_only (ret);
171 dst = SvPVX (ret); 201 dst = SvPVX (ret);
172 202
173 if (lzf_decompress (src, csize, dst, usize) != usize) 203 if (lzf_decompress (src, csize, dst, usize) != usize)
174 croak ("compressed data corrupted", csize, skip, usize); 204 croak ("compressed data corrupted (size mismatch)", csize, skip, usize);
175 } 205 }
176 else 206 else
177 { 207 {
178 usize = csize - 1; 208 usize = csize - 1;
179 ret = NEWSV (0, usize); 209 ret = NEWSV (0, usize | 1);
180 SvPOK_only (ret); 210 SvPOK_only (ret);
181 211
182 Move ((void *)(src + 1), (void *)SvPVX (ret), usize, unsigned char); 212 Move ((void *)(src + 1), (void *)SvPVX (ret), usize, unsigned char);
183 } 213 }
184 214
… …
192 222
193static void 223static void
194need_storable(void) 224need_storable(void)
195{ 225{
196#if PATCHLEVEL < 6 226#if PATCHLEVEL < 6
227 char req[8192];
228 sprintf (req, "require %s;", SvPV_nolen (serializer_package));
197 perl_eval_pv ("require Storable;", 1); 229 perl_eval_pv (req, 1);
198#else 230#else
199 load_module (PERL_LOADMOD_NOIMPORT, newSVpv ("Storable", 0), Nullsv); 231 load_module (PERL_LOADMOD_NOIMPORT, serializer_package, Nullsv);
200#endif 232#endif
201 233
202 storable_mstore = GvCV (gv_fetchpv ("Storable::mstore" , TRUE, SVt_PVCV)); 234 storable_mstore = GvCV (gv_fetchpv (SvPV_nolen (serializer_mstore ), TRUE, SVt_PVCV));
203 storable_mretrieve = GvCV (gv_fetchpv ("Storable::mretrieve", TRUE, SVt_PVCV)); 235 storable_mretrieve = GvCV (gv_fetchpv (SvPV_nolen (serializer_mretrieve), TRUE, SVt_PVCV));
204} 236}
205 237
206MODULE = Compress::LZF PACKAGE = Compress::LZF 238MODULE = Compress::LZF PACKAGE = Compress::LZF
239
240BOOT:
241 serializer_package = newSVpv ("Storable", 0);
242 serializer_mstore = newSVpv ("Storable::net_mstore", 0);
243 serializer_mretrieve = newSVpv ("Storable::mretrieve", 0);
244
245void
246set_serializer(package, mstore, mretrieve)
247 SV * package
248 SV * mstore
249 SV * mretrieve
250 PROTOTYPE: $$$
251 PPCODE:
252 SvSetSV (serializer_package , package );
253 SvSetSV (serializer_mstore , mstore );
254 SvSetSV (serializer_mretrieve, mretrieve);
255 storable_mstore =
256 storable_mretrieve = 0;
207 257
208void 258void
209compress(data) 259compress(data)
210 SV * data 260 SV * data
211 PROTOTYPE: $ 261 PROTOTYPE: $
212 PPCODE: 262 PPCODE:
213 XPUSHs (sv_2mortal (compress_sv (data, 0, 1))); 263 XPUSHs (sv_2mortal (compress_sv (data, 0, MAGIC_U)));
214 264
215void 265void
216decompress(data) 266decompress(data)
217 SV * data 267 SV * data
218 PROTOTYPE: $ 268 PROTOTYPE: $
… …
226 sfreeze_cr = 1 276 sfreeze_cr = 1
227 sfreeze_c = 2 277 sfreeze_c = 2
228 PROTOTYPE: $ 278 PROTOTYPE: $
229 PPCODE: 279 PPCODE:
230 280
281 SvGETMAGIC (sv);
282
231 if (!SvOK (sv)) 283 if (!SvOK (sv))
232 XPUSHs (sv_2mortal (newSVpvn ("\02", 1))); /* 02 == MAGIC_undef */ 284 XPUSHs (sv_2mortal (newSVpvn ("\02", 1))); /* 02 == MAGIC_undef */
285 else if (SvROK (sv)
233 else if (SvTYPE(sv) != SVt_IV 286 || (SvTYPE(sv) != SVt_IV
234 && SvTYPE(sv) != SVt_NV 287 && SvTYPE(sv) != SVt_NV
288 && SvTYPE(sv) != SVt_PV
289 && SvTYPE(sv) != SVt_PVIV
290 && SvTYPE(sv) != SVt_PVNV
235 && SvTYPE(sv) != SVt_PV) /* mstore */ 291 && SvTYPE(sv) != SVt_PVMG)) /* mstore */
236 { 292 {
237 int deref = !SvROK (sv); 293 int deref = !SvROK (sv);
238 294
239 if (!storable_mstore) 295 if (!storable_mstore)
240 need_storable (); 296 need_storable ();
… …
251 307
252 SPAGAIN; 308 SPAGAIN;
253 309
254 sv = POPs; 310 sv = POPs;
255 311
256 if (SvPVX (sv)[0] != MAGIC_R) 312 if (SvPVX (sv)[0] == MAGIC_R)
257 croak ("Storable format changed, need newer version of Compress::LZF");
258
259 if (ix) /* compress */
260 XPUSHs (sv_2mortal (compress_sv (sv, deref ? MAGIC_CR_deref : MAGIC_CR, 0)));
261 else
262 { 313 {
263 if (deref) 314 if (deref)
264 SvPVX (sv)[0] = MAGIC_R_deref; 315 SvPVX (sv)[0] = MAGIC_R_deref;
265
266 XPUSHs (sv);
267 } 316 }
317 else
318 {
319 char pfx[2];
320
321 pfx[0] = MAGIC_undef;
322 pfx[1] = deref ? MAGIC_R_deref : MAGIC_R;
323
324 sv_insert (sv, 0, 0, pfx, 2);
325 }
326
327 if (ix) /* compress */
328 sv = sv_2mortal (compress_sv (sv, deref ? MAGIC_CR_deref : MAGIC_CR, -1));
329
330 XPUSHs (sv);
268 } 331 }
269 else if (sv && IN_RANGE (SvPVX (sv)[0], MAGIC_LO, MAGIC_HI)) 332 else if (SvPOKp (sv) && IN_RANGE (SvPVX (sv)[0], MAGIC_LO, MAGIC_HI))
270 XPUSHs (sv_2mortal (compress_sv (sv, MAGIC_C, 1))); /* need to prefix only */ 333 XPUSHs (sv_2mortal (compress_sv (sv, MAGIC_C, MAGIC_U))); /* need to prefix only */
271 else if (ix == 2) /* compress always */ 334 else if (ix == 2) /* compress always */
272 XPUSHs (sv_2mortal (compress_sv (sv, MAGIC_C, 0))); 335 XPUSHs (sv_2mortal (compress_sv (sv, MAGIC_C, -1)));
336 else if (SvNIOK (sv)) /* don't compress */
337 {
338 STRLEN len;
339 char *s = SvPV (sv, len);
340 XPUSHs (sv_2mortal (newSVpvn (s, len)));
341 }
273 else /* don't compress */ 342 else /* don't compress */
274 XPUSHs (sv_2mortal (SvREFCNT_inc (sv))); 343 XPUSHs (sv_2mortal (newSVsv (sv)));
275 344
276void 345void
277sthaw(sv) 346sthaw(sv)
278 SV * sv 347 SV * sv
279 PROTOTYPE: $ 348 PROTOTYPE: $
280 PPCODE: 349 PPCODE:
281 350{
351 STRLEN svlen;
282 int deref = 0; 352 int deref = 0;
283 353
284 SvGETMAGIC (sv); 354 SvGETMAGIC (sv);
285 if (SvPOK (sv) && IN_RANGE (SvPV_nolen (sv)[0], MAGIC_LO, MAGIC_HI)) 355 if (SvPOK (sv) && IN_RANGE (SvPV (sv, svlen)[0], MAGIC_LO, MAGIC_HI))
286 { 356 {
357 redo:
358
287 switch (SvPVX (sv)[0]) 359 switch (SvPVX (sv)[0])
288 { 360 {
289 case MAGIC_undef: 361 case MAGIC_undef:
362 if (svlen <= 1)
290 XPUSHs (sv_2mortal (NEWSV (0, 0))); 363 XPUSHs (sv_2mortal (NEWSV (0, 0)));
364 else
365 {
366 if (SvPVX (sv)[1] == MAGIC_R_deref)
367 deref = 1;
368 else if (SvPVX (sv)[1] != MAGIC_R)
369 croak ("Compress::LZF::sthaw(): invalid data, maybe you need a newer version of Compress::LZF?");
370
371 sv_chop (sv, SvPVX (sv) + 2);
372
373 if (!storable_mstore)
374 need_storable ();
375
376 PUSHMARK (SP);
377 XPUSHs (sv);
378 PUTBACK;
379
380 if (1 != call_sv ((SV *)storable_mretrieve, G_SCALAR))
381 croak ("Storable::mstore didn't return a single scalar");
382
383 SPAGAIN;
384
385 if (deref)
386 SETs (sv_2mortal (SvREFCNT_inc (SvRV (TOPs))));
387 else
388 SETs (sv_2mortal (newSVsv (TOPs)));
389 }
291 break; 390 break;
292 391
293 case MAGIC_U: 392 case MAGIC_U:
294 XPUSHs (sv_2mortal (decompress_sv (sv, 0))); 393 XPUSHs (sv_2mortal (decompress_sv (sv, 0)));
295 break; 394 break;
296 395
297 case MAGIC_C: 396 case MAGIC_C:
298 XPUSHs (sv_2mortal (decompress_sv (sv, 1))); 397 XPUSHs (sv_2mortal (decompress_sv (sv, 1)));
299 break; 398 break;
399
400 case MAGIC_R_deref:
401 deref = 1;
402 SvPVX (sv)[0] = MAGIC_R;
403 goto handle_MAGIC_R;
300 404
301 case MAGIC_CR_deref: 405 case MAGIC_CR_deref:
302 deref = 1; 406 deref = 1;
303 case MAGIC_CR: 407 case MAGIC_CR:
304 sv = sv_2mortal (decompress_sv (sv, 1)); /* mortal could be optimized */ 408 sv = sv_2mortal (decompress_sv (sv, 1)); /* mortal could be optimized */
305 case MAGIC_R_deref: 409 if (deref)
306 if (SvPVX (sv)[0] == MAGIC_R_deref) 410 if (SvPVX (sv)[0] == MAGIC_R_deref)
307 {
308 deref = 1;
309 SvPVX (sv)[0] = MAGIC_R; 411 SvPVX (sv)[0] = MAGIC_R;
310 } 412
413 goto redo;
414
311 case MAGIC_R: 415 case MAGIC_R:
416 handle_MAGIC_R:
312 if (!storable_mstore) 417 if (!storable_mstore)
313 need_storable (); 418 need_storable ();
314 419
315 PUSHMARK (SP); 420 PUSHMARK (SP);
316 XPUSHs (sv); 421 XPUSHs (sv);
… …
321 426
322 SPAGAIN; 427 SPAGAIN;
323 428
324 if (deref) 429 if (deref)
325 { 430 {
326 SV *ref = SvREFCNT_inc (SvRV (TOPs)); 431 SETs (sv_2mortal (SvREFCNT_inc (SvRV (TOPs))));
327
328 SvREFCNT_dec (TOPs); /* destroy superfluous ref */
329 SETs (ref);
330 432
331 if (SvPVX (sv)[0] == MAGIC_R) 433 if (SvPVX (sv)[0] == MAGIC_R)
332 SvPVX (sv)[0] = MAGIC_R; 434 SvPVX (sv)[0] = MAGIC_R_deref;
333 } 435 }
334 436 else
335 XPUSHs (POPs); /* this is a nop, hopefully */ 437 SETs (sv_2mortal (newSVsv (TOPs)));
336 438
337 break; 439 break;
338 440
339 default: 441 default:
340 croak ("Compress::LZF::sthaw(): invalid data, maybe you need a newer version of Compress::LZF?"); 442 croak ("Compress::LZF::sthaw(): invalid data, maybe you need a newer version of Compress::LZF?");
341 } 443 }
342 } 444 }
343 else 445 else
344 XPUSHs (sv_2mortal (SvREFCNT_inc (sv))); 446 XPUSHs (sv_2mortal (newSVsv (sv)));
447}
448

Diff Legend

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