compiler-rt/test/msan/scandir.cpp

   1 // RUN: %clangxx_msan -O0 %s -o %t && %run %t %p
   2 // RUN: %clangxx_msan -O0 -D_FILE_OFFSET_BITS=64 %s -o %t && %run %t %p
   3 // RUN: %clangxx_msan -O3 %s -o %t && %run %t %p
   4
   5 #include <assert.h>
   6 #include <glob.h>
   7 #include <stdio.h>
   8 #include <stdlib.h>
   9 #include <string.h>
  10 #include <errno.h>
  11
  12 #include <sys/stat.h>
  13 #include <sys/types.h>
  14 #include <dirent.h>
  15 #include <unistd.h>
  16
  17 #include <sanitizer/msan_interface.h>
  18
  19
  20 static int my_filter(const struct dirent *a) {
  21   assert(__msan_test_shadow(&a, sizeof(a)) == (size_t)-1);
  22   printf("%s\n", a->d_name);
  23   __msan_print_shadow(a, a->d_reclen);
  24   assert(__msan_test_shadow(a, a->d_reclen) == (size_t)-1);
  25   printf("%s\n", a->d_name);
  26   return strlen(a->d_name) == 3 && a->d_name[2] == 'b';
  27 }
  28
  29 static int my_compar(const struct dirent **a, const struct dirent **b) {
  30   assert(__msan_test_shadow(a, sizeof(*a)) == (size_t)-1);
  31   assert(__msan_test_shadow(*a, (*a)->d_reclen) == (size_t)-1);
  32   assert(__msan_test_shadow(b, sizeof(*b)) == (size_t)-1);
  33   assert(__msan_test_shadow(*b, (*b)->d_reclen) == (size_t)-1);
  34   if ((*a)->d_name[1] == (*b)->d_name[1])
  35     return 0;
  36   return ((*a)->d_name[1] < (*b)->d_name[1]) ? 1 : -1;
  37 }
  38
  39 int main(int argc, char *argv[]) {
  40   assert(argc == 2);
  41   char buf[1024];
  42   snprintf(buf, sizeof(buf), "%s/%s", argv[1], "scandir_test_root/");
  43
  44   struct dirent **d;
  45   int res = scandir(buf, &d, my_filter, my_compar);
  46   assert(res == 2);
  47   assert(__msan_test_shadow(&d, sizeof(*d)) == (size_t)-1);
  48   for (int i = 0; i < res; ++i) {
  49     assert(__msan_test_shadow(&d[i], sizeof(d[i])) == (size_t)-1);
  50     assert(__msan_test_shadow(d[i], d[i]->d_reclen) == (size_t)-1);
  51   }
  52
  53   assert(strcmp(d[0]->d_name, "bbb") == 0);
  54   assert(strcmp(d[1]->d_name, "aab") == 0);
  55   return 0;
  56 }