1 files changed, 91 insertions, 0 deletions
diff --git a/vala-xmpp/src/module/xep/0280_message_carbons.vala b/vala-xmpp/src/module/xep/0280_message_carbons.vala
new file mode 100644
index 00000000..18b2ecdf
--- /dev/null
+++ b/vala-xmpp/src/module/xep/0280_message_carbons.vala
@@ -0,0 +1,91 @@
+using Xmpp.Core;
+
+namespace Xmpp.Xep.MessageCarbons {
+    private const string NS_URI = "urn:xmpp:carbons:2";
+
+    public class Module : XmppStreamModule {
+        public const string ID = "0280_message_carbons_module";
+
+        public void enable(XmppStream stream) {
+            Iq.Stanza iq = new Iq.Stanza.set(new StanzaNode.build("enable", NS_URI).add_self_xmlns());
+            Iq.Module.get_module(stream).send_iq(stream, iq);
+        }
+
+        public void disable(XmppStream stream) {
+            Iq.Stanza iq = new Iq.Stanza.set(new StanzaNode.build("disable", NS_URI).add_self_xmlns());
+            Iq.Module.get_module(stream).send_iq(stream, iq);
+        }
+
+        public override void attach(XmppStream stream) {
+            Bind.Module.require(stream);
+            Iq.Module.require(stream);
+            Message.Module.require(stream);
+            ServiceDiscovery.Module.require(stream);
+
+            stream.stream_negotiated.connect(enable);
+            Message.Module.get_module(stream).pre_received_message.connect(pre_received_message);
+            ServiceDiscovery.Module.get_module(stream).add_feature(stream, NS_URI);
+        }
+
+        public override void detach(XmppStream stream) {
+            stream.stream_negotiated.disconnect(enable);
+            Message.Module.get_module(stream).pre_received_message.disconnect(pre_received_message);
+        }
+
+        public static Module? get_module(XmppStream stream) {
+            return (Module?) stream.get_module(NS_URI, ID);
+        }
+
+        public static void require(XmppStream stream) {
+            if (get_module(stream) == null) stream.add_module(new Module());
+        }
+
+        public override string get_ns() { return NS_URI; }
+        public override string get_id() { return ID; }
+
+        private void pre_received_message(XmppStream stream, Message.Stanza message) {
+            StanzaNode? received_node = message.stanza.get_subnode("received", NS_URI);
+            StanzaNode? sent_node = received_node == null ? message.stanza.get_subnode("sent", NS_URI) : null;
+            StanzaNode? carbons_node = received_node != null ? received_node : sent_node;
+            if (carbons_node != null) {
+                StanzaNode? forwarded_node = carbons_node.get_subnode("forwarded", "urn:xmpp:forward:0");
+                if (forwarded_node != null) {
+                    StanzaNode? message_node = forwarded_node.get_subnode("message", Message.NS_URI);
+                    string? from_attribute = message_node.get_attribute("from", Message.NS_URI);
+                    // The security model assumed by this document is that all of the resources for a single user are in the same trust boundary.
+                    // Any forwarded copies received by a Carbons-enabled client MUST be from that user's bare JID; any copies that do not meet this requirement MUST be ignored.
+                    if (from_attribute != null && from_attribute == get_bare_jid(Bind.Flag.get_flag(stream).my_jid)) {
+                        if (received_node != null) {
+                            message.add_flag(new MessageFlag(MessageFlag.TYPE_RECEIVED));
+                        } else if (sent_node != null) {
+                            message.add_flag(new MessageFlag(MessageFlag.TYPE_SENT));
+                        }
+                        message.stanza = message_node;
+                        message.rerun_parsing = true;
+                    }
+                    message.stanza = message_node;
+                    message.rerun_parsing = true;
+                }
+            }
+        }
+    }
+
+    public class MessageFlag : Message.MessageFlag {
+        public const string id = "message_carbons";
+
+        public const string TYPE_RECEIVED = "received";
+        public const string TYPE_SENT = "sent";
+        private string type_;
+
+        public MessageFlag(string type) {
+            this.type_ = type;
+        }
+
+        public static MessageFlag? get_flag(Message.Stanza message) {
+            return (MessageFlag) message.get_flag(NS_URI, id);
+        }
+
+        public override string get_ns() { return NS_URI; }
+        public override string get_id() { return id; }
+    }
+}