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.55 by root, Fri Nov 25 06:13:16 2016 UTC

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

Diff Legend

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