jshiffer/slixmpp
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Renamed the _handle_on_recv to the on_recv method.

Browse Source 
Renamed requester_thread and target_thread to proxy. The send method is now simpler.
This commit is contained in:
Sandro Munda 2012-06-04 08:27:41 +02:00
parent b52d2768b0
commit 44ee0633f2
1 changed files with 21 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
sleekxmpp/plugins/xep_0065/proxy.py
View File

@ -123,12 +123,12 @@ class xep_0065(base_plugin):
# Next the Target attempts to open a standard TCP socket on # Next the Target attempts to open a standard TCP socket on
# the network address of the Proxy. # the network address of the Proxy.
self.target_thread = Proxy(sid, requester, target, self.proxy_host, self.proxy_thread = Proxy(sid, requester, target, self.proxy_host,
self.proxy_port, self._handle_on_recv) self.proxy_port, self.on_recv)
self.target_thread.start() self.proxy_thread.start()
# Wait until the proxy is connected # Wait until the proxy is connected
self.target_thread.connected.wait() self.proxy_thread.connected.wait()
# Replies to the incoming iq with a streamhost-used stanza. # Replies to the incoming iq with a streamhost-used stanza.
res_iq = iq.reply() res_iq = iq.reply()
@ -148,13 +148,13 @@ class xep_0065(base_plugin):
# The Requester will establish a connection to the SOCKS5 # The Requester will establish a connection to the SOCKS5
# proxy in the same way the Target did. # proxy in the same way the Target did.
self.requester_thread = Proxy(iq['q']['sid'], requester, target, self.proxy_thread = Proxy(iq['q']['sid'], requester, target,
self.proxy_host, self.proxy_port, self.proxy_host, self.proxy_port,
self._handle_on_recv) self.on_recv)
self.requester_thread.start() self.proxy_thread.start()
# Wait until the proxy is connected # Wait until the proxy is connected
self.requester_thread.connected.wait() self.proxy_thread.connected.wait()
# Requester sends IQ-set to StreamHost requesting that # Requester sends IQ-set to StreamHost requesting that
# StreamHost activate the bytestream associated with the # StreamHost activate the bytestream associated with the
@ -180,19 +180,21 @@ class xep_0065(base_plugin):
msg : The message data. msg : The message data.
""" """
if hasattr(self, 'requester_thread'): self.proxy_thread.send(msg)
self.requester_thread.send(msg)
elif hasattr(self, 'target_thread'):
self.target_thread.send(msg)
def _handle_on_recv(self, data): def on_recv(self, data):
""" A default callback when socket are receiving data. """ A default behavior when socket are receiving data.
This method should be overriden.
""" """
log.debug('Received: %s' % data) log.debug('Received data: %s' % data)
class Proxy(Thread): class Proxy(Thread):
""" Establishes in a thread a connection between the client and
the server-side Socks5 proxy.
"""
def __init__(self, sid, requester, target, proxy, proxy_port, def __init__(self, sid, requester, target, proxy, proxy_port,
on_recv): on_recv):
@ -210,6 +212,9 @@ class Proxy(Thread):
# Initializes the thread. # Initializes the thread.
Thread.__init__(self) Thread.__init__(self)
# This thread is a daemon thread.
self.daemon = True
# Because the xep_0065 plugin uses the proxy_port as string, # Because the xep_0065 plugin uses the proxy_port as string,
# the Proxy class accepts the proxy_port argument as a string # the Proxy class accepts the proxy_port argument as a string
# or an integer. Here, we force to use the port as an integer. # or an integer. Here, we force to use the port as an integer.
@ -266,7 +271,7 @@ class Proxy(Thread):
self.on_recv(data) self.on_recv(data)
def recv_size(self, the_socket): def recv_size(self, the_socket):
#data length is packed into 4 bytes # Data length is packed into 4 bytes.
total_len = 0 total_len = 0
total_data = [] total_data = []
size = sys.maxint size = sys.maxint