First Commit

externals/cpp-jwt/include/jwt/stack_alloc.hpp

@@ -0,0 +1,200 @@
+/*
+// The MIT License (MIT)
+// 
+// Copyright (c) 2015 Howard Hinnant
+// 
+// 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.
+ */
+
+#ifndef STACK_ALLOC_HPP
+#define STACK_ALLOC_HPP
+
+/*
+ * Based on Howard Hinnants awesome allocator boilerplate code
+ * https://howardhinnant.github.io/short_alloc.h
+ */
+
+#include <cstddef>
+#include <cassert>
+
+namespace jwt {
+
+/*
+ */
+template <
+  /// Size of the stack allocated byte buffer.
+  size_t N, 
+  /// The alignment required for the buffer.
+  size_t alignment = alignof(std::max_align_t)
+>
+class Arena
+{
+public: // 'tors
+  Arena() noexcept
+    : ptr_(buf_)
+  {
+    static_assert (alignment <= alignof(std::max_align_t),
+        "Alignment chosen is more than the maximum supported alignment");
+  }
+
+  /// Non copyable and assignable
+  Arena(const Arena&) = delete;
+  Arena& operator=(const Arena&) = delete;
+
+  ~Arena() 
+  { 
+    ptr_ = nullptr; 
+  }
+
+public: // Public APIs
+
+  /*
+   * Reserves space within the buffer of size atleast 'n'
+   * bytes.
+   * More bytes maybe reserved based on the alignment requirements.
+   *
+   * Returns:
+   * 1. The pointer within the storage buffer where the object can be constructed.
+   * 2. nullptr if space cannot be reserved for requested number of bytes 
+   *    (+ alignment padding if applicable)
+   */
+  template <
+    /// The requested alignment for this allocation.
+    /// Must be less than or equal to the 'alignment'.
+    size_t requested_alignment
+  >
+  char* allocate(size_t n) noexcept;
+
+  /*
+   * Free back the space pointed by p within the storage buffer.
+   */
+  void deallocate(char* p, size_t n) noexcept;
+
+  /*
+   * The size of the internal storage buffer.
+   */
+  constexpr static size_t size() noexcept
+  {
+    return N;
+  }
+
+  /*
+   * Returns number of remaining bytes within the storage buffer
+   * that can be used for further allocation requests.
+   */
+  size_t used() const noexcept
+  {
+    return static_cast<size_t>(ptr_ - buf_);
+  }
+
+private: // Private member functions
+
+  /*
+   * A check to determine if the pointer 'p'
+   * points to a region within storage.
+   */
+  bool pointer_in_storage(char* p) const noexcept
+  {
+    return (buf_ <= p) && (p <= (buf_ + N));
+  }
+
+  /*
+   * Rounds up the number to the next closest number
+   * as per the alignment.
+   */
+  constexpr static size_t align_up(size_t n) noexcept
+  {
+    return (n + (alignment - 1)) & ~(alignment - 1);
+  }
+
+private: // data members
+  /// Storage
+  alignas(alignment) char buf_[N];
+
+  /// Current allocation pointer within storage
+  char* ptr_ = nullptr;
+};
+
+
+
+/*
+ */
+template <
+  /// The allocator for type T
+  typename T,
+  /// Number of bytes for the arena
+  size_t N,
+  /// Alignment of the arena
+  size_t align = alignof(std::max_align_t)
+>
+class stack_alloc
+{
+public: // typedefs
+  using value_type = T;
+  using arena_type = Arena<N, align>;
+
+  static auto constexpr alignment = align;
+  static auto constexpr size = N;
+
+public: // 'tors
+  stack_alloc(arena_type& a)
+    : arena_(a)
+  {
+  }
+
+  stack_alloc(const stack_alloc&) = default;
+  stack_alloc& operator=(const stack_alloc&) = delete;
+
+  template <typename U>
+  stack_alloc(const stack_alloc<U, N, alignment>& other)
+    : arena_(other.arena_)
+  {
+  }
+
+  template <typename U>
+  struct rebind {
+    using other = stack_alloc<U, N, alignment>;
+  };
+
+public: // Exposed APIs
+
+  /*
+   * Allocate memory of 'n' bytes for object
+   * of type 'T'
+   */
+  T* allocate(size_t n) noexcept;
+
+  /*
+   * Deallocate the storage reserved for the object
+   * of type T pointed by pointer 'p'
+   */
+  void deallocate(T* p, size_t n) noexcept;
+
+private: // Private APIs
+
+private: // Private data members
+  /// The arena
+  arena_type& arena_;
+};
+
+} // END namespace jwt
+
+#include "jwt/impl/stack_alloc.ipp"
+
+#endif