compiler-rt/test/dfsan/libatomic.c

   1 // RUN: %clang_dfsan -g3 -DDATA_BYTES=3 %s -fno-exceptions %libatomic -o %t && %run %t
   2 // RUN: %clang_dfsan -g3 -DDATA_BYTES=3 -DORIGIN_TRACKING -mllvm -dfsan-track-origins=1 %s -fno-exceptions %libatomic -o %t && %run %t
   3 // RUN: %clang_dfsan -g3 -DDATA_BYTES=32 %s -fno-exceptions %libatomic -o %t && %run %t
   4 // RUN: %clang_dfsan -g3 -DDATA_BYTES=32 -DORIGIN_TRACKING -mllvm -dfsan-track-origins=1 %s -fno-exceptions %libatomic -o %t && %run %t
   5
   6 #include <assert.h>
   7 #include <sanitizer/dfsan_interface.h>
   8 #include <stdatomic.h>
   9
  10 typedef struct __attribute((packed)) {
  11   uint8_t val[DATA_BYTES];
  12 } idata;
  13
  14 void test_idata_load() {
  15   idata dest = {-1};
  16   idata init = {0};
  17
  18   dfsan_label i_label = 2;
  19   dfsan_set_label(i_label, &init, sizeof(init));
  20
  21   __atomic_load(&init, &dest, __ATOMIC_RELAXED);
  22
  23   dfsan_label read_label = dfsan_read_label(&dest, sizeof(dest));
  24   assert(read_label == i_label);
  25 #ifdef ORIGIN_TRACKING
  26   dfsan_origin read_origin =
  27       dfsan_read_origin_of_first_taint(&dest, sizeof(dest));
  28   assert(read_origin != 0);
  29 #endif
  30 }
  31
  32 void test_idata_store() {
  33   idata dest = {-1};
  34   idata init = {0};
  35
  36   dfsan_label i_label = 2;
  37   dfsan_set_label(i_label, &init, sizeof(init));
  38
  39   __atomic_store(&init, &dest, __ATOMIC_RELAXED);
  40
  41   dfsan_label read_label = dfsan_read_label(&dest, sizeof(dest));
  42   assert(read_label == i_label);
  43 #ifdef ORIGIN_TRACKING
  44   dfsan_origin read_origin =
  45       dfsan_read_origin_of_first_taint(&dest, sizeof(dest));
  46   assert(read_origin != 0);
  47 #endif
  48 }
  49
  50 void test_idata_exchange() {
  51   idata target = {-1};
  52   idata init = {0};
  53   idata dest = {3};
  54
  55   dfsan_label i_label = 1;
  56   dfsan_set_label(i_label, &init, sizeof(init));
  57   dfsan_label j_label = 2;
  58   dfsan_set_label(j_label, &target, sizeof(target));
  59
  60   dfsan_label dest0_label = dfsan_read_label(&dest, sizeof(dest));
  61   assert(dest0_label == 0);
  62 #ifdef ORIGIN_TRACKING
  63   dfsan_origin dest0_origin =
  64       dfsan_read_origin_of_first_taint(&dest, sizeof(dest));
  65   assert(dest0_origin == 0);
  66 #endif
  67
  68   __atomic_exchange(&target, &init, &dest, __ATOMIC_RELAXED);
  69
  70   dfsan_label dest_label = dfsan_read_label(&dest, sizeof(dest));
  71   assert(dest_label == j_label);
  72 #ifdef ORIGIN_TRACKING
  73   dfsan_origin dest_origin =
  74       dfsan_read_origin_of_first_taint(&dest, sizeof(dest));
  75   assert(dest_origin != 0);
  76 #endif
  77
  78   dfsan_label target_label = dfsan_read_label(&target, sizeof(target));
  79   assert(target_label == i_label);
  80 #ifdef ORIGIN_TRACKING
  81   dfsan_origin target_origin =
  82       dfsan_read_origin_of_first_taint(&target, sizeof(target));
  83   assert(target_origin != 0);
  84 #endif
  85 }
  86
  87 void test_idata_cmp_exchange_1() {
  88   idata target = {0};
  89   idata expected = {0}; // Target matches expected
  90   idata desired = {3};
  91
  92   dfsan_label i_label = 1;
  93   dfsan_set_label(i_label, &expected, sizeof(expected));
  94   dfsan_label j_label = 2;
  95   dfsan_set_label(j_label, &target, sizeof(target));
  96   dfsan_label k_label = 4;
  97   dfsan_set_label(k_label, &desired, sizeof(desired));
  98
  99   int r =
 100       __atomic_compare_exchange(&target, &expected, &desired, /*weak=false*/ 0,
 101                                 __ATOMIC_RELAXED, __ATOMIC_RELAXED);
 102   // Target matches expected => true
 103   assert(r);
 104
 105   // Copy desired to target.
 106   dfsan_label target_label = dfsan_read_label(&target, sizeof(target));
 107   assert(target_label == k_label);
 108 #ifdef ORIGIN_TRACKING
 109   dfsan_origin target_origin =
 110       dfsan_read_origin_of_first_taint(&target, sizeof(target));
 111   assert(target_origin != 0);
 112 #endif
 113 }
 114
 115 void test_idata_cmp_exchange_2() {
 116   idata target = {0};
 117   idata expected = {-1}; // Target does not match expected
 118   idata desired = {3};
 119
 120   dfsan_label i_label = 1;
 121   dfsan_set_label(i_label, &expected, sizeof(expected));
 122   dfsan_label j_label = 2;
 123   dfsan_set_label(j_label, &target, sizeof(target));
 124   dfsan_label k_label = 4;
 125   dfsan_set_label(k_label, &desired, sizeof(desired));
 126
 127   int r =
 128       __atomic_compare_exchange(&target, &expected, &desired, /*weak=false*/ 0,
 129                                 __ATOMIC_RELAXED, __ATOMIC_RELAXED);
 130   // Target does not match expected => false
 131   assert(!r);
 132
 133   // Copy target to expected
 134   dfsan_label expected_label = dfsan_read_label(&expected, sizeof(expected));
 135   assert(expected_label == j_label);
 136 #ifdef ORIGIN_TRACKING
 137   dfsan_origin expected_origin =
 138       dfsan_read_origin_of_first_taint(&expected, sizeof(expected));
 139   assert(expected_origin != 0);
 140 #endif
 141 }
 142
 143 int main() {
 144   test_idata_load();
 145   test_idata_store();
 146   test_idata_exchange();
 147   test_idata_cmp_exchange_1();
 148   test_idata_cmp_exchange_2();
 149
 150   return 0;
 151 }