libc_builtin_wrapper.c 32 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173
  1. /*
  2. * Copyright (C) 2019 Intel Corporation. All rights reserved.
  3. * SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
  4. */
  5. #include "bh_common.h"
  6. #include "bh_log.h"
  7. #include "wasm_export.h"
  8. #include "../interpreter/wasm.h"
  9. #if defined(_WIN32) || defined(_WIN32_)
  10. #define strncasecmp _strnicmp
  11. #define strcasecmp _stricmp
  12. #endif
  13. void
  14. wasm_runtime_set_exception(wasm_module_inst_t module, const char *exception);
  15. uint32
  16. wasm_runtime_module_realloc(wasm_module_inst_t module, uint64 ptr, uint64 size,
  17. void **p_native_addr);
  18. /* clang-format off */
  19. #define get_module_inst(exec_env) \
  20. wasm_runtime_get_module_inst(exec_env)
  21. #define validate_app_addr(offset, size) \
  22. wasm_runtime_validate_app_addr(module_inst, offset, size)
  23. #define validate_app_str_addr(offset) \
  24. wasm_runtime_validate_app_str_addr(module_inst, offset)
  25. #define validate_native_addr(addr, size) \
  26. wasm_runtime_validate_native_addr(module_inst, addr, size)
  27. #define addr_app_to_native(offset) \
  28. wasm_runtime_addr_app_to_native(module_inst, offset)
  29. #define addr_native_to_app(ptr) \
  30. wasm_runtime_addr_native_to_app(module_inst, ptr)
  31. #define module_malloc(size, p_native_addr) \
  32. wasm_runtime_module_malloc(module_inst, size, p_native_addr)
  33. #define module_free(offset) \
  34. wasm_runtime_module_free(module_inst, offset)
  35. /* clang-format on */
  36. typedef int (*out_func_t)(int c, void *ctx);
  37. typedef char *_va_list;
  38. #define _INTSIZEOF(n) (((uint32)sizeof(n) + 3) & (uint32)~3)
  39. #define _va_arg(ap, t) (*(t *)((ap += _INTSIZEOF(t)) - _INTSIZEOF(t)))
  40. #define CHECK_VA_ARG(ap, t) \
  41. do { \
  42. if ((uint8 *)ap + _INTSIZEOF(t) > native_end_addr) { \
  43. if (fmt_buf != temp_fmt) { \
  44. wasm_runtime_free(fmt_buf); \
  45. } \
  46. goto fail; \
  47. } \
  48. } while (0)
  49. /* clang-format off */
  50. #define PREPARE_TEMP_FORMAT() \
  51. char temp_fmt[32], *s, *fmt_buf = temp_fmt; \
  52. uint32 fmt_buf_len = (uint32)sizeof(temp_fmt); \
  53. int32 n; \
  54. \
  55. /* additional 2 bytes: one is the format char, \
  56. the other is `\0` */ \
  57. if ((uint32)(fmt - fmt_start_addr + 2) >= fmt_buf_len) { \
  58. bh_assert((uint32)(fmt - fmt_start_addr) <= \
  59. UINT32_MAX - 2); \
  60. fmt_buf_len = (uint32)(fmt - fmt_start_addr + 2); \
  61. if (!(fmt_buf = wasm_runtime_malloc(fmt_buf_len))) { \
  62. print_err(out, ctx); \
  63. break; \
  64. } \
  65. } \
  66. \
  67. memset(fmt_buf, 0, fmt_buf_len); \
  68. bh_memcpy_s(fmt_buf, fmt_buf_len, fmt_start_addr, \
  69. (uint32)(fmt - fmt_start_addr + 1));
  70. /* clang-format on */
  71. #define OUTPUT_TEMP_FORMAT() \
  72. do { \
  73. if (n > 0) { \
  74. s = buf; \
  75. while (*s) \
  76. out((int)(*s++), ctx); \
  77. } \
  78. \
  79. if (fmt_buf != temp_fmt) { \
  80. wasm_runtime_free(fmt_buf); \
  81. } \
  82. } while (0)
  83. static void
  84. print_err(out_func_t out, void *ctx)
  85. {
  86. out('E', ctx);
  87. out('R', ctx);
  88. out('R', ctx);
  89. }
  90. static bool
  91. _vprintf_wa(out_func_t out, void *ctx, const char *fmt, _va_list ap,
  92. wasm_module_inst_t module_inst)
  93. {
  94. int might_format = 0; /* 1 if encountered a '%' */
  95. int long_ctr = 0;
  96. uint8 *native_end_addr;
  97. const char *fmt_start_addr = NULL;
  98. if (!wasm_runtime_get_native_addr_range(module_inst, (uint8 *)ap, NULL,
  99. &native_end_addr))
  100. goto fail;
  101. /* fmt has already been adjusted if needed */
  102. while (*fmt) {
  103. if (!might_format) {
  104. if (*fmt != '%') {
  105. out((int)*fmt, ctx);
  106. }
  107. else {
  108. might_format = 1;
  109. long_ctr = 0;
  110. fmt_start_addr = fmt;
  111. }
  112. }
  113. else {
  114. switch (*fmt) {
  115. case '.':
  116. case '+':
  117. case '-':
  118. case ' ':
  119. case '#':
  120. case '0':
  121. case '1':
  122. case '2':
  123. case '3':
  124. case '4':
  125. case '5':
  126. case '6':
  127. case '7':
  128. case '8':
  129. case '9':
  130. goto still_might_format;
  131. case 't': /* ptrdiff_t */
  132. case 'z': /* size_t (32bit on wasm) */
  133. long_ctr = 1;
  134. goto still_might_format;
  135. case 'j':
  136. /* intmax_t/uintmax_t */
  137. long_ctr = 2;
  138. goto still_might_format;
  139. case 'l':
  140. long_ctr++;
  141. /* Fall through */
  142. case 'h':
  143. /* FIXME: do nothing for these modifiers */
  144. goto still_might_format;
  145. case 'o':
  146. case 'd':
  147. case 'i':
  148. case 'u':
  149. case 'p':
  150. case 'x':
  151. case 'X':
  152. case 'c':
  153. {
  154. char buf[64];
  155. PREPARE_TEMP_FORMAT();
  156. if (long_ctr < 2) {
  157. int32 d;
  158. CHECK_VA_ARG(ap, uint32);
  159. d = _va_arg(ap, int32);
  160. if (long_ctr == 1) {
  161. uint32 fmt_end_idx = (uint32)(fmt - fmt_start_addr);
  162. if (fmt_buf[fmt_end_idx - 1] == 'l'
  163. || fmt_buf[fmt_end_idx - 1] == 'z'
  164. || fmt_buf[fmt_end_idx - 1] == 't') {
  165. /* The %ld, %zd and %td should be treated as
  166. * 32bit integer in wasm */
  167. fmt_buf[fmt_end_idx - 1] = fmt_buf[fmt_end_idx];
  168. fmt_buf[fmt_end_idx] = '\0';
  169. }
  170. }
  171. n = snprintf(buf, sizeof(buf), fmt_buf, d);
  172. }
  173. else {
  174. int64 lld;
  175. /* Make 8-byte aligned */
  176. ap = (_va_list)(((uintptr_t)ap + 7) & ~(uintptr_t)7);
  177. CHECK_VA_ARG(ap, uint64);
  178. lld = _va_arg(ap, int64);
  179. n = snprintf(buf, sizeof(buf), fmt_buf, lld);
  180. }
  181. OUTPUT_TEMP_FORMAT();
  182. break;
  183. }
  184. case 's':
  185. {
  186. char buf_tmp[128], *buf = buf_tmp;
  187. char *start;
  188. uint32 s_offset, str_len, buf_len;
  189. PREPARE_TEMP_FORMAT();
  190. CHECK_VA_ARG(ap, int32);
  191. s_offset = _va_arg(ap, uint32);
  192. if (!validate_app_str_addr(s_offset)) {
  193. if (fmt_buf != temp_fmt) {
  194. wasm_runtime_free(fmt_buf);
  195. }
  196. return false;
  197. }
  198. s = start = addr_app_to_native((uint64)s_offset);
  199. str_len = (uint32)strlen(start);
  200. if (str_len >= UINT32_MAX - 64) {
  201. print_err(out, ctx);
  202. if (fmt_buf != temp_fmt) {
  203. wasm_runtime_free(fmt_buf);
  204. }
  205. break;
  206. }
  207. /* reserve 64 more bytes as there may be width description
  208. * in the fmt */
  209. buf_len = str_len + 64;
  210. if (buf_len > (uint32)sizeof(buf_tmp)) {
  211. if (!(buf = wasm_runtime_malloc(buf_len))) {
  212. print_err(out, ctx);
  213. if (fmt_buf != temp_fmt) {
  214. wasm_runtime_free(fmt_buf);
  215. }
  216. break;
  217. }
  218. }
  219. n = snprintf(buf, buf_len, fmt_buf,
  220. (s_offset == 0 && str_len == 0) ? NULL
  221. : start);
  222. OUTPUT_TEMP_FORMAT();
  223. if (buf != buf_tmp) {
  224. wasm_runtime_free(buf);
  225. }
  226. break;
  227. }
  228. case '%':
  229. {
  230. out((int)'%', ctx);
  231. break;
  232. }
  233. case 'e':
  234. case 'E':
  235. case 'g':
  236. case 'G':
  237. case 'f':
  238. case 'F':
  239. {
  240. float64 f64;
  241. char buf[64];
  242. PREPARE_TEMP_FORMAT();
  243. /* Make 8-byte aligned */
  244. ap = (_va_list)(((uintptr_t)ap + 7) & ~(uintptr_t)7);
  245. CHECK_VA_ARG(ap, float64);
  246. f64 = _va_arg(ap, float64);
  247. n = snprintf(buf, sizeof(buf), fmt_buf, f64);
  248. OUTPUT_TEMP_FORMAT();
  249. break;
  250. }
  251. case 'n':
  252. /* print nothing */
  253. break;
  254. default:
  255. out((int)'%', ctx);
  256. out((int)*fmt, ctx);
  257. break;
  258. }
  259. might_format = 0;
  260. }
  261. still_might_format:
  262. ++fmt;
  263. }
  264. return true;
  265. fail:
  266. wasm_runtime_set_exception(module_inst, "out of bounds memory access");
  267. return false;
  268. }
  269. struct str_context {
  270. char *str;
  271. uint32 max;
  272. uint32 count;
  273. };
  274. static int
  275. sprintf_out(int c, struct str_context *ctx)
  276. {
  277. if (!ctx->str || ctx->count >= ctx->max) {
  278. ctx->count++;
  279. return c;
  280. }
  281. if (ctx->count == ctx->max - 1) {
  282. ctx->str[ctx->count++] = '\0';
  283. }
  284. else {
  285. ctx->str[ctx->count++] = (char)c;
  286. }
  287. return c;
  288. }
  289. #ifndef BUILTIN_LIBC_BUFFERED_PRINTF
  290. #define BUILTIN_LIBC_BUFFERED_PRINTF 0
  291. #endif
  292. #ifndef BUILTIN_LIBC_BUFFERED_PRINT_SIZE
  293. #define BUILTIN_LIBC_BUFFERED_PRINT_SIZE 128
  294. #endif
  295. #ifndef BUILTIN_LIBC_BUFFERED_PRINT_PREFIX
  296. #define BUILTIN_LIBC_BUFFERED_PRINT_PREFIX
  297. #endif
  298. #if BUILTIN_LIBC_BUFFERED_PRINTF != 0
  299. BUILTIN_LIBC_BUFFERED_PRINT_PREFIX
  300. static char print_buf[BUILTIN_LIBC_BUFFERED_PRINT_SIZE] = { 0 };
  301. BUILTIN_LIBC_BUFFERED_PRINT_PREFIX
  302. static int print_buf_size = 0;
  303. static int
  304. printf_out(int c, struct str_context *ctx)
  305. {
  306. if (c == '\n') {
  307. print_buf[print_buf_size] = '\0';
  308. os_printf("%s\n", print_buf);
  309. print_buf_size = 0;
  310. }
  311. else if (print_buf_size >= sizeof(print_buf) - 2) {
  312. print_buf[print_buf_size++] = (char)c;
  313. print_buf[print_buf_size] = '\0';
  314. os_printf("%s\n", print_buf);
  315. print_buf_size = 0;
  316. }
  317. else {
  318. print_buf[print_buf_size++] = (char)c;
  319. }
  320. ctx->count++;
  321. return c;
  322. }
  323. #else
  324. static int
  325. printf_out(int c, struct str_context *ctx)
  326. {
  327. os_printf("%c", c);
  328. ctx->count++;
  329. return c;
  330. }
  331. #endif
  332. static int
  333. printf_wrapper(wasm_exec_env_t exec_env, const char *format, _va_list va_args)
  334. {
  335. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  336. struct str_context ctx = { NULL, 0, 0 };
  337. /* format has been checked by runtime */
  338. if (!validate_native_addr(va_args, (uint64)sizeof(int32)))
  339. return 0;
  340. if (!_vprintf_wa((out_func_t)printf_out, &ctx, format, va_args,
  341. module_inst))
  342. return 0;
  343. return (int)ctx.count;
  344. }
  345. static int
  346. sprintf_wrapper(wasm_exec_env_t exec_env, char *str, const char *format,
  347. _va_list va_args)
  348. {
  349. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  350. uint8 *native_end_offset;
  351. struct str_context ctx;
  352. /* str and format have been checked by runtime */
  353. if (!validate_native_addr(va_args, (uint64)sizeof(uint32)))
  354. return 0;
  355. if (!wasm_runtime_get_native_addr_range(module_inst, (uint8 *)str, NULL,
  356. &native_end_offset)) {
  357. wasm_runtime_set_exception(module_inst, "out of bounds memory access");
  358. return 0;
  359. }
  360. ctx.str = str;
  361. ctx.max = (uint32)(native_end_offset - (uint8 *)str);
  362. ctx.count = 0;
  363. if (!_vprintf_wa((out_func_t)sprintf_out, &ctx, format, va_args,
  364. module_inst))
  365. return 0;
  366. if (ctx.count < ctx.max) {
  367. str[ctx.count] = '\0';
  368. }
  369. return (int)ctx.count;
  370. }
  371. static int
  372. snprintf_wrapper(wasm_exec_env_t exec_env, char *str, uint32 size,
  373. const char *format, _va_list va_args)
  374. {
  375. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  376. struct str_context ctx;
  377. /* str and format have been checked by runtime */
  378. if (!validate_native_addr(va_args, (uint64)sizeof(uint32)))
  379. return 0;
  380. ctx.str = str;
  381. ctx.max = size;
  382. ctx.count = 0;
  383. if (!_vprintf_wa((out_func_t)sprintf_out, &ctx, format, va_args,
  384. module_inst))
  385. return 0;
  386. if (ctx.count < ctx.max) {
  387. str[ctx.count] = '\0';
  388. }
  389. return (int)ctx.count;
  390. }
  391. static int
  392. puts_wrapper(wasm_exec_env_t exec_env, const char *str)
  393. {
  394. (void)exec_env;
  395. return os_printf("%s\n", str);
  396. }
  397. static int
  398. putchar_wrapper(wasm_exec_env_t exec_env, int c)
  399. {
  400. (void)exec_env;
  401. os_printf("%c", c);
  402. return 1;
  403. }
  404. static uint32
  405. strdup_wrapper(wasm_exec_env_t exec_env, const char *str)
  406. {
  407. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  408. char *str_ret;
  409. uint32 len;
  410. uint32 str_ret_offset = 0;
  411. /* str has been checked by runtime */
  412. if (str) {
  413. len = (uint32)strlen(str) + 1;
  414. str_ret_offset = (uint32)module_malloc((uint64)len, (void **)&str_ret);
  415. if (str_ret_offset) {
  416. bh_memcpy_s(str_ret, len, str, len);
  417. }
  418. }
  419. return str_ret_offset;
  420. }
  421. static uint32
  422. _strdup_wrapper(wasm_exec_env_t exec_env, const char *str)
  423. {
  424. return strdup_wrapper(exec_env, str);
  425. }
  426. static int32
  427. memcmp_wrapper(wasm_exec_env_t exec_env, const void *s1, const void *s2,
  428. uint32 size)
  429. {
  430. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  431. /* s2 has been checked by runtime */
  432. if (!validate_native_addr((void *)s1, (uint64)size))
  433. return 0;
  434. return memcmp(s1, s2, size);
  435. }
  436. static uint32
  437. memcpy_wrapper(wasm_exec_env_t exec_env, void *dst, const void *src,
  438. uint32 size)
  439. {
  440. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  441. uint32 dst_offset = (uint32)addr_native_to_app(dst);
  442. if (size == 0)
  443. return dst_offset;
  444. /* src has been checked by runtime */
  445. if (!validate_native_addr(dst, (uint64)size))
  446. return dst_offset;
  447. bh_memcpy_s(dst, size, src, size);
  448. return dst_offset;
  449. }
  450. static uint32
  451. memmove_wrapper(wasm_exec_env_t exec_env, void *dst, void *src, uint32 size)
  452. {
  453. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  454. uint32 dst_offset = (uint32)addr_native_to_app(dst);
  455. if (size == 0)
  456. return dst_offset;
  457. /* src has been checked by runtime */
  458. if (!validate_native_addr(dst, (uint64)size))
  459. return dst_offset;
  460. memmove(dst, src, size);
  461. return dst_offset;
  462. }
  463. static uint32
  464. memset_wrapper(wasm_exec_env_t exec_env, void *s, int32 c, uint32 size)
  465. {
  466. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  467. uint32 s_offset = (uint32)addr_native_to_app(s);
  468. if (!validate_native_addr(s, (uint64)size))
  469. return s_offset;
  470. memset(s, c, size);
  471. return s_offset;
  472. }
  473. static uint32
  474. strchr_wrapper(wasm_exec_env_t exec_env, const char *s, int32 c)
  475. {
  476. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  477. char *ret;
  478. /* s has been checked by runtime */
  479. ret = strchr(s, c);
  480. return ret ? (uint32)addr_native_to_app(ret) : 0;
  481. }
  482. static int32
  483. strcmp_wrapper(wasm_exec_env_t exec_env, const char *s1, const char *s2)
  484. {
  485. (void)exec_env;
  486. /* s1 and s2 have been checked by runtime */
  487. return strcmp(s1, s2);
  488. }
  489. static int32
  490. strncmp_wrapper(wasm_exec_env_t exec_env, const char *s1, const char *s2,
  491. uint32 size)
  492. {
  493. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  494. /* s2 has been checked by runtime */
  495. if (!validate_native_addr((void *)s1, (uint64)size))
  496. return 0;
  497. return strncmp(s1, s2, size);
  498. }
  499. static uint32
  500. strcpy_wrapper(wasm_exec_env_t exec_env, char *dst, const char *src)
  501. {
  502. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  503. uint32 len = (uint32)strlen(src) + 1;
  504. /* src has been checked by runtime */
  505. if (!validate_native_addr(dst, (uint64)len))
  506. return 0;
  507. #ifndef BH_PLATFORM_WINDOWS
  508. strncpy(dst, src, len);
  509. #else
  510. strncpy_s(dst, len, src, len);
  511. #endif
  512. return (uint32)addr_native_to_app(dst);
  513. }
  514. static uint32
  515. strncpy_wrapper(wasm_exec_env_t exec_env, char *dst, const char *src,
  516. uint32 size)
  517. {
  518. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  519. /* src has been checked by runtime */
  520. if (!validate_native_addr(dst, (uint64)size))
  521. return 0;
  522. #ifndef BH_PLATFORM_WINDOWS
  523. strncpy(dst, src, size);
  524. #else
  525. strncpy_s(dst, size, src, size);
  526. #endif
  527. return (uint32)addr_native_to_app(dst);
  528. }
  529. static uint32
  530. strlen_wrapper(wasm_exec_env_t exec_env, const char *s)
  531. {
  532. (void)exec_env;
  533. /* s has been checked by runtime */
  534. return (uint32)strlen(s);
  535. }
  536. static uint32
  537. malloc_wrapper(wasm_exec_env_t exec_env, uint32 size)
  538. {
  539. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  540. return (uint32)module_malloc((uint64)size, NULL);
  541. }
  542. static uint32
  543. calloc_wrapper(wasm_exec_env_t exec_env, uint32 nmemb, uint32 size)
  544. {
  545. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  546. uint64 total_size = (uint64)nmemb * (uint64)size;
  547. uint32 ret_offset = 0;
  548. uint8 *ret_ptr;
  549. if (total_size >= UINT32_MAX)
  550. return 0;
  551. ret_offset = (uint32)module_malloc(total_size, (void **)&ret_ptr);
  552. if (ret_offset) {
  553. memset(ret_ptr, 0, (uint32)total_size);
  554. }
  555. return ret_offset;
  556. }
  557. static uint32
  558. realloc_wrapper(wasm_exec_env_t exec_env, uint32 ptr, uint32 new_size)
  559. {
  560. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  561. return wasm_runtime_module_realloc(module_inst, ptr, new_size, NULL);
  562. }
  563. static void
  564. free_wrapper(wasm_exec_env_t exec_env, void *ptr)
  565. {
  566. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  567. if (!validate_native_addr(ptr, (uint64)sizeof(uint32)))
  568. return;
  569. module_free(addr_native_to_app(ptr));
  570. }
  571. static int32
  572. atoi_wrapper(wasm_exec_env_t exec_env, const char *s)
  573. {
  574. (void)exec_env;
  575. /* s has been checked by runtime */
  576. return atoi(s);
  577. }
  578. static void
  579. exit_wrapper(wasm_exec_env_t exec_env, int32 status)
  580. {
  581. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  582. char buf[32];
  583. snprintf(buf, sizeof(buf), "env.exit(%" PRId32 ")", status);
  584. wasm_runtime_set_exception(module_inst, buf);
  585. }
  586. static int32
  587. strtol_wrapper(wasm_exec_env_t exec_env, const char *nptr, char **endptr,
  588. int32 base)
  589. {
  590. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  591. int32 num = 0;
  592. /* nptr has been checked by runtime */
  593. if (!validate_native_addr(endptr, (uint64)sizeof(uint32)))
  594. return 0;
  595. num = (int32)strtol(nptr, endptr, base);
  596. *(uint32 *)endptr = (uint32)addr_native_to_app(*endptr);
  597. return num;
  598. }
  599. static uint32
  600. strtoul_wrapper(wasm_exec_env_t exec_env, const char *nptr, char **endptr,
  601. int32 base)
  602. {
  603. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  604. uint32 num = 0;
  605. /* nptr has been checked by runtime */
  606. if (!validate_native_addr(endptr, (uint64)sizeof(uint32)))
  607. return 0;
  608. num = (uint32)strtoul(nptr, endptr, base);
  609. *(uint32 *)endptr = (uint32)addr_native_to_app(*endptr);
  610. return num;
  611. }
  612. static uint32
  613. memchr_wrapper(wasm_exec_env_t exec_env, const void *s, int32 c, uint32 n)
  614. {
  615. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  616. void *res;
  617. if (!validate_native_addr((void *)s, (uint64)n))
  618. return 0;
  619. res = memchr(s, c, n);
  620. return (uint32)addr_native_to_app(res);
  621. }
  622. static int32
  623. strncasecmp_wrapper(wasm_exec_env_t exec_env, const char *s1, const char *s2,
  624. uint32 n)
  625. {
  626. (void)exec_env;
  627. /* s1 and s2 have been checked by runtime */
  628. return strncasecmp(s1, s2, n);
  629. }
  630. static uint32
  631. strspn_wrapper(wasm_exec_env_t exec_env, const char *s, const char *accept)
  632. {
  633. (void)exec_env;
  634. /* s and accept have been checked by runtime */
  635. return (uint32)strspn(s, accept);
  636. }
  637. static uint32
  638. strcspn_wrapper(wasm_exec_env_t exec_env, const char *s, const char *reject)
  639. {
  640. (void)exec_env;
  641. /* s and reject have been checked by runtime */
  642. return (uint32)strcspn(s, reject);
  643. }
  644. static uint32
  645. strstr_wrapper(wasm_exec_env_t exec_env, const char *s, const char *find)
  646. {
  647. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  648. /* s and find have been checked by runtime */
  649. char *res = strstr(s, find);
  650. return (uint32)addr_native_to_app(res);
  651. }
  652. static int32
  653. isupper_wrapper(wasm_exec_env_t exec_env, int32 c)
  654. {
  655. (void)exec_env;
  656. return isupper(c);
  657. }
  658. static int32
  659. isalpha_wrapper(wasm_exec_env_t exec_env, int32 c)
  660. {
  661. (void)exec_env;
  662. return isalpha(c);
  663. }
  664. static int32
  665. isspace_wrapper(wasm_exec_env_t exec_env, int32 c)
  666. {
  667. (void)exec_env;
  668. return isspace(c);
  669. }
  670. static int32
  671. isgraph_wrapper(wasm_exec_env_t exec_env, int32 c)
  672. {
  673. (void)exec_env;
  674. return isgraph(c);
  675. }
  676. static int32
  677. isprint_wrapper(wasm_exec_env_t exec_env, int32 c)
  678. {
  679. (void)exec_env;
  680. return isprint(c);
  681. }
  682. static int32
  683. isdigit_wrapper(wasm_exec_env_t exec_env, int32 c)
  684. {
  685. (void)exec_env;
  686. return isdigit(c);
  687. }
  688. static int32
  689. isxdigit_wrapper(wasm_exec_env_t exec_env, int32 c)
  690. {
  691. (void)exec_env;
  692. return isxdigit(c);
  693. }
  694. static int32
  695. tolower_wrapper(wasm_exec_env_t exec_env, int32 c)
  696. {
  697. (void)exec_env;
  698. return tolower(c);
  699. }
  700. static int32
  701. toupper_wrapper(wasm_exec_env_t exec_env, int32 c)
  702. {
  703. (void)exec_env;
  704. return toupper(c);
  705. }
  706. static int32
  707. isalnum_wrapper(wasm_exec_env_t exec_env, int32 c)
  708. {
  709. (void)exec_env;
  710. return isalnum(c);
  711. }
  712. static uint32
  713. emscripten_memcpy_big_wrapper(wasm_exec_env_t exec_env, void *dst,
  714. const void *src, uint32 size)
  715. {
  716. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  717. uint32 dst_offset = (uint32)addr_native_to_app(dst);
  718. /* src has been checked by runtime */
  719. if (!validate_native_addr(dst, (uint64)size))
  720. return dst_offset;
  721. bh_memcpy_s(dst, size, src, size);
  722. return dst_offset;
  723. }
  724. static void
  725. abort_wrapper(wasm_exec_env_t exec_env, int32 code)
  726. {
  727. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  728. char buf[32];
  729. snprintf(buf, sizeof(buf), "env.abort(%" PRId32 ")", code);
  730. wasm_runtime_set_exception(module_inst, buf);
  731. }
  732. static void
  733. abortStackOverflow_wrapper(wasm_exec_env_t exec_env, int32 code)
  734. {
  735. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  736. char buf[32];
  737. snprintf(buf, sizeof(buf), "env.abortStackOverflow(%" PRId32 ")", code);
  738. wasm_runtime_set_exception(module_inst, buf);
  739. }
  740. static void
  741. nullFunc_X_wrapper(wasm_exec_env_t exec_env, int32 code)
  742. {
  743. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  744. char buf[32];
  745. snprintf(buf, sizeof(buf), "env.nullFunc_X(%" PRId32 ")", code);
  746. wasm_runtime_set_exception(module_inst, buf);
  747. }
  748. static uint32
  749. __cxa_allocate_exception_wrapper(wasm_exec_env_t exec_env, uint32 thrown_size)
  750. {
  751. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  752. uint32 exception = (uint32)module_malloc((uint64)thrown_size, NULL);
  753. if (!exception)
  754. return 0;
  755. return exception;
  756. }
  757. static void
  758. __cxa_begin_catch_wrapper(wasm_exec_env_t exec_env, void *exception_object)
  759. {
  760. (void)exec_env;
  761. (void)exception_object;
  762. }
  763. static void
  764. __cxa_throw_wrapper(wasm_exec_env_t exec_env, void *thrown_exception,
  765. void *tinfo, uint32 table_elem_idx)
  766. {
  767. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  768. char buf[32];
  769. (void)thrown_exception;
  770. (void)tinfo;
  771. (void)table_elem_idx;
  772. snprintf(buf, sizeof(buf), "%s", "exception thrown by stdc++");
  773. wasm_runtime_set_exception(module_inst, buf);
  774. }
  775. struct timespec_app {
  776. int64 tv_sec;
  777. int32 tv_nsec;
  778. };
  779. static uint32
  780. clock_gettime_wrapper(wasm_exec_env_t exec_env, uint32 clk_id,
  781. struct timespec_app *ts_app)
  782. {
  783. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  784. uint64 time;
  785. (void)clk_id;
  786. if (!validate_native_addr(ts_app, (uint64)sizeof(struct timespec_app)))
  787. return (uint32)-1;
  788. time = os_time_get_boot_us();
  789. ts_app->tv_sec = time / 1000000;
  790. ts_app->tv_nsec = (time % 1000000) * 1000;
  791. return (uint32)0;
  792. }
  793. static uint64
  794. clock_wrapper(wasm_exec_env_t exec_env)
  795. {
  796. (void)exec_env;
  797. /* Convert to nano seconds as CLOCKS_PER_SEC in wasi-sdk */
  798. return os_time_get_boot_us() * 1000;
  799. }
  800. #if WASM_ENABLE_SPEC_TEST != 0
  801. static void
  802. print_wrapper(wasm_exec_env_t exec_env)
  803. {
  804. os_printf("in specttest.print()\n");
  805. }
  806. static void
  807. print_i32_wrapper(wasm_exec_env_t exec_env, int32 i32)
  808. {
  809. os_printf("in specttest.print_i32(%" PRId32 ")\n", i32);
  810. }
  811. static void
  812. print_i32_f32_wrapper(wasm_exec_env_t exec_env, int32 i32, float f32)
  813. {
  814. os_printf("in specttest.print_i32_f32(%" PRId32 ", %f)\n", i32, f32);
  815. }
  816. static void
  817. print_f64_f64_wrapper(wasm_exec_env_t exec_env, double f64_1, double f64_2)
  818. {
  819. os_printf("in specttest.print_f64_f64(%f, %f)\n", f64_1, f64_2);
  820. }
  821. static void
  822. print_f32_wrapper(wasm_exec_env_t exec_env, float f32)
  823. {
  824. os_printf("in specttest.print_f32(%f)\n", f32);
  825. }
  826. static void
  827. print_f64_wrapper(wasm_exec_env_t exec_env, double f64)
  828. {
  829. os_printf("in specttest.print_f64(%f)\n", f64);
  830. }
  831. #endif /* WASM_ENABLE_SPEC_TEST */
  832. /* clang-format off */
  833. #define REG_NATIVE_FUNC(func_name, signature) \
  834. { #func_name, func_name##_wrapper, signature, NULL }
  835. /* clang-format on */
  836. static NativeSymbol native_symbols_libc_builtin[] = {
  837. REG_NATIVE_FUNC(printf, "($*)i"),
  838. REG_NATIVE_FUNC(sprintf, "($$*)i"),
  839. REG_NATIVE_FUNC(snprintf, "(*~$*)i"),
  840. { "vprintf", printf_wrapper, "($*)i", NULL },
  841. { "vsprintf", sprintf_wrapper, "($$*)i", NULL },
  842. { "vsnprintf", snprintf_wrapper, "(*~$*)i", NULL },
  843. REG_NATIVE_FUNC(puts, "($)i"),
  844. REG_NATIVE_FUNC(putchar, "(i)i"),
  845. REG_NATIVE_FUNC(memcmp, "(**~)i"),
  846. REG_NATIVE_FUNC(memcpy, "(**~)i"),
  847. REG_NATIVE_FUNC(memmove, "(**~)i"),
  848. REG_NATIVE_FUNC(memset, "(*ii)i"),
  849. REG_NATIVE_FUNC(strchr, "($i)i"),
  850. REG_NATIVE_FUNC(strcmp, "($$)i"),
  851. REG_NATIVE_FUNC(strcpy, "(*$)i"),
  852. REG_NATIVE_FUNC(strlen, "($)i"),
  853. REG_NATIVE_FUNC(strncmp, "(**~)i"),
  854. REG_NATIVE_FUNC(strncpy, "(**~)i"),
  855. REG_NATIVE_FUNC(malloc, "(i)i"),
  856. REG_NATIVE_FUNC(realloc, "(ii)i"),
  857. REG_NATIVE_FUNC(calloc, "(ii)i"),
  858. REG_NATIVE_FUNC(strdup, "($)i"),
  859. /* clang may introduce __strdup */
  860. REG_NATIVE_FUNC(_strdup, "($)i"),
  861. REG_NATIVE_FUNC(free, "(*)"),
  862. REG_NATIVE_FUNC(atoi, "($)i"),
  863. REG_NATIVE_FUNC(exit, "(i)"),
  864. REG_NATIVE_FUNC(strtol, "($*i)i"),
  865. REG_NATIVE_FUNC(strtoul, "($*i)i"),
  866. REG_NATIVE_FUNC(memchr, "(*ii)i"),
  867. REG_NATIVE_FUNC(strncasecmp, "($$i)i"),
  868. REG_NATIVE_FUNC(strspn, "($$)i"),
  869. REG_NATIVE_FUNC(strcspn, "($$)i"),
  870. REG_NATIVE_FUNC(strstr, "($$)i"),
  871. REG_NATIVE_FUNC(isupper, "(i)i"),
  872. REG_NATIVE_FUNC(isalpha, "(i)i"),
  873. REG_NATIVE_FUNC(isspace, "(i)i"),
  874. REG_NATIVE_FUNC(isgraph, "(i)i"),
  875. REG_NATIVE_FUNC(isprint, "(i)i"),
  876. REG_NATIVE_FUNC(isdigit, "(i)i"),
  877. REG_NATIVE_FUNC(isxdigit, "(i)i"),
  878. REG_NATIVE_FUNC(tolower, "(i)i"),
  879. REG_NATIVE_FUNC(toupper, "(i)i"),
  880. REG_NATIVE_FUNC(isalnum, "(i)i"),
  881. REG_NATIVE_FUNC(emscripten_memcpy_big, "(**~)i"),
  882. REG_NATIVE_FUNC(abort, "(i)"),
  883. REG_NATIVE_FUNC(abortStackOverflow, "(i)"),
  884. REG_NATIVE_FUNC(nullFunc_X, "(i)"),
  885. REG_NATIVE_FUNC(__cxa_allocate_exception, "(i)i"),
  886. REG_NATIVE_FUNC(__cxa_begin_catch, "(*)"),
  887. REG_NATIVE_FUNC(__cxa_throw, "(**i)"),
  888. REG_NATIVE_FUNC(clock_gettime, "(i*)i"),
  889. REG_NATIVE_FUNC(clock, "()I"),
  890. };
  891. #if WASM_ENABLE_SPEC_TEST != 0
  892. static NativeSymbol native_symbols_spectest[] = {
  893. REG_NATIVE_FUNC(print, "()"),
  894. REG_NATIVE_FUNC(print_i32, "(i)"),
  895. REG_NATIVE_FUNC(print_i32_f32, "(if)"),
  896. REG_NATIVE_FUNC(print_f64_f64, "(FF)"),
  897. REG_NATIVE_FUNC(print_f32, "(f)"),
  898. REG_NATIVE_FUNC(print_f64, "(F)")
  899. };
  900. #endif
  901. uint32
  902. get_libc_builtin_export_apis(NativeSymbol **p_libc_builtin_apis)
  903. {
  904. *p_libc_builtin_apis = native_symbols_libc_builtin;
  905. return sizeof(native_symbols_libc_builtin) / sizeof(NativeSymbol);
  906. }
  907. #if WASM_ENABLE_SPEC_TEST != 0
  908. uint32
  909. get_spectest_export_apis(NativeSymbol **p_libc_builtin_apis)
  910. {
  911. *p_libc_builtin_apis = native_symbols_spectest;
  912. return sizeof(native_symbols_spectest) / sizeof(NativeSymbol);
  913. }
  914. #endif
  915. /*************************************
  916. * Global Variables *
  917. *************************************/
  918. typedef struct WASMNativeGlobalDef {
  919. const char *module_name;
  920. const char *global_name;
  921. uint8 type;
  922. bool is_mutable;
  923. WASMValue value;
  924. } WASMNativeGlobalDef;
  925. static WASMNativeGlobalDef native_global_defs[] = {
  926. #if WASM_ENABLE_SPEC_TEST != 0
  927. { "spectest", "global_i32", VALUE_TYPE_I32, false, .value.i32 = 666 },
  928. { "spectest", "global_i64", VALUE_TYPE_I64, false, .value.i64 = 666 },
  929. { "spectest", "global_f32", VALUE_TYPE_F32, false, .value.f32 = 666.6 },
  930. { "spectest", "global_f64", VALUE_TYPE_F64, false, .value.f64 = 666.6 },
  931. { "test", "global-i32", VALUE_TYPE_I32, false, .value.i32 = 0 },
  932. { "test", "global-f32", VALUE_TYPE_F32, false, .value.f32 = 0 },
  933. { "test", "global-mut-i32", VALUE_TYPE_I32, true, .value.i32 = 0 },
  934. { "test", "global-mut-i64", VALUE_TYPE_I64, true, .value.i64 = 0 },
  935. #if WASM_ENABLE_GC != 0
  936. { "G", "g", VALUE_TYPE_I32, false, .value.i32 = 4 },
  937. { "M", "g", REF_TYPE_HT_NON_NULLABLE, false, .value.gc_obj = 0 },
  938. #endif
  939. #endif
  940. { "global", "NaN", VALUE_TYPE_F64, .value.u64 = 0x7FF8000000000000LL },
  941. { "global", "Infinity", VALUE_TYPE_F64, .value.u64 = 0x7FF0000000000000LL }
  942. };
  943. bool
  944. wasm_native_lookup_libc_builtin_global(const char *module_name,
  945. const char *global_name,
  946. WASMGlobalImport *global)
  947. {
  948. uint32 size = sizeof(native_global_defs) / sizeof(WASMNativeGlobalDef);
  949. WASMNativeGlobalDef *global_def = native_global_defs;
  950. WASMNativeGlobalDef *global_def_end = global_def + size;
  951. if (!module_name || !global_name || !global)
  952. return false;
  953. /* Lookup constant globals which can be defined by table */
  954. while (global_def < global_def_end) {
  955. if (!strcmp(global_def->module_name, module_name)
  956. && !strcmp(global_def->global_name, global_name)) {
  957. global->type = global_def->type;
  958. global->is_mutable = global_def->is_mutable;
  959. global->global_data_linked = global_def->value;
  960. return true;
  961. }
  962. global_def++;
  963. }
  964. return false;
  965. }