folly/portability/Memory.cpp

   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 #include <folly/portability/Memory.h>
  18
  19 #include <folly/portability/Config.h>
  20
  21 namespace folly {
  22 namespace detail {
  23 #if _POSIX_C_SOURCE >= 200112L || _XOPEN_SOURCE >= 600 ||                    \
  24     (defined(__ANDROID__) && (__ANDROID_API__ > 15)) ||                      \
  25     (defined(__APPLE__) && (__MAC_OS_X_VERSION_MIN_REQUIRED >= __MAC_10_6 || \
  26                             __IPHONE_OS_VERSION_MIN_REQUIRED >= __IPHONE_3_0))
  27 #include <errno.h>
  28
  29 // Use posix_memalign, but mimic the behaviour of memalign
  30 void* aligned_malloc(size_t size, size_t align) {
  31   void* ptr = nullptr;
  32   int rc = posix_memalign(&ptr, align, size);
  33   if (rc == 0) {
  34     return ptr;
  35   }
  36   errno = rc;
  37   return nullptr;
  38 }
  39
  40 void aligned_free(void* aligned_ptr) {
  41   free(aligned_ptr);
  42 }
  43 #elif defined(_WIN32)
  44 #include <malloc.h> // nolint
  45
  46 void* aligned_malloc(size_t size, size_t align) {
  47   return _aligned_malloc(size, align);
  48 }
  49
  50 void aligned_free(void* aligned_ptr) {
  51   _aligned_free(aligned_ptr);
  52 }
  53 #else
  54 #include <malloc.h> // nolint
  55
  56 void* aligned_malloc(size_t size, size_t align) {
  57   return memalign(align, size);
  58 }
  59
  60 void aligned_free(void* aligned_ptr) {
  61   free(aligned_ptr);
  62 }
  63 #endif
  64 }
  65 }