new.cpp

Go to the documentation of this file.

/*
 * This file is part of the µOS++ distribution.
 *   (https://github.com/micro-os-plus)
 * Copyright (c) 2016 Liviu Ionescu.
 *
 * Permission is hereby granted, free of charge, to any person
 * obtaining a copy of this software and associated documentation
 * files (the "Software"), to deal in the Software without
 * restriction, including without limitation the rights to use,
 * copy, modify, merge, publish, distribute, sublicense, and/or
 * sell copies of the Software, and to permit persons to whom
 * the Software is furnished to do so, subject to the following
 * conditions:
 *
 * The above copyright notice and this permission notice shall be
 * included in all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
 * OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
 * HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
 * WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
 * OTHER DEALINGS IN THE SOFTWARE.
 */
 
/*
 * [Partly inspired from the LLVM libcxx sources].
 * Copyright (c) 2009-2013 by the contributors listed in
 * 'LLVM libcxx Credits.txt'. See 'LLVM libcxx License.txt' for details.
 *
 * References are to ISO/IEC 14882:2011(E) Third edition (2011-09-01)
 */
 
#include <cmsis-plus/rtos/os.h>
#include <cmsis-plus/estd/memory_resource>
 
// ----------------------------------------------------------------------------
 
#if defined(__clang__)
#pragma clang diagnostic ignored "-Wc++98-compat"
#endif
 
// ----------------------------------------------------------------------------
 
using namespace os;
 
// ----------------------------------------------------------------------------
 
namespace
{
  std::new_handler new_handler_;
}
 
namespace std
{
  // Constant to be used as parameter to differentiate
  // the `noexcept` functions.
  const nothrow_t nothrow = nothrow_t
    { };
 
  new_handler
  set_new_handler (new_handler handler) noexcept
  {
    trace::printf ("std::%s(%p) \n", __func__, handler);
 
    new_handler prev_handler;
 
    prev_handler = new_handler_;
    new_handler_ = handler;
 
    return prev_handler;
  }
 
  new_handler
  get_new_handler () noexcept
  {
    return new_handler_;
  }
 
} /* namespace std */
 
// ----------------------------------------------------------------------------
 
void *
__attribute__((weak))
operator new (std::size_t bytes)
{
  assert(!rtos::interrupts::in_handler_mode ());
  if (bytes == 0)
    {
      bytes = 1;
    }
 
  // ----- Begin of critical section ------------------------------------------
  rtos::scheduler::critical_section scs;
 
  while (true)
    {
      void* mem = estd::pmr::get_default_resource ()->allocate (bytes);
 
      if (mem != nullptr)
        {
#if defined(OS_TRACE_LIBCPP_OPERATOR_NEW)
          trace::printf ("::%s(%d)=%p\n", __func__, bytes, mem);
#endif
          return mem;
        }
 
      // If allocate() fails and there is a new_handler,
      // call it to try free up memory.
      if (new_handler_)
        {
          new_handler_ ();
        }
      else
        {
          estd::__throw_bad_alloc ();
        }
    }
 
  // ----- End of critical section --------------------------------------------
}
 
void*
__attribute__((weak))
operator new (std::size_t bytes,
              const std::nothrow_t& nothrow __attribute__((unused))) noexcept
{
  assert(!rtos::interrupts::in_handler_mode ());
 
  if (bytes == 0)
    {
      bytes = 1;
    }
 
  // ----- Begin of critical section ------------------------------------------
  rtos::scheduler::critical_section scs;
 
  while (true)
    {
      void* mem = estd::pmr::get_default_resource ()->allocate (bytes);
 
      if (mem != nullptr)
        {
#if defined(OS_TRACE_LIBCPP_OPERATOR_NEW)
          trace::printf ("::%s(%d)=%p\n", __func__, bytes, mem);
#endif
          return mem;
        }
 
      // If allocate() fails and there is a new_handler,
      // call it to try free up memory.
      if (new_handler_)
        {
          new_handler_ ();
        }
      else
        {
          break; // return nullptr
        }
    }
 
  // ----- End of critical section --------------------------------------------
 
  return nullptr;
}
 
void*
__attribute__((weak))
operator new[] (std::size_t bytes)
{
  return ::operator new (bytes);
}
 
void*
__attribute__((weak))
operator new[] (std::size_t bytes,
                const std::nothrow_t& nothrow __attribute__((unused))) noexcept
{
  return ::operator new (bytes, std::nothrow);
}
 
// ----------------------------------------------------------------------------
 
void
__attribute__((weak))
operator delete (void* ptr) noexcept
{
#if defined(OS_TRACE_LIBCPP_OPERATOR_NEW)
  trace::printf ("::%s(%p)\n", __func__, ptr);
#endif
 
  assert(!rtos::interrupts::in_handler_mode ());
 
  if (ptr)
    {
      // ----- Begin of critical section --------------------------------------
      rtos::scheduler::critical_section scs;
 
      // The unknown size is passed as 0.
      estd::pmr::get_default_resource ()->deallocate (ptr, 0);
      // ----- End of critical section ----------------------------------------
    }
}
 
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wc++14-compat"
 
void
operator delete (void* ptr, std::size_t bytes) noexcept;
 
void
__attribute__((weak))
operator delete (void* ptr, std::size_t bytes) noexcept
{
#if defined(OS_TRACE_LIBCPP_OPERATOR_NEW)
  trace::printf ("::%s(%p,%u)\n", __func__, ptr, bytes);
#endif
 
  assert(!rtos::interrupts::in_handler_mode ());
 
  if (ptr)
    {
      // ----- Begin of critical section --------------------------------------
      rtos::scheduler::critical_section scs;
 
      estd::pmr::get_default_resource ()->deallocate (ptr, bytes);
      // ----- End of critical section ----------------------------------------
    }
}
 
#pragma GCC diagnostic pop
 
void
__attribute__((weak))
operator delete (void* ptr,
                 const std::nothrow_t& nothrow __attribute__((unused))) noexcept
{
#if defined(OS_TRACE_LIBCPP_OPERATOR_NEW)
  trace::printf ("::%s(%p)\n", __func__, ptr);
#endif
 
  assert(!rtos::interrupts::in_handler_mode ());
 
  if (ptr)
    {
      // ----- Begin of critical section --------------------------------------
      rtos::scheduler::critical_section scs;
 
      estd::pmr::get_default_resource ()->deallocate (ptr, 0);
      // ----- End of critical section ----------------------------------------
    }
}
 
void
__attribute__((weak))
operator delete[] (void* ptr) noexcept
{
  ::operator delete (ptr);
}
 
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wc++14-compat"
 
void
operator delete[] (void* ptr, std::size_t bytes) noexcept;
 
void
__attribute__((weak))
operator delete[] (void* ptr, std::size_t bytes) noexcept
{
  ::operator delete (ptr, bytes);
}
 
#pragma GCC diagnostic pop
 
void
__attribute__((weak))
operator delete[] (void* ptr, const std::nothrow_t& nothrow) noexcept
{
  ::operator delete (ptr, nothrow);
}
 
// ----------------------------------------------------------------------------