From 34a2dd333b605ce26fd2785a642c5fe02c1316ca Mon Sep 17 00:00:00 2001
From: Guilherme Maciel Ferreira <guilherme.maciel.ferreira@gmail.com>
Date: Fri, 4 Nov 2011 00:08:39 -0200
Subject: [PATCH] Providing read() and write() methods, once operators >> and << does work properly with smart pointers

---
 src/icmp/icmppacket.h     |   11 +++++++++++
 src/icmp/icmpv4packet.cpp |   12 ++++++++++++
 src/icmp/icmpv4packet.h   |    3 +++
 src/icmp/icmpv6packet.cpp |   12 ++++++++++++
 src/icmp/icmpv6packet.h   |    3 +++
 5 files changed, 41 insertions(+), 0 deletions(-)

diff --git a/src/icmp/icmppacket.h b/src/icmp/icmppacket.h
index 7949198..a2c1084 100644
--- a/src/icmp/icmppacket.h
+++ b/src/icmp/icmppacket.h
@@ -23,10 +23,12 @@
 
 #include <stdint.h>
 
+#include <iostream>
 #include <string>
 
 #include <boost/asio.hpp>
 #include <boost/date_time/posix_time/posix_time_types.hpp>
+#include <boost/shared_ptr.hpp>
 
 //-----------------------------------------------------------------------------
 // IcmpPacket
@@ -58,6 +60,15 @@ public:
     ) const = 0;
     virtual void print_destination_unreachable() const = 0;
 
+    virtual bool read( std::istream &is ) = 0;
+    virtual void write( std::ostream &os ) const = 0;
+
 };
 
+//-----------------------------------------------------------------------------
+// IcmpPacketItem
+//-----------------------------------------------------------------------------
+
+typedef boost::shared_ptr<IcmpPacket> IcmpPacketItem;
+
 #endif // ICMP_PACKET_H
diff --git a/src/icmp/icmpv4packet.cpp b/src/icmp/icmpv4packet.cpp
index 65155e2..0a60aa8 100644
--- a/src/icmp/icmpv4packet.cpp
+++ b/src/icmp/icmpv4packet.cpp
@@ -202,6 +202,18 @@ void Icmpv4Packet::print_destination_unreachable() const
          << " Destination Net Unreachable" << endl;
 }
 
+bool Icmpv4Packet::read( istream &is )
+{
+    is >> *this;
+
+    return is.fail();
+}
+
+void Icmpv4Packet::write( ostream &os ) const
+{
+    os << *this;
+}
+
 istream& operator>>(
         istream &is,
         Icmpv4Packet &packet
diff --git a/src/icmp/icmpv4packet.h b/src/icmp/icmpv4packet.h
index a2dc56d..51511a3 100644
--- a/src/icmp/icmpv4packet.h
+++ b/src/icmp/icmpv4packet.h
@@ -97,6 +97,9 @@ public:
     ) const;
     void print_destination_unreachable() const;
 
+    bool read( std::istream &is );
+    void write( std::ostream &os ) const;
+
     friend std::istream& operator>>(
             std::istream &is,
             Icmpv4Packet &packet
diff --git a/src/icmp/icmpv6packet.cpp b/src/icmp/icmpv6packet.cpp
index 0661c93..ecaff54 100644
--- a/src/icmp/icmpv6packet.cpp
+++ b/src/icmp/icmpv6packet.cpp
@@ -202,6 +202,18 @@ void Icmpv6Packet::print_destination_unreachable() const
          << " Destination Net Unreachable" << endl;
 }
 
+bool Icmpv6Packet::read( istream &is )
+{
+    is >> *this;
+
+    return is.fail();
+}
+
+void Icmpv6Packet::write( ostream &os ) const
+{
+    os << *this;
+}
+
 istream& operator>>(
         istream &is,
         Icmpv6Packet &packet
diff --git a/src/icmp/icmpv6packet.h b/src/icmp/icmpv6packet.h
index d3923b4..4e88da8 100644
--- a/src/icmp/icmpv6packet.h
+++ b/src/icmp/icmpv6packet.h
@@ -111,6 +111,9 @@ public:
     ) const;
     void print_destination_unreachable() const;
 
+    bool read( std::istream &is );
+    void write( std::ostream &os ) const;
+
     friend std::istream& operator>>(
             std::istream &is,
             Icmpv6Packet &packet
-- 
1.7.1