ViewVC Help

View File | Revision Log | Show Annotations | Download File

/cvs/CBOR-XS/XS.xs

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

Diff Legend

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