From 4f8fd5690355a4202e8f29eb175bb782d4417cef Mon Sep 17 00:00:00 2001
From: Treeston <treeston.mmoc@gmail.com>
Date: Tue, 25 Dec 2018 01:47:24 +0100
Subject: Core/Utils: Fix Unicode handling

(cherry picked from commit 6d6077e36fe9c5cb8ea8e4f981d637e72ee87037)
---
 src/common/Utilities/Containers.h | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)

(limited to 'src/common/Utilities/Containers.h')

diff --git a/src/common/Utilities/Containers.h b/src/common/Utilities/Containers.h
index ca4a00e4c29..cf4d31d4dc2 100644
--- a/src/common/Utilities/Containers.h
+++ b/src/common/Utilities/Containers.h
@@ -21,6 +21,8 @@
 #include "Define.h"
 #include "Random.h"
 #include <algorithm>
+#include <exception>
+#include <iterator>
 #include <utility>
 #include <vector>
 
@@ -38,6 +40,34 @@ namespace Trinity
         return std::addressof(not_ptr);
     }
 
+    template <class T>
+    class BufferWriteGuard
+    {
+        public:
+            using iterator_category = std::output_iterator_tag;
+            using value_type = void;
+            using pointer = T*;
+            using reference = T&;
+            using difference_type = std::ptrdiff_t;
+
+            BufferWriteGuard(T* buf, size_t n) : _buf(buf), _n(n) {}
+
+            T& operator*() const { check(); return *_buf; }
+            BufferWriteGuard& operator++() { check(); ++_buf; --_n; return *this; }
+            T* operator++(int) { check(); T* b = _buf; ++_buf; --_n; return b; }
+
+            size_t size() const { return _n; }
+
+        private:
+            T* _buf;
+            size_t _n;
+            void check()
+            {
+                if (!_n)
+                    throw std::out_of_range("index");
+            }
+    };
+
     namespace Containers
     {
         // replace with std::size in C++17
-- 
cgit v1.2.3