compiler-rt/test/dfsan/lookup_table.c

   1 // RUN: %clang_dfsan %s -mllvm -dfsan-combine-offset-labels-on-gep=false -mllvm -dfsan-combine-pointer-labels-on-load=false -mllvm -dfsan-combine-taint-lookup-table=remap_to_upper -DLOOKUP_TABLE -o %t && %run %t
   2 // RUN: %clang_dfsan %s -mllvm -dfsan-combine-offset-labels-on-gep=false -mllvm -dfsan-combine-pointer-labels-on-load=false -mllvm -dfsan-combine-taint-lookup-table=no_match -o %t && %run %t
   3 // RUN: %clang_dfsan %s -mllvm -dfsan-combine-offset-labels-on-gep=false -mllvm -dfsan-combine-pointer-labels-on-load=false -o %t && %run %t
   4 //
   5 // REQUIRES: x86_64-target-arch
   6
   7 #include <sanitizer/dfsan_interface.h>
   8 #include <assert.h>
   9
  10 const char remap_to_upper[256] = {
  11     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  12     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  13     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  14     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  15     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  16     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  17     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  18     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  19     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  20     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  21     '.', '.', '.', '.', '.', '.', '.', 'A', 'B',
  22     'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K',
  23     'L', 'M', 'N', 'O', 'P', 'Q', 'R', 'S', 'T',
  24     'U', 'V', 'W', 'X', 'Y', 'Z', '.', '.', '.',
  25     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  26     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  27     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  28     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  29     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  30     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  31     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  32     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  33     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  34     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  35     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  36     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  37     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  38     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  39     '.', '.', '.', '.',
  40 };
  41
  42 char character_mapping(unsigned char c) {
  43   return remap_to_upper[c];
  44 }
  45
  46 int main(void) {
  47   char a = 'b';
  48   dfsan_label i_label = 1;
  49   dfsan_set_label(i_label, &a, sizeof(a));
  50   assert(dfsan_read_label(&a, sizeof(a)) == i_label);
  51
  52   char b = character_mapping(a);
  53   assert(b == 'B');
  54
  55 #ifdef LOOKUP_TABLE
  56   assert(dfsan_read_label(&b, sizeof(b)) == i_label);
  57 #else
  58   assert(dfsan_read_label(&b, sizeof(b)) == 0);
  59 #endif
  60   return 0;
  61 }