X-Git-Url: http://git.tdb.fi/?p=libs%2Fnet.git;a=blobdiff_plain;f=source%2Fnet%2Fprotocol.cpp;h=52678af2d9009cf9279d88ce3e1856d0fee1be9b;hp=4d3f9628f9d1eddbc4a9bc81f9547bba6dbf153b;hb=8248f2912454229f4ded72b8fb4db3e76ea3ad15;hpb=df1b7561be0d4a57e964c783c01060c54864ec60

diff --git a/source/net/protocol.cpp b/source/net/protocol.cpp
index 4d3f962..52678af 100644
--- a/source/net/protocol.cpp
+++ b/source/net/protocol.cpp
@@ -5,6 +5,7 @@
 #include <msp/strings/format.h>
 #include <msp/strings/lexicalcast.h>
 #include "protocol.h"
+#include "protocol_impl.h"
 
 using namespace std;
 
@@ -29,9 +30,13 @@ void Protocol::add_packet(PacketDefBase *pdef)
 {
 	PacketDefBase *&ptr = packet_class_defs[pdef->get_class_id()];
 	if(ptr)
+	{
+		packet_id_defs.erase(ptr->get_id());
 		delete ptr;
+	}
 	ptr = pdef;
-	packet_id_defs[pdef->get_id()] = pdef;
+	if(unsigned id = pdef->get_id())
+		packet_id_defs[id] = pdef;
 }
 
 const Protocol::PacketDefBase &Protocol::get_packet_by_class_id(unsigned id) const