jerry_buffer.c 27 KB

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