HopefullyUnkillableStack/stack-demo.cpp at master · Lord-KA/HopefullyUnkillableStack · GitHub

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
108
109
110
111
#define STACK_TYPE int
#define ELEM_PRINTF_FORM "%d"

#include "gstack.h"
#include "gtest/gtest.h"

struct myFixture : public ::testing::Test {
    public:
        char LeftButcher[8];
        GENERIC(stack) S;
        char RightButcher[8];
        char Skip[800];
        GENERIC(stack) *Stack = &S;

        void SetUp() override {
            GENERIC(stack_ctor)(Stack);
            GENERIC(stack_push)(Stack, 10);
            GENERIC(stack_push)(Stack, 11);
            GENERIC(stack_push)(Stack, 12);
            GENERIC(stack_push)(Stack, 13);
            GENERIC(stack_push)(Stack, 14);
            GENERIC(stack_push)(Stack, 15);
            GENERIC(stack_push)(Stack, 16);
            GENERIC(stack_push)(Stack, 17);
            GENERIC(stack_push)(Stack, 24);
        }
        void TearDown() override {
            GENERIC(stack_dtor)(Stack);
        }
};

TEST(General, Modes)
  {
      printf("Debug modes:\n {\n");
      #ifdef STACK_USE_POISON
          printf("\tSTACK_USE_POISON\n");
      #endif
      #ifdef STACK_USE_WRAPPER
          printf("\tSTACK_USE_WRAPPER\n");
      #endif
      #ifdef STACK_USE_PTR_POISON
          printf("\tSTACK_USE_PTR_POISON\n");
      #endif
      #ifdef STACK_USE_CANARY
          printf("\tSTACK_USE_CANARY\n");
      #endif
      #ifdef STACK_USE_STRUCT_HASH
          printf("\tSTACK_USE_STRUCT_HASH\n");
      #endif
      #ifdef STACK_USE_DATA_HASH
          printf("\tSTACK_USE_DATA_HASH\n");
      #endif
      #ifdef FULL_DEBUG
          printf("\tFULL_DEBUG\n");
      #endif
      #ifdef __SANITIZE_ADDRESS__
          printf("\t__SANITIZE_ADDRESS__\n");
      #endif
      #ifdef NDEBUG
          printf("\tNDEBUG\n");
      #endif
      #ifdef STACK_VERBOSE
          printf("\tSTACK_VERBOSE = %d\n", STACK_VERBOSE);
      #endif
      printf(" }\n");
      #if defined(__SANITIZE_ADDRESS__)
          printf("\tERROR: do not compile demo with sanitizers!\n");
          exit(1);
      #endif

  }


TEST_F(myFixture, StructureCanary)
{
    strcpy(LeftButcher + 7, "This is a string11111");
    puts("Altered stack structure with left butcher:\n");
    GENERIC(stack_healthCheck)(Stack);
    strcpy(RightButcher - 36, "Another string!");
    puts("Altered stacks structure with right butcher:\n");
    GENERIC(stack_healthCheck)(Stack);
}


TEST_F(myFixture, StructureHashCapacity)
{
    Stack->capacity = 100;
    puts("Altered capacity:\n");
    GENERIC(stack_healthCheck)(Stack);
}

TEST_F(myFixture, StructureHashLen)
{
    Stack->len = 1000;
    puts("Altered len:\n");
    GENERIC(stack_healthCheck)(Stack);
}


TEST_F(myFixture, DataCanary)
{
    GENERIC(stack) *this_ = Stack;
    char* lButcher = (char*)(LEFT_CANARY_WRAPPER);
    char* rButcher = (char*)(RIGHT_CANARY_WRAPPER) - 16;
    puts("\nAltered data from left:\n");
    strcpy(lButcher, "Somewhat long string of a sorts...");
    GENERIC(stack_healthCheck)(Stack);
    puts("\nAltered data from right:\n");
    strcpy(rButcher, "Hamster running down the cliff...");
    GENERIC(stack_healthCheck)(Stack);
}