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.47 by root, Mon Apr 27 20:21:53 2015 UTC

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

Diff Legend

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