gcc/testsuite/gcc.dg/fstack-protector-strong.c

   1 /* Test that stack protection is done on chosen functions. */
   2
   3 /* { dg-do compile { target i?86-*-* x86_64-*-* rs6000-*-* s390x-*-* } } */
   4 /* { dg-options "-O2 -fstack-protector-strong" } */
   5
   6 #include<string.h>
   7
   8 extern int g0;
   9 extern int* pg0;
  10 int
  11 goo (int *);
  12 int
  13 hoo (int);
  14
  15 /* Function frame address escaped function call. */
  16 int
  17 foo1 ()
  18 {
  19   int i;
  20   return goo (&i);
  21 }
  22
  23 struct ArrayStruct
  24 {
  25   int a;
  26   int array[10];
  27 };
  28
  29 struct AA
  30 {
  31   int b;
  32   struct ArrayStruct as;
  33 };
  34
  35 /* Function frame contains array. */
  36 int
  37 foo2 ()
  38 {
  39   struct AA aa;
  40   int i;
  41   for (i = 0; i < 10; ++i)
  42     {
  43       aa.as.array[i] = i * (i-1) + i / 2;
  44     }
  45   return aa.as.array[5];
  46 }
  47
  48 /* Address computation based on a function frame address. */
  49 int
  50 foo3 ()
  51 {
  52   int a;
  53   int *p;
  54   p = &a + 5;
  55   return goo (p);
  56 }
  57
  58 /* Address cast based on a function frame address. */
  59 int
  60 foo4 ()
  61 {
  62   int a;
  63   return goo (g0 << 2 ? (int *)(3 * (long)(void *)(&a)) : 0);
  64 }
  65
  66 /* Address cast based on a local array. */
  67 int
  68 foo5 ()
  69 {
  70   short array[10];
  71   return goo ((int *)(array + 5));
  72 }
  73
  74 struct BB
  75 {
  76   int one;
  77   int two;
  78   int three;
  79 };
  80
  81 /* Address computaton based on a function frame address.*/
  82 int
  83 foo6 ()
  84 {
  85   struct BB bb;
  86   return goo (&bb.one + sizeof(int));
  87 }
  88
  89 /* Function frame address escaped via global variable. */
  90 int
  91 foo7 ()
  92 {
  93   int a;
  94   pg0 = &a;
  95   goo (pg0);
  96   return *pg0;
  97 }
  98
  99 /* Check that this covers -fstack-protector. */
 100 int
 101 foo8 ()
 102 {
 103   char base[100];
 104   memcpy ((void *)base, (const void *)pg0, 105);
 105   return (int)(base[32]);
 106 }
 107
 108 /* Check that this covers -fstack-protector. */
 109 int
 110 foo9 ()
 111 {
 112   char* p = __builtin_alloca (100);
 113   return goo ((int *)(p + 50));
 114 }
 115
 116 int
 117 global2 (struct BB* pbb);
 118
 119 /* Address taken on struct. */
 120 int
 121 foo10 ()
 122 {
 123   struct BB bb;
 124   int i;
 125   bb.one = global2 (&bb);
 126   for (i = 0; i < 10; ++i)
 127     {
 128       bb.two = bb.one + bb.two;
 129       bb.three = bb.one + bb.two + bb.three;
 130     }
 131   return bb.three;
 132 }
 133
 134 /* { dg-final { scan-assembler-times "stack_chk_fail" 10 } } */