From 78132e2a138f707b9522c41110079f92b39afc88 Mon Sep 17 00:00:00 2001
From: Martin Ling <martin-git@earth.li>
Date: Sat, 30 Aug 2014 12:16:51 +0100
Subject: [PATCH] C++: Add sanity checks in StructureWrapper.

---
 bindings/cxx/include/libsigrok/libsigrok.hpp | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/bindings/cxx/include/libsigrok/libsigrok.hpp b/bindings/cxx/include/libsigrok/libsigrok.hpp
index e16a9536..8c8b5549 100644
--- a/bindings/cxx/include/libsigrok/libsigrok.hpp
+++ b/bindings/cxx/include/libsigrok/libsigrok.hpp
@@ -148,6 +148,8 @@ public:
 	shared_ptr<StructureWrapper<Parent, Struct> >
 	get_shared_pointer(Parent *parent)
 	{
+		if (!parent)
+			throw Error(SR_ERR_BUG);
 		this->parent = static_pointer_cast<Parent>(parent->shared_from_this());
 		return shared_ptr<StructureWrapper<Parent, Struct> >(
 			this, reset_parent);
@@ -155,6 +157,8 @@ public:
 	shared_ptr<StructureWrapper<Parent, Struct> >
 	get_shared_pointer(shared_ptr<Parent> parent)
 	{
+		if (!parent)
+			throw Error(SR_ERR_BUG);
 		this->parent = parent;
 		return shared_ptr<StructureWrapper<Parent, Struct> >(
 			this, reset_parent);
@@ -162,6 +166,8 @@ public:
 protected:
 	static void reset_parent(StructureWrapper<Parent, Struct> *object)
 	{
+		if (!object->parent)
+			throw Error(SR_ERR_BUG);
 		object->parent.reset();
 	}