jerry_buffer.c 26 KB

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