ViewVC Help

View File | Revision Log | Show Annotations | Download File

/cvs/cvsroot/CBOR-XS/XS.xs

Comparing cvsroot/CBOR-XS/XS.xs (file contents):
Revision 1.5 by root, Sat Oct 26 21:14:20 2013 UTC vs.
Revision 1.66 by root, Sun Nov 29 21:32:01 2020 UTC

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

Diff Legend

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