jerry_buffer.c 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958
  1. /*
  2. * Licensed under the Apache License, Version 2.0 (the "License")
  3. * Copyright (c) 2016-2018, Intel Corporation.
  4. *
  5. * ported from zephyr.js
  6. * COPYRIGHT (C) 2018, RT-Thread Development Team
  7. */
  8. #include <string.h>
  9. #include <stdlib.h>
  10. #include <stdio.h>
  11. #include <jerry_util.h>
  12. #include "jerry_buffer.h"
  13. #include "jerry_utf8.h"
  14. #define strequal(a, b) !strcmp(a, b)
  15. #define DECL_FUNC_ARGS(name, ...) \
  16. jerry_value_t name(const jerry_value_t function_obj, \
  17. const jerry_value_t this, const jerry_value_t argv[], \
  18. const jerry_length_t args_cnt, __VA_ARGS__)
  19. static jerry_value_t jerry_buffer_prototype;
  20. js_buffer_t *jerry_buffer_find(const jerry_value_t obj);
  21. static void jerry_buffer_callback_free(void *handle)
  22. {
  23. js_buffer_t *item = (js_buffer_t *)handle;
  24. free(item->buffer);
  25. free(item);
  26. }
  27. static const jerry_object_native_info_t buffer_type_info =
  28. {
  29. .free_cb = jerry_buffer_callback_free
  30. };
  31. bool jerry_value_is_buffer(const jerry_value_t value)
  32. {
  33. if (jerry_value_is_object(value) && jerry_buffer_find(value))
  34. {
  35. return true;
  36. }
  37. return false;
  38. }
  39. js_buffer_t *jerry_buffer_find(const jerry_value_t obj)
  40. {
  41. // requires: obj should be the JS object associated with a buffer, created
  42. // in jerry_buffer
  43. // effects: looks up obj in our list of known buffer objects and returns
  44. // the associated list item struct, or NULL if not found
  45. js_buffer_t *handle;
  46. const jerry_object_native_info_t *tmp;
  47. if (jerry_get_object_native_pointer(obj, (void **)&handle, &tmp))
  48. {
  49. if (tmp == &buffer_type_info)
  50. {
  51. return handle;
  52. }
  53. }
  54. return NULL;
  55. }
  56. static DECL_FUNC_ARGS(jerry_buffer_read_bytes, int bytes, bool big_endian, bool sign)
  57. {
  58. // requires: this is a JS buffer object created with jerry_buffer_create,
  59. // argv[0] should be an offset into the buffer, but will treat
  60. // offset as 0 if not given, as node.js seems to
  61. // bytes is the number of bytes to read (expects 1, 2, or 4)
  62. // big_endian true reads the bytes in big endian order, false in
  63. // little endian order
  64. // effects: reads bytes from the buffer associated with this JS object, if
  65. // found, at the given offset, if within the bounds of the
  66. // buffer; otherwise returns an error
  67. // args: offset
  68. // ZJS_VALIDATE_ARGS(Z_OPTIONAL Z_NUMBER);
  69. uint32_t offset = 0;
  70. if (args_cnt >= 1)
  71. offset = (uint32_t)jerry_get_number_value(argv[0]);
  72. js_buffer_t *buf = jerry_buffer_find(this);
  73. if (!buf)
  74. return jerry_create_undefined();
  75. if (offset + bytes > buf->bufsize)
  76. return jerry_create_undefined();
  77. int dir = big_endian ? 1 : -1;
  78. if (!big_endian)
  79. offset += bytes - 1; // start on the big end
  80. uint32_t value = 0, i;
  81. for (i = 0; i < bytes; i++)
  82. {
  83. value <<= 8;
  84. value |= buf->buffer[offset];
  85. offset += dir;
  86. }
  87. if (sign)
  88. {
  89. if (bytes == 1)
  90. return jerry_create_number((int8_t)value);
  91. else if (bytes == 2)
  92. return jerry_create_number((int16_t)value);
  93. else if (bytes == 4)
  94. return jerry_create_number((int32_t)value);
  95. }
  96. return jerry_create_number(value);
  97. }
  98. static DECL_FUNC_ARGS(jerry_buffer_write_bytes, int bytes, bool big_endian)
  99. {
  100. // requires: this is a JS buffer object created with jerry_buffer_create,
  101. // argv[0] must be the value to be written, argv[1] should be
  102. // an offset into the buffer, but will treat offset as 0 if not
  103. // given, as node.js seems to
  104. // bytes is the number of bytes to write (expects 1, 2, or 4)
  105. // big_endian true writes the bytes in big endian order, false in
  106. // little endian order
  107. // effects: writes bytes into the buffer associated with this JS object, if
  108. // found, at the given offset, if within the bounds of the
  109. // buffer and returns the offset just beyond what was written;
  110. // otherwise returns an error
  111. // args: value[, offset]
  112. // ZJS_VALIDATE_ARGS(Z_NUMBER, Z_OPTIONAL Z_NUMBER);
  113. // technically negatives aren't supported but this makes them behave better
  114. double dval = jerry_get_number_value(argv[0]);
  115. uint32_t value = (uint32_t)(dval < 0 ? (int32_t)dval : dval);
  116. uint32_t offset = 0;
  117. if (args_cnt > 1)
  118. {
  119. offset = (uint32_t)jerry_get_number_value(argv[1]);
  120. }
  121. js_buffer_t *buf = jerry_buffer_find(this);
  122. if (!buf)
  123. {
  124. return jerry_create_undefined();
  125. }
  126. uint32_t beyond = offset + bytes;
  127. if (beyond > buf->bufsize)
  128. {
  129. printf("bufsize %d, write attempted from %d to %d\n", (int)buf->bufsize, (int)offset, (int)beyond);
  130. return jerry_create_undefined();
  131. }
  132. int dir = big_endian ? -1 : 1;
  133. if (big_endian)
  134. offset += bytes - 1; // start on the little end
  135. int i;
  136. for (i = 0; i < bytes; i++)
  137. {
  138. buf->buffer[offset] = value & 0xff;
  139. value >>= 8;
  140. offset += dir;
  141. }
  142. return jerry_create_number(beyond);
  143. }
  144. DECLARE_HANDLER(readUInt8)
  145. {
  146. return jerry_buffer_read_bytes(func_value, this_value, args, args_cnt, 1, true, false);
  147. }
  148. DECLARE_HANDLER(readInt8)
  149. {
  150. return jerry_buffer_read_bytes(func_value, this_value, args, args_cnt, 1, true, true);
  151. }
  152. DECLARE_HANDLER(readUInt16BE)
  153. {
  154. return jerry_buffer_read_bytes(func_value, this_value, args, args_cnt, 2, true, false);
  155. }
  156. DECLARE_HANDLER(readInt16BE)
  157. {
  158. return jerry_buffer_read_bytes(func_value, this_value, args, args_cnt, 2, true, true);
  159. }
  160. DECLARE_HANDLER(readUInt16LE)
  161. {
  162. return jerry_buffer_read_bytes(func_value, this_value, args, args_cnt, 2, false, false);
  163. }
  164. DECLARE_HANDLER(readInt16LE)
  165. {
  166. return jerry_buffer_read_bytes(func_value, this_value, args, args_cnt, 2, false, true);
  167. }
  168. DECLARE_HANDLER(readUInt32BE)
  169. {
  170. return jerry_buffer_read_bytes(func_value, this_value, args, args_cnt, 4, true, false);
  171. }
  172. DECLARE_HANDLER(readInt32BE)
  173. {
  174. return jerry_buffer_read_bytes(func_value, this_value, args, args_cnt, 4, true, true);
  175. }
  176. DECLARE_HANDLER(readUInt32LE)
  177. {
  178. return jerry_buffer_read_bytes(func_value, this_value, args, args_cnt, 4, false, false);
  179. }
  180. DECLARE_HANDLER(readInt32LE)
  181. {
  182. return jerry_buffer_read_bytes(func_value, this_value, args, args_cnt, 4, false, true);
  183. }
  184. DECLARE_HANDLER(writeUInt8)
  185. {
  186. return jerry_buffer_write_bytes(func_value, this_value, args, args_cnt, 1, true);
  187. }
  188. DECLARE_HANDLER(writeUInt16BE)
  189. {
  190. return jerry_buffer_write_bytes(func_value, this_value, args, args_cnt, 2, true);
  191. }
  192. DECLARE_HANDLER(writeUInt16LE)
  193. {
  194. return jerry_buffer_write_bytes(func_value, this_value, args, args_cnt, 2, false);
  195. }
  196. DECLARE_HANDLER(writeUInt32BE)
  197. {
  198. return jerry_buffer_write_bytes(func_value, this_value, args, args_cnt, 4, true);
  199. }
  200. DECLARE_HANDLER(writeUInt32LE)
  201. {
  202. return jerry_buffer_write_bytes(func_value, this_value, args, args_cnt, 4, false);
  203. }
  204. DECLARE_HANDLER(readFloatBE)
  205. {
  206. return jerry_create_undefined();
  207. }
  208. DECLARE_HANDLER(readDoubleBE)
  209. {
  210. return jerry_create_undefined();
  211. }
  212. char jerry_int_to_hex(int value)
  213. {
  214. if (value < 10)
  215. return '0' + value;
  216. return 'A' + value - 10;
  217. }
  218. static int hex2int(const char *str)
  219. {
  220. int i = 0;
  221. int r = 0;
  222. for (i = 0; i < 2; i++)
  223. {
  224. if (str[i] >= '0' && str[i] <= '9') r += str[i] - '0';
  225. else if (str[i] >= 'a' && str[i] <= 'f') r += str[i] - 'a' + 10;
  226. else if (str[i] >= 'A' && str[i] <= 'F') r += str[i] - 'A' + 10;
  227. if (!i) r *= 16;
  228. }
  229. return r;
  230. }
  231. /*
  232. * toString(encoding[opt], start[opt], end[opt])
  233. */
  234. DECLARE_HANDLER(toString)
  235. {
  236. int start, end;
  237. int optcount = args_cnt;
  238. if (args_cnt > 1 && !jerry_value_is_string(args[0])) return jerry_create_undefined();
  239. js_buffer_t *buf = jerry_buffer_find(this_value);
  240. if (!buf) return jerry_create_undefined();
  241. if (args_cnt > 1) start = jerry_get_number_value(args[1]);
  242. else start = 0;
  243. if (args_cnt > 2) end = jerry_get_number_value(args[2]);
  244. else end = buf->bufsize;
  245. int size;
  246. char *enc = NULL;
  247. const char *encoding = "utf8";
  248. if (optcount)
  249. {
  250. enc = js_value_to_string(args[0]);
  251. size = strlen(enc);
  252. if (!size)
  253. {
  254. free(enc);
  255. return jerry_create_undefined();
  256. }
  257. encoding = enc;
  258. }
  259. if (strequal(encoding, "utf8"))
  260. {
  261. jerry_char_t *str = (jerry_char_t *)buf->buffer;
  262. str += start;
  263. rt_free(enc);
  264. if (jerry_str_is_utf8(str, end - start))
  265. return jerry_create_string_sz_from_utf8(str, end - start);
  266. else
  267. return jerry_create_undefined();
  268. }
  269. else if (strequal(encoding, "ascii"))
  270. {
  271. char *str = malloc(buf->bufsize);
  272. if (!str)
  273. {
  274. free(enc);
  275. return jerry_create_undefined();
  276. }
  277. int index;
  278. for (index = start; index < end; ++index)
  279. {
  280. // strip off high bit if present
  281. str[index - start] = buf->buffer[index] & 0x7f;
  282. if (!str[index - start])
  283. break;
  284. }
  285. jerry_value_t jstr;
  286. if (jerry_str_is_utf8(str, index - start))
  287. jstr = jerry_create_string_sz_from_utf8((jerry_char_t *)str, index - start);
  288. else
  289. jstr = jerry_create_undefined();
  290. free(str);
  291. free(enc);
  292. return jstr;
  293. }
  294. else if (strequal(encoding, "hex"))
  295. {
  296. if (buf && end - start > 0)
  297. {
  298. int size = (end - start);
  299. char *hexbuf = malloc(size * 2 + 1);
  300. if (hexbuf)
  301. {
  302. int i;
  303. jerry_value_t ret;
  304. for (i = 0; i < size; i++)
  305. {
  306. int high = (0xf0 & buf->buffer[i + start]) >> 4;
  307. int low = 0xf & buf->buffer[i + start];
  308. hexbuf[2 * i] = jerry_int_to_hex(high);
  309. hexbuf[2 * i + 1] = jerry_int_to_hex(low);
  310. }
  311. hexbuf[size * 2] = '\0';
  312. free(enc);
  313. ret = jerry_create_string((jerry_char_t *)hexbuf);
  314. free(hexbuf);
  315. return ret;
  316. }
  317. else
  318. {
  319. free(enc);
  320. return jerry_create_undefined();
  321. }
  322. }
  323. }
  324. else
  325. {
  326. free(enc);
  327. return jerry_create_undefined();
  328. }
  329. free(enc);
  330. return jerry_create_undefined();
  331. }
  332. DECLARE_HANDLER(jsonParse)
  333. {
  334. int start, end;
  335. js_buffer_t *buf = jerry_buffer_find(this_value);
  336. if (!buf) return jerry_create_undefined();
  337. if (args_cnt > 0)
  338. start = jerry_get_number_value(args[0]);
  339. else
  340. start = 0;
  341. if (args_cnt > 1)
  342. end = jerry_get_number_value(args[1]);
  343. else
  344. end = buf->bufsize;
  345. if (end - start > 0)
  346. {
  347. jerry_value_t parsed_json;
  348. jerry_char_t *str = (jerry_char_t *)buf->buffer;
  349. str += start;
  350. parsed_json = jerry_json_parse(str, end - start);
  351. if (jerry_value_is_error(parsed_json))
  352. {
  353. jerry_release_value(parsed_json);
  354. return jerry_create_undefined();
  355. }
  356. else
  357. return parsed_json;
  358. }
  359. return jerry_create_undefined();
  360. }
  361. DECLARE_HANDLER(concat)
  362. {
  363. if (args_cnt == 1)
  364. {
  365. uint8_t *ptr;
  366. js_buffer_t *source = jerry_buffer_find(this_value);
  367. js_buffer_t *target = jerry_buffer_find(args[0]);
  368. if (!source || !target)
  369. {
  370. return jerry_create_undefined();
  371. }
  372. if (target->bufsize == 0 || source->bufsize == 0) return this_value;
  373. ptr = realloc(source->buffer, source->bufsize + target->bufsize);
  374. if (ptr)
  375. {
  376. source->buffer = ptr;
  377. memcpy(&source->buffer[source->bufsize], target->buffer, target->bufsize);
  378. source->bufsize += target->bufsize;
  379. js_set_property(this_value, "length", jerry_create_number(source->bufsize));
  380. }
  381. }
  382. return jerry_create_undefined();
  383. }
  384. DECLARE_HANDLER(copy)
  385. {
  386. // requires: this must be a JS buffer object and args[0] a target buffer
  387. // object
  388. // effects: copies buffer contents w/ optional offsets considered to the
  389. // target buffer (following Node v8.9.4 API)
  390. // args: target, [targetStart], [sourceStart], [sourceEnd]
  391. // ZJS_VALIDATE_ARGS_OPTCOUNT(optcount, Z_BUFFER,
  392. // Z_OPTIONAL Z_NUMBER Z_UNDEFINED,
  393. // Z_OPTIONAL Z_NUMBER Z_UNDEFINED,
  394. // Z_OPTIONAL Z_NUMBER Z_UNDEFINED);
  395. int optcount = args_cnt;
  396. if (optcount <= 0)
  397. return jerry_create_undefined();
  398. js_buffer_t *source = jerry_buffer_find(this_value);
  399. js_buffer_t *target = jerry_buffer_find(args[0]);
  400. if (!source || !target)
  401. {
  402. return jerry_create_undefined();
  403. }
  404. int targetStart = 0;
  405. int sourceStart = 0;
  406. int sourceEnd = -1;
  407. if (optcount > 1 && !jerry_value_is_undefined(args[1]))
  408. {
  409. targetStart = (int)jerry_get_number_value(args[1]);
  410. }
  411. if (optcount > 2 && !jerry_value_is_undefined(args[2]))
  412. {
  413. sourceStart = (int)jerry_get_number_value(args[2]);
  414. }
  415. if (optcount > 3 && !jerry_value_is_undefined(args[3]))
  416. {
  417. sourceEnd = (int)jerry_get_number_value(args[3]);
  418. }
  419. if (sourceEnd == -1)
  420. {
  421. sourceEnd = source->bufsize;
  422. }
  423. if (targetStart < 0 || targetStart >= target->bufsize ||
  424. sourceStart < 0 || sourceStart >= source->bufsize ||
  425. sourceEnd < 0 || sourceEnd <= sourceStart ||
  426. sourceEnd > source->bufsize)
  427. {
  428. return jerry_create_undefined();
  429. }
  430. if (sourceEnd - sourceStart > target->bufsize - targetStart)
  431. {
  432. return jerry_create_undefined();
  433. }
  434. int len = sourceEnd - sourceStart;
  435. memcpy(target->buffer + targetStart, source->buffer + sourceStart, len);
  436. return jerry_create_number(len);
  437. }
  438. DECLARE_HANDLER(write)
  439. {
  440. // requires: string - what will be written to buf
  441. // offset - where to start writing (Default: 0)
  442. // length - how many bytes to write (Default: buf.length -offset)
  443. // encoding - the character encoding of string. Currently only
  444. // supports the default of utf8
  445. // effects: writes string to buf at offset according to the character
  446. // encoding in encoding.
  447. // args: data[, offset[, length[, encoding]]]
  448. // ZJS_VALIDATE_ARGS(Z_STRING, Z_OPTIONAL Z_NUMBER, Z_OPTIONAL Z_NUMBER,
  449. // Z_OPTIONAL Z_STRING);
  450. js_buffer_t *buf = jerry_buffer_find(this_value);
  451. if (!buf)
  452. {
  453. return jerry_create_undefined();
  454. }
  455. // Check if the encoding string is anything other than utf8
  456. if (args_cnt > 3)
  457. {
  458. char *encoding = js_value_to_string(args[3]);
  459. if (!encoding)
  460. {
  461. return jerry_create_undefined();
  462. }
  463. // ask for one more char than needed to make sure not just prefix match
  464. const char *utf8_encoding = "utf8";
  465. int utf8_len = strlen(utf8_encoding);
  466. int rval = strncmp(encoding, utf8_encoding, utf8_len + 1);
  467. free(encoding);
  468. if (rval != 0)
  469. {
  470. return jerry_create_undefined();
  471. }
  472. }
  473. char *str = js_value_to_string(args[0]);
  474. if (!str)
  475. {
  476. return jerry_create_undefined();
  477. }
  478. jerry_size_t size = strlen(str);
  479. uint32_t offset = 0;
  480. if (args_cnt > 1)
  481. offset = (uint32_t)jerry_get_number_value(args[1]);
  482. uint32_t length = buf->bufsize - offset;
  483. if (args_cnt > 2)
  484. length = (uint32_t)jerry_get_number_value(args[2]);
  485. if (length > size)
  486. {
  487. free(str);
  488. return jerry_create_undefined();
  489. }
  490. if (offset + length > buf->bufsize)
  491. {
  492. free(str);
  493. return jerry_create_undefined();
  494. }
  495. memcpy(buf->buffer + offset, str, length);
  496. free(str);
  497. return jerry_create_number(length);
  498. }
  499. DECLARE_HANDLER(fill)
  500. {
  501. // requires: value - what will be written to buf
  502. // offset - where to start writing (Default: 0)
  503. // end - offset at which to stop writing (Default: buf.length)
  504. // encoding - the character encoding of value. Currently only
  505. // supports the default of utf8
  506. // effects: writes string to buf at offset according to the character
  507. // encoding in encoding.
  508. // args: data[, offset[, length[, encoding]]]
  509. // ZJS_VALIDATE_ARGS(Z_STRING Z_NUMBER Z_BUFFER, Z_OPTIONAL Z_NUMBER,
  510. // Z_OPTIONAL Z_NUMBER, Z_OPTIONAL Z_STRING);
  511. // TODO: support encodings other than 'utf8'
  512. uint32_t num;
  513. char *source = NULL;
  514. char *str = NULL;
  515. uint32_t srclen = 0;
  516. if (jerry_value_is_number(args[0]))
  517. {
  518. uint32_t srcnum = (uint32_t)jerry_get_number_value(args[0]);
  519. // convert in case of endian difference
  520. source = (char *)&num;
  521. source[0] = (0xff000000 & srcnum) >> 24;
  522. source[1] = (0x00ff0000 & srcnum) >> 16;
  523. source[2] = (0x0000ff00 & srcnum) >> 8;
  524. source[3] = 0x000000ff & srcnum;
  525. srclen = sizeof(uint32_t);
  526. }
  527. else if (jerry_value_is_buffer(args[0]))
  528. {
  529. js_buffer_t *srcbuf = jerry_buffer_find(args[0]);
  530. source = (char *)srcbuf->buffer;
  531. srclen = srcbuf->bufsize;
  532. }
  533. js_buffer_t *buf = jerry_buffer_find(this_value);
  534. if (!buf)
  535. {
  536. return jerry_create_undefined();
  537. }
  538. // Check if the encoding string is anything other than utf8
  539. if (args_cnt > 3)
  540. {
  541. char *encoding = js_value_to_string(args[3]);
  542. if (!encoding)
  543. {
  544. return jerry_create_undefined();
  545. }
  546. // ask for one more char than needed to make sure not just prefix match
  547. const char *utf8_encoding = "utf8";
  548. int utf8_len = strlen(utf8_encoding);
  549. int rval = strncmp(encoding, utf8_encoding, utf8_len + 1);
  550. free(encoding);
  551. if (rval != 0)
  552. {
  553. return jerry_create_undefined();
  554. }
  555. }
  556. uint32_t offset = 0;
  557. if (args_cnt > 1)
  558. offset = (uint32_t)jerry_get_number_value(args[1]);
  559. uint32_t end = buf->bufsize;
  560. if (args_cnt > 2)
  561. {
  562. end = (uint32_t)jerry_get_number_value(args[2]);
  563. if (end > buf->bufsize)
  564. {
  565. end = buf->bufsize;
  566. }
  567. }
  568. if (offset >= end)
  569. {
  570. // nothing to do
  571. return jerry_acquire_value(this_value);
  572. }
  573. if (jerry_value_is_string(args[0]))
  574. {
  575. char *str = js_value_to_string(args[0]);
  576. if (!str)
  577. {
  578. return jerry_create_undefined();
  579. }
  580. source = str;
  581. srclen = strlen(str);
  582. }
  583. uint32_t bytes_left = end - offset;
  584. while (bytes_left > 0)
  585. {
  586. uint32_t bytes = srclen;
  587. if (bytes > bytes_left)
  588. {
  589. bytes = bytes_left;
  590. }
  591. memcpy(buf->buffer + offset, source, bytes);
  592. offset += bytes;
  593. bytes_left -= bytes;
  594. }
  595. free(str);
  596. return jerry_acquire_value(this_value);
  597. }
  598. jerry_value_t jerry_buffer_create(uint32_t size, js_buffer_t **ret_buf)
  599. {
  600. // follow Node's Buffer.kMaxLength limits though we don't expose that
  601. uint32_t maxLength = (1UL << 31) - 1;
  602. if (sizeof(size_t) == 4)
  603. {
  604. // detected 32-bit architecture
  605. maxLength = (1 << 30) - 1;
  606. }
  607. if (size > maxLength)
  608. {
  609. printf("size: %d\n", (int)size);
  610. return jerry_create_undefined();
  611. }
  612. void *buf = malloc(size);
  613. js_buffer_t *buf_item = (js_buffer_t *)malloc(sizeof(js_buffer_t));
  614. if (!buf || !buf_item)
  615. {
  616. free(buf);
  617. free(buf_item);
  618. if (ret_buf)
  619. {
  620. *ret_buf = NULL;
  621. }
  622. return jerry_create_undefined();
  623. }
  624. jerry_value_t buf_obj = jerry_create_object();
  625. buf_item->buffer = buf;
  626. buf_item->bufsize = size;
  627. jerry_set_prototype(buf_obj, jerry_buffer_prototype);
  628. jerry_value_t length = jerry_create_number(size);
  629. js_set_property(buf_obj, "length", length);
  630. jerry_release_value(length);
  631. // watch for the object getting garbage collected, and clean up
  632. jerry_set_object_native_pointer(buf_obj, buf_item, &buffer_type_info);
  633. if (ret_buf)
  634. {
  635. *ret_buf = buf_item;
  636. }
  637. return buf_obj;
  638. }
  639. #define ENCODING_UTF8 0
  640. #define ENCODING_ASCII 1
  641. #define ENCODING_HEX 2
  642. #define ENCODING_BASE64 3
  643. int buffer_encoding_type(const char* encoding)
  644. {
  645. int ret = ENCODING_UTF8;
  646. if (strequal(encoding, "hex")) ret = ENCODING_HEX;
  647. else if (strequal(encoding, "ascii")) ret = ENCODING_ASCII;
  648. else if (strequal(encoding, "utf8")) ret = ENCODING_UTF8;
  649. else if (strequal(encoding, "base64")) ret = ENCODING_BASE64;
  650. return ret;
  651. }
  652. /*
  653. * Buffer(number);
  654. * Buffer(string, encoding[opt]);
  655. * Buffer(array);
  656. * Buffer(object);
  657. */
  658. DECLARE_HANDLER(Buffer)
  659. {
  660. int encoding_type = ENCODING_UTF8;
  661. if (args_cnt > 1)
  662. {
  663. char *encoding = js_value_to_string(args[1]);
  664. if (encoding)
  665. {
  666. encoding_type = buffer_encoding_type(encoding);
  667. free(encoding);
  668. }
  669. }
  670. if (jerry_value_is_number(args[0]))
  671. {
  672. double dnum = jerry_get_number_value(args[0]);
  673. uint32_t unum;
  674. if (dnum < 0)
  675. {
  676. unum = 0;
  677. }
  678. else if (dnum > 0xffffffff)
  679. {
  680. unum = 0xffffffff;
  681. }
  682. else
  683. {
  684. // round to the nearest integer
  685. unum = (uint32_t)(dnum + 0.5);
  686. }
  687. // treat a number argument as a length
  688. return jerry_buffer_create(unum, NULL);
  689. }
  690. else if (jerry_value_is_string(args[0]))
  691. {
  692. uint8_t *data = NULL;
  693. char *str = js_value_to_string(args[0]);
  694. if (!str)
  695. {
  696. return jerry_create_undefined();
  697. }
  698. js_buffer_t *buf;
  699. jerry_size_t size = strlen(str);
  700. if (encoding_type == ENCODING_HEX)
  701. {
  702. data = malloc(size / 2);
  703. if (data)
  704. {
  705. int index;
  706. for (index = 0; index < size / 2; index++)
  707. {
  708. data[index] = hex2int(&str[index * 2]);
  709. }
  710. }
  711. size = size / 2;
  712. }
  713. else
  714. {
  715. data = (uint8_t *)str;
  716. }
  717. jerry_value_t new_buf = jerry_buffer_create(size, &buf);
  718. if (buf)
  719. {
  720. memcpy(buf->buffer, data, size);
  721. }
  722. if ((char*)data != str) free(data);
  723. free(str);
  724. return new_buf;
  725. }
  726. else if (jerry_value_is_array(args[0]))
  727. {
  728. // treat array argument as byte initializers
  729. jerry_value_t array = args[0];
  730. uint32_t len = jerry_get_array_length(array);
  731. js_buffer_t *buf;
  732. jerry_value_t new_buf = jerry_buffer_create(len, &buf);
  733. if (buf)
  734. {
  735. int i;
  736. for (i = 0; i < len; i++)
  737. {
  738. jerry_value_t item = jerry_get_property_by_index(array, i);
  739. if (jerry_value_is_number(item))
  740. {
  741. buf->buffer[i] = (uint8_t)jerry_get_number_value(item);
  742. }
  743. else
  744. {
  745. printf("non-numeric value in array, treating as 0\n");
  746. buf->buffer[i] = 0;
  747. }
  748. }
  749. }
  750. return new_buf;
  751. }
  752. else if (jerry_value_is_object(args[0]))
  753. {
  754. jerry_value_t stringified = jerry_json_stringfy(args[0]);
  755. if (!jerry_value_is_error(stringified))
  756. {
  757. char *json_string = js_value_to_string(stringified);
  758. if (json_string)
  759. {
  760. js_buffer_t *buf;
  761. jerry_value_t new_buf = jerry_buffer_create(strlen(json_string), &buf);
  762. if (buf)
  763. {
  764. memcpy(buf->buffer, json_string, strlen(json_string));
  765. }
  766. free(json_string);
  767. jerry_release_value(stringified);
  768. return new_buf;
  769. }
  770. }
  771. jerry_release_value(stringified);
  772. }
  773. return jerry_create_undefined();
  774. }
  775. int js_buffer_cleanup(void)
  776. {
  777. jerry_release_value(jerry_buffer_prototype);
  778. return 0;
  779. }
  780. int js_buffer_init(void)
  781. {
  782. REGISTER_HANDLER(Buffer);
  783. jerry_buffer_prototype = jerry_create_object();
  784. REGISTER_METHOD(jerry_buffer_prototype, readUInt8);
  785. REGISTER_METHOD(jerry_buffer_prototype, readInt8);
  786. REGISTER_METHOD(jerry_buffer_prototype, writeUInt8);
  787. REGISTER_METHOD(jerry_buffer_prototype, readUInt16BE);
  788. REGISTER_METHOD(jerry_buffer_prototype, readInt16BE);
  789. REGISTER_METHOD(jerry_buffer_prototype, writeUInt16BE);
  790. REGISTER_METHOD(jerry_buffer_prototype, readUInt16LE);
  791. REGISTER_METHOD(jerry_buffer_prototype, readInt16LE);
  792. REGISTER_METHOD(jerry_buffer_prototype, writeUInt16LE);
  793. REGISTER_METHOD(jerry_buffer_prototype, readUInt32BE);
  794. REGISTER_METHOD(jerry_buffer_prototype, readInt32BE);
  795. REGISTER_METHOD(jerry_buffer_prototype, writeUInt32BE);
  796. REGISTER_METHOD(jerry_buffer_prototype, readUInt32LE);
  797. REGISTER_METHOD(jerry_buffer_prototype, readInt32LE);
  798. REGISTER_METHOD(jerry_buffer_prototype, writeUInt32LE);
  799. REGISTER_METHOD(jerry_buffer_prototype, readFloatBE);
  800. REGISTER_METHOD(jerry_buffer_prototype, readDoubleBE);
  801. REGISTER_METHOD(jerry_buffer_prototype, copy);
  802. REGISTER_METHOD(jerry_buffer_prototype, fill);
  803. REGISTER_METHOD(jerry_buffer_prototype, toString);
  804. REGISTER_METHOD(jerry_buffer_prototype, jsonParse);
  805. REGISTER_METHOD(jerry_buffer_prototype, write);
  806. REGISTER_METHOD(jerry_buffer_prototype, concat);
  807. return 0;
  808. }