libc_wasi_wrapper.c 72 KB

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