| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342 |
- /*
- * Copyright (C) 2019 Intel Corporation. All rights reserved.
- * SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
- */
- #include "aot_compiler.h"
- #include "aot_emit_compare.h"
- #include "aot_emit_conversion.h"
- #include "aot_emit_memory.h"
- #include "aot_emit_variable.h"
- #include "aot_emit_const.h"
- #include "aot_emit_exception.h"
- #include "aot_emit_numberic.h"
- #include "aot_emit_control.h"
- #include "aot_emit_function.h"
- #include "aot_emit_parametric.h"
- #include "aot_emit_table.h"
- #include "simd/simd_access_lanes.h"
- #include "simd/simd_bitmask_extracts.h"
- #include "simd/simd_bit_shifts.h"
- #include "simd/simd_bitwise_ops.h"
- #include "simd/simd_bool_reductions.h"
- #include "simd/simd_comparisons.h"
- #include "simd/simd_construct_values.h"
- #include "simd/simd_conversions.h"
- #include "simd/simd_floating_point.h"
- #include "simd/simd_int_arith.h"
- #include "simd/simd_load_store.h"
- #include "simd/simd_sat_int_arith.h"
- #include "../aot/aot_runtime.h"
- #include "../interpreter/wasm_opcode.h"
- #include <errno.h>
- #define CHECK_BUF(buf, buf_end, length) do { \
- if (buf + length > buf_end) { \
- aot_set_last_error("read leb failed: unexpected end."); \
- return false; \
- } \
- } while (0)
- static bool
- read_leb(const uint8 *buf, const uint8 *buf_end,
- uint32 *p_offset, uint32 maxbits,
- bool sign, uint64 *p_result)
- {
- uint64 result = 0;
- uint32 shift = 0;
- uint32 bcnt = 0;
- uint64 byte;
- while (true) {
- CHECK_BUF(buf, buf_end, 1);
- byte = buf[*p_offset];
- *p_offset += 1;
- result |= ((byte & 0x7f) << shift);
- shift += 7;
- if ((byte & 0x80) == 0) {
- break;
- }
- bcnt += 1;
- }
- if (bcnt > (((maxbits + 8) >> 3) - (maxbits + 8))) {
- aot_set_last_error("read leb failed: unsigned leb overflow.");
- return false;
- }
- if (sign && (shift < maxbits) && (byte & 0x40)) {
- /* Sign extend */
- result |= (uint64)(- (((uint64)1) << shift));
- }
- *p_result = result;
- return true;
- }
- #define read_leb_uint32(p, p_end, res) do { \
- uint32 off = 0; \
- uint64 res64; \
- if (!read_leb(p, p_end, &off, 32, false, &res64)) \
- return false; \
- p += off; \
- res = (uint32)res64; \
- } while (0)
- #define read_leb_int32(p, p_end, res) do { \
- uint32 off = 0; \
- uint64 res64; \
- if (!read_leb(p, p_end, &off, 32, true, &res64)) \
- return false; \
- p += off; \
- res = (int32)res64; \
- } while (0)
- #define read_leb_int64(p, p_end, res) do { \
- uint32 off = 0; \
- uint64 res64; \
- if (!read_leb(p, p_end, &off, 64, true, &res64)) \
- return false; \
- p += off; \
- res = (int64)res64; \
- } while (0)
- #define COMPILE_ATOMIC_RMW(OP, NAME) \
- case WASM_OP_ATOMIC_RMW_I32_##NAME: \
- bytes = 4; \
- op_type = VALUE_TYPE_I32; \
- goto OP_ATOMIC_##OP; \
- case WASM_OP_ATOMIC_RMW_I64_##NAME: \
- bytes = 8; \
- op_type = VALUE_TYPE_I64; \
- goto OP_ATOMIC_##OP; \
- case WASM_OP_ATOMIC_RMW_I32_##NAME##8_U: \
- bytes = 1; \
- op_type = VALUE_TYPE_I32; \
- goto OP_ATOMIC_##OP; \
- case WASM_OP_ATOMIC_RMW_I32_##NAME##16_U: \
- bytes = 2; \
- op_type = VALUE_TYPE_I32; \
- goto OP_ATOMIC_##OP; \
- case WASM_OP_ATOMIC_RMW_I64_##NAME##8_U: \
- bytes = 1; \
- op_type = VALUE_TYPE_I64; \
- goto OP_ATOMIC_##OP; \
- case WASM_OP_ATOMIC_RMW_I64_##NAME##16_U: \
- bytes = 2; \
- op_type = VALUE_TYPE_I64; \
- goto OP_ATOMIC_##OP; \
- case WASM_OP_ATOMIC_RMW_I64_##NAME##32_U: \
- bytes = 4; \
- op_type = VALUE_TYPE_I64; \
- OP_ATOMIC_##OP: \
- bin_op = LLVMAtomicRMWBinOp##OP; \
- goto build_atomic_rmw;
- static bool
- aot_compile_func(AOTCompContext *comp_ctx, uint32 func_index)
- {
- AOTFuncContext *func_ctx = comp_ctx->func_ctxes[func_index];
- uint8 *frame_ip = func_ctx->aot_func->code, opcode, *p_f32, *p_f64;
- uint8 *frame_ip_end = frame_ip + func_ctx->aot_func->code_size;
- uint8 *param_types = NULL;
- uint8 *result_types = NULL;
- uint8 value_type;
- uint16 param_count;
- uint16 result_count;
- uint32 br_depth, *br_depths, br_count;
- uint32 func_idx, type_idx, mem_idx, local_idx, global_idx, i;
- uint32 bytes = 4, align, offset;
- uint32 type_index;
- bool sign = true;
- int32 i32_const;
- int64 i64_const;
- float32 f32_const;
- float64 f64_const;
- AOTFuncType *func_type = NULL;
- /* Start to translate the opcodes */
- LLVMPositionBuilderAtEnd(comp_ctx->builder,
- func_ctx->block_stack.block_list_head
- ->llvm_entry_block);
- while (frame_ip < frame_ip_end) {
- opcode = *frame_ip++;
- switch (opcode) {
- case WASM_OP_UNREACHABLE:
- if (!aot_compile_op_unreachable(comp_ctx, func_ctx, &frame_ip))
- return false;
- break;
- case WASM_OP_NOP:
- break;
- case WASM_OP_BLOCK:
- case WASM_OP_LOOP:
- case WASM_OP_IF:
- value_type = *frame_ip++;
- if (value_type == VALUE_TYPE_I32
- || value_type == VALUE_TYPE_I64
- || value_type == VALUE_TYPE_F32
- || value_type == VALUE_TYPE_F64
- || value_type == VALUE_TYPE_V128
- || value_type == VALUE_TYPE_VOID
- || value_type == VALUE_TYPE_FUNCREF
- || value_type == VALUE_TYPE_EXTERNREF) {
- param_count = 0;
- param_types = NULL;
- if (value_type == VALUE_TYPE_VOID) {
- result_count = 0;
- result_types = NULL;
- }
- else {
- result_count = 1;
- result_types = &value_type;
- }
- }
- else {
- frame_ip--;
- read_leb_uint32(frame_ip, frame_ip_end, type_index);
- func_type = comp_ctx->comp_data->func_types[type_index];
- param_count = func_type->param_count;
- param_types = func_type->types;
- result_count = func_type->result_count;
- result_types = func_type->types + param_count;
- }
- if (!aot_compile_op_block(comp_ctx, func_ctx,
- &frame_ip, frame_ip_end,
- (uint32)(LABEL_TYPE_BLOCK + opcode - WASM_OP_BLOCK),
- param_count, param_types,
- result_count, result_types))
- return false;
- break;
- case WASM_OP_ELSE:
- if (!aot_compile_op_else(comp_ctx, func_ctx, &frame_ip))
- return false;
- break;
- case WASM_OP_END:
- if (!aot_compile_op_end(comp_ctx, func_ctx, &frame_ip))
- return false;
- break;
- case WASM_OP_BR:
- read_leb_uint32(frame_ip, frame_ip_end, br_depth);
- if (!aot_compile_op_br(comp_ctx, func_ctx, br_depth, &frame_ip))
- return false;
- break;
- case WASM_OP_BR_IF:
- read_leb_uint32(frame_ip, frame_ip_end, br_depth);
- if (!aot_compile_op_br_if(comp_ctx, func_ctx, br_depth, &frame_ip))
- return false;
- break;
- case WASM_OP_BR_TABLE:
- read_leb_uint32(frame_ip, frame_ip_end, br_count);
- if (!(br_depths =
- wasm_runtime_malloc((uint32)sizeof(uint32) * (br_count + 1)))) {
- aot_set_last_error("allocate memory failed.");
- goto fail;
- }
- for (i = 0; i <= br_count; i++)
- read_leb_uint32(frame_ip, frame_ip_end, br_depths[i]);
- if (!aot_compile_op_br_table(comp_ctx, func_ctx,
- br_depths, br_count, &frame_ip)) {
- wasm_runtime_free(br_depths);
- return false;
- }
- wasm_runtime_free(br_depths);
- break;
- case WASM_OP_RETURN:
- if (!aot_compile_op_return(comp_ctx, func_ctx, &frame_ip))
- return false;
- break;
- case WASM_OP_CALL:
- read_leb_uint32(frame_ip, frame_ip_end, func_idx);
- if (!aot_compile_op_call(comp_ctx, func_ctx, func_idx, false))
- return false;
- break;
- case WASM_OP_CALL_INDIRECT:
- {
- uint32 tbl_idx;
- read_leb_uint32(frame_ip, frame_ip_end, type_idx);
- #if WASM_ENABLE_REF_TYPES != 0
- if (comp_ctx->enable_ref_types) {
- read_leb_uint32(frame_ip, frame_ip_end, tbl_idx);
- }
- else
- #endif
- {
- frame_ip++;
- tbl_idx = 0;
- }
- if (!aot_compile_op_call_indirect(comp_ctx, func_ctx, type_idx,
- tbl_idx))
- return false;
- break;
- }
- #if WASM_ENABLE_TAIL_CALL != 0
- case WASM_OP_RETURN_CALL:
- if (!comp_ctx->enable_tail_call) {
- aot_set_last_error("unsupported opcode");
- return false;
- }
- read_leb_uint32(frame_ip, frame_ip_end, func_idx);
- if (!aot_compile_op_call(comp_ctx, func_ctx, func_idx, true))
- return false;
- if (!aot_compile_op_return(comp_ctx, func_ctx, &frame_ip))
- return false;
- break;
- case WASM_OP_RETURN_CALL_INDIRECT:
- {
- uint32 tbl_idx;
- if (!comp_ctx->enable_tail_call) {
- aot_set_last_error("unsupported opcode");
- return false;
- }
- read_leb_uint32(frame_ip, frame_ip_end, type_idx);
- #if WASM_ENABLE_REF_TYPES != 0
- if (comp_ctx->enable_ref_types) {
- read_leb_uint32(frame_ip, frame_ip_end, tbl_idx);
- }
- else
- #endif
- {
- frame_ip++;
- tbl_idx = 0;
- }
- if (!aot_compile_op_call_indirect(comp_ctx, func_ctx, type_idx,
- tbl_idx))
- return false;
- if (!aot_compile_op_return(comp_ctx, func_ctx, &frame_ip))
- return false;
- break;
- }
- #endif /* end of WASM_ENABLE_TAIL_CALL */
- case WASM_OP_DROP:
- if (!aot_compile_op_drop(comp_ctx, func_ctx, true))
- return false;
- break;
- case WASM_OP_DROP_64:
- if (!aot_compile_op_drop(comp_ctx, func_ctx, false))
- return false;
- break;
- case WASM_OP_SELECT:
- if (!aot_compile_op_select(comp_ctx, func_ctx, true))
- return false;
- break;
- case WASM_OP_SELECT_64:
- if (!aot_compile_op_select(comp_ctx, func_ctx, false))
- return false;
- break;
- #if WASM_ENABLE_REF_TYPES != 0
- case WASM_OP_SELECT_T:
- {
- uint32 vec_len;
- if (!comp_ctx->enable_ref_types) {
- goto unsupport_ref_types;
- }
- read_leb_uint32(frame_ip, frame_ip_end, vec_len);
- bh_assert(vec_len == 1);
- vec_len = vec_len;
- type_idx = *frame_ip++;
- if (!aot_compile_op_select(comp_ctx, func_ctx,
- (type_idx != VALUE_TYPE_I64)
- && (type_idx != VALUE_TYPE_F64)))
- return false;
- break;
- }
- case WASM_OP_TABLE_GET:
- {
- uint32 tbl_idx;
- if (!comp_ctx->enable_ref_types) {
- goto unsupport_ref_types;
- }
- read_leb_uint32(frame_ip, frame_ip_end, tbl_idx);
- if (!aot_compile_op_table_get(comp_ctx, func_ctx, tbl_idx))
- return false;
- break;
- }
- case WASM_OP_TABLE_SET:
- {
- uint32 tbl_idx;
- if (!comp_ctx->enable_ref_types) {
- goto unsupport_ref_types;
- }
- read_leb_uint32(frame_ip, frame_ip_end, tbl_idx);
- if (!aot_compile_op_table_set(comp_ctx, func_ctx, tbl_idx))
- return false;
- break;
- }
- case WASM_OP_REF_NULL:
- {
- uint32 type;
- if (!comp_ctx->enable_ref_types) {
- goto unsupport_ref_types;
- }
- read_leb_uint32(frame_ip, frame_ip_end, type);
- if (!aot_compile_op_ref_null(comp_ctx, func_ctx))
- return false;
- (void)type;
- break;
- }
- case WASM_OP_REF_IS_NULL:
- {
- if (!comp_ctx->enable_ref_types) {
- goto unsupport_ref_types;
- }
- if (!aot_compile_op_ref_is_null(comp_ctx, func_ctx))
- return false;
- break;
- }
- case WASM_OP_REF_FUNC:
- {
- uint32 func_idx;
- if (!comp_ctx->enable_ref_types) {
- goto unsupport_ref_types;
- }
- read_leb_uint32(frame_ip, frame_ip_end, func_idx);
- if (!aot_compile_op_ref_func(comp_ctx, func_ctx, func_idx))
- return false;
- break;
- }
- #endif
- case WASM_OP_GET_LOCAL:
- read_leb_uint32(frame_ip, frame_ip_end, local_idx);
- if (!aot_compile_op_get_local(comp_ctx, func_ctx, local_idx))
- return false;
- break;
- case WASM_OP_SET_LOCAL:
- read_leb_uint32(frame_ip, frame_ip_end, local_idx);
- if (!aot_compile_op_set_local(comp_ctx, func_ctx, local_idx))
- return false;
- break;
- case WASM_OP_TEE_LOCAL:
- read_leb_uint32(frame_ip, frame_ip_end, local_idx);
- if (!aot_compile_op_tee_local(comp_ctx, func_ctx, local_idx))
- return false;
- break;
- case WASM_OP_GET_GLOBAL:
- read_leb_uint32(frame_ip, frame_ip_end, global_idx);
- if (!aot_compile_op_get_global(comp_ctx, func_ctx, global_idx))
- return false;
- break;
- case WASM_OP_SET_GLOBAL:
- case WASM_OP_SET_GLOBAL_64:
- case WASM_OP_SET_GLOBAL_AUX_STACK:
- read_leb_uint32(frame_ip, frame_ip_end, global_idx);
- if (!aot_compile_op_set_global(comp_ctx, func_ctx, global_idx,
- opcode == WASM_OP_SET_GLOBAL_AUX_STACK
- ? true : false))
- return false;
- break;
- case WASM_OP_I32_LOAD:
- bytes = 4;
- sign = true;
- goto op_i32_load;
- case WASM_OP_I32_LOAD8_S:
- case WASM_OP_I32_LOAD8_U:
- bytes = 1;
- sign = (opcode == WASM_OP_I32_LOAD8_S) ? true : false;
- goto op_i32_load;
- case WASM_OP_I32_LOAD16_S:
- case WASM_OP_I32_LOAD16_U:
- bytes = 2;
- sign = (opcode == WASM_OP_I32_LOAD16_S) ? true : false;
- op_i32_load:
- read_leb_uint32(frame_ip, frame_ip_end, align);
- read_leb_uint32(frame_ip, frame_ip_end, offset);
- if (!aot_compile_op_i32_load(comp_ctx, func_ctx, align, offset,
- bytes, sign, false))
- return false;
- break;
- case WASM_OP_I64_LOAD:
- bytes = 8;
- sign = true;
- goto op_i64_load;
- case WASM_OP_I64_LOAD8_S:
- case WASM_OP_I64_LOAD8_U:
- bytes = 1;
- sign = (opcode == WASM_OP_I64_LOAD8_S) ? true : false;
- goto op_i64_load;
- case WASM_OP_I64_LOAD16_S:
- case WASM_OP_I64_LOAD16_U:
- bytes = 2;
- sign = (opcode == WASM_OP_I64_LOAD16_S) ? true : false;
- goto op_i64_load;
- case WASM_OP_I64_LOAD32_S:
- case WASM_OP_I64_LOAD32_U:
- bytes = 4;
- sign = (opcode == WASM_OP_I64_LOAD32_S) ? true : false;
- op_i64_load:
- read_leb_uint32(frame_ip, frame_ip_end, align);
- read_leb_uint32(frame_ip, frame_ip_end, offset);
- if (!aot_compile_op_i64_load(comp_ctx, func_ctx, align, offset,
- bytes, sign, false))
- return false;
- break;
- case WASM_OP_F32_LOAD:
- read_leb_uint32(frame_ip, frame_ip_end, align);
- read_leb_uint32(frame_ip, frame_ip_end, offset);
- if (!aot_compile_op_f32_load(comp_ctx, func_ctx, align, offset))
- return false;
- break;
- case WASM_OP_F64_LOAD:
- read_leb_uint32(frame_ip, frame_ip_end, align);
- read_leb_uint32(frame_ip, frame_ip_end, offset);
- if (!aot_compile_op_f64_load(comp_ctx, func_ctx, align, offset))
- return false;
- break;
- case WASM_OP_I32_STORE:
- bytes = 4;
- goto op_i32_store;
- case WASM_OP_I32_STORE8:
- bytes = 1;
- goto op_i32_store;
- case WASM_OP_I32_STORE16:
- bytes = 2;
- op_i32_store:
- read_leb_uint32(frame_ip, frame_ip_end, align);
- read_leb_uint32(frame_ip, frame_ip_end, offset);
- if (!aot_compile_op_i32_store(comp_ctx, func_ctx, align,
- offset, bytes, false))
- return false;
- break;
- case WASM_OP_I64_STORE:
- bytes = 8;
- goto op_i64_store;
- case WASM_OP_I64_STORE8:
- bytes = 1;
- goto op_i64_store;
- case WASM_OP_I64_STORE16:
- bytes = 2;
- goto op_i64_store;
- case WASM_OP_I64_STORE32:
- bytes = 4;
- op_i64_store:
- read_leb_uint32(frame_ip, frame_ip_end, align);
- read_leb_uint32(frame_ip, frame_ip_end, offset);
- if (!aot_compile_op_i64_store(comp_ctx, func_ctx, align,
- offset, bytes, false))
- return false;
- break;
- case WASM_OP_F32_STORE:
- read_leb_uint32(frame_ip, frame_ip_end, align);
- read_leb_uint32(frame_ip, frame_ip_end, offset);
- if (!aot_compile_op_f32_store(comp_ctx, func_ctx, align, offset))
- return false;
- break;
- case WASM_OP_F64_STORE:
- read_leb_uint32(frame_ip, frame_ip_end, align);
- read_leb_uint32(frame_ip, frame_ip_end, offset);
- if (!aot_compile_op_f64_store(comp_ctx, func_ctx, align, offset))
- return false;
- break;
- case WASM_OP_MEMORY_SIZE:
- read_leb_uint32(frame_ip, frame_ip_end, mem_idx);
- if (!aot_compile_op_memory_size(comp_ctx, func_ctx))
- return false;
- (void)mem_idx;
- break;
- case WASM_OP_MEMORY_GROW:
- read_leb_uint32(frame_ip, frame_ip_end, mem_idx);
- if (!aot_compile_op_memory_grow(comp_ctx, func_ctx))
- return false;
- break;
- case WASM_OP_I32_CONST:
- read_leb_int32(frame_ip, frame_ip_end, i32_const);
- if (!aot_compile_op_i32_const(comp_ctx, func_ctx, i32_const))
- return false;
- break;
- case WASM_OP_I64_CONST:
- read_leb_int64(frame_ip, frame_ip_end, i64_const);
- if (!aot_compile_op_i64_const(comp_ctx, func_ctx, i64_const))
- return false;
- break;
- case WASM_OP_F32_CONST:
- p_f32 = (uint8*)&f32_const;
- for (i = 0; i < sizeof(float32); i++)
- *p_f32++ = *frame_ip++;
- if (!aot_compile_op_f32_const(comp_ctx, func_ctx, f32_const))
- return false;
- break;
- case WASM_OP_F64_CONST:
- p_f64 = (uint8*)&f64_const;
- for (i = 0; i < sizeof(float64); i++)
- *p_f64++ = *frame_ip++;
- if (!aot_compile_op_f64_const(comp_ctx, func_ctx, f64_const))
- return false;
- break;
- case WASM_OP_I32_EQZ:
- case WASM_OP_I32_EQ:
- case WASM_OP_I32_NE:
- case WASM_OP_I32_LT_S:
- case WASM_OP_I32_LT_U:
- case WASM_OP_I32_GT_S:
- case WASM_OP_I32_GT_U:
- case WASM_OP_I32_LE_S:
- case WASM_OP_I32_LE_U:
- case WASM_OP_I32_GE_S:
- case WASM_OP_I32_GE_U:
- if (!aot_compile_op_i32_compare(comp_ctx, func_ctx,
- INT_EQZ + opcode - WASM_OP_I32_EQZ))
- return false;
- break;
- case WASM_OP_I64_EQZ:
- case WASM_OP_I64_EQ:
- case WASM_OP_I64_NE:
- case WASM_OP_I64_LT_S:
- case WASM_OP_I64_LT_U:
- case WASM_OP_I64_GT_S:
- case WASM_OP_I64_GT_U:
- case WASM_OP_I64_LE_S:
- case WASM_OP_I64_LE_U:
- case WASM_OP_I64_GE_S:
- case WASM_OP_I64_GE_U:
- if (!aot_compile_op_i64_compare(comp_ctx, func_ctx,
- INT_EQZ + opcode - WASM_OP_I64_EQZ))
- return false;
- break;
- case WASM_OP_F32_EQ:
- case WASM_OP_F32_NE:
- case WASM_OP_F32_LT:
- case WASM_OP_F32_GT:
- case WASM_OP_F32_LE:
- case WASM_OP_F32_GE:
- if (!aot_compile_op_f32_compare(comp_ctx, func_ctx,
- FLOAT_EQ + opcode - WASM_OP_F32_EQ))
- return false;
- break;
- case WASM_OP_F64_EQ:
- case WASM_OP_F64_NE:
- case WASM_OP_F64_LT:
- case WASM_OP_F64_GT:
- case WASM_OP_F64_LE:
- case WASM_OP_F64_GE:
- if (!aot_compile_op_f64_compare(comp_ctx, func_ctx,
- FLOAT_EQ + opcode - WASM_OP_F64_EQ))
- return false;
- break;
- case WASM_OP_I32_CLZ:
- if (!aot_compile_op_i32_clz(comp_ctx, func_ctx))
- return false;
- break;
- case WASM_OP_I32_CTZ:
- if (!aot_compile_op_i32_ctz(comp_ctx, func_ctx))
- return false;
- break;
- case WASM_OP_I32_POPCNT:
- if (!aot_compile_op_i32_popcnt(comp_ctx, func_ctx))
- return false;
- break;
- case WASM_OP_I32_ADD:
- case WASM_OP_I32_SUB:
- case WASM_OP_I32_MUL:
- case WASM_OP_I32_DIV_S:
- case WASM_OP_I32_DIV_U:
- case WASM_OP_I32_REM_S:
- case WASM_OP_I32_REM_U:
- if (!aot_compile_op_i32_arithmetic(comp_ctx, func_ctx,
- INT_ADD + opcode - WASM_OP_I32_ADD,
- &frame_ip))
- return false;
- break;
- case WASM_OP_I32_AND:
- case WASM_OP_I32_OR:
- case WASM_OP_I32_XOR:
- if (!aot_compile_op_i32_bitwise(comp_ctx, func_ctx,
- INT_SHL + opcode - WASM_OP_I32_AND))
- return false;
- break;
- case WASM_OP_I32_SHL:
- case WASM_OP_I32_SHR_S:
- case WASM_OP_I32_SHR_U:
- case WASM_OP_I32_ROTL:
- case WASM_OP_I32_ROTR:
- if (!aot_compile_op_i32_shift(comp_ctx, func_ctx,
- INT_SHL + opcode - WASM_OP_I32_SHL))
- return false;
- break;
- case WASM_OP_I64_CLZ:
- if (!aot_compile_op_i64_clz(comp_ctx, func_ctx))
- return false;
- break;
- case WASM_OP_I64_CTZ:
- if (!aot_compile_op_i64_ctz(comp_ctx, func_ctx))
- return false;
- break;
- case WASM_OP_I64_POPCNT:
- if (!aot_compile_op_i64_popcnt(comp_ctx, func_ctx))
- return false;
- break;
- case WASM_OP_I64_ADD:
- case WASM_OP_I64_SUB:
- case WASM_OP_I64_MUL:
- case WASM_OP_I64_DIV_S:
- case WASM_OP_I64_DIV_U:
- case WASM_OP_I64_REM_S:
- case WASM_OP_I64_REM_U:
- if (!aot_compile_op_i64_arithmetic(comp_ctx, func_ctx,
- INT_ADD + opcode - WASM_OP_I64_ADD,
- &frame_ip))
- return false;
- break;
- case WASM_OP_I64_AND:
- case WASM_OP_I64_OR:
- case WASM_OP_I64_XOR:
- if (!aot_compile_op_i64_bitwise(comp_ctx, func_ctx,
- INT_SHL + opcode - WASM_OP_I64_AND))
- return false;
- break;
- case WASM_OP_I64_SHL:
- case WASM_OP_I64_SHR_S:
- case WASM_OP_I64_SHR_U:
- case WASM_OP_I64_ROTL:
- case WASM_OP_I64_ROTR:
- if (!aot_compile_op_i64_shift(comp_ctx, func_ctx,
- INT_SHL + opcode - WASM_OP_I64_SHL))
- return false;
- break;
- case WASM_OP_F32_ABS:
- case WASM_OP_F32_NEG:
- case WASM_OP_F32_CEIL:
- case WASM_OP_F32_FLOOR:
- case WASM_OP_F32_TRUNC:
- case WASM_OP_F32_NEAREST:
- case WASM_OP_F32_SQRT:
- if (!aot_compile_op_f32_math(comp_ctx, func_ctx,
- FLOAT_ABS + opcode - WASM_OP_F32_ABS))
- return false;
- break;
- case WASM_OP_F32_ADD:
- case WASM_OP_F32_SUB:
- case WASM_OP_F32_MUL:
- case WASM_OP_F32_DIV:
- case WASM_OP_F32_MIN:
- case WASM_OP_F32_MAX:
- if (!aot_compile_op_f32_arithmetic(comp_ctx, func_ctx,
- FLOAT_ADD + opcode - WASM_OP_F32_ADD))
- return false;
- break;
- case WASM_OP_F32_COPYSIGN:
- if (!aot_compile_op_f32_copysign(comp_ctx, func_ctx))
- return false;
- break;
- case WASM_OP_F64_ABS:
- case WASM_OP_F64_NEG:
- case WASM_OP_F64_CEIL:
- case WASM_OP_F64_FLOOR:
- case WASM_OP_F64_TRUNC:
- case WASM_OP_F64_NEAREST:
- case WASM_OP_F64_SQRT:
- if (!aot_compile_op_f64_math(comp_ctx, func_ctx,
- FLOAT_ABS + opcode - WASM_OP_F64_ABS))
- return false;
- break;
- case WASM_OP_F64_ADD:
- case WASM_OP_F64_SUB:
- case WASM_OP_F64_MUL:
- case WASM_OP_F64_DIV:
- case WASM_OP_F64_MIN:
- case WASM_OP_F64_MAX:
- if (!aot_compile_op_f64_arithmetic(comp_ctx, func_ctx,
- FLOAT_ADD + opcode - WASM_OP_F64_ADD))
- return false;
- break;
- case WASM_OP_F64_COPYSIGN:
- if (!aot_compile_op_f64_copysign(comp_ctx, func_ctx))
- return false;
- break;
- case WASM_OP_I32_WRAP_I64:
- if (!aot_compile_op_i32_wrap_i64(comp_ctx, func_ctx))
- return false;
- break;
- case WASM_OP_I32_TRUNC_S_F32:
- case WASM_OP_I32_TRUNC_U_F32:
- sign = (opcode == WASM_OP_I32_TRUNC_S_F32) ? true : false;
- if (!aot_compile_op_i32_trunc_f32(comp_ctx, func_ctx, sign, false))
- return false;
- break;
- case WASM_OP_I32_TRUNC_S_F64:
- case WASM_OP_I32_TRUNC_U_F64:
- sign = (opcode == WASM_OP_I32_TRUNC_S_F64) ? true : false;
- if (!aot_compile_op_i32_trunc_f64(comp_ctx, func_ctx, sign, false))
- return false;
- break;
- case WASM_OP_I64_EXTEND_S_I32:
- case WASM_OP_I64_EXTEND_U_I32:
- sign = (opcode == WASM_OP_I64_EXTEND_S_I32) ? true : false;
- if (!aot_compile_op_i64_extend_i32(comp_ctx, func_ctx, sign))
- return false;
- break;
- case WASM_OP_I64_TRUNC_S_F32:
- case WASM_OP_I64_TRUNC_U_F32:
- sign = (opcode == WASM_OP_I64_TRUNC_S_F32) ? true : false;
- if (!aot_compile_op_i64_trunc_f32(comp_ctx, func_ctx, sign, false))
- return false;
- break;
- case WASM_OP_I64_TRUNC_S_F64:
- case WASM_OP_I64_TRUNC_U_F64:
- sign = (opcode == WASM_OP_I64_TRUNC_S_F64) ? true : false;
- if (!aot_compile_op_i64_trunc_f64(comp_ctx, func_ctx, sign, false))
- return false;
- break;
- case WASM_OP_F32_CONVERT_S_I32:
- case WASM_OP_F32_CONVERT_U_I32:
- sign = (opcode == WASM_OP_F32_CONVERT_S_I32) ? true : false;
- if (!aot_compile_op_f32_convert_i32(comp_ctx, func_ctx, sign))
- return false;
- break;
- case WASM_OP_F32_CONVERT_S_I64:
- case WASM_OP_F32_CONVERT_U_I64:
- sign = (opcode == WASM_OP_F32_CONVERT_S_I64) ? true : false;
- if (!aot_compile_op_f32_convert_i64(comp_ctx, func_ctx, sign))
- return false;
- break;
- case WASM_OP_F32_DEMOTE_F64:
- if (!aot_compile_op_f32_demote_f64(comp_ctx, func_ctx))
- return false;
- break;
- case WASM_OP_F64_CONVERT_S_I32:
- case WASM_OP_F64_CONVERT_U_I32:
- sign = (opcode == WASM_OP_F64_CONVERT_S_I32) ? true : false;
- if (!aot_compile_op_f64_convert_i32(comp_ctx, func_ctx, sign))
- return false;
- break;
- case WASM_OP_F64_CONVERT_S_I64:
- case WASM_OP_F64_CONVERT_U_I64:
- sign = (opcode == WASM_OP_F64_CONVERT_S_I64) ? true : false;
- if (!aot_compile_op_f64_convert_i64(comp_ctx, func_ctx, sign))
- return false;
- break;
- case WASM_OP_F64_PROMOTE_F32:
- if (!aot_compile_op_f64_promote_f32(comp_ctx, func_ctx))
- return false;
- break;
- case WASM_OP_I32_REINTERPRET_F32:
- if (!aot_compile_op_i32_reinterpret_f32(comp_ctx, func_ctx))
- return false;
- break;
- case WASM_OP_I64_REINTERPRET_F64:
- if (!aot_compile_op_i64_reinterpret_f64(comp_ctx, func_ctx))
- return false;
- break;
- case WASM_OP_F32_REINTERPRET_I32:
- if (!aot_compile_op_f32_reinterpret_i32(comp_ctx, func_ctx))
- return false;
- break;
- case WASM_OP_F64_REINTERPRET_I64:
- if (!aot_compile_op_f64_reinterpret_i64(comp_ctx, func_ctx))
- return false;
- break;
- case WASM_OP_I32_EXTEND8_S:
- if (!aot_compile_op_i32_extend_i32(comp_ctx, func_ctx, 8))
- return false;
- break;
- case WASM_OP_I32_EXTEND16_S:
- if (!aot_compile_op_i32_extend_i32(comp_ctx, func_ctx, 16))
- return false;
- break;
- case WASM_OP_I64_EXTEND8_S:
- if (!aot_compile_op_i64_extend_i64(comp_ctx, func_ctx, 8))
- return false;
- break;
- case WASM_OP_I64_EXTEND16_S:
- if (!aot_compile_op_i64_extend_i64(comp_ctx, func_ctx, 16))
- return false;
- break;
- case WASM_OP_I64_EXTEND32_S:
- if (!aot_compile_op_i64_extend_i64(comp_ctx, func_ctx, 32))
- return false;
- break;
- case WASM_OP_MISC_PREFIX:
- {
- uint32 opcode1;
- read_leb_uint32(frame_ip, frame_ip_end, opcode1);
- opcode = (uint32)opcode1;
- //TODO: --enable-bulk-memory ?
- #if WASM_ENABLE_REF_TYPES != 0
- if (WASM_OP_TABLE_INIT <= opcode && opcode <= WASM_OP_TABLE_FILL
- && !comp_ctx->enable_ref_types) {
- goto unsupport_ref_types;
- }
- #endif
- switch (opcode) {
- case WASM_OP_I32_TRUNC_SAT_S_F32:
- case WASM_OP_I32_TRUNC_SAT_U_F32:
- sign = (opcode == WASM_OP_I32_TRUNC_SAT_S_F32) ? true : false;
- if (!aot_compile_op_i32_trunc_f32(comp_ctx, func_ctx, sign, true))
- return false;
- break;
- case WASM_OP_I32_TRUNC_SAT_S_F64:
- case WASM_OP_I32_TRUNC_SAT_U_F64:
- sign = (opcode == WASM_OP_I32_TRUNC_SAT_S_F64) ? true : false;
- if (!aot_compile_op_i32_trunc_f64(comp_ctx, func_ctx, sign, true))
- return false;
- break;
- case WASM_OP_I64_TRUNC_SAT_S_F32:
- case WASM_OP_I64_TRUNC_SAT_U_F32:
- sign = (opcode == WASM_OP_I64_TRUNC_SAT_S_F32) ? true : false;
- if (!aot_compile_op_i64_trunc_f32(comp_ctx, func_ctx, sign, true))
- return false;
- break;
- case WASM_OP_I64_TRUNC_SAT_S_F64:
- case WASM_OP_I64_TRUNC_SAT_U_F64:
- sign = (opcode == WASM_OP_I64_TRUNC_SAT_S_F64) ? true : false;
- if (!aot_compile_op_i64_trunc_f64(comp_ctx, func_ctx, sign, true))
- return false;
- break;
- #if WASM_ENABLE_BULK_MEMORY != 0
- case WASM_OP_MEMORY_INIT:
- {
- uint32 seg_index;
- read_leb_uint32(frame_ip, frame_ip_end, seg_index);
- frame_ip ++;
- if (!aot_compile_op_memory_init(comp_ctx, func_ctx, seg_index))
- return false;
- break;
- }
- case WASM_OP_DATA_DROP:
- {
- uint32 seg_index;
- read_leb_uint32(frame_ip, frame_ip_end, seg_index);
- if (!aot_compile_op_data_drop(comp_ctx, func_ctx, seg_index))
- return false;
- break;
- }
- case WASM_OP_MEMORY_COPY:
- {
- frame_ip += 2;
- if (!aot_compile_op_memory_copy(comp_ctx, func_ctx))
- return false;
- break;
- }
- case WASM_OP_MEMORY_FILL:
- {
- frame_ip ++;
- if (!aot_compile_op_memory_fill(comp_ctx, func_ctx))
- return false;
- break;
- }
- #endif /* WASM_ENABLE_BULK_MEMORY */
- #if WASM_ENABLE_REF_TYPES != 0
- case WASM_OP_TABLE_INIT:
- {
- uint32 tbl_idx, tbl_seg_idx;
- read_leb_uint32(frame_ip, frame_ip_end, tbl_seg_idx);
- read_leb_uint32(frame_ip, frame_ip_end, tbl_idx);
- if (!aot_compile_op_table_init(comp_ctx, func_ctx, tbl_idx,
- tbl_seg_idx))
- return false;
- break;
- }
- case WASM_OP_ELEM_DROP:
- {
- uint32 tbl_seg_idx;
- read_leb_uint32(frame_ip, frame_ip_end, tbl_seg_idx);
- if (!aot_compile_op_elem_drop(comp_ctx, func_ctx, tbl_seg_idx))
- return false;
- break;
- }
- case WASM_OP_TABLE_COPY:
- {
- uint32 src_tbl_idx, dst_tbl_idx;
- read_leb_uint32(frame_ip, frame_ip_end, dst_tbl_idx);
- read_leb_uint32(frame_ip, frame_ip_end, src_tbl_idx);
- if (!aot_compile_op_table_copy(comp_ctx, func_ctx, src_tbl_idx,
- dst_tbl_idx))
- return false;
- break;
- }
- case WASM_OP_TABLE_GROW:
- {
- uint32 tbl_idx;
- read_leb_uint32(frame_ip, frame_ip_end, tbl_idx);
- if (!aot_compile_op_table_grow(comp_ctx, func_ctx, tbl_idx))
- return false;
- break;
- }
- case WASM_OP_TABLE_SIZE:
- {
- uint32 tbl_idx;
- read_leb_uint32(frame_ip, frame_ip_end, tbl_idx);
- if (!aot_compile_op_table_size(comp_ctx, func_ctx, tbl_idx))
- return false;
- break;
- }
- case WASM_OP_TABLE_FILL:
- {
- uint32 tbl_idx;
- read_leb_uint32(frame_ip, frame_ip_end, tbl_idx);
- if (!aot_compile_op_table_fill(comp_ctx, func_ctx, tbl_idx))
- return false;
- break;
- }
- #endif /* WASM_ENABLE_REF_TYPES */
- default:
- aot_set_last_error("unsupported opcode");
- return false;
- }
- break;
- }
- #if WASM_ENABLE_SHARED_MEMORY != 0
- case WASM_OP_ATOMIC_PREFIX:
- {
- uint8 bin_op, op_type;
- if (frame_ip < frame_ip_end) {
- opcode = *frame_ip++;
- }
- if (opcode != WASM_OP_ATOMIC_FENCE) {
- read_leb_uint32(frame_ip, frame_ip_end, align);
- read_leb_uint32(frame_ip, frame_ip_end, offset);
- }
- switch (opcode) {
- case WASM_OP_ATOMIC_WAIT32:
- if (!aot_compile_op_atomic_wait(comp_ctx, func_ctx, VALUE_TYPE_I32,
- align, offset, 4))
- return false;
- break;
- case WASM_OP_ATOMIC_WAIT64:
- if (!aot_compile_op_atomic_wait(comp_ctx, func_ctx, VALUE_TYPE_I64,
- align, offset, 8))
- return false;
- break;
- case WASM_OP_ATOMIC_NOTIFY:
- if (!aot_compiler_op_atomic_notify(comp_ctx, func_ctx, align,
- offset, bytes))
- return false;
- break;
- case WASM_OP_ATOMIC_I32_LOAD:
- bytes = 4;
- goto op_atomic_i32_load;
- case WASM_OP_ATOMIC_I32_LOAD8_U:
- bytes = 1;
- goto op_atomic_i32_load;
- case WASM_OP_ATOMIC_I32_LOAD16_U:
- bytes = 2;
- op_atomic_i32_load:
- if (!aot_compile_op_i32_load(comp_ctx, func_ctx, align,
- offset, bytes, sign, true))
- return false;
- break;
- case WASM_OP_ATOMIC_I64_LOAD:
- bytes = 8;
- goto op_atomic_i64_load;
- case WASM_OP_ATOMIC_I64_LOAD8_U:
- bytes = 1;
- goto op_atomic_i64_load;
- case WASM_OP_ATOMIC_I64_LOAD16_U:
- bytes = 2;
- goto op_atomic_i64_load;
- case WASM_OP_ATOMIC_I64_LOAD32_U:
- bytes = 4;
- op_atomic_i64_load:
- if (!aot_compile_op_i64_load(comp_ctx, func_ctx, align,
- offset, bytes, sign, true))
- return false;
- break;
- case WASM_OP_ATOMIC_I32_STORE:
- bytes = 4;
- goto op_atomic_i32_store;
- case WASM_OP_ATOMIC_I32_STORE8:
- bytes = 1;
- goto op_atomic_i32_store;
- case WASM_OP_ATOMIC_I32_STORE16:
- bytes = 2;
- op_atomic_i32_store:
- if (!aot_compile_op_i32_store(comp_ctx, func_ctx, align,
- offset, bytes, true))
- return false;
- break;
- case WASM_OP_ATOMIC_I64_STORE:
- bytes = 8;
- goto op_atomic_i64_store;
- case WASM_OP_ATOMIC_I64_STORE8:
- bytes = 1;
- goto op_atomic_i64_store;
- case WASM_OP_ATOMIC_I64_STORE16:
- bytes = 2;
- goto op_atomic_i64_store;
- case WASM_OP_ATOMIC_I64_STORE32:
- bytes = 4;
- op_atomic_i64_store:
- if (!aot_compile_op_i64_store(comp_ctx, func_ctx, align,
- offset, bytes, true))
- return false;
- break;
- case WASM_OP_ATOMIC_RMW_I32_CMPXCHG:
- bytes = 4;
- op_type = VALUE_TYPE_I32;
- goto op_atomic_cmpxchg;
- case WASM_OP_ATOMIC_RMW_I64_CMPXCHG:
- bytes = 8;
- op_type = VALUE_TYPE_I64;
- goto op_atomic_cmpxchg;
- case WASM_OP_ATOMIC_RMW_I32_CMPXCHG8_U:
- bytes = 1;
- op_type = VALUE_TYPE_I32;
- goto op_atomic_cmpxchg;
- case WASM_OP_ATOMIC_RMW_I32_CMPXCHG16_U:
- bytes = 2;
- op_type = VALUE_TYPE_I32;
- goto op_atomic_cmpxchg;
- case WASM_OP_ATOMIC_RMW_I64_CMPXCHG8_U:
- bytes = 1;
- op_type = VALUE_TYPE_I64;
- goto op_atomic_cmpxchg;
- case WASM_OP_ATOMIC_RMW_I64_CMPXCHG16_U:
- bytes = 2;
- op_type = VALUE_TYPE_I64;
- goto op_atomic_cmpxchg;
- case WASM_OP_ATOMIC_RMW_I64_CMPXCHG32_U:
- bytes = 4;
- op_type = VALUE_TYPE_I64;
- op_atomic_cmpxchg:
- if (!aot_compile_op_atomic_cmpxchg(comp_ctx, func_ctx,
- op_type, align,
- offset, bytes))
- return false;
- break;
- COMPILE_ATOMIC_RMW(Add, ADD);
- COMPILE_ATOMIC_RMW(Sub, SUB);
- COMPILE_ATOMIC_RMW(And, AND);
- COMPILE_ATOMIC_RMW(Or, OR);
- COMPILE_ATOMIC_RMW(Xor, XOR);
- COMPILE_ATOMIC_RMW(Xchg, XCHG);
- build_atomic_rmw:
- if (!aot_compile_op_atomic_rmw(comp_ctx, func_ctx,
- bin_op, op_type,
- align, offset, bytes))
- return false;
- break;
- default:
- aot_set_last_error("unsupported opcode");
- return false;
- }
- break;
- }
- #endif /* end of WASM_ENABLE_SHARED_MEMORY */
- #if WASM_ENABLE_SIMD != 0
- case WASM_OP_SIMD_PREFIX:
- {
- if (!comp_ctx->enable_simd) {
- goto unsupport_simd;
- }
- opcode = *frame_ip++;
- switch (opcode) {
- case SIMD_v128_load:
- {
- read_leb_uint32(frame_ip, frame_ip_end, align);
- read_leb_uint32(frame_ip, frame_ip_end, offset);
- if (!aot_compile_simd_v128_load(comp_ctx, func_ctx, align, offset))
- return false;
- break;
- }
- case SIMD_i16x8_load8x8_s:
- case SIMD_i16x8_load8x8_u:
- case SIMD_i32x4_load16x4_s:
- case SIMD_i32x4_load16x4_u:
- case SIMD_i64x2_load32x2_s:
- case SIMD_i64x2_load32x2_u:
- {
- read_leb_uint32(frame_ip, frame_ip_end, align);
- read_leb_uint32(frame_ip, frame_ip_end, offset);
- if (!aot_compile_simd_load_extend(comp_ctx, func_ctx,
- opcode, align, offset))
- return false;
- break;
- }
- case SIMD_v8x16_load_splat:
- case SIMD_v16x8_load_splat:
- case SIMD_v32x4_load_splat:
- case SIMD_v64x2_load_splat:
- {
- read_leb_uint32(frame_ip, frame_ip_end, align);
- read_leb_uint32(frame_ip, frame_ip_end, offset);
- if (!aot_compile_simd_load_splat(comp_ctx, func_ctx,
- opcode, align, offset))
- return false;
- break;
- }
- case SIMD_v128_store:
- {
- read_leb_uint32(frame_ip, frame_ip_end, align);
- read_leb_uint32(frame_ip, frame_ip_end, offset);
- if (!aot_compile_simd_v128_store(comp_ctx, func_ctx, align, offset))
- return false;
- break;
- }
- case SIMD_v128_const:
- {
- if (!aot_compile_simd_v128_const(comp_ctx, func_ctx, frame_ip))
- return false;
- frame_ip += 16;
- break;
- }
- case SIMD_v8x16_shuffle:
- {
- if (!aot_compile_simd_shuffle(comp_ctx, func_ctx, frame_ip))
- return false;
- frame_ip += 16;
- break;
- }
- case SIMD_v8x16_swizzle:
- {
- if (!aot_compile_simd_swizzle(comp_ctx, func_ctx))
- return false;
- break;
- }
- case SIMD_i8x16_splat:
- case SIMD_i16x8_splat:
- case SIMD_i32x4_splat:
- case SIMD_i64x2_splat:
- case SIMD_f32x4_splat:
- case SIMD_f64x2_splat:
- {
- if (!aot_compile_simd_splat(comp_ctx, func_ctx, opcode))
- return false;
- break;
- }
- case SIMD_i8x16_extract_lane_s:
- {
- if (!aot_compile_simd_extract_i8x16(comp_ctx, func_ctx, *frame_ip++,
- true))
- return false;
- break;
- }
- case SIMD_i8x16_extract_lane_u:
- {
- if (!aot_compile_simd_extract_i8x16(comp_ctx, func_ctx, *frame_ip++,
- false))
- return false;
- break;
- }
- case SIMD_i16x8_extract_lane_s:
- {
- if (!aot_compile_simd_extract_i16x8(comp_ctx, func_ctx, *frame_ip++,
- true))
- return false;
- break;
- }
- case SIMD_i16x8_extract_lane_u:
- {
- if (!aot_compile_simd_extract_i16x8(comp_ctx, func_ctx, *frame_ip++,
- false))
- return false;
- break;
- }
- case SIMD_i32x4_extract_lane:
- {
- if (!aot_compile_simd_extract_i32x4(comp_ctx, func_ctx, *frame_ip++))
- return false;
- break;
- }
- case SIMD_i64x2_extract_lane:
- {
- if (!aot_compile_simd_extract_i64x2(comp_ctx, func_ctx, *frame_ip++))
- return false;
- break;
- }
- case SIMD_f32x4_extract_lane:
- {
- if (!aot_compile_simd_extract_f32x4(comp_ctx, func_ctx, *frame_ip++))
- return false;
- break;
- }
- case SIMD_f64x2_extract_lane:
- {
- if (!aot_compile_simd_extract_f64x2(comp_ctx, func_ctx, *frame_ip++))
- return false;
- break;
- }
- case SIMD_i8x16_replace_lane:
- {
- if (!aot_compile_simd_replace_i8x16(comp_ctx, func_ctx, *frame_ip++))
- return false;
- break;
- }
- case SIMD_i16x8_replace_lane:
- {
- if (!aot_compile_simd_replace_i16x8(comp_ctx, func_ctx, *frame_ip++))
- return false;
- break;
- }
- case SIMD_i32x4_replace_lane:
- {
- if (!aot_compile_simd_replace_i32x4(comp_ctx, func_ctx, *frame_ip++))
- return false;
- break;
- }
- case SIMD_i64x2_replace_lane:
- {
- if (!aot_compile_simd_replace_i64x2(comp_ctx, func_ctx, *frame_ip++))
- return false;
- break;
- }
- case SIMD_f32x4_replace_lane:
- {
- if (!aot_compile_simd_replace_f32x4(comp_ctx, func_ctx, *frame_ip++))
- return false;
- break;
- }
- case SIMD_f64x2_replace_lane:
- {
- if (!aot_compile_simd_replace_f64x2(comp_ctx, func_ctx, *frame_ip++))
- return false;
- break;
- }
- case SIMD_i8x16_eq:
- case SIMD_i8x16_ne:
- case SIMD_i8x16_lt_s:
- case SIMD_i8x16_lt_u:
- case SIMD_i8x16_gt_s:
- case SIMD_i8x16_gt_u:
- case SIMD_i8x16_le_s:
- case SIMD_i8x16_le_u:
- case SIMD_i8x16_ge_s:
- case SIMD_i8x16_ge_u:
- {
- if (!aot_compile_simd_i8x16_compare(comp_ctx, func_ctx,
- INT_EQ + opcode - SIMD_i8x16_eq))
- return false;
- break;
- }
- case SIMD_i16x8_eq:
- case SIMD_i16x8_ne:
- case SIMD_i16x8_lt_s:
- case SIMD_i16x8_lt_u:
- case SIMD_i16x8_gt_s:
- case SIMD_i16x8_gt_u:
- case SIMD_i16x8_le_s:
- case SIMD_i16x8_le_u:
- case SIMD_i16x8_ge_s:
- case SIMD_i16x8_ge_u:
- {
- if (!aot_compile_simd_i16x8_compare(comp_ctx, func_ctx,
- INT_EQ + opcode - SIMD_i16x8_eq))
- return false;
- break;
- }
- case SIMD_i32x4_eq:
- case SIMD_i32x4_ne:
- case SIMD_i32x4_lt_s:
- case SIMD_i32x4_lt_u:
- case SIMD_i32x4_gt_s:
- case SIMD_i32x4_gt_u:
- case SIMD_i32x4_le_s:
- case SIMD_i32x4_le_u:
- case SIMD_i32x4_ge_s:
- case SIMD_i32x4_ge_u:
- {
- if (!aot_compile_simd_i32x4_compare(comp_ctx, func_ctx,
- INT_EQ + opcode - SIMD_i32x4_eq))
- return false;
- break;
- }
- case SIMD_f32x4_eq:
- case SIMD_f32x4_ne:
- case SIMD_f32x4_lt:
- case SIMD_f32x4_gt:
- case SIMD_f32x4_le:
- case SIMD_f32x4_ge:
- {
- if (!aot_compile_simd_f32x4_compare(comp_ctx, func_ctx,
- FLOAT_EQ + opcode - SIMD_f32x4_eq))
- return false;
- break;
- }
- case SIMD_f64x2_eq:
- case SIMD_f64x2_ne:
- case SIMD_f64x2_lt:
- case SIMD_f64x2_gt:
- case SIMD_f64x2_le:
- case SIMD_f64x2_ge:
- {
- if (!aot_compile_simd_f64x2_compare(comp_ctx, func_ctx,
- FLOAT_EQ + opcode - SIMD_f64x2_eq))
- return false;
- break;
- }
- case SIMD_v128_not:
- case SIMD_v128_and:
- case SIMD_v128_andnot:
- case SIMD_v128_or:
- case SIMD_v128_xor:
- case SIMD_v128_bitselect:
- {
- if (!aot_compile_simd_v128_bitwise(comp_ctx, func_ctx,
- V128_NOT + opcode - SIMD_v128_not))
- return false;
- break;
- }
- case SIMD_i8x16_add:
- case SIMD_i8x16_sub:
- {
- V128Arithmetic arith_op = (opcode == SIMD_i8x16_add)
- ? V128_ADD : V128_SUB;
- if (!aot_compile_simd_i8x16_arith(comp_ctx, func_ctx, arith_op))
- return false;
- break;
- }
- case SIMD_i16x8_add:
- case SIMD_i16x8_sub:
- case SIMD_i16x8_mul:
- {
- V128Arithmetic arith_op = V128_ADD;
- if (opcode == SIMD_i16x8_sub)
- arith_op = V128_SUB;
- else if (opcode == SIMD_i16x8_mul)
- arith_op = V128_MUL;
- if (!aot_compile_simd_i16x8_arith(comp_ctx, func_ctx, arith_op))
- return false;
- break;
- }
- case SIMD_i32x4_add:
- case SIMD_i32x4_sub:
- case SIMD_i32x4_mul:
- {
- V128Arithmetic arith_op = V128_ADD;
- if (opcode == SIMD_i32x4_sub)
- arith_op = V128_SUB;
- else if (opcode == SIMD_i32x4_mul)
- arith_op = V128_MUL;
- if (!aot_compile_simd_i32x4_arith(comp_ctx, func_ctx, arith_op))
- return false;
- break;
- }
- case SIMD_i64x2_add:
- case SIMD_i64x2_sub:
- case SIMD_i64x2_mul:
- {
- V128Arithmetic arith_op = V128_ADD;
- if (opcode == SIMD_i64x2_sub)
- arith_op = V128_SUB;
- else if (opcode == SIMD_i64x2_mul)
- arith_op = V128_MUL;
- if (!aot_compile_simd_i64x2_arith(comp_ctx, func_ctx, arith_op))
- return false;
- break;
- }
- case SIMD_i8x16_neg:
- {
- if (!aot_compile_simd_i8x16_neg(comp_ctx, func_ctx))
- return false;
- break;
- }
- case SIMD_i16x8_neg:
- {
- if (!aot_compile_simd_i16x8_neg(comp_ctx, func_ctx))
- return false;
- break;
- }
- case SIMD_i32x4_neg:
- {
- if (!aot_compile_simd_i32x4_neg(comp_ctx, func_ctx))
- return false;
- break;
- }
- case SIMD_i64x2_neg:
- {
- if (!aot_compile_simd_i64x2_neg(comp_ctx, func_ctx))
- return false;
- break;
- }
- case SIMD_i8x16_add_saturate_s:
- case SIMD_i8x16_add_saturate_u:
- {
- if (!aot_compile_simd_i8x16_saturate(comp_ctx, func_ctx, V128_ADD,
- opcode == SIMD_i8x16_add_saturate_s
- ? true : false))
- return false;
- break;
- }
- case SIMD_i8x16_sub_saturate_s:
- case SIMD_i8x16_sub_saturate_u:
- {
- if (!aot_compile_simd_i8x16_saturate(comp_ctx, func_ctx, V128_SUB,
- opcode == SIMD_i8x16_sub_saturate_s
- ? true : false))
- return false;
- break;
- }
- case SIMD_i16x8_add_saturate_s:
- case SIMD_i16x8_add_saturate_u:
- {
- if (!aot_compile_simd_i16x8_saturate(comp_ctx, func_ctx, V128_ADD,
- opcode == SIMD_i16x8_add_saturate_s
- ? true : false))
- return false;
- break;
- }
- case SIMD_i16x8_sub_saturate_s:
- case SIMD_i16x8_sub_saturate_u:
- {
- if (!aot_compile_simd_i16x8_saturate(comp_ctx, func_ctx, V128_SUB,
- opcode == SIMD_i16x8_sub_saturate_s
- ? true : false))
- return false;
- break;
- }
- case SIMD_i8x16_min_s:
- case SIMD_i8x16_min_u:
- {
- if (!aot_compile_simd_i8x16_cmp(comp_ctx, func_ctx, V128_MIN,
- opcode == SIMD_i8x16_min_s
- ? true : false))
- return false;
- break;
- }
- case SIMD_i8x16_max_s:
- case SIMD_i8x16_max_u:
- {
- if (!aot_compile_simd_i8x16_cmp(comp_ctx, func_ctx, V128_MAX,
- opcode == SIMD_i8x16_max_s
- ? true : false))
- return false;
- break;
- }
- case SIMD_i16x8_min_s:
- case SIMD_i16x8_min_u:
- {
- if (!aot_compile_simd_i16x8_cmp(comp_ctx, func_ctx, V128_MIN,
- opcode == SIMD_i16x8_min_s
- ? true : false))
- return false;
- break;
- }
- case SIMD_i16x8_max_s:
- case SIMD_i16x8_max_u:
- {
- if (!aot_compile_simd_i16x8_cmp(comp_ctx, func_ctx, V128_MAX,
- opcode == SIMD_i16x8_max_s
- ? true : false))
- return false;
- break;
- }
- case SIMD_i32x4_min_s:
- case SIMD_i32x4_min_u:
- {
- if (!aot_compile_simd_i32x4_cmp(comp_ctx, func_ctx, V128_MIN,
- opcode == SIMD_i32x4_min_s
- ? true : false))
- return false;
- break;
- }
- case SIMD_i32x4_max_s:
- case SIMD_i32x4_max_u:
- {
- if (!aot_compile_simd_i32x4_cmp(comp_ctx, func_ctx, V128_MAX,
- opcode == SIMD_i32x4_max_s
- ? true : false))
- return false;
- break;
- }
- case SIMD_i8x16_abs:
- {
- if (!aot_compile_simd_i8x16_abs(comp_ctx, func_ctx))
- return false;
- break;
- }
- case SIMD_i16x8_abs:
- {
- if (!aot_compile_simd_i16x8_abs(comp_ctx, func_ctx))
- return false;
- break;
- }
- case SIMD_i32x4_abs:
- {
- if (!aot_compile_simd_i32x4_abs(comp_ctx, func_ctx))
- return false;
- break;
- }
- case SIMD_i8x16_avgr_u:
- {
- if (!aot_compile_simd_i8x16_avgr_u(comp_ctx, func_ctx))
- return false;
- break;
- }
- case SIMD_i16x8_avgr_u:
- {
- if (!aot_compile_simd_i16x8_avgr_u(comp_ctx, func_ctx))
- return false;
- break;
- }
- case SIMD_i8x16_any_true:
- {
- if (!aot_compile_simd_i8x16_any_true(comp_ctx, func_ctx))
- return false;
- break;
- }
- case SIMD_i16x8_any_true:
- {
- if (!aot_compile_simd_i16x8_any_true(comp_ctx, func_ctx))
- return false;
- break;
- }
- case SIMD_i32x4_any_true:
- {
- if (!aot_compile_simd_i32x4_any_true(comp_ctx, func_ctx))
- return false;
- break;
- }
- case SIMD_i8x16_all_true:
- {
- if (!aot_compile_simd_i8x16_all_true(comp_ctx, func_ctx))
- return false;
- break;
- }
- case SIMD_i16x8_all_true:
- {
- if (!aot_compile_simd_i16x8_all_true(comp_ctx, func_ctx))
- return false;
- break;
- }
- case SIMD_i32x4_all_true:
- {
- if (!aot_compile_simd_i32x4_all_true(comp_ctx, func_ctx))
- return false;
- break;
- }
- case SIMD_i8x16_bitmask:
- {
- if (!aot_compile_simd_i8x16_bitmask(comp_ctx, func_ctx))
- return false;
- break;
- }
- case SIMD_i16x8_bitmask:
- {
- if (!aot_compile_simd_i16x8_bitmask(comp_ctx, func_ctx))
- return false;
- break;
- }
- case SIMD_i32x4_bitmask:
- {
- if (!aot_compile_simd_i32x4_bitmask(comp_ctx, func_ctx))
- return false;
- break;
- }
- case SIMD_i8x16_shl:
- case SIMD_i8x16_shr_s:
- case SIMD_i8x16_shr_u:
- {
- if (!aot_compile_simd_i8x16_shift(comp_ctx, func_ctx,
- INT_SHL + opcode - SIMD_i8x16_shl))
- return false;
- break;
- }
- case SIMD_i16x8_shl:
- case SIMD_i16x8_shr_s:
- case SIMD_i16x8_shr_u:
- {
- if (!aot_compile_simd_i16x8_shift(comp_ctx, func_ctx,
- INT_SHL + opcode - SIMD_i16x8_shl))
- return false;
- break;
- }
- case SIMD_i32x4_shl:
- case SIMD_i32x4_shr_s:
- case SIMD_i32x4_shr_u:
- {
- if (!aot_compile_simd_i32x4_shift(comp_ctx, func_ctx,
- INT_SHL + opcode - SIMD_i32x4_shl))
- return false;
- break;
- }
- case SIMD_i64x2_shl:
- case SIMD_i64x2_shr_s:
- case SIMD_i64x2_shr_u:
- {
- if (!aot_compile_simd_i64x2_shift(comp_ctx, func_ctx,
- INT_SHL + opcode - SIMD_i64x2_shl))
- return false;
- break;
- }
- case SIMD_i8x16_narrow_i16x8_s:
- case SIMD_i8x16_narrow_i16x8_u:
- {
- bool is_signed = (opcode == SIMD_i8x16_narrow_i16x8_s)
- ? true : false;
- if (!aot_compile_simd_i8x16_narrow_i16x8(comp_ctx, func_ctx,
- is_signed))
- return false;
- break;
- }
- case SIMD_i16x8_narrow_i32x4_s:
- case SIMD_i16x8_narrow_i32x4_u:
- {
- bool is_signed = (opcode == SIMD_i16x8_narrow_i32x4_s)
- ? true : false;
- if (!aot_compile_simd_i16x8_narrow_i32x4(comp_ctx, func_ctx,
- is_signed))
- return false;
- break;
- }
- case SIMD_i16x8_widen_low_i8x16_s:
- case SIMD_i16x8_widen_high_i8x16_s:
- {
- bool is_low = (opcode == SIMD_i16x8_widen_low_i8x16_s)
- ? true : false;
- if (!aot_compile_simd_i16x8_widen_i8x16(comp_ctx, func_ctx,
- is_low, true))
- return false;
- break;
- }
- case SIMD_i16x8_widen_low_i8x16_u:
- case SIMD_i16x8_widen_high_i8x16_u:
- {
- bool is_low = (opcode == SIMD_i16x8_widen_low_i8x16_u)
- ? true : false;
- if (!aot_compile_simd_i16x8_widen_i8x16(comp_ctx, func_ctx,
- is_low, false))
- return false;
- break;
- }
- case SIMD_i32x4_widen_low_i16x8_s:
- case SIMD_i32x4_widen_high_i16x8_s:
- {
- bool is_low = (opcode == SIMD_i32x4_widen_low_i16x8_s)
- ? true : false;
- if (!aot_compile_simd_i32x4_widen_i16x8(comp_ctx, func_ctx,
- is_low, true))
- return false;
- break;
- }
- case SIMD_i32x4_widen_low_i16x8_u:
- case SIMD_i32x4_widen_high_i16x8_u:
- {
- bool is_low = (opcode == SIMD_i32x4_widen_low_i16x8_u)
- ? true : false;
- if (!aot_compile_simd_i32x4_widen_i16x8(comp_ctx, func_ctx,
- is_low, false))
- return false;
- break;
- }
- case SIMD_i32x4_trunc_sat_f32x4_s:
- case SIMD_i32x4_trunc_sat_f32x4_u:
- {
- bool is_signed = (opcode == SIMD_i32x4_trunc_sat_f32x4_s)
- ? true : false;
- if (!aot_compile_simd_i32x4_trunc_sat_f32x4(comp_ctx, func_ctx,
- is_signed))
- return false;
- break;
- }
- case SIMD_f32x4_convert_i32x4_s:
- case SIMD_f32x4_convert_i32x4_u:
- {
- bool is_signed = (opcode == SIMD_f32x4_convert_i32x4_s)
- ? true : false;
- if (!aot_compile_simd_f32x4_convert_i32x4(comp_ctx, func_ctx,
- is_signed))
- return false;
- break;
- }
- case SIMD_f32x4_add:
- case SIMD_f32x4_sub:
- case SIMD_f32x4_mul:
- case SIMD_f32x4_div:
- case SIMD_f32x4_min:
- case SIMD_f32x4_max:
- {
- if (!aot_compile_simd_f32x4_arith(comp_ctx, func_ctx,
- FLOAT_ADD + opcode - SIMD_f32x4_add))
- return false;
- break;
- }
- case SIMD_f64x2_add:
- case SIMD_f64x2_sub:
- case SIMD_f64x2_mul:
- case SIMD_f64x2_div:
- case SIMD_f64x2_min:
- case SIMD_f64x2_max:
- {
- if (!aot_compile_simd_f64x2_arith(comp_ctx, func_ctx,
- FLOAT_ADD + opcode - SIMD_f64x2_add))
- return false;
- break;
- }
- case SIMD_f32x4_neg:
- {
- if (!aot_compile_simd_f32x4_neg(comp_ctx, func_ctx))
- return false;
- break;
- }
- case SIMD_f64x2_neg:
- {
- if (!aot_compile_simd_f64x2_neg(comp_ctx, func_ctx))
- return false;
- break;
- }
- case SIMD_f32x4_abs:
- {
- if (!aot_compile_simd_f32x4_abs(comp_ctx, func_ctx))
- return false;
- break;
- }
- case SIMD_f64x2_abs:
- {
- if (!aot_compile_simd_f64x2_abs(comp_ctx, func_ctx))
- return false;
- break;
- }
- case SIMD_f32x4_sqrt:
- {
- if (!aot_compile_simd_f32x4_sqrt(comp_ctx, func_ctx))
- return false;
- break;
- }
- case SIMD_f64x2_sqrt:
- {
- if (!aot_compile_simd_f64x2_sqrt(comp_ctx, func_ctx))
- return false;
- break;
- }
- case SIMD_f32x4_ceil:
- {
- if (!aot_compile_simd_f32x4_ceil(comp_ctx, func_ctx))
- return false;
- break;
- }
- case SIMD_f64x2_ceil:
- {
- if (!aot_compile_simd_f64x2_ceil(comp_ctx, func_ctx))
- return false;
- break;
- }
- case SIMD_f32x4_floor:
- {
- if (!aot_compile_simd_f32x4_floor(comp_ctx, func_ctx))
- return false;
- break;
- }
- case SIMD_f64x2_floor:
- {
- if (!aot_compile_simd_f64x2_floor(comp_ctx, func_ctx))
- return false;
- break;
- }
- case SIMD_f32x4_trunc:
- {
- if (!aot_compile_simd_f32x4_trunc(comp_ctx, func_ctx))
- return false;
- break;
- }
- case SIMD_f64x2_trunc:
- {
- if (!aot_compile_simd_f64x2_trunc(comp_ctx, func_ctx))
- return false;
- break;
- }
- case SIMD_f32x4_nearest:
- {
- if (!aot_compile_simd_f32x4_nearest(comp_ctx, func_ctx))
- return false;
- break;
- }
- case SIMD_f64x2_nearest:
- {
- if (!aot_compile_simd_f64x2_nearest(comp_ctx, func_ctx))
- return false;
- break;
- }
- default:
- aot_set_last_error("unsupported opcode");
- return false;
- }
- break;
- }
- #endif /* end of WASM_ENABLE_SIMD */
- default:
- aot_set_last_error("unsupported opcode");
- return false;
- }
- }
- /* Move func_return block to the bottom */
- if (func_ctx->func_return_block) {
- LLVMBasicBlockRef last_block =
- LLVMGetLastBasicBlock(func_ctx->func);
- if (last_block != func_ctx->func_return_block)
- LLVMMoveBasicBlockAfter(func_ctx->func_return_block,
- last_block);
- }
- /* Move got_exception block to the bottom */
- if (func_ctx->got_exception_block) {
- LLVMBasicBlockRef last_block =
- LLVMGetLastBasicBlock(func_ctx->func);
- if (last_block != func_ctx->got_exception_block)
- LLVMMoveBasicBlockAfter(func_ctx->got_exception_block,
- last_block);
- }
- return true;
- #if WASM_ENABLE_SIMD != 0
- unsupport_simd:
- aot_set_last_error("SIMD instruction was found, "
- "try adding --enable-simd option?");
- return false;
- #endif
- #if WASM_ENABLE_REF_TYPES != 0
- unsupport_ref_types:
- aot_set_last_error("reference type instruction was found, "
- "try adding --enable-ref-types option?");
- return false;
- #endif
- fail:
- return false;
- }
- bool
- aot_compile_wasm(AOTCompContext *comp_ctx)
- {
- char *msg = NULL;
- bool ret;
- uint32 i;
- bh_print_time("Begin to compile WASM bytecode to LLVM IR");
- for (i = 0; i < comp_ctx->func_ctx_count; i++)
- if (!aot_compile_func(comp_ctx, i)) {
- #if 0
- LLVMDumpModule(comp_ctx->module);
- char *err;
- LLVMTargetMachineEmitToFile(comp_ctx->target_machine, comp_ctx->module,
- "./test.o", LLVMObjectFile, &err);
- #endif
- return false;
- }
- #if 0
- LLVMDumpModule(comp_ctx->module);
- /* Clear error no, LLVMDumpModule may set errno */
- errno = 0;
- #endif
- bh_print_time("Begin to verify LLVM module");
- ret = LLVMVerifyModule(comp_ctx->module, LLVMPrintMessageAction, &msg);
- if (!ret && msg) {
- if (msg[0] != '\0') {
- aot_set_last_error(msg);
- LLVMDisposeMessage(msg);
- return false;
- }
- LLVMDisposeMessage(msg);
- }
- bh_print_time("Begin to run function optimization passes");
- if (comp_ctx->optimize) {
- LLVMInitializeFunctionPassManager(comp_ctx->pass_mgr);
- for (i = 0; i < comp_ctx->func_ctx_count; i++)
- LLVMRunFunctionPassManager(comp_ctx->pass_mgr,
- comp_ctx->func_ctxes[i]->func);
- }
- return true;
- }
- bool
- aot_emit_llvm_file(AOTCompContext *comp_ctx, const char *file_name)
- {
- char *err = NULL;
- bh_print_time("Begin to emit LLVM IR file");
- if (LLVMPrintModuleToFile(comp_ctx->module, file_name, &err) != 0) {
- if (err) {
- LLVMDisposeMessage(err);
- err = NULL;
- }
- aot_set_last_error("emit llvm ir to file failed.");
- return false;
- }
- return true;
- }
- bool
- aot_emit_object_file(AOTCompContext *comp_ctx, char *file_name)
- {
- char *err = NULL;
- bh_print_time("Begin to emit object file");
- if (LLVMTargetMachineEmitToFile(comp_ctx->target_machine,
- comp_ctx->module,
- file_name,
- LLVMObjectFile,
- &err) != 0) {
- if (err) {
- LLVMDisposeMessage(err);
- err = NULL;
- }
- aot_set_last_error("emit elf to memory buffer failed.");
- return false;
- }
- return true;
- }
- #if WASM_ENABLE_REF_TYPES != 0
- extern void
- wasm_set_ref_types_flag(bool enable);
- #endif
- typedef struct AOTFileMap {
- uint8 *wasm_file_buf;
- uint32 wasm_file_size;
- uint8 *aot_file_buf;
- uint32 aot_file_size;
- struct AOTFileMap *next;
- } AOTFileMap;
- static bool aot_compile_wasm_file_inited = false;
- static AOTFileMap *aot_file_maps = NULL;
- static korp_mutex aot_file_map_lock;
- bool
- aot_compile_wasm_file_init()
- {
- if (aot_compile_wasm_file_inited) {
- return true;
- }
- if (BHT_OK != os_mutex_init(&aot_file_map_lock)) {
- return false;
- }
- aot_file_maps = NULL;
- aot_compile_wasm_file_inited = true;
- return true;
- }
- void
- aot_compile_wasm_file_destroy()
- {
- AOTFileMap *file_map = aot_file_maps, *file_map_next;
- if (!aot_compile_wasm_file_inited) {
- return;
- }
- while (file_map) {
- file_map_next = file_map->next;
- wasm_runtime_free(file_map->wasm_file_buf);
- wasm_runtime_free(file_map->aot_file_buf);
- wasm_runtime_free(file_map);
- file_map = file_map_next;
- }
- aot_file_maps = NULL;
- os_mutex_destroy(&aot_file_map_lock);
- aot_compile_wasm_file_inited = false;
- }
- static void
- set_error_buf(char *error_buf, uint32 error_buf_size, const char *string)
- {
- if (error_buf != NULL) {
- snprintf(error_buf, error_buf_size,
- "WASM module load failed: %s", string);
- }
- }
- uint8*
- aot_compile_wasm_file(const uint8 *wasm_file_buf, uint32 wasm_file_size,
- uint32 opt_level, uint32 size_level,
- char *error_buf, uint32 error_buf_size,
- uint32 *p_aot_file_size)
- {
- WASMModule *wasm_module = NULL;
- AOTCompData *comp_data = NULL;
- AOTCompContext *comp_ctx = NULL;
- RuntimeInitArgs init_args;
- AOTCompOption option = { 0 };
- AOTFileMap *file_map = NULL, *file_map_next;
- uint8 *wasm_file_buf_cloned = NULL;
- uint8 *aot_file_buf = NULL;
- uint32 aot_file_size;
- option.is_jit_mode = false;
- option.opt_level = opt_level;
- option.size_level = size_level;
- option.output_format = AOT_FORMAT_FILE;
- /* default value, enable or disable depends on the platform */
- option.bounds_checks = 2;
- option.enable_aux_stack_check = true;
- #if WASM_ENABLE_BULK_MEMORY != 0
- option.enable_bulk_memory = true;
- #endif
- #if WASM_ENABLE_THREAD_MGR != 0
- option.enable_thread_mgr = true;
- #endif
- #if WASM_ENABLE_TAIL_CALL != 0
- option.enable_tail_call = true;
- #endif
- #if WASM_ENABLE_SIMD != 0
- option.enable_simd = true;
- #endif
- #if WASM_ENABLE_REF_TYPES != 0
- option.enable_ref_types = true;
- #endif
- #if (WASM_ENABLE_PERF_PROFILING != 0) || (WASM_ENABLE_DUMP_CALL_STACK != 0)
- option.enable_aux_stack_frame = true;
- #endif
- #if WASM_ENABLE_REF_TYPES != 0
- wasm_set_ref_types_flag(option.enable_ref_types);
- #endif
- memset(&init_args, 0, sizeof(RuntimeInitArgs));
- init_args.mem_alloc_type = Alloc_With_Allocator;
- init_args.mem_alloc_option.allocator.malloc_func = malloc;
- init_args.mem_alloc_option.allocator.realloc_func = realloc;
- init_args.mem_alloc_option.allocator.free_func = free;
- os_mutex_lock(&aot_file_map_lock);
- /* lookup the file maps */
- file_map = aot_file_maps;
- while (file_map) {
- file_map_next = file_map->next;
- if (wasm_file_size == file_map->wasm_file_size
- && memcmp(wasm_file_buf, file_map->wasm_file_buf,
- wasm_file_size) == 0) {
- os_mutex_unlock(&aot_file_map_lock);
- /* found */
- *p_aot_file_size = file_map->aot_file_size;
- return file_map->aot_file_buf;
- }
- file_map = file_map_next;
- }
- /* not found, initialize file map and clone wasm file */
- if (!(file_map = wasm_runtime_malloc(sizeof(AOTFileMap)))
- || !(wasm_file_buf_cloned = wasm_runtime_malloc(wasm_file_size))) {
- set_error_buf(error_buf, error_buf_size, "allocate memory failed");
- goto fail1;
- }
- bh_memcpy_s(wasm_file_buf_cloned, wasm_file_size,
- wasm_file_buf, wasm_file_size);
- memset(file_map, 0, sizeof(AOTFileMap));
- file_map->wasm_file_buf = wasm_file_buf_cloned;
- file_map->wasm_file_size = wasm_file_size;
- /* load WASM module */
- if (!(wasm_module = wasm_load(wasm_file_buf, wasm_file_size,
- error_buf, sizeof(error_buf)))) {
- goto fail1;
- }
- if (!(comp_data = aot_create_comp_data(wasm_module))) {
- set_error_buf(error_buf, error_buf_size, aot_get_last_error());
- goto fail2;
- }
- if (!(comp_ctx = aot_create_comp_context(comp_data, &option))) {
- set_error_buf(error_buf, error_buf_size, aot_get_last_error());
- goto fail3;
- }
- if (!aot_compile_wasm(comp_ctx)) {
- set_error_buf(error_buf, error_buf_size, aot_get_last_error());
- goto fail4;
- }
- if (!(aot_file_buf = aot_emit_aot_file_buf(comp_ctx, comp_data,
- &aot_file_size))) {
- set_error_buf(error_buf, error_buf_size, aot_get_last_error());
- goto fail4;
- }
- file_map->aot_file_buf = aot_file_buf;
- file_map->aot_file_size = aot_file_size;
- if (!aot_file_maps)
- aot_file_maps = file_map;
- else {
- file_map->next = aot_file_maps;
- aot_file_maps = file_map;
- }
- *p_aot_file_size = aot_file_size;
- fail4:
- /* Destroy compiler context */
- aot_destroy_comp_context(comp_ctx);
- fail3:
- /* Destroy compile data */
- aot_destroy_comp_data(comp_data);
- fail2:
- wasm_unload(wasm_module);
- fail1:
- if (!aot_file_buf) {
- if (wasm_file_buf_cloned)
- wasm_runtime_free(wasm_file_buf_cloned);
- if (file_map)
- wasm_runtime_free(file_map);
- }
- os_mutex_unlock(&aot_file_map_lock);
- return aot_file_buf;
- }
|