ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/CBOR-XS/XS.xs
Revision: 1.7
Committed: Sat Oct 26 23:02:55 2013 UTC (10 years, 6 months ago) by root
Branch: MAIN
CVS Tags: rel-0_03
Changes since 1.6: +6 -1 lines
Log Message:
0.03

File Contents

# User Rev Content
1 root 1.1 #include "EXTERN.h"
2     #include "perl.h"
3     #include "XSUB.h"
4    
5     #include <assert.h>
6     #include <string.h>
7     #include <stdlib.h>
8     #include <stdio.h>
9     #include <limits.h>
10     #include <float.h>
11    
12     #include "ecb.h"
13    
14     #define F_SHRINK 0x00000200UL
15     #define F_ALLOW_UNKNOWN 0x00002000UL
16    
17     #define INIT_SIZE 32 // initial scalar size to be allocated
18    
19     #define SB do {
20     #define SE } while (0)
21    
22     #define IN_RANGE_INC(type,val,beg,end) \
23     ((unsigned type)((unsigned type)(val) - (unsigned type)(beg)) \
24     <= (unsigned type)((unsigned type)(end) - (unsigned type)(beg)))
25    
26     #define ERR_NESTING_EXCEEDED "cbor text or perl structure exceeds maximum nesting level (max_depth set too low?)"
27    
28     #ifdef USE_ITHREADS
29     # define CBOR_SLOW 1
30     # define CBOR_STASH (cbor_stash ? cbor_stash : gv_stashpv ("CBOR::XS", 1))
31     #else
32     # define CBOR_SLOW 0
33     # define CBOR_STASH cbor_stash
34     #endif
35    
36 root 1.6 static HV *cbor_stash, *cbor_boolean_stash, *cbor_tagged_stash; // CBOR::XS::
37 root 1.1 static SV *cbor_true, *cbor_false;
38    
39     typedef struct {
40     U32 flags;
41     U32 max_depth;
42     STRLEN max_size;
43     } CBOR;
44    
45 root 1.5 ecb_inline void
46 root 1.1 cbor_init (CBOR *cbor)
47     {
48     Zero (cbor, 1, CBOR);
49     cbor->max_depth = 512;
50     }
51    
52     /////////////////////////////////////////////////////////////////////////////
53     // utility functions
54    
55 root 1.5 ecb_inline SV *
56 root 1.1 get_bool (const char *name)
57     {
58     SV *sv = get_sv (name, 1);
59    
60     SvREADONLY_on (sv);
61     SvREADONLY_on (SvRV (sv));
62    
63     return sv;
64     }
65    
66 root 1.5 ecb_inline void
67 root 1.1 shrink (SV *sv)
68     {
69     sv_utf8_downgrade (sv, 1);
70    
71     if (SvLEN (sv) > SvCUR (sv) + 1)
72     {
73     #ifdef SvPV_shrink_to_cur
74     SvPV_shrink_to_cur (sv);
75     #elif defined (SvPV_renew)
76     SvPV_renew (sv, SvCUR (sv) + 1);
77     #endif
78     }
79     }
80    
81     /////////////////////////////////////////////////////////////////////////////
82     // fp hell
83    
84     //TODO
85    
86     /////////////////////////////////////////////////////////////////////////////
87     // encoder
88    
89     // structure used for encoding CBOR
90     typedef struct
91     {
92     char *cur; // SvPVX (sv) + current output position
93     char *end; // SvEND (sv)
94     SV *sv; // result scalar
95     CBOR cbor;
96     U32 depth; // recursion level
97     } enc_t;
98    
99 root 1.5 ecb_inline void
100 root 1.1 need (enc_t *enc, STRLEN len)
101     {
102 root 1.5 if (ecb_expect_false (enc->cur + len >= enc->end))
103 root 1.1 {
104     STRLEN cur = enc->cur - (char *)SvPVX (enc->sv);
105     SvGROW (enc->sv, cur + (len < (cur >> 2) ? cur >> 2 : len) + 1);
106     enc->cur = SvPVX (enc->sv) + cur;
107     enc->end = SvPVX (enc->sv) + SvLEN (enc->sv) - 1;
108     }
109     }
110    
111 root 1.5 ecb_inline void
112 root 1.1 encode_ch (enc_t *enc, char ch)
113     {
114     need (enc, 1);
115     *enc->cur++ = ch;
116     }
117    
118     static void
119     encode_uint (enc_t *enc, int major, UV len)
120     {
121     need (enc, 9);
122    
123     if (len < 24)
124     *enc->cur++ = major | len;
125 root 1.4 else if (len <= 0xff)
126 root 1.1 {
127     *enc->cur++ = major | 24;
128     *enc->cur++ = len;
129     }
130 root 1.4 else if (len <= 0xffff)
131 root 1.1 {
132     *enc->cur++ = major | 25;
133     *enc->cur++ = len >> 8;
134     *enc->cur++ = len;
135     }
136 root 1.4 else if (len <= 0xffffffff)
137 root 1.1 {
138     *enc->cur++ = major | 26;
139     *enc->cur++ = len >> 24;
140     *enc->cur++ = len >> 16;
141     *enc->cur++ = len >> 8;
142     *enc->cur++ = len;
143     }
144 root 1.4 else
145 root 1.1 {
146     *enc->cur++ = major | 27;
147     *enc->cur++ = len >> 56;
148     *enc->cur++ = len >> 48;
149     *enc->cur++ = len >> 40;
150     *enc->cur++ = len >> 32;
151     *enc->cur++ = len >> 24;
152     *enc->cur++ = len >> 16;
153     *enc->cur++ = len >> 8;
154     *enc->cur++ = len;
155     }
156     }
157    
158     static void
159     encode_str (enc_t *enc, int utf8, char *str, STRLEN len)
160     {
161     encode_uint (enc, utf8 ? 0x60 : 0x40, len);
162     need (enc, len);
163     memcpy (enc->cur, str, len);
164     enc->cur += len;
165     }
166    
167     static void encode_sv (enc_t *enc, SV *sv);
168    
169     static void
170     encode_av (enc_t *enc, AV *av)
171     {
172     int i, len = av_len (av);
173    
174     if (enc->depth >= enc->cbor.max_depth)
175     croak (ERR_NESTING_EXCEEDED);
176    
177     ++enc->depth;
178    
179     encode_uint (enc, 0x80, len + 1);
180    
181     for (i = 0; i <= len; ++i)
182     {
183     SV **svp = av_fetch (av, i, 0);
184     encode_sv (enc, svp ? *svp : &PL_sv_undef);
185     }
186    
187     --enc->depth;
188     }
189    
190     static void
191     encode_hv (enc_t *enc, HV *hv)
192     {
193     HE *he;
194    
195     if (enc->depth >= enc->cbor.max_depth)
196     croak (ERR_NESTING_EXCEEDED);
197    
198     ++enc->depth;
199    
200     int pairs = hv_iterinit (hv);
201     int mg = SvMAGICAL (hv);
202    
203     if (mg)
204     encode_ch (enc, 0xa0 | 31);
205     else
206     encode_uint (enc, 0xa0, pairs);
207    
208     while ((he = hv_iternext (hv)))
209     {
210     if (HeKLEN (he) == HEf_SVKEY)
211     encode_sv (enc, HeSVKEY (he));
212     else
213     encode_str (enc, HeKUTF8 (he), HeKEY (he), HeKLEN (he));
214    
215 root 1.5 encode_sv (enc, ecb_expect_false (mg) ? hv_iterval (hv, he) : HeVAL (he));
216 root 1.1 }
217    
218     if (mg)
219     encode_ch (enc, 0xe0 | 31);
220    
221     --enc->depth;
222     }
223    
224     // encode objects, arrays and special \0=false and \1=true values.
225     static void
226     encode_rv (enc_t *enc, SV *sv)
227     {
228     svtype svt;
229    
230     SvGETMAGIC (sv);
231     svt = SvTYPE (sv);
232    
233 root 1.5 if (ecb_expect_false (SvOBJECT (sv)))
234 root 1.1 {
235 root 1.6 HV *boolean_stash = !CBOR_SLOW || cbor_boolean_stash
236     ? cbor_boolean_stash
237     : gv_stashpv ("CBOR::XS::Boolean", 1);
238     HV *tagged_stash = !CBOR_SLOW || cbor_tagged_stash
239     ? cbor_tagged_stash
240     : gv_stashpv ("CBOR::XS::Tagged" , 1);
241 root 1.1
242 root 1.6 if (SvSTASH (sv) == boolean_stash)
243 root 1.1 encode_ch (enc, SvIV (sv) ? 0xe0 | 21 : 0xe0 | 20);
244 root 1.6 else if (SvSTASH (sv) == tagged_stash)
245     {
246     if (svt != SVt_PVAV)
247     croak ("encountered CBOR::XS::Tagged object that isn't an array");
248    
249     encode_uint (enc, 0xc0, SvUV (*av_fetch ((AV *)sv, 0, 1)));
250     encode_sv (enc, *av_fetch ((AV *)sv, 1, 1));
251     }
252 root 1.1 else
253     {
254 root 1.6 // we re-bless the reference to get overload and other niceties right
255     GV *to_cbor = gv_fetchmethod_autoload (SvSTASH (sv), "TO_CBOR", 0);
256    
257     if (to_cbor)
258 root 1.1 {
259 root 1.6 dSP;
260 root 1.1
261 root 1.6 ENTER; SAVETMPS; PUSHMARK (SP);
262     XPUSHs (sv_bless (sv_2mortal (newRV_inc (sv)), SvSTASH (sv)));
263    
264     // calling with G_SCALAR ensures that we always get a 1 return value
265     PUTBACK;
266     call_sv ((SV *)GvCV (to_cbor), G_SCALAR);
267     SPAGAIN;
268    
269     // catch this surprisingly common error
270     if (SvROK (TOPs) && SvRV (TOPs) == sv)
271     croak ("%s::TO_CBOR method returned same object as was passed instead of a new one", HvNAME (SvSTASH (sv)));
272    
273     sv = POPs;
274     PUTBACK;
275    
276     encode_sv (enc, sv);
277    
278     FREETMPS; LEAVE;
279 root 1.1 }
280     else
281 root 1.6 croak ("encountered object '%s', but no TO_CBOR method available on it",
282 root 1.1 SvPV_nolen (sv_2mortal (newRV_inc (sv))));
283     }
284     }
285     else if (svt == SVt_PVHV)
286     encode_hv (enc, (HV *)sv);
287     else if (svt == SVt_PVAV)
288     encode_av (enc, (AV *)sv);
289     else if (svt < SVt_PVAV)
290     {
291     STRLEN len = 0;
292     char *pv = svt ? SvPV (sv, len) : 0;
293    
294     if (len == 1 && *pv == '1')
295     encode_ch (enc, 0xe0 | 21);
296     else if (len == 1 && *pv == '0')
297     encode_ch (enc, 0xe0 | 20);
298     else if (enc->cbor.flags & F_ALLOW_UNKNOWN)
299     encode_ch (enc, 0xe0 | 23);
300     else
301     croak ("cannot encode reference to scalar '%s' unless the scalar is 0 or 1",
302     SvPV_nolen (sv_2mortal (newRV_inc (sv))));
303     }
304     else if (enc->cbor.flags & F_ALLOW_UNKNOWN)
305     encode_ch (enc, 0xe0 | 23);
306     else
307     croak ("encountered %s, but CBOR can only represent references to arrays or hashes",
308     SvPV_nolen (sv_2mortal (newRV_inc (sv))));
309     }
310    
311     static void
312     encode_nv (enc_t *enc, SV *sv)
313     {
314     double nv = SvNVX (sv);
315    
316     need (enc, 9);
317    
318 root 1.5 if (ecb_expect_false (nv == (U32)nv))
319 root 1.1 encode_uint (enc, 0x00, (U32)nv);
320     //TODO: maybe I32?
321 root 1.5 else if (ecb_expect_false (nv == (float)nv))
322 root 1.1 {
323     uint32_t fp = ecb_float_to_binary32 (nv);
324    
325     *enc->cur++ = 0xe0 | 26;
326    
327     if (!ecb_big_endian ())
328     fp = ecb_bswap32 (fp);
329    
330     memcpy (enc->cur, &fp, 4);
331     enc->cur += 4;
332     }
333     else
334     {
335     uint64_t fp = ecb_double_to_binary64 (nv);
336    
337     *enc->cur++ = 0xe0 | 27;
338    
339     if (!ecb_big_endian ())
340     fp = ecb_bswap64 (fp);
341    
342     memcpy (enc->cur, &fp, 8);
343     enc->cur += 8;
344     }
345     }
346    
347     static void
348     encode_sv (enc_t *enc, SV *sv)
349     {
350     SvGETMAGIC (sv);
351    
352     if (SvPOKp (sv))
353     {
354     STRLEN len;
355     char *str = SvPV (sv, len);
356     encode_str (enc, SvUTF8 (sv), str, len);
357     }
358     else if (SvNOKp (sv))
359     encode_nv (enc, sv);
360     else if (SvIOKp (sv))
361     {
362     if (SvIsUV (sv))
363     encode_uint (enc, 0x00, SvUVX (sv));
364     else if (SvIVX (sv) >= 0)
365     encode_uint (enc, 0x00, SvIVX (sv));
366     else
367     encode_uint (enc, 0x20, -(SvIVX (sv) + 1));
368     }
369     else if (SvROK (sv))
370     encode_rv (enc, SvRV (sv));
371     else if (!SvOK (sv))
372     encode_ch (enc, 0xe0 | 22);
373     else if (enc->cbor.flags & F_ALLOW_UNKNOWN)
374     encode_ch (enc, 0xe0 | 23);
375     else
376     croak ("encountered perl type (%s,0x%x) that CBOR cannot handle, check your input data",
377     SvPV_nolen (sv), (unsigned int)SvFLAGS (sv));
378     }
379    
380     static SV *
381     encode_cbor (SV *scalar, CBOR *cbor)
382     {
383     enc_t enc;
384    
385     enc.cbor = *cbor;
386     enc.sv = sv_2mortal (NEWSV (0, INIT_SIZE));
387     enc.cur = SvPVX (enc.sv);
388     enc.end = SvEND (enc.sv);
389     enc.depth = 0;
390    
391     SvPOK_only (enc.sv);
392     encode_sv (&enc, scalar);
393    
394     SvCUR_set (enc.sv, enc.cur - SvPVX (enc.sv));
395     *SvEND (enc.sv) = 0; // many xs functions expect a trailing 0 for text strings
396    
397     if (enc.cbor.flags & F_SHRINK)
398     shrink (enc.sv);
399    
400     return enc.sv;
401     }
402    
403     /////////////////////////////////////////////////////////////////////////////
404     // decoder
405    
406     // structure used for decoding CBOR
407     typedef struct
408     {
409     U8 *cur; // current parser pointer
410     U8 *end; // end of input string
411     const char *err; // parse error, if != 0
412     CBOR cbor;
413     U32 depth; // recursion depth
414     U32 maxdepth; // recursion depth limit
415     } dec_t;
416    
417     #define ERR(reason) SB if (!dec->err) dec->err = reason; goto fail; SE
418    
419 root 1.5 #define WANT(len) if (ecb_expect_false (dec->cur + len > dec->end)) ERR ("unexpected end of CBOR data")
420 root 1.1
421     #define DEC_INC_DEPTH if (++dec->depth > dec->cbor.max_depth) ERR (ERR_NESTING_EXCEEDED)
422     #define DEC_DEC_DEPTH --dec->depth
423    
424     static UV
425     decode_uint (dec_t *dec)
426     {
427     switch (*dec->cur & 31)
428     {
429     case 0: case 1: case 2: case 3: case 4: case 5: case 6: case 7:
430     case 8: case 9: case 10: case 11: case 12: case 13: case 14: case 15:
431     case 16: case 17: case 18: case 19: case 20: case 21: case 22: case 23:
432     return *dec->cur++ & 31;
433    
434     case 24:
435     WANT (2);
436     dec->cur += 2;
437     return dec->cur[-1];
438    
439     case 25:
440     WANT (3);
441     dec->cur += 3;
442     return (((UV)dec->cur[-2]) << 8)
443     | ((UV)dec->cur[-1]);
444    
445     case 26:
446     WANT (5);
447     dec->cur += 5;
448     return (((UV)dec->cur[-4]) << 24)
449     | (((UV)dec->cur[-3]) << 16)
450     | (((UV)dec->cur[-2]) << 8)
451     | ((UV)dec->cur[-1]);
452    
453     case 27:
454     WANT (9);
455     dec->cur += 9;
456     return (((UV)dec->cur[-8]) << 56)
457     | (((UV)dec->cur[-7]) << 48)
458     | (((UV)dec->cur[-6]) << 40)
459     | (((UV)dec->cur[-5]) << 32)
460     | (((UV)dec->cur[-4]) << 24)
461     | (((UV)dec->cur[-3]) << 16)
462     | (((UV)dec->cur[-2]) << 8)
463     | ((UV)dec->cur[-1]);
464    
465     default:
466     ERR ("corrupted CBOR data (unsupported integer minor encoding)");
467     }
468    
469     fail:
470     return 0;
471     }
472    
473     static SV *decode_sv (dec_t *dec);
474    
475     static SV *
476     decode_av (dec_t *dec)
477     {
478     AV *av = newAV ();
479    
480     DEC_INC_DEPTH;
481    
482     if ((*dec->cur & 31) == 31)
483     {
484     ++dec->cur;
485    
486     for (;;)
487     {
488     WANT (1);
489    
490 root 1.2 if (*dec->cur == (0xe0 | 31))
491 root 1.1 {
492     ++dec->cur;
493     break;
494     }
495    
496     av_push (av, decode_sv (dec));
497     }
498     }
499     else
500     {
501     int i, len = decode_uint (dec);
502    
503     av_fill (av, len - 1);
504    
505     for (i = 0; i < len; ++i)
506     AvARRAY (av)[i] = decode_sv (dec);
507     }
508    
509     DEC_DEC_DEPTH;
510     return newRV_noinc ((SV *)av);
511    
512     fail:
513     SvREFCNT_dec (av);
514     DEC_DEC_DEPTH;
515     return &PL_sv_undef;
516     }
517    
518     static SV *
519     decode_hv (dec_t *dec)
520     {
521     HV *hv = newHV ();
522    
523     DEC_INC_DEPTH;
524    
525     if ((*dec->cur & 31) == 31)
526     {
527     ++dec->cur;
528    
529     for (;;)
530     {
531     WANT (1);
532    
533 root 1.2 if (*dec->cur == (0xe0 | 31))
534 root 1.1 {
535     ++dec->cur;
536     break;
537     }
538    
539     SV *k = decode_sv (dec);
540     SV *v = decode_sv (dec);
541    
542     hv_store_ent (hv, k, v, 0);
543     }
544     }
545     else
546     {
547     int len = decode_uint (dec);
548    
549     while (len--)
550     {
551     SV *k = decode_sv (dec);
552     SV *v = decode_sv (dec);
553    
554     hv_store_ent (hv, k, v, 0);
555     }
556     }
557    
558     DEC_DEC_DEPTH;
559     return newRV_noinc ((SV *)hv);
560    
561     fail:
562     SvREFCNT_dec (hv);
563     DEC_DEC_DEPTH;
564     return &PL_sv_undef;
565     }
566    
567     static SV *
568     decode_str (dec_t *dec, int utf8)
569     {
570 root 1.6 SV *sv = 0;
571 root 1.1
572     if ((*dec->cur & 31) == 31)
573     {
574     ++dec->cur;
575    
576     sv = newSVpvn ("", 0);
577    
578     // not very fast, and certainly not robust against illegal input
579     for (;;)
580     {
581     WANT (1);
582    
583 root 1.2 if (*dec->cur == (0xe0 | 31))
584 root 1.1 {
585     ++dec->cur;
586     break;
587     }
588    
589 root 1.6 sv_catsv (sv, decode_sv (dec));
590 root 1.1 }
591     }
592     else
593     {
594     STRLEN len = decode_uint (dec);
595    
596     WANT (len);
597     sv = newSVpvn (dec->cur, len);
598     dec->cur += len;
599     }
600    
601     if (utf8)
602     SvUTF8_on (sv);
603    
604     return sv;
605    
606     fail:
607 root 1.6 SvREFCNT_dec (sv);
608 root 1.1 return &PL_sv_undef;
609     }
610    
611     static SV *
612 root 1.3 decode_tagged (dec_t *dec)
613     {
614     UV tag = decode_uint (dec);
615     SV *sv = decode_sv (dec);
616    
617     if (tag == 55799) // 2.4.5 Self-Describe CBOR
618     return sv;
619    
620     AV *av = newAV ();
621     av_push (av, newSVuv (tag));
622     av_push (av, sv);
623 root 1.7
624     HV *tagged_stash = !CBOR_SLOW || cbor_tagged_stash
625     ? cbor_tagged_stash
626     : gv_stashpv ("CBOR::XS::Tagged" , 1);
627    
628     return sv_bless (newRV_noinc ((SV *)av), tagged_stash);
629 root 1.3 }
630    
631     static SV *
632 root 1.1 decode_sv (dec_t *dec)
633     {
634     WANT (1);
635    
636     switch (*dec->cur >> 5)
637     {
638     case 0: // unsigned int
639     return newSVuv (decode_uint (dec));
640     case 1: // negative int
641     return newSViv (-1 - (IV)decode_uint (dec));
642     case 2: // octet string
643     return decode_str (dec, 0);
644     case 3: // utf-8 string
645     return decode_str (dec, 1);
646     case 4: // array
647     return decode_av (dec);
648     case 5: // map
649     return decode_hv (dec);
650     case 6: // tag
651 root 1.3 return decode_tagged (dec);
652 root 1.1 case 7: // misc
653     switch (*dec->cur++ & 31)
654     {
655     case 20:
656     #if CBOR_SLOW
657     cbor_false = get_bool ("CBOR::XS::false");
658     #endif
659     return newSVsv (cbor_false);
660     case 21:
661     #if CBOR_SLOW
662     cbor_true = get_bool ("CBOR::XS::true");
663     #endif
664     return newSVsv (cbor_true);
665     case 22:
666     return newSVsv (&PL_sv_undef);
667    
668     case 25:
669 root 1.2 {
670     WANT (2);
671    
672     uint16_t fp = (dec->cur[0] << 8) | dec->cur[1];
673     dec->cur += 2;
674    
675     return newSVnv (ecb_binary16_to_float (fp));
676     }
677 root 1.1
678     case 26:
679     {
680     uint32_t fp;
681     WANT (4);
682     memcpy (&fp, dec->cur, 4);
683     dec->cur += 4;
684    
685     if (!ecb_big_endian ())
686     fp = ecb_bswap32 (fp);
687    
688     return newSVnv (ecb_binary32_to_float (fp));
689     }
690    
691     case 27:
692     {
693     uint64_t fp;
694     WANT (8);
695     memcpy (&fp, dec->cur, 8);
696     dec->cur += 8;
697    
698     if (!ecb_big_endian ())
699     fp = ecb_bswap64 (fp);
700    
701     return newSVnv (ecb_binary64_to_double (fp));
702     }
703    
704     // 0..19 unassigned
705     // 24 reserved + unassigned (reserved values are not encodable)
706     default:
707     ERR ("corrupted CBOR data (reserved/unassigned major 7 value)");
708     }
709    
710     break;
711     }
712    
713     fail:
714     return &PL_sv_undef;
715     }
716    
717     static SV *
718     decode_cbor (SV *string, CBOR *cbor, char **offset_return)
719     {
720     dec_t dec;
721     SV *sv;
722    
723     /* work around bugs in 5.10 where manipulating magic values
724     * makes perl ignore the magic in subsequent accesses.
725     * also make a copy of non-PV values, to get them into a clean
726     * state (SvPV should do that, but it's buggy, see below).
727     */
728     /*SvGETMAGIC (string);*/
729     if (SvMAGICAL (string) || !SvPOK (string))
730     string = sv_2mortal (newSVsv (string));
731    
732     SvUPGRADE (string, SVt_PV);
733    
734     /* work around a bug in perl 5.10, which causes SvCUR to fail an
735     * assertion with -DDEBUGGING, although SvCUR is documented to
736     * return the xpv_cur field which certainly exists after upgrading.
737     * according to nicholas clark, calling SvPOK fixes this.
738     * But it doesn't fix it, so try another workaround, call SvPV_nolen
739     * and hope for the best.
740     * Damnit, SvPV_nolen still trips over yet another assertion. This
741     * assertion business is seriously broken, try yet another workaround
742     * for the broken -DDEBUGGING.
743     */
744     {
745     #ifdef DEBUGGING
746     STRLEN offset = SvOK (string) ? sv_len (string) : 0;
747     #else
748     STRLEN offset = SvCUR (string);
749     #endif
750    
751     if (offset > cbor->max_size && cbor->max_size)
752     croak ("attempted decode of CBOR text of %lu bytes size, but max_size is set to %lu",
753     (unsigned long)SvCUR (string), (unsigned long)cbor->max_size);
754     }
755    
756     sv_utf8_downgrade (string, 0);
757    
758     dec.cbor = *cbor;
759     dec.cur = (U8 *)SvPVX (string);
760     dec.end = (U8 *)SvEND (string);
761     dec.err = 0;
762     dec.depth = 0;
763    
764     sv = decode_sv (&dec);
765    
766     if (offset_return)
767     *offset_return = dec.cur;
768    
769     if (!(offset_return || !sv))
770 root 1.2 if (dec.cur != dec.end && !dec.err)
771     dec.err = "garbage after CBOR object";
772    
773     if (dec.err)
774 root 1.1 {
775 root 1.2 SvREFCNT_dec (sv);
776     croak ("%s, at offset %d (octet 0x%02x)", dec.err, dec.cur - (U8 *)SvPVX (string), (int)(uint8_t)*dec.cur);
777 root 1.1 }
778    
779     sv = sv_2mortal (sv);
780    
781     return sv;
782     }
783    
784     /////////////////////////////////////////////////////////////////////////////
785     // XS interface functions
786    
787     MODULE = CBOR::XS PACKAGE = CBOR::XS
788    
789     BOOT:
790     {
791     cbor_stash = gv_stashpv ("CBOR::XS" , 1);
792     cbor_boolean_stash = gv_stashpv ("CBOR::XS::Boolean", 1);
793 root 1.6 cbor_tagged_stash = gv_stashpv ("CBOR::XS::Tagged" , 1);
794 root 1.1
795     cbor_true = get_bool ("CBOR::XS::true");
796     cbor_false = get_bool ("CBOR::XS::false");
797     }
798    
799     PROTOTYPES: DISABLE
800    
801     void CLONE (...)
802     CODE:
803     cbor_stash = 0;
804     cbor_boolean_stash = 0;
805 root 1.6 cbor_tagged_stash = 0;
806 root 1.1
807     void new (char *klass)
808     PPCODE:
809     {
810     SV *pv = NEWSV (0, sizeof (CBOR));
811     SvPOK_only (pv);
812     cbor_init ((CBOR *)SvPVX (pv));
813     XPUSHs (sv_2mortal (sv_bless (
814     newRV_noinc (pv),
815     strEQ (klass, "CBOR::XS") ? CBOR_STASH : gv_stashpv (klass, 1)
816     )));
817     }
818    
819     void shrink (CBOR *self, int enable = 1)
820     ALIAS:
821     shrink = F_SHRINK
822     allow_unknown = F_ALLOW_UNKNOWN
823     PPCODE:
824     {
825     if (enable)
826     self->flags |= ix;
827     else
828     self->flags &= ~ix;
829    
830     XPUSHs (ST (0));
831     }
832    
833     void get_shrink (CBOR *self)
834     ALIAS:
835     get_shrink = F_SHRINK
836     get_allow_unknown = F_ALLOW_UNKNOWN
837     PPCODE:
838     XPUSHs (boolSV (self->flags & ix));
839    
840     void max_depth (CBOR *self, U32 max_depth = 0x80000000UL)
841     PPCODE:
842     self->max_depth = max_depth;
843     XPUSHs (ST (0));
844    
845     U32 get_max_depth (CBOR *self)
846     CODE:
847     RETVAL = self->max_depth;
848     OUTPUT:
849     RETVAL
850    
851     void max_size (CBOR *self, U32 max_size = 0)
852     PPCODE:
853     self->max_size = max_size;
854     XPUSHs (ST (0));
855    
856     int get_max_size (CBOR *self)
857     CODE:
858     RETVAL = self->max_size;
859     OUTPUT:
860     RETVAL
861    
862     #if 0 //TODO
863    
864     void filter_cbor_object (CBOR *self, SV *cb = &PL_sv_undef)
865     PPCODE:
866     {
867     SvREFCNT_dec (self->cb_object);
868     self->cb_object = SvOK (cb) ? newSVsv (cb) : 0;
869    
870     XPUSHs (ST (0));
871     }
872    
873     void filter_cbor_single_key_object (CBOR *self, SV *key, SV *cb = &PL_sv_undef)
874     PPCODE:
875     {
876     if (!self->cb_sk_object)
877     self->cb_sk_object = newHV ();
878    
879     if (SvOK (cb))
880     hv_store_ent (self->cb_sk_object, key, newSVsv (cb), 0);
881     else
882     {
883     hv_delete_ent (self->cb_sk_object, key, G_DISCARD, 0);
884    
885     if (!HvKEYS (self->cb_sk_object))
886     {
887     SvREFCNT_dec (self->cb_sk_object);
888     self->cb_sk_object = 0;
889     }
890     }
891    
892     XPUSHs (ST (0));
893     }
894    
895     #endif
896    
897     void encode (CBOR *self, SV *scalar)
898     PPCODE:
899     PUTBACK; scalar = encode_cbor (scalar, self); SPAGAIN;
900     XPUSHs (scalar);
901    
902     void decode (CBOR *self, SV *cborstr)
903     PPCODE:
904     PUTBACK; cborstr = decode_cbor (cborstr, self, 0); SPAGAIN;
905     XPUSHs (cborstr);
906    
907     void decode_prefix (CBOR *self, SV *cborstr)
908     PPCODE:
909     {
910     SV *sv;
911     char *offset;
912     PUTBACK; sv = decode_cbor (cborstr, self, &offset); SPAGAIN;
913     EXTEND (SP, 2);
914     PUSHs (sv);
915     PUSHs (sv_2mortal (newSVuv (offset - SvPVX (cborstr))));
916     }
917    
918 root 1.6 #if 0
919    
920 root 1.1 void DESTROY (CBOR *self)
921     CODE:
922     SvREFCNT_dec (self->cb_sk_object);
923     SvREFCNT_dec (self->cb_object);
924    
925 root 1.6 #endif
926    
927 root 1.1 PROTOTYPES: ENABLE
928    
929     void encode_cbor (SV *scalar)
930     PPCODE:
931     {
932     CBOR cbor;
933     cbor_init (&cbor);
934     PUTBACK; scalar = encode_cbor (scalar, &cbor); SPAGAIN;
935     XPUSHs (scalar);
936     }
937    
938     void decode_cbor (SV *cborstr)
939     PPCODE:
940     {
941     CBOR cbor;
942     cbor_init (&cbor);
943     PUTBACK; cborstr = decode_cbor (cborstr, &cbor, 0); SPAGAIN;
944     XPUSHs (cborstr);
945     }
946