B ¨ô/bnyã @szddlmZyddlmZWn ek r<ddlmZYnXddlmZmZm Z ddl Z ddl m Z ddl Z ddlZddlmZddlZddlZddlZdZejdkrÜejd krÜy ddlZWnek rÚed ƒ‚YnXe  e¡Zd ZZd ZZd ZZeeedœZ e!e"e  #¡e  $¡ƒƒZ%ejZ&Z'dd„Z(Gdd„de)ƒZ*Gdd„de*ƒZ+Gdd„de*ƒZ,Gdd„de*ƒZ-Gdd„de*ƒZ.Gdd„de*ƒZ/Gdd„de*ƒZ0dd d!d"œZ1d#d$d%d&d'd(d)d*d+œZ2ed,ed,ed-iZ3d@d/d0„Z4d1d2„Z5d3d4„Z6e6Z7d5d6„Z8e8Z9dAd7d8„Z:Gd9d:„d:ejƒZ;d;d<„Ze?e>eƒs(e;j@ Ae¡eBe;eed?„d?e;ƒZCdS)Bé)Ú b64encode)ÚCallable)Ú EOPNOTSUPPÚEINVALÚEAGAINN)ÚBytesIO)ÚSEEK_CURz1.7.1Únt)érz8To run PySocks on Windows you must install win_inet_ptonéér )ÚSOCKS4ÚSOCKS5ÚHTTPcst ˆ¡‡fdd„ƒ}|S)Nc sp|d}zNy$| ¡}|dkr&| d¡ˆ||ŽStk rR}z‚Wdd}~XYnXWd|dkrj| d¡XdS)NrTF)Ú gettimeoutÚ setblockingÚ Exception)ÚargsÚkwargsÚselfZ _is_blockingÚe)Úfunction©ú)/usr/lib/python3.7/site-packages/socks.pyÚwrapper(s  z"set_self_blocking..wrapper)Ú functoolsÚwraps)rrr)rrÚset_self_blocking&src@s"eZdZdZddd„Zdd„ZdS)Ú ProxyErrorz4Socket_err contains original socket.error exception.NcCs(||_||_|r$|jd |¡7_dS)Nz: {})ÚmsgÚ socket_errÚformat)rrr rrrÚ__init__;szProxyError.__init__cCs|jS)N)r)rrrrÚ__str__BszProxyError.__str__)N)Ú__name__Ú __module__Ú __qualname__Ú__doc__r"r#rrrrr9s rc@s eZdZdS)ÚGeneralProxyErrorN)r$r%r&rrrrr(Fsr(c@s eZdZdS)ÚProxyConnectionErrorN)r$r%r&rrrrr)Jsr)c@s eZdZdS)ÚSOCKS5AuthErrorN)r$r%r&rrrrr*Nsr*c@s eZdZdS)Ú SOCKS5ErrorN)r$r%r&rrrrr+Rsr+c@s eZdZdS)Ú SOCKS4ErrorN)r$r%r&rrrrr,Vsr,c@s eZdZdS)Ú HTTPErrorN)r$r%r&rrrrr-Zsr-zRequest rejected or failedzLRequest rejected because SOCKS server cannot connect to identd on the clientzPRequest rejected because the client program and identd report different user-ids)é[é\é]zGeneral SOCKS server failurez!Connection not allowed by rulesetzNetwork unreachablezHost unreachablezConnection refusedz TTL expiredz(Command not supported, or protocol errorzAddress type not supported)r r r éééééi8iTcCs.|||||r| ¡nd|r"| ¡ndft_dS)z›Sets a default proxy. All further socksocket objects will use the default unless explicitly changed. All parameters are as for socket.set_proxy().N)ÚencodeÚ socksocketÚ default_proxy)Ú proxy_typeÚaddrÚportÚrdnsÚusernameÚpasswordrrrÚset_default_proxyssr?cOs d|kr| d¡|d<t||ŽS)NÚ proxytyper9)Úpopr?)rrrrrÚsetdefaultproxy~srBcCstjS)z4Returns the default proxy, set by set_default_proxy.)r7r8rrrrÚget_default_proxy„srCcCstjrt|j_ntdƒ‚dS)aAttempts to replace a module's socket library with a SOCKS socket. Must set a default proxy using set_default_proxy(...) first. This will only work on modules that import socket directly into the namespace; most of the Python Standard Library falls into this category.zNo default proxy specifiedN)r7r8Úsocketr()ÚmodulerrrÚ wrap_module‹s rFc  Cs8|\} } |  d¡r|  d¡} |r4| d¡r4| d¡}d} xæt ||dtj¡D]Ð} | \}}}}}d}yxt|||ƒ}| rŒx| D]}|j|ŽqzWt|tt fƒr¤|  |¡|r¼|  ||||||¡|rÊ|  |¡|  | | f¡|Stjtfk r}z|} |r | ¡d}Wdd}~XYqLXqLW| r*| ‚t d¡‚dS)aícreate_connection(dest_pair, *[, timeout], **proxy_args) -> socket object Like socket.create_connection(), but connects to proxy before returning the socket object. dest_pair - 2-tuple of (IP/hostname, port). **proxy_args - Same args passed to socksocket.set_proxy() if present. timeout - Optional socket timeout value, in seconds. source_address - tuple (host, port) for the socket to bind to as its source address before connecting (only for compatibility) ú[z[]Nrzgai returned empty list.)Ú startswithÚstriprDÚ getaddrinfoÚ SOCK_STREAMr7Ú setsockoptÚ isinstanceÚintÚfloatÚ settimeoutÚ set_proxyÚbindÚconnectÚerrorrÚclose)Ú dest_pairÚtimeoutÚsource_addressr9Ú proxy_addrÚ proxy_portZ proxy_rdnsZproxy_usernameZproxy_passwordÚsocket_optionsZ remote_hostZ remote_portÚerrÚrÚfamilyÚ socket_typeÚprotoÚ canonnameÚsaÚsockÚoptrrrrÚcreate_connection™s>        rec@seZdZdZdd„ZeƒZdS)Ú _BaseSocketzBAllows Python 2 delegated methods such as send() to be overridden.cOsHtj|f|ž|Žtƒ|_x(|jD]}t||ƒ|j|<t||ƒq"WdS)N)Ú _orig_socketr"ÚdictÚ _savedmethodsÚ _savenamesÚgetattrÚdelattr)rÚposÚkwÚnamerrrr"Øs  z_BaseSocket.__init__N)r$r%r&r'r"ÚlistrjrrrrrfÖsrfcs ‡fdd„S)Ncs|jˆ||ŽS)N)ri)rrmrn)rorrÚäóz_makemethod..r)ror)rorÚ _makemethodãsrs)ÚsendtoÚsendÚrecvfromÚrecvcs<eZdZdZdZejejdf‡fdd„ Zdd„Z ‡fdd „Z d d „Z d d „Z d7dd„Z dd„Z‡fdd„Z‡fdd„Zd8‡fdd„ Zd9‡fdd„ Zdd„Z‡fdd„Zdd „ZeZd!d"„ZeZd#d$„ZeZd%d&„Z‡fd'd(„Zd)d*„Zd+d,„Zd-d.„Zd/d0„Z e!ee"ee#e iZ$e%d:‡fd1d2„ ƒZ&e%d3d4„ƒZ'd5d6„Z(‡Z)S);r7a2socksocket([family[, type[, proto]]]) -> socket object Open a SOCKS enabled socket. The parameters are the same as those of the standard socket init. In order for SOCKS to work, you must specify family=AF_INET and proto=0. The "type" argument must be either SOCK_STREAM or SOCK_DGRAM. Nrcsp|tjtjfkr"d}t| |¡ƒ‚tt|ƒj|||f|ž|Žd|_|j rT|j |_ nd|_ d|_ d|_ d|_ dS)Nz0Socket type must be stream or datagram, not {!r})NNNNNN)rDrKÚ SOCK_DGRAMÚ ValueErrorr!Úsuperr7r"Ú _proxyconnr8ÚproxyÚproxy_socknameÚproxy_peernameÚ_timeout)rr^Útyper`rrr)Ú __class__rrr"üs zsocksocket.__init__cCs@d}x6t|ƒ|kr:| |t|ƒ¡}|s0tdƒ‚||7}qW|S)zŠReceive EXACTLY the number of bytes requested from the file object. Blocks until the required number of bytes have been received.rrzConnection closed unexpectedly)ÚlenÚreadr()rÚfileÚcountÚdataÚdrrrÚ_readalls zsocksocket._readallcs@||_y| ¡}tt|ƒ |j¡Wntjk r:YnXdS)N)rÚget_proxy_peernamerzr7rPrDrT)rrWZpeer)rrrrPs zsocksocket.settimeoutcCs|jS)N)r)rrrrr#szsocksocket.gettimeoutcCs|r| d¡n | d¡dS)Ng)rP)rÚvrrrr&s zsocksocket.setblockingTcCs.|||||r| ¡nd|r"| ¡ndf|_dS)a Sets the proxy to be used. proxy_type - The type of the proxy to be used. Three types are supported: PROXY_TYPE_SOCKS4 (including socks4a), PROXY_TYPE_SOCKS5 and PROXY_TYPE_HTTP addr - The address of the server (IP or DNS). port - The port of the server. Defaults to 1080 for SOCKS servers and 8080 for HTTP proxy servers. rdns - Should DNS queries be performed on the remote side (rather than the local side). The default is True. Note: This has no effect with SOCKS4 servers. username - Username to authenticate with to the server. The default is no authentication. password - Password to authenticate with to the server. Only relevant when username is also provided.N)r6r|)rr9r:r;r<r=r>rrrrQ,szsocksocket.set_proxycOs"d|kr| d¡|d<|j||ŽS)Nr@r9)rArQ)rrrrrrÚsetproxyAszsocksocket.setproxycsø|j\}}}}}}|r"|jtjkr4tj|f|ž|ŽS|jrFt td¡‚|t kr^d} t t | ¡‚t t |ƒj||Ž|  ¡\} } d| f} tƒ|_| ¡} |j | ¡d}| |j|| ¡\} }| \}} |\} } t t |ƒ || f¡t t |ƒ |j¡d|_dS)zVImplements proxy connection for UDP sockets. Happens during the bind() phase.z"Socket already bound to an addressz'UDP only supported by SOCKS5 proxy typeÚ0ó)z0.0.0.0rN)r|r€rDrxrgrRr{rTrrrrzr7Ú getsocknameÚ _proxy_addrrSÚ_SOCKS5_requestrPrr})rrmrnr9rYrZr<r=r>rÚ_r;Údstr|Z UDP_ASSOCIATEZrelayÚhost)rrrrRFs*    zsocksocket.bindc s¢|jtjkr$tt|ƒj|f|ž|ŽS|js4| d¡|d}|dd…}tƒ}d}|  |¡d}|  |¡|  ||¡tt|ƒj |  ¡|f|ž|Ž} | |  ¡S)N)Úréÿÿÿÿsó)r€rDrxrzr7rtr{rRrÚwriteÚ_write_SOCKS5_addressruÚgetvalueÚtell) rÚbytesrrÚaddressÚflagsÚheaderZRSVZ STANDALONEÚsent)rrrrtks      zsocksocket.sendtoc s:|jtjkr |j|||jf|ŽStt|ƒj||f|ŽSdS)N)r€rDrxrtr~rzr7ru)rr›rr)rrrrus zsocksocket.sendc s¾|jtjkrtt|ƒ ||¡S|js.| d¡ttt|ƒ  |d|¡ƒ}|  dt ¡|  d¡}t |ƒrntdƒ‚| |¡\}}|jr¬|j\}}||ks |d|fkr¬t td¡‚|  |¡||ffS)N)r”rir r zReceived UDP packet fragmentrzPacket filtered)r€rDrxrzr7rvr{rRrrwÚseekrrƒÚordÚNotImplementedErrorÚ_read_SOCKS5_addressr~rTr) rÚbufsizerÚbufÚfragZfromhostZfromportZpeerhostZpeerport)rrrrv…s      zsocksocket.recvfromcOs|j||Ž\}}|S)N)rv)rrmrnr›r‘rrrrw™szsocksocket.recvcs|jr|j ¡tt|ƒ ¡S)N)r{rUrzr7)r)rrrrUs zsocksocket.closecCs|jS)z:Returns the bound IP address and port number at the proxy.)r})rrrrÚget_proxy_sockname¢szsocksocket.get_proxy_socknamecCs| ¡S)z> Returns the IP and port number of the proxy. )Ú getpeername)rrrrr‰¨szsocksocket.get_proxy_peernamecCs|jS)zwReturns the IP address and port number of the destination machine. Note: get_proxy_peername returns the proxy.)r~)rrrrÚ get_peername°szsocksocket.get_peernamecGsd}| |||¡\|_|_dS)z7Negotiates a stream connection through a SOCKS5 server.óN)rr~r})rÚ dest_addrZCONNECTrrrÚ_negotiate_SOCKS5¸szsocksocket._negotiate_SOCKS5cs|j\}}}}}} | d¡} | dd¡} zÂ|r@| r@|  d¡n |  d¡|  ¡| | d¡} | dd…dkrvtd ƒ‚| dd…d kr|r| s˜td ƒ‚|  d tt|ƒƒ  ¡|tt| ƒƒ  ¡| ¡|  ¡| | d¡} | dd…d krötd ƒ‚| dd…d krHtdƒ‚n6| dd…d krH| dd…dkr@tdƒ‚ntd ƒ‚|  d|d ¡|  || ¡}|  ¡| | d¡}|dd…dkr”td ƒ‚t |dd…ƒ}|dkrÊt   |d¡}td ||¡ƒ‚| | ¡}tt|ƒ |j¡||fS|  ¡|  ¡XdS)z‘ Send SOCKS5 request with given command (CMD field) and address (DST field). Returns resolved DST address that was used. ÚwbÚrbrssr r óz%SOCKS5 proxy server sent invalid dataózPNo username/password supplied. Server requested username/password authenticationrªr–zSOCKS5 authentication failedóÿz7All offered SOCKS5 authentication methods were rejectedr z Unknown errorz {:#04x}: {}N)r|Úmakefiler—Úflushrˆr(r*Úchrr‚r6r˜r¡Ú SOCKS5_ERRORSÚgetr+r!r£rzr7rPrrU)rÚconnÚcmdr’r9r:r;r<r=r>ÚwriterÚreaderZ chosen_authZ auth_statusÚresolvedÚrespÚstatusrTZbnd)rrrr¾s\     .       zsocksocket._SOCKS5_requestc CsH|\}}|j\}}}}}} tjdtjdi} xrtjtjfD]b} yDt | |¡} | | | | ¡t | | ¡}| t d|¡¡||fStj k r˜w8Yq8Xq8W|rÌ|  d¡} | dt t | ƒƒ  ¡| ¡nbt  ||tjtjtjtj¡}|d}|d} |dd}t | |¡} | | | | ¡t | | ¡}| t d|¡¡||fS)z~ Return the host and port packed for the SOCKS5 protocol, and the resolved address as a tuple object. rªóz>HÚidnarrr1)r|rDÚAF_INETÚAF_INET6Ú inet_ptonr—Ú inet_ntopÚstructÚpackrTr6r´r‚rJÚ AF_UNSPECrKÚ IPPROTO_TCPÚ AI_ADDRCONFIG)rr:r„r“r;r9r‘r<r=r>Zfamily_to_byter^Ú addr_bytesZ host_bytesÚ addressesZ target_addrrrrr˜ s6         z socksocket._write_SOCKS5_addresscCs–| |d¡}|dkr(t | |d¡¡}nN|dkrN| |d¡}| |t|ƒ¡}n(|dkrnt tj| |d¡¡}ntdƒ‚t d| |d ¡¡d }||fS) Nr rªr1rr¾éz%SOCKS5 proxy server sent invalid dataz>Hr r) rˆrDÚ inet_ntoar¡rÃrÁr(rÄÚunpack)rr„Zatypr:Úlengthr;rrrr£Ms  zsocksocket._read_SOCKS5_addresscCsŽ|j\}}}}}}| d¡} | dd¡} zLd} yt |¡} Wn4tjk rr|r^d} d} nt t |¡¡} YnX|  t ddd |¡¡|  | ¡|r¢|  |¡|  d ¡| rÄ|  |  d ¡d ¡|   ¡|  | d ¡} | dd …d krðt d ƒ‚t | d d…ƒ}|dkr&t |d¡}td ||¡ƒ‚t | dd…¡t d| dd…¡df|_| rjt | ¡|f|_n ||f|_Wd|  ¡|  ¡XdS)z0Negotiates a connection through a SOCKS4 server.r­r®rFsTz>BBHr1r r–r¿r5z%SOCKS4 proxy server sent invalid datar éZz Unknown errorz {:#04x}: {}Nz>H)r|r²rDÚ inet_atonrTÚ gethostbynamer—rÄrÅr6r³rˆr(r¡Ú SOCKS4_ERRORSr¶r,r!rÌrÍr}r~rU)rr«Ú dest_portr9r:r;r<r=r>r¹rºZremote_resolverÉr¼r½rTrrrÚ_negotiate_SOCKS4\sH        zsocksocket._negotiate_SOCKS4cCsj|j\}}}}}}|r|nt |¡}d| d¡dt|ƒ ¡dd| d¡g} |rv|rv|  dt|d|ƒ¡|  d¡| d | ¡¡|  ¡} |   ¡} |   ¡| s´t dƒ‚y|   d d ¡\} } }Wntk ræt d ƒ‚YnX|  d ¡sút d ƒ‚y t| ƒ} Wntk r$tdƒ‚YnX| dkrVd | |¡}| dkrN|d7}t|ƒ‚d|_||f|_dS)zwNegotiates a connection through an HTTP server. NOTE: This currently only supports HTTP CONNECT-style proxies.sCONNECT r¿ó:s HTTP/1.1sHost: sProxy-Authorization: basic s zConnection closed unexpectedlyú r z'HTTP proxy server sent invalid responsezHTTP/z0Proxy server does not appear to be an HTTP proxyz4HTTP proxy server did not return a valid HTTP statuséÈz{}: {})ii“i•za [*] Note: The HTTP proxy server may not be supported by PySocks (must be a CONNECT tunnel proxy))s0.0.0.0rN)r|rDrÑr6ÚstrÚappendrÚsendallÚjoinr²ÚreadlinerUr(ÚsplitryrHrNr-r!r}r~)rr«rÓr9r:r;r<r=r>Z http_headersZfobjZ status_liner`Ú status_codeZ status_msgrTrrrÚ_negotiate_HTTP˜sB        zsocksocket._negotiate_HTTPc s$t|ƒdks|d d¡r,t dt|ƒ¡‚|\}}|jtjkr||jsP| d¡t  |¡}|dkrn|snd|_ n ||f|_ dS|j \}}}}} } t |t tfƒr¶t|ƒdks¶|r¶t |tƒs¾tdƒ‚tt|ƒ |j¡|dkr ||_ tt|ƒ |j¡tt|ƒ ||f¡dS| ¡}ytt|ƒ |¡Wnztjk r } zX| ¡|sŒ|\}}d  ||¡} t|} d  | | ¡}t d || ¡t|| ƒ‚n| ‚Wdd} ~ XYn€Xy|j|}||||ƒWnbtjk r} z"|sì| ¡td | ƒ‚n| ‚Wdd} ~ XYn tk r| ¡‚YnXdS) zã Connects to the specified destination through a proxy. Uses the same API as socket's connect(). To select the proxy server, use set_proxy(). dest_pair - 2-tuple of (IP/hostname, port). r rrGz PySocks doesn't support IPv6: %s)r”rz0.0.0.0Nz0Invalid destination-connection (host, port) pairz{}:{}zError connecting to {} proxy {}z %s due to: %sz Socket error)r‚rHrDrTrØr€rxr{rRrÑr~r|rMrpÚtuplerNr(rzr7rPrrSrrUr!ÚPRINTABLE_PROXY_TYPESÚlogÚdebugr)Ú_proxy_negotiatorsr)rrVÚ catch_errorsr«rÓr9rYrZr<r=r>rTZ proxy_serverZprintable_typerZ negotiate)rrrrSÙsd              zsocksocket.connectc CsHy|j|dddStk rB}z|jr0|jS‚Wdd}~XYnXdS)a& https://docs.python.org/3/library/socket.html#socket.socket.connect_ex Like connect(address), but return an error indicator instead of raising an exception for errors returned by the C-level connect() call (other problems, such as "host not found" can still raise exceptions). T)rårN)rSÚOSErrorÚerrno)rrVrrrrÚ connect_ex6szsocksocket.connect_excCs4|j\}}}}}}|pt |¡}|s,tdƒ‚||fS)zD Return proxy address to connect to as tuple object zInvalid proxy type)r|Ú DEFAULT_PORTSr¶r()rr9rYrZr<r=r>rrrrFs zsocksocket._proxy_addr)NNNTNN)r)r)N)*r$r%r&r'r8rDrÀrKr"rˆrPrrrQr‹rRrtrurvrwrUr§Zgetproxysocknamer‰Zgetproxypeernamer©r¨r¬rr˜r£rÔrßr rrrärrSrèrÚ __classcell__rr)rrr7ñsH   %   b-<<\ r7)NNNTNN) NNNNNTNNN)DÚbase64rÚcollections.abcrÚ ImportErrorÚ collectionsrçrrrrÚiorÚloggingÚosrrDrÄÚsysÚ __version__roÚ version_infoZ win_inet_ptonÚ getLoggerr$râZPROXY_TYPE_SOCKS4r ZPROXY_TYPE_SOCKS5rZPROXY_TYPE_HTTPrZ PROXY_TYPESrhÚzipÚvaluesÚkeysráZ _orgsocketrgrÚIOErrorrr(r)r*r+r,r-rÒrµrér?rBrCZgetdefaultproxyrFZ wrapmodulererfrsrkÚmethodrMrjrÙÚsetattrr7rrrrÚs†            8