B ߉]nyã @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©ú$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_STREAMr7Z setsockoptÚ isinstanceÚintÚfloatÚ settimeoutÚ set_proxyÚbindÚconnectÚerrorrÚclose)Ú dest_pairÚtimeoutZsource_addressr9Ú proxy_addrÚ proxy_portZ proxy_rdnsZproxy_usernameZproxy_passwordZsocket_optionsZ remote_hostZ remote_portÚerrÚrÚfamilyZ socket_typeÚprotoZ canonnameZsaZsockZoptrrrrÚcreate_connection™s>        r]c@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"Úlistrbrrrrr^Ösr^cs ‡fdd„S)Ncs|jˆ||ŽS)N)ra)rrerf)rgrrÚäóz_makemethod..r)rgr)rgrÚ _makemethodãsrk)Ú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.rjzConnection closed unexpectedly)ÚlenÚreadr()rÚfileÚcountÚdataÚdrrrÚ_readalls zsocksocket._readallcs@||_y| ¡}tt|ƒ |j¡Wntjk r:YnXdS)N)rwÚget_proxy_peernamerrr7rOrDrS)rrVZpeer)ryrrrOs zsocksocket.settimeoutcCs|jS)N)rw)rrrrr#szsocksocket.gettimeoutcCs|r| d¡n | d¡dS)Ng)rO)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)r6rt)rr9r:r;r<r=r>rrrrP,szsocksocket.set_proxycOs"d|kr| d¡|d<|j||ŽS)Nr@r9)rArP)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)rtrxrDrpr_rQrsrSrrrrrr7Z getsocknameÚ _proxy_addrrRÚ_SOCKS5_requestrOrwru)rrerfr9rWrXr<r=r>rÚ_r;ÚdstrtZ UDP_ASSOCIATEZrelayÚhost)ryrrrQFs*    zsocksocket.bindc s¢|jtjkr$tt|ƒj|f|ž|ŽS|js4| d¡|d}|dd…}tƒ}d}|  |¡d}|  |¡|  ||¡tt|ƒj |  ¡|f|ž|Ž} | |  ¡S)N)Úréÿÿÿÿsó)rxrDrprrr7rlrsrQrÚwriteÚ_write_SOCKS5_addressrmÚgetvalueÚtell) rÚbytesrrZaddressÚflagsÚheaderZRSVZ STANDALONEZsent)ryrrrlks      zsocksocket.sendtoc s:|jtjkr |j|||jf|ŽStt|ƒj||f|ŽSdS)N)rxrDrprlrvrrr7rm)rr’r“r)ryrrrms 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)rxrDrprrr7rnrsrQrroÚseekrr{ÚordÚNotImplementedErrorÚ_read_SOCKS5_addressrvrSr) rÚbufsizer“ZbufZfragZfromhostZfromportZpeerhostZpeerport)ryrrrn…s      zsocksocket.recvfromcOs|j||Ž\}}|S)N)rn)rrerfr’rˆrrrro™szsocksocket.recvcs|jr|j ¡tt|ƒ ¡S)N)rsrTrrr7)r)ryrrrTs zsocksocket.closecCs|jS)z:Returns the bound IP address and port number at the proxy.)ru)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.)rv)rrrrÚ get_peername°szsocksocket.get_peernamecGsd}| |||¡\|_|_dS)z7Negotiates a stream connection through a SOCKS5 server.óN)r‡rvru)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 authenticationrrzSOCKS5 authentication failedóÿz7All offered SOCKS5 authentication methods were rejectedr z Unknown errorz {:#04x}: {}N)rtÚmakefilerŽÚflushr€r(r*Úchrrzr6rr–Ú SOCKS5_ERRORSÚgetr+r!r˜rrr7rOrwrT)rZconnÚcmdr‰r9r:r;r<r=r>ÚwriterÚreaderZ chosen_authZ auth_statusZresolvedÚrespÚstatusrSZbnd)ryrrr‡¾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Úidnar…rr1)rtrDÚAF_INETÚAF_INET6Z inet_ptonrŽÚ inet_ntopÚstructÚpackrSr6r§rzrJZ AF_UNSPECrKZ IPPROTO_TCPZ AI_ADDRCONFIG)rr:r|rŠr;r9rˆr<r=r>Zfamily_to_byter[Ú addr_bytesZ host_bytesZ 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 rr1r…r¯é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 rr°r5z%SOCKS4 proxy server sent invalid datar éZz Unknown errorz {:#04x}: {}Nz>H)rtr¥rDZ inet_atonrSÚ gethostbynamerŽr´rµr6r¦r€r(r–Ú SOCKS4_ERRORSr©r,r!r¸r¹rurvrT)rržÚ dest_portr9r:r;r<r=r>r«r¬Zremote_resolver¶r­r®rSrrrÚ_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)rtrDr¼r6ÚstrÚappendrZsendallÚjoinr¥ÚreadlinerTr(ÚsplitrqrHrMr-r!rurv)rržr¾r9r:r;r<r=r>Z http_headersZfobjZ status_liner\Z status_codeZ status_msgrSrrrÚ_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)rzrHrDrSrÃrxrprsrQr¼rvrtrLrhÚtuplerMr(rrr7rOrwrRr†rTr!ÚPRINTABLE_PROXY_TYPESÚlogÚdebugr)Ú_proxy_negotiatorsr)rrUÚ catch_errorsržr¾r9rWrXr<r=r>rSZ proxy_serverZprintable_typerZ negotiate)ryrrrRÙ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)rRÚOSErrorÚerrno)rrUrrrrÚ connect_ex6szsocksocket.connect_excCs4|j\}}}}}}|pt |¡}|s,tdƒ‚||fS)zD Return proxy address to connect to as tuple object zInvalid proxy type)rtÚ DEFAULT_PORTSr©r()rr9rWrXr<r=r>rrrr†Fs zsocksocket._proxy_addr)NNNTNN)r)r)N)*r$r%r&r'r8rDr±rKr"r€rOrrrPrƒrQrlrmrnrorTršZgetproxysocknamerZgetproxypeernamerœr›rŸr‡rr˜r¿rÈr rrrÍrrRrÑr†Ú __classcell__rr)ryrr7ñsH   %   b-<<\ r7)NNNTNN) NNNNNTNNN)DÚbase64rZcollections.abcrÚ ImportErrorÚ collectionsrÐrrrrÚiorZloggingÚosrrDr´ÚsysÚ __version__rgÚ version_infoZ win_inet_ptonZ getLoggerr$rËZPROXY_TYPE_SOCKS4r ZPROXY_TYPE_SOCKS5rZPROXY_TYPE_HTTPrZ PROXY_TYPESr`ÚzipÚvaluesÚkeysrÊZ _orgsocketr_rÚIOErrorrr(r)r*r+r,r-r½r¨rÒr?rBrCZgetdefaultproxyrFZ wrapmoduler]r^rkrcÚmethodrLrbrÄÚsetattrr7rrrrÚs†            8