Name of filetransfer content is now random to be able to have 2 transfer in the same session. send and handle content-add in filetranfer
This commit is contained in:
parent
f03cdbbebf
commit
286d788da0
|
@ -48,7 +48,7 @@ class ConnectionJingle(object):
|
|||
|
||||
def __init__(self):
|
||||
# dictionary: sessionid => JingleSession object
|
||||
self.__sessions = {}
|
||||
self.__sessions__ = {}
|
||||
|
||||
# dictionary: (jid, iq stanza id) => JingleSession object,
|
||||
# one time callbacks
|
||||
|
@ -58,12 +58,12 @@ class ConnectionJingle(object):
|
|||
"""
|
||||
Remove a jingle session from a jingle stanza dispatcher
|
||||
"""
|
||||
if sid in self.__sessions:
|
||||
if sid in self.__sessions__:
|
||||
#FIXME: Move this elsewhere?
|
||||
for content in self.__sessions[sid].contents.values():
|
||||
for content in self.__sessions__[sid].contents.values():
|
||||
content.destroy()
|
||||
self.__sessions[sid].callbacks = []
|
||||
del self.__sessions[sid]
|
||||
self.__sessions__[sid].callbacks = []
|
||||
del self.__sessions__[sid]
|
||||
|
||||
def _JingleCB(self, con, stanza):
|
||||
"""
|
||||
|
@ -91,23 +91,23 @@ class ConnectionJingle(object):
|
|||
sid = jingle.getAttr('sid')
|
||||
else:
|
||||
sid = None
|
||||
for sesn in self.__sessions.values():
|
||||
for sesn in self.__sessions__.values():
|
||||
if id in sesn.iq_ids:
|
||||
sesn.on_stanza(stanza)
|
||||
return
|
||||
|
||||
# do we need to create a new jingle object
|
||||
if sid not in self.__sessions:
|
||||
if sid not in self.__sessions__:
|
||||
#TODO: tie-breaking and other things...
|
||||
newjingle = JingleSession(con=self, weinitiate=False, jid=jid,
|
||||
iq_id = id, sid=sid)
|
||||
self.__sessions[sid] = newjingle
|
||||
self.__sessions__[sid] = newjingle
|
||||
|
||||
# we already have such session in dispatcher...
|
||||
self.__sessions[sid].collect_iq_id(id)
|
||||
self.__sessions[sid].on_stanza(stanza)
|
||||
self.__sessions__[sid].collect_iq_id(id)
|
||||
self.__sessions__[sid].on_stanza(stanza)
|
||||
# Delete invalid/unneeded sessions
|
||||
if sid in self.__sessions and self.__sessions[sid].state == JingleStates.ended:
|
||||
if sid in self.__sessions__ and self.__sessions__[sid].state == JingleStates.ended:
|
||||
self.delete_jingle_session(sid)
|
||||
|
||||
raise xmpp.NodeProcessed
|
||||
|
@ -120,7 +120,7 @@ class ConnectionJingle(object):
|
|||
jingle.add_content('voice', JingleAudio(jingle))
|
||||
else:
|
||||
jingle = JingleSession(self, weinitiate=True, jid=jid)
|
||||
self.__sessions[jingle.sid] = jingle
|
||||
self.__sessions__[jingle.sid] = jingle
|
||||
jingle.add_content('voice', JingleAudio(jingle))
|
||||
jingle.start_session()
|
||||
return jingle.sid
|
||||
|
@ -133,7 +133,7 @@ class ConnectionJingle(object):
|
|||
jingle.add_content('video', JingleVideo(jingle))
|
||||
else:
|
||||
jingle = JingleSession(self, weinitiate=True, jid=jid)
|
||||
self.__sessions[jingle.sid] = jingle
|
||||
self.__sessions__[jingle.sid] = jingle
|
||||
jingle.add_content('video', JingleVideo(jingle))
|
||||
jingle.start_session()
|
||||
return jingle.sid
|
||||
|
@ -150,24 +150,25 @@ class ConnectionJingle(object):
|
|||
file_props['sid'] = jingle.sid
|
||||
c = JingleFileTransfer(jingle, file_props=file_props,
|
||||
use_security=use_security)
|
||||
jingle.add_content('file', c)
|
||||
jingle.add_content('file' + helpers.get_random_string_16(), c)
|
||||
jingle.on_session_state_changed(c)
|
||||
else:
|
||||
jingle = JingleSession(self, weinitiate=True, jid=jid)
|
||||
self.__sessions[jingle.sid] = jingle
|
||||
self.__sessions__[jingle.sid] = jingle
|
||||
file_props['sid'] = jingle.sid
|
||||
c = JingleFileTransfer(jingle, file_props=file_props,
|
||||
use_security=use_security)
|
||||
jingle.add_content('file', c)
|
||||
jingle.add_content('file' + helpers.get_random_string_16(), c)
|
||||
jingle.start_session()
|
||||
return c.transport.sid
|
||||
|
||||
|
||||
def iter_jingle_sessions(self, jid, sid=None, media=None):
|
||||
if sid:
|
||||
return (session for session in self.__sessions.values() if session.sid == sid)
|
||||
sessions = (session for session in self.__sessions.values() if session.peerjid == jid)
|
||||
return (session for session in self.__sessions__.values() if session.sid == sid)
|
||||
sessions = (session for session in self.__sessions__.values() if session.peerjid == jid)
|
||||
if media:
|
||||
if media not in ('audio', 'video'):
|
||||
if media not in ('audio', 'video', 'file'):
|
||||
return tuple()
|
||||
else:
|
||||
return (session for session in sessions if session.get_content(media))
|
||||
|
@ -177,14 +178,14 @@ class ConnectionJingle(object):
|
|||
|
||||
def get_jingle_session(self, jid, sid=None, media=None):
|
||||
if sid:
|
||||
if sid in self.__sessions:
|
||||
return self.__sessions[sid]
|
||||
if sid in self.__sessions__:
|
||||
return self.__sessions__[sid]
|
||||
else:
|
||||
return None
|
||||
elif media:
|
||||
if media not in ('audio', 'video'):
|
||||
if media not in ('audio', 'video', 'file'):
|
||||
return None
|
||||
for session in self.__sessions.values():
|
||||
for session in self.__sessions__.values():
|
||||
if session.peerjid == jid and session.get_content(media):
|
||||
return session
|
||||
|
||||
|
|
|
@ -44,6 +44,7 @@ class JingleFileTransfer(JingleContent):
|
|||
|
||||
# events we might be interested in
|
||||
self.callbacks['session-initiate'] += [self.__on_session_initiate]
|
||||
self.callbacks['content-add'] += [self.__on_session_initiate]
|
||||
self.callbacks['session-accept'] += [self.__on_session_accept]
|
||||
self.callbacks['session-terminate'] += [self.__on_session_terminate]
|
||||
self.callbacks['transport-accept'] += [self.__on_transport_accept]
|
||||
|
@ -249,7 +250,7 @@ class JingleFileTransfer(JingleContent):
|
|||
|
||||
content = xmpp.Node('content')
|
||||
content.setAttr('creator', 'initiator')
|
||||
content.setAttr('name', 'file')
|
||||
content.setAttr('name', self.name)
|
||||
|
||||
transport = xmpp.Node('transport')
|
||||
transport.setNamespace(xmpp.NS_JINGLE_BYTESTREAM)
|
||||
|
|
|
@ -340,7 +340,7 @@ class JingleSession(object):
|
|||
break
|
||||
elif child.getNamespace() == xmpp.NS_STANZAS:
|
||||
error_name = child.getName()
|
||||
self.__dispatch_error(error_name, text, error.getAttribute('type'))
|
||||
self.__dispatch_error(error_name, text, error.getAttr('type'))
|
||||
# FIXME: Not sure when we would want to do that...
|
||||
|
||||
def __on_transport_replace(self, stanza, jingle, error, action):
|
||||
|
@ -673,7 +673,8 @@ class JingleSession(object):
|
|||
stanza, jingle = self.__make_jingle('content-add')
|
||||
self.__append_content(jingle, content)
|
||||
self.__broadcast(stanza, jingle, None, 'content-add-sent')
|
||||
self.connection.connection.send(stanza)
|
||||
id_ = self.connection.connection.send(stanza)
|
||||
self.collect_iq_id(id_)
|
||||
|
||||
def __content_accept(self, content):
|
||||
# TODO: test
|
||||
|
@ -681,7 +682,8 @@ class JingleSession(object):
|
|||
stanza, jingle = self.__make_jingle('content-accept')
|
||||
self.__append_content(jingle, content)
|
||||
self.__broadcast(stanza, jingle, None, 'content-accept-sent')
|
||||
self.connection.connection.send(stanza)
|
||||
id_ = self.connection.connection.send(stanza)
|
||||
self.collect_iq_id(id_)
|
||||
|
||||
def __content_reject(self, content):
|
||||
assert self.state != JingleStates.ended
|
||||
|
|
|
@ -224,6 +224,13 @@ class JingleTransportSocks5(JingleTransport):
|
|||
proxy_cand.append(c)
|
||||
self.candidates += proxy_cand
|
||||
|
||||
def get_content(self):
|
||||
sesn = self.connection.get_jingle_session(self.ourjid,
|
||||
self.file_props['session-sid'])
|
||||
for content in sesn.contents.values():
|
||||
if content.transport == self:
|
||||
return content
|
||||
|
||||
def _on_proxy_auth_ok(self, proxy):
|
||||
log.info('proxy auth ok for ' + str(proxy))
|
||||
# send activate request to proxy, send activated confirmation to peer
|
||||
|
@ -242,15 +249,15 @@ class JingleTransportSocks5(JingleTransport):
|
|||
|
||||
content = xmpp.Node('content')
|
||||
content.setAttr('creator', 'initiator')
|
||||
content.setAttr('name', 'file')
|
||||
c = self.get_content()
|
||||
content.setAttr('name', c.name)
|
||||
transport = xmpp.Node('transport')
|
||||
transport.setNamespace(xmpp.NS_JINGLE_BYTESTREAM)
|
||||
activated = xmpp.Node('activated')
|
||||
cid = None
|
||||
for host in self.candidates:
|
||||
if host['host'] == proxy['host'] and \
|
||||
host['jid'] == proxy['jid'] and \
|
||||
host['port'] == proxy['port']:
|
||||
if host['host'] == proxy['host'] and host['jid'] == proxy['jid'] \
|
||||
and host['port'] == proxy['port']:
|
||||
cid = host['candidate_id']
|
||||
break
|
||||
if cid is None:
|
||||
|
|
|
@ -45,6 +45,8 @@ if PYOPENSSL_PRESENT:
|
|||
from OpenSSL import SSL
|
||||
from OpenSSL.SSL import Context
|
||||
from OpenSSL import crypto
|
||||
TYPE_RSA = crypto.TYPE_RSA
|
||||
TYPE_DSA = crypto.TYPE_DSA
|
||||
|
||||
SELF_SIGNED_CERTIFICATE = 'localcert'
|
||||
|
||||
|
@ -155,9 +157,6 @@ def send_cert_request(con, to_jid):
|
|||
|
||||
# the following code is partly due to pyopenssl examples
|
||||
|
||||
TYPE_RSA = crypto.TYPE_RSA
|
||||
TYPE_DSA = crypto.TYPE_DSA
|
||||
|
||||
def createKeyPair(type, bits):
|
||||
"""
|
||||
Create a public/private key pair.
|
||||
|
|
|
@ -148,6 +148,7 @@ class ConnectionBytestream:
|
|||
jingle_xtls.key_exchange_pend(id_, session)
|
||||
return
|
||||
session.approve_session()
|
||||
|
||||
session.approve_content('file')
|
||||
return
|
||||
|
||||
|
|
|
@ -914,7 +914,7 @@ class Socks5Listener(IdleObject):
|
|||
# try the different possibilities (ipv6, ipv4, etc.)
|
||||
try:
|
||||
self._serv = socket.socket(*ai[:3])
|
||||
if not self.fingerprint is None:
|
||||
if self.fingerprint is not None:
|
||||
self._serv = OpenSSL.SSL.Connection(
|
||||
jingle_xtls.get_context('server'), self._serv)
|
||||
except socket.error, e:
|
||||
|
|
Loading…
Reference in New Issue