libc_wasi_wrapper.c 72 KB

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