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 #include <sanitizer/dfsan_interface.h>
   6 #include <assert.h>
   7
   8 const char remap_to_upper[256] = {
   9     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  10     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  11     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  12     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  13     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  14     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  15     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  16     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  17     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  18     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  19     '.', '.', '.', '.', '.', '.', '.', 'A', 'B',
  20     'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K',
  21     'L', 'M', 'N', 'O', 'P', 'Q', 'R', 'S', 'T',
  22     'U', 'V', 'W', 'X', 'Y', 'Z', '.', '.', '.',
  23     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  24     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  25     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  26     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  27     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  28     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  29     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  30     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  31     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  32     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  33     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  34     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  35     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  36     '.', '.', '.', '.', '.', '.', '.', '.', '.',
  37     '.', '.', '.', '.',
  38 };
  39
  40 char character_mapping(unsigned char c) {
  41   return remap_to_upper[c];
  42 }
  43
  44 int main(void) {
  45   char a = 'b';
  46   dfsan_label i_label = 1;
  47   dfsan_set_label(i_label, &a, sizeof(a));
  48   assert(dfsan_read_label(&a, sizeof(a)) == i_label);
  49
  50   char b = character_mapping(a);
  51   assert(b == 'B');
  52
  53 #ifdef LOOKUP_TABLE
  54   assert(dfsan_read_label(&b, sizeof(b)) == i_label);
  55 #else
  56   assert(dfsan_read_label(&b, sizeof(b)) == 0);
  57 #endif
  58   return 0;
  59 }