stop creating icmp packet parts from istream if earlier part failed

diff --git a/src/icmp/icmpv4packet.cpp b/src/icmp/icmpv4packet.cpp
index 6a20879..9b52e38 100644 (file)
--- a/src/icmp/icmpv4packet.cpp
+++ b/src/icmp/icmpv4packet.cpp
@@ -247,7 +247,12 @@ istream& operator>>(
         Icmpv4Packet &packet
 )
 {
-    is >> packet.IpHeader >> packet.IcmpPayloadHeader >> packet.IcmpPayloadData;
+    if (is.good())
+        is >> packet.IpHeader;
+    if (is.good())
+        is >> packet.IcmpPayloadHeader;
+    if (is.good())
+        is >> packet.IcmpPayloadData;
 
     return is;
 }

diff --git a/src/icmp/icmpv6packet.cpp b/src/icmp/icmpv6packet.cpp
index f45356e..7f78170 100644 (file)
--- a/src/icmp/icmpv6packet.cpp
+++ b/src/icmp/icmpv6packet.cpp
@@ -262,9 +262,17 @@ istream& operator>>(
 {
 #ifdef IPV6_DATA_PRESENT_IN_ISTREAM
     //TODO WHY IPv6 does not come like IPv4????
-    is >> packet.IpHeader >> packet.IcmpPayloadHeader >> packet.IcmpPayloadData;
+    if (is.good())
+        is >> packet.IpHeader;
+    if (is.good())
+        is >> packet.IcmpPayloadHeader;
+    if (is.good())
+        is >> packet.IcmpPayloadData;
 #else
-    is >> packet.IcmpPayloadHeader >> packet.IcmpPayloadData;
+    if (is.good())
+        is >> packet.IcmpPayloadHeader;
+    if (is.good())
+        is >> packet.IcmpPayloadData;
 #endif
 
     return is;