From a59f728bdd7b81964a460b96b6c56da0d9dfda01 Mon Sep 17 00:00:00 2001 From: fiaxh Date: Sat, 12 Aug 2017 23:14:50 +0200 Subject: Stream Management --- xmpp-vala/src/module/bind.vala | 7 +- xmpp-vala/src/module/sasl.vala | 4 +- .../src/module/xep/0082_date_time_profiles.vala | 21 ++-- .../src/module/xep/0198_stream_management.vala | 112 +++++++++++++++++++++ 4 files changed, 128 insertions(+), 16 deletions(-) create mode 100644 xmpp-vala/src/module/xep/0198_stream_management.vala (limited to 'xmpp-vala/src/module') diff --git a/xmpp-vala/src/module/bind.vala b/xmpp-vala/src/module/bind.vala index 7b08e0f7..a13f9640 100644 --- a/xmpp-vala/src/module/bind.vala +++ b/xmpp-vala/src/module/bind.vala @@ -7,7 +7,7 @@ namespace Xmpp.Bind { public class Module : XmppStreamNegotiationModule { public static ModuleIdentity IDENTITY = new ModuleIdentity(NS_URI, "bind_module"); - private string requested_resource; + public string requested_resource { get; set; } public signal void bound_to_resource(XmppStream stream, string my_jid); @@ -28,15 +28,14 @@ namespace Xmpp.Bind { public void received_features_node(XmppStream stream) { if (stream.is_setup_needed()) return; + if (stream.is_negotiation_active()) return; var bind = stream.features.get_subnode("bind", NS_URI); if (bind != null) { var flag = new Flag(); StanzaNode bind_node = new StanzaNode.build("bind", NS_URI).add_self_xmlns(); bind_node.put_node(new StanzaNode.build("resource", NS_URI).put_node(new StanzaNode.text(requested_resource))); - stream.get_module(Iq.Module.IDENTITY).send_iq(stream, new Iq.Stanza.set(bind_node), (stream, iq) => { - stream.get_module(Module.IDENTITY).iq_response_stanza(stream, iq); - }); + stream.get_module(Iq.Module.IDENTITY).send_iq(stream, new Iq.Stanza.set(bind_node), iq_response_stanza); stream.add_flag(flag); } } diff --git a/xmpp-vala/src/module/sasl.vala b/xmpp-vala/src/module/sasl.vala index ee6a87f7..dee9d3e6 100644 --- a/xmpp-vala/src/module/sasl.vala +++ b/xmpp-vala/src/module/sasl.vala @@ -7,8 +7,8 @@ namespace Xmpp.PlainSasl { public static ModuleIdentity IDENTITY = new ModuleIdentity(NS_URI, "plain_module"); private const string MECHANISM = "PLAIN"; - private string name; - private string password; + public string name { get; set; } + public string password { get; set; } public bool use_full_name = false; public signal void received_auth_failure(XmppStream stream, StanzaNode node); diff --git a/xmpp-vala/src/module/xep/0082_date_time_profiles.vala b/xmpp-vala/src/module/xep/0082_date_time_profiles.vala index b2ce1077..57c7ec4d 100644 --- a/xmpp-vala/src/module/xep/0082_date_time_profiles.vala +++ b/xmpp-vala/src/module/xep/0082_date_time_profiles.vala @@ -1,13 +1,13 @@ namespace Xmpp.Xep.DateTimeProfiles { - public class Module { - public Regex DATETIME_REGEX; +public class Module { + public Regex DATETIME_REGEX; - public Module() { - DATETIME_REGEX = new Regex("""^(\d{4})-(\d{2})-(\d{2})T(\d{2}):(\d{2}):(\d{2})(\.(\d{3}))?(Z|((\+|\-)(\d{2}):(\d{2})))$"""); - } + public Module() { + DATETIME_REGEX = new Regex("""^(\d{4})-(\d{2})-(\d{2})T(\d{2}):(\d{2}):(\d{2})(\.(\d{3}))?(Z|((\+|\-)(\d{2})(:(\d{2}))?))$"""); + } - public DateTime? parse_string(string time_string) { + public DateTime? parse_string(string time_string) { MatchInfo match_info; if (DATETIME_REGEX.match(time_string, RegexMatchFlags.ANCHORED, out match_info)) { int year = int.parse(match_info.fetch(1)); @@ -33,9 +33,10 @@ namespace Xmpp.Xep.DateTimeProfiles { return null; } - public string to_datetime(DateTime time) { - return time.format("%Y-%m-%dT%H:%M:%SZ"); - } +public string to_datetime(DateTime time) { + return time.to_utc().format("%Y-%m-%dT%H:%M:%SZ"); } -} \ No newline at end of file +} + +} diff --git a/xmpp-vala/src/module/xep/0198_stream_management.vala b/xmpp-vala/src/module/xep/0198_stream_management.vala new file mode 100644 index 00000000..03be907c --- /dev/null +++ b/xmpp-vala/src/module/xep/0198_stream_management.vala @@ -0,0 +1,112 @@ +using Xmpp.Core; + +namespace Xmpp.Xep.StreamManagement { + +public const string NS_URI = "urn:xmpp:sm:3"; + +public class Module : XmppStreamNegotiationModule { + public static ModuleIdentity IDENTITY = new ModuleIdentity(NS_URI, "0313_message_archive_management"); + + public int h_inbound { get; private set; default=0; } + public string? session_id { get; set; default=null; } + public Gee.List flags = null; + + public override void attach(XmppStream stream) { + stream.get_module(Bind.Module.IDENTITY).bound_to_resource.connect(check_enable); + stream.received_features_node.connect(check_resume); + + stream.received_nonza.connect(on_received_nonza); + stream.received_message_stanza.connect(on_stanza_received); + stream.received_presence_stanza.connect(on_stanza_received); + stream.received_iq_stanza.connect(on_stanza_received); + } + + public override void detach(XmppStream stream) { } + + public static void require(XmppStream stream) { + if (stream.get_module(IDENTITY) == null) stream.add_module(new PrivateXmlStorage.Module()); + } + + public override bool mandatory_outstanding(XmppStream stream) { return false; } + + public override bool negotiation_active(XmppStream stream) { + return stream.has_flag(Flag.IDENTITY) && !stream.get_flag(Flag.IDENTITY).finished; + } + + public override string get_ns() { return NS_URI; } + public override string get_id() { return IDENTITY.id; } + + private void on_stanza_received(XmppStream stream, StanzaNode node) { + lock (h_inbound) h_inbound++; + } + + private void check_resume(XmppStream stream) { + if (stream_has_sm_feature(stream) && session_id != null) { + Tls.Flag? tls_flag = stream.get_flag(Tls.Flag.IDENTITY); + if (tls_flag != null && tls_flag.finished) { + StanzaNode node = new StanzaNode.build("resume", NS_URI).add_self_xmlns() + .put_attribute("h", h_inbound.to_string()) + .put_attribute("previd", session_id); + stream.write(node); + stream.add_flag(new Flag()); + } + } + } + + private void check_enable(XmppStream stream) { + if (stream_has_sm_feature(stream) && session_id == null) { + StanzaNode node = new StanzaNode.build("enable", NS_URI).add_self_xmlns().put_attribute("resume", "true"); + stream.write(node); + stream.add_flag(new Flag()); + } + } + + private void on_received_nonza(XmppStream stream, StanzaNode node) { + if (node.ns_uri == NS_URI) { + if (node.name == "r") { + send_ack(stream); + } else if (node.name == "a") { + handle_ack(stream, node); + } else if (node.name in new string[]{"enabled", "resumed", "failed"}) { + stream.get_flag(Flag.IDENTITY).finished = true; + + if (node.name == "enabled") { + lock(h_inbound) h_inbound = 0; + session_id = node.get_attribute("id", NS_URI); + flags = stream.flags; + } else if (node.name == "resumed") { + foreach (XmppStreamFlag flag in flags) { + stream.add_flag(flag); + } + stream.negotiation_complete = true; + } else if (node.name == "failed") { + stream.received_features_node(stream); + session_id = null; + } + } + } + } + + private void send_ack(XmppStream stream) { + StanzaNode node = new StanzaNode.build("a", NS_URI).add_self_xmlns().put_attribute("h", h_inbound.to_string()); + stream.write(node); + } + + private void handle_ack(XmppStream stream, StanzaNode node) { + + } + + private bool stream_has_sm_feature(XmppStream stream) { + return stream.features.get_subnode("sm", NS_URI) != null; + } +} + +public class Flag : XmppStreamFlag { + public static FlagIdentity IDENTITY = new FlagIdentity(NS_URI, "stream_management"); + public bool finished = false; + + public override string get_ns() { return NS_URI; } + public override string get_id() { return IDENTITY.id; } +} + +} -- cgit v1.2.3-54-g00ecf