libervia-backend: src/plugins/plugin_xep

annotate src/plugins/plugin_xep_0047.py @ 531:3bd8f84f920d

primitivus: fixed info messages timestamp

author	Goffi <goffi@goffi.org>
date	Sun, 28 Oct 2012 17:59:24 +0100
parents	2a072735e459
children	a31abb97310d

rev	line source
384 785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	1 #!/usr/bin/python
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	2 # -- coding: utf-8 --
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	3
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	4 """
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	5 SAT plugin for managing gateways (xep-0047)
459 cf005701624b copyleft date update Goffi <goffi@goffi.org> parents: 404 diff changeset	6 Copyright (C) 2009, 2010, 2011, 2012 Jérôme Poisson (goffi@goffi.org)
384 785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	7
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	8 This program is free software: you can redistribute it and/or modify
480 2a072735e459 Licence modification: the full project is now under AGPL v3+ instead of GPL v3+ Goffi <goffi@goffi.org> parents: 459 diff changeset	9 it under the terms of the GNU Affero General Public License as published by
384 785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	10 the Free Software Foundation, either version 3 of the License, or
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	11 (at your option) any later version.
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	12
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	13 This program is distributed in the hope that it will be useful,
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	14 but WITHOUT ANY WARRANTY; without even the implied warranty of
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	15 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
480 2a072735e459 Licence modification: the full project is now under AGPL v3+ instead of GPL v3+ Goffi <goffi@goffi.org> parents: 459 diff changeset	16 GNU Affero General Public License for more details.
384 785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	17
480 2a072735e459 Licence modification: the full project is now under AGPL v3+ instead of GPL v3+ Goffi <goffi@goffi.org> parents: 459 diff changeset	18 You should have received a copy of the GNU Affero General Public License
384 785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	19 along with this program. If not, see <http://www.gnu.org/licenses/>.
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	20 """
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	21
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	22 from logging import debug, info, warning, error
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	23 from twisted.words.protocols.jabber import client, jid
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	24 from twisted.words.protocols.jabber import error as jab_error
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	25 from twisted.words.xish import domish
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	26 import twisted.internet.error
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	27 from twisted.internet import reactor
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	28
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	29 from wokkel import disco, iwokkel
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	30
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	31 from zope.interface import implements
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	32
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	33 import base64
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	34
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	35 try:
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	36 from twisted.words.protocols.xmlstream import XMPPHandler
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	37 except ImportError:
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	38 from wokkel.subprotocols import XMPPHandler
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	39
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	40 MESSAGE = '/message'
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	41 IQ_SET = '/iq[@type="set"]'
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	42 NS_IBB = 'http://jabber.org/protocol/ibb'
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	43 IBB_OPEN = IQ_SET + '/open[@xmlns="' + NS_IBB + '"]'
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	44 IBB_CLOSE = IQ_SET + '/close[@xmlns="' + NS_IBB + '" and @sid="%s"]'
404 6a1c6c41b91b plugin xep-0047: removed gajim workaround as gajim has been fixed Goffi <goffi@goffi.org> parents: 403 diff changeset	45 IBB_IQ_DATA = IQ_SET + '/data[@xmlns="' + NS_IBB + '" and @sid="%s"]'
384 785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	46 IBB_MESSAGE_DATA = MESSAGE + '/data[@xmlns="' + NS_IBB + '" and @sid="%s"]'
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	47 TIMEOUT = 60 #timeout for workflow
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	48 BLOCK_SIZE = 4096
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	49
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	50 PLUGIN_INFO = {
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	51 "name": "In-Band Bytestream Plugin",
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	52 "import_name": "XEP-0047",
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	53 "type": "XEP",
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	54 "protocols": ["XEP-0047"],
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	55 "main": "XEP_0047",
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	56 "handler": "yes",
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	57 "description": _("""Implementation of In-Band Bytestreams""")
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	58 }
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	59
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	60 class XEP_0047():
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	61 NAMESPACE = NS_IBB
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	62
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	63 def __init__(self, host):
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	64 info(_("In-Band Bytestreams plugin initialization"))
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	65 self.host = host
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	66 self.current_stream = {} #key: stream_id, value: data(dict)
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	67
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	68 def getHandler(self, profile):
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	69 return XEP_0047_handler(self)
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	70
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	71 def _timeOut(self, sid):
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	72 """Delecte current_stream id, called after timeout
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	73 @param id: id of self.current_stream"""
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	74 info(_("In-Band Bytestream: TimeOut reached for id %s") % sid);
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	75 self._killId(sid, False, "TIMEOUT")
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	76
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	77 def _killId(self, sid, success=False, failure_reason="UNKNOWN"):
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	78 """Delete an current_stream id, clean up associated observers
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	79 @param sid: id of self.current_stream"""
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	80 if not self.current_stream.has_key(sid):
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	81 warning(_("kill id called on a non existant id"))
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	82 return
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	83 if self.current_stream[sid].has_key("observer_cb"):
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	84 xmlstream = self.current_stream[sid]["xmlstream"]
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	85 xmlstream.removeObserver(self.current_stream[sid]["event_data"], self.current_stream[sid]["observer_cb"])
394 8f3551ceee17 plugin XEP-0065: refactored and misc stuff fixed. Still not finished Goffi <goffi@goffi.org> parents: 391 diff changeset	86 if self.current_stream[sid]['timer'].active():
8f3551ceee17 plugin XEP-0065: refactored and misc stuff fixed. Still not finished Goffi <goffi@goffi.org> parents: 391 diff changeset	87 self.current_stream[sid]['timer'].cancel()
384 785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	88 if self.current_stream[sid].has_key("size"):
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	89 self.host.removeProgressCB(sid)
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	90
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	91 file_obj = self.current_stream[sid]['file_obj']
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	92 success_cb = self.current_stream[sid]['success_cb']
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	93 failure_cb = self.current_stream[sid]['failure_cb']
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	94
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	95 del self.current_stream[sid]
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	96
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	97 if success:
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	98 success_cb(sid, file_obj, NS_IBB)
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	99 else:
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	100 failure_cb(sid, file_obj, NS_IBB, failure_reason)
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	101
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	102 def getProgress(self, sid, data):
391 c34fd9d6242e spelling Goffi <goffi@goffi.org> parents: 389 diff changeset	103 """Fill data with position of current transfer"""
384 785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	104 try:
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	105 file_obj = self.current_stream[sid]["file_obj"]
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	106 data["position"] = str(file_obj.tell())
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	107 data["size"] = str(self.current_stream[sid]["size"])
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	108 except:
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	109 pass
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	110
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	111 def prepareToReceive(self, from_jid, sid, file_obj, size, success_cb, failure_cb):
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	112 """Called when a bytestream is imminent
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	113 @param from_jid: jid of the sender
394 8f3551ceee17 plugin XEP-0065: refactored and misc stuff fixed. Still not finished Goffi <goffi@goffi.org> parents: 391 diff changeset	114 @param sid: Stream id
8f3551ceee17 plugin XEP-0065: refactored and misc stuff fixed. Still not finished Goffi <goffi@goffi.org> parents: 391 diff changeset	115 @param file_obj: File object where data will be written
8f3551ceee17 plugin XEP-0065: refactored and misc stuff fixed. Still not finished Goffi <goffi@goffi.org> parents: 391 diff changeset	116 @param size: full size of the data, or None if unknown
8f3551ceee17 plugin XEP-0065: refactored and misc stuff fixed. Still not finished Goffi <goffi@goffi.org> parents: 391 diff changeset	117 @param success_cb: method to call when successfuly finished
8f3551ceee17 plugin XEP-0065: refactored and misc stuff fixed. Still not finished Goffi <goffi@goffi.org> parents: 391 diff changeset	118 @param failure_cb: method to call when something goes wrong"""
384 785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	119 data = self.current_stream[sid] = {}
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	120 data["from"] = from_jid
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	121 data["file_obj"] = file_obj
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	122 data["seq"] = -1
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	123 if size:
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	124 data["size"] = size
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	125 self.host.registerProgressCB(sid, self.getProgress)
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	126 data["timer"] = reactor.callLater(TIMEOUT, self._timeOut, sid)
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	127 data["success_cb"] = success_cb
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	128 data["failure_cb"] = failure_cb
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	129
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	130 def streamOpening(self, IQ, profile):
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	131 debug(_("IBB stream opening"))
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	132 IQ.handled=True
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	133 profile_jid, xmlstream = self.host.getJidNStream(profile)
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	134 open_elt = IQ.firstChildElement()
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	135 block_size = open_elt.getAttribute('block-size')
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	136 sid = open_elt.getAttribute('sid')
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	137 stanza = open_elt.getAttribute('stanza', 'iq')
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	138 if not sid or not block_size or int(block_size)>65535:
391 c34fd9d6242e spelling Goffi <goffi@goffi.org> parents: 389 diff changeset	139 warning(_("malformed IBB transfer: %s" % IQ['id']))
384 785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	140 self.sendNotAcceptableError(IQ['id'], IQ['from'], xmlstream)
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	141 return
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	142 if not sid in self.current_stream:
391 c34fd9d6242e spelling Goffi <goffi@goffi.org> parents: 389 diff changeset	143 warning(_("Ignoring unexpected IBB transfer: %s" % sid))
384 785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	144 self.sendNotAcceptableError(IQ['id'], IQ['from'], xmlstream)
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	145 return
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	146 if self.current_stream[sid]["from"] != jid.JID(IQ['from']):
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	147 warning(_("sended jid inconsistency (man in the middle attack attempt ?)"))
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	148 self.sendNotAcceptableError(IQ['id'], IQ['from'], xmlstream)
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	149 self._killId(sid, False, "PROTOCOL_ERROR")
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	150 return
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	151
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	152 #at this stage, the session looks ok and will be accepted
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	153
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	154 #we reset the timeout:
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	155 self.current_stream[sid]["timer"].reset(TIMEOUT)
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	156
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	157 #we save the xmlstream, events and observer data to allow observer removal
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	158 self.current_stream[sid]["xmlstream"] = xmlstream
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	159 self.current_stream[sid]["event_data"] = event_data = (IBB_MESSAGE_DATA if stanza=='message' else IBB_IQ_DATA) % sid
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	160 self.current_stream[sid]["observer_cb"] = observer_cb = self.messageData if stanza=='message' else self.iqData
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	161 event_close = IBB_CLOSE % sid
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	162 #we now set the stream observer to look after data packet
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	163 xmlstream.addObserver(event_data, observer_cb, profile = profile)
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	164 xmlstream.addOnetimeObserver(event_close, self.streamClosing, profile = profile)
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	165 #finally, we send the accept stanza
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	166 result = domish.Element(('', 'iq'))
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	167 result['type'] = 'result'
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	168 result['id'] = IQ['id']
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	169 result['to'] = IQ['from']
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	170 xmlstream.send(result)
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	171
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	172 def streamClosing(self, IQ, profile):
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	173 IQ.handled=True
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	174 debug(_("IBB stream closing"))
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	175 data_elt = IQ.firstChildElement()
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	176 sid = data_elt.getAttribute('sid')
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	177 result = domish.Element(('', 'iq'))
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	178 result['type'] = 'result'
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	179 result['id'] = IQ['id']
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	180 result['to'] = IQ['from']
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	181 self.current_stream[sid]["xmlstream"].send(result)
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	182 self._killId(sid, success=True)
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	183
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	184 def iqData(self, IQ, profile):
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	185 IQ.handled=True
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	186 data_elt = IQ.firstChildElement()
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	187
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	188 if self._manageDataElt(data_elt, 'iq', IQ['id'], jid.JID(IQ['from'])):
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	189 #and send a success answer
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	190 result = domish.Element(('', 'iq'))
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	191 result['type'] = 'result'
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	192 result['id'] = IQ['id']
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	193 result['to'] = IQ['from']
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	194 _jid, xmlstream = self.host.getJidNStream(profile)
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	195 xmlstream.send(result)
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	196
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	197 def messageData(self, message_elt, profile):
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	198 data_elt = message_elt.firstChildElement()
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	199 sid = message_elt.getAttribute('id','')
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	200 self._manageDataElt(message_elt, 'message', sid, jid.JID(message_elt['from']))
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	201
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	202 def _manageDataElt(self, data_elt, stanza, sid, stanza_from_jid):
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	203 """Manage the data elelement (check validity and write to the file_obj)
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	204 @param data_elt: "data" domish element
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	205 @return: True if success"""
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	206 sid = data_elt.getAttribute('sid')
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	207 if sid not in self.current_stream:
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	208 error(_("Received data for an unknown session id"))
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	209 return False
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	210 xmlstream = self.current_stream[sid]["xmlstream"]
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	211
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	212 from_jid = self.current_stream[sid]["from"]
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	213 file_obj = self.current_stream[sid]["file_obj"]
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	214
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	215 if stanza_from_jid != from_jid:
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	216 warning(_("sended jid inconsistency (man in the middle attack attempt ?)"))
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	217 if stanza=='iq':
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	218 self.sendNotAcceptableError(sid, from_jid, xmlstream)
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	219 return False
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	220
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	221 self.current_stream[sid]["seq"]+=1
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	222 if int(data_elt.getAttribute("seq",-1)) != self.current_stream[sid]["seq"]:
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	223 warning(_("Sequence error"))
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	224 if stanza=='iq':
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	225 self.sendNotAcceptableError(IQ["id"], from_jid, xmlstream)
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	226 return False
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	227
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	228 #we reset the timeout:
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	229 self.current_stream[sid]["timer"].reset(TIMEOUT)
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	230
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	231 #we can now decode the data
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	232 try:
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	233 file_obj.write(base64.b64decode(str(data_elt)))
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	234 except TypeError:
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	235 #The base64 data is invalid
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	236 warning(_("Invalid base64 data"))
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	237 if stanza=='iq':
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	238 self.sendNotAcceptableError(IQ["id"], from_jid, xmlstream)
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	239 return False
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	240 return True
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	241
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	242 def sendNotAcceptableError(self, iq_id, to_jid, xmlstream):
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	243 """Not acceptable error used when the stream is not expected or something is going wrong
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	244 @param iq_id: IQ id
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	245 @param to_jid: addressee
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	246 @param xmlstream: XML stream to use to send the error"""
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	247 result = domish.Element(('', 'iq'))
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	248 result['type'] = 'result'
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	249 result['id'] = iq_id
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	250 result['to'] = to_jid
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	251 error_el = result.addElement('error')
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	252 error_el['type'] = 'cancel'
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	253 error_el.addElement(('urn:ietf:params:xml:ns:xmpp-stanzas','not-acceptable'))
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	254 xmlstream.send(result)
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	255
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	256 def startStream(self, file_obj, to_jid, sid, length, successCb, failureCb, size = None, profile='@NONE@'):
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	257 """Launch the stream workflow
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	258 @param file_obj: file_obj to send
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	259 @param to_jid: JID of the recipient
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	260 @param sid: Stream session id
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	261 @param length: number of byte to send, or None to send until the end
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	262 @param successCb: method to call when stream successfuly finished
389 6ff50d609b16 plugin XEP-0047: typo Goffi <goffi@goffi.org> parents: 384 diff changeset	263 @param failureCb: method to call when something goes wrong
384 785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	264 @param profile: %(doc_profile)s"""
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	265 if length != None:
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	266 error(_('stream length not managed yet'))
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	267 return;
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	268 profile_jid, xmlstream = self.host.getJidNStream(profile)
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	269 data = self.current_stream[sid] = {}
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	270 data["timer"] = reactor.callLater(TIMEOUT, self._timeOut, sid)
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	271 data["file_obj"] = file_obj
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	272 data["to"] = to_jid
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	273 data["success_cb"] = successCb
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	274 data["failure_cb"] = failureCb
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	275 data["xmlstream"] = xmlstream
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	276 data["block_size"] = BLOCK_SIZE
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	277 if size:
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	278 data["size"] = size
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	279 self.host.registerProgressCB(sid, self.getProgress)
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	280 iq_elt = client.IQ(xmlstream,'set')
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	281 iq_elt['from'] = profile_jid.full()
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	282 iq_elt['to'] = to_jid.full()
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	283 open_elt = iq_elt.addElement('open',NS_IBB)
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	284 open_elt['block-size'] = str(BLOCK_SIZE)
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	285 open_elt['sid'] = sid
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	286 open_elt['stanza'] = 'iq'
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	287 iq_elt.addCallback(self.iqResult, sid, 0, length)
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	288 iq_elt.send()
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	289
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	290 def iqResult(self, sid, seq, length, iq_elt):
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	291 """Called when the result of open iq is received"""
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	292 data = self.current_stream[sid]
394 8f3551ceee17 plugin XEP-0065: refactored and misc stuff fixed. Still not finished Goffi <goffi@goffi.org> parents: 391 diff changeset	293 if iq_elt["type"] == "error":
384 785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	294 warning(_("Transfer failed"))
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	295 self.terminateStream(sid, "IQ_ERROR")
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	296 return
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	297
403 c513328ade9d plugins XEP-0047 and XEP-0065: timout bug fix Goffi <goffi@goffi.org> parents: 394 diff changeset	298 if data['timer'].active():
c513328ade9d plugins XEP-0047 and XEP-0065: timout bug fix Goffi <goffi@goffi.org> parents: 394 diff changeset	299 data['timer'].cancel()
c513328ade9d plugins XEP-0047 and XEP-0065: timout bug fix Goffi <goffi@goffi.org> parents: 394 diff changeset	300
384 785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	301 buffer = data["file_obj"].read(data["block_size"])
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	302 if buffer:
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	303 next_iq_elt = client.IQ(data["xmlstream"],'set')
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	304 next_iq_elt['to'] = data["to"].full()
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	305 data_elt = next_iq_elt.addElement('data', NS_IBB)
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	306 data_elt['seq'] = str(seq)
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	307 data_elt['sid'] = sid
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	308 data_elt.addContent(base64.b64encode(buffer))
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	309 next_iq_elt.addCallback(self.iqResult, sid, seq+1, length)
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	310 next_iq_elt.send()
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	311 else:
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	312 self.terminateStream(sid)
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	313
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	314 def terminateStream(self, sid, failure_reason = None):
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	315 """Terminate the stream session
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	316 @param to_jid: recipient
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	317 @param sid: Session id
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	318 @param file_obj: file object used
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	319 @param xmlstream: XML stream used with this session
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	320 @param progress_cb: True if we have to remove the progress callback
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	321 @param callback: method to call after finishing
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	322 @param failure_reason: reason of the failure, or None if steam was successful"""
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	323 data = self.current_stream[sid]
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	324 iq_elt = client.IQ(data["xmlstream"],'set')
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	325 iq_elt['to'] = data["to"].full()
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	326 close_elt = iq_elt.addElement('close',NS_IBB)
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	327 close_elt['sid'] = sid
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	328 iq_elt.send()
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	329 self.host.removeProgressCB(sid)
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	330 if failure_reason:
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	331 self._killId(sid, False, failure_reason)
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	332 else:
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	333 self._killId(sid, True)
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	334
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	335 class XEP_0047_handler(XMPPHandler):
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	336 implements(iwokkel.IDisco)
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	337
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	338 def __init__(self,parent):
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	339 self.plugin_parent = parent
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	340
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	341 def connectionInitialized(self):
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	342 self.xmlstream.addObserver(IBB_OPEN, self.plugin_parent.streamOpening, profile = self.parent.profile)
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	343
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	344 def getDiscoInfo(self, requestor, target, nodeIdentifier=''):
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	345 return [disco.DiscoFeature(NS_IBB)]
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	346
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	347 def getDiscoItems(self, requestor, target, nodeIdentifier=''):
785420cd63f7 plugins: In-Band Bytestreams (XEP-0047) implementation Goffi <goffi@goffi.org> parents: diff changeset	348 return []

Mercurial > libervia-backend

annotate src/plugins/plugin_xep_0047.py @ 531:3bd8f84f920d