pound/ballistic
0
0
Fork 0
mirror of https://github.com/pound-emu/ballistic.git synced 2026-06-17 04:16:48 -07:00
Code Issues Packages Projects Releases Wiki Activity

tests/x86/assembler: check for buffer overflows

Browse Source 
Signed-off-by: Ronald Caesar <github43132@proton.me>
This commit is contained in:
Ronald Caesar
2026-06-15 03:18:55 -04:00
parent cb7314dbcc
commit 247005edd5
1 changed files with 85 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/backend/x86/test_x86_assembler.cpp
+85
View File
@@ -311,6 +311,91 @@ TEST_F(Backendx86Assembler, Public_InvalidRegister_NoEmit)
TEST_F(Backendx86Assembler, Public_FullBuffer_NoEmit)
{
assembler.offset = sizeof(buffer);
bal_x86_emit_add_mem64_rbp_offset_imm(&assembler, 0, 0x1);
EXPECT_EQ(assembler.status, BAL_ERROR_INSTRUCTION_OVERFLOW);
EXPECT_EQ(assembler.offset, sizeof(buffer));
assembler.status = BAL_SUCCESS;
bal_x86_emit_add_r64_imm32(&assembler, BAL_X86_RAX, 0x1);
EXPECT_EQ(assembler.status, BAL_ERROR_INSTRUCTION_OVERFLOW);
EXPECT_EQ(assembler.offset, sizeof(buffer));
assembler.status = BAL_SUCCESS;
bal_x86_emit_add_r64_r64(&assembler, BAL_X86_RAX, BAL_X86_RAX);
EXPECT_EQ(assembler.status, BAL_ERROR_INSTRUCTION_OVERFLOW);
EXPECT_EQ(assembler.offset, sizeof(buffer));
assembler.status = BAL_SUCCESS;
bal_x86_emit_and_r64_r64(&assembler, BAL_X86_RAX, BAL_X86_RAX);
EXPECT_EQ(assembler.status, BAL_ERROR_INSTRUCTION_OVERFLOW);
EXPECT_EQ(assembler.offset, sizeof(buffer));
assembler.status = BAL_SUCCESS;
bal_x86_emit_jcc_rel32(&assembler, BAL_X86_COND_A, 0);
EXPECT_EQ(assembler.status, BAL_ERROR_INSTRUCTION_OVERFLOW);
EXPECT_EQ(assembler.offset, sizeof(buffer));
assembler.status = BAL_SUCCESS;
bal_x86_emit_jmp_r64(&assembler, BAL_X86_RAX);
EXPECT_EQ(assembler.status, BAL_ERROR_INSTRUCTION_OVERFLOW);
EXPECT_EQ(assembler.offset, sizeof(buffer));
assembler.status = BAL_SUCCESS;
bal_x86_emit_jmp_rel32(&assembler, 0);
EXPECT_EQ(assembler.status, BAL_ERROR_INSTRUCTION_OVERFLOW);
EXPECT_EQ(assembler.offset, sizeof(buffer));
assembler.status = BAL_SUCCESS;
bal_x86_emit_load_r64_rbp_offset(&assembler, BAL_X86_RAX, 0);
EXPECT_EQ(assembler.status, BAL_ERROR_INSTRUCTION_OVERFLOW);
EXPECT_EQ(assembler.offset, sizeof(buffer));
assembler.status = BAL_SUCCESS;
bal_x86_emit_mov_r64_r64(&assembler, BAL_X86_RAX, BAL_X86_RBX);
EXPECT_EQ(assembler.status, BAL_ERROR_INSTRUCTION_OVERFLOW);
EXPECT_EQ(assembler.offset, sizeof(buffer));
assembler.status = BAL_SUCCESS;
bal_x86_emit_mov_r64_imm64(&assembler, BAL_X86_RAX, 0);
EXPECT_EQ(assembler.status, BAL_ERROR_INSTRUCTION_OVERFLOW);
EXPECT_EQ(assembler.offset, sizeof(buffer));
assembler.status = BAL_SUCCESS;
bal_x86_emit_or_r64_r64(&assembler, BAL_X86_RAX, BAL_X86_RBP);
EXPECT_EQ(assembler.status, BAL_ERROR_INSTRUCTION_OVERFLOW);
EXPECT_EQ(assembler.offset, sizeof(buffer));
assembler.status = BAL_SUCCESS;
bal_x86_emit_pop_r64(&assembler, BAL_X86_RAX);
EXPECT_EQ(assembler.status, BAL_ERROR_INSTRUCTION_OVERFLOW);
EXPECT_EQ(assembler.offset, sizeof(buffer));
assembler.status = BAL_SUCCESS;
bal_x86_emit_push_r64(&assembler, BAL_X86_RAX);
EXPECT_EQ(assembler.status, BAL_ERROR_INSTRUCTION_OVERFLOW);
EXPECT_EQ(assembler.offset, sizeof(buffer));
assembler.status = BAL_SUCCESS;
bal_x86_emit_setcc_mem8_rbp_offset(&assembler, BAL_X86_COND_A, 0);
EXPECT_EQ(assembler.status, BAL_ERROR_INSTRUCTION_OVERFLOW);
EXPECT_EQ(assembler.offset, sizeof(buffer));
assembler.status = BAL_SUCCESS;
bal_x86_emit_store_r64_rbp_offset(&assembler, BAL_X86_RAX, 0);
EXPECT_EQ(assembler.status, BAL_ERROR_INSTRUCTION_OVERFLOW);
EXPECT_EQ(assembler.offset, sizeof(buffer));
assembler.status = BAL_SUCCESS;
bal_x86_emit_sub_r64_r64(&assembler, BAL_X86_RAX, BAL_X86_RBX);
EXPECT_EQ(assembler.status, BAL_ERROR_INSTRUCTION_OVERFLOW);
EXPECT_EQ(assembler.offset, sizeof(buffer));
assembler.status = BAL_SUCCESS;
bal_x86_emit_ret(&assembler);
EXPECT_EQ(assembler.status, BAL_ERROR_INSTRUCTION_OVERFLOW);
EXPECT_EQ(assembler.offset, sizeof(buffer));
}
TEST_F(Backendx86Assembler, Encode_Ret)