compiler-rt/test/dfsan/pair.cpp

   1 // RUN: %clangxx_dfsan %s -mllvm -dfsan-track-select-control-flow=false -mllvm -dfsan-combine-pointer-labels-on-load=false -O0 -DO0 -o %t && %run %t
   2 // RUN: %clangxx_dfsan %s -mllvm -dfsan-track-select-control-flow=false -mllvm -dfsan-combine-pointer-labels-on-load=false -O1 -o %t && %run %t
   3 //
   4 // REQUIRES: x86_64-target-arch
   5
   6 #include <algorithm>
   7 #include <assert.h>
   8 #include <sanitizer/dfsan_interface.h>
   9 #include <utility>
  10
  11 __attribute__((noinline))
  12 std::pair<int *, int>
  13 make_pair(int *p, int i) { return {p, i}; }
  14
  15 __attribute__((noinline))
  16 std::pair<int *, int>
  17 copy_pair1(const std::pair<int *, int> &pair) {
  18   return pair;
  19 }
  20
  21 __attribute__((noinline))
  22 std::pair<int *, int>
  23 copy_pair2(std::pair<int *, int> *pair) {
  24   return *pair;
  25 }
  26
  27 __attribute__((noinline))
  28 std::pair<int *, int>
  29 copy_pair3(std::pair<int *, int> &&pair) {
  30   return std::move(pair);
  31 }
  32
  33 __attribute__((noinline))
  34 std::pair<const char *, uint32_t>
  35 return_ptr_and_i32(const char *p, uint32_t res) {
  36   for (uint32_t i = 2; i < 5; i++) {
  37     uint32_t byte = static_cast<uint8_t>(p[i]);
  38     res += (byte - 1) << (7 * i);
  39     if (byte < 128) {
  40       return {p + i + 1, res};
  41     }
  42   }
  43   return {nullptr, 0};
  44 }
  45
  46 __attribute__((noinline))
  47 std::pair<const char *, uint64_t>
  48 return_ptr_and_i64(const char *p, uint32_t res32) {
  49   uint64_t res = res32;
  50   for (uint32_t i = 2; i < 10; i++) {
  51     uint64_t byte = static_cast<uint8_t>(p[i]);
  52     res += (byte - 1) << (7 * i);
  53     if (byte < 128) {
  54       return {p + i + 1, res};
  55     }
  56   }
  57   return {nullptr, 0};
  58 }
  59
  60 void test_simple_constructors() {
  61   int i = 1;
  62   int *ptr = NULL;
  63   dfsan_set_label(8, &i, sizeof(i));
  64   dfsan_set_label(2, &ptr, sizeof(ptr));
  65
  66   std::pair<int *, int> pair1 = make_pair(ptr, i);
  67   int i1 = pair1.second;
  68   int *ptr1 = pair1.first;
  69
  70 #ifdef O0
  71   assert(dfsan_read_label(&i1, sizeof(i1)) == 10);
  72   assert(dfsan_read_label(&ptr1, sizeof(ptr1)) == 10);
  73 #else
  74   assert(dfsan_read_label(&i1, sizeof(i1)) == 8);
  75   assert(dfsan_read_label(&ptr1, sizeof(ptr1)) == 2);
  76 #endif
  77
  78   std::pair<int *, int> pair2 = copy_pair1(pair1);
  79   int i2 = pair2.second;
  80   int *ptr2 = pair2.first;
  81
  82 #ifdef O0
  83   assert(dfsan_read_label(&i2, sizeof(i2)) == 10);
  84   assert(dfsan_read_label(&ptr2, sizeof(ptr2)) == 10);
  85 #else
  86   assert(dfsan_read_label(&i2, sizeof(i2)) == 8);
  87   assert(dfsan_read_label(&ptr2, sizeof(ptr2)) == 2);
  88 #endif
  89
  90   std::pair<int *, int> pair3 = copy_pair2(&pair1);
  91   int i3 = pair3.second;
  92   int *ptr3 = pair3.first;
  93
  94 #ifdef O0
  95   assert(dfsan_read_label(&i3, sizeof(i3)) == 10);
  96   assert(dfsan_read_label(&ptr3, sizeof(ptr3)) == 10);
  97 #else
  98   assert(dfsan_read_label(&i3, sizeof(i3)) == 8);
  99   assert(dfsan_read_label(&ptr3, sizeof(ptr3)) == 2);
 100 #endif
 101
 102   std::pair<int *, int> pair4 = copy_pair3(std::move(pair1));
 103   int i4 = pair4.second;
 104   int *ptr4 = pair4.first;
 105
 106 #ifdef O0
 107   assert(dfsan_read_label(&i4, sizeof(i4)) == 10);
 108   assert(dfsan_read_label(&ptr4, sizeof(ptr4)) == 10);
 109 #else
 110   assert(dfsan_read_label(&i4, sizeof(i4)) == 8);
 111   assert(dfsan_read_label(&ptr4, sizeof(ptr4)) == 2);
 112 #endif
 113 }
 114
 115 void test_branches() {
 116   uint32_t res = 4;
 117   dfsan_set_label(8, &res, sizeof(res));
 118
 119   char p[100];
 120   const char *q = p;
 121   dfsan_set_label(2, &q, sizeof(q));
 122
 123   {
 124     std::fill_n(p, 100, static_cast<char>(128));
 125
 126     {
 127       std::pair<const char *, uint32_t> r = return_ptr_and_i32(q, res);
 128       assert(dfsan_read_label(&r.first, sizeof(r.first)) == 0);
 129       assert(dfsan_read_label(&r.second, sizeof(r.second)) == 0);
 130     }
 131
 132     {
 133       std::pair<const char *, uint64_t> r = return_ptr_and_i64(q, res);
 134       assert(dfsan_read_label(&r.first, sizeof(r.first)) == 0);
 135       assert(dfsan_read_label(&r.second, sizeof(r.second)) == 0);
 136     }
 137   }
 138
 139   {
 140     std::fill_n(p, 100, 0);
 141
 142     {
 143       std::pair<const char *, uint32_t> r = return_ptr_and_i32(q, res);
 144 #ifdef O0
 145       assert(dfsan_read_label(&r.first, sizeof(r.first)) == 10);
 146       assert(dfsan_read_label(&r.second, sizeof(r.second)) == 10);
 147 #else
 148       assert(dfsan_read_label(&r.first, sizeof(r.first)) == 2);
 149       assert(dfsan_read_label(&r.second, sizeof(r.second)) == 8);
 150 #endif
 151     }
 152
 153     {
 154       std::pair<const char *, uint64_t> r = return_ptr_and_i64(q, res);
 155 #ifdef O0
 156       assert(dfsan_read_label(&r.first, sizeof(r.first)) == 10);
 157       assert(dfsan_read_label(&r.second, sizeof(r.second)) == 10);
 158 #else
 159       assert(dfsan_read_label(&r.first, sizeof(r.first)) == 2);
 160       assert(dfsan_read_label(&r.second, sizeof(r.second)) == 8);
 161 #endif
 162     }
 163   }
 164 }
 165
 166 int main(void) {
 167   test_simple_constructors();
 168   test_branches();
 169
 170   return 0;
 171 }