3
2;Za @ s" d Z ddlZyddlZddlZW n( ek
rH ddljZddljZY nX ddl Z dd Z
G dd dej Z G dd dej
ZG d d
d
ejejZed kr ddlZyeejd ZW n eefk
r d
ZY nX ejee jdeZe dej dj j e dej dj j dS )z
SocksiPy + urllib2 handler
version: 0.3
author: e
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 | j }|j| |S )N)copyupdate)abd r "/usr/lib/python3.6/sockshandler.py
merge_dict s
r c @ s e Zd ZdddZdd ZdS )SocksiPyConnectionNTc O s* ||||||f| _ tjj| f|| d S )N) proxyargshttplibHTTPConnection__init__) self proxytype proxyaddr proxyportrdnsusernamepasswordargskwargsr r r r s zSocksiPyConnection.__init__c C sP t j | _| jj| j t| jttfkr8| jj | j | jj
| j | j f d S )N)
socks
socksocketsocksetproxyr typetimeoutintfloat
settimeoutconnecthostport)r 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 )N)r r HTTPSConnectionr ) r r r r r r r r r r r r r ' s z SocksiPyConnectionS.__init__c C s\ t j }|j| j t| jttfkr2|j| j |j | j
| j f t j
|| j| j| _d S )N)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)NTNN)r$ r% r& r r! r r r r r' & s
r' c @ s$ e Zd Zdd Zdd Zdd ZdS )SocksiPyHandlerc O s || _ || _tjj| d S )N)r kwurllib2 HTTPHandlerr )r r r r r r r 4 s zSocksiPyHandler.__init__c s d fdd } j ||S )Nr c s( t j|}t j| ||d|}|S )N)r" r# r )r r+ r
r )r" r# r r r+ conn)r r r build: s z(SocksiPyHandler.http_open..build)Nr )do_open)r reqr/ r )r r http_open9 s zSocksiPyHandler.http_openc s d fdd } j ||S )Nr c s( t j|}t j| ||d|}|S )N)r" r# r )r r+ r' r )r" r# r r r+ r. )r r r r/ A s z)SocksiPyHandler.https_open..build)Nr )r0 )r r1 r/ r )r r
https_open@ s zSocksiPyHandler.https_openN)r$ r% r& r r2 r3 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_SOCKS5openerprintopenreaddecoder r r r s*