/* * Copyright (C) 2019 Intel Corporation. All rights reserved. * SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception */ #ifndef _AOT_COMPILER_H_ #define _AOT_COMPILER_H_ #include "aot.h" #include "aot_llvm.h" #ifdef __cplusplus extern "C" { #endif typedef enum IntCond { INT_EQZ = 0, INT_EQ, INT_NE, INT_LT_S, INT_LT_U, INT_GT_S, INT_GT_U, INT_LE_S, INT_LE_U, INT_GE_S, INT_GE_U } IntCond; typedef enum FloatCond { FLOAT_EQ = 0, FLOAT_NE, FLOAT_LT, FLOAT_GT, FLOAT_LE, FLOAT_GE } FloatCond; typedef enum IntArithmetic { INT_ADD = 0, INT_SUB, INT_MUL, INT_DIV_S, INT_DIV_U, INT_REM_S, INT_REM_U } IntArithmetic; typedef enum V128Arithmetic { V128_ADD = 0, V128_ADD_SATURATE_S, V128_ADD_SATURATE_U, V128_SUB, V128_SUB_SATURATE_S, V128_SUB_SATURATE_U, V128_MUL, V128_DIV, V128_NEG, V128_MIN, V128_MAX, } V128Arithmetic; typedef enum IntBitwise { INT_AND = 0, INT_OR, INT_XOR, } IntBitwise; typedef enum V128Bitwise { V128_NOT, V128_AND, V128_ANDNOT, V128_OR, V128_XOR, V128_BITSELECT } V128Bitwise; typedef enum IntShift { INT_SHL = 0, INT_SHR_S, INT_SHR_U, INT_ROTL, INT_ROTR } IntShift; typedef enum FloatMath { FLOAT_ABS = 0, FLOAT_NEG, FLOAT_CEIL, FLOAT_FLOOR, FLOAT_TRUNC, FLOAT_NEAREST, FLOAT_SQRT } FloatMath; typedef enum FloatArithmetic { FLOAT_ADD = 0, FLOAT_SUB, FLOAT_MUL, FLOAT_DIV, FLOAT_MIN, FLOAT_MAX } FloatArithmetic; #define CHECK_STACK() do { \ if (!func_ctx->block_stack.block_list_end) { \ aot_set_last_error("WASM block stack underflow."); \ goto fail; \ } \ if (!func_ctx->block_stack.block_list_end-> \ value_stack.value_list_end) { \ aot_set_last_error("WASM data stack underflow."); \ goto fail; \ } \ } while (0) #define POP(llvm_value, value_type) do { \ AOTValue *aot_value; \ CHECK_STACK(); \ aot_value = aot_value_stack_pop \ (&func_ctx->block_stack.block_list_end->value_stack); \ if ((value_type != VALUE_TYPE_I32 \ && aot_value->type != value_type) \ || (value_type == VALUE_TYPE_I32 \ && (aot_value->type != VALUE_TYPE_I32 \ && aot_value->type != VALUE_TYPE_I1))) { \ aot_set_last_error("invalid WASM stack data type."); \ wasm_runtime_free(aot_value); \ goto fail; \ } \ if (aot_value->type == value_type) \ llvm_value = aot_value->value; \ else { \ bh_assert(aot_value->type == VALUE_TYPE_I1); \ if (!(llvm_value = LLVMBuildZExt(comp_ctx->builder, \ aot_value->value, I32_TYPE, "i1toi32"))) { \ aot_set_last_error("invalid WASM stack data type.");\ wasm_runtime_free(aot_value); \ goto fail; \ } \ } \ wasm_runtime_free(aot_value); \ } while (0) #define POP_I32(v) POP(v, VALUE_TYPE_I32) #define POP_I64(v) POP(v, VALUE_TYPE_I64) #define POP_F32(v) POP(v, VALUE_TYPE_F32) #define POP_F64(v) POP(v, VALUE_TYPE_F64) #define POP_V128(v) POP(v, VALUE_TYPE_V128) #define POP_COND(llvm_value) do { \ AOTValue *aot_value; \ CHECK_STACK(); \ aot_value = aot_value_stack_pop \ (&func_ctx->block_stack.block_list_end->value_stack); \ if (aot_value->type != VALUE_TYPE_I1 \ && aot_value->type != VALUE_TYPE_I32) { \ aot_set_last_error("invalid WASM stack data type."); \ wasm_runtime_free(aot_value); \ goto fail; \ } \ if (aot_value->type == VALUE_TYPE_I1) \ llvm_value = aot_value->value; \ else { \ if (!(llvm_value = LLVMBuildICmp(comp_ctx->builder, \ LLVMIntNE, aot_value->value, I32_ZERO, \ "i1_cond"))){ \ aot_set_last_error("llvm build trunc failed."); \ wasm_runtime_free(aot_value); \ goto fail; \ } \ } \ wasm_runtime_free(aot_value); \ } while (0) #define PUSH(llvm_value, value_type) do { \ AOTValue *aot_value; \ if (!func_ctx->block_stack.block_list_end) { \ aot_set_last_error("WASM block stack underflow."); \ goto fail; \ } \ aot_value = wasm_runtime_malloc(sizeof(AOTValue)); \ if (!aot_value) { \ aot_set_last_error("allocate memory failed."); \ goto fail; \ } \ memset(aot_value, 0, sizeof(AOTValue)); \ aot_value->type = value_type; \ aot_value->value = llvm_value; \ aot_value_stack_push \ (&func_ctx->block_stack.block_list_end->value_stack,\ aot_value); \ } while (0) #define PUSH_I32(v) PUSH(v, VALUE_TYPE_I32) #define PUSH_I64(v) PUSH(v, VALUE_TYPE_I64) #define PUSH_F32(v) PUSH(v, VALUE_TYPE_F32) #define PUSH_F64(v) PUSH(v, VALUE_TYPE_F64) #define PUSH_V128(v) PUSH(v, VALUE_TYPE_V128) #define PUSH_COND(v) PUSH(v, VALUE_TYPE_I1) #define TO_LLVM_TYPE(wasm_type) \ wasm_type_to_llvm_type(&comp_ctx->basic_types, wasm_type) #define I32_TYPE comp_ctx->basic_types.int32_type #define I64_TYPE comp_ctx->basic_types.int64_type #define F32_TYPE comp_ctx->basic_types.float32_type #define F64_TYPE comp_ctx->basic_types.float64_type #define VOID_TYPE comp_ctx->basic_types.void_type #define INT1_TYPE comp_ctx->basic_types.int1_type #define INT8_TYPE comp_ctx->basic_types.int8_type #define INT16_TYPE comp_ctx->basic_types.int16_type #define MD_TYPE comp_ctx->basic_types.meta_data_type #define INT8_PTR_TYPE comp_ctx->basic_types.int8_ptr_type #define INT16_PTR_TYPE comp_ctx->basic_types.int16_ptr_type #define INT32_PTR_TYPE comp_ctx->basic_types.int32_ptr_type #define INT64_PTR_TYPE comp_ctx->basic_types.int64_ptr_type #define F32_PTR_TYPE comp_ctx->basic_types.float32_ptr_type #define F64_PTR_TYPE comp_ctx->basic_types.float64_ptr_type #define I32_CONST(v) LLVMConstInt(I32_TYPE, v, true) #define I64_CONST(v) LLVMConstInt(I64_TYPE, v, true) #define F32_CONST(v) LLVMConstReal(F32_TYPE, v) #define F64_CONST(v) LLVMConstReal(F64_TYPE, v) #define I8_CONST(v) LLVMConstInt(INT8_TYPE, v, true) #define I8_ZERO (comp_ctx->llvm_consts.i8_zero) #define I32_ZERO (comp_ctx->llvm_consts.i32_zero) #define I64_ZERO (comp_ctx->llvm_consts.i64_zero) #define F32_ZERO (comp_ctx->llvm_consts.f32_zero) #define F64_ZERO (comp_ctx->llvm_consts.f64_zero) #define I32_ONE (comp_ctx->llvm_consts.i32_one) #define I32_TWO (comp_ctx->llvm_consts.i32_two) #define I32_THREE (comp_ctx->llvm_consts.i32_three) #define I32_FOUR (comp_ctx->llvm_consts.i32_four) #define I32_EIGHT (comp_ctx->llvm_consts.i32_eight) #define I32_NEG_ONE (comp_ctx->llvm_consts.i32_neg_one) #define I64_NEG_ONE (comp_ctx->llvm_consts.i64_neg_one) #define I32_MIN (comp_ctx->llvm_consts.i32_min) #define I64_MIN (comp_ctx->llvm_consts.i64_min) #define I32_31 (comp_ctx->llvm_consts.i32_31) #define I32_32 (comp_ctx->llvm_consts.i32_32) #define I64_63 (comp_ctx->llvm_consts.i64_63) #define I64_64 (comp_ctx->llvm_consts.i64_64) #define V128_TYPE comp_ctx->basic_types.v128_type #define V128_PTR_TYPE comp_ctx->basic_types.v128_ptr_type #define V128_i8x16_TYPE comp_ctx->basic_types.i8x16_vec_type #define V128_i16x8_TYPE comp_ctx->basic_types.i16x8_vec_type #define V128_i32x4_TYPE comp_ctx->basic_types.i32x4_vec_type #define V128_i64x2_TYPE comp_ctx->basic_types.i64x2_vec_type #define V128_f32x4_TYPE comp_ctx->basic_types.f32x4_vec_type #define V128_f64x2_TYPE comp_ctx->basic_types.f64x2_vec_type #define V128_ZERO (comp_ctx->llvm_consts.v128_zero) #define V128_i8x16_ZERO (comp_ctx->llvm_consts.i8x16_vec_zero) #define V128_i16x8_ZERO (comp_ctx->llvm_consts.i16x8_vec_zero) #define V128_i32x4_ZERO (comp_ctx->llvm_consts.i32x4_vec_zero) #define V128_i64x2_ZERO (comp_ctx->llvm_consts.i64x2_vec_zero) #define V128_f32x4_ZERO (comp_ctx->llvm_consts.f32x4_vec_zero) #define V128_f64x2_ZERO (comp_ctx->llvm_consts.f64x2_vec_zero) #define TO_V128_i8x16(v) LLVMBuildBitCast(comp_ctx->builder, v, \ V128_i8x16_TYPE, "i8x16_val") #define TO_V128_i16x8(v) LLVMBuildBitCast(comp_ctx->builder, v, \ V128_i16x8_TYPE, "i16x8_val") #define TO_V128_i32x4(v) LLVMBuildBitCast(comp_ctx->builder, v, \ V128_i32x4_TYPE, "i32x4_val") #define TO_V128_i64x2(v) LLVMBuildBitCast(comp_ctx->builder, v, \ V128_i64x2_TYPE, "i64x2_val") #define TO_V128_f32x4(v) LLVMBuildBitCast(comp_ctx->builder, v, \ V128_f32x4_TYPE, "f32x4_val") #define TO_V128_f64x2(v) LLVMBuildBitCast(comp_ctx->builder, v, \ V128_f64x2_TYPE, "f64x2_val") #define CHECK_LLVM_CONST(v) do { \ if (!v) { \ aot_set_last_error("create llvm const failed."); \ goto fail; \ } \ } while (0) bool aot_compile_wasm(AOTCompContext *comp_ctx); bool aot_emit_llvm_file(AOTCompContext *comp_ctx, const char *file_name); bool aot_emit_aot_file(AOTCompContext *comp_ctx, AOTCompData *comp_data, const char *file_name); bool aot_emit_object_file(AOTCompContext *comp_ctx, char *file_name); #ifdef __cplusplus } /* end of extern "C" */ #endif #endif /* end of _AOT_COMPILER_H_ */