src/lib/support/CHIPMem.cpp

   1 /*
   2  *
   3  *    Copyright (c) 2020 Project CHIP Authors
   4  *    All rights reserved.
   5  *
   6  *    Licensed under the Apache License, Version 2.0 (the "License");
   7  *    you may not use this file except in compliance with the License.
   8  *    You may obtain a copy of the License at
   9  *
  10  *        http://www.apache.org/licenses/LICENSE-2.0
  11  *
  12  *    Unless required by applicable law or agreed to in writing, software
  13  *    distributed under the License is distributed on an "AS IS" BASIS,
  14  *    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15  *    See the License for the specific language governing permissions and
  16  *    limitations under the License.
  17  */
  18
  19 /**
  20  *    @file
  21  *      This file implements parts of the CHIP allocation API that are
  22  *      independent of the configured allocator.
  23  *
  24  */
  25
  26 #include <core/CHIPConfig.h>
  27 #include <support/CHIPMem.h>
  28 #include <support/CHIPPlatformMemory.h>
  29
  30 #include <atomic>
  31 #include <stdlib.h>
  32
  33 namespace chip {
  34 namespace Platform {
  35
  36 extern CHIP_ERROR MemoryAllocatorInit(void * buf, size_t bufSize);
  37 extern void MemoryAllocatorShutdown();
  38
  39 static std::atomic_int memoryInitializationCount{ 0 };
  40
  41 CHIP_ERROR MemoryInit(void * buf, size_t bufSize)
  42 {
  43     if (memoryInitializationCount++ > 0)
  44     {
  45         return CHIP_NO_ERROR;
  46     }
  47     // Initialize the allocator.
  48     CHIP_ERROR err = MemoryAllocatorInit(buf, bufSize);
  49     if (err != CHIP_NO_ERROR)
  50     {
  51         return err;
  52     }
  53     // Here we do things like mbedtls_platform_set_calloc_free(), depending on configuration.
  54     return err;
  55 }
  56
  57 void MemoryShutdown()
  58 {
  59     if ((memoryInitializationCount > 0) && (--memoryInitializationCount == 0))
  60     {
  61         // Here we undo things like mbedtls_platform_set_calloc_free()
  62         MemoryAllocatorShutdown();
  63     }
  64 }
  65
  66 } // namespace Platform
  67 } // namespace chip