[compiler-rt] r348027 - Add a new interceptor for getvfsstat(2) from NetBSD

[Kamil Rytarowski via llvm-commits]

Author: kamil
Date: Fri Nov 30 11:43:53 2018
New Revision: 348027

URL: http://llvm.org/viewvc/llvm-project?rev=348027&view=rev
Log:
Add a new interceptor for getvfsstat(2) from NetBSD

Summary:
getvfsstat - gets list of all mounted file systems.

Add a dedicated test.

Reviewers: vitalybuka, joerg

Reviewed By: vitalybuka

Subscribers: kubamracek, llvm-commits, mgorny, #sanitizers

Tags: #sanitizers

Differential Revision: https://reviews.llvm.org/D55014

Added:
    compiler-rt/trunk/test/sanitizer_common/TestCases/NetBSD/getvfsstat.cc
Modified:
    compiler-rt/trunk/lib/sanitizer_common/sanitizer_common_interceptors.inc
    compiler-rt/trunk/lib/sanitizer_common/sanitizer_platform_interceptors.h

Modified: compiler-rt/trunk/lib/sanitizer_common/sanitizer_common_interceptors.inc
URL: http://llvm.org/viewvc/llvm-project/compiler-rt/trunk/lib/sanitizer_common/sanitizer_common_interceptors.inc?rev=348027&r1=348026&r2=348027&view=diff
==============================================================================
--- compiler-rt/trunk/lib/sanitizer_common/sanitizer_common_interceptors.inc (original)
+++ compiler-rt/trunk/lib/sanitizer_common/sanitizer_common_interceptors.inc Fri Nov 30 11:43:53 2018
@@ -7348,6 +7348,20 @@ INTERCEPTOR(void, setlinebuf, __sanitize
 #define INIT_SETVBUF
 #endif
 
+#if SANITIZER_INTERCEPT_GETVFSSTAT
+INTERCEPTOR(int, getvfsstat, void *buf, SIZE_T bufsize, int flags) {
+  void *ctx;
+  COMMON_INTERCEPTOR_ENTER(ctx, getvfsstat, buf, bufsize, flags);
+  int ret = REAL(getvfsstat)(buf, bufsize, flags);
+  if (buf && ret > 0)
+    COMMON_INTERCEPTOR_WRITE_RANGE(ctx, buf, ret * struct_statvfs_sz);
+  return ret;
+}
+#define INIT_GETVFSSTAT COMMON_INTERCEPT_FUNCTION(getvfsstat)
+#else
+#define INIT_GETVFSSTAT
+#endif
+
 static void InitializeCommonInterceptors() {
   static u64 metadata_mem[sizeof(MetadataHashMap) / sizeof(u64) + 1];
   interceptor_metadata_map = new((void *)&metadata_mem) MetadataHashMap();
@@ -7604,6 +7618,7 @@ static void InitializeCommonInterceptors
   INIT_GETMNTINFO;
   INIT_MI_VECTOR_HASH;
   INIT_SETVBUF;
+  INIT_GETVFSSTAT;
 
   INIT___PRINTF_CHK;
 }

Modified: compiler-rt/trunk/lib/sanitizer_common/sanitizer_platform_interceptors.h
URL: http://llvm.org/viewvc/llvm-project/compiler-rt/trunk/lib/sanitizer_common/sanitizer_platform_interceptors.h?rev=348027&r1=348026&r2=348027&view=diff
==============================================================================
--- compiler-rt/trunk/lib/sanitizer_common/sanitizer_platform_interceptors.h (original)
+++ compiler-rt/trunk/lib/sanitizer_common/sanitizer_platform_interceptors.h Fri Nov 30 11:43:53 2018
@@ -520,5 +520,6 @@
   SI_LINUX || SI_MAC)
 #define SANITIZER_INTERCEPT_GETMNTINFO SI_NETBSD
 #define SANITIZER_INTERCEPT_MI_VECTOR_HASH SI_NETBSD
+#define SANITIZER_INTERCEPT_GETVFSSTAT SI_NETBSD
 
 #endif  // #ifndef SANITIZER_PLATFORM_INTERCEPTORS_H

Added: compiler-rt/trunk/test/sanitizer_common/TestCases/NetBSD/getvfsstat.cc
URL: http://llvm.org/viewvc/llvm-project/compiler-rt/trunk/test/sanitizer_common/TestCases/NetBSD/getvfsstat.cc?rev=348027&view=auto
==============================================================================
--- compiler-rt/trunk/test/sanitizer_common/TestCases/NetBSD/getvfsstat.cc (added)
+++ compiler-rt/trunk/test/sanitizer_common/TestCases/NetBSD/getvfsstat.cc Fri Nov 30 11:43:53 2018
@@ -0,0 +1,36 @@
+// RUN: %clangxx -O0 -g %s -o %t && %run %t 2>&1 | FileCheck %s
+
+#include <sys/types.h>
+
+#include <sys/statvfs.h>
+
+#include <assert.h>
+#include <stdio.h>
+#include <stdlib.h>
+
+int main(void) {
+  printf("getvfsstat\n");
+
+  int rv = getvfsstat(NULL, 0, ST_WAIT);
+  assert(rv != -1);
+
+  size_t sz = rv * sizeof(struct statvfs);
+  struct statvfs *buf = (struct statvfs *)malloc(sz);
+  assert(buf);
+
+  rv = getvfsstat(buf, sz, ST_WAIT);
+  assert(rv != -1);
+
+  for (int i = 0; i < rv; i++) {
+    printf("Filesystem %d\n", i);
+    printf("\tfstypename=%s\n", buf[i].f_fstypename);
+    printf("\tmntonname=%s\n", buf[i].f_mntonname);
+    printf("\tmntfromname=%s\n", buf[i].f_mntfromname);
+  }
+
+  free(buf);
+
+  // CHECK: getvfsstat
+
+  return 0;
+}