compiler-rt/test/msan/Linux/glob_altdirfunc.cpp

   1 // RUN: %clangxx_msan -O0 %s -o %t && %run %t %p 2>&1 | FileCheck %s
   2 // RUN: %clangxx_msan -O0 -D_FILE_OFFSET_BITS=64 %s -o %t && %run %t %p 2>&1 | FileCheck %s
   3 // RUN: %clangxx_msan -O3 %s -o %t && %run %t %p 2>&1 | FileCheck %s
   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 static void my_gl_closedir(void *dir) {
  20   if (!dir)
  21     exit(1);
  22   closedir((DIR *)dir);
  23 }
  24
  25 static struct dirent *my_gl_readdir(void *dir) {
  26   if (!dir)
  27     exit(1);
  28   struct dirent *d = readdir((DIR *)dir);
  29   if (d) __msan_poison(d, d->d_reclen); // hehe
  30   return d;
  31 }
  32
  33 static void *my_gl_opendir(const char *s) {
  34   assert(__msan_test_shadow(s, strlen(s) + 1) == (size_t)-1);
  35   return opendir(s);
  36 }
  37
  38 static int my_gl_lstat(const char *s, struct stat *st) {
  39   assert(__msan_test_shadow(s, strlen(s) + 1) == (size_t)-1);
  40   if (!st)
  41     exit(1);
  42   return lstat(s, st);
  43 }
  44
  45 static int my_gl_stat(const char *s, struct stat *st) {
  46   assert(__msan_test_shadow(s, strlen(s) + 1) == (size_t)-1);
  47   if (!st)
  48     exit(1);
  49   return lstat(s, st);
  50 }
  51
  52 int main(int argc, char *argv[]) {
  53   assert(argc == 2);
  54   char buf[1024];
  55   snprintf(buf, sizeof(buf), "%s/%s", argv[1], "glob_test_root/*a");
  56
  57   glob_t globbuf;
  58   globbuf.gl_closedir = my_gl_closedir;
  59   globbuf.gl_readdir = my_gl_readdir;
  60   globbuf.gl_opendir = my_gl_opendir;
  61   globbuf.gl_lstat = my_gl_lstat;
  62   globbuf.gl_stat = my_gl_stat;
  63   for (int i = 0; i < 10000; ++i) {
  64     int res = glob(buf, GLOB_ALTDIRFUNC | GLOB_MARK, 0, &globbuf);
  65     assert(res == 0);
  66     printf("%d %s\n", errno, strerror(errno));
  67     assert(globbuf.gl_pathc == 2);
  68     printf("%zu\n", strlen(globbuf.gl_pathv[0]));
  69     printf("%zu\n", strlen(globbuf.gl_pathv[1]));
  70     __msan_poison(globbuf.gl_pathv[0], strlen(globbuf.gl_pathv[0]) + 1);
  71     __msan_poison(globbuf.gl_pathv[1], strlen(globbuf.gl_pathv[1]) + 1);
  72     globfree(&globbuf);
  73   }
  74
  75   printf("PASS\n");
  76   // CHECK: PASS
  77   return 0;
  78 }