summaryrefslogtreecommitdiffstats
path: root/fluent-bit/lib/wasm-micro-runtime-WAMR-1.2.2/core/iwasm/compilation/aot_emit_parametric.c
blob: 8b1a9e6da379a22772dc54139d68d00447cc31a3 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
/*
 * Copyright (C) 2019 Intel Corporation. All rights reserved.
 * SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
 */

#include "aot_emit_parametric.h"

static bool
pop_value_from_wasm_stack(AOTCompContext *comp_ctx, AOTFuncContext *func_ctx,
                          LLVMValueRef *p_value, bool is_32, uint8 *p_type)
{
    AOTValue *aot_value;
    uint8 type;

    if (!func_ctx->block_stack.block_list_end) {
        aot_set_last_error("WASM block stack underflow.");
        return false;
    }
    if (!func_ctx->block_stack.block_list_end->value_stack.value_list_end) {
        aot_set_last_error("WASM data stack underflow.");
        return false;
    }

    aot_value =
        aot_value_stack_pop(&func_ctx->block_stack.block_list_end->value_stack);
    type = aot_value->type;

    if (aot_value->type == VALUE_TYPE_I1) {
        if (!(aot_value->value =
                  LLVMBuildZExt(comp_ctx->builder, aot_value->value, I32_TYPE,
                                "val_s_ext"))) {
            aot_set_last_error("llvm build sign ext failed.");
            return false;
        }
        type = aot_value->type = VALUE_TYPE_I32;
    }

    if (p_type != NULL) {
        *p_type = aot_value->type;
    }
    if (p_value != NULL) {
        *p_value = aot_value->value;
    }

    wasm_runtime_free(aot_value);

    /* is_32: i32, f32, ref.func, ref.extern, v128 */
    if (is_32
        && !(type == VALUE_TYPE_I32 || type == VALUE_TYPE_F32
             || type == VALUE_TYPE_FUNCREF || type == VALUE_TYPE_EXTERNREF
             || type == VALUE_TYPE_V128)) {
        aot_set_last_error("invalid WASM stack data type.");
        return false;
    }

    /* !is_32: i64, f64 */
    if (!is_32 && !(type == VALUE_TYPE_I64 || type == VALUE_TYPE_F64)) {
        aot_set_last_error("invalid WASM stack data type.");
        return false;
    }

    return true;
}

bool
aot_compile_op_drop(AOTCompContext *comp_ctx, AOTFuncContext *func_ctx,
                    bool is_drop_32)
{
    if (!pop_value_from_wasm_stack(comp_ctx, func_ctx, NULL, is_drop_32, NULL))
        return false;

    return true;
}

bool
aot_compile_op_select(AOTCompContext *comp_ctx, AOTFuncContext *func_ctx,
                      bool is_select_32)
{
    LLVMValueRef val1, val2, cond, selected;
    uint8 val1_type, val2_type;

    POP_COND(cond);

    if (!pop_value_from_wasm_stack(comp_ctx, func_ctx, &val2, is_select_32,
                                   &val2_type)
        || !pop_value_from_wasm_stack(comp_ctx, func_ctx, &val1, is_select_32,
                                      &val1_type))
        return false;

    if (val1_type != val2_type) {
        aot_set_last_error("invalid stack values with different type");
        return false;
    }

    if (!(selected =
              LLVMBuildSelect(comp_ctx->builder, cond, val1, val2, "select"))) {
        aot_set_last_error("llvm build select failed.");
        return false;
    }

    PUSH(selected, val1_type);

    return true;

fail:
    return false;
}