libc_wasi_wrapper.c 72 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323
  1. /*
  2. * Copyright (C) 2019 Intel Corporation. All rights reserved.
  3. * SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
  4. */
  5. #include "libc_wasi_wrapper.h"
  6. #include "bh_platform.h"
  7. #include "wasm_export.h"
  8. #include "wasm_runtime_common.h"
  9. #if WASM_ENABLE_THREAD_MGR != 0
  10. #include "../../../thread-mgr/thread_manager.h"
  11. #endif
  12. void
  13. wasm_runtime_set_exception(wasm_module_inst_t module, const char *exception);
  14. /* clang-format off */
  15. #define get_module_inst(exec_env) \
  16. wasm_runtime_get_module_inst(exec_env)
  17. #define get_wasi_ctx(module_inst) \
  18. wasm_runtime_get_wasi_ctx(module_inst)
  19. #define validate_app_addr(offset, size) \
  20. wasm_runtime_validate_app_addr(module_inst, offset, size)
  21. #define validate_native_addr(addr, size) \
  22. wasm_runtime_validate_native_addr(module_inst, addr, size)
  23. #define addr_app_to_native(offset) \
  24. wasm_runtime_addr_app_to_native(module_inst, offset)
  25. #define addr_native_to_app(ptr) \
  26. wasm_runtime_addr_native_to_app(module_inst, ptr)
  27. #define module_malloc(size, p_native_addr) \
  28. wasm_runtime_module_malloc(module_inst, size, p_native_addr)
  29. #define module_free(offset) \
  30. wasm_runtime_module_free(module_inst, offset)
  31. /* clang-format on */
  32. typedef struct wasi_prestat_app {
  33. wasi_preopentype_t pr_type;
  34. uint32 pr_name_len;
  35. } wasi_prestat_app_t;
  36. typedef struct iovec_app {
  37. uint32 buf_offset;
  38. uint32 buf_len;
  39. } iovec_app_t;
  40. typedef struct WASIContext *wasi_ctx_t;
  41. wasi_ctx_t
  42. wasm_runtime_get_wasi_ctx(wasm_module_inst_t module_inst);
  43. static inline size_t
  44. min(size_t a, size_t b)
  45. {
  46. return a > b ? b : a;
  47. }
  48. static inline struct fd_table *
  49. wasi_ctx_get_curfds(wasm_module_inst_t module_inst, wasi_ctx_t wasi_ctx)
  50. {
  51. if (!wasi_ctx)
  52. return NULL;
  53. return wasi_ctx->curfds;
  54. }
  55. static inline struct argv_environ_values *
  56. wasi_ctx_get_argv_environ(wasm_module_inst_t module_inst, wasi_ctx_t wasi_ctx)
  57. {
  58. if (!wasi_ctx)
  59. return NULL;
  60. return wasi_ctx->argv_environ;
  61. }
  62. static inline struct fd_prestats *
  63. wasi_ctx_get_prestats(wasm_module_inst_t module_inst, wasi_ctx_t wasi_ctx)
  64. {
  65. if (!wasi_ctx)
  66. return NULL;
  67. return wasi_ctx->prestats;
  68. }
  69. static inline struct addr_pool *
  70. wasi_ctx_get_addr_pool(wasm_module_inst_t module_inst, wasi_ctx_t wasi_ctx)
  71. {
  72. if (!wasi_ctx)
  73. return NULL;
  74. return wasi_ctx->addr_pool;
  75. }
  76. static inline char **
  77. wasi_ctx_get_ns_lookup_list(wasi_ctx_t wasi_ctx)
  78. {
  79. if (!wasi_ctx)
  80. return NULL;
  81. return wasi_ctx->ns_lookup_list;
  82. }
  83. static wasi_errno_t
  84. wasi_args_get(wasm_exec_env_t exec_env, uint32 *argv_offsets, char *argv_buf)
  85. {
  86. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  87. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  88. struct argv_environ_values *argv_environ =
  89. wasi_ctx_get_argv_environ(module_inst, wasi_ctx);
  90. size_t argc, argv_buf_size, i;
  91. char **argv;
  92. uint64 total_size;
  93. wasi_errno_t err;
  94. if (!wasi_ctx)
  95. return (wasi_errno_t)-1;
  96. err = wasmtime_ssp_args_sizes_get(argv_environ, &argc, &argv_buf_size);
  97. if (err)
  98. return err;
  99. total_size = sizeof(int32) * ((uint64)argc + 1);
  100. if (total_size >= UINT32_MAX
  101. || !validate_native_addr(argv_offsets, (uint32)total_size)
  102. || argv_buf_size >= UINT32_MAX
  103. || !validate_native_addr(argv_buf, (uint32)argv_buf_size))
  104. return (wasi_errno_t)-1;
  105. total_size = sizeof(char *) * ((uint64)argc + 1);
  106. if (total_size >= UINT32_MAX
  107. || !(argv = wasm_runtime_malloc((uint32)total_size)))
  108. return (wasi_errno_t)-1;
  109. err = wasmtime_ssp_args_get(argv_environ, argv, argv_buf);
  110. if (err) {
  111. wasm_runtime_free(argv);
  112. return err;
  113. }
  114. for (i = 0; i < argc; i++)
  115. argv_offsets[i] = addr_native_to_app(argv[i]);
  116. wasm_runtime_free(argv);
  117. return 0;
  118. }
  119. static wasi_errno_t
  120. wasi_args_sizes_get(wasm_exec_env_t exec_env, uint32 *argc_app,
  121. uint32 *argv_buf_size_app)
  122. {
  123. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  124. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  125. struct argv_environ_values *argv_environ;
  126. size_t argc, argv_buf_size;
  127. wasi_errno_t err;
  128. if (!wasi_ctx)
  129. return (wasi_errno_t)-1;
  130. if (!validate_native_addr(argc_app, sizeof(uint32))
  131. || !validate_native_addr(argv_buf_size_app, sizeof(uint32)))
  132. return (wasi_errno_t)-1;
  133. argv_environ = wasi_ctx->argv_environ;
  134. err = wasmtime_ssp_args_sizes_get(argv_environ, &argc, &argv_buf_size);
  135. if (err)
  136. return err;
  137. *argc_app = (uint32)argc;
  138. *argv_buf_size_app = (uint32)argv_buf_size;
  139. return 0;
  140. }
  141. static wasi_errno_t
  142. wasi_clock_res_get(wasm_exec_env_t exec_env,
  143. wasi_clockid_t clock_id, /* uint32 clock_id */
  144. wasi_timestamp_t *resolution /* uint64 *resolution */)
  145. {
  146. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  147. if (!validate_native_addr(resolution, sizeof(wasi_timestamp_t)))
  148. return (wasi_errno_t)-1;
  149. return wasmtime_ssp_clock_res_get(clock_id, resolution);
  150. }
  151. static wasi_errno_t
  152. wasi_clock_time_get(wasm_exec_env_t exec_env,
  153. wasi_clockid_t clock_id, /* uint32 clock_id */
  154. wasi_timestamp_t precision, /* uint64 precision */
  155. wasi_timestamp_t *time /* uint64 *time */)
  156. {
  157. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  158. if (!validate_native_addr(time, sizeof(wasi_timestamp_t)))
  159. return (wasi_errno_t)-1;
  160. return wasmtime_ssp_clock_time_get(clock_id, precision, time);
  161. }
  162. static wasi_errno_t
  163. wasi_environ_get(wasm_exec_env_t exec_env, uint32 *environ_offsets,
  164. char *environ_buf)
  165. {
  166. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  167. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  168. struct argv_environ_values *argv_environ =
  169. wasi_ctx_get_argv_environ(module_inst, wasi_ctx);
  170. size_t environ_count, environ_buf_size, i;
  171. uint64 total_size;
  172. char **environs;
  173. wasi_errno_t err;
  174. if (!wasi_ctx)
  175. return (wasi_errno_t)-1;
  176. err = wasmtime_ssp_environ_sizes_get(argv_environ, &environ_count,
  177. &environ_buf_size);
  178. if (err)
  179. return err;
  180. total_size = sizeof(int32) * ((uint64)environ_count + 1);
  181. if (total_size >= UINT32_MAX
  182. || !validate_native_addr(environ_offsets, (uint32)total_size)
  183. || environ_buf_size >= UINT32_MAX
  184. || !validate_native_addr(environ_buf, (uint32)environ_buf_size))
  185. return (wasi_errno_t)-1;
  186. total_size = sizeof(char *) * (((uint64)environ_count + 1));
  187. if (total_size >= UINT32_MAX
  188. || !(environs = wasm_runtime_malloc((uint32)total_size)))
  189. return (wasi_errno_t)-1;
  190. err = wasmtime_ssp_environ_get(argv_environ, environs, environ_buf);
  191. if (err) {
  192. wasm_runtime_free(environs);
  193. return err;
  194. }
  195. for (i = 0; i < environ_count; i++)
  196. environ_offsets[i] = addr_native_to_app(environs[i]);
  197. wasm_runtime_free(environs);
  198. return 0;
  199. }
  200. static wasi_errno_t
  201. wasi_environ_sizes_get(wasm_exec_env_t exec_env, uint32 *environ_count_app,
  202. uint32 *environ_buf_size_app)
  203. {
  204. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  205. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  206. struct argv_environ_values *argv_environ =
  207. wasi_ctx_get_argv_environ(module_inst, wasi_ctx);
  208. size_t environ_count, environ_buf_size;
  209. wasi_errno_t err;
  210. if (!wasi_ctx)
  211. return (wasi_errno_t)-1;
  212. if (!validate_native_addr(environ_count_app, sizeof(uint32))
  213. || !validate_native_addr(environ_buf_size_app, sizeof(uint32)))
  214. return (wasi_errno_t)-1;
  215. err = wasmtime_ssp_environ_sizes_get(argv_environ, &environ_count,
  216. &environ_buf_size);
  217. if (err)
  218. return err;
  219. *environ_count_app = (uint32)environ_count;
  220. *environ_buf_size_app = (uint32)environ_buf_size;
  221. return 0;
  222. }
  223. static wasi_errno_t
  224. wasi_fd_prestat_get(wasm_exec_env_t exec_env, wasi_fd_t fd,
  225. wasi_prestat_app_t *prestat_app)
  226. {
  227. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  228. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  229. struct fd_prestats *prestats = wasi_ctx_get_prestats(module_inst, wasi_ctx);
  230. wasi_prestat_t prestat;
  231. wasi_errno_t err;
  232. if (!wasi_ctx)
  233. return (wasi_errno_t)-1;
  234. if (!validate_native_addr(prestat_app, sizeof(wasi_prestat_app_t)))
  235. return (wasi_errno_t)-1;
  236. err = wasmtime_ssp_fd_prestat_get(prestats, fd, &prestat);
  237. if (err)
  238. return err;
  239. prestat_app->pr_type = prestat.pr_type;
  240. prestat_app->pr_name_len = (uint32)prestat.u.dir.pr_name_len;
  241. return 0;
  242. }
  243. static wasi_errno_t
  244. wasi_fd_prestat_dir_name(wasm_exec_env_t exec_env, wasi_fd_t fd, char *path,
  245. uint32 path_len)
  246. {
  247. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  248. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  249. struct fd_prestats *prestats = wasi_ctx_get_prestats(module_inst, wasi_ctx);
  250. if (!wasi_ctx)
  251. return (wasi_errno_t)-1;
  252. return wasmtime_ssp_fd_prestat_dir_name(prestats, fd, path, path_len);
  253. }
  254. static wasi_errno_t
  255. wasi_fd_close(wasm_exec_env_t exec_env, wasi_fd_t fd)
  256. {
  257. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  258. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  259. struct fd_table *curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  260. struct fd_prestats *prestats = wasi_ctx_get_prestats(module_inst, wasi_ctx);
  261. if (!wasi_ctx)
  262. return (wasi_errno_t)-1;
  263. return wasmtime_ssp_fd_close(curfds, prestats, fd);
  264. }
  265. static wasi_errno_t
  266. wasi_fd_datasync(wasm_exec_env_t exec_env, wasi_fd_t fd)
  267. {
  268. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  269. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  270. struct fd_table *curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  271. if (!wasi_ctx)
  272. return (wasi_errno_t)-1;
  273. return wasmtime_ssp_fd_datasync(curfds, fd);
  274. }
  275. static wasi_errno_t
  276. wasi_fd_pread(wasm_exec_env_t exec_env, wasi_fd_t fd, iovec_app_t *iovec_app,
  277. uint32 iovs_len, wasi_filesize_t offset, uint32 *nread_app)
  278. {
  279. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  280. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  281. struct fd_table *curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  282. wasi_iovec_t *iovec, *iovec_begin;
  283. uint64 total_size;
  284. size_t nread;
  285. uint32 i;
  286. wasi_errno_t err;
  287. if (!wasi_ctx)
  288. return (wasi_errno_t)-1;
  289. total_size = sizeof(iovec_app_t) * (uint64)iovs_len;
  290. if (!validate_native_addr(nread_app, (uint32)sizeof(uint32))
  291. || total_size >= UINT32_MAX
  292. || !validate_native_addr(iovec_app, (uint32)total_size))
  293. return (wasi_errno_t)-1;
  294. total_size = sizeof(wasi_iovec_t) * (uint64)iovs_len;
  295. if (total_size >= UINT32_MAX
  296. || !(iovec_begin = wasm_runtime_malloc((uint32)total_size)))
  297. return (wasi_errno_t)-1;
  298. iovec = iovec_begin;
  299. for (i = 0; i < iovs_len; i++, iovec_app++, iovec++) {
  300. if (!validate_app_addr(iovec_app->buf_offset, iovec_app->buf_len)) {
  301. err = (wasi_errno_t)-1;
  302. goto fail;
  303. }
  304. iovec->buf = (void *)addr_app_to_native(iovec_app->buf_offset);
  305. iovec->buf_len = iovec_app->buf_len;
  306. }
  307. err = wasmtime_ssp_fd_pread(curfds, fd, iovec_begin, iovs_len, offset,
  308. &nread);
  309. if (err)
  310. goto fail;
  311. *nread_app = (uint32)nread;
  312. /* success */
  313. err = 0;
  314. fail:
  315. wasm_runtime_free(iovec_begin);
  316. return err;
  317. }
  318. static wasi_errno_t
  319. wasi_fd_pwrite(wasm_exec_env_t exec_env, wasi_fd_t fd,
  320. const iovec_app_t *iovec_app, uint32 iovs_len,
  321. wasi_filesize_t offset, uint32 *nwritten_app)
  322. {
  323. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  324. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  325. struct fd_table *curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  326. wasi_ciovec_t *ciovec, *ciovec_begin;
  327. uint64 total_size;
  328. size_t nwritten;
  329. uint32 i;
  330. wasi_errno_t err;
  331. if (!wasi_ctx)
  332. return (wasi_errno_t)-1;
  333. total_size = sizeof(iovec_app_t) * (uint64)iovs_len;
  334. if (!validate_native_addr(nwritten_app, (uint32)sizeof(uint32))
  335. || total_size >= UINT32_MAX
  336. || !validate_native_addr((void *)iovec_app, (uint32)total_size))
  337. return (wasi_errno_t)-1;
  338. total_size = sizeof(wasi_ciovec_t) * (uint64)iovs_len;
  339. if (total_size >= UINT32_MAX
  340. || !(ciovec_begin = wasm_runtime_malloc((uint32)total_size)))
  341. return (wasi_errno_t)-1;
  342. ciovec = ciovec_begin;
  343. for (i = 0; i < iovs_len; i++, iovec_app++, ciovec++) {
  344. if (!validate_app_addr(iovec_app->buf_offset, iovec_app->buf_len)) {
  345. err = (wasi_errno_t)-1;
  346. goto fail;
  347. }
  348. ciovec->buf = (char *)addr_app_to_native(iovec_app->buf_offset);
  349. ciovec->buf_len = iovec_app->buf_len;
  350. }
  351. err = wasmtime_ssp_fd_pwrite(curfds, fd, ciovec_begin, iovs_len, offset,
  352. &nwritten);
  353. if (err)
  354. goto fail;
  355. *nwritten_app = (uint32)nwritten;
  356. /* success */
  357. err = 0;
  358. fail:
  359. wasm_runtime_free(ciovec_begin);
  360. return err;
  361. }
  362. static wasi_errno_t
  363. wasi_fd_read(wasm_exec_env_t exec_env, wasi_fd_t fd,
  364. const iovec_app_t *iovec_app, uint32 iovs_len, uint32 *nread_app)
  365. {
  366. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  367. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  368. struct fd_table *curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  369. wasi_iovec_t *iovec, *iovec_begin;
  370. uint64 total_size;
  371. size_t nread;
  372. uint32 i;
  373. wasi_errno_t err;
  374. if (!wasi_ctx)
  375. return (wasi_errno_t)-1;
  376. total_size = sizeof(iovec_app_t) * (uint64)iovs_len;
  377. if (!validate_native_addr(nread_app, (uint32)sizeof(uint32))
  378. || total_size >= UINT32_MAX
  379. || !validate_native_addr((void *)iovec_app, (uint32)total_size))
  380. return (wasi_errno_t)-1;
  381. total_size = sizeof(wasi_iovec_t) * (uint64)iovs_len;
  382. if (total_size >= UINT32_MAX
  383. || !(iovec_begin = wasm_runtime_malloc((uint32)total_size)))
  384. return (wasi_errno_t)-1;
  385. iovec = iovec_begin;
  386. for (i = 0; i < iovs_len; i++, iovec_app++, iovec++) {
  387. if (!validate_app_addr(iovec_app->buf_offset, iovec_app->buf_len)) {
  388. err = (wasi_errno_t)-1;
  389. goto fail;
  390. }
  391. iovec->buf = (void *)addr_app_to_native(iovec_app->buf_offset);
  392. iovec->buf_len = iovec_app->buf_len;
  393. }
  394. err = wasmtime_ssp_fd_read(curfds, fd, iovec_begin, iovs_len, &nread);
  395. if (err)
  396. goto fail;
  397. *nread_app = (uint32)nread;
  398. /* success */
  399. err = 0;
  400. fail:
  401. wasm_runtime_free(iovec_begin);
  402. return err;
  403. }
  404. static wasi_errno_t
  405. wasi_fd_renumber(wasm_exec_env_t exec_env, wasi_fd_t from, wasi_fd_t to)
  406. {
  407. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  408. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  409. struct fd_table *curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  410. struct fd_prestats *prestats = wasi_ctx_get_prestats(module_inst, wasi_ctx);
  411. if (!wasi_ctx)
  412. return (wasi_errno_t)-1;
  413. return wasmtime_ssp_fd_renumber(curfds, prestats, from, to);
  414. }
  415. static wasi_errno_t
  416. wasi_fd_seek(wasm_exec_env_t exec_env, wasi_fd_t fd, wasi_filedelta_t offset,
  417. wasi_whence_t whence, wasi_filesize_t *newoffset)
  418. {
  419. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  420. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  421. struct fd_table *curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  422. if (!wasi_ctx)
  423. return (wasi_errno_t)-1;
  424. if (!validate_native_addr(newoffset, sizeof(wasi_filesize_t)))
  425. return (wasi_errno_t)-1;
  426. return wasmtime_ssp_fd_seek(curfds, fd, offset, whence, newoffset);
  427. }
  428. static wasi_errno_t
  429. wasi_fd_tell(wasm_exec_env_t exec_env, wasi_fd_t fd, wasi_filesize_t *newoffset)
  430. {
  431. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  432. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  433. struct fd_table *curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  434. if (!wasi_ctx)
  435. return (wasi_errno_t)-1;
  436. if (!validate_native_addr(newoffset, sizeof(wasi_filesize_t)))
  437. return (wasi_errno_t)-1;
  438. return wasmtime_ssp_fd_tell(curfds, fd, newoffset);
  439. }
  440. static wasi_errno_t
  441. wasi_fd_fdstat_get(wasm_exec_env_t exec_env, wasi_fd_t fd,
  442. wasi_fdstat_t *fdstat_app)
  443. {
  444. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  445. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  446. struct fd_table *curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  447. wasi_fdstat_t fdstat;
  448. wasi_errno_t err;
  449. if (!wasi_ctx)
  450. return (wasi_errno_t)-1;
  451. if (!validate_native_addr(fdstat_app, sizeof(wasi_fdstat_t)))
  452. return (wasi_errno_t)-1;
  453. err = wasmtime_ssp_fd_fdstat_get(curfds, fd, &fdstat);
  454. if (err)
  455. return err;
  456. memcpy(fdstat_app, &fdstat, sizeof(wasi_fdstat_t));
  457. return 0;
  458. }
  459. static wasi_errno_t
  460. wasi_fd_fdstat_set_flags(wasm_exec_env_t exec_env, wasi_fd_t fd,
  461. wasi_fdflags_t flags)
  462. {
  463. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  464. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  465. struct fd_table *curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  466. if (!wasi_ctx)
  467. return (wasi_errno_t)-1;
  468. return wasmtime_ssp_fd_fdstat_set_flags(curfds, fd, flags);
  469. }
  470. static wasi_errno_t
  471. wasi_fd_fdstat_set_rights(wasm_exec_env_t exec_env, wasi_fd_t fd,
  472. wasi_rights_t fs_rights_base,
  473. wasi_rights_t fs_rights_inheriting)
  474. {
  475. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  476. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  477. struct fd_table *curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  478. if (!wasi_ctx)
  479. return (wasi_errno_t)-1;
  480. return wasmtime_ssp_fd_fdstat_set_rights(curfds, fd, fs_rights_base,
  481. fs_rights_inheriting);
  482. }
  483. static wasi_errno_t
  484. wasi_fd_sync(wasm_exec_env_t exec_env, wasi_fd_t fd)
  485. {
  486. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  487. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  488. struct fd_table *curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  489. if (!wasi_ctx)
  490. return (wasi_errno_t)-1;
  491. return wasmtime_ssp_fd_sync(curfds, fd);
  492. }
  493. static wasi_errno_t
  494. wasi_fd_write(wasm_exec_env_t exec_env, wasi_fd_t fd,
  495. const iovec_app_t *iovec_app, uint32 iovs_len,
  496. uint32 *nwritten_app)
  497. {
  498. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  499. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  500. struct fd_table *curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  501. wasi_ciovec_t *ciovec, *ciovec_begin;
  502. uint64 total_size;
  503. size_t nwritten;
  504. uint32 i;
  505. wasi_errno_t err;
  506. if (!wasi_ctx)
  507. return (wasi_errno_t)-1;
  508. total_size = sizeof(iovec_app_t) * (uint64)iovs_len;
  509. if (!validate_native_addr(nwritten_app, (uint32)sizeof(uint32))
  510. || total_size >= UINT32_MAX
  511. || !validate_native_addr((void *)iovec_app, (uint32)total_size))
  512. return (wasi_errno_t)-1;
  513. total_size = sizeof(wasi_ciovec_t) * (uint64)iovs_len;
  514. if (total_size >= UINT32_MAX
  515. || !(ciovec_begin = wasm_runtime_malloc((uint32)total_size)))
  516. return (wasi_errno_t)-1;
  517. ciovec = ciovec_begin;
  518. for (i = 0; i < iovs_len; i++, iovec_app++, ciovec++) {
  519. if (!validate_app_addr(iovec_app->buf_offset, iovec_app->buf_len)) {
  520. err = (wasi_errno_t)-1;
  521. goto fail;
  522. }
  523. ciovec->buf = (char *)addr_app_to_native(iovec_app->buf_offset);
  524. ciovec->buf_len = iovec_app->buf_len;
  525. }
  526. err = wasmtime_ssp_fd_write(curfds, fd, ciovec_begin, iovs_len, &nwritten);
  527. if (err)
  528. goto fail;
  529. *nwritten_app = (uint32)nwritten;
  530. /* success */
  531. err = 0;
  532. fail:
  533. wasm_runtime_free(ciovec_begin);
  534. return err;
  535. }
  536. static wasi_errno_t
  537. wasi_fd_advise(wasm_exec_env_t exec_env, wasi_fd_t fd, wasi_filesize_t offset,
  538. wasi_filesize_t len, wasi_advice_t advice)
  539. {
  540. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  541. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  542. struct fd_table *curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  543. if (!wasi_ctx)
  544. return (wasi_errno_t)-1;
  545. return wasmtime_ssp_fd_advise(curfds, fd, offset, len, advice);
  546. }
  547. static wasi_errno_t
  548. wasi_fd_allocate(wasm_exec_env_t exec_env, wasi_fd_t fd, wasi_filesize_t offset,
  549. wasi_filesize_t len)
  550. {
  551. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  552. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  553. struct fd_table *curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  554. if (!wasi_ctx)
  555. return (wasi_errno_t)-1;
  556. return wasmtime_ssp_fd_allocate(curfds, fd, offset, len);
  557. }
  558. static wasi_errno_t
  559. wasi_path_create_directory(wasm_exec_env_t exec_env, wasi_fd_t fd,
  560. const char *path, uint32 path_len)
  561. {
  562. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  563. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  564. struct fd_table *curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  565. if (!wasi_ctx)
  566. return (wasi_errno_t)-1;
  567. return wasmtime_ssp_path_create_directory(curfds, fd, path, path_len);
  568. }
  569. static wasi_errno_t
  570. wasi_path_link(wasm_exec_env_t exec_env, wasi_fd_t old_fd,
  571. wasi_lookupflags_t old_flags, const char *old_path,
  572. uint32 old_path_len, wasi_fd_t new_fd, const char *new_path,
  573. uint32 new_path_len)
  574. {
  575. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  576. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  577. struct fd_table *curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  578. struct fd_prestats *prestats = wasi_ctx_get_prestats(module_inst, wasi_ctx);
  579. if (!wasi_ctx)
  580. return (wasi_errno_t)-1;
  581. return wasmtime_ssp_path_link(curfds, prestats, old_fd, old_flags, old_path,
  582. old_path_len, new_fd, new_path, new_path_len);
  583. }
  584. static wasi_errno_t
  585. wasi_path_open(wasm_exec_env_t exec_env, wasi_fd_t dirfd,
  586. wasi_lookupflags_t dirflags, const char *path, uint32 path_len,
  587. wasi_oflags_t oflags, wasi_rights_t fs_rights_base,
  588. wasi_rights_t fs_rights_inheriting, wasi_fdflags_t fs_flags,
  589. wasi_fd_t *fd_app)
  590. {
  591. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  592. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  593. struct fd_table *curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  594. wasi_fd_t fd = (wasi_fd_t)-1; /* set fd_app -1 if path open failed */
  595. wasi_errno_t err;
  596. if (!wasi_ctx)
  597. return (wasi_errno_t)-1;
  598. if (!validate_native_addr(fd_app, sizeof(wasi_fd_t)))
  599. return (wasi_errno_t)-1;
  600. err = wasmtime_ssp_path_open(curfds, dirfd, dirflags, path, path_len,
  601. oflags, fs_rights_base, fs_rights_inheriting,
  602. fs_flags, &fd);
  603. *fd_app = fd;
  604. return err;
  605. }
  606. static wasi_errno_t
  607. wasi_fd_readdir(wasm_exec_env_t exec_env, wasi_fd_t fd, void *buf,
  608. uint32 buf_len, wasi_dircookie_t cookie, uint32 *bufused_app)
  609. {
  610. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  611. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  612. struct fd_table *curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  613. size_t bufused;
  614. wasi_errno_t err;
  615. if (!wasi_ctx)
  616. return (wasi_errno_t)-1;
  617. if (!validate_native_addr(bufused_app, sizeof(uint32)))
  618. return (wasi_errno_t)-1;
  619. err = wasmtime_ssp_fd_readdir(curfds, fd, buf, buf_len, cookie, &bufused);
  620. if (err)
  621. return err;
  622. *bufused_app = (uint32)bufused;
  623. return 0;
  624. }
  625. static wasi_errno_t
  626. wasi_path_readlink(wasm_exec_env_t exec_env, wasi_fd_t fd, const char *path,
  627. uint32 path_len, char *buf, uint32 buf_len,
  628. uint32 *bufused_app)
  629. {
  630. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  631. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  632. struct fd_table *curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  633. size_t bufused;
  634. wasi_errno_t err;
  635. if (!wasi_ctx)
  636. return (wasi_errno_t)-1;
  637. if (!validate_native_addr(bufused_app, sizeof(uint32)))
  638. return (wasi_errno_t)-1;
  639. err = wasmtime_ssp_path_readlink(curfds, fd, path, path_len, buf, buf_len,
  640. &bufused);
  641. if (err)
  642. return err;
  643. *bufused_app = (uint32)bufused;
  644. return 0;
  645. }
  646. static wasi_errno_t
  647. wasi_path_rename(wasm_exec_env_t exec_env, wasi_fd_t old_fd,
  648. const char *old_path, uint32 old_path_len, wasi_fd_t new_fd,
  649. const char *new_path, uint32 new_path_len)
  650. {
  651. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  652. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  653. struct fd_table *curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  654. if (!wasi_ctx)
  655. return (wasi_errno_t)-1;
  656. return wasmtime_ssp_path_rename(curfds, old_fd, old_path, old_path_len,
  657. new_fd, new_path, new_path_len);
  658. }
  659. static wasi_errno_t
  660. wasi_fd_filestat_get(wasm_exec_env_t exec_env, wasi_fd_t fd,
  661. wasi_filestat_t *filestat)
  662. {
  663. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  664. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  665. struct fd_table *curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  666. if (!wasi_ctx)
  667. return (wasi_errno_t)-1;
  668. if (!validate_native_addr(filestat, sizeof(wasi_filestat_t)))
  669. return (wasi_errno_t)-1;
  670. return wasmtime_ssp_fd_filestat_get(curfds, fd, filestat);
  671. }
  672. static wasi_errno_t
  673. wasi_fd_filestat_set_times(wasm_exec_env_t exec_env, wasi_fd_t fd,
  674. wasi_timestamp_t st_atim, wasi_timestamp_t st_mtim,
  675. wasi_fstflags_t fstflags)
  676. {
  677. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  678. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  679. struct fd_table *curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  680. if (!wasi_ctx)
  681. return (wasi_errno_t)-1;
  682. return wasmtime_ssp_fd_filestat_set_times(curfds, fd, st_atim, st_mtim,
  683. fstflags);
  684. }
  685. static wasi_errno_t
  686. wasi_fd_filestat_set_size(wasm_exec_env_t exec_env, wasi_fd_t fd,
  687. wasi_filesize_t st_size)
  688. {
  689. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  690. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  691. struct fd_table *curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  692. if (!wasi_ctx)
  693. return (wasi_errno_t)-1;
  694. return wasmtime_ssp_fd_filestat_set_size(curfds, fd, st_size);
  695. }
  696. static wasi_errno_t
  697. wasi_path_filestat_get(wasm_exec_env_t exec_env, wasi_fd_t fd,
  698. wasi_lookupflags_t flags, const char *path,
  699. uint32 path_len, wasi_filestat_t *filestat)
  700. {
  701. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  702. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  703. struct fd_table *curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  704. if (!wasi_ctx)
  705. return (wasi_errno_t)-1;
  706. if (!validate_native_addr(filestat, sizeof(wasi_filestat_t)))
  707. return (wasi_errno_t)-1;
  708. return wasmtime_ssp_path_filestat_get(curfds, fd, flags, path, path_len,
  709. filestat);
  710. }
  711. static wasi_errno_t
  712. wasi_path_filestat_set_times(wasm_exec_env_t exec_env, wasi_fd_t fd,
  713. wasi_lookupflags_t flags, const char *path,
  714. uint32 path_len, wasi_timestamp_t st_atim,
  715. wasi_timestamp_t st_mtim, wasi_fstflags_t fstflags)
  716. {
  717. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  718. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  719. struct fd_table *curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  720. if (!wasi_ctx)
  721. return (wasi_errno_t)-1;
  722. return wasmtime_ssp_path_filestat_set_times(
  723. curfds, fd, flags, path, path_len, st_atim, st_mtim, fstflags);
  724. }
  725. static wasi_errno_t
  726. wasi_path_symlink(wasm_exec_env_t exec_env, const char *old_path,
  727. uint32 old_path_len, wasi_fd_t fd, const char *new_path,
  728. uint32 new_path_len)
  729. {
  730. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  731. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  732. struct fd_table *curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  733. struct fd_prestats *prestats = wasi_ctx_get_prestats(module_inst, wasi_ctx);
  734. if (!wasi_ctx)
  735. return (wasi_errno_t)-1;
  736. return wasmtime_ssp_path_symlink(curfds, prestats, old_path, old_path_len,
  737. fd, new_path, new_path_len);
  738. }
  739. static wasi_errno_t
  740. wasi_path_unlink_file(wasm_exec_env_t exec_env, wasi_fd_t fd, const char *path,
  741. uint32 path_len)
  742. {
  743. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  744. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  745. struct fd_table *curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  746. if (!wasi_ctx)
  747. return (wasi_errno_t)-1;
  748. return wasmtime_ssp_path_unlink_file(curfds, fd, path, path_len);
  749. }
  750. static wasi_errno_t
  751. wasi_path_remove_directory(wasm_exec_env_t exec_env, wasi_fd_t fd,
  752. const char *path, uint32 path_len)
  753. {
  754. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  755. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  756. struct fd_table *curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  757. if (!wasi_ctx)
  758. return (wasi_errno_t)-1;
  759. return wasmtime_ssp_path_remove_directory(curfds, fd, path, path_len);
  760. }
  761. #if WASM_ENABLE_THREAD_MGR != 0
  762. static __wasi_timestamp_t
  763. get_timeout_for_poll_oneoff(const wasi_subscription_t *in,
  764. uint32 nsubscriptions)
  765. {
  766. __wasi_timestamp_t timeout = (__wasi_timestamp_t)-1;
  767. uint32 i = 0;
  768. for (i = 0; i < nsubscriptions; ++i) {
  769. const __wasi_subscription_t *s = &in[i];
  770. if (s->u.type == __WASI_EVENTTYPE_CLOCK
  771. && (s->u.u.clock.flags & __WASI_SUBSCRIPTION_CLOCK_ABSTIME) == 0) {
  772. timeout = min(timeout, s->u.u.clock.timeout);
  773. }
  774. }
  775. return timeout;
  776. }
  777. static void
  778. update_clock_subscription_data(wasi_subscription_t *in, uint32 nsubscriptions,
  779. const wasi_timestamp_t new_timeout)
  780. {
  781. uint32 i = 0;
  782. for (i = 0; i < nsubscriptions; ++i) {
  783. __wasi_subscription_t *s = &in[i];
  784. if (s->u.type == __WASI_EVENTTYPE_CLOCK) {
  785. s->u.u.clock.timeout = new_timeout;
  786. }
  787. }
  788. }
  789. static wasi_errno_t
  790. execute_interruptible_poll_oneoff(
  791. #if !defined(WASMTIME_SSP_STATIC_CURFDS)
  792. struct fd_table *curfds,
  793. #endif
  794. const __wasi_subscription_t *in, __wasi_event_t *out, size_t nsubscriptions,
  795. size_t *nevents, wasm_exec_env_t exec_env)
  796. {
  797. if (nsubscriptions == 0) {
  798. *nevents = 0;
  799. return __WASI_ESUCCESS;
  800. }
  801. wasi_errno_t err;
  802. __wasi_timestamp_t elapsed = 0;
  803. bool all_outs_are_type_clock;
  804. uint32 i;
  805. const __wasi_timestamp_t timeout = get_timeout_for_poll_oneoff(
  806. in, nsubscriptions),
  807. time_quant = 1e9;
  808. const uint64 size_to_copy =
  809. nsubscriptions * (uint64)sizeof(wasi_subscription_t);
  810. __wasi_subscription_t *in_copy = NULL;
  811. if (size_to_copy >= UINT32_MAX
  812. || !(in_copy = (__wasi_subscription_t *)wasm_runtime_malloc(
  813. (uint32)size_to_copy))) {
  814. return __WASI_ENOMEM;
  815. }
  816. bh_memcpy_s(in_copy, size_to_copy, in, size_to_copy);
  817. while (timeout == (__wasi_timestamp_t)-1 || elapsed <= timeout) {
  818. /* update timeout for clock subscription events */
  819. update_clock_subscription_data(in_copy, nsubscriptions,
  820. min(time_quant, timeout - elapsed));
  821. err = wasmtime_ssp_poll_oneoff(curfds, in_copy, out, nsubscriptions,
  822. nevents);
  823. elapsed += time_quant;
  824. if (err) {
  825. wasm_runtime_free(in_copy);
  826. return err;
  827. }
  828. if (wasm_cluster_is_thread_terminated(exec_env)) {
  829. wasm_runtime_free(in_copy);
  830. return EINTR;
  831. }
  832. else if (*nevents > 0) {
  833. all_outs_are_type_clock = true;
  834. for (i = 0; i < *nevents; i++) {
  835. if (out[i].type != __WASI_EVENTTYPE_CLOCK) {
  836. all_outs_are_type_clock = false;
  837. break;
  838. }
  839. }
  840. if (!all_outs_are_type_clock) {
  841. wasm_runtime_free(in_copy);
  842. return __WASI_ESUCCESS;
  843. }
  844. }
  845. }
  846. wasm_runtime_free(in_copy);
  847. return __WASI_ESUCCESS;
  848. }
  849. #endif
  850. static wasi_errno_t
  851. wasi_poll_oneoff(wasm_exec_env_t exec_env, const wasi_subscription_t *in,
  852. wasi_event_t *out, uint32 nsubscriptions, uint32 *nevents_app)
  853. {
  854. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  855. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  856. struct fd_table *curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  857. size_t nevents = 0;
  858. wasi_errno_t err;
  859. if (!wasi_ctx)
  860. return (wasi_errno_t)-1;
  861. if (!validate_native_addr((void *)in, sizeof(wasi_subscription_t))
  862. || !validate_native_addr(out, sizeof(wasi_event_t))
  863. || !validate_native_addr(nevents_app, sizeof(uint32)))
  864. return (wasi_errno_t)-1;
  865. #if WASM_ENABLE_THREAD_MGR == 0
  866. err = wasmtime_ssp_poll_oneoff(curfds, in, out, nsubscriptions, &nevents);
  867. #else
  868. err = execute_interruptible_poll_oneoff(curfds, in, out, nsubscriptions,
  869. &nevents, exec_env);
  870. #endif
  871. if (err)
  872. return err;
  873. *nevents_app = (uint32)nevents;
  874. return 0;
  875. }
  876. static void
  877. wasi_proc_exit(wasm_exec_env_t exec_env, wasi_exitcode_t rval)
  878. {
  879. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  880. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  881. /* Here throwing exception is just to let wasm app exit,
  882. the upper layer should clear the exception and return
  883. as normal */
  884. wasm_runtime_set_exception(module_inst, "wasi proc exit");
  885. wasi_ctx->exit_code = rval;
  886. }
  887. static wasi_errno_t
  888. wasi_proc_raise(wasm_exec_env_t exec_env, wasi_signal_t sig)
  889. {
  890. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  891. char buf[32];
  892. snprintf(buf, sizeof(buf), "%s%d", "wasi proc raise ", sig);
  893. wasm_runtime_set_exception(module_inst, buf);
  894. return 0;
  895. }
  896. static wasi_errno_t
  897. wasi_random_get(wasm_exec_env_t exec_env, void *buf, uint32 buf_len)
  898. {
  899. return wasmtime_ssp_random_get(buf, buf_len);
  900. }
  901. static wasi_errno_t
  902. wasi_sock_accept(wasm_exec_env_t exec_env, wasi_fd_t fd, wasi_fdflags_t flags,
  903. wasi_fd_t *fd_new)
  904. {
  905. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  906. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  907. struct fd_table *curfds = NULL;
  908. if (!wasi_ctx)
  909. return __WASI_EACCES;
  910. curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  911. return wasi_ssp_sock_accept(curfds, fd, flags, fd_new);
  912. }
  913. static wasi_errno_t
  914. wasi_sock_addr_local(wasm_exec_env_t exec_env, wasi_fd_t fd,
  915. __wasi_addr_t *addr)
  916. {
  917. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  918. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  919. struct fd_table *curfds = NULL;
  920. if (!wasi_ctx)
  921. return __WASI_EACCES;
  922. if (!validate_native_addr(addr, sizeof(__wasi_addr_t)))
  923. return __WASI_EINVAL;
  924. curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  925. return wasi_ssp_sock_addr_local(curfds, fd, addr);
  926. }
  927. static wasi_errno_t
  928. wasi_sock_addr_remote(wasm_exec_env_t exec_env, wasi_fd_t fd,
  929. __wasi_addr_t *addr)
  930. {
  931. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  932. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  933. struct fd_table *curfds = NULL;
  934. if (!wasi_ctx)
  935. return __WASI_EACCES;
  936. if (!validate_native_addr(addr, sizeof(__wasi_addr_t)))
  937. return __WASI_EINVAL;
  938. curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  939. return wasi_ssp_sock_addr_remote(curfds, fd, addr);
  940. }
  941. static wasi_errno_t
  942. wasi_sock_addr_resolve(wasm_exec_env_t exec_env, const char *host,
  943. const char *service, __wasi_addr_info_hints_t *hints,
  944. __wasi_addr_info_t *addr_info,
  945. __wasi_size_t addr_info_size,
  946. __wasi_size_t *max_info_size)
  947. {
  948. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  949. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  950. struct fd_table *curfds = NULL;
  951. char **ns_lookup_list = NULL;
  952. if (!wasi_ctx)
  953. return __WASI_EACCES;
  954. curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  955. ns_lookup_list = wasi_ctx_get_ns_lookup_list(wasi_ctx);
  956. return wasi_ssp_sock_addr_resolve(curfds, ns_lookup_list, host, service,
  957. hints, addr_info, addr_info_size,
  958. max_info_size);
  959. }
  960. static wasi_errno_t
  961. wasi_sock_bind(wasm_exec_env_t exec_env, wasi_fd_t fd, wasi_addr_t *addr)
  962. {
  963. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  964. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  965. struct fd_table *curfds = NULL;
  966. struct addr_pool *addr_pool = NULL;
  967. if (!wasi_ctx)
  968. return __WASI_EACCES;
  969. curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  970. addr_pool = wasi_ctx_get_addr_pool(module_inst, wasi_ctx);
  971. return wasi_ssp_sock_bind(curfds, addr_pool, fd, addr);
  972. }
  973. static wasi_errno_t
  974. wasi_sock_close(wasm_exec_env_t exec_env, wasi_fd_t fd)
  975. {
  976. return __WASI_ENOSYS;
  977. }
  978. static wasi_errno_t
  979. wasi_sock_connect(wasm_exec_env_t exec_env, wasi_fd_t fd, wasi_addr_t *addr)
  980. {
  981. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  982. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  983. struct fd_table *curfds = NULL;
  984. struct addr_pool *addr_pool = NULL;
  985. if (!wasi_ctx)
  986. return __WASI_EACCES;
  987. curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  988. addr_pool = wasi_ctx_get_addr_pool(module_inst, wasi_ctx);
  989. return wasi_ssp_sock_connect(curfds, addr_pool, fd, addr);
  990. }
  991. static wasi_errno_t
  992. wasi_sock_get_broadcast(wasm_exec_env_t exec_env, wasi_fd_t fd,
  993. bool *is_enabled)
  994. {
  995. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  996. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  997. struct fd_table *curfds = NULL;
  998. if (!wasi_ctx)
  999. return __WASI_EACCES;
  1000. if (!validate_native_addr(is_enabled, sizeof(bool)))
  1001. return __WASI_EINVAL;
  1002. curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  1003. return wasmtime_ssp_sock_get_broadcast(curfds, fd, is_enabled);
  1004. }
  1005. static wasi_errno_t
  1006. wasi_sock_get_keep_alive(wasm_exec_env_t exec_env, wasi_fd_t fd,
  1007. bool *is_enabled)
  1008. {
  1009. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  1010. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  1011. struct fd_table *curfds = NULL;
  1012. if (!wasi_ctx)
  1013. return __WASI_EACCES;
  1014. if (!validate_native_addr(is_enabled, sizeof(bool)))
  1015. return __WASI_EINVAL;
  1016. curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  1017. return wasmtime_ssp_sock_get_keep_alive(curfds, fd, is_enabled);
  1018. }
  1019. static wasi_errno_t
  1020. wasi_sock_get_linger(wasm_exec_env_t exec_env, wasi_fd_t fd, bool *is_enabled,
  1021. int *linger_s)
  1022. {
  1023. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  1024. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  1025. struct fd_table *curfds = NULL;
  1026. if (!wasi_ctx)
  1027. return __WASI_EACCES;
  1028. if (!validate_native_addr(is_enabled, sizeof(bool))
  1029. || !validate_native_addr(linger_s, sizeof(int)))
  1030. return __WASI_EINVAL;
  1031. curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  1032. return wasmtime_ssp_sock_get_linger(curfds, fd, is_enabled, linger_s);
  1033. }
  1034. static wasi_errno_t
  1035. wasi_sock_get_recv_buf_size(wasm_exec_env_t exec_env, wasi_fd_t fd,
  1036. size_t *size)
  1037. {
  1038. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  1039. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  1040. struct fd_table *curfds = NULL;
  1041. if (!wasi_ctx)
  1042. return __WASI_EACCES;
  1043. if (!validate_native_addr(size, sizeof(wasi_size_t)))
  1044. return __WASI_EINVAL;
  1045. curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  1046. return wasmtime_ssp_sock_get_recv_buf_size(curfds, fd, size);
  1047. }
  1048. static wasi_errno_t
  1049. wasi_sock_get_recv_timeout(wasm_exec_env_t exec_env, wasi_fd_t fd,
  1050. uint64_t *timeout_us)
  1051. {
  1052. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  1053. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  1054. struct fd_table *curfds = NULL;
  1055. if (!wasi_ctx)
  1056. return __WASI_EACCES;
  1057. if (!validate_native_addr(timeout_us, sizeof(uint64_t)))
  1058. return __WASI_EINVAL;
  1059. curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  1060. return wasmtime_ssp_sock_get_recv_timeout(curfds, fd, timeout_us);
  1061. }
  1062. static wasi_errno_t
  1063. wasi_sock_get_reuse_addr(wasm_exec_env_t exec_env, wasi_fd_t fd,
  1064. bool *is_enabled)
  1065. {
  1066. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  1067. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  1068. struct fd_table *curfds = NULL;
  1069. if (!wasi_ctx)
  1070. return __WASI_EACCES;
  1071. if (!validate_native_addr(is_enabled, sizeof(bool)))
  1072. return __WASI_EINVAL;
  1073. curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  1074. return wasmtime_ssp_sock_get_reuse_addr(curfds, fd, is_enabled);
  1075. }
  1076. static wasi_errno_t
  1077. wasi_sock_get_reuse_port(wasm_exec_env_t exec_env, wasi_fd_t fd,
  1078. bool *is_enabled)
  1079. {
  1080. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  1081. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  1082. struct fd_table *curfds = NULL;
  1083. if (!wasi_ctx)
  1084. return __WASI_EACCES;
  1085. if (!validate_native_addr(is_enabled, sizeof(bool)))
  1086. return __WASI_EINVAL;
  1087. curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  1088. return wasmtime_ssp_sock_get_reuse_port(curfds, fd, is_enabled);
  1089. }
  1090. static wasi_errno_t
  1091. wasi_sock_get_send_buf_size(wasm_exec_env_t exec_env, wasi_fd_t fd,
  1092. size_t *size)
  1093. {
  1094. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  1095. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  1096. struct fd_table *curfds = NULL;
  1097. if (!wasi_ctx)
  1098. return __WASI_EACCES;
  1099. if (!validate_native_addr(size, sizeof(__wasi_size_t)))
  1100. return __WASI_EINVAL;
  1101. curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  1102. return wasmtime_ssp_sock_get_send_buf_size(curfds, fd, size);
  1103. }
  1104. static wasi_errno_t
  1105. wasi_sock_get_send_timeout(wasm_exec_env_t exec_env, wasi_fd_t fd,
  1106. uint64_t *timeout_us)
  1107. {
  1108. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  1109. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  1110. struct fd_table *curfds = NULL;
  1111. if (!wasi_ctx)
  1112. return __WASI_EACCES;
  1113. if (!validate_native_addr(timeout_us, sizeof(uint64_t)))
  1114. return __WASI_EINVAL;
  1115. curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  1116. return wasmtime_ssp_sock_get_send_timeout(curfds, fd, timeout_us);
  1117. }
  1118. static wasi_errno_t
  1119. wasi_sock_get_tcp_fastopen_connect(wasm_exec_env_t exec_env, wasi_fd_t fd,
  1120. bool *is_enabled)
  1121. {
  1122. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  1123. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  1124. struct fd_table *curfds = NULL;
  1125. if (!wasi_ctx)
  1126. return __WASI_EACCES;
  1127. if (!validate_native_addr(is_enabled, sizeof(bool)))
  1128. return __WASI_EINVAL;
  1129. curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  1130. return wasmtime_ssp_sock_get_tcp_fastopen_connect(curfds, fd, is_enabled);
  1131. }
  1132. static wasi_errno_t
  1133. wasi_sock_get_tcp_no_delay(wasm_exec_env_t exec_env, wasi_fd_t fd,
  1134. bool *is_enabled)
  1135. {
  1136. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  1137. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  1138. struct fd_table *curfds = NULL;
  1139. if (!wasi_ctx)
  1140. return __WASI_EACCES;
  1141. if (!validate_native_addr(is_enabled, sizeof(bool)))
  1142. return __WASI_EINVAL;
  1143. curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  1144. return wasmtime_ssp_sock_get_tcp_no_delay(curfds, fd, is_enabled);
  1145. }
  1146. static wasi_errno_t
  1147. wasi_sock_get_tcp_quick_ack(wasm_exec_env_t exec_env, wasi_fd_t fd,
  1148. bool *is_enabled)
  1149. {
  1150. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  1151. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  1152. struct fd_table *curfds = NULL;
  1153. if (!wasi_ctx)
  1154. return __WASI_EACCES;
  1155. if (!validate_native_addr(is_enabled, sizeof(bool)))
  1156. return __WASI_EINVAL;
  1157. curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  1158. return wasmtime_ssp_sock_get_tcp_quick_ack(curfds, fd, is_enabled);
  1159. }
  1160. static wasi_errno_t
  1161. wasi_sock_get_tcp_keep_idle(wasm_exec_env_t exec_env, wasi_fd_t fd,
  1162. uint32_t *time_s)
  1163. {
  1164. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  1165. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  1166. struct fd_table *curfds = NULL;
  1167. if (!wasi_ctx)
  1168. return __WASI_EACCES;
  1169. if (!validate_native_addr(time_s, sizeof(uint32_t)))
  1170. return __WASI_EINVAL;
  1171. curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  1172. return wasmtime_ssp_sock_get_tcp_keep_idle(curfds, fd, time_s);
  1173. }
  1174. static wasi_errno_t
  1175. wasi_sock_get_tcp_keep_intvl(wasm_exec_env_t exec_env, wasi_fd_t fd,
  1176. uint32_t *time_s)
  1177. {
  1178. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  1179. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  1180. struct fd_table *curfds = NULL;
  1181. if (!wasi_ctx)
  1182. return __WASI_EACCES;
  1183. if (!validate_native_addr(time_s, sizeof(uint32_t)))
  1184. return __WASI_EINVAL;
  1185. curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  1186. return wasmtime_ssp_sock_get_tcp_keep_intvl(curfds, fd, time_s);
  1187. }
  1188. static wasi_errno_t
  1189. wasi_sock_get_ip_multicast_loop(wasm_exec_env_t exec_env, wasi_fd_t fd,
  1190. bool ipv6, bool *is_enabled)
  1191. {
  1192. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  1193. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  1194. struct fd_table *curfds = NULL;
  1195. if (!wasi_ctx)
  1196. return __WASI_EACCES;
  1197. if (!validate_native_addr(is_enabled, sizeof(bool)))
  1198. return __WASI_EINVAL;
  1199. curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  1200. return wasmtime_ssp_sock_get_ip_multicast_loop(curfds, fd, ipv6,
  1201. is_enabled);
  1202. }
  1203. static wasi_errno_t
  1204. wasi_sock_get_ip_ttl(wasm_exec_env_t exec_env, wasi_fd_t fd, uint8_t *ttl_s)
  1205. {
  1206. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  1207. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  1208. struct fd_table *curfds = NULL;
  1209. if (!wasi_ctx)
  1210. return __WASI_EACCES;
  1211. if (!validate_native_addr(ttl_s, sizeof(uint8_t)))
  1212. return __WASI_EINVAL;
  1213. curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  1214. return wasmtime_ssp_sock_get_ip_ttl(curfds, fd, ttl_s);
  1215. }
  1216. static wasi_errno_t
  1217. wasi_sock_get_ip_multicast_ttl(wasm_exec_env_t exec_env, wasi_fd_t fd,
  1218. uint8_t *ttl_s)
  1219. {
  1220. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  1221. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  1222. struct fd_table *curfds = NULL;
  1223. if (!wasi_ctx)
  1224. return __WASI_EACCES;
  1225. if (!validate_native_addr(ttl_s, sizeof(uint8_t)))
  1226. return __WASI_EINVAL;
  1227. curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  1228. return wasmtime_ssp_sock_get_ip_multicast_ttl(curfds, fd, ttl_s);
  1229. }
  1230. static wasi_errno_t
  1231. wasi_sock_get_ipv6_only(wasm_exec_env_t exec_env, wasi_fd_t fd,
  1232. bool *is_enabled)
  1233. {
  1234. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  1235. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  1236. struct fd_table *curfds = NULL;
  1237. if (!wasi_ctx)
  1238. return __WASI_EACCES;
  1239. if (!validate_native_addr(is_enabled, sizeof(bool)))
  1240. return __WASI_EINVAL;
  1241. curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  1242. return wasmtime_ssp_sock_get_ipv6_only(curfds, fd, is_enabled);
  1243. }
  1244. static wasi_errno_t
  1245. wasi_sock_listen(wasm_exec_env_t exec_env, wasi_fd_t fd, uint32 backlog)
  1246. {
  1247. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  1248. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  1249. struct fd_table *curfds = NULL;
  1250. if (!wasi_ctx)
  1251. return __WASI_EACCES;
  1252. curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  1253. return wasi_ssp_sock_listen(curfds, fd, backlog);
  1254. }
  1255. static wasi_errno_t
  1256. wasi_sock_open(wasm_exec_env_t exec_env, wasi_fd_t poolfd,
  1257. wasi_address_family_t af, wasi_sock_type_t socktype,
  1258. wasi_fd_t *sockfd)
  1259. {
  1260. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  1261. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  1262. struct fd_table *curfds = NULL;
  1263. if (!wasi_ctx)
  1264. return __WASI_EACCES;
  1265. curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  1266. return wasi_ssp_sock_open(curfds, poolfd, af, socktype, sockfd);
  1267. }
  1268. static wasi_errno_t
  1269. wasi_sock_set_broadcast(wasm_exec_env_t exec_env, wasi_fd_t fd, bool is_enabled)
  1270. {
  1271. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  1272. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  1273. struct fd_table *curfds = NULL;
  1274. if (!wasi_ctx)
  1275. return __WASI_EACCES;
  1276. curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  1277. return wasmtime_ssp_sock_set_broadcast(curfds, fd, is_enabled);
  1278. }
  1279. static wasi_errno_t
  1280. wasi_sock_set_keep_alive(wasm_exec_env_t exec_env, wasi_fd_t fd,
  1281. bool is_enabled)
  1282. {
  1283. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  1284. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  1285. struct fd_table *curfds = NULL;
  1286. if (!wasi_ctx)
  1287. return __WASI_EACCES;
  1288. curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  1289. return wasmtime_ssp_sock_set_keep_alive(curfds, fd, is_enabled);
  1290. }
  1291. static wasi_errno_t
  1292. wasi_sock_set_linger(wasm_exec_env_t exec_env, wasi_fd_t fd, bool is_enabled,
  1293. int linger_s)
  1294. {
  1295. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  1296. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  1297. struct fd_table *curfds = NULL;
  1298. if (!wasi_ctx)
  1299. return __WASI_EACCES;
  1300. curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  1301. return wasmtime_ssp_sock_set_linger(curfds, fd, is_enabled, linger_s);
  1302. }
  1303. static wasi_errno_t
  1304. wasi_sock_set_recv_buf_size(wasm_exec_env_t exec_env, wasi_fd_t fd, size_t size)
  1305. {
  1306. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  1307. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  1308. struct fd_table *curfds = NULL;
  1309. if (!wasi_ctx)
  1310. return __WASI_EACCES;
  1311. curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  1312. return wasmtime_ssp_sock_set_recv_buf_size(curfds, fd, size);
  1313. }
  1314. static wasi_errno_t
  1315. wasi_sock_set_recv_timeout(wasm_exec_env_t exec_env, wasi_fd_t fd,
  1316. uint64_t timeout_us)
  1317. {
  1318. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  1319. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  1320. struct fd_table *curfds = NULL;
  1321. if (!wasi_ctx)
  1322. return __WASI_EACCES;
  1323. curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  1324. return wasmtime_ssp_sock_set_recv_timeout(curfds, fd, timeout_us);
  1325. }
  1326. static wasi_errno_t
  1327. wasi_sock_set_reuse_addr(wasm_exec_env_t exec_env, wasi_fd_t fd,
  1328. bool is_enabled)
  1329. {
  1330. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  1331. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  1332. struct fd_table *curfds = NULL;
  1333. if (!wasi_ctx)
  1334. return __WASI_EACCES;
  1335. curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  1336. return wasmtime_ssp_sock_set_reuse_addr(curfds, fd, is_enabled);
  1337. }
  1338. static wasi_errno_t
  1339. wasi_sock_set_reuse_port(wasm_exec_env_t exec_env, wasi_fd_t fd,
  1340. bool is_enabled)
  1341. {
  1342. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  1343. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  1344. struct fd_table *curfds = NULL;
  1345. if (!wasi_ctx)
  1346. return __WASI_EACCES;
  1347. curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  1348. return wasmtime_ssp_sock_set_reuse_port(curfds, fd, is_enabled);
  1349. }
  1350. static wasi_errno_t
  1351. wasi_sock_set_send_buf_size(wasm_exec_env_t exec_env, wasi_fd_t fd, size_t size)
  1352. {
  1353. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  1354. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  1355. struct fd_table *curfds = NULL;
  1356. if (!wasi_ctx)
  1357. return __WASI_EACCES;
  1358. curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  1359. return wasmtime_ssp_sock_set_send_buf_size(curfds, fd, size);
  1360. }
  1361. static wasi_errno_t
  1362. wasi_sock_set_send_timeout(wasm_exec_env_t exec_env, wasi_fd_t fd,
  1363. uint64_t timeout_us)
  1364. {
  1365. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  1366. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  1367. struct fd_table *curfds = NULL;
  1368. if (!wasi_ctx)
  1369. return __WASI_EACCES;
  1370. curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  1371. return wasmtime_ssp_sock_set_send_timeout(curfds, fd, timeout_us);
  1372. }
  1373. static wasi_errno_t
  1374. wasi_sock_set_tcp_fastopen_connect(wasm_exec_env_t exec_env, wasi_fd_t fd,
  1375. bool is_enabled)
  1376. {
  1377. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  1378. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  1379. struct fd_table *curfds = NULL;
  1380. if (!wasi_ctx)
  1381. return __WASI_EACCES;
  1382. curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  1383. return wasmtime_ssp_sock_set_tcp_fastopen_connect(curfds, fd, is_enabled);
  1384. }
  1385. static wasi_errno_t
  1386. wasi_sock_set_tcp_no_delay(wasm_exec_env_t exec_env, wasi_fd_t fd,
  1387. bool is_enabled)
  1388. {
  1389. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  1390. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  1391. struct fd_table *curfds = NULL;
  1392. if (!wasi_ctx)
  1393. return __WASI_EACCES;
  1394. curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  1395. return wasmtime_ssp_sock_set_tcp_no_delay(curfds, fd, is_enabled);
  1396. }
  1397. static wasi_errno_t
  1398. wasi_sock_set_tcp_quick_ack(wasm_exec_env_t exec_env, wasi_fd_t fd,
  1399. bool is_enabled)
  1400. {
  1401. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  1402. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  1403. struct fd_table *curfds = NULL;
  1404. if (!wasi_ctx)
  1405. return __WASI_EACCES;
  1406. curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  1407. return wasmtime_ssp_sock_set_tcp_quick_ack(curfds, fd, is_enabled);
  1408. }
  1409. static wasi_errno_t
  1410. wasi_sock_set_tcp_keep_idle(wasm_exec_env_t exec_env, wasi_fd_t fd,
  1411. uint32_t time_s)
  1412. {
  1413. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  1414. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  1415. struct fd_table *curfds = NULL;
  1416. if (!wasi_ctx)
  1417. return __WASI_EACCES;
  1418. curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  1419. return wasmtime_ssp_sock_set_tcp_keep_idle(curfds, fd, time_s);
  1420. }
  1421. static wasi_errno_t
  1422. wasi_sock_set_tcp_keep_intvl(wasm_exec_env_t exec_env, wasi_fd_t fd,
  1423. uint32_t time_s)
  1424. {
  1425. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  1426. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  1427. struct fd_table *curfds = NULL;
  1428. if (!wasi_ctx)
  1429. return __WASI_EACCES;
  1430. curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  1431. return wasmtime_ssp_sock_set_tcp_keep_intvl(curfds, fd, time_s);
  1432. }
  1433. static wasi_errno_t
  1434. wasi_sock_set_ip_multicast_loop(wasm_exec_env_t exec_env, wasi_fd_t fd,
  1435. bool ipv6, bool is_enabled)
  1436. {
  1437. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  1438. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  1439. struct fd_table *curfds = NULL;
  1440. if (!wasi_ctx)
  1441. return __WASI_EACCES;
  1442. curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  1443. return wasmtime_ssp_sock_set_ip_multicast_loop(curfds, fd, ipv6,
  1444. is_enabled);
  1445. }
  1446. static wasi_errno_t
  1447. wasi_sock_set_ip_add_membership(wasm_exec_env_t exec_env, wasi_fd_t fd,
  1448. __wasi_addr_ip_t *imr_multiaddr,
  1449. uint32_t imr_interface)
  1450. {
  1451. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  1452. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  1453. struct fd_table *curfds = NULL;
  1454. if (!wasi_ctx)
  1455. return __WASI_EACCES;
  1456. if (!validate_native_addr(imr_multiaddr, sizeof(__wasi_addr_ip_t)))
  1457. return __WASI_EINVAL;
  1458. curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  1459. return wasmtime_ssp_sock_set_ip_add_membership(curfds, fd, imr_multiaddr,
  1460. imr_interface);
  1461. }
  1462. static wasi_errno_t
  1463. wasi_sock_set_ip_drop_membership(wasm_exec_env_t exec_env, wasi_fd_t fd,
  1464. __wasi_addr_ip_t *imr_multiaddr,
  1465. uint32_t imr_interface)
  1466. {
  1467. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  1468. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  1469. struct fd_table *curfds = NULL;
  1470. if (!wasi_ctx)
  1471. return __WASI_EACCES;
  1472. if (!validate_native_addr(imr_multiaddr, sizeof(__wasi_addr_ip_t)))
  1473. return __WASI_EINVAL;
  1474. curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  1475. return wasmtime_ssp_sock_set_ip_drop_membership(curfds, fd, imr_multiaddr,
  1476. imr_interface);
  1477. }
  1478. static wasi_errno_t
  1479. wasi_sock_set_ip_ttl(wasm_exec_env_t exec_env, wasi_fd_t fd, uint8_t ttl_s)
  1480. {
  1481. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  1482. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  1483. struct fd_table *curfds = NULL;
  1484. if (!wasi_ctx)
  1485. return __WASI_EACCES;
  1486. curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  1487. return wasmtime_ssp_sock_set_ip_ttl(curfds, fd, ttl_s);
  1488. }
  1489. static wasi_errno_t
  1490. wasi_sock_set_ip_multicast_ttl(wasm_exec_env_t exec_env, wasi_fd_t fd,
  1491. uint8_t ttl_s)
  1492. {
  1493. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  1494. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  1495. struct fd_table *curfds = NULL;
  1496. if (!wasi_ctx)
  1497. return __WASI_EACCES;
  1498. curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  1499. return wasmtime_ssp_sock_set_ip_multicast_ttl(curfds, fd, ttl_s);
  1500. }
  1501. static wasi_errno_t
  1502. wasi_sock_set_ipv6_only(wasm_exec_env_t exec_env, wasi_fd_t fd, bool is_enabled)
  1503. {
  1504. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  1505. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  1506. struct fd_table *curfds = NULL;
  1507. if (!wasi_ctx)
  1508. return __WASI_EACCES;
  1509. curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  1510. return wasmtime_ssp_sock_set_ipv6_only(curfds, fd, is_enabled);
  1511. }
  1512. static wasi_errno_t
  1513. allocate_iovec_app_buffer(wasm_module_inst_t module_inst,
  1514. const iovec_app_t *data, uint32 data_len,
  1515. uint8 **buf_ptr, uint64 *buf_len)
  1516. {
  1517. uint64 total_size = 0;
  1518. uint32 i;
  1519. uint8 *buf_begin = NULL;
  1520. if (data_len == 0) {
  1521. return __WASI_EINVAL;
  1522. }
  1523. total_size = sizeof(iovec_app_t) * (uint64)data_len;
  1524. if (total_size >= UINT32_MAX
  1525. || !validate_native_addr((void *)data, (uint32)total_size))
  1526. return __WASI_EINVAL;
  1527. for (total_size = 0, i = 0; i < data_len; i++, data++) {
  1528. total_size += data->buf_len;
  1529. }
  1530. if (total_size == 0) {
  1531. return __WASI_EINVAL;
  1532. }
  1533. if (total_size >= UINT32_MAX
  1534. || !(buf_begin = wasm_runtime_malloc((uint32)total_size))) {
  1535. return __WASI_ENOMEM;
  1536. }
  1537. *buf_len = total_size;
  1538. *buf_ptr = buf_begin;
  1539. return __WASI_ESUCCESS;
  1540. }
  1541. static wasi_errno_t
  1542. copy_buffer_to_iovec_app(wasm_module_inst_t module_inst, uint8 *buf_begin,
  1543. uint32 buf_size, iovec_app_t *data, uint32 data_len,
  1544. uint32 size_to_copy)
  1545. {
  1546. uint8 *buf = buf_begin;
  1547. uint32 i;
  1548. uint32 size_to_copy_into_iovec;
  1549. if (buf_size < size_to_copy) {
  1550. return __WASI_EINVAL;
  1551. }
  1552. for (i = 0; i < data_len; data++, i++) {
  1553. char *native_addr;
  1554. if (!validate_app_addr(data->buf_offset, data->buf_len)) {
  1555. return __WASI_EINVAL;
  1556. }
  1557. if (buf >= buf_begin + buf_size
  1558. || buf + data->buf_len < buf /* integer overflow */
  1559. || buf + data->buf_len > buf_begin + buf_size
  1560. || size_to_copy == 0) {
  1561. break;
  1562. }
  1563. /**
  1564. * If our app buffer size is smaller than the amount to be copied,
  1565. * only copy the amount in the app buffer. Otherwise, we fill the iovec
  1566. * buffer and reduce size to copy on the next iteration
  1567. */
  1568. size_to_copy_into_iovec = min(data->buf_len, size_to_copy);
  1569. native_addr = (void *)addr_app_to_native(data->buf_offset);
  1570. bh_memcpy_s(native_addr, size_to_copy_into_iovec, buf,
  1571. size_to_copy_into_iovec);
  1572. buf += size_to_copy_into_iovec;
  1573. size_to_copy -= size_to_copy_into_iovec;
  1574. }
  1575. return __WASI_ESUCCESS;
  1576. }
  1577. static wasi_errno_t
  1578. wasi_sock_recv_from(wasm_exec_env_t exec_env, wasi_fd_t sock,
  1579. iovec_app_t *ri_data, uint32 ri_data_len,
  1580. wasi_riflags_t ri_flags, __wasi_addr_t *src_addr,
  1581. uint32 *ro_data_len)
  1582. {
  1583. /**
  1584. * ri_data_len is the length of a list of iovec_app_t, which head is
  1585. * ri_data. ro_data_len is the number of bytes received
  1586. **/
  1587. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  1588. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  1589. struct fd_table *curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  1590. uint64 total_size;
  1591. uint8 *buf_begin = NULL;
  1592. wasi_errno_t err;
  1593. size_t recv_bytes = 0;
  1594. if (!wasi_ctx) {
  1595. return __WASI_EINVAL;
  1596. }
  1597. if (!validate_native_addr(ro_data_len, (uint32)sizeof(uint32)))
  1598. return __WASI_EINVAL;
  1599. err = allocate_iovec_app_buffer(module_inst, ri_data, ri_data_len,
  1600. &buf_begin, &total_size);
  1601. if (err != __WASI_ESUCCESS) {
  1602. goto fail;
  1603. }
  1604. memset(buf_begin, 0, total_size);
  1605. *ro_data_len = 0;
  1606. err = wasmtime_ssp_sock_recv_from(curfds, sock, buf_begin, total_size,
  1607. ri_flags, src_addr, &recv_bytes);
  1608. if (err != __WASI_ESUCCESS) {
  1609. goto fail;
  1610. }
  1611. *ro_data_len = (uint32)recv_bytes;
  1612. err = copy_buffer_to_iovec_app(module_inst, buf_begin, (uint32)total_size,
  1613. ri_data, ri_data_len, (uint32)recv_bytes);
  1614. fail:
  1615. if (buf_begin) {
  1616. wasm_runtime_free(buf_begin);
  1617. }
  1618. return err;
  1619. }
  1620. static wasi_errno_t
  1621. wasi_sock_recv(wasm_exec_env_t exec_env, wasi_fd_t sock, iovec_app_t *ri_data,
  1622. uint32 ri_data_len, wasi_riflags_t ri_flags, uint32 *ro_data_len,
  1623. wasi_roflags_t *ro_flags)
  1624. {
  1625. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  1626. __wasi_addr_t src_addr;
  1627. wasi_errno_t error;
  1628. if (!validate_native_addr(ro_flags, (uint32)sizeof(wasi_roflags_t)))
  1629. return __WASI_EINVAL;
  1630. error = wasi_sock_recv_from(exec_env, sock, ri_data, ri_data_len, ri_flags,
  1631. &src_addr, ro_data_len);
  1632. *ro_flags = ri_flags;
  1633. return error;
  1634. }
  1635. static wasi_errno_t
  1636. convert_iovec_app_to_buffer(wasm_module_inst_t module_inst,
  1637. const iovec_app_t *si_data, uint32 si_data_len,
  1638. uint8 **buf_ptr, uint64 *buf_len)
  1639. {
  1640. uint32 i;
  1641. const iovec_app_t *si_data_orig = si_data;
  1642. uint8 *buf = NULL;
  1643. wasi_errno_t error;
  1644. error = allocate_iovec_app_buffer(module_inst, si_data, si_data_len,
  1645. buf_ptr, buf_len);
  1646. if (error != __WASI_ESUCCESS) {
  1647. return error;
  1648. }
  1649. buf = *buf_ptr;
  1650. si_data = si_data_orig;
  1651. for (i = 0; i < si_data_len; i++, si_data++) {
  1652. char *native_addr;
  1653. if (!validate_app_addr(si_data->buf_offset, si_data->buf_len)) {
  1654. wasm_runtime_free(*buf_ptr);
  1655. return __WASI_EINVAL;
  1656. }
  1657. native_addr = (char *)addr_app_to_native(si_data->buf_offset);
  1658. bh_memcpy_s(buf, si_data->buf_len, native_addr, si_data->buf_len);
  1659. buf += si_data->buf_len;
  1660. }
  1661. return __WASI_ESUCCESS;
  1662. }
  1663. static wasi_errno_t
  1664. wasi_sock_send(wasm_exec_env_t exec_env, wasi_fd_t sock,
  1665. const iovec_app_t *si_data, uint32 si_data_len,
  1666. wasi_siflags_t si_flags, uint32 *so_data_len)
  1667. {
  1668. /**
  1669. * si_data_len is the length of a list of iovec_app_t, which head is
  1670. * si_data. so_data_len is the number of bytes sent
  1671. **/
  1672. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  1673. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  1674. struct fd_table *curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  1675. uint64 buf_size = 0;
  1676. uint8 *buf = NULL;
  1677. wasi_errno_t err;
  1678. size_t send_bytes = 0;
  1679. if (!wasi_ctx) {
  1680. return __WASI_EINVAL;
  1681. }
  1682. if (!validate_native_addr(so_data_len, sizeof(uint32)))
  1683. return __WASI_EINVAL;
  1684. err = convert_iovec_app_to_buffer(module_inst, si_data, si_data_len, &buf,
  1685. &buf_size);
  1686. if (err != __WASI_ESUCCESS)
  1687. return err;
  1688. *so_data_len = 0;
  1689. err = wasmtime_ssp_sock_send(curfds, sock, buf, buf_size, &send_bytes);
  1690. *so_data_len = (uint32)send_bytes;
  1691. wasm_runtime_free(buf);
  1692. return err;
  1693. }
  1694. static wasi_errno_t
  1695. wasi_sock_send_to(wasm_exec_env_t exec_env, wasi_fd_t sock,
  1696. const iovec_app_t *si_data, uint32 si_data_len,
  1697. wasi_siflags_t si_flags, const __wasi_addr_t *dest_addr,
  1698. uint32 *so_data_len)
  1699. {
  1700. /**
  1701. * si_data_len is the length of a list of iovec_app_t, which head is
  1702. * si_data. so_data_len is the number of bytes sent
  1703. **/
  1704. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  1705. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  1706. struct fd_table *curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  1707. uint64 buf_size = 0;
  1708. uint8 *buf = NULL;
  1709. wasi_errno_t err;
  1710. size_t send_bytes = 0;
  1711. struct addr_pool *addr_pool = wasi_ctx_get_addr_pool(module_inst, wasi_ctx);
  1712. if (!wasi_ctx) {
  1713. return __WASI_EINVAL;
  1714. }
  1715. if (!validate_native_addr(so_data_len, sizeof(uint32)))
  1716. return __WASI_EINVAL;
  1717. err = convert_iovec_app_to_buffer(module_inst, si_data, si_data_len, &buf,
  1718. &buf_size);
  1719. if (err != __WASI_ESUCCESS)
  1720. return err;
  1721. *so_data_len = 0;
  1722. err = wasmtime_ssp_sock_send_to(curfds, addr_pool, sock, buf, buf_size,
  1723. si_flags, dest_addr, &send_bytes);
  1724. *so_data_len = (uint32)send_bytes;
  1725. wasm_runtime_free(buf);
  1726. return err;
  1727. }
  1728. static wasi_errno_t
  1729. wasi_sock_shutdown(wasm_exec_env_t exec_env, wasi_fd_t sock, wasi_sdflags_t how)
  1730. {
  1731. wasm_module_inst_t module_inst = get_module_inst(exec_env);
  1732. wasi_ctx_t wasi_ctx = get_wasi_ctx(module_inst);
  1733. struct fd_table *curfds = wasi_ctx_get_curfds(module_inst, wasi_ctx);
  1734. if (!wasi_ctx)
  1735. return __WASI_EINVAL;
  1736. return wasmtime_ssp_sock_shutdown(curfds, sock);
  1737. }
  1738. static wasi_errno_t
  1739. wasi_sched_yield(wasm_exec_env_t exec_env)
  1740. {
  1741. return wasmtime_ssp_sched_yield();
  1742. }
  1743. /* clang-format off */
  1744. #define REG_NATIVE_FUNC(func_name, signature) \
  1745. { #func_name, wasi_##func_name, signature, NULL }
  1746. /* clang-format on */
  1747. static NativeSymbol native_symbols_libc_wasi[] = {
  1748. REG_NATIVE_FUNC(args_get, "(**)i"),
  1749. REG_NATIVE_FUNC(args_sizes_get, "(**)i"),
  1750. REG_NATIVE_FUNC(clock_res_get, "(i*)i"),
  1751. REG_NATIVE_FUNC(clock_time_get, "(iI*)i"),
  1752. REG_NATIVE_FUNC(environ_get, "(**)i"),
  1753. REG_NATIVE_FUNC(environ_sizes_get, "(**)i"),
  1754. REG_NATIVE_FUNC(fd_prestat_get, "(i*)i"),
  1755. REG_NATIVE_FUNC(fd_prestat_dir_name, "(i*~)i"),
  1756. REG_NATIVE_FUNC(fd_close, "(i)i"),
  1757. REG_NATIVE_FUNC(fd_datasync, "(i)i"),
  1758. REG_NATIVE_FUNC(fd_pread, "(i*iI*)i"),
  1759. REG_NATIVE_FUNC(fd_pwrite, "(i*iI*)i"),
  1760. REG_NATIVE_FUNC(fd_read, "(i*i*)i"),
  1761. REG_NATIVE_FUNC(fd_renumber, "(ii)i"),
  1762. REG_NATIVE_FUNC(fd_seek, "(iIi*)i"),
  1763. REG_NATIVE_FUNC(fd_tell, "(i*)i"),
  1764. REG_NATIVE_FUNC(fd_fdstat_get, "(i*)i"),
  1765. REG_NATIVE_FUNC(fd_fdstat_set_flags, "(ii)i"),
  1766. REG_NATIVE_FUNC(fd_fdstat_set_rights, "(iII)i"),
  1767. REG_NATIVE_FUNC(fd_sync, "(i)i"),
  1768. REG_NATIVE_FUNC(fd_write, "(i*i*)i"),
  1769. REG_NATIVE_FUNC(fd_advise, "(iIIi)i"),
  1770. REG_NATIVE_FUNC(fd_allocate, "(iII)i"),
  1771. REG_NATIVE_FUNC(path_create_directory, "(i*~)i"),
  1772. REG_NATIVE_FUNC(path_link, "(ii*~i*~)i"),
  1773. REG_NATIVE_FUNC(path_open, "(ii*~iIIi*)i"),
  1774. REG_NATIVE_FUNC(fd_readdir, "(i*~I*)i"),
  1775. REG_NATIVE_FUNC(path_readlink, "(i*~*~*)i"),
  1776. REG_NATIVE_FUNC(path_rename, "(i*~i*~)i"),
  1777. REG_NATIVE_FUNC(fd_filestat_get, "(i*)i"),
  1778. REG_NATIVE_FUNC(fd_filestat_set_times, "(iIIi)i"),
  1779. REG_NATIVE_FUNC(fd_filestat_set_size, "(iI)i"),
  1780. REG_NATIVE_FUNC(path_filestat_get, "(ii*~*)i"),
  1781. REG_NATIVE_FUNC(path_filestat_set_times, "(ii*~IIi)i"),
  1782. REG_NATIVE_FUNC(path_symlink, "(*~i*~)i"),
  1783. REG_NATIVE_FUNC(path_unlink_file, "(i*~)i"),
  1784. REG_NATIVE_FUNC(path_remove_directory, "(i*~)i"),
  1785. REG_NATIVE_FUNC(poll_oneoff, "(**i*)i"),
  1786. REG_NATIVE_FUNC(proc_exit, "(i)"),
  1787. REG_NATIVE_FUNC(proc_raise, "(i)i"),
  1788. REG_NATIVE_FUNC(random_get, "(*~)i"),
  1789. REG_NATIVE_FUNC(sock_accept, "(ii*)i"),
  1790. REG_NATIVE_FUNC(sock_addr_local, "(i*)i"),
  1791. REG_NATIVE_FUNC(sock_addr_remote, "(i*)i"),
  1792. REG_NATIVE_FUNC(sock_addr_resolve, "($$**i*)i"),
  1793. REG_NATIVE_FUNC(sock_bind, "(i*)i"),
  1794. REG_NATIVE_FUNC(sock_close, "(i)i"),
  1795. REG_NATIVE_FUNC(sock_connect, "(i*)i"),
  1796. REG_NATIVE_FUNC(sock_get_broadcast, "(i*)i"),
  1797. REG_NATIVE_FUNC(sock_get_keep_alive, "(i*)i"),
  1798. REG_NATIVE_FUNC(sock_get_linger, "(i**)i"),
  1799. REG_NATIVE_FUNC(sock_get_recv_buf_size, "(i*)i"),
  1800. REG_NATIVE_FUNC(sock_get_recv_timeout, "(i*)i"),
  1801. REG_NATIVE_FUNC(sock_get_reuse_addr, "(i*)i"),
  1802. REG_NATIVE_FUNC(sock_get_reuse_port, "(i*)i"),
  1803. REG_NATIVE_FUNC(sock_get_send_buf_size, "(i*)i"),
  1804. REG_NATIVE_FUNC(sock_get_send_timeout, "(i*)i"),
  1805. REG_NATIVE_FUNC(sock_get_tcp_fastopen_connect, "(i*)i"),
  1806. REG_NATIVE_FUNC(sock_get_tcp_keep_idle, "(i*)i"),
  1807. REG_NATIVE_FUNC(sock_get_tcp_keep_intvl, "(i*)i"),
  1808. REG_NATIVE_FUNC(sock_get_tcp_no_delay, "(i*)i"),
  1809. REG_NATIVE_FUNC(sock_get_tcp_quick_ack, "(i*)i"),
  1810. REG_NATIVE_FUNC(sock_get_ip_multicast_loop, "(ii*)i"),
  1811. REG_NATIVE_FUNC(sock_get_ip_multicast_ttl, "(i*)i"),
  1812. REG_NATIVE_FUNC(sock_get_ip_ttl, "(i*)i"),
  1813. REG_NATIVE_FUNC(sock_get_ipv6_only, "(i*)i"),
  1814. REG_NATIVE_FUNC(sock_listen, "(ii)i"),
  1815. REG_NATIVE_FUNC(sock_open, "(iii*)i"),
  1816. REG_NATIVE_FUNC(sock_recv, "(i*ii**)i"),
  1817. REG_NATIVE_FUNC(sock_recv_from, "(i*ii**)i"),
  1818. REG_NATIVE_FUNC(sock_send, "(i*ii*)i"),
  1819. REG_NATIVE_FUNC(sock_send_to, "(i*ii**)i"),
  1820. REG_NATIVE_FUNC(sock_set_broadcast, "(ii)i"),
  1821. REG_NATIVE_FUNC(sock_set_keep_alive, "(ii)i"),
  1822. REG_NATIVE_FUNC(sock_set_linger, "(iii)i"),
  1823. REG_NATIVE_FUNC(sock_set_recv_buf_size, "(ii)i"),
  1824. REG_NATIVE_FUNC(sock_set_recv_timeout, "(iI)i"),
  1825. REG_NATIVE_FUNC(sock_set_reuse_addr, "(ii)i"),
  1826. REG_NATIVE_FUNC(sock_set_reuse_port, "(ii)i"),
  1827. REG_NATIVE_FUNC(sock_set_send_buf_size, "(ii)i"),
  1828. REG_NATIVE_FUNC(sock_set_send_timeout, "(iI)i"),
  1829. REG_NATIVE_FUNC(sock_set_tcp_fastopen_connect, "(ii)i"),
  1830. REG_NATIVE_FUNC(sock_set_tcp_keep_idle, "(ii)i"),
  1831. REG_NATIVE_FUNC(sock_set_tcp_keep_intvl, "(ii)i"),
  1832. REG_NATIVE_FUNC(sock_set_tcp_no_delay, "(ii)i"),
  1833. REG_NATIVE_FUNC(sock_set_tcp_quick_ack, "(ii)i"),
  1834. REG_NATIVE_FUNC(sock_set_ip_multicast_loop, "(iii)i"),
  1835. REG_NATIVE_FUNC(sock_set_ip_multicast_ttl, "(ii)i"),
  1836. REG_NATIVE_FUNC(sock_set_ip_add_membership, "(i*i)i"),
  1837. REG_NATIVE_FUNC(sock_set_ip_drop_membership, "(i*i)i"),
  1838. REG_NATIVE_FUNC(sock_set_ip_ttl, "(ii)i"),
  1839. REG_NATIVE_FUNC(sock_set_ipv6_only, "(ii)i"),
  1840. REG_NATIVE_FUNC(sock_shutdown, "(ii)i"),
  1841. REG_NATIVE_FUNC(sched_yield, "()i"),
  1842. };
  1843. uint32
  1844. get_libc_wasi_export_apis(NativeSymbol **p_libc_wasi_apis)
  1845. {
  1846. *p_libc_wasi_apis = native_symbols_libc_wasi;
  1847. return sizeof(native_symbols_libc_wasi) / sizeof(NativeSymbol);
  1848. }