folly: fix sysMembarrier() with newer kernel headers
[folly.git] / folly / portability / SysMembarrier.cpp
index 16fa5fb..9b3735c 100644 (file)
@@ -1,5 +1,5 @@
 /*
- * Copyright 2016 Facebook, Inc.
+ * Copyright 2017 Facebook, Inc.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
 #include <mutex>
 
 #include <folly/Portability.h>
+#include <folly/portability/SysSyscall.h>
 #include <folly/portability/Unistd.h>
 
-#if !defined(__NR_membarrier) && FOLLY_X64 && !FOLLY_MOBILE
+#if FOLLY_X64 && !FOLLY_MOBILE && defined(__linux__)
+#define FOLLY_USE_SYS_MEMBARRIER 1
+#if !defined(__NR_membarrier)
 #define __NR_membarrier 324
+#endif
+#if FOLLY_HAVE_LINUX_MEMBARRIER_H
+#include <linux/membarrier.h> // @manual
+#else
 #define MEMBARRIER_CMD_QUERY 0
 #define MEMBARRIER_CMD_SHARED 1
 #endif
+#endif
 
 namespace folly {
 namespace detail {
@@ -35,7 +43,7 @@ bool sysMembarrierAvailable() {
     return false;
   }
 
-#ifdef __NR_membarrier
+#if FOLLY_USE_SYS_MEMBARRIER
   auto r = syscall(__NR_membarrier, MEMBARRIER_CMD_QUERY, /* flags = */ 0);
   if (r == -1) {
     return false;
@@ -48,11 +56,11 @@ bool sysMembarrierAvailable() {
 }
 
 int sysMembarrier() {
-#ifdef __NR_membarrier
+#if FOLLY_USE_SYS_MEMBARRIER
   return syscall(__NR_membarrier, MEMBARRIER_CMD_SHARED, /* flags = */ 0);
 #else
   return -1;
 #endif
 }
-}
-}
+} // namespace detail
+} // namespace folly