C++: Fix shared pointer handling for PacketPayload base class.

[libsigrok.git] / bindings / cxx / classes.cpp

diff --git a/bindings/cxx/classes.cpp b/bindings/cxx/classes.cpp
index 403499de31697e6a49db9fe17e8e00b4a2fc6adb..c90cac3f13ca69270457def55f71f06083448aee 100644 (file)
--- a/bindings/cxx/classes.cpp
+++ b/bindings/cxx/classes.cpp
@@ -973,6 +973,9 @@ Packet::Packet(shared_ptr<Device> device,
                                static_cast<const struct sr_datafeed_analog *>(
                                        structure->payload));
                        break;
+               default:
+                       payload = nullptr;
+                       break;
        }
 }
 
@@ -989,7 +992,10 @@ const PacketType *Packet::get_type()
 
 shared_ptr<PacketPayload> Packet::get_payload()
 {
-       return payload->get_shared_pointer(this);
+       if (payload)
+               return payload->get_shared_pointer(this);
+       else
+               throw Error(SR_ERR_NA);
 }
 
 PacketPayload::PacketPayload()
@@ -1001,8 +1007,8 @@ PacketPayload::~PacketPayload()
 }
 
 Header::Header(const struct sr_datafeed_header *structure) :
-       PacketPayload(),
-       StructureWrapper<Packet, const struct sr_datafeed_header>(structure)
+       StructureWrapper<Packet, const struct sr_datafeed_header>(structure),
+       PacketPayload()
 {
 }
 
@@ -1010,6 +1016,14 @@ Header::~Header()
 {
 }
 
+shared_ptr<PacketPayload> Header::get_shared_pointer(Packet *parent)
+{
+       return static_pointer_cast<PacketPayload>(
+               static_pointer_cast<Header>(
+               StructureWrapper<Packet, const struct sr_datafeed_header>::
+                       get_shared_pointer(parent)));
+}
+
 int Header::get_feed_version()
 {
        return structure->feed_version;
@@ -1023,8 +1037,8 @@ Glib::TimeVal Header::get_start_time()
 }
 
 Meta::Meta(const struct sr_datafeed_meta *structure) :
-       PacketPayload(),
-       StructureWrapper<Packet, const struct sr_datafeed_meta>(structure)
+       StructureWrapper<Packet, const struct sr_datafeed_meta>(structure),
+       PacketPayload()
 {
 }
 
@@ -1032,6 +1046,14 @@ Meta::~Meta()
 {
 }
 
+shared_ptr<PacketPayload> Meta::get_shared_pointer(Packet *parent)
+{
+       return static_pointer_cast<PacketPayload>(
+               static_pointer_cast<Meta>(
+               StructureWrapper<Packet, const struct sr_datafeed_meta>::
+                       get_shared_pointer(parent)));
+}
+
 map<const ConfigKey *, Glib::VariantBase> Meta::get_config()
 {
        map<const ConfigKey *, Glib::VariantBase> result;
@@ -1044,8 +1066,8 @@ map<const ConfigKey *, Glib::VariantBase> Meta::get_config()
 }
 
 Logic::Logic(const struct sr_datafeed_logic *structure) :
-       PacketPayload(),
-       StructureWrapper<Packet, const struct sr_datafeed_logic>(structure)
+       StructureWrapper<Packet, const struct sr_datafeed_logic>(structure),
+       PacketPayload()
 {
 }
 
@@ -1053,6 +1075,14 @@ Logic::~Logic()
 {
 }
 
+shared_ptr<PacketPayload> Logic::get_shared_pointer(Packet *parent)
+{
+       return static_pointer_cast<PacketPayload>(
+               static_pointer_cast<Logic>(
+               StructureWrapper<Packet, const struct sr_datafeed_logic>::
+                       get_shared_pointer(parent)));
+}
+
 void *Logic::get_data_pointer()
 {
        return structure->data;
@@ -1069,8 +1099,8 @@ unsigned int Logic::get_unit_size()
 }
 
 Analog::Analog(const struct sr_datafeed_analog *structure) :
-       PacketPayload(),
-       StructureWrapper<Packet, const struct sr_datafeed_analog>(structure)
+       StructureWrapper<Packet, const struct sr_datafeed_analog>(structure),
+       PacketPayload()
 {
 }
 
@@ -1078,6 +1108,14 @@ Analog::~Analog()
 {
 }
 
+shared_ptr<PacketPayload> Analog::get_shared_pointer(Packet *parent)
+{
+       return static_pointer_cast<PacketPayload>(
+               static_pointer_cast<Analog>(
+               StructureWrapper<Packet, const struct sr_datafeed_analog>::
+                       get_shared_pointer(parent)));
+}
+
 float *Analog::get_data_pointer()
 {
        return structure->data;