Line data    Source code

       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 <lib/core/CHIPConfig.h>
      27             : #include <lib/support/CHIPMem.h>
      28             : #include <lib/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         213 : CHIP_ERROR MemoryInit(void * buf, size_t bufSize)
      42             : {
      43         213 :     if (memoryInitializationCount++ > 0)
      44             :     {
      45         105 :         return CHIP_NO_ERROR;
      46             :     }
      47             :     // Initialize the allocator.
      48         108 :     CHIP_ERROR err = MemoryAllocatorInit(buf, bufSize);
      49         108 :     if (err != CHIP_NO_ERROR)
      50             :     {
      51           0 :         return err;
      52             :     }
      53             :     // Here we do things like mbedtls_platform_set_calloc_free(), depending on configuration.
      54         108 :     return err;
      55             : }
      56             : 
      57         211 : void MemoryShutdown()
      58             : {
      59         211 :     if ((memoryInitializationCount > 0) && (--memoryInitializationCount == 0))
      60             :     {
      61             :         // Here we undo things like mbedtls_platform_set_calloc_free()
      62         106 :         MemoryAllocatorShutdown();
      63             :     }
      64         211 : }
      65             : 
      66             : } // namespace Platform
      67             : } // namespace chip