Added ProtobufMessage.from_abstract_pointer (#390/#399).

Author: jordanbriere

src/core/modules/messages/messages.h

@@ -36,6 +36,7 @@
 #include "public/engine/iserverplugin.h"
 #include "bitbuf.h"
 
+#include "modules/memory/memory_utilities.h"
 
 #ifdef USE_PROTOBUF
 	#if defined(ENGINE_CSGO)
@@ -88,6 +89,14 @@
 			return boost::shared_ptr<google::protobuf::Message>(message->New());
 		}
 
+		static boost::shared_ptr<google::protobuf::Message> from_abstract_pointer(object oPtr)
+		{
+			google::protobuf::Message *pMessage = (google::protobuf::Message *)ExtractAddress(oPtr, true);
+			boost::shared_ptr<google::protobuf::Message> spMessage = __init__(pMessage->GetDescriptor()->name().c_str());
+			boost::get_pointer(spMessage)->ParseFromString(pMessage->SerializeAsString());
+			return spMessage;
+		}
+
 		static const google::protobuf::FieldDescriptor* GetFieldDescriptor(google::protobuf::Message* pMessage, const char* field_name)
 		{
 			const google::protobuf::Descriptor* descriptor = pMessage->GetDescriptor();

src/core/modules/messages/messages_wrap.cpp

@@ -105,7 +105,21 @@ void export_protobuf_message(scope _messages)
 	class_<google::protobuf::Message, boost::shared_ptr<google::protobuf::Message>, boost::noncopyable> ProtobufMessage("ProtobufMessage", no_init);
 
 	ProtobufMessage.def("__init__", make_constructor(&CProtobufMessageExt::__init__));
-	
+	ProtobufMessage.def(
+		"from_abstract_pointer",
+		&CProtobufMessageExt::from_abstract_pointer,
+		"Returns a new protobuf message instance filled with the fields contained in the given ProtobufMessage pointer.\n"
+		"\n"
+		":param object pointer:"
+		"	The pointer of the message to merge the fields from.\n"
+		"\n"
+		".. note::\n"
+		"	No checks is done on the given pointer and it is assumed to be valid. Do not use this "
+		"	constructor if you don't know what you are doing. This should really only be used to wrap "
+		"	the content of protobuf messages that do not have RTTI data (e.g. during the execution of certain hooks, etc.).",
+		args("pointer")
+	).staticmethod("from_abstract_pointer");
+
 	ProtobufMessage.def("get_int32", &CProtobufMessageExt::GetInt32);
 	ProtobufMessage.def("get_int64", &CProtobufMessageExt::GetInt64);
 	ProtobufMessage.def("get_uint32", &CProtobufMessageExt::GetUInt32);