ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/JSON-XS/XS.xs
Revision: 1.26
Committed: Fri Apr 6 21:17:09 2007 UTC (17 years, 1 month ago) by root
Branch: MAIN
Changes since 1.25: +5 -0 lines
Log Message:
*** empty log message ***

File Contents

# Content
1 #include "EXTERN.h"
2 #include "perl.h"
3 #include "XSUB.h"
4
5 #include "assert.h"
6 #include "string.h"
7 #include "stdlib.h"
8 #include "stdio.h"
9
10 #if defined(__BORLANDC__) || defined(_MSC_VER)
11 # define snprintf _snprintf // C compilers have this in stdio.h
12 #endif
13
14 #define F_ASCII 0x00000001UL
15 #define F_UTF8 0x00000002UL
16 #define F_INDENT 0x00000004UL
17 #define F_CANONICAL 0x00000008UL
18 #define F_SPACE_BEFORE 0x00000010UL
19 #define F_SPACE_AFTER 0x00000020UL
20 #define F_ALLOW_NONREF 0x00000080UL
21 #define F_SHRINK 0x00000100UL
22 #define F_MAXDEPTH 0xf8000000UL
23 #define S_MAXDEPTH 27
24
25 #define DEC_DEPTH(flags) (1UL << ((flags & F_MAXDEPTH) >> S_MAXDEPTH))
26
27 // F_SELFCONVERT? <=> to_json/toJson
28 // F_BLESSED? <=> { $__class__$ => }
29
30 #define F_PRETTY F_INDENT | F_SPACE_BEFORE | F_SPACE_AFTER
31 #define F_DEFAULT (9UL << S_MAXDEPTH)
32
33 #define INIT_SIZE 32 // initial scalar size to be allocated
34 #define INDENT_STEP 3 // spaces per indentation level
35
36 #define UTF8_MAX_LEN 11 // for perls UTF-X: max. number of octets per character
37 #define SHORT_STRING_LEN 512 // special-case strings of up to this size
38
39 #define SB do {
40 #define SE } while (0)
41
42 static HV *json_stash; // JSON::XS::
43
44 /////////////////////////////////////////////////////////////////////////////
45 // utility functions
46
47 static UV *
48 SvJSON (SV *sv)
49 {
50 if (!(SvROK (sv) && SvOBJECT (SvRV (sv)) && SvSTASH (SvRV (sv)) == json_stash))
51 croak ("object is not of type JSON::XS");
52
53 return &SvUVX (SvRV (sv));
54 }
55
56 static void
57 shrink (SV *sv)
58 {
59 sv_utf8_downgrade (sv, 1);
60 if (SvLEN (sv) > SvCUR (sv) + 1)
61 {
62 #ifdef SvPV_shrink_to_cur
63 SvPV_shrink_to_cur (sv);
64 #elif defined (SvPV_renew)
65 SvPV_renew (sv, SvCUR (sv) + 1);
66 #endif
67 }
68 }
69
70 // decode an utf-8 character and return it, or (UV)-1 in
71 // case of an error.
72 // we special-case "safe" characters from U+80 .. U+7FF,
73 // but use the very good perl function to parse anything else.
74 // note that we never call this function for a ascii codepoints
75 static UV
76 decode_utf8 (unsigned char *s, STRLEN len, STRLEN *clen)
77 {
78 if (s[0] > 0xdf || s[0] < 0xc2)
79 return utf8n_to_uvuni (s, len, clen, UTF8_CHECK_ONLY);
80 else if (len > 1 && s[1] >= 0x80 && s[1] <= 0xbf)
81 {
82 *clen = 2;
83 return ((s[0] & 0x1f) << 6) | (s[1] & 0x3f);
84 }
85 else
86 {
87 *clen = (STRLEN)-1;
88 return (UV)-1;
89 }
90 }
91
92 /////////////////////////////////////////////////////////////////////////////
93 // encoder
94
95 // structure used for encoding JSON
96 typedef struct
97 {
98 char *cur; // SvPVX (sv) + current output position
99 char *end; // SvEND (sv)
100 SV *sv; // result scalar
101 U32 flags; // F_*
102 U32 indent; // indentation level
103 U32 maxdepth; // max. indentation/recursion level
104 } enc_t;
105
106 static void
107 need (enc_t *enc, STRLEN len)
108 {
109 if (enc->cur + len >= enc->end)
110 {
111 STRLEN cur = enc->cur - SvPVX (enc->sv);
112 SvGROW (enc->sv, cur + len + 1);
113 enc->cur = SvPVX (enc->sv) + cur;
114 enc->end = SvPVX (enc->sv) + SvLEN (enc->sv);
115 }
116 }
117
118 static void
119 encode_ch (enc_t *enc, char ch)
120 {
121 need (enc, 1);
122 *enc->cur++ = ch;
123 }
124
125 static void
126 encode_str (enc_t *enc, char *str, STRLEN len, int is_utf8)
127 {
128 char *end = str + len;
129
130 need (enc, len);
131
132 while (str < end)
133 {
134 unsigned char ch = *(unsigned char *)str;
135
136 if (ch >= 0x20 && ch < 0x80) // most common case
137 {
138 if (ch == '"') // but with slow exceptions
139 {
140 need (enc, len += 1);
141 *enc->cur++ = '\\';
142 *enc->cur++ = '"';
143 }
144 else if (ch == '\\')
145 {
146 need (enc, len += 1);
147 *enc->cur++ = '\\';
148 *enc->cur++ = '\\';
149 }
150 else
151 *enc->cur++ = ch;
152
153 ++str;
154 }
155 else
156 {
157 switch (ch)
158 {
159 case '\010': need (enc, len += 1); *enc->cur++ = '\\'; *enc->cur++ = 'b'; ++str; break;
160 case '\011': need (enc, len += 1); *enc->cur++ = '\\'; *enc->cur++ = 't'; ++str; break;
161 case '\012': need (enc, len += 1); *enc->cur++ = '\\'; *enc->cur++ = 'n'; ++str; break;
162 case '\014': need (enc, len += 1); *enc->cur++ = '\\'; *enc->cur++ = 'f'; ++str; break;
163 case '\015': need (enc, len += 1); *enc->cur++ = '\\'; *enc->cur++ = 'r'; ++str; break;
164
165 default:
166 {
167 STRLEN clen;
168 UV uch;
169
170 if (is_utf8)
171 {
172 //uch = utf8n_to_uvuni (str, end - str, &clen, UTF8_CHECK_ONLY);
173 uch = decode_utf8 (str, end - str, &clen);
174 if (clen == (STRLEN)-1)
175 croak ("malformed or illegal unicode character in string [%.11s], cannot convert to JSON", str);
176 }
177 else
178 {
179 uch = ch;
180 clen = 1;
181 }
182
183 if (uch > 0x10FFFFUL)
184 croak ("out of range codepoint (0x%lx) encountered, unrepresentable in JSON", (unsigned long)uch);
185
186 if (uch < 0x80 || enc->flags & F_ASCII)
187 {
188 if (uch > 0xFFFFUL)
189 {
190 need (enc, len += 11);
191 sprintf (enc->cur, "\\u%04x\\u%04x",
192 (int)((uch - 0x10000) / 0x400 + 0xD800),
193 (int)((uch - 0x10000) % 0x400 + 0xDC00));
194 enc->cur += 12;
195 }
196 else
197 {
198 static char hexdigit [16] = "0123456789abcdef";
199 need (enc, len += 5);
200 *enc->cur++ = '\\';
201 *enc->cur++ = 'u';
202 *enc->cur++ = hexdigit [ uch >> 12 ];
203 *enc->cur++ = hexdigit [(uch >> 8) & 15];
204 *enc->cur++ = hexdigit [(uch >> 4) & 15];
205 *enc->cur++ = hexdigit [(uch >> 0) & 15];
206 }
207
208 str += clen;
209 }
210 else if (is_utf8)
211 {
212 need (enc, len += clen);
213 do
214 {
215 *enc->cur++ = *str++;
216 }
217 while (--clen);
218 }
219 else
220 {
221 need (enc, len += UTF8_MAX_LEN - 1); // never more than 11 bytes needed
222 enc->cur = uvuni_to_utf8_flags (enc->cur, uch, 0);
223 ++str;
224 }
225 }
226 }
227 }
228
229 --len;
230 }
231 }
232
233 static void
234 encode_indent (enc_t *enc)
235 {
236 if (enc->flags & F_INDENT)
237 {
238 int spaces = enc->indent * INDENT_STEP;
239
240 need (enc, spaces);
241 memset (enc->cur, ' ', spaces);
242 enc->cur += spaces;
243 }
244 }
245
246 static void
247 encode_space (enc_t *enc)
248 {
249 need (enc, 1);
250 encode_ch (enc, ' ');
251 }
252
253 static void
254 encode_nl (enc_t *enc)
255 {
256 if (enc->flags & F_INDENT)
257 {
258 need (enc, 1);
259 encode_ch (enc, '\n');
260 }
261 }
262
263 static void
264 encode_comma (enc_t *enc)
265 {
266 encode_ch (enc, ',');
267
268 if (enc->flags & F_INDENT)
269 encode_nl (enc);
270 else if (enc->flags & F_SPACE_AFTER)
271 encode_space (enc);
272 }
273
274 static void encode_sv (enc_t *enc, SV *sv);
275
276 static void
277 encode_av (enc_t *enc, AV *av)
278 {
279 int i, len = av_len (av);
280
281 if (enc->indent >= enc->maxdepth)
282 croak ("data structure too deep (hit recursion limit)");
283
284 encode_ch (enc, '['); encode_nl (enc);
285 ++enc->indent;
286
287 for (i = 0; i <= len; ++i)
288 {
289 encode_indent (enc);
290 encode_sv (enc, *av_fetch (av, i, 0));
291
292 if (i < len)
293 encode_comma (enc);
294 }
295
296 encode_nl (enc);
297
298 --enc->indent;
299 encode_indent (enc); encode_ch (enc, ']');
300 }
301
302 static void
303 encode_he (enc_t *enc, HE *he)
304 {
305 encode_ch (enc, '"');
306
307 if (HeKLEN (he) == HEf_SVKEY)
308 {
309 SV *sv = HeSVKEY (he);
310 STRLEN len;
311 char *str;
312
313 SvGETMAGIC (sv);
314 str = SvPV (sv, len);
315
316 encode_str (enc, str, len, SvUTF8 (sv));
317 }
318 else
319 encode_str (enc, HeKEY (he), HeKLEN (he), HeKUTF8 (he));
320
321 encode_ch (enc, '"');
322
323 if (enc->flags & F_SPACE_BEFORE) encode_space (enc);
324 encode_ch (enc, ':');
325 if (enc->flags & F_SPACE_AFTER ) encode_space (enc);
326 encode_sv (enc, HeVAL (he));
327 }
328
329 // compare hash entries, used when all keys are bytestrings
330 static int
331 he_cmp_fast (const void *a_, const void *b_)
332 {
333 int cmp;
334
335 HE *a = *(HE **)a_;
336 HE *b = *(HE **)b_;
337
338 STRLEN la = HeKLEN (a);
339 STRLEN lb = HeKLEN (b);
340
341 if (!(cmp = memcmp (HeKEY (a), HeKEY (b), la < lb ? la : lb)))
342 cmp = la - lb;
343
344 return cmp;
345 }
346
347 // compare hash entries, used when some keys are sv's or utf-x
348 static int
349 he_cmp_slow (const void *a, const void *b)
350 {
351 return sv_cmp (HeSVKEY_force (*(HE **)a), HeSVKEY_force (*(HE **)b));
352 }
353
354 static void
355 encode_hv (enc_t *enc, HV *hv)
356 {
357 int count, i;
358
359 if (enc->indent >= enc->maxdepth)
360 croak ("data structure too deep (hit recursion limit)");
361
362 encode_ch (enc, '{'); encode_nl (enc); ++enc->indent;
363
364 if ((count = hv_iterinit (hv)))
365 {
366 // for canonical output we have to sort by keys first
367 // actually, this is mostly due to the stupid so-called
368 // security workaround added somewhere in 5.8.x.
369 // that randomises hash orderings
370 if (enc->flags & F_CANONICAL)
371 {
372 HE *he, *hes [count]; // if your compiler dies here, you need to enable C99 mode
373 int fast = 1;
374
375 i = 0;
376 while ((he = hv_iternext (hv)))
377 {
378 hes [i++] = he;
379 if (HeKLEN (he) < 0 || HeKUTF8 (he))
380 fast = 0;
381 }
382
383 assert (i == count);
384
385 if (fast)
386 qsort (hes, count, sizeof (HE *), he_cmp_fast);
387 else
388 {
389 // hack to forcefully disable "use bytes"
390 COP cop = *PL_curcop;
391 cop.op_private = 0;
392
393 ENTER;
394 SAVETMPS;
395
396 SAVEVPTR (PL_curcop);
397 PL_curcop = &cop;
398
399 qsort (hes, count, sizeof (HE *), he_cmp_slow);
400
401 FREETMPS;
402 LEAVE;
403 }
404
405 for (i = 0; i < count; ++i)
406 {
407 encode_indent (enc);
408 encode_he (enc, hes [i]);
409
410 if (i < count - 1)
411 encode_comma (enc);
412 }
413
414 encode_nl (enc);
415 }
416 else
417 {
418 HE *he = hv_iternext (hv);
419
420 for (;;)
421 {
422 encode_indent (enc);
423 encode_he (enc, he);
424
425 if (!(he = hv_iternext (hv)))
426 break;
427
428 encode_comma (enc);
429 }
430
431 encode_nl (enc);
432 }
433 }
434
435 --enc->indent; encode_indent (enc); encode_ch (enc, '}');
436 }
437
438 // encode objects, arrays and special \0=false and \1=true values.
439 static void
440 encode_rv (enc_t *enc, SV *sv)
441 {
442 svtype svt;
443
444 SvGETMAGIC (sv);
445 svt = SvTYPE (sv);
446
447 if (svt == SVt_PVHV)
448 encode_hv (enc, (HV *)sv);
449 else if (svt == SVt_PVAV)
450 encode_av (enc, (AV *)sv);
451 else if (svt < SVt_PVAV)
452 {
453 if (SvNIOK (sv) && SvIV (sv) == 0)
454 encode_str (enc, "false", 5, 0);
455 else if (SvNIOK (sv) && SvIV (sv) == 1)
456 encode_str (enc, "true", 4, 0);
457 else
458 croak ("cannot encode reference to scalar '%s' unless the scalar is 0 or 1",
459 SvPV_nolen (sv_2mortal (newRV_inc (sv))));
460 }
461 else
462 croak ("encountered %s, but JSON can only represent references to arrays or hashes",
463 SvPV_nolen (sv_2mortal (newRV_inc (sv))));
464 }
465
466 static void
467 encode_sv (enc_t *enc, SV *sv)
468 {
469 SvGETMAGIC (sv);
470
471 if (SvPOKp (sv))
472 {
473 STRLEN len;
474 char *str = SvPV (sv, len);
475 encode_ch (enc, '"');
476 encode_str (enc, str, len, SvUTF8 (sv));
477 encode_ch (enc, '"');
478 }
479 else if (SvNOKp (sv))
480 {
481 need (enc, NV_DIG + 32);
482 Gconvert (SvNVX (sv), NV_DIG, 0, enc->cur);
483 enc->cur += strlen (enc->cur);
484 }
485 else if (SvIOKp (sv))
486 {
487 need (enc, 64);
488 enc->cur +=
489 SvIsUV(sv)
490 ? snprintf (enc->cur, 64, "%"UVuf, (UV)SvUVX (sv))
491 : snprintf (enc->cur, 64, "%"IVdf, (IV)SvIVX (sv));
492 }
493 else if (SvROK (sv))
494 encode_rv (enc, SvRV (sv));
495 else if (!SvOK (sv))
496 encode_str (enc, "null", 4, 0);
497 else
498 croak ("encountered perl type (%s,0x%x) that JSON cannot handle, you might want to report this",
499 SvPV_nolen (sv), SvFLAGS (sv));
500 }
501
502 static SV *
503 encode_json (SV *scalar, U32 flags)
504 {
505 enc_t enc;
506
507 if (!(flags & F_ALLOW_NONREF) && !SvROK (scalar))
508 croak ("hash- or arrayref expected (not a simple scalar, use allow_nonref to allow this)");
509
510 enc.flags = flags;
511 enc.sv = sv_2mortal (NEWSV (0, INIT_SIZE));
512 enc.cur = SvPVX (enc.sv);
513 enc.end = SvEND (enc.sv);
514 enc.indent = 0;
515 enc.maxdepth = DEC_DEPTH (flags);
516
517 SvPOK_only (enc.sv);
518 encode_sv (&enc, scalar);
519
520 if (!(flags & (F_ASCII | F_UTF8)))
521 SvUTF8_on (enc.sv);
522
523 SvCUR_set (enc.sv, enc.cur - SvPVX (enc.sv));
524
525 if (enc.flags & F_SHRINK)
526 shrink (enc.sv);
527
528 return enc.sv;
529 }
530
531 /////////////////////////////////////////////////////////////////////////////
532 // decoder
533
534 // structure used for decoding JSON
535 typedef struct
536 {
537 char *cur; // current parser pointer
538 char *end; // end of input string
539 const char *err; // parse error, if != 0
540 U32 flags; // F_*
541 U32 depth; // recursion depth
542 U32 maxdepth; // recursion depth limit
543 } dec_t;
544
545 static void
546 decode_ws (dec_t *dec)
547 {
548 for (;;)
549 {
550 char ch = *dec->cur;
551
552 if (ch > 0x20
553 || (ch != 0x20 && ch != 0x0a && ch != 0x0d && ch != 0x09))
554 break;
555
556 ++dec->cur;
557 }
558 }
559
560 #define ERR(reason) SB dec->err = reason; goto fail; SE
561
562 #define EXPECT_CH(ch) SB \
563 if (*dec->cur != ch) \
564 ERR (# ch " expected"); \
565 ++dec->cur; \
566 SE
567
568 #define DEC_INC_DEPTH if (++dec->depth > dec->maxdepth) ERR ("json datastructure exceeds maximum nesting level (set a higher max_depth)")
569 #define DEC_DEC_DEPTH --dec->depth
570
571 static SV *decode_sv (dec_t *dec);
572
573 static signed char decode_hexdigit[256];
574
575 static UV
576 decode_4hex (dec_t *dec)
577 {
578 signed char d1, d2, d3, d4;
579 unsigned char *cur = (unsigned char *)dec->cur;
580
581 d1 = decode_hexdigit [cur [0]]; if (d1 < 0) ERR ("four hexadecimal digits expected");
582 d2 = decode_hexdigit [cur [1]]; if (d2 < 0) ERR ("four hexadecimal digits expected");
583 d3 = decode_hexdigit [cur [2]]; if (d3 < 0) ERR ("four hexadecimal digits expected");
584 d4 = decode_hexdigit [cur [3]]; if (d4 < 0) ERR ("four hexadecimal digits expected");
585
586 dec->cur += 4;
587
588 return ((UV)d1) << 12
589 | ((UV)d2) << 8
590 | ((UV)d3) << 4
591 | ((UV)d4);
592
593 fail:
594 return (UV)-1;
595 }
596
597 static SV *
598 decode_str (dec_t *dec)
599 {
600 SV *sv = 0;
601 int utf8 = 0;
602
603 do
604 {
605 char buf [SHORT_STRING_LEN + UTF8_MAX_LEN];
606 char *cur = buf;
607
608 do
609 {
610 unsigned char ch = *(unsigned char *)dec->cur++;
611
612 if (ch == '"')
613 {
614 --dec->cur;
615 break;
616 }
617 else if (ch == '\\')
618 {
619 switch (*dec->cur)
620 {
621 case '\\':
622 case '/':
623 case '"': *cur++ = *dec->cur++; break;
624
625 case 'b': ++dec->cur; *cur++ = '\010'; break;
626 case 't': ++dec->cur; *cur++ = '\011'; break;
627 case 'n': ++dec->cur; *cur++ = '\012'; break;
628 case 'f': ++dec->cur; *cur++ = '\014'; break;
629 case 'r': ++dec->cur; *cur++ = '\015'; break;
630
631 case 'u':
632 {
633 UV lo, hi;
634 ++dec->cur;
635
636 hi = decode_4hex (dec);
637 if (hi == (UV)-1)
638 goto fail;
639
640 // possibly a surrogate pair
641 if (hi >= 0xd800)
642 if (hi < 0xdc00)
643 {
644 if (dec->cur [0] != '\\' || dec->cur [1] != 'u')
645 ERR ("missing low surrogate character in surrogate pair");
646
647 dec->cur += 2;
648
649 lo = decode_4hex (dec);
650 if (lo == (UV)-1)
651 goto fail;
652
653 if (lo < 0xdc00 || lo >= 0xe000)
654 ERR ("surrogate pair expected");
655
656 hi = (hi - 0xD800) * 0x400 + (lo - 0xDC00) + 0x10000;
657 }
658 else if (hi < 0xe000)
659 ERR ("missing high surrogate character in surrogate pair");
660
661 if (hi >= 0x80)
662 {
663 utf8 = 1;
664
665 cur = (char *)uvuni_to_utf8_flags (cur, hi, 0);
666 }
667 else
668 *cur++ = hi;
669 }
670 break;
671
672 default:
673 --dec->cur;
674 ERR ("illegal backslash escape sequence in string");
675 }
676 }
677 else if (ch >= 0x20 && ch <= 0x7f)
678 *cur++ = ch;
679 else if (ch >= 0x80)
680 {
681 STRLEN clen;
682 UV uch;
683
684 --dec->cur;
685
686 uch = decode_utf8 (dec->cur, dec->end - dec->cur, &clen);
687 if (clen == (STRLEN)-1)
688 ERR ("malformed UTF-8 character in JSON string");
689
690 do
691 *cur++ = *dec->cur++;
692 while (--clen);
693
694 utf8 = 1;
695 }
696 else
697 {
698 --dec->cur;
699
700 if (!ch)
701 ERR ("unexpected end of string while parsing JSON string");
702 else
703 ERR ("invalid character encountered while parsing JSON string");
704 }
705 }
706 while (cur < buf + SHORT_STRING_LEN);
707
708 {
709 STRLEN len = cur - buf;
710
711 if (sv)
712 {
713 SvGROW (sv, SvCUR (sv) + len + 1);
714 memcpy (SvPVX (sv) + SvCUR (sv), buf, len);
715 SvCUR_set (sv, SvCUR (sv) + len);
716 }
717 else
718 sv = newSVpvn (buf, len);
719 }
720 }
721 while (*dec->cur != '"');
722
723 ++dec->cur;
724
725 if (sv)
726 {
727 SvPOK_only (sv);
728 *SvEND (sv) = 0;
729
730 if (utf8)
731 SvUTF8_on (sv);
732 }
733 else
734 sv = newSVpvn ("", 0);
735
736 return sv;
737
738 fail:
739 return 0;
740 }
741
742 static SV *
743 decode_num (dec_t *dec)
744 {
745 int is_nv = 0;
746 char *start = dec->cur;
747
748 // [minus]
749 if (*dec->cur == '-')
750 ++dec->cur;
751
752 if (*dec->cur == '0')
753 {
754 ++dec->cur;
755 if (*dec->cur >= '0' && *dec->cur <= '9')
756 ERR ("malformed number (leading zero must not be followed by another digit)");
757 }
758 else if (*dec->cur < '0' || *dec->cur > '9')
759 ERR ("malformed number (no digits after initial minus)");
760 else
761 do
762 {
763 ++dec->cur;
764 }
765 while (*dec->cur >= '0' && *dec->cur <= '9');
766
767 // [frac]
768 if (*dec->cur == '.')
769 {
770 ++dec->cur;
771
772 if (*dec->cur < '0' || *dec->cur > '9')
773 ERR ("malformed number (no digits after decimal point)");
774
775 do
776 {
777 ++dec->cur;
778 }
779 while (*dec->cur >= '0' && *dec->cur <= '9');
780
781 is_nv = 1;
782 }
783
784 // [exp]
785 if (*dec->cur == 'e' || *dec->cur == 'E')
786 {
787 ++dec->cur;
788
789 if (*dec->cur == '-' || *dec->cur == '+')
790 ++dec->cur;
791
792 if (*dec->cur < '0' || *dec->cur > '9')
793 ERR ("malformed number (no digits after exp sign)");
794
795 do
796 {
797 ++dec->cur;
798 }
799 while (*dec->cur >= '0' && *dec->cur <= '9');
800
801 is_nv = 1;
802 }
803
804 if (!is_nv)
805 {
806 UV uv;
807 int numtype = grok_number (start, dec->cur - start, &uv);
808 if (numtype & IS_NUMBER_IN_UV)
809 if (numtype & IS_NUMBER_NEG)
810 {
811 if (uv < (UV)IV_MIN)
812 return newSViv (-(IV)uv);
813 }
814 else
815 return newSVuv (uv);
816 }
817
818 return newSVnv (Atof (start));
819
820 fail:
821 return 0;
822 }
823
824 static SV *
825 decode_av (dec_t *dec)
826 {
827 AV *av = newAV ();
828
829 DEC_INC_DEPTH;
830 decode_ws (dec);
831
832 if (*dec->cur == ']')
833 ++dec->cur;
834 else
835 for (;;)
836 {
837 SV *value;
838
839 value = decode_sv (dec);
840 if (!value)
841 goto fail;
842
843 av_push (av, value);
844
845 decode_ws (dec);
846
847 if (*dec->cur == ']')
848 {
849 ++dec->cur;
850 break;
851 }
852
853 if (*dec->cur != ',')
854 ERR (", or ] expected while parsing array");
855
856 ++dec->cur;
857 }
858
859 DEC_DEC_DEPTH;
860 return newRV_noinc ((SV *)av);
861
862 fail:
863 SvREFCNT_dec (av);
864 DEC_DEC_DEPTH;
865 return 0;
866 }
867
868 static SV *
869 decode_hv (dec_t *dec)
870 {
871 HV *hv = newHV ();
872
873 DEC_INC_DEPTH;
874 decode_ws (dec);
875
876 if (*dec->cur == '}')
877 ++dec->cur;
878 else
879 for (;;)
880 {
881 SV *key, *value;
882
883 decode_ws (dec); EXPECT_CH ('"');
884
885 key = decode_str (dec);
886 if (!key)
887 goto fail;
888
889 decode_ws (dec); EXPECT_CH (':');
890
891 value = decode_sv (dec);
892 if (!value)
893 {
894 SvREFCNT_dec (key);
895 goto fail;
896 }
897
898 hv_store_ent (hv, key, value, 0);
899 SvREFCNT_dec (key);
900
901 decode_ws (dec);
902
903 if (*dec->cur == '}')
904 {
905 ++dec->cur;
906 break;
907 }
908
909 if (*dec->cur != ',')
910 ERR (", or } expected while parsing object/hash");
911
912 ++dec->cur;
913 }
914
915 DEC_DEC_DEPTH;
916 return newRV_noinc ((SV *)hv);
917
918 fail:
919 SvREFCNT_dec (hv);
920 DEC_DEC_DEPTH;
921 return 0;
922 }
923
924 static SV *
925 decode_sv (dec_t *dec)
926 {
927 decode_ws (dec);
928 switch (*dec->cur)
929 {
930 case '"': ++dec->cur; return decode_str (dec);
931 case '[': ++dec->cur; return decode_av (dec);
932 case '{': ++dec->cur; return decode_hv (dec);
933
934 case '-':
935 case '0': case '1': case '2': case '3': case '4':
936 case '5': case '6': case '7': case '8': case '9':
937 return decode_num (dec);
938
939 case 't':
940 if (dec->end - dec->cur >= 4 && !memcmp (dec->cur, "true", 4))
941 {
942 dec->cur += 4;
943 return newSViv (1);
944 }
945 else
946 ERR ("'true' expected");
947
948 break;
949
950 case 'f':
951 if (dec->end - dec->cur >= 5 && !memcmp (dec->cur, "false", 5))
952 {
953 dec->cur += 5;
954 return newSViv (0);
955 }
956 else
957 ERR ("'false' expected");
958
959 break;
960
961 case 'n':
962 if (dec->end - dec->cur >= 4 && !memcmp (dec->cur, "null", 4))
963 {
964 dec->cur += 4;
965 return newSVsv (&PL_sv_undef);
966 }
967 else
968 ERR ("'null' expected");
969
970 break;
971
972 default:
973 ERR ("malformed JSON string, neither array, object, number, string or atom");
974 break;
975 }
976
977 fail:
978 return 0;
979 }
980
981 static SV *
982 decode_json (SV *string, U32 flags)
983 {
984 dec_t dec;
985 SV *sv;
986
987 SvUPGRADE (string, SVt_PV);
988
989 if (flags & F_UTF8)
990 sv_utf8_downgrade (string, 0);
991 else
992 sv_utf8_upgrade (string);
993
994 SvGROW (string, SvCUR (string) + 1); // should basically be a NOP
995
996 dec.flags = flags;
997 dec.cur = SvPVX (string);
998 dec.end = SvEND (string);
999 dec.err = 0;
1000 dec.depth = 0;
1001 dec.maxdepth = DEC_DEPTH (dec.flags);
1002
1003 *dec.end = 0; // this should basically be a nop, too, but make sure its there
1004 sv = decode_sv (&dec);
1005
1006 if (!sv)
1007 {
1008 IV offset = dec.flags & F_UTF8
1009 ? dec.cur - SvPVX (string)
1010 : utf8_distance (dec.cur, SvPVX (string));
1011 SV *uni = sv_newmortal ();
1012
1013 // horrible hack to silence warning inside pv_uni_display
1014 COP cop = *PL_curcop;
1015 cop.cop_warnings = pWARN_NONE;
1016 ENTER;
1017 SAVEVPTR (PL_curcop);
1018 PL_curcop = &cop;
1019 pv_uni_display (uni, dec.cur, dec.end - dec.cur, 20, UNI_DISPLAY_QQ);
1020 LEAVE;
1021
1022 croak ("%s, at character offset %d [\"%s\"]",
1023 dec.err,
1024 (int)offset,
1025 dec.cur != dec.end ? SvPV_nolen (uni) : "(end of string)");
1026 }
1027
1028 sv = sv_2mortal (sv);
1029
1030 if (!(dec.flags & F_ALLOW_NONREF) && !SvROK (sv))
1031 croak ("JSON text must be an object or array (but found number, string, true, false or null, use allow_nonref to allow this)");
1032
1033 return sv;
1034 }
1035
1036 /////////////////////////////////////////////////////////////////////////////
1037 // XS interface functions
1038
1039 MODULE = JSON::XS PACKAGE = JSON::XS
1040
1041 BOOT:
1042 {
1043 int i;
1044
1045 memset (decode_hexdigit, 0xff, 256);
1046
1047 for (i = 0; i < 256; ++i)
1048 decode_hexdigit [i] =
1049 i >= '0' && i <= '9' ? i - '0'
1050 : i >= 'a' && i <= 'f' ? i - 'a' + 10
1051 : i >= 'A' && i <= 'F' ? i - 'A' + 10
1052 : -1;
1053
1054 json_stash = gv_stashpv ("JSON::XS", 1);
1055 }
1056
1057 PROTOTYPES: DISABLE
1058
1059 SV *new (char *dummy)
1060 CODE:
1061 RETVAL = sv_bless (newRV_noinc (newSVuv (F_DEFAULT)), json_stash);
1062 OUTPUT:
1063 RETVAL
1064
1065 SV *ascii (SV *self, int enable = 1)
1066 ALIAS:
1067 ascii = F_ASCII
1068 utf8 = F_UTF8
1069 indent = F_INDENT
1070 canonical = F_CANONICAL
1071 space_before = F_SPACE_BEFORE
1072 space_after = F_SPACE_AFTER
1073 pretty = F_PRETTY
1074 allow_nonref = F_ALLOW_NONREF
1075 shrink = F_SHRINK
1076 CODE:
1077 {
1078 UV *uv = SvJSON (self);
1079 if (enable)
1080 *uv |= ix;
1081 else
1082 *uv &= ~ix;
1083
1084 RETVAL = newSVsv (self);
1085 }
1086 OUTPUT:
1087 RETVAL
1088
1089 SV *max_depth (SV *self, UV max_depth = 0x80000000UL)
1090 CODE:
1091 {
1092 UV *uv = SvJSON (self);
1093 UV log2 = 0;
1094
1095 if (max_depth > 0x80000000UL) max_depth = 0x80000000UL;
1096
1097 while ((1UL << log2) < max_depth)
1098 ++log2;
1099
1100 *uv = *uv & ~F_MAXDEPTH | (log2 << S_MAXDEPTH);
1101
1102 RETVAL = newSVsv (self);
1103 }
1104 OUTPUT:
1105 RETVAL
1106
1107 void encode (SV *self, SV *scalar)
1108 PPCODE:
1109 XPUSHs (encode_json (scalar, *SvJSON (self)));
1110
1111 void decode (SV *self, SV *jsonstr)
1112 PPCODE:
1113 XPUSHs (decode_json (jsonstr, *SvJSON (self)));
1114
1115 PROTOTYPES: ENABLE
1116
1117 void to_json (SV *scalar)
1118 ALIAS:
1119 objToJson = 0
1120 PPCODE:
1121 XPUSHs (encode_json (scalar, F_DEFAULT | F_UTF8));
1122
1123 void from_json (SV *jsonstr)
1124 ALIAS:
1125 jsonToObj = 0
1126 PPCODE:
1127 XPUSHs (decode_json (jsonstr, F_DEFAULT | F_UTF8));
1128