sqlmapproject/sqlmap
1
1
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2025-06-22 05:53:17 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix for a socks module - proper unwrapmodule (Issue #58)

Browse Source
This commit is contained in:
Miroslav Stampar 2013-08-20 19:48:03 +02:00
parent 23f2c5f166
commit 1d4e2d151d
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
thirdparty/socks/socks.py vendored
View File

@ -53,6 +53,7 @@ PROXY_TYPE_HTTP = 3
_defaultproxy = None _defaultproxy = None
_orgsocket = socket.socket _orgsocket = socket.socket
_orgcreateconnection = socket.create_connection
class ProxyError(Exception): pass class ProxyError(Exception): pass
class GeneralProxyError(ProxyError): pass class GeneralProxyError(ProxyError): pass
@ -113,8 +114,8 @@ def wrapmodule(module):
raise GeneralProxyError((4, "no proxy specified")) raise GeneralProxyError((4, "no proxy specified"))
def unwrapmodule(module): def unwrapmodule(module):
module.socket.socket = socket.socket module.socket.socket = _orgsocket
module.socket.create_connection = socket.create_connection module.socket.create_connection = _orgcreateconnection
class socksocket(socket.socket): class socksocket(socket.socket):
"""socksocket([family[, type[, proto]]]) -> socket object """socksocket([family[, type[, proto]]]) -> socket object