summaryrefslogtreecommitdiff
path: root/sleekxmpp/thirdparty/socks.py
diff options
context:
space:
mode:
authorMike Taylor <bear42@gmail.com>2015-04-11 19:49:43 -0400
committerMike Taylor <bear42@gmail.com>2015-04-11 19:49:43 -0400
commitcc145d20b077442b7d4118d4d2f4c27e1b0faf0c (patch)
treede848c1a7f60d42f427a5678aeab798e91ed386a /sleekxmpp/thirdparty/socks.py
parent881d9040c48417bc2e21922fead9e4213b2fab6d (diff)
parente94a73553d808f13708490024f73d4a5ca5a8bb3 (diff)
downloadslixmpp-cc145d20b077442b7d4118d4d2f4c27e1b0faf0c.tar.gz
slixmpp-cc145d20b077442b7d4118d4d2f4c27e1b0faf0c.tar.bz2
slixmpp-cc145d20b077442b7d4118d4d2f4c27e1b0faf0c.tar.xz
slixmpp-cc145d20b077442b7d4118d4d2f4c27e1b0faf0c.zip
Merge pull request #297 from keith-gray-powereng/develop
Fixed a unicode error in xep_0065 on Python 3
Diffstat (limited to 'sleekxmpp/thirdparty/socks.py')
-rw-r--r--sleekxmpp/thirdparty/socks.py13
1 files changed, 11 insertions, 2 deletions
diff --git a/sleekxmpp/thirdparty/socks.py b/sleekxmpp/thirdparty/socks.py
index 9239a7b9..f32eb21e 100644
--- a/sleekxmpp/thirdparty/socks.py
+++ b/sleekxmpp/thirdparty/socks.py
@@ -28,6 +28,9 @@ OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMANGE.
This module provides a standard socket-like interface for Python
for tunneling connections through SOCKS proxies.
+"""
+
+"""
Minor modifications made by Christopher Gilbert (http://motomastyle.com/)
for use in PyLoris (http://pyloris.sourceforge.net/)
@@ -35,10 +38,13 @@ for use in PyLoris (http://pyloris.sourceforge.net/)
Minor modifications made by Mario Vilas (http://breakingcode.wordpress.com/)
mainly to merge bug fixes found in Sourceforge
+Minor modifications made by Eugene Dementiev (http://www.dementiev.eu/)
+
"""
import socket
import struct
+import sys
PROXY_TYPE_SOCKS4 = 1
PROXY_TYPE_SOCKS5 = 2
@@ -208,7 +214,7 @@ class socksocket(socket.socket):
if self.__proxy[3]:
# Resolve remotely
ipaddr = None
- req = req + chr(0x03).encode() + chr(len(destaddr)).encode() + destaddr
+ req = req + chr(0x03).encode() + chr(len(destaddr)).encode() + destaddr.encode()
else:
# Resolve locally
ipaddr = socket.inet_aton(socket.gethostbyname(destaddr))
@@ -323,7 +329,10 @@ class socksocket(socket.socket):
# We read the response until we get the string "\r\n\r\n"
resp = self.recv(1)
while resp.find("\r\n\r\n".encode()) == -1:
- resp = resp + self.recv(1)
+ recv = self.recv(1)
+ if not recv:
+ raise GeneralProxyError((1, _generalerrors[1]))
+ resp = resp + recv
# We just need the first line to check if the connection
# was successful
statusline = resp.splitlines()[0].split(" ".encode(), 2)