Move ConnectionBytestream to common/protocols.
This commit is contained in:
Stephan Erb
parent
f52532f6f0
commit
d48202d221
|
|
@ -30,7 +30,6 @@
|
||||||
|
|
||||||
import os
|
import os
|
||||||
import base64
|
import base64
|
||||||
import socket
|
|
||||||
import sys
|
import sys
|
||||||
import operator
|
import operator
|
||||||
import hashlib
|
import hashlib
|
||||||
|
|
@ -41,7 +40,6 @@ from time import (altzone, daylight, gmtime, localtime, mktime, strftime,
|
||||||
from calendar import timegm
|
from calendar import timegm
|
||||||
import datetime
|
import datetime
|
||||||
|
|
||||||
import socks5
|
|
||||||
import common.xmpp
|
import common.xmpp
|
||||||
|
|
||||||
from common import helpers
|
from common import helpers
|
||||||
|
|
@ -51,6 +49,7 @@ from common.commands import ConnectionCommands
|
||||||
from common.pubsub import ConnectionPubSub
|
from common.pubsub import ConnectionPubSub
|
||||||
from common.pep import ConnectionPEP
|
from common.pep import ConnectionPEP
|
||||||
from common.protocol.caps import ConnectionCaps
|
from common.protocol.caps import ConnectionCaps
|
||||||
|
from common.protocol.bytestream import ConnectionBytestream
|
||||||
import common.caps_cache as capscache
|
import common.caps_cache as capscache
|
||||||
if gajim.HAVE_FARSIGHT:
|
if gajim.HAVE_FARSIGHT:
|
||||||
from common.jingle import ConnectionJingle
|
from common.jingle import ConnectionJingle
|
||||||
|
|
@ -84,613 +83,6 @@ except Exception:
|
||||||
log.debug(_('Unable to load idle module'))
|
log.debug(_('Unable to load idle module'))
|
||||||
HAS_IDLE = False
|
HAS_IDLE = False
|
||||||
|
|
||||||
class ConnectionBytestream:
|
|
||||||
def __init__(self):
|
|
||||||
self.files_props = {}
|
|
||||||
self.awaiting_xmpp_ping_id = None
|
|
||||||
|
|
||||||
def is_transfer_stopped(self, file_props):
|
|
||||||
if 'error' in file_props and file_props['error'] != 0:
|
|
||||||
return True
|
|
||||||
if 'completed' in file_props and file_props['completed']:
|
|
||||||
return True
|
|
||||||
if 'connected' in file_props and file_props['connected'] == False:
|
|
||||||
return True
|
|
||||||
if 'stopped' not in file_props or not file_props['stopped']:
|
|
||||||
return False
|
|
||||||
return True
|
|
||||||
|
|
||||||
def send_success_connect_reply(self, streamhost):
|
|
||||||
"""
|
|
||||||
Send reply to the initiator of FT that we made a connection
|
|
||||||
"""
|
|
||||||
if not self.connection or self.connected < 2:
|
|
||||||
return
|
|
||||||
if streamhost is None:
|
|
||||||
return None
|
|
||||||
iq = common.xmpp.Iq(to = streamhost['initiator'], typ = 'result',
|
|
||||||
frm = streamhost['target'])
|
|
||||||
iq.setAttr('id', streamhost['id'])
|
|
||||||
query = iq.setTag('query')
|
|
||||||
query.setNamespace(common.xmpp.NS_BYTESTREAM)
|
|
||||||
stream_tag = query.setTag('streamhost-used')
|
|
||||||
stream_tag.setAttr('jid', streamhost['jid'])
|
|
||||||
self.connection.send(iq)
|
|
||||||
|
|
||||||
def remove_transfers_for_contact(self, contact):
|
|
||||||
"""
|
|
||||||
Stop all active transfer for contact
|
|
||||||
"""
|
|
||||||
for file_props in self.files_props.values():
|
|
||||||
if self.is_transfer_stopped(file_props):
|
|
||||||
continue
|
|
||||||
receiver_jid = unicode(file_props['receiver'])
|
|
||||||
if contact.get_full_jid() == receiver_jid:
|
|
||||||
file_props['error'] = -5
|
|
||||||
self.remove_transfer(file_props)
|
|
||||||
self.dispatch('FILE_REQUEST_ERROR', (contact.jid, file_props, ''))
|
|
||||||
sender_jid = unicode(file_props['sender'])
|
|
||||||
if contact.get_full_jid() == sender_jid:
|
|
||||||
file_props['error'] = -3
|
|
||||||
self.remove_transfer(file_props)
|
|
||||||
|
|
||||||
def remove_all_transfers(self):
|
|
||||||
"""
|
|
||||||
Stop and remove all active connections from the socks5 pool
|
|
||||||
"""
|
|
||||||
for file_props in self.files_props.values():
|
|
||||||
self.remove_transfer(file_props, remove_from_list = False)
|
|
||||||
del(self.files_props)
|
|
||||||
self.files_props = {}
|
|
||||||
|
|
||||||
def remove_transfer(self, file_props, remove_from_list = True):
|
|
||||||
if file_props is None:
|
|
||||||
return
|
|
||||||
self.disconnect_transfer(file_props)
|
|
||||||
sid = file_props['sid']
|
|
||||||
gajim.socks5queue.remove_file_props(self.name, sid)
|
|
||||||
|
|
||||||
if remove_from_list:
|
|
||||||
if 'sid' in self.files_props:
|
|
||||||
del(self.files_props['sid'])
|
|
||||||
|
|
||||||
def disconnect_transfer(self, file_props):
|
|
||||||
if file_props is None:
|
|
||||||
return
|
|
||||||
if 'hash' in file_props:
|
|
||||||
gajim.socks5queue.remove_sender(file_props['hash'])
|
|
||||||
|
|
||||||
if 'streamhosts' in file_props:
|
|
||||||
for host in file_props['streamhosts']:
|
|
||||||
if 'idx' in host and host['idx'] > 0:
|
|
||||||
gajim.socks5queue.remove_receiver(host['idx'])
|
|
||||||
gajim.socks5queue.remove_sender(host['idx'])
|
|
||||||
|
|
||||||
def send_socks5_info(self, file_props, fast = True, receiver = None, sender
|
|
||||||
= None):
|
|
||||||
"""
|
|
||||||
Send iq for the present streamhosts and proxies
|
|
||||||
"""
|
|
||||||
if not self.connection or self.connected < 2:
|
|
||||||
return
|
|
||||||
if not isinstance(self.peerhost, tuple):
|
|
||||||
return
|
|
||||||
port = gajim.config.get('file_transfers_port')
|
|
||||||
ft_add_hosts_to_send = gajim.config.get('ft_add_hosts_to_send')
|
|
||||||
cfg_proxies = gajim.config.get_per('accounts', self.name,
|
|
||||||
'file_transfer_proxies')
|
|
||||||
if receiver is None:
|
|
||||||
receiver = file_props['receiver']
|
|
||||||
if sender is None:
|
|
||||||
sender = file_props['sender']
|
|
||||||
proxyhosts = []
|
|
||||||
if fast and cfg_proxies:
|
|
||||||
proxies = [e.strip() for e in cfg_proxies.split(',')]
|
|
||||||
default = gajim.proxy65_manager.get_default_for_name(self.name)
|
|
||||||
if default:
|
|
||||||
# add/move default proxy at top of the others
|
|
||||||
if proxies.__contains__(default):
|
|
||||||
proxies.remove(default)
|
|
||||||
proxies.insert(0, default)
|
|
||||||
|
|
||||||
for proxy in proxies:
|
|
||||||
(host, _port, jid) = gajim.proxy65_manager.get_proxy(proxy, self.name)
|
|
||||||
if host is None:
|
|
||||||
continue
|
|
||||||
host_dict={
|
|
||||||
'state': 0,
|
|
||||||
'target': unicode(receiver),
|
|
||||||
'id': file_props['sid'],
|
|
||||||
'sid': file_props['sid'],
|
|
||||||
'initiator': proxy,
|
|
||||||
'host': host,
|
|
||||||
'port': unicode(_port),
|
|
||||||
'jid': jid
|
|
||||||
}
|
|
||||||
proxyhosts.append(host_dict)
|
|
||||||
sha_str = helpers.get_auth_sha(file_props['sid'], sender,
|
|
||||||
receiver)
|
|
||||||
file_props['sha_str'] = sha_str
|
|
||||||
ft_add_hosts = []
|
|
||||||
if ft_add_hosts_to_send:
|
|
||||||
ft_add_hosts_to_send = [e.strip() for e in ft_add_hosts_to_send.split(',')]
|
|
||||||
for ft_host in ft_add_hosts_to_send:
|
|
||||||
ft_add_hosts.append(ft_host)
|
|
||||||
listener = gajim.socks5queue.start_listener(port,
|
|
||||||
sha_str, self._result_socks5_sid, file_props['sid'])
|
|
||||||
if listener is None:
|
|
||||||
file_props['error'] = -5
|
|
||||||
self.dispatch('FILE_REQUEST_ERROR', (unicode(receiver), file_props,
|
|
||||||
''))
|
|
||||||
self._connect_error(unicode(receiver), file_props['sid'],
|
|
||||||
file_props['sid'], code = 406)
|
|
||||||
return
|
|
||||||
|
|
||||||
iq = common.xmpp.Protocol(name = 'iq', to = unicode(receiver),
|
|
||||||
typ = 'set')
|
|
||||||
file_props['request-id'] = 'id_' + file_props['sid']
|
|
||||||
iq.setID(file_props['request-id'])
|
|
||||||
query = iq.setTag('query')
|
|
||||||
query.setNamespace(common.xmpp.NS_BYTESTREAM)
|
|
||||||
query.setAttr('mode', 'plain')
|
|
||||||
query.setAttr('sid', file_props['sid'])
|
|
||||||
for ft_host in ft_add_hosts:
|
|
||||||
# The streamhost, if set
|
|
||||||
ostreamhost = common.xmpp.Node(tag = 'streamhost')
|
|
||||||
query.addChild(node = ostreamhost)
|
|
||||||
ostreamhost.setAttr('port', unicode(port))
|
|
||||||
ostreamhost.setAttr('host', ft_host)
|
|
||||||
ostreamhost.setAttr('jid', sender)
|
|
||||||
try:
|
|
||||||
# The ip we're connected to server with
|
|
||||||
my_ips = [self.peerhost[0]]
|
|
||||||
# all IPs from local DNS
|
|
||||||
for addr in socket.getaddrinfo(socket.gethostname(), None):
|
|
||||||
if not addr[4][0] in my_ips and not addr[4][0].startswith('127'):
|
|
||||||
my_ips.append(addr[4][0])
|
|
||||||
for ip in my_ips:
|
|
||||||
streamhost = common.xmpp.Node(tag = 'streamhost')
|
|
||||||
query.addChild(node = streamhost)
|
|
||||||
streamhost.setAttr('port', unicode(port))
|
|
||||||
streamhost.setAttr('host', ip)
|
|
||||||
streamhost.setAttr('jid', sender)
|
|
||||||
except socket.gaierror:
|
|
||||||
self.dispatch('ERROR', (_('Wrong host'),
|
|
||||||
_('Invalid local address? :-O')))
|
|
||||||
|
|
||||||
if fast and proxyhosts != [] and gajim.config.get_per('accounts',
|
|
||||||
self.name, 'use_ft_proxies'):
|
|
||||||
file_props['proxy_receiver'] = unicode(receiver)
|
|
||||||
file_props['proxy_sender'] = unicode(sender)
|
|
||||||
file_props['proxyhosts'] = proxyhosts
|
|
||||||
for proxyhost in proxyhosts:
|
|
||||||
streamhost = common.xmpp.Node(tag = 'streamhost')
|
|
||||||
query.addChild(node=streamhost)
|
|
||||||
streamhost.setAttr('port', proxyhost['port'])
|
|
||||||
streamhost.setAttr('host', proxyhost['host'])
|
|
||||||
streamhost.setAttr('jid', proxyhost['jid'])
|
|
||||||
|
|
||||||
# don't add the proxy child tag for streamhosts, which are proxies
|
|
||||||
# proxy = streamhost.setTag('proxy')
|
|
||||||
# proxy.setNamespace(common.xmpp.NS_STREAM)
|
|
||||||
self.connection.send(iq)
|
|
||||||
|
|
||||||
def send_file_rejection(self, file_props, code='403', typ=None):
|
|
||||||
"""
|
|
||||||
Inform sender that we refuse to download the file
|
|
||||||
|
|
||||||
typ is used when code = '400', in this case typ can be 'strean' for
|
|
||||||
invalid stream or 'profile' for invalid profile
|
|
||||||
"""
|
|
||||||
# user response to ConfirmationDialog may come after we've disconneted
|
|
||||||
if not self.connection or self.connected < 2:
|
|
||||||
return
|
|
||||||
iq = common.xmpp.Protocol(name = 'iq',
|
|
||||||
to = unicode(file_props['sender']), typ = 'error')
|
|
||||||
iq.setAttr('id', file_props['request-id'])
|
|
||||||
if code == '400' and typ in ('stream', 'profile'):
|
|
||||||
name = 'bad-request'
|
|
||||||
text = ''
|
|
||||||
else:
|
|
||||||
name = 'forbidden'
|
|
||||||
text = 'Offer Declined'
|
|
||||||
err = common.xmpp.ErrorNode(code=code, typ='cancel', name=name, text=text)
|
|
||||||
if code == '400' and typ in ('stream', 'profile'):
|
|
||||||
if typ == 'stream':
|
|
||||||
err.setTag('no-valid-streams', namespace=common.xmpp.NS_SI)
|
|
||||||
else:
|
|
||||||
err.setTag('bad-profile', namespace=common.xmpp.NS_SI)
|
|
||||||
iq.addChild(node=err)
|
|
||||||
self.connection.send(iq)
|
|
||||||
|
|
||||||
def send_file_approval(self, file_props):
|
|
||||||
"""
|
|
||||||
Send iq, confirming that we want to download the file
|
|
||||||
"""
|
|
||||||
# user response to ConfirmationDialog may come after we've disconneted
|
|
||||||
if not self.connection or self.connected < 2:
|
|
||||||
return
|
|
||||||
iq = common.xmpp.Protocol(name = 'iq',
|
|
||||||
to = unicode(file_props['sender']), typ = 'result')
|
|
||||||
iq.setAttr('id', file_props['request-id'])
|
|
||||||
si = iq.setTag('si')
|
|
||||||
si.setNamespace(common.xmpp.NS_SI)
|
|
||||||
if 'offset' in file_props and file_props['offset']:
|
|
||||||
file_tag = si.setTag('file')
|
|
||||||
file_tag.setNamespace(common.xmpp.NS_FILE)
|
|
||||||
range_tag = file_tag.setTag('range')
|
|
||||||
range_tag.setAttr('offset', file_props['offset'])
|
|
||||||
feature = si.setTag('feature')
|
|
||||||
feature.setNamespace(common.xmpp.NS_FEATURE)
|
|
||||||
_feature = common.xmpp.DataForm(typ='submit')
|
|
||||||
feature.addChild(node=_feature)
|
|
||||||
field = _feature.setField('stream-method')
|
|
||||||
field.delAttr('type')
|
|
||||||
field.setValue(common.xmpp.NS_BYTESTREAM)
|
|
||||||
self.connection.send(iq)
|
|
||||||
|
|
||||||
def _ft_get_our_jid(self):
|
|
||||||
our_jid = gajim.get_jid_from_account(self.name)
|
|
||||||
resource = self.server_resource
|
|
||||||
return our_jid + '/' + resource
|
|
||||||
|
|
||||||
def _ft_get_receiver_jid(self, file_props):
|
|
||||||
return file_props['receiver'].jid + '/' + file_props['receiver'].resource
|
|
||||||
|
|
||||||
def send_file_request(self, file_props):
|
|
||||||
"""
|
|
||||||
Send iq for new FT request
|
|
||||||
"""
|
|
||||||
if not self.connection or self.connected < 2:
|
|
||||||
return
|
|
||||||
file_props['sender'] = self._ft_get_our_jid()
|
|
||||||
fjid = self._ft_get_receiver_jid(file_props)
|
|
||||||
iq = common.xmpp.Protocol(name = 'iq', to = fjid,
|
|
||||||
typ = 'set')
|
|
||||||
iq.setID(file_props['sid'])
|
|
||||||
self.files_props[file_props['sid']] = file_props
|
|
||||||
si = iq.setTag('si')
|
|
||||||
si.setNamespace(common.xmpp.NS_SI)
|
|
||||||
si.setAttr('profile', common.xmpp.NS_FILE)
|
|
||||||
si.setAttr('id', file_props['sid'])
|
|
||||||
file_tag = si.setTag('file')
|
|
||||||
file_tag.setNamespace(common.xmpp.NS_FILE)
|
|
||||||
file_tag.setAttr('name', file_props['name'])
|
|
||||||
file_tag.setAttr('size', file_props['size'])
|
|
||||||
desc = file_tag.setTag('desc')
|
|
||||||
if 'desc' in file_props:
|
|
||||||
desc.setData(file_props['desc'])
|
|
||||||
file_tag.setTag('range')
|
|
||||||
feature = si.setTag('feature')
|
|
||||||
feature.setNamespace(common.xmpp.NS_FEATURE)
|
|
||||||
_feature = common.xmpp.DataForm(typ='form')
|
|
||||||
feature.addChild(node=_feature)
|
|
||||||
field = _feature.setField('stream-method')
|
|
||||||
field.setAttr('type', 'list-single')
|
|
||||||
field.addOption(common.xmpp.NS_BYTESTREAM)
|
|
||||||
self.connection.send(iq)
|
|
||||||
|
|
||||||
def _result_socks5_sid(self, sid, hash_id):
|
|
||||||
"""
|
|
||||||
Store the result of SHA message from auth
|
|
||||||
"""
|
|
||||||
if sid not in self.files_props:
|
|
||||||
return
|
|
||||||
file_props = self.files_props[sid]
|
|
||||||
file_props['hash'] = hash_id
|
|
||||||
return
|
|
||||||
|
|
||||||
def _connect_error(self, to, _id, sid, code=404):
|
|
||||||
"""
|
|
||||||
Called when there is an error establishing BS connection, or when
|
|
||||||
connection is rejected
|
|
||||||
"""
|
|
||||||
if not self.connection or self.connected < 2:
|
|
||||||
return
|
|
||||||
msg_dict = {
|
|
||||||
404: 'Could not connect to given hosts',
|
|
||||||
405: 'Cancel',
|
|
||||||
406: 'Not acceptable',
|
|
||||||
}
|
|
||||||
msg = msg_dict[code]
|
|
||||||
iq = None
|
|
||||||
iq = common.xmpp.Protocol(name = 'iq', to = to,
|
|
||||||
typ = 'error')
|
|
||||||
iq.setAttr('id', _id)
|
|
||||||
err = iq.setTag('error')
|
|
||||||
err.setAttr('code', unicode(code))
|
|
||||||
err.setData(msg)
|
|
||||||
self.connection.send(iq)
|
|
||||||
if code == 404:
|
|
||||||
file_props = gajim.socks5queue.get_file_props(self.name, sid)
|
|
||||||
if file_props is not None:
|
|
||||||
self.disconnect_transfer(file_props)
|
|
||||||
file_props['error'] = -3
|
|
||||||
self.dispatch('FILE_REQUEST_ERROR', (to, file_props, msg))
|
|
||||||
|
|
||||||
def _proxy_auth_ok(self, proxy):
|
|
||||||
"""
|
|
||||||
Called after authentication to proxy server
|
|
||||||
"""
|
|
||||||
if not self.connection or self.connected < 2:
|
|
||||||
return
|
|
||||||
file_props = self.files_props[proxy['sid']]
|
|
||||||
iq = common.xmpp.Protocol(name = 'iq', to = proxy['initiator'],
|
|
||||||
typ = 'set')
|
|
||||||
auth_id = "au_" + proxy['sid']
|
|
||||||
iq.setID(auth_id)
|
|
||||||
query = iq.setTag('query')
|
|
||||||
query.setNamespace(common.xmpp.NS_BYTESTREAM)
|
|
||||||
query.setAttr('sid', proxy['sid'])
|
|
||||||
activate = query.setTag('activate')
|
|
||||||
activate.setData(file_props['proxy_receiver'])
|
|
||||||
iq.setID(auth_id)
|
|
||||||
self.connection.send(iq)
|
|
||||||
|
|
||||||
# register xmpppy handlers for bytestream and FT stanzas
|
|
||||||
def _bytestreamErrorCB(self, con, iq_obj):
|
|
||||||
log.debug('_bytestreamErrorCB')
|
|
||||||
id_ = unicode(iq_obj.getAttr('id'))
|
|
||||||
frm = helpers.get_full_jid_from_iq(iq_obj)
|
|
||||||
query = iq_obj.getTag('query')
|
|
||||||
gajim.proxy65_manager.error_cb(frm, query)
|
|
||||||
jid = helpers.get_jid_from_iq(iq_obj)
|
|
||||||
id_ = id_[3:]
|
|
||||||
if id_ not in self.files_props:
|
|
||||||
return
|
|
||||||
file_props = self.files_props[id_]
|
|
||||||
file_props['error'] = -4
|
|
||||||
self.dispatch('FILE_REQUEST_ERROR', (jid, file_props, ''))
|
|
||||||
raise common.xmpp.NodeProcessed
|
|
||||||
|
|
||||||
def _ft_get_from(self, iq_obj):
|
|
||||||
return helpers.get_full_jid_from_iq(iq_obj)
|
|
||||||
|
|
||||||
def _bytestreamSetCB(self, con, iq_obj):
|
|
||||||
log.debug('_bytestreamSetCB')
|
|
||||||
target = unicode(iq_obj.getAttr('to'))
|
|
||||||
id_ = unicode(iq_obj.getAttr('id'))
|
|
||||||
query = iq_obj.getTag('query')
|
|
||||||
sid = unicode(query.getAttr('sid'))
|
|
||||||
file_props = gajim.socks5queue.get_file_props(
|
|
||||||
self.name, sid)
|
|
||||||
streamhosts=[]
|
|
||||||
for item in query.getChildren():
|
|
||||||
if item.getName() == 'streamhost':
|
|
||||||
host_dict={
|
|
||||||
'state': 0,
|
|
||||||
'target': target,
|
|
||||||
'id': id_,
|
|
||||||
'sid': sid,
|
|
||||||
'initiator': self._ft_get_from(iq_obj)
|
|
||||||
}
|
|
||||||
for attr in item.getAttrs():
|
|
||||||
host_dict[attr] = item.getAttr(attr)
|
|
||||||
streamhosts.append(host_dict)
|
|
||||||
if file_props is None:
|
|
||||||
if sid in self.files_props:
|
|
||||||
file_props = self.files_props[sid]
|
|
||||||
file_props['fast'] = streamhosts
|
|
||||||
if file_props['type'] == 's': # FIXME: remove fast xmlns
|
|
||||||
# only psi do this
|
|
||||||
|
|
||||||
if 'streamhosts' in file_props:
|
|
||||||
file_props['streamhosts'].extend(streamhosts)
|
|
||||||
else:
|
|
||||||
file_props['streamhosts'] = streamhosts
|
|
||||||
if not gajim.socks5queue.get_file_props(self.name, sid):
|
|
||||||
gajim.socks5queue.add_file_props(self.name, file_props)
|
|
||||||
gajim.socks5queue.connect_to_hosts(self.name, sid,
|
|
||||||
self.send_success_connect_reply, None)
|
|
||||||
raise common.xmpp.NodeProcessed
|
|
||||||
|
|
||||||
file_props['streamhosts'] = streamhosts
|
|
||||||
if file_props['type'] == 'r':
|
|
||||||
gajim.socks5queue.connect_to_hosts(self.name, sid,
|
|
||||||
self.send_success_connect_reply, self._connect_error)
|
|
||||||
raise common.xmpp.NodeProcessed
|
|
||||||
|
|
||||||
def _ResultCB(self, con, iq_obj):
|
|
||||||
log.debug('_ResultCB')
|
|
||||||
# if we want to respect xep-0065 we have to check for proxy
|
|
||||||
# activation result in any result iq
|
|
||||||
real_id = unicode(iq_obj.getAttr('id'))
|
|
||||||
if real_id == self.awaiting_xmpp_ping_id:
|
|
||||||
self.awaiting_xmpp_ping_id = None
|
|
||||||
return
|
|
||||||
if not real_id.startswith('au_'):
|
|
||||||
return
|
|
||||||
frm = self._ft_get_from(iq_obj)
|
|
||||||
id_ = real_id[3:]
|
|
||||||
if id_ in self.files_props:
|
|
||||||
file_props = self.files_props[id_]
|
|
||||||
if file_props['streamhost-used']:
|
|
||||||
for host in file_props['proxyhosts']:
|
|
||||||
if host['initiator'] == frm and 'idx' in host:
|
|
||||||
gajim.socks5queue.activate_proxy(host['idx'])
|
|
||||||
raise common.xmpp.NodeProcessed
|
|
||||||
|
|
||||||
def _ft_get_streamhost_jid_attr(self, streamhost):
|
|
||||||
return helpers.parse_jid(streamhost.getAttr('jid'))
|
|
||||||
|
|
||||||
def _bytestreamResultCB(self, con, iq_obj):
|
|
||||||
log.debug('_bytestreamResultCB')
|
|
||||||
frm = self._ft_get_from(iq_obj)
|
|
||||||
real_id = unicode(iq_obj.getAttr('id'))
|
|
||||||
query = iq_obj.getTag('query')
|
|
||||||
gajim.proxy65_manager.resolve_result(frm, query)
|
|
||||||
|
|
||||||
try:
|
|
||||||
streamhost = query.getTag('streamhost-used')
|
|
||||||
except Exception: # this bytestream result is not what we need
|
|
||||||
pass
|
|
||||||
id_ = real_id[3:]
|
|
||||||
if id_ in self.files_props:
|
|
||||||
file_props = self.files_props[id_]
|
|
||||||
else:
|
|
||||||
raise common.xmpp.NodeProcessed
|
|
||||||
if streamhost is None:
|
|
||||||
# proxy approves the activate query
|
|
||||||
if real_id.startswith('au_'):
|
|
||||||
if 'streamhost-used' not in file_props or \
|
|
||||||
file_props['streamhost-used'] is False:
|
|
||||||
raise common.xmpp.NodeProcessed
|
|
||||||
if 'proxyhosts' not in file_props:
|
|
||||||
raise common.xmpp.NodeProcessed
|
|
||||||
for host in file_props['proxyhosts']:
|
|
||||||
if host['initiator'] == frm and \
|
|
||||||
unicode(query.getAttr('sid')) == file_props['sid']:
|
|
||||||
gajim.socks5queue.activate_proxy(host['idx'])
|
|
||||||
break
|
|
||||||
raise common.xmpp.NodeProcessed
|
|
||||||
jid = self._ft_get_streamhost_jid_attr(streamhost)
|
|
||||||
if 'streamhost-used' in file_props and \
|
|
||||||
file_props['streamhost-used'] is True:
|
|
||||||
raise common.xmpp.NodeProcessed
|
|
||||||
|
|
||||||
if real_id.startswith('au_'):
|
|
||||||
if 'stopped' in file and file_props['stopped']:
|
|
||||||
self.remove_transfer(file_props)
|
|
||||||
else:
|
|
||||||
gajim.socks5queue.send_file(file_props, self.name)
|
|
||||||
raise common.xmpp.NodeProcessed
|
|
||||||
|
|
||||||
proxy = None
|
|
||||||
if 'proxyhosts' in file_props:
|
|
||||||
for proxyhost in file_props['proxyhosts']:
|
|
||||||
if proxyhost['jid'] == jid:
|
|
||||||
proxy = proxyhost
|
|
||||||
|
|
||||||
if proxy is not None:
|
|
||||||
file_props['streamhost-used'] = True
|
|
||||||
if 'streamhosts' not in file_props:
|
|
||||||
file_props['streamhosts'] = []
|
|
||||||
file_props['streamhosts'].append(proxy)
|
|
||||||
file_props['is_a_proxy'] = True
|
|
||||||
receiver = socks5.Socks5Receiver(gajim.idlequeue, proxy,
|
|
||||||
file_props['sid'], file_props)
|
|
||||||
gajim.socks5queue.add_receiver(self.name, receiver)
|
|
||||||
proxy['idx'] = receiver.queue_idx
|
|
||||||
gajim.socks5queue.on_success = self._proxy_auth_ok
|
|
||||||
raise common.xmpp.NodeProcessed
|
|
||||||
|
|
||||||
else:
|
|
||||||
if 'stopped' in file_props and file_props['stopped']:
|
|
||||||
self.remove_transfer(file_props)
|
|
||||||
else:
|
|
||||||
gajim.socks5queue.send_file(file_props, self.name)
|
|
||||||
if 'fast' in file_props:
|
|
||||||
fasts = file_props['fast']
|
|
||||||
if len(fasts) > 0:
|
|
||||||
self._connect_error(frm, fasts[0]['id'], file_props['sid'],
|
|
||||||
code = 406)
|
|
||||||
|
|
||||||
raise common.xmpp.NodeProcessed
|
|
||||||
|
|
||||||
def _siResultCB(self, con, iq_obj):
|
|
||||||
log.debug('_siResultCB')
|
|
||||||
id_ = iq_obj.getAttr('id')
|
|
||||||
if id_ not in self.files_props:
|
|
||||||
# no such jid
|
|
||||||
return
|
|
||||||
file_props = self.files_props[id_]
|
|
||||||
if file_props is None:
|
|
||||||
# file properties for jid is none
|
|
||||||
return
|
|
||||||
if 'request-id' in file_props:
|
|
||||||
# we have already sent streamhosts info
|
|
||||||
return
|
|
||||||
file_props['receiver'] = self._ft_get_from(iq_obj)
|
|
||||||
si = iq_obj.getTag('si')
|
|
||||||
file_tag = si.getTag('file')
|
|
||||||
range_tag = None
|
|
||||||
if file_tag:
|
|
||||||
range_tag = file_tag.getTag('range')
|
|
||||||
if range_tag:
|
|
||||||
offset = range_tag.getAttr('offset')
|
|
||||||
if offset:
|
|
||||||
file_props['offset'] = int(offset)
|
|
||||||
length = range_tag.getAttr('length')
|
|
||||||
if length:
|
|
||||||
file_props['length'] = int(length)
|
|
||||||
feature = si.setTag('feature')
|
|
||||||
if feature.getNamespace() != common.xmpp.NS_FEATURE:
|
|
||||||
return
|
|
||||||
form_tag = feature.getTag('x')
|
|
||||||
form = common.xmpp.DataForm(node=form_tag)
|
|
||||||
field = form.getField('stream-method')
|
|
||||||
if field.getValue() != common.xmpp.NS_BYTESTREAM:
|
|
||||||
return
|
|
||||||
self.send_socks5_info(file_props, fast = True)
|
|
||||||
raise common.xmpp.NodeProcessed
|
|
||||||
|
|
||||||
def _siSetCB(self, con, iq_obj):
|
|
||||||
log.debug('_siSetCB')
|
|
||||||
jid = self._ft_get_from(iq_obj)
|
|
||||||
file_props = {'type': 'r'}
|
|
||||||
file_props['sender'] = jid
|
|
||||||
file_props['request-id'] = unicode(iq_obj.getAttr('id'))
|
|
||||||
si = iq_obj.getTag('si')
|
|
||||||
profile = si.getAttr('profile')
|
|
||||||
mime_type = si.getAttr('mime-type')
|
|
||||||
if profile != common.xmpp.NS_FILE:
|
|
||||||
self.send_file_rejection(file_props, code='400', typ='profile')
|
|
||||||
raise common.xmpp.NodeProcessed
|
|
||||||
feature_tag = si.getTag('feature', namespace=common.xmpp.NS_FEATURE)
|
|
||||||
if not feature_tag:
|
|
||||||
return
|
|
||||||
form_tag = feature_tag.getTag('x', namespace=common.xmpp.NS_DATA)
|
|
||||||
if not form_tag:
|
|
||||||
return
|
|
||||||
form = common.dataforms.ExtendForm(node=form_tag)
|
|
||||||
for f in form.iter_fields():
|
|
||||||
if f.var == 'stream-method' and f.type == 'list-single':
|
|
||||||
values = [o[1] for o in f.options]
|
|
||||||
if common.xmpp.NS_BYTESTREAM in values:
|
|
||||||
break
|
|
||||||
else:
|
|
||||||
self.send_file_rejection(file_props, code='400', typ='stream')
|
|
||||||
raise common.xmpp.NodeProcessed
|
|
||||||
file_tag = si.getTag('file')
|
|
||||||
for attribute in file_tag.getAttrs():
|
|
||||||
if attribute in ('name', 'size', 'hash', 'date'):
|
|
||||||
val = file_tag.getAttr(attribute)
|
|
||||||
if val is None:
|
|
||||||
continue
|
|
||||||
file_props[attribute] = val
|
|
||||||
file_desc_tag = file_tag.getTag('desc')
|
|
||||||
if file_desc_tag is not None:
|
|
||||||
file_props['desc'] = file_desc_tag.getData()
|
|
||||||
|
|
||||||
if mime_type is not None:
|
|
||||||
file_props['mime-type'] = mime_type
|
|
||||||
our_jid = gajim.get_jid_from_account(self.name)
|
|
||||||
resource = self.server_resource
|
|
||||||
file_props['receiver'] = self._ft_get_our_jid()
|
|
||||||
file_props['sid'] = unicode(si.getAttr('id'))
|
|
||||||
file_props['transfered_size'] = []
|
|
||||||
gajim.socks5queue.add_file_props(self.name, file_props)
|
|
||||||
self.dispatch('FILE_REQUEST', (jid, file_props))
|
|
||||||
raise common.xmpp.NodeProcessed
|
|
||||||
|
|
||||||
def _siErrorCB(self, con, iq_obj):
|
|
||||||
log.debug('_siErrorCB')
|
|
||||||
si = iq_obj.getTag('si')
|
|
||||||
profile = si.getAttr('profile')
|
|
||||||
if profile != common.xmpp.NS_FILE:
|
|
||||||
return
|
|
||||||
id_ = iq_obj.getAttr('id')
|
|
||||||
if id_ not in self.files_props:
|
|
||||||
# no such jid
|
|
||||||
return
|
|
||||||
file_props = self.files_props[id_]
|
|
||||||
if file_props is None:
|
|
||||||
# file properties for jid is none
|
|
||||||
return
|
|
||||||
jid = self._ft_get_from(iq_obj)
|
|
||||||
file_props['error'] = -3
|
|
||||||
self.dispatch('FILE_REQUEST_ERROR', (jid, file_props, ''))
|
|
||||||
raise common.xmpp.NodeProcessed
|
|
||||||
|
|
||||||
class ConnectionDisco:
|
class ConnectionDisco:
|
||||||
"""
|
"""
|
||||||
|
|
|
||||||
|
|
@ -0,0 +1,662 @@
|
||||||
|
# -*- coding:utf-8 -*-
|
||||||
|
## src/common/connection_handlers.py
|
||||||
|
##
|
||||||
|
## Copyright (C) 2006 Dimitur Kirov <dkirov AT gmail.com>
|
||||||
|
## Junglecow J <junglecow AT gmail.com>
|
||||||
|
## Copyright (C) 2006-2007 Tomasz Melcer <liori AT exroot.org>
|
||||||
|
## Travis Shirk <travis AT pobox.com>
|
||||||
|
## Nikos Kouremenos <kourem AT gmail.com>
|
||||||
|
## Copyright (C) 2006-2008 Yann Leboulanger <asterix AT lagaule.org>
|
||||||
|
## Copyright (C) 2007 Julien Pivotto <roidelapluie AT gmail.com>
|
||||||
|
## Copyright (C) 2007-2008 Brendan Taylor <whateley AT gmail.com>
|
||||||
|
## Jean-Marie Traissard <jim AT lapin.org>
|
||||||
|
## Stephan Erb <steve-e AT h3c.de>
|
||||||
|
## Copyright (C) 2008 Jonathan Schleifer <js-gajim AT webkeks.org>
|
||||||
|
##
|
||||||
|
## This file is part of Gajim.
|
||||||
|
##
|
||||||
|
## Gajim is free software; you can redistribute it and/or modify
|
||||||
|
## it under the terms of the GNU General Public License as published
|
||||||
|
## by the Free Software Foundation; version 3 only.
|
||||||
|
##
|
||||||
|
## Gajim is distributed in the hope that it will be useful,
|
||||||
|
## but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
## GNU General Public License for more details.
|
||||||
|
##
|
||||||
|
## You should have received a copy of the GNU General Public License
|
||||||
|
## along with Gajim. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
##
|
||||||
|
|
||||||
|
import socket
|
||||||
|
|
||||||
|
import common.xmpp
|
||||||
|
from common import gajim
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
class ConnectionBytestream:
|
||||||
|
|
||||||
|
def __init__(self):
|
||||||
|
self.files_props = {}
|
||||||
|
self.awaiting_xmpp_ping_id = None
|
||||||
|
|
||||||
|
def is_transfer_stopped(self, file_props):
|
||||||
|
if 'error' in file_props and file_props['error'] != 0:
|
||||||
|
return True
|
||||||
|
if 'completed' in file_props and file_props['completed']:
|
||||||
|
return True
|
||||||
|
if 'connected' in file_props and file_props['connected'] == False:
|
||||||
|
return True
|
||||||
|
if 'stopped' not in file_props or not file_props['stopped']:
|
||||||
|
return False
|
||||||
|
return True
|
||||||
|
|
||||||
|
def send_success_connect_reply(self, streamhost):
|
||||||
|
"""
|
||||||
|
Send reply to the initiator of FT that we made a connection
|
||||||
|
"""
|
||||||
|
if not self.connection or self.connected < 2:
|
||||||
|
return
|
||||||
|
if streamhost is None:
|
||||||
|
return None
|
||||||
|
iq = common.xmpp.Iq(to = streamhost['initiator'], typ = 'result',
|
||||||
|
frm = streamhost['target'])
|
||||||
|
iq.setAttr('id', streamhost['id'])
|
||||||
|
query = iq.setTag('query')
|
||||||
|
query.setNamespace(common.xmpp.NS_BYTESTREAM)
|
||||||
|
stream_tag = query.setTag('streamhost-used')
|
||||||
|
stream_tag.setAttr('jid', streamhost['jid'])
|
||||||
|
self.connection.send(iq)
|
||||||
|
|
||||||
|
def remove_transfers_for_contact(self, contact):
|
||||||
|
"""
|
||||||
|
Stop all active transfer for contact
|
||||||
|
"""
|
||||||
|
for file_props in self.files_props.values():
|
||||||
|
if self.is_transfer_stopped(file_props):
|
||||||
|
continue
|
||||||
|
receiver_jid = unicode(file_props['receiver'])
|
||||||
|
if contact.get_full_jid() == receiver_jid:
|
||||||
|
file_props['error'] = -5
|
||||||
|
self.remove_transfer(file_props)
|
||||||
|
self.dispatch('FILE_REQUEST_ERROR', (contact.jid, file_props, ''))
|
||||||
|
sender_jid = unicode(file_props['sender'])
|
||||||
|
if contact.get_full_jid() == sender_jid:
|
||||||
|
file_props['error'] = -3
|
||||||
|
self.remove_transfer(file_props)
|
||||||
|
|
||||||
|
def remove_all_transfers(self):
|
||||||
|
"""
|
||||||
|
Stop and remove all active connections from the socks5 pool
|
||||||
|
"""
|
||||||
|
for file_props in self.files_props.values():
|
||||||
|
self.remove_transfer(file_props, remove_from_list = False)
|
||||||
|
del(self.files_props)
|
||||||
|
self.files_props = {}
|
||||||
|
|
||||||
|
def remove_transfer(self, file_props, remove_from_list = True):
|
||||||
|
if file_props is None:
|
||||||
|
return
|
||||||
|
self.disconnect_transfer(file_props)
|
||||||
|
sid = file_props['sid']
|
||||||
|
gajim.socks5queue.remove_file_props(self.name, sid)
|
||||||
|
|
||||||
|
if remove_from_list:
|
||||||
|
if 'sid' in self.files_props:
|
||||||
|
del(self.files_props['sid'])
|
||||||
|
|
||||||
|
def disconnect_transfer(self, file_props):
|
||||||
|
if file_props is None:
|
||||||
|
return
|
||||||
|
if 'hash' in file_props:
|
||||||
|
gajim.socks5queue.remove_sender(file_props['hash'])
|
||||||
|
|
||||||
|
if 'streamhosts' in file_props:
|
||||||
|
for host in file_props['streamhosts']:
|
||||||
|
if 'idx' in host and host['idx'] > 0:
|
||||||
|
gajim.socks5queue.remove_receiver(host['idx'])
|
||||||
|
gajim.socks5queue.remove_sender(host['idx'])
|
||||||
|
|
||||||
|
def send_socks5_info(self, file_props, fast = True, receiver = None, sender
|
||||||
|
= None):
|
||||||
|
"""
|
||||||
|
Send iq for the present streamhosts and proxies
|
||||||
|
"""
|
||||||
|
if not self.connection or self.connected < 2:
|
||||||
|
return
|
||||||
|
if not isinstance(self.peerhost, tuple):
|
||||||
|
return
|
||||||
|
port = gajim.config.get('file_transfers_port')
|
||||||
|
ft_add_hosts_to_send = gajim.config.get('ft_add_hosts_to_send')
|
||||||
|
cfg_proxies = gajim.config.get_per('accounts', self.name,
|
||||||
|
'file_transfer_proxies')
|
||||||
|
if receiver is None:
|
||||||
|
receiver = file_props['receiver']
|
||||||
|
if sender is None:
|
||||||
|
sender = file_props['sender']
|
||||||
|
proxyhosts = []
|
||||||
|
if fast and cfg_proxies:
|
||||||
|
proxies = [e.strip() for e in cfg_proxies.split(',')]
|
||||||
|
default = gajim.proxy65_manager.get_default_for_name(self.name)
|
||||||
|
if default:
|
||||||
|
# add/move default proxy at top of the others
|
||||||
|
if proxies.__contains__(default):
|
||||||
|
proxies.remove(default)
|
||||||
|
proxies.insert(0, default)
|
||||||
|
|
||||||
|
for proxy in proxies:
|
||||||
|
(host, _port, jid) = gajim.proxy65_manager.get_proxy(proxy, self.name)
|
||||||
|
if host is None:
|
||||||
|
continue
|
||||||
|
host_dict={
|
||||||
|
'state': 0,
|
||||||
|
'target': unicode(receiver),
|
||||||
|
'id': file_props['sid'],
|
||||||
|
'sid': file_props['sid'],
|
||||||
|
'initiator': proxy,
|
||||||
|
'host': host,
|
||||||
|
'port': unicode(_port),
|
||||||
|
'jid': jid
|
||||||
|
}
|
||||||
|
proxyhosts.append(host_dict)
|
||||||
|
sha_str = helpers.get_auth_sha(file_props['sid'], sender,
|
||||||
|
receiver)
|
||||||
|
file_props['sha_str'] = sha_str
|
||||||
|
ft_add_hosts = []
|
||||||
|
if ft_add_hosts_to_send:
|
||||||
|
ft_add_hosts_to_send = [e.strip() for e in ft_add_hosts_to_send.split(',')]
|
||||||
|
for ft_host in ft_add_hosts_to_send:
|
||||||
|
ft_add_hosts.append(ft_host)
|
||||||
|
listener = gajim.socks5queue.start_listener(port,
|
||||||
|
sha_str, self._result_socks5_sid, file_props['sid'])
|
||||||
|
if listener is None:
|
||||||
|
file_props['error'] = -5
|
||||||
|
self.dispatch('FILE_REQUEST_ERROR', (unicode(receiver), file_props,
|
||||||
|
''))
|
||||||
|
self._connect_error(unicode(receiver), file_props['sid'],
|
||||||
|
file_props['sid'], code = 406)
|
||||||
|
return
|
||||||
|
|
||||||
|
iq = common.xmpp.Protocol(name = 'iq', to = unicode(receiver),
|
||||||
|
typ = 'set')
|
||||||
|
file_props['request-id'] = 'id_' + file_props['sid']
|
||||||
|
iq.setID(file_props['request-id'])
|
||||||
|
query = iq.setTag('query')
|
||||||
|
query.setNamespace(common.xmpp.NS_BYTESTREAM)
|
||||||
|
query.setAttr('mode', 'plain')
|
||||||
|
query.setAttr('sid', file_props['sid'])
|
||||||
|
for ft_host in ft_add_hosts:
|
||||||
|
# The streamhost, if set
|
||||||
|
ostreamhost = common.xmpp.Node(tag = 'streamhost')
|
||||||
|
query.addChild(node = ostreamhost)
|
||||||
|
ostreamhost.setAttr('port', unicode(port))
|
||||||
|
ostreamhost.setAttr('host', ft_host)
|
||||||
|
ostreamhost.setAttr('jid', sender)
|
||||||
|
try:
|
||||||
|
# The ip we're connected to server with
|
||||||
|
my_ips = [self.peerhost[0]]
|
||||||
|
# all IPs from local DNS
|
||||||
|
for addr in socket.getaddrinfo(socket.gethostname(), None):
|
||||||
|
if not addr[4][0] in my_ips and not addr[4][0].startswith('127'):
|
||||||
|
my_ips.append(addr[4][0])
|
||||||
|
for ip in my_ips:
|
||||||
|
streamhost = common.xmpp.Node(tag = 'streamhost')
|
||||||
|
query.addChild(node = streamhost)
|
||||||
|
streamhost.setAttr('port', unicode(port))
|
||||||
|
streamhost.setAttr('host', ip)
|
||||||
|
streamhost.setAttr('jid', sender)
|
||||||
|
except socket.gaierror:
|
||||||
|
self.dispatch('ERROR', (_('Wrong host'),
|
||||||
|
_('Invalid local address? :-O')))
|
||||||
|
|
||||||
|
if fast and proxyhosts != [] and gajim.config.get_per('accounts',
|
||||||
|
self.name, 'use_ft_proxies'):
|
||||||
|
file_props['proxy_receiver'] = unicode(receiver)
|
||||||
|
file_props['proxy_sender'] = unicode(sender)
|
||||||
|
file_props['proxyhosts'] = proxyhosts
|
||||||
|
for proxyhost in proxyhosts:
|
||||||
|
streamhost = common.xmpp.Node(tag = 'streamhost')
|
||||||
|
query.addChild(node=streamhost)
|
||||||
|
streamhost.setAttr('port', proxyhost['port'])
|
||||||
|
streamhost.setAttr('host', proxyhost['host'])
|
||||||
|
streamhost.setAttr('jid', proxyhost['jid'])
|
||||||
|
|
||||||
|
# don't add the proxy child tag for streamhosts, which are proxies
|
||||||
|
# proxy = streamhost.setTag('proxy')
|
||||||
|
# proxy.setNamespace(common.xmpp.NS_STREAM)
|
||||||
|
self.connection.send(iq)
|
||||||
|
|
||||||
|
def send_file_rejection(self, file_props, code='403', typ=None):
|
||||||
|
"""
|
||||||
|
Inform sender that we refuse to download the file
|
||||||
|
|
||||||
|
typ is used when code = '400', in this case typ can be 'strean' for
|
||||||
|
invalid stream or 'profile' for invalid profile
|
||||||
|
"""
|
||||||
|
# user response to ConfirmationDialog may come after we've disconneted
|
||||||
|
if not self.connection or self.connected < 2:
|
||||||
|
return
|
||||||
|
iq = common.xmpp.Protocol(name = 'iq',
|
||||||
|
to = unicode(file_props['sender']), typ = 'error')
|
||||||
|
iq.setAttr('id', file_props['request-id'])
|
||||||
|
if code == '400' and typ in ('stream', 'profile'):
|
||||||
|
name = 'bad-request'
|
||||||
|
text = ''
|
||||||
|
else:
|
||||||
|
name = 'forbidden'
|
||||||
|
text = 'Offer Declined'
|
||||||
|
err = common.xmpp.ErrorNode(code=code, typ='cancel', name=name, text=text)
|
||||||
|
if code == '400' and typ in ('stream', 'profile'):
|
||||||
|
if typ == 'stream':
|
||||||
|
err.setTag('no-valid-streams', namespace=common.xmpp.NS_SI)
|
||||||
|
else:
|
||||||
|
err.setTag('bad-profile', namespace=common.xmpp.NS_SI)
|
||||||
|
iq.addChild(node=err)
|
||||||
|
self.connection.send(iq)
|
||||||
|
|
||||||
|
def send_file_approval(self, file_props):
|
||||||
|
"""
|
||||||
|
Send iq, confirming that we want to download the file
|
||||||
|
"""
|
||||||
|
# user response to ConfirmationDialog may come after we've disconneted
|
||||||
|
if not self.connection or self.connected < 2:
|
||||||
|
return
|
||||||
|
iq = common.xmpp.Protocol(name = 'iq',
|
||||||
|
to = unicode(file_props['sender']), typ = 'result')
|
||||||
|
iq.setAttr('id', file_props['request-id'])
|
||||||
|
si = iq.setTag('si')
|
||||||
|
si.setNamespace(common.xmpp.NS_SI)
|
||||||
|
if 'offset' in file_props and file_props['offset']:
|
||||||
|
file_tag = si.setTag('file')
|
||||||
|
file_tag.setNamespace(common.xmpp.NS_FILE)
|
||||||
|
range_tag = file_tag.setTag('range')
|
||||||
|
range_tag.setAttr('offset', file_props['offset'])
|
||||||
|
feature = si.setTag('feature')
|
||||||
|
feature.setNamespace(common.xmpp.NS_FEATURE)
|
||||||
|
_feature = common.xmpp.DataForm(typ='submit')
|
||||||
|
feature.addChild(node=_feature)
|
||||||
|
field = _feature.setField('stream-method')
|
||||||
|
field.delAttr('type')
|
||||||
|
field.setValue(common.xmpp.NS_BYTESTREAM)
|
||||||
|
self.connection.send(iq)
|
||||||
|
|
||||||
|
def _ft_get_our_jid(self):
|
||||||
|
our_jid = gajim.get_jid_from_account(self.name)
|
||||||
|
resource = self.server_resource
|
||||||
|
return our_jid + '/' + resource
|
||||||
|
|
||||||
|
def _ft_get_receiver_jid(self, file_props):
|
||||||
|
return file_props['receiver'].jid + '/' + file_props['receiver'].resource
|
||||||
|
|
||||||
|
def send_file_request(self, file_props):
|
||||||
|
"""
|
||||||
|
Send iq for new FT request
|
||||||
|
"""
|
||||||
|
if not self.connection or self.connected < 2:
|
||||||
|
return
|
||||||
|
file_props['sender'] = self._ft_get_our_jid()
|
||||||
|
fjid = self._ft_get_receiver_jid(file_props)
|
||||||
|
iq = common.xmpp.Protocol(name = 'iq', to = fjid,
|
||||||
|
typ = 'set')
|
||||||
|
iq.setID(file_props['sid'])
|
||||||
|
self.files_props[file_props['sid']] = file_props
|
||||||
|
si = iq.setTag('si')
|
||||||
|
si.setNamespace(common.xmpp.NS_SI)
|
||||||
|
si.setAttr('profile', common.xmpp.NS_FILE)
|
||||||
|
si.setAttr('id', file_props['sid'])
|
||||||
|
file_tag = si.setTag('file')
|
||||||
|
file_tag.setNamespace(common.xmpp.NS_FILE)
|
||||||
|
file_tag.setAttr('name', file_props['name'])
|
||||||
|
file_tag.setAttr('size', file_props['size'])
|
||||||
|
desc = file_tag.setTag('desc')
|
||||||
|
if 'desc' in file_props:
|
||||||
|
desc.setData(file_props['desc'])
|
||||||
|
file_tag.setTag('range')
|
||||||
|
feature = si.setTag('feature')
|
||||||
|
feature.setNamespace(common.xmpp.NS_FEATURE)
|
||||||
|
_feature = common.xmpp.DataForm(typ='form')
|
||||||
|
feature.addChild(node=_feature)
|
||||||
|
field = _feature.setField('stream-method')
|
||||||
|
field.setAttr('type', 'list-single')
|
||||||
|
field.addOption(common.xmpp.NS_BYTESTREAM)
|
||||||
|
self.connection.send(iq)
|
||||||
|
|
||||||
|
def _result_socks5_sid(self, sid, hash_id):
|
||||||
|
"""
|
||||||
|
Store the result of SHA message from auth
|
||||||
|
"""
|
||||||
|
if sid not in self.files_props:
|
||||||
|
return
|
||||||
|
file_props = self.files_props[sid]
|
||||||
|
file_props['hash'] = hash_id
|
||||||
|
return
|
||||||
|
|
||||||
|
def _connect_error(self, to, _id, sid, code=404):
|
||||||
|
"""
|
||||||
|
Called when there is an error establishing BS connection, or when
|
||||||
|
connection is rejected
|
||||||
|
"""
|
||||||
|
if not self.connection or self.connected < 2:
|
||||||
|
return
|
||||||
|
msg_dict = {
|
||||||
|
404: 'Could not connect to given hosts',
|
||||||
|
405: 'Cancel',
|
||||||
|
406: 'Not acceptable',
|
||||||
|
}
|
||||||
|
msg = msg_dict[code]
|
||||||
|
iq = None
|
||||||
|
iq = common.xmpp.Protocol(name = 'iq', to = to,
|
||||||
|
typ = 'error')
|
||||||
|
iq.setAttr('id', _id)
|
||||||
|
err = iq.setTag('error')
|
||||||
|
err.setAttr('code', unicode(code))
|
||||||
|
err.setData(msg)
|
||||||
|
self.connection.send(iq)
|
||||||
|
if code == 404:
|
||||||
|
file_props = gajim.socks5queue.get_file_props(self.name, sid)
|
||||||
|
if file_props is not None:
|
||||||
|
self.disconnect_transfer(file_props)
|
||||||
|
file_props['error'] = -3
|
||||||
|
self.dispatch('FILE_REQUEST_ERROR', (to, file_props, msg))
|
||||||
|
|
||||||
|
def _proxy_auth_ok(self, proxy):
|
||||||
|
"""
|
||||||
|
Called after authentication to proxy server
|
||||||
|
"""
|
||||||
|
if not self.connection or self.connected < 2:
|
||||||
|
return
|
||||||
|
file_props = self.files_props[proxy['sid']]
|
||||||
|
iq = common.xmpp.Protocol(name = 'iq', to = proxy['initiator'],
|
||||||
|
typ = 'set')
|
||||||
|
auth_id = "au_" + proxy['sid']
|
||||||
|
iq.setID(auth_id)
|
||||||
|
query = iq.setTag('query')
|
||||||
|
query.setNamespace(common.xmpp.NS_BYTESTREAM)
|
||||||
|
query.setAttr('sid', proxy['sid'])
|
||||||
|
activate = query.setTag('activate')
|
||||||
|
activate.setData(file_props['proxy_receiver'])
|
||||||
|
iq.setID(auth_id)
|
||||||
|
self.connection.send(iq)
|
||||||
|
|
||||||
|
# register xmpppy handlers for bytestream and FT stanzas
|
||||||
|
def _bytestreamErrorCB(self, con, iq_obj):
|
||||||
|
log.debug('_bytestreamErrorCB')
|
||||||
|
id_ = unicode(iq_obj.getAttr('id'))
|
||||||
|
frm = helpers.get_full_jid_from_iq(iq_obj)
|
||||||
|
query = iq_obj.getTag('query')
|
||||||
|
gajim.proxy65_manager.error_cb(frm, query)
|
||||||
|
jid = helpers.get_jid_from_iq(iq_obj)
|
||||||
|
id_ = id_[3:]
|
||||||
|
if id_ not in self.files_props:
|
||||||
|
return
|
||||||
|
file_props = self.files_props[id_]
|
||||||
|
file_props['error'] = -4
|
||||||
|
self.dispatch('FILE_REQUEST_ERROR', (jid, file_props, ''))
|
||||||
|
raise common.xmpp.NodeProcessed
|
||||||
|
|
||||||
|
def _ft_get_from(self, iq_obj):
|
||||||
|
return helpers.get_full_jid_from_iq(iq_obj)
|
||||||
|
|
||||||
|
def _bytestreamSetCB(self, con, iq_obj):
|
||||||
|
log.debug('_bytestreamSetCB')
|
||||||
|
target = unicode(iq_obj.getAttr('to'))
|
||||||
|
id_ = unicode(iq_obj.getAttr('id'))
|
||||||
|
query = iq_obj.getTag('query')
|
||||||
|
sid = unicode(query.getAttr('sid'))
|
||||||
|
file_props = gajim.socks5queue.get_file_props(
|
||||||
|
self.name, sid)
|
||||||
|
streamhosts=[]
|
||||||
|
for item in query.getChildren():
|
||||||
|
if item.getName() == 'streamhost':
|
||||||
|
host_dict={
|
||||||
|
'state': 0,
|
||||||
|
'target': target,
|
||||||
|
'id': id_,
|
||||||
|
'sid': sid,
|
||||||
|
'initiator': self._ft_get_from(iq_obj)
|
||||||
|
}
|
||||||
|
for attr in item.getAttrs():
|
||||||
|
host_dict[attr] = item.getAttr(attr)
|
||||||
|
streamhosts.append(host_dict)
|
||||||
|
if file_props is None:
|
||||||
|
if sid in self.files_props:
|
||||||
|
file_props = self.files_props[sid]
|
||||||
|
file_props['fast'] = streamhosts
|
||||||
|
if file_props['type'] == 's': # FIXME: remove fast xmlns
|
||||||
|
# only psi do this
|
||||||
|
|
||||||
|
if 'streamhosts' in file_props:
|
||||||
|
file_props['streamhosts'].extend(streamhosts)
|
||||||
|
else:
|
||||||
|
file_props['streamhosts'] = streamhosts
|
||||||
|
if not gajim.socks5queue.get_file_props(self.name, sid):
|
||||||
|
gajim.socks5queue.add_file_props(self.name, file_props)
|
||||||
|
gajim.socks5queue.connect_to_hosts(self.name, sid,
|
||||||
|
self.send_success_connect_reply, None)
|
||||||
|
raise common.xmpp.NodeProcessed
|
||||||
|
|
||||||
|
file_props['streamhosts'] = streamhosts
|
||||||
|
if file_props['type'] == 'r':
|
||||||
|
gajim.socks5queue.connect_to_hosts(self.name, sid,
|
||||||
|
self.send_success_connect_reply, self._connect_error)
|
||||||
|
raise common.xmpp.NodeProcessed
|
||||||
|
|
||||||
|
def _ResultCB(self, con, iq_obj):
|
||||||
|
log.debug('_ResultCB')
|
||||||
|
# if we want to respect xep-0065 we have to check for proxy
|
||||||
|
# activation result in any result iq
|
||||||
|
real_id = unicode(iq_obj.getAttr('id'))
|
||||||
|
if real_id == self.awaiting_xmpp_ping_id:
|
||||||
|
self.awaiting_xmpp_ping_id = None
|
||||||
|
return
|
||||||
|
if not real_id.startswith('au_'):
|
||||||
|
return
|
||||||
|
frm = self._ft_get_from(iq_obj)
|
||||||
|
id_ = real_id[3:]
|
||||||
|
if id_ in self.files_props:
|
||||||
|
file_props = self.files_props[id_]
|
||||||
|
if file_props['streamhost-used']:
|
||||||
|
for host in file_props['proxyhosts']:
|
||||||
|
if host['initiator'] == frm and 'idx' in host:
|
||||||
|
gajim.socks5queue.activate_proxy(host['idx'])
|
||||||
|
raise common.xmpp.NodeProcessed
|
||||||
|
|
||||||
|
def _ft_get_streamhost_jid_attr(self, streamhost):
|
||||||
|
return helpers.parse_jid(streamhost.getAttr('jid'))
|
||||||
|
|
||||||
|
def _bytestreamResultCB(self, con, iq_obj):
|
||||||
|
log.debug('_bytestreamResultCB')
|
||||||
|
frm = self._ft_get_from(iq_obj)
|
||||||
|
real_id = unicode(iq_obj.getAttr('id'))
|
||||||
|
query = iq_obj.getTag('query')
|
||||||
|
gajim.proxy65_manager.resolve_result(frm, query)
|
||||||
|
|
||||||
|
try:
|
||||||
|
streamhost = query.getTag('streamhost-used')
|
||||||
|
except Exception: # this bytestream result is not what we need
|
||||||
|
pass
|
||||||
|
id_ = real_id[3:]
|
||||||
|
if id_ in self.files_props:
|
||||||
|
file_props = self.files_props[id_]
|
||||||
|
else:
|
||||||
|
raise common.xmpp.NodeProcessed
|
||||||
|
if streamhost is None:
|
||||||
|
# proxy approves the activate query
|
||||||
|
if real_id.startswith('au_'):
|
||||||
|
if 'streamhost-used' not in file_props or \
|
||||||
|
file_props['streamhost-used'] is False:
|
||||||
|
raise common.xmpp.NodeProcessed
|
||||||
|
if 'proxyhosts' not in file_props:
|
||||||
|
raise common.xmpp.NodeProcessed
|
||||||
|
for host in file_props['proxyhosts']:
|
||||||
|
if host['initiator'] == frm and \
|
||||||
|
unicode(query.getAttr('sid')) == file_props['sid']:
|
||||||
|
gajim.socks5queue.activate_proxy(host['idx'])
|
||||||
|
break
|
||||||
|
raise common.xmpp.NodeProcessed
|
||||||
|
jid = self._ft_get_streamhost_jid_attr(streamhost)
|
||||||
|
if 'streamhost-used' in file_props and \
|
||||||
|
file_props['streamhost-used'] is True:
|
||||||
|
raise common.xmpp.NodeProcessed
|
||||||
|
|
||||||
|
if real_id.startswith('au_'):
|
||||||
|
if 'stopped' in file and file_props['stopped']:
|
||||||
|
self.remove_transfer(file_props)
|
||||||
|
else:
|
||||||
|
gajim.socks5queue.send_file(file_props, self.name)
|
||||||
|
raise common.xmpp.NodeProcessed
|
||||||
|
|
||||||
|
proxy = None
|
||||||
|
if 'proxyhosts' in file_props:
|
||||||
|
for proxyhost in file_props['proxyhosts']:
|
||||||
|
if proxyhost['jid'] == jid:
|
||||||
|
proxy = proxyhost
|
||||||
|
|
||||||
|
if proxy is not None:
|
||||||
|
file_props['streamhost-used'] = True
|
||||||
|
if 'streamhosts' not in file_props:
|
||||||
|
file_props['streamhosts'] = []
|
||||||
|
file_props['streamhosts'].append(proxy)
|
||||||
|
file_props['is_a_proxy'] = True
|
||||||
|
receiver = socks5.Socks5Receiver(gajim.idlequeue, proxy,
|
||||||
|
file_props['sid'], file_props)
|
||||||
|
gajim.socks5queue.add_receiver(self.name, receiver)
|
||||||
|
proxy['idx'] = receiver.queue_idx
|
||||||
|
gajim.socks5queue.on_success = self._proxy_auth_ok
|
||||||
|
raise common.xmpp.NodeProcessed
|
||||||
|
|
||||||
|
else:
|
||||||
|
if 'stopped' in file_props and file_props['stopped']:
|
||||||
|
self.remove_transfer(file_props)
|
||||||
|
else:
|
||||||
|
gajim.socks5queue.send_file(file_props, self.name)
|
||||||
|
if 'fast' in file_props:
|
||||||
|
fasts = file_props['fast']
|
||||||
|
if len(fasts) > 0:
|
||||||
|
self._connect_error(frm, fasts[0]['id'], file_props['sid'],
|
||||||
|
code = 406)
|
||||||
|
|
||||||
|
raise common.xmpp.NodeProcessed
|
||||||
|
|
||||||
|
def _siResultCB(self, con, iq_obj):
|
||||||
|
log.debug('_siResultCB')
|
||||||
|
id_ = iq_obj.getAttr('id')
|
||||||
|
if id_ not in self.files_props:
|
||||||
|
# no such jid
|
||||||
|
return
|
||||||
|
file_props = self.files_props[id_]
|
||||||
|
if file_props is None:
|
||||||
|
# file properties for jid is none
|
||||||
|
return
|
||||||
|
if 'request-id' in file_props:
|
||||||
|
# we have already sent streamhosts info
|
||||||
|
return
|
||||||
|
file_props['receiver'] = self._ft_get_from(iq_obj)
|
||||||
|
si = iq_obj.getTag('si')
|
||||||
|
file_tag = si.getTag('file')
|
||||||
|
range_tag = None
|
||||||
|
if file_tag:
|
||||||
|
range_tag = file_tag.getTag('range')
|
||||||
|
if range_tag:
|
||||||
|
offset = range_tag.getAttr('offset')
|
||||||
|
if offset:
|
||||||
|
file_props['offset'] = int(offset)
|
||||||
|
length = range_tag.getAttr('length')
|
||||||
|
if length:
|
||||||
|
file_props['length'] = int(length)
|
||||||
|
feature = si.setTag('feature')
|
||||||
|
if feature.getNamespace() != common.xmpp.NS_FEATURE:
|
||||||
|
return
|
||||||
|
form_tag = feature.getTag('x')
|
||||||
|
form = common.xmpp.DataForm(node=form_tag)
|
||||||
|
field = form.getField('stream-method')
|
||||||
|
if field.getValue() != common.xmpp.NS_BYTESTREAM:
|
||||||
|
return
|
||||||
|
self.send_socks5_info(file_props, fast = True)
|
||||||
|
raise common.xmpp.NodeProcessed
|
||||||
|
|
||||||
|
def _siSetCB(self, con, iq_obj):
|
||||||
|
log.debug('_siSetCB')
|
||||||
|
jid = self._ft_get_from(iq_obj)
|
||||||
|
file_props = {'type': 'r'}
|
||||||
|
file_props['sender'] = jid
|
||||||
|
file_props['request-id'] = unicode(iq_obj.getAttr('id'))
|
||||||
|
si = iq_obj.getTag('si')
|
||||||
|
profile = si.getAttr('profile')
|
||||||
|
mime_type = si.getAttr('mime-type')
|
||||||
|
if profile != common.xmpp.NS_FILE:
|
||||||
|
self.send_file_rejection(file_props, code='400', typ='profile')
|
||||||
|
raise common.xmpp.NodeProcessed
|
||||||
|
feature_tag = si.getTag('feature', namespace=common.xmpp.NS_FEATURE)
|
||||||
|
if not feature_tag:
|
||||||
|
return
|
||||||
|
form_tag = feature_tag.getTag('x', namespace=common.xmpp.NS_DATA)
|
||||||
|
if not form_tag:
|
||||||
|
return
|
||||||
|
form = common.dataforms.ExtendForm(node=form_tag)
|
||||||
|
for f in form.iter_fields():
|
||||||
|
if f.var == 'stream-method' and f.type == 'list-single':
|
||||||
|
values = [o[1] for o in f.options]
|
||||||
|
if common.xmpp.NS_BYTESTREAM in values:
|
||||||
|
break
|
||||||
|
else:
|
||||||
|
self.send_file_rejection(file_props, code='400', typ='stream')
|
||||||
|
raise common.xmpp.NodeProcessed
|
||||||
|
file_tag = si.getTag('file')
|
||||||
|
for attribute in file_tag.getAttrs():
|
||||||
|
if attribute in ('name', 'size', 'hash', 'date'):
|
||||||
|
val = file_tag.getAttr(attribute)
|
||||||
|
if val is None:
|
||||||
|
continue
|
||||||
|
file_props[attribute] = val
|
||||||
|
file_desc_tag = file_tag.getTag('desc')
|
||||||
|
if file_desc_tag is not None:
|
||||||
|
file_props['desc'] = file_desc_tag.getData()
|
||||||
|
|
||||||
|
if mime_type is not None:
|
||||||
|
file_props['mime-type'] = mime_type
|
||||||
|
our_jid = gajim.get_jid_from_account(self.name)
|
||||||
|
resource = self.server_resource
|
||||||
|
file_props['receiver'] = self._ft_get_our_jid()
|
||||||
|
file_props['sid'] = unicode(si.getAttr('id'))
|
||||||
|
file_props['transfered_size'] = []
|
||||||
|
gajim.socks5queue.add_file_props(self.name, file_props)
|
||||||
|
self.dispatch('FILE_REQUEST', (jid, file_props))
|
||||||
|
raise common.xmpp.NodeProcessed
|
||||||
|
|
||||||
|
def _siErrorCB(self, con, iq_obj):
|
||||||
|
log.debug('_siErrorCB')
|
||||||
|
si = iq_obj.getTag('si')
|
||||||
|
profile = si.getAttr('profile')
|
||||||
|
if profile != common.xmpp.NS_FILE:
|
||||||
|
return
|
||||||
|
id_ = iq_obj.getAttr('id')
|
||||||
|
if id_ not in self.files_props:
|
||||||
|
# no such jid
|
||||||
|
return
|
||||||
|
file_props = self.files_props[id_]
|
||||||
|
if file_props is None:
|
||||||
|
# file properties for jid is none
|
||||||
|
return
|
||||||
|
jid = self._ft_get_from(iq_obj)
|
||||||
|
file_props['error'] = -3
|
||||||
|
self.dispatch('FILE_REQUEST_ERROR', (jid, file_props, ''))
|
||||||
|
raise common.xmpp.NodeProcessed
|
||||||
|
|
||||||
|
|
||||||
|
class ConnectionBytestreamZeroconf(ConnectionBytestream):
|
||||||
|
|
||||||
|
def _ft_get_from(self, iq_obj):
|
||||||
|
return unicode(iq_obj.getFrom())
|
||||||
|
|
||||||
|
def _ft_get_our_jid(self):
|
||||||
|
return gajim.get_jid_from_account(self.name)
|
||||||
|
|
||||||
|
def _ft_get_receiver_jid(self, file_props):
|
||||||
|
return file_props['receiver'].jid
|
||||||
|
|
||||||
|
def _ft_get_streamhost_jid_attr(self, streamhost):
|
||||||
|
return streamhost.getAttr('jid')
|
||||||
|
|
||||||
|
# vim: se ts=3:
|
||||||
|
|
@ -28,7 +28,6 @@ import socket
|
||||||
|
|
||||||
from calendar import timegm
|
from calendar import timegm
|
||||||
|
|
||||||
from common import socks5
|
|
||||||
import common.xmpp
|
import common.xmpp
|
||||||
|
|
||||||
from common import helpers
|
from common import helpers
|
||||||
|
|
@ -36,6 +35,7 @@ from common import gajim
|
||||||
from common.zeroconf import zeroconf
|
from common.zeroconf import zeroconf
|
||||||
from common.commands import ConnectionCommands
|
from common.commands import ConnectionCommands
|
||||||
from common.pep import ConnectionPEP
|
from common.pep import ConnectionPEP
|
||||||
|
from common.protocol.bytestream import ConnectionBytestreamZeroconf
|
||||||
|
|
||||||
import logging
|
import logging
|
||||||
log = logging.getLogger('gajim.c.z.connection_handlers_zeroconf')
|
log = logging.getLogger('gajim.c.z.connection_handlers_zeroconf')
|
||||||
|
|
@ -69,25 +69,12 @@ class ConnectionVcard(connection_handlers.ConnectionVcard):
|
||||||
def send_vcard(self, vcard):
|
def send_vcard(self, vcard):
|
||||||
pass
|
pass
|
||||||
|
|
||||||
class ConnectionBytestream(connection_handlers.ConnectionBytestream):
|
|
||||||
def _ft_get_from(self, iq_obj):
|
|
||||||
return unicode(iq_obj.getFrom())
|
|
||||||
|
|
||||||
def _ft_get_our_jid(self):
|
class ConnectionHandlersZeroconf(ConnectionVcard, ConnectionBytestreamZeroconf,
|
||||||
return gajim.get_jid_from_account(self.name)
|
|
||||||
|
|
||||||
def _ft_get_receiver_jid(self, file_props):
|
|
||||||
return file_props['receiver'].jid
|
|
||||||
|
|
||||||
def _ft_get_streamhost_jid_attr(self, streamhost):
|
|
||||||
return streamhost.getAttr('jid')
|
|
||||||
|
|
||||||
|
|
||||||
class ConnectionHandlersZeroconf(ConnectionVcard, ConnectionBytestream,
|
|
||||||
ConnectionCommands, ConnectionPEP, connection_handlers.ConnectionHandlersBase):
|
ConnectionCommands, ConnectionPEP, connection_handlers.ConnectionHandlersBase):
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
ConnectionVcard.__init__(self)
|
ConnectionVcard.__init__(self)
|
||||||
ConnectionBytestream.__init__(self)
|
ConnectionBytestreamZeroconf.__init__(self)
|
||||||
ConnectionCommands.__init__(self)
|
ConnectionCommands.__init__(self)
|
||||||
connection_handlers.ConnectionHandlersBase.__init__(self)
|
connection_handlers.ConnectionHandlersBase.__init__(self)
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue