ViewVC Help

View File | Revision Log | Show Annotations | Download File

/cvs/CBOR-XS/XS.xs

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

Diff Legend

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