No puede seleccionar más de 25 temas Los temas deben comenzar con una letra o número, pueden incluir guiones ('-') y pueden tener hasta 35 caracteres de largo.

1063 líneas
30 KiB

  1. // This file is part of Jiffy released under the MIT license.
  2. // See the LICENSE file for more information.
  3. #include <assert.h>
  4. #include <errno.h>
  5. #include <stdio.h>
  6. #include <stdlib.h>
  7. #include <string.h>
  8. #include "erl_nif.h"
  9. #include "jiffy.h"
  10. #define U(c) ((unsigned char) (c))
  11. #define ERROR(i, msg) make_error(st, env, msg)
  12. #define STACK_SIZE_INC 64
  13. #define NUM_BUF_LEN 32
  14. #if WINDOWS || WIN32
  15. #define snprintf _snprintf
  16. #endif
  17. enum {
  18. st_value=0,
  19. st_object,
  20. st_array,
  21. st_key,
  22. st_colon,
  23. st_comma,
  24. st_done,
  25. st_invalid
  26. } JsonState;
  27. enum {
  28. nst_init=0,
  29. nst_sign,
  30. nst_mantissa,
  31. nst_frac0,
  32. nst_frac1,
  33. nst_frac,
  34. nst_esign,
  35. nst_edigit
  36. } JsonNumState;
  37. typedef struct {
  38. ErlNifEnv* env;
  39. jiffy_st* atoms;
  40. ERL_NIF_TERM arg;
  41. ErlNifBinary bin;
  42. size_t bytes_per_iter;
  43. int is_partial;
  44. int return_maps;
  45. int use_nil;
  46. int with_trailer;
  47. char* p;
  48. unsigned char* u;
  49. int i;
  50. int len;
  51. char* st_data;
  52. int st_size;
  53. int st_top;
  54. } Decoder;
  55. Decoder*
  56. dec_new(ErlNifEnv* env)
  57. {
  58. jiffy_st* st = (jiffy_st*) enif_priv_data(env);
  59. Decoder* d = enif_alloc_resource(st->res_dec, sizeof(Decoder));
  60. int i;
  61. if(d == NULL) {
  62. return NULL;
  63. }
  64. d->atoms = st;
  65. d->bytes_per_iter = DEFAULT_BYTES_PER_ITER;
  66. d->is_partial = 0;
  67. d->return_maps = 0;
  68. d->use_nil = 0;
  69. d->with_trailer = 0;
  70. d->p = NULL;
  71. d->u = NULL;
  72. d->len = -1;
  73. d->i = -1;
  74. d->st_data = (char*) enif_alloc(STACK_SIZE_INC * sizeof(char));
  75. d->st_size = STACK_SIZE_INC;
  76. d->st_top = 0;
  77. for(i = 0; i < d->st_size; i++) {
  78. d->st_data[i] = st_invalid;
  79. }
  80. d->st_data[0] = st_value;
  81. d->st_top++;
  82. return d;
  83. }
  84. void
  85. dec_init(Decoder* d, ErlNifEnv* env, ERL_NIF_TERM arg, ErlNifBinary* bin)
  86. {
  87. d->env = env;
  88. d->arg = arg;
  89. d->p = (char*) bin->data;
  90. d->u = bin->data;
  91. d->len = bin->size;
  92. // I'd like to be more forceful on this check so that when
  93. // we run a second iteration of the decoder we are sure
  94. // that we're using the same binary. Unfortunately, I don't
  95. // think there's a value to base this assertion on.
  96. if(d->i < 0) {
  97. d->i = 0;
  98. } else {
  99. assert(d->i <= d->len && "mismatched binary lengths");
  100. }
  101. }
  102. void
  103. dec_destroy(ErlNifEnv* env, void* obj)
  104. {
  105. Decoder* d = (Decoder*) obj;
  106. if(d->st_data != NULL) {
  107. enif_free(d->st_data);
  108. }
  109. }
  110. ERL_NIF_TERM
  111. dec_error(Decoder* d, const char* atom)
  112. {
  113. ERL_NIF_TERM pos = enif_make_int(d->env, d->i+1);
  114. ERL_NIF_TERM msg = make_atom(d->env, atom);
  115. ERL_NIF_TERM ret = enif_make_tuple2(d->env, pos, msg);
  116. return enif_make_tuple2(d->env, d->atoms->atom_error, ret);
  117. }
  118. char
  119. dec_curr(Decoder* d)
  120. {
  121. return d->st_data[d->st_top-1];
  122. }
  123. int
  124. dec_top(Decoder* d)
  125. {
  126. return d->st_top;
  127. }
  128. void
  129. dec_push(Decoder* d, char val)
  130. {
  131. char* tmp;
  132. int new_sz;
  133. int i;
  134. if(d->st_top >= d->st_size) {
  135. new_sz = d->st_size + STACK_SIZE_INC;
  136. tmp = (char*) enif_alloc(new_sz * sizeof(char));
  137. memcpy(tmp, d->st_data, d->st_size * sizeof(char));
  138. enif_free(d->st_data);
  139. d->st_data = tmp;
  140. d->st_size = new_sz;
  141. for(i = d->st_top; i < d->st_size; i++) {
  142. d->st_data[i] = st_invalid;
  143. }
  144. }
  145. d->st_data[d->st_top++] = val;
  146. }
  147. void
  148. dec_pop(Decoder* d, char val)
  149. {
  150. assert(d->st_data[d->st_top-1] == val && "popped invalid state.");
  151. d->st_data[d->st_top-1] = st_invalid;
  152. d->st_top--;
  153. }
  154. int
  155. dec_string(Decoder* d, ERL_NIF_TERM* value)
  156. {
  157. int has_escape = 0;
  158. int num_escapes = 0;
  159. int st;
  160. int ulen;
  161. int ui;
  162. int hi;
  163. int lo;
  164. char* chrbuf;
  165. int chrpos;
  166. if(d->p[d->i] != '\"') {
  167. return 0;
  168. }
  169. d->i++;
  170. st = d->i;
  171. while(d->i < d->len) {
  172. if(d->u[d->i] < 0x20) {
  173. return 0;
  174. } else if(d->p[d->i] == '\"') {
  175. d->i++;
  176. goto parse;
  177. } else if(d->p[d->i] == '\\') {
  178. if(d->i+1 >= d->len) {
  179. return 0;
  180. }
  181. has_escape = 1;
  182. num_escapes += 1;
  183. d->i++;
  184. switch(d->p[d->i]) {
  185. case '\"':
  186. case '\\':
  187. case '/':
  188. case 'b':
  189. case 'f':
  190. case 'n':
  191. case 'r':
  192. case 't':
  193. d->i++;
  194. break;
  195. case 'u':
  196. hi = 0;
  197. lo = 0;
  198. d->i++;
  199. if(d->i + 4 >= d->len) {
  200. return 0;
  201. }
  202. hi = int_from_hex(&(d->u[d->i]));
  203. if(hi < 0) {
  204. return 0;
  205. }
  206. d->i += 4;
  207. if(hi >= 0xD800 && hi < 0xDC00) {
  208. if(d->i + 6 >= d->len) {
  209. return 0;
  210. }
  211. if(d->p[d->i++] != '\\') {
  212. return 0;
  213. } else if(d->p[d->i++] != 'u') {
  214. return 0;
  215. }
  216. lo = int_from_hex(&(d->u[d->i]));
  217. if(lo < 0) {
  218. return 0;
  219. }
  220. hi = unicode_from_pair(hi, lo);
  221. if(hi < 0) {
  222. return 0;
  223. }
  224. }
  225. hi = utf8_len(hi);
  226. if(hi < 0) {
  227. return 0;
  228. }
  229. if(lo == 0) {
  230. num_escapes += 5 - hi;
  231. } else {
  232. num_escapes += 11 - hi;
  233. }
  234. break;
  235. default:
  236. return 0;
  237. }
  238. } else if(d->u[d->i] < 0x80) {
  239. d->i++;
  240. } else {
  241. ulen = utf8_validate(&(d->u[d->i]), d->len - d->i);
  242. if(ulen < 0) {
  243. return 0;
  244. }
  245. d->i += ulen;
  246. }
  247. }
  248. // The goto above ensures that we only
  249. // hit this when a string is not terminated
  250. // correctly.
  251. return 0;
  252. parse:
  253. if(!has_escape) {
  254. *value = enif_make_sub_binary(d->env, d->arg, st, (d->i - st - 1));
  255. return 1;
  256. }
  257. hi = 0;
  258. lo = 0;
  259. ulen = (d->i - 1) - st - num_escapes;
  260. chrbuf = (char*) enif_make_new_binary(d->env, ulen, value);
  261. chrpos = 0;
  262. ui = st;
  263. while(ui < d->i - 1) {
  264. if(d->p[ui] != '\\') {
  265. chrbuf[chrpos++] = d->p[ui++];
  266. continue;
  267. }
  268. ui++;
  269. switch(d->p[ui]) {
  270. case '\"':
  271. case '\\':
  272. case '/':
  273. chrbuf[chrpos++] = d->p[ui];
  274. ui++;
  275. break;
  276. case 'b':
  277. chrbuf[chrpos++] = '\b';
  278. ui++;
  279. break;
  280. case 'f':
  281. chrbuf[chrpos++] = '\f';
  282. ui++;
  283. break;
  284. case 'n':
  285. chrbuf[chrpos++] = '\n';
  286. ui++;
  287. break;
  288. case 'r':
  289. chrbuf[chrpos++] = '\r';
  290. ui++;
  291. break;
  292. case 't':
  293. chrbuf[chrpos++] = '\t';
  294. ui++;
  295. break;
  296. case 'u':
  297. ui++;
  298. hi = int_from_hex(&(d->u[ui]));
  299. if(hi < 0) {
  300. return 0;
  301. }
  302. if(hi >= 0xD800 && hi < 0xDC00) {
  303. lo = int_from_hex(&(d->u[ui+6]));
  304. if(lo < 0) {
  305. return 0;
  306. }
  307. hi = unicode_from_pair(hi, lo);
  308. ui += 10;
  309. } else {
  310. ui += 4;
  311. }
  312. hi = unicode_to_utf8(hi, (unsigned char*) chrbuf+chrpos);
  313. if(hi < 0) {
  314. return 0;
  315. }
  316. chrpos += hi;
  317. break;
  318. default:
  319. return 0;
  320. }
  321. }
  322. return 1;
  323. }
  324. int
  325. dec_number(Decoder* d, ERL_NIF_TERM* value)
  326. {
  327. ERL_NIF_TERM num_type = d->atoms->atom_error;
  328. char state = nst_init;
  329. char nbuf[NUM_BUF_LEN];
  330. int st = d->i;
  331. int has_frac = 0;
  332. int has_exp = 0;
  333. double dval;
  334. long lval;
  335. while(d->i < d->len) {
  336. switch(state) {
  337. case nst_init:
  338. switch(d->p[d->i]) {
  339. case '-':
  340. state = nst_sign;
  341. d->i++;
  342. break;
  343. case '0':
  344. state = nst_frac0;
  345. d->i++;
  346. break;
  347. case '1':
  348. case '2':
  349. case '3':
  350. case '4':
  351. case '5':
  352. case '6':
  353. case '7':
  354. case '8':
  355. case '9':
  356. state = nst_mantissa;
  357. d->i++;
  358. break;
  359. default:
  360. return 0;
  361. }
  362. break;
  363. case nst_sign:
  364. switch(d->p[d->i]) {
  365. case '0':
  366. state = nst_frac0;
  367. d->i++;
  368. break;
  369. case '1':
  370. case '2':
  371. case '3':
  372. case '4':
  373. case '5':
  374. case '6':
  375. case '7':
  376. case '8':
  377. case '9':
  378. state = nst_mantissa;
  379. d->i++;
  380. break;
  381. default:
  382. return 0;
  383. }
  384. break;
  385. case nst_mantissa:
  386. switch(d->p[d->i]) {
  387. case '.':
  388. state = nst_frac1;
  389. d->i++;
  390. break;
  391. case 'e':
  392. case 'E':
  393. state = nst_esign;
  394. d->i++;
  395. break;
  396. case '0':
  397. case '1':
  398. case '2':
  399. case '3':
  400. case '4':
  401. case '5':
  402. case '6':
  403. case '7':
  404. case '8':
  405. case '9':
  406. d->i++;
  407. break;
  408. default:
  409. goto parse;
  410. }
  411. break;
  412. case nst_frac0:
  413. switch(d->p[d->i]) {
  414. case '.':
  415. state = nst_frac1;
  416. d->i++;
  417. break;
  418. case 'e':
  419. case 'E':
  420. state = nst_esign;
  421. d->i++;
  422. break;
  423. default:
  424. goto parse;
  425. }
  426. break;
  427. case nst_frac1:
  428. has_frac = 1;
  429. switch(d->p[d->i]) {
  430. case '0':
  431. case '1':
  432. case '2':
  433. case '3':
  434. case '4':
  435. case '5':
  436. case '6':
  437. case '7':
  438. case '8':
  439. case '9':
  440. state = nst_frac;
  441. d->i++;
  442. break;
  443. default:
  444. goto parse;
  445. }
  446. break;
  447. case nst_frac:
  448. switch(d->p[d->i]) {
  449. case 'e':
  450. case 'E':
  451. state = nst_esign;
  452. d->i++;
  453. break;
  454. case '0':
  455. case '1':
  456. case '2':
  457. case '3':
  458. case '4':
  459. case '5':
  460. case '6':
  461. case '7':
  462. case '8':
  463. case '9':
  464. d->i++;
  465. break;
  466. default:
  467. goto parse;
  468. }
  469. break;
  470. case nst_esign:
  471. has_exp = 1;
  472. switch(d->p[d->i]) {
  473. case '-':
  474. case '+':
  475. case '0':
  476. case '1':
  477. case '2':
  478. case '3':
  479. case '4':
  480. case '5':
  481. case '6':
  482. case '7':
  483. case '8':
  484. case '9':
  485. state = nst_edigit;
  486. d->i++;
  487. break;
  488. default:
  489. return 0;
  490. }
  491. break;
  492. case nst_edigit:
  493. switch(d->p[d->i]) {
  494. case '0':
  495. case '1':
  496. case '2':
  497. case '3':
  498. case '4':
  499. case '5':
  500. case '6':
  501. case '7':
  502. case '8':
  503. case '9':
  504. d->i++;
  505. break;
  506. default:
  507. goto parse;
  508. }
  509. break;
  510. default:
  511. return 0;
  512. }
  513. }
  514. parse:
  515. switch(state) {
  516. case nst_init:
  517. case nst_sign:
  518. case nst_frac1:
  519. case nst_esign:
  520. return 0;
  521. default:
  522. break;
  523. }
  524. errno = 0;
  525. if(d->i - st < NUM_BUF_LEN) {
  526. memset(nbuf, 0, NUM_BUF_LEN);
  527. memcpy(nbuf, &(d->p[st]), d->i - st);
  528. if(has_frac || has_exp) {
  529. dval = strtod(nbuf, NULL);
  530. if(errno != ERANGE) {
  531. *value = enif_make_double(d->env, dval);
  532. return 1;
  533. }
  534. } else {
  535. lval = strtol(nbuf, NULL, 10);
  536. if(errno != ERANGE) {
  537. *value = enif_make_int64(d->env, lval);
  538. return 1;
  539. }
  540. }
  541. }
  542. if(!has_frac && !has_exp) {
  543. num_type = d->atoms->atom_bignum;
  544. } else if(!has_frac && has_exp) {
  545. num_type = d->atoms->atom_bignum_e;
  546. } else {
  547. num_type = d->atoms->atom_bigdbl;
  548. }
  549. d->is_partial = 1;
  550. *value = enif_make_sub_binary(d->env, d->arg, st, d->i - st);
  551. *value = enif_make_tuple2(d->env, num_type, *value);
  552. return 1;
  553. }
  554. ERL_NIF_TERM
  555. make_empty_object(ErlNifEnv* env, int ret_map)
  556. {
  557. #if MAP_TYPE_PRESENT
  558. if(ret_map) {
  559. return enif_make_new_map(env);
  560. }
  561. #endif
  562. return enif_make_tuple1(env, enif_make_list(env, 0));
  563. }
  564. int
  565. make_object(ErlNifEnv* env, ERL_NIF_TERM pairs, ERL_NIF_TERM* out, int ret_map)
  566. {
  567. ERL_NIF_TERM ret;
  568. ERL_NIF_TERM key;
  569. ERL_NIF_TERM val;
  570. #if MAP_TYPE_PRESENT
  571. if(ret_map) {
  572. ret = enif_make_new_map(env);
  573. while(enif_get_list_cell(env, pairs, &val, &pairs)) {
  574. if(!enif_get_list_cell(env, pairs, &key, &pairs)) {
  575. assert(0 == 1 && "Unbalanced object pairs.");
  576. }
  577. if(!enif_make_map_put(env, ret, key, val, &ret)) {
  578. return 0;
  579. }
  580. }
  581. *out = ret;
  582. return 1;
  583. }
  584. #endif
  585. ret = enif_make_list(env, 0);
  586. while(enif_get_list_cell(env, pairs, &val, &pairs)) {
  587. if(!enif_get_list_cell(env, pairs, &key, &pairs)) {
  588. assert(0 == 1 && "Unbalanced object pairs.");
  589. }
  590. val = enif_make_tuple2(env, key, val);
  591. ret = enif_make_list_cell(env, val, ret);
  592. }
  593. *out = enif_make_tuple1(env, ret);
  594. return 1;
  595. }
  596. ERL_NIF_TERM
  597. make_array(ErlNifEnv* env, ERL_NIF_TERM list)
  598. {
  599. ERL_NIF_TERM ret = enif_make_list(env, 0);
  600. ERL_NIF_TERM item;
  601. while(enif_get_list_cell(env, list, &item, &list)) {
  602. ret = enif_make_list_cell(env, item, ret);
  603. }
  604. return ret;
  605. }
  606. ERL_NIF_TERM
  607. decode_init(ErlNifEnv* env, int argc, const ERL_NIF_TERM argv[])
  608. {
  609. Decoder* d;
  610. jiffy_st* st = (jiffy_st*) enif_priv_data(env);
  611. ERL_NIF_TERM tmp_argv[5];
  612. ERL_NIF_TERM opts;
  613. ERL_NIF_TERM val;
  614. if(argc != 2) {
  615. return enif_make_badarg(env);
  616. }
  617. d = dec_new(env);
  618. if(d == NULL) {
  619. return make_error(st, env, "internal_error");
  620. }
  621. tmp_argv[0] = argv[0];
  622. tmp_argv[1] = enif_make_resource(env, d);
  623. tmp_argv[2] = st->atom_error;
  624. tmp_argv[3] = enif_make_list(env, 0);
  625. tmp_argv[4] = enif_make_list(env, 0);
  626. enif_release_resource(d);
  627. opts = argv[1];
  628. if(!enif_is_list(env, opts)) {
  629. return enif_make_badarg(env);
  630. }
  631. while(enif_get_list_cell(env, opts, &val, &opts)) {
  632. if(get_bytes_per_iter(env, val, &(d->bytes_per_iter))) {
  633. continue;
  634. } else if(enif_compare(val, d->atoms->atom_return_maps) == 0) {
  635. #if MAP_TYPE_PRESENT
  636. d->return_maps = 1;
  637. #else
  638. return enif_make_badarg(env);
  639. #endif
  640. } else if(enif_compare(val, d->atoms->atom_use_nil) == 0) {
  641. d->use_nil = 1;
  642. } else if(enif_compare(val, d->atoms->atom_with_trailer) == 0) {
  643. d->with_trailer = 1;
  644. } else {
  645. return enif_make_badarg(env);
  646. }
  647. }
  648. return decode_iter(env, 5, tmp_argv);
  649. }
  650. ERL_NIF_TERM
  651. decode_iter(ErlNifEnv* env, int argc, const ERL_NIF_TERM argv[])
  652. {
  653. Decoder* d;
  654. jiffy_st* st = (jiffy_st*) enif_priv_data(env);
  655. ErlNifBinary bin;
  656. ERL_NIF_TERM objs;
  657. ERL_NIF_TERM curr;
  658. ERL_NIF_TERM val = argv[2];
  659. ERL_NIF_TERM ret;
  660. size_t start;
  661. if(argc != 5) {
  662. return enif_make_badarg(env);
  663. } else if(!enif_inspect_binary(env, argv[0], &bin)) {
  664. return enif_make_badarg(env);
  665. } else if(!enif_get_resource(env, argv[1], st->res_dec, (void**) &d)) {
  666. return enif_make_badarg(env);
  667. } else if(!enif_is_list(env, argv[3])) {
  668. return enif_make_badarg(env);
  669. } else if(!enif_is_list(env, argv[4])) {
  670. return enif_make_badarg(env);
  671. }
  672. dec_init(d, env, argv[0], &bin);
  673. objs = argv[3];
  674. curr = argv[4];
  675. //fprintf(stderr, "Parsing:\r\n");
  676. start = d->i;
  677. while(d->i < bin.size) {
  678. //fprintf(stderr, "state: %d\r\n", dec_curr(d));
  679. if(should_yield(d->i - start, d->bytes_per_iter)
  680. /* A system could handle roughly 100kb per millisecond on a single core.
  681. * So the total amount of work per millisecond is 100kb.
  682. * We report the percentage of the time every (bytes_per_iter) bytes
  683. * in hope that the system will ask as to yield. We don't yield until
  684. * asked by the system according to our feedback (of questionable accuracy).
  685. */
  686. && consume_timeslice(env, d->i - start, 100000)) {
  687. return enif_make_tuple5(
  688. env,
  689. st->atom_iter,
  690. argv[1],
  691. val,
  692. objs,
  693. curr
  694. );
  695. }
  696. switch(dec_curr(d)) {
  697. case st_value:
  698. switch(d->p[d->i]) {
  699. case ' ':
  700. case '\n':
  701. case '\r':
  702. case '\t':
  703. d->i++;
  704. break;
  705. case 'n':
  706. if(d->i + 3 >= d->len) {
  707. ret = dec_error(d, "invalid_literal");
  708. goto done;
  709. }
  710. if(memcmp(&(d->p[d->i]), "null", 4) != 0) {
  711. ret = dec_error(d, "invalid_literal");
  712. goto done;
  713. }
  714. val = d->use_nil ? d->atoms->atom_nil : d->atoms->atom_null;
  715. dec_pop(d, st_value);
  716. d->i += 4;
  717. break;
  718. case 't':
  719. if(d->i + 3 >= d->len) {
  720. ret = dec_error(d, "invalid_literal");
  721. goto done;
  722. }
  723. if(memcmp(&(d->p[d->i]), "true", 4) != 0) {
  724. ret = dec_error(d, "invalid_literal");
  725. goto done;
  726. }
  727. val = d->atoms->atom_true;
  728. dec_pop(d, st_value);
  729. d->i += 4;
  730. break;
  731. case 'f':
  732. if(d->i + 4 >= bin.size) {
  733. ret = dec_error(d, "invalid_literal");
  734. goto done;
  735. }
  736. if(memcmp(&(d->p[d->i]), "false", 5) != 0) {
  737. ret = dec_error(d, "invalid_literal");
  738. goto done;
  739. }
  740. val = d->atoms->atom_false;
  741. dec_pop(d, st_value);
  742. d->i += 5;
  743. break;
  744. case '\"':
  745. if(!dec_string(d, &val)) {
  746. ret = dec_error(d, "invalid_string");
  747. goto done;
  748. }
  749. dec_pop(d, st_value);
  750. break;
  751. case '-':
  752. case '0':
  753. case '1':
  754. case '2':
  755. case '3':
  756. case '4':
  757. case '5':
  758. case '6':
  759. case '7':
  760. case '8':
  761. case '9':
  762. if(!dec_number(d, &val)) {
  763. ret = dec_error(d, "invalid_number");
  764. goto done;
  765. }
  766. dec_pop(d, st_value);
  767. break;
  768. case '{':
  769. dec_push(d, st_object);
  770. dec_push(d, st_key);
  771. objs = enif_make_list_cell(env, curr, objs);
  772. curr = enif_make_list(env, 0);
  773. d->i++;
  774. break;
  775. case '[':
  776. dec_push(d, st_array);
  777. dec_push(d, st_value);
  778. objs = enif_make_list_cell(env, curr, objs);
  779. curr = enif_make_list(env, 0);
  780. d->i++;
  781. break;
  782. case ']':
  783. if(!enif_is_empty_list(env, curr)) {
  784. ret = dec_error(d, "invalid_json");
  785. goto done;
  786. }
  787. dec_pop(d, st_value);
  788. if(dec_curr(d) != st_array) {
  789. ret = dec_error(d, "invalid_json");
  790. goto done;
  791. }
  792. dec_pop(d, st_array);
  793. dec_pop(d, st_value);
  794. val = curr; // curr is []
  795. if(!enif_get_list_cell(env, objs, &curr, &objs)) {
  796. ret = dec_error(d, "internal_error");
  797. goto done;
  798. }
  799. d->i++;
  800. break;
  801. default:
  802. ret = dec_error(d, "invalid_json");
  803. goto done;
  804. }
  805. if(dec_top(d) == 0) {
  806. dec_push(d, st_done);
  807. } else if(dec_curr(d) != st_value && dec_curr(d) != st_key) {
  808. dec_push(d, st_comma);
  809. curr = enif_make_list_cell(env, val, curr);
  810. }
  811. break;
  812. case st_key:
  813. switch(d->p[d->i]) {
  814. case ' ':
  815. case '\n':
  816. case '\r':
  817. case '\t':
  818. d->i++;
  819. break;
  820. case '\"':
  821. if(!dec_string(d, &val)) {
  822. ret = dec_error(d, "invalid_string");
  823. goto done;
  824. }
  825. dec_pop(d, st_key);
  826. dec_push(d, st_colon);
  827. curr = enif_make_list_cell(env, val, curr);
  828. break;
  829. case '}':
  830. if(!enif_is_empty_list(env, curr)) {
  831. ret = dec_error(d, "invalid_json");
  832. goto done;
  833. }
  834. dec_pop(d, st_key);
  835. dec_pop(d, st_object);
  836. dec_pop(d, st_value);
  837. val = make_empty_object(env, d->return_maps);
  838. if(!enif_get_list_cell(env, objs, &curr, &objs)) {
  839. ret = dec_error(d, "internal_error");
  840. goto done;
  841. }
  842. if(dec_top(d) == 0) {
  843. dec_push(d, st_done);
  844. } else {
  845. dec_push(d, st_comma);
  846. curr = enif_make_list_cell(env, val, curr);
  847. }
  848. d->i++;
  849. break;
  850. default:
  851. ret = dec_error(d, "invalid_json");
  852. goto done;
  853. }
  854. break;
  855. case st_colon:
  856. switch(d->p[d->i]) {
  857. case ' ':
  858. case '\n':
  859. case '\r':
  860. case '\t':
  861. d->i++;
  862. break;
  863. case ':':
  864. dec_pop(d, st_colon);
  865. dec_push(d, st_value);
  866. d->i++;
  867. break;
  868. default:
  869. ret = dec_error(d, "invalid_json");
  870. goto done;
  871. }
  872. break;
  873. case st_comma:
  874. switch(d->p[d->i]) {
  875. case ' ':
  876. case '\n':
  877. case '\r':
  878. case '\t':
  879. d->i++;
  880. break;
  881. case ',':
  882. dec_pop(d, st_comma);
  883. switch(dec_curr(d)) {
  884. case st_object:
  885. dec_push(d, st_key);
  886. break;
  887. case st_array:
  888. dec_push(d, st_value);
  889. break;
  890. default:
  891. ret = dec_error(d, "internal_error");
  892. goto done;
  893. }
  894. d->i++;
  895. break;
  896. case '}':
  897. dec_pop(d, st_comma);
  898. if(dec_curr(d) != st_object) {
  899. ret = dec_error(d, "invalid_json");
  900. goto done;
  901. }
  902. dec_pop(d, st_object);
  903. dec_pop(d, st_value);
  904. if(!make_object(env, curr, &val, d->return_maps)) {
  905. ret = dec_error(d, "internal_object_error");
  906. goto done;
  907. }
  908. if(!enif_get_list_cell(env, objs, &curr, &objs)) {
  909. ret = dec_error(d, "internal_error");
  910. goto done;
  911. }
  912. if(dec_top(d) > 0) {
  913. dec_push(d, st_comma);
  914. curr = enif_make_list_cell(env, val, curr);
  915. } else {
  916. dec_push(d, st_done);
  917. }
  918. d->i++;
  919. break;
  920. case ']':
  921. dec_pop(d, st_comma);
  922. if(dec_curr(d) != st_array) {
  923. ret = dec_error(d, "invalid_json");
  924. goto done;
  925. }
  926. dec_pop(d, st_array);
  927. dec_pop(d, st_value);
  928. val = make_array(env, curr);
  929. if(!enif_get_list_cell(env, objs, &curr, &objs)) {
  930. ret = dec_error(d, "internal_error");
  931. goto done;
  932. }
  933. if(dec_top(d) > 0) {
  934. dec_push(d, st_comma);
  935. curr = enif_make_list_cell(env, val, curr);
  936. } else {
  937. dec_push(d, st_done);
  938. }
  939. d->i++;
  940. break;
  941. default:
  942. ret = dec_error(d, "invalid_json");
  943. goto done;
  944. }
  945. break;
  946. case st_done:
  947. switch(d->p[d->i]) {
  948. case ' ':
  949. case '\n':
  950. case '\r':
  951. case '\t':
  952. d->i++;
  953. break;
  954. default:
  955. ret = dec_error(d, "invalid_trailing_data");
  956. goto done;
  957. }
  958. break;
  959. default:
  960. ret = dec_error(d, "invalid_internal_state");
  961. goto done;
  962. }
  963. }
  964. if(dec_curr(d) != st_done) {
  965. ret = dec_error(d, "truncated_json");
  966. } else if(d->is_partial) {
  967. ret = enif_make_tuple2(env, d->atoms->atom_partial, val);
  968. } else {
  969. ret = val;
  970. }
  971. done:
  972. consume_timeslice(env, d->i - start, d->bytes_per_iter);
  973. return ret;
  974. }