From 8604ba0f1cfc1f2cd73a9d98085143614e587888 Mon Sep 17 00:00:00 2001
From: Rodion Malinovsky <rodionmalino@gmail.com>
Date: Wed, 16 Sep 2015 14:53:12 +0300
Subject: [PATCH] Add asserts to prevent UB

---
 include/rapidjson/document.h       | 5 ++++-
 include/rapidjson/internal/stack.h | 5 ++++-
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/include/rapidjson/document.h b/include/rapidjson/document.h
index f5c1be9..6275f96 100644
--- a/include/rapidjson/document.h
+++ b/include/rapidjson/document.h
@@ -1989,7 +1989,10 @@ public:
     //!@}
 
     //! Get the allocator of this document.
-    Allocator& GetAllocator() { return *allocator_; }
+    Allocator& GetAllocator() {
+        RAPIDJSON_ASSERT(allocator_);
+        return *allocator_;
+    }
 
     //! Get the capacity of stack in bytes.
     size_t GetStackCapacity() const { return stack_.GetCapacity(); }
diff --git a/include/rapidjson/internal/stack.h b/include/rapidjson/internal/stack.h
index 0d4a7f7..4b98e38 100644
--- a/include/rapidjson/internal/stack.h
+++ b/include/rapidjson/internal/stack.h
@@ -134,7 +134,10 @@ public:
     template<typename T>
     T* Bottom() { return (T*)stack_; }
 
-    Allocator& GetAllocator() { return *allocator_; }
+    Allocator& GetAllocator() {
+        RAPIDJSON_ASSERT(allocator_);
+        return *allocator_;
+    }
     bool Empty() const { return stackTop_ == stack_; }
     size_t GetSize() const { return static_cast<size_t>(stackTop_ - stack_); }
     size_t GetCapacity() const { return static_cast<size_t>(stackEnd_ - stack_); }