folly/portability/Builtins.h

   1 /*
   2  * Copyright 2017 Facebook, Inc.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *   http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 #pragma once
  18
  19 #ifdef _WIN32
  20 #include <assert.h>
  21 #include <folly/Portability.h>
  22 #include <intrin.h>
  23 #include <stdint.h>
  24
  25 namespace folly {
  26 namespace portability {
  27 namespace detail {
  28 void call_flush_instruction_cache_self_pid(void* begin, size_t size);
  29 }
  30 }
  31 }
  32
  33 FOLLY_ALWAYS_INLINE void __builtin___clear_cache(char* begin, char* end) {
  34   if (folly::kIsArchAmd64) {
  35     // x86_64 doesn't require the instruction cache to be flushed after
  36     // modification.
  37   } else {
  38     // Default to flushing it for everything else, such as ARM.
  39     folly::portability::detail::call_flush_instruction_cache_self_pid(
  40         static_cast<void*>(begin), static_cast<size_t>(end - begin));
  41   }
  42 }
  43
  44 FOLLY_ALWAYS_INLINE int __builtin_clz(unsigned int x) {
  45   unsigned long index;
  46   return (int)(_BitScanReverse(&index, (unsigned long)x) ? 31 - index : 32);
  47 }
  48
  49 FOLLY_ALWAYS_INLINE int __builtin_clzl(unsigned long x) {
  50   return __builtin_clz((unsigned int)x);
  51 }
  52
  53 FOLLY_ALWAYS_INLINE int __builtin_clzll(unsigned long long x) {
  54   unsigned long index;
  55   return (int)(_BitScanReverse64(&index, x) ? 63 - index : 64);
  56 }
  57
  58 FOLLY_ALWAYS_INLINE int __builtin_ctzll(unsigned long long x) {
  59   unsigned long index;
  60   return (int)(_BitScanForward64(&index, x) ? index : 64);
  61 }
  62
  63 FOLLY_ALWAYS_INLINE int __builtin_ffs(int x) {
  64   unsigned long index;
  65   return (int)(_BitScanForward(&index, (unsigned long)x) ? index + 1 : 0);
  66 }
  67
  68 FOLLY_ALWAYS_INLINE int __builtin_ffsl(long x) { return __builtin_ffs((int)x); }
  69
  70 FOLLY_ALWAYS_INLINE int __builtin_ffsll(long long x) {
  71   unsigned long index;
  72   return (int)(_BitScanForward64(&index, (unsigned long long)x) ? index + 1 : 0);
  73 }
  74
  75 FOLLY_ALWAYS_INLINE int __builtin_popcountll(unsigned long long x) {
  76   return (int)__popcnt64(x);
  77 }
  78
  79 FOLLY_ALWAYS_INLINE void* __builtin_return_address(unsigned int frame) {
  80   // I really hope frame is zero...
  81   (void)frame;
  82   assert(frame == 0);
  83   return _ReturnAddress();
  84 }
  85 #endif