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

Comparing CBOR-XS/XS.xs (file contents):
Revision 1.16 by root, Tue Oct 29 20:59:16 2013 UTC vs.
Revision 1.57 by root, Fri Nov 25 12:16:12 2016 UTC

7#include <stdlib.h> 7#include <stdlib.h>
8#include <stdio.h> 8#include <stdio.h>
9#include <limits.h> 9#include <limits.h>
10#include <float.h> 10#include <float.h>
11 11
12#define ECB_NO_THREADS 1
12#include "ecb.h" 13#include "ecb.h"
13 14
14// compatibility with perl <5.18 15// compatibility with perl <5.18
15#ifndef HvNAMELEN_get 16#ifndef HvNAMELEN_get
16# define HvNAMELEN_get(hv) strlen (HvNAME (hv)) 17# define HvNAMELEN_get(hv) strlen (HvNAME (hv))
19# define HvNAMELEN(hv) HvNAMELEN_get (hv) 20# define HvNAMELEN(hv) HvNAMELEN_get (hv)
20#endif 21#endif
21#ifndef HvNAMEUTF8 22#ifndef HvNAMEUTF8
22# define HvNAMEUTF8(hv) 0 23# define HvNAMEUTF8(hv) 0
23#endif 24#endif
25#ifndef SvREFCNT_dec_NN
26# define SvREFCNT_dec_NN(sv) SvREFCNT_dec (sv)
27#endif
28
29// known major and minor types
30enum cbor_type
31{
32 MAJOR_SHIFT = 5,
33 MINOR_MASK = 0x1f,
34
35 MAJOR_POS_INT = 0 << MAJOR_SHIFT,
36 MAJOR_NEG_INT = 1 << MAJOR_SHIFT,
37 MAJOR_BYTES = 2 << MAJOR_SHIFT,
38 MAJOR_TEXT = 3 << MAJOR_SHIFT,
39 MAJOR_ARRAY = 4 << MAJOR_SHIFT,
40 MAJOR_MAP = 5 << MAJOR_SHIFT,
41 MAJOR_TAG = 6 << MAJOR_SHIFT,
42 MAJOR_MISC = 7 << MAJOR_SHIFT,
43
44 // INT/STRING/ARRAY/MAP subtypes
45 LENGTH_EXT1 = 24,
46 LENGTH_EXT2 = 25,
47 LENGTH_EXT4 = 26,
48 LENGTH_EXT8 = 27,
49
50 // SIMPLE types (effectively MISC subtypes)
51 SIMPLE_FALSE = 20,
52 SIMPLE_TRUE = 21,
53 SIMPLE_NULL = 22,
54 SIMPLE_UNDEF = 23,
55
56 // MISC subtype (unused)
57 MISC_EXT1 = 24,
58 MISC_FLOAT16 = 25,
59 MISC_FLOAT32 = 26,
60 MISC_FLOAT64 = 27,
61
62 // BYTES/TEXT/ARRAY/MAP
63 MINOR_INDEF = 31,
64};
24 65
25// known tags 66// known tags
26enum cbor_tag 67enum cbor_tag
27{ 68{
28 // inofficial extensions (pending iana registration) 69 // extensions
29 CBOR_TAG_PERL_OBJECT = 256, 70 CBOR_TAG_STRINGREF = 25, // http://cbor.schmorp.de/stringref
30 CBOR_TAG_GENERIC_OBJECT = 257, 71 CBOR_TAG_PERL_OBJECT = 26, // http://cbor.schmorp.de/perl-object
72 CBOR_TAG_GENERIC_OBJECT = 27, // http://cbor.schmorp.de/generic-object
73 CBOR_TAG_VALUE_SHAREABLE = 28, // http://cbor.schmorp.de/value-sharing
74 CBOR_TAG_VALUE_SHAREDREF = 29, // http://cbor.schmorp.de/value-sharing
75 CBOR_TAG_STRINGREF_NAMESPACE = 256, // http://cbor.schmorp.de/stringref
76 CBOR_TAG_INDIRECTION = 22098, // http://cbor.schmorp.de/indirection
31 77
32 // rfc7049 78 // rfc7049
33 CBOR_TAG_DATETIME = 0, // rfc4287, utf-8 79 CBOR_TAG_DATETIME = 0, // rfc4287, utf-8
34 CBOR_TAG_TIMESTAMP = 1, // unix timestamp, any 80 CBOR_TAG_TIMESTAMP = 1, // unix timestamp, any
35 CBOR_TAG_POS_BIGNUM = 2, // byte string 81 CBOR_TAG_POS_BIGNUM = 2, // byte string
36 CBOR_TAG_NEG_BIGNUM = 3, // byte string 82 CBOR_TAG_NEG_BIGNUM = 3, // byte string
37 CBOR_TAG_DECIMAL = 4, // decimal fraction, array 83 CBOR_TAG_DECIMAL = 4, // decimal fraction, array
38 CBOR_TAG_BIGFLOAT = 5, // array 84 CBOR_TAG_BIGFLOAT = 5, // array
39 85
40 CBOR_TAG_CONV_B64U = 21, // base64url, any 86 CBOR_TAG_CONV_B64U = 21, // base64url, any
41 CBOR_TAG_CONV_B64 = 22, // base64, any 87 CBOR_TAG_CONV_B64 = 22, // base64, any
42 CBOR_TAG_CONV_HEX = 23, // base16, any 88 CBOR_TAG_CONV_HEX = 23, // base16, any
43 CBOR_TAG_CBOR = 24, // embedded cbor, byte string 89 CBOR_TAG_CBOR = 24, // embedded cbor, byte string
44 90
45 CBOR_TAG_URI = 32, // URI rfc3986, utf-8 91 CBOR_TAG_URI = 32, // URI rfc3986, utf-8
46 CBOR_TAG_B64U = 33, // base64url rfc4648, utf-8 92 CBOR_TAG_B64U = 33, // base64url rfc4648, utf-8
47 CBOR_TAG_B64 = 34, // base6 rfc46484, utf-8 93 CBOR_TAG_B64 = 34, // base6 rfc46484, utf-8
48 CBOR_TAG_REGEX = 35, // regex pcre/ecma262, utf-8 94 CBOR_TAG_REGEX = 35, // regex pcre/ecma262, utf-8
49 CBOR_TAG_MIME = 36, // mime message rfc2045, utf-8 95 CBOR_TAG_MIME = 36, // mime message rfc2045, utf-8
50 96
51 CBOR_TAG_MAGIC = 55799 // self-describe cbor 97 CBOR_TAG_MAGIC = 55799, // self-describe cbor
52}; 98};
53 99
54#define F_SHRINK 0x00000200UL 100#define F_SHRINK 0x00000001UL
55#define F_ALLOW_UNKNOWN 0x00002000UL 101#define F_ALLOW_UNKNOWN 0x00000002UL
102#define F_ALLOW_SHARING 0x00000004UL
103#define F_ALLOW_CYCLES 0x00000008UL
104#define F_PACK_STRINGS 0x00000010UL
105#define F_TEXT_KEYS 0x00000020UL
106#define F_TEXT_STRINGS 0x00000040UL
107#define F_VALIDATE_UTF8 0x00000080UL
56 108
57#define INIT_SIZE 32 // initial scalar size to be allocated 109#define INIT_SIZE 32 // initial scalar size to be allocated
58 110
59#define SB do { 111#define SB do {
60#define SE } while (0) 112#define SE } while (0)
72# define CBOR_SLOW 0 124# define CBOR_SLOW 0
73# define CBOR_STASH cbor_stash 125# define CBOR_STASH cbor_stash
74#endif 126#endif
75 127
76static HV *cbor_stash, *types_boolean_stash, *types_error_stash, *cbor_tagged_stash; // CBOR::XS:: 128static HV *cbor_stash, *types_boolean_stash, *types_error_stash, *cbor_tagged_stash; // CBOR::XS::
77static SV *types_true, *types_false, *types_error, *sv_cbor; 129static SV *types_true, *types_false, *types_error, *sv_cbor, *default_filter;
78 130
79typedef struct { 131typedef struct {
80 U32 flags; 132 U32 flags;
81 U32 max_depth; 133 U32 max_depth;
82 STRLEN max_size; 134 STRLEN max_size;
135 SV *filter;
136
137 // for the incremental parser
138 STRLEN incr_pos; // the current offset into the text
139 STRLEN incr_need; // minimum bytes needed to decode
140 AV *incr_count; // for every nesting level, the number of outstanding values, or -1 for indef.
83} CBOR; 141} CBOR;
84 142
85ecb_inline void 143ecb_inline void
86cbor_init (CBOR *cbor) 144cbor_init (CBOR *cbor)
87{ 145{
88 Zero (cbor, 1, CBOR); 146 Zero (cbor, 1, CBOR);
89 cbor->max_depth = 512; 147 cbor->max_depth = 512;
148}
149
150ecb_inline void
151cbor_free (CBOR *cbor)
152{
153 SvREFCNT_dec (cbor->filter);
154 SvREFCNT_dec (cbor->incr_count);
90} 155}
91 156
92///////////////////////////////////////////////////////////////////////////// 157/////////////////////////////////////////////////////////////////////////////
93// utility functions 158// utility functions
94 159
116 SvPV_renew (sv, SvCUR (sv) + 1); 181 SvPV_renew (sv, SvCUR (sv) + 1);
117#endif 182#endif
118 } 183 }
119} 184}
120 185
121///////////////////////////////////////////////////////////////////////////// 186// minimum length of a string to be registered for stringref
122// fp hell 187ecb_inline int
123 188minimum_string_length (UV idx)
124//TODO 189{
190 return idx > 23
191 ? idx > 0xffU
192 ? idx > 0xffffU
193 ? idx > 0xffffffffU
194 ? 11
195 : 7
196 : 5
197 : 4
198 : 3;
199}
125 200
126///////////////////////////////////////////////////////////////////////////// 201/////////////////////////////////////////////////////////////////////////////
127// encoder 202// encoder
128 203
129// structure used for encoding CBOR 204// structure used for encoding CBOR
132 char *cur; // SvPVX (sv) + current output position 207 char *cur; // SvPVX (sv) + current output position
133 char *end; // SvEND (sv) 208 char *end; // SvEND (sv)
134 SV *sv; // result scalar 209 SV *sv; // result scalar
135 CBOR cbor; 210 CBOR cbor;
136 U32 depth; // recursion level 211 U32 depth; // recursion level
212 HV *stringref[2]; // string => index, or 0 ([0] = bytes, [1] = utf-8)
213 UV stringref_idx;
214 HV *shareable; // ptr => index, or 0
215 UV shareable_idx;
137} enc_t; 216} enc_t;
138 217
139ecb_inline void 218ecb_inline void
140need (enc_t *enc, STRLEN len) 219need (enc_t *enc, STRLEN len)
141{ 220{
158static void 237static void
159encode_uint (enc_t *enc, int major, UV len) 238encode_uint (enc_t *enc, int major, UV len)
160{ 239{
161 need (enc, 9); 240 need (enc, 9);
162 241
163 if (len < 24) 242 if (ecb_expect_true (len < LENGTH_EXT1))
164 *enc->cur++ = major | len; 243 *enc->cur++ = major | len;
165 else if (len <= 0xff) 244 else if (ecb_expect_true (len <= 0xffU))
166 { 245 {
167 *enc->cur++ = major | 24; 246 *enc->cur++ = major | LENGTH_EXT1;
168 *enc->cur++ = len; 247 *enc->cur++ = len;
169 } 248 }
170 else if (len <= 0xffff) 249 else if (len <= 0xffffU)
171 { 250 {
172 *enc->cur++ = major | 25; 251 *enc->cur++ = major | LENGTH_EXT2;
173 *enc->cur++ = len >> 8; 252 *enc->cur++ = len >> 8;
174 *enc->cur++ = len; 253 *enc->cur++ = len;
175 } 254 }
176 else if (len <= 0xffffffff) 255 else if (len <= 0xffffffffU)
177 { 256 {
178 *enc->cur++ = major | 26; 257 *enc->cur++ = major | LENGTH_EXT4;
179 *enc->cur++ = len >> 24; 258 *enc->cur++ = len >> 24;
180 *enc->cur++ = len >> 16; 259 *enc->cur++ = len >> 16;
181 *enc->cur++ = len >> 8; 260 *enc->cur++ = len >> 8;
182 *enc->cur++ = len; 261 *enc->cur++ = len;
183 } 262 }
184 else 263 else
185 { 264 {
186 *enc->cur++ = major | 27; 265 *enc->cur++ = major | LENGTH_EXT8;
187 *enc->cur++ = len >> 56; 266 *enc->cur++ = len >> 56;
188 *enc->cur++ = len >> 48; 267 *enc->cur++ = len >> 48;
189 *enc->cur++ = len >> 40; 268 *enc->cur++ = len >> 40;
190 *enc->cur++ = len >> 32; 269 *enc->cur++ = len >> 32;
191 *enc->cur++ = len >> 24; 270 *enc->cur++ = len >> 24;
193 *enc->cur++ = len >> 8; 272 *enc->cur++ = len >> 8;
194 *enc->cur++ = len; 273 *enc->cur++ = len;
195 } 274 }
196} 275}
197 276
198static void 277ecb_inline void
278encode_tag (enc_t *enc, UV tag)
279{
280 encode_uint (enc, MAJOR_TAG, tag);
281}
282
283// exceptional (hopefully) slow path for byte strings that need to be utf8-encoded
284ecb_noinline static void
199encode_str (enc_t *enc, int utf8, char *str, STRLEN len) 285encode_str_utf8 (enc_t *enc, int utf8, char *str, STRLEN len)
200{ 286{
201 encode_uint (enc, utf8 ? 0x60 : 0x40, len); 287 STRLEN ulen = len;
288 U8 *p, *pend = (U8 *)str + len;
289
290 for (p = (U8 *)str; p < pend; ++p)
291 ulen += *p >> 7; // count set high bits
292
293 encode_uint (enc, MAJOR_TEXT, ulen);
294
295 need (enc, ulen);
296 for (p = (U8 *)str; p < pend; ++p)
297 if (*p < 0x80)
298 *enc->cur++ = *p;
299 else
300 {
301 *enc->cur++ = 0xc0 + (*p >> 6);
302 *enc->cur++ = 0x80 + (*p & 63);
303 }
304}
305
306ecb_inline void
307encode_str (enc_t *enc, int upgrade_utf8, int utf8, char *str, STRLEN len)
308{
309 if (ecb_expect_false (upgrade_utf8))
310 if (!utf8)
311 {
312 encode_str_utf8 (enc, utf8, str, len);
313 return;
314 }
315
316 encode_uint (enc, utf8 ? MAJOR_TEXT : MAJOR_BYTES, len);
202 need (enc, len); 317 need (enc, len);
203 memcpy (enc->cur, str, len); 318 memcpy (enc->cur, str, len);
204 enc->cur += len; 319 enc->cur += len;
205} 320}
206 321
322ecb_inline void
323encode_strref (enc_t *enc, int upgrade_utf8, int utf8, char *str, STRLEN len)
324{
325 if (ecb_expect_false (enc->cbor.flags & F_PACK_STRINGS))
326 {
327 SV **svp = hv_fetch (enc->stringref[!!utf8], str, len, 1);
328
329 if (SvOK (*svp))
330 {
331 // already registered, use stringref
332 encode_tag (enc, CBOR_TAG_STRINGREF);
333 encode_uint (enc, MAJOR_POS_INT, SvUV (*svp));
334 return;
335 }
336 else if (len >= minimum_string_length (enc->stringref_idx))
337 {
338 // register only
339 sv_setuv (*svp, enc->stringref_idx);
340 ++enc->stringref_idx;
341 }
342 }
343
344 encode_str (enc, upgrade_utf8, utf8, str, len);
345}
346
207static void encode_sv (enc_t *enc, SV *sv); 347static void encode_sv (enc_t *enc, SV *sv);
208 348
209static void 349static void
210encode_av (enc_t *enc, AV *av) 350encode_av (enc_t *enc, AV *av)
211{ 351{
214 if (enc->depth >= enc->cbor.max_depth) 354 if (enc->depth >= enc->cbor.max_depth)
215 croak (ERR_NESTING_EXCEEDED); 355 croak (ERR_NESTING_EXCEEDED);
216 356
217 ++enc->depth; 357 ++enc->depth;
218 358
219 encode_uint (enc, 0x80, len + 1); 359 encode_uint (enc, MAJOR_ARRAY, len + 1);
220 360
361 if (SvMAGICAL (av))
221 for (i = 0; i <= len; ++i) 362 for (i = 0; i <= len; ++i)
222 { 363 {
223 SV **svp = av_fetch (av, i, 0); 364 SV **svp = av_fetch (av, i, 0);
224 encode_sv (enc, svp ? *svp : &PL_sv_undef); 365 encode_sv (enc, svp ? *svp : &PL_sv_undef);
225 } 366 }
367 else
368 for (i = 0; i <= len; ++i)
369 {
370 SV *sv = AvARRAY (av)[i];
371 encode_sv (enc, sv ? sv : &PL_sv_undef);
372 }
226 373
227 --enc->depth; 374 --enc->depth;
228} 375}
229 376
230static void 377static void
239 386
240 int pairs = hv_iterinit (hv); 387 int pairs = hv_iterinit (hv);
241 int mg = SvMAGICAL (hv); 388 int mg = SvMAGICAL (hv);
242 389
243 if (mg) 390 if (mg)
244 encode_ch (enc, 0xa0 | 31); 391 encode_ch (enc, MAJOR_MAP | MINOR_INDEF);
245 else 392 else
246 encode_uint (enc, 0xa0, pairs); 393 encode_uint (enc, MAJOR_MAP, pairs);
247 394
248 while ((he = hv_iternext (hv))) 395 while ((he = hv_iternext (hv)))
249 { 396 {
250 if (HeKLEN (he) == HEf_SVKEY) 397 if (HeKLEN (he) == HEf_SVKEY)
251 encode_sv (enc, HeSVKEY (he)); 398 encode_sv (enc, HeSVKEY (he));
252 else 399 else
253 encode_str (enc, HeKUTF8 (he), HeKEY (he), HeKLEN (he)); 400 encode_strref (enc, enc->cbor.flags & (F_TEXT_KEYS | F_TEXT_STRINGS), HeKUTF8 (he), HeKEY (he), HeKLEN (he));
254 401
255 encode_sv (enc, ecb_expect_false (mg) ? hv_iterval (hv, he) : HeVAL (he)); 402 encode_sv (enc, ecb_expect_false (mg) ? hv_iterval (hv, he) : HeVAL (he));
256 } 403 }
257 404
258 if (mg) 405 if (mg)
259 encode_ch (enc, 0xe0 | 31); 406 encode_ch (enc, MAJOR_MISC | MINOR_INDEF);
260 407
261 --enc->depth; 408 --enc->depth;
262} 409}
263 410
264// encode objects, arrays and special \0=false and \1=true values. 411// encode objects, arrays and special \0=false and \1=true values.
265static void 412static void
266encode_rv (enc_t *enc, SV *sv) 413encode_rv (enc_t *enc, SV *sv)
267{ 414{
268 svtype svt;
269
270 SvGETMAGIC (sv); 415 SvGETMAGIC (sv);
416
271 svt = SvTYPE (sv); 417 svtype svt = SvTYPE (sv);
272 418
273 if (ecb_expect_false (SvOBJECT (sv))) 419 if (ecb_expect_false (SvOBJECT (sv)))
274 { 420 {
275 HV *boolean_stash = !CBOR_SLOW || types_boolean_stash 421 HV *boolean_stash = !CBOR_SLOW || types_boolean_stash
276 ? types_boolean_stash 422 ? types_boolean_stash
281 HV *tagged_stash = !CBOR_SLOW || cbor_tagged_stash 427 HV *tagged_stash = !CBOR_SLOW || cbor_tagged_stash
282 ? cbor_tagged_stash 428 ? cbor_tagged_stash
283 : gv_stashpv ("CBOR::XS::Tagged" , 1); 429 : gv_stashpv ("CBOR::XS::Tagged" , 1);
284 430
285 HV *stash = SvSTASH (sv); 431 HV *stash = SvSTASH (sv);
286 GV *method;
287 432
288 if (stash == boolean_stash) 433 if (stash == boolean_stash)
289 encode_ch (enc, SvIV (sv) ? 0xe0 | 21 : 0xe0 | 20); 434 {
435 encode_ch (enc, SvIV (sv) ? MAJOR_MISC | SIMPLE_TRUE : MAJOR_MISC | SIMPLE_FALSE);
436 return;
437 }
290 else if (stash == error_stash) 438 else if (stash == error_stash)
291 encode_ch (enc, 0xe0 | 23); 439 {
440 encode_ch (enc, MAJOR_MISC | SIMPLE_UNDEF);
441 return;
442 }
292 else if (stash == tagged_stash) 443 else if (stash == tagged_stash)
293 { 444 {
294 if (svt != SVt_PVAV) 445 if (svt != SVt_PVAV)
295 croak ("encountered CBOR::XS::Tagged object that isn't an array"); 446 croak ("encountered CBOR::XS::Tagged object that isn't an array");
296 447
297 encode_uint (enc, 0xc0, SvUV (*av_fetch ((AV *)sv, 0, 1))); 448 encode_uint (enc, MAJOR_TAG, SvUV (*av_fetch ((AV *)sv, 0, 1)));
298 encode_sv (enc, *av_fetch ((AV *)sv, 1, 1)); 449 encode_sv (enc, *av_fetch ((AV *)sv, 1, 1));
450
451 return;
452 }
453 }
454
455 if (ecb_expect_false (SvREFCNT (sv) > 1)
456 && ecb_expect_false (enc->cbor.flags & F_ALLOW_SHARING))
457 {
458 if (!enc->shareable)
459 enc->shareable = (HV *)sv_2mortal ((SV *)newHV ());
460
461 SV **svp = hv_fetch (enc->shareable, (char *)&sv, sizeof (sv), 1);
462
463 if (SvOK (*svp))
299 } 464 {
465 encode_tag (enc, CBOR_TAG_VALUE_SHAREDREF);
466 encode_uint (enc, MAJOR_POS_INT, SvUV (*svp));
467 return;
468 }
469 else
470 {
471 sv_setuv (*svp, enc->shareable_idx);
472 ++enc->shareable_idx;
473 encode_tag (enc, CBOR_TAG_VALUE_SHAREABLE);
474 }
475 }
476
477 if (ecb_expect_false (SvOBJECT (sv)))
478 {
479 HV *stash = SvSTASH (sv);
480 GV *method;
481
300 else if ((method = gv_fetchmethod_autoload (stash, "TO_CBOR", 0))) 482 if ((method = gv_fetchmethod_autoload (stash, "TO_CBOR", 0)))
301 { 483 {
302 dSP; 484 dSP;
303 485
304 ENTER; SAVETMPS; PUSHMARK (SP); 486 ENTER; SAVETMPS;
487 PUSHMARK (SP);
305 // we re-bless the reference to get overload and other niceties right 488 // we re-bless the reference to get overload and other niceties right
306 XPUSHs (sv_bless (sv_2mortal (newRV_inc (sv)), stash)); 489 XPUSHs (sv_bless (sv_2mortal (newRV_inc (sv)), stash));
307 490
308 PUTBACK; 491 PUTBACK;
309 // G_SCALAR ensures that return value is 1 492 // G_SCALAR ensures that return value is 1
322 } 505 }
323 else if ((method = gv_fetchmethod_autoload (stash, "FREEZE", 0)) != 0) 506 else if ((method = gv_fetchmethod_autoload (stash, "FREEZE", 0)) != 0)
324 { 507 {
325 dSP; 508 dSP;
326 509
327 ENTER; SAVETMPS; PUSHMARK (SP); 510 ENTER; SAVETMPS;
511 SAVESTACK_POS ();
512 PUSHMARK (SP);
328 EXTEND (SP, 2); 513 EXTEND (SP, 2);
329 // we re-bless the reference to get overload and other niceties right 514 // we re-bless the reference to get overload and other niceties right
330 PUSHs (sv_bless (sv_2mortal (newRV_inc (sv)), stash)); 515 PUSHs (sv_bless (sv_2mortal (newRV_inc (sv)), stash));
331 PUSHs (sv_cbor); 516 PUSHs (sv_cbor);
332 517
336 521
337 // catch this surprisingly common error 522 // catch this surprisingly common error
338 if (count == 1 && SvROK (TOPs) && SvRV (TOPs) == sv) 523 if (count == 1 && SvROK (TOPs) && SvRV (TOPs) == sv)
339 croak ("%s::FREEZE(CBOR) method returned same object as was passed instead of a new one", HvNAME (stash)); 524 croak ("%s::FREEZE(CBOR) method returned same object as was passed instead of a new one", HvNAME (stash));
340 525
341 encode_uint (enc, 0xc0, CBOR_TAG_PERL_OBJECT); 526 encode_tag (enc, CBOR_TAG_PERL_OBJECT);
342 encode_uint (enc, 0x80, count + 1); 527 encode_uint (enc, MAJOR_ARRAY, count + 1);
343 encode_str (enc, HvNAMEUTF8 (stash), HvNAME (stash), HvNAMELEN (stash)); 528 encode_strref (enc, 0, HvNAMEUTF8 (stash), HvNAME (stash), HvNAMELEN (stash));
344 529
345 while (count) 530 while (count)
346 encode_sv (enc, SP[1 - count--]); 531 encode_sv (enc, SP[1 - count--]);
347 532
348 PUTBACK; 533 PUTBACK;
355 } 540 }
356 else if (svt == SVt_PVHV) 541 else if (svt == SVt_PVHV)
357 encode_hv (enc, (HV *)sv); 542 encode_hv (enc, (HV *)sv);
358 else if (svt == SVt_PVAV) 543 else if (svt == SVt_PVAV)
359 encode_av (enc, (AV *)sv); 544 encode_av (enc, (AV *)sv);
360 else if (svt < SVt_PVAV)
361 {
362 STRLEN len = 0;
363 char *pv = svt ? SvPV (sv, len) : 0;
364
365 if (len == 1 && *pv == '1')
366 encode_ch (enc, 0xe0 | 21);
367 else if (len == 1 && *pv == '0')
368 encode_ch (enc, 0xe0 | 20);
369 else if (enc->cbor.flags & F_ALLOW_UNKNOWN)
370 encode_ch (enc, 0xe0 | 23);
371 else
372 croak ("cannot encode reference to scalar '%s' unless the scalar is 0 or 1",
373 SvPV_nolen (sv_2mortal (newRV_inc (sv))));
374 }
375 else if (enc->cbor.flags & F_ALLOW_UNKNOWN)
376 encode_ch (enc, 0xe0 | 23);
377 else 545 else
378 croak ("encountered %s, but CBOR can only represent references to arrays or hashes", 546 {
379 SvPV_nolen (sv_2mortal (newRV_inc (sv)))); 547 encode_tag (enc, CBOR_TAG_INDIRECTION);
548 encode_sv (enc, sv);
549 }
380} 550}
381 551
382static void 552static void
383encode_nv (enc_t *enc, SV *sv) 553encode_nv (enc_t *enc, SV *sv)
384{ 554{
385 double nv = SvNVX (sv); 555 double nv = SvNVX (sv);
386 556
387 need (enc, 9); 557 need (enc, 9);
388 558
389 if (ecb_expect_false (nv == (U32)nv)) 559 if (ecb_expect_false (nv == (NV)(U32)nv))
390 encode_uint (enc, 0x00, (U32)nv); 560 encode_uint (enc, MAJOR_POS_INT, (U32)nv);
391 //TODO: maybe I32? 561 //TODO: maybe I32?
392 else if (ecb_expect_false (nv == (float)nv)) 562 else if (ecb_expect_false (nv == (float)nv))
393 { 563 {
394 uint32_t fp = ecb_float_to_binary32 (nv); 564 uint32_t fp = ecb_float_to_binary32 (nv);
395 565
396 *enc->cur++ = 0xe0 | 26; 566 *enc->cur++ = MAJOR_MISC | MISC_FLOAT32;
397 567
398 if (!ecb_big_endian ()) 568 if (!ecb_big_endian ())
399 fp = ecb_bswap32 (fp); 569 fp = ecb_bswap32 (fp);
400 570
401 memcpy (enc->cur, &fp, 4); 571 memcpy (enc->cur, &fp, 4);
403 } 573 }
404 else 574 else
405 { 575 {
406 uint64_t fp = ecb_double_to_binary64 (nv); 576 uint64_t fp = ecb_double_to_binary64 (nv);
407 577
408 *enc->cur++ = 0xe0 | 27; 578 *enc->cur++ = MAJOR_MISC | MISC_FLOAT64;
409 579
410 if (!ecb_big_endian ()) 580 if (!ecb_big_endian ())
411 fp = ecb_bswap64 (fp); 581 fp = ecb_bswap64 (fp);
412 582
413 memcpy (enc->cur, &fp, 8); 583 memcpy (enc->cur, &fp, 8);
422 592
423 if (SvPOKp (sv)) 593 if (SvPOKp (sv))
424 { 594 {
425 STRLEN len; 595 STRLEN len;
426 char *str = SvPV (sv, len); 596 char *str = SvPV (sv, len);
427 encode_str (enc, SvUTF8 (sv), str, len); 597 encode_strref (enc, enc->cbor.flags & F_TEXT_STRINGS, SvUTF8 (sv), str, len);
428 } 598 }
429 else if (SvNOKp (sv)) 599 else if (SvNOKp (sv))
430 encode_nv (enc, sv); 600 encode_nv (enc, sv);
431 else if (SvIOKp (sv)) 601 else if (SvIOKp (sv))
432 { 602 {
433 if (SvIsUV (sv)) 603 if (SvIsUV (sv))
434 encode_uint (enc, 0x00, SvUVX (sv)); 604 encode_uint (enc, MAJOR_POS_INT, SvUVX (sv));
435 else if (SvIVX (sv) >= 0) 605 else if (SvIVX (sv) >= 0)
436 encode_uint (enc, 0x00, SvIVX (sv)); 606 encode_uint (enc, MAJOR_POS_INT, SvIVX (sv));
437 else 607 else
438 encode_uint (enc, 0x20, -(SvIVX (sv) + 1)); 608 encode_uint (enc, MAJOR_NEG_INT, -(SvIVX (sv) + 1));
439 } 609 }
440 else if (SvROK (sv)) 610 else if (SvROK (sv))
441 encode_rv (enc, SvRV (sv)); 611 encode_rv (enc, SvRV (sv));
442 else if (!SvOK (sv)) 612 else if (!SvOK (sv))
443 encode_ch (enc, 0xe0 | 22); 613 encode_ch (enc, MAJOR_MISC | SIMPLE_NULL);
444 else if (enc->cbor.flags & F_ALLOW_UNKNOWN) 614 else if (enc->cbor.flags & F_ALLOW_UNKNOWN)
445 encode_ch (enc, 0xe0 | 23); 615 encode_ch (enc, MAJOR_MISC | SIMPLE_UNDEF);
446 else 616 else
447 croak ("encountered perl type (%s,0x%x) that CBOR cannot handle, check your input data", 617 croak ("encountered perl type (%s,0x%x) that CBOR cannot handle, check your input data",
448 SvPV_nolen (sv), (unsigned int)SvFLAGS (sv)); 618 SvPV_nolen (sv), (unsigned int)SvFLAGS (sv));
449} 619}
450 620
451static SV * 621static SV *
452encode_cbor (SV *scalar, CBOR *cbor) 622encode_cbor (SV *scalar, CBOR *cbor)
453{ 623{
454 enc_t enc; 624 enc_t enc = { 0 };
455 625
456 enc.cbor = *cbor; 626 enc.cbor = *cbor;
457 enc.sv = sv_2mortal (NEWSV (0, INIT_SIZE)); 627 enc.sv = sv_2mortal (NEWSV (0, INIT_SIZE));
458 enc.cur = SvPVX (enc.sv); 628 enc.cur = SvPVX (enc.sv);
459 enc.end = SvEND (enc.sv); 629 enc.end = SvEND (enc.sv);
460 enc.depth = 0;
461 630
462 SvPOK_only (enc.sv); 631 SvPOK_only (enc.sv);
632
633 if (cbor->flags & F_PACK_STRINGS)
634 {
635 encode_tag (&enc, CBOR_TAG_STRINGREF_NAMESPACE);
636 enc.stringref[0]= (HV *)sv_2mortal ((SV *)newHV ());
637 enc.stringref[1]= (HV *)sv_2mortal ((SV *)newHV ());
638 }
639
463 encode_sv (&enc, scalar); 640 encode_sv (&enc, scalar);
464 641
465 SvCUR_set (enc.sv, enc.cur - SvPVX (enc.sv)); 642 SvCUR_set (enc.sv, enc.cur - SvPVX (enc.sv));
466 *SvEND (enc.sv) = 0; // many xs functions expect a trailing 0 for text strings 643 *SvEND (enc.sv) = 0; // many xs functions expect a trailing 0 for text strings
467 644
481 U8 *end; // end of input string 658 U8 *end; // end of input string
482 const char *err; // parse error, if != 0 659 const char *err; // parse error, if != 0
483 CBOR cbor; 660 CBOR cbor;
484 U32 depth; // recursion depth 661 U32 depth; // recursion depth
485 U32 maxdepth; // recursion depth limit 662 U32 maxdepth; // recursion depth limit
663 AV *shareable;
664 AV *stringref;
665 SV *decode_tagged;
486} dec_t; 666} dec_t;
487 667
488#define ERR(reason) SB if (!dec->err) dec->err = reason; goto fail; SE 668#define ERR(reason) SB if (!dec->err) dec->err = reason; goto fail; SE
489 669
490#define WANT(len) if (ecb_expect_false (dec->cur + len > dec->end)) ERR ("unexpected end of CBOR data") 670#define WANT(len) if (ecb_expect_false ((UV)(dec->end - dec->cur) < (UV)len)) ERR ("unexpected end of CBOR data")
491 671
492#define DEC_INC_DEPTH if (++dec->depth > dec->cbor.max_depth) ERR (ERR_NESTING_EXCEEDED) 672#define DEC_INC_DEPTH if (ecb_expect_false (++dec->depth > dec->cbor.max_depth)) ERR (ERR_NESTING_EXCEEDED)
493#define DEC_DEC_DEPTH --dec->depth 673#define DEC_DEC_DEPTH --dec->depth
494 674
495static UV 675static UV
496decode_uint (dec_t *dec) 676decode_uint (dec_t *dec)
497{ 677{
498 switch (*dec->cur & 31) 678 U8 m = *dec->cur & MINOR_MASK;
499 { 679 ++dec->cur;
500 case 0: case 1: case 2: case 3: case 4: case 5: case 6: case 7:
501 case 8: case 9: case 10: case 11: case 12: case 13: case 14: case 15:
502 case 16: case 17: case 18: case 19: case 20: case 21: case 22: case 23:
503 return *dec->cur++ & 31;
504 680
505 case 24: 681 if (ecb_expect_true (m < LENGTH_EXT1))
682 return m;
683 else if (ecb_expect_true (m == LENGTH_EXT1))
684 {
506 WANT (2); 685 WANT (1);
507 dec->cur += 2; 686 dec->cur += 1;
508 return dec->cur[-1]; 687 return dec->cur[-1];
509 688 }
510 case 25: 689 else if (ecb_expect_true (m == LENGTH_EXT2))
690 {
511 WANT (3); 691 WANT (2);
512 dec->cur += 3; 692 dec->cur += 2;
513 return (((UV)dec->cur[-2]) << 8) 693 return (((UV)dec->cur[-2]) << 8)
514 | ((UV)dec->cur[-1]); 694 | ((UV)dec->cur[-1]);
515 695 }
516 case 26: 696 else if (ecb_expect_true (m == LENGTH_EXT4))
697 {
517 WANT (5); 698 WANT (4);
518 dec->cur += 5; 699 dec->cur += 4;
519 return (((UV)dec->cur[-4]) << 24) 700 return (((UV)dec->cur[-4]) << 24)
520 | (((UV)dec->cur[-3]) << 16) 701 | (((UV)dec->cur[-3]) << 16)
521 | (((UV)dec->cur[-2]) << 8) 702 | (((UV)dec->cur[-2]) << 8)
522 | ((UV)dec->cur[-1]); 703 | ((UV)dec->cur[-1]);
523 704 }
524 case 27: 705 else if (ecb_expect_true (m == LENGTH_EXT8))
706 {
525 WANT (9); 707 WANT (8);
526 dec->cur += 9; 708 dec->cur += 8;
709
710 return
711#if UVSIZE < 8
712 0
713#else
527 return (((UV)dec->cur[-8]) << 56) 714 (((UV)dec->cur[-8]) << 56)
528 | (((UV)dec->cur[-7]) << 48) 715 | (((UV)dec->cur[-7]) << 48)
529 | (((UV)dec->cur[-6]) << 40) 716 | (((UV)dec->cur[-6]) << 40)
530 | (((UV)dec->cur[-5]) << 32) 717 | (((UV)dec->cur[-5]) << 32)
718#endif
531 | (((UV)dec->cur[-4]) << 24) 719 | (((UV)dec->cur[-4]) << 24)
532 | (((UV)dec->cur[-3]) << 16) 720 | (((UV)dec->cur[-3]) << 16)
533 | (((UV)dec->cur[-2]) << 8) 721 | (((UV)dec->cur[-2]) << 8)
534 | ((UV)dec->cur[-1]); 722 | ((UV)dec->cur[-1]);
535 723 }
536 default: 724 else
537 ERR ("corrupted CBOR data (unsupported integer minor encoding)"); 725 ERR ("corrupted CBOR data (unsupported integer minor encoding)");
538 }
539 726
540fail: 727fail:
541 return 0; 728 return 0;
542} 729}
543 730
548{ 735{
549 AV *av = newAV (); 736 AV *av = newAV ();
550 737
551 DEC_INC_DEPTH; 738 DEC_INC_DEPTH;
552 739
553 if ((*dec->cur & 31) == 31) 740 if (*dec->cur == (MAJOR_ARRAY | MINOR_INDEF))
554 { 741 {
555 ++dec->cur; 742 ++dec->cur;
556 743
557 for (;;) 744 for (;;)
558 { 745 {
559 WANT (1); 746 WANT (1);
560 747
561 if (*dec->cur == (0xe0 | 31)) 748 if (*dec->cur == (MAJOR_MISC | MINOR_INDEF))
562 { 749 {
563 ++dec->cur; 750 ++dec->cur;
564 break; 751 break;
565 } 752 }
566 753
567 av_push (av, decode_sv (dec)); 754 av_push (av, decode_sv (dec));
568 } 755 }
569 } 756 }
570 else 757 else
571 { 758 {
572 int i, len = decode_uint (dec); 759 UV i, len = decode_uint (dec);
573 760
761 WANT (len); // complexity check for av_fill - need at least one byte per value, do not allow supersize arrays
574 av_fill (av, len - 1); 762 av_fill (av, len - 1);
575 763
576 for (i = 0; i < len; ++i) 764 for (i = 0; i < len; ++i)
577 AvARRAY (av)[i] = decode_sv (dec); 765 AvARRAY (av)[i] = decode_sv (dec);
578 } 766 }
588 776
589static void 777static void
590decode_he (dec_t *dec, HV *hv) 778decode_he (dec_t *dec, HV *hv)
591{ 779{
592 // for speed reasons, we specialcase single-string 780 // for speed reasons, we specialcase single-string
593 // byte or utf-8 strings as keys. 781 // byte or utf-8 strings as keys, but only when !stringref
594 782
595 if (*dec->cur >= 0x40 && *dec->cur <= 0x40 + 27) 783 if (ecb_expect_true (!dec->stringref))
784 if (ecb_expect_true ((U8)(*dec->cur - MAJOR_BYTES) <= LENGTH_EXT8))
596 { 785 {
597 I32 len = decode_uint (dec); 786 STRLEN len = decode_uint (dec);
598 char *key = (char *)dec->cur; 787 char *key = (char *)dec->cur;
599 788
789 WANT (len);
600 dec->cur += len; 790 dec->cur += len;
601 791
602 hv_store (hv, key, len, decode_sv (dec), 0); 792 hv_store (hv, key, len, decode_sv (dec), 0);
793
794 return;
603 } 795 }
604 else if (*dec->cur >= 0x60 && *dec->cur <= 0x60 + 27) 796 else if (ecb_expect_true ((U8)(*dec->cur - MAJOR_TEXT) <= LENGTH_EXT8))
605 { 797 {
606 I32 len = decode_uint (dec); 798 STRLEN len = decode_uint (dec);
607 char *key = (char *)dec->cur; 799 char *key = (char *)dec->cur;
608 800
801 WANT (len);
609 dec->cur += len; 802 dec->cur += len;
610 803
804 if (ecb_expect_false (dec->cbor.flags & F_VALIDATE_UTF8))
805 if (!is_utf8_string (key, len))
806 ERR ("corrupted CBOR data (invalid UTF-8 in map key)");
807
611 hv_store (hv, key, -len, decode_sv (dec), 0); 808 hv_store (hv, key, -len, decode_sv (dec), 0);
809
810 return;
612 } 811 }
613 else 812
614 {
615 SV *k = decode_sv (dec); 813 SV *k = decode_sv (dec);
616 SV *v = decode_sv (dec); 814 SV *v = decode_sv (dec);
617 815
618 hv_store_ent (hv, k, v, 0); 816 hv_store_ent (hv, k, v, 0);
619 SvREFCNT_dec (k); 817 SvREFCNT_dec (k);
620 } 818
819fail:
820 ;
621} 821}
622 822
623static SV * 823static SV *
624decode_hv (dec_t *dec) 824decode_hv (dec_t *dec)
625{ 825{
626 HV *hv = newHV (); 826 HV *hv = newHV ();
627 827
628 DEC_INC_DEPTH; 828 DEC_INC_DEPTH;
629 829
630 if ((*dec->cur & 31) == 31) 830 if (*dec->cur == (MAJOR_MAP | MINOR_INDEF))
631 { 831 {
632 ++dec->cur; 832 ++dec->cur;
633 833
634 for (;;) 834 for (;;)
635 { 835 {
636 WANT (1); 836 WANT (1);
637 837
638 if (*dec->cur == (0xe0 | 31)) 838 if (*dec->cur == (MAJOR_MISC | MINOR_INDEF))
639 { 839 {
640 ++dec->cur; 840 ++dec->cur;
641 break; 841 break;
642 } 842 }
643 843
644 decode_he (dec, hv); 844 decode_he (dec, hv);
645 } 845 }
646 } 846 }
647 else 847 else
648 { 848 {
649 int pairs = decode_uint (dec); 849 UV pairs = decode_uint (dec);
850
851 WANT (pairs); // complexity check - need at least one byte per value, do not allow supersize hashes
650 852
651 while (pairs--) 853 while (pairs--)
652 decode_he (dec, hv); 854 decode_he (dec, hv);
653 } 855 }
654 856
664static SV * 866static SV *
665decode_str (dec_t *dec, int utf8) 867decode_str (dec_t *dec, int utf8)
666{ 868{
667 SV *sv = 0; 869 SV *sv = 0;
668 870
669 if ((*dec->cur & 31) == 31) 871 if ((*dec->cur & MINOR_MASK) == MINOR_INDEF)
670 { 872 {
873 // indefinite length strings
671 ++dec->cur; 874 ++dec->cur;
672 875
876 U8 major = *dec->cur & MAJOR_MISC;
877
673 sv = newSVpvn ("", 0); 878 sv = newSVpvn ("", 0);
674 879
675 // not very fast, and certainly not robust against illegal input
676 for (;;) 880 for (;;)
677 { 881 {
678 WANT (1); 882 WANT (1);
679 883
680 if (*dec->cur == (0xe0 | 31)) 884 if ((*dec->cur - major) > LENGTH_EXT8)
885 if (*dec->cur == (MAJOR_MISC | MINOR_INDEF))
681 { 886 {
682 ++dec->cur; 887 ++dec->cur;
683 break; 888 break;
684 } 889 }
890 else
891 ERR ("corrupted CBOR data (invalid chunks in indefinite length string)");
685 892
686 sv_catsv (sv, decode_sv (dec)); 893 STRLEN len = decode_uint (dec);
894
895 WANT (len);
896 sv_catpvn (sv, dec->cur, len);
897 dec->cur += len;
687 } 898 }
688 } 899 }
689 else 900 else
690 { 901 {
691 STRLEN len = decode_uint (dec); 902 STRLEN len = decode_uint (dec);
692 903
693 WANT (len); 904 WANT (len);
694 sv = newSVpvn (dec->cur, len); 905 sv = newSVpvn (dec->cur, len);
695 dec->cur += len; 906 dec->cur += len;
907
908 if (ecb_expect_false (dec->stringref)
909 && SvCUR (sv) >= minimum_string_length (AvFILLp (dec->stringref) + 1))
910 av_push (dec->stringref, SvREFCNT_inc_NN (sv));
696 } 911 }
697 912
698 if (utf8) 913 if (utf8)
914 {
915 if (ecb_expect_false (dec->cbor.flags & F_VALIDATE_UTF8))
916 if (!is_utf8_string (SvPVX (sv), SvCUR (sv)))
917 ERR ("corrupted CBOR data (invalid UTF-8 in text string)");
918
699 SvUTF8_on (sv); 919 SvUTF8_on (sv);
920 }
700 921
701 return sv; 922 return sv;
702 923
703fail: 924fail:
704 SvREFCNT_dec (sv); 925 SvREFCNT_dec (sv);
706} 927}
707 928
708static SV * 929static SV *
709decode_tagged (dec_t *dec) 930decode_tagged (dec_t *dec)
710{ 931{
932 SV *sv = 0;
711 UV tag = decode_uint (dec); 933 UV tag = decode_uint (dec);
934
935 WANT (1);
936
937 switch (tag)
938 {
939 case CBOR_TAG_MAGIC:
712 SV *sv = decode_sv (dec); 940 sv = decode_sv (dec);
941 break;
713 942
714 if (tag == CBOR_TAG_MAGIC) 943 case CBOR_TAG_INDIRECTION:
715 return sv; 944 sv = newRV_noinc (decode_sv (dec));
716 else if (tag == CBOR_TAG_PERL_OBJECT) 945 break;
717 { 946
947 case CBOR_TAG_STRINGREF_NAMESPACE:
948 {
949 // do nmot use SAVETMPS/FREETMPS, as these will
950 // erase mortalised caches, e.g. "shareable"
951 ENTER;
952
953 SAVESPTR (dec->stringref);
954 dec->stringref = (AV *)sv_2mortal ((SV *)newAV ());
955
956 sv = decode_sv (dec);
957
958 LEAVE;
959 }
960 break;
961
962 case CBOR_TAG_STRINGREF:
963 {
964 if ((*dec->cur >> MAJOR_SHIFT) != (MAJOR_POS_INT >> MAJOR_SHIFT))
965 ERR ("corrupted CBOR data (stringref index not an unsigned integer)");
966
967 UV idx = decode_uint (dec);
968
969 if (!dec->stringref || (int)idx > AvFILLp (dec->stringref))
970 ERR ("corrupted CBOR data (stringref index out of bounds or outside namespace)");
971
972 sv = newSVsv (AvARRAY (dec->stringref)[idx]);
973 }
974 break;
975
976 case CBOR_TAG_VALUE_SHAREABLE:
977 {
978 if (ecb_expect_false (!dec->shareable))
979 dec->shareable = (AV *)sv_2mortal ((SV *)newAV ());
980
981 if (dec->cbor.flags & F_ALLOW_CYCLES)
982 {
983 sv = newSV (0);
984 av_push (dec->shareable, SvREFCNT_inc_NN (sv));
985
986 SV *osv = decode_sv (dec);
987 sv_setsv (sv, osv);
988 SvREFCNT_dec_NN (osv);
989 }
990 else
991 {
992 av_push (dec->shareable, &PL_sv_undef);
993 int idx = AvFILLp (dec->shareable);
994 sv = decode_sv (dec);
995 av_store (dec->shareable, idx, SvREFCNT_inc_NN (sv));
996 }
997 }
998 break;
999
1000 case CBOR_TAG_VALUE_SHAREDREF:
1001 {
1002 if ((*dec->cur >> MAJOR_SHIFT) != (MAJOR_POS_INT >> MAJOR_SHIFT))
1003 ERR ("corrupted CBOR data (sharedref index not an unsigned integer)");
1004
1005 UV idx = decode_uint (dec);
1006
1007 if (!dec->shareable || (int)idx > AvFILLp (dec->shareable))
1008 ERR ("corrupted CBOR data (sharedref index out of bounds)");
1009
1010 sv = SvREFCNT_inc_NN (AvARRAY (dec->shareable)[idx]);
1011
1012 if (sv == &PL_sv_undef)
1013 ERR ("cyclic CBOR data structure found, but allow_cycles is not enabled");
1014 }
1015 break;
1016
1017 case CBOR_TAG_PERL_OBJECT:
1018 {
1019 sv = decode_sv (dec);
1020
718 if (!SvROK (sv) || SvTYPE (SvRV (sv)) != SVt_PVAV) 1021 if (!SvROK (sv) || SvTYPE (SvRV (sv)) != SVt_PVAV)
719 ERR ("corrupted CBOR data (non-array perl object)"); 1022 ERR ("corrupted CBOR data (non-array perl object)");
720 1023
721 AV *av = (AV *)SvRV (sv); 1024 AV *av = (AV *)SvRV (sv);
722 int len = av_len (av) + 1; 1025 int len = av_len (av) + 1;
723 HV *stash = gv_stashsv (*av_fetch (av, 0, 1), 0); 1026 HV *stash = gv_stashsv (*av_fetch (av, 0, 1), 0);
724 1027
725 if (!stash) 1028 if (!stash)
726 ERR ("cannot decode perl-object (package does not exist)"); 1029 ERR ("cannot decode perl-object (package does not exist)");
727 1030
728 GV *method = gv_fetchmethod_autoload (stash, "THAW", 0); 1031 GV *method = gv_fetchmethod_autoload (stash, "THAW", 0);
729 1032
730 if (!method) 1033 if (!method)
731 ERR ("cannot decode perl-object (package does not have a THAW method)"); 1034 ERR ("cannot decode perl-object (package does not have a THAW method)");
732 1035
733 dSP; 1036 dSP;
734 1037
735 ENTER; SAVETMPS; PUSHMARK (SP); 1038 ENTER; SAVETMPS;
1039 PUSHMARK (SP);
736 EXTEND (SP, len + 1); 1040 EXTEND (SP, len + 1);
737 // we re-bless the reference to get overload and other niceties right 1041 // we re-bless the reference to get overload and other niceties right
738 PUSHs (*av_fetch (av, 0, 1)); 1042 PUSHs (*av_fetch (av, 0, 1));
739 PUSHs (sv_cbor); 1043 PUSHs (sv_cbor);
740 1044
741 int i; 1045 int i;
742 1046
743 for (i = 1; i < len; ++i) 1047 for (i = 1; i < len; ++i)
744 PUSHs (*av_fetch (av, i, 1)); 1048 PUSHs (*av_fetch (av, i, 1));
745 1049
746 PUTBACK; 1050 PUTBACK;
747 call_sv ((SV *)GvCV (method), G_SCALAR); 1051 call_sv ((SV *)GvCV (method), G_SCALAR | G_EVAL);
748 SPAGAIN; 1052 SPAGAIN;
749 1053
1054 if (SvTRUE (ERRSV))
1055 {
1056 FREETMPS; LEAVE;
1057 ERR (SvPVutf8_nolen (sv_2mortal (SvREFCNT_inc (ERRSV))));
1058 }
1059
750 SvREFCNT_dec (sv); 1060 SvREFCNT_dec (sv);
751 sv = SvREFCNT_inc (POPs); 1061 sv = SvREFCNT_inc (POPs);
752 1062
753 PUTBACK; 1063 PUTBACK;
754 1064
755 FREETMPS; LEAVE; 1065 FREETMPS; LEAVE;
1066 }
1067 break;
756 1068
757 return sv; 1069 default:
758 } 1070 {
759 else 1071 sv = decode_sv (dec);
760 { 1072
1073 dSP;
1074 ENTER; SAVETMPS;
1075 SAVESTACK_POS ();
1076 PUSHMARK (SP);
1077 EXTEND (SP, 2);
1078 PUSHs (newSVuv (tag));
1079 PUSHs (sv);
1080
1081 PUTBACK;
1082 int count = call_sv (dec->cbor.filter ? dec->cbor.filter : default_filter, G_ARRAY | G_EVAL);
1083 SPAGAIN;
1084
1085 if (SvTRUE (ERRSV))
1086 {
1087 FREETMPS; LEAVE;
1088 ERR (SvPVutf8_nolen (sv_2mortal (SvREFCNT_inc (ERRSV))));
1089 }
1090
1091 if (count)
1092 {
1093 SvREFCNT_dec (sv);
1094 sv = SvREFCNT_inc (POPs);
1095 }
1096 else
1097 {
761 AV *av = newAV (); 1098 AV *av = newAV ();
762 av_push (av, newSVuv (tag)); 1099 av_push (av, newSVuv (tag));
763 av_push (av, sv); 1100 av_push (av, sv);
764 1101
765 HV *tagged_stash = !CBOR_SLOW || cbor_tagged_stash 1102 HV *tagged_stash = !CBOR_SLOW || cbor_tagged_stash
766 ? cbor_tagged_stash 1103 ? cbor_tagged_stash
767 : gv_stashpv ("CBOR::XS::Tagged" , 1); 1104 : gv_stashpv ("CBOR::XS::Tagged" , 1);
768
769 return sv_bless (newRV_noinc ((SV *)av), tagged_stash); 1105 sv = sv_bless (newRV_noinc ((SV *)av), tagged_stash);
1106 }
1107
1108 PUTBACK;
1109
1110 FREETMPS; LEAVE;
1111 }
1112 break;
770 } 1113 }
1114
1115 return sv;
771 1116
772fail: 1117fail:
773 SvREFCNT_dec (sv); 1118 SvREFCNT_dec (sv);
774 return &PL_sv_undef; 1119 return &PL_sv_undef;
775} 1120}
777static SV * 1122static SV *
778decode_sv (dec_t *dec) 1123decode_sv (dec_t *dec)
779{ 1124{
780 WANT (1); 1125 WANT (1);
781 1126
782 switch (*dec->cur >> 5) 1127 switch (*dec->cur >> MAJOR_SHIFT)
783 { 1128 {
784 case 0: // unsigned int 1129 case MAJOR_POS_INT >> MAJOR_SHIFT: return newSVuv (decode_uint (dec));
785 return newSVuv (decode_uint (dec)); 1130 case MAJOR_NEG_INT >> MAJOR_SHIFT: return newSViv (-1 - (IV)decode_uint (dec));
786 case 1: // negative int 1131 case MAJOR_BYTES >> MAJOR_SHIFT: return decode_str (dec, 0);
787 return newSViv (-1 - (IV)decode_uint (dec)); 1132 case MAJOR_TEXT >> MAJOR_SHIFT: return decode_str (dec, 1);
788 case 2: // octet string 1133 case MAJOR_ARRAY >> MAJOR_SHIFT: return decode_av (dec);
789 return decode_str (dec, 0); 1134 case MAJOR_MAP >> MAJOR_SHIFT: return decode_hv (dec);
790 case 3: // utf-8 string 1135 case MAJOR_TAG >> MAJOR_SHIFT: return decode_tagged (dec);
791 return decode_str (dec, 1); 1136
792 case 4: // array 1137 case MAJOR_MISC >> MAJOR_SHIFT:
793 return decode_av (dec);
794 case 5: // map
795 return decode_hv (dec);
796 case 6: // tag
797 return decode_tagged (dec);
798 case 7: // misc
799 switch (*dec->cur++ & 31) 1138 switch (*dec->cur++ & MINOR_MASK)
800 { 1139 {
801 case 20: 1140 case SIMPLE_FALSE:
802#if CBOR_SLOW 1141#if CBOR_SLOW
803 types_false = get_bool ("Types::Serialiser::false"); 1142 types_false = get_bool ("Types::Serialiser::false");
804#endif 1143#endif
805 return newSVsv (types_false); 1144 return newSVsv (types_false);
806 case 21: 1145 case SIMPLE_TRUE:
807#if CBOR_SLOW 1146#if CBOR_SLOW
808 types_true = get_bool ("Types::Serialiser::true"); 1147 types_true = get_bool ("Types::Serialiser::true");
809#endif 1148#endif
810 return newSVsv (types_true); 1149 return newSVsv (types_true);
811 case 22: 1150 case SIMPLE_NULL:
812 return newSVsv (&PL_sv_undef); 1151 return newSVsv (&PL_sv_undef);
813 case 23: 1152 case SIMPLE_UNDEF:
814#if CBOR_SLOW 1153#if CBOR_SLOW
815 types_error = get_bool ("Types::Serialiser::error"); 1154 types_error = get_bool ("Types::Serialiser::error");
816#endif 1155#endif
817 return newSVsv (types_error); 1156 return newSVsv (types_error);
818 1157
819 case 25: 1158 case MISC_FLOAT16:
820 { 1159 {
821 WANT (2); 1160 WANT (2);
822 1161
823 uint16_t fp = (dec->cur[0] << 8) | dec->cur[1]; 1162 uint16_t fp = (dec->cur[0] << 8) | dec->cur[1];
824 dec->cur += 2; 1163 dec->cur += 2;
825 1164
826 return newSVnv (ecb_binary16_to_float (fp)); 1165 return newSVnv (ecb_binary16_to_float (fp));
827 } 1166 }
828 1167
829 case 26: 1168 case MISC_FLOAT32:
830 { 1169 {
831 uint32_t fp; 1170 uint32_t fp;
832 WANT (4); 1171 WANT (4);
833 memcpy (&fp, dec->cur, 4); 1172 memcpy (&fp, dec->cur, 4);
834 dec->cur += 4; 1173 dec->cur += 4;
837 fp = ecb_bswap32 (fp); 1176 fp = ecb_bswap32 (fp);
838 1177
839 return newSVnv (ecb_binary32_to_float (fp)); 1178 return newSVnv (ecb_binary32_to_float (fp));
840 } 1179 }
841 1180
842 case 27: 1181 case MISC_FLOAT64:
843 { 1182 {
844 uint64_t fp; 1183 uint64_t fp;
845 WANT (8); 1184 WANT (8);
846 memcpy (&fp, dec->cur, 8); 1185 memcpy (&fp, dec->cur, 8);
847 dec->cur += 8; 1186 dec->cur += 8;
850 fp = ecb_bswap64 (fp); 1189 fp = ecb_bswap64 (fp);
851 1190
852 return newSVnv (ecb_binary64_to_double (fp)); 1191 return newSVnv (ecb_binary64_to_double (fp));
853 } 1192 }
854 1193
855 // 0..19 unassigned 1194 // 0..19 unassigned simple
856 // 24 reserved + unassigned (reserved values are not encodable) 1195 // 24 reserved + unassigned simple (reserved values are not encodable)
1196 // 28-30 unassigned misc
1197 // 31 break code
857 default: 1198 default:
858 ERR ("corrupted CBOR data (reserved/unassigned major 7 value)"); 1199 ERR ("corrupted CBOR data (reserved/unassigned/unexpected major 7 value)");
859 } 1200 }
860 1201
861 break; 1202 break;
862 } 1203 }
863 1204
866} 1207}
867 1208
868static SV * 1209static SV *
869decode_cbor (SV *string, CBOR *cbor, char **offset_return) 1210decode_cbor (SV *string, CBOR *cbor, char **offset_return)
870{ 1211{
871 dec_t dec; 1212 dec_t dec = { 0 };
872 SV *sv; 1213 SV *sv;
873 STRLEN len; 1214 STRLEN len;
874 char *data = SvPVbyte (string, len); 1215 char *data = SvPVbyte (string, len);
875 1216
876 if (len > cbor->max_size && cbor->max_size) 1217 if (len > cbor->max_size && cbor->max_size)
878 (unsigned long)len, (unsigned long)cbor->max_size); 1219 (unsigned long)len, (unsigned long)cbor->max_size);
879 1220
880 dec.cbor = *cbor; 1221 dec.cbor = *cbor;
881 dec.cur = (U8 *)data; 1222 dec.cur = (U8 *)data;
882 dec.end = (U8 *)data + len; 1223 dec.end = (U8 *)data + len;
883 dec.err = 0;
884 dec.depth = 0;
885 1224
886 sv = decode_sv (&dec); 1225 sv = decode_sv (&dec);
887 1226
888 if (offset_return) 1227 if (offset_return)
889 *offset_return = dec.cur; 1228 *offset_return = dec.cur;
892 if (dec.cur != dec.end && !dec.err) 1231 if (dec.cur != dec.end && !dec.err)
893 dec.err = "garbage after CBOR object"; 1232 dec.err = "garbage after CBOR object";
894 1233
895 if (dec.err) 1234 if (dec.err)
896 { 1235 {
1236 if (dec.shareable)
1237 {
1238 // need to break cyclic links, which whould all be in shareable
1239 int i;
1240 SV **svp;
1241
1242 for (i = av_len (dec.shareable) + 1; i--; )
1243 if ((svp = av_fetch (dec.shareable, i, 0)))
1244 sv_setsv (*svp, &PL_sv_undef);
1245 }
1246
897 SvREFCNT_dec (sv); 1247 SvREFCNT_dec (sv);
898 croak ("%s, at offset %d (octet 0x%02x)", dec.err, dec.cur - (U8 *)data, (int)(uint8_t)*dec.cur); 1248 croak ("%s, at offset %d (octet 0x%02x)", dec.err, dec.cur - (U8 *)data, (int)(uint8_t)*dec.cur);
899 } 1249 }
900 1250
901 sv = sv_2mortal (sv); 1251 sv = sv_2mortal (sv);
902 1252
903 return sv; 1253 return sv;
904} 1254}
905 1255
1256/////////////////////////////////////////////////////////////////////////////
1257// incremental parser
1258
1259#define INCR_DONE(cbor) (AvFILLp (cbor->incr_count) < 0)
1260
1261// returns 0 for notyet, 1 for success or error
1262static int
1263incr_parse (CBOR *self, SV *cborstr)
1264{
1265 STRLEN cur;
1266 SvPV (cborstr, cur);
1267
1268 while (ecb_expect_true (self->incr_need <= cur))
1269 {
1270 // table of integer count bytes
1271 static I8 incr_len[MINOR_MASK + 1] = {
1272 0, 0, 0, 0, 0, 0, 0, 0,
1273 0, 0, 0, 0, 0, 0, 0, 0,
1274 0, 0, 0, 0, 0, 0, 0, 0,
1275 1, 2, 4, 8,-1,-1,-1,-2
1276 };
1277
1278 const U8 *p = SvPVX (cborstr) + self->incr_pos;
1279 U8 m = *p & MINOR_MASK;
1280 IV count = SvIVX (AvARRAY (self->incr_count)[AvFILLp (self->incr_count)]);
1281 I8 ilen = incr_len[m];
1282
1283 self->incr_need = self->incr_pos + 1;
1284
1285 if (ecb_expect_false (ilen < 0))
1286 {
1287 if (m != MINOR_INDEF)
1288 return 1; // error
1289
1290 if (*p == (MAJOR_MISC | MINOR_INDEF))
1291 {
1292 if (count >= 0)
1293 return 1; // error
1294
1295 count = 1;
1296 }
1297 else
1298 {
1299 av_push (self->incr_count, newSViv (-1)); //TODO: nest
1300 count = -1;
1301 }
1302 }
1303 else
1304 {
1305 self->incr_need += ilen;
1306 if (ecb_expect_false (self->incr_need > cur))
1307 return 0;
1308
1309 int major = *p >> MAJOR_SHIFT;
1310
1311 switch (major)
1312 {
1313 case MAJOR_TAG >> MAJOR_SHIFT:
1314 ++count; // tags merely prefix another value
1315 break;
1316
1317 case MAJOR_BYTES >> MAJOR_SHIFT:
1318 case MAJOR_TEXT >> MAJOR_SHIFT:
1319 case MAJOR_ARRAY >> MAJOR_SHIFT:
1320 case MAJOR_MAP >> MAJOR_SHIFT:
1321 {
1322 UV len;
1323
1324 if (ecb_expect_false (ilen))
1325 {
1326 len = 0;
1327
1328 do {
1329 len = (len << 8) | *++p;
1330 } while (--ilen);
1331 }
1332 else
1333 len = m;
1334
1335 switch (major)
1336 {
1337 case MAJOR_BYTES >> MAJOR_SHIFT:
1338 case MAJOR_TEXT >> MAJOR_SHIFT:
1339 self->incr_need += len;
1340 if (ecb_expect_false (self->incr_need > cur))
1341 return 0;
1342
1343 break;
1344
1345 case MAJOR_MAP >> MAJOR_SHIFT:
1346 len <<= 1;
1347 case MAJOR_ARRAY >> MAJOR_SHIFT:
1348 if (len)
1349 {
1350 av_push (self->incr_count, newSViv (len + 1)); //TODO: nest
1351 count = len + 1;
1352 }
1353 break;
1354 }
1355 }
1356 }
1357 }
1358
1359 self->incr_pos = self->incr_need;
1360
1361 if (count > 0)
1362 {
1363 while (!--count)
1364 {
1365 if (!AvFILLp (self->incr_count))
1366 return 1; // done
1367
1368 SvREFCNT_dec_NN (av_pop (self->incr_count));
1369 count = SvIVX (AvARRAY (self->incr_count)[AvFILLp (self->incr_count)]);
1370 }
1371
1372 SvIVX (AvARRAY (self->incr_count)[AvFILLp (self->incr_count)]) = count;
1373 }
1374 }
1375
1376 return 0;
1377}
1378
1379
906///////////////////////////////////////////////////////////////////////////// 1380/////////////////////////////////////////////////////////////////////////////
907// XS interface functions 1381// XS interface functions
908 1382
909MODULE = CBOR::XS PACKAGE = CBOR::XS 1383MODULE = CBOR::XS PACKAGE = CBOR::XS
910 1384
918 1392
919 types_true = get_bool ("Types::Serialiser::true" ); 1393 types_true = get_bool ("Types::Serialiser::true" );
920 types_false = get_bool ("Types::Serialiser::false"); 1394 types_false = get_bool ("Types::Serialiser::false");
921 types_error = get_bool ("Types::Serialiser::error"); 1395 types_error = get_bool ("Types::Serialiser::error");
922 1396
1397 default_filter = newSVpv ("CBOR::XS::default_filter", 0);
1398
923 sv_cbor = newSVpv ("CBOR", 0); 1399 sv_cbor = newSVpv ("CBOR", 0);
924 SvREADONLY_on (sv_cbor); 1400 SvREADONLY_on (sv_cbor);
1401
1402 assert (("STRLEN must be an unsigned type", 0 <= (STRLEN)-1));
925} 1403}
926 1404
927PROTOTYPES: DISABLE 1405PROTOTYPES: DISABLE
928 1406
929void CLONE (...) 1407void CLONE (...)
947 1425
948void shrink (CBOR *self, int enable = 1) 1426void shrink (CBOR *self, int enable = 1)
949 ALIAS: 1427 ALIAS:
950 shrink = F_SHRINK 1428 shrink = F_SHRINK
951 allow_unknown = F_ALLOW_UNKNOWN 1429 allow_unknown = F_ALLOW_UNKNOWN
1430 allow_sharing = F_ALLOW_SHARING
1431 allow_cycles = F_ALLOW_CYCLES
1432 pack_strings = F_PACK_STRINGS
1433 text_keys = F_TEXT_KEYS
1434 text_strings = F_TEXT_STRINGS
1435 validate_utf8 = F_VALIDATE_UTF8
952 PPCODE: 1436 PPCODE:
953{ 1437{
954 if (enable) 1438 if (enable)
955 self->flags |= ix; 1439 self->flags |= ix;
956 else 1440 else
961 1445
962void get_shrink (CBOR *self) 1446void get_shrink (CBOR *self)
963 ALIAS: 1447 ALIAS:
964 get_shrink = F_SHRINK 1448 get_shrink = F_SHRINK
965 get_allow_unknown = F_ALLOW_UNKNOWN 1449 get_allow_unknown = F_ALLOW_UNKNOWN
1450 get_allow_sharing = F_ALLOW_SHARING
1451 get_allow_cycles = F_ALLOW_CYCLES
1452 get_pack_strings = F_PACK_STRINGS
1453 get_text_keys = F_TEXT_KEYS
1454 get_text_strings = F_TEXT_STRINGS
1455 get_validate_utf8 = F_VALIDATE_UTF8
966 PPCODE: 1456 PPCODE:
967 XPUSHs (boolSV (self->flags & ix)); 1457 XPUSHs (boolSV (self->flags & ix));
968 1458
969void max_depth (CBOR *self, U32 max_depth = 0x80000000UL) 1459void max_depth (CBOR *self, U32 max_depth = 0x80000000UL)
970 PPCODE: 1460 PPCODE:
983 XPUSHs (ST (0)); 1473 XPUSHs (ST (0));
984 1474
985int get_max_size (CBOR *self) 1475int get_max_size (CBOR *self)
986 CODE: 1476 CODE:
987 RETVAL = self->max_size; 1477 RETVAL = self->max_size;
1478 OUTPUT:
1479 RETVAL
1480
1481void filter (CBOR *self, SV *filter = 0)
1482 PPCODE:
1483 SvREFCNT_dec (self->filter);
1484 self->filter = filter ? newSVsv (filter) : filter;
1485 XPUSHs (ST (0));
1486
1487SV *get_filter (CBOR *self)
1488 CODE:
1489 RETVAL = self->filter ? self->filter : NEWSV (0, 0);
988 OUTPUT: 1490 OUTPUT:
989 RETVAL 1491 RETVAL
990 1492
991void encode (CBOR *self, SV *scalar) 1493void encode (CBOR *self, SV *scalar)
992 PPCODE: 1494 PPCODE:
1007 EXTEND (SP, 2); 1509 EXTEND (SP, 2);
1008 PUSHs (sv); 1510 PUSHs (sv);
1009 PUSHs (sv_2mortal (newSVuv (offset - SvPVX (cborstr)))); 1511 PUSHs (sv_2mortal (newSVuv (offset - SvPVX (cborstr))));
1010} 1512}
1011 1513
1514void incr_parse (CBOR *self, SV *cborstr)
1515 ALIAS:
1516 incr_parse_multiple = 1
1517 PPCODE:
1518{
1519 if (SvUTF8 (cborstr))
1520 sv_utf8_downgrade (cborstr, 0);
1521
1522 if (!self->incr_count)
1523 {
1524 self->incr_count = newAV ();
1525 self->incr_pos = 0;
1526 self->incr_need = 1;
1527
1528 av_push (self->incr_count, newSViv (1));
1529 }
1530
1531 do
1532 {
1533 if (!incr_parse (self, cborstr))
1534 {
1535 if (self->incr_need > self->max_size && self->max_size)
1536 croak ("attempted decode of CBOR text of %lu bytes size, but max_size is set to %lu",
1537 (unsigned long)self->incr_need, (unsigned long)self->max_size);
1538
1539 break;
1540 }
1541
1542 SV *sv;
1543 char *offset;
1544
1545 PUTBACK; sv = decode_cbor (cborstr, self, &offset); SPAGAIN;
1546 XPUSHs (sv);
1547
1548 sv_chop (cborstr, offset);
1549
1550 av_clear (self->incr_count);
1551 av_push (self->incr_count, newSViv (1));
1552
1553 self->incr_pos = 0;
1554 self->incr_need = self->incr_pos + 1;
1555 }
1556 while (ix);
1557}
1558
1559void incr_reset (CBOR *self)
1560 CODE:
1561{
1562 SvREFCNT_dec (self->incr_count);
1563 self->incr_count = 0;
1564}
1565
1566void DESTROY (CBOR *self)
1567 PPCODE:
1568 cbor_free (self);
1569
1012PROTOTYPES: ENABLE 1570PROTOTYPES: ENABLE
1013 1571
1014void encode_cbor (SV *scalar) 1572void encode_cbor (SV *scalar)
1573 ALIAS:
1574 encode_cbor = 0
1575 encode_cbor_sharing = F_ALLOW_SHARING
1015 PPCODE: 1576 PPCODE:
1016{ 1577{
1017 CBOR cbor; 1578 CBOR cbor;
1018 cbor_init (&cbor); 1579 cbor_init (&cbor);
1580 cbor.flags |= ix;
1019 PUTBACK; scalar = encode_cbor (scalar, &cbor); SPAGAIN; 1581 PUTBACK; scalar = encode_cbor (scalar, &cbor); SPAGAIN;
1020 XPUSHs (scalar); 1582 XPUSHs (scalar);
1021} 1583}
1022 1584
1023void decode_cbor (SV *cborstr) 1585void decode_cbor (SV *cborstr)

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines