Fixed #314.

Author: jordanbriere

addons/source-python/packages/source-python/messages/hooks.py

@@ -28,6 +28,8 @@
 from listeners import ListenerManager
 #   Memory
 from memory import make_object
+from memory import get_object_pointer
+from memory import get_size
 from memory import get_virtual_function
 from memory.hooks import PreHook
 from memory.hooks import PostHook
@@ -171,9 +173,16 @@ def _pre_send_user_message(args):
         if not user_message_hooks and not protobuf_user_message_hooks:
             return
 
-        # Replace original recipients filter
-        tmp_recipients = make_object(BaseRecipientFilter, args[1])
-        _recipients.update(*tuple(tmp_recipients), clear=True)
+        try:
+            # Replace original recipients filter
+            tmp_recipients = make_object(BaseRecipientFilter, args[1])
+            _recipients.update(*tuple(tmp_recipients), clear=True)
+        except RuntimeError:
+            # Patch for issue #314
+            tmp_recipients = RecipientFilter()
+            (args[1] + 4).copy(get_object_pointer(tmp_recipients) + 4,
+                get_size(RecipientFilter) - 4)
+            _recipients.update(*tuple(tmp_recipients), clear=True)
         args[1] = _recipients
 
         buffer = make_object(ProtobufMessage, args[3])
@@ -200,9 +209,16 @@ def _pre_send_user_message(args):
 else:
     @PreHook(get_virtual_function(engine_server, 'UserMessageBegin'))
     def _pre_user_message_begin(args):
-        # Replace original recipients filter
-        tmp_recipients = make_object(BaseRecipientFilter, args[1])
-        _recipients.update(*tuple(tmp_recipients), clear=True)
+        try:
+            # Replace original recipients filter
+            tmp_recipients = make_object(BaseRecipientFilter, args[1])
+            _recipients.update(*tuple(tmp_recipients), clear=True)
+        except RuntimeError:
+            # Patch for issue #314
+            tmp_recipients = RecipientFilter()
+            (args[1] + 4).copy(get_object_pointer(tmp_recipients) + 4,
+                get_size(RecipientFilter) - 4)
+            _recipients.update(*tuple(tmp_recipients), clear=True)
         args[1] = _recipients
 
     @PostHook(get_virtual_function(engine_server, 'UserMessageBegin'))
@@ -260,4 +276,4 @@ def _pre_message_end(args):
         # Update buffer if data has been changed
         if data.has_been_changed():
             buffer_write.seek_to_bit(0)
-            impl.write(buffer_write, data)
+            impl.write(buffer_write, data)