summaryrefslogtreecommitdiff
path: root/Tests/LibC/TestStackSmash.cpp
blob: bcea97d5017276e2a29da444fbce031a5c87e379 (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
/*
 * Copyright (c) 2021, Brian Gianforcaro <bgianf@serenityos.org>
 *
 * SPDX-License-Identifier: BSD-2-Clause
 */

#include <AK/Format.h>
#include <LibTest/TestCase.h>

// Note: Needs to be 'noline' so stack canary isn't optimized out.
static void __attribute__((noinline)) smasher(char* string)
{
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Warray-bounds"
    for (int i = 0; i < 256; i++) {
        string[i] = 'A';
    }
#pragma GCC diagnostic pop
}

// Note: Needs to be 'noline' so stack canary isn't optimized out.
static void __attribute__((noinline)) stack_to_smash()
{
    char string[8] = {};
    smasher(string);
}

TEST_CASE(stack_smash)
{
    EXPECT_CRASH("Smash the stack and trigger __stack_chk_fail", [] {
        outln("[+] Starting the stack smash...");
        stack_to_smash();
        outln("[+] Stack smash wasn't detected!");
        return Test::Crash::Failure::DidNotCrash;
    });
}