using Xmpp; using Xmpp.Xep; using Gee; namespace Xmpp.Xep.HttpFileUpload { private const string NS_URI = "urn:xmpp:http:upload"; private const string NS_URI_0 = "urn:xmpp:http:upload:0"; public errordomain HttpFileTransferError { SLOT_REQUEST } public class Module : XmppStreamModule { public static Xmpp.ModuleIdentity IDENTITY = new Xmpp.ModuleIdentity(NS_URI, "0363_http_file_upload"); public signal void feature_available(XmppStream stream, long max_file_size); public signal void received_url(XmppStream stream, MessageStanza message); public delegate void OnSlotOk(XmppStream stream, string url_get, string url_put); public delegate void OnError(XmppStream stream, string error); public struct SlotResult { public string url_get { get; set; } public string url_put { get; set; } public HashMap headers { get; set; } } public async SlotResult request_slot(XmppStream stream, string filename, int64 file_size, string? content_type) throws HttpFileTransferError { Flag? flag = stream.get_flag(Flag.IDENTITY); if (flag == null) { throw new HttpFileTransferError.SLOT_REQUEST("No flag"); } StanzaNode? request_node = null; switch (flag.ns_ver) { case NS_URI_0: request_node = new StanzaNode.build("request", NS_URI_0).add_self_xmlns(); request_node.put_attribute("filename", filename).put_attribute("size", file_size.to_string()); if (content_type != null) request_node.put_attribute("content-type", content_type); break; case NS_URI: request_node = new StanzaNode.build("request", NS_URI).add_self_xmlns() .put_node(new StanzaNode.build("filename", NS_URI).put_node(new StanzaNode.text(filename))) .put_node(new StanzaNode.build("size", NS_URI).put_node(new StanzaNode.text(file_size.to_string()))); if (content_type != null) { request_node.put_node(new StanzaNode.build("content-type", NS_URI).put_node(new StanzaNode.text(content_type))); } break; } SourceFunc callback = request_slot.callback; var slot_result = SlotResult(); Iq.Stanza iq = new Iq.Stanza.get(request_node) { to=flag.file_store_jid }; HttpFileTransferError? e = null; stream.get_module(Iq.Module.IDENTITY).send_iq(stream, iq, (stream, iq) => { if (iq.is_error()) { e = new HttpFileTransferError.SLOT_REQUEST("Error getting upload/download url (Error Iq)"); Idle.add((owned) callback); return; } string? url_get = null, url_put = null; // FIXME change back to switch on version in a while (prosody bug) url_get = iq.stanza.get_deep_attribute(flag.ns_ver + ":slot", flag.ns_ver + ":get", flag.ns_ver + ":url"); url_put = iq.stanza.get_deep_attribute(flag.ns_ver + ":slot", flag.ns_ver + ":put", flag.ns_ver + ":url"); if (url_get == null && url_put == null) { url_get = iq.stanza.get_deep_string_content(flag.ns_ver + ":slot", flag.ns_ver + ":get"); url_put = iq.stanza.get_deep_string_content(flag.ns_ver + ":slot", flag.ns_ver + ":put"); } if (url_get == null || url_put == null) { e = new HttpFileTransferError.SLOT_REQUEST("Error getting upload/download url: %s".printf(iq.stanza.to_string())); Idle.add((owned) callback); return; } if (!url_get.down().has_prefix("https://") || !url_put.down().has_prefix("https://")) { e = new HttpFileTransferError.SLOT_REQUEST("Error getting upload/download url: Received non-https URL from server"); Idle.add((owned) callback); return; } slot_result.headers = new HashMap(); foreach (StanzaNode node in iq.stanza.get_deep_subnodes(flag.ns_ver + ":slot", flag.ns_ver + ":put", flag.ns_ver + ":header")) { string header_name = node.get_attribute("name"); if (header_name == "Authorization" || header_name == "Cookie" || header_name == "Expires") { string? header_val = node.get_string_content(); if (header_val != null && header_val.length < 8192) { header_val = header_val.replace("\n", "").replace("\r", ""); slot_result.headers[header_name] = header_val; } } } slot_result.url_get = url_get; slot_result.url_put = url_put; Idle.add((owned) callback); }); yield; if (e != null) { throw e; } return slot_result; } public override void attach(XmppStream stream) { stream.stream_negotiated.connect(query_availability); } public override void detach(XmppStream stream) { stream.stream_negotiated.disconnect(query_availability); } public override string get_ns() { return NS_URI; } public override string get_id() { return IDENTITY.id; } private async void query_availability(XmppStream stream) { ServiceDiscovery.InfoResult? info_result = yield stream.get_module(ServiceDiscovery.Module.IDENTITY).request_info(stream, stream.remote_name); bool available = check_ns_in_info(stream, stream.remote_name, info_result); if (!available) { ServiceDiscovery.ItemsResult? items_result = yield stream.get_module(ServiceDiscovery.Module.IDENTITY).request_items(stream, stream.remote_name); if (items_result == null) return; for (int i = 0; i < 2; i++) { foreach (Xep.ServiceDiscovery.Item item in items_result.items) { // First try the promising items and only afterwards all the others bool promising_upload_item = item.jid.to_string().has_prefix("upload"); if ((i == 0 && !promising_upload_item) || (i == 1) && promising_upload_item) continue; ServiceDiscovery.InfoResult? info_result2 = yield stream.get_module(ServiceDiscovery.Module.IDENTITY).request_info(stream, item.jid); bool available2 = check_ns_in_info(stream, item.jid, info_result2); if (available2) return; } } } } private bool check_ns_in_info(XmppStream stream, Jid jid, Xep.ServiceDiscovery.InfoResult? info_result) { if (info_result == null) return false; bool ver_available = false; bool ver_0_available = false; foreach (string feature in info_result.features) { if (feature == NS_URI_0) { ver_0_available = true; break; } else if (feature == NS_URI) { ver_available = true; } } if (ver_available || ver_0_available) { long max_file_size = extract_max_file_size(info_result); if (ver_0_available) { stream.add_flag(new Flag(jid, NS_URI_0)); } else if (ver_available) { stream.add_flag(new Flag(jid, NS_URI)); } feature_available(stream, max_file_size); return true; } return false; } private long extract_max_file_size(Xep.ServiceDiscovery.InfoResult info_result) { string? max_file_size_str = null; Gee.List x_nodes = info_result.iq.stanza.get_deep_subnodes("http://jabber.org/protocol/disco#info:query", "jabber:x:data:x"); foreach(StanzaNode x_node in x_nodes) { Gee.List field_nodes = x_node.get_subnodes("field", "jabber:x:data"); foreach (StanzaNode node in field_nodes) { string? var_attr = node.get_attribute("var"); if (var_attr == "max-file-size") { StanzaNode value_node = node.get_subnode("value", "jabber:x:data"); max_file_size_str = value_node.get_string_content(); break; } } } if (max_file_size_str != null) return long.parse(max_file_size_str); // If there is no max-file-size node, there is no file size limit return long.MAX; } } public class ReceivedPipelineListener : StanzaListener { private string[] after_actions_const = {"EXTRACT_MESSAGE_2"}; public override string action_group { get { return "EXTRACT_MESSAGE_2"; } } public override string[] after_actions { get { return after_actions_const; } } public override async bool run(XmppStream stream, MessageStanza message) { string? oob_url = OutOfBandData.get_url_from_message(message); if (oob_url != null && oob_url == message.body) { stream.get_module(Module.IDENTITY).received_url(stream, message); } return false; } } public class Flag : XmppStreamFlag { public static FlagIdentity IDENTITY = new FlagIdentity(NS_URI, "http_file_upload"); public Jid file_store_jid; public string ns_ver; public int? max_file_size; public Flag(Jid file_store_jid, string ns_ver) { this.file_store_jid = file_store_jid; this.ns_ver = ns_ver; } public override string get_ns() { return NS_URI; } public override string get_id() { return IDENTITY.id; } } }