U
"@Wa @ s& d Z ddlZzddlZddlZW n, ek
rL ddlmZ ddlmZ Y nX ddl Z dd Z
G dd dej Z G dd dej
ZG d d
d
ejejZed kr"ddlZzeejd ZW n eefk
r d
ZY nX eee jdeZe de d e de d dS )z
SocksiPy + urllib2 handler
version: 0.3
author: e<e@tr0ll.in>
This module provides a Handler which you can use with urllib2 to allow it to tunnel your connection through a socks.sockssocket socket, with out monkey patching the original socket...
Nc C s | }|| |S N)copyupdate)abd r /sockshandler.py
merge_dict s
r
c @ s e Zd ZdddZdd ZdS )SocksiPyConnectionNTc O s* ||||||f| _ tjj| f|| d S r ) proxyargshttplibHTTPConnection__init__ selfZ proxytypeZ proxyaddrZ proxyportZrdnsZusernameZpasswordargskwargsr r r r s zSocksiPyConnection.__init__c C sP t | _| jj| j t| jttfkr8| j | j | j
| j | j f d S r )
socks
socksocketsocksetproxyr typetimeoutintfloat
settimeoutconnecthostportr r r r r s
zSocksiPyConnection.connect)NTNN__name__
__module__ __qualname__r r r r r r r s
r c @ s e Zd ZdddZdd ZdS )SocksiPyConnectionSNTc O s* ||||||f| _ tjj| f|| d S r )r r
HTTPSConnectionr r r r r r ' s z SocksiPyConnectionS.__init__c C s\ t }|j| j t| jttfkr2|| j | | j
| j f t
|| j| j| _d S r )r r r r r r r r r r r r sslZ wrap_socketZkey_fileZ cert_filer )r r r r r r + s zSocksiPyConnectionS.connect)NTNNr! r r r r r% & s
r% c @ s$ e Zd Zdd Zdd Zdd ZdS )SocksiPyHandlerc O s || _ || _tj| d S r )r kwurllib2 HTTPHandlerr )r r r r r r r 4 s zSocksiPyHandler.__init__c s d fdd } ||S )Nr c s( t j|}t j| ||d|}|S N)r r r )r
r) r r r r r r r) Zconnr r r build: s z(SocksiPyHandler.http_open.<locals>.build)Nr Zdo_openr Zreqr. r r r http_open9 s zSocksiPyHandler.http_openc s d fdd } ||S )Nr c s( t j|}t j| ||d|}|S r, )r
r) r% r r- r r r r. A s z)SocksiPyHandler.https_open.<locals>.build)Nr r/ r0 r r r
https_open@ s zSocksiPyHandler.https_openN)r" r# r$ r r1 r2 r r r r r( 3 s r( __main__ iZ# Z localhostzHTTP: zhttp://httpbin.org/ipzHTTPS: zhttps://httpbin.org/ip) __doc__r' r* r
ImportErrorZurllib.requestZrequestZ http.clientZclientr r
r r r& r% r+ Z HTTPSHandlerr( r" sysr argvr
ValueError
IndexErrorZ build_openerZPROXY_TYPE_SOCKS5Zopenerprintopenreaddecoder r r r <module> s*