Add unit test suites (#3490)
This commit is contained in:
Zhang, Yi
335
tests/unit/compilation/aot_emit_memory_test.cc
Normal file
335
tests/unit/compilation/aot_emit_memory_test.cc
Normal file
@ -0,0 +1,335 @@
|
||||
/*
|
||||
* Copyright (C) 2019 Intel Corporation. All rights reserved.
|
||||
* SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
||||
*/
|
||||
|
||||
#include "gtest/gtest.h"
|
||||
#include "bh_platform.h"
|
||||
#include "bh_read_file.h"
|
||||
#include "aot_emit_memory.h"
|
||||
#include "test_helper.h"
|
||||
|
||||
#define DEFAULT_CYCLE_TIMES 0xFFFF
|
||||
#define DEFAULT_MAX_RAND_NUM 0xFFFFFFFF
|
||||
|
||||
static std::string CWD;
|
||||
static std::string MAIN_WASM = "/main.wasm";
|
||||
static char *WASM_FILE;
|
||||
|
||||
static std::string
|
||||
get_binary_path()
|
||||
{
|
||||
char cwd[1024];
|
||||
memset(cwd, 0, 1024);
|
||||
|
||||
if (readlink("/proc/self/exe", cwd, 1024) <= 0) {
|
||||
}
|
||||
|
||||
char *path_end = strrchr(cwd, '/');
|
||||
if (path_end != NULL) {
|
||||
*path_end = '\0';
|
||||
}
|
||||
|
||||
return std::string(cwd);
|
||||
}
|
||||
|
||||
class compilation_aot_emit_memory_test : public testing::Test
|
||||
{
|
||||
protected:
|
||||
void SetUp() override
|
||||
{
|
||||
CWD = get_binary_path();
|
||||
WASM_FILE = strdup((CWD + MAIN_WASM).c_str());
|
||||
AOTCompOption option = { 0 };
|
||||
|
||||
option.opt_level = 3;
|
||||
option.size_level = 3;
|
||||
option.output_format = AOT_FORMAT_FILE;
|
||||
/* default value, enable or disable depends on the platform */
|
||||
option.bounds_checks = 2;
|
||||
/* default value, enable or disable depends on the platform */
|
||||
option.stack_bounds_checks = 2;
|
||||
option.enable_simd = true;
|
||||
option.enable_aux_stack_check = true;
|
||||
option.enable_bulk_memory = true;
|
||||
option.enable_ref_types = true;
|
||||
|
||||
const char *wasm_file = WASM_FILE;
|
||||
unsigned int wasm_file_size = 0;
|
||||
unsigned char *wasm_file_buf = nullptr;
|
||||
char error_buf[128] = { 0 };
|
||||
|
||||
wasm_file_buf =
|
||||
(unsigned char *)bh_read_file_to_buffer(wasm_file, &wasm_file_size);
|
||||
EXPECT_NE(wasm_file_buf, nullptr);
|
||||
wasm_module = reinterpret_cast<WASMModule *>(wasm_runtime_load(
|
||||
wasm_file_buf, wasm_file_size, error_buf, sizeof(error_buf)));
|
||||
EXPECT_NE(wasm_module, nullptr);
|
||||
comp_data = aot_create_comp_data(wasm_module, NULL, false);
|
||||
EXPECT_NE(comp_data, nullptr);
|
||||
|
||||
// properly init compilation and function context, to do that,
|
||||
// use as a dummy module(instead of compile the function in it, simply
|
||||
// test the APIs)
|
||||
comp_ctx = aot_create_comp_context(comp_data, &option);
|
||||
EXPECT_NE(comp_ctx, nullptr);
|
||||
func_ctx = comp_ctx->func_ctxes[0];
|
||||
EXPECT_NE(func_ctx, nullptr);
|
||||
}
|
||||
|
||||
void TearDown() override
|
||||
{
|
||||
aot_destroy_comp_context(comp_ctx);
|
||||
aot_destroy_comp_data(comp_data);
|
||||
wasm_runtime_unload(reinterpret_cast<WASMModuleCommon *>(wasm_module));
|
||||
}
|
||||
|
||||
public:
|
||||
WASMModule *wasm_module = nullptr;
|
||||
AOTCompData *comp_data = nullptr;
|
||||
AOTCompContext *comp_ctx = nullptr;
|
||||
AOTFuncContext *func_ctx = nullptr;
|
||||
|
||||
WAMRRuntimeRAII<512 * 1024> runtime;
|
||||
};
|
||||
|
||||
TEST_F(compilation_aot_emit_memory_test, aot_check_memory_overflow)
|
||||
{
|
||||
uint32 offset = 64;
|
||||
uint32 bytes = 4;
|
||||
|
||||
for (uint32 i = 0; i < DEFAULT_CYCLE_TIMES; i++) {
|
||||
offset = (1 + (rand() % (DEFAULT_MAX_RAND_NUM - 1 + 1)));
|
||||
aot_check_memory_overflow(comp_ctx, func_ctx, offset, bytes, false);
|
||||
}
|
||||
}
|
||||
|
||||
TEST_F(compilation_aot_emit_memory_test, aot_compile_op_i32_load)
|
||||
{
|
||||
uint32 align = 0;
|
||||
uint32 offset = 1024;
|
||||
uint32 bytes = 0;
|
||||
bool sign = false;
|
||||
bool atomic = false;
|
||||
|
||||
for (uint32 i = 0; i < DEFAULT_CYCLE_TIMES; i++) {
|
||||
align = (1 + (rand() % (DEFAULT_MAX_RAND_NUM - 1 + 1)));
|
||||
offset = (1 + (rand() % (DEFAULT_MAX_RAND_NUM - 1 + 1)));
|
||||
bytes = (1 + (rand() % (4 - 1 + 1)));
|
||||
printf("---%d", aot_compile_op_i32_load(comp_ctx, func_ctx, align,
|
||||
offset, bytes, sign, atomic));
|
||||
}
|
||||
}
|
||||
|
||||
TEST_F(compilation_aot_emit_memory_test, aot_compile_op_i64_load)
|
||||
{
|
||||
uint32 align = 0;
|
||||
uint32 offset = 1024;
|
||||
uint32 bytes = 0;
|
||||
bool sign = false;
|
||||
bool atomic = false;
|
||||
|
||||
for (uint32 i = 0; i < DEFAULT_CYCLE_TIMES; i++) {
|
||||
align = (1 + (rand() % (DEFAULT_MAX_RAND_NUM - 1 + 1)));
|
||||
offset = (1 + (rand() % (DEFAULT_MAX_RAND_NUM - 1 + 1)));
|
||||
bytes = (1 + (rand() % (4 - 1 + 1)));
|
||||
sign = !sign;
|
||||
atomic = !atomic;
|
||||
aot_compile_op_i64_load(comp_ctx, func_ctx, align, offset, bytes, sign,
|
||||
atomic);
|
||||
}
|
||||
}
|
||||
|
||||
TEST_F(compilation_aot_emit_memory_test, aot_compile_op_f32_load)
|
||||
{
|
||||
uint32 align = 10;
|
||||
uint32 offset = 10;
|
||||
|
||||
for (uint32 i = 0; i < DEFAULT_CYCLE_TIMES; i++) {
|
||||
align = (1 + (rand() % (DEFAULT_MAX_RAND_NUM - 1 + 1)));
|
||||
offset = (1 + (rand() % (DEFAULT_MAX_RAND_NUM - 1 + 1)));
|
||||
aot_compile_op_f32_load(comp_ctx, func_ctx, align, offset);
|
||||
}
|
||||
}
|
||||
|
||||
TEST_F(compilation_aot_emit_memory_test, aot_compile_op_f64_load)
|
||||
{
|
||||
uint32 align = 10;
|
||||
uint32 offset = 10;
|
||||
|
||||
for (uint32 i = 0; i < DEFAULT_CYCLE_TIMES; i++) {
|
||||
align = (1 + (rand() % (DEFAULT_MAX_RAND_NUM - 1 + 1)));
|
||||
offset = (1 + (rand() % (DEFAULT_MAX_RAND_NUM - 1 + 1)));
|
||||
aot_compile_op_f64_load(comp_ctx, func_ctx, align, offset);
|
||||
}
|
||||
}
|
||||
|
||||
TEST_F(compilation_aot_emit_memory_test, aot_compile_op_i32_store)
|
||||
{
|
||||
uint32 align = 0;
|
||||
uint32 offset = 0;
|
||||
uint32 bytes = 0;
|
||||
bool atomic = false;
|
||||
|
||||
EXPECT_FALSE(aot_compile_op_i32_store(comp_ctx, func_ctx, align, offset,
|
||||
bytes, atomic));
|
||||
|
||||
/* Generate random number range：[m,n] int a=m+rand()%(n-m+1); */
|
||||
for (uint32 i = 0; i < DEFAULT_CYCLE_TIMES; i++) {
|
||||
bytes = (1 + (rand() % (4 - 1 + 1)));
|
||||
offset = (1 + (rand() % (0xFFFFFFFF - 1 + 1)));
|
||||
align = (1 + (rand() % (0xFFFFFFFF - 1 + 1)));
|
||||
atomic = !atomic;
|
||||
|
||||
EXPECT_FALSE(aot_compile_op_i32_store(comp_ctx, func_ctx, align, offset,
|
||||
bytes, atomic));
|
||||
}
|
||||
}
|
||||
|
||||
TEST_F(compilation_aot_emit_memory_test, aot_compile_op_i64_store)
|
||||
{
|
||||
uint32 align = 0;
|
||||
uint32 offset = 0;
|
||||
uint32 bytes = 0;
|
||||
bool atomic = false;
|
||||
|
||||
EXPECT_FALSE(aot_compile_op_i64_store(comp_ctx, func_ctx, align, offset,
|
||||
bytes, atomic));
|
||||
|
||||
/* Generate random number range：[m,n] int a=m+rand()%(n-m+1); */
|
||||
for (uint32 i = 0; i < DEFAULT_CYCLE_TIMES; i++) {
|
||||
bytes = (1 + (rand() % (8 - 1 + 1)));
|
||||
offset = (1 + (rand() % (0xFFFFFFFF - 1 + 1)));
|
||||
align = (1 + (rand() % (0xFFFFFFFF - 1 + 1)));
|
||||
atomic = !atomic;
|
||||
|
||||
EXPECT_FALSE(aot_compile_op_i64_store(comp_ctx, func_ctx, align, offset,
|
||||
bytes, atomic));
|
||||
}
|
||||
}
|
||||
|
||||
TEST_F(compilation_aot_emit_memory_test, aot_compile_op_f32_store)
|
||||
{
|
||||
uint32 align = 0;
|
||||
uint32 offset = 0;
|
||||
|
||||
EXPECT_FALSE(aot_compile_op_f32_store(comp_ctx, func_ctx, align, offset));
|
||||
|
||||
/* Generate random number range：[m,n] int a=m+rand()%(n-m+1); */
|
||||
for (uint32 i = 0; i < DEFAULT_CYCLE_TIMES; i++) {
|
||||
offset = (1 + (rand() % (0xFFFFFFFF - 1 + 1)));
|
||||
align = (1 + (rand() % (0xFFFFFFFF - 1 + 1)));
|
||||
|
||||
EXPECT_FALSE(
|
||||
aot_compile_op_f32_store(comp_ctx, func_ctx, align, offset));
|
||||
}
|
||||
}
|
||||
|
||||
TEST_F(compilation_aot_emit_memory_test, aot_compile_op_f64_store)
|
||||
{
|
||||
uint32 align = 0;
|
||||
uint32 offset = 0;
|
||||
|
||||
EXPECT_FALSE(aot_compile_op_f64_store(comp_ctx, func_ctx, align, offset));
|
||||
|
||||
/* Generate random number range：[m,n] int a=m+rand()%(n-m+1); */
|
||||
for (uint32 i = 0; i < DEFAULT_CYCLE_TIMES; i++) {
|
||||
offset = (1 + (rand() % (0xFFFFFFFF - 1 + 1)));
|
||||
align = (1 + (rand() % (0xFFFFFFFF - 1 + 1)));
|
||||
|
||||
EXPECT_FALSE(
|
||||
aot_compile_op_f64_store(comp_ctx, func_ctx, align, offset));
|
||||
}
|
||||
}
|
||||
|
||||
TEST_F(compilation_aot_emit_memory_test, aot_compile_op_memory_size)
|
||||
{
|
||||
aot_compile_op_memory_size(comp_ctx, func_ctx);
|
||||
}
|
||||
|
||||
TEST_F(compilation_aot_emit_memory_test, aot_compile_op_memory_grow)
|
||||
{
|
||||
aot_compile_op_memory_grow(comp_ctx, func_ctx);
|
||||
}
|
||||
|
||||
#if WASM_ENABLE_BULK_MEMORY != 0
|
||||
TEST_F(compilation_aot_emit_memory_test, aot_compile_op_memory_init)
|
||||
{
|
||||
uint32 seg_index = 0;
|
||||
|
||||
/* Generate random number range：[m,n] int a=m+rand()%(n-m+1); */
|
||||
for (uint32 i = 0; i < DEFAULT_CYCLE_TIMES; i++) {
|
||||
seg_index = (1 + (rand() % (0xFFFFFFFF - 1 + 1)));
|
||||
aot_compile_op_memory_init(comp_ctx, func_ctx, seg_index);
|
||||
}
|
||||
}
|
||||
|
||||
TEST_F(compilation_aot_emit_memory_test, aot_compile_op_data_drop)
|
||||
{
|
||||
uint32 seg_index = 0;
|
||||
|
||||
/* Generate random number range：[m,n] int a=m+rand()%(n-m+1); */
|
||||
for (uint32 i = 0; i < DEFAULT_CYCLE_TIMES; i++) {
|
||||
seg_index = (1 + (rand() % (0xFFFFFFFF - 1 + 1)));
|
||||
aot_compile_op_data_drop(comp_ctx, func_ctx, seg_index);
|
||||
}
|
||||
}
|
||||
|
||||
TEST_F(compilation_aot_emit_memory_test, aot_compile_op_memory_copy)
|
||||
{
|
||||
aot_compile_op_memory_copy(comp_ctx, func_ctx);
|
||||
}
|
||||
|
||||
TEST_F(compilation_aot_emit_memory_test, aot_compile_op_memory_fill)
|
||||
{
|
||||
aot_compile_op_memory_fill(comp_ctx, func_ctx);
|
||||
}
|
||||
#endif
|
||||
|
||||
#if WASM_ENABLE_SHARED_MEMORY != 0
|
||||
TEST_F(compilation_aot_emit_memory_test, aot_compile_op_atomic_rmw)
|
||||
{
|
||||
uint8 atomic_op = LLVMAtomicRMWBinOpAdd;
|
||||
uint8 op_type = VALUE_TYPE_I32;
|
||||
uint32 align = 4;
|
||||
uint32 offset = 64;
|
||||
uint32 bytes = 4;
|
||||
|
||||
aot_compile_op_atomic_rmw(comp_ctx, func_ctx, atomic_op, op_type, align,
|
||||
offset, bytes);
|
||||
}
|
||||
|
||||
TEST_F(compilation_aot_emit_memory_test, aot_compile_op_atomic_cmpxchg)
|
||||
{
|
||||
|
||||
uint8 op_type = VALUE_TYPE_I32;
|
||||
uint32 align = 4;
|
||||
uint32 offset = 64;
|
||||
uint32 bytes = 4;
|
||||
|
||||
aot_compile_op_atomic_cmpxchg(comp_ctx, func_ctx, op_type, align, offset,
|
||||
bytes);
|
||||
}
|
||||
|
||||
TEST_F(compilation_aot_emit_memory_test, aot_compile_op_atomic_wait)
|
||||
{
|
||||
|
||||
uint8 op_type = VALUE_TYPE_I32;
|
||||
uint32 align = 4;
|
||||
uint32 offset = 64;
|
||||
uint32 bytes = 4;
|
||||
|
||||
aot_compile_op_atomic_wait(comp_ctx, func_ctx, op_type, align, offset,
|
||||
bytes);
|
||||
}
|
||||
|
||||
TEST_F(compilation_aot_emit_memory_test, aot_compiler_op_atomic_notify)
|
||||
{
|
||||
|
||||
uint32 align = 4;
|
||||
uint32 offset = 64;
|
||||
uint32 bytes = 4;
|
||||
|
||||
aot_compiler_op_atomic_notify(comp_ctx, func_ctx, align, offset, bytes);
|
||||
}
|
||||
#endif
|
||||
Reference in New Issue
Block a user