[XRay] Fixing one test case for FreeBSD

Reviewers: dberris

Reviewed By: dberris

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

git-svn-id: https://llvm.org/svn/llvm-project/compiler-rt/trunk@370209 91177308-0d34-0410-b5e6-96231b3b80d8

1 files changed, 10 insertions, 0 deletions

diff --git a/test/xray/TestCases/Posix/fork_basic_logging.cpp b/test/xray/TestCases/Posix/fork_basic_logging.cpp
index 3873325ef..fd8b781b8 100644
--- a/test/xray/TestCases/Posix/fork_basic_logging.cpp
+++ b/test/xray/TestCases/Posix/fork_basic_logging.cpp
@@ -18,14 +18,24 @@
 #include <stdio.h>
 #include <unistd.h>
 #include <stdint.h>
+#if defined(__linux__)
 #include <sys/syscall.h>
+#elif defined(__FreeBSD__)
+#include <sys/thr.h>
+#endif
 
 //modified from sanitizer
 
 static uintptr_t syscall_gettid() {
   uint64_t retval;
+#ifdef __linux__
   asm volatile("syscall" : "=a"(retval) : "a"(__NR_gettid) : "rcx", "r11",
                "memory", "cc");
+#else
+  long t;
+  thr_self(&t);
+  retval = static_cast<uint64_t>(t);
+#endif
   return retval;
 }