added posix calls for testing

hariharan-devarajan · Oct 10, 2023 · 11949bf · 11949bf
1 parent e4e8f4c
commit 11949bf
Show file tree

Hide file tree

Showing 4 changed files with 67 additions and 8 deletions.
diff --git a/src/dlio_profiler/core/dlio_profiler_main.cpp b/src/dlio_profiler/core/dlio_profiler_main.cpp
@@ -154,10 +154,7 @@ dlio_profiler::DLIOProfilerCore::initlialize(bool is_init, bool _bind, const cha
       } else {
         this->data_dirs = _data_dirs;
       }
-
       DLIO_PROFILER_LOGINFO("Setting data_dirs to %s", this->data_dirs.c_str());
-
-
       dlio_profiler::Singleton<DLIOLogger>::get_instance()->update_log_file(this->log_file, this->process_id);
       if (bind) {
         brahma_gotcha_wrap("dlio_profiler", this->gotcha_priority);

diff --git a/test/CMakeLists.txt b/test/CMakeLists.txt
@@ -31,7 +31,7 @@ set_property(TEST ${test_name} APPEND PROPERTY ENVIRONMENT DLIO_PROFILER_INIT=PR
 set_property(TEST ${test_name} APPEND PROPERTY ENVIRONMENT DLIO_PROFILER_DATA_DIR=${CMAKE_CURRENT_BINARY_DIR}/data)
 set_property(TEST ${test_name} APPEND PROPERTY ENVIRONMENT DLIO_PROFILER_LOG_FILE=${CMAKE_CURRENT_BINARY_DIR}/${test_name})
 
-dlp_add_test(check_file_exists_${test_name} ${BASH_PROGRAM} ${CMAKE_CURRENT_SOURCE_DIR}/check_file.sh ${CMAKE_CURRENT_BINARY_DIR}/${test_name}* 7)
+dlp_add_test(check_file_exists_${test_name} ${BASH_PROGRAM} ${CMAKE_CURRENT_SOURCE_DIR}/check_file.sh ${CMAKE_CURRENT_BINARY_DIR}/${test_name}* 39)
 set_tests_properties(check_file_exists_${test_name} PROPERTIES DEPENDS ${test_name})
 
 set(test_name test_cpp_basic_app_only)
@@ -42,7 +42,7 @@ set_property(TEST ${test_name} APPEND PROPERTY ENVIRONMENT DLIO_PROFILER_ENABLE=
 set_property(TEST ${test_name} APPEND PROPERTY ENVIRONMENT DLIO_PROFILER_DATA_DIR=${CMAKE_CURRENT_BINARY_DIR}/data)
 set_property(TEST ${test_name} APPEND PROPERTY ENVIRONMENT DLIO_PROFILER_LOG_FILE=${CMAKE_CURRENT_BINARY_DIR}/${test_name})
 
-dlp_add_test(check_file_exists_${test_name} ${BASH_PROGRAM} ${CMAKE_CURRENT_SOURCE_DIR}/check_file.sh ${CMAKE_CURRENT_BINARY_DIR}/${test_name}* 7)
+dlp_add_test(check_file_exists_${test_name} ${BASH_PROGRAM} ${CMAKE_CURRENT_SOURCE_DIR}/check_file.sh ${CMAKE_CURRENT_BINARY_DIR}/${test_name}* 38)
 set_tests_properties(check_file_exists_${test_name} PROPERTIES DEPENDS ${test_name})
 
 set(test_name test_cpp_basic_app_disable_only)

diff --git a/test/c/test.c b/test/c/test.c
@@ -27,7 +27,7 @@ int main(int argc, char* argv[]) {
     }
   }
   char filename[1024];
-  sprintf(filename, "%s/demofile.txt", argv[1]);
+  sprintf(filename, "%s/demofile_c.txt", argv[1]);
   foo();
   FILE* fh = fopen(filename, "w+");
   fwrite("hello", sizeof("hello"), 1, fh);

diff --git a/test/cpp/test.cpp b/test/cpp/test.cpp
@@ -27,10 +27,72 @@ int main(int argc, char* argv[]) {
   }
   char filename[1024];
   sprintf(filename, "%s/demofile.txt", argv[1]);
+  char filename_link[1024];
+  sprintf(filename_link, "%s/demofile_link.txt", argv[1]);
   foo();
+  truncate(filename,0);
   FILE* fh = fopen(filename, "w+");
-  fwrite("hello", sizeof("hello"), 1, fh);
-  fclose(fh);
+  if (fh != NULL) {
+    fwrite("hello", sizeof("hello"), 1, fh);
+    fclose(fh);
+  }
+  struct stat stat_buf;
+  struct stat64 stat_buf64;
+  __xstat(0, filename, &stat_buf);
+  __lxstat(0, filename, &stat_buf);
+  __xstat64(0, filename, &stat_buf64);
+  __lxstat64(0, filename, &stat_buf64);
+
+  link(filename, filename_link);
+  unlink(filename_link);
+  symlink(filename, filename_link);
+  chmod(filename, S_ISUID);
+  chown(filename, 0, 0);
+  lchown(filename, 0, 0);
+  struct utimbuf utimbuf1;
+  utime(filename, &utimbuf1);
+  char dir[1024];
+  sprintf(dir, "%s", argv[1]);
+  int dd = open(dir, O_PATH);
+  assert(dd != -1);
+  fcntl(dd, F_DUPFD);
+  fcntl(dd, F_GETFD);
+  fcntl(dd, F_GETOWN_EX);
+  int dd2;
+  dup2(dd, dd2);
+  umask(0);
+  mkfifo(filename, 0);
+  symlinkat(filename, dd, filename_link);
+  __fxstat(0, dd, &stat_buf);
+  __fxstat64(0, dd, &stat_buf64);
+  faccessat(dd,"demofile.txt",O_RDONLY,0);
+  linkat(dd,"demofile.txt", dd, "demofile_link2.txt", 0);
+  chdir(dir);
+  int fd = openat(dd, "demofile.txt", O_RDONLY);
+  if (fd != -1) close(fd);
+  fd = openat(dd, "demofile2.txt", O_WRONLY | O_CREAT);
+  if (fd != -1) close(fd);
+  close(dd);
+  char filename2[1024];
+  sprintf(filename, "%s/demofile2.txt", argv[1]);
+  fd = creat64(filename, O_RDWR) ;
+  if (fd != -1) close(fd);
+  fd = open(filename, O_RDWR);
+  int set_offset = lseek(fd, 1, SEEK_SET);
+  char buf[1];
+  pread(fd, buf, 1, 1);
+  pread64(fd, buf, 1, 1);
+  pwrite(fd, buf, 1, 1);
+  pwrite64(fd, buf, 1, 1);
+  fsync(fd);
+  fdatasync(fd);
+  readlinkat(fd, filename, buf, 1);
+  ftruncate(fd,0);
+  close(fd);
+  remove(filename2);
+  remove(filename);
+  remove(filename_link);
+  remove("demofile_link2.txt");
   if (init ==  1) {
     DLIO_PROFILER_CPP_FINI();
   }