U
|
Z ±d~ ã @ s< d Z ddlZddlZzddlZddlZW n, ek
|
rT ddlmZ ddlm Z Y nX ddl
|
Z
|
dd Z dd Z e ZG dd dejZG d d
|
d
|
ejZG d d d ejejZed kr8ddlZzeejd ZW n ee fk
|
rî dZY nX e ee
|
j de¡Z e 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)ÚcopyÚupdate)ÚaÚbÚd© r úCd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\sockshandler.pyÚ
|
merge_dict s
|
r
|
c C sN z6d| krt t j| ¡ nd| kr.t | ¡ nW dS W n Y dS X dS d S )Nú:Ú.FT)ÚsocketÚ inet_ptonÚAF_INET6Ú inet_aton)Úsr r r Úis_ip s
|
r c @ s e Zd ZdddZdd ZdS )ÚSocksiPyConnectionNTc O s* ||||||f| _ tjj| f|| d S r )Ú proxyargsÚhttplibÚHTTPConnectionÚ__init__© ÚselfÚ proxytypeÚ proxyaddrÚ proxyportÚrdnsÚusernameÚpasswordÚargsÚkwargsr r r r + s zSocksiPyConnection.__init__c C s¶ | j \}}}}}}|o |tk}z:t | j| jf| jd ||||||tjtj dff¡
|
}W q¬W q tj
|
k
|
r¨ } z0|rdt |krt | jsd}t |¡ n W 5 d }~X Y q X q || _d S )Né Z0x5bF)r Úsocks4_no_rdnsÚsocksÚcreate_connectionÚhostÚportÚtimeoutr Ú IPPROTO_TCPÚ TCP_NODELAYZ SOCKS4ErrorÚstrr ÚaddÚsock) r r r r r r r r- Úer r r Úconnect/ 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 D s z SocksiPyConnectionS.__init__c C sz t | ¡ | jj| j| jd| _| jjsv| jrvzt | j
|
¡ | j¡ W n. t k
|
rt | j t j¡ | j ¡ Y nX d S )N)Úserver_hostname)r r/ Ú_contextÚ wrap_socketr- r& Úcheck_hostnameÚ_check_hostnameÚsslÚmatch_hostnameÚ getpeercertÚ ExceptionÚshutdownr Ú SHUT_RDWRÚclose©r r r r r/ H s
|
|
zSocksiPyConnectionS.connect)NTNNr0 r r r r r4 C s
|
r4 c @ s$ e Zd Zdd Zdd Zdd ZdS )ÚSocksiPyHandlerc O s || _ || _tj | ¡ d S r )r ÚkwÚurllib2Ú HTTPHandlerr )r r r! r r r r T s zSocksiPyHandler.__init__c s d fdd } ||¡S )Nr c s( t j|}t j| ||d|}|S ©N)r& r' r( )r
|
rD r r ©r& r' r( r! rD ÚconnrB r r ÚbuildZ s z(SocksiPyHandler.http_open.<locals>.build)Nr ©Údo_open©r ÚreqrJ r rB r Ú http_openY s zSocksiPyHandler.http_openc s d fdd } ||¡S )Nr c s( t j|}t j| ||d|}|S rG )r
|
rD r4 r rH rB r r rJ a s z)SocksiPyHandler.https_open.<locals>.build)Nr rK rM r rB r Ú
|
https_open` s zSocksiPyHandler.https_openN)r1 r2 r3 r rO rP r r r r rC S s rC Ú__main__r" iZ# Ú localhostzHTTP: zhttp://httpbin.org/ipzHTTPS: zhttps://httpbin.org/ip)$Ú__doc__r r; rE r Ú ImportErrorÚurllib.requestÚrequestÚ http.clientÚclientr$ r
|
r Úsetr# r r r5 r4 rF Ú HTTPSHandlerrC r1 ÚsysÚintÚargvr' Ú
|
ValueErrorÚ
|
IndexErrorÚ build_openerZPROXY_TYPE_SOCKS5ÚopenerÚprintÚopenÚreadÚdecoder r r r Ú<module> s0
|
|
|
