ViewVC Help

View File | Revision Log | Show Annotations | Download File

/cvs/CBOR-XS/XS.xs

Comparing CBOR-XS/XS.xs (file contents):
Revision 1.16 by root, Tue Oct 29 20:59:16 2013 UTC vs.
Revision 1.67 by root, Sun Nov 29 21:35:06 2020 UTC

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

Diff Legend

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